/
client.go
92 lines (76 loc) · 2.47 KB
/
client.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 client
import (
"os"
ingressnodefwv1alpha1 "github.com/openshift/ingress-node-firewall/api/v1alpha1"
"github.com/golang/glog"
apiext "k8s.io/apiextensions-apiserver/pkg/apis/apiextensions/v1"
"k8s.io/apimachinery/pkg/runtime"
discovery "k8s.io/client-go/discovery"
"k8s.io/client-go/kubernetes/scheme"
appsv1client "k8s.io/client-go/kubernetes/typed/apps/v1"
corev1client "k8s.io/client-go/kubernetes/typed/core/v1"
networkv1client "k8s.io/client-go/kubernetes/typed/networking/v1"
rbacv1client "k8s.io/client-go/kubernetes/typed/rbac/v1"
"k8s.io/client-go/rest"
"k8s.io/client-go/tools/clientcmd"
"sigs.k8s.io/controller-runtime/pkg/client"
)
// Client defines the client set that will be used for testing
var Client *ClientSet
func init() {
Client = New("")
}
// ClientSet provides the struct to talk with relevant API
type ClientSet struct {
client.Client
corev1client.CoreV1Interface
networkv1client.NetworkingV1Client
appsv1client.AppsV1Interface
rbacv1client.RbacV1Interface
discovery.DiscoveryInterface
Config *rest.Config
}
// New returns a *ClientBuilder with the given kubeconfig.
func New(kubeconfig string) *ClientSet {
var config *rest.Config
var err error
if kubeconfig == "" {
kubeconfig = os.Getenv("KUBECONFIG")
}
if kubeconfig != "" {
glog.V(4).Infof("Loading kube client config from path %q", kubeconfig)
config, err = clientcmd.BuildConfigFromFlags("", kubeconfig)
} else {
glog.V(4).Infof("Using in-cluster kube client config")
config, err = rest.InClusterConfig()
}
if err != nil {
glog.Infof("Failed to init kubernetes client, please check the $KUBECONFIG environment variable")
panic(err)
}
clientSet := &ClientSet{}
clientSet.CoreV1Interface = corev1client.NewForConfigOrDie(config)
clientSet.AppsV1Interface = appsv1client.NewForConfigOrDie(config)
clientSet.RbacV1Interface = rbacv1client.NewForConfigOrDie(config)
clientSet.DiscoveryInterface = discovery.NewDiscoveryClientForConfigOrDie(config)
clientSet.NetworkingV1Client = *networkv1client.NewForConfigOrDie(config)
clientSet.Config = config
myScheme := runtime.NewScheme()
if err = scheme.AddToScheme(myScheme); err != nil {
panic(err)
}
// Setup Scheme for all resources
if err := apiext.AddToScheme(myScheme); err != nil {
panic(err)
}
if err := ingressnodefwv1alpha1.AddToScheme(myScheme); err != nil {
panic(err)
}
clientSet.Client, err = client.New(config, client.Options{
Scheme: myScheme,
})
if err != nil {
return nil
}
return clientSet
}