gistfile1.cs

    public class SessionHandlerPipelineModule : HubPipelineModule
    {
        private readonly ISession _session;
        private ITransaction _transaction;

        public SessionHandlerPipelineModule(ISession session)
        {
            _session = session;
        }

        protected override bool OnBeforeConnect(IHub hub)
        {
            var baseHub = hub as BaseHub;
            if (baseHub != null)
            {
                baseHub.Session = _session;
                _transaction = _session.BeginTransaction();
            }

            return base.OnBeforeConnect(hub);
        }

        protected override bool OnBeforeDisconnect(IHub hub)
        {
            if(_transaction != null)
                _transaction.Commit();

            return base.OnBeforeDisconnect(hub);
        }
    }