info.archinnov.achilles.entity.ThriftEntityMapper Maven / Gradle / Ivy

Go to download
package info.archinnov.achilles.entity;

import static info.archinnov.achilles.serializer.ThriftSerializerUtils.STRING_SRZ;
import info.archinnov.achilles.entity.metadata.EntityMeta;
import info.archinnov.achilles.entity.metadata.PropertyMeta;
import info.archinnov.achilles.helper.EntityMapper;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import me.prettyprint.hector.api.beans.Composite;
import org.apache.cassandra.utils.Pair;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * ThriftEntityMapper
 * 
 * @author DuyHai DOAN
 * 
 */
public class ThriftEntityMapper extends EntityMapper
{
    private static final Logger log = LoggerFactory.getLogger(ThriftEntityMapper.class);

    public void setEagerPropertiesToEntity(Object key, List> columns,
            EntityMeta entityMeta,
            Object entity)
    {
        log.trace("Set eager properties to entity {} ", entityMeta.getClassName());

        Map> listProperties = new HashMap>();
        Map> setProperties = new HashMap>();
        Map> mapProperties = new HashMap>();

        setIdToEntity(key, entityMeta.getIdMeta(), entity);

        Map propertyMetas = entityMeta.getPropertyMetas();

        for (Pair pair : columns)
        {
            String propertyName = pair.left.get(1, STRING_SRZ);

            PropertyMeta propertyMeta = propertyMetas.get(propertyName);

            if (propertyMeta != null)
            {

                switch (propertyMeta.type())
                {
                    case SIMPLE:
                        setSimplePropertyToEntity(pair.right, propertyMeta, entity);
                        break;
                    case LIST:
                        addToList(listProperties, propertyMeta,
                                propertyMeta.getValueFromString(pair.right));
                        break;
                    case SET:
                        addToSet(setProperties, propertyMeta,
                                propertyMeta.getValueFromString(pair.right));
                        break;
                    case MAP:
                        addToMap(mapProperties, propertyMeta,
                                propertyMeta.getKeyValueFromString(pair.right));
                        break;
                    default:
                        log.debug("Property {} is lazy or of proxy type, do not set to entity now");
                        break;
                }
            }
            else
            {
                log.warn("No field mapping for property {}", propertyName);
            }
        }

        setMultiValuesProperties(entity, listProperties, setProperties, mapProperties,
                propertyMetas);
    }

    public  T initClusteredEntity(Class entityClass, PropertyMeta idMeta,
            Object embeddedId)
    {
        T clusteredEntity = null;
        clusteredEntity = invoker.instanciate(entityClass);
        invoker.setValueToField(clusteredEntity, idMeta.getSetter(), embeddedId);
        return clusteredEntity;
    }

    public  T createClusteredEntityWithValue(Class entityClass, PropertyMeta idMeta,
            PropertyMeta pm,
            Object embeddedId, Object clusteredValue)
    {
        T clusteredEntity = null;
        clusteredEntity = invoker.instanciate(entityClass);
        invoker.setValueToField(clusteredEntity, idMeta.getSetter(), embeddedId);
        invoker.setValueToField(clusteredEntity, pm.getSetter(), clusteredValue);

        return clusteredEntity;
    }

    private void setMultiValuesProperties(Object entity, Map> listProperties,
            Map> setProperties, Map> mapProperties,
            Map propertyMetas)
    {
        for (Entry> entry : listProperties.entrySet())
        {
            setListPropertyToEntity(entry.getValue(), propertyMetas.get(entry.getKey()), entity);
        }

        for (Entry> entry : setProperties.entrySet())
        {
            setSetPropertyToEntity(entry.getValue(), propertyMetas.get(entry.getKey()), entity);
        }

        for (Entry> entry : mapProperties.entrySet())
        {
            setMapPropertyToEntity(entry.getValue(), propertyMetas.get(entry.getKey()), entity);
        }
    }
}