@@ -36,6 +36,7 @@ import (
36
36
clientset "k8s.io/client-go/kubernetes"
37
37
"k8s.io/klog"
38
38
"k8s.io/kubernetes/pkg/api/legacyscheme"
39
+ podutil "k8s.io/kubernetes/pkg/api/v1/pod"
39
40
v1helper "k8s.io/kubernetes/pkg/apis/core/v1/helper"
40
41
"k8s.io/kubernetes/pkg/features"
41
42
"k8s.io/kubernetes/pkg/util/mount"
@@ -555,13 +556,7 @@ func GetPodVolumeNames(pod *v1.Pod) (mounts sets.String, devices sets.String) {
555
556
mounts = sets .NewString ()
556
557
devices = sets .NewString ()
557
558
558
- addContainerVolumes (pod .Spec .Containers , mounts , devices )
559
- addContainerVolumes (pod .Spec .InitContainers , mounts , devices )
560
- return
561
- }
562
-
563
- func addContainerVolumes (containers []v1.Container , mounts , devices sets.String ) {
564
- for _ , container := range containers {
559
+ podutil .VisitContainers (& pod .Spec , func (container * v1.Container ) bool {
565
560
if container .VolumeMounts != nil {
566
561
for _ , mount := range container .VolumeMounts {
567
562
mounts .Insert (mount .Name )
@@ -574,5 +569,7 @@ func addContainerVolumes(containers []v1.Container, mounts, devices sets.String)
574
569
devices .Insert (device .Name )
575
570
}
576
571
}
577
- }
572
+ return true
573
+ })
574
+ return
578
575
}
0 commit comments