From da4c542724604c082d6d76115ada83ea607d77b2 Mon Sep 17 00:00:00 2001 From: TortleWortle Date: Wed, 2 Dec 2020 14:27:37 +0100 Subject: [PATCH] Update allocations.go --- environment/allocations.go | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/environment/allocations.go b/environment/allocations.go index d301e30..b741f8a 100644 --- a/environment/allocations.go +++ b/environment/allocations.go @@ -43,17 +43,11 @@ func (a *Allocations) Bindings() nat.PortMap { HostPort: strconv.Itoa(port), } - if v, ok := out[nat.Port(fmt.Sprintf("%d/tcp", port))]; ok { - out[nat.Port(fmt.Sprintf("%d/tcp", port))] = append(v, binding) - } else { - out[nat.Port(fmt.Sprintf("%d/tcp", port))] = []nat.PortBinding{binding} - } + tcp := nat.Port(fmt.Sprintf("%d/tcp", port)) + udp := nat.Port(fmt.Sprintf("%d/udp", port)) - if v, ok := out[nat.Port(fmt.Sprintf("%d/udp", port))]; ok { - out[nat.Port(fmt.Sprintf("%d/udp", port))] = append(v, binding) - } else { - out[nat.Port(fmt.Sprintf("%d/udp", port))] = []nat.PortBinding{binding} - } + out[tcp] = append(out[tcp], binding) + out[udp] = append(out[udp], binding) } }