Add "# Containers" column to container images as children

Side effect is that it also adds it to the container image details panel
weaveworks · Feb 5, 2016 · 8770d56 · 8770d56
1 parent 2ed3968
commit 8770d56
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 10 deletions.
diff --git a/render/detailed/labels.go b/render/detailed/labels.go
@@ -8,6 +8,7 @@ import (
 	"github.com/weaveworks/scope/probe/kubernetes"
 	"github.com/weaveworks/scope/probe/overlay"
 	"github.com/weaveworks/scope/probe/process"
+	"github.com/weaveworks/scope/render"
 )
 
 var labels = map[string]string{
@@ -44,6 +45,7 @@ var labels = map[string]string{
 	process.PID:                  "PID",
 	process.PPID:                 "Parent PID",
 	process.Threads:              "# Threads",
+	render.ContainersKey:         "# Containers",
 }
 
 // Label maps from the internal keys to the human-readable label for a piece

diff --git a/render/detailed/metadata.go b/render/detailed/metadata.go
@@ -2,13 +2,15 @@ package detailed
 
 import (
 	"encoding/json"
+	"strconv"
 	"strings"
 
 	"github.com/weaveworks/scope/probe/docker"
 	"github.com/weaveworks/scope/probe/host"
 	"github.com/weaveworks/scope/probe/kubernetes"
 	"github.com/weaveworks/scope/probe/overlay"
 	"github.com/weaveworks/scope/probe/process"
+	"github.com/weaveworks/scope/render"
 	"github.com/weaveworks/scope/report"
 )
 
@@ -34,6 +36,7 @@ var (
 	)
 	containerImageNodeMetadata = renderMetadata(
 		ltst(docker.ImageID),
+		counter(render.ContainersKey),
 	)
 	podNodeMetadata = renderMetadata(
 		ltst(kubernetes.PodID),
@@ -120,3 +123,12 @@ func ltst(id string) func(report.Node) []MetadataRow {
 		return nil
 	}
 }
+
+func counter(id string) func(report.Node) []MetadataRow {
+	return func(n report.Node) []MetadataRow {
+		if val, ok := n.Counters.Lookup(id); ok {
+			return []MetadataRow{{ID: id, Value: strconv.Itoa(val)}}
+		}
+		return nil
+	}
+}
diff --git a/render/detailed/node.go b/render/detailed/node.go
@@ -93,7 +93,7 @@ var (
 	}{
 		{report.Host, NodeSummaryGroup{TopologyID: "hosts", Label: "Hosts", Columns: []string{host.CPUUsage, host.MemoryUsage}}},
 		{report.Pod, NodeSummaryGroup{TopologyID: "pods", Label: "Pods", Columns: []string{}}},
-		{report.ContainerImage, NodeSummaryGroup{TopologyID: "containers-by-image", Label: "Container Images", Columns: []string{}}},
+		{report.ContainerImage, NodeSummaryGroup{TopologyID: "containers-by-image", Label: "Container Images", Columns: []string{render.ContainersKey}}},
 		{report.Container, NodeSummaryGroup{TopologyID: "containers", Label: "Containers", Columns: []string{docker.CPUTotalUsage, docker.MemoryUsage}}},
 		{report.Process, NodeSummaryGroup{TopologyID: "processes", Label: "Processes", Columns: []string{process.PID, process.CPUUsage, process.MemoryUsage}}},
 	}

diff --git a/render/mapping.go b/render/mapping.go
@@ -23,7 +23,7 @@ const (
 	TheInternetID    = "theinternet"
 	TheInternetMajor = "The Internet"
 
-	containersKey = "containers"
+	ContainersKey = "containers"
 	ipsKey        = "ips"
 	podsKey       = "pods"
 	processesKey  = "processes"
@@ -520,7 +520,7 @@ func MapContainer2ContainerImage(n RenderableNode, _ report.Networks) Renderable
 	// Add container id key to the counters, which will later be counted to produce the minor label
 	id := MakeContainerImageID(imageID)
 	result := NewDerivedNode(id, n.WithParents(report.EmptySets))
-	result.Node.Counters = result.Node.Counters.Add(containersKey, 1)
+	result.Node.Counters = result.Node.Counters.Add(ContainersKey, 1)
 
 	// Add the container as a child of the new image node
 	result.Children = result.Children.Add(n.Node)
@@ -653,7 +653,7 @@ func MapContainer2Pod(n RenderableNode, _ report.Networks) RenderableNodes {
 	// Add container-<id> key to NMD, which will later be counted to produce the
 	// minor label
 	result := NewRenderableNodeWith(id, "", "", podID, n.WithParents(report.EmptySets))
-	result.Counters = result.Counters.Add(containersKey, 1)
+	result.Counters = result.Counters.Add(ContainersKey, 1)
 
 	// Due to a bug in kubernetes, addon pods on the master node are not returned
 	// from the API. This is a workaround until
@@ -690,7 +690,7 @@ func MapContainer2Hostname(n RenderableNode, _ report.Networks) RenderableNodes
 	result.Rank = id
 
 	// Add container id key to the counters, which will later be counted to produce the minor label
-	result.Counters = result.Counters.Add(containersKey, 1)
+	result.Counters = result.Counters.Add(ContainersKey, 1)
 
 	result.Node.Topology = "container_hostname"
 	result.Node.ID = id
@@ -708,7 +708,7 @@ func MapCountContainers(n RenderableNode, _ report.Networks) RenderableNodes {
 		return RenderableNodes{n.ID: n}
 	}
 
-	containers, _ := n.Node.Counters.Lookup(containersKey)
+	containers, _ := n.Node.Counters.Lookup(ContainersKey)
 	if containers == 1 {
 		n.LabelMinor = "1 container"
 	} else {

diff --git a/render/topologies.go b/render/topologies.go
@@ -192,10 +192,7 @@ var AddressRenderer = MakeMap(
 var HostRenderer = MakeReduce(
 	MakeMap(
 		MapX2Host,
-		MakeMap(
-			MapContainerImageIdentity,
-			SelectContainerImage,
-		),
+		FilterPseudo(ContainerImageRenderer),
 	),
 	MakeMap(
 		MapX2Host,