Function GetQuerySpan

backend/plugin/parser/mongodb/query_span.go:18–42

GetQuerySpan returns a QuerySpan for a MongoDB shell statement. It classifies the statement type and populates PredicatePaths from predicate fields extracted by masking analysis.

(_ context.Context, _ base.GetQuerySpanContext, stmt base.Statement, _ string, _ string, _ bool)

// It classifies the statement type and populates PredicatePaths from
// predicate fields extracted by masking analysis.
func GetQuerySpan(_ context.Context, _ base.GetQuerySpanContext, stmt base.Statement, _ string, _ string, _ bool) (*base.QuerySpan, error) {
	queryType := GetQueryType(stmt.Text)
	span := &base.QuerySpan{
		Type: queryType,
	}

	analysis, err := AnalyzeMaskingStatement(stmt.Text)
	if err != nil {
		return span, err
	}
	if analysis == nil {
		return span, nil
	}

	span.MongoDBAnalysis = analysis

	if len(analysis.PredicateFields) > 0 {
		span.PredicatePaths = make(map[string]*base.PathAST, len(analysis.PredicateFields))
		for _, field := range analysis.PredicateFields {
			span.PredicatePaths[field] = dotPathToPathAST(field)
		}
	}

	return span, nil
}

// dotPathToPathAST converts a dot-delimited field path (e.g. "contact.phone")
// into a PathAST with ItemSelector nodes.