增加websocket支持

internal/core/application.go

@@ -9,6 +9,7 @@ import (
 	"git.huangwc.com/pig/pig-farm-controller/internal/api"
 	"git.huangwc.com/pig/pig-farm-controller/internal/config"
 	"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/db"
 	"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
 	"git.huangwc.com/pig/pig-farm-controller/internal/task"
@@ -38,6 +39,9 @@ type Application struct {
 	// DeviceRepo 设备仓库实例
 	DeviceRepo repository.DeviceRepo
 
+	// WebSocketService WebSocket服务实例
+	WebSocketService *service.WebSocketService
+
 	// Config 应用配置
 	Config *config.Config
 
@@ -71,8 +75,11 @@ func NewApplication(cfg *config.Config) *Application {
 	// 初始化设备仓库
 	deviceRepo := repository.NewDeviceRepo(store.GetDB())
 
+	// 初始化WebSocket服务
+	websocketService := service.NewWebSocketService()
+
 	// 初始化API组件
-	apiInstance := api.NewAPI(cfg, userRepo, operationHistoryRepo, deviceControlRepo, deviceRepo)
+	apiInstance := api.NewAPI(cfg, userRepo, operationHistoryRepo, deviceControlRepo, deviceRepo, websocketService)
 
 	// 初始化任务执行器组件(使用5个工作协程)
 	taskExecutor := task.NewExecutor(5)
@@ -85,6 +92,7 @@ func NewApplication(cfg *config.Config) *Application {
 		OperationHistoryRepo: operationHistoryRepo,
 		DeviceControlRepo:    deviceControlRepo,
 		DeviceRepo:           deviceRepo,
+		WebSocketService:     websocketService,
 		Config:               cfg,
 		logger:               logs.NewLogger(),
 	}