From 00c23de1ca3538128e7a59a665685e01f7118c2f Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Tue, 11 Jun 2024 15:27:33 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E8=AE=BE=E5=A4=87=E9=87=8D?=
 =?UTF-8?q?=E5=90=AF=E8=AF=B7=E6=B1=82id=E8=8E=B7=E5=8F=96=E4=B8=8D?=
 =?UTF-8?q?=E5=88=B0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 handler/device.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/handler/device.go b/handler/device.go
index 94021a8..3051fe2 100644
--- a/handler/device.go
+++ b/handler/device.go
@@ -68,7 +68,6 @@ func UpdateDevice(c *gin.Context) {
 	var req DeviceUpdateReq
 	user_id, _ := c.Get("id")
 	if err := c.ShouldBindJSON(&req); err == nil {
-		fmt.Println("req:", req)
 		res := service.UpdateDevice(req.DeviceName, req.DeviceIP, req.DeviceStatus, req.DeviceInfo, req.DeviceType, req.DeviceLocation, req.ID, int(user_id.(float64)))
 		if res {
 			c.JSON(200, gin.H{
@@ -137,13 +136,18 @@ func GetDeviceList(c *gin.Context) {
 	c.JSON(200, gin.H{
 		"code":    0,
 		"message": "success",
-		"data": devices,
+		"data":    devices,
 	})
 }
 
 func RestartDevice(c *gin.Context) {
 	user_id, _ := c.Get("id")
-	device_id := c.GetInt("device_id")
+	var req DeviceDelReq
+	if err := c.ShouldBind(&req); err != nil {
+		c.JSON(200, gin.H{"code": 1, "message": "failed"})
+		return
+	}
+	device_id := req.ID
 	device := service.GetDevice(device_id, int(user_id.(float64)))
 	if device.ID != 0 {
 		if device.DeviceIP != "" {