Examples with AtlasStructType org.apache.atlas.type.AtlasStructType used on opensource projects

Example 6 with AtlasStructType

use of org.apache.atlas.type.AtlasStructType in project atlas by apache.

the class AtlasEntityStoreV1Test method validateEntity.

private void validateEntity(AtlasEntityExtInfo entityExtInfo, AtlasStruct actual, AtlasStruct expected) throws AtlasBaseException, AtlasException {
    if (expected == null) {
        Assert.assertNull(actual, "expected null instance. Found " + actual);
        return;
    }
    Assert.assertNotNull(actual, "found null instance");
    AtlasStructType entityType = (AtlasStructType) typeRegistry.getType(actual.getTypeName());
    for (String attrName : expected.getAttributes().keySet()) {
        Object expectedVal = expected.getAttribute(attrName);
        Object actualVal = actual.getAttribute(attrName);
        AtlasType attrType = entityType.getAttributeType(attrName);
        validateAttribute(entityExtInfo, actualVal, expectedVal, attrType, attrName);
    }
}

Example 7 with AtlasStructType

use of org.apache.atlas.type.AtlasStructType in project atlas by apache.

the class RestUtilsTest method convertToJsonAndBack.

private AtlasAttributeDef convertToJsonAndBack(AtlasTypeRegistry registry, AtlasStructDef structDef, AtlasAttributeDef attributeDef, boolean compositeExpected) throws AtlasBaseException {
    AtlasTypeDefGraphStoreV1 typeDefStore = makeTypeStore(registry);
    AtlasStructType structType = (AtlasStructType) registry.getType(structDef.getName());
    AtlasAttribute attribute = structType.getAttribute(attributeDef.getName());
    String attribJson = AtlasStructDefStoreV1.toJsonFromAttribute(attribute);
    Map attrInfo = AtlasType.fromJson(attribJson, Map.class);
    Assert.assertEquals(attrInfo.get("isComposite"), compositeExpected);
    return AtlasStructDefStoreV1.toAttributeDefFromJson(structDef, attrInfo, typeDefStore);
}

Example 8 with AtlasStructType

use of org.apache.atlas.type.AtlasStructType in project atlas by apache.

the class ModelTestUtil method newStruct.

public static AtlasStruct newStruct(AtlasStructDef structDef, AtlasTypeRegistry typesRegistry) {
    AtlasStruct ret = null;
    AtlasStructType structType = typesRegistry.getStructTypeByName(structDef.getName());
    if (structType != null) {
        ret = structType.createDefaultValue();
    } else {
        LOG.error("failed to get struct-type {}", structDef.getName());
    }
    return ret;
}

Example 9 with AtlasStructType

use of org.apache.atlas.type.AtlasStructType in project incubator-atlas by apache.

the class EntityGraphMapper method mapAttributes.

private void mapAttributes(AtlasStruct struct, AtlasVertex vertex, EntityOperation op, EntityMutationContext context) throws AtlasBaseException {
    if (LOG.isDebugEnabled()) {
        LOG.debug("==> mapAttributes({}, {})", op, struct.getTypeName());
    }
    if (MapUtils.isNotEmpty(struct.getAttributes())) {
        AtlasStructType structType = getStructType(struct.getTypeName());
        if (op.equals(CREATE)) {
            for (AtlasAttribute attribute : structType.getAllAttributes().values()) {
                Object attrValue = struct.getAttribute(attribute.getName());
                mapAttribute(attribute, attrValue, vertex, op, context);
            }
        } else if (op.equals(UPDATE)) {
            for (String attrName : struct.getAttributes().keySet()) {
                AtlasAttribute attribute = structType.getAttribute(attrName);
                if (attribute != null) {
                    Object attrValue = struct.getAttribute(attrName);
                    mapAttribute(attribute, attrValue, vertex, op, context);
                } else {
                    LOG.warn("mapAttributes(): invalid attribute {}.{}. Ignored..", struct.getTypeName(), attrName);
                }
            }
        }
        updateModificationMetadata(vertex);
    }
    if (LOG.isDebugEnabled()) {
        LOG.debug("<== mapAttributes({}, {})", op, struct.getTypeName());
    }
}

Example 10 with AtlasStructType

use of org.apache.atlas.type.AtlasStructType in project incubator-atlas by apache.

the class AtlasEntityStoreV1Test method validateEntity.

private void validateEntity(AtlasEntityExtInfo entityExtInfo, AtlasStruct actual, AtlasStruct expected) throws AtlasBaseException, AtlasException {
    if (expected == null) {
        Assert.assertNull(actual, "expected null instance. Found " + actual);
        return;
    }
    Assert.assertNotNull(actual, "found null instance");
    AtlasStructType entityType = (AtlasStructType) typeRegistry.getType(actual.getTypeName());
    for (String attrName : expected.getAttributes().keySet()) {
        Object expectedVal = expected.getAttribute(attrName);
        Object actualVal = actual.getAttribute(attrName);
        AtlasType attrType = entityType.getAttributeType(attrName);
        validateAttribute(entityExtInfo, actualVal, expectedVal, attrType, attrName);
    }
}