package org.act.tstream.daemon.worker;

import java.util.concurrent.ConcurrentHashMap;

import org.apache.log4j.Logger;

import backtype.storm.messaging.IConnection;
import backtype.storm.messaging.TaskMessage;
import backtype.storm.utils.DisruptorQueue;

import org.act.tstream.metric.JStormTimer;
import org.act.tstream.metric.MetricDef;
import org.act.tstream.metric.Metrics;
import org.act.tstream.utils.DisruptorRunable;

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

/**
 * Message dispatcher
 * 
 * @author yannian/Longda
 * 
 */
public class VirtualPortDispatch extends DisruptorRunable {
	private final static Logger LOG = Logger
			.getLogger(VirtualPortDispatch.class);

	private ConcurrentHashMap<Integer, DisruptorQueue> deserializeQueues;
	private IConnection recvConnection;

	private static JStormTimer timer = Metrics.registerTimer(null, 
			MetricDef.DISPATCH_TIME, null, Metrics.MetricType.WORKER);
	
	public VirtualPortDispatch(WorkerData workerData,
			IConnection recvConnection, DisruptorQueue recvQueue) {
		super(recvQueue, timer, VirtualPortDispatch.class.getSimpleName(), 
				workerData.getActive());

		this.recvConnection = recvConnection;
		this.deserializeQueues = workerData.getDeserializeQueues();

		Metrics.registerQueue(null, MetricDef.DISPATCH_QUEUE, queue, null, Metrics.MetricType.WORKER);
	}

	public void cleanup() {
		LOG.info("Begin to shutdown VirtualPortDispatch");
		// don't need send shutdown command to every task
		// due to every task has been shutdown by workerData.active
		// at the same time queue has been fulll
//		byte shutdownCmd[] = { TaskStatus.SHUTDOWN };
//		for (DisruptorQueue queue : deserializeQueues.values()) {
//
//			queue.publish(shutdownCmd);
//		}

		try {
			recvConnection.close();
		}catch(Exception e) {
			
		}
		recvConnection = null;
		LOG.info("Successfully shudown VirtualPortDispatch");
	}

	@Override
	public void handleEvent(Object event, boolean endOfBatch)
			throws Exception {
		TaskMessage message = (TaskMessage) event;

		int task = message.task();

		DisruptorQueue queue = deserializeQueues.get(task);
		if (queue == null) {
			LOG.warn("Received invalid message directed at port " + task
					+ ". Dropping...");
			return;
		}

		queue.publish(message.message());

	}

}