yx lin
1 year ago
committed by
GitHub
GitHub
36 changed files with 1521 additions and 1576 deletions
@ -1,14 +1,13 @@ |
|||
namespace LINGYUN.Abp.EventBus.CAP |
|||
namespace LINGYUN.Abp.EventBus.CAP; |
|||
|
|||
/// <summary>
|
|||
/// 过期消息清理配置项
|
|||
/// </summary>
|
|||
public class AbpCAPEventBusOptions |
|||
{ |
|||
/// <summary>
|
|||
/// 过期消息清理配置项
|
|||
/// 发布消息处理失败通知
|
|||
/// default: false
|
|||
/// </summary>
|
|||
public class AbpCAPEventBusOptions |
|||
{ |
|||
/// <summary>
|
|||
/// 发布消息处理失败通知
|
|||
/// default: false
|
|||
/// </summary>
|
|||
public bool NotifyFailedCallback { get; set; } = false; |
|||
} |
|||
public bool NotifyFailedCallback { get; set; } = false; |
|||
} |
|||
|
|||
@ -1,13 +1,12 @@ |
|||
namespace LINGYUN.Abp.EventBus.CAP |
|||
namespace LINGYUN.Abp.EventBus.CAP; |
|||
|
|||
public static class AbpCAPHeaders |
|||
{ |
|||
public static class AbpCAPHeaders |
|||
{ |
|||
public static string ClientId { get; set; } = "cap-abp-client-id"; |
|||
public static string ClientId { get; set; } = "cap-abp-client-id"; |
|||
|
|||
public static string UserId { get; set; } = "cap-abp-user-id"; |
|||
public static string UserId { get; set; } = "cap-abp-user-id"; |
|||
|
|||
public static string TenantId { get; set; } = "cap-abp-tenant-id"; |
|||
public static string TenantId { get; set; } = "cap-abp-tenant-id"; |
|||
|
|||
public static string MessageId { get; set; } = "cap-abp-message-id"; |
|||
} |
|||
public static string MessageId { get; set; } = "cap-abp-message-id"; |
|||
} |
|||
|
|||
@ -1,47 +1,46 @@ |
|||
using DotNetCore.CAP.Messages; |
|||
using System; |
|||
|
|||
namespace LINGYUN.Abp.EventBus.CAP |
|||
namespace LINGYUN.Abp.EventBus.CAP; |
|||
|
|||
/// <summary>
|
|||
/// CAP消息扩展
|
|||
/// </summary>
|
|||
public static class AbpCAPMessageExtensions |
|||
{ |
|||
/// <summary>
|
|||
/// CAP消息扩展
|
|||
/// 尝试获取消息标头中的租户标识
|
|||
/// </summary>
|
|||
public static class AbpCAPMessageExtensions |
|||
/// <param name="message"></param>
|
|||
/// <param name="tenantId"></param>
|
|||
/// <returns></returns>
|
|||
public static bool TryGetTenantId( |
|||
this Message message, |
|||
out Guid? tenantId) |
|||
{ |
|||
/// <summary>
|
|||
/// 尝试获取消息标头中的租户标识
|
|||
/// </summary>
|
|||
/// <param name="message"></param>
|
|||
/// <param name="tenantId"></param>
|
|||
/// <returns></returns>
|
|||
public static bool TryGetTenantId( |
|||
this Message message, |
|||
out Guid? tenantId) |
|||
if (message.Headers.TryGetValue(AbpCAPHeaders.TenantId, out string tenantStr)) |
|||
{ |
|||
if (message.Headers.TryGetValue(AbpCAPHeaders.TenantId, out string tenantStr)) |
|||
if (Guid.TryParse(tenantStr, out Guid id)) |
|||
{ |
|||
if (Guid.TryParse(tenantStr, out Guid id)) |
|||
{ |
|||
tenantId = id; |
|||
return true; |
|||
} |
|||
tenantId = id; |
|||
return true; |
|||
} |
|||
tenantId = null; |
|||
return false; |
|||
} |
|||
/// <summary>
|
|||
/// 获取消息标头中的租户标识
|
|||
/// </summary>
|
|||
/// <param name="message"></param>
|
|||
/// <returns></returns>
|
|||
public static Guid? GetTenantIdOrNull( |
|||
this Message message) |
|||
tenantId = null; |
|||
return false; |
|||
} |
|||
/// <summary>
|
|||
/// 获取消息标头中的租户标识
|
|||
/// </summary>
|
|||
/// <param name="message"></param>
|
|||
/// <returns></returns>
|
|||
public static Guid? GetTenantIdOrNull( |
|||
this Message message) |
|||
{ |
|||
if (message.TryGetTenantId(out Guid? tenantId)) |
|||
{ |
|||
if (message.TryGetTenantId(out Guid? tenantId)) |
|||
{ |
|||
return tenantId; |
|||
} |
|||
return null; |
|||
return tenantId; |
|||
} |
|||
return null; |
|||
} |
|||
} |
|||
|
|||
@ -1,24 +1,23 @@ |
|||
using System; |
|||
using Volo.Abp.EventBus; |
|||
|
|||
namespace LINGYUN.Abp.EventBus.CAP |
|||
namespace LINGYUN.Abp.EventBus.CAP; |
|||
|
|||
/// <summary>
|
|||
/// 自定义事件订阅者
|
|||
/// </summary>
|
|||
public interface ICustomDistributedEventSubscriber |
|||
{ |
|||
/// <summary>
|
|||
/// 自定义事件订阅者
|
|||
/// 订阅事件
|
|||
/// </summary>
|
|||
/// <param name="eventType"></param>
|
|||
/// <param name="factory"></param>
|
|||
void Subscribe(Type eventType, IEventHandlerFactory factory); |
|||
/// <summary>
|
|||
/// 取消订阅
|
|||
/// </summary>
|
|||
public interface ICustomDistributedEventSubscriber |
|||
{ |
|||
/// <summary>
|
|||
/// 订阅事件
|
|||
/// </summary>
|
|||
/// <param name="eventType"></param>
|
|||
/// <param name="factory"></param>
|
|||
void Subscribe(Type eventType, IEventHandlerFactory factory); |
|||
/// <summary>
|
|||
/// 取消订阅
|
|||
/// </summary>
|
|||
/// <param name="eventType"></param>
|
|||
/// <param name="factory"></param>
|
|||
void UnSubscribe(Type eventType, IEventHandlerFactory factory); |
|||
} |
|||
/// <param name="eventType"></param>
|
|||
/// <param name="factory"></param>
|
|||
void UnSubscribe(Type eventType, IEventHandlerFactory factory); |
|||
} |
|||
|
|||
@ -1,9 +1,8 @@ |
|||
using System.Threading.Tasks; |
|||
|
|||
namespace LINGYUN.Abp.EventBus.CAP |
|||
namespace LINGYUN.Abp.EventBus.CAP; |
|||
|
|||
public interface IFailedThresholdCallbackNotifier |
|||
{ |
|||
public interface IFailedThresholdCallbackNotifier |
|||
{ |
|||
Task NotifyAsync(AbpCAPExecutionFailedException exception); |
|||
} |
|||
Task NotifyAsync(AbpCAPExecutionFailedException exception); |
|||
} |
|||
|
|||
@ -1,337 +1,329 @@ |
|||
// Copyright (c) .NET Foundation. All rights reserved.
|
|||
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|||
|
|||
#nullable enable |
|||
|
|||
using System; |
|||
using System.Collections.Generic; |
|||
using System.Diagnostics; |
|||
using System.Linq.Expressions; |
|||
using System.Reflection; |
|||
|
|||
namespace LINGYUN.Abp.EventBus.CAP.Internal |
|||
namespace LINGYUN.Abp.EventBus.CAP.Internal; |
|||
|
|||
internal class ObjectMethodExecutor |
|||
{ |
|||
internal class ObjectMethodExecutor |
|||
{ |
|||
// ReSharper disable once InconsistentNaming
|
|||
private static readonly ConstructorInfo _objectMethodExecutorAwaitableConstructor = |
|||
typeof(ObjectMethodExecutorAwaitable).GetConstructor(new[] |
|||
{ |
|||
typeof(object), // customAwaitable
|
|||
typeof(Func<object, object>), // getAwaiterMethod
|
|||
typeof(Func<object, bool>), // isCompletedMethod
|
|||
typeof(Func<object, object>), // getResultMethod
|
|||
typeof(Action<object, Action>), // onCompletedMethod
|
|||
typeof(Action<object, Action>) // unsafeOnCompletedMethod
|
|||
}); |
|||
|
|||
private readonly MethodExecutor _executor; |
|||
private readonly MethodExecutorAsync _executorAsync; |
|||
private readonly object[] _parameterDefaultValues; |
|||
|
|||
private ObjectMethodExecutor(MethodInfo methodInfo, TypeInfo targetTypeInfo, object[] parameterDefaultValues) |
|||
private static readonly ConstructorInfo _objectMethodExecutorAwaitableConstructor = |
|||
typeof(ObjectMethodExecutorAwaitable).GetConstructor(new[] |
|||
{ |
|||
if (methodInfo == null) |
|||
{ |
|||
throw new ArgumentNullException(nameof(methodInfo)); |
|||
} |
|||
typeof(object), // customAwaitable
|
|||
typeof(Func<object, object>), // getAwaiterMethod
|
|||
typeof(Func<object, bool>), // isCompletedMethod
|
|||
typeof(Func<object, object>), // getResultMethod
|
|||
typeof(Action<object, Action>), // onCompletedMethod
|
|||
typeof(Action<object, Action>) // unsafeOnCompletedMethod
|
|||
})!; |
|||
|
|||
private readonly MethodExecutor? _executor; |
|||
private readonly MethodExecutorAsync? _executorAsync; |
|||
private readonly object?[]? _parameterDefaultValues; |
|||
|
|||
private ObjectMethodExecutor(MethodInfo methodInfo, TypeInfo targetTypeInfo, object?[]? parameterDefaultValues) |
|||
{ |
|||
if (methodInfo == null) throw new ArgumentNullException(nameof(methodInfo)); |
|||
|
|||
MethodInfo = methodInfo; |
|||
MethodParameters = methodInfo.GetParameters(); |
|||
TargetTypeInfo = targetTypeInfo; |
|||
MethodReturnType = methodInfo.ReturnType; |
|||
MethodInfo = methodInfo; |
|||
MethodParameters = methodInfo.GetParameters(); |
|||
TargetTypeInfo = targetTypeInfo; |
|||
MethodReturnType = methodInfo.ReturnType; |
|||
|
|||
var isAwaitable = CoercedAwaitableInfo.IsTypeAwaitable(MethodReturnType, out var coercedAwaitableInfo); |
|||
var isAwaitable = CoercedAwaitableInfo.IsTypeAwaitable(MethodReturnType, out var coercedAwaitableInfo); |
|||
|
|||
IsMethodAsync = isAwaitable; |
|||
AsyncResultType = isAwaitable ? coercedAwaitableInfo.AwaitableInfo.ResultType : null; |
|||
IsMethodAsync = isAwaitable; |
|||
AsyncResultType = isAwaitable ? coercedAwaitableInfo.AwaitableInfo.ResultType : null; |
|||
|
|||
// Upstream code may prefer to use the sync-executor even for async methods, because if it knows
|
|||
// that the result is a specific Task<T> where T is known, then it can directly cast to that type
|
|||
// and await it without the extra heap allocations involved in the _executorAsync code path.
|
|||
_executor = GetExecutor(methodInfo, targetTypeInfo); |
|||
// Upstream code may prefer to use the sync-executor even for async methods, because if it knows
|
|||
// that the result is a specific Task<T> where T is known, then it can directly cast to that type
|
|||
// and await it without the extra heap allocations involved in the _executorAsync code path.
|
|||
_executor = GetExecutor(methodInfo, targetTypeInfo); |
|||
|
|||
if (IsMethodAsync) |
|||
{ |
|||
_executorAsync = GetExecutorAsync(methodInfo, targetTypeInfo, coercedAwaitableInfo); |
|||
} |
|||
if (IsMethodAsync) _executorAsync = GetExecutorAsync(methodInfo, targetTypeInfo, coercedAwaitableInfo); |
|||
|
|||
_parameterDefaultValues = parameterDefaultValues; |
|||
} |
|||
_parameterDefaultValues = parameterDefaultValues; |
|||
} |
|||
|
|||
public MethodInfo MethodInfo { get; } |
|||
public MethodInfo MethodInfo { get; } |
|||
|
|||
public ParameterInfo[] MethodParameters { get; } |
|||
public ParameterInfo[] MethodParameters { get; } |
|||
|
|||
public TypeInfo TargetTypeInfo { get; } |
|||
public TypeInfo TargetTypeInfo { get; } |
|||
|
|||
public Type AsyncResultType { get; } |
|||
public Type? AsyncResultType { get; } |
|||
|
|||
// This field is made internal set because it is set in unit tests.
|
|||
public Type MethodReturnType { get; internal set; } |
|||
// This field is made internal set because it is set in unit tests.
|
|||
public Type MethodReturnType { get; internal set; } |
|||
|
|||
public bool IsMethodAsync { get; } |
|||
public bool IsMethodAsync { get; } |
|||
|
|||
public static ObjectMethodExecutor Create(MethodInfo methodInfo, TypeInfo targetTypeInfo) |
|||
{ |
|||
return new ObjectMethodExecutor(methodInfo, targetTypeInfo, null); |
|||
} |
|||
public static ObjectMethodExecutor Create(MethodInfo methodInfo, TypeInfo targetTypeInfo) |
|||
{ |
|||
return new ObjectMethodExecutor(methodInfo, targetTypeInfo, null); |
|||
} |
|||
|
|||
public static ObjectMethodExecutor Create(MethodInfo methodInfo, TypeInfo targetTypeInfo, |
|||
object[] parameterDefaultValues) |
|||
{ |
|||
if (parameterDefaultValues == null) |
|||
{ |
|||
throw new ArgumentNullException(nameof(parameterDefaultValues)); |
|||
} |
|||
public static ObjectMethodExecutor Create(MethodInfo methodInfo, TypeInfo targetTypeInfo, |
|||
object?[] parameterDefaultValues) |
|||
{ |
|||
if (parameterDefaultValues == null) throw new ArgumentNullException(nameof(parameterDefaultValues)); |
|||
|
|||
return new ObjectMethodExecutor(methodInfo, targetTypeInfo, parameterDefaultValues); |
|||
} |
|||
return new ObjectMethodExecutor(methodInfo, targetTypeInfo, parameterDefaultValues); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Executes the configured method on <paramref name="target" />. This can be used whether or not
|
|||
/// the configured method is asynchronous.
|
|||
/// </summary>
|
|||
/// <remarks>
|
|||
/// Even if the target method is asynchronous, it's desirable to invoke it using Execute rather than
|
|||
/// ExecuteAsync if you know at compile time what the return type is, because then you can directly
|
|||
/// "await" that value (via a cast), and then the generated code will be able to reference the
|
|||
/// resulting awaitable as a value-typed variable. If you use ExecuteAsync instead, the generated
|
|||
/// code will have to treat the resulting awaitable as a boxed object, because it doesn't know at
|
|||
/// compile time what type it would be.
|
|||
/// </remarks>
|
|||
/// <param name="target">The object whose method is to be executed.</param>
|
|||
/// <param name="parameters">Parameters to pass to the method.</param>
|
|||
/// <returns>The method return value.</returns>
|
|||
public object? Execute(object target, object?[]? parameters) |
|||
{ |
|||
Debug.Assert(_executor != null, "Sync execution is not supported."); |
|||
return _executor(target, parameters); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Executes the configured method on <paramref name="target" />. This can only be used if the configured
|
|||
/// method is asynchronous.
|
|||
/// </summary>
|
|||
/// <remarks>
|
|||
/// If you don't know at compile time the type of the method's returned awaitable, you can use ExecuteAsync,
|
|||
/// which supplies an awaitable-of-object. This always works, but can incur several extra heap allocations
|
|||
/// as compared with using Execute and then using "await" on the result value typecasted to the known
|
|||
/// awaitable type. The possible extra heap allocations are for:
|
|||
/// 1. The custom awaitable (though usually there's a heap allocation for this anyway, since normally
|
|||
/// it's a reference type, and you normally create a new instance per call).
|
|||
/// 2. The custom awaiter (whether or not it's a value type, since if it's not, you need a new instance
|
|||
/// of it, and if it is, it will have to be boxed so the calling code can reference it as an object).
|
|||
/// 3. The async result value, if it's a value type (it has to be boxed as an object, since the calling
|
|||
/// code doesn't know what type it's going to be).
|
|||
/// </remarks>
|
|||
/// <param name="target">The object whose method is to be executed.</param>
|
|||
/// <param name="parameters">Parameters to pass to the method.</param>
|
|||
/// <returns>An object that you can "await" to get the method return value.</returns>
|
|||
public ObjectMethodExecutorAwaitable ExecuteAsync(object target, object?[]? parameters) |
|||
{ |
|||
Debug.Assert(_executorAsync != null, "Async execution is not supported."); |
|||
return _executorAsync(target, parameters); |
|||
} |
|||
|
|||
public object? GetDefaultValueForParameter(int index) |
|||
{ |
|||
if (_parameterDefaultValues == null) |
|||
throw new InvalidOperationException( |
|||
$"Cannot call {nameof(GetDefaultValueForParameter)}, because no parameter default values were supplied."); |
|||
|
|||
if (index < 0 || index > MethodParameters.Length - 1) throw new ArgumentOutOfRangeException(nameof(index)); |
|||
|
|||
/// <summary>
|
|||
/// Executes the configured method on <paramref name="target" />. This can be used whether or not
|
|||
/// the configured method is asynchronous.
|
|||
/// </summary>
|
|||
/// <remarks>
|
|||
/// Even if the target method is asynchronous, it's desirable to invoke it using Execute rather than
|
|||
/// ExecuteAsync if you know at compile time what the return type is, because then you can directly
|
|||
/// "await" that value (via a cast), and then the generated code will be able to reference the
|
|||
/// resulting awaitable as a value-typed variable. If you use ExecuteAsync instead, the generated
|
|||
/// code will have to treat the resulting awaitable as a boxed object, because it doesn't know at
|
|||
/// compile time what type it would be.
|
|||
/// </remarks>
|
|||
/// <param name="target">The object whose method is to be executed.</param>
|
|||
/// <param name="parameters">Parameters to pass to the method.</param>
|
|||
/// <returns>The method return value.</returns>
|
|||
public object Execute(object target, params object[] parameters) |
|||
return _parameterDefaultValues[index]; |
|||
} |
|||
|
|||
private static MethodExecutor GetExecutor(MethodInfo methodInfo, TypeInfo targetTypeInfo) |
|||
{ |
|||
// Parameters to executor
|
|||
var targetParameter = Expression.Parameter(typeof(object), "target"); |
|||
var parametersParameter = Expression.Parameter(typeof(object?[]), "parameters"); |
|||
|
|||
// Build parameter list
|
|||
var paramInfos = methodInfo.GetParameters(); |
|||
var parameters = new List<Expression>(paramInfos.Length); |
|||
for (var i = 0; i < paramInfos.Length; i++) |
|||
{ |
|||
return _executor(target, parameters); |
|||
var paramInfo = paramInfos[i]; |
|||
var valueObj = Expression.ArrayIndex(parametersParameter, Expression.Constant(i)); |
|||
var valueCast = Expression.Convert(valueObj, paramInfo.ParameterType); |
|||
|
|||
// valueCast is "(Ti) parameters[i]"
|
|||
parameters.Add(valueCast); |
|||
} |
|||
|
|||
/// <summary>
|
|||
/// Executes the configured method on <paramref name="target" />. This can only be used if the configured
|
|||
/// method is asynchronous.
|
|||
/// </summary>
|
|||
/// <remarks>
|
|||
/// If you don't know at compile time the type of the method's returned awaitable, you can use ExecuteAsync,
|
|||
/// which supplies an awaitable-of-object. This always works, but can incur several extra heap allocations
|
|||
/// as compared with using Execute and then using "await" on the result value typecasted to the known
|
|||
/// awaitable type. The possible extra heap allocations are for:
|
|||
/// 1. The custom awaitable (though usually there's a heap allocation for this anyway, since normally
|
|||
/// it's a reference type, and you normally create a new instance per call).
|
|||
/// 2. The custom awaiter (whether or not it's a value type, since if it's not, you need a new instance
|
|||
/// of it, and if it is, it will have to be boxed so the calling code can reference it as an object).
|
|||
/// 3. The async result value, if it's a value type (it has to be boxed as an object, since the calling
|
|||
/// code doesn't know what type it's going to be).
|
|||
/// </remarks>
|
|||
/// <param name="target">The object whose method is to be executed.</param>
|
|||
/// <param name="parameters">Parameters to pass to the method.</param>
|
|||
/// <returns>An object that you can "await" to get the method return value.</returns>
|
|||
public ObjectMethodExecutorAwaitable ExecuteAsync(object target, params object[] parameters) |
|||
// Call method
|
|||
var instanceCast = Expression.Convert(targetParameter, targetTypeInfo.AsType()); |
|||
var methodCall = Expression.Call(instanceCast, methodInfo, parameters); |
|||
|
|||
// methodCall is "((Ttarget) target) method((T0) parameters[0], (T1) parameters[1], ...)"
|
|||
// Create function
|
|||
if (methodCall.Type == typeof(void)) |
|||
{ |
|||
return _executorAsync(target, parameters); |
|||
var lambda = Expression.Lambda<VoidMethodExecutor>(methodCall, targetParameter, parametersParameter); |
|||
var voidExecutor = lambda.Compile(); |
|||
return WrapVoidMethod(voidExecutor); |
|||
} |
|||
|
|||
public object GetDefaultValueForParameter(int index) |
|||
else |
|||
{ |
|||
if (_parameterDefaultValues == null) |
|||
{ |
|||
throw new InvalidOperationException( |
|||
$"Cannot call {nameof(GetDefaultValueForParameter)}, because no parameter default values were supplied."); |
|||
} |
|||
|
|||
if (index < 0 || index > MethodParameters.Length - 1) |
|||
{ |
|||
throw new ArgumentOutOfRangeException(nameof(index)); |
|||
} |
|||
|
|||
return _parameterDefaultValues[index]; |
|||
// must coerce methodCall to match ActionExecutor signature
|
|||
var castMethodCall = Expression.Convert(methodCall, typeof(object)); |
|||
var lambda = Expression.Lambda<MethodExecutor>(castMethodCall, targetParameter, parametersParameter); |
|||
return lambda.Compile(); |
|||
} |
|||
} |
|||
|
|||
private static MethodExecutor GetExecutor(MethodInfo methodInfo, TypeInfo targetTypeInfo) |
|||
private static MethodExecutor WrapVoidMethod(VoidMethodExecutor executor) |
|||
{ |
|||
return delegate (object target, object?[]? parameters) |
|||
{ |
|||
// Parameters to executor
|
|||
var targetParameter = Expression.Parameter(typeof(object), "target"); |
|||
var parametersParameter = Expression.Parameter(typeof(object[]), "parameters"); |
|||
|
|||
// Build parameter list
|
|||
var parameters = new List<Expression>(); |
|||
var paramInfos = methodInfo.GetParameters(); |
|||
for (var i = 0; i < paramInfos.Length; i++) |
|||
{ |
|||
var paramInfo = paramInfos[i]; |
|||
var valueObj = Expression.ArrayIndex(parametersParameter, Expression.Constant(i)); |
|||
var valueCast = Expression.Convert(valueObj, paramInfo.ParameterType); |
|||
|
|||
// valueCast is "(Ti) parameters[i]"
|
|||
parameters.Add(valueCast); |
|||
} |
|||
|
|||
// Call method
|
|||
var instanceCast = Expression.Convert(targetParameter, targetTypeInfo.AsType()); |
|||
var methodCall = Expression.Call(instanceCast, methodInfo, parameters); |
|||
|
|||
// methodCall is "((Ttarget) target) method((T0) parameters[0], (T1) parameters[1], ...)"
|
|||
// Create function
|
|||
if (methodCall.Type == typeof(void)) |
|||
{ |
|||
var lambda = Expression.Lambda<VoidMethodExecutor>(methodCall, targetParameter, parametersParameter); |
|||
var voidExecutor = lambda.Compile(); |
|||
return WrapVoidMethod(voidExecutor); |
|||
} |
|||
else |
|||
{ |
|||
// must coerce methodCall to match ActionExecutor signature
|
|||
var castMethodCall = Expression.Convert(methodCall, typeof(object)); |
|||
var lambda = Expression.Lambda<MethodExecutor>(castMethodCall, targetParameter, parametersParameter); |
|||
return lambda.Compile(); |
|||
} |
|||
} |
|||
executor(target, parameters); |
|||
return null; |
|||
}; |
|||
} |
|||
|
|||
private static MethodExecutor WrapVoidMethod(VoidMethodExecutor executor) |
|||
private static MethodExecutorAsync GetExecutorAsync( |
|||
MethodInfo methodInfo, |
|||
TypeInfo targetTypeInfo, |
|||
CoercedAwaitableInfo coercedAwaitableInfo) |
|||
{ |
|||
// Parameters to executor
|
|||
var targetParameter = Expression.Parameter(typeof(object), "target"); |
|||
var parametersParameter = Expression.Parameter(typeof(object[]), "parameters"); |
|||
|
|||
// Build parameter list
|
|||
var paramInfos = methodInfo.GetParameters(); |
|||
var parameters = new List<Expression>(paramInfos.Length); |
|||
for (var i = 0; i < paramInfos.Length; i++) |
|||
{ |
|||
return delegate (object target, object[] parameters) |
|||
{ |
|||
executor(target, parameters); |
|||
return null; |
|||
}; |
|||
var paramInfo = paramInfos[i]; |
|||
var valueObj = Expression.ArrayIndex(parametersParameter, Expression.Constant(i)); |
|||
var valueCast = Expression.Convert(valueObj, paramInfo.ParameterType); |
|||
|
|||
// valueCast is "(Ti) parameters[i]"
|
|||
parameters.Add(valueCast); |
|||
} |
|||
|
|||
private static MethodExecutorAsync GetExecutorAsync( |
|||
MethodInfo methodInfo, |
|||
TypeInfo targetTypeInfo, |
|||
CoercedAwaitableInfo coercedAwaitableInfo) |
|||
{ |
|||
// Parameters to executor
|
|||
var targetParameter = Expression.Parameter(typeof(object), "target"); |
|||
var parametersParameter = Expression.Parameter(typeof(object[]), "parameters"); |
|||
|
|||
// Build parameter list
|
|||
var parameters = new List<Expression>(); |
|||
var paramInfos = methodInfo.GetParameters(); |
|||
for (var i = 0; i < paramInfos.Length; i++) |
|||
{ |
|||
var paramInfo = paramInfos[i]; |
|||
var valueObj = Expression.ArrayIndex(parametersParameter, Expression.Constant(i)); |
|||
var valueCast = Expression.Convert(valueObj, paramInfo.ParameterType); |
|||
|
|||
// valueCast is "(Ti) parameters[i]"
|
|||
parameters.Add(valueCast); |
|||
} |
|||
|
|||
// Call method
|
|||
var instanceCast = Expression.Convert(targetParameter, targetTypeInfo.AsType()); |
|||
var methodCall = Expression.Call(instanceCast, methodInfo, parameters); |
|||
|
|||
// Using the method return value, construct an ObjectMethodExecutorAwaitable based on
|
|||
// the info we have about its implementation of the awaitable pattern. Note that all
|
|||
// the funcs/actions we construct here are precompiled, so that only one instance of
|
|||
// each is preserved throughout the lifetime of the ObjectMethodExecutor.
|
|||
|
|||
// var getAwaiterFunc = (object awaitable) =>
|
|||
// (object)((CustomAwaitableType)awaitable).GetAwaiter();
|
|||
var customAwaitableParam = Expression.Parameter(typeof(object), "awaitable"); |
|||
var awaitableInfo = coercedAwaitableInfo.AwaitableInfo; |
|||
var postCoercionMethodReturnType = coercedAwaitableInfo.CoercerResultType ?? methodInfo.ReturnType; |
|||
var getAwaiterFunc = Expression.Lambda<Func<object, object>>( |
|||
// Call method
|
|||
var instanceCast = Expression.Convert(targetParameter, targetTypeInfo.AsType()); |
|||
var methodCall = Expression.Call(instanceCast, methodInfo, parameters); |
|||
|
|||
// Using the method return value, construct an ObjectMethodExecutorAwaitable based on
|
|||
// the info we have about its implementation of the awaitable pattern. Note that all
|
|||
// the funcs/actions we construct here are precompiled, so that only one instance of
|
|||
// each is preserved throughout the lifetime of the ObjectMethodExecutor.
|
|||
|
|||
// var getAwaiterFunc = (object awaitable) =>
|
|||
// (object)((CustomAwaitableType)awaitable).GetAwaiter();
|
|||
var customAwaitableParam = Expression.Parameter(typeof(object), "awaitable"); |
|||
var awaitableInfo = coercedAwaitableInfo.AwaitableInfo; |
|||
var postCoercionMethodReturnType = coercedAwaitableInfo.CoercerResultType ?? methodInfo.ReturnType; |
|||
var getAwaiterFunc = Expression.Lambda<Func<object, object>>( |
|||
Expression.Convert( |
|||
Expression.Call( |
|||
Expression.Convert(customAwaitableParam, postCoercionMethodReturnType), |
|||
awaitableInfo.GetAwaiterMethod), |
|||
typeof(object)), |
|||
customAwaitableParam).Compile(); |
|||
|
|||
// var isCompletedFunc = (object awaiter) =>
|
|||
// ((CustomAwaiterType)awaiter).IsCompleted;
|
|||
var isCompletedParam = Expression.Parameter(typeof(object), "awaiter"); |
|||
var isCompletedFunc = Expression.Lambda<Func<object, bool>>( |
|||
Expression.MakeMemberAccess( |
|||
Expression.Convert(isCompletedParam, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterIsCompletedProperty), |
|||
isCompletedParam).Compile(); |
|||
|
|||
var getResultParam = Expression.Parameter(typeof(object), "awaiter"); |
|||
Func<object, object> getResultFunc; |
|||
if (awaitableInfo.ResultType == typeof(void)) |
|||
// var getResultFunc = (object awaiter) =>
|
|||
// {
|
|||
// ((CustomAwaiterType)awaiter).GetResult(); // We need to invoke this to surface any exceptions
|
|||
// return (object)null;
|
|||
// };
|
|||
getResultFunc = Expression.Lambda<Func<object, object>>( |
|||
Expression.Block( |
|||
Expression.Call( |
|||
Expression.Convert(getResultParam, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterGetResultMethod), |
|||
Expression.Constant(null) |
|||
), |
|||
getResultParam).Compile(); |
|||
else |
|||
// var getResultFunc = (object awaiter) =>
|
|||
// (object)((CustomAwaiterType)awaiter).GetResult();
|
|||
getResultFunc = Expression.Lambda<Func<object, object>>( |
|||
Expression.Convert( |
|||
Expression.Call( |
|||
Expression.Convert(customAwaitableParam, postCoercionMethodReturnType), |
|||
awaitableInfo.GetAwaiterMethod), |
|||
Expression.Convert(getResultParam, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterGetResultMethod), |
|||
typeof(object)), |
|||
customAwaitableParam).Compile(); |
|||
|
|||
// var isCompletedFunc = (object awaiter) =>
|
|||
// ((CustomAwaiterType)awaiter).IsCompleted;
|
|||
var isCompletedParam = Expression.Parameter(typeof(object), "awaiter"); |
|||
var isCompletedFunc = Expression.Lambda<Func<object, bool>>( |
|||
Expression.MakeMemberAccess( |
|||
Expression.Convert(isCompletedParam, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterIsCompletedProperty), |
|||
isCompletedParam).Compile(); |
|||
|
|||
var getResultParam = Expression.Parameter(typeof(object), "awaiter"); |
|||
Func<object, object> getResultFunc; |
|||
if (awaitableInfo.ResultType == typeof(void)) |
|||
{ |
|||
getResultFunc = Expression.Lambda<Func<object, object>>( |
|||
Expression.Block( |
|||
Expression.Call( |
|||
Expression.Convert(getResultParam, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterGetResultMethod), |
|||
Expression.Constant(null) |
|||
), |
|||
getResultParam).Compile(); |
|||
} |
|||
else |
|||
{ |
|||
getResultFunc = Expression.Lambda<Func<object, object>>( |
|||
Expression.Convert( |
|||
Expression.Call( |
|||
Expression.Convert(getResultParam, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterGetResultMethod), |
|||
typeof(object)), |
|||
getResultParam).Compile(); |
|||
} |
|||
|
|||
// var onCompletedFunc = (object awaiter, Action continuation) => {
|
|||
// ((CustomAwaiterType)awaiter).OnCompleted(continuation);
|
|||
getResultParam).Compile(); |
|||
|
|||
// var onCompletedFunc = (object awaiter, Action continuation) => {
|
|||
// ((CustomAwaiterType)awaiter).OnCompleted(continuation);
|
|||
// };
|
|||
var onCompletedParam1 = Expression.Parameter(typeof(object), "awaiter"); |
|||
var onCompletedParam2 = Expression.Parameter(typeof(Action), "continuation"); |
|||
var onCompletedFunc = Expression.Lambda<Action<object, Action>>( |
|||
Expression.Call( |
|||
Expression.Convert(onCompletedParam1, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterOnCompletedMethod, |
|||
onCompletedParam2), |
|||
onCompletedParam1, |
|||
onCompletedParam2).Compile(); |
|||
|
|||
Action<object, Action>? unsafeOnCompletedFunc = null; |
|||
if (awaitableInfo.AwaiterUnsafeOnCompletedMethod != null) |
|||
{ |
|||
// var unsafeOnCompletedFunc = (object awaiter, Action continuation) => {
|
|||
// ((CustomAwaiterType)awaiter).UnsafeOnCompleted(continuation);
|
|||
// };
|
|||
var onCompletedParam1 = Expression.Parameter(typeof(object), "awaiter"); |
|||
var onCompletedParam2 = Expression.Parameter(typeof(Action), "continuation"); |
|||
var onCompletedFunc = Expression.Lambda<Action<object, Action>>( |
|||
var unsafeOnCompletedParam1 = Expression.Parameter(typeof(object), "awaiter"); |
|||
var unsafeOnCompletedParam2 = Expression.Parameter(typeof(Action), "continuation"); |
|||
unsafeOnCompletedFunc = Expression.Lambda<Action<object, Action>>( |
|||
Expression.Call( |
|||
Expression.Convert(onCompletedParam1, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterOnCompletedMethod, |
|||
onCompletedParam2), |
|||
onCompletedParam1, |
|||
onCompletedParam2).Compile(); |
|||
|
|||
Action<object, Action> unsafeOnCompletedFunc = null; |
|||
if (awaitableInfo.AwaiterUnsafeOnCompletedMethod != null) |
|||
{ |
|||
// var unsafeOnCompletedFunc = (object awaiter, Action continuation) => {
|
|||
// ((CustomAwaiterType)awaiter).UnsafeOnCompleted(continuation);
|
|||
// };
|
|||
var unsafeOnCompletedParam1 = Expression.Parameter(typeof(object), "awaiter"); |
|||
var unsafeOnCompletedParam2 = Expression.Parameter(typeof(Action), "continuation"); |
|||
unsafeOnCompletedFunc = Expression.Lambda<Action<object, Action>>( |
|||
Expression.Call( |
|||
Expression.Convert(unsafeOnCompletedParam1, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterUnsafeOnCompletedMethod, |
|||
unsafeOnCompletedParam2), |
|||
unsafeOnCompletedParam1, |
|||
unsafeOnCompletedParam2).Compile(); |
|||
} |
|||
|
|||
// If we need to pass the method call result through a coercer function to get an
|
|||
// awaitable, then do so.
|
|||
var coercedMethodCall = coercedAwaitableInfo.RequiresCoercion |
|||
? Expression.Invoke(coercedAwaitableInfo.CoercerExpression, methodCall) |
|||
: (Expression)methodCall; |
|||
|
|||
// return new ObjectMethodExecutorAwaitable(
|
|||
// (object)coercedMethodCall,
|
|||
// getAwaiterFunc,
|
|||
// isCompletedFunc,
|
|||
// getResultFunc,
|
|||
// onCompletedFunc,
|
|||
// unsafeOnCompletedFunc);
|
|||
var returnValueExpression = Expression.New( |
|||
_objectMethodExecutorAwaitableConstructor, |
|||
Expression.Convert(coercedMethodCall, typeof(object)), |
|||
Expression.Constant(getAwaiterFunc), |
|||
Expression.Constant(isCompletedFunc), |
|||
Expression.Constant(getResultFunc), |
|||
Expression.Constant(onCompletedFunc), |
|||
Expression.Constant(unsafeOnCompletedFunc, typeof(Action<object, Action>))); |
|||
|
|||
var lambda = |
|||
Expression.Lambda<MethodExecutorAsync>(returnValueExpression, targetParameter, parametersParameter); |
|||
return lambda.Compile(); |
|||
Expression.Convert(unsafeOnCompletedParam1, awaitableInfo.AwaiterType), |
|||
awaitableInfo.AwaiterUnsafeOnCompletedMethod, |
|||
unsafeOnCompletedParam2), |
|||
unsafeOnCompletedParam1, |
|||
unsafeOnCompletedParam2).Compile(); |
|||
} |
|||
|
|||
private delegate ObjectMethodExecutorAwaitable MethodExecutorAsync(object target, params object[] parameters); |
|||
// If we need to pass the method call result through a coercer function to get an
|
|||
// awaitable, then do so.
|
|||
var coercedMethodCall = coercedAwaitableInfo.RequiresCoercion |
|||
? Expression.Invoke(coercedAwaitableInfo.CoercerExpression, methodCall) |
|||
: (Expression)methodCall; |
|||
|
|||
// return new ObjectMethodExecutorAwaitable(
|
|||
// (object)coercedMethodCall,
|
|||
// getAwaiterFunc,
|
|||
// isCompletedFunc,
|
|||
// getResultFunc,
|
|||
// onCompletedFunc,
|
|||
// unsafeOnCompletedFunc);
|
|||
var returnValueExpression = Expression.New( |
|||
_objectMethodExecutorAwaitableConstructor, |
|||
Expression.Convert(coercedMethodCall, typeof(object)), |
|||
Expression.Constant(getAwaiterFunc), |
|||
Expression.Constant(isCompletedFunc), |
|||
Expression.Constant(getResultFunc), |
|||
Expression.Constant(onCompletedFunc), |
|||
Expression.Constant(unsafeOnCompletedFunc, typeof(Action<object, Action>))); |
|||
|
|||
var lambda = |
|||
Expression.Lambda<MethodExecutorAsync>(returnValueExpression, targetParameter, parametersParameter); |
|||
return lambda.Compile(); |
|||
} |
|||
|
|||
private delegate object MethodExecutor(object target, params object[] parameters); |
|||
private delegate ObjectMethodExecutorAwaitable MethodExecutorAsync(object target, object?[]? parameters); |
|||
|
|||
private delegate void VoidMethodExecutor(object target, object[] parameters); |
|||
} |
|||
private delegate object? MethodExecutor(object target, object?[]? parameters); |
|||
|
|||
private delegate void VoidMethodExecutor(object target, object?[]? parameters); |
|||
} |
|||
|
|||
@ -1,115 +1,114 @@ |
|||
using System; |
|||
using System.Runtime.CompilerServices; |
|||
|
|||
namespace LINGYUN.Abp.EventBus.CAP.Internal |
|||
namespace LINGYUN.Abp.EventBus.CAP.Internal; |
|||
|
|||
/// <summary>
|
|||
/// Provides a common awaitable structure that <see cref="ObjectMethodExecutor.ExecuteAsync" /> can
|
|||
/// return, regardless of whether the underlying value is a System.Task, an FSharpAsync, or an
|
|||
/// application-defined custom awaitable.
|
|||
/// </summary>
|
|||
internal readonly struct ObjectMethodExecutorAwaitable |
|||
{ |
|||
/// <summary>
|
|||
/// Provides a common awaitable structure that <see cref="ObjectMethodExecutor.ExecuteAsync" /> can
|
|||
/// return, regardless of whether the underlying value is a System.Task, an FSharpAsync, or an
|
|||
/// application-defined custom awaitable.
|
|||
/// </summary>
|
|||
internal struct ObjectMethodExecutorAwaitable |
|||
private readonly object _customAwaitable; |
|||
private readonly Func<object, object> _getAwaiterMethod; |
|||
private readonly Func<object, bool> _isCompletedMethod; |
|||
private readonly Func<object, object> _getResultMethod; |
|||
private readonly Action<object, Action> _onCompletedMethod; |
|||
private readonly Action<object, Action> _unsafeOnCompletedMethod; |
|||
|
|||
// Perf note: since we're requiring the customAwaitable to be supplied here as an object,
|
|||
// this will trigger a further allocation if it was a value type (i.e., to box it). We can't
|
|||
// fix this by making the customAwaitable type generic, because the calling code typically
|
|||
// does not know the type of the awaitable/awaiter at compile-time anyway.
|
|||
//
|
|||
// However, we could fix it by not passing the customAwaitable here at all, and instead
|
|||
// passing a func that maps directly from the target object (e.g., controller instance),
|
|||
// target method (e.g., action method info), and params array to the custom awaiter in the
|
|||
// GetAwaiter() method below. In effect, by delaying the actual method call until the
|
|||
// upstream code calls GetAwaiter on this ObjectMethodExecutorAwaitable instance.
|
|||
// This optimization is not currently implemented because:
|
|||
// [1] It would make no difference when the awaitable was an object type, which is
|
|||
// by far the most common scenario (e.g., System.Task<T>).
|
|||
// [2] It would be complex - we'd need some kind of object pool to track all the parameter
|
|||
// arrays until we needed to use them in GetAwaiter().
|
|||
// We can reconsider this in the future if there's a need to optimize for ValueTask<T>
|
|||
// or other value-typed awaitables.
|
|||
|
|||
public ObjectMethodExecutorAwaitable( |
|||
object customAwaitable, |
|||
Func<object, object> getAwaiterMethod, |
|||
Func<object, bool> isCompletedMethod, |
|||
Func<object, object> getResultMethod, |
|||
Action<object, Action> onCompletedMethod, |
|||
Action<object, Action> unsafeOnCompletedMethod) |
|||
{ |
|||
_customAwaitable = customAwaitable; |
|||
_getAwaiterMethod = getAwaiterMethod; |
|||
_isCompletedMethod = isCompletedMethod; |
|||
_getResultMethod = getResultMethod; |
|||
_onCompletedMethod = onCompletedMethod; |
|||
_unsafeOnCompletedMethod = unsafeOnCompletedMethod; |
|||
} |
|||
|
|||
public Awaiter GetAwaiter() |
|||
{ |
|||
var customAwaiter = _getAwaiterMethod(_customAwaitable); |
|||
return new Awaiter(customAwaiter, _isCompletedMethod, _getResultMethod, _onCompletedMethod, |
|||
_unsafeOnCompletedMethod); |
|||
} |
|||
|
|||
public readonly struct Awaiter : ICriticalNotifyCompletion |
|||
{ |
|||
private readonly object _customAwaitable; |
|||
private readonly Func<object, object> _getAwaiterMethod; |
|||
private readonly object _customAwaiter; |
|||
private readonly Func<object, bool> _isCompletedMethod; |
|||
private readonly Func<object, object> _getResultMethod; |
|||
private readonly Action<object, Action> _onCompletedMethod; |
|||
private readonly Action<object, Action> _unsafeOnCompletedMethod; |
|||
|
|||
// Perf note: since we're requiring the customAwaitable to be supplied here as an object,
|
|||
// this will trigger a further allocation if it was a value type (i.e., to box it). We can't
|
|||
// fix this by making the customAwaitable type generic, because the calling code typically
|
|||
// does not know the type of the awaitable/awaiter at compile-time anyway.
|
|||
//
|
|||
// However, we could fix it by not passing the customAwaitable here at all, and instead
|
|||
// passing a func that maps directly from the target object (e.g., controller instance),
|
|||
// target method (e.g., action method info), and params array to the custom awaiter in the
|
|||
// GetAwaiter() method below. In effect, by delaying the actual method call until the
|
|||
// upstream code calls GetAwaiter on this ObjectMethodExecutorAwaitable instance.
|
|||
// This optimization is not currently implemented because:
|
|||
// [1] It would make no difference when the awaitable was an object type, which is
|
|||
// by far the most common scenario (e.g., System.Task<T>).
|
|||
// [2] It would be complex - we'd need some kind of object pool to track all the parameter
|
|||
// arrays until we needed to use them in GetAwaiter().
|
|||
// We can reconsider this in the future if there's a need to optimize for ValueTask<T>
|
|||
// or other value-typed awaitables.
|
|||
|
|||
public ObjectMethodExecutorAwaitable( |
|||
object customAwaitable, |
|||
Func<object, object> getAwaiterMethod, |
|||
public Awaiter( |
|||
object customAwaiter, |
|||
Func<object, bool> isCompletedMethod, |
|||
Func<object, object> getResultMethod, |
|||
Action<object, Action> onCompletedMethod, |
|||
Action<object, Action> unsafeOnCompletedMethod) |
|||
{ |
|||
_customAwaitable = customAwaitable; |
|||
_getAwaiterMethod = getAwaiterMethod; |
|||
_customAwaiter = customAwaiter; |
|||
_isCompletedMethod = isCompletedMethod; |
|||
_getResultMethod = getResultMethod; |
|||
_onCompletedMethod = onCompletedMethod; |
|||
_unsafeOnCompletedMethod = unsafeOnCompletedMethod; |
|||
} |
|||
|
|||
public Awaiter GetAwaiter() |
|||
public bool IsCompleted => _isCompletedMethod(_customAwaiter); |
|||
|
|||
public object GetResult() |
|||
{ |
|||
var customAwaiter = _getAwaiterMethod(_customAwaitable); |
|||
return new Awaiter(customAwaiter, _isCompletedMethod, _getResultMethod, _onCompletedMethod, |
|||
_unsafeOnCompletedMethod); |
|||
return _getResultMethod(_customAwaiter); |
|||
} |
|||
|
|||
public struct Awaiter : ICriticalNotifyCompletion |
|||
public void OnCompleted(Action continuation) |
|||
{ |
|||
private readonly object _customAwaiter; |
|||
private readonly Func<object, bool> _isCompletedMethod; |
|||
private readonly Func<object, object> _getResultMethod; |
|||
private readonly Action<object, Action> _onCompletedMethod; |
|||
private readonly Action<object, Action> _unsafeOnCompletedMethod; |
|||
|
|||
public Awaiter( |
|||
object customAwaiter, |
|||
Func<object, bool> isCompletedMethod, |
|||
Func<object, object> getResultMethod, |
|||
Action<object, Action> onCompletedMethod, |
|||
Action<object, Action> unsafeOnCompletedMethod) |
|||
{ |
|||
_customAwaiter = customAwaiter; |
|||
_isCompletedMethod = isCompletedMethod; |
|||
_getResultMethod = getResultMethod; |
|||
_onCompletedMethod = onCompletedMethod; |
|||
_unsafeOnCompletedMethod = unsafeOnCompletedMethod; |
|||
} |
|||
|
|||
public bool IsCompleted => _isCompletedMethod(_customAwaiter); |
|||
|
|||
public object GetResult() |
|||
{ |
|||
return _getResultMethod(_customAwaiter); |
|||
} |
|||
|
|||
public void OnCompleted(Action continuation) |
|||
{ |
|||
_onCompletedMethod(_customAwaiter, continuation); |
|||
} |
|||
_onCompletedMethod(_customAwaiter, continuation); |
|||
} |
|||
|
|||
public void UnsafeOnCompleted(Action continuation) |
|||
{ |
|||
// If the underlying awaitable implements ICriticalNotifyCompletion, use its UnsafeOnCompleted.
|
|||
// If not, fall back on using its OnCompleted.
|
|||
//
|
|||
// Why this is safe:
|
|||
// - Implementing ICriticalNotifyCompletion is a way of saying the caller can choose whether it
|
|||
// needs the execution context to be preserved (which it signals by calling OnCompleted), or
|
|||
// that it doesn't (which it signals by calling UnsafeOnCompleted). Obviously it's faster *not*
|
|||
// to preserve and restore the context, so we prefer that where possible.
|
|||
// - If a caller doesn't need the execution context to be preserved and hence calls UnsafeOnCompleted,
|
|||
// there's no harm in preserving it anyway - it's just a bit of wasted cost. That's what will happen
|
|||
// if a caller sees that the proxy implements ICriticalNotifyCompletion but the proxy chooses to
|
|||
// pass the call on to the underlying awaitable's OnCompleted method.
|
|||
public void UnsafeOnCompleted(Action continuation) |
|||
{ |
|||
// If the underlying awaitable implements ICriticalNotifyCompletion, use its UnsafeOnCompleted.
|
|||
// If not, fall back on using its OnCompleted.
|
|||
//
|
|||
// Why this is safe:
|
|||
// - Implementing ICriticalNotifyCompletion is a way of saying the caller can choose whether it
|
|||
// needs the execution context to be preserved (which it signals by calling OnCompleted), or
|
|||
// that it doesn't (which it signals by calling UnsafeOnCompleted). Obviously it's faster *not*
|
|||
// to preserve and restore the context, so we prefer that where possible.
|
|||
// - If a caller doesn't need the execution context to be preserved and hence calls UnsafeOnCompleted,
|
|||
// there's no harm in preserving it anyway - it's just a bit of wasted cost. That's what will happen
|
|||
// if a caller sees that the proxy implements ICriticalNotifyCompletion but the proxy chooses to
|
|||
// pass the call on to the underlying awaitable's OnCompleted method.
|
|||
|
|||
var underlyingMethodToUse = _unsafeOnCompletedMethod ?? _onCompletedMethod; |
|||
underlyingMethodToUse(_customAwaiter, continuation); |
|||
} |
|||
var underlyingMethodToUse = _unsafeOnCompletedMethod ?? _onCompletedMethod; |
|||
underlyingMethodToUse(_customAwaiter, continuation); |
|||
} |
|||
} |
|||
} |
|||
Loading…
Reference in new issue