Skip to content

Commit 249edca

Browse files
Merge pull request #1333 from dineshsalunke/tree-repo-fix
fix(provider): return proper repository
2 parents 78f30a7 + 5b666f2 commit 249edca

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

lib/typeorm.providers.ts

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,13 @@ export function createTypeOrmProviders(
1010
return (entities || []).map((entity) => ({
1111
provide: getRepositoryToken(entity, dataSource),
1212
useFactory: (dataSource: DataSource) => {
13-
return dataSource.options.type === 'mongodb'
14-
? dataSource.getMongoRepository(entity)
15-
: dataSource.getRepository(entity);
13+
const enitityMetadata = dataSource.entityMetadatas.find((meta) => meta.target === entity)
14+
const isTreeEntity = typeof enitityMetadata?.treeType !== 'undefined'
15+
return isTreeEntity
16+
? dataSource.getTreeRepository(entity)
17+
: dataSource.options.type === 'mongodb'
18+
? dataSource.getMongoRepository(entity)
19+
: dataSource.getRepository(entity);
1620
},
1721
inject: [getDataSourceToken(dataSource)],
1822
/**

0 commit comments

Comments
 (0)