mongo-job-scheduler 0.1.3 → 0.1.4

package/README.md

@@ -21,6 +21,7 @@ Designed for distributed systems that need:
 - **Interval jobs**
 - **Resume on restart**
 - **Stale lock recovery**
+- **Automatic Lock Renewal** (Heartbeats): Long-running jobs automatically extend their lock.
 - **Sharding-safe design**
 
 ## Distributed Systems
@@ -76,6 +77,20 @@ await scheduler.schedule(
 );
 ```
 
+## Interval Scheduling
+
+Run a job repeatedly with a fixed delay between executions (e.g., every 5 minutes).
+
+```typescript
+await scheduler.schedule({
+  name: "cleanup-logs",
+  data: {},
+  repeat: {
+    every: 5 * 60 * 1000, // 5 minutes in milliseconds
+  },
+});
+```
+
 ## Job Deduplication
 
 Prevent duplicate jobs using `dedupeKey`. If a job with the same key exists, it returns the existing job instead of creating a new one.

package/dist/store/in-memory-job-store.d.ts

@@ -22,4 +22,5 @@ export declare class InMemoryJobStore implements JobStore {
     }): Promise<number>;
     cancel(jobId: unknown): Promise<void>;
     findById(jobId: unknown): Promise<Job | null>;
+    renewLock(jobId: unknown, workerId: string): Promise<void>;
 }

package/dist/store/in-memory-job-store.js

@@ -119,5 +119,17 @@ class InMemoryJobStore {
         const job = this.jobs.get(String(jobId));
         return job ? { ...job } : null;
     }
+    async renewLock(jobId, workerId) {
+        const job = this.jobs.get(String(jobId));
+        if (!job)
+            throw new store_errors_1.JobNotFoundError();
+        if (job.status === "running" && job.lockedBy === workerId) {
+            job.lockedAt = new Date();
+            job.updatedAt = new Date();
+        }
+        else {
+            throw new Error("Job lock lost or owner changed");
+        }
+    }
 }
 exports.InMemoryJobStore = InMemoryJobStore;

package/dist/store/job-store.d.ts

@@ -47,4 +47,8 @@ export interface JobStore {
      * Get job by ID
      */
     findById(jobId: unknown): Promise<Job | null>;
+    /**
+     * Renew the lock for a running job (heartbeat)
+     */
+    renewLock(jobId: unknown, workerId: string): Promise<void>;
 }

package/dist/store/mongo/mongo-job-store.d.ts

@@ -28,4 +28,5 @@ export declare class MongoJobStore implements JobStore {
         now: Date;
         lockTimeoutMs: number;
     }): Promise<number>;
+    renewLock(id: ObjectId, workerId: string): Promise<void>;
 }

package/dist/store/mongo/mongo-job-store.js

@@ -165,5 +165,21 @@ class MongoJobStore {
         });
         return result.modifiedCount;
     }
+    async renewLock(id, workerId) {
+        const now = new Date();
+        const result = await this.collection.updateOne({
+            _id: id,
+            lockedBy: workerId,
+            status: "running",
+        }, {
+            $set: {
+                lockedAt: now,
+                updatedAt: now,
+            },
+        });
+        if (result.matchedCount === 0) {
+            throw new Error("Job lock lost or owner changed");
+        }
+    }
 }
 exports.MongoJobStore = MongoJobStore;

package/dist/worker/worker.js

@@ -80,10 +80,35 @@ class Worker {
             job.lastScheduledAt = next;
             await this.store.reschedule(job._id, next);
         }
+        // ---------------------------
+        // HEARTBEAT
+        // ---------------------------
+        const heartbeatIntervalMs = Math.max(50, this.lockTimeout / 2);
+        const heartbeatParams = {
+            jobId: job._id,
+            workerId: this.workerId,
+        };
+        let stopHeartbeat = false;
+        const heartbeatLoop = async () => {
+            while (!stopHeartbeat) {
+                await this.sleep(heartbeatIntervalMs);
+                if (stopHeartbeat)
+                    break;
+                try {
+                    await this.store.renewLock(heartbeatParams.jobId, heartbeatParams.workerId);
+                }
+                catch (err) {
+                    this.emitter.emitSafe("worker:error", new Error(`Heartbeat failed for job ${heartbeatParams.jobId}: ${err}`));
+                    break;
+                }
+            }
+        };
+        const heartbeatPromise = heartbeatLoop();
         try {
             const current = await this.store.findById(job._id);
             if (current && current.status === "cancelled") {
                 this.emitter.emitSafe("job:complete", job);
+                stopHeartbeat = true; // stop fast
                 return;
             }
             await this.handler(job);
@@ -112,10 +137,14 @@ class Worker {
                     attempts,
                     lastError: error.message,
                 });
-                return;
             }
-            await this.store.markFailed(job._id, error.message);
-            this.emitter.emitSafe("job:fail", { job, error });
+            else {
+                await this.store.markFailed(job._id, error.message);
+                this.emitter.emitSafe("job:fail", { job, error });
+            }
+        }
+        finally {
+            stopHeartbeat = true;
         }
     }
     sleep(ms) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "mongo-job-scheduler",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "Production-grade MongoDB-backed job scheduler with retries, cron, timezone support, and crash recovery",
   "license": "MIT",
   "author": "Darshan Bhut",