62 lines
2 KiB
Java
62 lines
2 KiB
Java
package io.kokuwa.keycloak.metrics.junit;
|
|
|
|
import java.util.HashSet;
|
|
import java.util.Map;
|
|
import java.util.Objects;
|
|
import java.util.Set;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Stream;
|
|
|
|
import org.keycloak.events.EventType;
|
|
|
|
/**
|
|
* Client to access Prometheus metric values:
|
|
*
|
|
* @author Stephan Schnabel
|
|
*/
|
|
public class Prometheus {
|
|
|
|
private final Set<PrometheusMetric> state = new HashSet<>();
|
|
private final PrometheusClient client;
|
|
|
|
public Prometheus(PrometheusClient client) {
|
|
this.client = client;
|
|
}
|
|
|
|
public int userEvent(EventType type) {
|
|
return state.stream()
|
|
.filter(metric -> Objects.equals(metric.name(), "keycloak_event_user_total"))
|
|
.filter(metric -> Objects.equals(metric.tags().get("type"), type.toString()))
|
|
.mapToInt(metric -> metric.value().intValue())
|
|
.sum();
|
|
}
|
|
|
|
public int userEvent(EventType type, String realmName, String clientId) {
|
|
return state.stream()
|
|
.filter(metric -> Objects.equals(metric.name(), "keycloak_event_user_total"))
|
|
.filter(metric -> Objects.equals(metric.tags().get("type"), type.toString()))
|
|
.filter(metric -> Objects.equals(metric.tags().get("realm"), realmName))
|
|
.filter(metric -> Objects.equals(metric.tags().get("client"), clientId))
|
|
.mapToInt(metric -> metric.value().intValue())
|
|
.sum();
|
|
}
|
|
|
|
public void scrap() {
|
|
state.clear();
|
|
Stream.of(client.scrap().split("[\\r\\n]+"))
|
|
.filter(line -> !line.startsWith("#"))
|
|
.filter(line -> line.startsWith("keycloak"))
|
|
.map(line -> {
|
|
var name = line.substring(0, line.contains("{") ? line.indexOf("{") : line.lastIndexOf(" "));
|
|
var tags = line.contains("{")
|
|
? Stream.of(line.substring(line.indexOf("{") + 1, line.indexOf("}")).split(","))
|
|
.map(tag -> tag.split("="))
|
|
.filter(tag -> tag.length >= 2)
|
|
.collect(Collectors.toMap(tag -> tag[0], tag -> tag[1].replace("\"", "")))
|
|
: Map.<String, String>of();
|
|
var value = Double.parseDouble(line.substring(line.lastIndexOf(" ")));
|
|
return new PrometheusMetric(name, tags, value);
|
|
})
|
|
.forEach(state::add);
|
|
}
|
|
}
|