Function toBaseStatements

backend/plugin/parser/tsql/split.go:53–87

toBaseStatements converts omni's statement list into []base.Statement while enforcing byte-contiguity and position adjacency. Any bytes that omni did not attribute to a statement (gaps) are absorbed into the following statement so the output remains contiguous.

(sql string, omniStmts []omnimssql.Statement)

// not attribute to a statement (gaps) are absorbed into the following
// statement so the output remains contiguous.
func toBaseStatements(sql string, omniStmts []omnimssql.Statement) []base.Statement {
	if len(omniStmts) == 0 {
		return nil
	}
	result := make([]base.Statement, 0, len(omniStmts))
	positionMapper := base.NewByteOffsetPositionMapper(sql)
	prevEnd := &storepb.Position{Line: 1, Column: 1}
	prevByte := 0
	for _, os := range omniStmts {
		startByte := prevByte
		endByte := os.ByteEnd
		if endByte < startByte {
			endByte = startByte
		}
		text := sql[startByte:endByte]
		endPos := positionMapper.Position(endByte)

		_, isGo := os.AST.(*ast.GoStmt)
		empty := isGo || os.Empty() || strings.TrimSpace(text) == ""

		result = append(result, base.Statement{
			Text: text,
			Range: &storepb.Range{
				Start: int32(startByte),
				End:   int32(endByte),
			},
			Start: prevEnd,
			End:   endPos,
			Empty: empty,
		})
		prevEnd = endPos
		prevByte = endByte
	}
	return result
}