Merge pull request #2750 from atlanhq/dev/feat/removeRelationshipAttr…

…ibute PLT-573| Bug | Fix mutationResponse
atlanhq · Jan 11, 2024 · db9cb41 · db9cb41
2 parents 9ee63d1 + 49f8dfd
commit db9cb41
Showing 1 changed file with 6 additions and 2 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
@@ -1771,10 +1771,14 @@ private AtlasEdge getEdgeUsingRelationship(AttributeMutationContext ctx, EntityM
                 AtlasEdge edge = relationshipStore.getRelationship(fromVertex, toVertex, new AtlasRelationship(relationshipName));
 
                 if (edge != null && getStatus(edge) != DELETED) {
-                    return edge;
+                    ret = edge;
                 }
+
+                RequestContext requestContext = RequestContext.get();
+                requestContext.recordEntityUpdate(entityRetriever.toAtlasEntityHeader(toVertex));
             }
         }
+
         if (LOG.isDebugEnabled()) {
             LOG.debug("<== getEdgeUsingRelationship({})", ctx);
         }
@@ -2166,7 +2170,7 @@ public List removeArrayValue(AttributeMutationContext ctx, EntityMutationContext
             LOG.debug("<== removeArrayValue({})", ctx);
         }
 
-        return new ArrayList<>();
+        return entityRelationsDeleted;
     }
 
     private void addEdgesToContext(String guid, List<Object> newElementsCreated, List<AtlasEdge> removedElements) {