Add bulk and remove attribute support in bulk API

atlanhq · Jan 17, 2024 · 71dbaeb · 71dbaeb
1 parent acbfee6
commit 71dbaeb
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
@@ -1770,10 +1770,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);
         }
@@ -2165,7 +2169,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) {