finallycoffee/keycloak-metrics
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: migrate to using an own SPI and implementing that

Browse Source
This commit is contained in:
transcaffeine 2024-08-24 17:18:14 +02:00
parent 23bc2ec41d
commit 89d758ecae
Signed by: transcaffeine
GPG Key ID: 03624C433676E465
19 changed files with 151 additions and 79 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.idea/runConfigurations/keycloak_metrics__package_.xml → .idea/runConfigurations/keycloak_metrics__clean_package_.xml generated
View File

@ -1,5 +1,5 @@
<component name="ProjectRunConfigurationManager">
<configuration default="false" name="keycloak-metrics [package]" type="MavenRunConfiguration" factoryName="Maven" nameIsGenerated="true">
<configuration default="false" name="keycloak-metrics [clean,package]" type="MavenRunConfiguration" factoryName="Maven" nameIsGenerated="true">
<MavenSettings>
<option name="myGeneralSettings" />
<option name="myRunnerSettings" />
@ -11,6 +11,7 @@
</option>
<option name="goals">
<list>
<option value="clean" />
<option value="package" />
</list>
</option>

27
.idea/workspace.xml generated
View File

@ -5,13 +5,17 @@
</component>
<component name="ChangeListManager">
<list default="true" id="0b19344d-a6be-4de4-b466-20f9855520b4" name="Changes" comment="">
<change afterPath="$PROJECT_DIR$/.gitignore" afterDir="false" />
<change afterPath="$PROJECT_DIR$/.idea/encodings.xml" afterDir="false" />
<change afterPath="$PROJECT_DIR$/.idea/misc.xml" afterDir="false" />
<change afterPath="$PROJECT_DIR$/.idea/runConfigurations/keycloak_metrics__package_.xml" afterDir="false" />
<change afterPath="$PROJECT_DIR$/.idea/vcs.xml" afterDir="false" />
<change afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
<change afterPath="$PROJECT_DIR$/pom.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/runConfigurations/keycloak_metrics__package_.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/runConfigurations/keycloak_metrics__clean_package_.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/pom.xml" beforeDir="false" afterPath="$PROJECT_DIR$/pom.xml" afterDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/KeycloakMetricsServer.java" beforeDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/Metrics.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/Metrics.java" afterDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/MetricsEndpoint.java" beforeDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/MetricsEndpointFactory.java" beforeDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListener.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListenerProvider.java" afterDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListenerFactory.java" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListenerProviderFactory.java" afterDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/resources/META-INF/services/org.keycloak.events.EventListenerProviderFactory" beforeDir="false" afterPath="$PROJECT_DIR$/src/main/resources/META-INF/services/org.keycloak.events.EventListenerProviderFactory" afterDir="false" />
<change beforePath="$PROJECT_DIR$/src/main/resources/META-INF/services/org.keycloak.services.resource.RealmResourceProviderFactory" beforeDir="false" />
</list>
<option name="SHOW_DIALOG" value="false" />
<option name="HIGHLIGHT_CONFLICTS" value="true" />
@ -22,12 +26,18 @@
<option name="RECENT_TEMPLATES">
<list>
<option value="Class" />
<option value="Interface" />
</list>
</option>
</component>
<component name="Git.Settings">
<option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
</component>
<component name="HighlightingSettingsPerFile">
<setting file="jar://$MAVEN_REPOSITORY$/org/keycloak/keycloak-server-spi-private/24.0.0/keycloak-server-spi-private-24.0.0-sources.jar!/org/keycloak/events/admin/AdminEvent.java" root0="FORCE_HIGHLIGHTING" />
<setting file="jar://$MAVEN_REPOSITORY$/org/keycloak/keycloak-server-spi-private/24.0.0/keycloak-server-spi-private-24.0.0-sources.jar!/org/keycloak/events/admin/OperationType.java" root0="FORCE_HIGHLIGHTING" />
<setting file="jar://$MAVEN_REPOSITORY$/org/keycloak/keycloak-server-spi-private/24.0.0/keycloak-server-spi-private-24.0.0-sources.jar!/org/keycloak/events/admin/ResourceType.java" root0="FORCE_HIGHLIGHTING" />
</component>
<component name="MavenImportPreferences">
<option name="importingSettings">
<MavenImportingSettings>
@ -52,6 +62,7 @@
<component name="PropertiesComponent"><![CDATA[{
"keyToString": {
"Downloaded.Files.Path.Enabled": "false",
"Maven.keycloak-metrics [clean,package].executor": "Run",
"Maven.keycloak-metrics [compile].executor": "Run",
"Maven.keycloak-metrics [package].executor": "Run",
"Maven.keycloak-metrics.executor": "Run",
@ -65,7 +76,7 @@
"project.structure.last.edited": "Libraries",
"project.structure.proportion": "0.15",
"project.structure.side.proportion": "0.3908046",
"settings.editor.selected.configurable": "preferences.lookFeel"
"settings.editor.selected.configurable": "reference.settings.project.maven.repository.indices"
}
}]]></component>
<component name="RecentsManager">

