Merge pull request #960 from weaveworks/no-ips-for-no-network

Don't show blank IPs metadata row for containers with no IP

render/topologies.go

@@ -2,8 +2,10 @@ package render
 
 import (
 	"fmt"
+	"net"
 
 	"github.com/weaveworks/scope/probe/docker"
+	"github.com/weaveworks/scope/probe/host"
 	"github.com/weaveworks/scope/probe/process"
 	"github.com/weaveworks/scope/report"
 )
@@ -116,6 +118,49 @@ var ContainerRenderer = MakeReduce(
 	),
 )
 
+type containerWithHostIPsRenderer struct {
+	Renderer
+}
+
+// Render produces a process graph where the ips for host network mode are set
+// to the host's IPs.
+func (r containerWithHostIPsRenderer) Render(rpt report.Report) RenderableNodes {
+	containers := r.Renderer.Render(rpt)
+	hosts := MakeMap(
+		MapHostIdentity,
+		SelectHost,
+	).Render(rpt)
+
+	for id, c := range containers {
+		networkMode, ok := c.Node.Latest.Lookup(docker.ContainerNetworkMode)
+		if !ok || networkMode != docker.NetworkModeHost {
+			continue
+		}
+
+		h, ok := hosts[MakeHostID(report.ExtractHostID(c.Node))]
+		if !ok {
+			continue
+		}
+
+		newIPs := report.MakeStringSet()
+		hostNetworks, _ := h.Sets.Lookup(host.LocalNetworks)
+		for _, cidr := range hostNetworks {
+			if ip, _, err := net.ParseCIDR(cidr); err == nil {
+				newIPs = newIPs.Add(ip.String())
+			}
+		}
+
+		c.Sets = c.Sets.Add(docker.ContainerIPs, newIPs)
+		containers[id] = c
+	}
+
+	return containers
+}
+
+// ContainerWithHostIPsRenderer is a Renderer which produces a container graph
+// enriched with host IPs on containers where NetworkMode is Host
+var ContainerWithHostIPsRenderer = containerWithHostIPsRenderer{ContainerRenderer}
+
 type containerWithImageNameRenderer struct {
 	Renderer
 }
@@ -149,7 +194,7 @@ func (r containerWithImageNameRenderer) Render(rpt report.Report) RenderableNode
 
 // ContainerWithImageNameRenderer is a Renderer which produces a container
 // graph where the ranks are the image names, not their IDs
-var ContainerWithImageNameRenderer = containerWithImageNameRenderer{ContainerRenderer}
+var ContainerWithImageNameRenderer = containerWithImageNameRenderer{ContainerWithHostIPsRenderer}
 
 // ContainerImageRenderer is a Renderer which produces a renderable container
 // image graph by merging the container graph and the container image topology.

render/topologies_test.go

@@ -8,6 +8,7 @@ import (
 	"github.com/weaveworks/scope/probe/kubernetes"
 	"github.com/weaveworks/scope/render"
 	"github.com/weaveworks/scope/render/expected"
+	"github.com/weaveworks/scope/report"
 	"github.com/weaveworks/scope/test"
 	"github.com/weaveworks/scope/test/fixture"
 )
@@ -51,6 +52,25 @@ func TestContainerFilterRenderer(t *testing.T) {
 	}
 }
 
+func TestContainerWithHostIPsRenderer(t *testing.T) {
+	input := fixture.Report.Copy()
+	input.Container.Nodes[fixture.ClientContainerNodeID] = input.Container.Nodes[fixture.ClientContainerNodeID].WithLatests(map[string]string{
+		docker.ContainerNetworkMode: "host",
+	})
+	nodes := render.ContainerWithHostIPsRenderer.Render(input)
+
+	// Test host network nodes get the host IPs added.
+	haveNode, ok := nodes[render.MakeContainerID(fixture.ClientContainerID)]
+	if !ok {
+		t.Fatal("Expected output to have the client container node")
+	}
+	have, _ := haveNode.Sets.Lookup(docker.ContainerIPs)
+	want := report.MakeStringSet("10.10.10.0")
+	if !reflect.DeepEqual(want, have) {
+		t.Error(test.Diff(want, have))
+	}
+}
+
 func TestContainerFilterRendererImageName(t *testing.T) {
 	// Test nodes are filtered by image name as well.
 	input := fixture.Report.Copy()