nest中如何对typeorm 的repo设置总的center

小山不高2025-07-06 11:29

问题:在nestjs 框架中使用typeorm 每一个server的entity repo都要手动写一个注入，很麻烦，10个repo和10个server如果都使用就是写100遍注入

思考:如何只写一次注入，在这个注入的实体里面调用每一个repo

step1：

对typeorm有过了解的写的熟练的朋友可以知道手动实例化的方法，DataSource.getRepository 那么我们可以先定一个数组或者对象映射，方法无所谓，根本在于如何知道你需要的实例化的所有repo和其对应的名称，我这里简单用对象来表达。

ts 复制代码

const RepoEntityMap = {
  tagEntity: TagEntity,
  fileEntity: FileEntity,
};

@Injectable()
export class RepoCenter implements OnModuleInit {
  @InjectDataSource() dataSource: DataSource;

  onModuleInit(): any {
    for (const RepoEntityMapKey of Object.keys(RepoEntityMap)) {
      this[`${RepoEntityMapKey}Repo`] = this.dataSource.getRepository(RepoEntityMap[RepoEntityMapKey]);
    }
  }
}

step2:

前一步我们手动装配了每一个repo,但是对于开发还有重要的一步，我们如何知道这个center装配了哪些repo呢？参考我前一篇typescript的文章，我们来定义ts的类型，让ts提示我们

ts 复制代码

type RepoEntityMapType = {
  [K in keyof typeof RepoEntityMap as `${K}Repo`]?: Repository<InstanceType<typeof RepoEntityMap[K]>>;
};

declare module './RepoCenter' {
  interface RepoCenter extends RepoEntityMapType {}
}

如此，短短两步，很少的代码我们就可以使用center只需在每个server注入一次就可以使用所有的repo

当然了你也可以说我可以写一个center，在center中注入每一个repo,这个当然是可以了，但是难道说在TypeOrmModule.forFeature进行注册的时候也要来一遍吗？我可以使用我前面的对象映射或者定义一个数组（形式不重要，根本在于如何知道需要哪些repo）直接复用