FIX tests err + ADD cluster mode

Author: tacxou

service/src/app.cluster.service.ts

@@ -0,0 +1,33 @@
+import cluster from 'node:cluster'
+import * as os from 'node:os'
+import { Injectable, Logger } from '@nestjs/common'
+import process from 'node:process'
+
+const CLUSTER_DEFAULT_FORKS = os.cpus().length
+
+@Injectable()
+export class AppClusterService {
+  public static async clusterize(callback: () => Promise<void>): Promise<void> {
+    if (!process.env.TK_SERVICE_CLUSTERIZE) {
+      await callback()
+      return
+    }
+    if (cluster.isPrimary) {
+      const numCPUs = process.env.TK_SERVICE_CLUSTERIZE_FORKS || CLUSTER_DEFAULT_FORKS
+      Logger.log(`Master server started on <${process.pid}> with pid <${numCPUs}> forks 🏁`, AppClusterService.name)
+      for (let i = 0; i < numCPUs; i++) {
+        cluster.fork()
+      }
+      cluster.on('online', (worker) => {
+        Logger.log(`Worker ${worker.process['pid']} starting... 🟠`, AppClusterService.name)
+      })
+      cluster.on('exit', (worker, code, signal) => {
+        Logger.error(`Worker ${worker.process['pid']} died. Restarting`, AppClusterService.name)
+        cluster.fork()
+      })
+    } else {
+      Logger.log(`Worker server started on pid <${process.pid}> 🟢`, AppClusterService.name)
+      await callback()
+    }
+  }
+}

service/src/config.ts

@@ -8,7 +8,6 @@ import * as process from 'process'
 import { StorageManagerConfig } from '@streamkits/nestjs_module_factorydrive'
 import { AmazonWebServicesS3StorageConfig } from '@streamkits/nestjs_module_factorydrive-s3'
 import { Settings } from '~/core/settings/settings.interface'
-import { MailsSettingsInterface } from '~/tickets/mails/_interfaces/mails-settings.interface'
 
 export interface MongoosePlugin {
   package: string

service/src/extensions/extensions.service.setup.d.ts

@@ -1,4 +1,4 @@
-// noinspection JSUnresolvedReference
+// noinspection JSUnresolvedReference,JSIgnoredPromiseFromCall
 
 import { Logger } from '@nestjs/common'
 import { NestFactory } from '@nestjs/core'
@@ -12,16 +12,19 @@ import { ShutdownService } from '~/shutdown.service'
 import process from 'process'
 import { rawBodyBuffer } from '~/_common/middlewares/raw-body-buffer.middleware'
 import config from '~/config'
+import { AppClusterService } from '~/app.cluster.service'
 
 // eslint-disable-next-line
 declare const module: any
-;(async (): Promise<void> => {
+;AppClusterService.clusterize(async (): Promise<void> => {
   const cfg = config()
   const app = await NestFactory.create<NestExpressApplication>(AppModule, {
     cors: true,
+    // logger: false,
     bodyParser: false,
     rawBody: true,
   })
+  // app.useLogger(['error', 'warn', 'log', 'debug', 'verbose'])
   app.get(ShutdownService).subscribeToShutdown(async () => {
     await app.close()
     process.exit(0)
@@ -47,4 +50,4 @@ declare const module: any
     module.hot.accept()
     module.hot.dispose((): Promise<void> => app.close())
   }
-})()
+})

service/src/extensions/extensions.service.setup.js

@@ -2,7 +2,7 @@ import { DeleteResult } from 'mongodb'
 import { Test, TestingModule } from '@nestjs/testing'
 import { SourceRequestController } from './source-request.controller'
 import { SourceRequestService } from './source-request.service'
-import { SourceRequestDto } from '~/tickets/source-request/_dto/source-request._dto'
+import { SourceRequestDto } from '~/tickets/source-request/_dto/source-request.dto'
 import { SourceRequest } from '~/tickets/source-request/_schemas/source-request.schema'
 import { HttpException, HttpStatus } from '@nestjs/common'
 import { Types } from 'mongoose'

service/src/main.ts

@@ -84,11 +84,11 @@ describe('SourceRequestService', () => {
 
   describe('update', () => {
     it('should update a SourceRequest record by ID with metadata', async () => {
-      const source-requestDto = { info: { key: 'updated value' } }
+      const sourcerequestDto = { info: { key: 'updated value' } }
       const newObject = { ...mockSourceRequest }
-      const expected = newObject.info.push(source-requestDto.info)
+      const expected = newObject.info.push(sourcerequestDto.info)
       jest.spyOn(model, 'findByIdAndUpdate').mockResolvedValueOnce(expected)
-      const result = await service.update(_id.toString(), source-requestDto)
+      const result = await service.update(_id.toString(), sourcerequestDto)
       expect(result).toEqual(expected)
     })
   })