Function NewColumnTableDef

postgres/parser/sem/tree/create_table.go:334–423

NewColumnTableDef constructs a column definition for a CreateTable statement.

(
	name Name,
	typRef ResolvableTypeReference,
	compression string,
	collation string,
	qualifications []NamedColumnQualification,
)

// NewColumnTableDef constructs a column definition for a CreateTable statement.
func NewColumnTableDef(
	name Name,
	typRef ResolvableTypeReference,
	compression string,
	collation string,
	qualifications []NamedColumnQualification,
) (*ColumnTableDef, error) {
	var isSerial bool
	if typRef != nil {
		isSerial = IsReferenceSerialType(typRef)
	}
	d := &ColumnTableDef{
		Name:     name,
		Type:     typRef,
		IsSerial: isSerial,
	}
	d.Nullable.Nullability = SilentNull
	if collation != "" {
		_, err := language.Parse(collation)
		if err != nil {
			return nil, pgerror.Wrapf(err, pgcode.Syntax, "invalid locale %s", collation)
		}
		collatedTyp, err := processCollationOnType(name, d.Type, collation)
		if err != nil {
			// TODO: currently ignore as the test used dummy value // return nil, err
		} else {
			d.Type = collatedTyp
		}
	}
	for _, c := range qualifications {
		switch t := c.Qualification.(type) {
		case *ColumnDefault:
			if d.HasDefaultExpr() {
				return nil, pgerror.Newf(pgcode.Syntax,
					"multiple default values specified for column %q", name)
			}
			d.DefaultExpr.Expr = t.Expr
			d.DefaultExpr.ConstraintName = c.Name
		case NotNullConstraint:
			if d.Nullable.Nullability == Null {
				return nil, pgerror.Newf(pgcode.Syntax,
					"conflicting NULL/NOT NULL declarations for column %q", name)
			}
			d.Nullable.Nullability = NotNull
			d.Nullable.ConstraintName = c.Name
		case NullConstraint:
			if d.Nullable.Nullability == NotNull {
				return nil, pgerror.Newf(pgcode.Syntax,
					"conflicting NULL/NOT NULL declarations for column %q", name)
			}
			d.Nullable.Nullability = Null
			d.Nullable.ConstraintName = c.Name
		case UniqueConstraint:
			if t.IsPrimary {
				d.PrimaryKey.IsPrimaryKey = true
			} else {
				d.Unique = true
			}