wanglei/EchoChat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(ws): WebSocket 核心模块（Hub + Client + PubSub + Handler）

Browse Source 
- pkg/ws/message.go: 统一消息协议（Message/Response/PushMessage）
- pkg/ws/hub.go: Hub 连接管理（注册/注销/按 userID 查找/在线计数）
- pkg/ws/client.go: 客户端连接封装（readPump/writePump/心跳 30s）
- pkg/ws/pubsub.go: Redis Pub/Sub 消息路由（按用户频道发布/订阅）
- app/ws/handler.go: WebSocket 升级处理（JWT 认证 + 消息分发）
- app/ws/router.go: GET /ws 路由注册
- app/ws/provider.go: Wire Provider Set
- 更新 provider/router 集成 WebSocket 模块

Made-with: Cursor
This commit is contained in:
bujinyuan
2026-03-02 16:46:17 +08:00
parent d392653446
commit bfb4fb0ccb
13 changed files with 582 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
backend/go-service/app/provider/wire_gen.go
View File

@@ -7,12 +7,13 @@
package provider
import (
adminController "github.com/echochat/backend/app/admin/controller"
adminDAO "github.com/echochat/backend/app/admin/dao"
adminService "github.com/echochat/backend/app/admin/service"
controller2 "github.com/echochat/backend/app/admin/controller"
dao2 "github.com/echochat/backend/app/admin/dao"
service2 "github.com/echochat/backend/app/admin/service"
"github.com/echochat/backend/app/auth/controller"
"github.com/echochat/backend/app/auth/dao"
"github.com/echochat/backend/app/auth/service"
"github.com/echochat/backend/app/ws"
"github.com/echochat/backend/config"
"github.com/echochat/backend/pkg/db"
)
@@ -38,9 +39,12 @@ func InitializeApp(cfg *config.Config) (*App, error) {
authService := service.NewAuthService(userDAO, roleDAO, jwtConfig, tokenStore)
authController := controller.NewAuthController(authService)
adminAuthController := controller.NewAdminAuthController(authService)
userManageDAO := adminDAO.NewUserManageDAO(gormDB)
userManageService := adminService.NewUserManageService(userManageDAO, userDAO, roleDAO)
userManageController := adminController.NewUserManageController(userManageService)
app := NewApp(cfg, gormDB, client, authService, authController, adminAuthController, userManageController)
userManageDAO := dao2.NewUserManageDAO(gormDB)
userManageService := service2.NewUserManageService(userManageDAO, userDAO, roleDAO)
userManageController := controller2.NewUserManageController(userManageService)
hub := ws.ProvideHub()
pubSub := ws.ProvidePubSub(client, hub)
handler := ws.ProvideWSHandler(hub, pubSub, jwtConfig)
app := NewApp(cfg, gormDB, client, authService, authController, adminAuthController, userManageController, handler, hub, pubSub)
return app, nil
}