@trigger.dev/core 0.0.0-prerelease-20250228095546 → 0.0.0-prerelease-20250321122618

package/dist/commonjs/v3/runEngineWorker/supervisor/queueConsumer.js

@@ -0,0 +1,89 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RunQueueConsumer = void 0;
+class RunQueueConsumer {
+    client;
+    preDequeue;
+    preSkip;
+    onDequeue;
+    intervalMs;
+    isEnabled;
+    constructor(opts) {
+        this.isEnabled = false;
+        this.intervalMs = opts.intervalMs ?? 5_000;
+        this.preDequeue = opts.preDequeue;
+        this.preSkip = opts.preSkip;
+        this.onDequeue = opts.onDequeue;
+        this.client = opts.client;
+    }
+    start() {
+        if (this.isEnabled) {
+            return;
+        }
+        this.isEnabled = true;
+        this.dequeue();
+    }
+    stop() {
+        if (!this.isEnabled) {
+            return;
+        }
+        this.isEnabled = false;
+    }
+    async dequeue() {
+        // Incredibly verbose logging for debugging purposes
+        // console.debug("[RunQueueConsumer] dequeue()", { enabled: this.isEnabled });
+        if (!this.isEnabled) {
+            return;
+        }
+        let preDequeueResult;
+        if (this.preDequeue) {
+            // console.debug("[RunQueueConsumer] preDequeue()");
+            try {
+                preDequeueResult = await this.preDequeue();
+            }
+            catch (preDequeueError) {
+                console.error("[RunQueueConsumer] preDequeue error", { error: preDequeueError });
+            }
+        }
+        // console.debug("[RunQueueConsumer] preDequeueResult", { preDequeueResult });
+        if (preDequeueResult?.skipDequeue ||
+            preDequeueResult?.maxResources?.cpu === 0 ||
+            preDequeueResult?.maxResources?.memory === 0) {
+            if (this.preSkip) {
+                console.debug("[RunQueueConsumer] preSkip()");
+                try {
+                    await this.preSkip();
+                }
+                catch (preSkipError) {
+                    console.error("[RunQueueConsumer] preSkip error", { error: preSkipError });
+                }
+            }
+            return this.scheduleNextDequeue();
+        }
+        try {
+            const response = await this.client.dequeue({
+                maxResources: preDequeueResult?.maxResources,
+            });
+            if (!response.success) {
+                console.error("[RunQueueConsumer] Failed to dequeue", { error: response.error });
+            }
+            else {
+                try {
+                    await this.onDequeue(response.data);
+                }
+                catch (handlerError) {
+                    console.error("[RunQueueConsumer] onDequeue error", { error: handlerError });
+                }
+            }
+        }
+        catch (clientError) {
+            console.error("[RunQueueConsumer] client.dequeue error", { error: clientError });
+        }
+        this.scheduleNextDequeue();
+    }
+    scheduleNextDequeue(delay = this.intervalMs) {
+        setTimeout(this.dequeue.bind(this), delay);
+    }
+}
+exports.RunQueueConsumer = RunQueueConsumer;
+//# sourceMappingURL=queueConsumer.js.map

package/dist/commonjs/v3/runEngineWorker/supervisor/queueConsumer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"queueConsumer.js","sourceRoot":"","sources":["../../../../../src/v3/runEngineWorker/supervisor/queueConsumer.ts"],"names":[],"mappings":";;;AAYA,MAAa,gBAAgB;IACV,MAAM,CAAuB;IAC7B,UAAU,CAAgB;IAC1B,OAAO,CAAa;IACpB,SAAS,CAA4D;IAE9E,UAAU,CAAS;IACnB,SAAS,CAAU;IAE3B,YAAY,IAA6B;QACvC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,IAAI,KAAK,CAAC;QAC3C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC;QAClC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAC5B,CAAC;IAED,KAAK;QACH,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED,IAAI;QACF,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QAED,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;IACzB,CAAC;IAEO,KAAK,CAAC,OAAO;QACnB,oDAAoD;QACpD,8EAA8E;QAE9E,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,OAAO;QACT,CAAC;QAED,IAAI,gBAA+D,CAAC;QACpE,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpB,oDAAoD;YAEpD,IAAI,CAAC;gBACH,gBAAgB,GAAG,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;YAC7C,CAAC;YAAC,OAAO,eAAe,EAAE,CAAC;gBACzB,OAAO,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;YACnF,CAAC;QACH,CAAC;QAED,8EAA8E;QAE9E,IACE,gBAAgB,EAAE,WAAW;YAC7B,gBAAgB,EAAE,YAAY,EAAE,GAAG,KAAK,CAAC;YACzC,gBAAgB,EAAE,YAAY,EAAE,MAAM,KAAK,CAAC,EAC5C,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,8BAA8B,CAAC,CAAC;gBAE9C,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,OAAO,EAAE,CAAC;gBACvB,CAAC;gBAAC,OAAO,YAAY,EAAE,CAAC;oBACtB,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC7E,CAAC;YACH,CAAC;YAED,OAAO,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACpC,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;gBACzC,YAAY,EAAE,gBAAgB,EAAE,YAAY;aAC7C,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACtB,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;YACnF,CAAC;iBAAM,CAAC;gBACN,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBACtC,CAAC;gBAAC,OAAO,YAAY,EAAE,CAAC;oBACtB,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC/E,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,WAAW,EAAE,CAAC;YACrB,OAAO,CAAC,KAAK,CAAC,yCAAyC,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;QACnF,CAAC;QAED,IAAI,CAAC,mBAAmB,EAAE,CAAC;IAC7B,CAAC;IAED,mBAAmB,CAAC,QAAgB,IAAI,CAAC,UAAU;QACjD,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,KAAK,CAAC,CAAC;IAC7C,CAAC;CACF;AAlGD,4CAkGC"}