com.hmsonline.trident.cql.incremental.CassandraCqlIncrementalStateFactory Maven / Gradle / Ivy

Go to download
package com.hmsonline.trident.cql.incremental;

import backtype.storm.task.IMetricsContext;
import com.hmsonline.trident.cql.CqlClientFactory;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import storm.trident.operation.CombinerAggregator;
import storm.trident.state.State;
import storm.trident.state.StateFactory;

import java.util.Map;

@SuppressWarnings("rawtypes")
// TODO: Is it worth subclassing from CassandraCqlStateFactory?
public class CassandraCqlIncrementalStateFactory implements StateFactory {
    private static final long serialVersionUID = 1L;
    private static final Logger LOG = LoggerFactory.getLogger(CassandraCqlIncrementalStateFactory.class);
    public static String TRIDENT_CASSANDRA_CQL_HOSTS = "trident.cassandra.cql.hosts";
    private static CqlClientFactory clientFactory;
    private CombinerAggregator aggregator;
    private CqlIncrementMapper mapper;

    public CassandraCqlIncrementalStateFactory(CombinerAggregator aggregator, CqlIncrementMapper mapper) {
        this.aggregator = aggregator;
        this.mapper = mapper;
    }

    @Override
    public State makeState(Map configuration, IMetricsContext metrics, int partitionIndex, int numPartitions) {
        // worth synchronizing here?
        if (clientFactory == null) {
            clientFactory = new CqlClientFactory(configuration);
        }
        LOG.debug("Creating State for partition [{}] of [{}]", new Object[]{partitionIndex, numPartitions});
        return new CassandraCqlIncrementalState(CassandraCqlIncrementalStateFactory.clientFactory, aggregator, mapper, partitionIndex);
    }
}