Action/AnyCable + Apartment

connection.rb

module ApplicationCable
  class Connection < ActionCable::Base::Connection
    # we need to keep tenant information for subsequent messages,
    # so let's store it as an identifier
    identified_by :tenant
    
    def connect
      # assuming you store current tenant in session
      self.tenant = request.session[:current_tenant]
      reject_unauthorized_connection unless tenant
      
      using_current_tenant do
        # set additional identifiers        
      end
    end
    
    # Override to make all channel commands tenant-aware
    # There is not middleware-like functionality for Action Cable,
    # so we need to patch it
    def dispatch_websocket_message(*)
      using_tenant { super }
    end

    # The same override for AnyCable, which uses a different method though
    def handle_channel_command(*)
      using_tenant { super }
    end
    
    private
    
    def using_current_tenant(&block)
      Apartment::Tenant.switch(&block)
    end
  end
end