Custom Decorator

fastcampus 강의

// user-id.decorator.ts

export const UserId = createParamDecorator(
  (data: unknown, context: ExecutionContext) => {
    const request = context.switchToHttp().getRequest();

    if (!request || !request.user || !request.user.sub)
      throw new UnauthorizedException('사용자 정보를 찾을 수 없습니다!');

    return request.user.sub;
  },
);

user.sub 있는지 체크 후 반환

@Post()
  @RBAC(Role.admin)
  @UseInterceptors(TransactionInterceptor)
  postMovie(
    @Body() body: CreateMovieDto, //
    @Req() req,
    @UserId() userId: number,
  ) {
    return this.movieService.create(body, userId, req.queryRunner);
  }

 

- 

// query-runner.decorator.ts

export const QueryRunnerDeco = createParamDecorator(
  (data: any, context: ExecutionContext) => {
    const request = context.switchToHttp().getRequest();

    if (!request || !request.queryRunner)
      throw new InternalServerErrorException(
        'Query Runner 객체를 찾을 수 없습니다!',
      );

    return request.queryRunner;
  },
);

쿼리 러너 들어있느지 체크 후 반환

@Post()
  @RBAC(Role.admin)
  @UseInterceptors(TransactionInterceptor)
  postMovie(
    @Body() body: CreateMovieDto, //
    @QueryRunnerDeco() queryRunner: QueryRunner,
    @UserId() userId: number,
  ) {
    return this.movieService.create(body, userId, queryRunner);
  }

 

---

https://fastcampus.co.kr/classroom/239666

커리어 성장을 위한 최고의 실무교육 아카데미 | 패스트캠퍼스