-
Notifications
You must be signed in to change notification settings - Fork 70
/
port_allocator.go
91 lines (74 loc) · 2.04 KB
/
port_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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package port_allocator
import (
"errors"
"fmt"
"lib/serial"
"code.cloudfoundry.org/filelock"
)
//go:generate counterfeiter -o ../fakes/file_locker.go --fake-name FileLocker . FileLocker
type FileLocker interface {
filelock.FileLocker
}
//go:generate counterfeiter -o ../fakes/locked_file.go --fake-name LockedFile . LockedFile
type LockedFile interface {
filelock.LockedFile
}
//go:generate counterfeiter -o ../fakes/tracker.go --fake-name Tracker . tracker
type tracker interface {
AcquireOne(pool *Pool, handle string) (int, error)
ReleaseAll(pool *Pool, handle string) error
InRange(port int) bool
}
type PortAllocator struct {
Tracker tracker
Serializer serial.Serializer
Locker filelock.FileLocker
}
func (p *PortAllocator) AllocatePort(handle string, port int) (int, error) {
if port != 0 {
if p.Tracker.InRange(port) {
return -1, errors.New("cannot specify port from allocation range")
} else {
return port, nil
}
}
file, err := p.Locker.Open()
if err != nil {
return -1, fmt.Errorf("open lock: %s", err)
}
defer file.Close() // defer not tested
pool := &Pool{}
err = p.Serializer.DecodeAll(file, pool)
if err != nil {
return -1, fmt.Errorf("decoding state file: %s", err)
}
newPort, err := p.Tracker.AcquireOne(pool, handle)
if err != nil {
return -1, fmt.Errorf("acquire port: %s", err)
}
err = p.Serializer.EncodeAndOverwrite(file, pool)
if err != nil {
return -1, fmt.Errorf("encode and overwrite: %s", err)
}
return newPort, nil
}
func (p *PortAllocator) ReleaseAllPorts(handle string) error {
file, err := p.Locker.Open()
if err != nil {
return fmt.Errorf("open lock: %s", err)
}
defer file.Close() // defer not tested
pool := &Pool{}
err = p.Serializer.DecodeAll(file, pool)
if err != nil {
return fmt.Errorf("decoding state file: %s", err)
}
if err := p.Tracker.ReleaseAll(pool, handle); err != nil {
return fmt.Errorf("release all ports: %s", err)
}
err = p.Serializer.EncodeAndOverwrite(file, pool)
if err != nil {
return fmt.Errorf("encode and overwrite: %s", err)
}
return nil
}