Adding the APIMetricsRegistry the metric factory
The MetricsRegistry in yammer used to create metrics and register them in the JMX. The APIMetricsRegistry extends the functionality by accepting a URL that the API will perform to retrieve the relevant data, it also creates the API version of each of the requested Metric (i.e. APIMeter, APICounter or APIHistogram) Signed-off-by: Amnon Heiman <amnon@cloudius-systems.com>
This commit is contained in:
Amnon Heiman
parent
dd2d50a45c
commit
1212e09ffd
234
src/main/java/com/yammer/metrics/core/APIMetricsRegistry.java
Normal file
234
src/main/java/com/yammer/metrics/core/APIMetricsRegistry.java
Normal file
|
|
@ -0,0 +1,234 @@
|
|||
package com.yammer.metrics.core;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.concurrent.ConcurrentMap;
|
||||
import java.util.concurrent.ScheduledExecutorService;
|
||||
import java.util.concurrent.TimeUnit;
|
||||
|
||||
import com.yammer.metrics.core.APICounter;
|
||||
import com.yammer.metrics.core.APIMeter;
|
||||
import com.yammer.metrics.core.Clock;
|
||||
import com.yammer.metrics.core.Counter;
|
||||
import com.yammer.metrics.core.Meter;
|
||||
import com.yammer.metrics.core.Metric;
|
||||
import com.yammer.metrics.core.MetricName;
|
||||
import com.yammer.metrics.core.MetricsRegistry;
|
||||
import com.yammer.metrics.core.ThreadPools;
|
||||
import com.yammer.metrics.core.Histogram.SampleType;
|
||||
|
||||
/*
|
||||
* Copyright 2015 Cloudius Systems
|
||||
*
|
||||
* Modified by Cloudius Systems
|
||||
*/
|
||||
|
||||
public class APIMetricsRegistry extends MetricsRegistry {
|
||||
Field field_metrics;
|
||||
Field field_clock;
|
||||
Field field_thread_pool;
|
||||
|
||||
public APIMetricsRegistry() {
|
||||
try {
|
||||
field_metrics = MetricsRegistry.class.getDeclaredField("metrics");
|
||||
field_metrics.setAccessible(true);
|
||||
field_clock = MetricsRegistry.class.getDeclaredField("clock");
|
||||
field_clock.setAccessible(true);
|
||||
field_thread_pool = MetricsRegistry.class.getDeclaredField("threadPools");
|
||||
field_thread_pool.setAccessible(true);
|
||||
} catch (NoSuchFieldException | SecurityException e) {
|
||||
// TODO Auto-generated catch block
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
public ThreadPools getThreadPools() {
|
||||
try {
|
||||
return (ThreadPools)field_thread_pool.get(this);
|
||||
} catch (IllegalArgumentException | IllegalAccessException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
public Clock getClock() {
|
||||
try {
|
||||
return (Clock)field_clock.get(this);
|
||||
} catch (IllegalArgumentException | IllegalAccessException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
public ConcurrentMap<MetricName, Metric> get_metrics() {
|
||||
try {
|
||||
return (ConcurrentMap<MetricName, Metric>)field_metrics.get(this);
|
||||
} catch (IllegalArgumentException | IllegalAccessException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
/**
|
||||
* Creates a new {@link Counter} and registers it under the given class and name.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @return a new {@link Counter}
|
||||
*/
|
||||
public Counter newCounter(String url, Class<?> klass,
|
||||
String name) {
|
||||
return newCounter(url, klass, name, null);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new {@link Counter} and registers it under the given class and name.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @param scope the scope of the metric
|
||||
* @return a new {@link Counter}
|
||||
*/
|
||||
public Counter newCounter(String url, Class<?> klass,
|
||||
String name,
|
||||
String scope) {
|
||||
return newCounter(url, createName(klass, name, scope));
|
||||
}
|
||||
/**
|
||||
* Creates a new {@link Counter} and registers it under the given metric name.
|
||||
*
|
||||
* @param metricName the name of the metric
|
||||
* @return a new {@link Counter}
|
||||
*/
|
||||
public Counter newCounter(String url, MetricName metricName) {
|
||||
return getOrAdd(metricName, new APICounter(url));
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new {@link Meter} and registers it under the given class and name.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @param eventType the plural name of the type of events the meter is measuring (e.g., {@code
|
||||
* "requests"})
|
||||
* @param unit the rate unit of the new meter
|
||||
* @return a new {@link Meter}
|
||||
*/
|
||||
public Meter newMeter(String url, Class<?> klass,
|
||||
String name,
|
||||
String eventType,
|
||||
TimeUnit unit) {
|
||||
return newMeter(url, klass, name, null, eventType, unit);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new {@link Meter} and registers it under the given class, name, and scope.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @param scope the scope of the metric
|
||||
* @param eventType the plural name of the type of events the meter is measuring (e.g., {@code
|
||||
* "requests"})
|
||||
* @param unit the rate unit of the new meter
|
||||
* @return a new {@link Meter}
|
||||
*/
|
||||
public Meter newMeter(String url,
|
||||
Class<?> klass,
|
||||
String name,
|
||||
String scope,
|
||||
String eventType,
|
||||
TimeUnit unit) {
|
||||
return newMeter(url, createName(klass, name, scope), eventType, unit);
|
||||
}
|
||||
|
||||
private ScheduledExecutorService newMeterTickThreadPool() {
|
||||
return getThreadPools().newScheduledThreadPool(2, "meter-tick");
|
||||
}
|
||||
/**
|
||||
* Creates a new {@link Meter} and registers it under the given metric name.
|
||||
*
|
||||
* @param metricName the name of the metric
|
||||
* @param eventType the plural name of the type of events the meter is measuring (e.g., {@code
|
||||
* "requests"})
|
||||
* @param unit the rate unit of the new meter
|
||||
* @return a new {@link Meter}
|
||||
*/
|
||||
public Meter newMeter(String url, MetricName metricName,
|
||||
String eventType,
|
||||
TimeUnit unit) {
|
||||
final Metric existingMetric = get_metrics().get(metricName);
|
||||
if (existingMetric != null) {
|
||||
return (Meter) existingMetric;
|
||||
}
|
||||
return getOrAdd(metricName, new APIMeter(url, newMeterTickThreadPool(), eventType, unit, getClock()));
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new {@link Histogram} and registers it under the given class and name.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @param biased whether or not the histogram should be biased
|
||||
* @return a new {@link Histogram}
|
||||
*/
|
||||
public Histogram newHistogram(String url, Class<?> klass,
|
||||
String name,
|
||||
boolean biased) {
|
||||
return newHistogram(url, klass, name, null, biased);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new {@link Histogram} and registers it under the given class, name, and scope.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @param scope the scope of the metric
|
||||
* @param biased whether or not the histogram should be biased
|
||||
* @return a new {@link Histogram}
|
||||
*/
|
||||
public Histogram newHistogram(String url, Class<?> klass,
|
||||
String name,
|
||||
String scope,
|
||||
boolean biased) {
|
||||
return newHistogram(url, createName(klass, name, scope), biased);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new non-biased {@link Histogram} and registers it under the given class and name.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @return a new {@link Histogram}
|
||||
*/
|
||||
public Histogram newHistogram(String url, Class<?> klass,
|
||||
String name) {
|
||||
return newHistogram(url, klass, name, false);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new non-biased {@link Histogram} and registers it under the given class, name, and
|
||||
* scope.
|
||||
*
|
||||
* @param klass the class which owns the metric
|
||||
* @param name the name of the metric
|
||||
* @param scope the scope of the metric
|
||||
* @return a new {@link Histogram}
|
||||
*/
|
||||
public Histogram newHistogram(String url, Class<?> klass,
|
||||
String name,
|
||||
String scope) {
|
||||
return newHistogram(url, klass, name, scope, false);
|
||||
}
|
||||
|
||||
/**
|
||||
* Creates a new {@link Histogram} and registers it under the given metric name.
|
||||
*
|
||||
* @param metricName the name of the metric
|
||||
* @param biased whether or not the histogram should be biased
|
||||
* @return a new {@link Histogram}
|
||||
*/
|
||||
public Histogram newHistogram(String url, MetricName metricName,
|
||||
boolean biased) {
|
||||
return getOrAdd(metricName,
|
||||
new APIHistogram(url, biased ? SampleType.BIASED : SampleType.UNIFORM));
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user