89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"io"
|
|||
|
|
"net/http"
|
|||
|
|
"net/url"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"model-asynch/model/entity"
|
|||
|
|
|
|||
|
|
"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
|
|||
|
|
}
|
|||
|
|
callbackURL := strings.TrimSpace(t.CallbackURL)
|
|||
|
|
bizName := strings.TrimSpace(t.BizName)
|
|||
|
|
if callbackURL == "" || bizName == "" {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
u, err := url.Parse(callbackURL)
|
|||
|
|
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)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 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)
|
|||
|
|
if err != nil {
|
|||
|
|
g.Log().Warningf(ctx, "[callback] build request failed url=%s err=%v", u.String(), 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)
|
|||
|
|
}
|
|||
|
|
|