Added system info

Author: igorrendulic

docker/docker.go

@@ -640,6 +640,27 @@ func (cl *Client) ContainerRename(containerID string, newContainerName string) e
 	return nil
 }
 
+func (cl *Client) SystemWideInfo() (types.Info, types.DiskUsage, error) {
+	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
+	defer cancel()
+
+	info, err := cl.client.Info(ctx)
+	if err != nil {
+		if cl.log != nil {
+			cl.log.Error("failed to retrieve docker info", err)
+		}
+		return types.Info{}, types.DiskUsage{}, err
+	}
+	diskUsage, err := cl.client.DiskUsage(ctx)
+	if err != nil {
+		if cl.log != nil {
+			cl.log.Error("failed to retrieve docker info", err)
+		}
+		return types.Info{}, types.DiskUsage{}, err
+	}
+	return info, diskUsage, nil
+}
+
 func calculateBlockIO(blkio types.BlkioStats) (blkRead uint64, blkWrite uint64) {
 	for _, bioEntry := range blkio.IoServiceBytesRecursive {
 		switch strings.ToLower(bioEntry.Op) {

docker/docker_interfaces.go

@@ -51,4 +51,7 @@ type Docker interface {
 	ContainerRemove(containerID string) error
 	ContainerRename(containerID string, newContainerName string) error
 	ContainerReplace(containerID string, image string, tag string) error
+
+	// host system information
+	SystemWideInfo() (types.Info, types.DiskUsage, error)
 }

docker/docker_test.go

@@ -15,11 +15,13 @@
 package docker
 
 import (
+	"fmt"
 	"io/ioutil"
 	"strings"
 	"testing"
 
 	mclog "github.com/chryscloud/go-microkit-plugins/log"
+	"github.com/docker/docker/api/types"
 )
 
 var (
@@ -52,6 +54,70 @@ func TestContainerReplace(t *testing.T) {
 	}
 }
 
+func TestSystemWideInfo(t *testing.T) {
+	cl := NewSocketClient(Log(zl), Host("unix:///var/run/docker.sock"))
+	systemInfo, diskUsage, err := cl.SystemWideInfo()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	fmt.Printf("Container running: %v\n", systemInfo.ContainersRunning)
+	fmt.Printf("Container paused: %v\n", systemInfo.ContainersPaused)
+	fmt.Printf("Container stopped: %v\n", systemInfo.ContainersStopped)
+	fmt.Printf("Containers total: %v\n", systemInfo.Containers)
+
+	imgNum := len(diskUsage.Images)
+	totalImgSize := int64(0)
+	activeImages := int64(0)
+	for _, im := range diskUsage.Images {
+		activeImages += im.Containers
+		totalImgSize += im.SharedSize
+	}
+	containerTotalSize := int64(0)
+	for _, c := range diskUsage.Containers {
+		containerTotalSize += c.SizeRw
+	}
+	totalVolumeSize := int64(0)
+	activeVolumes := int64(0)
+	for _, v := range diskUsage.Volumes {
+		activeVolumes += v.UsageData.RefCount
+		totalVolumeSize += v.UsageData.Size
+	}
+	fmt.Printf("disk size images: %v, size: %v, active images: %v\n", imgNum, totalImgSize, activeImages)
+	fmt.Printf("Containers total disk size: %v, volume size: %v, Active volumes: %v\n", containerTotalSize, totalVolumeSize, activeVolumes)
+
+	opts := types.ContainerListOptions{
+		All:   true,
+		Size:  true,
+		Quiet: false,
+	}
+	containers, err := cl.ContainersListWithOptions(opts)
+	if err != nil {
+		t.Fatal(err)
+	}
+	for _, c := range containers {
+		if len(c.Names) > 0 {
+			skip := false
+			for _, n := range c.Names {
+				if strings.Contains(n, "chrysedgeportal") || strings.Contains(n, "chrysedgeserver") || strings.Contains(n, "redis") {
+					skip = true
+				}
+			}
+			if skip {
+				continue
+			}
+			stats, err := cl.ContainerStats(c.ID)
+			if err != nil {
+				t.Fatal(err)
+			}
+			calculated := cl.CalculateStats(stats)
+			calculated.Status = c.Status
+			calculated.State = c.State
+			fmt.Printf("Stats: %v\n", calculated)
+		}
+	}
+}
+
 //TODO: tests need to be modified to run without actual docker config
 // func TestSocketClient(t *testing.T) {
 // 	cl := NewSocketClient(Log(zl), Host("unix:///var/run/docker.sock"))