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
33 lines
650 B
Go
33 lines
650 B
Go
package ast
|
|
|
|
import "github.com/sqlc-dev/sqlc/internal/sql/format"
|
|
|
|
type DoStmt struct {
|
|
Tag NodeTag[DoStmt] `json:"tag"`
|
|
|
|
Args *List `json:"args,omitempty"`
|
|
}
|
|
|
|
func (n *DoStmt) Pos() int {
|
|
return 0
|
|
}
|
|
|
|
func (n *DoStmt) Format(buf *TrackedBuffer, d format.Dialect) {
|
|
if n == nil {
|
|
return
|
|
}
|
|
buf.WriteString("DO ")
|
|
// Find the "as" argument which contains the body
|
|
if items(n.Args) {
|
|
for _, arg := range n.Args.Items {
|
|
if de, ok := arg.(*DefElem); ok && de.Defname != nil && *de.Defname == "as" {
|
|
if s, ok := de.Arg.(*String); ok {
|
|
buf.WriteString("$$")
|
|
buf.WriteString(s.Str)
|
|
buf.WriteString("$$")
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|