53 lines
1.5 KiB
Go
53 lines
1.5 KiB
Go
|
package docker
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
docker "github.com/fsouza/go-dockerclient"
|
||
|
|
"github.com/hashicorp/go-hclog"
|
||
|
|
"github.com/hashicorp/nomad/helper/pluginutils/hclutils"
|
||
|
|
)
|
||
|
|
|
||
|
|
type publishedPorts struct {
|
||
|
|
logger hclog.Logger
|
||
|
|
publishedPorts map[docker.Port][]docker.PortBinding
|
||
|
|
exposedPorts map[docker.Port]struct{}
|
||
|
|
}
|
||
|
|
|
||
|
|
func newPublishedPorts(logger hclog.Logger) *publishedPorts {
|
||
|
|
return &publishedPorts{
|
||
|
|
logger: logger,
|
||
|
|
publishedPorts: map[docker.Port][]docker.PortBinding{},
|
||
|
|
exposedPorts: map[docker.Port]struct{}{},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// adds the port to the structures the Docker API expects for declaring mapped ports
|
||
|
|
func (p *publishedPorts) addMapped(label, ip string, port int, portMap hclutils.MapStrInt) {
|
||
|
|
// By default we will map the allocated port 1:1 to the container
|
||
|
|
containerPortInt := port
|
||
|
|
|
||
|
|
// If the user has mapped a port using port_map we'll change it here
|
||
|
|
if mapped, ok := portMap[label]; ok {
|
||
|
|
containerPortInt = mapped
|
||
|
|
}
|
||
|
|
|
||
|
|
p.add(label, ip, port, containerPortInt)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *publishedPorts) add(label, ip string, port, to int) {
|
||
|
|
if to == 0 {
|
||
|
|
to = port
|
||
|
|
}
|
||
|
|
hostPortStr := strconv.Itoa(port)
|
||
|
|
containerPort := docker.Port(strconv.Itoa(to))
|
||
|
|
|
||
|
|
p.publishedPorts[containerPort+"/tcp"] = getPortBinding(ip, hostPortStr)
|
||
|
|
p.publishedPorts[containerPort+"/udp"] = getPortBinding(ip, hostPortStr)
|
||
|
|
p.logger.Debug("allocated static port", "ip", ip, "port", port)
|
||
|
|
|
||
|
|
p.exposedPorts[containerPort+"/tcp"] = struct{}{}
|
||
|
|
p.exposedPorts[containerPort+"/udp"] = struct{}{}
|
||
|
|
p.logger.Debug("exposed port", "port", port)
|
||
|
|
}
|