From 33a468c0bceff7841abe168a6bc825d6ccf96a6f Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 12 十二月 2024 14:25:28 +0800
Subject: [PATCH] 查询模型任务时只查询启用的模型
---
service/task.go | 15 ++++++++++++++-
1 files changed, 14 insertions(+), 1 deletions(-)
diff --git a/service/task.go b/service/task.go
index b0afcfb..f47d7d2 100644
--- a/service/task.go
+++ b/service/task.go
@@ -3,5 +3,18 @@
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).FindAll()
}
--
Gitblit v1.8.0