From 8d52ad75342297a2e68f99f4e1f2ad8b6a28178f Mon Sep 17 00:00:00 2001
From: wubijie <wubijie@kylinos.cn>
Date: Fri, 30 Jun 2023 14:51:06 +0800
Subject: [PATCH] Add database-related operations

---
 server/config.yml.templete | 10 +++++-----
 server/db/redisdao.go      | 31 +++++++++++++++++++++++++++++++
 server/service/agent.go    |  7 -------
 3 files changed, 36 insertions(+), 12 deletions(-)
 create mode 100644 server/db/redisdao.go

diff --git a/server/config.yml.templete b/server/config.yml.templete
index 3c0a8be..32a7606 100644
--- a/server/config.yml.templete
+++ b/server/config.yml.templete
@@ -10,8 +10,8 @@ log:
   max_file: 1
   max_size: 10485760
 mysql:
-  host:localhost
-  port:3306
-  user:''
-  password:''
-  database:PluginRedis
\ No newline at end of file
+  host: localhost
+  port: 3306
+  user: ''
+  password: ''
+  database: PluginRedis
\ No newline at end of file
diff --git a/server/db/redisdao.go b/server/db/redisdao.go
new file mode 100644
index 0000000..563bc98
--- /dev/null
+++ b/server/db/redisdao.go
@@ -0,0 +1,31 @@
+package db
+
+import (
+	"fmt"
+	"time"
+
+	"openeuler.org/PilotGo/redis-plugin/global"
+)
+
+type RedisExportTarget struct {
+	ID        uint   `gorm:"primary_key;AUTO_INCREMENT" json:"id"`
+	UUID      string `json:"uuid"`
+	Status    string `json:"status"` //install or remove
+	UpdatedAt time.Time
+}
+
+func AddRedisExporter(ret RedisExportTarget) error {
+	if len(ret.UUID) == 0 {
+		return fmt.Errorf("机器不能为空")
+	}
+	return global.GlobalDB.Save(&ret).Error
+}
+
+func GetRedisExporter() ([]RedisExportTarget, error) {
+	var ret []RedisExportTarget
+	err := global.GlobalDB.Where("status=?", "install").Find(&ret).Error
+	if err != nil {
+		return nil, err
+	}
+	return ret, nil
+}
diff --git a/server/service/agent.go b/server/service/agent.go
index c0c21c4..8af29c2 100644
--- a/server/service/agent.go
+++ b/server/service/agent.go
@@ -8,13 +8,6 @@ import (
 	"openeuler.org/PilotGo/redis-plugin/plugin"
 )
 
-type RedisExportTarget struct {
-	ID        uint   `gorm:"primary_key;AUTO_INCREMENT" json:"id"`
-	UUID      string `json:"uuid"`
-	Status    string `json:"status"` //install or remove
-	UpdatedAt time.Time
-}
-
 func FormatData(cmdResults []*client.CmdResult) ([]string, []interface{}, error) {
 	ret := []interface{}{}
 	monitorTargets := []string{}
-- 
Gitee