config-model/src/main/java/com/yahoo/vespa/model/admin/metricsproxy/VespaServicesConfigGenerator.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.

package com.yahoo.vespa.model.admin.metricsproxy;

import ai.vespa.metricsproxy.service.VespaServicesConfig;
import com.yahoo.vespa.model.Service;

import java.util.List;
import java.util.stream.Collectors;

/**
 * @author gjoranv
 */
public class VespaServicesConfigGenerator {

    public static List<VespaServicesConfig.Service.Builder> generate(List<Service> services) {
        return services.stream()
                .filter(VespaServicesConfigGenerator::doIncludeServiceMetrics)
                .map(VespaServicesConfigGenerator::toServiceBuilder)
                .toList();
    }

    private static boolean doIncludeServiceMetrics(Service s) {
        return s.getStartupCommand().isPresent() || s.getServiceType().equals("configserver") || s.getServiceType().equals("config-sentinel");
    }

    private static VespaServicesConfig.Service.Builder toServiceBuilder(Service service) {
        VespaServicesConfig.Service.Builder builder = new VespaServicesConfig.Service.Builder()
                .configId(service.getConfigId())
                .name(service.getServiceName())
                .port(service.getHealthPort());

        service.getDefaultMetricDimensions().forEach((name, value) -> builder.dimension(toServiceDimensionBuilder(name, value)));
        return builder;
    }

    private static VespaServicesConfig.Service.Dimension.Builder toServiceDimensionBuilder(String name, String value) {
        return new VespaServicesConfig.Service.Dimension.Builder()
                .key(name)
                .value(value);
    }

}