@clipboard-health/mongo-jobs 0.1.0

package/src/lib/internal/worker.js

@@ -0,0 +1,342 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Worker = void 0;
+const tracing_1 = require("../tracing");
+const duplicateRunningError_1 = require("./duplicateRunningError");
+const mongoDuplicate_1 = require("./mongoDuplicate");
+const fairQueueConsumer_1 = require("./worker/fairQueueConsumer");
+const MILLIS_IN_SECOND = 1000;
+const DEFAULT_LOCK_TIMEOUT = 600 * MILLIS_IN_SECOND; // 10 minutes
+const DEFAULT_MAX_CONCURRENCY = 10;
+const DEFAULT_NEW_JOB_CHECK_WAIT = 10 * MILLIS_IN_SECOND; // 10 seconds
+const DEFAULT_USE_CHANGE_STREAM = true;
+const DEFAULT_UNLOCK_JOBS_INTERVAL = 60 * MILLIS_IN_SECOND; // 1 minute
+const DEFAULT_REFRESH_QUEUES_INTERVAL = 30 * MILLIS_IN_SECOND; // 30 seconds
+const DUPLICATE_RESCHEDULE_TIME = 5 * MILLIS_IN_SECOND; // 5 seconds
+const DEFAULT_MAX_ATTEMPTS = 10;
+const GRACEFUL_SHUTDOWN_WAIT = 30 * MILLIS_IN_SECOND; // 30 seconds
+function errorMessage(error) {
+    if (error instanceof Error) {
+        return error.message;
+    }
+    if (typeof error === "string") {
+        return error;
+    }
+    return JSON.stringify(error);
+}
+function errorToString(error) {
+    if (error instanceof Error) {
+        return error.stack ?? error.message;
+    }
+    if (typeof error === "string") {
+        return error;
+    }
+    return JSON.stringify(error);
+}
+function fromNow(millis) {
+    return new Date(Date.now() + millis);
+}
+async function delay(time) {
+    await new Promise((resolve) => {
+        setTimeout(resolve, time);
+    });
+}
+function removeItemFromArray(array, item) {
+    const index = array.indexOf(item);
+    if (index > -1) {
+        array.splice(index, 1);
+    }
+}
+class Worker {
+    stopped = true;
+    lockTimeoutMS;
+    maxConcurrency;
+    newJobCheckWaitMS;
+    useChangeStream;
+    unlockJobsIntervalMS;
+    refreshQueuesIntervalMS;
+    jobsRepo;
+    runningJobs = new Map();
+    metrics;
+    registry;
+    cron;
+    logger;
+    queueConsumer;
+    unlockJobsInterval;
+    workerLoopInterval;
+    fetchingJobs = false;
+    newJobsWatched = false;
+    nextAvailableJobCheck = new Date();
+    constructor(options) {
+        this.metrics = options.metrics;
+        this.lockTimeoutMS = options.lockTimeoutMS ?? DEFAULT_LOCK_TIMEOUT;
+        this.maxConcurrency = options.maxConcurrency ?? DEFAULT_MAX_CONCURRENCY;
+        this.newJobCheckWaitMS = options.newJobCheckWaitMS ?? DEFAULT_NEW_JOB_CHECK_WAIT;
+        this.useChangeStream = options.useChangeStream ?? DEFAULT_USE_CHANGE_STREAM;
+        this.unlockJobsIntervalMS = options.unlockJobsIntervalMS ?? DEFAULT_UNLOCK_JOBS_INTERVAL;
+        this.refreshQueuesIntervalMS =
+            options.refreshQueuesIntervalMS ?? DEFAULT_REFRESH_QUEUES_INTERVAL;
+        this.jobsRepo = options.jobsRepo;
+        this.registry = options.registry;
+        this.cron = options.cron;
+        this.logger = options.logger;
+        const queues = this.registry.getQueuesForGroups(options.groups);
+        if (options.exclude) {
+            for (const excludedQueue of options.exclude) {
+                removeItemFromArray(queues, excludedQueue);
+            }
+        }
+        this.queueConsumer = new fairQueueConsumer_1.FairQueueConsumer(queues, options.jobsRepo);
+    }
+    async start() {
+        this.stopped = false;
+        this.logger?.info("Starting Background Processing");
+        this.unlockJobsInterval = setInterval(() => {
+            void this.unlockStuckJobs();
+        }, this.unlockJobsIntervalMS);
+        this.workerLoopInterval = setInterval(() => {
+            void this.workerLoop();
+        }, this.newJobCheckWaitMS);
+        await this.queueConsumer.start({
+            useChangeStream: this.useChangeStream,
+            refreshQueuesIntervalMS: this.refreshQueuesIntervalMS,
+        });
+        void this.unlockStuckJobs();
+        void this.workerLoop();
+    }
+    async acquireNextJob() {
+        try {
+            return await this.queueConsumer.acquireNextJob();
+        }
+        catch (error) {
+            this.logger?.error(`Error while getting job: ${errorMessage(error)}`);
+            return undefined;
+        }
+    }
+    async runJobHandler(job) {
+        this.reportExecutionDelay(job);
+        await (0, tracing_1.withConsumerTrace)(job, async () => {
+            const { handlerName } = job;
+            const { handler } = this.registry.getRegisteredHandler(handlerName);
+            const { data } = job;
+            await handler.perform(data, job);
+        });
+    }
+    async stop(waitTime) {
+        this.logger?.info("Background Jobs: Stopping");
+        this.stopped = true;
+        clearInterval(this.workerLoopInterval);
+        clearInterval(this.unlockJobsInterval);
+        // Stop metrics reporting first to prevent further database operations
+        this.metrics.stopReporting();
+        await this.queueConsumer.stop();
+        const shutdownWaitPromise = delay(waitTime ?? GRACEFUL_SHUTDOWN_WAIT);
+        const jobsWaitPromise = Promise.all(this.runningJobs.values());
+        await Promise.race([shutdownWaitPromise, jobsWaitPromise]);
+        if (this.runningJobs.size > 0) {
+            const runningJobsIds = [...this.runningJobs.keys()].join(", ");
+            this.logger?.error(`Following jobs didn't end on time during shutdown: ${runningJobsIds}`);
+        }
+        this.logger?.info("Background Jobs: Stopped");
+    }
+    async markJobCompleted(job) {
+        await this.jobsRepo.deleteJobs([job._id]);
+    }
+    getConsumedQueues() {
+        return this.queueConsumer.getConsumedQueues();
+    }
+    async updateUniqueKey(job) {
+        try {
+            const uniqueOptions = job.options?.unique;
+            if (!uniqueOptions) {
+                return;
+            }
+            if (job.uniqueKey === uniqueOptions.runningKey) {
+                return;
+            }
+            const operation = uniqueOptions.runningKey
+                ? { $set: { uniqueKey: uniqueOptions.runningKey } }
+                : { $unset: { uniqueKey: "" } };
+            await this.jobsRepo.updateOne(job._id, operation);
+        }
+        catch (error) {
+            throw (0, mongoDuplicate_1.isMongoDuplicateError)(error) ? new duplicateRunningError_1.DuplicateRunningError(error.message) : error;
+        }
+    }
+    async workerLoop() {
+        if (this.stopped ||
+            this.fetchingJobs ||
+            this.runningJobs.size >= this.maxConcurrency ||
+            new Date() < this.nextAvailableJobCheck) {
+            return;
+        }
+        while (this.runningJobs.size < this.maxConcurrency) {
+            this.fetchingJobs = true;
+            /* I know that fetching jobs one by one here is terrible,
+               but unfortunately Mongo doesn't support updateMany with limit.
+               So from all bad solutions this seemed the least terrible. */
+            // eslint-disable-next-line no-await-in-loop
+            const job = await this.acquireNextJob();
+            this.fetchingJobs = false;
+            if (!job) {
+                this.scheduleWorkerLoopAfterWait();
+                break;
+            }
+            void this.performJob(job);
+        }
+    }
+    handleNewJobEvent() {
+        this.newJobsWatched = false;
+        this.nextAvailableJobCheck = new Date();
+        void this.workerLoop();
+    }
+    scheduleWorkerLoopAfterWait() {
+        this.nextAvailableJobCheck = fromNow(this.newJobCheckWaitMS);
+        if (!this.newJobsWatched) {
+            this.newJobsWatched = true;
+            this.queueConsumer.addEventListener("newJob", () => {
+                this.handleNewJobEvent();
+            }, { once: true });
+        }
+        setTimeout(() => {
+            void this.workerLoop();
+        }, this.newJobCheckWaitMS);
+    }
+    async performJob(job) {
+        let jobPromiseResolve;
+        const jobPromise = new Promise((resolve) => {
+            jobPromiseResolve = resolve;
+        });
+        await (0, tracing_1.withInternalsTrace)("performJob", async () => {
+            try {
+                this.runningJobs.set(job._id, jobPromise);
+                await this.cron.maybeScheduleNextJob(job);
+                await this.updateUniqueKey(job);
+                await this.runJobHandler(job);
+                await this.markJobCompleted(job);
+            }
+            catch (error) {
+                if (error instanceof duplicateRunningError_1.DuplicateRunningError) {
+                    void this.duplicateReschedule(job);
+                }
+                else {
+                    this.logger?.error(`Error while performing job ${job._id.toString()}: ${errorMessage(error)}`);
+                    void this.handleJobError(job, errorToString(error));
+                }
+            }
+            finally {
+                this.runningJobs.delete(job._id);
+                jobPromiseResolve();
+                void this.workerLoop();
+            }
+        });
+    }
+    async handleJobError(job, error) {
+        try {
+            const attemptsCount = job.attemptsCount + 1;
+            const maxAttempts = this.getMaxAttemptsForJob(job);
+            // eslint-disable-next-line unicorn/prefer-ternary
+            if (attemptsCount < maxAttempts) {
+                await this.scheduleRetry(job, attemptsCount, error);
+            }
+            else {
+                await this.markJobFailed(job, attemptsCount, error);
+            }
+        }
+        catch {
+            this.logger?.error(`Error while marking job as failed. Job id: ${job._id.toString()}, error: ${errorMessage(error)}`);
+        }
+    }
+    async duplicateReschedule(job) {
+        await this.jobsRepo.updateOne(job._id, {
+            $set: {
+                nextRunAt: fromNow(DUPLICATE_RESCHEDULE_TIME),
+            },
+            $unset: {
+                lockedAt: "",
+            },
+        });
+    }
+    async unlockStuckJobs() {
+        const lockedAtThreshold = fromNow(-this.lockTimeoutMS);
+        for (;;) {
+            // eslint-disable-next-line no-await-in-loop
+            const job = await this.jobsRepo.unlockFirstExpiredJob(lockedAtThreshold);
+            if (!job) {
+                return;
+            }
+            if (job.queue) {
+                this.metrics.increment(job.queue, "expired");
+            }
+            this.logger?.error("Background job expired", this.logContext(job));
+        }
+    }
+    reportExecutionDelay(job) {
+        if (!job.nextRunAt) {
+            return;
+        }
+        if (job.queue) {
+            this.metrics.timing(job.queue, "delay", job.nextRunAt);
+        }
+    }
+    logContext(job) {
+        return {
+            jobId: job._id.toString(),
+            queue: job.queue,
+            handlerName: job.handlerName,
+            attemptsCount: job.attemptsCount,
+            createdAt: job.createdAt,
+            updatedAt: job.updatedAt,
+            nextRunAt: job.nextRunAt,
+            lockedAt: job.lockedAt,
+        };
+    }
+    getMaxAttemptsForJob(job) {
+        try {
+            const { handlerName } = job;
+            const { handler } = this.registry.getRegisteredHandler(handlerName);
+            return handler.maxAttempts ?? DEFAULT_MAX_ATTEMPTS;
+        }
+        catch {
+            return DEFAULT_MAX_ATTEMPTS;
+        }
+    }
+    async markJobFailed(job, attemptsCount, error) {
+        this.logger?.error(`Background Job Failed: ${job.handlerName}`, {
+            ...this.logContext(job),
+            error,
+        });
+        await this.jobsRepo.updateOne(job._id, {
+            $set: {
+                lastError: error,
+                failedAt: new Date(),
+                attemptsCount,
+                originalQueue: job.queue,
+            },
+            $unset: {
+                queue: "",
+                nextRunAt: "",
+                uniqueKey: "",
+            },
+        });
+    }
+    async scheduleRetry(job, attemptsCount, error) {
+        const exponentialBackoffTime = 2 ** attemptsCount * MILLIS_IN_SECOND;
+        const nextRunAt = fromNow(exponentialBackoffTime);
+        if (job.queue) {
+            this.metrics.increment(job.queue, "retry");
+        }
+        await this.jobsRepo.updateOne(job._id, {
+            $set: {
+                lastError: error.toString(),
+                nextRunAt,
+                attemptsCount,
+            },
+            $unset: {
+                lockedAt: "",
+            },
+        });
+    }
+}
+exports.Worker = Worker;
+//# sourceMappingURL=worker.js.map

