minor optimization

datahub-project · Jul 22, 2024 · 69a1222 · 69a1222
1 parent b8759e2
commit 69a1222
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/...n/java/com/linkedin/datahub/graphql/resolvers/load/EntityRelationshipsResultResolver.java b/...n/java/com/linkedin/datahub/graphql/resolvers/load/EntityRelationshipsResultResolver.java
@@ -90,13 +90,12 @@ private EntityRelationshipsResult mapEntityRelationships(
       final boolean includeSoftDelete) {
     final EntityRelationshipsResult result = new EntityRelationshipsResult();
 
-    Set<Urn> allRelatedUrns =
-        entityRelationships.getRelationships().stream()
-            .map(EntityRelationship::getEntity)
-            .collect(Collectors.toSet());
-
     final Set<Urn> existentUrns;
     if (context != null && _entityService != null && !includeSoftDelete) {
+      Set<Urn> allRelatedUrns =
+              entityRelationships.getRelationships().stream()
+                      .map(EntityRelationship::getEntity)
+                      .collect(Collectors.toSet());
       existentUrns = _entityService.exists(context.getOperationContext(), allRelatedUrns, false);
     } else {
       existentUrns = null;