服务器准备

rabbitmq/queue_setup.go

@@ -0,0 +1,111 @@
+// Package rabbitmq 提供 RabbitMQ 队列初始化的封装方法
+//
+// 本文件包含常用队列的声明和绑定逻辑，简化业务层的队列配置代码
+package rabbitmq
+
+import (
+	"context"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/glog"
+)
+
+// SetupResponseQueue 初始化 RAGFlow 响应队列
+//
+// 功能：
+//  1. 声明持久化队列（从配置文件读取队列名，默认 ragflow.response.queue）
+//  2. 绑定到 ragflow.response Exchange（Topic 类型）
+//  3. 使用通配符 # 匹配所有 routing key（userId）
+//
+// 参数：
+//
+//	ctx: 上下文
+//
+// 返回：
+//
+//	err: 错误信息，成功返回 nil
+//
+// 配置示例（config.yml）：
+//
+//	rabbitmq:
+//	  responseQueue: "ragflow.response.queue"  # 可选，默认值
+func SetupResponseQueue(ctx context.Context) (err error) {
+	// 从配置文件读取队列名（支持每个开发者配置独立队列名）
+	responseQueue := g.Cfg().MustGet(ctx, "rabbitmq.responseQueue", "ragflow.response.queue").String()
+
+	// 声明持久化队列（服务器重启后队列仍存在）
+	if err = DeclareQueue(ctx, &QueueConfig{
+		Name:    responseQueue,
+		Durable: true, // 持久化，防止数据丢失
+	}); err != nil {
+		glog.Errorf(ctx, "声明响应队列失败: %v", err)
+		return
+	}
+
+	// 绑定队列到 Exchange
+	// Exchange 类型为 topic，routing key 使用通配符 # 匹配所有 userId
+	if err = BindQueue(ctx, &BindingConfig{
+		Queue:      responseQueue,
+		Exchange:   "ragflow.response", // RAGFlow 响应 Exchange
+		RoutingKey: "#",                // 通配符，匹配所有消息
+	}); err != nil {
+		glog.Errorf(ctx, "绑定响应队列失败: %v", err)
+		return
+	}
+
+	glog.Infof(ctx, "响应队列已绑定: %s -> ragflow.response (routingKey=#)", responseQueue)
+	return
+}
+
+// SetupDelayedFlushQueue 初始化延时落库队列
+//
+// 功能：
+//  1. 声明延时 Exchange（x-delayed-message 插件）
+//  2. 声明持久化队列 conversation.flush.queue
+//  3. 绑定队列到延时 Exchange
+//
+// 用途：
+//
+//	对话缓存延时落库机制的兜底策略
+//	当对话少于5句时，10分钟后触发延时消息将缓存写入MongoDB
+//
+// 参数：
+//
+//	ctx: 上下文
+//
+// 返回：
+//
+//	err: 错误信息，成功返回 nil
+//
+// 相关：
+//   - service/conversation_service.go: handleResponse()
+//   - service/conversation_service.go: handleDelayedFlush()
+func SetupDelayedFlushQueue(ctx context.Context) (err error) {
+	// 声明延时 Exchange（需要 RabbitMQ 安装 x-delayed-message 插件）
+	if err = SetupDelayExchange(ctx, "conversation.flush.delayed"); err != nil {
+		glog.Warningf(ctx, "声明延时落库 Exchange 失败: %v", err)
+		return
+	}
+
+	// 声明持久化队列
+	if err = DeclareQueue(ctx, &QueueConfig{
+		Name:    "conversation.flush.queue",
+		Durable: true, // 持久化，防止延时消息丢失
+	}); err != nil {
+		glog.Warningf(ctx, "声明延时落库 Queue 失败: %v", err)
+		return
+	}
+
+	// 绑定队列到延时 Exchange
+	if err = BindQueue(ctx, &BindingConfig{
+		Queue:      "conversation.flush.queue",
+		Exchange:   "conversation.flush.delayed",
+		RoutingKey: "flush", // 延时落库消息的 routing key
+	}); err != nil {
+		glog.Warningf(ctx, "绑定延时落库 Queue 失败: %v", err)
+		return
+	}
+
+	glog.Info(ctx, "延时落库队列已配置")
+	return
+}