This repository has been archived by the owner on Apr 8, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 443
/
k8sutil.go
92 lines (80 loc) · 2.41 KB
/
k8sutil.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
package utils
import (
"os"
"github.com/sirupsen/logrus"
apps_v1 "k8s.io/api/apps/v1"
batch_v1 "k8s.io/api/batch/v1"
api_v1 "k8s.io/api/core/v1"
ext_v1beta1 "k8s.io/api/extensions/v1beta1"
meta_v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
rbac_v1beta1 "k8s.io/api/rbac/v1beta1"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/rest"
"k8s.io/client-go/tools/clientcmd"
)
// GetClient returns a k8s clientset to the request from inside of cluster
func GetClient() kubernetes.Interface {
config, err := rest.InClusterConfig()
if err != nil {
logrus.Fatalf("Can not get kubernetes config: %v", err)
}
clientset, err := kubernetes.NewForConfig(config)
if err != nil {
logrus.Fatalf("Can not create kubernetes client: %v", err)
}
return clientset
}
func buildOutOfClusterConfig() (*rest.Config, error) {
kubeconfigPath := os.Getenv("KUBECONFIG")
if kubeconfigPath == "" {
kubeconfigPath = os.Getenv("HOME") + "/.kube/config"
}
return clientcmd.BuildConfigFromFlags("", kubeconfigPath)
}
// GetClientOutOfCluster returns a k8s clientset to the request from outside of cluster
func GetClientOutOfCluster() kubernetes.Interface {
config, err := buildOutOfClusterConfig()
if err != nil {
logrus.Fatalf("Can not get kubernetes config: %v", err)
}
clientset, err := kubernetes.NewForConfig(config)
if err != nil {
logrus.Fatalf("Can not get kubernetes config: %v", err)
}
return clientset
}
// GetObjectMetaData returns metadata of a given k8s object
func GetObjectMetaData(obj interface{}) meta_v1.ObjectMeta {
var objectMeta meta_v1.ObjectMeta
switch object := obj.(type) {
case *apps_v1.Deployment:
objectMeta = object.ObjectMeta
case *api_v1.ReplicationController:
objectMeta = object.ObjectMeta
case *apps_v1.ReplicaSet:
objectMeta = object.ObjectMeta
case *apps_v1.DaemonSet:
objectMeta = object.ObjectMeta
case *api_v1.Service:
objectMeta = object.ObjectMeta
case *api_v1.Pod:
objectMeta = object.ObjectMeta
case *batch_v1.Job:
objectMeta = object.ObjectMeta
case *api_v1.PersistentVolume:
objectMeta = object.ObjectMeta
case *api_v1.Namespace:
objectMeta = object.ObjectMeta
case *api_v1.Secret:
objectMeta = object.ObjectMeta
case *ext_v1beta1.Ingress:
objectMeta = object.ObjectMeta
case *api_v1.Node:
objectMeta = object.ObjectMeta
case *rbac_v1beta1.ClusterRole:
objectMeta = object.ObjectMeta
case *api_v1.ServiceAccount:
objectMeta = object.ObjectMeta
}
return objectMeta
}