refactor: add dedicated consumer entrypoint and graceful shutdown

Author: renganathc

apps/consumer/index.js

@@ -0,0 +1,35 @@
+const dotenv = require('dotenv');
+dotenv.config({ path: require('path').join(__dirname, '../../.env') });
+
+const { validateEnv } = require('@urbackend/common');
+
+if (process.env.NODE_ENV !== 'test') {
+    validateEnv();
+}
+
+const { initExportWorker } = require('./src/workers/export.worker');
+
+const { connectDB } = require('@urbackend/common');
+
+    (async () => {
+        try {
+            await connectDB();
+
+            const worker = initExportWorker();
+
+            console.log('[CONSUMER] Export worker started and listening for jobs...');
+
+            const shutdown = async () => {
+                console.log('Shutting down worker...');
+                await worker.close();
+                process.exit(0);
+            };
+
+            process.on('SIGINT', shutdown);
+            process.on('SIGTERM', shutdown);
+
+        } catch (err) {
+            console.error('Failed to start worker:', err);
+            process.exit(1);
+        }
+    })();

apps/consumer/package.json

@@ -4,8 +4,8 @@
   "description": "",
   "main": "src/app.js",
   "scripts": {
-    "dev": "node src/exportWorker.js",
-    "start": "node src/exportWorker.js",
+    "dev": "node index.js",
+    "start": "node index.js",
     "test": "jest --testPathPatterns=src/"
   },
   "keywords": [],

apps/consumer/src/exportWorker.js …ps/consumer/src/workers/export.worker.jsapps/consumer/src/exportWorker.js renamed to apps/consumer/src/workers/export.worker.js apps/consumer/src/exportWorker.js renamed to apps/consumer/src/workers/export.worker.js

@@ -1,11 +1,8 @@
-const dotenv = require('dotenv');
-dotenv.config({ path: require('path').join(__dirname, '../../../.env') });
-
 const { Worker } = require('bullmq');
-const mongoose = require('mongoose')
 const fs = require('fs');
 const path = require('path');
 const os = require('os');
+
 const { 
     redis, 
     exportQueue, 
@@ -17,12 +14,6 @@ const {
     getBucket 
 } = require('@urbackend/common');
 
-const { validateEnv } = require('@urbackend/common');
-
-if (process.env.NODE_ENV !== 'test') {
-    validateEnv();
-}
-
 const initExportWorker = () => {
     const worker = new Worker(exportQueue.name, async (job) => {
         const { projectId, userId, email } = job.data;
@@ -109,21 +100,4 @@ const initExportWorker = () => {
     return worker;
 };
 
-if (require.main === module) {
-
-    const { connectDB } = require('@urbackend/common');
-
-    (async () => {
-        try {
-            await connectDB();
-
-            initExportWorker();
-
-            console.log('[CONSUMER] Export worker started and listening for jobs...');
-        } catch (err) {
-            console.error('Failed to start worker:', err);
-            process.exit(1);
-        }
-    })();
-}
 module.exports = { initExportWorker };