轻量DI容器实现方法与技巧

前端依赖注入（DI）容器的核心目标是解耦组件与依赖的创建过程，提升可测试性和可维护性。实现一个轻量级的 DI 容器并不需要复杂的设计模式或大量代码，关键在于理解“注册-解析”机制。

定义容器的基本结构

一个最简 DI 容器应支持：

• 注册依赖：将服务标识（token）与构造函数或工厂函数绑定
• 解析依赖：根据标识获取实例，自动处理依赖关系
• 生命周期管理：支持单例和瞬时模式

下面是一个基础实现：

class Container {
  constructor() {
    this.registry = new Map();
    this.instances = new Map(); // 缓存单例
  }

  register(token, provider, { singleton = true } = {}) {
    this.registry.set(token, { provider, singleton });
  }

  resolve(token) {
    if (this.instances.has(token)) {
      return this.instances.get(token);
    }

    const record = this.registry.get(token);
    if (!record) throw new Error(`No provider registered for ${String(token)}`);

    const instance = typeof record.provider === 'function'
      ? new record.provider(this)
      : record.provider;

    if (record.singleton) {
      this.instances.set(token, instance);
    }

    return instance;
  }
}

利用构造函数参数自动注入

通过 Function.prototype.toString() 解析构造函数参数，实现自动依赖查找。这是轻量级框架常见的手段。

示例：为类添加元数据标记

function inject(...dependencies) {
  return function(target) {
    target.inject = dependencies;
  };
}

// 使用装饰器语法模拟
@inject('userService', 'logger')
class UserController {
  constructor(userService, logger) {
    this.userService = userService;
    this.logger = logger;
  }
}

在容器中解析带注入元数据的类：

resolve(token) {
  if (this.instances.has(token)) {
    return this.instances.get(token);
  }

  const record = this.registry.get(token);
  if (!record) throw new Error(...);

  let instance;
  if (typeof record.provider === 'function' && record.provider.inject) {
    const deps = record.provider.inject.map(dep => this.resolve(dep));
    instance = new record.provider(...deps);
  } else if (typeof record.provider === 'function') {
    instance = new record.provider(this);
  } else {
    instance = record.provider;
  }

  if (record.singleton) {
    this.instances.set(token, instance);
  }

  return instance;
}

简化使用方式

避免手动注册所有服务，可通过扫描或约定式注册减少配置。

例如，提供快捷注册方法：

bindClass(token, clazz, options) {
  this.register(token, clazz, options);
}

// 使用
container.bindClass('logger', ConsoleLogger);
container.bindClass('userService', UserService, { singleton: true });
container.bindClass('userController', UserController);

也可以支持异步依赖加载（如动态 import），适用于模块懒加载场景。

实际应用场景

这类容器适合中小型项目或工具库内部使用，比如：

• 插件系统中管理服务实例
• 测试环境中替换真实依赖为 mock
• 跨模块共享状态或工具类（日志、请求客户端）

不建议在大型应用中替代状态管理库或全局 context，但作为辅助解耦工具非常有效。

基本上就这些。一个不到 100 行的容器就能满足多数轻量需求，重点是清晰的注册与解析逻辑，以及合理的依赖描述方式。