POLICY-166 Bulk linking APi improved version for rules

atlanhq · Dec 11, 2024 · 97d8534 · 97d8534
1 parent 47b5449
commit 97d8534
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphMapper.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphMapper.java
@@ -4820,7 +4820,11 @@ private final Set<String> calculateEffectivePolicies(Set<String> existing, Set<S
     }
 
     private void updateVertexPolicies(AtlasVertex vertex, String propertyKey, Set<String> policies) {
-        vertex.setProperty(propertyKey, policies.stream().collect(Collectors.joining(",")));
+        if(ASSET_POLICY_GUIDS.equals(propertyKey)){
+            policies.forEach(policyGuid -> vertex.setProperty(ASSET_POLICY_GUIDS, policyGuid));
+        }else {
+            policies.forEach(policyGuid -> vertex.setProperty(NON_COMPLIANT_ASSET_POLICY_GUIDS, policyGuid));
+        }
     }
 
     private int countPoliciesExcluding(Set<String> policies, String substring) {