private static List<AtlasEntityDef> toAtlasEntityDefs(List<HierarchicalTypeDefinition<ClassType>> classTypeDefinitions, AtlasTypeRegistry registry) throws AtlasBaseException { List<AtlasEntityDef> atlasEntityDefs = new ArrayList<AtlasEntityDef>(); for (HierarchicalTypeDefinition<ClassType> classType : classTypeDefinitions) { List<AtlasAttributeDef> attrDefs = new ArrayList<AtlasAttributeDef>(); AtlasEntityDef atlasEntityDef = new AtlasEntityDef(); String classTypeDefName = classType.typeName; atlasEntityDef.setName(classTypeDefName); atlasEntityDef.setDescription(classType.typeDescription); atlasEntityDef.setTypeVersion(classType.typeVersion); atlasEntityDef.setSuperTypes(classType.superTypes); AttributeDefinition[] attrDefinitions = classType.attributeDefinitions; for (AttributeDefinition oldAttr : attrDefinitions) { AtlasAttributeDef newAttr = toAtlasAttributeDef(oldAttr); attrDefs.add(newAttr); } atlasEntityDef.setAttributeDefs(attrDefs); atlasEntityDefs.add(atlasEntityDef); } return atlasEntityDefs; }
private static List<AtlasStructDef> toAtlasStructDefs(List<StructTypeDefinition> structTypeDefinitions) throws AtlasBaseException { List<AtlasStructDef> ret = new ArrayList<AtlasStructDef>(); for (StructTypeDefinition structType : structTypeDefinitions) { AtlasStructDef structDef = new AtlasStructDef(); List<AtlasAttributeDef> attrDefs = new ArrayList<AtlasAttributeDef>(); structDef.setName(structType.typeName); structDef.setDescription(structType.typeDescription); structDef.setTypeVersion(structType.typeVersion); AttributeDefinition[] attrDefinitions = structType.attributeDefinitions; for (AttributeDefinition attrDefinition : attrDefinitions) { attrDefs.add(toAtlasAttributeDef(attrDefinition)); } structDef.setAttributeDefs(attrDefs); ret.add(structDef); } return ret; }
private static List<AtlasClassificationDef> toAtlasClassificationDefs(List<HierarchicalTypeDefinition<TraitType>> traitTypeDefinitions) throws AtlasBaseException { List<AtlasClassificationDef> ret = new ArrayList<AtlasClassificationDef>(); for (HierarchicalTypeDefinition<TraitType> traitType : traitTypeDefinitions) { AtlasClassificationDef classifDef = new AtlasClassificationDef(); List<AtlasAttributeDef> attrDefs = new ArrayList<AtlasAttributeDef>(); classifDef.setName(traitType.typeName); classifDef.setDescription(traitType.typeDescription); classifDef.setTypeVersion(traitType.typeVersion); classifDef.setSuperTypes(traitType.superTypes); AttributeDefinition[] attrDefinitions = traitType.attributeDefinitions; for (AttributeDefinition attrDefinition : attrDefinitions) { attrDefs.add(toAtlasAttributeDef(attrDefinition)); } classifDef.setAttributeDefs(attrDefs); ret.add(classifDef); } 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; }
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; }
public static AtlasAttributeDef toAtlasAttributeDef(final AttributeDefinition attrDefinition) { AtlasAttributeDef ret = new AtlasAttributeDef(); ret.setName(attrDefinition.name); ret.setTypeName(attrDefinition.dataTypeName); ret.setIsIndexable(attrDefinition.isIndexable);
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; }