在自定义拦截器中访问 Nest“注入器”

Question

我需要在intercept 函数内访问一个服务（由Nest TypeOrmModule 提供）（重要提示：不是constructor 参数！！！）因为这取决于传递的选项（在这种情况下为entity）。

服务注入令牌由getRepositoryToken 函数提供。

export class PaginationInterceptor {
  constructor(private readonly entity: Function) {}

  intercept(context: ExecutionContext, call$: Observable<any>): Observable<any> {
    // Here I want to inject the TypeORM repository.
    // I know how to get the injection token, but not HOW to
    // get the "injector" itself.
    const repository = getRepositoryToken(this.entity);

    // stuff...

    return call$;
  }
}

Nest 中有“服务容器”的概念吗？这个github issue 对我没有帮助。

示例用法（控制器操作）：

  @Get()
  @UseInterceptors(new PaginationInterceptor(Customer))
  async getAll() {
    // stuff...
  }

Answer 1

您可以使用NestFactory 创建NestApplication 实例来注入任何提供程序或控制器。

示例

假设您的AppModule 中有以下提供程序：

providers: [{provide: 'MyToken', useValue: 'my-value'}]

然后你可以在任何地方注入这个提供者：

// Creates an instance of the NestApplication
const application = await NestFactory.create(AppModule);

// Retrieves an instance of either injectable or controller available anywhere, otherwise, throws exception.
const myValue = application.get('MyToken');

console.log(myValue); // -> my-value

Answer 2

关于依赖注入（如果你真的想要/需要它），我想使用mixin 类可以解决问题。请参阅v4 documentation（高级 > Mixin 类）。

import { NestInterceptor, ExecutionContext, mixin, Inject } from '@nestjs/common';
import { getRepositoryToken } from '@nestjs/typeorm';
import { Observable } from 'rxjs';
import { Repository } from 'typeorm';

export function mixinPaginationInterceptor<T extends new (...args: any[]) => any>(entityClass: T) {
  // declare the class here as we can't give it "as-is" to `mixin` because of the decorator in its constructor
  class PaginationInterceptor implements NestInterceptor {
    constructor(@Inject(getRepositoryToken(entityClass)) private readonly repository: Repository<T>) {}

    intercept(context: ExecutionContext, $call: Observable<any>) {
      // do your things with `this.repository`
      return $call;
    }
  }

  return mixin(PaginationInterceptor);
}

免责声明：这是有效的 TypeScript 代码，但我没有机会在实际项目中对其进行测试，因此可能需要进行一些返工。我们的想法是像这样使用它：

@UseInterceptors(mixinPaginationInterceptor(YourEntityClass))

如果您对代码有任何疑问，请告诉我。但是我觉得mixin 的文档还不错！

---

OR 你也可以使用getRepository from typeorm（传递实体类）。 这不是 DI，因此，您必须使用 spyOn 和 getRepository 函数才能进行正确的测试。

---

关于容器，我几乎可以肯定访问它的唯一方法是使用 Execution Context，正如 Kim 所指出的那样。