ATLAS-3084: setting default value to the relationship attributes

Signed-off-by: Madhan Neethiraj <madhan@apache.org>

ATLAS-3084: setting default value to the relationship attributes
f070d924 · kirankumar D G · Madhan Neethiraj · 39b059e0 · f070d924
Commit f070d924 authored Apr 10, 2019 by kirankumar D G Committed by Madhan Neethiraj Apr 10, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 11 additions and 2 deletions

AtlasRelationshipStoreV2.java ...s/repository/store/graph/v2/AtlasRelationshipStoreV2.java +11 -2

No files found.
--- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasRelationshipStoreV2.java
+++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasRelationshipStoreV2.java
@@ -427,6 +427,16 @@ public class AtlasRelationshipStoreV2 implements AtlasRelationshipStore {

                if (relationship.hasAttribute(attrName)) {
                    AtlasGraphUtilsV2.setEncodedProperty(relationshipEdge, attrVertexProperty, relationship.getAttribute(attrName));
+                } else {
+                    String defaultValue = attr.getAttributeDef().getDefaultValue();
+
+                    if (StringUtils.isNotEmpty(defaultValue)) {
+                        Object attrValue = attr.getAttributeType().createDefaultValue(defaultValue);
+
+                        if (attrValue != null) {
+                            AtlasGraphUtilsV2.setEncodedProperty(relationshipEdge, attrVertexProperty, attrValue);
+                        }
+                    }
                }
            }
        }
@@ -880,4 +890,4 @@ public class AtlasRelationshipStoreV2 implements AtlasRelationshipStore {
            entityChangeNotifier.notifyRelationshipMutation(ret, relationshipUpdate);
        }
    }
-}
\ No newline at end of file
+}