首页  编辑  

类似Java/Springboot的ThreadLocal变量使用: AsyncLocalStorage

Tags: /Node & JS/   Date Created:
参考资料： 异步本地存储 |NestJS - 渐进式 Node.js 框架

我们可以使用AsyncLocalStorage，实现类似Java语言中的ThreadLocal变量，从而在整个线程范围内扩散对象和数据，而无需在方法参数列表中传递对象。例如实现类似 SpringBoot中RequestContextHolder类似的效果。

首先创建 als.module.ts，这个是类似实现一个单例的对象，该对象在单次前端请求整个生命期间有效：

  1. import { Inject, Injectable, Module } from '@nestjs/common';
  2. import { AsyncLocalStorage } from 'async_hooks';

  4. @Module({
  5.     providers: [
  6.         {
  7.             provide: AsyncLocalStorage,
  8.             useValue: new AsyncLocalStorage(),
  9.         },
  10.     ],
  11.     exports: [AsyncLocalStorage],
  12. })
  14. export class AlsModule { }
其次，修改 app.module.ts，路由拦截所有请求，并处理相关数据，然后存入 storage：
  1. import { AlsModule } from './als.module';
  2. import { AlsLogger } from 'AlsLogger';
  4. @Module({
  5.   imports: [
  6.     AlsModule,
  7.     //...
  8.   ],
  9.   // ...
  10. })
  12. export class AppModule implements NestModule {
  13.   private readonly logger: Logger = new AlsLogger("AppModule");
  14.   constructor(
  15.     private readonly als: AsyncLocalStorage<object>
  16.   ) { }
  18.   configure(consumer: MiddlewareConsumer) {
  19.     //...
  21.     consumer.apply((req, res, next) => {
  22.       const token = req.headers['accesstoken'];
  23.       let info = "";
  24.       if (token) {
  25.         const headers = JSON.parse(token);
  26.         info = this.logger.genInfo(headers.a, headers.b, headers.c, headers.d);
  27.       } else {
  28.         info = req.headers['token'] || req.headers['access_token'] || req.headers['access-token'] || this.logger.genInfo("-9", "-9", "-9", "-9");
  29.       }
  30.       const store = { info: info };
  31.       this.als.run(store, () => next());
  32.     }).forRoutes('*');
  33.   }
  34. }

最后，在任意地方使用：

  1. import { AsyncLocalStorage } from 'async_hooks';
  3. @Injectable()
  4. export class XxxxService {
  5.   constructor(
  6.     private readonly als: AsyncLocalStorage<object>,
  7.     // ....
  8.   ) {}
  10.   async xxx(a: string) {
  11.     const passAlong = this.als.getStore()['passAlong'] as string;
  12.     console.log("我们可以拿到 pass along: ", passAlong);
  13.     // ...
  14.   }
  15. }

这样我们就可以实现线程全局变量了。