Backup_Repos/stash
1
0
Fork 0
mirror of https://github.com/stashapp/stash.git synced 2026-04-18 04:53:57 +02:00
Code Issues Projects Releases Wiki Activity

Integrate progress into TaskQueue

Browse source
This commit is contained in:
WithoutPants 2026-03-16 17:37:06 +11:00
parent 17aa605574
commit 2f230e5897
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
pkg/job/task.go
View file

@ -32,6 +32,7 @@ func NewTaskQueue(ctx context.Context, p *Progress, queueSize int, processes int
}
func (tq *TaskQueue) Add(description string, fn func(ctx context.Context)) {
tq.p.AddTotal(1)
tq.tasks <- taskExec{
task: task{
description: description,
@ -42,6 +43,10 @@ func (tq *TaskQueue) Add(description string, fn func(ctx context.Context)) {
func (tq *TaskQueue) Close() {
close(tq.tasks)
tq.p.Definite()
}
func (tq *TaskQueue) Wait() {
// wait for all tasks to finish
<-tq.done
}
@ -59,6 +64,7 @@ func (tq *TaskQueue) executer(ctx context.Context) {
tq.wg.Add()
go func() {
defer tq.wg.Done()
defer tq.p.Increment()
tq.p.ExecuteTask(tt.description, func() {
tt.fn(ctx)
})