io.dropwizard.redis.metrics.event.visitor.CommandLatencyEventVisitor Maven / Gradle / Ivy

Go to download
package io.dropwizard.redis.metrics.event.visitor;

import com.codahale.metrics.MetricRegistry;
import io.lettuce.core.cluster.event.ClusterTopologyChangedEvent;
import io.lettuce.core.event.connection.ConnectedEvent;
import io.lettuce.core.event.connection.ConnectionActivatedEvent;
import io.lettuce.core.event.connection.ConnectionDeactivatedEvent;
import io.lettuce.core.event.connection.DisconnectedEvent;
import io.lettuce.core.event.metrics.CommandLatencyEvent;

import java.util.HashMap;
import java.util.Map;

import static java.util.Objects.requireNonNull;

public class CommandLatencyEventVisitor implements EventVisitor {
    private final String name;
    private final MetricRegistry metrics;
    private final Map metricNames;

    public CommandLatencyEventVisitor(final String name, final MetricRegistry metrics) {
        this(name, metrics, new HashMap<>());
    }

    // Visible for testing
    CommandLatencyEventVisitor(final String name, final MetricRegistry metrics, final Map metricNames) {
        this.name = MetricRegistry.name(requireNonNull(name), "event.latency");
        this.metrics = requireNonNull(metrics);
        this.metricNames = requireNonNull(metricNames);
    }

    @Override
    public void visit(final CommandLatencyEvent event) {
        event.getLatencies().forEach((commandLatencyId, commandMetrics) -> {
            final String commandType = commandLatencyId.commandType().name();
            final String metricName = metricNames.computeIfAbsent(commandType, (type) -> MetricRegistry.name(name, type));
            metrics.timer(metricName).update(commandMetrics.getCompletion().getMax(), commandMetrics.getTimeUnit());
        });
    }

    @Override
    public void visit(final DisconnectedEvent event) {
        // do nothing
    }

    @Override
    public void visit(final ClusterTopologyChangedEvent event) {
        // do nothing
    }

    @Override
    public void visit(final ConnectedEvent event) {
        // do nothing
    }

    @Override
    public void visit(final ConnectionActivatedEvent event) {
        // do nothing
    }

    @Override
    public void visit(final ConnectionDeactivatedEvent event) {
        // do nothing
    }
}