mirror of https://github.com/ErsatzTV/ErsatzTV.git
Jason Dove
2 years ago
committed by
GitHub
28 changed files with 323 additions and 259 deletions
@ -0,0 +1,5 @@ |
|||||||
|
namespace ErsatzTV.Application; |
||||||
|
|
||||||
|
public interface IScannerBackgroundServiceRequest |
||||||
|
{ |
||||||
|
} |
||||||
@ -1,3 +1,3 @@ |
|||||||
namespace ErsatzTV.Application.Search; |
namespace ErsatzTV.Application.Search; |
||||||
|
|
||||||
public record RebuildSearchIndex : IRequest, IBackgroundServiceRequest; |
public record RebuildSearchIndex : IRequest, IScannerBackgroundServiceRequest; |
||||||
|
|||||||
@ -0,0 +1,259 @@ |
|||||||
|
using System.Threading.Channels; |
||||||
|
using Bugsnag; |
||||||
|
using ErsatzTV.Application; |
||||||
|
using ErsatzTV.Application.Emby; |
||||||
|
using ErsatzTV.Application.Jellyfin; |
||||||
|
using ErsatzTV.Application.MediaSources; |
||||||
|
using ErsatzTV.Application.Plex; |
||||||
|
using ErsatzTV.Core; |
||||||
|
using ErsatzTV.Core.Errors; |
||||||
|
using ErsatzTV.Core.Interfaces.Locking; |
||||||
|
using MediatR; |
||||||
|
|
||||||
|
namespace ErsatzTV.Services; |
||||||
|
|
||||||
|
public class ScannerService : BackgroundService |
||||||
|
{ |
||||||
|
private readonly ChannelReader<IScannerBackgroundServiceRequest> _channel; |
||||||
|
private readonly IServiceScopeFactory _serviceScopeFactory; |
||||||
|
private readonly ILogger<ScannerService> _logger; |
||||||
|
|
||||||
|
public ScannerService( |
||||||
|
ChannelReader<IScannerBackgroundServiceRequest> channel, |
||||||
|
IServiceScopeFactory serviceScopeFactory, |
||||||
|
ILogger<ScannerService> logger) |
||||||
|
{ |
||||||
|
_channel = channel; |
||||||
|
_serviceScopeFactory = serviceScopeFactory; |
||||||
|
_logger = logger; |
||||||
|
} |
||||||
|
|
||||||
|
protected override async Task ExecuteAsync(CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
try |
||||||
|
{ |
||||||
|
_logger.LogInformation("Scanner service started"); |
||||||
|
|
||||||
|
await foreach (IScannerBackgroundServiceRequest request in _channel.ReadAllAsync(cancellationToken)) |
||||||
|
{ |
||||||
|
try |
||||||
|
{ |
||||||
|
Task requestTask; |
||||||
|
switch (request) |
||||||
|
{ |
||||||
|
case ISynchronizePlexLibraryById synchronizePlexLibraryById: |
||||||
|
requestTask = SynchronizePlexLibrary(synchronizePlexLibraryById, cancellationToken); |
||||||
|
break; |
||||||
|
case SynchronizeJellyfinAdminUserId synchronizeJellyfinAdminUserId: |
||||||
|
requestTask = SynchronizeAdminUserId(synchronizeJellyfinAdminUserId, cancellationToken); |
||||||
|
break; |
||||||
|
case SynchronizeJellyfinLibraries synchronizeJellyfinLibraries: |
||||||
|
requestTask = SynchronizeLibraries(synchronizeJellyfinLibraries, cancellationToken); |
||||||
|
break; |
||||||
|
case ISynchronizeJellyfinLibraryById synchronizeJellyfinLibraryById: |
||||||
|
requestTask = SynchronizeJellyfinLibrary(synchronizeJellyfinLibraryById, cancellationToken); |
||||||
|
break; |
||||||
|
case ISynchronizeEmbyLibraryById synchronizeEmbyLibraryById: |
||||||
|
requestTask = SynchronizeEmbyLibrary(synchronizeEmbyLibraryById, cancellationToken); |
||||||
|
break; |
||||||
|
case IScanLocalLibrary scanLocalLibrary: |
||||||
|
requestTask = SynchronizeLocalLibrary(scanLocalLibrary, cancellationToken); |
||||||
|
break; |
||||||
|
default: |
||||||
|
throw new NotSupportedException($"Unsupported request type: {request.GetType().Name}"); |
||||||
|
} |
||||||
|
|
||||||
|
await requestTask; |
||||||
|
} |
||||||
|
catch (Exception ex) |
||||||
|
{ |
||||||
|
_logger.LogWarning(ex, "Failed to process scanner background service request"); |
||||||
|
|
||||||
|
try |
||||||
|
{ |
||||||
|
using IServiceScope scope = _serviceScopeFactory.CreateScope(); |
||||||
|
IClient client = scope.ServiceProvider.GetRequiredService<IClient>(); |
||||||
|
client.Notify(ex); |
||||||
|
} |
||||||
|
catch (Exception) |
||||||
|
{ |
||||||
|
// do nothing
|
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
catch (Exception ex) when (ex is TaskCanceledException or OperationCanceledException) |
||||||
|
{ |
||||||
|
_logger.LogInformation("Plex service shutting down"); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private async Task SynchronizeLocalLibrary(IScanLocalLibrary request, CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
using IServiceScope scope = _serviceScopeFactory.CreateScope(); |
||||||
|
IMediator mediator = scope.ServiceProvider.GetRequiredService<IMediator>(); |
||||||
|
IEntityLocker entityLocker = scope.ServiceProvider.GetRequiredService<IEntityLocker>(); |
||||||
|
|
||||||
|
Either<BaseError, string> scanResult = await mediator.Send(request, cancellationToken); |
||||||
|
scanResult.BiIter( |
||||||
|
name => _logger.LogDebug( |
||||||
|
"Done scanning local library {Library}", |
||||||
|
name), |
||||||
|
error => |
||||||
|
{ |
||||||
|
if (error is ScanIsNotRequired) |
||||||
|
{ |
||||||
|
_logger.LogDebug( |
||||||
|
"Scan is not required for local library {LibraryId} at this time", |
||||||
|
request.LibraryId); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
_logger.LogWarning( |
||||||
|
"Unable to scan local library {LibraryId}: {Error}", |
||||||
|
request.LibraryId, |
||||||
|
error.Value); |
||||||
|
} |
||||||
|
}); |
||||||
|
|
||||||
|
if (entityLocker.IsLibraryLocked(request.LibraryId)) |
||||||
|
{ |
||||||
|
entityLocker.UnlockLibrary(request.LibraryId); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private async Task SynchronizePlexLibrary( |
||||||
|
ISynchronizePlexLibraryById request, |
||||||
|
CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
using IServiceScope scope = _serviceScopeFactory.CreateScope(); |
||||||
|
IMediator mediator = scope.ServiceProvider.GetRequiredService<IMediator>(); |
||||||
|
IEntityLocker entityLocker = scope.ServiceProvider.GetRequiredService<IEntityLocker>(); |
||||||
|
|
||||||
|
Either<BaseError, string> result = await mediator.Send(request, cancellationToken); |
||||||
|
result.BiIter( |
||||||
|
name => _logger.LogDebug("Done synchronizing plex library {Name}", name), |
||||||
|
error => |
||||||
|
{ |
||||||
|
if (error is ScanIsNotRequired) |
||||||
|
{ |
||||||
|
_logger.LogDebug( |
||||||
|
"Scan is not required for plex library {LibraryId} at this time", |
||||||
|
request.PlexLibraryId); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
_logger.LogWarning( |
||||||
|
"Unable to synchronize plex library {LibraryId}: {Error}", |
||||||
|
request.PlexLibraryId, |
||||||
|
error.Value); |
||||||
|
} |
||||||
|
}); |
||||||
|
|
||||||
|
if (entityLocker.IsLibraryLocked(request.PlexLibraryId)) |
||||||
|
{ |
||||||
|
entityLocker.UnlockLibrary(request.PlexLibraryId); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private async Task SynchronizeAdminUserId( |
||||||
|
SynchronizeJellyfinAdminUserId request, |
||||||
|
CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
using IServiceScope scope = _serviceScopeFactory.CreateScope(); |
||||||
|
IMediator mediator = scope.ServiceProvider.GetRequiredService<IMediator>(); |
||||||
|
|
||||||
|
Either<BaseError, Unit> result = await mediator.Send(request, cancellationToken); |
||||||
|
result.BiIter( |
||||||
|
_ => _logger.LogInformation( |
||||||
|
"Successfully synchronized Jellyfin admin user id for source {MediaSourceId}", |
||||||
|
request.JellyfinMediaSourceId), |
||||||
|
error => _logger.LogWarning( |
||||||
|
"Unable to synchronize Jellyfin admin user id for source {MediaSourceId}: {Error}", |
||||||
|
request.JellyfinMediaSourceId, |
||||||
|
error.Value)); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task SynchronizeLibraries(SynchronizeJellyfinLibraries request, CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
using IServiceScope scope = _serviceScopeFactory.CreateScope(); |
||||||
|
IMediator mediator = scope.ServiceProvider.GetRequiredService<IMediator>(); |
||||||
|
|
||||||
|
Either<BaseError, Unit> result = await mediator.Send(request, cancellationToken); |
||||||
|
result.BiIter( |
||||||
|
_ => _logger.LogInformation( |
||||||
|
"Successfully synchronized Jellyfin libraries for source {MediaSourceId}", |
||||||
|
request.JellyfinMediaSourceId), |
||||||
|
error => _logger.LogWarning( |
||||||
|
"Unable to synchronize Jellyfin libraries for source {MediaSourceId}: {Error}", |
||||||
|
request.JellyfinMediaSourceId, |
||||||
|
error.Value)); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task SynchronizeJellyfinLibrary( |
||||||
|
ISynchronizeJellyfinLibraryById request, |
||||||
|
CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
using IServiceScope scope = _serviceScopeFactory.CreateScope(); |
||||||
|
IMediator mediator = scope.ServiceProvider.GetRequiredService<IMediator>(); |
||||||
|
IEntityLocker entityLocker = scope.ServiceProvider.GetRequiredService<IEntityLocker>(); |
||||||
|
|
||||||
|
Either<BaseError, string> result = await mediator.Send(request, cancellationToken); |
||||||
|
result.BiIter( |
||||||
|
name => _logger.LogDebug("Done synchronizing jellyfin library {Name}", name), |
||||||
|
error => |
||||||
|
{ |
||||||
|
if (error is ScanIsNotRequired) |
||||||
|
{ |
||||||
|
_logger.LogDebug( |
||||||
|
"Scan is not required for jellyfin library {LibraryId} at this time", |
||||||
|
request.JellyfinLibraryId); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
_logger.LogWarning( |
||||||
|
"Unable to synchronize jellyfin library {LibraryId}: {Error}", |
||||||
|
request.JellyfinLibraryId, |
||||||
|
error.Value); |
||||||
|
} |
||||||
|
}); |
||||||
|
|
||||||
|
if (entityLocker.IsLibraryLocked(request.JellyfinLibraryId)) |
||||||
|
{ |
||||||
|
entityLocker.UnlockLibrary(request.JellyfinLibraryId); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private async Task SynchronizeEmbyLibrary(ISynchronizeEmbyLibraryById request, CancellationToken cancellationToken) |
||||||
|
{ |
||||||
|
using IServiceScope scope = _serviceScopeFactory.CreateScope(); |
||||||
|
IMediator mediator = scope.ServiceProvider.GetRequiredService<IMediator>(); |
||||||
|
IEntityLocker entityLocker = scope.ServiceProvider.GetRequiredService<IEntityLocker>(); |
||||||
|
|
||||||
|
Either<BaseError, string> result = await mediator.Send(request, cancellationToken); |
||||||
|
result.BiIter( |
||||||
|
name => _logger.LogDebug("Done synchronizing emby library {Name}", name), |
||||||
|
error => |
||||||
|
{ |
||||||
|
if (error is ScanIsNotRequired) |
||||||
|
{ |
||||||
|
_logger.LogDebug( |
||||||
|
"Scan is not required for emby library {LibraryId} at this time", |
||||||
|
request.EmbyLibraryId); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
_logger.LogWarning( |
||||||
|
"Unable to synchronize emby library {LibraryId}: {Error}", |
||||||
|
request.EmbyLibraryId, |
||||||
|
error.Value); |
||||||
|
} |
||||||
|
}); |
||||||
|
|
||||||
|
if (entityLocker.IsLibraryLocked(request.EmbyLibraryId)) |
||||||
|
{ |
||||||
|
entityLocker.UnlockLibrary(request.EmbyLibraryId); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
Loading…
Reference in new issue