Track ongoing search tasks (elastic#107129)

jimczi · Apr 8, 2024 · 073048a · 073048a
1 parent ab52ef1
commit 073048a
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/docs/changelog/107129.yaml b/docs/changelog/107129.yaml
@@ -0,0 +1,5 @@
+pr: 107129
+summary: Track ongoing search tasks
+area: Search
+type: enhancement
+issues: []
diff --git a/server/src/main/java/org/elasticsearch/threadpool/ThreadPool.java b/server/src/main/java/org/elasticsearch/threadpool/ThreadPool.java
@@ -151,6 +151,8 @@ public static ThreadPoolType fromType(String type) {
         entry(Names.SYSTEM_CRITICAL_WRITE, ThreadPoolType.FIXED)
     );
 
+    public static final double searchAutoscalingEWMA = 0.1;
+
     private final Map<String, ExecutorHolder> executors;
 
     private final ThreadPoolInfo threadPoolInfo;
@@ -222,15 +224,27 @@ public ThreadPool(final Settings settings, MeterRegistry meterRegistry, final Ex
         builders.put(Names.ANALYZE, new FixedExecutorBuilder(settings, Names.ANALYZE, 1, 16, TaskTrackingConfig.DO_NOT_TRACK));
         builders.put(
             Names.SEARCH,
-            new FixedExecutorBuilder(settings, Names.SEARCH, searchOrGetThreadPoolSize, 1000, TaskTrackingConfig.DEFAULT)
+            new FixedExecutorBuilder(
+                settings,
+                Names.SEARCH,
+                searchOrGetThreadPoolSize,
+                1000,
+                new TaskTrackingConfig(true, searchAutoscalingEWMA)
+            )
         );
         builders.put(
             Names.SEARCH_WORKER,
             new FixedExecutorBuilder(settings, Names.SEARCH_WORKER, searchOrGetThreadPoolSize, -1, TaskTrackingConfig.DEFAULT)
         );
         builders.put(
             Names.SEARCH_COORDINATION,
-            new FixedExecutorBuilder(settings, Names.SEARCH_COORDINATION, halfProc, 1000, TaskTrackingConfig.DEFAULT)
+            new FixedExecutorBuilder(
+                settings,
+                Names.SEARCH_COORDINATION,
+                halfProc,
+                1000,
+                new TaskTrackingConfig(true, searchAutoscalingEWMA)
+            )
         );
         builders.put(
             Names.AUTO_COMPLETE,