NestJS 允许您访问本机 Mongoose Connection,这反过来又提供了对连接的 db-object 的访问,因此您可以创建以下服务:
@Injectable()
export class DynamicMongoDbService {
constructor (@InjectConnection() private connection: Connection) {
}
async insert(collectionName: string, data: any) {
return this.connection.db.collection(collectionName).insert(data);
}
}
然后相应地使用此服务:
this.dynamicMongoDbService.insert('collectionA', data);
this.dynamicMongoDbService.insert('collectionB', data);
编辑:
如果模型在编译时是已知的,您还可以创建一个服务来注入所有必需的模型并将它们存储在地图中。然后,在使用服务时,您可以动态决定选择哪个模型并将其委托给它:
@Injectable()
export class DynamicMongoDbService {
private modelMap: Record<string, Model<any>>;
constructor (
@InjectModel(Cat.name) catModel: Model<Cat>,
@InjectModel(Dog.name) dogModel: Model<Dog>) {
this.modelMap = {
[Cat.name]: catModel,
[Dog.name]: dogModel
};
}
async insertDynamically<M, T> (modelType: typeof M, data: T) {
const model = this.modelMap[modelType.name];
return model.save(data);
}
}
像这样使用它:
this.dynamicMongoDbService.insert(Cat, data);
this.dynamicMongoDbService.insert(Dog, data);