MessageRouter.cs

이 파일의 문서화 페이지로 가기

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using System.Threading;
using System.Threading.Tasks;
using Dreamine.Communication.Abstractions.Interfaces;
using Dreamine.Communication.Abstractions.Models;
 
namespace Dreamine.Communication.Core.Routing;

public sealed class MessageRouter : IMessageRouter
{
    private readonly ConcurrentDictionary<string, List<Func<MessageEnvelope, CancellationToken, Task>>> _handlers = new();

    public void Register(
        string route,
        Func<MessageEnvelope, CancellationToken, Task> handler)
    {
        ArgumentException.ThrowIfNullOrWhiteSpace(route);
        ArgumentNullException.ThrowIfNull(handler);
 
        var handlers = _handlers.GetOrAdd(
            route,
            _ => new List<Func<MessageEnvelope, CancellationToken, Task>>());
 
        lock (handlers)
        {
            handlers.Add(handler);
        }
    }

    public bool Unregister(
        string route,
        Func<MessageEnvelope, CancellationToken, Task> handler)
    {
        ArgumentException.ThrowIfNullOrWhiteSpace(route);
        ArgumentNullException.ThrowIfNull(handler);
 
        if (!_handlers.TryGetValue(route, out var handlers))
        {
            return false;
        }
 
        lock (handlers)
        {
            var removed = handlers.Remove(handler);
 
            if (handlers.Count == 0)
            {
                _handlers.TryRemove(route, out _);
            }
 
            return removed;
        }
    }

    public async Task RouteAsync(
        MessageEnvelope message,
        CancellationToken cancellationToken = default)
    {
        ArgumentNullException.ThrowIfNull(message);
 
        if (!_handlers.TryGetValue(message.Route, out var handlers))
        {
            return;
        }
 
        List<Func<MessageEnvelope, CancellationToken, Task>> snapshot;
 
        lock (handlers)
        {
            snapshot = new List<Func<MessageEnvelope, CancellationToken, Task>>(handlers);
        }
 
        foreach (var handler in snapshot)
        {
            cancellationToken.ThrowIfCancellationRequested();
            await handler(message, cancellationToken).ConfigureAwait(false);
        }
    }

    public void Clear()
    {
        _handlers.Clear();
    }
}