From 33a468c0bceff7841abe168a6bc825d6ccf96a6f Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期四, 12 十二月 2024 14:25:28 +0800
Subject: [PATCH] 查询模型任务时只查询启用的模型
---
db/task.go | 10 ++++++++++
1 files changed, 10 insertions(+), 0 deletions(-)
diff --git a/db/task.go b/db/task.go
index 573558c..7f21a2a 100644
--- a/db/task.go
+++ b/db/task.go
@@ -32,6 +32,7 @@
PageNum int
PageSize int
Keyword string
+ ModelIDs []string
}
)
@@ -102,6 +103,11 @@
return slf
}
+func (slf *ModelTaskSearch) SetModelIDs(ids []string) *ModelTaskSearch {
+ slf.ModelIDs = ids
+ return slf
+}
+
func (slf *ModelTaskSearch) SetKeyword(kw string) *ModelTaskSearch {
slf.Keyword = kw
return slf
@@ -123,6 +129,10 @@
db = db.Where("model_id = ?", slf.ModelID)
}
+ if len(slf.ModelIDs) != 0 {
+ db = db.Where("model_id in ?", slf.ModelIDs)
+ }
+
if slf.Keyword != "" {
kw := "%" + slf.Keyword + "%"
db = db.Where("name like ?", kw)
--
Gitblit v1.8.0