All files / src/app/shared/services prometheus-notification.service.ts

100% Statements 31/31
85.71% Branches 12/14
100% Functions 9/9
100% Lines 26/26

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 546x   6x   6x     6x         6x   15x     15x 15x   15x     26x 26x 1x   25x     24x 1x       24x 24x 2x   22x 13x 15x     22x     15x 15x   21x   6x  
import { Injectable } from '@angular/core';
 
import * as _ from 'lodash';
 
import { PrometheusService } from '../api/prometheus.service';
import { CdNotificationConfig } from '../models/cd-notification';
import { AlertmanagerNotification } from '../models/prometheus-alerts';
import { PrometheusAlertFormatter } from './prometheus-alert-formatter';
 
@Injectable({
  providedIn: 'root'
})
export class PrometheusNotificationService {
  private notifications: AlertmanagerNotification[];
  private backendFailure = false;
 
  constructor(
    private alertFormatter: PrometheusAlertFormatter,
    private prometheusService: PrometheusService
  ) {
    this.notifications = [];
  }
 
  refresh() {
    if (this.backendFailure) {
      return;
    }
    this.prometheusService
      .getNotifications(_.last(this.notifications))
      .subscribe(
        (notifications) => this.handleNotifications(notifications),
        () => (this.backendFailure = true)
      );
  }
 
  private handleNotifications(notifications: AlertmanagerNotification[]) {
    if (notifications.length === 0) {
      return;
    }
    if (this.notifications.length > 0) {
      this.alertFormatter.sendNotifications(
        _.flatten(notifications.map((notification) => this.formatNotification(notification)))
      );
    }
    this.notifications = this.notifications.concat(notifications);
  }
 
  private formatNotification(notification: AlertmanagerNotification): CdNotificationConfig[] {
    return this.alertFormatter
      .convertToCustomAlerts(notification.alerts)
      .map((alert) => this.alertFormatter.convertAlertToNotification(alert));
  }
}