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

Example 6 with AtlasEnumType

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

the class AtlasEnumFormatConverter method fromV2ToV1.

@Override
public Object fromV2ToV1(Object v2Obj, AtlasType type, ConverterContext ctx) throws AtlasBaseException {
    EnumValue ret = null;
    if (v2Obj == null || !(type instanceof AtlasEnumType)) {
        return ret;
    }
    AtlasEnumType enumType = (AtlasEnumType) type;
    AtlasEnumElementDef elementDef = enumType.getEnumElementDef(v2Obj.toString());
    if (elementDef != null) {
        ret = new EnumValue(elementDef.getValue(), elementDef.getOrdinal());
    }
    return ret;
}

Example 7 with AtlasEnumType

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

the class AtlasEnumFormatConverter method fromV1ToV2.

@Override
public Object fromV1ToV2(Object v1Obj, AtlasType type, ConverterContext ctx) throws AtlasBaseException {
    String ret = null;
    if (v1Obj == null || !(type instanceof AtlasEnumType)) {
        return ret;
    }
    Object v1Value = null;
    if (v1Obj instanceof EnumValue) {
        EnumValue enumValue = (EnumValue) v1Obj;
        v1Value = enumValue.value;
        if (v1Value == null) {
            v1Value = enumValue.ordinal;
        }
    } else if (v1Obj instanceof Map) {
        Map mapValue = (Map) v1Obj;
        v1Value = mapValue.get("value");
        if (v1Value == null) {
            v1Value = mapValue.get("ordinal");
        }
    }
    if (v1Value == null) {
        // could be 'value' or 'ordinal'
        v1Value = v1Obj;
    }
    AtlasEnumElementDef elementDef;
    if (v1Value instanceof Number) {
        elementDef = ((AtlasEnumType) type).getEnumElementDef((Number) v1Value);
    } else {
        elementDef = ((AtlasEnumType) type).getEnumElementDef(v1Value.toString());
    }
    if (elementDef != null) {
        ret = elementDef.getValue();
    }
    return ret;
}