ASP.NET Core 實作 MediatR 的 Pipeline 功能

之前的兩篇文章已經把 MediatR 的基本功能都介紹過了，現在要介紹 Pipeline 的功能

範例 Repository 在 Github

新增 LogPreProcessor 實作 IRequestPreProcessor

IRequestPreProcessor 指的是每一個 Request 之前都會執行一次，泛型的型別就是實作 IRequest 的型別，因為是在 Request 之前，所以在 Process 裡面只拿的到 Request 的資料

public class LogPreProcessor<TRequest> : IRequestPreProcessor<TRequest>
{
    private readonly ILogger<TRequest> _logger;

    public LogPreProcessor(ILogger<TRequest> logger)
    {
        _logger = logger;
    }

    public Task Process(TRequest request, CancellationToken cancellationToken)
    {
        _logger.LogInformation($"{nameof(LogPreProcessor<TRequest>)} Request -- {JsonConvert.SerializeObject(request)}");

        return Task.CompletedTask;
    }
}

在 Startup 裡面去註冊剛才新增的 LogPreProcessor

public void ConfigureServices(IServiceCollection services)
{
    services.AddMediatR(typeof(AddPersonCommand).GetTypeInfo().Assembly);
    services.AddTransient(typeof(IRequestPreProcessor<>), typeof(LogPreProcessor<>));
}

在這裡使用之前的新增 Person 來作測試，可以看到有把 Request 給 Log 下來

新增 LogPostProcessor 實作 IRequestPostProcessor

IRequestPostProcessor 指的是每一個 Request 之後都會執行一次，泛型的型別有兩個，就是實作 IRequest 的型別和 Response 的型別，因為是在 Request 之後，所以在 Process 可以拿的的到 Request 和 Response 的資料

public class LogPostProcessor<TRequest, TResponse> : IRequestPostProcessor<TRequest, TResponse>
{
    private readonly ILogger<TRequest> _logger;

    public LogPostProcessor(ILogger<TRequest> logger)
    {
        _logger = logger;
    }

    public Task Process(TRequest request, TResponse response)
    {
        _logger.LogInformation($"{nameof(LogPostProcessor<TRequest, TRequest>)} Request -- {JsonConvert.SerializeObject(request)}");
        _logger.LogInformation($"{nameof(LogPostProcessor<TRequest, TResponse>)} Response -- {JsonConvert.SerializeObject(response)}");

        return Task.CompletedTask;
    }
}

在 Startup 裡面把原本註冊的 LogPreProcessor 改成 LogPostProcessor

public void ConfigureServices(IServiceCollection services)
{
    services.AddMediatR(typeof(AddPersonCommand).GetTypeInfo().Assembly);
    services.AddTransient(typeof(IRequestPostProcessor<>), typeof(LogPostProcessor<>));
}

測試

新增 LogBehavior 實作 IPipelineBehavior

IPipelineBehavior 和 ASP.NET Core 的 Middleware 很像，裡面有 next 可以使用，可以自行決定在呼叫 IRequestHandler 的前或後要作什麼，泛型的型別有兩個，實作 IRequest 和 Response 的型別

   public class LogBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TResponse>
    {
        private readonly ILogger<TRequest> _logger;

        public LogBehavior(ILogger<TRequest> logger)
        {
            _logger = logger;
        }

        public async Task<TResponse> Handle(TRequest request, CancellationToken cancellationToken, RequestHandlerDelegate<TResponse> next)
        {
            _logger.LogInformation($" LogBehavior Request -- {JsonConvert.SerializeObject(request)}");

            var response = await next();

            _logger.LogInformation($" LogBehavior Response -- {JsonConvert.SerializeObject(response)}");

            return response;
        }
    }

在 Startup 裡面把註冊的型別換成 LogBehavior

public void ConfigureServices(IServiceCollection services)
{
    services.AddMediatR(typeof(AddPersonCommand).GetTypeInfo().Assembly);
    services.AddTransient(typeof(IPipelineBehavior<,>), typeof(LogBehavior<,>));
}

測試

組合不同的 IPipelineBehavior

在新增一個 RequestPerformanceBehavior，用來測量 Request 的執行時間

public class RequestPerformanceBehavior<TRequest, TResponse> : IPipelineBehavior<TRequest, TResponse>
{
    private readonly Stopwatch _timer;
    private readonly ILogger<TRequest> _logger;

    public RequestPerformanceBehavior(ILogger<TRequest> logger)
    {
        _timer = new Stopwatch();

        _logger = logger;
    }

    public async Task<TResponse> Handle(TRequest request, CancellationToken cancellationToken, RequestHandlerDelegate<TResponse> next)
    {
        _timer.Start();

        var response = await next();

        _timer.Stop();

        _logger.LogInformation($"{nameof(RequestPerformanceBehavior<TRequest, TRequest>)} running time - {_timer.ElapsedMilliseconds}");

        return response;
    }
}

在 Startup 裡面註冊 RequestPerformanceBehavior，加到最後面

public void ConfigureServices(IServiceCollection services)
{
    services.AddMediatR(typeof(AddPersonCommand).GetTypeInfo().Assembly);
    services.AddTransient(typeof(IPipelineBehavior<,>), typeof(LogBehavior<,>));
	services.AddTransient(typeof(IPipelineBehavior<,>), typeof(RequestPerformanceBehavior<,>));
}

測試，可以拿到執行 Request 的時間出現在 Log 前面

如果調整順序把 RequestPerformanceBehavior 移到 Pipeline 的第一個

public void ConfigureServices(IServiceCollection services)
{
    services.AddMediatR(typeof(AddPersonCommand).GetTypeInfo().Assembly);
	services.AddTransient(typeof(IPipelineBehavior<,>), typeof(RequestPerformanceBehavior<,>));
	services.AddTransient(typeof(IPipelineBehavior<,>), typeof(LogBehavior<,>));
}

再一次測試，可以拿到執行 Request 的時間出現在 Log 後面，也就是放的順序和出現的順序是相反的

後記

MediatR 有提供 Pipeline 的功能，某種程度上面還蠻方便的，不過也因為它的功能沒有那麼強大，所以使用的範圍來說還是有限的，如果需要作到比較多的事情，使用 ActionFilter 和 Middleware 還是比較方便一些