com.hmsonline.cassandra.triggers.ConfigurationStore Maven / Gradle / Ivy

Go to download
package com.hmsonline.cassandra.triggers;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.cassandra.thrift.ColumnOrSuperColumn;
import org.apache.cassandra.thrift.ColumnParent;
import org.apache.cassandra.thrift.ConsistencyLevel;
import org.apache.cassandra.thrift.InvalidRequestException;
import org.apache.cassandra.thrift.KeyRange;
import org.apache.cassandra.thrift.KeySlice;
import org.apache.cassandra.thrift.SlicePredicate;
import org.apache.cassandra.thrift.SliceRange;
import org.apache.cassandra.thrift.TimedOutException;
import org.apache.cassandra.thrift.UnavailableException;
import org.apache.cassandra.utils.ByteBufferUtil;
import org.apache.thrift.TException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ConfigurationStore extends CassandraStore {
    public static final String KEYSPACE = "triggers";
    public static final String COLUMN_FAMILY = "Configuration";
    private static Logger logger = LoggerFactory.getLogger(ConfigurationStore.class);
    private static ConfigurationStore instance = null;
    private long lastFetchTime = -1;
    private static final int REFRESH_INTERVAL = 1000 * 30; // 30 seconds
    private Map> cache = new HashMap>();
    boolean refreshEnabled = true;

    public ConfigurationStore(String keyspace, String columnFamily) throws Exception {
        super(keyspace, columnFamily);
        logger.debug("Instantiated configuration store.");
    }

    public static synchronized ConfigurationStore getStore() throws Exception {
        if (instance == null)
            instance = new ConfigurationStore(KEYSPACE, COLUMN_FAMILY);
        return instance;
    }

    public boolean isCommitLogEnabled() throws InvalidRequestException, UnavailableException, TimedOutException,
            TException, Exception {
        Map commitLogProperties = this.getConfiguration().get("CommitLog");
        if (commitLogProperties != null) {
            String enabled = commitLogProperties.get("enabled");
            return (enabled != null && enabled.equals("true"));
        }
        return false;
    }

    public boolean shouldWriteColumns() throws Throwable {
        Map commitLogProperties = this.getConfiguration().get("CommitLog");
        if (commitLogProperties != null) {
            String writeColumns = commitLogProperties.get("writeColumns");
            return (writeColumns != null && writeColumns.equals("true"));
        }
        return false;
    }

    public Map> getConfiguration() throws InvalidRequestException, UnavailableException,
            TimedOutException, TException, Exception {
        long currentTime = System.currentTimeMillis();
        long timeSinceRefresh = currentTime - this.lastFetchTime;
        if (refreshEnabled && timeSinceRefresh > REFRESH_INTERVAL) {
            logger.debug("Refreshing trigger configuration.");
            SlicePredicate predicate = new SlicePredicate();
            SliceRange range = new SliceRange(ByteBufferUtil.bytes(""), ByteBufferUtil.bytes(""), false, 10);
            predicate.setSlice_range(range);

            Map> configuration = new HashMap>();
            KeyRange keyRange = new KeyRange(1000);
            keyRange.setStart_key(ByteBufferUtil.bytes(""));
            keyRange.setEnd_key(ByteBufferUtil.EMPTY_BYTE_BUFFER);
            ColumnParent parent = new ColumnParent(COLUMN_FAMILY);
            List rows = getConnection(KEYSPACE).get_range_slices(parent, predicate, keyRange,
                    ConsistencyLevel.ALL);
            for (KeySlice slice : rows) {
                String component = ByteBufferUtil.string(slice.key);
                Map properties = new HashMap();
                for (ColumnOrSuperColumn column : slice.columns) {
                    String key = ByteBufferUtil.string(column.column.name);
                    String value = ByteBufferUtil.string(column.column.value);
                    properties.put(key, value);
                }
                configuration.put(component, properties);
            }

            this.lastFetchTime = currentTime;
            cache = configuration;
        }
        return cache;
    }

    public void enableCommitLog() throws InvalidRequestException, UnavailableException, TimedOutException, TException,
            Exception {
        Map commitLogProperties = this.getConfiguration().get("CommitLog");
        if (commitLogProperties == null) {
            commitLogProperties = new HashMap();
            this.getConfiguration().put("CommitLog", commitLogProperties);
        }
        commitLogProperties.put("enabled", "true");
    }

    public void disableRefresh() {
        this.refreshEnabled = false;
    }
}