feat: add app cluster

Author: mdwitr0

src/app-cluster.service.ts

@@ -0,0 +1,27 @@
+/* eslint-disable @typescript-eslint/ban-ts-comment */
+// @ts-nocheck
+import * as cluster from 'cluster';
+import * as os from 'os';
+import { Injectable } from '@nestjs/common';
+
+const numCPUs = os.cpus().length;
+
+@Injectable()
+export class AppClusterService {
+  // eslint-disable-next-line @typescript-eslint/ban-types
+  static clusterize(callback: Function): void {
+    if (cluster.isMaster) {
+      console.log(`Master server started on ${process.pid}`);
+      for (let i = 0; i < numCPUs; i++) {
+        cluster.fork();
+      }
+      cluster.on('exit', (worker, code, signal) => {
+        console.log(`Worker ${worker.process.pid} died. Restarting`);
+        cluster.fork();
+      });
+    } else {
+      console.log(`Cluster server started on ${process.pid}`);
+      callback();
+    }
+  }
+}

src/main.ts

@@ -5,6 +5,7 @@ import { AppModule } from './app.module';
 import { FastifyAdapter, NestFastifyApplication } from '@nestjs/platform-fastify';
 import fastifyHelmet from '@fastify/helmet';
 import compression from '@fastify/compress';
+import { AppClusterService } from './app-cluster.service';
 
 async function bootstrap() {
   const docGlobalPrefix = 'documentation';
@@ -64,4 +65,4 @@ async function bootstrap() {
   logger.log(`📑 API Documentation is running on: http://localhost:${port}/${docGlobalPrefix}`);
 }
 
-bootstrap();
+AppClusterService.clusterize(bootstrap);