@terreno/jobs 57.4.0 → 57.6.0

package/dist/runners/mongoRunner.js

@@ -1,204 +1,110 @@
 "use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
-    });
-};
-var __generator = (this && this.__generator) || function (thisArg, body) {
-    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
-    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
-    function verb(n) { return function (v) { return step([n, v]); }; }
-    function step(op) {
-        if (f) throw new TypeError("Generator is already executing.");
-        while (g && (g = 0, op[0] && (_ = 0)), _) try {
-            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
-            if (y = 0, t) op = [op[0] & 2, t.value];
-            switch (op[0]) {
-                case 0: case 1: t = op; break;
-                case 4: _.label++; return { value: op[1], done: false };
-                case 5: _.label++; y = op[1]; op = [0]; continue;
-                case 7: op = _.ops.pop(); _.trys.pop(); continue;
-                default:
-                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
-                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
-                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
-                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
-                    if (t[2]) _.ops.pop();
-                    _.trys.pop(); continue;
-            }
-            op = body.call(thisArg, _);
-        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
-        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
-    }
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MongoJobRunner = exports.hasWorkerIdPrefix = exports.getWorkerId = void 0;
-var luxon_1 = require("luxon");
-var claimLock_1 = require("../claimLock");
-var jobExecutor_1 = require("../jobExecutor");
-var job_1 = require("../models/job");
-var DEFAULT_POLL_INTERVAL_MS = 1000;
+const luxon_1 = require("luxon");
+const claimLock_1 = require("../claimLock");
+const jobExecutor_1 = require("../jobExecutor");
+const job_1 = require("../models/job");
+const DEFAULT_POLL_INTERVAL_MS = 1_000;
 /** @internal */
 var claimLock_2 = require("../claimLock");
 Object.defineProperty(exports, "getWorkerId", { enumerable: true, get: function () { return claimLock_2.getWorkerId; } });
 Object.defineProperty(exports, "hasWorkerIdPrefix", { enumerable: true, get: function () { return claimLock_2.hasWorkerIdPrefix; } });
-var buildLockExpiry = function (lockTtlMs) {
-    return luxon_1.DateTime.utc().minus({ milliseconds: lockTtlMs }).toJSDate();
-};
-var waitForAbortOrTimeout = function (signal, timeoutMs) {
-    return new Promise(function (resolve) {
-        if (signal.aborted) {
-            resolve();
-            return;
-        }
-        var timer = setTimeout(function () {
-            signal.removeEventListener("abort", onAbort);
-            resolve();
-        }, timeoutMs);
-        var onAbort = function () {
-            clearTimeout(timer);
-            resolve();
-        };
-        signal.addEventListener("abort", onAbort, { once: true });
-    });
-};
-var readClaimLock = function (job) {
+const buildLockExpiry = (lockTtlMs) => luxon_1.DateTime.utc().minus({ milliseconds: lockTtlMs }).toJSDate();
+const waitForAbortOrTimeout = (signal, timeoutMs) => new Promise((resolve) => {
+    if (signal.aborted) {
+        resolve();
+        return;
+    }
+    const timer = setTimeout(() => {
+        signal.removeEventListener("abort", onAbort);
+        resolve();
+    }, timeoutMs);
+    const onAbort = () => {
+        clearTimeout(timer);
+        resolve();
+    };
+    signal.addEventListener("abort", onAbort, { once: true });
+});
+const readClaimLock = (job) => {
     if (!job.lockedAt || !job.lockedBy) {
-        throw new Error("Job ".concat(job._id.toString(), " is missing claim lock metadata"));
+        throw new Error(`Job ${job._id.toString()} is missing claim lock metadata`);
     }
     return {
         lockedAt: job.lockedAt,
         lockedBy: job.lockedBy,
     };
 };
-var MongoJobRunner = /** @class */ (function () {
-    function MongoJobRunner() {
-        this.id = "mongo";
-        this.pollIntervalMs = DEFAULT_POLL_INTERVAL_MS;
-        this.running = false;
+class MongoJobRunner {
+    id = "mongo";
+    pollIntervalMs = DEFAULT_POLL_INTERVAL_MS;
+    running = false;
+    stopResolve;
+    async enqueue(_job) {
+        // Mongo persistence happens before dispatch; the poller picks up the row.
     }
-    MongoJobRunner.prototype.enqueue = function (_job) {
-        return __awaiter(this, void 0, void 0, function () {
-            return __generator(this, function (_a) {
-                return [2 /*return*/];
-            });
-        });
-    };
-    MongoJobRunner.prototype.start = function (options) {
-        return __awaiter(this, void 0, void 0, function () {
-            var processedAny, processed;
-            var _a, _b;
-            return __generator(this, function (_c) {
-                switch (_c.label) {
-                    case 0:
-                        this.pollIntervalMs = (_a = options.pollIntervalMs) !== null && _a !== void 0 ? _a : DEFAULT_POLL_INTERVAL_MS;
-                        this.running = true;
-                        _c.label = 1;
-                    case 1:
-                        _c.trys.push([1, , 11, 12]);
-                        _c.label = 2;
-                    case 2:
-                        if (!!options.signal.aborted) return [3 /*break*/, 10];
-                        return [4 /*yield*/, options.jobs.reconcileSchedulesIfDirty()];
-                    case 3:
-                        _c.sent();
-                        return [4 /*yield*/, options.jobs.tickSchedules()];
-                    case 4:
-                        _c.sent();
-                        processedAny = false;
-                        _c.label = 5;
-                    case 5:
-                        if (!!options.signal.aborted) return [3 /*break*/, 7];
-                        return [4 /*yield*/, this.processNext(options)];
-                    case 6:
-                        processed = _c.sent();
-                        if (!processed) {
-                            return [3 /*break*/, 7];
-                        }
-                        processedAny = true;
-                        return [3 /*break*/, 5];
-                    case 7:
-                        if (options.signal.aborted) {
-                            return [3 /*break*/, 10];
-                        }
-                        if (!!processedAny) return [3 /*break*/, 9];
-                        return [4 /*yield*/, waitForAbortOrTimeout(options.signal, this.pollIntervalMs)];
-                    case 8:
-                        _c.sent();
-                        _c.label = 9;
-                    case 9: return [3 /*break*/, 2];
-                    case 10: return [3 /*break*/, 12];
-                    case 11:
-                        this.running = false;
-                        (_b = this.stopResolve) === null || _b === void 0 ? void 0 : _b.call(this);
-                        this.stopResolve = undefined;
-                        return [7 /*endfinally*/];
-                    case 12: return [2 /*return*/];
+    async start(options) {
+        this.pollIntervalMs = options.pollIntervalMs ?? DEFAULT_POLL_INTERVAL_MS;
+        this.running = true;
+        try {
+            while (!options.signal.aborted) {
+                await options.jobs.reconcileSchedulesIfDirty();
+                await options.jobs.tickSchedules();
+                let processedAny = false;
+                while (!options.signal.aborted) {
+                    const processed = await this.processNext(options);
+                    if (!processed) {
+                        break;
+                    }
+                    processedAny = true;
                 }
-            });
-        });
-    };
-    MongoJobRunner.prototype.stop = function () {
-        return __awaiter(this, void 0, void 0, function () {
-            var _this = this;
-            return __generator(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        if (!this.running) {
-                            return [2 /*return*/];
-                        }
-                        return [4 /*yield*/, new Promise(function (resolve) {
-                                _this.stopResolve = resolve;
-                            })];
-                    case 1:
-                        _a.sent();
-                        return [2 /*return*/];
+                if (options.signal.aborted) {
+                    break;
                 }
-            });
-        });
-    };
-    MongoJobRunner.prototype.processNext = function (options) {
-        return __awaiter(this, void 0, void 0, function () {
-            var now, lockExpiry, claim, claimed;
-            return __generator(this, function (_a) {
-                switch (_a.label) {
-                    case 0:
-                        now = luxon_1.DateTime.utc().toJSDate();
-                        lockExpiry = buildLockExpiry(options.jobs.getLockTtlMs());
-                        claim = (0, claimLock_1.createClaimLock)(now);
-                        return [4 /*yield*/, job_1.Job.findOneAndUpdate({
-                                $or: [{ lockedAt: { $exists: false } }, { lockedAt: null }, { lockedAt: { $lte: lockExpiry } }],
-                                runAt: { $lte: now },
-                                status: { $in: ["pending", "running", "scheduled"] },
-                            }, {
-                                $set: {
-                                    lockedAt: claim.lockedAt,
-                                    lockedBy: claim.lockedBy,
-                                    status: "running",
-                                },
-                            }, { returnDocument: "after", sort: { runAt: 1 } })];
-                    case 1:
-                        claimed = _a.sent();
-                        if (!claimed) {
-                            return [2 /*return*/, false];
-                        }
-                        return [4 /*yield*/, (0, jobExecutor_1.runClaimedJob)({
-                                claim: readClaimLock(claimed),
-                                host: options.jobs,
-                                job: claimed,
-                                signal: options.signal,
-                            })];
-                    case 2:
-                        _a.sent();
-                        return [2 /*return*/, true];
+                if (!processedAny) {
+                    await waitForAbortOrTimeout(options.signal, this.pollIntervalMs);
                 }
-            });
+            }
+        }
+        finally {
+            this.running = false;
+            this.stopResolve?.();
+            this.stopResolve = undefined;
+        }
+    }
+    async stop() {
+        if (!this.running) {
+            return;
+        }
+        await new Promise((resolve) => {
+            this.stopResolve = resolve;
         });
-    };
-    return MongoJobRunner;
-}());
+    }
+    async processNext(options) {
+        const now = luxon_1.DateTime.utc().toJSDate();
+        const lockExpiry = buildLockExpiry(options.jobs.getLockTtlMs());
+        const claim = (0, claimLock_1.createClaimLock)(now);
+        const claimed = await job_1.Job.findOneAndUpdate({
+            $or: [{ lockedAt: { $exists: false } }, { lockedAt: null }, { lockedAt: { $lte: lockExpiry } }],
+            runAt: { $lte: now },
+            status: { $in: ["pending", "running", "scheduled"] },
+        }, {
+            $set: {
+                lockedAt: claim.lockedAt,
+                lockedBy: claim.lockedBy,
+                status: "running",
+            },
+        }, { returnDocument: "after", sort: { runAt: 1 } });
+        if (!claimed) {
+            return false;
+        }
+        await (0, jobExecutor_1.runClaimedJob)({
+            claim: readClaimLock(claimed),
+            host: options.jobs,
+            job: claimed,
+            signal: options.signal,
+        });
+        return true;
+    }
+}
 exports.MongoJobRunner = MongoJobRunner;