2019-04-19 18:29:37 +02:00
|
|
|
|
using Bit.Core.Abstractions;
|
|
|
|
|
using Bit.Core.Models.Domain;
|
|
|
|
|
using System;
|
|
|
|
|
using System.Collections.Generic;
|
2019-05-30 17:40:33 +02:00
|
|
|
|
using System.Threading.Tasks;
|
2019-04-19 18:29:37 +02:00
|
|
|
|
|
|
|
|
|
namespace Bit.App.Services
|
|
|
|
|
{
|
|
|
|
|
public class BroadcasterService : IBroadcasterService
|
|
|
|
|
{
|
|
|
|
|
private readonly Dictionary<string, Action<Message>> _subscribers = new Dictionary<string, Action<Message>>();
|
2019-06-05 03:28:50 +02:00
|
|
|
|
private object _myLock = new object();
|
2019-04-19 18:29:37 +02:00
|
|
|
|
|
|
|
|
|
public void Send(Message message, string id = null)
|
|
|
|
|
{
|
2020-03-28 14:16:28 +01:00
|
|
|
|
lock (_myLock)
|
2019-04-19 18:29:37 +02:00
|
|
|
|
{
|
2020-03-28 14:16:28 +01:00
|
|
|
|
if (!string.IsNullOrWhiteSpace(id))
|
2019-04-19 18:29:37 +02:00
|
|
|
|
{
|
2020-03-28 14:16:28 +01:00
|
|
|
|
if (_subscribers.ContainsKey(id))
|
2019-06-05 03:28:50 +02:00
|
|
|
|
{
|
|
|
|
|
Task.Run(() => _subscribers[id].Invoke(message));
|
|
|
|
|
}
|
|
|
|
|
return;
|
|
|
|
|
}
|
2020-03-28 14:16:28 +01:00
|
|
|
|
foreach (var sub in _subscribers)
|
2019-06-05 03:28:50 +02:00
|
|
|
|
{
|
|
|
|
|
Task.Run(() => sub.Value.Invoke(message));
|
2019-04-19 18:29:37 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Subscribe(string id, Action<Message> messageCallback)
|
|
|
|
|
{
|
2020-03-28 14:16:28 +01:00
|
|
|
|
lock (_myLock)
|
2019-04-19 18:29:37 +02:00
|
|
|
|
{
|
2020-03-28 14:16:28 +01:00
|
|
|
|
if (_subscribers.ContainsKey(id))
|
2019-06-05 03:28:50 +02:00
|
|
|
|
{
|
2019-07-22 15:21:00 +02:00
|
|
|
|
_subscribers[id] = messageCallback;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
_subscribers.Add(id, messageCallback);
|
2019-06-05 03:28:50 +02:00
|
|
|
|
}
|
2019-04-19 18:29:37 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public void Unsubscribe(string id)
|
|
|
|
|
{
|
2020-03-28 14:16:28 +01:00
|
|
|
|
lock (_myLock)
|
2019-04-19 18:29:37 +02:00
|
|
|
|
{
|
2020-03-28 14:16:28 +01:00
|
|
|
|
if (_subscribers.ContainsKey(id))
|
2019-06-05 03:28:50 +02:00
|
|
|
|
{
|
|
|
|
|
_subscribers.Remove(id);
|
|
|
|
|
}
|
2019-04-19 18:29:37 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|