-
Notifications
You must be signed in to change notification settings - Fork 51
/
allocator.go
58 lines (46 loc) · 995 Bytes
/
allocator.go
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
52
53
54
55
56
57
58
package allocator
import (
"strconv"
)
// allocator
type allocator struct {
allocate chan int
}
// New provides a new allocator
func New(start, size int) Allocator {
a := &allocator{
allocate: make(chan int, size),
}
for i := start; i < (start + size); i++ {
a.allocate <- i
}
return a
}
// Allocate allocates an item
func (p *allocator) Allocate() string {
return strconv.Itoa(<-p.allocate)
}
// Release releases an item
func (p *allocator) Release(item string) {
// Do not release when the channel is full. These can happen when we resync
// stopped containers.
if len(p.allocate) == cap(p.allocate) {
return
}
intItem, err := strconv.Atoi(item)
if err != nil {
return
}
p.allocate <- intItem
}
// AllocateInt allocates an integer.
func (p *allocator) AllocateInt() int {
return <-p.allocate
}
// ReleaseInt releases an int.
func (p *allocator) ReleaseInt(item int) {
if len(p.allocate) == cap(p.allocate) || item == 0 {
return
}
p.allocate <- item
}