blob: 565b1b694c3d84c6401ad4aee5239e0aad0b7de9 (
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
|
package container
import (
"sync/atomic"
"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() (worker.BaseProcess, bool) {
/*
if *addr == old {
*addr = new
return true
}
*/
if atomic.CompareAndSwapUint64(&v.destroy, 1, 1) {
return nil, true
}
return <-v.workers, false
}
func (v *Vec) Destroy() {
atomic.StoreUint64(&v.destroy, 1)
}
|
