open-nomad/ui/app/models/task-group.js
2022-01-20 09:54:56 -05:00

97 lines
2.4 KiB
JavaScript

import { computed } from '@ember/object';
import Fragment from 'ember-data-model-fragments/fragment';
import { attr } from '@ember-data/model';
import {
fragmentOwner,
fragmentArray,
fragment,
} from 'ember-data-model-fragments/attributes';
import sumAggregation from '../utils/properties/sum-aggregation';
import classic from 'ember-classic-decorator';
const maybe = (arr) => arr || [];
@classic
export default class TaskGroup extends Fragment {
@fragmentOwner() job;
@attr('string') name;
@attr('number') count;
@fragmentArray('task') tasks;
@fragmentArray('service') services;
@fragmentArray('volume-definition') volumes;
@fragment('group-scaling') scaling;
@attr() meta;
@computed('job.meta.raw', 'meta')
get mergedMeta() {
return {
...this.job.get('meta.raw'),
...this.meta,
};
}
@computed('tasks.@each.driver')
get drivers() {
return this.tasks.mapBy('driver').uniq();
}
@computed('job.allocations.@each.taskGroup', 'name')
get allocations() {
return maybe(this.get('job.allocations')).filterBy(
'taskGroupName',
this.name
);
}
@sumAggregation('tasks', 'reservedCPU') reservedCPU;
@sumAggregation('tasks', 'reservedMemory') reservedMemory;
@sumAggregation('tasks', 'reservedDisk') reservedDisk;
@computed('tasks.@each.{reservedMemory,reservedMemoryMax}')
get reservedMemoryMax() {
return this.get('tasks')
.map((t) => t.get('reservedMemoryMax') || t.get('reservedMemory'))
.reduce((sum, count) => sum + count, 0);
}
@attr('number') reservedEphemeralDisk;
@computed('job.latestFailureEvaluation.failedTGAllocs.[]', 'name')
get placementFailures() {
const placementFailures = this.get(
'job.latestFailureEvaluation.failedTGAllocs'
);
return placementFailures && placementFailures.findBy('name', this.name);
}
@computed('summary.{queuedAllocs,startingAllocs}')
get queuedOrStartingAllocs() {
return (
this.get('summary.queuedAllocs') + this.get('summary.startingAllocs')
);
}
@computed('job.taskGroupSummaries.[]', 'name')
get summary() {
return maybe(this.get('job.taskGroupSummaries')).findBy('name', this.name);
}
@computed('job.scaleState.taskGroupScales.[]', 'name')
get scaleState() {
return maybe(this.get('job.scaleState.taskGroupScales')).findBy(
'name',
this.name
);
}
scale(count, message) {
return this.job.scale(this.name, count, message);
}
}