POLICY-165 Policy rules related attributes updated

atlanhq · Nov 13, 2024 · 063d578 · 063d578
1 parent ac4f849
commit 063d578
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
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
@@ -4801,9 +4801,7 @@ private Optional<AtlasVertex> processAssetPolicy(String assetGuid, Set<String> n
     private Optional<AtlasVertex> updateAssetPolicies(AtlasVertex vertex, Set<String> currentPolicies, Set<String> policiesToAdd) {
         Set<String> existingPolicies = new HashSet<>(currentPolicies);
         Set<String> updatedPolicies = new HashSet<>(currentPolicies);
-        updatedPolicies.addAll(policiesToAdd);
-
-        vertex.setProperty(ASSET_POLICY_GUIDS, updatedPolicies);
+        updatedPolicies.forEach(policyGuid -> vertex.setProperty(ASSET_POLICY_GUIDS, updatedPolicies));
         vertex.setProperty(ASSET_POLICIES_COUNT, updatedPolicies.size());
         updateModificationMetadata(vertex);
 

diff --git a/webapp/src/main/java/org/apache/atlas/web/rest/BusinessPolicyREST.java b/webapp/src/main/java/org/apache/atlas/web/rest/BusinessPolicyREST.java
@@ -43,13 +43,11 @@ public BusinessPolicyREST(AtlasEntityStore entitiesStore) {
 
     /**
      * Links a business policy to entities.
-     *
-     * @param policyGuid the ID of the policy to be linked
      * @param request    the request containing the GUIDs of the assets to link the policy to
      * @throws AtlasBaseException if there is an error during the linking process
      */
     @POST
-    @Path("/{policyId}/link-business-policy")
+    @Path("/link-business-policy")
     @Timed
     public void linkBusinessPolicy(final LinkBusinessPolicyRequest request) throws AtlasBaseException {
         AtlasPerfMetrics.MetricRecorder metric = RequestContext.get().startMetricRecord("linkBusinessPolicy");