From e4fc307d58856a0b319c41f0930cd3b731cb0b24 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期二, 11 二月 2025 10:19:47 +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