2
0

ActivityManager.cs 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  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 IDbContextFactory<JellyfinDbContext> _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(IDbContextFactory<JellyfinDbContext> 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. var dbContext = await _provider.CreateDbContextAsync().ConfigureAwait(false);
  32. await using (dbContext.ConfigureAwait(false))
  33. {
  34. dbContext.ActivityLogs.Add(entry);
  35. await dbContext.SaveChangesAsync().ConfigureAwait(false);
  36. }
  37. EntryCreated?.Invoke(this, new GenericEventArgs<ActivityLogEntry>(ConvertToOldModel(entry)));
  38. }
  39. /// <inheritdoc/>
  40. public async Task<QueryResult<ActivityLogEntry>> GetPagedResultAsync(ActivityLogQuery query)
  41. {
  42. var dbContext = await _provider.CreateDbContextAsync().ConfigureAwait(false);
  43. await using (dbContext.ConfigureAwait(false))
  44. {
  45. var entries = dbContext.ActivityLogs
  46. .OrderByDescending(entry => entry.DateCreated)
  47. .Where(entry => query.MinDate == null || entry.DateCreated >= query.MinDate)
  48. .Where(entry => !query.HasUserId.HasValue || entry.UserId.Equals(default) != query.HasUserId.Value);
  49. return new QueryResult<ActivityLogEntry>(
  50. query.Skip,
  51. await entries.CountAsync().ConfigureAwait(false),
  52. await entries
  53. .Skip(query.Skip ?? 0)
  54. .Take(query.Limit ?? 100)
  55. .Select(entity => new ActivityLogEntry(entity.Name, entity.Type, entity.UserId)
  56. {
  57. Id = entity.Id,
  58. Overview = entity.Overview,
  59. ShortOverview = entity.ShortOverview,
  60. ItemId = entity.ItemId,
  61. Date = entity.DateCreated,
  62. Severity = entity.LogSeverity
  63. })
  64. .ToListAsync()
  65. .ConfigureAwait(false));
  66. }
  67. }
  68. /// <inheritdoc />
  69. public async Task CleanAsync(DateTime startDate)
  70. {
  71. var dbContext = await _provider.CreateDbContextAsync().ConfigureAwait(false);
  72. await using (dbContext.ConfigureAwait(false))
  73. {
  74. await dbContext.ActivityLogs
  75. .Where(entry => entry.DateCreated <= startDate)
  76. .ExecuteDeleteAsync()
  77. .ConfigureAwait(false);
  78. }
  79. }
  80. private static ActivityLogEntry ConvertToOldModel(ActivityLog entry)
  81. {
  82. return new ActivityLogEntry(entry.Name, entry.Type, entry.UserId)
  83. {
  84. Id = entry.Id,
  85. Overview = entry.Overview,
  86. ShortOverview = entry.ShortOverview,
  87. ItemId = entry.ItemId,
  88. Date = entry.DateCreated,
  89. Severity = entry.LogSeverity
  90. };
  91. }
  92. }
  93. }