lan
c630cbf4d0
Introduce a new WebSocket messaging architecture that supports both standalone and cluster modes. This allows for horizontal scaling of WebSocket servers by using Redis Pub/Sub to synchronize messages across multiple instances. Key changes: - Added `ws.MessagePublisher` interface to abstract message distribution. - Implemented `ws.Bus` to handle cluster-mode messaging via Redis. - Added `ws.OnlineTracker` to manage user online status across the cluster. - Refactored multiple services (Chat, Group, Push, Call, etc.) to use the new `MessagePublisher` instead of a concrete `ws.Hub`. - Added WebSocket configuration options (mode, instance ID, channel, TTL, heartbeat) to `config.yaml` and `config.go`. - Updated dependency injection with Wire to support the new publisher and Redis client. - Improved logging by replacing standard `log` with `zap` in several service components.
119 lines
3.4 KiB
Go
119 lines
3.4 KiB
Go
package wire
|
|
|
|
import (
|
|
"with_you/internal/handler"
|
|
"with_you/internal/pkg/ws"
|
|
"with_you/internal/repository"
|
|
"with_you/internal/service"
|
|
|
|
"github.com/google/wire"
|
|
)
|
|
|
|
// HandlerSet Handler 层 Provider Set
|
|
var HandlerSet = wire.NewSet(
|
|
// 直接使用构造函数的 Handler
|
|
handler.NewCommentHandler,
|
|
handler.NewNotificationHandler,
|
|
handler.NewUploadHandler,
|
|
handler.NewPushHandler,
|
|
handler.NewStickerHandler,
|
|
handler.NewVoteHandler,
|
|
handler.NewChannelHandler,
|
|
handler.NewRoleHandler,
|
|
handler.NewAdminUserHandler,
|
|
handler.NewAdminPostHandler,
|
|
handler.NewAdminCommentHandler,
|
|
handler.NewAdminGroupHandler,
|
|
handler.NewAdminDashboardHandler,
|
|
handler.NewAdminLogHandler,
|
|
handler.NewQRCodeHandler,
|
|
handler.NewMaterialHandler,
|
|
handler.NewCallHandler,
|
|
handler.NewReportHandler,
|
|
handler.NewAdminReportHandler,
|
|
handler.NewVerificationHandler,
|
|
handler.NewAdminProfileAuditHandler,
|
|
handler.NewPostHandler,
|
|
handler.NewTradeHandler,
|
|
|
|
// 需要特殊处理的 Handler
|
|
ProvideUserHandler,
|
|
ProvideMessageHandler,
|
|
ProvideSystemMessageHandler,
|
|
ProvideGroupHandler,
|
|
ProvideScheduleHandler,
|
|
ProvideWSHandler,
|
|
ProvideAdminVerificationHandler,
|
|
ProvideSetupHandler,
|
|
)
|
|
|
|
func ProvideUserHandler(
|
|
userService service.UserService,
|
|
activityService service.UserActivityService,
|
|
logService *service.LogService,
|
|
profileAuditService service.UserProfileAuditService,
|
|
) *handler.UserHandler {
|
|
h := handler.NewUserHandler(userService, activityService, logService)
|
|
h.SetProfileAuditService(profileAuditService)
|
|
return h
|
|
}
|
|
|
|
func ProvideMessageHandler(
|
|
chatService service.ChatService,
|
|
messageService *service.MessageService,
|
|
userService service.UserService,
|
|
groupService service.GroupService,
|
|
publisher ws.MessagePublisher,
|
|
) *handler.MessageHandler {
|
|
return handler.NewMessageHandler(chatService, messageService, userService, groupService, publisher)
|
|
}
|
|
|
|
// ProvideSystemMessageHandler 提供系统消息处理器
|
|
func ProvideSystemMessageHandler(
|
|
systemMsgService service.SystemMessageService,
|
|
) *handler.SystemMessageHandler {
|
|
return handler.NewSystemMessageHandler(systemMsgService)
|
|
}
|
|
|
|
// ProvideGroupHandler 提供群组处理器
|
|
func ProvideGroupHandler(
|
|
groupService service.GroupService,
|
|
userService service.UserService,
|
|
) *handler.GroupHandler {
|
|
return handler.NewGroupHandler(groupService, userService)
|
|
}
|
|
|
|
// ProvideScheduleHandler 提供课表处理器
|
|
func ProvideScheduleHandler(
|
|
scheduleService service.ScheduleService,
|
|
scheduleSyncService service.ScheduleSyncService,
|
|
) *handler.ScheduleHandler {
|
|
return handler.NewScheduleHandler(scheduleService, scheduleSyncService)
|
|
}
|
|
|
|
// ProvideWSHandler 提供WebSocket处理器
|
|
func ProvideWSHandler(
|
|
publisher ws.MessagePublisher,
|
|
chatService service.ChatService,
|
|
groupService service.GroupService,
|
|
jwtService *service.JWTService,
|
|
callService service.CallService,
|
|
userRepo repository.UserRepository,
|
|
) *handler.WSHandler {
|
|
return handler.NewWSHandler(publisher, chatService, groupService, jwtService, callService, userRepo)
|
|
}
|
|
|
|
// ProvideAdminVerificationHandler 提供管理端身份认证处理器
|
|
func ProvideAdminVerificationHandler(
|
|
adminVerificationService service.AdminVerificationService,
|
|
userRepo repository.UserRepository,
|
|
) *handler.AdminVerificationHandler {
|
|
return handler.NewAdminVerificationHandler(adminVerificationService, userRepo)
|
|
}
|
|
|
|
func ProvideSetupHandler(
|
|
setupService service.SetupService,
|
|
) *handler.SetupHandler {
|
|
return handler.NewSetupHandler(setupService)
|
|
}
|