yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-09-25 17:04:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Dao Get Call based on type (#15429)

Browse Source 
* Add Dao Get Call based on type

* Keep Jdbi as Attr
This commit is contained in:
Mohit Yadav 2024-03-02 23:43:28 +05:30 committed by GitHub
parent ec475bce03
commit f2b1622c0f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
openmetadata-service/src/main/java/org/openmetadata/service/Entity.java
View File

@ -579,4 +579,8 @@ public final class Entity {
} }
throw new BadRequestException("searchrepository not initialized"); throw new BadRequestException("searchrepository not initialized");
} }
public static <T> T getDao() {
return (T) collectionDAO;
}
} }

4
openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplication.java
View File

@ -123,6 +123,8 @@ public class OpenMetadataApplication extends Application<OpenMetadataApplication
private Authorizer authorizer; private Authorizer authorizer;
private AuthenticatorHandler authenticatorHandler; private AuthenticatorHandler authenticatorHandler;
protected Jdbi jdbi;
@Override @Override
public void run(OpenMetadataApplicationConfig catalogConfig, Environment environment) public void run(OpenMetadataApplicationConfig catalogConfig, Environment environment)
throws ClassNotFoundException, throws ClassNotFoundException,
@ -143,7 +145,7 @@ public class OpenMetadataApplication extends Application<OpenMetadataApplication
// init for dataSourceFactory // init for dataSourceFactory
DatasourceConfig.initialize(catalogConfig.getDataSourceFactory().getDriverClass()); DatasourceConfig.initialize(catalogConfig.getDataSourceFactory().getDriverClass());
final Jdbi jdbi = createAndSetupJDBI(environment, catalogConfig.getDataSourceFactory()); jdbi = createAndSetupJDBI(environment, catalogConfig.getDataSourceFactory());
CollectionDAO collectionDAO = jdbi.onDemand(CollectionDAO.class); CollectionDAO collectionDAO = jdbi.onDemand(CollectionDAO.class);
Entity.setCollectionDAO(collectionDAO); Entity.setCollectionDAO(collectionDAO);