run es bulk in async (#18649)

open-metadata · Nov 18, 2024 · ae1e7e6 · ae1e7e6
1 parent f0b76c4
commit ae1e7e6
Show file tree

Hide file tree

Showing 7 changed files with 460 additions and 206 deletions.
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/Entity.java b/openmetadata-service/src/main/java/org/openmetadata/service/Entity.java
@@ -77,6 +77,7 @@
 public final class Entity {
   private static volatile boolean initializedRepositories = false;
   @Getter @Setter private static CollectionDAO collectionDAO;
+  @Getter @Setter private static Jdbi jdbi;
   public static final String SEPARATOR = "."; // Fully qualified name separator
 
   // Canonical entity name to corresponding EntityRepository map

diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplication.java b/openmetadata-service/src/main/java/org/openmetadata/service/OpenMetadataApplication.java
@@ -162,6 +162,7 @@ public void run(OpenMetadataApplicationConfig catalogConfig, Environment environ
 
     jdbi = createAndSetupJDBI(environment, catalogConfig.getDataSourceFactory());
     Entity.setCollectionDAO(getDao(jdbi));
+    Entity.setJdbi(jdbi);
 
     initializeSearchRepository(catalogConfig.getElasticSearchConfiguration());
     // Initialize the MigrationValidationClient, used in the Settings Repository