CoreAgent/CoreAgent.WebSocketTransport/Extensions/WebSocketTransportExtensions.cs

using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using CoreAgent.WebSocketTransport.Interfaces;
using CoreAgent.WebSocketTransport.Models;
using CoreAgent.WebSocketTransport.Services;
using CoreAgent.WebSocketTransport.Middleware;
using Microsoft.Extensions.Caching.Memory;

namespace CoreAgent.WebSocketTransport.Extensions;

/// <summary>
/// WebSocket 传输服务注册扩展
/// 遵循依赖注入原则，注册所有组件
/// </summary>
public static class WebSocketTransportExtensions
{
    /// <summary>
    /// 添加 WebSocket 传输服务
    /// </summary>
    /// <param name="services">服务集合</param>
    /// <param name="configuration">配置</param>
    /// <param name="configSection">配置节名称</param>
    /// <returns>服务集合</returns>
    public static IServiceCollection AddWebSocketTransport(
        this IServiceCollection services,
        IConfiguration configuration,
        string configSection = "WebSocket")
    {
        // 注册配置
        services.Configure<WebSocketConfig>(options => 
        {
            configuration.GetSection(configSection).Bind(options);
        });
        
        // 注册核心组件
        services.AddSingleton<IWebSocketConnection, WebSocketConnection>();
        services.AddSingleton<IMessageSerializer, JsonMessageSerializer>();

        // 注册消息通道管理器
        services.AddSingleton<IMessageChannelManager>(provider =>
        {
            var logger = provider.GetRequiredService<ILogger<MessageChannelManager>>();
            var config = configuration.GetSection(configSection).Get<WebSocketConfig>() ?? new WebSocketConfig();
            return new MessageChannelManager(logger, config.QueueCapacity, config.QueueCapacity, 100);
        });

        // 注册 WebSocket 传输
        services.AddSingleton<IWebSocketTransport>(provider =>
        {
            var logger = provider.GetRequiredService<ILogger<CoreAgent.WebSocketTransport.Services.WebSocketTransport>>();
            var connection = provider.GetRequiredService<IWebSocketConnection>();
            var serializer = provider.GetRequiredService<IMessageSerializer>();
            var middlewares = provider.GetServices<IMessageMiddleware>();
            var config = configuration.GetSection(configSection).Get<WebSocketConfig>() ?? new WebSocketConfig();
            var channelManager = provider.GetRequiredService<IMessageChannelManager>();
            
            return new CoreAgent.WebSocketTransport.Services.WebSocketTransport(logger, connection, serializer, middlewares, config, channelManager);
        });
        
        // 注册默认中间件
        services.AddWebSocketMiddleware<LoggingMiddleware>();
        // services.AddWebSocketMiddleware<CacheMiddleware>();
        services.AddScoped<IMessageMiddleware>(provider =>
        {
            var cache = provider.GetRequiredService<IMemoryCache>();
            var logger = provider.GetRequiredService<ILogger<CacheMiddleware>>();
            var config = provider.GetRequiredService<IOptions<WebSocketConfig>>().Value;
            return new CacheMiddleware(cache, logger, config);
        });

        return services;
    }

    /// <summary>
    /// 添加 WebSocket 传输服务（使用委托配置）
    /// </summary>
    /// <param name="services">服务集合</param>
    /// <param name="configure">配置委托</param>
    /// <returns>服务集合</returns>
    public static IServiceCollection AddWebSocketTransport(
        this IServiceCollection services,
        Action<WebSocketConfig> configure)
    {
        // 注册配置
        services.Configure<WebSocketConfig>(options => configure(options));
        
        // 注册核心组件
        services.AddSingleton<IWebSocketConnection, WebSocketConnection>();
        services.AddSingleton<IMessageSerializer, JsonMessageSerializer>();

        // 注册消息通道管理器
        services.AddSingleton<IMessageChannelManager>(provider =>
        {
            var logger = provider.GetRequiredService<ILogger<MessageChannelManager>>();
            var config = new WebSocketConfig();
            configure(config);
            return new MessageChannelManager(logger, config.QueueCapacity, config.QueueCapacity, 100);
        });

        // 注册 WebSocket 传输
        services.AddSingleton<IWebSocketTransport>(provider =>
        {
            var logger = provider.GetRequiredService<ILogger<CoreAgent.WebSocketTransport.Services.WebSocketTransport>>();
            var connection = provider.GetRequiredService<IWebSocketConnection>();
            var serializer = provider.GetRequiredService<IMessageSerializer>();
            var middlewares = provider.GetServices<IMessageMiddleware>();
            var config = new WebSocketConfig();
            configure(config);
            var channelManager = provider.GetRequiredService<IMessageChannelManager>();
            
            return new CoreAgent.WebSocketTransport.Services.WebSocketTransport(logger, connection, serializer, middlewares, config, channelManager);
        });
        
        // 注册默认中间件
        services.AddWebSocketMiddleware<LoggingMiddleware>();
        // services.AddWebSocketMiddleware<CacheMiddleware>();
        services.AddScoped<IMessageMiddleware>(provider =>
        {
            var cache = provider.GetRequiredService<IMemoryCache>();
            var logger = provider.GetRequiredService<ILogger<CacheMiddleware>>();
            var config = provider.GetRequiredService<IOptions<WebSocketConfig>>().Value;
            return new CacheMiddleware(cache, logger, config);
        });

        return services;
    }

    /// <summary>
    /// 添加 WebSocket 中间件
    /// </summary>
    /// <typeparam name="T">中间件类型</typeparam>
    /// <param name="services">服务集合</param>
    /// <returns>服务集合</returns>
    public static IServiceCollection AddWebSocketMiddleware<T>(this IServiceCollection services)
        where T : class, IMessageMiddleware
    {
        services.AddScoped<IMessageMiddleware, T>();
        return services;
    }
}