private static String getClassificationFilter(AtlasTypeRegistry typeRegistry, String classificationName, int maxTypesLengthInIdxQuery) { AtlasClassificationType type = typeRegistry.getClassificationTypeByName(classificationName); String typeAndSubTypesQryStr = type != null ? type.getTypeAndAllSubTypesQryStr() : null; if(StringUtils.isNotEmpty(typeAndSubTypesQryStr) && typeAndSubTypesQryStr.length() <= maxTypesLengthInIdxQuery) { return typeAndSubTypesQryStr; } return ""; }
private void addClassificationType(String typeName, ExportService.ExportContext context) { if (!context.classificationTypes.contains(typeName)) { AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(typeName); addClassificationType(classificationType, context); } }
private void addClassificationType(String typeName, ExportContext context) { if (!context.classificationTypes.contains(typeName)) { AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(typeName); addClassificationType(classificationType, context); } }
public Set<String> getClassificationTypeAndAllSuperTypes(String classificationName, AtlasTypeRegistry typeRegistry) { final Set<String> ret; if (classificationName == null) { ret = Collections.emptySet(); } else if (typeRegistry == null) { ret = Collections.singleton(classificationName); } else { AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classificationName); return classificationType != null ? classificationType.getTypeAndAllSuperTypes() : Collections.singleton(classificationName); } return ret; }
private AtlasClassificationType getClassificationType(String classificationName) { AtlasClassificationType ret; if (StringUtils.equals(classificationName, MATCH_ALL_WILDCARD_CLASSIFICATION.getTypeName())) { ret = MATCH_ALL_WILDCARD_CLASSIFICATION; } else if (StringUtils.equals(classificationName, MATCH_ALL_CLASSIFIED.getTypeName())) { ret = MATCH_ALL_CLASSIFIED; } else if (StringUtils.equals(classificationName, MATCH_ALL_NOT_CLASSIFIED.getTypeName())) { ret = MATCH_ALL_NOT_CLASSIFIED; } else { ret = typeRegistry.getClassificationTypeByName(classificationName); } return ret; }
private AtlasClassificationType ensureClassificationType(String typeName) throws AtlasBaseException { AtlasClassificationType ret = typeRegistry.getClassificationTypeByName(typeName); if (ret == null) { throw new AtlasBaseException(AtlasErrorCode.TYPE_NAME_INVALID, TypeCategory.CLASSIFICATION.name(), typeName); } return ret; }
@Override public void resolveReferencesPhase2(AtlasTypeRegistry typeRegistry) throws AtlasBaseException { super.resolveReferencesPhase2(typeRegistry); for (String superTypeName : allSuperTypes) { AtlasClassificationType superType = typeRegistry.getClassificationTypeByName(superTypeName); superType.addSubType(this); } }
@Override void resolveReferencesPhase2(AtlasTypeRegistry typeRegistry) throws AtlasBaseException { super.resolveReferencesPhase2(typeRegistry); for (AtlasClassificationType superType : superTypes) { superType.addSubType(this); } for (String superTypeName : allSuperTypes) { AtlasClassificationType superType = typeRegistry.getClassificationTypeByName(superTypeName); superType.addToAllSubTypes(this); } }
private void validateAndNormalize(AtlasClassification classification) throws AtlasBaseException { AtlasClassificationType type = typeRegistry.getClassificationTypeByName(classification.getTypeName()); if (type == null) { throw new AtlasBaseException(AtlasErrorCode.CLASSIFICATION_NOT_FOUND, classification.getTypeName()); } List<String> messages = new ArrayList<>(); type.validateValue(classification, classification.getTypeName(), messages); if (!messages.isEmpty()) { throw new AtlasBaseException(AtlasErrorCode.INVALID_PARAMETERS, messages); } type.getNormalizedValue(classification); }
private void validateAndNormalize(AtlasClassification classification) throws AtlasBaseException { AtlasClassificationType type = typeRegistry.getClassificationTypeByName(classification.getTypeName()); if (type == null) { throw new AtlasBaseException(AtlasErrorCode.CLASSIFICATION_NOT_FOUND, classification.getTypeName()); } List<String> messages = new ArrayList<>(); type.validateValue(classification, classification.getTypeName(), messages); if (!messages.isEmpty()) { throw new AtlasBaseException(AtlasErrorCode.INVALID_PARAMETERS, messages); } type.getNormalizedValue(classification); }
private AtlasVertex createClassificationVertex(AtlasClassification classification) { if (LOG.isDebugEnabled()) { LOG.debug("==> createVertex({})", classification.getTypeName()); } AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classification.getTypeName()); AtlasVertex ret = createStructVertex(classification); AtlasGraphUtilsV2.addEncodedProperty(ret, SUPER_TYPES_PROPERTY_KEY, classificationType.getAllSuperTypes()); AtlasGraphUtilsV2.setEncodedProperty(ret, CLASSIFICATION_ENTITY_GUID, classification.getEntityGuid()); AtlasGraphUtilsV2.setEncodedProperty(ret, CLASSIFICATION_ENTITY_STATUS, classification.getEntityStatus().name()); 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; }
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 AtlasVertex createClassificationVertex(AtlasClassification classification) { if (LOG.isDebugEnabled()) { LOG.debug("==> createVertex({})", classification.getTypeName()); } AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classification.getTypeName()); AtlasVertex ret = createStructVertex(classification); AtlasGraphUtilsV2.addEncodedProperty(ret, SUPER_TYPES_PROPERTY_KEY, classificationType.getAllSuperTypes()); AtlasGraphUtilsV2.setEncodedProperty(ret, CLASSIFICATION_ENTITY_GUID, classification.getEntityGuid()); AtlasGraphUtilsV2.setEncodedProperty(ret, CLASSIFICATION_ENTITY_STATUS, classification.getEntityStatus().name()); return ret; }
public AtlasClassification toAtlasClassification(Struct classification) throws AtlasBaseException { AtlasFormatConverter converter = instanceFormatters.getConverter(TypeCategory.CLASSIFICATION); AtlasClassificationType classificationType = typeRegistry.getClassificationTypeByName(classification.getTypeName()); if (classificationType == null) { throw new AtlasBaseException(AtlasErrorCode.TYPE_NAME_INVALID, TypeCategory.CLASSIFICATION.name(), classification.getTypeName()); } AtlasClassification ret = (AtlasClassification)converter.fromV1ToV2(classification, classificationType, new AtlasFormatConverter.ConverterContext()); return ret; }
@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 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 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"); } }