PeerConnectiveChat/PCC.Shared/Common/EventBus/InProcessEventBusBase.cs

using PCC.Common.AssemblyInject.Interfaces;

namespace PCC.Common.EventBus;

public abstract class InProcessEventBusBase(ILogger<InProcessEventBusBase> logger) : IEventBus, IAssemblyInjectSingleton<IEventBus>
{
    private readonly Dictionary<Type, HashSet<Delegate>> _dicTypeToHandlers = [];

    public bool Subscript<T>(Action<T> callBack)
    {
        var type = typeof(T);
        lock (_dicTypeToHandlers)
        {
            if (!_dicTypeToHandlers.TryGetValue(type, out var handlers))
            {
                handlers = _dicTypeToHandlers[type] = [];
            }

            return handlers.Add(callBack); // 忽略重复
        }
    }

    public bool UnSubscript<T>(Action<T> callBack)
    {
        lock (_dicTypeToHandlers)
        {
            if (_dicTypeToHandlers.TryGetValue(typeof(T), out var handlers))
            {
                var unSubscript = handlers.Remove(callBack);

                if (handlers.Count == 0) _dicTypeToHandlers.Remove(typeof(T));

                return unSubscript;
            }

            return false;
        }
    }

    public bool Publish<T>()
    {
        PublishInternal(new AnyPublishEvent(typeof(T), default));
        return PublishInternal<T?>(default);
    }

    public bool Publish<T>(T obj)
    {
        PublishInternal(new AnyPublishEvent(typeof(T), obj));
        return PublishInternal(obj);
    }

    private bool PublishInternal<T>(T eventValue)
    {
        var type = typeof(T);

        Delegate[] subscripts;
        lock (_dicTypeToHandlers)
        {
            if (!_dicTypeToHandlers.TryGetValue(type, out var handlers)) return false;
            subscripts = [.. handlers];
        }

        foreach (var del in subscripts)
        {
            try
            {
                ((Action<T>)del)(eventValue);
            }
            catch (Exception e)
            {
                logger.LogError(e, nameof(Publish));
            }
        }

        return true;
    }
}