zhangqian
2024-12-12 33a468c0bceff7841abe168a6bc825d6ccf96a6f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
package service
 
import "model-engine/db"
 
func GetTasks() (tasks []*db.ModelTask, err error) {
    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()
}