ceph/src/pybind/mgr/dashboard/frontend/src/app/shared/services/task-manager.service.ts

   1 import { Injectable } from '@angular/core';
   2
   3 import * as _ from 'lodash';
   4
   5 import { ExecutingTask } from '../models/executing-task';
   6 import { FinishedTask } from '../models/finished-task';
   7 import { Task } from '../models/task';
   8 import { SummaryService } from './summary.service';
   9
  10 class TaskSubscription {
  11   name: string;
  12   metadata: object;
  13   onTaskFinished: (finishedTask: FinishedTask) => any;
  14
  15   constructor(name: string, metadata: object, onTaskFinished: any) {
  16     this.name = name;
  17     this.metadata = metadata;
  18     this.onTaskFinished = onTaskFinished;
  19   }
  20 }
  21
  22 @Injectable({
  23   providedIn: 'root'
  24 })
  25 export class TaskManagerService {
  26   subscriptions: Array<TaskSubscription> = [];
  27
  28   constructor(summaryService: SummaryService) {
  29     summaryService.subscribe((data: any) => {
  30       if (!data) {
  31         return;
  32       }
  33       const executingTasks = data.executing_tasks;
  34       const finishedTasks = data.finished_tasks;
  35       const newSubscriptions: Array<TaskSubscription> = [];
  36       for (const subscription of this.subscriptions) {
  37         const finishedTask = <FinishedTask>this._getTask(subscription, finishedTasks);
  38         const executingTask = <ExecutingTask>this._getTask(subscription, executingTasks);
  39         if (finishedTask !== null && executingTask === null) {
  40           subscription.onTaskFinished(finishedTask);
  41         }
  42         if (executingTask !== null) {
  43           newSubscriptions.push(subscription);
  44         }
  45         this.subscriptions = newSubscriptions;
  46       }
  47     });
  48   }
  49
  50   subscribe(name: string, metadata: object, onTaskFinished: (finishedTask: FinishedTask) => any) {
  51     this.subscriptions.push(new TaskSubscription(name, metadata, onTaskFinished));
  52   }
  53
  54   _getTask(subscription: TaskSubscription, tasks: Array<Task>): Task {
  55     for (const task of tasks) {
  56       if (task.name === subscription.name && _.isEqual(task.metadata, subscription.metadata)) {
  57         return task;
  58       }
  59     }
  60     return null;
  61   }
  62 }