k8s.go

package kubernetes

import (
	"context"
	"os"
	"strings"

	"github.com/bryopsida/http-healthcheck-sidecar/config"
	"github.com/gofiber/fiber/v2/log"
	v1 "k8s.io/api/core/v1"
	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
	"k8s.io/client-go/kubernetes"
	"k8s.io/client-go/rest"
)

func getNamespace() string {
	// namespace will be in /var/run/secrets/kubernetes.io/serviceaccount/namespace
	data, err := os.ReadFile("/var/run/secrets/kubernetes.io/serviceaccount/namespace")
	if err != nil {
		panic(err.Error())
	}
	return string(data)
}
func returnOverrideState() bool {
	return config.GetStatusOverrideState()
}

func fetchPodState(podName string) *v1.Pod {
	config, err := rest.InClusterConfig()
	if err != nil {
		panic(err.Error())
	}
	clientset, err := kubernetes.NewForConfig(config)
	if err != nil {
		panic(err.Error())
	}
	pod, err := clientset.CoreV1().Pods(getNamespace()).Get(context.TODO(), podName, metav1.GetOptions{})
	if err != nil {
		panic(err.Error())
	}
	return pod
}

func IsPodHealthy(podName string) bool {
	log.Info("Fetching status for pod %s", podName)
	if config.IsStatusOverriden() {
		return config.GetStatusOverrideState()
	} else {
		podState := fetchPodState(podName)
		for _, containerStatus := range podState.Status.ContainerStatuses {
			if !containerStatus.Ready {
				return false
			}
		}
		return true
	}
}

func IsPodContainerHealthy(podName string, containerName string) bool {
	log.Info("Fetching status for container %s on pod %s", containerName, podName)
	if config.IsStatusOverriden() {
		return config.GetStatusOverrideState()
	} else {
		podState := fetchPodState(podName)
		for _, containerStatus := range podState.Status.ContainerStatuses {
			if strings.ToLower(containerStatus.Name) == strings.ToLower(containerName) {
				return containerStatus.Ready
			}
		}
		return false
	}
}