@eventista/ticketing-common 1.0.0 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@eventista/ticketing-common",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "common package for ticketing",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/cluster/config/cluster.config.ts

@@ -0,0 +1,6 @@
+import { registerAs } from '@nestjs/config';
+
+export default registerAs('cluster', () => ({
+  enabled: process.env.ENABLE_CLUSTER?.toLowerCase() === 'true',
+  maxWorkers: parseInt(process.env.MAX_WORKERS || '0', 10) || undefined,
+}));

package/src/cluster/index.ts

@@ -0,0 +1,2 @@
+export * from './modules/cluster.module';
+export * from './services/cluster.service';

package/src/cluster/modules/cluster.module.ts

@@ -0,0 +1,9 @@
+import { Module, Global } from '@nestjs/common';
+import { ClusterService } from '../services/cluster.service';
+
+@Global()
+@Module({
+  providers: [ClusterService],
+  exports: [ClusterService],
+})
+export class ClusterModule {}

package/src/cluster/services/cluster.service.ts

@@ -0,0 +1,76 @@
+import { Injectable } from '@nestjs/common';
+import { CustomLogger } from 'src/logger';
+
+
+// Import node:cluster và node:os
+const cluster = require('node:cluster');
+const { cpus } = require('node:os');
+
+@Injectable()
+export class ClusterService {
+  private logger: CustomLogger;
+
+  constructor() {
+    this.logger = new CustomLogger('ClusterService');
+  }
+
+  /**
+   * Khởi động ứng dụng với chế độ cluster hoặc đơn luồng dựa trên cấu hình
+   * @param bootstrapFunction Hàm khởi động ứng dụng
+   */
+  public startApplication(bootstrapFunction: () => Promise<void>): void {
+    // Lấy giá trị ENABLE_CLUSTER từ biến môi trường, mặc định là 'false'
+    const enableCluster = process.env.ENABLE_CLUSTER?.toLowerCase() === 'true';
+
+    if (enableCluster && cluster.isPrimary) {
+      this.startClusterMode();
+    } else {
+      this.startSingleProcessMode(bootstrapFunction, enableCluster);
+    }
+  }
+
+  /**
+   * Khởi động ứng dụng trong chế độ cluster
+   */
+  private startClusterMode(): void {
+    this.logger.log(`Cluster mode enabled. Primary ${process.pid} is running`);
+
+    // Fork workers for each CPU
+    const numCPUs = cpus().length;
+
+    // Có thể giới hạn số lượng worker thông qua biến môi trường
+    const maxWorkers = parseInt(process.env.MAX_WORKERS || String(numCPUs), 10);
+    const workerCount = Math.min(numCPUs, maxWorkers);
+
+    this.logger.log(`Starting ${workerCount} workers out of ${numCPUs} available CPUs`);
+
+    for (let i = 0; i < workerCount; i++) {
+      cluster.fork();
+    }
+
+    cluster.on('exit', (worker, code, signal) => {
+      this.logger.warn(`Worker ${worker.process.pid} died with code ${code} and signal ${signal}`);
+      // Thay thế worker đã chết
+      this.logger.log('Forking a new worker...');
+      cluster.fork();
+    });
+  }
+
+  /**
+   * Khởi động ứng dụng trong chế độ đơn luồng hoặc như một worker
+   * @param bootstrapFunction Hàm khởi động ứng dụng
+   * @param isWorker Có phải là worker trong chế độ cluster không
+   */
+  private startSingleProcessMode(bootstrapFunction: () => Promise<void>, isWorker: boolean): void {
+    bootstrapFunction().catch(err => {
+      this.logger.error('Failed to start application', err);
+      process.exit(1);
+    });
+
+    if (isWorker) {
+      this.logger.log(`Worker ${process.pid} started`);
+    } else {
+      this.logger.log(`Running in single process mode. Process ${process.pid} started`);
+    }
+  }
+}

package/src/database/mongodb/mongodb.service.ts

@@ -9,14 +9,14 @@ export class MongodbService implements OnModuleInit {
   constructor(
     @InjectConnection()
     private readonly connection: Connection,
-  ) { }
+  ) {}
 
   async onModuleInit() {
     try {
       // Kiểm tra kết nối khi khởi động
       if (this.connection.readyState === 1) {
         this.logger.log('MongoDB connected successfully');
-
+        
         // Log thông tin về connection pool
         const poolInfo = this.getConnectionPoolInfo();
         this.logger.log(`MongoDB connection pool initialized: ${JSON.stringify(poolInfo)}`);
@@ -79,7 +79,7 @@ export class MongodbService implements OnModuleInit {
           readyState: this.getReadyStateText(this.connection.readyState),
         };
       }
-
+      
       // Cách 2: Nếu không có getClient, sử dụng cách khác
       return {
         readyState: this.getReadyStateText(this.connection.readyState),