package/src/lib/internal/worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.js","sourceRoot":"","sources":["../../../../../../packages/mongo-jobs/src/lib/internal/worker.ts"],"names":[],"mappings":";;;AAGA,wCAAmE;AAEnE,mEAAgE;AAIhE,qDAAyD;AAEzD,kEAA+D;AAsB/D,MAAM,gBAAgB,GAAG,IAAI,CAAC;AAC9B,MAAM,oBAAoB,GAAG,GAAG,GAAG,gBAAgB,CAAC,CAAC,aAAa;AAClE,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,MAAM,0BAA0B,GAAG,EAAE,GAAG,gBAAgB,CAAC,CAAC,aAAa;AACvE,MAAM,yBAAyB,GAAG,IAAI,CAAC;AACvC,MAAM,4BAA4B,GAAG,EAAE,GAAG,gBAAgB,CAAC,CAAC,WAAW;AACvE,MAAM,+BAA+B,GAAG,EAAE,GAAG,gBAAgB,CAAC,CAAC,aAAa;AAC5E,MAAM,yBAAyB,GAAG,CAAC,GAAG,gBAAgB,CAAC,CAAC,YAAY;AACpE,MAAM,oBAAoB,GAAG,EAAE,CAAC;AAChC,MAAM,sBAAsB,GAAG,EAAE,GAAG,gBAAgB,CAAC,CAAC,aAAa;AAEnE,SAAS,YAAY,CAAC,KAAc;IAClC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,OAAO,CAAC;IACvB,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,OAAO,CAAC;IACtC,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,OAAO,CAAC,MAAc;IAC7B,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,CAAC;AACvC,CAAC;AAED,KAAK,UAAU,KAAK,CAAC,IAAY;IAC/B,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QAC5B,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;IAC5B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAI,KAAU,EAAE,IAAO;IACjD,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,KAAK,GAAG,CAAC,CAAC,EAAE,CAAC;QACf,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IACzB,CAAC;AACH,CAAC;AAED,MAAa,MAAM;IACV,OAAO,GAAG,IAAI,CAAC;IAEL,aAAa,CAAS;IACtB,cAAc,CAAS;IACvB,iBAAiB,CAAS;IAC1B,eAAe,CAAU;IAEzB,oBAAoB,CAAS;IAC7B,uBAAuB,CAAS;IAChC,QAAQ,CAAiB;IACzB,WAAW,GAAG,IAAI,GAAG,EAA6C,CAAC;IACnE,OAAO,CAAU;IACjB,QAAQ,CAAW;IACnB,IAAI,CAAO;IACX,MAAM,CAAqB;IAC3B,aAAa,CAAgB;IAEtC,kBAAkB,CAAkB;IACpC,kBAAkB,CAAkB;IACpC,YAAY,GAAG,KAAK,CAAC;IACrB,cAAc,GAAG,KAAK,CAAC;IACvB,qBAAqB,GAAG,IAAI,IAAI,EAAE,CAAC;IAE3C,YAAY,OAA2B;QACrC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,oBAAoB,CAAC;QACnE,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,uBAAuB,CAAC;QACxE,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,iBAAiB,IAAI,0BAA0B,CAAC;QACjF,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,eAAe,IAAI,yBAAyB,CAAC;QAC5E,IAAI,CAAC,oBAAoB,GAAG,OAAO,CAAC,oBAAoB,IAAI,4BAA4B,CAAC;QACzF,IAAI,CAAC,uBAAuB;YAC1B,OAAO,CAAC,uBAAuB,IAAI,+BAA+B,CAAC;QACrE,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QACzB,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAE7B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,kBAAkB,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAEhE,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,KAAK,MAAM,aAAa,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC5C,mBAAmB,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;YAC7C,CAAC;QACH,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAiB,CAAC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvE,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,gCAAgC,CAAC,CAAC;QAEpD,IAAI,CAAC,kBAAkB,GAAG,WAAW,CAAC,GAAG,EAAE;YACzC,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;QAC9B,CAAC,EAAE,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAE9B,IAAI,CAAC,kBAAkB,GAAG,WAAW,CAAC,GAAG,EAAE;YACzC,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;QACzB,CAAC,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE3B,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YAC7B,eAAe,EAAE,IAAI,CAAC,eAAe;YACrC,uBAAuB,EAAE,IAAI,CAAC,uBAAuB;SACtD,CAAC,CAAC;QAEH,KAAK,IAAI,CAAC,eAAe,EAAE,CAAC;QAC5B,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;IACzB,CAAC;IAEM,KAAK,CAAC,cAAc;QACzB,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,aAAa,CAAC,cAAc,EAAE,CAAC;QACnD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,4BAA4B,YAAY,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YACtE,OAAO,SAAS,CAAC;QACnB,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,aAAa,CAAC,GAA+B;QACxD,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAE/B,MAAM,IAAA,2BAAiB,EAAC,GAAG,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,EAAE,WAAW,EAAE,GAAG,GAAG,CAAC;YAC5B,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;YACpE,MAAM,EAAE,IAAI,EAAE,GAAG,GAAG,CAAC;YACrB,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,QAAiB;QACjC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,aAAa,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACvC,aAAa,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAEvC,sEAAsE;QACtE,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC;QAE7B,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;QAEhC,MAAM,mBAAmB,GAAG,KAAK,CAAC,QAAQ,IAAI,sBAAsB,CAAC,CAAC;QACtE,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QAE/D,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,mBAAmB,EAAE,eAAe,CAAC,CAAC,CAAC;QAE3D,IAAI,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC9B,MAAM,cAAc,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/D,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,sDAAsD,cAAc,EAAE,CAAC,CAAC;QAC7F,CAAC;QAED,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,0BAA0B,CAAC,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,GAA+B;QAC3D,MAAM,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5C,CAAC;IAEM,iBAAiB;QACtB,OAAO,IAAI,CAAC,aAAa,CAAC,iBAAiB,EAAE,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,eAAe,CAAC,GAA+B;QAC1D,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC;YAE1C,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,IAAI,GAAG,CAAC,SAAS,KAAK,aAAa,CAAC,UAAU,EAAE,CAAC;gBAC/C,OAAO;YACT,CAAC;YAED,MAAM,SAAS,GAAG,aAAa,CAAC,UAAU;gBACxC,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,SAAS,EAAE,aAAa,CAAC,UAAU,EAAE,EAAE;gBACnD,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,CAAC;YAElC,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACpD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAA,sCAAqB,EAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,6CAAqB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QACxF,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,UAAU;QACtB,IACE,IAAI,CAAC,OAAO;YACZ,IAAI,CAAC,YAAY;YACjB,IAAI,CAAC,WAAW,CAAC,IAAI,IAAI,IAAI,CAAC,cAAc;YAC5C,IAAI,IAAI,EAAE,GAAG,IAAI,CAAC,qBAAqB,EACvC,CAAC;YACD,OAAO;QACT,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,CAAC,IAAI,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;YACnD,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;YACzB;;2EAE+D;YAE/D,4CAA4C;YAC5C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;YACxC,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;YAE1B,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,IAAI,CAAC,2BAA2B,EAAE,CAAC;gBACnC,MAAM;YACR,CAAC;YAED,KAAK,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC;QAC5B,IAAI,CAAC,qBAAqB,GAAG,IAAI,IAAI,EAAE,CAAC;QACxC,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;IACzB,CAAC;IAEO,2BAA2B;QACjC,IAAI,CAAC,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAE7D,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YACzB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;YAE3B,IAAI,CAAC,aAAa,CAAC,gBAAgB,CACjC,QAAQ,EACR,GAAG,EAAE;gBACH,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC,EACD,EAAE,IAAI,EAAE,IAAI,EAAE,CACf,CAAC;QACJ,CAAC;QAED,UAAU,CAAC,GAAG,EAAE;YACd,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;QACzB,CAAC,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,GAA+B;QACtD,IAAI,iBAA4C,CAAC;QACjD,MAAM,UAAU,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YACzC,iBAAiB,GAAG,OAAO,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,IAAA,4BAAkB,EAAC,YAAY,EAAE,KAAK,IAAI,EAAE;YAChD,IAAI,CAAC;gBACH,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;gBAC1C,MAAM,IAAI,CAAC,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;gBAC1C,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;gBAChC,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;gBAC9B,MAAM,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;YACnC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,6CAAqB,EAAE,CAAC;oBAC3C,KAAK,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC;gBACrC,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,MAAM,EAAE,KAAK,CAChB,8BAA8B,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,KAAK,YAAY,CAAC,KAAK,CAAC,EAAE,CAC3E,CAAC;oBACF,KAAK,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;gBACjC,iBAAiB,EAAE,CAAC;gBACpB,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;YACzB,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,cAAc,CAAC,GAA+B,EAAE,KAAa;QACzE,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,GAAG,CAAC,aAAa,GAAG,CAAC,CAAC;YAC5C,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,CAAC;YAEnD,kDAAkD;YAClD,IAAI,aAAa,GAAG,WAAW,EAAE,CAAC;gBAChC,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;YACtD,CAAC;iBAAM,CAAC;gBACN,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,aAAa,EAAE,KAAK,CAAC,CAAC;YACtD,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,CAAC,MAAM,EAAE,KAAK,CAChB,8CAA8C,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,YAAY,YAAY,CACtF,KAAK,CACN,EAAE,CACJ,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,GAA+B;QAC/D,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE;YACrC,IAAI,EAAE;gBACJ,SAAS,EAAE,OAAO,CAAC,yBAAyB,CAAC;aAC9C;YACD,MAAM,EAAE;gBACN,QAAQ,EAAE,EAAE;aACb;SACF,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,eAAe;QAC3B,MAAM,iBAAiB,GAAG,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAEvD,SAAS,CAAC;YACR,4CAA4C;YAC5C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,qBAAqB,CAAC,iBAAiB,CAAC,CAAC;YAEzE,IAAI,CAAC,GAAG,EAAE,CAAC;gBACT,OAAO;YACT,CAAC;YAED,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACd,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;YAC/C,CAAC;YAED,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,wBAAwB,EAAE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAEO,oBAAoB,CAAC,GAA+B;QAC1D,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC;YACnB,OAAO;QACT,CAAC;QAED,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,GAA+B;QAChD,OAAO;YACL,KAAK,EAAE,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE;YACzB,KAAK,EAAE,GAAG,CAAC,KAAK;YAChB,WAAW,EAAE,GAAG,CAAC,WAAW;YAC5B,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,SAAS,EAAE,GAAG,CAAC,SAAS;YACxB,QAAQ,EAAE,GAAG,CAAC,QAAQ;SACvB,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,GAA+B;QAC1D,IAAI,CAAC;YACH,MAAM,EAAE,WAAW,EAAE,GAAG,GAAG,CAAC;YAC5B,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;YACpE,OAAO,OAAO,CAAC,WAAW,IAAI,oBAAoB,CAAC;QACrD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,oBAAoB,CAAC;QAC9B,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,GAA+B,EAC/B,aAAqB,EACrB,KAAa;QAEb,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,0BAA0B,GAAG,CAAC,WAAW,EAAE,EAAE;YAC9D,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YACvB,KAAK;SACN,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE;YACrC,IAAI,EAAE;gBACJ,SAAS,EAAE,KAAK;gBAChB,QAAQ,EAAE,IAAI,IAAI,EAAE;gBACpB,aAAa;gBACb,aAAa,EAAE,GAAG,CAAC,KAAK;aACzB;YACD,MAAM,EAAE;gBACN,KAAK,EAAE,EAAE;gBACT,SAAS,EAAE,EAAE;gBACb,SAAS,EAAE,EAAE;aACd;SACF,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,aAAa,CACzB,GAA+B,EAC/B,aAAqB,EACrB,KAAa;QAEb,MAAM,sBAAsB,GAAG,CAAC,IAAI,aAAa,GAAG,gBAAgB,CAAC;QACrE,MAAM,SAAS,GAAG,OAAO,CAAC,sBAAsB,CAAC,CAAC;QAElD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,GAAG,CAAC,GAAG,EAAE;YACrC,IAAI,EAAE;gBACJ,SAAS,EAAE,KAAK,CAAC,QAAQ,EAAE;gBAC3B,SAAS;gBACT,aAAa;aACd;YACD,MAAM,EAAE;gBACN,QAAQ,EAAE,EAAE;aACb;SACF,CAAC,CAAC;IACL,CAAC;CACF;AAxWD,wBAwWC"}

