请求上下文

$ yarn add cls-hooked
$ yarn add -D @types/cls-hooked点击复制复制失败已复制

import { getNamespace } from 'cls-hooked';
import { Request, Response } from 'express';
import { v4 as uuid } from 'uuid';
import { JwtPayload } from '../../user/interfaces/jwt-payload.interface';
export class RequestContext {
  public static nsid = uuid();
  public uuid: string;
  public request: Request;
  public response: Response;
  /**
   * 请求上下文构造函数
   * @param request 请求对象
   * @param response 返回对象
   */
  constructor(request: Request, response: Response) {
    this.uuid = uuid();
    this.request = request;
    this.response = response;
  }
  /**
   * 获取请求上下文
   * @returns 当前请求上下文
   */
  public static currentRequestContext(): RequestContext {
    const namespace = getNamespace(RequestContext.nsid);
    return namespace && namespace.active ? namespace.get(RequestContext.name) : null;
  }
  /**
   * 获取当前请求
   * @returns 当前请求
   */
  public static currentRequest(): Request {
    const requestContext = RequestContext.currentRequestContext();
    return requestContext?.request;
  }
  /**
   * 获取当前用户
   * @returns 当前用户
   */
  public static currentUser(): JwtPayload {
    const requestContext = RequestContext.currentRequestContext();
    return requestContext?.request?.user;
  }
}点击复制复制失败已复制

import { NextFunction, Response, Request } from 'express';
import { RequestContext } from '../context/request.context';
import { getNamespace, createNamespace } from 'cls-hooked';
/**
 * 请求上下文中间件
 * @param req 请求参数
 * @param res 返回参数
 * @param next next函数
 */
export function RequestContextMiddleware(req: Request, res: Response, next: NextFunction) {
  const requestContext = new RequestContext(req, res);
  const namespace = getNamespace(RequestContext.nsid) || createNamespace(RequestContext.nsid);
  namespace.run(() => {
    namespace.set(RequestContext.name, requestContext);
    next();
  });
}点击复制复制失败已复制

export class AppModule implements NestModule {
  configure(consumer: MiddlewareConsumer) {
    consumer.apply(RequestContextMiddleware).forRoutes({ path: '*', method: RequestMethod.ALL });
  }
}点击复制复制失败已复制