CleanMusicArtist.cs 1.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. using System.Linq;
  2. using System.Threading;
  3. using System.Threading.Tasks;
  4. using Jellyfin.Data.Enums;
  5. using Jellyfin.Database.Implementations;
  6. using Jellyfin.Server.ServerSetupApp;
  7. using Microsoft.EntityFrameworkCore;
  8. using Microsoft.Extensions.Logging;
  9. namespace Jellyfin.Server.Migrations.Routines;
  10. /// <summary>
  11. /// Cleans up all Music artists that have been migrated in the 10.11 RC migrations.
  12. /// </summary>
  13. public class CleanMusicArtist : IAsyncMigrationRoutine
  14. {
  15. private readonly IStartupLogger<CleanMusicArtist> _startupLogger;
  16. private readonly IDbContextFactory<JellyfinDbContext> _dbContextFactory;
  17. /// <summary>
  18. /// Initializes a new instance of the <see cref="CleanMusicArtist"/> class.
  19. /// </summary>
  20. /// <param name="startupLogger">The startup logger.</param>
  21. /// <param name="dbContextFactory">The Db context factory.</param>
  22. public CleanMusicArtist(IStartupLogger<CleanMusicArtist> startupLogger, IDbContextFactory<JellyfinDbContext> dbContextFactory)
  23. {
  24. _startupLogger = startupLogger;
  25. _dbContextFactory = dbContextFactory;
  26. }
  27. /// <inheritdoc/>
  28. public async Task PerformAsync(CancellationToken cancellationToken)
  29. {
  30. var context = await _dbContextFactory.CreateDbContextAsync(cancellationToken).ConfigureAwait(false);
  31. await using (context.ConfigureAwait(false))
  32. {
  33. var peoples = context.Peoples.Where(e => e.PersonType == nameof(PersonKind.Artist) || e.PersonType == nameof(PersonKind.AlbumArtist));
  34. _startupLogger.LogInformation("Delete {Number} Artist and Album Artist person types from db", await peoples.CountAsync(cancellationToken).ConfigureAwait(false));
  35. await peoples
  36. .ExecuteDeleteAsync(cancellationToken)
  37. .ConfigureAwait(false);
  38. }
  39. }
  40. }