TaskMetricInfo.java

package org.act.tstream.task;

import java.io.Serializable;
import java.util.Map;
import java.util.HashMap;
import java.util.List;
import java.util.ArrayList;

import org.apache.log4j.Logger;

import com.codahale.metrics.Metric;
import com.codahale.metrics.Gauge;
import com.codahale.metrics.Timer;
import com.codahale.metrics.Counter;
import com.codahale.metrics.Histogram;
import com.codahale.metrics.Meter;
import com.codahale.metrics.Snapshot;
import org.act.tstream.metric.MetricDef;
import org.act.tstream.metric.MetricInfo;
import org.act.tstream.metric.Metrics.QueueGauge;
import org.act.tstream.utils.JStormUtils;
import org.act.tstream.task.execute.spout.TimerRatio;

/**
 * /storm-zk-root/Monitor/{topologyid}/{taskid} data
 */
public class TaskMetricInfo implements Serializable {
	public static Logger LOG = Logger.getLogger(TaskMetricInfo.class);
	
	private static final long serialVersionUID = 7645367099257857979L;
	private String taskId;
	private String component;

    private Map<String, Double> gaugeData;
    private Map<String, Double> counterData;
    private Map<String, Double> meterData;
    private Map<String, Double> timerData;
    private Map<String, Double> histogramData;

	private static final double FULL_RATIO = 100.0;
	
	public static final String QEUEU_IS_FULL = "queue is full";

	public TaskMetricInfo(String taskId, String component) {
		this.taskId    = taskId;
		this.component = component;
		
		this.gaugeData     = new HashMap<String, Double>();
		this.counterData   = new HashMap<String, Double>();
		this.meterData     = new HashMap<String, Double>();
		this.timerData     = new HashMap<String, Double>();
		this.histogramData = new HashMap<String, Double>();
	}
	
	public String getTaskId() {
		return taskId;
	}
	
	public void setTaskId(String taskId) {
		this.taskId = taskId;
	}
	
	public String getComponent() {
		return this.component;
	}
	
	public void setComponent(String component) {
		this.component = component;
	}
	
	public Map<String, Double> getGaugeData() {
		return gaugeData;
	}
	
	public Map<String, Double> getCounterData() {
		return counterData;
	}
	
	public Map<String, Double> getMeterData() {
		return meterData;
	}
	
	public Map<String, Double> getTimerData() {
		return timerData;
	}
	
	public Map<String, Double> getHistogramData() {
		return histogramData;
	}
	
	public void updateMetricData(MetricInfo metricInfo) {
		String name = metricInfo.getName();
		Metric metric = metricInfo.getMetric();
		if (metric instanceof QueueGauge) {
			//covert to %
			float queueRatio = (((QueueGauge) metric).getValue())*100;
			double value = JStormUtils.formatDoubleDecPoint2((double)queueRatio);
            gaugeData.put(name, value);
		} else if ( (metric instanceof Gauge<?>) ||
				(metric instanceof TimerRatio)) {
			Double value = JStormUtils.convertToDouble(((Gauge) metric).getValue());
			if (value == null) {
				LOG.warn("gauge value is null or unknow type.");
			} else {
				value = JStormUtils.formatDoubleDecPoint4(value);
				gaugeData.put(name, value);
			}
	    } else if (metric instanceof Timer) {
			Snapshot snapshot = ((Timer) metric).getSnapshot();
			//Covert from ns to ms
			Double value = JStormUtils.formatDoubleDecPoint4(
					(snapshot.getMean())/1000000);
			timerData.put(name, value);
		} else if (metric instanceof Counter) {
		    Long value = ((Counter) metric).getCount();
		    counterData.put(name, value.doubleValue());
		} else if (metric instanceof Meter) {
			Double value = JStormUtils.formatDoubleDecPoint4(
					((Meter) metric).getMeanRate());
			meterData.put(name, value);
		} else if (metric instanceof Histogram) {
			Snapshot snapshot = ((Histogram) metric).getSnapshot();
			Double value = JStormUtils.formatDoubleDecPoint4(
					snapshot.getMean());
			histogramData.put(name, value);
		} else {
			LOG.warn("Unknown metric type, name:" + name);
		}
	}
	
	public List<String> anyQueueFull() {
		List<String> ret = new ArrayList<String>();
		String taskInfo = component + "-" + taskId + ": ";
	    if (gaugeData.get(MetricDef.DESERIALIZE_QUEUE) == FULL_RATIO) {
	    	ret.add(taskInfo + "deserialize-" + QEUEU_IS_FULL);
	    } else if (gaugeData.get(MetricDef.SERIALIZE_QUEUE) == FULL_RATIO)
	    {
	    	ret.add(taskInfo + "serialize-" + QEUEU_IS_FULL);
	    } else if (gaugeData.get(MetricDef.EXECUTE_QUEUE) == FULL_RATIO) {
	    	ret.add(taskInfo + "execute-" + QEUEU_IS_FULL);
	    }
	    return ret;
	}
}