修复ValueTask处理异常

liguoliang · liguoliang · commit 79a5ededa258 · 2023-04-03T15:11:54.000+08:00
diff --git a/src/DotNetCoreRpc.Client/HttpRequestInterceptor.cs b/src/DotNetCoreRpc.Client/HttpRequestInterceptor.cs
@@ -34,12 +34,18 @@ public void Intercept(IInvocation invocation)
                 return;
             }
 
-            if (methodReturnType.IsTask() || methodReturnType.IsValueTask())
+            if (methodReturnType.IsTask())
             {
                 invocation.ReturnValue = _requestHandler.TaskValueTaskWithoutResultHandle(invocation.Method, invocation.Arguments);
                 return;
             }
 
+            if (methodReturnType.IsValueTask())
+            {
+                invocation.ReturnValue = new ValueTask(_requestHandler.TaskValueTaskWithoutResultHandle(invocation.Method, invocation.Arguments));
+                return;
+            }
+
             if (methodReturnType.IsTaskWithResult())
             {
                 invocation.ReturnValue = _requestHandler.GetTaskResultHandleFunc(methodReturnType).Invoke(invocation.Method, invocation.Arguments);

Original file line number	Diff line number	Diff line change
`@@ -34,12 +34,18 @@ public void Intercept(IInvocation invocation)`
`34`	`34`	`return;`
`35`	`35`	`}`
`36`	`36`
`37`		`- if (methodReturnType.IsTask() \|\| methodReturnType.IsValueTask())`
	`37`	`+ if (methodReturnType.IsTask())`
`38`	`38`	`{`
`39`	`39`	`invocation.ReturnValue = _requestHandler.TaskValueTaskWithoutResultHandle(invocation.Method, invocation.Arguments);`
`40`	`40`	`return;`
`41`	`41`	`}`
`42`	`42`
	`43`	`+ if (methodReturnType.IsValueTask())`
	`44`	`+ {`
	`45`	`+ invocation.ReturnValue = new ValueTask(_requestHandler.TaskValueTaskWithoutResultHandle(invocation.Method, invocation.Arguments));`
	`46`	`+ return;`
	`47`	`+ }`
	`48`	`+`
`43`	`49`	`if (methodReturnType.IsTaskWithResult())`
`44`	`50`	`{`
`45`	`51`	`invocation.ReturnValue = _requestHandler.GetTaskResultHandleFunc(methodReturnType).Invoke(invocation.Method, invocation.Arguments);`