feat(websocket): integrate WebSocket support and refactor SSE handling

- Added WebSocket support by introducing a new WSHandler and related infrastructure.
- Replaced SSE implementations with WebSocket equivalents across message and QR code handlers.
- Updated service and repository layers to utilize WebSocket for real-time messaging.
- Removed obsolete SSE hub and related code, streamlining the application for WebSocket usage.
- Enhanced router to include WebSocket endpoints for real-time communication.

internal/service/chat_service.go

@@ -10,7 +10,7 @@ import (
 	"carrot_bbs/internal/cache"
 	"carrot_bbs/internal/dto"
 	"carrot_bbs/internal/model"
-	"carrot_bbs/internal/pkg/sse"
+	"carrot_bbs/internal/pkg/ws"
 	"carrot_bbs/internal/repository"
 
 	"go.uber.org/zap"
@@ -59,7 +59,7 @@ type chatServiceImpl struct {
 	repo      repository.MessageRepository
 	userRepo  repository.UserRepository
 	sensitive SensitiveService
-	sseHub    *sse.Hub
+	wsHub     *ws.Hub
 
 	// 缓存相关字段
 	conversationCache *cache.ConversationCache
@@ -71,7 +71,7 @@ func NewChatService(
 	repo repository.MessageRepository,
 	userRepo repository.UserRepository,
 	sensitive SensitiveService,
-	sseHub *sse.Hub,
+	wsHub *ws.Hub,
 	cacheBackend cache.Cache,
 	uploadService *UploadService,
 ) ChatService {
@@ -91,17 +91,17 @@ func NewChatService(
 		repo:              repo,
 		userRepo:          userRepo,
 		sensitive:         sensitive,
-		sseHub:            sseHub,
+		wsHub:             wsHub,
 		conversationCache: conversationCache,
 		uploadService:     uploadService,
 	}
 }
 
-func (s *chatServiceImpl) publishSSEToUsers(userIDs []string, event string, payload interface{}) {
-	if s.sseHub == nil || len(userIDs) == 0 {
+func (s *chatServiceImpl) publishToUsers(userIDs []string, event string, payload interface{}) {
+	if s.wsHub == nil || len(userIDs) == 0 {
 		return
 	}
-	s.sseHub.PublishToUsers(userIDs, event, payload)
+	s.wsHub.PublishToUsers(userIDs, event, payload)
 }
 
 // GetOrCreateConversation 获取或创建私聊会话
@@ -314,12 +314,12 @@ func (s *chatServiceImpl) SendMessage(ctx context.Context, senderID string, conv
 		}
 	}()
 
-	// 获取会话中的参与者并发送 SSE
+	// 获取会话中的参与者并发送消息
 	participants, err := s.getParticipants(ctx, conversationID)
 	if err == nil {
 		targetIDs := make([]string, 0, len(participants))
 		for _, p := range participants {
-			// 私聊场景下，发送者已经从 HTTP 响应拿到消息，避免再通过 SSE 回推导致本端重复展示。
+			// 私聊场景下，发送者已经从 HTTP 响应拿到消息，避免再通过 WebSocket 回推导致本端重复展示。
 			if conv.Type == model.ConversationTypePrivate && p.UserID == senderID {
 				continue
 			}
@@ -329,7 +329,7 @@ func (s *chatServiceImpl) SendMessage(ctx context.Context, senderID string, conv
 		if conv.Type == model.ConversationTypeGroup {
 			detailType = "group"
 		}
-		s.publishSSEToUsers(targetIDs, "chat_message", map[string]interface{}{
+		s.publishToUsers(targetIDs, "chat_message", map[string]interface{}{
 			"detail_type": detailType,
 			"message":     dto.ConvertMessageToResponse(message),
 		})
@@ -342,7 +342,7 @@ func (s *chatServiceImpl) SendMessage(ctx context.Context, senderID string, conv
 				s.conversationCache.InvalidateUnreadCount(p.UserID, conversationID)
 			}
 			if totalUnread, uErr := s.repo.GetAllUnreadCount(p.UserID); uErr == nil {
-				s.publishSSEToUsers([]string{p.UserID}, "conversation_unread", map[string]interface{}{
+				s.publishToUsers([]string{p.UserID}, "conversation_unread", map[string]interface{}{
 					"conversation_id": conversationID,
 					"total_unread":    totalUnread,
 				})
@@ -490,7 +490,7 @@ func (s *chatServiceImpl) MarkAsRead(ctx context.Context, conversationID string,
 		for _, p := range participants {
 			targetIDs = append(targetIDs, p.UserID)
 		}
-		s.publishSSEToUsers(targetIDs, "message_read", map[string]interface{}{
+		s.publishToUsers(targetIDs, "message_read", map[string]interface{}{
 			"detail_type":     detailType,
 			"conversation_id": conversationID,
 			"group_id":        groupID,
@@ -499,7 +499,7 @@ func (s *chatServiceImpl) MarkAsRead(ctx context.Context, conversationID string,
 		})
 	}
 	if totalUnread, uErr := s.repo.GetAllUnreadCount(userID); uErr == nil {
-		s.publishSSEToUsers([]string{userID}, "conversation_unread", map[string]interface{}{
+		s.publishToUsers([]string{userID}, "conversation_unread", map[string]interface{}{
 			"conversation_id": conversationID,
 			"total_unread":    totalUnread,
 		})
@@ -579,7 +579,7 @@ func (s *chatServiceImpl) RecallMessage(ctx context.Context, messageID string, u
 		for _, p := range participants {
 			targetIDs = append(targetIDs, p.UserID)
 		}
-		s.publishSSEToUsers(targetIDs, "message_recall", map[string]interface{}{
+		s.publishToUsers(targetIDs, "message_recall", map[string]interface{}{
 			"detail_type":     detailType,
 			"conversation_id": message.ConversationID,
 			"group_id":        groupID,
@@ -627,7 +627,7 @@ func (s *chatServiceImpl) DeleteMessage(ctx context.Context, messageID string, u
 
 // SendTyping 发送正在输入状态
 func (s *chatServiceImpl) SendTyping(ctx context.Context, senderID string, conversationID string) {
-	if s.sseHub == nil {
+	if s.wsHub == nil {
 		return
 	}
 
@@ -651,8 +651,8 @@ func (s *chatServiceImpl) SendTyping(ctx context.Context, senderID string, conve
 		if p.UserID == senderID {
 			continue
 		}
-		if s.sseHub != nil {
-			s.sseHub.PublishToUser(p.UserID, "typing", map[string]interface{}{
+		if s.wsHub != nil {
+			s.wsHub.PublishToUser(p.UserID, "typing", map[string]interface{}{
 				"detail_type":     detailType,
 				"conversation_id": conversationID,
 				"user_id":         senderID,
@@ -664,8 +664,8 @@ func (s *chatServiceImpl) SendTyping(ctx context.Context, senderID string, conve
 
 // IsUserOnline 检查用户是否在线
 func (s *chatServiceImpl) IsUserOnline(userID string) bool {
-	if s.sseHub != nil {
-		return s.sseHub.HasSubscribers(userID)
+	if s.wsHub != nil {
+		return s.wsHub.HasClients(userID)
 	}
 	return false
 }