do branch based deployment

Author: aarshi0301

.github/workflows/maven.yml

@@ -33,6 +33,7 @@ on:
       - interceptapis
       - mlh-1240-improve-cm-refresh-master
       - mlh1432
+      - mlh-1620
 
 jobs:
   build:

repository/src/main/java/org/apache/atlas/repository/store/graph/v1/DeleteHandlerV1.java

@@ -37,6 +37,7 @@
 import org.apache.atlas.model.tasks.TaskSearchResult;
 import org.apache.atlas.model.typedef.AtlasRelationshipDef;
 import org.apache.atlas.model.typedef.AtlasRelationshipDef.PropagateTags;
+import org.apache.atlas.model.typedef.AtlasRelationshipEndDef;
 import org.apache.atlas.model.typedef.AtlasStructDef.AtlasAttributeDef;
 import org.apache.atlas.repository.graph.GraphHelper;
 import org.apache.atlas.repository.graphdb.AtlasEdge;
@@ -1728,6 +1729,8 @@ public void resetHasLineageOnInputOutputDelete(Collection<AtlasEdge> removedEdge
                     AtlasGraphUtilsV2.setEncodedProperty(processVertex, HAS_LINEAGE, false);
                     AtlasEntity diffEntity = getOrInitializeDiffEntity(processVertex);
                     diffEntity.setAttribute(HAS_LINEAGE, false);
+                    // Add removed relationship attribute for notification
+                    addRemovedProcessRelationshipToDiffEntity(diffEntity, atlasEdge, removedEdges);
 
                     String oppositeEdgeLabel = isOutputEdge ? PROCESS_INPUTS : PROCESS_OUTPUTS;
 
@@ -1845,7 +1848,6 @@ private void updateAssetHasLineageStatus(AtlasVertex assetVertex, AtlasEdge curr
     }
     private void updateAssetHasLineageStatusV1(AtlasVertex assetVertex, AtlasEdge currentEdge, Collection<AtlasEdge> removedEdges) {
         AtlasPerfMetrics.MetricRecorder metricRecorder = RequestContext.get().startMetricRecord("updateAssetHasLineageStatusV1");
-
         removedEdges.forEach(edge -> RequestContext.get().addToDeletedEdgesIdsForResetHasLineage(edge.getIdForDisplay()));
 
         Iterator<AtlasEdge> edgeIterator = assetVertex.query()
@@ -1873,6 +1875,8 @@ private void updateAssetHasLineageStatusV1(AtlasVertex assetVertex, AtlasEdge cu
             AtlasGraphUtilsV2.setEncodedProperty(assetVertex, HAS_LINEAGE, false);
             AtlasEntity diffEntity = getOrInitializeDiffEntity(assetVertex);
             diffEntity.setAttribute(HAS_LINEAGE, false);
+            // Add removed relationship attribute for notification
+            addRemovedProcessRelationshipToDiffEntity(diffEntity, currentEdge, removedEdges);
         }
 
         RequestContext.get().endMetricRecord(metricRecorder);
@@ -1890,6 +1894,67 @@ private AtlasEntity getOrInitializeDiffEntity(AtlasVertex vertex) {
         return diffEntity;
     }
 
+    /**
+     * Helper method to add removed process relationship to diff entity's removedRelationshipAttributes
+     * when hasLineage changes to false
+     */
+    private void addRemovedProcessRelationshipToDiffEntity(AtlasEntity diffEntity, AtlasEdge removedEdge, Collection<AtlasEdge> removedEdgesList) {
+        try {
+            // Null checks for input parameters
+            if (diffEntity == null) {
+                LOG.warn("Cannot add removed process relationship: diffEntity is null");
+                return;
+            }
+            
+            if (removedEdge == null) {
+                LOG.warn("Cannot add removed process relationship: currentEdge is null");
+                return;
+            }
+
+
+            String edgeIdForDisplay = removedEdge.getIdForDisplay();
+            boolean isEdgeInDeletedList = false;
+            if (edgeIdForDisplay != null && RequestContext.get() != null) {
+                isEdgeInDeletedList = RequestContext.get().getDeletedEdgesIdsForResetHasLineage().contains(edgeIdForDisplay);
+            }
+
+            Set<AtlasEdge> removedEdges = new HashSet<>(removedEdgesList);
+            if (!isEdgeInDeletedList && !removedEdges.contains(removedEdge)) {
+                // Edge is not deleted and not in the deleted list, skip processing
+                return;
+            }
+            
+            // Get vertices from the edge
+            AtlasVertex vertexA = removedEdge.getOutVertex();
+            AtlasVertex vertexB = removedEdge.getInVertex();
+            AtlasVertex otherVertex;
+            if (diffEntity.getGuid().equals(GraphHelper.getGuid(vertexA))) {
+                otherVertex = vertexB;
+            } else {
+                otherVertex = vertexA;
+            }
+            
+            if (otherVertex == null) {
+                LOG.warn("Cannot add removed process relationship: processVertex is null");
+                return;
+            }
+
+
+            String edgeLabel = removedEdge.getLabel();
+
+            if (edgeLabel == null) {
+                LOG.warn("Cannot add removed process relationship: edgeLabel is null");
+                return;
+            }
+
+            String attributeName = typeRegistry.getRelationshipDefByLabel(edgeLabel).getEndDef1().getName();
+            // Add to removedRelationshipAttributes
+            diffEntity.setRemovedRelationshipAttribute(attributeName, new AtlasObjectId(getGuid(otherVertex), getTypeName(otherVertex)));
+        } catch (Exception e) {
+            LOG.warn("Failed to add removed process relationship to diff entity for notification: {}", e.getMessage(), e);
+        }
+    }
+
     private void updateAssetHasLineageStatusV2(AtlasVertex assetVertex, AtlasEdge currentEdge, Collection<AtlasEdge> removedEdges) {
         AtlasPerfMetrics.MetricRecorder metricRecorder = RequestContext.get().startMetricRecord("updateAssetHasLineageStatusV2");
 
@@ -1909,6 +1974,8 @@ private void updateAssetHasLineageStatusV2(AtlasVertex assetVertex, AtlasEdge cu
             AtlasGraphUtilsV2.setEncodedProperty(assetVertex, HAS_LINEAGE, false);
             AtlasEntity diffEntity = getOrInitializeDiffEntity(assetVertex);
             diffEntity.setAttribute(HAS_LINEAGE, false);
+            // Add removed relationship attribute for notification
+            addRemovedProcessRelationshipToDiffEntity(diffEntity, currentEdge, removedEdges);
         }
 
         RequestContext.get().endMetricRecord(metricRecorder);