naive fix for #25 (#26)

Thanks!
kevinjqiu · Jun 30, 2022 · 5660bed · 5660bed
1 parent af45ca7
commit 5660bed
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/dockerdiscovery.go b/dockerdiscovery.go
@@ -145,10 +145,18 @@ func (dd *DockerDiscovery) getContainerAddress(container *dockerapi.Container) (
 		}
 	}
 
-	network, ok := container.NetworkSettings.Networks[networkMode]
+	var (
+		network dockerapi.ContainerNetwork
+		ok      = false
+	)
+
 	if hasNetName {
 		log.Printf("[docker] network name %s specified (%s)", netName, container.ID[:12])
 		network, ok = container.NetworkSettings.Networks[netName]
+	} else if len(container.NetworkSettings.Networks) == 1 {
+		for netName, network = range container.NetworkSettings.Networks {
+			ok = true
+		}
 	}
 
 	if !ok { // sometime while "network:disconnect" event fire