增加websocket支持

internal/api/api.go

@@ -13,9 +13,12 @@ import (
 	"git.huangwc.com/pig/pig-farm-controller/internal/config"
 	"git.huangwc.com/pig/pig-farm-controller/internal/controller/device"
 	"git.huangwc.com/pig/pig-farm-controller/internal/controller/operation"
+	"git.huangwc.com/pig/pig-farm-controller/internal/controller/remote"
 	"git.huangwc.com/pig/pig-farm-controller/internal/controller/user"
 	"git.huangwc.com/pig/pig-farm-controller/internal/logs"
+	"git.huangwc.com/pig/pig-farm-controller/internal/service"
 	"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
+	"git.huangwc.com/pig/pig-farm-controller/internal/websocket"
 	"github.com/gin-gonic/gin"
 )
 
@@ -40,16 +43,25 @@ type API struct {
 	// deviceController 设备控制控制器
 	deviceController *device.Controller
 
+	// remoteController 远程控制控制器
+	remoteController *remote.Controller
+
 	// authMiddleware 鉴权中间件
 	authMiddleware *middleware.AuthMiddleware
 
+	// websocketManager WebSocket管理器
+	websocketManager *websocket.Manager
+
+	// websocketService WebSocket服务
+	websocketService *service.WebSocketService
+
 	// logger 日志记录器
 	logger *logs.Logger
 }
 
 // NewAPI 创建并返回一个新的API实例
 // 初始化Gin引擎和相关配置
-func NewAPI(cfg *config.Config, userRepo repository.UserRepo, operationHistoryRepo repository.OperationHistoryRepo, deviceControlRepo repository.DeviceControlRepo, deviceRepo repository.DeviceRepo) *API {
+func NewAPI(cfg *config.Config, userRepo repository.UserRepo, operationHistoryRepo repository.OperationHistoryRepo, deviceControlRepo repository.DeviceControlRepo, deviceRepo repository.DeviceRepo, websocketService *service.WebSocketService) *API {
 	// 设置Gin为发布模式
 	gin.SetMode(gin.ReleaseMode)
 
@@ -80,7 +92,13 @@ func NewAPI(cfg *config.Config, userRepo repository.UserRepo, operationHistoryRe
 	operationController := operation.NewController(operationHistoryRepo)
 
 	// 创建设备控制控制器
-	deviceController := device.NewController(deviceControlRepo, deviceRepo)
+	deviceController := device.NewController(deviceControlRepo, deviceRepo, websocketService)
+
+	// 创建WebSocket管理器
+	websocketManager := websocket.NewManager(websocketService)
+
+	// 创建远程控制控制器
+	remoteController := remote.NewController(websocketService)
 
 	// 创建鉴权中间件
 	authMiddleware := middleware.NewAuthMiddleware(userRepo)
@@ -91,7 +109,10 @@ func NewAPI(cfg *config.Config, userRepo repository.UserRepo, operationHistoryRe
 		userController:      userController,
 		operationController: operationController,
 		deviceController:    deviceController,
+		remoteController:    remoteController,
 		authMiddleware:      authMiddleware,
+		websocketManager:    websocketManager,
+		websocketService:    websocketService,
 		logger:              logs.NewLogger(),
 	}
 }
@@ -146,6 +167,9 @@ func (a *API) setupRoutes() {
 	// 基础路由示例
 	a.engine.GET("/health", a.healthHandler)
 
+	// WebSocket路由
+	a.engine.GET("/ws/device", a.websocketManager.HandleConnection)
+
 	// 用户相关路由
 	userGroup := a.engine.Group("/api/v1/user")
 	{
@@ -170,6 +194,13 @@ func (a *API) setupRoutes() {
 		{
 			deviceGroup.POST("/switch", a.deviceController.Switch)
 		}
+
+		// 远程控制相关路由
+		remoteGroup := protectedGroup.Group("/remote")
+		{
+			remoteGroup.POST("/command", a.remoteController.SendCommand)
+			remoteGroup.GET("/devices", a.remoteController.ListConnectedDevices)
+		}
 	}
 
 	// TODO: 添加更多路由