Merge pull request #1333 from dineshsalunke/tree-repo-fix

fix(provider): return proper repository
nestjs · Aug 11, 2022 · 249edca · 249edca
2 parents 78f30a7 + 5b666f2
commit 249edca
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/lib/typeorm.providers.ts b/lib/typeorm.providers.ts
@@ -10,9 +10,13 @@ export function createTypeOrmProviders(
   return (entities || []).map((entity) => ({
     provide: getRepositoryToken(entity, dataSource),
     useFactory: (dataSource: DataSource) => {
-      return dataSource.options.type === 'mongodb'
-        ? dataSource.getMongoRepository(entity)
-        : dataSource.getRepository(entity);
+      const enitityMetadata = dataSource.entityMetadatas.find((meta) => meta.target === entity)
+      const isTreeEntity = typeof enitityMetadata?.treeType !== 'undefined'
+      return isTreeEntity 
+        ? dataSource.getTreeRepository(entity)
+        : dataSource.options.type === 'mongodb'
+          ? dataSource.getMongoRepository(entity)
+          : dataSource.getRepository(entity);
     },
     inject: [getDataSourceToken(dataSource)],
     /**