From fca319958029fa924308e50cb61202d7d6ff5008 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 19 二月 2025 13:33:37 +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