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
34 lines
709 B
Go
34 lines
709 B
Go
package compiler
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/sql/ast"
|
|
"github.com/sqlc-dev/sqlc/internal/sql/astutils"
|
|
)
|
|
|
|
func arrayDims(n *ast.TypeName) int {
|
|
if n == nil || n.ArrayBounds == nil {
|
|
return 0
|
|
}
|
|
return len(n.ArrayBounds.Items)
|
|
}
|
|
|
|
func toColumn(n *ast.TypeName) *Column {
|
|
if n == nil {
|
|
panic("can't build column for nil type name")
|
|
}
|
|
typ, err := ParseTypeName(n)
|
|
if err != nil {
|
|
panic("toColumn: " + err.Error())
|
|
}
|
|
arrayDims := arrayDims(n)
|
|
return &Column{
|
|
Type: typ,
|
|
DataType: strings.TrimPrefix(astutils.Join(n.Names, "."), "."),
|
|
NotNull: true, // XXX: How do we know if this should be null?
|
|
IsArray: arrayDims > 0,
|
|
ArrayDims: arrayDims,
|
|
}
|
|
}
|