1 files changed, 174 insertions, 0 deletions
diff --git a/src/go/collectors/go.d.plugin/modules/docker/charts.go b/src/go/collectors/go.d.plugin/modules/docker/charts.go
new file mode 100644
index 000000000..2dd26c0e3
--- /dev/null
+++ b/src/go/collectors/go.d.plugin/modules/docker/charts.go
@@ -0,0 +1,174 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package docker
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/netdata/netdata/go/go.d.plugin/agent/module"
+)
+
+const (
+	prioContainersState = module.Priority + iota
+	prioContainersHealthy
+
+	prioContainerState
+	prioContainerHealthStatus
+	prioContainerWritableLayerSize
+
+	prioImagesCount
+	prioImagesSize
+)
+
+var summaryCharts = module.Charts{
+	containersStateChart.Copy(),
+	containersHealthyChart.Copy(),
+
+	imagesCountChart.Copy(),
+	imagesSizeChart.Copy(),
+}
+
+var (
+	containersStateChart = module.Chart{
+		ID:       "containers_state",
+		Title:    "Total number of Docker containers in various states",
+		Units:    "containers",
+		Fam:      "containers",
+		Ctx:      "docker.containers_state",
+		Priority: prioContainersState,
+		Type:     module.Stacked,
+		Dims: module.Dims{
+			{ID: "containers_state_running", Name: "running"},
+			{ID: "containers_state_paused", Name: "paused"},
+			{ID: "containers_state_exited", Name: "exited"},
+		},
+	}
+	containersHealthyChart = module.Chart{
+		ID:       "healthy_containers",
+		Title:    "Total number of Docker containers in various health states",
+		Units:    "containers",
+		Fam:      "containers",
+		Ctx:      "docker.containers_health_status",
+		Priority: prioContainersHealthy,
+		Dims: module.Dims{
+			{ID: "containers_health_status_healthy", Name: "healthy"},
+			{ID: "containers_health_status_unhealthy", Name: "unhealthy"},
+			{ID: "containers_health_status_not_running_unhealthy", Name: "not_running_unhealthy"},
+			{ID: "containers_health_status_starting", Name: "starting"},
+			{ID: "containers_health_status_none", Name: "no_healthcheck"},
+		},
+	}
+)
+
+var (
+	imagesCountChart = module.Chart{
+		ID:       "images_count",
+		Title:    "Total number of Docker images in various states",
+		Units:    "images",
+		Fam:      "images",
+		Ctx:      "docker.images",
+		Priority: prioImagesCount,
+		Type:     module.Stacked,
+		Dims: module.Dims{
+			{ID: "images_active", Name: "active"},
+			{ID: "images_dangling", Name: "dangling"},
+		},
+	}
+	imagesSizeChart = module.Chart{
+		ID:       "images_size",
+		Title:    "Total size of all Docker images",
+		Units:    "bytes",
+		Fam:      "images",
+		Ctx:      "docker.images_size",
+		Priority: prioImagesSize,
+		Dims: module.Dims{
+			{ID: "images_size", Name: "size"},
+		},
+	}
+)
+
+var (
+	containerChartsTmpl = module.Charts{
+		containerStateChartTmpl.Copy(),
+		containerHealthStatusChartTmpl.Copy(),
+		containerWritableLayerSizeChartTmpl.Copy(),
+	}
+
+	containerStateChartTmpl = module.Chart{
+		ID:       "container_%s_state",
+		Title:    "Docker container state",
+		Units:    "state",
+		Fam:      "containers",
+		Ctx:      "docker.container_state",
+		Priority: prioContainerState,
+		Dims: module.Dims{
+			{ID: "container_%s_state_running", Name: "running"},
+			{ID: "container_%s_state_paused", Name: "paused"},
+			{ID: "container_%s_state_exited", Name: "exited"},
+			{ID: "container_%s_state_created", Name: "created"},
+			{ID: "container_%s_state_restarting", Name: "restarting"},
+			{ID: "container_%s_state_removing", Name: "removing"},
+			{ID: "container_%s_state_dead", Name: "dead"},
+		},
+	}
+	containerHealthStatusChartTmpl = module.Chart{
+		ID:       "container_%s_health_status",
+		Title:    "Docker container health status",
+		Units:    "status",
+		Fam:      "containers",
+		Ctx:      "docker.container_health_status",
+		Priority: prioContainerHealthStatus,
+		Dims: module.Dims{
+			{ID: "container_%s_health_status_healthy", Name: "healthy"},
+			{ID: "container_%s_health_status_unhealthy", Name: "unhealthy"},
+			{ID: "container_%s_health_status_not_running_unhealthy", Name: "not_running_unhealthy"},
+			{ID: "container_%s_health_status_starting", Name: "starting"},
+			{ID: "container_%s_health_status_none", Name: "no_healthcheck"},
+		},
+	}
+	containerWritableLayerSizeChartTmpl = module.Chart{
+		ID:       "container_%s_writable_layer_size",
+		Title:    "Docker container writable layer size",
+		Units:    "bytes",
+		Fam:      "containers",
+		Ctx:      "docker.container_writeable_layer_size",
+		Priority: prioContainerWritableLayerSize,
+		Dims: module.Dims{
+			{ID: "container_%s_size_rw", Name: "writable_layer"},
+		},
+	}
+)
+
+func (d *Docker) addContainerCharts(name, image string) {
+	charts := containerChartsTmpl.Copy()
+	if !d.CollectContainerSize {
+		_ = charts.Remove(containerWritableLayerSizeChartTmpl.ID)
+	}
+
+	for _, chart := range *charts {
+		chart.ID = fmt.Sprintf(chart.ID, name)
+		chart.Labels = []module.Label{
+			{Key: "container_name", Value: name},
+			{Key: "image", Value: image},
+		}
+		for _, dim := range chart.Dims {
+			dim.ID = fmt.Sprintf(dim.ID, name)
+		}
+	}
+
+	if err := d.Charts().Add(*charts...); err != nil {
+		d.Warning(err)
+	}
+}
+
+func (d *Docker) removeContainerCharts(name string) {
+	px := fmt.Sprintf("container_%s", name)
+
+	for _, chart := range *d.Charts() {
+		if strings.HasPrefix(chart.ID, px) {
+			chart.MarkRemove()
+			chart.MarkNotCreated()
+		}
+	}
+}