refactor(model): 重构模型实体和数据访问层

common/util/files.go

@@ -0,0 +1,69 @@
+package util
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// DetectFileType 根据返回的二进制内容推断 contentType + 扩展名（尽量稳定）
+func DetectFileType(data []byte) (contentType string, ext string) {
+	if len(data) == 0 {
+		return "application/octet-stream", ""
+	}
+	ct := http.DetectContentType(data)
+	// gateway.DetectContentType 可能带 charset 等参数：text/plain; charset=utf-8
+	if idx := strings.Index(ct, ";"); idx > 0 {
+		ct = strings.TrimSpace(ct[:idx])
+	}
+	switch ct {
+	case "audio/mpeg":
+		return ct, ".mp3"
+	case "audio/wave", "audio/wav", "audio/x-wav":
+		return ct, ".wav"
+	case "video/mp4":
+		return ct, ".mp4"
+	case "image/png":
+		return ct, ".png"
+	case "image/jpeg":
+		return ct, ".jpg"
+	case "application/pdf":
+		return ct, ".pdf"
+	case "text/plain":
+		return ct, ".txt"
+	case "application/json":
+		return ct, ".json"
+	default:
+		// 兜底：尝试从 ct 截取 subtype 作为后缀（例如 application/json）
+		if parts := strings.Split(ct, "/"); len(parts) == 2 {
+			sub := parts[1]
+			// 避免出现 "plain; charset=utf-8" 之类的后缀
+			if idx := strings.Index(sub, ";"); idx > 0 {
+				sub = strings.TrimSpace(sub[:idx])
+			}
+			return ct, "." + sub
+		}
+		return ct, ""
+	}
+}
+
+// SaveTmpResult 将模型输出写入临时文件，用于 OSS 上传失败后的“仅重试 OSS”。
+func SaveTmpResult(taskID string, data []byte, ext string) (string, error) {
+	dir := filepath.Join(os.TempDir(), "model-asynch")
+	if err := os.MkdirAll(dir, 0o755); err != nil {
+		return "", err
+	}
+	if ext == "" {
+		ext = ".bin"
+	}
+	if ext[0] != '.' {
+		ext = "." + ext
+	}
+	path := filepath.Join(dir, fmt.Sprintf("%s%s", taskID, ext))
+	if err := os.WriteFile(path, data, 0o644); err != nil {
+		return "", err
+	}
+	return path, nil
+}

common/util/headers.go

@@ -0,0 +1,100 @@
+package util
+
+import (
+	"context"
+
+	"gitea.com/red-future/common/utils"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+// AsyncCtx 固化异步上下文中的 token 和用户信息，避免请求结束后丢失
+func AsyncCtx(ctx context.Context) context.Context {
+	asyncCtx := context.WithoutCancel(ctx)
+
+	if r := g.RequestFromCtx(ctx); r != nil {
+		if token := r.Header.Get("Authorization"); token != "" {
+			asyncCtx = context.WithValue(asyncCtx, "token", token)
+		}
+		if userInfo := r.Header.Get("X-User-Info"); userInfo != "" {
+			asyncCtx = context.WithValue(asyncCtx, "xUserInfo", userInfo)
+		}
+	}
+
+	if user, err := utils.GetUserInfo(ctx); err == nil && user != nil {
+		asyncCtx = context.WithValue(asyncCtx, "user", user)
+	}
+
+	return asyncCtx
+}
+
+// ForwardHeaders 透传调用链路的头信息，优先使用 ctx 中的固化值
+func ForwardHeaders(ctx context.Context) map[string]string {
+	headers := make(map[string]string)
+	SetHeaderFromContext(headers, ctx, "Authorization", "token")
+	SetHeaderFromContext(headers, ctx, "X-User-Info", "xUserInfo")
+	FallbackToRequestHeaders(headers, ctx)
+	return headers
+}
+
+// SetHeaderFromContext 从上下文中设置 header
+func SetHeaderFromContext(headers map[string]string, ctx context.Context, headerKey, ctxKey string) {
+	if value, ok := ctx.Value(ctxKey).(string); ok && value != "" {
+		headers[headerKey] = value
+	}
+}
+
+// FallbackToRequestHeaders 从请求头中获取作为兜底
+func FallbackToRequestHeaders(headers map[string]string, ctx context.Context) {
+	r := g.RequestFromCtx(ctx)
+	if r == nil {
+		return
+	}
+
+	if headers["Authorization"] == "" {
+		if token := r.Header.Get("Authorization"); token != "" {
+			headers["Authorization"] = token
+		}
+	}
+
+	if headers["X-User-Info"] == "" {
+		if userInfo := r.Header.Get("X-User-Info"); userInfo != "" {
+			headers["X-User-Info"] = userInfo
+		}
+	}
+}
+
+// SetTaskHeadersToCtx 把任务入库时保存的 header 信息注入 ctx，给 worker 调 OSS 用
+func SetTaskHeadersToCtx(ctx context.Context, headers map[string]string) context.Context {
+	if headers == nil {
+		return ctx
+	}
+	if v := gconv.String(headers["Authorization"]); v != "" {
+		ctx = context.WithValue(ctx, "token", v)
+	}
+	if v := gconv.String(headers["X-User-Info"]); v != "" {
+		ctx = context.WithValue(ctx, "xUserInfo", v)
+	}
+	return ctx
+}
+
+// ParseStoredPayload 解析入库的 request_payload，拆出模型调用 payload 与透传 headers
+// 入库格式：{"payload": <any>, "headers": {"Authorization": "...", "X-User-Info":"..."}}
+func ParseStoredPayload(v any) (payload any, headers map[string]string) {
+	if v == nil {
+		return nil, nil
+	}
+	m := gconv.Map(v)
+	if len(m) == 0 {
+		return v, nil
+	}
+	if h, ok := m["headers"]; ok {
+		headers = gconv.MapStrStr(h)
+	}
+	if p, ok := m["payload"]; ok {
+		payload = p
+	} else {
+		payload = v
+	}
+	return
+}

common/util/json.go

@@ -0,0 +1,28 @@
+package util
+
+import (
+	"encoding/json"
+
+	"github.com/gogf/gf/v2/container/gvar"
+)
+
+func ParseJSONField(field any) any {
+	var v *gvar.Var
+	switch val := field.(type) {
+	case *gvar.Var:
+		v = val
+	default:
+		return field
+	}
+
+	if v == nil || v.IsNil() || v.IsEmpty() {
+		return nil
+	}
+
+	str := v.String()
+	var result any
+	if json.Unmarshal([]byte(str), &result) == nil {
+		return result
+	}
+	return str
+}