0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
73 lines
2.5 KiB
Go
73 lines
2.5 KiB
Go
package ast
|
|
|
|
import "github.com/sqlc-dev/sqlc/internal/sql/format"
|
|
|
|
type ColumnDef struct {
|
|
Tag NodeTag[ColumnDef] `json:"tag"`
|
|
|
|
Colname string `json:"colname"`
|
|
// TypeName is the column's type for the catalog. When Typeless is set
|
|
// the author wrote no type at all — SQLite allows it — and the column
|
|
// prints without one, whatever TypeName carries.
|
|
TypeName *TypeName `json:"type_name,omitempty"`
|
|
Typeless bool `json:"typeless"`
|
|
IsNotNull bool `json:"is_not_null"`
|
|
IsUnsigned bool `json:"is_unsigned"`
|
|
IsArray bool `json:"is_array"`
|
|
ArrayDims int `json:"array_dims"`
|
|
Vals *List `json:"vals,omitempty"`
|
|
Length *int `json:"length,omitempty"`
|
|
PrimaryKey bool `json:"primary_key"`
|
|
// IsHidden marks a column a relation offers by name without listing it,
|
|
// like the column an sqlite fts5 table names after itself. The legacy
|
|
// catalog drops hidden columns; the core catalog keeps them out of star
|
|
// expansions and models.
|
|
IsHidden bool `json:"is_hidden"`
|
|
|
|
// From pg.ColumnDef
|
|
Inhcount int `json:"inhcount"`
|
|
IsLocal bool `json:"is_local"`
|
|
IsFromType bool `json:"is_from_type"`
|
|
IsFromParent bool `json:"is_from_parent"`
|
|
Storage byte `json:"storage"`
|
|
RawDefault Node `json:"raw_default,omitempty"`
|
|
CookedDefault Node `json:"cooked_default,omitempty"`
|
|
Identity byte `json:"identity"`
|
|
CollClause *CollateClause `json:"coll_clause,omitempty"`
|
|
CollOid Oid `json:"coll_oid"`
|
|
Constraints *List `json:"constraints,omitempty"`
|
|
Fdwoptions *List `json:"fdwoptions,omitempty"`
|
|
Location int `json:"location"`
|
|
Comment string `json:"comment"`
|
|
}
|
|
|
|
func (n *ColumnDef) Pos() int {
|
|
return n.Location
|
|
}
|
|
|
|
func (n *ColumnDef) Format(buf *TrackedBuffer, d format.Dialect) {
|
|
if n == nil {
|
|
return
|
|
}
|
|
buf.WriteString(n.Colname)
|
|
if !n.Typeless {
|
|
buf.WriteString(" ")
|
|
buf.astFormat(n.TypeName, d)
|
|
// MySQL integer types: signedness is part of the type.
|
|
if n.IsUnsigned {
|
|
buf.WriteString(" unsigned")
|
|
}
|
|
}
|
|
// Use IsArray from ColumnDef since TypeName.ArrayBounds may not be set
|
|
// (for type resolution compatibility)
|
|
if n.IsArray && !items(n.TypeName.ArrayBounds) {
|
|
buf.WriteString("[]")
|
|
}
|
|
if n.PrimaryKey {
|
|
buf.WriteString(" PRIMARY KEY")
|
|
} else if n.IsNotNull {
|
|
buf.WriteString(" NOT NULL")
|
|
}
|
|
buf.astFormat(n.Constraints, d)
|
|
}
|