完成websocket对话演示和main方法的功能抽离

rabbitmq/consumer.go

@@ -83,6 +83,34 @@ func (c *Consumer) Start(ctx context.Context) (err error) {
 		return err
 	}
 
+	// 声明队列（如果不存在则创建）
+	_, err = ch.QueueDeclare(
+		c.queue, // name
+		true,    // durable（持久化）
+		false,   // autoDelete（不自动删除）
+		false,   // exclusive（非独占）
+		false,   // noWait
+		nil,     // arguments
+	)
+	if err != nil {
+		return gerror.Newf("声明队列失败: %v", err)
+	}
+
+	// TODO: 队列绑定逻辑暂时注释，避免重复binding导致消息重复投递
+	// 绑定队列到Exchange（使用队列名作为routingKey，支持多租户）
+	// Exchange类型应该是topic，绑定模式为 #（接收所有消息）
+	// err = ch.QueueBind(
+	// 	c.queue,           // queue name
+	// 	"#",               // routing key（通配符，接收所有消息）
+	// 	"ragflow.response", // exchange name
+	// 	false,             // noWait
+	// 	nil,               // arguments
+	// )
+	// if err != nil {
+	// 	g.Log().Warningf(ctx, "绑定队列到Exchange失败（可能Exchange不存在或类型不匹配）: %v", err)
+	// 	// 不返回错误，继续启动消费者（可能是direct exchange或队列已绑定）
+	// }
+
 	// 设置 QoS（并发控制）
 	err = ch.Qos(
 		c.prefetchCount, // prefetchCount: 每个 consumer 最多同时处理的消息数

rabbitmq/instance.go

@@ -0,0 +1,41 @@
+package rabbitmq
+
+import (
+	"fmt"
+	"os"
+	"sync"
+
+	"github.com/gogf/gf/v2/util/guid"
+)
+
+var (
+	instanceId   string
+	instanceOnce sync.Once
+)
+
+// getInstanceId 获取当前实例的唯一标识（单例）
+// 格式：{hostname}.{uuid8}
+func getInstanceId() string {
+	instanceOnce.Do(func() {
+		// 获取主机名
+		hostname, err := os.Hostname()
+		if err != nil || hostname == "" {
+			hostname = "unknown"
+		}
+
+		// 生成8位UUID
+		uuid := guid.S()[:8]
+
+		instanceId = fmt.Sprintf("%s.%s", hostname, uuid)
+	})
+	return instanceId
+}
+
+// GetInstanceQueueName 获取当前实例的响应队列名
+// 格式：{baseQueue}.{hostname}.{uuid8}
+func GetInstanceQueueName(baseQueue string) string {
+	if baseQueue == "" {
+		baseQueue = "ragflow.response"
+	}
+	return fmt.Sprintf("%s.%s", baseQueue, getInstanceId())
+}