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

package/functions/abort/context-promise-patch.d.ts

@@ -7,10 +7,15 @@ export declare class ContextPromisePatch {
     private static originalClearInterval;
     private static contextTimers;
     private static timerContextMap;
+    private static contextPromises;
     /**
      * 특정 컨텍스트의 모든 타이머 정리
      */
     static clearContextTimers(contextId: string): void;
+    /**
+     * 특정 컨텍스트의 모든 Promise가 완료될 때까지 대기
+     */
+    static waitForContextPromises(contextId: string): Promise<void>;
     static patch(): void;
     private static patchPromise;
     private static patchTimeouts;

package/functions/abort/context-promise-patch.js

@@ -17,6 +17,16 @@ class ContextPromisePatch {
             this.contextTimers.delete(contextId);
         }
     }
+    /**
+     * 특정 컨텍스트의 모든 Promise가 완료될 때까지 대기
+     */
+    static async waitForContextPromises(contextId) {
+        const promises = this.contextPromises.get(contextId);
+        if (promises && promises.size > 0) {
+            // 모든 Promise가 완료될 때까지 기다림
+            await Promise.allSettled(Array.from(promises));
+        }
+    }
     static patch() {
         if (this.isPatched) {
             return;
@@ -46,7 +56,71 @@ class ContextPromisePatch {
                 if (abortSignal.aborted) {
                     return Promise.reject(new Error(`REQUEST_ABORTED: Promise가 중단되었습니다 (${contextId})`));
                 }
-                const promise = Reflect.construct(target, args);
+                // Promise 생성
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                let promise;
+                // executor 함수를 래핑하여 abort 체크 추가
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                const originalExecutor = args[0];
+                if (originalExecutor) {
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    const wrappedExecutor = (resolve, reject) => {
+                        // abort 체크 함수
+                        const checkAbort = () => {
+                            if (abortSignal.aborted) {
+                                reject(new Error(`REQUEST_ABORTED: Promise가 중단되었습니다 (${contextId})`));
+                            }
+                        };
+                        // abort 이벤트 리스너 등록
+                        abortSignal.addEventListener('abort', checkAbort, { once: false });
+                        // 원본 executor 호출
+                        try {
+                            originalExecutor(
+                            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                            (value) => {
+                                abortSignal.removeEventListener('abort', checkAbort);
+                                resolve(value);
+                            }, 
+                            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                            (reason) => {
+                                abortSignal.removeEventListener('abort', checkAbort);
+                                reject(reason);
+                            });
+                        }
+                        catch (error) {
+                            abortSignal.removeEventListener('abort', checkAbort);
+                            reject(error);
+                        }
+                    };
+                    // 새로운 args 배열 생성
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    const newArgs = [
+                        wrappedExecutor,
+                        ...args.slice(1),
+                    ];
+                    promise = Reflect.construct(target, newArgs);
+                }
+                else {
+                    promise = Reflect.construct(target, args);
+                }
+                // Promise 추적 추가
+                if (!ContextPromisePatch.contextPromises.has(contextId)) {
+                    ContextPromisePatch.contextPromises.set(contextId, new Set());
+                }
+                ContextPromisePatch.contextPromises.get(contextId).add(promise);
+                // abort 이벤트 리스너 등록 (Promise가 abort되면 추적에서 제거하도록)
+                const abortListener = () => {
+                    var _a;
+                    // Promise가 abort되었을 때 추적에서 제거
+                    (_a = ContextPromisePatch.contextPromises.get(contextId)) === null || _a === void 0 ? void 0 : _a.delete(promise);
+                };
+                abortSignal.addEventListener('abort', abortListener, { once: true });
+                // Promise 완료 시 추적에서 제거 및 리스너 해제
+                promise.finally(() => {
+                    var _a;
+                    (_a = ContextPromisePatch.contextPromises.get(contextId)) === null || _a === void 0 ? void 0 : _a.delete(promise);
+                    abortSignal.removeEventListener('abort', abortListener);
+                });
                 // Promise 체인에 자동으로 중단 확인 추가
                 return promise.then((result) => {
                     // 컨텍스트가 여전히 유효한지 확인
@@ -289,3 +363,6 @@ ContextPromisePatch.contextTimers = new Map();
 // timerId -> contextId 매핑
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 ContextPromisePatch.timerContextMap = new Map();
+// 컨텍스트별 활성 Promise 추적
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+ContextPromisePatch.contextPromises = new Map();

package/interceptors/rpc.interceptor.js

@@ -355,15 +355,14 @@ let RPCInterceptor = class RPCInterceptor {
                 }
             },
             error: localErrorProcessor,
-        }), (0, operators_1.finalize)(() => {
+        }), (0, operators_1.finalize)(async () => {
             this.gracefulShutdownService.popActiveInterceptors(execInterceptor);
             // 활성 타이머를 모두 정리
             context_promise_patch_1.ContextPromisePatch.clearContextTimers(execInterceptor.traceId);
             abortController = null;
-            // disable을 충분히 지연시켜서 모든 이벤트 리스너가 완료될 때까지 기다림
-            setTimeout(() => {
-                functions_1.AbortContext.disable();
-            }, 100);
+            // 모든 Promise가 완료될 때까지 기다림
+            await context_promise_patch_1.ContextPromisePatch.waitForContextPromises(execInterceptor.traceId);
+            functions_1.AbortContext.disable();
             return rxjs_1.EMPTY;
         })));
     }

package/package.json

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