You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@atlas.apache.org by ma...@apache.org on 2018/06/10 15:17:43 UTC

[3/3] atlas git commit: ATLAS-2745 - AtlasEnumDefStore should remember default value

ATLAS-2745 - AtlasEnumDefStore should remember default value

Signed-off-by: David Radley <da...@uk.ibm.com>
(cherry picked from commit ce5ffeb710721b78135e8b3c3ebe593bd70d2cdf)


Project: http://git-wip-us.apache.org/repos/asf/atlas/repo
Commit: http://git-wip-us.apache.org/repos/asf/atlas/commit/d5ca279b
Tree: http://git-wip-us.apache.org/repos/asf/atlas/tree/d5ca279b
Diff: http://git-wip-us.apache.org/repos/asf/atlas/diff/d5ca279b

Branch: refs/heads/branch-1.0
Commit: d5ca279bcf3c23d6e0e8e9a9dd2fbca4199f83b7
Parents: 6acbb02
Author: Graham Wallis <gr...@uk.ibm.com>
Authored: Thu Jun 7 14:25:53 2018 +0100
Committer: Madhan Neethiraj <ma...@apache.org>
Committed: Sun Jun 10 08:06:08 2018 -0700

----------------------------------------------------------------------
 .../store/graph/v2/AtlasEnumDefStoreV2.java     | 42 ++++++++++++--------
 1 file changed, 25 insertions(+), 17 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/atlas/blob/d5ca279b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEnumDefStoreV2.java
----------------------------------------------------------------------
diff --git a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEnumDefStoreV2.java b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEnumDefStoreV2.java
index 7d7233f..2abfcf8 100644
--- a/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEnumDefStoreV2.java
+++ b/repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEnumDefStoreV2.java
@@ -37,7 +37,7 @@ import java.util.Iterator;
 import java.util.List;
 
 /**
- * EnumDef store in v1 format.
+ * EnumDef store in v2 format.
  */
 class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     private static final Logger LOG = LoggerFactory.getLogger(AtlasEnumDefStoreV2.class);
@@ -49,7 +49,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public AtlasVertex preCreate(AtlasEnumDef enumDef) throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-          LOG.debug("==> AtlasEnumDefStoreV1.preCreate({})", enumDef);
+          LOG.debug("==> AtlasEnumDefStoreV2.preCreate({})", enumDef);
         }
 
         validateType(enumDef);
@@ -65,7 +65,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         toVertex(enumDef, vertex);
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEnumDefStoreV1.preCreate({}): {}", enumDef, vertex);
+            LOG.debug("<== AtlasEnumDefStoreV2.preCreate({}): {}", enumDef, vertex);
         }
 
         return vertex;
@@ -74,7 +74,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public AtlasEnumDef create(AtlasEnumDef enumDef, AtlasVertex preCreateResult) throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-          LOG.debug("==> AtlasEnumDefStoreV1.create({}, {})", enumDef, preCreateResult);
+          LOG.debug("==> AtlasEnumDefStoreV2.create({}, {})", enumDef, preCreateResult);
         }
 
         AtlasAuthorizationUtils.verifyAccess(new AtlasTypeAccessRequest(AtlasPrivilege.TYPE_CREATE, enumDef), "create enum-def ", enumDef.getName());
@@ -84,7 +84,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         AtlasEnumDef ret = toEnumDef(vertex);
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEntityDefStoreV1.create({}, {}): {}", enumDef, preCreateResult, ret);
+            LOG.debug("<== AtlasEntityDefStoreV2.create({}, {}): {}", enumDef, preCreateResult, ret);
         }
 
         return ret;
@@ -93,7 +93,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public List<AtlasEnumDef> getAll() throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-            LOG.debug("==> AtlasEnumDefStoreV1.getAll()");
+            LOG.debug("==> AtlasEnumDefStoreV2.getAll()");
         }
 
         List<AtlasEnumDef> ret = new ArrayList<>();
@@ -104,7 +104,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         }
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEnumDefStoreV1.getAll(): count={}", ret.size());
+            LOG.debug("<== AtlasEnumDefStoreV2.getAll(): count={}", ret.size());
         }
 
         return ret;
@@ -113,7 +113,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public AtlasEnumDef getByName(String name) throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-            LOG.debug("==> AtlasEnumDefStoreV1.getByName({})", name);
+            LOG.debug("==> AtlasEnumDefStoreV2.getByName({})", name);
         }
 
         AtlasVertex vertex = typeDefStore.findTypeVertexByNameAndCategory(name, TypeCategory.ENUM);
