17370845950

如何实现一个支持依赖注入的Node.js框架?

实现一个支持依赖注入(Dependency Injection, DI)的 Node.js 框架,核心在于解耦模块之间的创建与使用关系,让对象的依赖由外部容器管理,而不是在代码中硬编码。以下是构建这样一个轻量级框架的关键思路和实现步骤。

理解依赖注入的基本原理

依赖注入有三种常见形式:构造函数注入、属性注入和方法注入。在 Node.js 中,最常用的是构造函数注入。其核心思想是:

  • 不主动 new 依赖实例,而是通过参数接收
  • 由一个统一的容器负责实例化和生命周期管理
  • 模块只关注自身逻辑,不关心依赖从哪来

例如,一个服务依赖数据库连接,传统写法是在类内部 require 或创建实例,而 DI 模式下,数据库连接作为参数传入。

设计依赖注入容器

容器是 DI 的核心,它需要能注册、解析和管理依赖。可以实现一个简单的 Container 类:

class Container {
  constructor() {
    this.bindings = new Map();
    this.instances = new Map();
  }

  // 注册依赖,支持类或工厂函数
  bind(name, value) {
    this.bindings.set(name, value);
  }

  // 单例模式获取实例
  get(name) {
    if (this.instances.has(name)) {
      return this.instances.get(name);
    }

    const binding = this.bindings.get(name);
    const instance = typeof binding === 'function' ? new binding(this) : binding;
    
    this.instances.set(name, instance);
    return instance;
  }
}

这里传入 this(即容器本身)到构造函数,使得类可以在内部通过容器获取其他依赖,实现自动装配。

实现基于装饰器的自动注入(可选)

为了更接近主流框架(如 NestJS)的体验,可以使用 TypeScript 装饰器标记依赖:

function Injectable() {
  return target => {};
}

function Inject(token) {
  return (target, key, index) => {
    Reflect.defineMetadata('design:paramtypes', [token], target);
  };
}

结合 Reflect Metadata API,可以在运行时读取构造函数的参数类型,并由容器自动解析。虽然原生 Node.js 不支持装饰器,但通过 Babel 或 TypeScript 可启用。

构建基础框架结构

将容器集成到应用启动流程中。比如创建一个 App 类:

class App {
  constructor(container) {
    this.container = container;
  }

  use(serviceName) {
    const service = this.container.get(serviceName);
    // 执行中间件、控制器等逻辑
    return this;
  }

  start() {
    console.log('App started');
  }
}

在入口文件中完成依赖注册:

const container = new Container();
container.bind('Database', DatabaseService);
container.bind('UserService', UserService);

const app = new App(container);
app.use('UserService').start();

基本上就这些。一个轻量但完整的依赖注入框架,重点在于容器的设计和依赖解析机制。不需要一开始就追求复杂功能,先实现基本的注册与获取,再逐步扩展作用域、异步加载、模块化组织等特性。关键是保持松耦合和可测试性,这才是 DI 的真正价值。