resync indexes of dataAssets based on database & databaseSchema displ…

…ayName change
open-metadata · Sep 21, 2024 · e4c6db8 · e4c6db8
1 parent a36b041
commit e4c6db8
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseRepository.java b/openmetadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseRepository.java
@@ -24,6 +24,7 @@
 import java.util.ArrayList;
 import java.util.Comparator;
 import java.util.List;
+import java.util.Objects;
 import java.util.UUID;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.csv.CSVPrinter;
@@ -105,6 +106,18 @@ public EntityInterface getParentEntity(Database entity, String fields) {
     return Entity.getEntity(entity.getService(), fields, Include.ALL);
   }
 
+  @Override
+  public void entityRelationshipReindex(Database original, Database updated) {
+    super.entityRelationshipReindex(original, updated);
+
+    // Update search indexes of assets and entity on database displayName change
+    if (!Objects.equals(original.getDisplayName(), updated.getDisplayName())) {
+      searchRepository
+          .getSearchClient()
+          .reindexAcrossIndices("database.fullyQualifiedName", original.getEntityReference());
+    }
+  }
+
   @Override
   public String exportToCsv(String name, String user) throws IOException {
     Database database = getByName(null, name, Fields.EMPTY_FIELDS); // Validate database name

diff --git a/...tadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseSchemaRepository.java b/...tadata-service/src/main/java/org/openmetadata/service/jdbi3/DatabaseSchemaRepository.java
@@ -28,6 +28,7 @@
 import java.util.Collections;
 import java.util.Comparator;
 import java.util.List;
+import java.util.Objects;
 import java.util.UUID;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.csv.CSVPrinter;
@@ -179,6 +180,18 @@ private void populateDatabase(DatabaseSchema schema) {
         .withServiceType(database.getServiceType());
   }
 
+  @Override
+  public void entityRelationshipReindex(DatabaseSchema original, DatabaseSchema updated) {
+    super.entityRelationshipReindex(original, updated);
+
+    // Update search indexes of assets and entity on databaseSchema displayName change
+    if (!Objects.equals(original.getDisplayName(), updated.getDisplayName())) {
+      searchRepository
+          .getSearchClient()
+          .reindexAcrossIndices("databaseSchema.fullyQualifiedName", original.getEntityReference());
+    }
+  }
+
   @Override
   public String exportToCsv(String name, String user) throws IOException {
     DatabaseSchema schema = getByName(null, name, Fields.EMPTY_FIELDS); // Validate database schema