@yolo-croket-dev/core 1.7.38-slowquery.3 → 1.7.38-slowquery.4

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

@@ -11,7 +11,7 @@ interface RequestContext {
 export declare class AbortContext {
     private static asyncLocalStorage;
     private static contextCounter;
-    static run<T>(abortSignal: AbortSignal, requestId: string, callback: () => T): T;
+    static run<T>(abortSignal: AbortSignal, requestId: string, callback: (context: RequestContext) => T): T;
     static getCurrentAbortSignal(): AbortSignal | null;
     static getCurrentRequestId(): string | null;
     static getCurrentContext(): {

package/functions/abort/abort-context.js

@@ -12,7 +12,7 @@ class AbortContext {
             createdAt: Date.now(),
             parentContext: currentContext || undefined,
         };
-        return this.asyncLocalStorage.run(newContext, callback);
+        return this.asyncLocalStorage.run(newContext, callback.bind(this, newContext));
     }
     static getCurrentAbortSignal() {
         const context = this.asyncLocalStorage.getStore();

package/interceptors/rpc.interceptor.js

@@ -329,7 +329,7 @@ let RPCInterceptor = RPCInterceptor_1 = class RPCInterceptor {
         }
         this.gracefulShutdownService.pushActiveInterceptors(execInterceptor);
         // eslint-disable-next-line consistent-return
-        return functions_1.AbortContext.run(abortController.signal, execInterceptor.traceId, () => next
+        return functions_1.AbortContext.run(abortController.signal, execInterceptor.traceId, (context) => next
             .handle()
             .pipe((0, operators_1.timeout)({
             each: this.DEFAULT_TIMEOUT,
@@ -362,7 +362,7 @@ let RPCInterceptor = RPCInterceptor_1 = class RPCInterceptor {
         }), (0, operators_1.finalize)(() => {
             this.gracefulShutdownService.popActiveInterceptors(execInterceptor);
             console.log('finalize called', execInterceptor.traceId);
-            functions_1.AbortContext.endContext();
+            functions_1.AbortContext.endContext(context);
             abortController = null;
             return new rxjs_1.Observable();
         })));

package/package.json

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