typescript nodejs 依赖注入实现方法代码详解

Cady ·

更新时间:2024-11-10

· 932 次阅读

依赖注入通常也是我们所说的ioc模式，今天分享的是用typescript语言实现的ioc模式，这边用到的主要组件是 reflect-metadata 这个组件可以获取或者设置元数据信息，它的作用是拿到原数据后进行对象创建类似C#中的反射，先看第一段代码：


import "reflect-metadata";
/**
 * 对象管理器
 */
const _partialContainer = new Map<string, any>();
const PARAMTYPES = "design:paramtypes";//需要反射的原数据，有很多种选择，我们这里选择的是拿到构造函数的参数类型，为了后续判断
/**
 * 局部注入器，注入的是全局服务，实例是全局共享
 */
export function Inject(): ClassDecorator {
  return target => {
    const params: Array<any> = Reflect.getMetadata(PARAMTYPES, target);
    if (params)
      for (const item of params) {
        if (item === target) throw new Error("不能注入自己");
      }
    _partialContainer.set(target.name, target);//加入到对象管理器中，这个时候对象还没有被创建
  }
}

　上面的代码是创建一个类级别的装饰器，表示凡是使用了这个装饰器的类都会被依赖注入对象管理器管理，这里没有马上创建服务，原因是reflect-metadata的执行有先机是最高的，而这个依赖注入是支持手动注入一些实例对象，所有为了防止出现注入参数为undefined所以创建实例的工作是放在后面的，请看接下来的代码：


/**
 *
 * @param type 已创建的实例对象
 */
export function addServiceInGlobal(...types: Array<Object>) {
  for (const iterator of types) {
    _partialContainer.set(iterator.constructor.name, iterator);
  }
}

　上面的方法是手动注入实例对象时调用的，我们需要提高这个方法的执行优先级，具体的实例会在后面演示，接下来是最重要部分，创建实例部分：


export function serviceProvider<T>(service: ServiceType<T>): T {
  if (_partialContainer.has(service.name) && !_partialContainer.get(service.name).name)
    return _partialContainer.get(service.name);// 如果实例已经被创建就直接返回
  const params: Array<any> = Reflect.getMetadata(PARAMTYPES, service);// 反射拿到构造函数的参数类型
  const constrparams = params.map(item => { // 实例化参数中的依赖
    if (!_partialContainer.has(item.name)) throw new Error(`${item}没有被注入`);// 如果没有注入就抛出异常
    if (item.length)// 表示这个类型还有其它依赖
      return serviceProvider(item);// 递归继续获取其他依赖
    if (_partialContainer.has(item.name) && !_partialContainer.get(item.name).name)
      return _partialContainer.get(item.name);// 如果实例已经被创建就直接返回
    const obj = new item();// 已经没有其他依赖了 开始创建实例
    _partialContainer.set(item.name, obj);// 替换对象管理器中原来没有实例化的对象
    return obj;
  });
  const obj = new service(...constrparams); // 这里表示对象没有被创建，开始创建对象
  _partialContainer.set(service.name, obj);// 替换对象管理器中原来没有实例化的对象
  return obj;
}

　上面代码写的稍微有一点点复杂，其他理解起来也不困难，大白话讲就是如果已经实例化了直接返回实例不然就开始对象以及创建出所有的依赖。接下来是例子：


import { serviceProvider, addServiceInGlobal, Inject } from './core/injectable/injector';
import "reflect-metadata";
import moment = require('moment');
@Inject()
export class ServiceA{
  property?:string;
  msg(){
    return "ServiceA";
  }
}
@Inject()
export class ServiceC {
  constructor(private service: ServiceA) { }
  print() {
   console.log( this.service.property);
    return "调用了我";
  }
}
@Inject()
export class ServiceD{
  print(){
   console.log("我在测试注入");
  }
}
@Inject()
export class GlobalService {
  constructor(private service: ServiceC) { }
  msg!: string;
  print() {
    console.log(`共享模块${this.service.print()}`)
  }
}
@Inject()
export class Init {
  constructor(private service: ServiceA,
    private serviceD: ServiceD,
    private global: GlobalService,
    private date: Date,
    private strList: string[],
    private serviceC: ServiceC,
  ) { }
  start() {
    console.log(this.service.msg());
    this.service.property = "A模块设置的共享数据"
    console.log(moment(this.date).format("YYYY-MM-DD"))
    console.log(this.strList);
    this.serviceD.print();
    this.serviceC.print();
    this.global.print();
  }
}
const obj = new Date("2017-1-1");
const str = ['吕顺彬','菜鸟','豆豆','大铁','CC哥','码农之家的一群人'];
addServiceInGlobal(obj, str); // 添加手动创建的实例对象到对象管理器
const service = serviceProvider(Init); // 开始创建实例
service.start()// 执行

上面的实例中得到一下执行结果：