在nestjs中自定义注入

Question

我创建了一个简单的日志记录服务，它通过 debug 在后台提供调试日志记录功能，可按如下方式使用：

@Resolver('Category')
export class CategoryResolver {
  private log: DebugLog;

  constructor(private readonly categoryService: CategoryService, loggingService: LoggingService) {
    // needs this to determine logger prefix
    this.log = loggingService.getDebugLogger(this);
  }

  @ResolveProperty()
  async mainCategory(category: Category, args, context): Promise<MainCategory> {
    this.log('resolving mainCategory');
    /* ... */
  }
}

现在，我想通过创建一个类似于@InjectRepository for @nestjs/typeorm 的@InjectLogger 装饰器来摆脱构造函数中的样板代码。基本上我想实现能够做到这一点：

@Resolver('Category')
export class CategoryResolver {

  constructor(
    private readonly categoryService: CategoryService, 
    @InjectLogger(CategoryResolver) private readonly log: DebugLog
  ) { }

  @ResolveProperty()
  async mainCategory(category: Category, args, context): Promise<MainCategory> {
    this.log('resolving mainCategory');
    /* ... */
  }
}

我将如何实现这样的目标？

Answer 1

@nestjs/typeorm 的@InjectRepository() 注入通过为每个注册实体添加单独的 DI 提供程序令牌来工作 - 请参阅 @nestjs/typeorm providers 以了解提供程序逻辑。该逻辑迭代每个已注册的@Entity，并为每个@Entity 生成并添加一个令牌提供程序。

要执行与您的示例相匹配的类似操作，您需要能够为您将作为参数传递给@InjectLogger(<type>) 的每种类型识别并生成令牌和关联的提供程序。