module ActionCable::Connection::InternalChannel

def internal_channel
  "action_cable/#{connection_identifier}"
end

def process_internal_message(message)
  case message["type"]
  when "disconnect"
    logger.info "Removing connection (#{connection_identifier})"
    websocket.close
  end
rescue Exception => e
  logger.error "There was an exception - #{e.class}(#{e.message})"
  logger.error e.backtrace.join("\n")
  close
end

def subscribe_to_internal_channel
  if connection_identifier.present?
    callback = -> (message) { process_internal_message decode(message) }
    @_internal_subscriptions ||= []
    @_internal_subscriptions << [ internal_channel, callback ]
    server.event_loop.post { pubsub.subscribe(internal_channel, callback) }
    logger.info "Registered connection (#{connection_identifier})"
  end
end

def unsubscribe_from_internal_channel
  if @_internal_subscriptions.present?
    @_internal_subscriptions.each { |channel, callback| server.event_loop.post { pubsub.unsubscribe(channel, callback) } }
  end
end