Add Airlift JMX modules

trinodb · Jan 11, 2025 · 17ccb2f · 17ccb2f
1 parent 63471d4
commit 17ccb2f
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 7 deletions.
diff --git a/trino-aws-proxy/pom.xml b/trino-aws-proxy/pom.xml
@@ -84,6 +84,16 @@
             <artifactId>jaxrs</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>io.airlift</groupId>
+            <artifactId>jmx</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>io.airlift</groupId>
+            <artifactId>jmx-http</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>io.airlift</groupId>
             <artifactId>json</artifactId>
@@ -94,11 +104,21 @@
             <artifactId>log</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>io.airlift</groupId>
+            <artifactId>log-manager</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>io.airlift</groupId>
             <artifactId>node</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>io.airlift</groupId>
+            <artifactId>openmetrics</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>io.airlift</groupId>
             <artifactId>units</artifactId>
@@ -129,6 +149,11 @@
             <artifactId>jersey-server</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>org.weakref</groupId>
+            <artifactId>jmxutils</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>software.amazon.awssdk</groupId>
             <artifactId>annotations</artifactId>
@@ -174,12 +199,6 @@
             <artifactId>utils</artifactId>
         </dependency>
 
-        <dependency>
-            <groupId>io.airlift</groupId>
-            <artifactId>log-manager</artifactId>
-            <scope>runtime</scope>
-        </dependency>
-
         <dependency>
             <groupId>jakarta.servlet</groupId>
             <artifactId>jakarta.servlet-api</artifactId>

diff --git a/trino-aws-proxy/src/main/java/io/trino/aws/proxy/server/TrinoAwsProxyServer.java b/trino-aws-proxy/src/main/java/io/trino/aws/proxy/server/TrinoAwsProxyServer.java
@@ -18,9 +18,14 @@
 import io.airlift.bootstrap.Bootstrap;
 import io.airlift.http.server.HttpServerModule;
 import io.airlift.jaxrs.JaxrsModule;
+import io.airlift.jmx.JmxHttpModule;
+import io.airlift.jmx.JmxModule;
 import io.airlift.json.JsonModule;
+import io.airlift.log.LogJmxModule;
 import io.airlift.log.Logger;
 import io.airlift.node.NodeModule;
+import io.airlift.openmetrics.JmxOpenMetricsModule;
+import org.weakref.jmx.guice.MBeanModule;
 
 public final class TrinoAwsProxyServer
 {
@@ -35,7 +40,12 @@ public static void main(String[] args)
                 .add(new NodeModule())
                 .add(new HttpServerModule())
                 .add(new JsonModule())
-                .add(new JaxrsModule());
+                .add(new JaxrsModule())
+                .add(new JmxModule())
+                .add(new JmxHttpModule())
+                .add(new JmxOpenMetricsModule())
+                .add(new LogJmxModule())
+                .add(new MBeanModule());
 
         Bootstrap app = new Bootstrap(modules.build());
         app.initialize();