From 51aa4c73cdd577d96f7cf99278d22bab5d07eff5 Mon Sep 17 00:00:00 2001 From: Tortle Date: Tue, 1 Dec 2020 21:20:01 +0100 Subject: [PATCH] Update allocations.go --- environment/allocations.go | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/environment/allocations.go b/environment/allocations.go index 6e25f69..cb8aad7 100644 --- a/environment/allocations.go +++ b/environment/allocations.go @@ -43,17 +43,13 @@ func (a *Allocations) Bindings() nat.PortMap { HostPort: strconv.Itoa(port), } - tcpPort, ok := out[nat.Port(fmt.Sprintf("%d/tcp", port))] - - if !ok { + if tcpPort, ok := out[nat.Port(fmt.Sprintf("%d/tcp", port))]; !ok { out[nat.Port(fmt.Sprintf("%d/tcp", port))] = []nat.PortBinding{binding} } else { out[nat.Port(fmt.Sprintf("%d/tcp", port))] = append(tcpPort, binding) } - udpPort, ok := out[nat.Port(fmt.Sprintf("%d/udp", port))] - - if !ok { + if udpPort, ok := out[nat.Port(fmt.Sprintf("%d/udp", port))]; !ok { out[nat.Port(fmt.Sprintf("%d/udp", port))] = []nat.PortBinding{binding} } else { out[nat.Port(fmt.Sprintf("%d/udp", port))] = append(udpPort, binding)