48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
|
package consumer
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
|
|||
|
|
"github.com/gogf/gf/v2/encoding/gjson"
|
|||
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|||
|
|
"github.com/gogf/gf/v2/frame/g"
|
|||
|
|
"github.com/gogf/gf/v2/os/glog"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// recreateChatIfNeeded 检测并重建Chat配置
|
|||
|
|
// 调用customerservice的recreateRAGFlow接口，由customerservice负责MongoDB操作
|
|||
|
|
func recreateChatIfNeeded(ctx context.Context, tenantId, accountName, platform string) error {
|
|||
|
|
glog.Infof(ctx, "开始自动重建Chat - accountName: %s, platform: %s, tenantId: %s", accountName, platform, tenantId)
|
|||
|
|
|
|||
|
|
// 1. 从配置获取customerservice地址（避免consul.GetInstanceAddr在Stream context中panic）
|
|||
|
|
customerserviceAddr := g.Cfg().MustGet(ctx, "customerservice.address", "customerservice:3000").String()
|
|||
|
|
|
|||
|
|
// 2. 调用customerservice的recreateRAGFlow接口
|
|||
|
|
url := "http://" + customerserviceAddr + "/customer/service/account/recreateRAGFlow"
|
|||
|
|
reqBody := g.Map{
|
|||
|
|
"accountName": accountName,
|
|||
|
|
"platform": platform,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
glog.Infof(ctx, "调用customerservice recreateRAGFlow接口 - URL: %s, 请求: %+v", url, reqBody)
|
|||
|
|
|
|||
|
|
client := g.Client()
|
|||
|
|
resp, err := client.Post(ctx, url, reqBody)
|
|||
|
|
if err != nil {
|
|||
|
|
return gerror.Wrapf(err, "调用recreateRAGFlow接口失败")
|
|||
|
|
}
|
|||
|
|
defer resp.Close()
|
|||
|
|
|
|||
|
|
// 3. 解析响应
|
|||
|
|
respBody := resp.ReadAllString()
|
|||
|
|
result := gjson.New(respBody)
|
|||
|
|
|
|||
|
|
if result.Get("code").Int() != 0 {
|
|||
|
|
errMsg := result.Get("message").String()
|
|||
|
|
return gerror.Newf("recreateRAGFlow失败: %s", errMsg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
glog.Infof(ctx, "Chat自动重建完成 - accountName: %s", accountName)
|
|||
|
|
return nil
|
|||
|
|
}
|