add update hosts

apache · Dec 16, 2024 · c297683 · c297683
1 parent b6bae5a
commit c297683
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/...c/main/java/org/apache/bigtop/manager/stack/infra/v1_0_0/prometheus/PrometheusParams.java b/...c/main/java/org/apache/bigtop/manager/stack/infra/v1_0_0/prometheus/PrometheusParams.java
@@ -41,9 +41,9 @@
 @NoArgsConstructor
 public class PrometheusParams extends InfraParams {
 
-    private final String PROMETHEUS_SELF_JOB_NAME = "prometheus";
-    private final String BM_AGENT_JOB_NAME = "bm-agent";
-    private final String BM_AGENT_PORT = "8081";
+    protected final String PROMETHEUS_SELF_JOB_NAME = "prometheus";
+    protected final String BM_AGENT_JOB_NAME = "bm-agent";
+    protected final String BM_AGENT_PORT = "8081";
 
     private Map<String, Object> prometheusScrapeJob;
     private Map<String, Object> agentScrapeJob;
@@ -78,7 +78,7 @@ public String getServiceName() {
         return "prometheus";
     }
 
-    private List<String> getAllHost() {
+    protected List<String> getAllHost() {
         List<String> ips = LocalSettings.hosts().get("all");
         List<String> hosts = new ArrayList<>();
         for (String ip : ips) {

diff --git a/...rc/main/java/org/apache/bigtop/manager/stack/infra/v1_0_0/prometheus/PrometheusSetup.java b/...rc/main/java/org/apache/bigtop/manager/stack/infra/v1_0_0/prometheus/PrometheusSetup.java
@@ -67,4 +67,18 @@ public static ShellResult config(Params params) {
         }
         return ShellResult.success("Prometheus Configure success!");
     }
+
+    public static ShellResult updateHosts(Params params) {
+        PrometheusParams prometheusParams = (PrometheusParams) params;
+        String user = prometheusParams.user();
+        String group = prometheusParams.group();
+        LinuxFileUtils.toFile(
+                ConfigType.JSON,
+                prometheusParams.targetsConfigFile(prometheusParams.PROMETHEUS_SELF_JOB_NAME),
+                user,
+                group,
+                Constants.PERMISSION_644,
+                prometheusParams.getAllHost());
+        return ShellResult.success("Update hosts success!");
+    }
 }