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
41 lines
709 B
Go
41 lines
709 B
Go
package ast
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/sql/format"
|
|
)
|
|
|
|
type ColumnRef struct {
|
|
Tag NodeTag[ColumnRef] `json:"tag"`
|
|
|
|
Name string `json:"name"`
|
|
|
|
// From pg.ColumnRef
|
|
Fields *List `json:"fields,omitempty"`
|
|
Location int `json:"location"`
|
|
}
|
|
|
|
func (n *ColumnRef) Pos() int {
|
|
return n.Location
|
|
}
|
|
|
|
func (n *ColumnRef) Format(buf *TrackedBuffer, d format.Dialect) {
|
|
if n == nil {
|
|
return
|
|
}
|
|
|
|
if n.Fields != nil {
|
|
var items []string
|
|
for _, item := range n.Fields.Items {
|
|
switch nn := item.(type) {
|
|
case *String:
|
|
items = append(items, d.QuoteIdent(nn.Str))
|
|
case *A_Star:
|
|
items = append(items, "*")
|
|
}
|
|
}
|
|
buf.WriteString(strings.Join(items, "."))
|
|
}
|
|
}
|