Function SplitSQL

backend/plugin/parser/cosmosdb/split.go:18–49

SplitSQL splits the given SQL statement into individual statements. CosmosDB only supports single SELECT statements.

(statement string)

// SplitSQL splits the given SQL statement into individual statements.
// CosmosDB only supports single SELECT statements.
func SplitSQL(statement string) ([]base.Statement, error) {
	if strings.TrimSpace(statement) == "" {
		return nil, nil
	}

	stmts, err := omnicosmosdb.Parse(statement)
	if err != nil {
		return nil, err
	}
	if len(stmts) == 0 {
		return nil, nil
	}

	var result []base.Statement
	positionMapper := base.NewByteOffsetPositionMapper(statement)
	for _, stmt := range stmts {
		startPos := positionMapper.Position(stmt.ByteStart)
		endPos := positionMapper.Position(stmt.ByteEnd)

		result = append(result, base.Statement{
			Text: stmt.Text,
			Range: &storepb.Range{
				Start: int32(stmt.ByteStart),
				End:   int32(stmt.ByteEnd),
			},
			Start: startPos,
			End:   endPos,
			Empty: stmt.Empty(),
		})
	}
	return result, nil
}