✔ Fix Start-Up

bifeldy · Oct 12, 2023 · 03e26c4 · 03e26c4
1 parent eeb6998
commit 03e26c4
Showing 1 changed file with 32 additions and 40 deletions.
diff --git a/src/api/main.ts b/src/api/main.ts
@@ -97,47 +97,39 @@ declare const __non_webpack_require__: NodeRequire;
 const mainModule = __non_webpack_require__.main;
 const moduleFilename = (mainModule && mainModule.filename) || '';
 if (moduleFilename === __filename || moduleFilename.includes('iisnode')) {
-  const numCPUs = Number.parseInt(process.env['MAX_CPUS']) || os.cpus().length;
-  if (numCPUs === 1) {
-    app().then((nestApp) => {
-      const gs = nestApp.get(GlobalService);
-      nestApp.listen(process.env['PORT'] || 4200, async () => {
-        gs.log(`[APP_BOOTSTRAP_PID] 💘`, process.pid);
-      });
-    });
-  } else {
-    ctx().then(
-      async (nestCtx) => {
-        try {
-          const nestApp = await app();
-          const workers = [];
-          if (cluster.isMaster) {
-            nestCtx.get(ClusterMasterSlaveService).masterHandleMessages();
-            const gs = nestCtx.get(GlobalService);
-            gs.log(`[APP_MASTER_PID] 💻`, process.pid);
-            for (let i = 0; i < numCPUs - 1; i++) {
-              const worker = cluster.fork({ FSID: `${i}` });
-              workers.push(worker);
-              gs.log(`[WORKER_${i}] Spawned`, worker.id);
-            }
-            cluster.on('exit', (worker, code, signal) => {
-              let msg = `[WORKER_${worker.id}]`;
-              if (signal) {
-                gs.log(`${msg} Killed`, signal, 'error');
-              } else {
-                gs.log(`${msg} Exited`, code, 'error');
-              }
-            });
-          } else {
-            const gs = nestApp.get(GlobalService);
-            nestApp.listen(process.env['PORT'] || 4200, async () => {
-              gs.log(`[APP_SLAVE_PID] 💘`, process.pid);
-            });
+  app().then(async (nestApp) => {
+    const numCPUs = Number.parseInt(process.env['MAX_CPUS']) || os.cpus().length;
+    const gs = nestApp.get(GlobalService);
+    if (numCPUs >= 1) {
+      try {
+        if (cluster.isMaster) {
+          const nestCtx = await ctx();
+          nestCtx.get(ClusterMasterSlaveService).masterHandleMessages();
+          gs.log(`[APP_MASTER_PID] 💻`, process.pid);
+          for (let i = 0; i < numCPUs - 1; i++) {
+            const worker = cluster.fork({ FSID: `${i}` });
+            gs.log(`[WORKER_${i}] Spawned`, worker.id);
           }
-        } catch (err) {
-          console.error('[APP_WORKER] 💢', cluster.worker.id);
+          cluster.on('exit', (worker, code, signal) => {
+            let msg = `[WORKER_${worker.id}]`;
+            if (signal) {
+              gs.log(`${msg} Killed`, signal, 'error');
+            } else {
+              gs.log(`${msg} Exited`, code, 'error');
+            }
+          });
+        } else {
+          nestApp.listen(process.env['PORT'] || 4200, async () => {
+            gs.log(`[APP_SLAVE_PID] 💘`, process.pid);
+          });
         }
+      } catch (e) {
+        gs.log('[APP_WORKER] 💢', e, 'error');
       }
-    ).catch(err => console.error('[APP_CONTEXT] 💢', err));
-  }
+    } else {
+      nestApp.listen(process.env['PORT'] || 4200, async () => {
+        gs.log(`[APP_BOOTSTRAP_PID] 💘`, process.pid);
+      });
+    }
+  }).catch(err => console.error('[APP_CONTEXT] 💢', err));
 }