ai.vespa.metricsproxy.metric.model.json.YamasJsonModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metrics-proxy 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 ai.vespa.metricsproxy.metric.model.json;

import ai.vespa.metricsproxy.metric.Metric;
import ai.vespa.metricsproxy.metric.model.ConsumerId;
import ai.vespa.metricsproxy.metric.model.DimensionId;
import ai.vespa.metricsproxy.metric.model.MetricId;
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonPropertyOrder;

import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

import static com.yahoo.stream.CustomCollectors.toLinkedMap;

/**
 * Datamodel for Yamas execute output
 * 
 * Used to read from original yamas checks and annotate with routing information.
 */
@JsonIgnoreProperties(ignoreUnknown = true)
@JsonInclude(JsonInclude.Include.NON_NULL)
@JsonPropertyOrder({  "status_code", "timestamp", "application", "metrics", "dimensions", "routing", "status_msg"})
public class YamasJsonModel {
    @JsonProperty("status_code")
    public Integer status_code;
    @JsonProperty("status_msg")
    public String status_msg;
    @JsonProperty("timestamp")
    public Long timestamp;
    @JsonProperty("application")
    public String application;
    @JsonProperty("metrics")
    public Map metrics;
    @JsonProperty("dimensions")
    public Map dimensions;
    @JsonProperty("routing")
    public Map routing;

    public static class YamasJsonNamespace {
        @JsonProperty("namespaces")
        public List namespaces;
    }

    // NOTE: do not rename to 'setMetrics', as jackson will try to use it.
    public void resetMetrics(List newMetrics) {
        metrics = new LinkedHashMap<>();
        newMetrics.forEach(metric -> metrics.put(metric.getName().id, metric.getValue().doubleValue()));
    }

    /**
     * Convenience method to add dimensions
     */
    public void addDimensions(Map additionalDimensions, boolean replace) {
        additionalDimensions.forEach((k,v) -> {
            addDimension(k.id, v, replace);
        });
    }

    /**
     * Convenience method to add dimensions
     */
    public void addDimension(String key, String value, boolean replace) {
        if (dimensions == null) {
            dimensions = new HashMap<>();
        }
        if (!dimensions.containsKey(key) || replace) {
            dimensions.put(key, value);
        }
    }

    List getMetricsList() {
        if (metrics == null) return List.of();

        return metrics.keySet().stream()
                .map(name -> new Metric(MetricId.toMetricId(name), metrics.get(name)))
                .toList();
    }

    Map getDimensionsById() {
        if (dimensions == null) return Map.of();

        return dimensions.keySet().stream().collect(toLinkedMap(DimensionId::toDimensionId,
                                                                name -> dimensions.get(name)));
    }

    Set getYamasConsumers() {
        if (routing == null || routing.get("yamas") == null) return Set.of();

        return routing.get("yamas").namespaces.stream()
                .map(ConsumerId::toConsumerId)
                .collect(Collectors.toCollection(LinkedHashSet::new));
    }
}