feat: 实现 #12

Author: Cyaim

Cyaim.WebSocketServer/Cyaim.WebSocketServer/Infrastructure/Configures/WebSocketRouteOption.cs

@@ -120,6 +120,12 @@ public class WebSocketRouteOption
         /// </summary>
         public BandwidthLimitPolicy BandwidthLimitPolicy { get; set; }
 
+        /// <summary>
+        /// 是否要求请求必须包含Id属性。如果为true，未包含Id的请求将被拒绝响应。
+        /// 默认为true，因为客户端需要Id来区分响应来源。
+        /// </summary>
+        public bool RequireRequestId { get; set; } = true;
+
         #region Event
 
         /// <summary>

Cyaim.WebSocketServer/Cyaim.WebSocketServer/Infrastructure/Handlers/MvcHandler/MvcChannelHandler.cs

@@ -489,6 +489,37 @@ await bandwidthLimitManager.WaitForBandwidthAsync(
                             requestBody = body.ValueKind == JsonValueKind.Undefined ? null : (JsonNode.Parse(body.GetRawText())?.AsObject());
                         }
 
+                        // 检查请求是否包含Id属性
+                        if (webSocketOption.RequireRequestId && (requestScheme == null || string.IsNullOrWhiteSpace(requestScheme.Id)))
+                        {
+                            // 创建错误响应
+                            MvcResponseScheme errorResponse = new MvcResponseScheme()
+                            {
+                                Status = 1,
+                                RequestTime = requestTime,
+                                CompleteTime = DateTime.Now.Ticks,
+                                Target = requestScheme?.Target,
+                                Id = requestScheme?.Id,
+                                Msg = string.Format(I18nText.WS_INTERACTIVE_TEXT_TEMPALTE, context.Connection.RemoteIpAddress, context.Connection.RemotePort, context.Connection.Id, I18nText.MvcForwardSendData_RequestIdRequired)
+                            };
+
+                            // 发送错误响应
+                            string serialJson = JsonSerializer.Serialize(errorResponse, webSocketOption.DefaultResponseJsonSerializerOptions);
+                            var responseBytes = Encoding.UTF8.GetBytes(serialJson);
+                            await webSocket.SendAsync(new ArraySegment<byte>(responseBytes), result.MessageType, result.EndOfMessage, CancellationToken.None);
+
+                            logger.LogInformation(string.Format(I18nText.WS_INTERACTIVE_TEXT_TEMPALTE, context.Connection.RemoteIpAddress, context.Connection.RemotePort, context.Connection.Id, I18nText.MvcForwardSendData_RequestIdRequired));
+
+                            // 记录消息发送指标
+                            var currentNodeId = Infrastructure.Cluster.GlobalClusterCenter.ClusterContext?.NodeId;
+                            _metricsCollector?.RecordMessageSent(responseBytes.Length, currentNodeId, context.Request.Path);
+
+                            // 记录统计信息（如果统计记录器可用）
+                            Infrastructure.Cluster.GlobalClusterCenter.StatisticsRecorder?.RecordBytesSent(context.Connection.Id, responseBytes.Length);
+
+                            continue;
+                        }
+
                         // 执行管道 BeforeForwardingData
                         _ = await InvokePipeline(RequestPipelineStage.BeforeForwardingData, PipelineContext.CreateForward(context, webSocket, result, wsReceiveReader.GetBuffer(), requestScheme, requestBody, webSocketOption));
 

Cyaim.WebSocketServer/Cyaim.WebSocketServer/Infrastructure/I18nText.cs

@@ -142,6 +142,8 @@ public static void UpdateI18nText(string i18nResourcePath)
 
         public static readonly string MvcDistributeAsync_EndPointNotFound = "Endpoint not found: ";
 
+        public static readonly string MvcForwardSendData_RequestIdRequired = "Request must contain an Id property. The client cannot distinguish the response source without an Id." + Environment.NewLine;
+
 
     }
 }