open-nomad/ui/app/components/primary-metric/task.js
2022-01-20 09:54:56 -05:00

68 lines
1.7 KiB
JavaScript

import Ember from 'ember';
import Component from '@glimmer/component';
import { tracked } from '@glimmer/tracking';
import { task, timeout } from 'ember-concurrency';
import { assert } from '@ember/debug';
import { inject as service } from '@ember/service';
import { action } from '@ember/object';
export default class TaskPrimaryMetric extends Component {
@service('stats-trackers-registry') statsTrackersRegistry;
/** Args
taskState = null;
metric null; (one of 'cpu' or 'memory'
*/
@tracked tracker = null;
@tracked taskState = null;
get metric() {
assert('metric is a required argument', this.args.metric);
return this.args.metric;
}
get data() {
if (!this.tracker) return [];
const task = this.tracker.tasks.findBy('task', this.taskState.name);
return task && task[this.metric];
}
get reservedAmount() {
if (!this.tracker) return null;
const task = this.tracker.tasks.findBy('task', this.taskState.name);
if (this.metric === 'cpu') return task.reservedCPU;
if (this.metric === 'memory') return task.reservedMemory;
return null;
}
get chartClass() {
if (this.metric === 'cpu') return 'is-info';
if (this.metric === 'memory') return 'is-danger';
return 'is-primary';
}
@task(function* () {
do {
this.tracker.poll.perform();
yield timeout(100);
} while (!Ember.testing);
})
poller;
@action
start() {
this.taskState = this.args.taskState;
this.tracker = this.statsTrackersRegistry.getTracker(
this.args.taskState.allocation
);
this.poller.perform();
}
willDestroy() {
super.willDestroy(...arguments);
this.poller.cancelAll();
this.tracker.signalPause.perform();
}
}