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;
@@ -976,24 +976,26 @@ public static async Task<MvcResponseScheme> MvcDistributeAsync(WebSocketRouteOpt
                 invokeResult = methodInvoker.Invoke(inst, args);
 
                 // Async api support
-                if (invokeResult is Task)
+                if (invokeResult is Task task)
                 {
-                    dynamic invokeResultTask = invokeResult;
-                    //await invokeResultTask;
-                    await Task.WhenAny(invokeResultTask, Task.Delay(Timeout.Infinite, appLifetime.ApplicationStopping));
+                    await Task.WhenAny(task, Task.Delay(Timeout.Infinite, appLifetime.ApplicationStopping));
 
-                    if (invokeResultTask.Exception != null)
+                    if (task.Exception != null)
                     {
-                        throw invokeResultTask.Exception;
+                        throw task.Exception;
                     }
 
-                    try
+                    if (method.ReturnType == typeof(Task))
                     {
-                        invokeResult = invokeResultTask.Result;
+                        invokeResult = null;
                     }
-                    catch (RuntimeBinderException)
+                    else
                     {
-                        invokeResult = null;
+                        Func<Task, object> taskResultGetter = null;
+                        webSocketOptions.WatchAssemblyContext.MethodTaskResultGetters?.TryGetValue(method, out taskResultGetter);
+                        invokeResult = taskResultGetter != null
+                            ? taskResultGetter(task)
+                            : null;
                     }
                 }
 

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>