From 9b40dcaf523aff25d8c379800a7ff99efcadc5a2 Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Tue, 4 Jul 2023 16:57:42 +0800
Subject: [PATCH] Throttle the request interface

---
 server/httphandler/agent.go | 13 ++++++-------
 server/main.go              |  4 ++--
 server/router/router.go     |  7 ++-----
 3 files changed, 10 insertions(+), 14 deletions(-)

diff --git a/server/httphandler/agent.go b/server/httphandler/agent.go
index bbccade..b8eeb5c 100644
--- a/server/httphandler/agent.go
+++ b/server/httphandler/agent.go
@@ -10,14 +10,13 @@ import (
 // 安装运行
 func InstallRedisExporter(c *gin.Context) {
 	// TODOs
-	var param *common.Batch
-
-	if err := c.BindJSON(param); err != nil {
+	var param common.Batch
+	if err := c.BindJSON(&param); err != nil {
 		response.Fail(c, nil, err.Error())
 		return
 	}
 
-	ret, err := service.Install(param)
+	ret, err := service.Install(&param)
 	if err != nil {
 		response.Fail(c, nil, err.Error())
 		return
@@ -26,12 +25,12 @@ func InstallRedisExporter(c *gin.Context) {
 }
 
 func UnInstallRedisExporter(c *gin.Context) {
-	var param *common.Batch
-	if err := c.BindJSON(param); err != nil {
+	var param common.Batch
+	if err := c.BindJSON(&param); err != nil {
 		response.Fail(c, nil, err.Error())
 		return
 	}
-	ret, err := service.UnInstall(param)
+	ret, err := service.UnInstall(&param)
 	if err != nil {
 		response.Fail(c, nil, err.Error())
 		return
diff --git a/server/main.go b/server/main.go
index 4a757e5..cb5f30e 100644
--- a/server/main.go
+++ b/server/main.go
@@ -35,10 +35,10 @@ func main() {
 
 	server := router.InitRouter()
 	global.GlobalClient = client.DefaultClient(plugin.Init(config.Config().Redis))
-	router.RegisterAPIs(server)
 	global.GlobalClient.Server = config.Config().Http.Addr
+	router.RegisterAPIs(server)
 
-	if err := server.Run(config.Config().Http.Addr); err != nil {
+	if err := server.Run(global.GlobalClient.Server); err != nil {
 		logger.Fatal("failed to run server")
 	}
 }
diff --git a/server/router/router.go b/server/router/router.go
index bfb8acb..18db944 100644
--- a/server/router/router.go
+++ b/server/router/router.go
@@ -21,11 +21,8 @@ func RegisterAPIs(router *gin.Engine) {
 
 	pg := router.Group("/plugin/" + global.GlobalClient.PluginInfo.Name)
 	{
-		pg.GET("/query", func(c *gin.Context) {
-			c.Set("query", global.GlobalClient.PluginInfo.ReverseDest)
-			httphandler.InstallRedisExporter(c)
-		})
-
+		pg.POST("/install", httphandler.InstallRedisExporter)
+		pg.POST("/remove", httphandler.UnInstallRedisExporter)
 	}
 
 }
-- 
Gitee