"修复WebSocket传输服务依赖注入问题，添加IMemoryCache服务注册"

CoreAgent.API/Startup.cs

@@ -43,6 +43,9 @@ namespace CoreAgent.API
             // 添加网络命令配置
             services.Configure<NetworkCommandConfig>(Configuration.GetSection("networkCommand"));
 
+            // 添加内存缓存服务（WebSocket中间件需要）
+            services.AddMemoryCache();
+
             // 添加 WebSocket 传输服务
             services.AddWebSocketTransport(Configuration, "WebSocket");
         }

CoreAgent.WebSocketTransport/Extensions/WebSocketTransportExtensions.cs

@@ -2,129 +2,121 @@ using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using Microsoft.Extensions.Options;
+using Microsoft.Extensions.Caching.Memory;
 using CoreAgent.WebSocketTransport.Interfaces;
-using CoreAgent.WebSocketTransport.Models;
 using CoreAgent.WebSocketTransport.Services;
 using CoreAgent.WebSocketTransport.Middleware;
-using Microsoft.Extensions.Caching.Memory;
+using CoreAgent.WebSocketTransport.Models;
-
-namespace CoreAgent.WebSocketTransport.Extensions;
 
-/// <summary>
+namespace CoreAgent.WebSocketTransport.Extensions
-/// WebSocket 传输服务注册扩展
-/// 遵循依赖注入原则，注册所有组件
-/// </summary>
-public static class WebSocketTransportExtensions
 {
     /// <summary>
-    /// 添加 WebSocket 传输服务
+    /// WebSocket 传输服务扩展
     /// </summary>
-    /// <param name="services">服务集合</param>
+    public static class WebSocketTransportExtensions
-    /// <param name="configuration">配置</param>
-    /// <param name="configSection">配置节名称</param>
-    /// <returns>服务集合</returns>
-    public static IServiceCollection AddWebSocketTransport(
-        this IServiceCollection services,
-        IConfiguration configuration,
-        string configSection = "WebSocket")
     {
-        if (services == null)
+        /// <summary>
-            throw new ArgumentNullException(nameof(services));
+        /// 添加 WebSocket 传输服务
-        if (configuration == null)
+        /// </summary>
-            throw new ArgumentNullException(nameof(configuration));
+        /// <param name="services">服务集合</param>
-
+        /// <param name="configuration">配置</param>
-        // 注册配置
+        /// <param name="configSection">配置节名称</param>
-        services.Configure<WebSocketConfig>(options => 
+        /// <returns>服务集合</returns>
+        public static IServiceCollection AddWebSocketTransport(
+            this IServiceCollection services,
+            IConfiguration configuration,
+            string configSection = "WebSocket")
         {
-            configuration.GetSection(configSection).Bind(options);
+            if (services == null)
-        });
+                throw new ArgumentNullException(nameof(services));
+            if (configuration == null)
+                throw new ArgumentNullException(nameof(configuration));
 
-        // 注册核心服务
+            // 注册配置
-        RegisterCoreServices(services);
+            services.Configure<WebSocketConfig>(options => 
-
+            {
-        // 注册默认中间件
+                configuration.GetSection(configSection).Bind(options);
-        RegisterDefaultMiddleware(services);
+            });
-
-        return services;
-    }
 
+            // 注册默认中间件（在核心服务之前）
+            RegisterDefaultMiddleware(services);
 
+            // 注册核心服务
+            RegisterCoreServices(services);
 
-    /// <summary>
+            return services;
-    /// 添加 WebSocket 中间件
+        }
-    /// </summary>
-    /// <typeparam name="T">中间件类型</typeparam>
-    /// <param name="services">服务集合</param>
-    /// <returns>服务集合</returns>
-    public static IServiceCollection AddWebSocketMiddleware<T>(this IServiceCollection services)
-        where T : class, IMessageMiddleware
-    {
-        if (services == null)
-            throw new ArgumentNullException(nameof(services));
 
-        services.AddScoped<IMessageMiddleware, T>();
+        /// <summary>
-        return services;
+        /// 添加 WebSocket 中间件
-    }
+        /// </summary>
+        /// <typeparam name="T">中间件类型</typeparam>
+        /// <param name="services">服务集合</param>
+        /// <returns>服务集合</returns>
+        public static IServiceCollection AddWebSocketMiddleware<T>(this IServiceCollection services)
+            where T : class, IMessageMiddleware
+        {
+            if (services == null)
+                throw new ArgumentNullException(nameof(services));
 
-    /// <summary>
+            services.AddScoped<IMessageMiddleware, T>();
-    /// 注册核心服务
+            return services;
-    /// </summary>
+        }
-    /// <param name="services">服务集合</param>
-    private static void RegisterCoreServices(IServiceCollection services)
-    {
-        // 注册核心组件
-        services.AddSingleton<IWebSocketConnection, WebSocketConnection>();
-        services.AddSingleton<IMessageSerializer, JsonMessageSerializer>();
 
-        // 注册消息通道管理器
+        /// <summary>
-        services.AddSingleton<IMessageChannelManager>(provider =>
+        /// 注册核心服务
+        /// </summary>
+        /// <param name="services">服务集合</param>
+        private static void RegisterCoreServices(IServiceCollection services)
         {
-            var logger = provider.GetRequiredService<ILogger<MessageChannelManager>>();
+            // 注册核心组件
-            var config = provider.GetRequiredService<IOptions<WebSocketConfig>>().Value;
+            services.AddSingleton<IWebSocketConnection, WebSocketConnection>();
-            
+            services.AddSingleton<IMessageSerializer, JsonMessageSerializer>();
-            return new MessageChannelManager(
-                logger, 
-                config.QueueCapacity, 
-                config.QueueCapacity, 
-                100);
-        });
 
-        // 注册 WebSocket 传输
+            // 注册消息通道管理器
-        services.AddSingleton<IWebSocketTransport>(provider =>
+            services.AddSingleton<IMessageChannelManager>(provider =>
-        {
+            {
-            var logger = provider.GetRequiredService<ILogger<CoreAgent.WebSocketTransport.Services.WebSocketTransport>>();
+                var logger = provider.GetRequiredService<ILogger<MessageChannelManager>>();
-            var connection = provider.GetRequiredService<IWebSocketConnection>();
+                var config = provider.GetRequiredService<IOptions<WebSocketConfig>>().Value;
-            var serializer = provider.GetRequiredService<IMessageSerializer>();
+                
-            var middlewares = provider.GetServices<IMessageMiddleware>();
+                return new MessageChannelManager(
-            var config = provider.GetRequiredService<IOptions<WebSocketConfig>>().Value;
+                    logger, 
-            var channelManager = provider.GetRequiredService<IMessageChannelManager>();
+                    config.QueueCapacity, 
-            
+                    config.QueueCapacity, 
-            return new CoreAgent.WebSocketTransport.Services.WebSocketTransport(
+                    100);
-                logger, 
+            });
-                connection, 
-                serializer, 
-                middlewares, 
-                config, 
-                channelManager);
-        });
-    }
 
