fix for prop text update

atlanhq · Jan 28, 2025 · 2ea2057 · 2ea2057
1 parent a118278
commit 2ea2057
Showing 1 changed file with 2 additions and 5 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
@@ -4090,7 +4090,7 @@ public void updateClassificationTextPropagation(String classificationVertexId) t
 
         // update the 'assetsCountToPropagate' on in memory java object.
         AtlasTask currentTask = RequestContext.get().getCurrentTask();
-        currentTask.setAssetsCountToPropagate((long) impactedVertices.size() - 1);
+        currentTask.setAssetsCountToPropagate((long) impactedVertices.size());
 
         //update the 'assetsCountToPropagate' in the current task vertex.
         AtlasVertex currentTaskVertex = (AtlasVertex) graph.query().has(TASK_GUID, currentTask.getGuid()).vertices().iterator().next();
@@ -4112,10 +4112,7 @@ public void updateClassificationTextPropagation(String classificationVertexId) t
                 }
             }
 
-            //substract one to exclude direct attachement
-            int batchSizeToAdd = (end == impactedVertices.size()) ? batch.size() - 1 : batch.size();
-
-            currentTask.setAssetsCountPropagated(currentTask.getAssetsCountPropagated() + batchSizeToAdd);
+            currentTask.setAssetsCountPropagated(currentTask.getAssetsCountPropagated() + end);
             currentTaskVertex.setProperty(TASK_ASSET_COUNT_PROPAGATED, currentTask.getAssetsCountPropagated());
 
             transactionInterceptHelper.intercept();