SignalR - 多租戶依賴注入

Question

我需要根據租戶的 owin 值解析 DbContext。 但是在 hub 的 OnDisconnected 方法的管道中， HttpContext 是不可訪問的。

我的樞紐班：

public class UserTrackingHub : Hub
{
    private readonly UserContext _context;

    public UserTrackingHub(UserContext context) { ... }

    public override async Task OnConnected() { /* OK HERE...*/ }

    public override async Task OnDisconnected(bool stopCalled)
    {
        // NEVER FIRES WITH IF I USE THE CTOR INJECTION.

        var connection = await _context.Connections.FindAsync(Context.ConnectionId);
        if (connection != null)
        {
            _context.Connections.Remove(connection);
            await _context.SaveChangesAsync();
        }
    }
}

這是我的 Autofac 配置：

    public static IContainer Register(IAppBuilder app)
    {
        var builder = new ContainerBuilder();

        // Other registers...

        builder.Register<UserContext>(c =>
        {    
            // Details and conditions omitted for brevity.

            var context = HttpContext.Current; // NULL in OnDisconnected pipeline.
            var owinContext = context.GetOwinContext();
            var tenant = owinContext.Environment["app.tenant"].ToString();
            var connection = GetConnectionString(tenant); 

            return new UserContext(connection);
        });

        var container = builder.Build();
        var config = new HubConfiguration
        {
            Resolver = new AutofacDependencyResolver(container)
        };

        app.MapSignalR(config);

        return container;
    }

有人可以幫助我以這種方式或任何其他方式識別租戶 OnDisconnected 嗎？
謝謝！

Answer 1

對於任何感興趣的人，我最終會注入一個上下文工廠而不是上下文本身：

public class UserTrackingHub : Hub
{
    private readonly Func<string, UserContext> _contextFactory;

    public UserTrackingHub(Func<string, UserContext> contextFactory) { ... }

    public override async Task OnConnected() { ... }

    public override async Task OnDisconnected(bool stopCalled)
    {
        var tenant = Context.Request.Cookies["app.tenant"].Value;

        using (var context = _contextFactory.Invoke(tenant))
        {
            var connection = await context.Connections.FindAsync(Context.ConnectionId);
            if (connection != null)
            {
                context.Connections.Remove(connection);
                await context.SaveChangesAsync();
            }
        }
    }
}

和 Autofac 的配置：

 // Resolve context based on tenant
 builder.Register<Func<string, UserContext>>(c => new Func<string, UserContext>(tenant => UserContextResolver.Resolve(tenant)));