scylla-jmx/src/main/java/com/yammer/metrics/core/APIMetricsRegistry.java

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 fieldMetrics;
    Field fieldClock;
    Field fieldThreadPool;

    public APIMetricsRegistry() {
        try {
            fieldMetrics = MetricsRegistry.class.getDeclaredField("metrics");
            fieldMetrics.setAccessible(true);
            fieldClock = MetricsRegistry.class.getDeclaredField("clock");
            fieldClock.setAccessible(true);
            fieldThreadPool = MetricsRegistry.class
                    .getDeclaredField("threadPools");
            fieldThreadPool.setAccessible(true);
        } catch (NoSuchFieldException | SecurityException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    public ThreadPools getThreadPools() {
        try {
            return (ThreadPools) fieldThreadPool.get(this);
        } catch (IllegalArgumentException | IllegalAccessException e) {
            e.printStackTrace();
        }
        return null;
    }

    public Clock getClock() {
        try {
            return (Clock) fieldClock.get(this);
        } catch (IllegalArgumentException | IllegalAccessException e) {
            e.printStackTrace();
        }
        return null;
    }

    @SuppressWarnings("unchecked")
    public ConcurrentMap<MetricName, Metric> getMetrics() {
        try {
            return (ConcurrentMap<MetricName, Metric>) fieldMetrics.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 APIMeter 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 APIMeter 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 APIMeter newMeter(String url, MetricName metricName, String eventType,
            TimeUnit unit) {
        final Metric existingMetric = getMetrics().get(metricName);
        if (existingMetric != null) {
            return (APIMeter) existingMetric;
        }
        return getOrAdd(metricName, new APIMeter(url, newMeterTickThreadPool(),
                eventType, unit));
    }

    /**
     * Creates a new {@link APISettableMeter} 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 newSettableMeter(MetricName metricName, String eventType,
            TimeUnit unit) {
        final Metric existingMetric = getMetrics().get(metricName);
        if (existingMetric != null) {
            return (Meter) existingMetric;
        }
        return getOrAdd(metricName, new APISettableMeter(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));
    }

    /**
     * Creates a new {@link Timer} and registers it under the given class and
     * name, measuring elapsed time in milliseconds and invocations per second.
     *
     * @param klass
     *            the class which owns the metric
     * @param name
     *            the name of the metric
     * @return a new {@link Timer}
     */
    public Timer newTimer(String url, Class<?> klass, String name) {
        return newTimer(url, klass, name, null, TimeUnit.MILLISECONDS,
                TimeUnit.SECONDS);
    }

    /**
     * Creates a new {@link Timer} 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 durationUnit
     *            the duration scale unit of the new timer
     * @param rateUnit
     *            the rate scale unit of the new timer
     * @return a new {@link Timer}
     */
    public Timer newTimer(String url, Class<?> klass, String name,
            TimeUnit durationUnit, TimeUnit rateUnit) {
        return newTimer(url, klass, name, null, durationUnit, rateUnit);
    }

    /**
     * Creates a new {@link Timer} and registers it under the given class, name,
     * and scope, measuring elapsed time in milliseconds and invocations per
     * second.
     *
     * @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 Timer}
     */
    public Timer newTimer(String url, Class<?> klass, String name, String scope) {
        return newTimer(url, klass, name, scope, TimeUnit.MILLISECONDS,
                TimeUnit.SECONDS);
    }

    /**
     * Creates a new {@link Timer} 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 durationUnit
     *            the duration scale unit of the new timer
     * @param rateUnit
     *            the rate scale unit of the new timer
     * @return a new {@link Timer}
     */
    public Timer newTimer(String url, Class<?> klass, String name,
            String scope, TimeUnit durationUnit, TimeUnit rateUnit) {
        return newTimer(url, createName(klass, name, scope), durationUnit,
                rateUnit);
    }

    /**
     * Creates a new {@link Timer} and registers it under the given metric name.
     *
     * @param metricName
     *            the name of the metric
     * @param durationUnit
     *            the duration scale unit of the new timer
     * @param rateUnit
     *            the rate scale unit of the new timer
     * @return a new {@link Timer}
     */
    public Timer newTimer(String url, MetricName metricName,
            TimeUnit durationUnit, TimeUnit rateUnit) {
        final Metric existingMetric = getMetrics().get(metricName);
        if (existingMetric != null) {
            return (Timer) existingMetric;
        }
        return getOrAdd(metricName, new APITimer(url, newMeterTickThreadPool(),
                durationUnit, rateUnit));
    }

}