DG-1697: Adding endpoint for linking/unlink policy

atlanhq · Jul 12, 2024 · 861d095 · 861d095
1 parent 8241f31
commit 861d095
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityStoreV2.java
@@ -104,6 +104,7 @@
 import javax.inject.Inject;
 import java.io.InputStream;
 import java.util.*;
+import java.util.concurrent.CompletableFuture;
 import java.util.stream.Collectors;
 
 import static java.lang.Boolean.FALSE;
@@ -2748,7 +2749,15 @@ public void linkBusinessPolicy(String guid, List<String> linkGuids) throws Atlas
             aeh = entityRetriever.toAtlasEntityHeader(v);
             entityMutationResponse.addEntity(UPDATE, aeh);
         }
-        entityChangeNotifier.onEntitiesMutated(entityMutationResponse, false);
+        // Send notifications for the entity changes asynchronously
+        CompletableFuture.runAsync(() -> {
+            try {
+                entityChangeNotifier.onEntitiesMutated(entityMutationResponse, false);
+            } catch (AtlasBaseException e) {
+                // Handle exception
+                LOG.error("Error in processing notification for policy link updates for asset ids {} ", linkGuids, e);
+            }
+        });
         RequestContext.get().endMetricRecord(metric);
     }