116 lines
3.6 KiB
JavaScript
116 lines
3.6 KiB
JavaScript
import EmberObject, { computed, get } from '@ember/object';
|
|
import { alias } from '@ember/object/computed';
|
|
import RollingArray from 'nomad-ui/utils/classes/rolling-array';
|
|
import AbstractStatsTracker from 'nomad-ui/utils/classes/abstract-stats-tracker';
|
|
|
|
const percent = (numerator, denominator) => {
|
|
if (!numerator || !denominator) {
|
|
return 0;
|
|
}
|
|
return numerator / denominator;
|
|
};
|
|
|
|
const empty = ts => ({ timestamp: ts, used: null, percent: null });
|
|
|
|
const AllocationStatsTracker = EmberObject.extend(AbstractStatsTracker, {
|
|
// Set via the stats computed property macro
|
|
allocation: null,
|
|
|
|
url: computed('allocation', function() {
|
|
return `/v1/client/allocation/${this.get('allocation.id')}/stats`;
|
|
}),
|
|
|
|
append(frame) {
|
|
const timestamp = new Date(Math.floor(frame.Timestamp / 1000000));
|
|
|
|
const cpuUsed = Math.floor(frame.ResourceUsage.CpuStats.TotalTicks) || 0;
|
|
this.get('cpu').pushObject({
|
|
timestamp,
|
|
used: cpuUsed,
|
|
percent: percent(cpuUsed, this.get('reservedCPU')),
|
|
});
|
|
|
|
const memoryUsed = frame.ResourceUsage.MemoryStats.RSS;
|
|
this.get('memory').pushObject({
|
|
timestamp,
|
|
used: memoryUsed,
|
|
percent: percent(memoryUsed / 1024 / 1024, this.get('reservedMemory')),
|
|
});
|
|
|
|
for (var taskName in frame.Tasks) {
|
|
const taskFrame = frame.Tasks[taskName];
|
|
const stats = this.get('tasks').findBy('task', taskName);
|
|
|
|
// If for whatever reason there is a task in the frame data that isn't in the
|
|
// allocation, don't attempt to append data for the task.
|
|
if (!stats) continue;
|
|
|
|
const frameTimestamp = new Date(Math.floor(taskFrame.Timestamp / 1000000));
|
|
|
|
const taskCpuUsed = Math.floor(taskFrame.ResourceUsage.CpuStats.TotalTicks) || 0;
|
|
stats.cpu.pushObject({
|
|
timestamp: frameTimestamp,
|
|
used: taskCpuUsed,
|
|
percent: percent(taskCpuUsed, stats.reservedCPU),
|
|
});
|
|
|
|
const taskMemoryUsed = taskFrame.ResourceUsage.MemoryStats.RSS;
|
|
stats.memory.pushObject({
|
|
timestamp: frameTimestamp,
|
|
used: taskMemoryUsed,
|
|
percent: percent(taskMemoryUsed / 1024 / 1024, stats.reservedMemory),
|
|
});
|
|
}
|
|
},
|
|
|
|
pause() {
|
|
const ts = new Date();
|
|
this.get('memory').pushObject(empty(ts));
|
|
this.get('cpu').pushObject(empty(ts));
|
|
this.get('tasks').forEach(task => {
|
|
task.memory.pushObject(empty(ts));
|
|
task.cpu.pushObject(empty(ts));
|
|
});
|
|
},
|
|
|
|
// Static figures, denominators for stats
|
|
reservedCPU: alias('allocation.taskGroup.reservedCPU'),
|
|
reservedMemory: alias('allocation.taskGroup.reservedMemory'),
|
|
|
|
// Dynamic figures, collected over time
|
|
// []{ timestamp: Date, used: Number, percent: Number }
|
|
cpu: computed('allocation', function() {
|
|
return RollingArray(this.get('bufferSize'));
|
|
}),
|
|
memory: computed('allocation', function() {
|
|
return RollingArray(this.get('bufferSize'));
|
|
}),
|
|
|
|
tasks: computed('allocation', function() {
|
|
const bufferSize = this.get('bufferSize');
|
|
return this.get('allocation.taskGroup.tasks').map(task => ({
|
|
task: get(task, 'name'),
|
|
|
|
// Static figures, denominators for stats
|
|
reservedCPU: get(task, 'reservedCPU'),
|
|
reservedMemory: get(task, 'reservedMemory'),
|
|
|
|
// Dynamic figures, collected over time
|
|
// []{ timestamp: Date, used: Number, percent: Number }
|
|
cpu: RollingArray(bufferSize),
|
|
memory: RollingArray(bufferSize),
|
|
}));
|
|
}),
|
|
});
|
|
|
|
export default AllocationStatsTracker;
|
|
|
|
export function stats(allocationProp, fetch) {
|
|
return computed(allocationProp, function() {
|
|
return AllocationStatsTracker.create({
|
|
fetch: fetch.call(this),
|
|
allocation: this.get(allocationProp),
|
|
});
|
|
});
|
|
}
|