public AtlasEnumDef(String name, String description, String typeVersion, List<AtlasEnumElementDef> elementDefs, String defaultValue, Map<String, String> options) { super(TypeCategory.ENUM, name, description, typeVersion, options); setElementDefs(elementDefs); setDefaultValue(defaultValue); }
public AtlasEnumDef(String name, String description, String typeVersion, List<AtlasEnumElementDef> elementDefs, String defaultValue, Map<String, String> options) { super(TypeCategory.ENUM, name, description, typeVersion, options); setElementDefs(elementDefs); setDefaultValue(defaultValue); }
public AtlasEnumDef(AtlasEnumDef other) { super(other); if (other != null) { setElementDefs(other.getElementDefs()); setDefaultValue(other.getDefaultValue()); } }
public AtlasEnumDef(AtlasEnumDef other) { super(other); if (other != null) { setElementDefs(other.getElementDefs()); setDefaultValue(other.getDefaultValue()); } }
private static List<AtlasEnumDef> toAtlasEnumDefs(List<EnumTypeDefinition> enumTypeDefinitions) { List<AtlasEnumDef> ret = new ArrayList<AtlasEnumDef>(); for (EnumTypeDefinition enumType : enumTypeDefinitions) { AtlasEnumDef enumDef = new AtlasEnumDef(); enumDef.setName(enumType.name); enumDef.setDescription(enumType.description); enumDef.setTypeVersion(enumType.version); enumDef.setElementDefs(getAtlasEnumElementDefs(enumType.enumValues)); ret.add(enumDef); } return ret; }
private static List<AtlasEnumDef> toAtlasEnumDefs(List<EnumTypeDefinition> enumTypeDefinitions) { List<AtlasEnumDef> ret = new ArrayList<AtlasEnumDef>(); for (EnumTypeDefinition enumType : enumTypeDefinitions) { AtlasEnumDef enumDef = new AtlasEnumDef(); enumDef.setName(enumType.getName()); enumDef.setDescription(enumType.getDescription()); enumDef.setTypeVersion(enumType.getVersion()); enumDef.setElementDefs(getAtlasEnumElementDefs(enumType.getEnumValues())); ret.add(enumDef); } return ret; }
@Test public void testEnumDefSetElementDefs() { AtlasEnumDef enumDef = ModelTestUtil.newEnumDef(); List<AtlasEnumElementDef> oldElements = enumDef.getElementDefs(); List<AtlasEnumElementDef> newElements = new ArrayList<>(); newElements.add(new AtlasEnumElementDef("newElement", "new Element", 100)); enumDef.setElementDefs(newElements); for (AtlasEnumElementDef elementDef : oldElements) { assertFalse(enumDef.hasElement(elementDef.getValue())); } for (AtlasEnumElementDef elementDef : newElements) { assertTrue(enumDef.hasElement(elementDef.getValue())); } } }
@Test public void testEnumDefSetElementDefs() { AtlasEnumDef enumDef = ModelTestUtil.newEnumDef(); List<AtlasEnumElementDef> oldElements = enumDef.getElementDefs(); List<AtlasEnumElementDef> newElements = new ArrayList<>(); newElements.add(new AtlasEnumElementDef("newElement", "new Element", 100)); enumDef.setElementDefs(newElements); for (AtlasEnumElementDef elementDef : oldElements) { assertFalse(enumDef.hasElement(elementDef.getValue())); } for (AtlasEnumElementDef elementDef : newElements) { assertTrue(enumDef.hasElement(elementDef.getValue())); } } }
private static AtlasEnumDef toEnumDef(AtlasVertex vertex, AtlasEnumDef enumDef, AtlasTypeDefGraphStoreV2 typeDefStore) { AtlasEnumDef ret = enumDef != null ? enumDef : new AtlasEnumDef(); typeDefStore.vertexToTypeDef(vertex, ret); List<AtlasEnumElementDef> elements = new ArrayList<>(); List<String> elemValues = vertex.getProperty(AtlasGraphUtilsV2.getTypeDefPropertyKey(ret), List.class); for (String elemValue : elemValues) { String elemKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(ret, elemValue); String descKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(elemKey, "description"); Integer ordinal = AtlasGraphUtilsV2.getProperty(vertex, elemKey, Integer.class); String desc = AtlasGraphUtilsV2.getProperty(vertex, descKey, String.class); elements.add(new AtlasEnumElementDef(elemValue, desc, ordinal)); } ret.setElementDefs(elements); String defaultValueKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(ret, "defaultValue"); String defaultValue = AtlasGraphUtilsV2.getProperty(vertex, defaultValueKey, String.class); ret.setDefaultValue(defaultValue); return ret; } }
private static AtlasEnumDef toEnumDef(AtlasVertex vertex, AtlasEnumDef enumDef, AtlasTypeDefGraphStoreV2 typeDefStore) { AtlasEnumDef ret = enumDef != null ? enumDef : new AtlasEnumDef(); typeDefStore.vertexToTypeDef(vertex, ret); List<AtlasEnumElementDef> elements = new ArrayList<>(); List<String> elemValues = vertex.getProperty(AtlasGraphUtilsV2.getTypeDefPropertyKey(ret), List.class); for (String elemValue : elemValues) { String elemKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(ret, elemValue); String descKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(elemKey, "description"); Integer ordinal = AtlasGraphUtilsV2.getProperty(vertex, elemKey, Integer.class); String desc = AtlasGraphUtilsV2.getProperty(vertex, descKey, String.class); elements.add(new AtlasEnumElementDef(elemValue, desc, ordinal)); } ret.setElementDefs(elements); String defaultValueKey = AtlasGraphUtilsV2.getTypeDefPropertyKey(ret, "defaultValue"); String defaultValue = AtlasGraphUtilsV2.getProperty(vertex, defaultValueKey, String.class); ret.setDefaultValue(defaultValue); return ret; } }