Function nodeAlterTableComputed

server/ast/alter_table.go:447–494

nodeAlterTableComputed converts a tree.AlterTableComputed instance into an equivalent CREATE/ALTER SEQUENCE statement.

(ctx *Context, tableName tree.TableName, node *tree.AlterTableComputed)

// nodeAlterTableComputed converts a tree.AlterTableComputed instance into an equivalent CREATE/ALTER SEQUENCE statement.
func nodeAlterTableComputed(ctx *Context, tableName tree.TableName, node *tree.AlterTableComputed) (vitess.Statement, error) {
	if len(node.Defs) > 0 {
		return NotYetSupportedError("ALTER TABLE variant is not yet supported")
	}
	var persistence tree.Persistence
	var seqName tree.TableName
	var trimmedOptions []tree.SequenceOption
	switch def := node.AddDefs.(type) {
	case *tree.ColumnComputedDef:
		if def.Expr != nil {
			return NotYetSupportedError("ALTER TABLE variant is not yet supported")
		}
		trimmedOptions = make([]tree.SequenceOption, 0, len(def.Options)+2)
		for _, opt := range def.Options {
			switch opt.Name {
			case tree.SeqOptOwnedBy:
				return NotYetSupportedError("OWNED BY is invalid here")
			case tree.SeqOptRestart:
				return NotYetSupportedError("RESTART is not yet supported")
			case tree.SeqOptName:
				seqName = opt.SeqName
			case tree.SeqOptLogged:
				persistence = tree.PersistencePermanent
			case tree.SeqOptUnlogged:
				persistence = tree.PersistenceUnlogged
			default:
				trimmedOptions = append(trimmedOptions, opt)
			}
		}
	default:
		return NotYetSupportedError("ALTER TABLE variant is not yet supported")
	}
	trimmedOptions = append(trimmedOptions,
		tree.SequenceOption{
			Name:          tree.SeqOptOwnedBy,
			ColumnItemVal: tree.NewColumnItem(&tableName, node.Column),
		},
		tree.SequenceOption{
			Name: tree.SeqOptViaAlterTable,
		},
	)
	return nodeCreateSequence(ctx, &tree.CreateSequence{
		IfNotExists: false,
		Name:        seqName,
		Persistence: persistence,
		Options:     trimmedOptions,
	})
}