perf: 优化调用异步终结点性能

Author: Cyaim

Cyaim.WebSocketServer/Cyaim.WebSocketServer/Infrastructure/Configures/WatchAssemblyContext.cs

@@ -1,7 +1,8 @@
-using System;
+using System;
 using System.Collections.Concurrent;
 using System.Collections.Generic;
 using System.Reflection;
+using System.Threading.Tasks;
 
 namespace Cyaim.WebSocketServer.Infrastructure.Configures
 {
@@ -52,5 +53,11 @@ public class WatchAssemblyContext
         /// K MethodInfo,V ParameterInfo
         /// </summary>
         public Dictionary<MethodInfo, ParameterInfo[]> MethodParameters { get; set; }
+
+        /// <summary>
+        /// Task result getter cache in endpoint method
+        /// K endpoint MethodInfo,V Task result getter
+        /// </summary>
+        public Dictionary<MethodInfo, Func<Task, object>> MethodTaskResultGetters { get; set; }
     }
 }

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

@@ -1,4 +1,4 @@
-using Cyaim.WebSocketServer.Infrastructure.AccessControl;
+using Cyaim.WebSocketServer.Infrastructure.AccessControl;
 using Cyaim.WebSocketServer.Infrastructure.Configures;
 using Cyaim.WebSocketServer.Infrastructure.Injectors;
 using Cyaim.WebSocketServer.Infrastructure.Metrics;
