Poll for allocation stats

ui/app/components/allocation-row.js

@@ -1,7 +1,9 @@
+import Ember from 'ember';
 import { inject as service } from '@ember/service';
 import Component from '@ember/component';
 import { run } from '@ember/runloop';
 import { lazyClick } from '../helpers/lazy-click';
+import { task, timeout } from 'ember-concurrency';
 
 export default Component.extend({
   store: service(),
@@ -15,6 +17,10 @@ export default Component.extend({
   // Used to determine whether the row should mention the node or the job
   context: null,
 
+  // Internal state
+  stats: null,
+  statsError: false,
+
   onClick() {},
 
   click(event) {
@@ -45,8 +51,32 @@ export default Component.extend({
     // being resolved through the store (store.findAll('job')). The
     // workaround is to persist the jobID as a string on the allocation
     // and manually re-link the two records here.
+    const allocation = this.get('allocation');
+
+    if (allocation) {
+      this.get('fetchStats').perform(allocation);
+    } else {
+      this.get('fetchStats').cancelAll();
+      this.set('stats', null);
+    }
     run.scheduleOnce('afterRender', this, qualifyJob);
   },
+
+  fetchStats: task(function*(allocation) {
+    const maxTiming = 5500;
+    const backoffSequence = [500, 800, 1300, 2100, 3400];
+
+    do {
+      try {
+        const stats = yield allocation.fetchStats();
+        this.set('stats', stats);
+      } catch (error) {
+        this.set('statsError', true);
+        break;
+      }
+      yield timeout(backoffSequence.shift() || maxTiming);
+    } while (!Ember.testing);
+  }).drop(),
 });
 
 function qualifyJob() {

ui/app/templates/components/allocation-row.hbs

@@ -23,37 +23,41 @@
   <td data-test-job-version>{{allocation.jobVersion}}</td>
 {{/if}}
 <td data-test-cpu class="has-text-centered">
-  {{#if allocation.stats.isPending}}
+  {{#if (and (not stats) fetchStats.isRunning)}}
     ...
-  {{else if allocation.stats.isRejected}}
-    <span class="tooltip text-center" aria-label="Couldn't connect to client">
+  {{else if (not allocation)}}
+    {{! nothing when there's no allocation}}
+  {{else if statsError}}
+    <span class="tooltip text-center" aria-label="Couldn't collect stats">
       {{x-icon "warning" class="is-warning"}}
     </span>
   {{else}}
-    <div class="inline-chart tooltip" aria-label="{{allocation.cpuUsed}} / {{allocation.taskGroup.reservedCPU}} MHz">
+    <div class="inline-chart tooltip" aria-label="{{stats.cpuUsed}} / {{stats.reservedCPU}} MHz">
       <progress
         class="progress is-info is-small"
-        value="{{allocation.percentCPU}}"
+        value="{{stats.percentCPU}}"
         max="1">
-        {{allocation.percentCPU}}
+        {{stats.percentCPU}}
       </progress>
     </div>
   {{/if}}
 </td>
 <td data-test-mem class="has-text-centered">
-  {{#if allocation.stats.isPending}}
+  {{#if (and (not stats) fetchStats.isRunning)}}
     ...
-  {{else if allocation.stats.isRejected}}
-    <span class="tooltip text-center" aria-label="Couldn't connect to client">
+  {{else if (not allocation)}}
+    {{! nothing when there's no allocation}}
+  {{else if statsError}}
+    <span class="tooltip text-center" aria-label="Couldn't collect stats">
       {{x-icon "warning" class="is-warning"}}
     </span>
   {{else}}
-    <div class="inline-chart tooltip" aria-label="{{format-bytes allocation.memoryUsed}} / {{allocation.taskGroup.reservedMemory}} MiB">
+    <div class="inline-chart tooltip" aria-label="{{format-bytes stats.memoryUsed}} / {{stats.reservedMemory}} MiB">
       <progress
         class="progress is-danger is-small"
-        value="{{allocation.percentMemory}}"
+        value="{{stats.percentMemory}}"
         max="1">
-        {{allocation.percentMemory}}
+        {{stats.percentMemory}}
       </progress>
     </div>
   {{/if}}