修复冻结用户强制下线

backend/internal/modules/chathub/hub.go

@@ -86,6 +86,22 @@ func (h *Hub) Unsubscribe(pType string, pID uint64, ch <-chan *ChatEvent) {
 	h.mu.Unlock()
 }
 
+// DisconnectUser 关闭指定用户当前进程内的全部实时连接。
+func (h *Hub) DisconnectUser(userID uint64) {
+	h.disconnect("user", userID)
+}
+
+func (h *Hub) disconnect(pType string, pID uint64) {
+	key := principal{Type: pType, ID: pID}
+	h.mu.Lock()
+	clients := h.clients[key]
+	delete(h.clients, key)
+	for ch := range clients {
+		close(ch)
+	}
+	h.mu.Unlock()
+}
+
 // NotifyConversation 查询会话参与者，向所有在线参与者推送事件。
 func (h *Hub) NotifyConversation(conversationID uint64, event *ChatEvent) {
 	if h.db == nil {

backend/internal/modules/chathub/hub_test.go

@@ -0,0 +1,25 @@
+package chathub
+
+import "testing"
+
+func TestDisconnectUserClosesAllConnections(t *testing.T) {
+	hub := NewHub(nil)
+	first := hub.Subscribe("user", 7)
+	second := hub.Subscribe("user", 7)
+	hub.Subscribe("user", 8)
+
+	hub.DisconnectUser(7)
+	for name, ch := range map[string]<-chan *ChatEvent{"first": first, "second": second} {
+		select {
+		case _, ok := <-ch:
+			if ok {
+				t.Fatalf("%s connection was not closed", name)
+			}
+		default:
+			t.Fatalf("%s connection did not close immediately", name)
+		}
+	}
+	if got := hub.OnlineCount(); got != 1 {
+		t.Fatalf("online connection count = %d, want 1", got)
+	}
+}