Function FetchTableColumns

src/go/plugin/go.d/pkg/sqlquery/columns.go:38–64

FetchTableColumns returns the set of column names for schema.table. If transform is non-nil, it is applied to each column name before insertion.

(ctx context.Context, q Queryer, schema, table string, style PlaceholderStyle, transform func(string) string)

// FetchTableColumns returns the set of column names for schema.table.
// If transform is non-nil, it is applied to each column name before insertion.
func FetchTableColumns(ctx context.Context, q Queryer, schema, table string, style PlaceholderStyle, transform func(string) string) (map[string]bool, error) {
	query, err := tableColumnsQuery(style)
	if err != nil {
		return nil, err
	}
	rows, err := q.QueryContext(ctx, query, schema, table)
	if err != nil {
		return nil, err
	}
	defer func() { _ = rows.Close() }()

	cols := make(map[string]bool)
	for rows.Next() {
		var name string
		if err := rows.Scan(&name); err != nil {
			return nil, err
		}
		if transform != nil {
			name = transform(name)
		}
		cols[name] = true
	}
	if err := rows.Err(); err != nil {
		return nil, err
	}
	return cols, nil
}