Skip to content

Commit

Permalink
Merge pull request #7 from k8spacket/disable-k8s-api
Browse files Browse the repository at this point in the history
Disable k8s api
  • Loading branch information
k8spacket authored Aug 20, 2024
2 parents 3716c34 + bc2577e commit 16a3750
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions client.go
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,8 @@ import (
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/rest"
"os"
"strconv"
)

type IPResourceInfo struct {
Expand All @@ -16,8 +18,15 @@ type IPResourceInfo struct {

var _, clientset = configClusterClient()

var disabledK8sResource, _ = strconv.ParseBool(os.Getenv("K8S_PACKET_K8S_RESOURCES_DISABLED"))

func FetchK8SInfo() map[string]IPResourceInfo {

if disabledK8sResource {
fmt.Println("Getting k8s resources is disabled")
return map[string]IPResourceInfo{}
}

fmt.Println("Getting k8s resources")

m := make(map[string]IPResourceInfo)
Expand Down Expand Up @@ -72,6 +81,10 @@ func FetchK8SInfo() map[string]IPResourceInfo {

func GetPodIPsBySelectors(fieldSelector string, labelSelector string) []string {

if disabledK8sResource {
return []string{}
}

list := make([]string, 0)

pods, err := clientset.CoreV1().Pods("").List(context.TODO(), metav1.ListOptions{FieldSelector: fieldSelector, LabelSelector: labelSelector})
Expand All @@ -88,6 +101,11 @@ func GetPodIPsBySelectors(fieldSelector string, labelSelector string) []string {
}

func configClusterClient() (error, *kubernetes.Clientset) {

if disabledK8sResource {
return nil, nil
}

config, err := rest.InClusterConfig()

if err != nil {
Expand Down

0 comments on commit 16a3750

Please sign in to comment.