36 lines
707 B
Go
36 lines
707 B
Go
|
package k8s
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"k8s.io/client-go/kubernetes"
|
||
|
|
"k8s.io/client-go/rest"
|
||
|
|
"k8s.io/client-go/tools/clientcmd"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Client struct {
|
||
|
|
Clientset *kubernetes.Clientset
|
||
|
|
}
|
||
|
|
|
||
|
|
func New() (*Client, error) {
|
||
|
|
cfg, err := rest.InClusterConfig()
|
||
|
|
if err != nil {
|
||
|
|
kubeconfig := os.Getenv("KUBECONFIG")
|
||
|
|
if kubeconfig == "" {
|
||
|
|
return nil, fmt.Errorf("load in-cluster config: %w", err)
|
||
|
|
}
|
||
|
|
cfg, err = clientcmd.BuildConfigFromFlags("", kubeconfig)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("load kubeconfig: %w", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
clientset, err := kubernetes.NewForConfig(cfg)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("build clientset: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Client{Clientset: clientset}, nil
|
||
|
|
}
|