From f96286cde8d273ab6cfed6f0a6196ebf79dccded Mon Sep 17 00:00:00 2001
From: giaogiao <giaogiao>
Date: Fri, 15 Sep 2023 10:57:29 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E7=BC=96=E8=BE=91=E7=9A=84?=
 =?UTF-8?q?=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md            |  1 +
 controller/ws.go     | 12 ++++++++++++
 controller/wsmain.go | 12 ++++++++++++
 database/database.go | 24 +++++++++++++++++++-----
 main_test.go         | 27 +++++++++++++++++++++++++++
 5 files changed, 71 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index e69de29..69ef7b5 100644
--- a/README.md
+++ b/README.md
@@ -0,0 +1 @@
+[![Build Status](https://drone.shagain.club/api/badges/lingling/go_echo_wol/status.svg)](https://drone.shagain.club/lingling/go_echo_wol)
\ No newline at end of file
diff --git a/controller/ws.go b/controller/ws.go
index 571a4b3..440e75c 100644
--- a/controller/ws.go
+++ b/controller/ws.go
@@ -71,3 +71,15 @@ func Add(ws **websocket.Conn, uuid string, name string, mac string) {
 		return
 	}
 }
+
+func Edi(ws **websocket.Conn, uuid string, id int64, name string, mac string) {
+	wol := new(database.Wol)
+	wol.Mac = mac
+	wol.Name = name
+	database.Edi(wol, id)
+	err := websocket.Message.Send(*ws, res.Get_res_string(200, "", uuid, ""))
+	if err != nil {
+		// c.Logger().Error(err)
+		return
+	}
+}
diff --git a/controller/wsmain.go b/controller/wsmain.go
index 05f3cec..36215b1 100644
--- a/controller/wsmain.go
+++ b/controller/wsmain.go
@@ -18,6 +18,11 @@ type AddData struct {
 	Nmae string `json:"name"`
 	Mac  string `json:"mac"`
 }
+type EdiData struct {
+	ID   int64  `json:"id"`
+	Nmae string `json:"name"`
+	Mac  string `json:"mac"`
+}
 type TooLTT struct {
 	Method string      `json:"method"`
 	Data   interface{} `json:"data"`
@@ -72,6 +77,13 @@ func Hello(c echo.Context) error {
 					return
 				}
 				Add(&ws, wsValue.Uuid, AddData.Nmae, AddData.Mac)
+			case "edi":
+				var EdiData EdiData
+				if err := json.Unmarshal(json_data, &EdiData); err != nil {
+					c.Logger().Error(err.Error())
+					return
+				}
+				Edi(&ws, wsValue.Uuid, EdiData.ID, EdiData.Nmae, EdiData.Mac)
 			default:
 				c.Logger().Error("没有找到方法")
 
diff --git a/database/database.go b/database/database.go
index 4041903..494ae08 100644
--- a/database/database.go
+++ b/database/database.go
@@ -54,17 +54,31 @@ func Getallwoltext() {
 	println(len(everyone))
 }
 
+// 返回插入的数量
 func Insert(wol *Wol) {
-	affected, err := engine.Insert(wol)
+	_, err := engine.Insert(wol)
 	if err != nil {
 		println(err.Error())
 	}
-	println(affected)
+
 }
 func Del(id int64) {
-	affected, err := engine.Where("i_d = ?", id).Delete(&Wol{})
+	_, err := engine.Where("i_d = ?", id).Delete(&Wol{})
 	if err != nil {
-		println(err)
+		println(err.Error())
+	}
+}
+func Edi(wol *Wol, id int64) {
+	_, err := engine.Where("i_d = ?", id).Update(wol)
+	if err != nil {
+		println(err.Error())
+	}
+}
+
+// 根据id查找
+func GetWolById(id int64, wol *Wol) {
+	_, err := engine.Where("i_d = ?", id).Get(wol)
+	if err != nil {
+		println(err.Error())
 	}
-	println(affected, "ok")
 }
diff --git a/main_test.go b/main_test.go
index efc73af..bad2960 100644
--- a/main_test.go
+++ b/main_test.go
@@ -28,3 +28,30 @@ func TestAdd_Delwol(t *testing.T) {
 	}
 	t.Errorf("并没有找到")
 }
+
+func TestEdi_wol(t *testing.T) {
+	wol := new(database.Wol)
+	wol.Mac = "8F:26:58:C1:85:83"
+	wol.Name = "text1"
+	database.Insert(wol)
+	wolarray := make([]database.Wol, 0)
+	database.Getallwol(&wolarray)
+	var idReturnedOnInsertion int64 = -1
+	for _, item := range wolarray {
+		if item.Mac == wol.Mac && item.Name == wol.Name {
+			idReturnedOnInsertion = item.ID
+			break
+		}
+	}
+	wol2 := new(database.Wol)
+	wol2.Name = "testchang"
+	wol2.Mac = "97:88:90:99:8D:93"
+	database.Edi(wol2, idReturnedOnInsertion)
+	changwoled := new(database.Wol)
+	database.GetWolById(idReturnedOnInsertion, changwoled)
+	if changwoled.Mac == wol2.Mac && changwoled.Name == wol2.Name {
+		database.Del(idReturnedOnInsertion)
+		return
+	}
+	t.Errorf("并没有找到,或者失败")
+}