From ae46e5bc802eba4bfc8b845ba33fb6a23fad340e Mon Sep 17 00:00:00 2001
From: aarshi <aarshi.arora@atlan.com>
Date: Wed, 23 Oct 2024 12:16:00 +0530
Subject: [PATCH] Minor optimisation

---
 .../store/graph/v1/DeleteHandlerV1.java       | 21 ++++++++++---------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v1/DeleteHandlerV1.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v1/DeleteHandlerV1.java
index 716fd87eb9..537f27339f 100644
--- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v1/DeleteHandlerV1.java
+++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v1/DeleteHandlerV1.java
@@ -1489,7 +1489,6 @@ public void resetHasLineageOnInputOutputDelete(Collection<AtlasEdge> removedEdge
 
             boolean isOutputEdge = PROCESS_OUTPUTS.equals(atlasEdge.getLabel());
 
-            AtlasVertex processVertex = atlasEdge.getOutVertex();
             AtlasVertex assetVertex = atlasEdge.getInVertex();
             String assetEdgeLabel = getLabel(getGuid(assetVertex), atlasEdge.getLabel());
 
@@ -1501,18 +1500,20 @@ public void resetHasLineageOnInputOutputDelete(Collection<AtlasEdge> removedEdge
                     updateAssetHasLineageStatus(assetVertex, atlasEdge, removedEdges);
                 }
             }
-                if (getStatus(processVertex) == ACTIVE && !processVertex.equals(deletedVertex)) {
-                String edgeLabel = isOutputEdge ? PROCESS_OUTPUTS : PROCESS_INPUTS;
 
-                    String processId = getGuid(processVertex);
-                    String processEdgeLabel = getLabel(processId,edgeLabel);
-                    boolean processLabelPairAlreadyProcessed = RequestContext.get().isEdgeLabelAlreadyProcessed(processEdgeLabel);
+            AtlasVertex processVertex = atlasEdge.getOutVertex();
+            String processId = getGuid(processVertex);
+            String edgeLabel = isOutputEdge ? PROCESS_OUTPUTS : PROCESS_INPUTS;
+            String processEdgeLabel = getLabel(processId, edgeLabel);
+            boolean processLabelPairAlreadyProcessed = RequestContext.get().isEdgeLabelAlreadyProcessed(processEdgeLabel);
 
-                    if (processLabelPairAlreadyProcessed) {
-                        continue;
-                    }
-                    RequestContext.get().addEdgeLabel(processEdgeLabel);
+            if (processLabelPairAlreadyProcessed) {
+                continue;
+            }
 
+            RequestContext.get().addEdgeLabel(processEdgeLabel);
+
+                if (getStatus(processVertex) == ACTIVE && !processVertex.equals(deletedVertex)) {
                 Iterator<AtlasEdge> edgeIterator = GraphHelper.getActiveEdges(processVertex, edgeLabel, AtlasEdgeDirection.BOTH);
 
                 boolean activeEdgeFound = false;