From a56e948bec7ee3cc440f9230a74da6aa4d376bc4 Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Mon, 10 Jul 2023 11:21:06 +0800
Subject: [PATCH] Add the targets data interface required by the Peometheus
 configuration file

---
 server/httphandler/agent.go | 19 +++++++++++++++++++
 server/router/router.go     |  1 +
 server/service/agent.go     | 13 +++++++++++++
 3 files changed, 33 insertions(+)

diff --git a/server/httphandler/agent.go b/server/httphandler/agent.go
index 6776331..51cd4f3 100644
--- a/server/httphandler/agent.go
+++ b/server/httphandler/agent.go
@@ -1,6 +1,8 @@
 package httphandler
 
 import (
+	"net/http"
+
 	"gitee.com/openeuler/PilotGo-plugins/sdk/common"
 	"gitee.com/openeuler/PilotGo-plugins/sdk/response"
 	"github.com/gin-gonic/gin"
@@ -69,3 +71,20 @@ func StopRedisExporter(c *gin.Context) {
 }
 
 // 查询数据库安装情况
+func GetTargets(c *gin.Context) {
+	targets, err := service.GetRedisExporterIp()
+	if err != nil {
+		response.Fail(c, nil, err.Error())
+		return
+	}
+	objs := []RedisexporterObject{
+		{
+			Targets: targets,
+		},
+	}
+	c.JSON(http.StatusOK, objs)
+}
+
+type RedisexporterObject struct {
+	Targets []string `json:"targets"`
+}
diff --git a/server/router/router.go b/server/router/router.go
index fcd8bc0..27aa627 100644
--- a/server/router/router.go
+++ b/server/router/router.go
@@ -31,5 +31,6 @@ func RegisterAPIs(router *gin.Engine) {
 		pg.POST("/remove", httphandler.UnInstallRedisExporter)
 		pg.POST("/restart", httphandler.RestartRedisExporter)
 		pg.POST("/stop", httphandler.StopRedisExporter)
+		pg.GET("/targets", httphandler.GetTargets)
 	}
 }
diff --git a/server/service/agent.go b/server/service/agent.go
index 873cb92..e9d2410 100644
--- a/server/service/agent.go
+++ b/server/service/agent.go
@@ -91,3 +91,16 @@ func Stop(param *common.Batch) ([]db.RedisExportTarget, error) {
 	ret, err := FormatData(cmdResults)
 	return ret, err
 }
+
+func GetRedisExporterIp() ([]string, error) {
+	ret, err := db.GetRedisExporter()
+	if err != nil {
+		return nil, err
+	}
+	var targets []string
+	for _, tm := range ret {
+		target := tm.MachineIP + ":9121"
+		targets = append(targets, target)
+	}
+	return targets, nil
+}
-- 
Gitee