🐛 onnx运行路径问题

internal/application/classifier/classifier.go

@@ -5,9 +5,14 @@ import (
 	"fmt"
 	_ "image/jpeg"
 	_ "image/png"
+	"likedream-goframe/internal/framework/logger"
 	"os"
+	"path/filepath"
+	"runtime"
+	"strings"
 
 	ort "github.com/yalue/onnxruntime_go"
+	"go.uber.org/zap"
 )
 
 type ImageClassifier struct {
@@ -21,7 +26,25 @@ type ImageClassifier struct {
 // NewImageClassifier 创建新的图片分类器
 func NewImageClassifier(modelPath, classesPath string) (*ImageClassifier, error) {
 	// 初始化ONNX Runtime
-	err := ort.InitializeEnvironment()
+	exePath := strings.ToLower(os.Args[0])
+	// 获取绝对路径
+	absPath, err := filepath.Abs(exePath)
+	dir := filepath.Dir(absPath)
+	systemOS := runtime.GOOS
+	switch systemOS {
+	case "windows":
+		dir = filepath.Join(dir, "onnxruntime.dll")
+		break
+	case "darwin":
+		dir = filepath.Join(dir, "onnxruntime.dylib")
+		break
+	case "linux":
+		dir = filepath.Join(dir, "onnxruntime.so")
+		break
+	}
+	logger.Logger.Info("初始化onnx：", zap.String("绝对路径", absPath), zap.String("父文件夹", dir))
+	ort.SetSharedLibraryPath(dir)
+	err = ort.InitializeEnvironment()
 	if err != nil {
 		return nil, fmt.Errorf("初始化ONNX Runtime失败: %v", err)
 	}

internal/framework/schedule/cpdaily_handler.go

@@ -64,7 +64,7 @@ func (c *CpdailyTaskHandler) ProcessTask(ctx context.Context, task *asynq.Task)
 		e = db.Global.CpTaskLog.Create().
 			SetCpTaskID(taskPayload.CpTaskID).
 			SetCpUserID(targetTask.Edges.CpUser.ID).
-			SetSystemUserID(targetTask.Edges.CpUser.Edges.Sysuser.ID).
+			SetSystemUserID(targetTask.Edges.SysUser.ID).
 			SetIsSuccess(false).
 			SetRealName("").
 			SetErrorMessage(errorMessage).