/*
 * Copyright 2014 Red Hat, Inc.
 *
 *  All rights reserved. This program and the accompanying materials
 *  are made available under the terms of the Eclipse Public License v1.0
 *  and Apache License v2.0 which accompanies this distribution.
 *
 *  The Eclipse Public License is available at
 *  http://www.eclipse.org/legal/epl-v10.html
 *
 *  The Apache License v2.0 is available at
 *  http://www.opensource.org/licenses/apache2.0.php
 *
 *  You may elect to redistribute this code under either of these licenses.
 */

package io.vertx.ext.dropwizard.impl;

import com.codahale.metrics.Counter;
import com.codahale.metrics.Meter;
import io.vertx.core.eventbus.Message;
import io.vertx.core.eventbus.ReplyFailure;
import io.vertx.core.spi.metrics.EventBusMetrics;
import io.vertx.ext.dropwizard.DropwizardMetricsOptions;
import io.vertx.ext.dropwizard.ThroughputMeter;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

Author:Nick Scavelli
/** * @author <a href="mailto:nscavell@redhat.com">Nick Scavelli</a> */
class EventBusMetricsImpl extends AbstractMetrics implements EventBusMetrics<EventBusMetricsImpl.HandlerMetric> { private final HandlerMetric ignoredHandler = new HandlerMetric(null, false, true); private final HandlerMetric noMatchHandler = new HandlerMetric(null, true, false); private final ConcurrentMap<String, HandlerCounter> handlerTimers = new ConcurrentHashMap<>(); private final Matcher handlerMatcher; private final Counter handlerCount; private final Counter pending; private final Counter pendingLocal; private final Counter pendingRemote; private final Counter discarded; private final Counter discardedLocal; private final Counter discardedRemote; private final Meter bytesRead; private final Meter bytesWritten; private final ThroughputMeter receivedMessages; private final ThroughputMeter receivedLocalMessages; private final ThroughputMeter receivedRemoteMessages; private final ThroughputMeter sentMessages; private final ThroughputMeter sentLocalMessages; private final ThroughputMeter sentRemoteMessages; private final ThroughputMeter publishedMessages; private final ThroughputMeter publishedLocalMessages; private final ThroughputMeter publishedRemoteMessages; private final ThroughputMeter deliveredMessages; private final ThroughputMeter deliveredLocalMessages; private final ThroughputMeter deliveredRemoteMessages; private final Meter replyFailures; EventBusMetricsImpl(AbstractMetrics metrics, String baseName, DropwizardMetricsOptions options) { super(metrics.registry(), baseName); handlerCount = counter("handlers"); pending = counter("messages", "pending"); pendingLocal = counter("messages", "pending-local"); pendingRemote = counter("messages", "pending-remote"); discarded = counter("messages", "discarded"); discardedLocal = counter("messages", "discarded-local"); discardedRemote = counter("messages", "discarded-remote"); receivedMessages = throughputMeter("messages", "received"); receivedLocalMessages = throughputMeter("messages", "received-local"); receivedRemoteMessages = throughputMeter("messages", "received-remote"); deliveredMessages = throughputMeter("messages", "delivered"); deliveredLocalMessages = throughputMeter("messages", "delivered-local"); deliveredRemoteMessages = throughputMeter("messages", "delivered-remote"); sentMessages = throughputMeter("messages", "sent"); sentLocalMessages = throughputMeter("messages", "sent-local"); sentRemoteMessages = throughputMeter("messages", "sent-remote"); publishedMessages = throughputMeter("messages", "published"); publishedLocalMessages = throughputMeter("messages", "published-local"); publishedRemoteMessages = throughputMeter("messages", "published-remote"); replyFailures = meter("messages", "reply-failures"); bytesRead = meter("messages", "bytes-read"); bytesWritten = meter("messages", "bytes-written"); handlerMatcher = new Matcher(options.getMonitoredEventBusHandlers()); } private static boolean isInternal(String address) { return address.startsWith("__vertx."); } @Override public void messageWritten(String address, int size) { if (!isInternal(address)) { bytesWritten.mark(size); } } @Override public void messageRead(String address, int size) { if (!isInternal(address)) { bytesRead.mark(size); } } @Override public void close() { } @Override public HandlerMetric handlerRegistered(String address, String repliedAddress) { if (isInternal(address)) { return ignoredHandler; } handlerCount.inc(); String match = handlerMatcher.matches(address); if (match != null) { return new HandlerMetric(match, false, false); } return noMatchHandler; } @Override public void handlerUnregistered(HandlerMetric handler) { if (handler.ignored) { return; } handlerCount.dec(); if (!handler.noMatch) { handler.remove(); } } @Override public void scheduleMessage(HandlerMetric handler, boolean local) { if (handler.ignored) { return; } pending.inc(); if (local) { pendingLocal.inc(); if (!handler.noMatch) { handler.pendingLocalCount++; } } else { pendingRemote.inc(); if (!handler.noMatch) { handler.pendingRemoteCount++; } } } @Override public void discardMessage(HandlerMetric handler, boolean local, Message<?> msg) { if (handler.ignored) { return; } pending.dec(); discarded.inc(); if (local) { discardedLocal.inc(); pendingLocal.dec(); if (!handler.noMatch) { handler.pendingLocalCount--; } } else { discardedRemote.inc(); pendingRemote.dec(); if (!handler.noMatch) { handler.pendingRemoteCount--; } } } @Override public void messageDelivered(HandlerMetric handler, boolean local) { if (handler.ignored) { return; } pending.dec(); if (local) { pendingLocal.dec(); } else { pendingRemote.dec(); } if (!handler.noMatch) { if (local) { handler.pendingLocalCount--; } else { handler.pendingRemoteCount--; } handler.counter.inc(); } } @Override public void messageSent(String address, boolean publish, boolean local, boolean remote) { if (!isInternal(address)) { if (publish) { publishedMessages.mark(); if (local) { publishedLocalMessages.mark(); } else { publishedRemoteMessages.mark(); } } else { sentMessages.mark(); if (local) { sentLocalMessages.mark(); } else { sentRemoteMessages.mark(); } } } } @Override public void messageReceived(String address, boolean publish, boolean local, int handlers) { if (!isInternal(address)) { receivedMessages.mark(); if (local) { receivedLocalMessages.mark(); } else { receivedRemoteMessages.mark(); } if (handlers > 0) { deliveredMessages.mark(); if (local) { deliveredLocalMessages.mark(); } else { deliveredRemoteMessages.mark(); } } } } public class HandlerMetric { final String address; final Counter counter; final String name; final boolean noMatch; final boolean ignored; long pendingLocalCount; long pendingRemoteCount; public HandlerMetric(String address, boolean noMatch, boolean ignored) { this.address = address; this.noMatch = noMatch; this.ignored = ignored; if (noMatch || ignored) { this.counter = null; this.name = null; return; } this.name = nameOf("handlers", address); while (true) { HandlerCounter existing = handlerTimers.get(address); if (existing != null) { HandlerCounter next = existing.inc(); if (handlerTimers.replace(address, existing, next)) { counter = next.counter; break; } } else { HandlerCounter created = new HandlerCounter(); if (handlerTimers.putIfAbsent(address, created) == null) { registry.register(name, created.counter); counter = created.counter; break; } } } } void remove() { if (!noMatch && !ignored) { while (true) { HandlerCounter existing = handlerTimers.get(address); HandlerCounter next = existing.dec(); if (next.refCount == 0) { if (handlerTimers.remove(address, existing)) { registry.remove(name); break; } } else { if (handlerTimers.replace(address, existing, next)) { break; } } } } } } @Override public void replyFailure(String address, ReplyFailure failure) { replyFailures.mark(); meter("messages", "reply-failures", failure.name()).mark(); } static class HandlerCounter { final int refCount; final Counter counter; public HandlerCounter(int refCount, Counter counter) { this.refCount = refCount; this.counter = counter; } public HandlerCounter() { this(1, new Counter()); } HandlerCounter inc() { return new HandlerCounter(refCount + 1, counter); } HandlerCounter dec() { return new HandlerCounter(refCount - 1, counter); } @Override public boolean equals(Object obj) { HandlerCounter that = (HandlerCounter) obj; return refCount == that.refCount; } } }