private void createClassificationNameIndexMap(List<AtlasClassificationDef> classificationDefs) { for (int i = 0; i < classificationDefs.size(); i++) { AtlasClassificationDef classificationDef = classificationDefs.get(i); classificationIndexMap.put(classificationDef.getName(), i); } }
@Override public AtlasClassification createDefaultValue() { AtlasClassification ret = new AtlasClassification(classificationDef.getName()); populateDefaultValues(ret); return ret; }
@Override public AtlasClassification createDefaultValue() { AtlasClassification ret = new AtlasClassification(classificationDef.getName()); populateDefaultValues(ret); return ret; }
@Override public AtlasClassificationDef update(AtlasClassificationDef classifiDef) throws AtlasBaseException { if (LOG.isDebugEnabled()) { LOG.debug("==> AtlasClassificationDefStoreV1.update({})", classifiDef); } validateType(classifiDef); AtlasClassificationDef ret = StringUtils.isNotBlank(classifiDef.getGuid()) ? updateByGuid(classifiDef.getGuid(), classifiDef) : updateByName(classifiDef.getName(), classifiDef); if (LOG.isDebugEnabled()) { LOG.debug("<== AtlasClassificationDefStoreV1.update({}): {}", classifiDef, ret); } return ret; }
@Override public AtlasClassificationDef update(AtlasClassificationDef classifiDef) throws AtlasBaseException { if (LOG.isDebugEnabled()) { LOG.debug("==> AtlasClassificationDefStoreV1.update({})", classifiDef); } validateType(classifiDef); AtlasClassificationDef ret = StringUtils.isNotBlank(classifiDef.getGuid()) ? updateByGuid(classifiDef.getGuid(), classifiDef) : updateByName(classifiDef.getName(), classifiDef); if (LOG.isDebugEnabled()) { LOG.debug("<== AtlasClassificationDefStoreV1.update({}): {}", classifiDef, ret); } return ret; }
private void updateClassificationDef(AtlasTypesDef typeDefinitionMap, AtlasImportResult result) throws AtlasBaseException { for (AtlasClassificationDef def : typeDefinitionMap.getClassificationDefs()) { AtlasClassificationDef existing = typeRegistry.getClassificationDefByName(def.getName()); if (existing != null && addAttributes(existing, def)) { typeDefStore.updateClassificationDefByName(existing.getName(), existing); result.incrementMeticsCounter("typedef:classification:update"); } } }
public static AtlasClassification newClassification(AtlasClassificationDef classificationDef, AtlasTypeRegistry typesRegistry) { AtlasClassification ret = null; AtlasClassificationType classificationType = typesRegistry.getClassificationTypeByName(classificationDef.getName()); if (classificationType != null) { ret = classificationType.createDefaultValue(); } else { LOG.error("failed to get classification-type {}", classificationDef.getName()); } return ret; }
public static AtlasClassification newClassification(AtlasClassificationDef classificationDef, AtlasTypeRegistry typesRegistry) { AtlasClassification ret = null; AtlasClassificationType classificationType = typesRegistry.getClassificationTypeByName(classificationDef.getName()); if (classificationType != null) { ret = classificationType.createDefaultValue(); } else { LOG.error("failed to get classification-type {}", classificationDef.getName()); } return ret; }
private ClassificationToStructDefName createLegacyStructDefFromClassification(String typeName) { AtlasClassificationDef classificationDef = getClassificationDefByName(typeName); if (classificationDef == null) { return null; } AtlasStructDef structDef = getStructDefFromClassificationDef(classificationDef); addStructDefToTypesDef(structDef); return new ClassificationToStructDefName(classificationDef.getName(), structDef.getName()); }
private static TypesDef classificationToTypesDef(AtlasClassificationType classificationType, AtlasTypeRegistry registry) { String typeName = classificationType.getClassificationDef().getName(); String typeDesc = classificationType.getClassificationDef().getDescription(); String typeVersion = classificationType.getClassificationDef().getTypeVersion(); Set<String> superTypes = new HashSet<>(classificationType.getClassificationDef().getSuperTypes()); List<AttributeDefinition> attributes = getAttributes(classificationType, registry); TraitTypeDefinition traitTypeDef = new TraitTypeDefinition(typeName, typeDesc, typeVersion, attributes, superTypes); TypesDef ret = new TypesDef(null, null, Arrays.asList(traitTypeDef), null); return ret; }
private AtlasStructDef getStructDefFromClassificationDef(AtlasClassificationDef classificationDef) { String legacyTypeName = getLegacyTypeNameForStructDef(classificationDef.getName()); return new AtlasStructDef(legacyTypeName, classificationDef.getDescription(), classificationDef.getTypeVersion(), getDefaultAttributeDefsIfNecessary(classificationDef.getAttributeDefs())); }
private void updateClassificationDef(AtlasTypesDef typeDefinitionMap, AtlasImportResult result) throws AtlasBaseException { for (AtlasClassificationDef def: typeDefinitionMap.getClassificationDefs()) { AtlasClassificationDef existing = typeRegistry.getClassificationDefByName(def.getName()); if(existing != null && addAttributes(existing, def)) { typeDefStore.updateClassificationDefByName(existing.getName(), existing); result.incrementMeticsCounter("typedef:classification:update"); } } }
private void updateEnumDef(AtlasTypesDef typeDefinitionMap, AtlasImportResult result) throws AtlasBaseException { for (AtlasEnumDef def: typeDefinitionMap.getEnumDefs()) { AtlasEnumDef existing = typeRegistry.getEnumDefByName(def.getName()); if(existing != null && addElements(existing, def)) { typeDefStore.updateEnumDefByName(existing.getName(), existing); result.incrementMeticsCounter("typedef:enum:update"); } } }
private void assertTraitMap(TypesDefScrubber typesDefScrubber, AtlasTypesDef td, int classificationIndex, String expectedStructName, int attrIndex) { String label = typesDefScrubber.getEdgeLabel(td.getEntityDefs().get(0).getName(), td.getEntityDefs().get(0).getAttributeDefs().get(attrIndex).getName()); assertTrue(typesDefScrubber.getTraitToTypeMap().containsKey(label)); assertEquals(typesDefScrubber.getTraitToTypeMap().get(label).getTypeName(), td.getClassificationDefs().get(classificationIndex).getName()); assertEquals(typesDefScrubber.getTraitToTypeMap().get(label).getLegacyTypeName(), expectedStructName); }
private void assertTraitMap(TypesDefScrubber typesDefScrubber, AtlasTypesDef td, int classificationIndex, String expectedStructName, int attrIndex) { String label = typesDefScrubber.getEdgeLabel(td.getEntityDefs().get(0).getName(), td.getEntityDefs().get(0).getAttributeDefs().get(attrIndex).getName()); assertTrue(typesDefScrubber.getTraitToTypeMap().containsKey(label)); assertEquals(typesDefScrubber.getTraitToTypeMap().get(label).getTypeName(), td.getClassificationDefs().get(classificationIndex).getName()); assertEquals(typesDefScrubber.getTraitToTypeMap().get(label).getLegacyTypeName(), expectedStructName); }
@Test public void testClassificationSerDe() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDef(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationDef.getName()); assertNotNull(classificationType); AtlasClassification ent1 = ModelTestUtil.newClassification(classificationDef, typeRegistry); String jsonString = AtlasType.toJson(ent1); AtlasClassification ent2 = AtlasType.fromJson(jsonString, AtlasClassification.class); classificationType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasClassification"); }
@Test public void testClassificationSerDe() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDef(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationDef.getName()); assertNotNull(classificationType); AtlasClassification ent1 = ModelTestUtil.newClassification(classificationDef, typeRegistry); String jsonString = AtlasType.toJson(ent1); AtlasClassification ent2 = AtlasType.fromJson(jsonString, AtlasClassification.class); classificationType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasClassification"); }
@Test public void testClassificationSerDeWithSuperType() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperType(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationDef.getName()); assertNotNull(classificationType); AtlasClassification ent1 = classificationType.createDefaultValue(); String jsonString = AtlasType.toJson(ent1); AtlasClassification ent2 = AtlasType.fromJson(jsonString, AtlasClassification.class); classificationType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasClassification with superType"); }
@Test public void testClassificationSerDeWithSuperType() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperType(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationDef.getName()); assertNotNull(classificationType); AtlasClassification ent1 = classificationType.createDefaultValue(); String jsonString = AtlasType.toJson(ent1); AtlasClassification ent2 = AtlasType.fromJson(jsonString, AtlasClassification.class); classificationType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasClassification with superType"); }
@Test public void testClassificationSerDeWithSuperTypes() throws AtlasBaseException { AtlasClassificationDef classificationDef = ModelTestUtil.getClassificationDefWithSuperTypes(); AtlasTypeRegistry typeRegistry = ModelTestUtil.getTypesRegistry(); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationDef.getName()); assertNotNull(classificationType); AtlasClassification ent1 = classificationType.createDefaultValue(); String jsonString = AtlasType.toJson(ent1); AtlasClassification ent2 = AtlasType.fromJson(jsonString, AtlasClassification.class); classificationType.normalizeAttributeValues(ent2); assertEquals(ent2, ent1, "Incorrect serialization/deserialization of AtlasClassification with superTypes"); } }