info.archinnov.achilles.iterator.ThriftKeyValueIteratorImpl Maven / Gradle / Ivy

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

import info.archinnov.achilles.context.ThriftPersistenceContext;
import info.archinnov.achilles.entity.metadata.PropertyMeta;
import info.archinnov.achilles.iterator.factory.ThriftKeyValueFactory;
import info.archinnov.achilles.type.KeyValue;
import info.archinnov.achilles.type.KeyValueIterator;

import java.util.NoSuchElementException;

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

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

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

	private ThriftKeyValueFactory factory = new ThriftKeyValueFactory();
	protected ThriftAbstractSliceIterator> achillesSliceIterator;
	private PropertyMeta propertyMeta;
	private ThriftPersistenceContext context;

	protected ThriftKeyValueIteratorImpl() {}

	public ThriftKeyValueIteratorImpl(ThriftPersistenceContext context,
			ThriftAbstractSliceIterator> columnSliceIterator,
			PropertyMeta propertyMeta)
	{
		this.context = context;
		this.achillesSliceIterator = columnSliceIterator;
		this.propertyMeta = propertyMeta;
	}

	@Override
	public boolean hasNext()
	{
		log.trace("Does the {} has next value ? ", achillesSliceIterator.type());
		return this.achillesSliceIterator.hasNext();
	}

	@Override
	public KeyValue next()
	{
		log.trace("Get next key/value from the {} ", achillesSliceIterator.type());
		KeyValue keyValue = null;
		if (this.achillesSliceIterator.hasNext())
		{
			HColumn column = this.achillesSliceIterator.next();
			keyValue = factory.createKeyValue(context, propertyMeta, column);
		}
		else
		{
			throw new NoSuchElementException();
		}
		return keyValue;
	}

	@Override
	public K nextKey()
	{
		log.trace("Get next key from the {} ", achillesSliceIterator.type());
		K key = null;
		if (this.achillesSliceIterator.hasNext())
		{
			HColumn column = this.achillesSliceIterator.next();
			key = factory.createKey(propertyMeta, column);
		}
		else
		{
			throw new NoSuchElementException();
		}
		return key;
	}

	@Override
	public V nextValue()
	{
		log.trace("Get next value from the {} ", achillesSliceIterator.type());
		V value = null;
		if (this.achillesSliceIterator.hasNext())
		{
			HColumn column = this.achillesSliceIterator.next();
			value = factory.createValue(context, propertyMeta, column);
		}
		else
		{
			throw new NoSuchElementException();
		}
		return value;
	}

	@Override
	public Integer nextTtl()
	{
		log.trace("Get next ttl from the {} ", achillesSliceIterator.type());
		Integer ttl = null;
		if (this.achillesSliceIterator.hasNext())
		{
			HColumn column = this.achillesSliceIterator.next();
			ttl = factory.createTtl(column);
		}
		else
		{
			throw new NoSuchElementException();
		}
		return ttl;
	}

	@Override
	public void remove()
	{
		throw new UnsupportedOperationException(
				"Remove from iterator is not supported. Please use removeValue() or removeValues() instead");
	}

}