Method ModifyColumn

catalog/table.go:349–479

ModifyColumn implements sql.AlterableTable.

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

	}
	// Update the PK ordinals only after the column is successfully added.
	if column.PrimaryKey {
		t.hasPrimaryKey = true
		t.comment.Meta.PkOrdinals = tableInfo.PkOrdinals
	}
	return t.withSchema(ctx)
}

// DropColumn implements sql.AlterableTable.
func (t *Table) DropColumn(ctx *sql.Context, columnName string) error {
	t.mu.Lock()
	defer t.mu.Unlock()

	// Check if the column is AUTO_INCREMENT
	autoIncrement := false
	for _, column := range t.schema.Schema {
		if column.AutoIncrement && strings.EqualFold(column.Name, columnName) {
			autoIncrement = true
			break
		}
	}

	sql := `ALTER TABLE ` + FullTableName(t.db.catalog, t.db.name, t.name) + ` DROP COLUMN ` + QuoteIdentifierANSI(columnName)

	if autoIncrement {
		// Drop the sequence
		sql += `; DROP SEQUENCE IF EXISTS ` + t.comment.Meta.Sequence
		// Remove the sequence name from the table comment
		extraInfo := t.comment.Meta
		extraInfo.Sequence = ""
		comment := NewCommentWithMeta(t.comment.Text, extraInfo)
		sql += `; COMMENT ON TABLE ` + FullTableName(t.db.catalog, t.db.name, t.name) + ` IS '` + comment.Encode() + `'`
	}

	_, err := adapter.Exec(ctx, sql)
	if err != nil {
		return ErrDuckDB.New(err)
	}

	// Update the sequence name only after the column is successfully dropped.
	if autoIncrement {
		t.comment.Meta.Sequence = ""
	}
	return t.withSchema(ctx)
}

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

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

	// Find existing column to check for AUTO_INCREMENT and PRIMARY KEY
	var oldColumn *sql.Column
	var oldColumnIndex int