open-nomad/ui/app/components/allocation-row.js
2020-06-10 16:18:56 -05:00

102 lines
2.7 KiB
JavaScript

import Ember from 'ember';
import { inject as service } from '@ember/service';
import Component from '@ember/component';
import { computed } from '@ember/object';
import { computed as overridable } from 'ember-overridable-computed';
import { alias } from '@ember/object/computed';
import { run } from '@ember/runloop';
import { task, timeout } from 'ember-concurrency';
import { lazyClick } from '../helpers/lazy-click';
import AllocationStatsTracker from 'nomad-ui/utils/classes/allocation-stats-tracker';
import classic from 'ember-classic-decorator';
import { classNames, tagName } from '@ember-decorators/component';
@classic
@tagName('tr')
@classNames('allocation-row', 'is-interactive')
export default class AllocationRow extends Component {
@service store;
@service token;
allocation = null;
// Used to determine whether the row should mention the node or the job
context = null;
// Internal state
statsError = false;
@overridable(() => !Ember.testing) enablePolling;
@computed('allocation', 'allocation.isRunning')
get stats() {
if (!this.get('allocation.isRunning')) return undefined;
return AllocationStatsTracker.create({
fetch: url => this.token.authorizedRequest(url),
allocation: this.allocation,
});
}
@alias('stats.cpu.lastObject') cpu;
@alias('stats.memory.lastObject') memory;
onClick() {}
click(event) {
lazyClick([this.onClick, event]);
}
didReceiveAttrs() {
this.updateStatsTracker();
}
updateStatsTracker() {
const allocation = this.allocation;
if (allocation) {
run.scheduleOnce('afterRender', this, qualifyAllocation);
} else {
this.fetchStats.cancelAll();
}
}
@(task(function*() {
do {
if (this.stats) {
try {
yield this.get('stats.poll').perform();
this.set('statsError', false);
} catch (error) {
this.set('statsError', true);
}
}
yield timeout(500);
} while (this.enablePolling);
}).drop())
fetchStats;
}
async function qualifyAllocation() {
const allocation = this.allocation;
// Make sure the allocation is a complete record and not a partial so we
// can show information such as preemptions and rescheduled allocation.
if (allocation.isPartial) {
await allocation.reload();
}
if (allocation.get('job.isPending')) {
// Make sure the job is loaded before starting the stats tracker
await allocation.get('job');
} else if (!allocation.get('taskGroup')) {
// Make sure that the job record in the store for this allocation
// is complete and not a partial from the list endpoint
const job = allocation.get('job.content');
if (job) await job.reload();
}
this.fetchStats.perform();
}