mirror of https://github.com/stashapp/stash.git
53 lines
1.0 KiB
Go
53 lines
1.0 KiB
Go
|
package api
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
"github.com/stashapp/stash/pkg/job"
|
||
|
|
"github.com/stashapp/stash/pkg/manager"
|
||
|
|
"github.com/stashapp/stash/pkg/models"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (r *queryResolver) JobQueue(ctx context.Context) ([]*models.Job, error) {
|
||
|
|
queue := manager.GetInstance().JobManager.GetQueue()
|
||
|
|
|
||
|
|
var ret []*models.Job
|
||
|
|
for _, j := range queue {
|
||
|
|
ret = append(ret, jobToJobModel(j))
|
||
|
|
}
|
||
|
|
|
||
|
|
return ret, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *queryResolver) FindJob(ctx context.Context, input models.FindJobInput) (*models.Job, error) {
|
||
|
|
jobID, err := strconv.Atoi(input.ID)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
j := manager.GetInstance().JobManager.GetJob(jobID)
|
||
|
|
if j == nil {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return jobToJobModel(*j), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func jobToJobModel(j job.Job) *models.Job {
|
||
|
|
ret := &models.Job{
|
||
|
|
ID: strconv.Itoa(j.ID),
|
||
|
|
Status: models.JobStatus(j.Status),
|
||
|
|
Description: j.Description,
|
||
|
|
SubTasks: j.Details,
|
||
|
|
StartTime: j.StartTime,
|
||
|
|
EndTime: j.EndTime,
|
||
|
|
AddTime: j.AddTime,
|
||
|
|
}
|
||
|
|
|
||
|
|
if j.Progress != -1 {
|
||
|
|
ret.Progress = &j.Progress
|
||
|
|
}
|
||
|
|
|
||
|
|
return ret
|
||
|
|
}
|