1 files changed, 59 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/frontend/src/app/shared/services/task-manager.service.ts b/src/pybind/mgr/dashboard/frontend/src/app/shared/services/task-manager.service.ts
new file mode 100644
index 000000000..0310a7826
--- /dev/null
+++ b/src/pybind/mgr/dashboard/frontend/src/app/shared/services/task-manager.service.ts
@@ -0,0 +1,59 @@
+import { Injectable } from '@angular/core';
+
+import _ from 'lodash';
+
+import { ExecutingTask } from '../models/executing-task';
+import { FinishedTask } from '../models/finished-task';
+import { Task } from '../models/task';
+import { SummaryService } from './summary.service';
+
+class TaskSubscription {
+  name: string;
+  metadata: object;
+  onTaskFinished: (finishedTask: FinishedTask) => any;
+
+  constructor(name: string, metadata: object, onTaskFinished: any) {
+    this.name = name;
+    this.metadata = metadata;
+    this.onTaskFinished = onTaskFinished;
+  }
+}
+
+@Injectable({
+  providedIn: 'root'
+})
+export class TaskManagerService {
+  subscriptions: Array<TaskSubscription> = [];
+
+  init(summaryService: SummaryService) {
+    return summaryService.subscribe((summary) => {
+      const executingTasks = summary.executing_tasks;
+      const finishedTasks = summary.finished_tasks;
+      const newSubscriptions: Array<TaskSubscription> = [];
+      for (const subscription of this.subscriptions) {
+        const finishedTask = <FinishedTask>this._getTask(subscription, finishedTasks);
+        const executingTask = <ExecutingTask>this._getTask(subscription, executingTasks);
+        if (finishedTask !== null && executingTask === null) {
+          subscription.onTaskFinished(finishedTask);
+        }
+        if (executingTask !== null) {
+          newSubscriptions.push(subscription);
+        }
+        this.subscriptions = newSubscriptions;
+      }
+    });
+  }
+
+  subscribe(name: string, metadata: object, onTaskFinished: (finishedTask: FinishedTask) => any) {
+    this.subscriptions.push(new TaskSubscription(name, metadata, onTaskFinished));
+  }
+
+  private _getTask(subscription: TaskSubscription, tasks: Array<Task>): Task {
+    for (const task of tasks) {
+      if (task.name === subscription.name && _.isEqual(task.metadata, subscription.metadata)) {
+        return task;
+      }
+    }
+    return null;
+  }
+}