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

zhangqian

2024-12-12 33a468c0bceff7841abe168a6bc825d6ccf96a6f

 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()
}