@yolo-croket-dev/core 1.4.53-slowquery.5 → 1.4.53-slowquery.7

package/interceptors/rpc.interceptor.js

@@ -266,7 +266,6 @@ let RPCInterceptor = class RPCInterceptor {
         else {
             // logger.debug(`=> (${traceId}): ${JSON.stringify(this.keyCutter(JSON.stringify(content)))}`);
         }
-        msaLogger.debug('RPCInterceptor IncActiveInterceptors called', { traceId: 'INC_RPC_ACTIVE_INTERCEPTORS' });
         this.gracefulShutdownService.incActiveInterceptors();
         // eslint-disable-next-line consistent-return
         return next
@@ -292,7 +291,6 @@ let RPCInterceptor = class RPCInterceptor {
             },
             error: localErrorProcessor,
         }), (0, operators_1.finalize)(() => {
-            msaLogger.debug('RPCInterceptor DecActiveInterceptors called', { traceId: 'DEC_RPC_ACTIVE_INTERCEPTORS' });
             this.gracefulShutdownService.decActiveInterceptors();
         }));
     }

package/modules/graceful-shutdown/graceful-shutdown.module.js

@@ -13,6 +13,7 @@ let GracefulShutdownModule = class GracefulShutdownModule {
 };
 exports.GracefulShutdownModule = GracefulShutdownModule;
 exports.GracefulShutdownModule = GracefulShutdownModule = __decorate([
+    (0, common_1.Global)(),
     (0, common_1.Module)({
         providers: [graceful_shutdown_service_1.GracefulShutdownService],
         exports: [graceful_shutdown_service_1.GracefulShutdownService],

package/modules/graceful-shutdown/graceful-shutdown.rmq.d.ts

@@ -0,0 +1,20 @@
+import { RmqOptions, ServerRMQ } from '@nestjs/microservices';
+import { Channel } from 'amqp-connection-manager';
+import { ConsumeMessage } from 'amqplib';
+export type GracefulRmqOptions = RmqOptions['options'] & {
+    waitingEndingHandlersTimeoutMs?: number;
+    waitingEndingHandlersIntervalMs?: number;
+};
+export declare class GracefulServerRMQ extends ServerRMQ {
+    protected readonly options: GracefulRmqOptions;
+    protected runningMessages: number;
+    protected closing: boolean;
+    protected consumerTag: string | null;
+    protected readonly waitingEndingHandlersTimeoutMs: number;
+    protected readonly waitingEndingHandlersIntervalMs: number;
+    constructor(options: GracefulRmqOptions);
+    setupChannel(channel: Channel, callback: () => void): Promise<void>;
+    protected waitingHandlers(): Promise<void>;
+    handleMessage(message: ConsumeMessage, channel: Channel): Promise<void>;
+    close(): Promise<void>;
+}

package/modules/graceful-shutdown/graceful-shutdown.rmq.js

@@ -0,0 +1,59 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GracefulServerRMQ = void 0;
+const microservices_1 = require("@nestjs/microservices");
+const node_util_1 = require("node:util");
+const sleep = (0, node_util_1.promisify)(setTimeout);
+const RMQ_DEFAULT_WAITING_ENDING_HANDLERS_TIMEOUT_MS = 5000;
+const RMQ_DEFAULT_WAITING_ENDING_HANDLERS_INTERVAL_MS = 500;
+class GracefulServerRMQ extends microservices_1.ServerRMQ {
+    constructor(options) {
+        super(options);
+        this.options = options;
+        this.runningMessages = 0;
+        this.closing = false;
+        this.consumerTag = null;
+        this.waitingEndingHandlersTimeoutMs = this.getOptionsProp(this.options, 'waitingEndingHandlersTimeoutMs')
+            || RMQ_DEFAULT_WAITING_ENDING_HANDLERS_TIMEOUT_MS;
+        this.waitingEndingHandlersIntervalMs = this.getOptionsProp(this.options, 'waitingEndingHandlersIntervalMs')
+            || RMQ_DEFAULT_WAITING_ENDING_HANDLERS_INTERVAL_MS;
+    }
+    async setupChannel(channel, callback) {
+        if (this.closing) {
+            return;
+        }
+        if (!this.queueOptions.noAssert) {
+            await channel.assertQueue(this.queue, this.queueOptions);
+        }
+        await channel.prefetch(this.prefetchCount, this.isGlobalPrefetchCount);
+        const { consumerTag } = await channel.consume(this.queue, (msg) => this.handleMessage(msg, channel), { noAck: false });
+        this.consumerTag = consumerTag;
+        callback();
+    }
+    async waitingHandlers() {
+        while (this.runningMessages > 0) {
+            await sleep(this.waitingEndingHandlersIntervalMs);
+        }
+    }
+    async handleMessage(message, channel) {
+        this.runningMessages += 1;
+        return super.handleMessage(message, channel).finally(() => {
+            this.runningMessages -= 1;
+        });
+    }
+    async close() {
+        this.closing = true;
+        if (this.channel) {
+            await this.channel.removeSetup(undefined, (channel) => channel.cancel(this.consumerTag));
+        }
+        this.consumerTag = null;
+        await Promise.race([
+            this.waitingHandlers(),
+            this.waitingEndingHandlersTimeoutMs > 0
+                && sleep(this.waitingEndingHandlersTimeoutMs),
+        ]);
+        this.runningMessages = 0;
+        super.close();
+    }
+}
+exports.GracefulServerRMQ = GracefulServerRMQ;

package/modules/graceful-shutdown/graceful-shutdown.service.js

@@ -39,39 +39,32 @@ let GracefulShutdownService = class GracefulShutdownService {
         this.isApplicationExit = false;
     }
     onApplicationBootstrap() {
-        msaLogger.debug('GracefulShutdownService.onApplicationBootstrap', { traceId: 'ON_APPLICATION_BOOTSTRAP' });
         this.activeInterceptors = 0;
     }
     incActiveInterceptors() {
         this.activeInterceptors += 1;
-        msaLogger.debug(`GracefulShutdownService.incActiveInterceptors: ${this.activeInterceptors}`, { traceId: 'INC_ACTIVE_INTERCEPTORS' });
     }
     decActiveInterceptors() {
         this.activeInterceptors -= 1;
         msaLogger.debug(`GracefulShutdownService.decActiveInterceptors: ${this.activeInterceptors}`, { traceId: 'DEC_ACTIVE_INTERCEPTORS' });
     }
     async onApplicationShutdown(signal) {
-        this.isApplicationExit = true;
-        msaLogger.debug(`GracefulShutdownService.onApplicationShutdown: ${signal}`, { traceId: 'ON_APPLICATION_SHUTDOWN' });
-        // eslint-disable-next-line max-len
-        msaLogger.debug(`GracefulShutdownService.onApplicationShutdown: activeInterceptors=${this.activeInterceptors}`, { traceId: 'ON_APPLICATION_SHUTDOWN' });
-        while (this.activeInterceptors > 0) {
-            // eslint-disable-next-line max-len
-            msaLogger.debug(`GracefulShutdownService.onApplicationShutdown: waiting for ${this.activeInterceptors} interceptors to finish`, { traceId: 'ON_APPLICATION_SHUTDOWN' });
-            await new Promise((resolve) => {
-                setTimeout(resolve, 100);
-            });
-        }
+        msaLogger.debug(`GracefulShutdownService.onApplicationShutdown signal=${signal}`, { traceId: 'ON_APPLICATION_SHUTDOWN' });
+        process.exit(0);
     }
     async onModuleDestroy() {
         this.isApplicationExit = true;
         msaLogger.debug('GracefulShutdownService.onModuleDestroy', { traceId: 'ON_MODULE_DESTROY' });
         msaLogger.debug(`GracefulShutdownService.onModuleDestroy: activeInterceptors=${this.activeInterceptors}`, { traceId: 'ON_MODULE_DESTROY' });
+        let prevExecInterceptorCount = 0;
         while (this.activeInterceptors > 0) {
-            // eslint-disable-next-line max-len
-            msaLogger.debug(`GracefulShutdownService.onModuleDestroy: waiting for ${this.activeInterceptors} interceptors to finish`, { traceId: 'ON_MODULE_DESTROY' });
+            if (prevExecInterceptorCount !== this.activeInterceptors) {
+                prevExecInterceptorCount = this.activeInterceptors;
+                // eslint-disable-next-line max-len
+                msaLogger.debug(`GracefulShutdownService.onModuleDestroy: waiting for ${this.activeInterceptors} interceptors to finish`, { traceId: 'ON_MODULE_DESTROY' });
+            }
             await new Promise((resolve) => {
-                setTimeout(resolve, 100);
+                setTimeout(resolve, 1000);
             });
         }
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yolo-croket-dev/core",
-  "version": "1.4.53-slowquery.5",
+  "version": "1.4.53-slowquery.7",
   "author": "Yolo Co., Ltd.",
   "description": "Yolo commonization module",
   "main": "index.js",