feat: kubesphere 4.0 (#6115)
* feat: kubesphere 4.0 Signed-off-by: ci-bot <ci-bot@kubesphere.io> * feat: kubesphere 4.0 Signed-off-by: ci-bot <ci-bot@kubesphere.io> --------- Signed-off-by: ci-bot <ci-bot@kubesphere.io> Co-authored-by: ks-ci-bot <ks-ci-bot@example.com> Co-authored-by: joyceliu <joyceliu@yunify.com>
This commit is contained in:
committed by
GitHub
parent
b5015ec7b9
commit
447a51f08b
81
pkg/controller/kubectl/kubectl.go
Normal file
81
pkg/controller/kubectl/kubectl.go
Normal file
@@ -0,0 +1,81 @@
|
||||
/*
|
||||
* Please refer to the LICENSE file in the root directory of the project.
|
||||
* https://github.com/kubesphere/kubesphere/blob/master/LICENSE
|
||||
*/
|
||||
|
||||
package kubectl
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
coordinationv1 "k8s.io/api/coordination/v1"
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
"k8s.io/apimachinery/pkg/api/errors"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/util/wait"
|
||||
"k8s.io/klog/v2"
|
||||
"sigs.k8s.io/controller-runtime/pkg/client"
|
||||
|
||||
"kubesphere.io/kubesphere/pkg/constants"
|
||||
"kubesphere.io/kubesphere/pkg/controller"
|
||||
)
|
||||
|
||||
const controllerName = "kubectl"
|
||||
|
||||
type Reconciler struct {
|
||||
client.Client
|
||||
|
||||
resyncPeriod time.Duration
|
||||
renewPeriod time.Duration
|
||||
}
|
||||
|
||||
func (r *Reconciler) Name() string {
|
||||
return controllerName
|
||||
}
|
||||
|
||||
func (r *Reconciler) NeedLeaderElection() bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (r *Reconciler) SetupWithManager(mgr *controller.Manager) error {
|
||||
r.Client = mgr.GetClient()
|
||||
r.resyncPeriod = time.Minute
|
||||
r.renewPeriod = time.Minute
|
||||
|
||||
return mgr.Add(r)
|
||||
}
|
||||
|
||||
func (r *Reconciler) Start(ctx context.Context) error {
|
||||
go wait.UntilWithContext(ctx, func(ctx context.Context) {
|
||||
if err := r.reconcile(ctx); err != nil {
|
||||
klog.Errorf("%s controller reconcile error: %s\n", controllerName, err.Error())
|
||||
}
|
||||
}, r.resyncPeriod)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (r *Reconciler) reconcile(ctx context.Context) error {
|
||||
leases := &coordinationv1.LeaseList{}
|
||||
if err := r.List(ctx, leases, client.MatchingLabels{constants.KubectlPodLabel: ""}); err != nil {
|
||||
return err
|
||||
}
|
||||
// The minimum required heartbeat time, the heartbeat time of all leases must be greater than this
|
||||
heartbeatTime := time.Now().Add(-r.renewPeriod)
|
||||
for i := range leases.Items {
|
||||
lease := &leases.Items[i]
|
||||
if lease.Spec.RenewTime.After(heartbeatTime) {
|
||||
continue
|
||||
}
|
||||
pod := &corev1.Pod{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Namespace: lease.Namespace,
|
||||
Name: lease.Name,
|
||||
},
|
||||
}
|
||||
if err := r.Delete(ctx, pod, client.GracePeriodSeconds(0)); err != nil && !errors.IsNotFound(err) {
|
||||
klog.Errorf("deleting Pod %s/%s failed: %s, will retry", pod.Namespace, pod.Name, err.Error())
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
77
pkg/controller/kubectl/lease/lease.go
Normal file
77
pkg/controller/kubectl/lease/lease.go
Normal file
@@ -0,0 +1,77 @@
|
||||
/*
|
||||
* Please refer to the LICENSE file in the root directory of the project.
|
||||
* https://github.com/kubesphere/kubesphere/blob/master/LICENSE
|
||||
*/
|
||||
|
||||
package lease
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
coordinationv1 "k8s.io/api/coordination/v1"
|
||||
corev1 "k8s.io/api/core/v1"
|
||||
"k8s.io/apimachinery/pkg/api/errors"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/client-go/kubernetes"
|
||||
"k8s.io/client-go/util/retry"
|
||||
|
||||
"kubesphere.io/kubesphere/pkg/constants"
|
||||
)
|
||||
|
||||
type Operator struct {
|
||||
client kubernetes.Interface
|
||||
}
|
||||
|
||||
func NewOperator(client kubernetes.Interface) *Operator {
|
||||
return &Operator{
|
||||
client: client,
|
||||
}
|
||||
}
|
||||
|
||||
func (o *Operator) Create(ctx context.Context, owner *corev1.Pod) error {
|
||||
now := metav1.NowMicro()
|
||||
lease := &coordinationv1.Lease{
|
||||
ObjectMeta: metav1.ObjectMeta{
|
||||
Namespace: owner.Namespace,
|
||||
Name: owner.Name,
|
||||
Labels: map[string]string{
|
||||
constants.KubectlPodLabel: "",
|
||||
constants.KubeSphereManagedLabel: "true",
|
||||
},
|
||||
OwnerReferences: []metav1.OwnerReference{
|
||||
{
|
||||
APIVersion: "v1",
|
||||
Kind: "Pod",
|
||||
UID: owner.GetUID(),
|
||||
Name: owner.GetName(),
|
||||
},
|
||||
},
|
||||
},
|
||||
Spec: coordinationv1.LeaseSpec{
|
||||
AcquireTime: &now,
|
||||
RenewTime: &now,
|
||||
},
|
||||
}
|
||||
if _, err := o.client.CoordinationV1().Leases(owner.Namespace).Create(ctx, lease, metav1.CreateOptions{}); err != nil {
|
||||
if errors.IsAlreadyExists(err) {
|
||||
return o.Renew(ctx, lease.Namespace, lease.Name)
|
||||
}
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (o *Operator) Renew(ctx context.Context, namespace, name string) error {
|
||||
return retry.RetryOnConflict(retry.DefaultRetry, func() error {
|
||||
lease, err := o.client.CoordinationV1().Leases(namespace).Get(ctx, name, metav1.GetOptions{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
now := metav1.NowMicro()
|
||||
lease.Spec.RenewTime = &now
|
||||
if _, err = o.client.CoordinationV1().Leases(namespace).Update(ctx, lease, metav1.UpdateOptions{}); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user