asp.net core 自定义中间件【以dapper为例】

在asp.net core开发中。按照国际案例开始。都是先在Nuget安装XXX包。比如我们今天要用到的Dapper

nuget里面安装Dapper

1.然后新建一个类文件DapperExtensions.cs

因为Dapper是IDbConnection扩展出来的，所以我们必须给IDbConnection一个默认的实现

/// <summary>
/// 注入服务
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="service"></param>
/// <returns></returns>
public static IServiceCollection AddDapper<T>(this IServiceCollection service) where T:class,IDbConnection
{
service.AddScoped<IDbConnection, T>();
return service;
}

如何使用呢？在Startup里面加入

services.AddDapper();

理论上到这里，就已经可以勉强使用了。但是本文是记录中间件的学习，所以我们还是得在后面学习一下中间件的写法

2.新建一个DapperMiddleWareExtensions.cs和DapperMiddleWare.cs文件

public class DapperMiddleWare
{

private readonly RequestDelegate _next;
private DapperOption _option;

public DapperMiddleWare(RequestDelegate next, DapperOption option)
{
_next = next;
this._option = option;
}

public async Task InvokeAsync(HttpContext context)
{

var conn = context.RequestServices.GetService<IDbConnection>();

if (_option != default(DapperOption))
{
if (!_option.connStr.IsNull())
{
conn.ConnectionString = _option.connStr;
}
}
// Call the next delegate/middleware in the pipeline
await _next(context);
}
}

public static class DapperMiddleWareExtensions
{
public static IApplicationBuilder UseDapper(this IApplicationBuilder builder, Action<DapperOption> option = null)
{
DapperOption opt = new DapperOption();
if (option != null)
{
option(opt);
}
return builder.UseMiddleware<DapperMiddleWare>(opt);
}
}

使用：

app.UseDapper(opt =>
{
opt.connStr = Configuration[“db:mysqlmaster”];
});

这两段代码非常简单。就是编写一个IApplicationBuilder的扩展方法，然后再扩展方法里面获取到注入的IDbconnection的接口，然后把委托方法传递进来的默认配置参数赋值进去，就可以了。

实际上，也可以在AddService的时候就可以把委托配置给赋值进去，很多第三方的库就是这么做的。