class Multiwoven::Integrations::Source::AwsAthena::Client

def group_by_table(records)
  result = {}
  records.each_with_index do |entry, index|
    table_name = entry["table_name"]
    column_data = {
      column_name: entry["column_name"],
      data_type: entry["data_type"],
      is_nullable: entry["is_nullable"] == "YES"
    }
    result[index] ||= {}
    result[index][:tablename] = table_name
    result[index][:columns] = [column_data]
  end
  result
end