@yolo-croket-dev/core 1.5.0 → 1.5.1-jun.1

package/interceptors/logging-service.interceptor.js

@@ -90,7 +90,11 @@ let LoggingServiceInterceptor = class LoggingServiceInterceptor {
             traceInfo: { from },
         });
         return next.handle()
-            .pipe((0, operators_1.tap)(() => {
+            .pipe((0, operators_1.map)(data => {
+            console.log('@data', data);
+            return data;
+        }), (0, operators_1.tap)(() => {
+            console.log('@top');
             const elapsed = Date.now() - now;
             msaLogger.printOutService({
                 traceId,

package/interceptors/rpc.interceptor.d.ts

@@ -1,9 +1,11 @@
 import { ExecutionContext, NestInterceptor, CallHandler } from '@nestjs/common';
 import { Reflector } from '@nestjs/core';
 import { Observable } from 'rxjs';
+import { GracefulShutdownService } from '../modules/graceful-shutdown/graceful-shutdown.service';
 export declare class RPCInterceptor implements NestInterceptor {
     private readonly reflector;
-    constructor(reflector: Reflector);
+    private readonly gracefulShutdownService;
+    constructor(reflector: Reflector, gracefulShutdownService: GracefulShutdownService);
     private maxLengthOverShorten;
     private updateMaxLengthArrayShorten;
     intercept(context: ExecutionContext, next: CallHandler): Promise<Observable<unknown>>;

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);
 };
@@ -16,14 +39,19 @@ 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 {
-    constructor(reflector) {
+    constructor(reflector, gracefulShutdownService) {
         this.reflector = reflector;
+        this.gracefulShutdownService = gracefulShutdownService;
     }
     maxLengthOverShorten(data) {
         if (typeof data !== 'string') {
@@ -107,7 +135,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];
@@ -119,13 +147,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,
@@ -206,7 +246,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) {
@@ -235,10 +275,12 @@ let RPCInterceptor = class RPCInterceptor {
         else {
             // logger.debug(`=> (${traceId}): ${JSON.stringify(this.keyCutter(JSON.stringify(content)))}`);
         }
+        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;
@@ -257,11 +299,14 @@ let RPCInterceptor = class RPCInterceptor {
                 }
             },
             error: localErrorProcessor,
+        }), (0, operators_1.finalize)(() => {
+            this.gracefulShutdownService.popActiveInterceptors(execInterceptor);
         }));
     }
 };
 exports.RPCInterceptor = RPCInterceptor;
 exports.RPCInterceptor = RPCInterceptor = __decorate([
     (0, common_1.Injectable)(),
-    __metadata("design:paramtypes", [core_1.Reflector])
+    __metadata("design:paramtypes", [core_1.Reflector,
+        graceful_shutdown_service_1.GracefulShutdownService])
 ], RPCInterceptor);

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

@@ -0,0 +1,2 @@
+export declare class GracefulShutdownModule {
+}

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

@@ -0,0 +1,21 @@
+"use strict";
+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;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GracefulShutdownModule = void 0;
+const common_1 = require("@nestjs/common");
+const graceful_shutdown_service_1 = require("./graceful-shutdown.service");
+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],
+    })
+], GracefulShutdownModule);

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

@@ -0,0 +1,16 @@
+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;
+    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

@@ -0,0 +1,81 @@
+"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;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GracefulShutdownService = void 0;
+const common_1 = require("@nestjs/common");
+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 = [];
+        this.isApplicationExit = false;
+    }
+    onApplicationBootstrap() {
+        this.activeInterceptors = [];
+    }
+    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=${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 });
+        let prevExecInterceptorCount = 0;
+        let count = 0;
+        while (this.activeInterceptors.length) {
+            count += 1;
+            if ((prevExecInterceptorCount !== this.activeInterceptors.length) || (count % 100 === 0)) {
+                prevExecInterceptorCount = this.activeInterceptors.length;
+                // eslint-disable-next-line max-len
+                msaLogger.debug('GracefulShutdownService.onModuleDestroy: waiting for interceptors to finish', { traceId: 'ON_MODULE_DESTROY' }, {
+                    activeInterceptors: this.activeInterceptors,
+                    count,
+                });
+            }
+            await new Promise((resolve) => {
+                setTimeout(resolve, 1000);
+            });
+        }
+    }
+};
+exports.GracefulShutdownService = GracefulShutdownService;
+exports.GracefulShutdownService = GracefulShutdownService = __decorate([
+    (0, common_1.Injectable)()
+], GracefulShutdownService);

package/modules/index.d.ts

@@ -1 +1,2 @@
 export * from './redis.module';
+export * from './graceful-shutdown/graceful-shutdown.module';

package/modules/index.js

@@ -15,3 +15,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./redis.module"), exports);
+__exportStar(require("./graceful-shutdown/graceful-shutdown.module"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yolo-croket-dev/core",
-  "version": "1.5.0",
+  "version": "1.5.1-jun.1",
   "author": "Yolo Co., Ltd.",
   "description": "Yolo commonization module",
   "main": "index.js",