【修改】k8s 配置

apps/devops/services/k8s/common/pod.go

@@ -0,0 +1,151 @@
+package common
+
+import (
+	apps "k8s.io/api/apps/v1"
+	batch "k8s.io/api/batch/v1"
+	v1 "k8s.io/api/core/v1"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+)
+
+// FilterDeploymentPodsByOwnerReference returns a subset of pods controlled by given deployment.
+func FilterDeploymentPodsByOwnerReference(deployment apps.Deployment, allRS []apps.ReplicaSet,
+	allPods []v1.Pod) []v1.Pod {
+	var matchingPods []v1.Pod
+	for _, rs := range allRS {
+		if metav1.IsControlledBy(&rs, &deployment) {
+			matchingPods = append(matchingPods, FilterPodsByControllerRef(&rs, allPods)...)
+		}
+	}
+
+	return matchingPods
+}
+
+// FilterPodsByControllerRef returns a subset of pods controlled by given controller resource, excluding deployments.
+func FilterPodsByControllerRef(owner metav1.Object, allPods []v1.Pod) []v1.Pod {
+	var matchingPods []v1.Pod
+	for _, pod := range allPods {
+		if metav1.IsControlledBy(&pod, owner) {
+			matchingPods = append(matchingPods, pod)
+		}
+	}
+	return matchingPods
+}
+
+// GetContainerImages returns container image strings from the given pod spec.
+func GetContainerImages(podTemplate *v1.PodSpec) []string {
+	var containerImages []string
+	for _, container := range podTemplate.Containers {
+		containerImages = append(containerImages, container.Image)
+	}
+	return containerImages
+}
+
+// GetInitContainerImages returns init container image strings from the given pod spec.
+func GetInitContainerImages(podTemplate *v1.PodSpec) []string {
+	var initContainerImages []string
+	for _, initContainer := range podTemplate.InitContainers {
+		initContainerImages = append(initContainerImages, initContainer.Image)
+	}
+	return initContainerImages
+}
+
+// FilterPodsForJob returns a list of pods that matches to a job controller's selector
+func FilterPodsForJob(job batch.Job, pods []v1.Pod) []v1.Pod {
+	result := make([]v1.Pod, 0)
+	for _, pod := range pods {
+		if pod.Namespace == job.Namespace {
+			selectorMatch := true
+			for key, value := range job.Spec.Selector.MatchLabels {
+				if pod.Labels[key] != value {
+					selectorMatch = false
+					break
+				}
+			}
+			if selectorMatch {
+				result = append(result, pod)
+			}
+		}
+	}
+
+	return result
+}
+
+// GetContainerNames returns the container image name without the version number from the given pod spec.
+func GetContainerNames(podTemplate *v1.PodSpec) []string {
+	var containerNames []string
+	for _, container := range podTemplate.Containers {
+		containerNames = append(containerNames, container.Name)
+	}
+	return containerNames
+}
+
+// GetInitContainerNames returns the init container image name without the version number from the given pod spec.
+func GetInitContainerNames(podTemplate *v1.PodSpec) []string {
+	var initContainerNames []string
+	for _, initContainer := range podTemplate.InitContainers {
+		initContainerNames = append(initContainerNames, initContainer.Name)
+	}
+	return initContainerNames
+}
+
+// GetNonduplicateContainerImages returns list of container image strings without duplicates
+func GetNonduplicateContainerImages(podList []v1.Pod) []string {
+	var containerImages []string
+	for _, pod := range podList {
+		for _, container := range pod.Spec.Containers {
+			if noStringInSlice(container.Image, containerImages) {
+				containerImages = append(containerImages, container.Image)
+			}
+		}
+	}
+	return containerImages
+}
+
+// GetNonduplicateInitContainerImages returns list of init container image strings without duplicates
+func GetNonduplicateInitContainerImages(podList []v1.Pod) []string {
+	var initContainerImages []string
+	for _, pod := range podList {
+		for _, initContainer := range pod.Spec.InitContainers {
+			if noStringInSlice(initContainer.Image, initContainerImages) {
+				initContainerImages = append(initContainerImages, initContainer.Image)
+			}
+		}
+	}
+	return initContainerImages
+}
+
+// GetNonduplicateContainerNames returns list of container names strings without duplicates
+func GetNonduplicateContainerNames(podList []v1.Pod) []string {
+	var containerNames []string
+	for _, pod := range podList {
+		for _, container := range pod.Spec.Containers {
+			if noStringInSlice(container.Name, containerNames) {
+				containerNames = append(containerNames, container.Name)
+			}
+		}
+	}
+	return containerNames
+}
+
+// GetNonduplicateInitContainerNames returns list of init container names strings without duplicates
+func GetNonduplicateInitContainerNames(podList []v1.Pod) []string {
+	var initContainerNames []string
+	for _, pod := range podList {
+		for _, initContainer := range pod.Spec.InitContainers {
+			if noStringInSlice(initContainer.Name, initContainerNames) {
+				initContainerNames = append(initContainerNames, initContainer.Name)
+			}
+		}
+	}
+	return initContainerNames
+}
+
+//noStringInSlice checks if string in array
+func noStringInSlice(str string, array []string) bool {
+	for _, alreadystr := range array {
+		if alreadystr == str {
+			return false
+		}
+	}
+	return true
+}