public static AtlasEnumDef createEnumTypeDef(String name, String description, AtlasEnumElementDef... enumValues) { return new AtlasEnumDef(name, description, "1.0", Arrays.asList(enumValues)); }
public static AtlasEnumDef createEnumTypeDef(String name, String description, AtlasEnumElementDef... enumValues) { return new AtlasEnumDef(name, description, "1.0", Arrays.asList(enumValues)); }
private AtlasEnumDef toEnumDef(AtlasVertex vertex) { AtlasEnumDef ret = null; if (vertex != null && typeDefStore.isTypeVertex(vertex, TypeCategory.ENUM)) { ret = toEnumDef(vertex, new AtlasEnumDef(), typeDefStore); } return ret; }
private AtlasEnumDef toEnumDef(AtlasVertex vertex) { AtlasEnumDef ret = null; if (vertex != null && typeDefStore.isTypeVertex(vertex, TypeCategory.ENUM)) { ret = toEnumDef(vertex, new AtlasEnumDef(), typeDefStore); } 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.name); enumDef.setDescription(enumType.description); enumDef.setTypeVersion(enumType.version); enumDef.setElementDefs(getAtlasEnumElementDefs(enumType.enumValues)); ret.add(enumDef); } return ret; }
@Test public void testEnumDefSerDeEmpty() { AtlasEnumDef enumDef1 = new AtlasEnumDef(); String jsonString = AtlasType.toJson(enumDef1); AtlasEnumDef enumDef2 = AtlasType.fromJson(jsonString, AtlasEnumDef.class); assertEquals(enumDef1, enumDef2, "Incorrect serialization/deserialization of AtlasEnumDef"); }
@Test public void testEnumDefSerDeEmpty() { AtlasEnumDef enumDef1 = new AtlasEnumDef(); String jsonString = AtlasType.toJson(enumDef1); AtlasEnumDef enumDef2 = AtlasType.fromJson(jsonString, AtlasEnumDef.class); assertEquals(enumDef1, enumDef2, "Incorrect serialization/deserialization of AtlasEnumDef"); }
private AtlasEnumDef getAtlasEnumWithAttributes(String... elements) { AtlasEnumDef enumDef = new AtlasEnumDef(); for (AtlasEnumDef.AtlasEnumElementDef ed : getEnumElementDefs(0, elements)) { enumDef.addElement(ed); } return enumDef; }
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; }
private AtlasEnumDef getAtlasEnumWithAttributes(String... elements) { AtlasEnumDef enumDef = new AtlasEnumDef(); for (AtlasEnumDef.AtlasEnumElementDef ed : getEnumElementDefs(0, elements)) { enumDef.addElement(ed); } return enumDef; }
public static AtlasTypesDef simpleType(){ AtlasEntityDef superTypeDefinition = createClassTypeDef("h_type", Collections.<String>emptySet(), createOptionalAttrDef("attr", "string")); AtlasStructDef structTypeDefinition = new AtlasStructDef("s_type", "structType", "1.0", Arrays.asList(createRequiredAttrDef("name", "string"))); AtlasClassificationDef traitTypeDefinition = AtlasTypeUtil.createTraitTypeDef("t_type", "traitType", Collections.<String>emptySet()); AtlasEnumDef enumTypeDefinition = new AtlasEnumDef("e_type", "enumType", "1.0", Arrays.asList(new AtlasEnumElementDef("ONE", "Element Description", 1))); AtlasTypesDef ret = AtlasTypeUtil.getTypesDef(Collections.singletonList(enumTypeDefinition), Collections.singletonList(structTypeDefinition), Collections.singletonList(traitTypeDefinition), Collections.singletonList(superTypeDefinition)); populateSystemAttributes(ret); 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; } }
public static AtlasTypesDef simpleTypeUpdated(){ AtlasEntityDef superTypeDefinition = createClassTypeDef("h_type", Collections.<String>emptySet(), createOptionalAttrDef("attr", "string")); AtlasEntityDef newSuperTypeDefinition = createClassTypeDef("new_h_type", Collections.<String>emptySet(), createOptionalAttrDef("attr", "string")); AtlasStructDef structTypeDefinition = new AtlasStructDef("s_type", "structType", "1.0", Arrays.asList(createRequiredAttrDef("name", "string"))); AtlasClassificationDef traitTypeDefinition = AtlasTypeUtil.createTraitTypeDef("t_type", "traitType", Collections.<String>emptySet()); AtlasEnumDef enumTypeDefinition = new AtlasEnumDef("e_type", "enumType", Arrays.asList(new AtlasEnumElementDef("ONE", "Element Description", 1))); AtlasTypesDef ret = AtlasTypeUtil.getTypesDef(Collections.singletonList(enumTypeDefinition), Collections.singletonList(structTypeDefinition), Collections.singletonList(traitTypeDefinition), Arrays.asList(superTypeDefinition, newSuperTypeDefinition)); populateSystemAttributes(ret); return ret; }
@Override public Object call() throws Exception { for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasEnumDef(enumTypePrefix + i)); } for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasStructDef(structTypePrefix + i)); } for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasClassificationDef(classificationPrefix + i)); } for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasEntityDef(entityTypePrefix + i)); } return null; } });
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; } }
@Override public Object call() throws Exception { for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasEnumDef(enumTypePrefix + i)); } for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasStructDef(structTypePrefix + i)); } for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasClassificationDef(classificationPrefix + i)); } for (int i = 0; i < numOfTypesPerKind; i++) { addType(typeRegistry, new AtlasEntityDef(entityTypePrefix + i)); } return null; } });
public static AtlasEnumDef newEnumDef(AtlasTypeRegistry typesRegistry, boolean hasDefaultValue) { int enumDefIdx = IDX_ENUM_DEF.getAndIncrement(); AtlasEnumDef ret = new AtlasEnumDef(); ret.setName(PREFIX_ENUM_DEF + enumDefIdx); ret.setDescription(ret.getName()); int numElements = ThreadLocalRandom.current().nextInt(1, MAX_ENUM_ELEMENT_COUNT); for (int i = 0; i < numElements; i++) { String elementName = "element-" + i; ret.addElement(new AtlasEnumElementDef(elementName, elementName.toUpperCase(), i)); } if (hasDefaultValue) { int idxDefault = ThreadLocalRandom.current().nextInt(0, numElements); ret.setDefaultValue(ret.getElementDefs().get(idxDefault).getValue()); } AtlasTransientTypeRegistry ttr = null; boolean commit = false; try { ttr = typesRegistry.lockTypeRegistryForUpdate(); ttr.addType(ret); commit = true; } catch (AtlasBaseException excp) { LOG.error("failed to create enum-def", excp); ret = null; } finally { typesRegistry.releaseTypeRegistryForUpdate(ttr, commit); } return ret; }
public static AtlasEnumDef newEnumDef(AtlasTypeRegistry typesRegistry, boolean hasDefaultValue) { int enumDefIdx = IDX_ENUM_DEF.getAndIncrement(); AtlasEnumDef ret = new AtlasEnumDef(); ret.setName(PREFIX_ENUM_DEF + enumDefIdx); ret.setDescription(ret.getName()); int numElements = ThreadLocalRandom.current().nextInt(1, MAX_ENUM_ELEMENT_COUNT); for (int i = 0; i < numElements; i++) { String elementName = "element-" + i; ret.addElement(new AtlasEnumElementDef(elementName, elementName.toUpperCase(), i)); } if (hasDefaultValue) { int idxDefault = ThreadLocalRandom.current().nextInt(0, numElements); ret.setDefaultValue(ret.getElementDefs().get(idxDefault).getValue()); } AtlasTransientTypeRegistry ttr = null; boolean commit = false; try { ttr = typesRegistry.lockTypeRegistryForUpdate(); ttr.addType(ret); commit = true; } catch (AtlasBaseException excp) { LOG.error("failed to create enum-def", excp); ret = null; } finally { typesRegistry.releaseTypeRegistryForUpdate(ttr, commit); } return ret; }
AtlasEnumDef updatedDef = new AtlasEnumDef((AtlasEnumDef)typeDef);
createRequiredAttrDef("city", "string")); AtlasEnumDef orgLevelType = new AtlasEnumDef(ORG_LEVEL_TYPE, description(ORG_LEVEL_TYPE), DEFAULT_VERSION, getOrgLevelElements());