package/src/lib/job.d.ts

@@ -0,0 +1,37 @@
+import mongoose from "mongoose";
+import type { TraceHeaders } from "./tracing";
+export interface JobUniqueOptions {
+    enqueuedKey: string | undefined;
+    runningKey: string | undefined;
+}
+interface JobOptions {
+    unique?: JobUniqueOptions;
+}
+export interface BackgroundJobType<T> {
+    _id: mongoose.Types.ObjectId;
+    createdAt: Date;
+    updatedAt: Date;
+    queue: string | undefined;
+    handlerName: string;
+    data: T & TraceHeaders;
+    nextRunAt: Date | undefined;
+    lockedAt: Date | undefined;
+    failedAt: Date | undefined;
+    attemptsCount: number;
+    lastError: string | undefined;
+    options: JobOptions | undefined;
+    uniqueKey: string | undefined;
+    scheduleName: string | undefined;
+    originalQueue: string | undefined;
+}
+export declare const BackgroundJobSchemaName = "BackgroundJob";
+declare const BackgroundJobSchema: mongoose.Schema<BackgroundJobType<unknown>, mongoose.Model<BackgroundJobType<unknown>, any, any, any, mongoose.Document<unknown, any, BackgroundJobType<unknown>, any, {}> & BackgroundJobType<unknown> & Required<{
+    _id: mongoose.Types.ObjectId;
+}> & {
+    __v: number;
+}, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, BackgroundJobType<unknown>, mongoose.Document<unknown, {}, mongoose.FlatRecord<BackgroundJobType<unknown>>, {}, mongoose.ResolveSchemaOptions<mongoose.DefaultSchemaOptions>> & mongoose.FlatRecord<BackgroundJobType<unknown>> & Required<{
+    _id: mongoose.Types.ObjectId;
+}> & {
+    __v: number;
+}>;
+export { BackgroundJobSchema };

