From 82329e71b7b4bea80f8444438ea7c5f5917ef2cc Mon Sep 17 00:00:00 2001
From: Stephen Dodson <steve.dodson@elastic.co>
Date: Fri, 4 Mar 2022 11:47:48 +0000
Subject: [PATCH] Adding #126839 to #126214

---
 x-pack/plugins/task_manager/server/plugin.ts                | 5 +++++
 x-pack/plugins/task_manager/server/polling_lifecycle.ts     | 4 ++++
 x-pack/plugins/task_manager/server/queries/task_claiming.ts | 2 +-
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/x-pack/plugins/task_manager/server/plugin.ts b/x-pack/plugins/task_manager/server/plugin.ts
index b58b0665c10c0f..4c506b73f60f1c 100644
--- 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
index a452c8a3f82fbe..659e34bb2e8cee 100644
--- 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
index 1b4f0fdb73683c..dad435ac2b10d7 100644
--- 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[];
 
   /**