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
117 lines
3.1 KiB
Go
117 lines
3.1 KiB
Go
package ast
|
|
|
|
import "github.com/sqlc-dev/sqlc/internal/sql/format"
|
|
|
|
type WindowDef struct {
|
|
Tag NodeTag[WindowDef] `json:"tag"`
|
|
|
|
Name *string `json:"name,omitempty"`
|
|
Refname *string `json:"refname,omitempty"`
|
|
PartitionClause *List `json:"partition_clause,omitempty"`
|
|
OrderClause *List `json:"order_clause,omitempty"`
|
|
FrameOptions int `json:"frame_options"`
|
|
StartOffset Node `json:"start_offset,omitempty"`
|
|
EndOffset Node `json:"end_offset,omitempty"`
|
|
Location int `json:"location"`
|
|
}
|
|
|
|
func (n *WindowDef) Pos() int {
|
|
return n.Location
|
|
}
|
|
|
|
// Frame option constants (from PostgreSQL's parsenodes.h)
|
|
const (
|
|
FrameOptionNonDefault = 0x00001
|
|
FrameOptionRange = 0x00002
|
|
FrameOptionRows = 0x00004
|
|
FrameOptionGroups = 0x00008
|
|
FrameOptionBetween = 0x00010
|
|
FrameOptionStartUnboundedPreceding = 0x00020
|
|
FrameOptionEndUnboundedPreceding = 0x00040
|
|
FrameOptionStartUnboundedFollowing = 0x00080
|
|
FrameOptionEndUnboundedFollowing = 0x00100
|
|
FrameOptionStartCurrentRow = 0x00200
|
|
FrameOptionEndCurrentRow = 0x00400
|
|
FrameOptionStartOffset = 0x00800
|
|
FrameOptionEndOffset = 0x01000
|
|
FrameOptionExcludeCurrentRow = 0x02000
|
|
FrameOptionExcludeGroup = 0x04000
|
|
FrameOptionExcludeTies = 0x08000
|
|
)
|
|
|
|
func (n *WindowDef) Format(buf *TrackedBuffer, d format.Dialect) {
|
|
if n == nil {
|
|
return
|
|
}
|
|
|
|
// Named window reference
|
|
if n.Refname != nil && *n.Refname != "" {
|
|
buf.WriteString(*n.Refname)
|
|
return
|
|
}
|
|
|
|
buf.WriteString("(")
|
|
needSpace := false
|
|
|
|
if items(n.PartitionClause) {
|
|
buf.WriteString("PARTITION BY ")
|
|
buf.join(n.PartitionClause, d, ", ")
|
|
needSpace = true
|
|
}
|
|
|
|
if items(n.OrderClause) {
|
|
if needSpace {
|
|
buf.WriteString(" ")
|
|
}
|
|
buf.WriteString("ORDER BY ")
|
|
buf.join(n.OrderClause, d, ", ")
|
|
needSpace = true
|
|
}
|
|
|
|
// Frame clause
|
|
if n.FrameOptions&FrameOptionNonDefault != 0 {
|
|
if needSpace {
|
|
buf.WriteString(" ")
|
|
}
|
|
|
|
// Frame type
|
|
if n.FrameOptions&FrameOptionRows != 0 {
|
|
buf.WriteString("ROWS ")
|
|
} else if n.FrameOptions&FrameOptionRange != 0 {
|
|
buf.WriteString("RANGE ")
|
|
} else if n.FrameOptions&FrameOptionGroups != 0 {
|
|
buf.WriteString("GROUPS ")
|
|
}
|
|
|
|
if n.FrameOptions&FrameOptionBetween != 0 {
|
|
buf.WriteString("BETWEEN ")
|
|
}
|
|
|
|
// Start bound
|
|
if n.FrameOptions&FrameOptionStartUnboundedPreceding != 0 {
|
|
buf.WriteString("UNBOUNDED PRECEDING")
|
|
} else if n.FrameOptions&FrameOptionStartCurrentRow != 0 {
|
|
buf.WriteString("CURRENT ROW")
|
|
} else if n.FrameOptions&FrameOptionStartOffset != 0 {
|
|
buf.astFormat(n.StartOffset, d)
|
|
buf.WriteString(" PRECEDING")
|
|
}
|
|
|
|
if n.FrameOptions&FrameOptionBetween != 0 {
|
|
buf.WriteString(" AND ")
|
|
|
|
// End bound
|
|
if n.FrameOptions&FrameOptionEndUnboundedFollowing != 0 {
|
|
buf.WriteString("UNBOUNDED FOLLOWING")
|
|
} else if n.FrameOptions&FrameOptionEndCurrentRow != 0 {
|
|
buf.WriteString("CURRENT ROW")
|
|
} else if n.FrameOptions&FrameOptionEndOffset != 0 {
|
|
buf.astFormat(n.EndOffset, d)
|
|
buf.WriteString(" FOLLOWING")
|
|
}
|
|
}
|
|
}
|
|
|
|
buf.WriteString(")")
|
|
}
|