package/src/lib/job.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BackgroundJobSchema = exports.BackgroundJobSchemaName = void 0;
+const tslib_1 = require("tslib");
+const mongoose_1 = tslib_1.__importDefault(require("mongoose"));
+exports.BackgroundJobSchemaName = "BackgroundJob";
+const BackgroundJobSchema = new mongoose_1.default.Schema({
+    queue: String,
+    handlerName: String,
+    data: {},
+    nextRunAt: Date,
+    lockedAt: Date,
+    failedAt: Date,
+    attemptsCount: { type: Number, default: 0 },
+    lastError: String,
+    options: {},
+    uniqueKey: String,
+    scheduleName: String,
+    originalQueue: String,
+}, {
+    timestamps: true,
+});
+exports.BackgroundJobSchema = BackgroundJobSchema;
+BackgroundJobSchema.index({ queue: 1, lockedAt: 1, nextRunAt: 1 });
+BackgroundJobSchema.index({ failedAt: 1, lockedAt: 1 });
+BackgroundJobSchema.index({ originalQueue: 1, failedAt: 1 });
+BackgroundJobSchema.index({ uniqueKey: 1 }, { unique: true, sparse: true });
+BackgroundJobSchema.index({ scheduleName: 1 });
+//# sourceMappingURL=job.js.map

