Method Format

pkg/sql/sem/tree/create_routine.go:96–166

Format implements the NodeFormatter interface.

(ctx *FmtCtx)

// Format implements the NodeFormatter interface.
func (node *CreateRoutine) Format(ctx *FmtCtx) {
	ctx.WriteString("CREATE ")
	if node.Replace {
		ctx.WriteString("OR REPLACE ")
	}
	if node.IsProcedure {
		ctx.WriteString("PROCEDURE ")
	} else {
		ctx.WriteString("FUNCTION ")
	}
	ctx.FormatNode(&node.Name)
	ctx.WriteByte('(')
	ctx.FormatNode(node.Params)
	ctx.WriteString(")\n\t")
	if !node.IsProcedure && node.ReturnType != nil {
		ctx.WriteString("RETURNS ")
		if node.ReturnType.SetOf {
			ctx.WriteString("SETOF ")
		}
		ctx.FormatTypeReference(node.ReturnType.Type)
		ctx.WriteString("\n\t")
	}
	var isPLpgSQL bool
	var funcBody RoutineBodyStr
	for _, option := range node.Options {
		switch t := option.(type) {
		case RoutineBodyStr:
			funcBody = t
			continue
		case RoutineLeakproof, RoutineVolatility, RoutineNullInputBehavior:
			if node.IsProcedure {
				continue
			}
		case RoutineLanguage:
			isPLpgSQL = t == RoutineLangPLpgSQL
		}
		ctx.FormatNode(option)
		ctx.WriteString("\n\t")
	}

	if ctx.HasFlags(FmtMarkRedactionNode) {
		ctx.WriteString("AS ")
		ctx.WriteString("$$")
		for i, stmt := range node.BodyStatements {
			if i > 0 {
				ctx.WriteByte(' ')
			}
			ctx.WithAnnotations(node.BodyAnnotations[i], func() {
				ctx.FormatNode(stmt)
				if !isPLpgSQL {
					// PL/pgSQL statements handle printing semicolons themselves.
					ctx.WriteByte(';')
				}
			})
		}
		ctx.WriteString("$$")
	} else if node.RoutineBody != nil {
		ctx.WriteString("BEGIN ATOMIC ")