ActivityManager.cs 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  1. using System;
  2. using System.Linq;
  3. using System.Threading.Tasks;
  4. using Jellyfin.Data.Entities;
  5. using Jellyfin.Data.Events;
  6. using MediaBrowser.Model.Activity;
  7. using MediaBrowser.Model.Querying;
  8. namespace Jellyfin.Server.Implementations.Activity
  9. {
  10. /// <summary>
  11. /// Manages the storage and retrieval of <see cref="ActivityLog"/> instances.
  12. /// </summary>
  13. public class ActivityManager : IActivityManager
  14. {
  15. private readonly JellyfinDbProvider _provider;
  16. /// <summary>
  17. /// Initializes a new instance of the <see cref="ActivityManager"/> class.
  18. /// </summary>
  19. /// <param name="provider">The Jellyfin database provider.</param>
  20. public ActivityManager(JellyfinDbProvider provider)
  21. {
  22. _provider = provider;
  23. }
  24. /// <inheritdoc/>
  25. public event EventHandler<GenericEventArgs<ActivityLogEntry>> EntryCreated;
  26. /// <inheritdoc/>
  27. public async Task CreateAsync(ActivityLog entry)
  28. {
  29. await using var dbContext = _provider.CreateContext();
  30. dbContext.ActivityLogs.Add(entry);
  31. await dbContext.SaveChangesAsync().ConfigureAwait(false);
  32. EntryCreated?.Invoke(this, new GenericEventArgs<ActivityLogEntry>(ConvertToOldModel(entry)));
  33. }
  34. /// <inheritdoc/>
  35. public QueryResult<ActivityLogEntry> GetPagedResult(
  36. Func<IQueryable<ActivityLog>, IQueryable<ActivityLog>> func,
  37. int? startIndex,
  38. int? limit)
  39. {
  40. using var dbContext = _provider.CreateContext();
  41. var query = func(dbContext.ActivityLogs.OrderByDescending(entry => entry.DateCreated));
  42. if (startIndex.HasValue)
  43. {
  44. query = query.Skip(startIndex.Value);
  45. }
  46. if (limit.HasValue)
  47. {
  48. query = query.Take(limit.Value);
  49. }
  50. // This converts the objects from the new database model to the old for compatibility with the existing API.
  51. var list = query.Select(ConvertToOldModel).ToList();
  52. return new QueryResult<ActivityLogEntry>
  53. {
  54. Items = list,
  55. TotalRecordCount = func(dbContext.ActivityLogs).Count()
  56. };
  57. }
  58. /// <inheritdoc/>
  59. public QueryResult<ActivityLogEntry> GetPagedResult(int? startIndex, int? limit)
  60. {
  61. return GetPagedResult(logs => logs, startIndex, limit);
  62. }
  63. private static ActivityLogEntry ConvertToOldModel(ActivityLog entry)
  64. {
  65. return new ActivityLogEntry
  66. {
  67. Id = entry.Id,
  68. Name = entry.Name,
  69. Overview = entry.Overview,
  70. ShortOverview = entry.ShortOverview,
  71. Type = entry.Type,
  72. ItemId = entry.ItemId,
  73. UserId = entry.UserId,
  74. Date = entry.DateCreated,
  75. Severity = entry.LogSeverity
  76. };
  77. }
  78. }
  79. }