From bc37c44c45d08c2cb42822e08adf429f4bcffb51 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期一, 10 二月 2025 11:36:20 +0800
Subject: [PATCH] 测试 场景分析 bug

---
 service/task_results.go |   25 +++++++++++++++++++++++--
 1 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/service/task_results.go b/service/task_results.go
index bdeef8f..2d911af 100644
--- a/service/task_results.go
+++ b/service/task_results.go
@@ -1,7 +1,28 @@
 package service
 
-import "model-engine/db"
+import (
+	"model-engine/db"
+	"model-engine/pkg/logger"
+)
 
 func SaveTaskResults(results []*db.ModelTaskResults) error {
-	return db.NewModelTaskResultsSearch().BatchCreate(results)
+	for _, v := range results {
+		err := db.NewModelTaskResultsSearch().SetTaskId(v.ModelTaskID).SetPicDate(v.PicDate).SetFirstPersonId(v.FirstPersonID).FirstOrCreate(v)
+		if err != nil {
+			logger.Errorf("save task result err:%v", err)
+			return err
+		}
+	}
+	return nil
+}
+
+func SaveLocationTaskResults(results []*db.ModelTaskResults) error {
+	for _, v := range results {
+		err := db.NewModelTaskResultsSearch().SetTaskId(v.ModelTaskID).SetFirstLocation(v.Location).FirstOrCreate(v)
+		if err != nil {
+			logger.Errorf("save task result err:%v", err)
+			return err
+		}
+	}
+	return nil
 }

--
Gitblit v1.8.0