EventConsumerStats.java

package com.netflix.eventbus.impl;

import com.netflix.servo.DefaultMonitorRegistry;
import com.netflix.servo.monitor.BasicCounter;
import com.netflix.servo.monitor.BasicGauge;
import com.netflix.servo.monitor.Counter;
import com.netflix.servo.monitor.MonitorConfig;
import com.netflix.servo.monitor.StatsTimer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.concurrent.Callable;
import java.util.concurrent.atomic.AtomicLong;

import static com.netflix.eventbus.utils.EventBusUtils.newStatsTimer;

/**
 * @author Nitesh Kant (nkant@netflix.com)
 */
public class EventConsumerStats {

    private static final Logger LOGGER = LoggerFactory.getLogger(EventConsumerStats.class);

    final StatsTimer enqueueStats;
    final StatsTimer consumptionStats;
    final StatsTimer filterStats;
    final AtomicLong QUEUE_SIZE_COUNTER;
    final BasicGauge<Long> QUEUE_SIZE_GAUGE;
    final Counter QUEUE_OFFER_RETRY_COUNTER;
    final Counter EVENT_ENQUEUE_REJECTED_COUNTER;

    public EventConsumerStats(String consumerName, long collectionDurationInMillis) {
        String statsPrefix = "eventbus_consumer_" + consumerName;
        QUEUE_SIZE_COUNTER = new AtomicLong();
        QUEUE_SIZE_GAUGE = new BasicGauge<Long>(MonitorConfig.builder(statsPrefix + "_queue_size").build(), new Callable<Long>() {
            @Override
            public Long call() throws Exception {
                return QUEUE_SIZE_COUNTER.get();
            }
        });
        QUEUE_OFFER_RETRY_COUNTER = new BasicCounter(MonitorConfig.builder(statsPrefix + "_queue_retry").build());
        EVENT_ENQUEUE_REJECTED_COUNTER = new BasicCounter(MonitorConfig.builder(statsPrefix + "_enqueue_reject").build());

        enqueueStats = newStatsTimer(statsPrefix + "_enqueue", collectionDurationInMillis);
        consumptionStats = newStatsTimer(statsPrefix + "_consumption", collectionDurationInMillis);
        filterStats = newStatsTimer(statsPrefix + "_filter", collectionDurationInMillis);
        try {
            DefaultMonitorRegistry.getInstance().register(QUEUE_SIZE_GAUGE);
            DefaultMonitorRegistry.getInstance().register(QUEUE_OFFER_RETRY_COUNTER);
            DefaultMonitorRegistry.getInstance().register(EVENT_ENQUEUE_REJECTED_COUNTER);
            DefaultMonitorRegistry.getInstance().register(enqueueStats);
            DefaultMonitorRegistry.getInstance().register(consumptionStats);
            DefaultMonitorRegistry.getInstance().register(filterStats);
        } catch (Throwable th) {
            LOGGER.error("Unable to register to event bus consumer stats to servo.", th);
        }
    }
}