kubeshark/kubernetes/utils.go

package kubernetes

import (
	"regexp"

	"github.com/kubeshark/worker/models"
	core "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)

func GetNodeHostToTargettedPodsMap(targettedPods []core.Pod) models.NodeToPodsMap {
	nodeToTargettedPodsMap := make(models.NodeToPodsMap)
	for _, pod := range targettedPods {
		minimizedPod := getMinimizedPod(pod)

		existingList := nodeToTargettedPodsMap[pod.Spec.NodeName]
		if existingList == nil {
			nodeToTargettedPodsMap[pod.Spec.NodeName] = []core.Pod{minimizedPod}
		} else {
			nodeToTargettedPodsMap[pod.Spec.NodeName] = append(nodeToTargettedPodsMap[pod.Spec.NodeName], minimizedPod)
		}
	}
	return nodeToTargettedPodsMap
}

func getMinimizedPod(fullPod core.Pod) core.Pod {
	return core.Pod{
		ObjectMeta: metav1.ObjectMeta{
			Name:      fullPod.Name,
			Namespace: fullPod.Namespace,
		},
		Status: core.PodStatus{
			PodIP:             fullPod.Status.PodIP,
			ContainerStatuses: getMinimizedContainerStatuses(fullPod),
		},
	}
}

func getMinimizedContainerStatuses(fullPod core.Pod) []core.ContainerStatus {
	result := make([]core.ContainerStatus, len(fullPod.Status.ContainerStatuses))

	for i, container := range fullPod.Status.ContainerStatuses {
		result[i] = core.ContainerStatus{
			ContainerID: container.ContainerID,
		}
	}

	return result
}

func excludeSelfPods(pods []core.Pod) []core.Pod {
	kubesharkPrefixRegex := regexp.MustCompile("^" + KubesharkResourcesPrefix)

	nonKubesharkPods := make([]core.Pod, 0)
	for _, pod := range pods {
		if !kubesharkPrefixRegex.MatchString(pod.Name) {
			nonKubesharkPods = append(nonKubesharkPods, pod)
		}
	}

	return nonKubesharkPods
}

func getPodArrayDiff(oldPods []core.Pod, newPods []core.Pod) (added []core.Pod, removed []core.Pod) {
	added = getMissingPods(newPods, oldPods)
	removed = getMissingPods(oldPods, newPods)

	return added, removed
}

//returns pods present in pods1 array and missing in pods2 array
func getMissingPods(pods1 []core.Pod, pods2 []core.Pod) []core.Pod {
	missingPods := make([]core.Pod, 0)
	for _, pod1 := range pods1 {
		var found = false
		for _, pod2 := range pods2 {
			if pod1.UID == pod2.UID {
				found = true
				break
			}
		}
		if !found {
			missingPods = append(missingPods, pod1)
		}
	}
	return missingPods
}

func GetPodInfosForPods(pods []core.Pod) []*models.PodInfo {
	podInfos := make([]*models.PodInfo, 0)
	for _, pod := range pods {
		podInfos = append(podInfos, &models.PodInfo{Name: pod.Name, Namespace: pod.Namespace, NodeName: pod.Spec.NodeName})
	}
	return podInfos
}