in internal/database/postgres/postgres.go [150:178]
func (h postgresHandler) ListColumns(db *database.DB, tableName string) ([]database.ColumnInfo, error) {
query := `
SELECT column_name, data_type
FROM information_schema.columns
WHERE table_schema = 'public'
AND table_name = $1
ORDER BY ordinal_position;`
rows, err := db.Query(query, tableName)
if err != nil {
return nil, fmt.Errorf("error querying columns for table %s: %w", tableName, err)
}
defer rows.Close()
var columns []database.ColumnInfo
for rows.Next() {
var colInfo database.ColumnInfo
if err := rows.Scan(&colInfo.Name, &colInfo.DataType); err != nil {
return nil, fmt.Errorf("error scanning column name and data type: %w", err)
}
columns = append(columns, colInfo)
}
if err := rows.Err(); err != nil {
return nil, fmt.Errorf("error iterating column rows: %w", err)
}
return columns, nil
}