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 { Injectable } from '@angular/core';
  22 import { MatTableDataSource } from '@angular/material';
  23
  24 import { BehaviorSubject } from 'rxjs/BehaviorSubject';
  25
  26 import { EIJob } from '../interfaces/ei.types';
  27 import { EIService } from '../services/ei/ei.service';
  28
  29 @Injectable({
  30     providedIn: 'root'
  31 })
  32
  33 export class EIJobDataSource extends MatTableDataSource<EIJob> {
  34
  35     eiJobsSubject = new BehaviorSubject<EIJob[]>([]);
  36
  37     private loadingSubject = new BehaviorSubject<boolean>(false);
  38
  39     public loading$ = this.loadingSubject.asObservable();
  40
  41     public rowCount = 1; // hide footer during intial load
  42
  43     constructor(
  44         private eiSvc: EIService) {
  45         super();
  46     }
  47
  48     getJobs() {
  49         this.loadingSubject.next(true);
  50         this.eiSvc.getProducerIds()
  51             .subscribe((producerIds: string[]) => {
  52                 producerIds.forEach(id => {
  53                     this.getJobsForProducer(id);
  54                 });
  55             });
  56     }
  57
  58     private getJobsForProducer(id: string) {
  59         console.log('Getting jobs for producer ID: ', id);
  60         this.eiSvc.getJobsForProducer(id).subscribe(jobs => {
  61             this.addJobsToSubject(jobs);
  62             this.rowCount = this.eiJobsSubject.getValue().length;
  63         });
  64     }
  65
  66     private addJobsToSubject(jobs: EIJob[]) {
  67         const currentValue = this.eiJobsSubject.value;
  68         const updatedValue = [...currentValue, ...jobs];
  69         this.eiJobsSubject.next(updatedValue);
  70     }
  71
  72     connect(): BehaviorSubject<EIJob[]> {
  73         return this.eiJobsSubject;
  74     }
  75
  76     disconnect(): void {
  77         this.eiJobsSubject.complete();
  78         this.loadingSubject.complete();
  79     }
  80 }