hobots 1.9.1 → 1.9.2

package/dist/index.cjs

@@ -161,7 +161,7 @@ function scrubSecrets(text) {
 
 // src/utils.ts
 var SDK_NAME = "hobots";
-var SDK_VERSION = true ? "1.9.0" : "0.0.0-dev";
+var SDK_VERSION = true ? "1.9.1" : "0.0.0-dev";
 function isNode() {
   return typeof process !== "undefined" && typeof process.versions?.node === "string";
 }
@@ -1559,6 +1559,7 @@ var MIN_POLL_INTERVAL_MS = 1e3;
 var MAX_CONCURRENCY = 5;
 var POLL_RETRY_DELAYS_MS = [1e3, 4e3, 16e3];
 var AUTH_FAILURE_INTERVAL_MS = 3e4;
+var DEFAULT_CANCEL_GRACE_MS = 1e4;
 function hostname() {
   if (!isNode()) return void 0;
   return process.env.HOSTNAME || process.env.COMPUTERNAME || void 0;
@@ -1577,6 +1578,7 @@ var TaskAgent = class {
     this.taskTimeoutMs = options.taskTimeoutMs;
     this.logFlushIntervalMs = options.logFlushIntervalMs;
     this.cancelCheckIntervalMs = options.cancelCheckIntervalMs;
+    this.cancelGraceMs = options.cancelGraceMs ?? DEFAULT_CANCEL_GRACE_MS;
     this.logger.debug("agente configurado", { baseUrl: config.tasksBaseUrl });
   }
   logger;
@@ -1588,6 +1590,7 @@ var TaskAgent = class {
   taskTimeoutMs;
   logFlushIntervalMs;
   cancelCheckIntervalMs;
+  cancelGraceMs;
   state = "idle";
   timer;
   pollAbort;
@@ -1772,14 +1775,26 @@ var TaskAgent = class {
   }
   async invokeWithTimeout(handler, run, reporter) {
     const invocation = Promise.resolve().then(() => handler(run.params, reporter.context));
-    if (!this.taskTimeoutMs || this.taskTimeoutMs <= 0) {
+    const races = [];
+    if (this.taskTimeoutMs && this.taskTimeoutMs > 0) {
+      races.push(sleep(this.taskTimeoutMs).then(() => TIMEOUT_SENTINEL));
+    }
+    if (this.cancelGraceMs > 0) {
+      races.push(waitForCancelGrace(reporter.context.signal, this.cancelGraceMs).then(() => CANCEL_SENTINEL));
+    }
+    if (races.length === 0) {
       await invocation;
       return;
     }
+    const outcome = await Promise.race([invocation, ...races]);
+    if (outcome !== TIMEOUT_SENTINEL && outcome !== CANCEL_SENTINEL) return;
+    if (outcome === CANCEL_SENTINEL) {
+      this.logger.warn(
+        `run ${run.id} cancelada \u2014 handler n\xE3o respondeu a ctx.signal em ${this.cancelGraceMs}ms, liberando o slot (o handler pode continuar rodando em segundo plano)`
+      );
+      return;
+    }
     const timeoutMs = this.taskTimeoutMs;
-    const timeout = sleep(timeoutMs).then(() => TIMEOUT_SENTINEL);
-    const outcome = await Promise.race([invocation, timeout]);
-    if (outcome !== TIMEOUT_SENTINEL) return;
     reporter.abort();
     invocation.then(() => this.logger.warn(`run ${run.id} terminou depois do timeout de ${timeoutMs}ms`)).catch(() => this.logger.warn(`run ${run.id} falhou depois do timeout de ${timeoutMs}ms (erro descartado)`));
     throw Object.assign(new Error(`timeout ap\xF3s ${timeoutMs}ms (o handler pode continuar em background)`), {
@@ -1788,6 +1803,16 @@ var TaskAgent = class {
   }
 };
 var TIMEOUT_SENTINEL = /* @__PURE__ */ Symbol("task-timeout");
+var CANCEL_SENTINEL = /* @__PURE__ */ Symbol("task-canceled");
+function waitForCancelGrace(signal, graceMs) {
+  return new Promise((resolve) => {
+    if (signal.aborted) {
+      setTimeout(resolve, graceMs);
+      return;
+    }
+    signal.addEventListener("abort", () => setTimeout(resolve, graceMs), { once: true });
+  });
+}
 var MAX_ERROR_MESSAGE_CHARS = 1e4;
 var MAX_ERROR_STACK_CHARS = 2e4;
 function toRunError(error) {