增加websocket支持

internal/controller/device/device.go

@@ -7,6 +7,7 @@ import (
 	"git.huangwc.com/pig/pig-farm-controller/internal/controller"
 	"git.huangwc.com/pig/pig-farm-controller/internal/logs"
 	"git.huangwc.com/pig/pig-farm-controller/internal/model"
+	"git.huangwc.com/pig/pig-farm-controller/internal/service"
 	"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
 	"github.com/gin-gonic/gin"
 )
@@ -15,14 +16,16 @@ import (
 type Controller struct {
 	deviceControlRepo repository.DeviceControlRepo
 	deviceRepo        repository.DeviceRepo
+	websocketService  *service.WebSocketService
 	logger            *logs.Logger
 }
 
 // NewController 创建设备控制控制器实例
-func NewController(deviceControlRepo repository.DeviceControlRepo, deviceRepo repository.DeviceRepo) *Controller {
+func NewController(deviceControlRepo repository.DeviceControlRepo, deviceRepo repository.DeviceRepo, websocketService *service.WebSocketService) *Controller {
 	return &Controller{
 		deviceControlRepo: deviceControlRepo,
 		deviceRepo:        deviceRepo,
+		websocketService:  websocketService,
 		logger:            logs.NewLogger(),
 	}
 }
@@ -63,8 +66,20 @@ func (c *Controller) Switch(ctx *gin.Context) {
 		return
 	}
 
-	// TODO: 实际的设备控制逻辑
-	// 这里暂时用TODO代替具体逻辑
+	// 通过WebSocket向中继设备发送控制指令
+	// 这里假设中继设备ID为"relay-001"，在实际应用中应该根据设备层级结构动态获取
+	controlData := map[string]interface{}{
+		"device_type": req.DeviceType,
+		"device_id":   req.DeviceID,
+		"action":      req.Action,
+	}
+
+	err := c.websocketService.SendCommand("relay-001", "control_device", controlData)
+	if err != nil {
+		c.logger.Error("通过WebSocket发送设备控制指令失败: " + err.Error())
+		controller.SendErrorResponse(ctx, controller.InternalServerErrorCode, "设备控制失败")
+		return
+	}
 
 	// 创建设备控制记录
 	if err := c.createDeviceControlRecord(