Evaluate attrName in appendRelationshipAttribute

atlanhq · Dec 27, 2023 · 06a1467 · 06a1467
1 parent 1ab182a
commit 06a1467
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityGraphDiscoveryV2.java b/...src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEntityGraphDiscoveryV2.java
@@ -358,6 +358,7 @@ private void visitRelationships(AtlasEntityType entityType, AtlasEntity entity,
         for (String attrName : entityType.getRelationshipAttributes().keySet()) {
 
             // if attribute is not in 'relationshipAttributes', try 'attributes'
+            // if appendRelationshipAttribute exists try attrName there
             if (entity.hasRelationshipAttribute(attrName)) {
                 Object         attrVal          = entity.getRelationshipAttribute(attrName);
                 String         relationshipType = AtlasEntityUtil.getRelationshipType(attrVal);
@@ -374,6 +375,15 @@ private void visitRelationships(AtlasEntityType entityType, AtlasEntity entity,
                 visitAttribute(attribute.getAttributeType(), attrVal, entity.getGuid());
 
                 visitedAttributes.add(attrName);
+            } else if (entity.hasAppendRelationshipAttribute(attrName)){
+                Object         attrVal          = entity.getRelationshipAttribute(attrName);
+                String         relationshipType = AtlasEntityUtil.getRelationshipType(attrVal);
+                AtlasAttribute attribute        = entityType.getRelationshipAttribute(attrName, relationshipType);
+
+                visitAttribute(attribute.getAttributeType(), attrVal, entity.getGuid());
+
+                visitedAttributes.add(attrName);
+
             }
         }
     }