MetricsInterceptor.java

/**
 *  Copyright (c) 2015 The original author or authors
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0

 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package org.reveno.atp.metrics;

import org.reveno.atp.api.domain.WriteableRepository;
import org.reveno.atp.api.transaction.TransactionInterceptor;
import org.reveno.atp.api.transaction.TransactionStage;
import org.reveno.atp.metrics.impl.GraphiteSink;
import org.reveno.atp.metrics.impl.Slf4jSink;
import org.reveno.atp.metrics.meter.Counter;
import org.reveno.atp.metrics.meter.Histogram;
import org.reveno.atp.metrics.meter.impl.SimpleCounter;
import org.reveno.atp.metrics.meter.impl.TwoBufferHistogram;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class MetricsInterceptor implements TransactionInterceptor {

	@Override
	public void intercept(long transactionId, long time, long systemFlag, WriteableRepository repository, TransactionStage stage) {
		if (systemFlag == 0) {
			counter.inc();
			histogram.update(System.nanoTime() - time);
		}
	}
	
	@Override
	public void destroy() {
		if (!executor.isShutdown())
			executor.shutdown();
	}

	public MetricsInterceptor(ConfigurationImpl config) {
		String prefix = "reveno.instances." + config.hostName().replace(".", "_") + "." + config.instanceName() + ".";
		this.latency = prefix + "latency";
		this.throughput = prefix + "throughput";
		this.config = config;
		this.counter = new SimpleCounter(throughput);
		this.histogram = new TwoBufferHistogram(latency, config.metricBufferSize());
	}
	
	public void init() {
		if (config.sendToGraphite()) {
			sinks.add(new GraphiteSink(config.graphiteServer(), config.graphitePort()));
		}
		if (config.sendToLog()) {
			sinks.add(new Slf4jSink());
		}
		sinks.forEach(Sink::init);
		executor.scheduleAtFixedRate(() -> {
			histogram.sendTo(sinks, true);
			counter.sendTo(sinks, true);
		}, 15, 15, TimeUnit.SECONDS);
	}
	
	public void shutdown() {
		sinks.forEach(Sink::close);
		if (histogram != null) {
			histogram.destroy();
		}
		executor.shutdown();
	}
	
	protected Counter counter;
	protected Histogram histogram;
	protected List<Sink> sinks = new ArrayList<>();
	protected final String latency, throughput;
	protected final ConfigurationImpl config;
	protected final ScheduledExecutorService executor = Executors.newSingleThreadScheduledExecutor();
}