diff --git a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/action/EsqlExecutionInfo.java b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/action/EsqlExecutionInfo.java index 05abd4b107f6d..96aa0a398a594 100644 --- a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/action/EsqlExecutionInfo.java +++ b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/action/EsqlExecutionInfo.java @@ -275,6 +275,19 @@ public Map getClusters() { return clusterInfo; } + /** + * This creates an initial Cluster object with indexExpression and skipUnavailable. + */ + public void initCluster(String clusterAlias, String indexExpression) { + swapCluster(clusterAlias, (ca, previous) -> { + var expr = indexExpression; + if (previous != null) { + expr = previous.getIndexExpression() + "," + indexExpression; + } + return new Cluster(clusterAlias, expr, shouldSkipOnFailure(clusterAlias)); + }); + } + /** * Utility to swap a Cluster object. Guidelines for the remapping function: *