org.simpleflatmapper.converter.impl.ObjectToEnumConverter Maven / Gradle / Ivy

Go to download

package org.simpleflatmapper.converter.impl;

import org.simpleflatmapper.converter.Context;
import org.simpleflatmapper.converter.ContextualConverter;

public class ObjectToEnumConverter> implements ContextualConverter {
    private final Class enumClass;

    private final NumberToEnumConverter numberToEnumConverter;

    public ObjectToEnumConverter(Class enumClass) {
        this.enumClass = enumClass;
        this.numberToEnumConverter = new NumberToEnumConverter(enumClass);
    }

    @Override
    public E convert(Object in, Context context) throws Exception {
        if (in == null) return null;
        if (in instanceof Number) {
            return numberToEnumConverter.convert((Number) in, context);
        } else {
            return Enum.valueOf(enumClass, String.valueOf(in));
        }
    }
}