From 3c033759200ad7c02dd59521b1aebbbdc35b98fa Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期六, 08 二月 2025 16:18:50 +0800 Subject: [PATCH] 长场景人员分析模型 --- models/gather_model.go | 9 ++++++++- 1 files changed, 8 insertions(+), 1 deletions(-) diff --git a/models/gather_model.go b/models/gather_model.go index c576627..59fe8a4 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,9 +82,10 @@ } } - fmt.Printf("GatherModel init finish ...task id:%s, name:%s, rule:%+v\n", task.ID, task.Name, task.Rules) + 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 { + logger.Warnf("invalid parameters. task id:%s, name:%s\n", task.ID, task.Name) return errors.New("invalid parameters") } @@ -188,6 +190,11 @@ 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