info.archinnov.achilles.entity.operations.ThriftEntityLoader Maven / Gradle / Ivy

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

import info.archinnov.achilles.context.ThriftPersistenceContext;
import info.archinnov.achilles.entity.metadata.EntityMeta;
import info.archinnov.achilles.entity.metadata.PropertyMeta;
import info.archinnov.achilles.entity.operations.impl.ThriftJoinLoaderImpl;
import info.archinnov.achilles.entity.operations.impl.ThriftLoaderImpl;
import info.archinnov.achilles.exception.AchillesException;
import info.archinnov.achilles.proxy.ReflectionInvoker;
import info.archinnov.achilles.validation.Validator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * ThriftEntityLoader
 * 
 * @author DuyHai DOAN
 * 
 */
public class ThriftEntityLoader implements EntityLoader
{
    private static final Logger log = LoggerFactory.getLogger(ThriftEntityLoader.class);

    private ReflectionInvoker invoker = new ReflectionInvoker();
    private ThriftJoinLoaderImpl joinLoaderImpl = new ThriftJoinLoaderImpl();
    private ThriftLoaderImpl loaderImpl = new ThriftLoaderImpl();

    @Override
    public  T load(ThriftPersistenceContext context, Class entityClass)
    {
        log.debug("Loading entity of class {} with primary key {}", context
                .getEntityClass()
                .getCanonicalName(), context.getPrimaryKey());

        EntityMeta entityMeta = context.getEntityMeta();
        Object primaryKey = context.getPrimaryKey();

        Validator.validateNotNull(entityClass, "Entity class should not be null");
        Validator.validateNotNull(primaryKey, "Entity '%s' key should not be null", entityClass.getCanonicalName());
        Validator.validateNotNull(entityMeta, "Entity meta for '%s' should not be null",
                entityClass.getCanonicalName());

        T entity = null;
        try
        {
            if (context.isLoadEagerFields())
            {
                entity = loaderImpl.load(context, entityClass);
            }
            else
            {
                log.debug("Get reference called, just instanciate the entity with primary key");
                entity = invoker.instanciate(entityClass);
                invoker.setValueToField(entity, entityMeta.getIdMeta().getSetter(), primaryKey);
            }
        } catch (Exception e)
        {
            throw new AchillesException("Error when loading entity type '"
                    + entityClass.getCanonicalName() + "' with key '" + primaryKey + "'. Cause : "
                    + e.getMessage(), e);
        }
        return entity;
    }

    @Override
    public  void loadPropertyIntoObject(ThriftPersistenceContext context, Object realObject,
            PropertyMeta propertyMeta)
    {
        log.debug("Loading eager properties into entity of class {} with primary key {}", context
                .getEntityClass()
                .getCanonicalName(), context.getPrimaryKey());

        Object value = null;
        switch (propertyMeta.type())
        {
            case SIMPLE:
            case LAZY_SIMPLE:
                value = loaderImpl.loadSimpleProperty(context, propertyMeta);
                break;
            case LIST:
            case LAZY_LIST:
                value = loaderImpl.loadListProperty(context, propertyMeta);
                break;
            case SET:
            case LAZY_SET:
                value = loaderImpl.loadSetProperty(context, propertyMeta);
                break;
            case MAP:
            case LAZY_MAP:
                value = loaderImpl.loadMapProperty(context, propertyMeta);
                break;
            case JOIN_SIMPLE:
                value = loaderImpl.loadJoinSimple(context, propertyMeta, this);
                break;
            case JOIN_LIST:
                value = joinLoaderImpl.loadJoinListProperty(context, propertyMeta);
                break;
            case JOIN_SET:
                value = joinLoaderImpl.loadJoinSetProperty(context, propertyMeta);
                break;
            case JOIN_MAP:
                value = joinLoaderImpl.loadJoinMapProperty(context, propertyMeta);
                break;
            default:
                return;
        }
        invoker.setValueToField(realObject, propertyMeta.getSetter(), value);
    }

    protected Object loadPrimaryKey(ThriftPersistenceContext context, PropertyMeta propertyMeta)
    {
        return loaderImpl.loadSimpleProperty(context, propertyMeta);
    }
}