From 3c033759200ad7c02dd59521b1aebbbdc35b98fa Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期六, 08 二月 2025 16:18:50 +0800 Subject: [PATCH] 长场景人员分析模型 --- service/task.go | 19 ++++++++++++++++++- 1 files changed, 18 insertions(+), 1 deletions(-) diff --git a/service/task.go b/service/task.go index b0afcfb..449970a 100644 --- a/service/task.go +++ b/service/task.go @@ -3,5 +3,22 @@ import "model-engine/db" func GetTasks() (tasks []*db.ModelTask, err error) { - return db.NewModelTaskSearch().FindAll() + models, err := db.NewModelSearch().SetEnabled(true).FindAll() + if err != nil { + return nil, err + } + if len(models) == 0 { + return nil, nil + } + + modelIds := make([]string, 0, len(models)) + for _, model := range models { + modelIds = append(modelIds, model.ID) + } + + return db.NewModelTaskSearch(). + SetModelIDs(modelIds). + SetEnabled(true). + SetUnexpired(). + FindAll() } -- Gitblit v1.8.0