public void normalizeAttributeValuesForUpdate(Map<String, Object> obj) { if (obj != null) { for (AtlasAttributeDef attrDef : structDef.getAttributeDefs()) { String attrName = attrDef.getName(); Object attrValue = obj.get(attrName); if (obj.containsKey(attrName)) { attrValue = getNormalizedValueForUpdate(attrValue, attrDef); obj.put(attrName, attrValue); } } } }
public void normalizeAttributeValuesForUpdate(Map<String, Object> obj) { if (obj != null) { for (AtlasAttributeDef attrDef : structDef.getAttributeDefs()) { String attrName = attrDef.getName(); Object attrValue = obj.get(attrName); if (obj.containsKey(attrName)) { attrValue = getNormalizedValueForUpdate(attrValue, attrDef); obj.put(attrName, attrValue); } } } }
public void normalizeAttributeValues(Map<String, Object> obj) { if (obj != null) { for (AtlasAttributeDef attributeDef : structDef.getAttributeDefs()) { String attributeName = attributeDef.getName(); if (obj.containsKey(attributeName)) { Object attributeValue = getNormalizedValue(obj.get(attributeName), attributeDef); obj.put(attributeName, attributeValue); } else if (!attributeDef.getIsOptional()) { obj.put(attributeName, createDefaultValue(attributeDef)); } } } }
private void addOrUpdateAttributes(AtlasStructDef structDef, List<AtlasAttributeDef> attributesToUpdate) { for (AtlasAttributeDef attributeToUpdate : attributesToUpdate) { String attrName = attributeToUpdate.getName(); if (structDef.hasAttribute(attrName)) { structDef.removeAttribute(attrName); } structDef.addAttribute(attributeToUpdate); } } }
public static AtlasAttributeDef findAttribute(Collection<AtlasAttributeDef> attributeDefs, String attrName) { AtlasAttributeDef ret = null; if (CollectionUtils.isNotEmpty(attributeDefs)) { for (AtlasAttributeDef attributeDef : attributeDefs) { if (StringUtils.equalsIgnoreCase(attributeDef.getName(), attrName)) { ret = attributeDef; break; } } } return ret; }
@Test public void testStructDefHasAttribute() { AtlasStructDef structDef = ModelTestUtil.getStructDef(); for (AtlasAttributeDef attributeDef : structDef.getAttributeDefs()) { assertTrue(structDef.hasAttribute(attributeDef.getName())); } assertFalse(structDef.hasAttribute("01234-xyzabc-;''-)(")); }
private void addOrUpdateAttributes(AtlasStructDef structDef, List<AtlasAttributeDef> attributesToUpdate) { for (AtlasAttributeDef attributeToUpdate : attributesToUpdate) { String attrName = attributeToUpdate.getName(); if (structDef.hasAttribute(attrName)) { structDef.removeAttribute(attrName); } structDef.addAttribute(attributeToUpdate); } } }
private Object createDefaultValue(AtlasAttributeDef attributeDef) { Object ret = null; if (attributeDef != null) { AtlasAttribute attribute = allAttributes.get(attributeDef.getName()); if (attribute != null) { AtlasType dataType = attribute.getAttributeType(); ret = dataType.createDefaultValue(attributeDef.getDefaultValue()); } } return ret; }
private Object getNormalizedValue(Object value, AtlasAttributeDef attributeDef) { AtlasAttribute attribute = relationshipAttributes.get(attributeDef.getName()); if (attribute != null) { AtlasType attrType = attribute.getAttributeType(); if (isValidRelationshipType(attrType) && value != null) { return attrType.getNormalizedValue(value); } } return null; }
private Object createDefaultValue(AtlasAttributeDef attributeDef) { Object ret = null; if (attributeDef != null) { AtlasAttribute attribute = allAttributes.get(attributeDef.getName()); if (attribute != null) { AtlasType dataType = attribute.getAttributeType(); ret = dataType.createDefaultValue(attributeDef.getDefaultValue()); } } return ret; }
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); }
public void normalizeAttributeValuesForUpdate(AtlasStruct obj) { if (obj != null) { for (AtlasAttributeDef attributeDef : structDef.getAttributeDefs()) { String attributeName = attributeDef.getName(); if (obj.hasAttribute(attributeName)) { Object attributeValue = getNormalizedValueForUpdate(obj.getAttribute(attributeName), attributeDef); obj.setAttribute(attributeName, attributeValue); } } } }
@Test public void testStructDefRemoveAttribute() { AtlasStructDef structDef = ModelTestUtil.newStructDef(); String attrName = structDef.getAttributeDefs().get(0).getName(); assertTrue(structDef.hasAttribute(attrName)); structDef.removeAttribute(attrName); assertFalse(structDef.hasAttribute(attrName)); }
@Test public void testStructDefRemoveAttribute() { AtlasStructDef structDef = ModelTestUtil.newStructDef(); String attrName = structDef.getAttributeDefs().get(0).getName(); assertTrue(structDef.hasAttribute(attrName)); structDef.removeAttribute(attrName); assertFalse(structDef.hasAttribute(attrName)); }
public static AtlasAttributeDef findAttribute(Collection<AtlasAttributeDef> attributeDefs, String attrName) { AtlasAttributeDef ret = null; if (CollectionUtils.isNotEmpty(attributeDefs)) { for (AtlasAttributeDef attributeDef : attributeDefs) { if (StringUtils.equalsIgnoreCase(attributeDef.getName(), attrName)) { ret = attributeDef; break; } } } return ret; }
private Object getNormalizedValueForUpdate(Object value, AtlasAttributeDef attributeDef) { AtlasAttribute attribute = allAttributes.get(attributeDef.getName()); if (attribute != null) { AtlasType attrType = attribute.getAttributeType(); if (value != null) { return attrType.getNormalizedValueForUpdate(value); } } return null; }
public void normalizeAttributeValuesForUpdate(AtlasStruct obj) { if (obj != null) { for (AtlasAttributeDef attributeDef : structDef.getAttributeDefs()) { String attributeName = attributeDef.getName(); if (obj.hasAttribute(attributeName)) { Object attributeValue = getNormalizedValueForUpdate(obj.getAttribute(attributeName), attributeDef); obj.setAttribute(attributeName, attributeValue); } } } }
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 Object getNormalizedValueForUpdate(Object value, AtlasAttributeDef attributeDef) { AtlasAttribute attribute = allAttributes.get(attributeDef.getName()); if (attribute != null) { AtlasType attrType = attribute.getAttributeType(); if (value != null) { return attrType.getNormalizedValueForUpdate(value); } } return null; }
@Test public void testStructDefHasAttribute() { AtlasStructDef structDef = ModelTestUtil.getStructDef(); for (AtlasAttributeDef attributeDef : structDef.getAttributeDefs()) { assertTrue(structDef.hasAttribute(attributeDef.getName())); } assertFalse(structDef.hasAttribute("01234-xyzabc-;''-)(")); }