zhangqian
2024-12-13 ccd7a3175cd160275955636673b5c4c43d9691a3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
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).
        SetEnabled(true).
        SetUnexpired().
        FindAll()
}