Function scanRelations

internal/tools/sqlc-pg-gen/relation.go:53–102

(rows pgx.Rows)

}

func scanRelations(rows pgx.Rows) ([]Relation, error) {
	defer rows.Close()
	// Iterate through the result set
	var relations []Relation
	var prevRel *Relation

	for rows.Next() {
		var schemaName string
		var tableName string
		var columnName string
		var columnNotNull bool
		var columnType string
		var columnLength *int
		var columnIsArray bool
		err := rows.Scan(
			&schemaName,
			&tableName,
			&columnName,
			&columnNotNull,
			&columnType,
			&columnLength,
			&columnIsArray,
		)
		if err != nil {
			return nil, err
		}

		if prevRel == nil || tableName != prevRel.Name {
			// We are on the same table, just keep adding columns
			r := Relation{
				Catalog:    "pg_catalog",
				SchemaName: schemaName,
				Name:       tableName,
			}

			relations = append(relations, r)
			prevRel = &relations[len(relations)-1]
		}

		prevRel.Columns = append(prevRel.Columns, RelationColumn{
			Name:      columnName,
			Type:      columnType,
			IsNotNull: columnNotNull,
			IsArray:   columnIsArray,
			Length:    columnLength,
		})
	}

	return relations, rows.Err()
}

func readRelations(ctx context.Context, conn *pgx.Conn, schemaName string) ([]Relation, error) {
	rows, err := conn.Query(ctx, relationQuery, schemaName)