24 lines
1.0 KiB
Go
24 lines
1.0 KiB
Go
package repository
|
|
|
|
import (
|
|
"yingce/backend/internal/model"
|
|
)
|
|
|
|
type TaskPage struct {
|
|
Tasks []model.Task `json:"tasks"`
|
|
NextCursor string `json:"nextCursor,omitempty"`
|
|
}
|
|
|
|
func (r *Repository) WorkbenchTasks(userID, kind string, activeOnly bool, cursor string, limit int) (TaskPage, error) {
|
|
if limit < 1 || limit > 100 { limit = 30 }
|
|
if kind != model.WorkbenchKindImage && kind != model.WorkbenchKindVideo { return TaskPage{}, nil }
|
|
query := r.db.Model(&model.Task{}).Where("user_id = ? AND workbench_kind = ?", userID, kind)
|
|
if activeOnly { query = query.Where("status IN ?", []model.TaskStatus{model.TaskStatusQueued, model.TaskStatusRunning}) }
|
|
if cursor != "" { query = query.Where("id < ?", cursor) }
|
|
var tasks []model.Task
|
|
if err := query.Order("created_at desc, id desc").Limit(limit + 1).Find(&tasks).Error; err != nil { return TaskPage{}, err }
|
|
page := TaskPage{Tasks: tasks}
|
|
if len(tasks) > limit { page.Tasks = tasks[:limit]; page.NextCursor = page.Tasks[len(page.Tasks)-1].ID }
|
|
return page, nil
|
|
}
|