restrict policy fetch

datahub-project · Apr 9, 2024 · 6c0e871 · 6c0e871
1 parent 68ba65e
commit 6c0e871
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/metadata-service/auth-impl/src/main/java/com/datahub/authorization/PolicyFetcher.java b/metadata-service/auth-impl/src/main/java/com/datahub/authorization/PolicyFetcher.java
@@ -21,6 +21,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
+import java.util.Set;
 import java.util.concurrent.CompletableFuture;
 import java.util.stream.Collectors;
 import javax.annotation.Nullable;
@@ -107,7 +108,11 @@ public PolicyFetchResult fetchPolicies(
 
     // Fetch DataHubPolicyInfo aspects for each urn
     final Map<Urn, EntityResponse> policyEntities =
-        entityClient.batchGetV2(opContext, POLICY_ENTITY_NAME, new HashSet<>(policyUrns), null);
+        entityClient.batchGetV2(
+            opContext,
+            POLICY_ENTITY_NAME,
+            new HashSet<>(policyUrns),
+            Set.of(DATAHUB_POLICY_INFO_ASPECT_NAME));
     return new PolicyFetchResult(
         policyUrns.stream()
             .map(policyEntities::get)