当承诺在 Nest 中解析为 undefined 时，如何返回 404 HTTP 状态代码？

Question

为了避免样板代码（一遍又一遍地检查每个控制器中的未定义），当承诺getOne返回未定义时，如何自动返回 404 错误？

@Controller('/customers')
export class CustomerController {
  constructor (
      @InjectRepository(Customer) private readonly repository: Repository<Customer>
  ) { }

  @Get('/:id')
  async getOne(@Param('id') id): Promise<Customer|undefined> {
      return this.repository.findOne(id)
         .then(result => {
             if (typeof result === 'undefined') {
                 throw new NotFoundException();
             }

             return result;
         });
  }
}

Nestjs 提供了与 TypeORM 的集成，在示例存储库中是一个 TypeORMRepository实例。

Answer 1

您可以编写一个抛出on的拦截器：NotFoundExceptionundefined

@Injectable()
export class NotFoundInterceptor implements NestInterceptor {
  intercept(context: ExecutionContext, next: CallHandler): Observable<any> { {
    // next.handle() is an Observable of the controller's result value
    return next.handle()
      .pipe(tap(data => {
        if (data === undefined) throw new NotFoundException();
      }));
  }
}

然后在控制器中使用拦截器。您可以在每个类或方法中使用它：

// Apply the interceptor to *all* endpoints defined in this controller
@Controller('user')
@UseInterceptors(NotFoundInterceptor)
export class UserController {
  

或者

// Apply the interceptor only to this endpoint
@Get()
@UseInterceptors(NotFoundInterceptor)
getUser() {
  return Promise.resolve(undefined);
}