Function SplitSQL

backend/plugin/parser/pg/split.go:18–36

SplitSQL splits the given SQL statement into multiple SQL statements. It uses omni's lexical splitter which handles quotes, dollar-quotes, comments, and BEGIN ATOMIC blocks without requiring valid SQL.

(statement string)

// It uses omni's lexical splitter which handles quotes, dollar-quotes,
// comments, and BEGIN ATOMIC blocks without requiring valid SQL.
func SplitSQL(statement string) ([]base.Statement, error) {
	segments := omnipg.Split(statement)

	result := make([]base.Statement, 0, len(segments))
	positionMapper := base.NewByteOffsetPositionMapper(statement)
	for _, seg := range segments {
		result = append(result, base.Statement{
			Text:  seg.Text,
			Empty: seg.Empty(),
			Start: positionMapper.Position(seg.ByteStart),
			End:   positionMapper.Position(seg.ByteEnd),
			Range: &storepb.Range{
				Start: int32(seg.ByteStart),
				End:   int32(seg.ByteEnd),
			},
		})
	}
	return result, nil
}