class ActiveRecord::Middleware::ShardSelector

def call(env)
  request = ActionDispatch::Request.new(env)
  shard = selected_shard(request)
  set_shard(shard) do
    @app.call(env)
  end
end