com.tinkerpop.rexster.server.metrics.GangliaReporterConfig Maven / Gradle / Ivy

Go to download
package com.tinkerpop.rexster.server.metrics;

import com.codahale.metrics.MetricRegistry;
import com.codahale.metrics.ganglia.GangliaReporter;
import com.tinkerpop.rexster.Tokens;
import info.ganglia.gmetric4j.gmetric.GMetric;
import org.apache.commons.configuration.HierarchicalConfiguration;
import org.apache.log4j.Logger;

import java.util.ArrayList;
import java.util.List;

/**
 * Configures Ganglia as a reporter.
 *
 * @author Stephen Mallette (http://stephen.genoprime.com)
 */
class GangliaReporterConfig extends AbstractHostPortReporterConfig {
    private static final Logger logger = Logger.getLogger(GangliaReporterConfig.class);

    private List reporters = new ArrayList();

    public GangliaReporterConfig(final HierarchicalConfiguration config, final MetricRegistry metricRegistry) {
        super(config, metricRegistry);
        this.hostsString = this.registryConfiguration.getString(Tokens.REXSTER_REPORTER_HOSTS, "localhost:8649");

        logger.info(String.format("Configured Ganglia Metric Reporter [%s].", this.hostsString));
    }

    @Override
    public List getFullHostList() {
        return getHostListAndStringList();
    }

    @Override
    public boolean enable() {
        final List hosts = getFullHostList();
        if (hosts == null || hosts.isEmpty())
        {
            logger.error("No hosts specified, cannot enable GangliaReporter");
            return false;
        }

        try
        {
            for (HostPort hostPort : hosts) {
                final GMetric ganglia = new GMetric(hostPort.getHost(), hostPort.getPort(), GMetric.UDPAddressingMode.MULTICAST, 1);
                final GangliaReporter reporter = GangliaReporter.forRegistry(this.metricRegistry)
                            .convertDurationsTo(this.getRealDurationTimeUnitConversion())
                            .convertRatesTo(this.getRealRateTimeUnitConversion())
                            .filter(new RegexMetricFilter(this.inclusion, this.exclusion))
                            .build(ganglia);
                reporter.start(this.period, this.getRealTimeUnit());
                reporters.add(reporter);
            }
        }
        catch (Exception e)
        {
            logger.error("Failure while enabling ganglia reporter", e);
            return false;
        }
        return true;
    }

    @Override
    public void disable() {
        for (GangliaReporter reporter : reporters) {
            reporter.stop();
        }

        reporters.clear();
    }
}