com.hmsonline.cassandra.index.Configuration Maven / Gradle / Ivy

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

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

import org.apache.commons.lang.StringUtils;

public class Configuration {
    public static final String KEYSPACE = "keyspace";
    public static final String COLUMN_FAMILY = "column_family";
    public static final String COLUMNS = "columns";
    public static final String COLUMN_DELIM = ",";
    public static final String FIELD_DELIM = ":";

    private Map>> config = new HashMap>>();

    public void addIndex(String indexName, Map indexProperties) {
        String keyspace = indexProperties.get(KEYSPACE);
        String columnFamily = indexProperties.get(COLUMN_FAMILY);
        String columns = indexProperties.get(COLUMNS);

        if (StringUtils.isEmpty(keyspace) || StringUtils.isEmpty(columnFamily) || StringUtils.isEmpty(columns)) {
            return;
        }

        List indexColumns = new ArrayList();
        for (String column : columns.split(COLUMN_DELIM)) {
            if (StringUtils.isNotEmpty(column)) {
                indexColumns.add(column.trim());
            }
        }

        String key = generateKey(keyspace, columnFamily);
        if (!config.containsKey(key)) {
            config.put(key, new HashMap>());
        }
        config.get(key).put(indexName, indexColumns);
    }

    public Map> getIndexes(String keyspace, String columnFamily) {
        String key = generateKey(keyspace, columnFamily);
        return config.containsKey(key) ? config.get(key) : new HashMap>();
    }

    public Set getIndexNames(String keyspace, String columnFamily) {
        return getIndexes(keyspace, columnFamily).keySet();
    }

    public List getIndexColumns(String keyspace, String columnFamily, String indexName) {
        return getIndexes(keyspace, columnFamily).get(indexName);
    }

    public boolean isEmpty() {
        return config.isEmpty();
    }

    public void clear() {
        config.clear();
    }

    private String generateKey(String keyspace, String columnFamily) {
        return keyspace.hashCode() + "_" + columnFamily.hashCode();
    }
}