130 lines
3.1 KiB
Go
130 lines
3.1 KiB
Go
package k8s
|
|
|
|
import (
|
|
applicationclientset "github.com/kubernetes-sigs/application/pkg/client/clientset/versioned"
|
|
s2i "github.com/kubesphere/s2ioperator/pkg/client/clientset/versioned"
|
|
"k8s.io/client-go/discovery"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
kubesphere "kubesphere.io/kubesphere/pkg/client/clientset/versioned"
|
|
"strings"
|
|
)
|
|
|
|
type KubernetesClient struct {
|
|
// kubernetes client interface
|
|
k8s *kubernetes.Clientset
|
|
|
|
// discovery client
|
|
discoveryClient *discovery.DiscoveryClient
|
|
|
|
// generated clientset
|
|
ks *kubesphere.Clientset
|
|
|
|
s2i *s2i.Clientset
|
|
|
|
application *applicationclientset.Clientset
|
|
|
|
master string
|
|
|
|
config *rest.Config
|
|
}
|
|
|
|
// NewKubernetesClientOrDie creates KubernetesClient and panic if there is an error
|
|
func NewKubernetesClientOrDie(options *KubernetesOptions) *KubernetesClient {
|
|
config, err := clientcmd.BuildConfigFromFlags("", options.KubeConfig)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
config.QPS = options.QPS
|
|
config.Burst = options.Burst
|
|
|
|
k := &KubernetesClient{
|
|
k8s: kubernetes.NewForConfigOrDie(config),
|
|
discoveryClient: discovery.NewDiscoveryClientForConfigOrDie(config),
|
|
ks: kubesphere.NewForConfigOrDie(config),
|
|
s2i: s2i.NewForConfigOrDie(config),
|
|
application: applicationclientset.NewForConfigOrDie(config),
|
|
master: config.Host,
|
|
config: config,
|
|
}
|
|
|
|
if options.Master != "" {
|
|
k.master = options.Master
|
|
}
|
|
// The https prefix is automatically added when using sa.
|
|
// But it will not be set automatically when reading from kubeconfig
|
|
// which may cause some problems in the client of other languages.
|
|
if !strings.HasPrefix(k.master, "http://") && !strings.HasPrefix(k.master, "https://") {
|
|
k.master = "https://" + k.master
|
|
}
|
|
return k
|
|
}
|
|
|
|
// NewKubernetesClient creates a KubernetesClient
|
|
func NewKubernetesClient(options *KubernetesOptions) (*KubernetesClient, error) {
|
|
config, err := clientcmd.BuildConfigFromFlags("", options.KubeConfig)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
config.QPS = options.QPS
|
|
config.Burst = options.Burst
|
|
|
|
var k KubernetesClient
|
|
k.k8s, err = kubernetes.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
k.ks, err = kubesphere.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
k.s2i, err = s2i.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
k.application, err = applicationclientset.NewForConfig(config)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
k.master = options.Master
|
|
k.config = config
|
|
|
|
return &k, nil
|
|
}
|
|
|
|
func (k *KubernetesClient) Kubernetes() kubernetes.Interface {
|
|
return k.k8s
|
|
}
|
|
|
|
func (k *KubernetesClient) Discovery() discovery.DiscoveryInterface {
|
|
return k.discoveryClient
|
|
}
|
|
|
|
func (k *KubernetesClient) KubeSphere() kubesphere.Interface {
|
|
return k.ks
|
|
}
|
|
|
|
func (k *KubernetesClient) S2i() s2i.Interface {
|
|
return k.s2i
|
|
}
|
|
|
|
func (k *KubernetesClient) Application() applicationclientset.Interface {
|
|
return k.application
|
|
}
|
|
|
|
// master address used to generate kubeconfig for downloading
|
|
func (k *KubernetesClient) Master() string {
|
|
return k.master
|
|
}
|
|
|
|
func (k *KubernetesClient) Config() *rest.Config {
|
|
return k.config
|
|
}
|