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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
// 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.prometheus;
import ai.vespa.metricsproxy.metric.model.DimensionId;
import ai.vespa.metricsproxy.metric.model.MetricId;
import ai.vespa.metricsproxy.metric.model.MetricsPacket;
import ai.vespa.metricsproxy.metric.model.ServiceId;
import io.prometheus.client.Collector;
import io.prometheus.client.Collector.MetricFamilySamples;
import io.prometheus.client.Collector.MetricFamilySamples.Sample;
import io.prometheus.client.exporter.common.TextFormat;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
/**
* @author yj-jtakagi
* @author gjoranv
*/
public class PrometheusModel implements Enumeration<MetricFamilySamples> {
private final Map<ServiceId, List<MetricsPacket>> packetsByServiceId;
private final Iterator<MetricId> metricIterator;
private final Iterator<MetricFamilySamples> statusMetrics;
PrometheusModel(Set<MetricId> metricNames, Map<ServiceId,
List<MetricsPacket>> packetsByServiceId,
List<MetricFamilySamples> statusMetrics)
{
metricIterator = metricNames.iterator();
this.packetsByServiceId = packetsByServiceId;
this.statusMetrics = statusMetrics.iterator();
}
@Override
public boolean hasMoreElements() {
return metricIterator.hasNext() || statusMetrics.hasNext();
}
@Override
public MetricFamilySamples nextElement() {
return metricIterator.hasNext()
? createMetricFamily(metricIterator.next())
: statusMetrics.next();
}
public String serialize() {
var writer = new StringWriter();
try {
TextFormat.write004(writer, this);
} catch (Exception e) {
throw new PrometheusRenderingException("Could not render metrics. Check the log for details.", e);
}
return writer.toString();
}
private MetricFamilySamples createMetricFamily(MetricId metricId) {
List<MetricFamilySamples.Sample> sampleList = new ArrayList<>();
packetsByServiceId.forEach(((serviceId, packets) -> {
for (var packet : packets) {
Number metric = packet.metrics().get(metricId);
if (metric != null) {
sampleList.add(createSample(serviceId, metricId, metric, packet.timestamp, packet.dimensions()));
}
}
}));
return new MetricFamilySamples(metricId.getIdForPrometheus(), Collector.Type.UNKNOWN, "", sampleList);
}
private static Sample createSample(ServiceId serviceId, MetricId metricId, Number metric,
Long timeStamp, Map<DimensionId, String> dimensions)
{
List<String> labels = new ArrayList<>(dimensions.size());
List<String> labelValues = new ArrayList<>(dimensions.size());
for (var entry : dimensions.entrySet()) {
var labelName = entry.getKey().getIdForPrometheus();
labels.add(labelName);
labelValues.add(entry.getValue());
}
labels.add("vespa_service");
labelValues.add(serviceId.getIdForPrometheus());
return new Sample(metricId.getIdForPrometheus(), labels, labelValues, metric.doubleValue(), timeStamp);
}
}
|