package/src/lib/job.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"job.js","sourceRoot":"","sources":["../../../../../packages/mongo-jobs/src/lib/job.ts"],"names":[],"mappings":";;;;AAAA,gEAAgC;AAoCnB,QAAA,uBAAuB,GAAG,eAAe,CAAC;AAEvD,MAAM,mBAAmB,GAAG,IAAI,kBAAQ,CAAC,MAAM,CAC7C;IACE,KAAK,EAAE,MAAM;IACb,WAAW,EAAE,MAAM;IACnB,IAAI,EAAE,EAAE;IACR,SAAS,EAAE,IAAI;IACf,QAAQ,EAAE,IAAI;IACd,QAAQ,EAAE,IAAI;IACd,aAAa,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE;IAC3C,SAAS,EAAE,MAAM;IACjB,OAAO,EAAE,EAAE;IACX,SAAS,EAAE,MAAM;IACjB,YAAY,EAAE,MAAM;IACpB,aAAa,EAAE,MAAM;CACtB,EACD;IACE,UAAU,EAAE,IAAI;CACjB,CACF,CAAC;AAQO,kDAAmB;AAN5B,mBAAmB,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;AACnE,mBAAmB,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;AACxD,mBAAmB,CAAC,KAAK,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;AAC7D,mBAAmB,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;AAC5E,mBAAmB,CAAC,KAAK,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,CAAC"}

