98 lines
2.9 KiB
Go
98 lines
2.9 KiB
Go
package util
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/dynamic"
|
|
kubeclientset "k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
|
|
"github.com/huawei-cloudnative/karmada/pkg/apis/membercluster/v1alpha1"
|
|
)
|
|
|
|
const (
|
|
// kubeAPIQPS is the maximum QPS to the master from this client
|
|
kubeAPIQPS = 20.0
|
|
// kubeAPIBurst is the maximum burst for throttle
|
|
kubeAPIBurst = 30
|
|
tokenKey = "token"
|
|
cADataKey = "caBundle"
|
|
)
|
|
|
|
// ClusterClient stands for a cluster Clientset for the given member cluster
|
|
type ClusterClient struct {
|
|
KubeClient *kubeclientset.Clientset
|
|
ClusterName string
|
|
}
|
|
|
|
// DynamicClusterClient stands for a dynamic client for the given member cluster
|
|
type DynamicClusterClient struct {
|
|
DynamicClientSet dynamic.Interface
|
|
ClusterName string
|
|
}
|
|
|
|
// NewClusterClientSet returns a ClusterClient for the given member cluster.
|
|
func NewClusterClientSet(c *v1alpha1.MemberCluster, client kubeclientset.Interface) (*ClusterClient, error) {
|
|
clusterConfig, err := buildMemberClusterConfig(c, client)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var clusterClientSet = ClusterClient{ClusterName: c.Name}
|
|
|
|
if clusterConfig != nil {
|
|
clusterClientSet.KubeClient = kubeclientset.NewForConfigOrDie(clusterConfig)
|
|
}
|
|
return &clusterClientSet, nil
|
|
}
|
|
|
|
// NewClusterDynamicClientSet returns a dynamic client for the given member cluster.
|
|
func NewClusterDynamicClientSet(c *v1alpha1.MemberCluster, client kubeclientset.Interface) (*DynamicClusterClient, error) {
|
|
clusterConfig, err := buildMemberClusterConfig(c, client)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var clusterClientSet = DynamicClusterClient{ClusterName: c.Name}
|
|
|
|
if clusterConfig != nil {
|
|
clusterClientSet.DynamicClientSet = dynamic.NewForConfigOrDie(clusterConfig)
|
|
}
|
|
return &clusterClientSet, nil
|
|
}
|
|
|
|
func buildMemberClusterConfig(memberCluster *v1alpha1.MemberCluster, client kubeclientset.Interface) (*rest.Config, error) {
|
|
clusterName := memberCluster.Name
|
|
apiEndpoint := memberCluster.Spec.APIEndpoint
|
|
if apiEndpoint == "" {
|
|
return nil, fmt.Errorf("the api endpoint of cluster %s is empty", clusterName)
|
|
}
|
|
|
|
secretName := memberCluster.Spec.SecretRef.Name
|
|
if secretName == "" {
|
|
return nil, fmt.Errorf("cluster %s does not have a secret name", clusterName)
|
|
}
|
|
|
|
secret, err := client.CoreV1().Secrets(memberCluster.Spec.SecretRef.Namespace).Get(context.TODO(), secretName, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
token, tokenFound := secret.Data[tokenKey]
|
|
if !tokenFound || len(token) == 0 {
|
|
return nil, fmt.Errorf("the secret for cluster %s is missing a non-empty value for %q", clusterName, tokenKey)
|
|
}
|
|
|
|
clusterConfig, err := clientcmd.BuildConfigFromFlags(apiEndpoint, "")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
clusterConfig.CAData = secret.Data[cADataKey]
|
|
clusterConfig.BearerToken = string(token)
|
|
clusterConfig.QPS = kubeAPIQPS
|
|
clusterConfig.Burst = kubeAPIBurst
|
|
|
|
return clusterConfig, nil
|
|
}
|