webapp-frontend/src/app/ei-coordinator/ei-job.datasource.ts

   1 /*-
   2  * ========================LICENSE_START=================================
   3  * O-RAN-SC
   4  * %%
   5  * Copyright (C) 2019 Nordix Foundation
   6  * %%
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ========================LICENSE_END===================================
  19  */
  20
  21 import { HttpErrorResponse } from '@angular/common/http';
  22 import { Injectable } from '@angular/core';
  23 import { MatTableDataSource } from '@angular/material';
  24
  25 import { BehaviorSubject } from 'rxjs/BehaviorSubject';
  26 import { of } from 'rxjs/observable/of';
  27 import { catchError, finalize, map } from 'rxjs/operators';
  28
  29 import { EIJob } from '../interfaces/ei.types';
  30 import { EIService } from '../services/ei/ei.service';
  31 import { NotificationService } from '../services/ui/notification.service';
  32
  33 @Injectable({
  34     providedIn: 'root'
  35 })
  36
  37 export class EIJobDataSource extends MatTableDataSource<EIJob> {
  38
  39     eiJobsSubject = new BehaviorSubject<EIJob[]>([]);
  40
  41     private loadingSubject = new BehaviorSubject<boolean>(false);
  42
  43     public loading$ = this.loadingSubject.asObservable();
  44
  45     public rowCount = 1; // hide footer during intial load
  46
  47     constructor(
  48         private eiSvc: EIService,
  49         private notificationService: NotificationService) {
  50         super();
  51     }
  52
  53     getJobs() {
  54         this.loadingSubject.next(true);
  55         this.eiSvc.getProducerIds()
  56             .pipe(
  57                 catchError((her: HttpErrorResponse) => {
  58                     this.notificationService.error('Failed to get EI jobs: ' + her.error);
  59                     return of([]);
  60                 }),
  61                 finalize(() => this.loadingSubject.next(false))
  62             )
  63             .subscribe((producerIds: string[]) => {
  64                 producerIds.forEach(id => {
  65                     this.getJobsForProducer(id);
  66                 });
  67             });
  68     }
  69
  70     private getJobsForProducer(id: string) {
  71         console.log('Getting jobs for producer ID: ', id);
  72         this.eiSvc.getJobsForProducer(id).subscribe(jobs => {
  73             this.addJobsToSubject(jobs);
  74             this.rowCount = this.eiJobsSubject.getValue().length;
  75         });
  76     }
  77
  78     private addJobsToSubject(jobs: EIJob[]) {
  79         const currentValue = this.eiJobsSubject.value;
  80         const updatedValue = [...currentValue, ...jobs];
  81         this.eiJobsSubject.next(updatedValue);
  82     }
  83
  84     connect(): BehaviorSubject<EIJob[]> {
  85         return this.eiJobsSubject;
  86     }
  87
  88     disconnect(): void {
  89         this.eiJobsSubject.complete();
  90         this.loadingSubject.complete();
  91     }
  92 }