如何在asp.net核心中间件中进行DI？

Question

我试图将依赖注入我的中间件构造函数,如下所示

public class CreateCompanyMiddleware
{
    private readonly RequestDelegate _next;
    private readonly UserManager<ApplicationUser> _userManager;

    public CreateCompanyMiddleware(RequestDelegate next
        , UserManager<ApplicationUser> userManager
        )
    {
        _next = next;
    }

    public async Task Invoke(HttpContext context)
    {
        await _next.Invoke(context);
    }
}

我的Startup.cs文件看起来像

public void ConfigureServices(IServiceCollection services)
{
    services.AddDbContext<ApplicationDbContext>(options =>
        options.UseMySql(Configuration.GetConnectionString("IdentityConnection")));

    services.AddIdentity<ApplicationUser, IdentityRole>()
        .AddEntityFrameworkStores<ApplicationDbContext>()
        .AddDefaultTokenProviders();
    ...

    app.UseMiddleware<CreateCompanyMiddleware>();

    ...

但是我收到了这个错误

启动应用程序时发生错误.InvalidOperationException:无法从根提供程序解析作用域服务'Microsoft.AspNetCore.Identity.UserManager`1 [Common.Models.ApplicationUser]'.Microsoft.Extensions.DependencyInjection.ServiceLookup.CallSiteValidator.ValidateResolution(类型serviceType,IServiceScope范围,IServiceScope rootScope)

Answer 1

UserManager<ApplicationUser>是(默认情况下)注册为作用域依赖项,而您的CreateCompanyMiddleware中间件是在应用程序启动时构建的(有效地使其成为单例).这是一个相当标准的错误,表示您不能将作用域依赖项放入单例类中.

在这种情况下,修复很简单 - 您可以将注入UserManager<ApplicationUser>到您的Invoke方法中:

public async Task Invoke(HttpContext context, UserManager<ApplicationUser> userManager)
{
    await _next.Invoke(context);
}

这在ASP.NET核心中间件中有记录:每请求依赖项:

由于中间件是在应用程序启动时构建的,而不是按请求构建的,因此中间件构造函数使用的作用域生存期服务在每个请求期间不会与其他依赖注入的类型共享.如果必须在中间件和其他类型之间共享作用域服务,请将这些服务添加到Invoke方法的签名中.该Invoke方法可以接受由DI填充的其他参数.

Answer 2

另一种方法是通过IMiddleware接口创建中间件并将其注册为服务

例如，中间件

public class CreateCompanyMiddlewareByInterface : IMiddleware
{
    private readonly UserManager<ApplicationUser> _userManager;

    public CreateCompanyMiddlewareByInterface(UserManager<ApplicationUser> userManager )
    {
        this._userManager = userManager;
    }


    public Task InvokeAsync(HttpContext context, RequestDelegate next)
    {
        return next(context);
    }
} 

和服务注册：

services.AddScoped<CreateCompanyMiddlewareByInterface>();

3. 那为什么会发生呢？

使用的中间件IMiddleware由UseMiddlewareInterface(appBuilder, middlewareType type) 以下组件构建：

private static IApplicationBuilder UseMiddlewareInterface(IApplicationBuilder app, Type middlewareType)
{
    return app.Use(next =>
    {
        return async context =>
        {
            var middlewareFactory = (IMiddlewareFactory)context.RequestServices.GetService(typeof(IMiddlewareFactory));
            if (middlewareFactory == null) { /* throw ... */ }

            var middleware = middlewareFactory.Create(middlewareType);
            if (middleware == null) { /* throw ... */ }

            try{
                await middleware.InvokeAsync(context, next);
            }
            finally{
                middlewareFactory.Release(middleware);
            }
        };
    });
}

此处的代码context=>{}是按请求执行的。因此，每次有传入请求时，var middleware = middlewareFactory.Create(middlewareType);都会执行，然后从中请求中间件middlewareType（已注册为服务）ServiceProvider。

至于约定俗成的中间件，没有工厂创建它们。

这些实例都是ActivatorUtilities.CreateInstance()在启动时创建的。以及任何Invoke约定俗成的中间件方法，例如

Task Invoke(HttpContext context,UserManager<ApplicationUser> userManage, ILoggerFactory loggeryFactory , ... )

将被编译成如下功能：

Task Invoke(Middleware instance, HttpContext httpContext, IServiceprovider provider)
{
    var useManager  /* = get service from service provider */ ;
    var log = /* = get service from service provider */ ;
    // ... 
    return instance.Invoke(httpContext,userManager,log, ...);
}

如您所见，这里是在启动时创建实例的，Invoke每个请求都请求method的那些服务。