如何在NestJS拦截器中获取存储库

时间:2020-09-09 18:27:33

标签: nestjs nestjs-gateways

我已经在模块中的Interceptor上创建了。我想在拦截器中获取存储库[ LocumRepository ],并在调用后进行一些处理。 以下是我的拦截器课程:

import { CallHandler, ExecutionContext, Injectable, NestInterceptor } from '@nestjs/common';
import { LocumEntity } from '../../locum/entities/locum.entity';
import { getRepository, Like, Repository } from 'typeorm';
import { Observable, combineLatest } from 'rxjs';
import { tap } from 'rxjs/operators';
import { map } from 'rxjs/operators';


@Injectable()
export class ApprovalInterceptor implements NestInterceptor {


  intercept(context: ExecutionContext, next: CallHandler): Observable<any> {
    return next
      .handle()
      .pipe(
        map(value => this.updateLocumStatus(value, context))
      );
  }

  async updateLocumStatus(value, context) {
    if (context.switchToHttp().getResponse().statusCode) {
      
      let locumData = await getRepository(LocumEntity)
        .createQueryBuilder('locum')
        .where('locum.id = :id', { id: value.locumId })
        .getOne();

    }
    return value;
  }
}

我收到以下错误:

未找到“ LocumRepository”的存储库。看起来该实体未在当前的“默认”连接中注册?

当LocumRepository在模块文件中声明时,我正在Interceptor类之外使用它

1 个答案:

答案 0 :(得分:0)

由于interceptor@Injectable(),因此您可以像通常使用@InjectRepository(Locum)(或您的实体所称的那样)一样,采用DI方法并注入它,然后进行常规服务this.repo.repoMethod()。这样,您仍然可以获得使用DI的好处,并且能够在测试过程中提供模拟。使用这种方法唯一要确保的是,您将在使用拦截器的当前模块中拥有此存储库。