You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.8 KiB
80 lines
2.8 KiB
7 years ago
|
using System;
|
||
7 years ago
|
using System.Linq;
|
||
4 years ago
|
using Microsoft.AspNetCore.Mvc;
|
||
7 years ago
|
using NzbDrone.Common.TPL;
|
||
7 years ago
|
using NzbDrone.Core.Datastore.Events;
|
||
|
using NzbDrone.Core.Download.Pending;
|
||
5 years ago
|
using NzbDrone.Core.Download.TrackedDownloads;
|
||
7 years ago
|
using NzbDrone.Core.Messaging.Events;
|
||
|
using NzbDrone.Core.Queue;
|
||
|
using NzbDrone.SignalR;
|
||
5 years ago
|
using Readarr.Http;
|
||
4 years ago
|
using Readarr.Http.REST;
|
||
7 years ago
|
|
||
5 years ago
|
namespace Readarr.Api.V1.Queue
|
||
7 years ago
|
{
|
||
4 years ago
|
[V1ApiController("queue/status")]
|
||
|
public class QueueStatusController : RestControllerWithSignalR<QueueStatusResource, NzbDrone.Core.Queue.Queue>,
|
||
7 years ago
|
IHandle<QueueUpdatedEvent>, IHandle<PendingReleasesUpdatedEvent>
|
||
|
{
|
||
|
private readonly IQueueService _queueService;
|
||
|
private readonly IPendingReleaseService _pendingReleaseService;
|
||
7 years ago
|
private readonly Debouncer _broadcastDebounce;
|
||
|
|
||
4 years ago
|
public QueueStatusController(IBroadcastSignalRMessage broadcastSignalRMessage, IQueueService queueService, IPendingReleaseService pendingReleaseService)
|
||
|
: base(broadcastSignalRMessage)
|
||
7 years ago
|
{
|
||
|
_queueService = queueService;
|
||
|
_pendingReleaseService = pendingReleaseService;
|
||
7 years ago
|
|
||
|
_broadcastDebounce = new Debouncer(BroadcastChange, TimeSpan.FromSeconds(5));
|
||
7 years ago
|
}
|
||
|
|
||
4 years ago
|
public override QueueStatusResource GetResourceById(int id)
|
||
7 years ago
|
{
|
||
4 years ago
|
throw new NotImplementedException();
|
||
7 years ago
|
}
|
||
|
|
||
4 years ago
|
[HttpGet]
|
||
|
public QueueStatusResource GetQueueStatus()
|
||
7 years ago
|
{
|
||
7 years ago
|
_broadcastDebounce.Pause();
|
||
|
|
||
7 years ago
|
var queue = _queueService.GetQueue();
|
||
|
var pending = _pendingReleaseService.GetPendingQueue();
|
||
|
|
||
7 years ago
|
var resource = new QueueStatusResource
|
||
7 years ago
|
{
|
||
6 years ago
|
TotalCount = queue.Count + pending.Count,
|
||
4 years ago
|
Count = queue.Count(q => q.Author != null) + pending.Count,
|
||
|
UnknownCount = queue.Count(q => q.Author == null),
|
||
|
Errors = queue.Any(q => q.Author != null && q.TrackedDownloadStatus == TrackedDownloadStatus.Error),
|
||
|
Warnings = queue.Any(q => q.Author != null && q.TrackedDownloadStatus == TrackedDownloadStatus.Warning),
|
||
|
UnknownErrors = queue.Any(q => q.Author == null && q.TrackedDownloadStatus == TrackedDownloadStatus.Error),
|
||
|
UnknownWarnings = queue.Any(q => q.Author == null && q.TrackedDownloadStatus == TrackedDownloadStatus.Warning)
|
||
7 years ago
|
};
|
||
7 years ago
|
|
||
|
_broadcastDebounce.Resume();
|
||
|
|
||
|
return resource;
|
||
7 years ago
|
}
|
||
|
|
||
7 years ago
|
private void BroadcastChange()
|
||
7 years ago
|
{
|
||
|
BroadcastResourceChange(ModelAction.Updated, GetQueueStatus());
|
||
|
}
|
||
|
|
||
4 years ago
|
[NonAction]
|
||
7 years ago
|
public void Handle(QueueUpdatedEvent message)
|
||
|
{
|
||
|
_broadcastDebounce.Execute();
|
||
|
}
|
||
5 years ago
|
|
||
4 years ago
|
[NonAction]
|
||
7 years ago
|
public void Handle(PendingReleasesUpdatedEvent message)
|
||
|
{
|
||
7 years ago
|
_broadcastDebounce.Execute();
|
||
7 years ago
|
}
|
||
|
}
|
||
7 years ago
|
}
|