com.yahoo.jdisc.test.MockMetric Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jdisc_core Show documentation

There is a newer version: 8.441.21

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.jdisc.test;

import com.yahoo.jdisc.Metric;

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

/**
 * Metric implementation for tests.
 *
 * @author jonmv
 */
public class MockMetric implements Metric {

    private final Map, Double>> metrics = new ConcurrentHashMap<>();

    @Override
    public void set(String key, Number val, Context ctx) {
        metrics.computeIfAbsent(key, k -> new ConcurrentHashMap<>())
               .put(MapContext.emptyIfNull(ctx).dimensions, val.doubleValue());
    }

    @Override
    public void add(String key, Number val, Context ctx) {
        metrics.computeIfAbsent(key, k -> new ConcurrentHashMap<>())
               .merge(MapContext.emptyIfNull(ctx).dimensions, val.doubleValue(), Double::sum);
    }

    @Override
    public Context createContext(Map properties) {
        return properties == null ? MapContext.empty : new MapContext(properties);
    }

    public Map, Double>> metrics() { return metrics; }

    @Override
    public String toString() {
        return "MockMetric{" +
                "metrics=" + metrics +
                '}';
    }

    private static class MapContext implements Context {

        private static final MapContext empty = new MapContext(Map.of());

        private final Map dimensions;

        private MapContext(Map dimensions) {
            this.dimensions = new HashMap<>(dimensions.size());
            this.dimensions.putAll(dimensions);
        }

        private static MapContext emptyIfNull(Context context) {
            return context == null ? empty : (MapContext) context;
        }

    }

}