blob: b9150c43bcf227555bd8670c5d2d7ac49b236539 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
package container
import (
"context"
"sync/atomic"
"github.com/spiral/errors"
"github.com/spiral/roadrunner/v2/pkg/worker"
)
type Vec struct {
destroy uint64
workers chan worker.BaseProcess
}
func NewVector(initialNumOfWorkers uint64) Vector {
vec := &Vec{
destroy: 0,
workers: make(chan worker.BaseProcess, initialNumOfWorkers),
}
return vec
}
func (v *Vec) Enqueue(w worker.BaseProcess) {
v.workers <- w
}
func (v *Vec) Dequeue(ctx context.Context) (worker.BaseProcess, error) {
/*
if *addr == old {
*addr = new
return true
}
*/
if atomic.CompareAndSwapUint64(&v.destroy, 1, 1) {
return nil, errors.E(errors.WatcherStopped)
}
select {
case w := <-v.workers:
return w, nil
case <-ctx.Done():
return nil, errors.E(ctx.Err(), errors.NoFreeWorkers)
}
}
func (v *Vec) Destroy() {
atomic.StoreUint64(&v.destroy, 1)
}
|
