mirror of https://github.com/k3s-io/k3s
351 lines
12 KiB
Go
351 lines
12 KiB
Go
/*
|
|
Copyright 2017 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
// This file contains adapters that convert between RC and RS,
|
|
// as if ReplicationController were an older API version of ReplicaSet.
|
|
// It allows ReplicaSetController to directly replace the old ReplicationManager,
|
|
// which was previously a manually-maintained copy-paste of RSC.
|
|
|
|
package replication
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
apps "k8s.io/api/apps/v1"
|
|
autoscalingv1 "k8s.io/api/autoscaling/v1"
|
|
"k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
|
|
"k8s.io/apimachinery/pkg/watch"
|
|
coreinformers "k8s.io/client-go/informers/core/v1"
|
|
clientset "k8s.io/client-go/kubernetes"
|
|
appsv1client "k8s.io/client-go/kubernetes/typed/apps/v1"
|
|
v1client "k8s.io/client-go/kubernetes/typed/core/v1"
|
|
appslisters "k8s.io/client-go/listers/apps/v1"
|
|
v1listers "k8s.io/client-go/listers/core/v1"
|
|
"k8s.io/client-go/tools/cache"
|
|
appsinternal "k8s.io/kubernetes/pkg/apis/apps"
|
|
appsconversion "k8s.io/kubernetes/pkg/apis/apps/v1"
|
|
apiv1 "k8s.io/kubernetes/pkg/apis/core/v1"
|
|
"k8s.io/kubernetes/pkg/controller"
|
|
)
|
|
|
|
// informerAdapter implements ReplicaSetInformer by wrapping ReplicationControllerInformer
|
|
// and converting objects.
|
|
type informerAdapter struct {
|
|
rcInformer coreinformers.ReplicationControllerInformer
|
|
}
|
|
|
|
func (i informerAdapter) Informer() cache.SharedIndexInformer {
|
|
return conversionInformer{i.rcInformer.Informer()}
|
|
}
|
|
|
|
func (i informerAdapter) Lister() appslisters.ReplicaSetLister {
|
|
return conversionLister{i.rcInformer.Lister()}
|
|
}
|
|
|
|
type conversionInformer struct {
|
|
cache.SharedIndexInformer
|
|
}
|
|
|
|
func (i conversionInformer) AddEventHandler(handler cache.ResourceEventHandler) {
|
|
i.SharedIndexInformer.AddEventHandler(conversionEventHandler{handler})
|
|
}
|
|
|
|
func (i conversionInformer) AddEventHandlerWithResyncPeriod(handler cache.ResourceEventHandler, resyncPeriod time.Duration) {
|
|
i.SharedIndexInformer.AddEventHandlerWithResyncPeriod(conversionEventHandler{handler}, resyncPeriod)
|
|
}
|
|
|
|
type conversionLister struct {
|
|
rcLister v1listers.ReplicationControllerLister
|
|
}
|
|
|
|
func (l conversionLister) List(selector labels.Selector) ([]*apps.ReplicaSet, error) {
|
|
rcList, err := l.rcLister.List(selector)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return convertSlice(rcList)
|
|
}
|
|
|
|
func (l conversionLister) ReplicaSets(namespace string) appslisters.ReplicaSetNamespaceLister {
|
|
return conversionNamespaceLister{l.rcLister.ReplicationControllers(namespace)}
|
|
}
|
|
|
|
func (l conversionLister) GetPodReplicaSets(pod *v1.Pod) ([]*apps.ReplicaSet, error) {
|
|
rcList, err := l.rcLister.GetPodControllers(pod)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return convertSlice(rcList)
|
|
}
|
|
|
|
type conversionNamespaceLister struct {
|
|
rcLister v1listers.ReplicationControllerNamespaceLister
|
|
}
|
|
|
|
func (l conversionNamespaceLister) List(selector labels.Selector) ([]*apps.ReplicaSet, error) {
|
|
rcList, err := l.rcLister.List(selector)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return convertSlice(rcList)
|
|
}
|
|
|
|
func (l conversionNamespaceLister) Get(name string) (*apps.ReplicaSet, error) {
|
|
rc, err := l.rcLister.Get(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return convertRCtoRS(rc, nil)
|
|
}
|
|
|
|
type conversionEventHandler struct {
|
|
handler cache.ResourceEventHandler
|
|
}
|
|
|
|
func (h conversionEventHandler) OnAdd(obj interface{}) {
|
|
rs, err := convertRCtoRS(obj.(*v1.ReplicationController), nil)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("dropping RC OnAdd event: can't convert object %#v to RS: %v", obj, err))
|
|
return
|
|
}
|
|
h.handler.OnAdd(rs)
|
|
}
|
|
|
|
func (h conversionEventHandler) OnUpdate(oldObj, newObj interface{}) {
|
|
oldRS, err := convertRCtoRS(oldObj.(*v1.ReplicationController), nil)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("dropping RC OnUpdate event: can't convert old object %#v to RS: %v", oldObj, err))
|
|
return
|
|
}
|
|
newRS, err := convertRCtoRS(newObj.(*v1.ReplicationController), nil)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("dropping RC OnUpdate event: can't convert new object %#v to RS: %v", newObj, err))
|
|
return
|
|
}
|
|
h.handler.OnUpdate(oldRS, newRS)
|
|
}
|
|
|
|
func (h conversionEventHandler) OnDelete(obj interface{}) {
|
|
rc, ok := obj.(*v1.ReplicationController)
|
|
if !ok {
|
|
// Convert the Obj inside DeletedFinalStateUnknown.
|
|
tombstone, ok := obj.(cache.DeletedFinalStateUnknown)
|
|
if !ok {
|
|
utilruntime.HandleError(fmt.Errorf("dropping RC OnDelete event: couldn't get object from tombstone %+v", obj))
|
|
return
|
|
}
|
|
rc, ok = tombstone.Obj.(*v1.ReplicationController)
|
|
if !ok {
|
|
utilruntime.HandleError(fmt.Errorf("dropping RC OnDelete event: tombstone contained object that is not a RC %#v", obj))
|
|
return
|
|
}
|
|
rs, err := convertRCtoRS(rc, nil)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("dropping RC OnDelete event: can't convert object %#v to RS: %v", obj, err))
|
|
return
|
|
}
|
|
h.handler.OnDelete(cache.DeletedFinalStateUnknown{Key: tombstone.Key, Obj: rs})
|
|
return
|
|
}
|
|
|
|
// It's a regular RC object.
|
|
rs, err := convertRCtoRS(rc, nil)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("dropping RC OnDelete event: can't convert object %#v to RS: %v", obj, err))
|
|
return
|
|
}
|
|
h.handler.OnDelete(rs)
|
|
}
|
|
|
|
type clientsetAdapter struct {
|
|
clientset.Interface
|
|
}
|
|
|
|
func (c clientsetAdapter) AppsV1() appsv1client.AppsV1Interface {
|
|
return conversionAppsV1Client{c.Interface, c.Interface.AppsV1()}
|
|
}
|
|
|
|
func (c clientsetAdapter) Apps() appsv1client.AppsV1Interface {
|
|
return conversionAppsV1Client{c.Interface, c.Interface.AppsV1()}
|
|
}
|
|
|
|
type conversionAppsV1Client struct {
|
|
clientset clientset.Interface
|
|
appsv1client.AppsV1Interface
|
|
}
|
|
|
|
func (c conversionAppsV1Client) ReplicaSets(namespace string) appsv1client.ReplicaSetInterface {
|
|
return conversionClient{c.clientset.CoreV1().ReplicationControllers(namespace)}
|
|
}
|
|
|
|
type conversionClient struct {
|
|
v1client.ReplicationControllerInterface
|
|
}
|
|
|
|
func (c conversionClient) Create(ctx context.Context, rs *apps.ReplicaSet, opts metav1.CreateOptions) (*apps.ReplicaSet, error) {
|
|
return convertCall(func(rc *v1.ReplicationController) (*v1.ReplicationController, error) {
|
|
return c.ReplicationControllerInterface.Create(ctx, rc, opts)
|
|
}, rs)
|
|
}
|
|
|
|
func (c conversionClient) Update(ctx context.Context, rs *apps.ReplicaSet, opts metav1.UpdateOptions) (*apps.ReplicaSet, error) {
|
|
return convertCall(func(rc *v1.ReplicationController) (*v1.ReplicationController, error) {
|
|
return c.ReplicationControllerInterface.Update(ctx, rc, opts)
|
|
}, rs)
|
|
}
|
|
|
|
func (c conversionClient) UpdateStatus(ctx context.Context, rs *apps.ReplicaSet, opts metav1.UpdateOptions) (*apps.ReplicaSet, error) {
|
|
return convertCall(func(rc *v1.ReplicationController) (*v1.ReplicationController, error) {
|
|
return c.ReplicationControllerInterface.UpdateStatus(ctx, rc, opts)
|
|
}, rs)
|
|
}
|
|
|
|
func (c conversionClient) Get(ctx context.Context, name string, options metav1.GetOptions) (*apps.ReplicaSet, error) {
|
|
rc, err := c.ReplicationControllerInterface.Get(context.TODO(), name, options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return convertRCtoRS(rc, nil)
|
|
}
|
|
|
|
func (c conversionClient) List(ctx context.Context, opts metav1.ListOptions) (*apps.ReplicaSetList, error) {
|
|
rcList, err := c.ReplicationControllerInterface.List(context.TODO(), opts)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return convertList(rcList)
|
|
}
|
|
|
|
func (c conversionClient) Watch(ctx context.Context, opts metav1.ListOptions) (watch.Interface, error) {
|
|
// This is not used by RSC because we wrap the shared informer instead.
|
|
return nil, errors.New("Watch() is not implemented for conversionClient")
|
|
}
|
|
|
|
func (c conversionClient) Patch(ctx context.Context, name string, pt types.PatchType, data []byte, opts metav1.PatchOptions, subresources ...string) (result *apps.ReplicaSet, err error) {
|
|
// This is not used by RSC.
|
|
return nil, errors.New("Patch() is not implemented for conversionClient")
|
|
}
|
|
|
|
func (c conversionClient) GetScale(ctx context.Context, name string, options metav1.GetOptions) (result *autoscalingv1.Scale, err error) {
|
|
// This is not used by RSC.
|
|
return nil, errors.New("GetScale() is not implemented for conversionClient")
|
|
}
|
|
|
|
func (c conversionClient) UpdateScale(ctx context.Context, name string, scale *autoscalingv1.Scale, opts metav1.UpdateOptions) (result *autoscalingv1.Scale, err error) {
|
|
// This is not used by RSC.
|
|
return nil, errors.New("UpdateScale() is not implemented for conversionClient")
|
|
}
|
|
|
|
func convertSlice(rcList []*v1.ReplicationController) ([]*apps.ReplicaSet, error) {
|
|
rsList := make([]*apps.ReplicaSet, 0, len(rcList))
|
|
for _, rc := range rcList {
|
|
rs, err := convertRCtoRS(rc, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
rsList = append(rsList, rs)
|
|
}
|
|
return rsList, nil
|
|
}
|
|
|
|
func convertList(rcList *v1.ReplicationControllerList) (*apps.ReplicaSetList, error) {
|
|
rsList := &apps.ReplicaSetList{Items: make([]apps.ReplicaSet, len(rcList.Items))}
|
|
for i := range rcList.Items {
|
|
rc := &rcList.Items[i]
|
|
_, err := convertRCtoRS(rc, &rsList.Items[i])
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
return rsList, nil
|
|
}
|
|
|
|
func convertCall(fn func(*v1.ReplicationController) (*v1.ReplicationController, error), rs *apps.ReplicaSet) (*apps.ReplicaSet, error) {
|
|
rc, err := convertRStoRC(rs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
result, err := fn(rc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return convertRCtoRS(result, nil)
|
|
}
|
|
|
|
func convertRCtoRS(rc *v1.ReplicationController, out *apps.ReplicaSet) (*apps.ReplicaSet, error) {
|
|
var rsInternal appsinternal.ReplicaSet
|
|
if err := apiv1.Convert_v1_ReplicationController_To_apps_ReplicaSet(rc, &rsInternal, nil); err != nil {
|
|
return nil, fmt.Errorf("can't convert ReplicationController %v/%v to ReplicaSet: %v", rc.Namespace, rc.Name, err)
|
|
}
|
|
if out == nil {
|
|
out = new(apps.ReplicaSet)
|
|
}
|
|
if err := appsconversion.Convert_apps_ReplicaSet_To_v1_ReplicaSet(&rsInternal, out, nil); err != nil {
|
|
return nil, fmt.Errorf("can't convert ReplicaSet (converted from ReplicationController %v/%v) from internal to apps/v1: %v", rc.Namespace, rc.Name, err)
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func convertRStoRC(rs *apps.ReplicaSet) (*v1.ReplicationController, error) {
|
|
var rsInternal appsinternal.ReplicaSet
|
|
if err := appsconversion.Convert_v1_ReplicaSet_To_apps_ReplicaSet(rs, &rsInternal, nil); err != nil {
|
|
return nil, fmt.Errorf("can't convert ReplicaSet (converting to ReplicationController %v/%v) from apps/v1 to internal: %v", rs.Namespace, rs.Name, err)
|
|
}
|
|
var rc v1.ReplicationController
|
|
if err := apiv1.Convert_apps_ReplicaSet_To_v1_ReplicationController(&rsInternal, &rc, nil); err != nil {
|
|
return nil, fmt.Errorf("can't convert ReplicaSet to ReplicationController %v/%v: %v", rs.Namespace, rs.Name, err)
|
|
}
|
|
return &rc, nil
|
|
}
|
|
|
|
type podControlAdapter struct {
|
|
controller.PodControlInterface
|
|
}
|
|
|
|
func (pc podControlAdapter) CreatePods(namespace string, template *v1.PodTemplateSpec, object runtime.Object) error {
|
|
// This is not used by RSC.
|
|
return errors.New("CreatePods() is not implemented for podControlAdapter")
|
|
}
|
|
|
|
func (pc podControlAdapter) CreatePodsOnNode(nodeName, namespace string, template *v1.PodTemplateSpec, object runtime.Object, controllerRef *metav1.OwnerReference) error {
|
|
// This is not used by RSC.
|
|
return errors.New("CreatePodsOnNode() is not implemented for podControlAdapter")
|
|
}
|
|
|
|
func (pc podControlAdapter) CreatePodsWithControllerRef(namespace string, template *v1.PodTemplateSpec, object runtime.Object, controllerRef *metav1.OwnerReference) error {
|
|
rc, err := convertRStoRC(object.(*apps.ReplicaSet))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return pc.PodControlInterface.CreatePodsWithControllerRef(namespace, template, rc, controllerRef)
|
|
}
|
|
|
|
func (pc podControlAdapter) DeletePod(namespace string, podID string, object runtime.Object) error {
|
|
rc, err := convertRStoRC(object.(*apps.ReplicaSet))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return pc.PodControlInterface.DeletePod(namespace, podID, rc)
|
|
}
|