ActivityManager.cs 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. using System;
  2. using System.Linq;
  3. using System.Threading.Tasks;
  4. using Jellyfin.Data.Entities;
  5. using Jellyfin.Data.Events;
  6. using Jellyfin.Data.Queries;
  7. using MediaBrowser.Model.Activity;
  8. using MediaBrowser.Model.Querying;
  9. using Microsoft.EntityFrameworkCore;
  10. namespace Jellyfin.Server.Implementations.Activity
  11. {
  12. /// <summary>
  13. /// Manages the storage and retrieval of <see cref="ActivityLog"/> instances.
  14. /// </summary>
  15. public class ActivityManager : IActivityManager
  16. {
  17. private readonly JellyfinDbProvider _provider;
  18. /// <summary>
  19. /// Initializes a new instance of the <see cref="ActivityManager"/> class.
  20. /// </summary>
  21. /// <param name="provider">The Jellyfin database provider.</param>
  22. public ActivityManager(JellyfinDbProvider provider)
  23. {
  24. _provider = provider;
  25. }
  26. /// <inheritdoc/>
  27. public event EventHandler<GenericEventArgs<ActivityLogEntry>> EntryCreated;
  28. /// <inheritdoc/>
  29. public async Task CreateAsync(ActivityLog entry)
  30. {
  31. await using var dbContext = _provider.CreateContext();
  32. dbContext.ActivityLogs.Add(entry);
  33. await dbContext.SaveChangesAsync().ConfigureAwait(false);
  34. EntryCreated?.Invoke(this, new GenericEventArgs<ActivityLogEntry>(ConvertToOldModel(entry)));
  35. }
  36. /// <inheritdoc/>
  37. public async Task<QueryResult<ActivityLogEntry>> GetPagedResultAsync(ActivityLogQuery query)
  38. {
  39. await using var dbContext = _provider.CreateContext();
  40. IQueryable<ActivityLog> entries = dbContext.ActivityLogs
  41. .AsQueryable()
  42. .OrderByDescending(entry => entry.DateCreated);
  43. if (query.MinDate.HasValue)
  44. {
  45. entries = entries.Where(entry => entry.DateCreated >= query.MinDate);
  46. }
  47. if (query.HasUserId.HasValue)
  48. {
  49. entries = entries.Where(entry => entry.UserId != Guid.Empty == query.HasUserId.Value );
  50. }
  51. return new QueryResult<ActivityLogEntry>
  52. {
  53. Items = await entries
  54. .Skip(query.StartIndex ?? 0)
  55. .Take(query.Limit ?? 100)
  56. .AsAsyncEnumerable()
  57. .Select(ConvertToOldModel)
  58. .ToListAsync()
  59. .ConfigureAwait(false),
  60. TotalRecordCount = await entries.CountAsync().ConfigureAwait(false)
  61. };
  62. }
  63. /// <inheritdoc />
  64. public async Task CleanAsync(DateTime startDate)
  65. {
  66. await using var dbContext = _provider.CreateContext();
  67. var entries = dbContext.ActivityLogs
  68. .AsQueryable()
  69. .Where(entry => entry.DateCreated <= startDate);
  70. dbContext.RemoveRange(entries);
  71. await dbContext.SaveChangesAsync().ConfigureAwait(false);
  72. }
  73. private static ActivityLogEntry ConvertToOldModel(ActivityLog entry)
  74. {
  75. return new ActivityLogEntry
  76. {
  77. Id = entry.Id,
  78. Name = entry.Name,
  79. Overview = entry.Overview,
  80. ShortOverview = entry.ShortOverview,
  81. Type = entry.Type,
  82. ItemId = entry.ItemId,
  83. UserId = entry.UserId,
  84. Date = entry.DateCreated,
  85. Severity = entry.LogSeverity
  86. };
  87. }
  88. }
  89. }