问题:在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)直接复用