forked from TrueCloudLab/frostfs-node
79 lines
1.3 KiB
Go
79 lines
1.3 KiB
Go
package worker
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
)
|
|
|
|
type (
|
|
// Workers is an interface of worker tool.
|
|
Workers interface {
|
|
Start(context.Context)
|
|
Stop()
|
|
|
|
Add(Job Handler)
|
|
}
|
|
|
|
workers struct {
|
|
cancel context.CancelFunc
|
|
started *int32
|
|
wg *sync.WaitGroup
|
|
jobs []Handler
|
|
}
|
|
|
|
// Handler is a worker's handling function.
|
|
Handler func(ctx context.Context)
|
|
|
|
// Jobs is a map of worker names to handlers.
|
|
Jobs map[string]Handler
|
|
|
|
// Job groups the parameters of worker's job.
|
|
Job struct {
|
|
Disabled bool
|
|
Immediately bool
|
|
Timer time.Duration
|
|
Ticker time.Duration
|
|
Handler Handler
|
|
}
|
|
)
|
|
|
|
// New is a constructor of workers.
|
|
func New() Workers {
|
|
return &workers{
|
|
started: new(int32),
|
|
wg: new(sync.WaitGroup),
|
|
}
|
|
}
|
|
|
|
func (w *workers) Add(job Handler) {
|
|
w.jobs = append(w.jobs, job)
|
|
}
|
|
|
|
func (w *workers) Stop() {
|
|
if !atomic.CompareAndSwapInt32(w.started, 1, 0) {
|
|
// already stopped
|
|
return
|
|
}
|
|
|
|
w.cancel()
|
|
w.wg.Wait()
|
|
}
|
|
|
|
func (w *workers) Start(ctx context.Context) {
|
|
if !atomic.CompareAndSwapInt32(w.started, 0, 1) {
|
|
// already started
|
|
return
|
|
}
|
|
|
|
ctx, w.cancel = context.WithCancel(ctx)
|
|
for _, job := range w.jobs {
|
|
w.wg.Add(1)
|
|
|
|
go func(handler Handler) {
|
|
defer w.wg.Done()
|
|
handler(ctx)
|
|
}(job)
|
|
}
|
|
}
|