在装饰器中使用全局嵌套模块

Question

我在嵌套中有一个全局记录器模块，它记录到云记录服务。我正在尝试创建一个添加日志记录功能的类方法装饰器。但是我正在努力如何在装饰器中注入全局嵌套模块的服务，因为我在文档中找到的所有依赖注入机制都依赖于基于类或类属性的注入。

export function logDecorator() {

  // I would like to inject a LoggerService that is a provider of a global logger module
  let logger = ???

  return (target: any, propertyKey: string, propertyDescriptor: PropertyDescriptor) => {
    //get original method
    const originalMethod = propertyDescriptor.value;

    //redefine descriptor value within own function block
    propertyDescriptor.value = function(...args: any[]) {
      logger.log(`${propertyKey} method called with args.`);
      
      //attach original method implementation
      const result = originalMethod.apply(this, args);

      //log result of method
      logger.log(`${propertyKey} method return value`);
    };
  };
}

更新：根据要求提供一个简单示例 基本示例是使用我的自定义记录器（在我的情况下记录到云服务）记录对服务方法的调用：

class MyService {
    @logDecorator()
    someMethod(name: string) {
        // calls to this method as well as method return values would be logged to CloudWatch
        return `Hello ${name}`
    }
}

另一个扩展用例是捕获一些错误，然后记录它们。我有很多这种逻辑可以在我的所有服务中重复使用。

Answer 1

好的，找到了解决方案。以防其他人偶然发现这一点。首先请记住装饰器是如何工作的——它们是基于类构造函数的，而不是基于实例的。

在我的例子中，我想让我的记录器服务注入到类实例中。所以解决方法是告诉装饰器中的Nest，将LoggerService注入到包含被装饰方法的类的实例中。

import { Inject } from '@nestjs/common';
import { LoggerService } from '../../logger/logger.service';

export function logErrorDecorator(bubble = true) {
  const injectLogger = Inject(LoggerService);

  return (target: any, propertyKey: string, propertyDescriptor: PropertyDescriptor) => {
    injectLogger(target, 'logger'); // this is the same as using constructor(private readonly logger: LoggerService) in a class

    //get original method
    const originalMethod = propertyDescriptor.value;

    //redefine descriptor value within own function block
    propertyDescriptor.value = async function(...args: any[]) {
      try {
        return await originalMethod.apply(this, args);
      } catch (error) {
        const logger: LoggerService = this.logger;

        logger.setContext(target.constructor.name);
        logger.error(error.message, error.stack);

        // rethrow error, so it can bubble up
        if (bubble) {
          throw error;
        }
      }
    };
  };
}

这提供了在方法中捕获错误、在服务上下文中记录它们并重新抛出它们（以便您的控制器可以处理用户响应）或不重新抛出它们的可能性。就我而言，我还必须在这里实现一些与事务相关的逻辑。

export class FoobarService implements OnModuleInit {
  onModuleInit() {
    this.test();
  }

  @logErrorDecorator()
  test() {
    throw new Error('Oh my');
  }
}