Adding elastic#126839 to elastic#126214

stevedodson · Mar 4, 2022 · 82329e7 · 82329e7
1 parent 7317156
commit 82329e7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/x-pack/plugins/task_manager/server/plugin.ts b/x-pack/plugins/task_manager/server/plugin.ts
@@ -116,6 +116,11 @@ export class TaskManagerPlugin
         startServicesPromise.then(({ elasticsearch }) => elasticsearch.client),
     });
 
+    router.post({ path: '/api/task_manager/_stop', validate: false }, async (context, req, res) => {
+      this.taskPollingLifecycle!.stop();
+      return res.ok({ body: `Aye, aye, captain` });
+    });
+
     core.status.derivedStatus$.subscribe((status) =>
       this.logger.debug(`status core.status.derivedStatus now set to ${status.level}`)
     );

diff --git a/x-pack/plugins/task_manager/server/polling_lifecycle.ts b/x-pack/plugins/task_manager/server/polling_lifecycle.ts
@@ -216,6 +216,10 @@ export class TaskPollingLifecycle {
     });
   }
 
+  public stop() {
+    this.taskClaiming.excludedTaskTypes = ['*'];
+  }
+
   public get events(): Observable<TaskLifecycleEvent> {
     return this.events$;
   }

diff --git a/x-pack/plugins/task_manager/server/queries/task_claiming.ts b/x-pack/plugins/task_manager/server/queries/task_claiming.ts
@@ -121,7 +121,7 @@ export class TaskClaiming {
   private logger: Logger;
   private readonly taskClaimingBatchesByType: TaskClaimingBatches;
   private readonly taskMaxAttempts: Record<string, number>;
-  private readonly excludedTaskTypes: string[];
+  public readonly excludedTaskTypes: string[];
   private readonly unusedTypes: string[];
 
   /**