class ActionCable::Server::Worker

def work(connection, &block)
  self.connection = connection
  run_callbacks :work, &block
ensure
  self.connection = nil
end