feat: Add container metrics to getAllContainers function

cmd/sentinel/containers.go

@@ -3,8 +3,11 @@ package main
 import (
 	"context"
 	"encoding/json"
+	"fmt"
+	"io"
 	"log"
 
+	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/container"
 	"github.com/docker/docker/client"
 )
@@ -16,6 +19,20 @@ type Container struct {
 	State        string            `json:"state"`
 	Labels       map[string]string `json:"labels"`
 	HealthStatus string            `json:"health_status"`
+	Metrics      ContainerMetrics  `json:"metrics"`
+}
+
+type ContainerMetrics struct {
+	CPUUsagePercentage    float64       `json:"cpu_usage_percentage"`
+	MemoryUsagePercentage float64       `json:"memory_usage_percentage"`
+	MemoryUsed            uint64        `json:"memory_used"`
+	MemoryAvailable       uint64        `json:"available_memory"`
+	NetworkUsage          NetworkDevice `json:"network_usage_in"`
+}
+type NetworkDevice struct {
+	Name    string `json:"name"`
+	RxBytes uint64 `json:"rx_bytes"`
+	TxBytes uint64 `json:"tx_bytes"`
 }
 
 func getAllContainers() (string, error) {
@@ -41,13 +58,53 @@ func getAllContainers() (string, error) {
 		if inspectData.State.Health != nil {
 			healthStatus = inspectData.State.Health.Status
 		}
+		// Get container stats
+		metrics := ContainerMetrics{
+			CPUUsagePercentage:    0,
+			MemoryUsagePercentage: 0,
+			MemoryUsed:            0,
+			MemoryAvailable:       0,
+			NetworkUsage:          NetworkDevice{},
+		}
+		if container.State == "running" {
+			stats, err := apiClient.ContainerStatsOneShot(ctx, container.ID)
+			if err != nil {
+				fmt.Printf("Error getting container stats: %v\n", err)
+				continue
+			}
+			var v *types.StatsJSON
+			dec := json.NewDecoder(stats.Body)
+			if err := dec.Decode(&v); err != nil {
+				if err != io.EOF {
+					fmt.Printf("Error decoding container stats: %v\n", err)
+				}
+			}
+			network_devices := v.Networks
+			for _, device := range network_devices {
+				metrics.NetworkUsage = NetworkDevice{
+					Name:    device.InstanceID,
+					RxBytes: device.RxBytes,
+					TxBytes: device.TxBytes,
+				}
+			}
+
+			metrics = ContainerMetrics{
+				CPUUsagePercentage:    calculateCPUPercent(v),
+				MemoryUsagePercentage: calculateMmemoryPercent(v),
+				MemoryUsed:            v.MemoryStats.Usage,
+				MemoryAvailable:       v.MemoryStats.Limit,
+				NetworkUsage:          metrics.NetworkUsage,
+			}
+		}
+
 		containersData = append(containersData, Container{
 			ID:           container.ID,
 			Image:        container.Image,
 			Labels:       container.Labels,
 			Name:         container.Names[0][1:],
 			State:        container.State,
 			HealthStatus: healthStatus,
+			Metrics:      metrics,
 		})
 	}
 	jsonData, err := json.MarshalIndent(containersData, "", "    ")
@@ -57,3 +114,21 @@ func getAllContainers() (string, error) {
 	return string(jsonData), nil
 
 }
+func calculateCPUPercent(stat *types.StatsJSON) float64 {
+	cpuPercent := 0.0
+	cpuDelta := float64(stat.CPUStats.CPUUsage.TotalUsage) - float64(stat.PreCPUStats.CPUUsage.TotalUsage)
+	systemDelta := float64(stat.CPUStats.SystemUsage) - float64(stat.PreCPUStats.SystemUsage)
+	if systemDelta > 0.0 {
+		cpuPercent = (cpuDelta / systemDelta) * float64(len(stat.CPUStats.CPUUsage.PercpuUsage)) * 100.0
+	}
+	return cpuPercent
+}
+
+func calculateMmemoryPercent(stat *types.StatsJSON) float64 {
+	usageMemory := float64(stat.MemoryStats.Usage)
+	cachedMemory := float64(stat.MemoryStats.Stats["cache"])
+	availableMemory := float64(stat.MemoryStats.Limit)
+	usedMemory := usageMemory - cachedMemory
+	usedMemoryPercentage := (usedMemory / availableMemory) * 100
+	return usedMemoryPercentage
+}