Method getPrimaryKeyColumnsForTables

database/postgres/database.go:1324–1352

(tableNames []string)

}

func (d *PostgresDatabase) getPrimaryKeyColumnsForTables(tableNames []string) (map[string][]string, error) {
	const query = `SELECT
	tc.table_schema || '.' || tc.table_name AS qualified_table_name,
	kcu.column_name
FROM
	information_schema.table_constraints AS tc
	JOIN information_schema.key_column_usage AS kcu
		USING (table_schema, table_name, constraint_name)
WHERE constraint_type = 'PRIMARY KEY'
AND tc.table_schema || '.' || tc.table_name = ANY($1::text[])
ORDER BY tc.table_schema, tc.table_name, kcu.ordinal_position`

	rows, err := d.db.Query(query, pq.Array(tableNames))
	if err != nil {
		return nil, err
	}
	defer rows.Close()

	result := make(map[string][]string, len(tableNames))
	for rows.Next() {
		var tableName, columnName string
		err = rows.Scan(&tableName, &columnName)
		if err != nil {
			return nil, err
		}
		result[tableName] = append(result[tableName], columnName)
	}
	return result, nil
}

func (d *PostgresDatabase) getPrimaryKeyInfosForTables(tableNames []string) (map[string]primaryKeyInfo, error) {
	var selectCols string