5

我想在NestJs Interceptor中使用AsyncLocalStorage：

export interface CallHandler<T = any> {
    handle(): Observable<T>;
}
export interface NestInterceptor<T = any, R = any> {
    intercept(context: ExecutionContext, next: CallHandler<T>): Observable<R> | Promise<Observable<R>>;
}

拦截器函数获取一个next CallHandler返回一个Observable.

在这种情况下我不能使用runcallHandler.handle() （run 回调将在observable 完成之前立即退出）：

  intercept(context: ExecutionContext, callHandler: CallHandler): Observable<any> | Promise<Observable<any>> {
    const asyncLocalStorage = new AsyncLocalStorage();
    const myStore = {  some: 'data'};
    return asyncLocalStorage.run(myStore, () => callHandler.handle());
  }

请参阅损坏的复制示例

我想出的解决方案是这样的：

const localStorage = new AsyncLocalStorage();

export class MyInterceptor implements NestInterceptor {
  intercept(context: ExecutionContext, callHandler: CallHandler): Observable<any> | Promise<Observable<any>> {
    const resource = new AsyncResource('AsyncLocalStorage', { requireManualDestroy: true });
    const myStore = { some: 'data' };

    localStorage.enterWith(myStore);
    return callHandler.handle().pipe(
      finalize(() => resource.emitDestroy())
    );
  }
}

请参阅工作复制示例

这似乎工作正常，但我不确定这是否真的正确 - 它看起来很混乱且容易出错。所以我想知道：

  1. 这完全正确吗？
  2. 有没有更好/更清洁的方法来处理这个问题？
4

1 回答 1

-1

我如何使用cls-hooks，我找到的解决方案是：

return new Observable(observer => {
  namespace.runAndReturn(async () => {
    namespace.set("some", "data")   
    next.handle()
        .subscribe(
          res => observer.next(res), 
          error => observer.error(error),
          () => observer.complete()
        )
  })
})
于 2021-06-23T22:37:51.267 回答