添加任务运行状态更新

zhangzengfei

2024-12-20 ead55fe84e115e08423ebe1d2bdf28ae5d2cac49

添加任务运行状态更新

 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")

 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")

 models/model.go

@@ -9,6 +9,7 @@
type Model interface {
    Init(task *db.ModelTask) error
    Run() error
    KeepAlive() error
    Shutdown() error
}