Function SplitSQL

backend/plugin/parser/plsql/split.go:20–50

SplitSQL splits the given SQL statement into multiple SQL statements. It uses omni's lexical Oracle splitter, which handles strings, comments, SQL*Plus separators, and PL/SQL blocks without requiring valid SQL.

(statement string)

// It uses omni's lexical Oracle splitter, which handles strings, comments,
// SQL*Plus separators, and PL/SQL blocks without requiring valid SQL.
func SplitSQL(statement string) ([]base.Statement, error) {
	segments := oracleparser.Split(statement)

	result := make([]base.Statement, 0, len(segments))
	positionMapper := base.NewByteOffsetPositionMapper(statement)
	for i := 0; i < len(segments); i++ {
		seg := segments[i]
		if seg.Kind == oracleparser.SegmentSQLPlusCommand {
			continue
		}
		byteStart := seg.ByteStart
		byteEnd := seg.ByteEnd
		if end, ok := matchRecognizeDefineMergeEnd(statement, segments, i); ok {
			byteEnd = end
			i += 2
		}
		byteEnd = byteStart + trimOracleSegmentTrailingHidden(statement[byteStart:byteEnd])
		text := statement[byteStart:byteEnd]
		result = append(result, base.Statement{
			Text:  text,
			Start: positionMapper.Position(byteStart),
			End:   positionMapper.Position(byteEnd),
			Empty: seg.Empty(),
			Range: &storepb.Range{
				Start: int32(byteStart),
				End:   int32(byteEnd),
			},
		})
	}
	return result, nil
}

func matchRecognizeDefineMergeEnd(statement string, segments []oracleparser.Segment, index int) (int, bool) {
	if index+2 >= len(segments) {