- 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
51 lines
1.8 KiB
Go
51 lines
1.8 KiB
Go
// Code generated by Wire. DO NOT EDIT.
|
||
|
||
//go:generate go run -mod=mod github.com/google/wire/cmd/wire
|
||
//go:build !wireinject
|
||
// +build !wireinject
|
||
|
||
package provider
|
||
|
||
import (
|
||
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"
|
||
)
|
||
|
||
// Injectors from wire.go:
|
||
|
||
// InitializeApp 初始化整个应用(Wire 自动生成实现)
|
||
func InitializeApp(cfg *config.Config) (*App, error) {
|
||
databaseConfig := provideDBConfig(cfg)
|
||
gormDB, err := db.NewPostgres(databaseConfig)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
redisConfig := provideRedisConfig(cfg)
|
||
client, err := db.NewRedis(redisConfig)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
userDAO := dao.NewUserDAO(gormDB)
|
||
roleDAO := dao.NewRoleDAO(gormDB)
|
||
jwtConfig := provideJWTConfig(cfg)
|
||
tokenStore := service.NewTokenStore(client, jwtConfig)
|
||
authService := service.NewAuthService(userDAO, roleDAO, jwtConfig, tokenStore)
|
||
authController := controller.NewAuthController(authService)
|
||
adminAuthController := controller.NewAdminAuthController(authService)
|
||
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
|
||
}
|