-    /// <summary>
+            // 注册 WebSocket 传输
-    /// 注册默认中间件
+            services.AddSingleton<IWebSocketTransport>(provider =>
-    /// </summary>
+            {
-    /// <param name="services">服务集合</param>
+                var logger = provider.GetRequiredService<ILogger<CoreAgent.WebSocketTransport.Services.WebSocketTransport>>();
-    private static void RegisterDefaultMiddleware(IServiceCollection services)
+                var connection = provider.GetRequiredService<IWebSocketConnection>();
-    {
+                var serializer = provider.GetRequiredService<IMessageSerializer>();
-        // 注册日志中间件
+                var middlewares = provider.GetServices<IMessageMiddleware>();
-        services.AddWebSocketMiddleware<LoggingMiddleware>();
+                var config = provider.GetRequiredService<IOptions<WebSocketConfig>>().Value;
+                var channelManager = provider.GetRequiredService<IMessageChannelManager>();
+                
+                return new CoreAgent.WebSocketTransport.Services.WebSocketTransport(
+                    logger, 
+                    connection, 
+                    serializer, 
+                    middlewares, 
+                    config, 
+                    channelManager);
+            });
+        }
 
-        // 注册缓存中间件（可选）
+        /// <summary>
-        services.AddScoped<IMessageMiddleware>(provider =>
+        /// 注册默认中间件
+        /// </summary>
+        /// <param name="services">服务集合</param>
+        private static void RegisterDefaultMiddleware(IServiceCollection services)
         {
-            var cache = provider.GetRequiredService<IMemoryCache>();
+            // 注册日志中间件
-            var logger = provider.GetRequiredService<ILogger<CacheMiddleware>>();
+            services.AddWebSocketMiddleware<LoggingMiddleware>();
-            var config = provider.GetRequiredService<IOptions<WebSocketConfig>>().Value;
+
-            return new CacheMiddleware(cache, logger, config);
+            // 注册缓存中间件（使用正确的注册方式）
-        });
+            services.AddWebSocketMiddleware<CacheMiddleware>();
+        }
     }
 } 

