From f2b1622c0f274bffd2dd528e208d34dc5db3f77a Mon Sep 17 00:00:00 2001
From: Mohit Yadav <105265192+mohityadav766@users.noreply.github.com>
Date: Sat, 2 Mar 2024 23:43:28 +0530
Subject: [PATCH] Add Dao Get Call based on type (#15429)

* Add Dao Get Call based on type

* Keep Jdbi as Attr
---
 .../src/main/java/org/openmetadata/service/Entity.java        | 4 ++++
 .../org/openmetadata/service/OpenMetadataApplication.java     | 4 +++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/Entity.java b/openmetadata-service/src/main/java/org/openmetadata/service/Entity.java
index 67a75d67d97..ed0a81c9973 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/Entity.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/Entity.java
@@ -579,4 +579,8 @@ public final class Entity {
     }
     throw new BadRequestException("searchrepository not initialized");
   }
+
+  public static <T> T getDao() {
+    return (T) collectionDAO;
+  }
 }
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplication.java b/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplication.java
index 418e67086a9..bfca970e143 100644
--- a/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplication.java
+++ b/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplication.java
@@ -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);