frostfs-node/pkg/util/worker_pool.go
Evgenii Stratonikov 0beaed2ef4 [] util: implement Release for PseudWorkerPool
Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
2021-10-25 18:53:36 +03:00

54 lines
1.3 KiB
Go

package util
import (
"github.com/panjf2000/ants/v2"
"go.uber.org/atomic"
)
// WorkerPool represents the tool for control
// the execution of go-routine pool.
type WorkerPool interface {
// Submit queues a function for execution
// in a separate routine.
//
// Implementation must return any error encountered
// that prevented the function from being queued.
Submit(func()) error
// Release releases worker pool resources. All `Submit` calls will
// finish with ErrPoolClosed. It doesn't wait until all submitted
// functions have returned so synchronization must be achieved
// via other means (e.g. sync.WaitGroup).
Release()
}
// pseudoWorkerPool represents pseudo worker pool which executes submitted job immediately in the caller's routine.
type pseudoWorkerPool struct {
closed atomic.Bool
}
// ErrPoolClosed is returned when submitting task to a closed pool.
var ErrPoolClosed = ants.ErrPoolClosed
// NewPseudoWorkerPool returns new instance of a synchronous worker pool.
func NewPseudoWorkerPool() WorkerPool {
return &pseudoWorkerPool{}
}
// Submit executes passed function immediately.
//
// Always returns nil.
func (p *pseudoWorkerPool) Submit(fn func()) error {
if p.closed.Load() {
return ErrPoolClosed
}
fn()
return nil
}
// Release implements WorkerPool interface.
func (p *pseudoWorkerPool) Release() {
p.closed.Store(true)
}