module WolfCore::NoSqlDbDataSource

def query(table_name:, key_condition_expression:, expression_attribute_values:, index_name: nil, limit: nil)
  query_params = {
    table_name: table_name,
    key_condition_expression: key_condition_expression,
    expression_attribute_values: expression_attribute_values
  }
  query_params[:index_name] = index_name if index_name
  query_params[:limit] = limit if limit
  instance.query(query_params).items
end