mirror of
https://github.com/Prowlarr/Prowlarr.git
synced 2025-09-17 17:14:18 +02:00
fixed ninjet's race condition
This commit is contained in:
56
NzbDrone.Core/Providers/Fakes/FakeNotificationProvider.cs
Normal file
56
NzbDrone.Core/Providers/Fakes/FakeNotificationProvider.cs
Normal file
@@ -0,0 +1,56 @@
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using NzbDrone.Core.Entities.Notification;
|
||||
|
||||
namespace NzbDrone.Core.Providers.Fakes
|
||||
{
|
||||
class FakeNotificationProvider : INotificationProvider
|
||||
{
|
||||
private readonly Dictionary<Guid, BasicNotification> _basicNotifications = new Dictionary<Guid, BasicNotification>();
|
||||
private readonly Dictionary<Guid, ProgressNotification> _progressNotification = new Dictionary<Guid, ProgressNotification>();
|
||||
private readonly Object _lock = new object();
|
||||
|
||||
|
||||
ProgressNotification fakeNotification = new ProgressNotification("Updating Series");
|
||||
public void Register(ProgressNotification notification)
|
||||
{
|
||||
_progressNotification.Add(notification.Id, notification);
|
||||
}
|
||||
|
||||
public void Register(BasicNotification notification)
|
||||
{
|
||||
_basicNotifications.Add(notification.Id, notification);
|
||||
}
|
||||
|
||||
public List<BasicNotification> BasicNotifications
|
||||
{
|
||||
get { return new List<BasicNotification>(_basicNotifications.Values); }
|
||||
}
|
||||
|
||||
public List<ProgressNotification> ProgressNotifications
|
||||
{
|
||||
|
||||
get
|
||||
{
|
||||
fakeNotification.Status = NotificationStatus.InProgress;
|
||||
fakeNotification.CurrentStatus = DateTime.Now.ToString();
|
||||
return new List<ProgressNotification> { fakeNotification };
|
||||
}
|
||||
}
|
||||
|
||||
public void Dismiss(Guid notificationId)
|
||||
{
|
||||
lock (_lock)
|
||||
{
|
||||
if (_basicNotifications.ContainsKey(notificationId))
|
||||
{
|
||||
_basicNotifications.Remove(notificationId);
|
||||
}
|
||||
else if (_progressNotification.ContainsKey(notificationId))
|
||||
{
|
||||
_progressNotification.Remove(notificationId);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user