fix agent losing track of container after restart

agent/main.go

@@ -178,14 +178,21 @@ func getDockerStats() ([]*ContainerStats, error) {
 	var wg sync.WaitGroup
 
 	for _, ctr := range containers {
-		ctr.IdShort = ctr.ID[:12]
+		ctr.IdShort = ctr.Id[:12]
 		validIds[ctr.IdShort] = struct{}{}
+		// check if container is less than 1 minute old (possible restart)
+		// note: can't use Created field because it's not updated on restart
+		if strings.HasSuffix(ctr.Status, "seconds") {
+			// if so, remove old container data
+			delete(containerCpuMap, ctr.IdShort)
+		}
 		wg.Add(1)
 		go func() {
 			defer wg.Done()
 			cstats, err := getContainerStats(ctr)
 			if err != nil {
-				// retry once
+				// delete container from map and retry once
+				delete(containerCpuMap, ctr.IdShort)
 				cstats, err = getContainerStats(ctr)
 				if err != nil {
 					log.Printf("Error getting container stats: %+v\n", err)

agent/types.go

@@ -42,23 +42,23 @@ type ContainerStats struct {
 }
 
 type Container struct {
-	ID      string `json:"Id"`
+	Id      string
 	IdShort string
 	Names   []string
-	Image   string
-	ImageID string
-	Command string
-	Created int64
-	// Ports      []Port
-	SizeRw     int64 `json:",omitempty"`
-	SizeRootFs int64 `json:",omitempty"`
-	Labels     map[string]string
-	State      string
 	Status  string
-	HostConfig struct {
-		NetworkMode string            `json:",omitempty"`
-		Annotations map[string]string `json:",omitempty"`
-	}
+	// Image   string
+	// ImageID string
+	// Command string
+	// Created int64
+	// Ports      []Port
+	// SizeRw     int64 `json:",omitempty"`
+	// SizeRootFs int64 `json:",omitempty"`
+	// Labels     map[string]string
+	// State      string
+	// HostConfig struct {
+	// 	NetworkMode string            `json:",omitempty"`
+	// 	Annotations map[string]string `json:",omitempty"`
+	// }
 	// NetworkSettings *SummaryNetworkSettings
 	// Mounts          []MountPoint
 }