Skip to content

A
atlas
Commit ce5ffeb7 by Graham Wallis Committed by David Radley
Options

ATLAS-2745 - AtlasEnumDefStore should remember default value

Signed-off-by: 's avatarDavid Radley <david_radley@uk.ibm.com>
parent 7174257b
Showing with 25 additions and 17 deletions
repository/src/main/java/org/apache/atlas/repository/store/graph/v2/AtlasEnumDefStoreV2.java
...@@ -37,7 +37,7 @@ import java.util.Iterator; ...@@ -37,7 +37,7 @@ import java.util.Iterator;
import java.util.List; import java.util.List;
/** /**
* EnumDef store in v1 format. * EnumDef store in v2 format.
*/ */
class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
private static final Logger LOG = LoggerFactory.getLogger(AtlasEnumDefStoreV2.class); private static final Logger LOG = LoggerFactory.getLogger(AtlasEnumDefStoreV2.class);
...@@ -49,7 +49,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -49,7 +49,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public AtlasVertex preCreate(AtlasEnumDef enumDef) throws AtlasBaseException { public AtlasVertex preCreate(AtlasEnumDef enumDef) throws AtlasBaseException {
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("==> AtlasEnumDefStoreV1.preCreate({})", enumDef); LOG.debug("==> AtlasEnumDefStoreV2.preCreate({})", enumDef);
} }
validateType(enumDef); validateType(enumDef);
...@@ -65,7 +65,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -65,7 +65,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
toVertex(enumDef, vertex); toVertex(enumDef, vertex);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEnumDefStoreV1.preCreate({}): {}", enumDef, vertex); LOG.debug("<== AtlasEnumDefStoreV2.preCreate({}): {}", enumDef, vertex);
} }
return vertex; return vertex;
...@@ -74,7 +74,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -74,7 +74,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public AtlasEnumDef create(AtlasEnumDef enumDef, AtlasVertex preCreateResult) throws AtlasBaseException { public AtlasEnumDef create(AtlasEnumDef enumDef, AtlasVertex preCreateResult) throws AtlasBaseException {
if (LOG.isDebugEnabled()) { 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()); AtlasAuthorizationUtils.verifyAccess(new AtlasTypeAccessRequest(AtlasPrivilege.TYPE_CREATE, enumDef), "create enum-def ", enumDef.getName());
...@@ -84,7 +84,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -84,7 +84,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
AtlasEnumDef ret = toEnumDef(vertex); AtlasEnumDef ret = toEnumDef(vertex);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEntityDefStoreV1.create({}, {}): {}", enumDef, preCreateResult, ret); LOG.debug("<== AtlasEntityDefStoreV2.create({}, {}): {}", enumDef, preCreateResult, ret);
} }
return ret; return ret;
...@@ -93,7 +93,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -93,7 +93,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public List<AtlasEnumDef> getAll() throws AtlasBaseException { public List<AtlasEnumDef> getAll() throws AtlasBaseException {
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("==> AtlasEnumDefStoreV1.getAll()"); LOG.debug("==> AtlasEnumDefStoreV2.getAll()");
} }
List<AtlasEnumDef> ret = new ArrayList<>(); List<AtlasEnumDef> ret = new ArrayList<>();
...@@ -104,7 +104,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -104,7 +104,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
} }
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEnumDefStoreV1.getAll(): count={}", ret.size()); LOG.debug("<== AtlasEnumDefStoreV2.getAll(): count={}", ret.size());
} }
return ret; return ret;
...@@ -113,7 +113,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -113,7 +113,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public AtlasEnumDef getByName(String name) throws AtlasBaseException { public AtlasEnumDef getByName(String name) throws AtlasBaseException {
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("==> AtlasEnumDefStoreV1.getByName({})", name); LOG.debug("==> AtlasEnumDefStoreV2.getByName({})", name);
} }
AtlasVertex vertex = typeDefStore.findTypeVertexByNameAndCategory(name, TypeCategory.ENUM); AtlasVertex vertex = typeDefStore.findTypeVertexByNameAndCategory(name, TypeCategory.ENUM);
...@@ -127,7 +127,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -127,7 +127,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
AtlasEnumDef ret = toEnumDef(vertex); AtlasEnumDef ret = toEnumDef(vertex);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEnumDefStoreV1.getByName({}): {}", name, ret); LOG.debug("<== AtlasEnumDefStoreV2.getByName({}): {}", name, ret);
} }
return ret; return ret;
...@@ -136,7 +136,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -136,7 +136,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public AtlasEnumDef getByGuid(String guid) throws AtlasBaseException { public AtlasEnumDef getByGuid(String guid) throws AtlasBaseException {
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("==> AtlasEnumDefStoreV1.getByGuid({})", guid); LOG.debug("==> AtlasEnumDefStoreV2.getByGuid({})", guid);
} }
AtlasVertex vertex = typeDefStore.findTypeVertexByGuidAndCategory(guid, TypeCategory.ENUM); AtlasVertex vertex = typeDefStore.findTypeVertexByGuidAndCategory(guid, TypeCategory.ENUM);
...@@ -148,7 +148,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -148,7 +148,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
AtlasEnumDef ret = toEnumDef(vertex); AtlasEnumDef ret = toEnumDef(vertex);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEnumDefStoreV1.getByGuid({}): {}", guid, ret); LOG.debug("<== AtlasEnumDefStoreV2.getByGuid({}): {}", guid, ret);
} }
return ret; return ret;
...@@ -157,7 +157,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -157,7 +157,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public AtlasEnumDef update(AtlasEnumDef enumDef) throws AtlasBaseException { public AtlasEnumDef update(AtlasEnumDef enumDef) throws AtlasBaseException {
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("==> AtlasEnumDefStoreV1.update({})", enumDef); LOG.debug("==> AtlasEnumDefStoreV2.update({})", enumDef);
} }
validateType(enumDef); validateType(enumDef);
...@@ -166,7 +166,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -166,7 +166,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
: updateByName(enumDef.getName(), enumDef); : updateByName(enumDef.getName(), enumDef);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEnumDefStoreV1.update({}): {}", enumDef, ret); LOG.debug("<== AtlasEnumDefStoreV2.update({}): {}", enumDef, ret);
} }
return ret; return ret;
...@@ -175,7 +175,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -175,7 +175,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public AtlasEnumDef updateByName(String name, AtlasEnumDef enumDef) throws AtlasBaseException { public AtlasEnumDef updateByName(String name, AtlasEnumDef enumDef) throws AtlasBaseException {
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("==> AtlasEnumDefStoreV1.updateByName({}, {})", name, enumDef); LOG.debug("==> AtlasEnumDefStoreV2.updateByName({}, {})", name, enumDef);
} }
AtlasEnumDef existingDef = typeRegistry.getEnumDefByName(name); AtlasEnumDef existingDef = typeRegistry.getEnumDefByName(name);
...@@ -197,7 +197,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -197,7 +197,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
AtlasEnumDef ret = toEnumDef(vertex); AtlasEnumDef ret = toEnumDef(vertex);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEnumDefStoreV1.updateByName({}, {}): {}", name, enumDef, ret); LOG.debug("<== AtlasEnumDefStoreV2.updateByName({}, {}): {}", name, enumDef, ret);
} }
return ret; return ret;
...@@ -206,7 +206,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -206,7 +206,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
@Override @Override
public AtlasEnumDef updateByGuid(String guid, AtlasEnumDef enumDef) throws AtlasBaseException { public AtlasEnumDef updateByGuid(String guid, AtlasEnumDef enumDef) throws AtlasBaseException {
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("==> AtlasEnumDefStoreV1.updateByGuid({})", guid); LOG.debug("==> AtlasEnumDefStoreV2.updateByGuid({})", guid);
} }
AtlasEnumDef existingDef = typeRegistry.getEnumDefByGuid(guid); AtlasEnumDef existingDef = typeRegistry.getEnumDefByGuid(guid);
...@@ -228,7 +228,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -228,7 +228,7 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
AtlasEnumDef ret = toEnumDef(vertex); AtlasEnumDef ret = toEnumDef(vertex);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("<== AtlasEnumDefStoreV1.updateByGuid({}): {}", guid, ret); LOG.debug("<== AtlasEnumDefStoreV2.updateByGuid({}): {}", guid, ret);
} }
return ret; return ret;
...@@ -290,6 +290,10 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -290,6 +290,10 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
values.add(element.getValue()); values.add(element.getValue());
} }
AtlasGraphUtilsV2.setProperty(vertex, AtlasGraphUtilsV2.getTypeDefPropertyKey(enumDef), values); AtlasGraphUtilsV2.setProperty(vertex, AtlasGraphUtilsV2.getTypeDefPropertyKey(enumDef), values);
String defaultValueKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(enumDef, "defaultValue");
AtlasGraphUtilsV2.setProperty(vertex, defaultValueKey, enumDef.getDefaultValue());
} }
private AtlasEnumDef toEnumDef(AtlasVertex vertex) { private AtlasEnumDef toEnumDef(AtlasVertex vertex) {
...@@ -320,6 +324,10 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> { ...@@ -320,6 +324,10 @@ class AtlasEnumDefStoreV2 extends AtlasAbstractDefStoreV2<AtlasEnumDef> {
} }
ret.setElementDefs(elements); ret.setElementDefs(elements);
String defaultValueKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(ret, "defaultValue");
String defaultValue = AtlasGraphUtilsV2.getProperty(vertex, defaultValueKey, String.class);
ret.setDefaultValue(defaultValue);
return ret; return ret;
} }
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment