Function AnalyzeDoBlockOptions

pkg/sql/sem/tree/do.go:48–85

AnalyzeDoBlockOptions checks the list of DO block options for validity, and returns the code string.

(options DoBlockOptions)

// AnalyzeDoBlockOptions checks the list of DO block options for validity, and
// returns the code string.
func AnalyzeDoBlockOptions(options DoBlockOptions) (RoutineBodyStr, error) {
	var doBody RoutineBodyStr
	var lang RoutineLanguage
	var foundLanguage, foundBody bool
	for _, doOption := range options {
		switch t := doOption.(type) {
		case RoutineBodyStr:
			if foundBody {
				return "", pgerror.New(pgcode.Syntax, "conflicting or redundant options")
			}
			doBody = t
			foundBody = true
		case RoutineLanguage:
			if foundLanguage {
				return "", pgerror.New(pgcode.Syntax, "conflicting or redundant options")
			}
			lang = t
			foundLanguage = true
		default:
			return "", errors.AssertionFailedf("unexpected DO statement option: %v", t)
		}
	}
	if !foundBody {
		return "", pgerror.New(pgcode.Syntax, "no inline code specified")
	}
	switch lang {
	case "", RoutineLangPLpgSQL:
		// PL/pgSQL is the default, and currently the only supported language.
	case RoutineLangSQL:
		// Postgres disallows SQL for DO statements.
		return "", pgerror.Newf(pgcode.FeatureNotSupported,
			"language \"%s\" does not support inline code execution", lang,
		)
	default:
		return "", pgerror.Newf(pgcode.UndefinedObject, "language \"%s\" does not exist", lang)
	}
	return doBody, nil
}