package/src/lib/schedule.d.ts

@@ -0,0 +1,21 @@
+import mongoose from "mongoose";
+export interface ScheduleType<T> {
+    _id: mongoose.Types.ObjectId;
+    name: string;
+    cronExpression: string;
+    timeZone: string;
+    handlerName: string;
+    queue: string;
+    data: T;
+}
+export declare const ScheduleSchemaName = "BackgroundJobSchedule";
+declare const ScheduleSchema: mongoose.Schema<ScheduleType<unknown>, mongoose.Model<ScheduleType<unknown>, any, any, any, mongoose.Document<unknown, any, ScheduleType<unknown>, any, {}> & ScheduleType<unknown> & Required<{
+    _id: mongoose.Types.ObjectId;
+}> & {
+    __v: number;
+}, any>, {}, {}, {}, {}, mongoose.DefaultSchemaOptions, ScheduleType<unknown>, mongoose.Document<unknown, {}, mongoose.FlatRecord<ScheduleType<unknown>>, {}, mongoose.ResolveSchemaOptions<mongoose.DefaultSchemaOptions>> & mongoose.FlatRecord<ScheduleType<unknown>> & Required<{
+    _id: mongoose.Types.ObjectId;
+}> & {
+    __v: number;
+}>;
+export { ScheduleSchema };

