com.redislabs.riot.redis.LatencyCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of riot-redis Show documentation
RIOT Core module
The newest version!
package com.redislabs.riot.redis;

import io.lettuce.core.api.sync.BaseRedisCommands;
import io.lettuce.core.metrics.CommandMetrics;
import io.lettuce.core.metrics.DefaultCommandLatencyCollectorOptions;
import org.HdrHistogram.Histogram;
import org.LatencyUtils.LatencyStats;
import picocli.CommandLine.Command;
import picocli.CommandLine.Option;

import java.util.Map;
import java.util.TreeMap;
import java.util.concurrent.TimeUnit;

@Command(name = "latency", description = "Calculate latency stats")
public class LatencyCommand extends AbstractRedisCommand {

	@Option(names = "--iterations", description = "Number of latency tests (default: ${DEFAULT-VALUE})", paramLabel = "")
	private int iterations = 1000;
	@Option(names = "--sleep", description = "Sleep duration between calls (default: ${DEFAULT-VALUE})", paramLabel = "")
	private long sleep = 1;
	@Option(names = "--unit", description = "Latency unit (default: ${DEFAULT-VALUE})", paramLabel = "")
	private TimeUnit unit = TimeUnit.MILLISECONDS;
	@Option(names = "--show-distribution", description = "Show latency distribution")
	private boolean showDistribution = false;

	@Override
	protected void execute(BaseRedisCommands commands) {
		LatencyStats stats = new LatencyStats();
		for (int index = 0; index < iterations; index++) {
			long startTime = System.nanoTime();
			commands.ping();
			stats.recordLatency(System.nanoTime() - startTime);
			try {
				Thread.sleep(sleep);
			} catch (InterruptedException e) {
				return;
			}
		}
		Histogram histogram = stats.getIntervalHistogram();
		if (showDistribution) {
			histogram.outputPercentileDistribution(System.out, 1000000.0);
		} else {
			DefaultCommandLatencyCollectorOptions options = DefaultCommandLatencyCollectorOptions.create();
			Map percentiles = new TreeMap<>();
			for (double targetPercentile : options.targetPercentiles()) {
				percentiles.put(targetPercentile,
						unit.convert(histogram.getValueAtPercentile(targetPercentile), TimeUnit.NANOSECONDS));
			}
			CommandMetrics.CommandLatency latency = new CommandMetrics.CommandLatency(
					unit.convert(histogram.getMinValue(), TimeUnit.NANOSECONDS),
					unit.convert(histogram.getMaxValue(), TimeUnit.NANOSECONDS), percentiles);
			System.out.println(latency.toString());
		}
	}
}