info.archinnov.achilles.internals.codec.EnumNameCodec Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of achilles-core Show documentation
CQL implementation for Achilles using Datastax Java driver
There is a newer version: 6.1.0
/*
 * Copyright (C) 2012-2016 DuyHai DOAN
 *
 * Licensed 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 info.archinnov.achilles.internals.codec;

import static java.lang.String.format;

import java.util.List;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import info.archinnov.achilles.exception.AchillesTranscodingException;
import info.archinnov.achilles.type.codec.Codec;

public class EnumNameCodec implements Codec {

    private static final Logger LOGGER = LoggerFactory.getLogger(EnumNameCodec.class);

    private final List enumValues;
    private final Class sourceType;

    public EnumNameCodec(List enumValues, Class sourceType) {
        this.enumValues = enumValues;
        this.sourceType = sourceType;
    }

    public static  EnumNameCodec create(List enumTypes, Class sourceType) {
        return new EnumNameCodec<>(enumTypes, sourceType);
    }

    @Override
    public Class sourceType() {
        return sourceType;
    }

    @Override
    public Class targetType() {
        return String.class;
    }

    @Override
    public String encode(ENUM fromJava) throws AchillesTranscodingException {
        if (fromJava == null) return null;
        if (LOGGER.isTraceEnabled()) {
            LOGGER.trace(String.format("Encoding enum %s to String", fromJava));
        }
        if (!fromJava.getClass().isEnum()) {
            throw new AchillesTranscodingException(format("Object '%s' to be encoded should be an enum", fromJava));
        }
        if (!enumValues.contains(fromJava)) {
            throw new AchillesTranscodingException(format("Cannot find matching enum values for '%s' from possible enum constants '%s' ", fromJava, enumValues));
        }
        return ((Enum) fromJava).name();
    }

    @Override
    public ENUM decode(String fromCassandra) throws AchillesTranscodingException {
        if (fromCassandra == null) return null;
        if (LOGGER.isTraceEnabled()) {
            LOGGER.trace(String.format("Decoding enum type %s from String %s", sourceType.getCanonicalName(), fromCassandra));
        }
        for (ENUM enumValue : enumValues) {
            if (((Enum) enumValue).name().equals(fromCassandra)) return enumValue;
        }
        throw new AchillesTranscodingException(format("Cannot find matching enum values for '%s' from possible enum constants '%s' ", fromCassandra, enumValues));
    }


}