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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of achilles-thrift Show documentation
Thrift implementation for Achilles using Hector library
The newest version!
/**
 *
 * Copyright (C) 2012-2013 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.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;

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>();

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

		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:
					propertyMeta.setValueToField(entity, propertyMeta.forceDecodeFromJSON(pair.right));
					break;
				case LIST:
					addToList(listProperties, propertyMeta, propertyMeta.forceDecodeFromJSON(pair.right));
					break;
				case SET:
					addToSet(setProperties, propertyMeta, propertyMeta.decode(pair.left.get(2, STRING_SRZ)));
					break;
				case MAP:
					Object decodedKey = propertyMeta.forceDecodeFromJSON(pair.left.get(2, STRING_SRZ),
							propertyMeta.getKeyClass());
					Object decodedValue = propertyMeta.decode(pair.right);
					addToMap(mapProperties, propertyMeta, decodedKey, decodedValue);
					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, EntityMeta meta, Object embeddedId) {
		T clusteredEntity = null;
		clusteredEntity = meta. instanciate();
		meta.setPrimaryKey(clusteredEntity, embeddedId);
		return clusteredEntity;
	}

	public  T createClusteredEntityWithValue(Class entityClass, EntityMeta meta, PropertyMeta pm,
			Object embeddedId, Object clusteredValue) {
		T clusteredEntity = null;
		clusteredEntity = meta. instanciate();
		meta.setPrimaryKey(clusteredEntity, embeddedId);
		pm.setValueToField(clusteredEntity, clusteredValue);

		return clusteredEntity;
	}

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

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

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