intg/src/main/java/org/apache/atlas/type/AtlasEnumType.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.apache.atlas.type;


import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import org.apache.atlas.exception.AtlasBaseException;
import org.apache.atlas.model.typedef.AtlasEnumDef;
import org.apache.atlas.model.typedef.AtlasEnumDef.AtlasEnumElementDef;


/**
 * class that implements behaviour of an enum-type.
 */
public class AtlasEnumType extends AtlasType {
    private final AtlasEnumDef                     enumDef;
    private final Map<String, AtlasEnumElementDef> elementDefs;
    private final String                           defaultValue;

    public AtlasEnumType(AtlasEnumDef enumDef) {
        super(enumDef);

        Map<String, AtlasEnumElementDef> e = new HashMap<>();

        for (AtlasEnumElementDef elementDef : enumDef.getElementDefs()) {
            e.put(elementDef.getValue().toLowerCase(), elementDef);
        }

        String d = enumDef.getDefaultValue();

        if (d == null) {
            AtlasEnumElementDef defElem = enumDef.getElementDefs().size() > 0 ? enumDef.getElementDefs().get(0) : null;

            if (defElem != null) {
                d = defElem.getValue();
            }
        }

        this.enumDef      = enumDef;
        this.elementDefs  = Collections.unmodifiableMap(e);
        this.defaultValue = d;
    }

    public AtlasEnumDef getEnumDef() { return enumDef; }

    @Override
    public void resolveReferences(AtlasTypeRegistry typeRegistry) throws AtlasBaseException {
    }

    @Override
    public Object createDefaultValue() {
        return defaultValue;
    }

    @Override
    public boolean isValidValue(Object obj) {
        if (obj != null) {
            return elementDefs.containsKey(obj.toString().toLowerCase());
        }

        return true;
    }

    @Override
    public Object getNormalizedValue(Object obj) {
        if (obj != null) {
            AtlasEnumElementDef elementDef = elementDefs.get(obj.toString().toLowerCase());

            if (elementDef != null) {
                return elementDef.getValue();
            }
        }

        return null;
    }

    public AtlasEnumElementDef getEnumElementDef(String value) {
        if (value != null) {
            return elementDefs.get(value.toLowerCase());
        }

        return null;
    }

    public AtlasEnumElementDef getEnumElementDef(Number ordinal) {
        if (ordinal != null) {
            for (AtlasEnumElementDef elementDef : elementDefs.values()) {
                if (elementDef.getOrdinal().longValue() == ordinal.longValue()) {
                    return elementDef;
                }
            }
        }

        return null;
    }
}