gatway

service/callback.go

@@ -2,87 +2,86 @@ package service
 
 import (
 	"context"
-	"fmt"
-	"io"
-	"net/http"
-	"net/url"
-	"strings"
-	"time"
+	"encoding/json"
 
 	"model-asynch/model/entity"
 
+	"gitea.com/red-future/common/http"
 	"github.com/gogf/gf/v2/frame/g"
 )
 
-// triggerSuccessCallback 任务成功后的回调钩子：
-// - 使用 GET 请求
-// - 回调地址为 callbackUrl + "/" + bizName
-// - query 参数：task_id/state/oss_file/file_type
-// 注意：回调失败不影响任务主流程，只记录日志。
-func triggerSuccessCallback(ctx context.Context, t *entity.AsynchTask) {
-	if t == nil {
-		return
+// triggerCallback 任务成功后的回调：
+// - JSON body 参数：task_id/state/oss_file/file_type/text（可选）
+func triggerCallback(ctx context.Context, t *entity.AsynchTask) {
+	callbackURL := t.BizName + t.CallbackURL
+	headers := forwardHeaders(ctx)
+	var req struct{}
+	payload := map[string]interface{}{
+		"task_id":   t.TaskID,
+		"state":     t.State,
+		"oss_file":  t.OssFile,
+		"file_type": t.FileType,
+		"text":      t.TextResult,
+		"error_msg": t.ErrorMsg,
 	}
-	callbackURL := strings.TrimSpace(t.CallbackURL)
-	bizName := strings.TrimSpace(t.BizName)
-	if callbackURL == "" || bizName == "" {
-		return
-	}
-
-	u, err := url.Parse(callbackURL)
+	jsonData, err := json.Marshal(payload)
 	if err != nil {
-		g.Log().Warningf(ctx, "[callback] invalid callbackUrl=%s err=%v", callbackURL, err)
-		return
-	}
-	// 必须是可发起 HTTP 请求的绝对地址
-	if u.Scheme == "" || u.Host == "" {
-		g.Log().Warningf(ctx, "[callback] callbackUrl must be absolute http(s) url, got=%s", callbackURL)
+		g.Log().Warningf(ctx, "[回调] JSON序列化失败 taskId=%s 错误=%v", t.TaskID, err)
 		return
 	}
+	g.Log().Infof(ctx, "[回调] 开始发送 taskId=%s 回调地址=%s 请求头数量=%d 消息体大小=%d字节",
+		t.TaskID, callbackURL, len(headers), len(jsonData))
 
-	// path 末尾拼接 bizName
-	bizSeg := url.PathEscape(bizName)
-	if strings.HasSuffix(u.Path, "/") || u.Path == "" {
-		u.Path = strings.TrimRight(u.Path, "/") + "/" + bizSeg
-	} else {
-		u.Path = u.Path + "/" + bizSeg
-	}
-
-	q := u.Query()
-	q.Set("task_id", t.TaskID)
-	q.Set("state", fmt.Sprintf("%d", t.State))
-	q.Set("oss_file", t.OssFile)
-	q.Set("file_type", t.FileType)
-	u.RawQuery = q.Encode()
-
-	req, err := http.NewRequestWithContext(ctx, http.MethodGet, u.String(), nil)
+	err = http.Post(ctx, callbackURL, headers, &req, jsonData)
 	if err != nil {
-		g.Log().Warningf(ctx, "[callback] build request failed url=%s err=%v", u.String(), err)
+		g.Log().Warningf(ctx, "[回调] 发送失败 taskId=%s 回调地址=%s 错误=%v", t.TaskID, callbackURL, err)
 		return
 	}
-	// 透传必要头部（如 Authorization / X-User-Info）
-	for k, v := range forwardHeaders(ctx) {
-		if v != "" {
-			req.Header.Set(k, v)
-		}
-	}
-
-	client := &http.Client{Timeout: 5 * time.Second}
-	resp, err := client.Do(req)
-	if err != nil {
-		g.Log().Warningf(ctx, "[callback] request failed url=%s err=%v", u.String(), err)
-		return
-	}
-	defer resp.Body.Close()
-	b, _ := io.ReadAll(resp.Body)
-	if resp.StatusCode < 200 || resp.StatusCode >= 300 {
-		msg := string(b)
-		if len(msg) > 2000 {
-			msg = msg[:2000]
-		}
-		g.Log().Warningf(ctx, "[callback] non-2xx url=%s code=%d body=%s", u.String(), resp.StatusCode, msg)
-		return
-	}
-	g.Log().Infof(ctx, "[callback] success url=%s code=%d", u.String(), resp.StatusCode)
+	g.Log().Infof(ctx, "[回调] 发送成功 taskId=%s 回调地址=%s 消息体大小=%d字节", t.TaskID, callbackURL, len(jsonData))
 }
 
+// triggerPromptsCallback 任务成功后的提示词回调
+// - JSON body 参数：epicycleId（轮次id）/textResult（模型回答消息）
+func triggerPromptsCallback(ctx context.Context, t *entity.AsynchTask, epicycleId int64) {
+	callbackURL := "prompts-core/session/sessionCallback"
+	headers := forwardHeaders(ctx)
+	var req struct{}
+	payload := map[string]interface{}{
+		"epicycleId": epicycleId,
+		"text":       t.TextResult,
+	}
+	jsonData, err := json.Marshal(payload)
+	if err != nil {
+		g.Log().Warningf(ctx, "[提示词回调] JSON序列化失败 epicycleId=%d 错误=%v", epicycleId, err)
+		return
+	}
+	g.Log().Infof(ctx, "[提示词回调] 开始发送 epicycleId=%d 回调地址=%s 请求头数量=%d 消息体大小=%d字节",
+		t.EpicycleId, callbackURL, len(headers), len(jsonData))
+
+	err = http.Post(ctx, callbackURL, headers, &req, jsonData)
+	if err != nil {
+		g.Log().Warningf(ctx, "[提示词回调] 发送失败 epicycleId=%d 回调地址=%s 错误=%v", t.EpicycleId, callbackURL, err)
+		return
+	}
+	g.Log().Infof(ctx, "[提示词回调] 发送成功 epicycleId=%d 回调地址=%s 消息体大小=%d字节", t.EpicycleId, callbackURL, len(jsonData))
+}
+
+// IsSuperAdmin 调用admin-go服务检查是否是超级管理员
+func IsSuperAdmin(ctx context.Context) (res bool, err error) {
+	headers := forwardHeaders(ctx)
+	var r = make(map[string]bool)
+	if err = http.Get(ctx, "admin-go/api/v1/system/user/checkIsSuperAdmin", headers, &r); err != nil {
+		return false, err
+	}
+	return r["isSuperAdmin"], err
+}
+
+// IsAdmin 调用admin-go服务检查是否是管理员
+func IsAdmin(ctx context.Context) (res bool, err error) {
+	headers := forwardHeaders(ctx)
+	var r = make(map[string]bool)
+	if err = http.Get(ctx, "admin-go/api/v1/system/user/checkIsSuperAdmin", headers, &r); err != nil {
+		return false, err
+	}
+	return r["isSuperAdmin"], err
+}