handler/tool.go

@@ -47,12 +47,12 @@ func SetDeviceStatusV2(c *gin.Context) {
 		c.JSON(200, gin.H{"code": 400, "message": "参数错误"})
 		return
 	} else {
-		token := c.Request.Header.Get("token")
-		if token == "" {
+		token, _ := c.Get("token")
+		if token == nil {
 			c.JSON(200, gin.H{"code": 401, "message": "token为空"})
 			return
 		}
-		devices := worker.GetRedisSetMembers(token)
+		devices := worker.GetRedisSetMembers(token.(string))
 		if len(devices) == 0 {
 			c.JSON(200, gin.H{"code": 402, "message": "设备为空"})
 			return