SignalR - 多租户依赖注入

时间:2015-07-20 21:56:58

标签: c# dependency-injection signalr autofac

我需要根据租户的owin值来解析DbContext。但是在Hub的OnDisconnected方法的管道中,HttpContext是not accessible

我的中心课程:

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吗? 谢谢!

1 个答案:

答案 0 :(得分:2)

对于任何有兴趣的人,我最终会注入一个上下文工厂而不是上下文本身:

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)));