org.mockserver.metrics.Metrics Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockserver-core Show documentation
Functionality used by all MockServer modules for matching and expectations
There is a newer version: 5.15.0
package org.mockserver.metrics;

import org.mockserver.configuration.ConfigurationProperties;
import org.mockserver.model.Action;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author jamesdbloom
 */
@SuppressWarnings("SynchronizationOnLocalVariableOrMethodParameter")
public class Metrics {

    private static Map metrics = new ConcurrentHashMap<>();

    public static void clear() {
        metrics.clear();
    }

    public static void set(Name name, Integer value) {
        metrics.put(name, value);
    }

    public static Integer get(Name name) {
        Integer value = metrics.get(name);
        return value != null ? value : 0;
    }

    public static void increment(Name name) {
        if (ConfigurationProperties.metricsEnabled()) {
            synchronized (name) {
                metrics.merge(name, 1, Integer::sum);
            }
        }
    }

    public static void decrement(Name name) {
        if (ConfigurationProperties.metricsEnabled()) {
            synchronized (name) {
                final Integer currentValue = metrics.get(name);
                if (currentValue != null) {
                    metrics.put(name, currentValue - 1);
                } else {
                    throw new IllegalArgumentException("Can not decrement metric \"" + name + "\" because it not exist");
                }
            }
        }
    }

    public static void increment(Action.Type type) {
        if (ConfigurationProperties.metricsEnabled()) {
            Name name = Name.valueOf("ACTION_" + type.name() + "_COUNT");
            synchronized (name) {
                metrics.merge(name, 1, Integer::sum);
            }
        }
    }

    public static void decrement(Action.Type type) {
        if (ConfigurationProperties.metricsEnabled()) {
            Name name = Name.valueOf("ACTION_" + type.name() + "_COUNT");
            synchronized (name) {
                final Integer currentValue = metrics.get(name);
                if (currentValue != null) {
                    metrics.put(name, currentValue - 1);
                } else {
                    throw new IllegalArgumentException("Can not decrement metric \"" + name + "\" because it not exist");
                }
            }
        }
    }

    public static void clearActionMetrics() {
        metrics.remove(Name.ACTION_FORWARD_COUNT);
        metrics.remove(Name.ACTION_FORWARD_TEMPLATE_COUNT);
        metrics.remove(Name.ACTION_FORWARD_CLASS_CALLBACK_COUNT);
        metrics.remove(Name.ACTION_FORWARD_OBJECT_CALLBACK_COUNT);
        metrics.remove(Name.ACTION_FORWARD_REPLACE_COUNT);
        metrics.remove(Name.ACTION_RESPONSE_COUNT);
        metrics.remove(Name.ACTION_RESPONSE_TEMPLATE_COUNT);
        metrics.remove(Name.ACTION_RESPONSE_CLASS_CALLBACK_COUNT);
        metrics.remove(Name.ACTION_RESPONSE_OBJECT_CALLBACK_COUNT);
        metrics.remove(Name.ACTION_ERROR_COUNT);
    }

    public static void clearWebSocketMetrics() {
        metrics.remove(Name.WEBSOCKET_CALLBACK_CLIENT_COUNT);
        metrics.remove(Name.WEBSOCKET_CALLBACK_RESPONSE_HANDLER_COUNT);
        metrics.remove(Name.WEBSOCKET_CALLBACK_FORWARD_HANDLER_COUNT);
    }

    public enum Name {
        EXPECTATION_NOT_MATCHED_COUNT,
        RESPONSE_EXPECTATION_MATCHED_COUNT,
        FORWARD_EXPECTATION_MATCHED_COUNT,
        ACTION_FORWARD_COUNT,
        ACTION_FORWARD_TEMPLATE_COUNT,
        ACTION_FORWARD_CLASS_CALLBACK_COUNT,
        ACTION_FORWARD_OBJECT_CALLBACK_COUNT,
        ACTION_FORWARD_REPLACE_COUNT,
        ACTION_RESPONSE_COUNT,
        ACTION_RESPONSE_TEMPLATE_COUNT,
        ACTION_RESPONSE_CLASS_CALLBACK_COUNT,
        ACTION_RESPONSE_OBJECT_CALLBACK_COUNT,
        ACTION_ERROR_COUNT,
        WEBSOCKET_CALLBACK_CLIENT_COUNT,
        WEBSOCKET_CALLBACK_RESPONSE_HANDLER_COUNT,
        WEBSOCKET_CALLBACK_FORWARD_HANDLER_COUNT
    }
}