735eb34829
This refactors the v1 lease controller. It makes two functional differences to the lease controller: * It no longer ties lease updates to node pings or node status updates * There is no fallback mechanism to status updates This also moves vk_envtest, allowing for future brown-box testing of the lease controller with envtest
66 lines
2.0 KiB
Go
66 lines
2.0 KiB
Go
package nodeutil
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/pkg/errors"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/fields"
|
|
kubeinformers "k8s.io/client-go/informers"
|
|
"k8s.io/client-go/kubernetes"
|
|
coordclientset "k8s.io/client-go/kubernetes/typed/coordination/v1"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/client-go/tools/clientcmd"
|
|
)
|
|
|
|
// ClientsetFromEnv returns a kuberentes client set from:
|
|
// 1. the passed in kubeconfig path
|
|
// 2. If the kubeconfig path is empty or non-existent, then the in-cluster config is used.
|
|
func ClientsetFromEnv(kubeConfigPath string) (*kubernetes.Clientset, error) {
|
|
var (
|
|
config *rest.Config
|
|
err error
|
|
)
|
|
|
|
if kubeConfigPath != "" {
|
|
config, err = clientsetFromEnvKubeConfigPath(kubeConfigPath)
|
|
} else {
|
|
config, err = rest.InClusterConfig()
|
|
}
|
|
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "error getting rest client config")
|
|
}
|
|
|
|
return kubernetes.NewForConfig(config)
|
|
}
|
|
|
|
func clientsetFromEnvKubeConfigPath(kubeConfigPath string) (*rest.Config, error) {
|
|
_, err := os.Stat(kubeConfigPath)
|
|
if os.IsNotExist(err) {
|
|
return rest.InClusterConfig()
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return clientcmd.NewNonInteractiveDeferredLoadingClientConfig(
|
|
&clientcmd.ClientConfigLoadingRules{ExplicitPath: kubeConfigPath},
|
|
&clientcmd.ConfigOverrides{},
|
|
).ClientConfig()
|
|
}
|
|
|
|
// PodInformerFilter is a filter that you should use when creating a pod informer for use with the pod controller.
|
|
func PodInformerFilter(node string) kubeinformers.SharedInformerOption {
|
|
return kubeinformers.WithTweakListOptions(func(options *metav1.ListOptions) {
|
|
options.FieldSelector = fields.OneTermEqualSelector("spec.nodeName", node).String()
|
|
})
|
|
}
|
|
|
|
// NodeLeaseV1Client creates a V1 Lease client for use with node leases from the passed in client.
|
|
//
|
|
// Use this with node.WithNodeEnableLeaseV1Beta1 when creating a node controller.
|
|
func NodeLeaseV1Client(client kubernetes.Interface) coordclientset.LeaseInterface {
|
|
return client.CoordinationV1().Leases(corev1.NamespaceNodeLease)
|
|
}
|