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
50 lines
1007 B
Go
50 lines
1007 B
Go
package ast
|
|
|
|
import "github.com/sqlc-dev/sqlc/internal/sql/format"
|
|
|
|
type FuncParamMode int
|
|
|
|
const (
|
|
FuncParamIn FuncParamMode = iota
|
|
FuncParamOut
|
|
FuncParamInOut
|
|
FuncParamVariadic
|
|
FuncParamTable
|
|
FuncParamDefault
|
|
)
|
|
|
|
type FuncParam struct {
|
|
Tag NodeTag[FuncParam] `json:"tag"`
|
|
|
|
Name *string `json:"name,omitempty"`
|
|
Type *TypeName `json:"type,omitempty"`
|
|
DefExpr Node `json:"def_expr,omitempty"` // Will always be &ast.TODO
|
|
Mode FuncParamMode `json:"mode"`
|
|
}
|
|
|
|
func (n *FuncParam) Pos() int {
|
|
return 0
|
|
}
|
|
|
|
func (n *FuncParam) Format(buf *TrackedBuffer, d format.Dialect) {
|
|
if n == nil {
|
|
return
|
|
}
|
|
// Parameter mode prefix (OUT, INOUT, VARIADIC)
|
|
switch n.Mode {
|
|
case FuncParamOut:
|
|
buf.WriteString("OUT ")
|
|
case FuncParamInOut:
|
|
buf.WriteString("INOUT ")
|
|
case FuncParamVariadic:
|
|
buf.WriteString("VARIADIC ")
|
|
}
|
|
// Parameter name (if present)
|
|
if n.Name != nil {
|
|
buf.WriteString(*n.Name)
|
|
buf.WriteString(" ")
|
|
}
|
|
// Parameter type
|
|
buf.astFormat(n.Type, d)
|
|
}
|