class Restforce::Concerns::Streaming::ReplayExtension

def replay_id(channel)
  handler = @replay_handlers[channel]
  if handler.is_a?(Integer)
    handler # treat it as a scalar
  elsif handler.respond_to?(:[])
    # Ask for the latest replayId for this channel
    handler[channel]
  else
    # Just pass it along
    handler
  end
end