Method AddColumn

catalog/table.go:223–308

AddColumn implements sql.AlterableTable.

(ctx *sql.Context, column *sql.Column, order *sql.ColumnOrder)

func setPrimaryKeyColumns(schema sql.Schema, ordinals []int) {
	for _, idx := range ordinals {
		schema[idx].PrimaryKey = true
	}
}

// String implements sql.Table.
func (t *Table) String() string {
	return t.name
}

// PrimaryKeySchema implements sql.PrimaryKeyTable.
func (t *Table) PrimaryKeySchema() sql.PrimaryKeySchema {
	return t.schema
}

func getPrimaryKeyOrdinals(ctx *sql.Context, catalogName, dbName, tableName string) []int {
	rows, err := adapter.QueryCatalog(ctx, `
		SELECT constraint_column_indexes FROM duckdb_constraints() WHERE ((database_name = ? AND schema_name = ? AND table_name = ?) OR (database_name = 'temp' AND schema_name = 'main' AND table_name = ?)) AND constraint_type = 'PRIMARY KEY' LIMIT 1
	`, catalogName, dbName, tableName, tableName)
	if err != nil {
		panic(ErrDuckDB.New(err))
	}
	defer rows.Close()

	var ordinals []int
	if rows.Next() {
		var arr duckdb.Composite[[]int]
		if err := rows.Scan(&arr); err != nil {
			panic(ErrDuckDB.New(err))
		}
		ordinals = arr.Get()
	}
	if err := rows.Err(); err != nil {
		panic(ErrDuckDB.New(err))
	}
	return ordinals
}

func getCreateSequence(temporary bool, sequenceName string) (createStmt, fullName string) {
	if temporary {
		return `CREATE TEMP SEQUENCE "` + sequenceName + `"`, `temp.main."` + sequenceName + `"`
	}
	fullName = InternalSchemas.SYS.Schema + `."` + sequenceName + `"`
	return `CREATE SEQUENCE ` + fullName, fullName
}

// AddColumn implements sql.AlterableTable.
func (t *Table) AddColumn(ctx *sql.Context, column *sql.Column, order *sql.ColumnOrder) error {
	t.mu.Lock()
	defer t.mu.Unlock()

	// TODO: Column order is ignored as DuckDB does not support it.

	typ, err := DuckdbDataType(column.Type)
	if err != nil {
		return err
	}

	var sqls []string