.NET Core 自定义中间件 Middleware
引言
很多看了上一章的朋友私信博主,问如何自定义,自己的中间件(Middleware),毕竟在实际的项目中,大家会有很多需求要用到中间件,比如防盗链、缓存、日志等等功能,于是博主这边就简单讲解一下框架、组件惯用的优雅手法,官方也推荐这种写法,这样会使得我们扩展性更好,也不会破坏原本结构。
什么是中间件
中间件是一种装配到应用管道以处理请求和响应的软件。 每个组件:
- 选择是否将请求传递到管道中的下一个组件。
- 可在管道中的下一个组件前后执行工作。
使用 RunMap 和 Use 扩展方法来配置请求委托,请求委托用于生成请求管道。 请求委托处理每个 HTTP 请求。
简单的说,我们按需求决定使用哪些组件,程序运行时,一个HTTP请求过来,程序执行流程,是按照我们定义的组件顺序执行的。所以我们项目上的中间件放置顺序是不能乱的,并且不用的也不要装配,避免消耗性能。
概念图:
如果想做其他相关了解,博主建议直接在官网上看文档,微软的文档写的还是很好的,还提供的Demo下载,比很多网上博客讲的好。
微软官网地址:ASP.NET Core 中间件 | Microsoft Docs
接下来进入正题,我们写一个,把所有http请求地址发送到MQ的中间件:
1.编写MsgMiddleware类
这里我们就使用直接编写Middleware类的方式,大家也可以使用实现IMiddleware接口的方式。两种底层原理不一样:前者是框架启动时就实例化;后者是请求来时才实例化,用完立即释放。
编写Middleware类注意:
- 编写好InvokeAsync或者Invoke方法
- 构造函数参数需要一个RequestDelegate类型的委托
因为这块源码是直接通过反射创建和调用的,不过源码也会对我们的类进行规范校验。
代码逻辑:
这里我们就实现一个简单逻辑,根据Options配置SendFlag是否开启发送MQ,如果是,就调用我们的ISendMessage发送服务。服务获取方式大家也可以使用注入的方式
ISendMessage服务、Options配置类代码,我放到了后面讲解,毕竟逻辑简单,而且也不是重点。
查看代码
public class MsgMiddleware
{
private readonly RequestDelegate _next;
private readonly MsgOptions options;
/// <summary>
/// 管道执行到该中间件时候下一个中间件的RequestDelegate请求委托,如果有其它参数,也同样通过注入的方式获得
/// </summary>
/// <param name="next"></param>
public MsgMiddleware(RequestDelegate next, IOptions<MsgOptions> options)
{
//通过注入方式获得对象
_next = next;
this.options = options.Value;
}
/// <summary>
/// 自定义中间件要执行的逻辑
/// </summary>
/// <param name="context"></param>
/// <returns></returns>
public async Task Invoke(HttpContext context)
{
if (options.SendFlag)
{
//通过IOC获取ISendMessage
ISendMessage _message = context.RequestServices.GetService<ISendMessage>();
_message.Send(context.Request.Path.Value);
}
//把context传进去执行下一个中间件
await _next(context);
}
}
2.编写IApplicationBuilder扩展方法
这里我们就定义两个扩展方法,用来应对在Use时候直接配置Options,或者后面通过IOC方式配置Options
注意:如果在Use时候直接配置参数,我们的Options需要通过Options.Create(op)帮我们包裹成IOptions<>类型,因为编写的中间件参数是Options模式的一个IOptions接口
查看代码
public static class MsgBuilderMiddlewareExtensions
{
//没有Option,依靠IOC的Add的方式设置
public static IApplicationBuilder UseMsgSend(this IApplicationBuilder app)
{
if (app == null)
{
throw new ArgumentNullException(nameof(app));
}
return app.UseMiddleware<MsgMiddleware>();
}
//Use直接配置Options
public static IApplicationBuilder UseMsgSend(this IApplicationBuilder app, Action<MsgOptions> optionsAction)
{
if (app == null)
{
throw new ArgumentNullException(nameof(app));
}
//1 不能直接初始化Option
//2 也不能找到ServiceCollection去初始化了
MsgOptions op = new MsgOptions();
optionsAction(op);
return app.UseMiddleware<MsgMiddleware>(Options.Create(op));
}
}
3.编写IServiceCollection容器扩展方法
这样的好处是,我们这样可以集中注册内部映射,隐藏实现细节,外部不需要关心,这个好处博主就不多说了,因为大家使用别人写的组件也心有体会,对于使用者来说只需要关心怎么用,他内部有多么复杂的逻辑是不知道的,也不需要知道.
这里也是编写两个扩展方法,用来应对在Use时候直接配置Options,或者后面通过IOC方式配置Options
代码逻辑:
这两个方法里面逻辑就是,注册好业务服务、Options 等等。。。
查看代码
/// <summary>
/// 这样可以集中注册内部映射,外部不需要关心
/// </summary>
public static class ServiceCollectionExtensions
{
/// <summary>
/// 配置信息初始化由Middleware
/// </summary>
/// <param name="services"></param>
/// <returns></returns>
public static IServiceCollection AddSendMessage(this IServiceCollection services)
{
return services.AddSingleton<ISendMessage, SendMessage>();
}
/// <summary>
/// 配置信息直接用Option的模式去初始化
/// </summary>
/// <param name="services"></param>
/// <param name="configure"></param>
/// <returns></returns>
public static IServiceCollection AddSendMessage(this IServiceCollection services, Action<MsgOptions> configure)
{
MsgOptions msg = new MsgOptions();
configure(msg);
services.Configure(configure);
services.Configure(msg.RabbitMQOptions);
return services.AddSendMessage();
}
}
4.定义Options类
定义好我们业务逻辑需要的配置信息Options类
查看代码
//MQ配置类
public class RabbitMQOptions
{
public string IP { get; set; }
public string Port { get; set; }
}
//Msg配置类
public class MsgOptions
{
//是否发送
public bool SendFlag { get; set; }
//MQ配置
internal Action<RabbitMQOptions>? RabbitMQOptions { get; private set; }
public void Register(Action<RabbitMQOptions> action)
{
RabbitMQOptions = action;
}
}
//Msg配置扩展方法,用来设置其MQ配置信息
public static class MsgOptionsExtensions
{
public static void UseRabbitMQ(this MsgOptions options, Action<RabbitMQOptions> configure)
{
if (configure == null)
{
throw new ArgumentNullException(nameof(configure));
}
options.Register(configure);
}
}
5.编写Msg服务
这里我们就用打印信息的方式来表示我们将数据发送至MQ了
public interface ISendMessage
{
void Send(string message);
}
public class SendMessage : ISendMessage
{
private readonly RabbitMQOptions rabbitMQ;
public SendMessage(IOptions<RabbitMQOptions> rabbitMQ)
{
this.rabbitMQ = rabbitMQ.Value;
}
public void Send(string message)
{
Console.WriteLine($"发送消息:{message},至--->{rabbitMQ.IP}:{rabbitMQ.Port}服务器");
}
}
6.使用自定义中间件
我们只需要调用我们定义的扩展方法即可,博主这里用的.NET 6,使用的顶级语句。老版本的朋友就在Startup类里配置,调用方式是一样的
如下:
- Configure方法里使用,app.UseMsgSend()
- ConfigureServices方法里使用,services.AddSendMessage(x=>{...})
查看代码
using WebApplication1.MiddlewareExp;
using WebApplication1.MiddlewareExp.Middleware;
var builder = WebApplication.CreateBuilder(args);
// Add services to the container.
builder.Services.AddControllers();
// Learn more about configuring Swagger/OpenAPI at https://aka.ms/aspnetcore/swashbuckle
builder.Services.AddEndpointsApiExplorer();
builder.Services.AddSwaggerGen();
//配置中间件配置信息
builder.Services.AddSendMessage(c =>
{
c.SendFlag = true;
c.UseRabbitMQ(x =>
{
x.IP = "127.0.0.1";
x.Port = "8080";
});
});
var app = builder.Build();
// Configure the HTTP request pipeline.
if (app.Environment.IsDevelopment())
{
app.UseSwagger();
app.UseSwaggerUI();
}
// 使用自定义的Msg中间件
app.UseMsgSend();
app.UseAuthorization();
app.MapControllers();
app.Run();
运行效果
如图,我们请求多少次,请求都会经过我们中间件。