[Task Manager] Using track_total_hits in workload agg (elastic#99360)

* Using track_total_hits in workload agg * Increasing size of terms aggregation
yctercero · May 11, 2021 · e20741d · e20741d
1 parent a60474b
commit e20741d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/x-pack/plugins/task_manager/server/monitoring/workload_statistics.test.ts b/x-pack/plugins/task_manager/server/monitoring/workload_statistics.test.ts
@@ -102,7 +102,7 @@ describe('Workload Statistics Aggregator', () => {
         expect(taskStore.aggregate).toHaveBeenCalledWith({
           aggs: {
             taskType: {
-              terms: { field: 'task.taskType' },
+              terms: { size: 100, field: 'task.taskType' },
               aggs: {
                 status: {
                   terms: { field: 'task.status' },

diff --git a/x-pack/plugins/task_manager/server/monitoring/workload_statistics.ts b/x-pack/plugins/task_manager/server/monitoring/workload_statistics.ts
@@ -116,7 +116,7 @@ export function createWorkloadAggregator(
       taskStore.aggregate({
         aggs: {
           taskType: {
-            terms: { field: 'task.taskType' },
+            terms: { size: 100, field: 'task.taskType' },
             aggs: {
               status: {
                 terms: { field: 'task.status' },

diff --git a/x-pack/plugins/task_manager/server/task_store.ts b/x-pack/plugins/task_manager/server/task_store.ts
@@ -340,6 +340,7 @@ export class TaskStore {
     const { body } = await this.esClient.search<ConcreteTaskInstance>({
       index: this.index,
       ignore_unavailable: true,
+      track_total_hits: true,
       body: ensureAggregationOnlyReturnsTaskObjects({
         query,
         aggs,