From 577363287d447c2cb1ab6dbf769769a2ec5dc5aa Mon Sep 17 00:00:00 2001
From: ideaa <ideaa@qq.com>
Date: Mon, 29 Sep 2025 09:40:25 +0800
Subject: [PATCH] refactor(ws/pubsub): simplify topic message processing loop

Remove redundant channel check and flatten select statement into a range loop for cleaner message handling
---
 ws/pubsub.go | 32 ++++++++++++--------------------
 1 file changed, 12 insertions(+), 20 deletions(-)

diff --git a/ws/pubsub.go b/ws/pubsub.go
index dadab06..bf8e064 100644
--- a/ws/pubsub.go
+++ b/ws/pubsub.go
@@ -39,7 +39,7 @@ func (a *PubSub) initTopic(topicId string) *Topic {
 // Pub 发布主题
 func (a *PubSub) Pub(topicId string, data any) {
 	msg := Action{
-		Action: "subscriber",
+		Action: topicId,
 		Data: H{
 			"topicId": topicId,
 			"message": data,
@@ -66,25 +66,17 @@ func (a *PubSub) SubFunc(topicId string, f func(msg *TopicMsg)) {
 }
 
 func (a *PubSub) Start() {
-	for {
-		select {
-		case msg, ok := <-a.TopicMsgQueue:
-			if !ok {
-				logger.SugarLog.Info("从订阅主题队列取数据失败")
-				continue
-			}
-
-			t, hasTopic := a.Topics.Load(msg.TopicId)
-			if !hasTopic {
-				logger.SugarLog.Info("未发布订阅主题收到消息")
-				continue
-			}
-
-			//订阅消息的函数处理
-			t.(*Topic).ApplyFunc(msg)
-
-			//订阅消息的用户处理
-			t.(*Topic).SendToSubUser(msg.Msg)
+	for msg := range a.TopicMsgQueue {
+		t, hasTopic := a.Topics.Load(msg.TopicId)
+		if !hasTopic {
+			logger.SugarLog.Info("未发布订阅主题收到消息")
+			continue
 		}
+
+		//订阅消息的函数处理
+		t.(*Topic).ApplyFunc(msg)
+
+		//订阅消息的用户处理
+		t.(*Topic).SendToSubUser(msg.Msg)
 	}
 }