查询模型任务时只查询启用的模型

zhangqian

2024-12-12 33a468c0bceff7841abe168a6bc825d6ccf96a6f

 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)