Function splitStatement

backend/plugin/db/util/statement_parser.go:114–192

(sql string)

}

func splitStatement(sql string) ([]string, error) {
	var stmts []string
	const singleQuote = '\''
	const doubleQuote = '"'
	const backtick = '`'
	const delimiter = ';'
	isInQuoted := false
	var startQuote rune
	lastCharWasEscapeChar := false
	isTripleQuoted := false
	res := strings.Builder{}
	res.Grow(len(sql))
	index := 0
	runes := []rune(sql)
	for index < len(runes) {
		c := runes[index]
		if isInQuoted {
			if (c == '\n' || c == '\r') && !isTripleQuoted {
				return nil, spanner.ToSpannerError(connect.NewError(connect.CodeInvalidArgument, errors.Errorf("statement contains an unclosed literal: %s", sql)))
			} else if c == startQuote {
				if lastCharWasEscapeChar {
					lastCharWasEscapeChar = false
				} else if isTripleQuoted {
					if len(runes) > index+2 && runes[index+1] == startQuote && runes[index+2] == startQuote {
						isInQuoted = false
						startQuote = 0
						isTripleQuoted = false
						_, _ = res.WriteRune(c)
						_, _ = res.WriteRune(c)
						index += 2
					}
				} else {
					isInQuoted = false
					startQuote = 0
				}
			} else if c == '\\' {
				lastCharWasEscapeChar = true
			} else {
				lastCharWasEscapeChar = false
			}
			_, _ = res.WriteRune(c)
		} else {
			// We are not in a quoted string.
			switch c {
			case singleQuote, doubleQuote, backtick:
				isInQuoted = true
				startQuote = c
				// Check whether it is a triple-quote.
				if len(runes) > index+2 && runes[index+1] == startQuote && runes[index+2] == startQuote {
					isTripleQuoted = true
					_, _ = res.WriteRune(c)
					_, _ = res.WriteRune(c)
					index += 2
				}
				_, _ = res.WriteRune(c)
			case delimiter:
				stmt := strings.Trim(res.String(), " \n\t")
				if stmt != "" {