task_manager.ts rename

tsullivan · Sep 5, 2018 · 1e60301 · 1e60301
1 parent 3630e8c
commit 1e60301
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 13 deletions.
diff --git a/src/server/task_manager/index.ts b/src/server/task_manager/index.ts
@@ -17,4 +17,4 @@
  * under the License.
  */
 
-export { taskManagerMixin } from './task_manager';
+export { taskManagerMixin } from './task_manager_mixin';
diff --git a/src/server/task_manager/task_manager.ts → ...server/task_manager/task_manager_mixin.ts b/src/server/task_manager/task_manager.ts → ...server/task_manager/task_manager_mixin.ts
@@ -29,12 +29,16 @@ import {
 } from './task_pool';
 import { SanitizedTaskDefinition } from './task_pool/task';
 
-export async function taskManagerMixin(kbnServer: any, server: any, config: any) {
-  const logger = new Logger((...args) => server.log(...args));
-  const numWorkers = config.get('taskManager.num_workers');
+function taskManagerFactory(kbnServer: any, server: any, config: any) {
+  let taskManager: TaskManager;
+  return async function getTaskManager() {
+    if (taskManager) {
+      return taskManager;
+    }
 
-  kbnServer.afterPluginsInit(async () => {
+    const logger = new Logger((...args) => server.log(...args));
     const callCluster = server.plugins.elasticsearch.getCluster('admin').callWithInternalUser;
+    const numWorkers = config.get('taskManager.num_workers');
     const store = new TaskStore({
       index: config.get('taskManager.index'),
       callCluster,
@@ -58,14 +62,18 @@ export async function taskManagerMixin(kbnServer: any, server: any, config: any)
 
     pool.start();
 
-    server.decorate(
-      'server',
-      'taskManager',
-      new TaskManager({
-        store,
-        pool,
-      })
-    );
+    taskManager = new TaskManager({
+      store,
+      pool,
+    });
+
+    return taskManager;
+  };
+}
+
+export async function taskManagerMixin(kbnServer: any, server: any, config: any) {
+  server.decorate('server', 'taskManager', {
+    getTaskManager: taskManagerFactory(kbnServer, server, config),
   });
 }