Update traefik dependencies (docker/docker and related) - Update dependencies - Fix compilation problems - Remove vdemeester/docker-events (in docker api now) - Remove `integration/vendor` - Use `testImport` - update some deps. - regenerate the lock from scratch (after a `glide cc`)
179 lines
3.9 KiB
Go
179 lines
3.9 KiB
Go
package docker
|
|
|
|
import (
|
|
docker "github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/api/types/network"
|
|
"github.com/docker/docker/api/types/swarm"
|
|
"github.com/docker/go-connections/nat"
|
|
)
|
|
|
|
func containerJSON(ops ...func(*docker.ContainerJSON)) docker.ContainerJSON {
|
|
c := &docker.ContainerJSON{
|
|
ContainerJSONBase: &docker.ContainerJSONBase{
|
|
Name: "fake",
|
|
HostConfig: &container.HostConfig{},
|
|
},
|
|
Config: &container.Config{},
|
|
NetworkSettings: &docker.NetworkSettings{
|
|
NetworkSettingsBase: docker.NetworkSettingsBase{},
|
|
},
|
|
}
|
|
|
|
for _, op := range ops {
|
|
op(c)
|
|
}
|
|
|
|
return *c
|
|
}
|
|
|
|
func name(name string) func(*docker.ContainerJSON) {
|
|
return func(c *docker.ContainerJSON) {
|
|
c.ContainerJSONBase.Name = name
|
|
}
|
|
}
|
|
|
|
func networkMode(mode string) func(*docker.ContainerJSON) {
|
|
return func(c *docker.ContainerJSON) {
|
|
c.ContainerJSONBase.HostConfig.NetworkMode = container.NetworkMode(mode)
|
|
}
|
|
}
|
|
|
|
func labels(labels map[string]string) func(*docker.ContainerJSON) {
|
|
return func(c *docker.ContainerJSON) {
|
|
c.Config.Labels = labels
|
|
}
|
|
}
|
|
|
|
func ports(portMap nat.PortMap) func(*docker.ContainerJSON) {
|
|
return func(c *docker.ContainerJSON) {
|
|
c.NetworkSettings.NetworkSettingsBase.Ports = portMap
|
|
}
|
|
}
|
|
|
|
func withNetwork(name string, ops ...func(*network.EndpointSettings)) func(*docker.ContainerJSON) {
|
|
return func(c *docker.ContainerJSON) {
|
|
if c.NetworkSettings.Networks == nil {
|
|
c.NetworkSettings.Networks = map[string]*network.EndpointSettings{}
|
|
}
|
|
c.NetworkSettings.Networks[name] = &network.EndpointSettings{}
|
|
for _, op := range ops {
|
|
op(c.NetworkSettings.Networks[name])
|
|
}
|
|
}
|
|
}
|
|
|
|
func ipv4(ip string) func(*network.EndpointSettings) {
|
|
return func(s *network.EndpointSettings) {
|
|
s.IPAddress = ip
|
|
}
|
|
}
|
|
|
|
func swarmTask(id string, ops ...func(*swarm.Task)) swarm.Task {
|
|
task := &swarm.Task{
|
|
ID: id,
|
|
}
|
|
|
|
for _, op := range ops {
|
|
op(task)
|
|
}
|
|
|
|
return *task
|
|
}
|
|
|
|
func taskSlot(slot int) func(*swarm.Task) {
|
|
return func(task *swarm.Task) {
|
|
task.Slot = slot
|
|
}
|
|
}
|
|
|
|
func taskStatus(ops ...func(*swarm.TaskStatus)) func(*swarm.Task) {
|
|
return func(task *swarm.Task) {
|
|
status := &swarm.TaskStatus{}
|
|
|
|
for _, op := range ops {
|
|
op(status)
|
|
}
|
|
|
|
task.Status = *status
|
|
}
|
|
}
|
|
|
|
func taskState(state swarm.TaskState) func(*swarm.TaskStatus) {
|
|
return func(status *swarm.TaskStatus) {
|
|
status.State = state
|
|
}
|
|
}
|
|
|
|
func swarmService(ops ...func(*swarm.Service)) swarm.Service {
|
|
service := &swarm.Service{
|
|
ID: "serviceID",
|
|
Spec: swarm.ServiceSpec{
|
|
Annotations: swarm.Annotations{
|
|
Name: "defaultServiceName",
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, op := range ops {
|
|
op(service)
|
|
}
|
|
|
|
return *service
|
|
}
|
|
|
|
func serviceName(name string) func(service *swarm.Service) {
|
|
return func(service *swarm.Service) {
|
|
service.Spec.Annotations.Name = name
|
|
}
|
|
}
|
|
|
|
func serviceLabels(labels map[string]string) func(service *swarm.Service) {
|
|
return func(service *swarm.Service) {
|
|
service.Spec.Annotations.Labels = labels
|
|
}
|
|
}
|
|
|
|
func withEndpoint(ops ...func(*swarm.Endpoint)) func(*swarm.Service) {
|
|
return func(service *swarm.Service) {
|
|
endpoint := &swarm.Endpoint{}
|
|
|
|
for _, op := range ops {
|
|
op(endpoint)
|
|
}
|
|
|
|
service.Endpoint = *endpoint
|
|
}
|
|
}
|
|
|
|
func virtualIP(networkID, addr string) func(*swarm.Endpoint) {
|
|
return func(endpoint *swarm.Endpoint) {
|
|
if endpoint.VirtualIPs == nil {
|
|
endpoint.VirtualIPs = []swarm.EndpointVirtualIP{}
|
|
}
|
|
endpoint.VirtualIPs = append(endpoint.VirtualIPs, swarm.EndpointVirtualIP{
|
|
NetworkID: networkID,
|
|
Addr: addr,
|
|
})
|
|
}
|
|
}
|
|
|
|
func withEndpointSpec(ops ...func(*swarm.EndpointSpec)) func(*swarm.Service) {
|
|
return func(service *swarm.Service) {
|
|
endpointSpec := &swarm.EndpointSpec{}
|
|
|
|
for _, op := range ops {
|
|
op(endpointSpec)
|
|
}
|
|
|
|
service.Spec.EndpointSpec = endpointSpec
|
|
}
|
|
}
|
|
|
|
func modeDNSSR(spec *swarm.EndpointSpec) {
|
|
spec.Mode = swarm.ResolutionModeDNSRR
|
|
}
|
|
|
|
func modeVIP(spec *swarm.EndpointSpec) {
|
|
spec.Mode = swarm.ResolutionModeVIP
|
|
}
|