Method readTripleQuotedString

pkg/sql/tokenizer/tokenizer.go:1076–1133

readTripleQuotedString reads a triple-quoted string (e.g. "'abc"' or """abc""")

(quote rune)

// readTripleQuotedString reads a triple-quoted string (e.g. "'abc"' or """abc""")
func (t *Tokenizer) readTripleQuotedString(quote rune) (models.Token, error) {
	// Store start position for error reporting
	startPos := t.pos

	// Skip opening triple quotes
	for i := 0; i < 3; i++ {
		r, size := utf8.DecodeRune(t.input[t.pos.Index:])
		t.pos.AdvanceRune(r, size)
	}

	var buf bytes.Buffer
	for t.pos.Index < len(t.input) {
		// Check for closing triple quotes
		if t.pos.Index+2 < len(t.input) {
			r1, s1 := utf8.DecodeRune(t.input[t.pos.Index:])
			r2, s2 := utf8.DecodeRune(t.input[t.pos.Index+s1:])
			r3, s3 := utf8.DecodeRune(t.input[t.pos.Index+s1+s2:])
			if r1 == quote && r2 == quote && r3 == quote {
				// Skip closing quotes
				t.pos.Index += s1 + s2 + s3
				t.pos.Column += 3

				value := buf.String()
				if quote == '\'' {
					return models.Token{
						Type:  models.TokenTypeTripleSingleQuotedString,
						Value: value,
						Quote: quote,
					}, nil
				}
				return models.Token{
					Type:  models.TokenTypeTripleDoubleQuotedString,
					Value: value,
					Quote: quote,
				}, nil
			}
		}

		// Handle regular characters
		r, size := utf8.DecodeRune(t.input[t.pos.Index:])
		if r == '\n' {
			buf.WriteRune(r)
			t.pos.Index += size
			t.pos.Line++
			t.pos.Column = 1
			continue
		}

		buf.WriteRune(r)
		t.pos.Index += size
		t.pos.Column++
	}

	return models.Token{}, errors.UnterminatedStringError(
		t.toSQLPosition(startPos),
		string(t.input),
	)
}