info.archinnov.achilles.internal.context.AbstractFlushContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of achilles-core Show documentation
CQL implementation for Achilles using Datastax Java driver
There is a newer version: 6.1.0
/*
 * Copyright (C) 2012-2014 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.internal.context;

import static com.google.common.base.Optional.fromNullable;
import java.util.ArrayList;
import java.util.List;
import com.datastax.driver.core.BatchStatement;
import com.datastax.driver.core.ResultSet;
import com.google.common.base.Optional;
import com.google.common.util.concurrent.ListenableFuture;
import info.archinnov.achilles.interceptor.Event;
import info.archinnov.achilles.internal.async.AsyncUtils;
import info.archinnov.achilles.internal.metadata.holder.EntityMeta;
import info.archinnov.achilles.internal.statement.wrapper.AbstractStatementWrapper;
import info.archinnov.achilles.internal.statement.wrapper.BatchStatementWrapper;
import info.archinnov.achilles.type.ConsistencyLevel;
import info.archinnov.achilles.type.Empty;

public abstract class AbstractFlushContext {
    protected AsyncUtils asyncUtils = AsyncUtils.Singleton.INSTANCE.get();
    protected DaoContext daoContext;

    protected List statementWrappers = new ArrayList<>();
    protected List counterStatementWrappers = new ArrayList<>();

    protected ConsistencyLevel consistencyLevel;
		protected Optional serialConsistencyLevel = Optional.absent();

    public AbstractFlushContext(DaoContext daoContext, ConsistencyLevel consistencyLevel, Optional serialConsistencyLevel) {
        this.daoContext = daoContext;
        this.consistencyLevel = consistencyLevel;
        this.serialConsistencyLevel = serialConsistencyLevel;
    }

    protected AbstractFlushContext(DaoContext daoContext, List statementWrappers,
            ConsistencyLevel consistencyLevel,Optional serialConsistencyLevel) {
        this.statementWrappers = statementWrappers;
        this.daoContext = daoContext;
        this.consistencyLevel = consistencyLevel;
		this.serialConsistencyLevel = serialConsistencyLevel;
    }

    protected ListenableFuture executeBatch(BatchStatement.Type batchType, List statementWrappers) {
        if (statementWrappers.isEmpty()) {
            return null;
        }

        final AbstractStatementWrapper batchStatement;
        if (statementWrappers.size() == 1 && batchType != BatchStatement.Type.LOGGED) {
            batchStatement = statementWrappers.get(0);
        } else {
            batchStatement = new BatchStatementWrapper(batchType, statementWrappers, fromNullable(consistencyLevel), serialConsistencyLevel);
        }

        return daoContext.execute(batchStatement);
    }

    public void pushStatement(AbstractStatementWrapper statementWrapper) {
        statementWrappers.add(statementWrapper);
    }

    public void pushCounterStatement(AbstractStatementWrapper statementWrapper) {
        counterStatementWrappers.add(statementWrapper);
    }

    public ListenableFuture execute(AbstractStatementWrapper statementWrapper) {
        return daoContext.execute(statementWrapper);
    }

    public void setConsistencyLevel(ConsistencyLevel consistencyLevel) {
        this.consistencyLevel = consistencyLevel;
    }

    public ConsistencyLevel getConsistencyLevel() {
        return consistencyLevel;
    }

    public abstract void startBatch();

    public abstract ListenableFuture> flush();

    public abstract ListenableFuture flushBatch();

    public abstract FlushType type();

    public abstract AbstractFlushContext duplicate();

    public abstract void triggerInterceptor(EntityMeta meta, Object entity, Event event);

    public enum FlushType {
        BATCH, IMMEDIATE
    }

    @Override
    public String toString() {
        return type().toString();
    }
}