From ead55fe84e115e08423ebe1d2bdf28ae5d2cac49 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期五, 20 十二月 2024 16:17:46 +0800
Subject: [PATCH] 添加任务运行状态更新
---
models/gather_model.go | 5 +++++
models/disappear.go | 5 +++++
models/model.go | 1 +
3 files changed, 11 insertions(+), 0 deletions(-)
diff --git a/models/disappear.go b/models/disappear.go
index 08ceda2..542bd0f 100644
--- a/models/disappear.go
+++ b/models/disappear.go
@@ -243,6 +243,11 @@
return service.SaveTaskResults(results)
}
+func (m *DisappearModel) KeepAlive() error {
+ db.GetDB().Model(m.Task).Where("id = ?", m.Task.ID).Update("last_run_time", time.Now())
+ return nil
+}
+
func (m *DisappearModel) Shutdown() error {
// 娓呯悊璧勬簮
fmt.Println("Shutting down Disappear Model")
diff --git a/models/gather_model.go b/models/gather_model.go
index 64aa815..59fe8a4 100644
--- a/models/gather_model.go
+++ b/models/gather_model.go
@@ -190,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")
diff --git a/models/model.go b/models/model.go
index a0567ab..5a0e98e 100644
--- a/models/model.go
+++ b/models/model.go
@@ -9,6 +9,7 @@
type Model interface {
Init(task *db.ModelTask) error
Run() error
+ KeepAlive() error
Shutdown() error
}
--
Gitblit v1.8.0