class Gladys::Container

def query(table = nil, &block)
  message = if table.nil?
              "Calling :query"
            else
              "Calling :query with table: #{table}"
            end
  # Do not remove the handle_metrics call. It is VERY important.
  handle_metrics(message) do
    results = if block.nil?
                load_db(table)
              else
                load_db(table).instance_eval(&block)
              end
    metrics&.increment_qps
    results
  end
end