65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
// Package controller - RAGFlow控制器
|
||
// 功能：接收来自外部的RAGFlow请求（测试/调试用）
|
||
package controller
|
||
|
||
import (
|
||
"customer-server/consumer"
|
||
|
||
"gitea.com/red-future/common/redis"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/gogf/gf/v2/net/ghttp"
|
||
)
|
||
|
||
type ragflow struct{}
|
||
|
||
var RAGFlow = new(ragflow)
|
||
|
||
// Process 处理RAGFlow请求
|
||
// 参数: r - HTTP请求对象
|
||
// 功能: 接收外部RAGFlow消息请求，直接调用consumer处理逻辑（用于测试）
|
||
// 注意: 正常流程不经过此接口，而是直接消费Redis Stream
|
||
func (c *ragflow) Process(r *ghttp.Request) {
|
||
ctx := r.Context()
|
||
|
||
// 调试：打印原始请求体
|
||
bodyBytes := r.GetBody()
|
||
g.Log().Infof(ctx, "收到原始请求体: %s", string(bodyBytes))
|
||
|
||
var req redis.SendStreamMessage
|
||
if err := r.Parse(&req); err != nil {
|
||
r.Response.WriteJsonExit(g.Map{
|
||
"code": 400,
|
||
"msg": "参数错误: " + err.Error(),
|
||
})
|
||
return
|
||
}
|
||
|
||
g.Log().Infof(ctx, "收到RAGFlow请求 - 用户: %s, 内容: %s", req.UserId, req.Content)
|
||
|
||
// 直接调用consumer的处理逻辑
|
||
message := map[string]interface{}{
|
||
"UserId": req.UserId,
|
||
"Content": req.Content,
|
||
"MessageId": req.MessageId,
|
||
"Platform": req.Platform,
|
||
"TenantId": req.TenantId,
|
||
"AccountName": req.AccountName,
|
||
"ChatId": req.ChatId,
|
||
"ReplyQueue": req.ReplyQueue,
|
||
"History": req.History,
|
||
}
|
||
|
||
if err := consumer.HandleMessageHTTP(ctx, message); err != nil {
|
||
r.Response.WriteJsonExit(g.Map{
|
||
"code": 500,
|
||
"msg": "处理失败: " + err.Error(),
|
||
})
|
||
return
|
||
}
|
||
|
||
r.Response.WriteJsonExit(g.Map{
|
||
"code": 0,
|
||
"msg": "success",
|
||
})
|
||
}
|