Add Dao Get Call based on type (#15429)

* 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

View File

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

View File

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