@@ -127,7 +127,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         AtlasEnumDef ret = toEnumDef(vertex);
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEnumDefStoreV1.getByName({}): {}", name, ret);
+            LOG.debug("<== AtlasEnumDefStoreV2.getByName({}): {}", name, ret);
         }
 
         return ret;
@@ -136,7 +136,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public AtlasEnumDef getByGuid(String guid) throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-            LOG.debug("==> AtlasEnumDefStoreV1.getByGuid({})", guid);
+            LOG.debug("==> AtlasEnumDefStoreV2.getByGuid({})", guid);
         }
 
         AtlasVertex vertex = typeDefStore.findTypeVertexByGuidAndCategory(guid, TypeCategory.ENUM);
@@ -148,7 +148,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         AtlasEnumDef ret = toEnumDef(vertex);
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEnumDefStoreV1.getByGuid({}): {}", guid, ret);
+            LOG.debug("<== AtlasEnumDefStoreV2.getByGuid({}): {}", guid, ret);
         }
 
         return ret;
@@ -157,7 +157,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public AtlasEnumDef update(AtlasEnumDef enumDef) throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-            LOG.debug("==> AtlasEnumDefStoreV1.update({})", enumDef);
+            LOG.debug("==> AtlasEnumDefStoreV2.update({})", enumDef);
         }
 
         validateType(enumDef);
@@ -166,7 +166,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
                                                                      : updateByName(enumDef.getName(), enumDef);
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEnumDefStoreV1.update({}): {}", enumDef, ret);
+            LOG.debug("<== AtlasEnumDefStoreV2.update({}): {}", enumDef, ret);
         }
 
         return ret;
@@ -175,7 +175,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public AtlasEnumDef updateByName(String name, AtlasEnumDef enumDef) throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-            LOG.debug("==> AtlasEnumDefStoreV1.updateByName({}, {})", name, enumDef);
+            LOG.debug("==> AtlasEnumDefStoreV2.updateByName({}, {})", name, enumDef);
         }
 
         AtlasEnumDef existingDef = typeRegistry.getEnumDefByName(name);
@@ -197,7 +197,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         AtlasEnumDef ret = toEnumDef(vertex);
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEnumDefStoreV1.updateByName({}, {}): {}", name, enumDef, ret);
+            LOG.debug("<== AtlasEnumDefStoreV2.updateByName({}, {}): {}", name, enumDef, ret);
         }
 
         return ret;
@@ -206,7 +206,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
     @Override
     public AtlasEnumDef updateByGuid(String guid, AtlasEnumDef enumDef) throws AtlasBaseException {
         if (LOG.isDebugEnabled()) {
-            LOG.debug("==> AtlasEnumDefStoreV1.updateByGuid({})", guid);
+            LOG.debug("==> AtlasEnumDefStoreV2.updateByGuid({})", guid);
         }
 
         AtlasEnumDef existingDef = typeRegistry.getEnumDefByGuid(guid);
@@ -228,7 +228,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         AtlasEnumDef ret = toEnumDef(vertex);
 
         if (LOG.isDebugEnabled()) {
-            LOG.debug("<== AtlasEnumDefStoreV1.updateByGuid({}): {}", guid, ret);
+            LOG.debug("<== AtlasEnumDefStoreV2.updateByGuid({}): {}", guid, ret);
         }
 
         return ret;
@@ -290,6 +290,10 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
             values.add(element.getValue());
         }
         AtlasGraphUtilsV2.setProperty(vertex, AtlasGraphUtilsV2.getTypeDefPropertyKey(enumDef), values);
+
+        String defaultValueKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(enumDef, "defaultValue");
+        AtlasGraphUtilsV2.setProperty(vertex, defaultValueKey, enumDef.getDefaultValue());
+
     }
 
     private AtlasEnumDef toEnumDef(AtlasVertex vertex) {
@@ -320,6 +324,10 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
         }
         ret.setElementDefs(elements);
 
+        String defaultValueKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(ret, "defaultValue");
+        String defaultValue = AtlasGraphUtilsV2.getProperty(vertex, defaultValueKey, String.class);
+        ret.setDefaultValue(defaultValue);
+
         return ret;
     }
 }