[INLONG-1712][Feature][Agent] Add prometheus enable

issue: apache#1712
shink · Dec 8, 2021 · 7d3628b · 7d3628b
1 parent c279da4
commit 7d3628b
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 5 deletions.
diff --git a/...ng-agent/agent-common/src/main/java/org/apache/inlong/agent/constants/AgentConstants.java b/...ng-agent/agent-common/src/main/java/org/apache/inlong/agent/constants/AgentConstants.java
@@ -160,6 +160,9 @@ public class AgentConstants {
 
     public static final String AGENT_LOCAL_IP = "agent.local.ip";
 
+    public static final String PROMETHEUS_ENABLE = "agent.prometheus.enable";
+    public static final boolean DEFAULT_PROMETHEUS_ENABLE = false;
+
     public static final String PROMETHEUS_EXPORTER_PORT = "agent.prometheus.exporter.port";
     public static final int DEFAULT_PROMETHEUS_EXPORTER_PORT = 8080;
 

diff --git a/inlong-agent/agent-core/src/main/java/org/apache/inlong/agent/core/AgentMain.java b/inlong-agent/agent-core/src/main/java/org/apache/inlong/agent/core/AgentMain.java
@@ -31,7 +31,9 @@
 
 import java.util.Iterator;
 
+import static org.apache.inlong.agent.constants.AgentConstants.DEFAULT_PROMETHEUS_ENABLE;
 import static org.apache.inlong.agent.constants.AgentConstants.DEFAULT_PROMETHEUS_EXPORTER_PORT;
+import static org.apache.inlong.agent.constants.AgentConstants.PROMETHEUS_ENABLE;
 import static org.apache.inlong.agent.constants.AgentConstants.PROMETHEUS_EXPORTER_PORT;
 
 /**
@@ -125,11 +127,15 @@ public static void main(String[] args) throws Exception {
             manager.start();
             stopManagerIfKilled(manager);
 
-            // starting metrics server
-            int metricsServerPort = AgentConfiguration.getAgentConf()
-                    .getInt(PROMETHEUS_EXPORTER_PORT, DEFAULT_PROMETHEUS_EXPORTER_PORT);
-            LOGGER.info("Starting prometheus metrics server on port {}", metricsServerPort);
-            metricsServer = new HTTPServer(metricsServerPort);
+            boolean enablePrometheus = AgentConfiguration.getAgentConf()
+                    .getBoolean(PROMETHEUS_ENABLE, DEFAULT_PROMETHEUS_ENABLE);
+            if (enablePrometheus) {
+                // starting metrics server
+                int metricsServerPort = AgentConfiguration.getAgentConf()
+                        .getInt(PROMETHEUS_EXPORTER_PORT, DEFAULT_PROMETHEUS_EXPORTER_PORT);
+                LOGGER.info("Starting prometheus metrics server on port {}", metricsServerPort);
+                metricsServer = new HTTPServer(metricsServerPort);
+            }
 
             manager.join();
         } catch (Exception ex) {

diff --git a/inlong-agent/conf/agent.properties b/inlong-agent/conf/agent.properties
@@ -106,5 +106,7 @@ agent.manager.vip.http.port=
 ############################
 # prometheus config
 ############################
+# whether to enable prometheus
+agent.prometheus.enable=false
 # prometheus exporter server default port
 agent.prometheus.exporter.port=8080