info.archinnov.achilles.context.ThriftAbstractFlushContext Maven / Gradle / Ivy

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

import info.archinnov.achilles.consistency.AchillesConsistencyLevelPolicy;
import info.archinnov.achilles.counter.AchillesCounter;
import info.archinnov.achilles.dao.ThriftAbstractDao;
import info.archinnov.achilles.dao.ThriftCounterDao;
import info.archinnov.achilles.dao.ThriftGenericEntityDao;
import info.archinnov.achilles.dao.ThriftGenericWideRowDao;
import info.archinnov.achilles.type.ConsistencyLevel;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import me.prettyprint.hector.api.mutation.Mutator;
import org.apache.cassandra.utils.Pair;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.google.common.base.Optional;

/**
 * ThriftAbstractFlushContext
 * 
 * @author DuyHai DOAN
 * 
 */
public abstract class ThriftAbstractFlushContext> extends
        FlushContext
{
    protected static final Logger log = LoggerFactory.getLogger(ThriftAbstractFlushContext.class);

    protected ThriftDaoContext thriftDaoContext;
    protected ThriftConsistencyContext consistencyContext;

    protected Map, ThriftAbstractDao>> mutatorMap = new HashMap, ThriftAbstractDao>>();
    protected boolean hasCustomConsistencyLevels = false;

    protected ThriftAbstractFlushContext(ThriftDaoContext thriftDaoContext,
            AchillesConsistencyLevelPolicy policy, Optional readLevelO,
            Optional writeLevelO, Optional ttlO)
    {
        super(ttlO);
        this.thriftDaoContext = thriftDaoContext;
        this.consistencyContext = new ThriftConsistencyContext(policy, readLevelO, writeLevelO);
    }

    protected ThriftAbstractFlushContext(ThriftDaoContext thriftDaoContext,
            ThriftConsistencyContext consistencyContext,
            Map, ThriftAbstractDao>> mutatorMap,
            boolean hasCustomConsistencyLevels,
            Optional ttlO)
    {
        super(ttlO);
        this.thriftDaoContext = thriftDaoContext;
        this.consistencyContext = consistencyContext;
        this.mutatorMap = mutatorMap;
        this.hasCustomConsistencyLevels = hasCustomConsistencyLevels;
    }

    protected void doFlush()
    {
        log.debug("Execute mutations flush");
        try
        {
            for (Entry, ThriftAbstractDao>> entry : mutatorMap
                    .entrySet())
            {
                ThriftAbstractDao dao = entry.getValue().right;
                Mutator mutator = entry.getValue().left;
                dao.executeMutator(mutator);
            }
        } finally
        {
            cleanUp();
        }
    }

    @Override
    public void cleanUp()
    {
        log.debug("Cleaning up flush context");
        consistencyContext.reinitConsistencyLevels();
        hasCustomConsistencyLevels = false;
        mutatorMap.clear();
    }

    @Override
    public void setWriteConsistencyLevel(Optional writeLevelO)
    {
        hasCustomConsistencyLevels = true;
        consistencyContext.setWriteConsistencyLevel(writeLevelO);
    }

    @Override
    public void setReadConsistencyLevel(Optional readLevelO)
    {
        hasCustomConsistencyLevels = true;
        consistencyContext.setReadConsistencyLevel(readLevelO);
    }

    @Override
    public void reinitConsistencyLevels()
    {
        if (!hasCustomConsistencyLevels)
        {
            consistencyContext.reinitConsistencyLevels();
        }
    }

    public Mutator