Require consecutive settled checks in wait

Author: firecow

src/commands/wait-cmd.ts

@@ -26,12 +26,15 @@ export async function handler (args: ArgumentsCamelCase) {
     assertNumber(timeout, "timeout must be a number in ms");
     const interval = args.interval;
     assertNumber(interval, "interval must be a number in ms");
+    const stableChecks = args.stableChecks;
+    assertNumber(stableChecks, "stableChecks must be a number");
 
     const dockerode = new Docker();
 
     console.log(`Awaiting task reconciliation for ${timeout}ms`);
 
     let services: Service[], tasks: Task[], timedout, bail, serviceStateMap;
+    let stableStreak = 0;
     const start = Date.now();
     do {
         // To prevent high cpu usage
@@ -57,15 +60,19 @@ export async function handler (args: ArgumentsCamelCase) {
 
         const servicesUpdating = [...serviceStateMap.entries()];
 
-        bail = servicesUpdating.length === 0;
-        if (!bail) {
+        if (servicesUpdating.length === 0) {
+            stableStreak++;
+        } else {
+            stableStreak = 0;
             for (const [serviceId, state] of servicesUpdating) {
                 const serviceName = services.find((s) => s.ID === serviceId)?.Spec?.Name;
                 assert(serviceName != null, "serviceName must be a string");
                 const errMsg = tasks.find((t) => t.ServiceID === serviceId && t.Status.Err)?.Status.Err;
                 console.log(`${serviceName} is in ${state} state${errMsg ? ", error: '" + errMsg + "'" : ""}`);
             }
         }
+
+        bail = stableStreak >= stableChecks;
     } while (!timedout && !bail);
 
     if (timedout) {
@@ -88,6 +95,11 @@ export function builder (yargs: Argv) {
         description: "How often reconciliation should run",
         default: 5000,
     });
+    yargs.positional("stableChecks", {
+        type: "number",
+        description: "Consecutive settled checks required before success",
+        default: 3,
+    });
     yargs.hide("help");
     yargs.hide("version");
     return yargs;