diff --git a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ComputeService.java b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ComputeService.java index 2cb7943231938..d72fe46169809 100644 --- a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ComputeService.java +++ b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/ComputeService.java @@ -65,6 +65,7 @@ import org.elasticsearch.xpack.esql.session.Configuration; import org.elasticsearch.xpack.esql.session.EsqlCCSUtils; import org.elasticsearch.xpack.esql.session.Result; +import org.elasticsearch.xpack.ml.MachineLearning; import java.util.ArrayList; import java.util.Collections; @@ -196,7 +197,8 @@ public void execute( TcpTransport.TRANSPORT_WORKER_THREAD_NAME_PREFIX, ThreadPool.Names.SYSTEM_READ, ThreadPool.Names.SEARCH, - ThreadPool.Names.SEARCH_COORDINATION + ThreadPool.Names.SEARCH_COORDINATION, + MachineLearning.NATIVE_INFERENCE_COMMS_THREAD_POOL_NAME ); Tuple, PhysicalPlan> subplansAndMainPlan = PlannerUtils.breakPlanIntoSubPlansAndMainPlan(physicalPlan); diff --git a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSender.java b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSender.java index 8d34b90c1ab5b..388594f21a215 100644 --- a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSender.java +++ b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/plugin/DataNodeRequestSender.java @@ -39,6 +39,7 @@ import org.elasticsearch.transport.TransportRequestOptions; import org.elasticsearch.transport.TransportService; import org.elasticsearch.xpack.esql.action.EsqlSearchShardsAction; +import org.elasticsearch.xpack.ml.MachineLearning; import java.util.ArrayList; import java.util.Collections; @@ -135,7 +136,8 @@ final void startComputeOnDataNodes(Set concreteIndices, Runnable runOnTa TcpTransport.TRANSPORT_WORKER_THREAD_NAME_PREFIX, ThreadPool.Names.SYSTEM_READ, ThreadPool.Names.SEARCH, - ThreadPool.Names.SEARCH_COORDINATION + ThreadPool.Names.SEARCH_COORDINATION, + MachineLearning.NATIVE_INFERENCE_COMMS_THREAD_POOL_NAME ); final long startTimeInNanos = System.nanoTime(); searchShards(concreteIndices, ActionListener.wrap(targetShards -> { diff --git a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/session/EsqlSession.java b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/session/EsqlSession.java index 1faf810cd6a90..a98b0f3c52735 100644 --- a/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/session/EsqlSession.java +++ b/x-pack/plugin/esql/src/main/java/org/elasticsearch/xpack/esql/session/EsqlSession.java @@ -76,6 +76,7 @@ import org.elasticsearch.xpack.esql.planner.premapper.PreMapper; import org.elasticsearch.xpack.esql.plugin.TransportActionServices; import org.elasticsearch.xpack.esql.telemetry.PlanTelemetry; +import org.elasticsearch.xpack.ml.MachineLearning; import java.util.ArrayList; import java.util.Collection; @@ -202,7 +203,8 @@ public void executeOptimizedPlan( TcpTransport.TRANSPORT_WORKER_THREAD_NAME_PREFIX, ThreadPool.Names.SYSTEM_READ, ThreadPool.Names.SEARCH, - ThreadPool.Names.SEARCH_COORDINATION + ThreadPool.Names.SEARCH_COORDINATION, + MachineLearning.NATIVE_INFERENCE_COMMS_THREAD_POOL_NAME ); if (explainMode) {// TODO: INLINESTATS come back to the explain mode branch and reevaluate PhysicalPlan physicalPlan = logicalPlanToPhysicalPlan(optimizedPlan, request);