@yolo-croket-dev/core 1.4.53-slowquery.1 → 1.4.53-slowquery.10

package/interceptors/rpc.interceptor.js

@@ -1,10 +1,33 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
 var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
     var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
     if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
     else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 };
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 var __metadata = (this && this.__metadata) || function (k, v) {
     if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 };
@@ -15,10 +38,13 @@ const common_1 = require("@nestjs/common");
 const core_1 = require("@nestjs/core");
 const rxjs_1 = require("rxjs");
 const operators_1 = require("rxjs/operators");
+const msa_logger_1 = __importStar(require("@yolo-croket-dev/msa-logger"));
 // import Logger from '@yolo-croket-dev/logger';
 const functions_1 = require("../functions");
 const exceptions_1 = require("../exceptions");
 const graceful_shutdown_service_1 = require("../modules/graceful-shutdown/graceful-shutdown.service");
+// const logger = Logger.createLogger('RPCInterceptor');
+const msaLogger = new msa_logger_1.default(msa_logger_1.ServerTypeEnum.SERVICE);
 // const MAX_CUT_LENGTH = 500;
 const MAX_LENGTH = 300;
 let RPCInterceptor = class RPCInterceptor {
@@ -108,7 +134,7 @@ let RPCInterceptor = class RPCInterceptor {
     //   return data;
     // }
     async intercept(context, next) {
-        var _a;
+        var _a, _b;
         const options = this.reflector.get('rpcInterceptorScope', context.getHandler());
         const args = context.getArgs();
         const arg = args[0];
@@ -120,13 +146,25 @@ let RPCInterceptor = class RPCInterceptor {
         // const traceId = content.data.param ? content.data.param.traceId : null;
         const isInternal = content.data.param ? content.data.param.isInternal : false;
         const isAdmin = content.data.param ? content.data.param.isAdmin : false;
-        // const method = content.pattern;
         const reDeliveryCount = Number(message.properties.headers['x-delivery-count']) || 0; /** 몇번 재전달되었는지 */
         const staticRetryCount = Number(options === null || options === void 0 ? void 0 : options.messageRetryCount) || 0; /** 최대 재시도 갯수 */
+        const method = content.pattern;
+        const correlationId = message.properties.correlationId;
+        const execInterceptor = {
+            method,
+            traceId: ((_a = content.data.param) === null || _a === void 0 ? void 0 : _a.traceId) || '-',
+            correlationId,
+        };
         // const logParam = {
         //   traceId,
         //   method,
         // };
+        /** consume 메시지 전부 cancel */
+        if (this.gracefulShutdownService.isApplicationExit) {
+            msaLogger.debug('RPCInterceptor isApplicationExit called', { traceId: 'IS_APPLICATION_EXIT' });
+            channel.cancel(message.fields.consumerTag);
+            return new rxjs_1.Observable();
+        }
         if (!staticRetryCount && reDeliveryCount > 0) {
             channel.sendToQueue(message.properties.replyTo, Buffer.from(JSON.stringify({
                 isError: true,
@@ -205,7 +243,7 @@ let RPCInterceptor = class RPCInterceptor {
         };
         localErrorProcessor = localErrorProcessor.bind(this);
         /** 파일 객체 처리 */
-        (0, functions_1.reWriteFileValue)((_a = content === null || content === void 0 ? void 0 : content.data) === null || _a === void 0 ? void 0 : _a.file);
+        (0, functions_1.reWriteFileValue)((_b = content === null || content === void 0 ? void 0 : content.data) === null || _b === void 0 ? void 0 : _b.file);
         if (options) {
             if (options.maxLengthKeys && options.maxLengthKeys.length) {
                 for (const key of options.maxLengthKeys) {
@@ -234,11 +272,12 @@ let RPCInterceptor = class RPCInterceptor {
         else {
             // logger.debug(`=> (${traceId}): ${JSON.stringify(this.keyCutter(JSON.stringify(content)))}`);
         }
-        this.gracefulShutdownService.incActiveInterceptors();
+        this.gracefulShutdownService.pushActiveInterceptors(execInterceptor);
         // eslint-disable-next-line consistent-return
         return next
             .handle()
-            .pipe((0, operators_1.tap)({
+            .pipe((0, operators_1.timeout)(1000 * 60 * 30), // 30분 타임아웃
+        (0, operators_1.tap)({
             next: async (val) => {
                 val.isError = false;
                 // const duration = Date.now() - now;
@@ -258,7 +297,7 @@ let RPCInterceptor = class RPCInterceptor {
             },
             error: localErrorProcessor,
         }), (0, operators_1.finalize)(() => {
-            this.gracefulShutdownService.decActiveInterceptors();
+            this.gracefulShutdownService.popActiveInterceptors(execInterceptor);
         }));
     }
 };

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.d.ts

@@ -1,7 +1,16 @@
-import { OnApplicationShutdown } from '@nestjs/common';
-export declare class GracefulShutdownService implements OnApplicationShutdown {
+import { OnApplicationShutdown, OnModuleDestroy, OnApplicationBootstrap } from '@nestjs/common';
+interface GracefulExecInterceptor {
+    method: string;
+    traceId?: string;
+    correlationId: string;
+}
+export declare class GracefulShutdownService implements OnApplicationShutdown, OnModuleDestroy, OnApplicationBootstrap {
     private activeInterceptors;
-    incActiveInterceptors(): void;
-    decActiveInterceptors(): void;
+    isApplicationExit: boolean;
+    onApplicationBootstrap(): void;
+    pushActiveInterceptors(execInterceptor: GracefulExecInterceptor): void;
+    popActiveInterceptors(execInterceptor: GracefulExecInterceptor): void;
     onApplicationShutdown(signal?: string): Promise<void>;
+    onModuleDestroy(): Promise<void>;
 }
+export {};

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

@@ -35,19 +35,42 @@ const msa_logger_1 = __importStar(require("@yolo-croket-dev/msa-logger"));
 const msaLogger = new msa_logger_1.default(msa_logger_1.ServerTypeEnum.SERVICE);
 let GracefulShutdownService = class GracefulShutdownService {
     constructor() {
-        this.activeInterceptors = 0;
+        this.activeInterceptors = [];
+        this.isApplicationExit = false;
     }
-    incActiveInterceptors() {
-        this.activeInterceptors += 1;
+    onApplicationBootstrap() {
+        this.activeInterceptors = [];
     }
-    decActiveInterceptors() {
-        this.activeInterceptors -= 1;
+    pushActiveInterceptors(execInterceptor) {
+        this.activeInterceptors.push(execInterceptor);
+    }
+    popActiveInterceptors(execInterceptor) {
+        // eslint-disable-next-line max-len
+        this.activeInterceptors = this.activeInterceptors.filter((item) => item.method === execInterceptor.method && item.traceId === execInterceptor.traceId && item.correlationId === execInterceptor.correlationId);
     }
     async onApplicationShutdown(signal) {
-        msaLogger.debug(`GracefulShutdownService.onApplicationShutdown: ${signal}`, { traceId: 'ON_APPLICATION_SHUTDOWN' });
-        while (this.activeInterceptors > 0) {
+        console.debug(`GracefulShutdownService.onApplicationShutdown signal=${signal}`, { traceId: 'ON_APPLICATION_SHUTDOWN' });
+        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' });
+        // eslint-disable-next-line max-len
+        msaLogger.debug('GracefulShutdownService.onModuleDestroy activeInterceptors', { traceId: 'ON_MODULE_DESTROY' }, { activeInterceptors: this.activeInterceptors });
+        // eslint-disable-next-line max-len
+        console.debug('GracefulShutdownService.onModuleDestroy activeInterceptors', { traceId: 'ON_MODULE_DESTROY' }, { activeInterceptors: this.activeInterceptors });
+        let prevExecInterceptorCount = 0;
+        while (this.activeInterceptors.length) {
+            if (prevExecInterceptorCount !== this.activeInterceptors.length) {
+                prevExecInterceptorCount = this.activeInterceptors.length;
+                // eslint-disable-next-line max-len
+                console.log(`GracefulShutdownService.onModuleDestroy: waiting for ${this.activeInterceptors} interceptors to finish`, { traceId: 'ON_MODULE_DESTROY' }, { activeInterceptors: this.activeInterceptors });
+                // eslint-disable-next-line max-len
+                msaLogger.debug(`GracefulShutdownService.onModuleDestroy: waiting for ${this.activeInterceptors} interceptors to finish`, { traceId: 'ON_MODULE_DESTROY' }, { activeInterceptors: this.activeInterceptors });
+            }
             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.1",
+  "version": "1.4.53-slowquery.10",
   "author": "Yolo Co., Ltd.",
   "description": "Yolo commonization module",
   "main": "index.js",