open-nomad/ui/app/utils/resources-diffs.js

import d3Format from 'd3-format';

import { reduceToLargestUnit } from 'nomad-ui/helpers/format-bytes';

const formatPercent = d3Format.format('+.0%');
const sumAggregate = (total, val) => total + val;

export default class ResourcesDiffs {
  constructor(model, multiplier, recommendations, excludedRecommendations) {
    this.model = model;
    this.multiplier = multiplier;
    this.recommendations = recommendations;
    this.excludedRecommendations = excludedRecommendations.filter(r => recommendations.includes(r));
  }

  get cpu() {
    const included = this.includedRecommendations.filterBy('resource', 'CPU');
    const excluded = this.excludedRecommendations.filterBy('resource', 'CPU');

    return new ResourceDiffs(
      this.model.reservedCPU,
      'reservedCPU',
      'MHz',
      this.multiplier,
      included,
      excluded
    );
  }

  get memory() {
    const included = this.includedRecommendations.filterBy('resource', 'MemoryMB');
    const excluded = this.excludedRecommendations.filterBy('resource', 'MemoryMB');

    return new ResourceDiffs(
      this.model.reservedMemory,
      'reservedMemory',
      'MiB',
      this.multiplier,
      included,
      excluded
    );
  }

  get includedRecommendations() {
    return this.recommendations.reject(r => this.excludedRecommendations.includes(r));
  }
}

class ResourceDiffs {
  constructor(
    base,
    baseTaskPropertyName,
    units,
    multiplier,
    includedRecommendations,
    excludedRecommendations
  ) {
    this.base = base;
    this.baseTaskPropertyName = baseTaskPropertyName;
    this.units = units;
    this.multiplier = multiplier;
    this.included = includedRecommendations;
    this.excluded = excludedRecommendations;
  }

  get recommended() {
    if (this.included.length) {
      return (
        this.included.mapBy('value').reduce(sumAggregate, 0) +
        this.excluded.mapBy(`task.${this.baseTaskPropertyName}`).reduce(sumAggregate, 0)
      );
    } else {
      return this.base;
    }
  }

  get delta() {
    return this.recommended - this.base;
  }

  get aggregateDiff() {
    return this.delta * this.multiplier;
  }

  get absoluteAggregateDiff() {
    const delta = Math.abs(this.aggregateDiff);

    if (this.units === 'MiB') {
      if (delta === 0) {
        return '0 MiB';
      }

      const [memory, units] = reduceToLargestUnit(delta * 1024 * 1024);
      const formattedMemory = Number.isInteger(memory) ? memory : memory.toFixed(2);

      return `${formattedMemory} ${units}`;
    } else {
      return `${delta} ${this.units}`;
    }
  }

  get signedDiff() {
    const delta = this.aggregateDiff;
    return `${signForDelta(delta)}${this.absoluteAggregateDiff}`;
  }

  get percentDiff() {
    return formatPercent(this.delta / this.base);
  }
}

function signForDelta(delta) {
  if (delta > 0) {
    return '+';
  } else if (delta < 0) {
    return '-';
  }

  return '';
}