forked from remind101/empire
/
ports.go
49 lines (41 loc) · 1.11 KB
/
ports.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
package cloudformation
import (
"fmt"
"strconv"
"github.com/remind101/empire/pkg/cloudformation/customresources"
"golang.org/x/net/context"
)
type portAllocator interface {
Get() (int64, error)
Put(port int64) error
}
// InstancePortsProvisioner is a Provisioner that allocates instance ports.
type InstancePortsProvisioner struct {
ports portAllocator
}
func (p *InstancePortsProvisioner) Properties() interface{} {
return nil
}
func (p *InstancePortsProvisioner) Provision(_ context.Context, req customresources.Request) (id string, data interface{}, err error) {
switch req.RequestType {
case customresources.Create:
var port int64
port, err = p.ports.Get()
if err != nil {
return
}
id = fmt.Sprintf("%d", port)
data = map[string]int64{"InstancePort": port}
case customresources.Delete:
port, err2 := strconv.Atoi(req.PhysicalResourceId)
if err2 != nil {
err = fmt.Errorf("physical resource id should have been a port number: %v", err2)
return
}
id = req.PhysicalResourceId
err = p.ports.Put(int64(port))
default:
err = fmt.Errorf("%s is not supported", req.RequestType)
}
return
}