5
pom.xml
View File

@ -48,6 +48,11 @@
<artifactId>simpleclient_common</artifactId>
<version>0.16.0</version>
</dependency>
<dependency>
<groupId>io.prometheus</groupId>
<artifactId>simpleclient_httpserver</artifactId>
<version>0.16.0</version>
</dependency>
</dependencies>
<build>

2
src/main/java/KeycloakMetricsServer.java
View File

@ -1,2 +0,0 @@
public class KeycloakMetricsServer {
}

4
src/main/java/coffee/_finally/keycloak_metrics/Metrics.java
View File

@ -52,6 +52,10 @@ public class Metrics {
return METRICS_INSTANCE;
}
public CollectorRegistry getRegistry() {
return registry;
}
public static void increment(String metric) {
Collector collector = metrics.get(metric);
if (collector == null) {

28
src/main/java/coffee/_finally/keycloak_metrics/MetricsEndpoint.java
View File

@ -1,28 +0,0 @@
package coffee._finally.keycloak_metrics;
import jakarta.ws.rs.GET;
import jakarta.ws.rs.Produces;
import jakarta.ws.rs.core.*;
import org.keycloak.services.resource.RealmResourceProvider;
public class MetricsEndpoint implements RealmResourceProvider {
public static final String ID = "metrics";
@Override
public Object getResource() {
return this;
}
@Override
public void close() {
}
@GET
@Produces(MediaType.TEXT_PLAIN)
public Response get(@Context HttpHeaders headers) {
final StreamingOutput stream = outputStream -> Metrics.getInstance().export(outputStream);
return Response.ok(stream).build();
}
}

34
src/main/java/coffee/_finally/keycloak_metrics/MetricsEndpointFactory.java
View File

@ -1,34 +0,0 @@
package coffee._finally.keycloak_metrics;
import org.keycloak.Config;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.KeycloakSessionFactory;
import org.keycloak.services.resource.RealmResourceProvider;
import org.keycloak.services.resource.RealmResourceProviderFactory;
public class MetricsEndpointFactory implements RealmResourceProviderFactory {
@Override
public RealmResourceProvider create(KeycloakSession keycloakSession) {
return new MetricsEndpoint();
}
@Override
public void init(Config.Scope scope) {
// empty on purpose
}
@Override
public void postInit(KeycloakSessionFactory keycloakSessionFactory) {
// empty on purpose
}
@Override
public void close() {
// empty on purpose as nothing needs to be closed manually
}
@Override
public String getId() {
return MetricsEndpoint.ID;
}
}

2
src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListener.java → src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListenerProvider.java
View File

@ -4,7 +4,7 @@ import org.keycloak.events.Event;
import org.keycloak.events.EventListenerProvider;
import org.keycloak.events.admin.AdminEvent;
public class MetricsEventListener implements EventListenerProvider {
public class MetricsEventListenerProvider implements EventListenerProvider {
public static final String ID = "metrics-listener";

6
src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListenerFactory.java → src/main/java/coffee/_finally/keycloak_metrics/MetricsEventListenerProviderFactory.java
View File

@ -6,10 +6,10 @@ import org.keycloak.events.EventListenerProviderFactory;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.KeycloakSessionFactory;
public class MetricsEventListenerFactory implements EventListenerProviderFactory {
public class MetricsEventListenerProviderFactory implements EventListenerProviderFactory {
@Override
public EventListenerProvider create(KeycloakSession keycloakSession) {
return new MetricsEventListener();
return new MetricsEventListenerProvider();
}
@Override
@ -29,6 +29,6 @@ public class MetricsEventListenerFactory implements EventListenerProviderFactory
@Override
public String getId() {
return MetricsEventListener.ID;
return MetricsEventListenerProvider.ID;
}
}

8
src/main/java/coffee/_finally/keycloak_metrics/MetricsServerProvider.java Normal file
View File

@ -0,0 +1,8 @@
package coffee._finally.keycloak_metrics;
public class MetricsServerProvider implements ServerProvider {
@Override
public void close() {
}
}

64
src/main/java/coffee/_finally/keycloak_metrics/MetricsServerProviderFactory.java Normal file
View File

@ -0,0 +1,64 @@
package coffee._finally.keycloak_metrics;
import io.prometheus.client.exporter.HTTPServer;
import org.keycloak.Config;
import org.keycloak.models.KeycloakSession;
import org.keycloak.models.KeycloakSessionFactory;
import org.keycloak.provider.Provider;
import org.keycloak.provider.ProviderFactory;
import java.io.IOException;
public class MetricsServerProviderFactory implements ServerProviderFactory {
private static HTTPServer metricsServer = null;
@Override
public Provider create(KeycloakSession session) {
return new MetricsServerProvider();
}
/**
* Only called once when the factory is first created. This config is pulled from keycloak_server.json
*
* @param config
*/
@Override
public void init(Config.Scope config) {
// intentionally left blank
}
/**
* Called after all provider factories have been initialized
*
* @param factory
*/
@Override
public void postInit(KeycloakSessionFactory factory) {
if (metricsServer == null) {
try {
metricsServer = new HTTPServer.Builder()
.withPort(9400)
.withRegistry(Metrics.getInstance().getRegistry())
.build();
} catch (IOException io) {
System.err.println("Unable to start HTTP server for metrics on port 9400: " + io.toString());
}
}
}
/**
* This is called when the server shuts down.
*/
@Override
public void close() {
if (metricsServer != null) {
metricsServer.close();
}
}
@Override
public String getId() {
return "metrics-server-provider-spi";
}
}

6
src/main/java/coffee/_finally/keycloak_metrics/ServerProvider.java Normal file
View File

@ -0,0 +1,6 @@
package coffee._finally.keycloak_metrics;
import org.keycloak.provider.Provider;
public interface ServerProvider extends Provider {
}

6
src/main/java/coffee/_finally/keycloak_metrics/ServerProviderFactory.java Normal file
View File

@ -0,0 +1,6 @@
package coffee._finally.keycloak_metrics;
import org.keycloak.provider.ProviderFactory;
public interface ServerProviderFactory extends ProviderFactory {
}

27
src/main/java/coffee/_finally/keycloak_metrics/ServerSpi.java Normal file
View File

@ -0,0 +1,27 @@
package coffee._finally.keycloak_metrics;
import org.keycloak.provider.Provider;
import org.keycloak.provider.ProviderFactory;
import org.keycloak.provider.Spi;
public class ServerSpi implements Spi {
@Override
public boolean isInternal() {
return false;
}
@Override
public String getName() {
return "metrics-server";
}
@Override
public Class<? extends Provider> getProviderClass() {
return ServerProvider.class;
}
@Override
public Class<? extends ProviderFactory> getProviderFactoryClass() {
return ServerProviderFactory.class;
}
}

3
src/main/resources/META-INF/services/MANIFEST.MF Normal file
View File

@ -0,0 +1,3 @@
Manifest-Version: 1.0
Dependencies: org.keycloak.keycloak-services
Sealed: true

1
src/main/resources/META-INF/services/coffee._finally.keycloak_metrics.ServerSpi Normal file
View File

@ -0,0 +1 @@
coffee._finally.keycloak_metrics.MetricsServerProviderFactory

2
src/main/resources/META-INF/services/org.keycloak.events.EventListenerProviderFactory
View File

@ -1 +1 @@
coffee._finally.keycloak_metrics.MetricsEventListenerFactory
coffee._finally.keycloak_metrics.MetricsEventListenerProviderFactory

1
src/main/resources/META-INF/services/org.keycloak.provider.Spi Normal file
View File

@ -0,0 +1 @@
coffee._finally.keycloak_metrics.ServerSpi

1
src/main/resources/META-INF/services/org.keycloak.services.resource.RealmResourceProviderFactory
View File

@ -1 +0,0 @@
coffee._finally.keycloak_metrics.MetricsEndpointFactory