From fca319958029fa924308e50cb61202d7d6ff5008 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 19 二月 2025 13:33:37 +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