com.pinterest.doctorkafka.stats.KafkaMetricRetrievingTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kafkastats Show documentation

Collectoring the kafka stats for DoctorKafka

There is a newer version: 0.3.0-rc.2

package com.pinterest.doctorkafka.stats;

import java.util.concurrent.Callable;
import javax.management.MBeanServerConnection;
import javax.management.ObjectName;

public class KafkaMetricRetrievingTask implements Callable {

  private MBeanServerConnection mbs;
  private String metricName;
  private String attributeName;

  public KafkaMetricRetrievingTask(MBeanServerConnection mbs,
                                   String metricName, String attributeName) {
    this.mbs = mbs;
    this.metricName = metricName;
    this.attributeName = attributeName;
  }


  @Override
  public KafkaMetricValue call() throws Exception {
    try {
      Object obj = mbs.getAttribute(new ObjectName(metricName), attributeName);
      return new KafkaMetricValue(obj);
    } catch (Exception e)  {
      return new KafkaMetricValue(e);
    }
  }
}