enhance(k8s): pod tracker.TrackContainers can be called more than once

cognifloyd · cognifloyd · commit 9b46c1f330dc · 2022-04-12T00:43:54.000-05:00
diff --git a/runtime/kubernetes/pod_tracker.go b/runtime/kubernetes/pod_tracker.go
@@ -142,16 +142,17 @@ func (p podTracker) Start(ctx context.Context) {
 }
 
 // TrackContainers creates a containerTracker for each container.
-func (p podTracker) TrackContainers(containers []v1.Container) {
-	ctnTrackers := map[string]*containerTracker{}
+func (p *podTracker) TrackContainers(containers []v1.Container) {
+	if p.Containers == nil {
+		p.Containers := map[string]*containerTracker{}
+	}
+
 	for _, ctn := range containers {
-		ctnTrackers[ctn.Name] = &containerTracker{
+		p.Containers[ctn.Name] = &containerTracker{
 			Name:       ctn.Name,
 			Terminated: make(chan struct{}),
 		}
 	}
-
-	p.Containers = ctnTrackers
 }
 
 // newPodTracker initializes a podTracker with a given clientset for a given pod.

Original file line number	Diff line number	Diff line change
`@@ -142,16 +142,17 @@ func (p podTracker) Start(ctx context.Context) {`
`142`	`142`	`}`
`143`	`143`
`144`	`144`	`// TrackContainers creates a containerTracker for each container.`
`145`		`-func (p podTracker) TrackContainers(containers []v1.Container) {`
`146`		`- ctnTrackers := map[string]*containerTracker{}`
	`145`	`+func (p *podTracker) TrackContainers(containers []v1.Container) {`
	`146`	`+ if p.Containers == nil {`
	`147`	`+ p.Containers := map[string]*containerTracker{}`
	`148`	`+ }`
	`149`	`+`
`147`	`150`	`for _, ctn := range containers {`
`148`		`- ctnTrackers[ctn.Name] = &containerTracker{`
	`151`	`+ p.Containers[ctn.Name] = &containerTracker{`
`149`	`152`	`Name: ctn.Name,`
`150`	`153`	`Terminated: make(chan struct{}),`
`151`	`154`	`}`
`152`	`155`	`}`
`153`		`-`
`154`		`- p.Containers = ctnTrackers`
`155`	`156`	`}`
`156`	`157`
`157`	`158`	`// newPodTracker initializes a podTracker with a given clientset for a given pod.`