From 2ea2057182d6875f1cfc093c0f6769d22002c452 Mon Sep 17 00:00:00 2001 From: Abhijeet Kumar Date: Tue, 28 Jan 2025 18:26:06 +0530 Subject: [PATCH] fix for prop text update --- .../atlas/repository/store/graph/v2/EntityGraphMapper.java | 7 ++----- 1 file changed, 2 insertions(+), 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 index 9a2440feac..e868af6c06 100644 --- 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();