package org.act.tstream.utils;

import java.util.concurrent.atomic.AtomicBoolean;

import org.act.tstream.callback.RunnableCallback;
import org.apache.log4j.Logger;

import backtype.storm.utils.DisruptorQueue;

import org.act.tstream.metric.JStormTimer;
import org.act.tstream.metric.Metrics;
import com.codahale.metrics.Timer;
import com.lmax.disruptor.EventHandler;

//import org.act.tstream.message.zeroMq.ISendConnection;

/**
 * 
 * Disruptor Consumer thread
 * 
 * @author yannian
 * 
 */
public abstract class DisruptorRunable extends RunnableCallback implements EventHandler {
	private final static Logger LOG = Logger.getLogger(DisruptorRunable.class);

	protected DisruptorQueue queue;
	protected String        idStr;
	protected AtomicBoolean active;
	protected JStormTimer   timer;


	public DisruptorRunable(DisruptorQueue queue, JStormTimer timer, String idStr,
			AtomicBoolean active) {
		this.queue  = queue;
		this.timer  = timer;
		this.idStr  = idStr;
		this.active = active;
	}

	public abstract void handleEvent(Object event, boolean endOfBatch) throws Exception;
	/**
	 * This function need to be implements
	 * @see com.lmax.disruptor.EventHandler#onEvent(java.lang.Object, long, boolean)
	 */
	@Override
	public void onEvent(Object event, long sequence, boolean endOfBatch) 
			throws Exception{
		if (event == null) {
			return ;
		}
		timer.start();
		try {
			handleEvent(event, endOfBatch);
		}finally {
			timer.stop();
		}
	}

	@Override
	public void run() {
		LOG.info("Successfully start thread " + idStr);
		queue.consumerStarted();

		while (active.get()) {
			try {

				queue.consumeBatchWhenAvailable(this);

			} catch (Exception e) {
				if (active.get() == true) {
					LOG.error("DrainerRunable send error", e);
					throw new RuntimeException(e);
				}
			}
		}
		
		LOG.info("Successfully exit thread " + idStr);
	}

	@Override
	public Object getResult() {
		if (active.get())
			return 0;
		else
			return -1;
	}

}