mirror of
https://github.com/immense/Remotely.git
synced 2025-10-26 11:27:15 +00:00
48 lines
1.3 KiB
C#
48 lines
1.3 KiB
C#
using Immense.SimpleMessenger;
|
|
using Microsoft.AspNetCore.Components;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
using System.Threading.Tasks;
|
|
|
|
namespace Remotely.Server.Components;
|
|
|
|
public class MessengerSubscriber : ComponentBase, IDisposable
|
|
{
|
|
protected readonly ConcurrentQueue<IDisposable> _registrations = new();
|
|
|
|
[Inject]
|
|
protected IMessenger Messenger { get; init; } = null!;
|
|
|
|
public void Dispose()
|
|
{
|
|
while (_registrations.TryDequeue(out var registration))
|
|
{
|
|
try
|
|
{
|
|
registration.Dispose();
|
|
}
|
|
catch { }
|
|
}
|
|
GC.SuppressFinalize(this);
|
|
}
|
|
|
|
protected Task Register<TMessage, TChannel>(TChannel channel, RegistrationCallback<TMessage> handler)
|
|
where TMessage : class
|
|
where TChannel : IEquatable<TChannel>
|
|
{
|
|
var registration = Messenger.Register(this, channel, handler);
|
|
_registrations.Enqueue(registration);
|
|
return Task.CompletedTask;
|
|
}
|
|
|
|
protected Task Register<TMessage>(RegistrationCallback<TMessage> handler)
|
|
where TMessage : class
|
|
{
|
|
var registration = Messenger.Register(this, handler);
|
|
_registrations.Enqueue(registration);
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|