package com.hmsonline.trident.cql;

import com.datastax.driver.core.Statement;
import com.hmsonline.trident.cql.mappers.CqlTupleMapper;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import storm.trident.operation.TridentCollector;
import storm.trident.operation.TridentOperationContext;
import storm.trident.state.StateUpdater;
import storm.trident.tuple.TridentTuple;

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

public class CassandraCqlStateUpdater<K,V> implements StateUpdater<CassandraCqlState> {
    private static final long serialVersionUID = 1L;
    private static final Logger LOG = LoggerFactory.getLogger(CassandraCqlStateUpdater.class);
    private CqlTupleMapper<K,V> mapper = null;
	private boolean propagateTuples;

    public CassandraCqlStateUpdater(CqlTupleMapper<K,V> mapper) {
        this(mapper, false);
    }

	public CassandraCqlStateUpdater(CqlTupleMapper<K,V> mapper, boolean propagateTuples) {
        this.mapper = mapper;
		this.propagateTuples = propagateTuples;
	}

    @SuppressWarnings("rawtypes")
    @Override
    public void prepare(Map configuration, TridentOperationContext context) {
        LOG.debug("Preparing updater with [{}]", configuration);
    }

    @Override
    public void cleanup() {
    }

    @Override
    public void updateState(CassandraCqlState state, List<TridentTuple> tuples, TridentCollector collector) {
        for (TridentTuple tuple : tuples) {
            Statement statement = this.mapper.map(tuple);
            state.addStatement(statement);
			if (propagateTuples) {
				collector.emit(tuple);
			}
        }
    }
}