package/src/lib/schedule.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ScheduleSchema = exports.ScheduleSchemaName = void 0;
+const tslib_1 = require("tslib");
+const mongoose_1 = tslib_1.__importDefault(require("mongoose"));
+exports.ScheduleSchemaName = "BackgroundJobSchedule";
+const ScheduleSchema = new mongoose_1.default.Schema({
+    name: String,
+    cronExpression: String,
+    timeZone: String,
+    handlerName: String,
+    data: {},
+});
+exports.ScheduleSchema = ScheduleSchema;
+ScheduleSchema.index({ name: 1 }, { unique: true });
+//# sourceMappingURL=schedule.js.map

package/src/lib/schedule.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schedule.js","sourceRoot":"","sources":["../../../../../packages/mongo-jobs/src/lib/schedule.ts"],"names":[],"mappings":";;;;AAAA,gEAAgC;AAYnB,QAAA,kBAAkB,GAAG,uBAAuB,CAAC;AAE1D,MAAM,cAAc,GAAG,IAAI,kBAAQ,CAAC,MAAM,CAAwB;IAChE,IAAI,EAAE,MAAM;IACZ,cAAc,EAAE,MAAM;IACtB,QAAQ,EAAE,MAAM;IAChB,WAAW,EAAE,MAAM;IACnB,IAAI,EAAE,EAAE;CACT,CAAC,CAAC;AAIM,wCAAc;AAFvB,cAAc,CAAC,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC"}

package/src/lib/testing.d.ts

