从 C# 中的 Mediatr Behaviour 获取命令处理程序

Question

在命令处理程序上，我想设置事务处理。我不想在所有方法上设置事务，因为它存在性能泄漏。本质上，我想设置此属性来检查是否应该设置事务。目前，我可以访问 request 属性，但我想访问handler 属性。

[ SqlTransactionAttribute ]只是一个简单的标记属性

[SqlTransactionAttribute]
public class LoginUserCommandHandler : IRequestHandler<LoginUserCommand, TokenDTO>
{
 //Command handler implementation
}

现在的问题是，在管道行为中如何确定一个处理程序是否具有 SqlTransactionAttribute

public class TransactionBehavior<TRequest, TResponse> :
IPipelineBehavior<TRequest, TResponse>
where TRequest : notnull
where TResponse : notnull
{
public async Task<TResponse> Handle(TRequest request, CancellationToken cancellationToken, RequestHandlerDelegate<TResponse> next)
{
    //how to get command handler 
    if (Attribute.GetCustomAttribute(typeof(handler), typeof(SqlTransactionAttribute)) != null)
    {
        await HandleSqlTransaction(next);
    }
}
}

将此属性添加到命令中允许我读取它（从 TRequest），但是，我更喜欢从处理程序中读取它，因为逻辑必须位于处理程序中，并且它将更具可读性。

Answer 1

将 RequestHandler 注入到构造函数中。

public class TransactionBehavior<TRequest, TResponse> :
    IPipelineBehavior<TRequest, TResponse>
    where TRequest : IRequest<TResponse>
    where TResponse : notnull
{
    private readonly IRequestHandler<TRequest, TResponse> requestHandler;

    public TransactionBehavior(IRequestHandler<TRequest, TResponse> requestHandler)
    {
        this.requestHandler = requestHandler;
    }

    public async Task<TResponse> Handle(TRequest request, CancellationToken cancellationToken, RequestHandlerDelegate<TResponse> next)
    {
        var hasSqlTransactionAttribute = requestHandler.GetType().GetCustomAttributes<SqlTransactionAttribute>().Any();

        if (hasSqlTransactionAttribute)
        {
            await HandleSqlTransaction(next);
        }
    }
}