mirror of
https://github.com/NginxProxyManager/nginx-proxy-manager.git
synced 2025-04-29 02:22:28 +00:00
61 lines
1.1 KiB
Go
61 lines
1.1 KiB
Go
package jobqueue
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
)
|
|
|
|
// Queue holds name, list of jobs and context with cancel.
|
|
type Queue struct {
|
|
jobs chan Job
|
|
ctx context.Context
|
|
cancel context.CancelFunc
|
|
mu sync.Mutex
|
|
}
|
|
|
|
// Job - holds logic to perform some operations during queue execution.
|
|
type Job struct {
|
|
Name string
|
|
Action func() error // A function that should be executed when the job is running.
|
|
Done chan bool // A channel that should be closed when the job is done.
|
|
}
|
|
|
|
// AddJobs adds jobs to the queue and cancels channel.
|
|
func (q *Queue) AddJobs(jobs []Job) {
|
|
var wg sync.WaitGroup
|
|
wg.Add(len(jobs))
|
|
|
|
for _, job := range jobs {
|
|
// Goroutine which adds job to the queue.
|
|
go func(job Job) {
|
|
q.AddJob(job)
|
|
wg.Done()
|
|
}(job)
|
|
}
|
|
|
|
go func() {
|
|
wg.Wait()
|
|
// Cancel queue channel, when all goroutines were done.
|
|
q.cancel()
|
|
}()
|
|
}
|
|
|
|
// AddJob sends job to the channel.
|
|
func (q *Queue) AddJob(job Job) {
|
|
q.mu.Lock()
|
|
defer q.mu.Unlock()
|
|
q.jobs <- job
|
|
}
|
|
|
|
// Run performs job execution.
|
|
func (j *Job) Run() error {
|
|
err := j.Action()
|
|
if err != nil {
|
|
j.Done <- true
|
|
return err
|
|
}
|
|
|
|
j.Done <- true
|
|
return nil
|
|
}
|