info.archinnov.achilles.context.ThriftAbstractFlushContext 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.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;

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 ConsistencyLevel consistencyLevel;

	protected ThriftAbstractFlushContext(ThriftDaoContext thriftDaoContext, AchillesConsistencyLevelPolicy policy,
			ConsistencyLevel consistencyLevel) {
		this.thriftDaoContext = thriftDaoContext;
		this.consistencyContext = new ThriftConsistencyContext(policy, consistencyLevel);
	}

	protected ThriftAbstractFlushContext(ThriftDaoContext thriftDaoContext,
			ThriftConsistencyContext consistencyContext,
			Map, ThriftAbstractDao>> mutatorMap, boolean hasCustomConsistencyLevels) {
		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 setConsistencyLevel(ConsistencyLevel consistencyLevel) {
		hasCustomConsistencyLevels = true;
		consistencyContext.setConsistencyLevel(consistencyLevel);
	}

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

	public Mutator