com.version1.metrics.MetricRegistryHelper Maven / Gradle / Ivy

Go to download
package com.version1.metrics;

import com.codahale.metrics.MetricFilter;
import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.graphite.Graphite;
import com.codahale.metrics.graphite.GraphiteReporter;
import com.codahale.metrics.jmx.JmxReporter;
import com.version1.webdriver.configuration.TestConfigHelper;

import java.net.InetSocketAddress;
import java.util.concurrent.TimeUnit;

public class MetricRegistryHelper {

    private static final MetricRegistry metricRegistry = new MetricRegistry();
    private static MetricRegistryHelper metricRegistryHelper;
    private JmxReporter jmxReporter;
    private GraphiteReporter graphiteReporter;
    private Graphite graphite;

    /**
     *
     */
    private MetricRegistryHelper() {
        if(TestConfigHelper.get().getMetricsConfig().getJmxConfig().isEnabled()) {
            jmxReporter = JmxReporter.forRegistry(metricRegistry).build();
            jmxReporter.start();
        }
        if(TestConfigHelper.get().getMetricsConfig().getGraphiteConfig().isEnabled()) {
            graphite = new Graphite(new InetSocketAddress(
                    TestConfigHelper.get().getMetricsConfig().getGraphiteConfig().getHost(),
                    TestConfigHelper.get().getMetricsConfig().getGraphiteConfig().getPort()));
            graphiteReporter = GraphiteReporter.forRegistry(metricRegistry)
                    .prefixedWith("selenium-pom-framework")
                    .convertRatesTo(TimeUnit.SECONDS)
                    .convertDurationsTo(TimeUnit.MILLISECONDS)
                    .filter(MetricFilter.ALL)
                    .build(graphite);
            graphiteReporter.start(30, TimeUnit.SECONDS);
        }
    }

    /**
     *
     * @return MetricRegistryHelper returns the instance of the singleton to access the member methods
     */
    public static MetricRegistry get() {
        if(null == metricRegistryHelper) {
            metricRegistryHelper = new MetricRegistryHelper();
        }
        return metricRegistry;
    }
}