From f99b8a6d2ec0fc2414b939c3fef4d9338d13fce4 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期三, 12 二月 2025 14:07:07 +0800
Subject: [PATCH] 调整场所分析身份条件

---
 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