@@ -0,0 +1,9 @@
+import type { BackgroundJobsService } from "./backgroundJobs";
+import type { AnyHandlerClass } from "./internal/registry";
+interface DrainOptions {
+    jobsScheduledUntil?: Date;
+    recursive?: boolean;
+}
+export declare function drainQueues(backgroundJobs: BackgroundJobsService, groups: string[], options?: DrainOptions): Promise<void>;
+export declare function drainHandlers(backgroundJobs: BackgroundJobsService, handlers: Array<AnyHandlerClass<unknown> | string>, options?: DrainOptions): Promise<void>;
+export {};

package/src/lib/testing.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.drainQueues = drainQueues;
+exports.drainHandlers = drainHandlers;
+async function executeJobs(backgroundJobs, worker, queues, options = {}) {
+    for (;;) {
+        const pendingJobs = await backgroundJobs.jobModel.find({
+            nextRunAt: { $lte: options.jobsScheduledUntil ?? new Date() },
+            queue: { $in: queues },
+        });
+        if (pendingJobs.length === 0) {
+            return;
+        }
+        for (const job of pendingJobs) {
+            await worker.updateUniqueKey(job);
+            await worker.runJobHandler(job);
+            await worker.markJobCompleted(job);
+        }
+        if (options.recursive === false) {
+            return;
+        }
+    }
+}
+// Drain jobs for all queues in given groups
+async function drainQueues(backgroundJobs, groups, options = {}) {
+    const worker = backgroundJobs.buildWorker(groups, { useChangeStream: false });
+    const consumedQueues = worker.getConsumedQueues();
+    await executeJobs(backgroundJobs, worker, consumedQueues, options);
+}
+// Drain jobs for given handlers
+async function drainHandlers(backgroundJobs, handlers, options = {}) {
+    const queues = handlers.map((handler) => backgroundJobs.registry.getRegisteredHandler(handler).queue);
+    const worker = backgroundJobs.buildWorker([], { useChangeStream: false });
+    await executeJobs(backgroundJobs, worker, queues, options);
+}
+/* eslint-enable no-await-in-loop */
+//# sourceMappingURL=testing.js.map

package/src/lib/testing.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"testing.js","sourceRoot":"","sources":["../../../../../packages/mongo-jobs/src/lib/testing.ts"],"names":[],"mappings":";;AAuCA,kCASC;AAGD,sCAWC;AApDD,KAAK,UAAU,WAAW,CACxB,cAAqC,EACrC,MAAc,EACd,MAAgB,EAChB,UAAwB,EAAE;IAE1B,SAAS,CAAC;QACR,MAAM,WAAW,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC;YACrD,SAAS,EAAE,EAAE,IAAI,EAAE,OAAO,CAAC,kBAAkB,IAAI,IAAI,IAAI,EAAE,EAAE;YAC7D,KAAK,EAAE,EAAE,GAAG,EAAE,MAAM,EAAE;SACvB,CAAC,CAAC;QAEH,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QAED,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC9B,MAAM,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAClC,MAAM,MAAM,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;YAChC,MAAM,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;QACrC,CAAC;QAED,IAAI,OAAO,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;IACH,CAAC;AACH,CAAC;AAED,4CAA4C;AACrC,KAAK,UAAU,WAAW,CAC/B,cAAqC,EACrC,MAAgB,EAChB,UAAwB,EAAE;IAE1B,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAC;IAC9E,MAAM,cAAc,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC;IAElD,MAAM,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,CAAC,CAAC;AACrE,CAAC;AAED,gCAAgC;AACzB,KAAK,UAAU,aAAa,CACjC,cAAqC,EACrC,QAAkD,EAClD,UAAwB,EAAE;IAE1B,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CACzB,CAAC,OAAO,EAAE,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,KAAK,CACzE,CAAC;IACF,MAAM,MAAM,GAAG,cAAc,CAAC,WAAW,CAAC,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,CAAC,CAAC;IAE1E,MAAM,WAAW,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;AAC7D,CAAC;AACD,oCAAoC"}

package/src/lib/tracing.d.ts

@@ -0,0 +1,8 @@
+import type { HandlerInterface } from "./handler";
+import type { BackgroundJobType } from "./job";
+export interface TraceHeaders {
+    _traceHeaders?: Record<string, string>;
+}
+export declare function withProducerTrace<T>(handler: HandlerInterface<T>, data: T, callback: (data: T & TraceHeaders) => Promise<BackgroundJobType<T> | undefined>): Promise<BackgroundJobType<T> | undefined>;
+export declare function withConsumerTrace(job: BackgroundJobType<unknown>, callback: () => Promise<void>): Promise<void>;
+export declare function withInternalsTrace<T>(resource: string, callback: () => Promise<T>): Promise<T>;