package backtype.storm.metric;
import backtype.storm.Config;
import backtype.storm.metric.api.IMetricsConsumer;
import backtype.storm.task.IBolt;
import backtype.storm.task.IErrorReporter;
import backtype.storm.task.OutputCollector;
import backtype.storm.task.TopologyContext;
import backtype.storm.tuple.Tuple;
import java.util.Collection;
import java.util.Map;
public class MetricsConsumerBolt implements IBolt {
IMetricsConsumer _metricsConsumer;
String _consumerClassName;
OutputCollector _collector;
Object _registrationArgument;
public MetricsConsumerBolt(String consumerClassName,
Object registrationArgument) {
_consumerClassName = consumerClassName;
_registrationArgument = registrationArgument;
}
@Override
public void prepare(Map stormConf, TopologyContext context,
OutputCollector collector) {
try {
_metricsConsumer = (IMetricsConsumer) Class.forName(
_consumerClassName).newInstance();
} catch (Exception e) {
throw new RuntimeException(
"Could not instantiate a class listed in config under section "
+ Config.TOPOLOGY_METRICS_CONSUMER_REGISTER
+ " with fully qualified name "
+ _consumerClassName, e);
}
_metricsConsumer.prepare(stormConf, _registrationArgument, context,
(IErrorReporter) collector);
_collector = collector;
}
@Override
public void execute(Tuple input) {
_metricsConsumer.handleDataPoints(
(IMetricsConsumer.TaskInfo) input.getValue(0),
(Collection) input.getValue(1));
_collector.ack(input);
}
@Override
public void cleanup() {
_metricsConsumer.cleanup();
}
}