@@ -1076,53 +1076,33 @@ public static async Task<MvcResponseScheme> MvcDistributeAsync(WebSocketRouteOpt
                 // Async api support
                 if (invokeResult is Task task)
                 {
-                    var taskType = task.GetType();
-                    bool isGenericTask = taskType.IsGenericType && taskType.GetGenericTypeDefinition() == typeof(Task<>);
+                    await Task.WhenAny(task, Task.Delay(Timeout.Infinite, appLifetime.ApplicationStopping));
 
-                    // 预先准备好反射所需的 PropertyInfo（如果是 Task<T>）
-                    PropertyInfo resultProperty = null;
-                    if (isGenericTask) resultProperty = taskType.GetProperty("Result");
-
-                    try
-                    {
-                        // 等待任务完成（异步操作，不阻塞线程）
-                        await task.ConfigureAwait(false);
-                    }
-                    catch (Exception ex)
-                    {
-                        // await 会抛出 AggregateException，提取内部异常
-                        if (ex is AggregateException aggEx && aggEx.InnerException != null)
-                        {
-                            ex = aggEx.InnerException;
-                        }
-                        mvcResponse = await webSocketOptions.OnException(ex, request, mvcResponse, context, webSocketOptions, context.Request.Path, logger).ConfigureAwait(false);
-                    }
-
-                    // 检查任务状态（await 后任务已完成，但需要检查是否有异常或取消）
-                    if (task.IsFaulted && task.Exception != null)
+                    if (task.IsCanceled || task.IsFaulted)
                     {
-                        // 抛出内部异常（AggregateException 的第一个内部异常）
-                        var innerException = task.Exception.InnerException ?? task.Exception;
-                        mvcResponse = await webSocketOptions.OnException(innerException, request, mvcResponse, context, webSocketOptions, context.Request.Path, logger).ConfigureAwait(false);
+                        await task;
                     }
 
-                    if (task.IsCanceled)
+                    if (task.Exception != null)
                     {
-                        mvcResponse = await webSocketOptions.OnException(new TaskCanceledException(task), request, mvcResponse, context, webSocketOptions, context.Request.Path, logger).ConfigureAwait(false);
+                        throw task.Exception;
                     }
 
-                    // 检查是否是 Task<T> 类型，需要获取返回值
-                    if (isGenericTask && resultProperty != null)
+                    if (method.ReturnType == typeof(Task))
                     {
-                        // 使用预先准备好的 PropertyInfo 获取结果（此时任务已完成，不会阻塞）
-                        invokeResult = resultProperty.GetValue(task);
+                        invokeResult = null;
                     }
                     else
                     {
-                        invokeResult = null;
+                        Func<Task, object> taskResultGetter = null;
+                        webSocketOptions.WatchAssemblyContext.MethodTaskResultGetters?.TryGetValue(method, out taskResultGetter);
+                        invokeResult = taskResultGetter != null
+                            ? taskResultGetter(task)
+                            : null;
                     }
                 }
 
+
                 #endregion
 
 
@@ -1137,7 +1117,11 @@ public static async Task<MvcResponseScheme> MvcDistributeAsync(WebSocketRouteOpt
             {
                 MvcResponseScheme resp = new MvcResponseScheme() { Id = request.Id, Status = 1, Target = request.Target, RequestTime = requestTime, CompleteTime = DateTime.Now.Ticks };
 
-                ex = (ex.InnerException ?? ex);
+                if (ex is AggregateException aggEx && aggEx.InnerException != null)
+                {
+                    ex = aggEx.InnerException;
+                }
+
                 resp.Msg = string.Format(I18nText.WS_INTERACTIVE_TEXT_TEMPALTE, context.Connection.RemoteIpAddress, context.Connection.RemotePort, context.Connection.Id, I18nText.MvcDistributeAsync_Target + requestPath + Environment.NewLine + ex.Message + Environment.NewLine + ex.StackTrace);
                 logger.LogInformation(resp.Msg);
 

Cyaim.WebSocketServer/Cyaim.WebSocketServer/Infrastructure/WebSocketRouteServiceCollectionExtensions.cs

@@ -1,4 +1,4 @@
-using Cyaim.WebSocketServer.Infrastructure.Attributes;
+using Cyaim.WebSocketServer.Infrastructure.Attributes;
 using Cyaim.WebSocketServer.Infrastructure.Configures;
 using Microsoft.Extensions.Configuration;
 using Microsoft.Extensions.DependencyInjection;
@@ -10,7 +10,9 @@
 using System.Collections.Generic;
 using System.Globalization;
 using System.Linq;
+using System.Linq.Expressions;
 using System.Reflection;
+using System.Threading.Tasks;
 
 namespace Cyaim.WebSocketServer.Infrastructure
 {
@@ -182,6 +184,20 @@ public static void ConfigureWebSocketRoute(this IServiceCollection services, ICo
 
             #endregion
 
+            #region 计算终结点返回值Task<TResult>结果读取器缓存
+
+            var methodTaskResultGetters = new Dictionary<MethodInfo, Func<Task, object>>();
+            foreach (var item in points.Select(x => x.MethodInfo))
+            {
+                if (TryBuildTaskResultGetter(item.ReturnType, out var taskResultGetter))
+                {
+                    methodTaskResultGetters[item] = taskResultGetter;
+                }
+            }
+            wsrOptions.WatchAssemblyContext.MethodTaskResultGetters = methodTaskResultGetters;
+
+            #endregion
+
             services.AddSingleton(x => wsrOptions);
 
             //services.TryAddSingleton<IHttpContextAccessor, HttpContextAccessor>();
@@ -194,6 +210,41 @@ public static void ConfigureWebSocketRoute(this IServiceCollection services, ICo
             var defaultCultureInfo = CultureInfo.CurrentCulture;
         }
 
+        /// <summary>
+        /// 编译表达式树以获取 Task<TResult> 的结果，避免每次通过反射访问 Result 属性的性能开销。
+        /// </summary>
+        /// <param name="returnType"></param>
+        /// <param name="getter"></param>
+        /// <returns></returns>
+        private static bool TryBuildTaskResultGetter(Type returnType, out Func<Task, object> getter)
+        {
+            getter = null;
+            if (returnType == null || !returnType.IsGenericType)
+            {
+                return false;
+            }
+
+            var genericTypeDef = returnType.GetGenericTypeDefinition();
+            if (genericTypeDef != typeof(Task<>))
+            {
+                return false;
+            }
+
+            var resultType = returnType.GenericTypeArguments[0];
+            if (resultType.Name == "VoidTaskResult" && resultType.Namespace == "System.Threading.Tasks")
+            {
+                return false;
+            }
+
+            var taskParam = Expression.Parameter(typeof(Task), "task");
+            var castTask = Expression.Convert(taskParam, returnType);
+            var resultProperty = Expression.Property(castTask, "Result");
+            var castResult = Expression.Convert(resultProperty, typeof(object));
+
+            getter = Expression.Lambda<Func<Task, object>>(castResult, taskParam).Compile();
+            return true;
+        }
+
         /// <summary>
         /// Get assembly controller "WebSocket" EndPoint
         /// </summary>