From fca319958029fa924308e50cb61202d7d6ff5008 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 19 二月 2025 13:33:37 +0800
Subject: [PATCH] 暂停聚集模型

---
 models/gather_model.go |   11 +++++++++--
 1 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/models/gather_model.go b/models/gather_model.go
index 3724c0d..d0c2756 100644
--- a/models/gather_model.go
+++ b/models/gather_model.go
@@ -15,6 +15,7 @@
 
 	"model-engine/config"
 	"model-engine/db"
+	"model-engine/pkg/logger"
 	"model-engine/pkg/set"
 	"model-engine/service"
 )
@@ -81,10 +82,10 @@
 		}
 	}
 
-	fmt.Printf("GatherModel init finish ...task id:%s, name:%s, rule:%+v\n", task.ID, task.Name, m)
+	logger.Debugf("GatherModel init finish ...task id:%s, name:%s, rule:%+v\n", task.ID, task.Name, m)
 
 	if m.GatherPersons == 0 || m.AppearInterval == 0 || m.DaysWindow == 0 || m.Threshold == 0 {
-		fmt.Printf("invalid parameters. task id:%s, name:%s\n", task.ID, task.Name)
+		logger.Warnf("invalid parameters. task id:%s, name:%s\n", task.ID, task.Name)
 		return errors.New("invalid parameters")
 	}
 
@@ -184,11 +185,17 @@
 			PicDate:       lt.Time,
 			FirstPersonID: personIds[0],
 		}
+
 		results = append(results, result)
 	}
 	return service.SaveTaskResults(results)
 }
 
+func (m *GatherModel) KeepAlive() error {
+	db.GetDB().Model(m.Task).Where("id = ?", m.Task.ID).Update("last_run_time", time.Now())
+	return nil
+}
+
 func (m *GatherModel) Shutdown() error {
 	// 娓呯悊璧勬簮
 	fmt.Println("Shutting down GatherModel Model")

--
Gitblit v1.8.0