Function ApplyASTRewrites

pkg/traceql/ast_rewriter.go:46–62

ApplyASTRewrites applies all registered transformations except those whose names appear in skip. Passing [TransformationAll] in skip disables every transformation.

(r *RootExpr, skip []string)

// ApplyASTRewrites applies all registered transformations except those whose names appear in skip.
// Passing [TransformationAll] in skip disables every transformation.
func ApplyASTRewrites(r *RootExpr, skip []string) *RootExpr {
	if slices.Contains(skip, TransformationAll) {
		return r
	}

	var chain rewriteChain
	for _, nr := range defaultTransformations {
		if !slices.Contains(skip, nr.name) {
			chain = append(chain, nr.rewrite)
		}
	}
	if len(chain) == 0 {
		return r
	}

	return chain.RewriteRoot(r)
}

// ASTRewriter modifies a TraceQL AST without changing its meaning.
type ASTRewriter interface {