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

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

import info.archinnov.achilles.dao.ThriftGenericEntityDao;
import info.archinnov.achilles.entity.ThriftEntityMapper;
import info.archinnov.achilles.entity.metadata.EntityMeta;
import info.archinnov.achilles.proxy.ReflectionInvoker;
import org.apache.cassandra.utils.Pair;
import info.archinnov.achilles.validation.Validator;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import me.prettyprint.hector.api.beans.Composite;

import org.apache.commons.lang.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

	private ThriftEntityMapper mapper = new ThriftEntityMapper();
	private ReflectionInvoker invoker = new ReflectionInvoker();

	public  Map loadJoinEntities(Class entityClass, List keys,
			EntityMeta entityMeta, ThriftGenericEntityDao joinEntityDao)
	{
		if (log.isTraceEnabled())
		{
			log.trace("Load join entities of class {} with primary keys {}",
					entityClass.getCanonicalName(), StringUtils.join(keys, ","));
		}

		Validator.validateNotNull(entityClass, "Entity class should not be null");
		Validator.validateNotEmpty(keys, "List of join primary keys '" + keys
				+ "' should not be empty");
		Validator.validateNotNull(entityMeta, "Entity meta for '" + entityClass.getCanonicalName()
				+ "' should not be null");

		Map entitiesByKey = new HashMap();
		Map>> rows = joinEntityDao.eagerFetchEntities(keys);

		for (Entry>> entry : rows.entrySet())
		{
			T entity;
			entity = invoker.instanciate(entityClass);

			ID key = entry.getKey();
			List> columns = entry.getValue();
			if (columns.size() > 0)
			{
				mapper.setEagerPropertiesToEntity(key, columns, entityMeta, entity);
				invoker.setValueToField(entity, entityMeta.getIdMeta().getSetter(), key);
				entitiesByKey.put(key, entity);
			}
		}
		return entitiesByKey;
	}
}