Fix missing FoldContexts

Author: ivancea

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/planner/PlannerUtils.java

@@ -183,7 +183,7 @@ public static PhysicalPlan localPlan(
     /**
      * Extracts a filter that can be used to skip unmatched shards on the coordinator.
      */
-    public static QueryBuilder canMatchFilter(PhysicalPlan plan) {
+    public static QueryBuilder canMatchFilter(PhysicalPlan plan, FoldContext foldContext) {
         return detectFilter(plan, CoordinatorRewriteContext.SUPPORTED_FIELDS::contains, foldContext);
     }
 

x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/session/QueryBuilderResolver.java

@@ -16,6 +16,7 @@
 import org.elasticsearch.index.query.Rewriteable;
 import org.elasticsearch.search.SearchService;
 import org.elasticsearch.transport.TransportService;
+import org.elasticsearch.xpack.esql.core.expression.FoldContext;
 import org.elasticsearch.xpack.esql.core.util.Holder;
 import org.elasticsearch.xpack.esql.expression.function.fulltext.FullTextFunction;
 import org.elasticsearch.xpack.esql.plan.logical.EsRelation;
@@ -33,7 +34,7 @@
 /**
  * Some {@link FullTextFunction} implementations such as {@link org.elasticsearch.xpack.esql.expression.function.fulltext.Match}
  * will be translated to a {@link QueryBuilder} that require a rewrite phase on the coordinator.
- * {@link QueryBuilderResolver#resolveQueryBuilders(LogicalPlan, ActionListener, BiConsumer)} will rewrite the plan by replacing
+ * {@link QueryBuilderResolver#resolveQueryBuilders(LogicalPlan, FoldContext, ActionListener, BiConsumer)} will rewrite the plan by replacing
  * {@link FullTextFunction} expression with new ones that hold rewritten {@link QueryBuilder}s.
  */
 public class QueryBuilderResolver {
@@ -56,6 +57,7 @@ public QueryBuilderResolver(
 
     public void resolveQueryBuilders(
         LogicalPlan plan,
+        FoldContext foldContext,
         ActionListener<Result> listener,
         BiConsumer<LogicalPlan, ActionListener<Result>> callback
     ) {
@@ -72,7 +74,7 @@ public void resolveQueryBuilders(
             return;
         }
         QueryRewriteContext ctx = queryRewriteContext(indexNames);
-        FullTextFunctionsRewritable rewritable = new FullTextFunctionsRewritable(unresolved);
+        FullTextFunctionsRewritable rewritable = new FullTextFunctionsRewritable(unresolved, foldContext);
         Rewriteable.rewriteAndFetch(rewritable, ctx, new ActionListener<FullTextFunctionsRewritable>() {
             @Override
             public void onResponse(FullTextFunctionsRewritable fullTextFunctionsRewritable) {
@@ -136,11 +138,11 @@ private class FullTextFunctionsRewritable implements Rewriteable<FullTextFunctio
             this.queryBuilderMap = queryBuilderMap;
         }
 
-        FullTextFunctionsRewritable(Set<FullTextFunction> functions) {
+        FullTextFunctionsRewritable(Set<FullTextFunction> functions, FoldContext foldContext) {
             this.queryBuilderMap = new HashMap<>();
 
             for (FullTextFunction func : functions) {
-                queryBuilderMap.put(func, TRANSLATOR_HANDLER.asQuery(func).asBuilder());
+                queryBuilderMap.put(func, TRANSLATOR_HANDLER.asQuery(func, foldContext).asBuilder());
             }
         }