Method getPrimaryKeyInfosForTables

database/postgres/database.go:1354–1390

(tableNames []string)

}

func (d *PostgresDatabase) getPrimaryKeyInfosForTables(tableNames []string) (map[string]primaryKeyInfo, error) {
	var selectCols string
	if d.supportsConperiod() {
		selectCols = "con.conname, con.conperiod"
	} else {
		selectCols = "con.conname, false"
	}
	query := fmt.Sprintf(`
		SELECT nsp.nspname || '.' || cls.relname AS qualified_table_name, %s
		FROM pg_constraint con
		JOIN pg_class cls ON cls.oid = con.conrelid
		JOIN pg_namespace nsp ON nsp.oid = cls.relnamespace
		WHERE nsp.nspname || '.' || cls.relname = ANY($1::text[])
		AND con.contype = 'p'
	`, selectCols)

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

	result := make(map[string]primaryKeyInfo, len(tableNames))
	for rows.Next() {
		var tableName, keyName string
		var period bool
		err = rows.Scan(&tableName, &keyName, &period)
		if err != nil {
			return nil, err
		}
		result[tableName] = primaryKeyInfo{
			name:   NewIdentWithQuoteDetected(keyName),
			period: period,
		}
	}
	return result, nil
}

func (d *PostgresDatabase) getIndexDefsForTables(tableNames []string) (map[string][]string, error) {
	// Exclude indexes that are implicitly created for primary keys or unique constraints or exclusion constraints.