modify.md

@@ -472,4 +472,77 @@
 - 调用方代码需要处理返回值
 - 网络启动和停止流程的状态判断
 - 日志记录详细程度提升
-- 错误处理和状态监控能力增强
+- 错误处理和状态监控能力增强
+
+### WebSocket传输服务依赖注入修复
+
+**修改时间**: 2024年
+**修改文件**: 
+- `CoreAgent.WebSocketTransport/Extensions/WebSocketTransportExtensions.cs`
+- `CoreAgent.API/Startup.cs`
+
+**修改内容**:
+
+1. **修复依赖注入顺序问题**
+   - 将 `RegisterDefaultMiddleware` 移到 `RegisterCoreServices` 之前调用
+   - 确保中间件在核心服务注册之前就已经注册到容器中
+   - 解决 `provider.GetServices<IMessageMiddleware>()` 无法找到服务的问题
+
+2. **修复CacheMiddleware注册方式**
+   - 将 `CacheMiddleware` 的注册方式从手动工厂方法改为使用 `AddWebSocketMiddleware<T>`
+   - 简化注册逻辑，确保依赖注入容器能正确处理构造函数参数
+   - 移除复杂的工厂方法注册，使用标准的依赖注入模式
+
+3. **添加IMemoryCache服务注册**
+   - 在 `Startup.cs` 的 `ConfigureServices` 方法中添加 `services.AddMemoryCache()`
+   - 确保 `CacheMiddleware` 能够正确获取 `IMemoryCache` 依赖
+   - 在 WebSocket 传输服务注册之前添加内存缓存服务
+
+4. **具体实现**:
+   ```csharp
+   // WebSocketTransportExtensions.cs - 调整注册顺序
+   public static IServiceCollection AddWebSocketTransport(...)
+   {
+       // 注册配置
+       services.Configure<WebSocketConfig>(...);
+
+       // 注册默认中间件（在核心服务之前）
+       RegisterDefaultMiddleware(services);
+
+       // 注册核心服务
+       RegisterCoreServices(services);
+
+       return services;
+   }
+
+   // Startup.cs - 添加内存缓存服务
+   public void ConfigureServices(IServiceCollection services)
+   {
+       // ... 其他服务注册 ...
+
+       // 添加内存缓存服务（WebSocket中间件需要）
+       services.AddMemoryCache();
+
+       // 添加 WebSocket 传输服务
+       services.AddWebSocketTransport(Configuration, "WebSocket");
+   }
+   ```
+
+5. **设计优势**:
+   - **依赖顺序正确**：确保中间件在核心服务之前注册
+   - **简化注册逻辑**：使用标准的依赖注入模式
+   - **完整的服务注册**：包含所有必要的依赖服务
+   - **错误预防**：避免运行时依赖注入异常
+   - **代码清晰**：注册逻辑更加直观和易于理解
+
+6. **修复的问题**:
+   - `System.InvalidOperationException: Cannot resolve scoped service 'System.Collections.Generic.IEnumerable`1[CoreAgent.WebSocketTransport.Middleware.IMessageMiddleware]' from root provider`
+   - 依赖注入容器无法找到 `IMessageMiddleware` 服务
+   - `CacheMiddleware` 无法获取 `IMemoryCache` 依赖
+
+**影响范围**: 
+- WebSocket传输服务的依赖注入配置
+- 中间件注册和初始化顺序
+- 应用程序启动时的服务注册
+- 内存缓存服务的可用性
+- 错误处理和异常预防