mirror of https://github.com/k3s-io/k3s
835 lines
31 KiB
Go
835 lines
31 KiB
Go
/*
|
|
Copyright 2016 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.
|
|
*/
|
|
|
|
// ### ATTENTION ###
|
|
//
|
|
// This code implements both ReplicaSet and ReplicationController.
|
|
//
|
|
// For RC, the objects are converted on the way in and out (see ../replication/),
|
|
// as if ReplicationController were just an older API version of ReplicaSet.
|
|
// However, RC and RS still have separate storage and separate instantiations
|
|
// of the ReplicaSetController object.
|
|
//
|
|
// Use rsc.Kind in log messages rather than hard-coding "ReplicaSet".
|
|
|
|
package replicaset
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
"sort"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
apps "k8s.io/api/apps/v1"
|
|
"k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/errors"
|
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
|
|
"k8s.io/apimachinery/pkg/util/wait"
|
|
appsinformers "k8s.io/client-go/informers/apps/v1"
|
|
coreinformers "k8s.io/client-go/informers/core/v1"
|
|
clientset "k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/kubernetes/scheme"
|
|
v1core "k8s.io/client-go/kubernetes/typed/core/v1"
|
|
appslisters "k8s.io/client-go/listers/apps/v1"
|
|
corelisters "k8s.io/client-go/listers/core/v1"
|
|
"k8s.io/client-go/tools/cache"
|
|
"k8s.io/client-go/tools/record"
|
|
"k8s.io/client-go/util/workqueue"
|
|
"k8s.io/component-base/metrics/prometheus/ratelimiter"
|
|
"k8s.io/klog/v2"
|
|
podutil "k8s.io/kubernetes/pkg/api/v1/pod"
|
|
"k8s.io/kubernetes/pkg/controller"
|
|
"k8s.io/utils/integer"
|
|
)
|
|
|
|
const (
|
|
// Realistic value of the burstReplica field for the replica set manager based off
|
|
// performance requirements for kubernetes 1.0.
|
|
BurstReplicas = 500
|
|
|
|
// The number of times we retry updating a ReplicaSet's status.
|
|
statusUpdateRetries = 1
|
|
)
|
|
|
|
// ReplicaSetController is responsible for synchronizing ReplicaSet objects stored
|
|
// in the system with actual running pods.
|
|
type ReplicaSetController struct {
|
|
// GroupVersionKind indicates the controller type.
|
|
// Different instances of this struct may handle different GVKs.
|
|
// For example, this struct can be used (with adapters) to handle ReplicationController.
|
|
schema.GroupVersionKind
|
|
|
|
kubeClient clientset.Interface
|
|
podControl controller.PodControlInterface
|
|
|
|
// A ReplicaSet is temporarily suspended after creating/deleting these many replicas.
|
|
// It resumes normal action after observing the watch events for them.
|
|
burstReplicas int
|
|
// To allow injection of syncReplicaSet for testing.
|
|
syncHandler func(rsKey string) error
|
|
|
|
// A TTLCache of pod creates/deletes each rc expects to see.
|
|
expectations *controller.UIDTrackingControllerExpectations
|
|
|
|
// A store of ReplicaSets, populated by the shared informer passed to NewReplicaSetController
|
|
rsLister appslisters.ReplicaSetLister
|
|
// rsListerSynced returns true if the pod store has been synced at least once.
|
|
// Added as a member to the struct to allow injection for testing.
|
|
rsListerSynced cache.InformerSynced
|
|
|
|
// A store of pods, populated by the shared informer passed to NewReplicaSetController
|
|
podLister corelisters.PodLister
|
|
// podListerSynced returns true if the pod store has been synced at least once.
|
|
// Added as a member to the struct to allow injection for testing.
|
|
podListerSynced cache.InformerSynced
|
|
|
|
// Controllers that need to be synced
|
|
queue workqueue.RateLimitingInterface
|
|
}
|
|
|
|
// NewReplicaSetController configures a replica set controller with the specified event recorder
|
|
func NewReplicaSetController(rsInformer appsinformers.ReplicaSetInformer, podInformer coreinformers.PodInformer, kubeClient clientset.Interface, burstReplicas int) *ReplicaSetController {
|
|
eventBroadcaster := record.NewBroadcaster()
|
|
eventBroadcaster.StartStructuredLogging(0)
|
|
eventBroadcaster.StartRecordingToSink(&v1core.EventSinkImpl{Interface: kubeClient.CoreV1().Events("")})
|
|
return NewBaseController(rsInformer, podInformer, kubeClient, burstReplicas,
|
|
apps.SchemeGroupVersion.WithKind("ReplicaSet"),
|
|
"replicaset_controller",
|
|
"replicaset",
|
|
controller.RealPodControl{
|
|
KubeClient: kubeClient,
|
|
Recorder: eventBroadcaster.NewRecorder(scheme.Scheme, v1.EventSource{Component: "replicaset-controller"}),
|
|
},
|
|
)
|
|
}
|
|
|
|
// NewBaseController is the implementation of NewReplicaSetController with additional injected
|
|
// parameters so that it can also serve as the implementation of NewReplicationController.
|
|
func NewBaseController(rsInformer appsinformers.ReplicaSetInformer, podInformer coreinformers.PodInformer, kubeClient clientset.Interface, burstReplicas int,
|
|
gvk schema.GroupVersionKind, metricOwnerName, queueName string, podControl controller.PodControlInterface) *ReplicaSetController {
|
|
if kubeClient != nil && kubeClient.CoreV1().RESTClient().GetRateLimiter() != nil {
|
|
ratelimiter.RegisterMetricAndTrackRateLimiterUsage(metricOwnerName, kubeClient.CoreV1().RESTClient().GetRateLimiter())
|
|
}
|
|
|
|
rsc := &ReplicaSetController{
|
|
GroupVersionKind: gvk,
|
|
kubeClient: kubeClient,
|
|
podControl: podControl,
|
|
burstReplicas: burstReplicas,
|
|
expectations: controller.NewUIDTrackingControllerExpectations(controller.NewControllerExpectations()),
|
|
queue: workqueue.NewNamedRateLimitingQueue(workqueue.DefaultControllerRateLimiter(), queueName),
|
|
}
|
|
|
|
rsInformer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
|
|
AddFunc: rsc.addRS,
|
|
UpdateFunc: rsc.updateRS,
|
|
DeleteFunc: rsc.deleteRS,
|
|
})
|
|
rsc.rsLister = rsInformer.Lister()
|
|
rsc.rsListerSynced = rsInformer.Informer().HasSynced
|
|
|
|
podInformer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
|
|
AddFunc: rsc.addPod,
|
|
// This invokes the ReplicaSet for every pod change, eg: host assignment. Though this might seem like
|
|
// overkill the most frequent pod update is status, and the associated ReplicaSet will only list from
|
|
// local storage, so it should be ok.
|
|
UpdateFunc: rsc.updatePod,
|
|
DeleteFunc: rsc.deletePod,
|
|
})
|
|
rsc.podLister = podInformer.Lister()
|
|
rsc.podListerSynced = podInformer.Informer().HasSynced
|
|
|
|
rsc.syncHandler = rsc.syncReplicaSet
|
|
|
|
return rsc
|
|
}
|
|
|
|
// SetEventRecorder replaces the event recorder used by the ReplicaSetController
|
|
// with the given recorder. Only used for testing.
|
|
func (rsc *ReplicaSetController) SetEventRecorder(recorder record.EventRecorder) {
|
|
// TODO: Hack. We can't cleanly shutdown the event recorder, so benchmarks
|
|
// need to pass in a fake.
|
|
rsc.podControl = controller.RealPodControl{KubeClient: rsc.kubeClient, Recorder: recorder}
|
|
}
|
|
|
|
// Run begins watching and syncing.
|
|
func (rsc *ReplicaSetController) Run(workers int, stopCh <-chan struct{}) {
|
|
defer utilruntime.HandleCrash()
|
|
defer rsc.queue.ShutDown()
|
|
|
|
controllerName := strings.ToLower(rsc.Kind)
|
|
klog.Infof("Starting %v controller", controllerName)
|
|
defer klog.Infof("Shutting down %v controller", controllerName)
|
|
|
|
if !cache.WaitForNamedCacheSync(rsc.Kind, stopCh, rsc.podListerSynced, rsc.rsListerSynced) {
|
|
return
|
|
}
|
|
|
|
for i := 0; i < workers; i++ {
|
|
go wait.Until(rsc.worker, time.Second, stopCh)
|
|
}
|
|
|
|
<-stopCh
|
|
}
|
|
|
|
// getReplicaSetsWithSameController returns a list of ReplicaSets with the same
|
|
// owner as the given ReplicaSet.
|
|
func (rsc *ReplicaSetController) getReplicaSetsWithSameController(rs *apps.ReplicaSet) []*apps.ReplicaSet {
|
|
controllerRef := metav1.GetControllerOf(rs)
|
|
if controllerRef == nil {
|
|
utilruntime.HandleError(fmt.Errorf("ReplicaSet has no controller: %v", rs))
|
|
return nil
|
|
}
|
|
|
|
allRSs, err := rsc.rsLister.ReplicaSets(rs.Namespace).List(labels.Everything())
|
|
if err != nil {
|
|
utilruntime.HandleError(err)
|
|
return nil
|
|
}
|
|
|
|
var relatedRSs []*apps.ReplicaSet
|
|
for _, r := range allRSs {
|
|
if ref := metav1.GetControllerOf(r); ref != nil && ref.UID == controllerRef.UID {
|
|
relatedRSs = append(relatedRSs, r)
|
|
}
|
|
}
|
|
|
|
if klog.V(2).Enabled() {
|
|
var relatedNames []string
|
|
for _, r := range relatedRSs {
|
|
relatedNames = append(relatedNames, r.Name)
|
|
}
|
|
klog.InfoS("Found related ReplicaSets", "replicaSet", klog.KObj(rs), "relatedReplicaSets", relatedNames)
|
|
}
|
|
|
|
return relatedRSs
|
|
}
|
|
|
|
// getPodReplicaSets returns a list of ReplicaSets matching the given pod.
|
|
func (rsc *ReplicaSetController) getPodReplicaSets(pod *v1.Pod) []*apps.ReplicaSet {
|
|
rss, err := rsc.rsLister.GetPodReplicaSets(pod)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
if len(rss) > 1 {
|
|
// ControllerRef will ensure we don't do anything crazy, but more than one
|
|
// item in this list nevertheless constitutes user error.
|
|
utilruntime.HandleError(fmt.Errorf("user error! more than one %v is selecting pods with labels: %+v", rsc.Kind, pod.Labels))
|
|
}
|
|
return rss
|
|
}
|
|
|
|
// resolveControllerRef returns the controller referenced by a ControllerRef,
|
|
// or nil if the ControllerRef could not be resolved to a matching controller
|
|
// of the correct Kind.
|
|
func (rsc *ReplicaSetController) resolveControllerRef(namespace string, controllerRef *metav1.OwnerReference) *apps.ReplicaSet {
|
|
// We can't look up by UID, so look up by Name and then verify UID.
|
|
// Don't even try to look up by Name if it's the wrong Kind.
|
|
if controllerRef.Kind != rsc.Kind {
|
|
return nil
|
|
}
|
|
rs, err := rsc.rsLister.ReplicaSets(namespace).Get(controllerRef.Name)
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
if rs.UID != controllerRef.UID {
|
|
// The controller we found with this Name is not the same one that the
|
|
// ControllerRef points to.
|
|
return nil
|
|
}
|
|
return rs
|
|
}
|
|
|
|
func (rsc *ReplicaSetController) enqueueRS(rs *apps.ReplicaSet) {
|
|
key, err := controller.KeyFunc(rs)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get key for object %#v: %v", rs, err))
|
|
return
|
|
}
|
|
|
|
rsc.queue.Add(key)
|
|
}
|
|
|
|
func (rsc *ReplicaSetController) enqueueRSAfter(rs *apps.ReplicaSet, duration time.Duration) {
|
|
key, err := controller.KeyFunc(rs)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get key for object %#v: %v", rs, err))
|
|
return
|
|
}
|
|
|
|
rsc.queue.AddAfter(key, duration)
|
|
}
|
|
|
|
func (rsc *ReplicaSetController) addRS(obj interface{}) {
|
|
rs := obj.(*apps.ReplicaSet)
|
|
klog.V(4).Infof("Adding %s %s/%s", rsc.Kind, rs.Namespace, rs.Name)
|
|
rsc.enqueueRS(rs)
|
|
}
|
|
|
|
// callback when RS is updated
|
|
func (rsc *ReplicaSetController) updateRS(old, cur interface{}) {
|
|
oldRS := old.(*apps.ReplicaSet)
|
|
curRS := cur.(*apps.ReplicaSet)
|
|
|
|
// TODO: make a KEP and fix informers to always call the delete event handler on re-create
|
|
if curRS.UID != oldRS.UID {
|
|
key, err := controller.KeyFunc(oldRS)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get key for object %#v: %v", oldRS, err))
|
|
return
|
|
}
|
|
rsc.deleteRS(cache.DeletedFinalStateUnknown{
|
|
Key: key,
|
|
Obj: oldRS,
|
|
})
|
|
}
|
|
|
|
// You might imagine that we only really need to enqueue the
|
|
// replica set when Spec changes, but it is safer to sync any
|
|
// time this function is triggered. That way a full informer
|
|
// resync can requeue any replica set that don't yet have pods
|
|
// but whose last attempts at creating a pod have failed (since
|
|
// we don't block on creation of pods) instead of those
|
|
// replica sets stalling indefinitely. Enqueueing every time
|
|
// does result in some spurious syncs (like when Status.Replica
|
|
// is updated and the watch notification from it retriggers
|
|
// this function), but in general extra resyncs shouldn't be
|
|
// that bad as ReplicaSets that haven't met expectations yet won't
|
|
// sync, and all the listing is done using local stores.
|
|
if *(oldRS.Spec.Replicas) != *(curRS.Spec.Replicas) {
|
|
klog.V(4).Infof("%v %v updated. Desired pod count change: %d->%d", rsc.Kind, curRS.Name, *(oldRS.Spec.Replicas), *(curRS.Spec.Replicas))
|
|
}
|
|
rsc.enqueueRS(curRS)
|
|
}
|
|
|
|
func (rsc *ReplicaSetController) deleteRS(obj interface{}) {
|
|
rs, ok := obj.(*apps.ReplicaSet)
|
|
if !ok {
|
|
tombstone, ok := obj.(cache.DeletedFinalStateUnknown)
|
|
if !ok {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get object from tombstone %#v", obj))
|
|
return
|
|
}
|
|
rs, ok = tombstone.Obj.(*apps.ReplicaSet)
|
|
if !ok {
|
|
utilruntime.HandleError(fmt.Errorf("tombstone contained object that is not a ReplicaSet %#v", obj))
|
|
return
|
|
}
|
|
}
|
|
|
|
key, err := controller.KeyFunc(rs)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get key for object %#v: %v", rs, err))
|
|
return
|
|
}
|
|
|
|
klog.V(4).Infof("Deleting %s %q", rsc.Kind, key)
|
|
|
|
// Delete expectations for the ReplicaSet so if we create a new one with the same name it starts clean
|
|
rsc.expectations.DeleteExpectations(key)
|
|
|
|
rsc.queue.Add(key)
|
|
}
|
|
|
|
// When a pod is created, enqueue the replica set that manages it and update its expectations.
|
|
func (rsc *ReplicaSetController) addPod(obj interface{}) {
|
|
pod := obj.(*v1.Pod)
|
|
|
|
if pod.DeletionTimestamp != nil {
|
|
// on a restart of the controller manager, it's possible a new pod shows up in a state that
|
|
// is already pending deletion. Prevent the pod from being a creation observation.
|
|
rsc.deletePod(pod)
|
|
return
|
|
}
|
|
|
|
// If it has a ControllerRef, that's all that matters.
|
|
if controllerRef := metav1.GetControllerOf(pod); controllerRef != nil {
|
|
rs := rsc.resolveControllerRef(pod.Namespace, controllerRef)
|
|
if rs == nil {
|
|
return
|
|
}
|
|
rsKey, err := controller.KeyFunc(rs)
|
|
if err != nil {
|
|
return
|
|
}
|
|
klog.V(4).Infof("Pod %s created: %#v.", pod.Name, pod)
|
|
rsc.expectations.CreationObserved(rsKey)
|
|
rsc.queue.Add(rsKey)
|
|
return
|
|
}
|
|
|
|
// Otherwise, it's an orphan. Get a list of all matching ReplicaSets and sync
|
|
// them to see if anyone wants to adopt it.
|
|
// DO NOT observe creation because no controller should be waiting for an
|
|
// orphan.
|
|
rss := rsc.getPodReplicaSets(pod)
|
|
if len(rss) == 0 {
|
|
return
|
|
}
|
|
klog.V(4).Infof("Orphan Pod %s created: %#v.", pod.Name, pod)
|
|
for _, rs := range rss {
|
|
rsc.enqueueRS(rs)
|
|
}
|
|
}
|
|
|
|
// When a pod is updated, figure out what replica set/s manage it and wake them
|
|
// up. If the labels of the pod have changed we need to awaken both the old
|
|
// and new replica set. old and cur must be *v1.Pod types.
|
|
func (rsc *ReplicaSetController) updatePod(old, cur interface{}) {
|
|
curPod := cur.(*v1.Pod)
|
|
oldPod := old.(*v1.Pod)
|
|
if curPod.ResourceVersion == oldPod.ResourceVersion {
|
|
// Periodic resync will send update events for all known pods.
|
|
// Two different versions of the same pod will always have different RVs.
|
|
return
|
|
}
|
|
|
|
labelChanged := !reflect.DeepEqual(curPod.Labels, oldPod.Labels)
|
|
if curPod.DeletionTimestamp != nil {
|
|
// when a pod is deleted gracefully it's deletion timestamp is first modified to reflect a grace period,
|
|
// and after such time has passed, the kubelet actually deletes it from the store. We receive an update
|
|
// for modification of the deletion timestamp and expect an rs to create more replicas asap, not wait
|
|
// until the kubelet actually deletes the pod. This is different from the Phase of a pod changing, because
|
|
// an rs never initiates a phase change, and so is never asleep waiting for the same.
|
|
rsc.deletePod(curPod)
|
|
if labelChanged {
|
|
// we don't need to check the oldPod.DeletionTimestamp because DeletionTimestamp cannot be unset.
|
|
rsc.deletePod(oldPod)
|
|
}
|
|
return
|
|
}
|
|
|
|
curControllerRef := metav1.GetControllerOf(curPod)
|
|
oldControllerRef := metav1.GetControllerOf(oldPod)
|
|
controllerRefChanged := !reflect.DeepEqual(curControllerRef, oldControllerRef)
|
|
if controllerRefChanged && oldControllerRef != nil {
|
|
// The ControllerRef was changed. Sync the old controller, if any.
|
|
if rs := rsc.resolveControllerRef(oldPod.Namespace, oldControllerRef); rs != nil {
|
|
rsc.enqueueRS(rs)
|
|
}
|
|
}
|
|
|
|
// If it has a ControllerRef, that's all that matters.
|
|
if curControllerRef != nil {
|
|
rs := rsc.resolveControllerRef(curPod.Namespace, curControllerRef)
|
|
if rs == nil {
|
|
return
|
|
}
|
|
klog.V(4).Infof("Pod %s updated, objectMeta %+v -> %+v.", curPod.Name, oldPod.ObjectMeta, curPod.ObjectMeta)
|
|
rsc.enqueueRS(rs)
|
|
// TODO: MinReadySeconds in the Pod will generate an Available condition to be added in
|
|
// the Pod status which in turn will trigger a requeue of the owning replica set thus
|
|
// having its status updated with the newly available replica. For now, we can fake the
|
|
// update by resyncing the controller MinReadySeconds after the it is requeued because
|
|
// a Pod transitioned to Ready.
|
|
// Note that this still suffers from #29229, we are just moving the problem one level
|
|
// "closer" to kubelet (from the deployment to the replica set controller).
|
|
if !podutil.IsPodReady(oldPod) && podutil.IsPodReady(curPod) && rs.Spec.MinReadySeconds > 0 {
|
|
klog.V(2).Infof("%v %q will be enqueued after %ds for availability check", rsc.Kind, rs.Name, rs.Spec.MinReadySeconds)
|
|
// Add a second to avoid milliseconds skew in AddAfter.
|
|
// See https://github.com/kubernetes/kubernetes/issues/39785#issuecomment-279959133 for more info.
|
|
rsc.enqueueRSAfter(rs, (time.Duration(rs.Spec.MinReadySeconds)*time.Second)+time.Second)
|
|
}
|
|
return
|
|
}
|
|
|
|
// Otherwise, it's an orphan. If anything changed, sync matching controllers
|
|
// to see if anyone wants to adopt it now.
|
|
if labelChanged || controllerRefChanged {
|
|
rss := rsc.getPodReplicaSets(curPod)
|
|
if len(rss) == 0 {
|
|
return
|
|
}
|
|
klog.V(4).Infof("Orphan Pod %s updated, objectMeta %+v -> %+v.", curPod.Name, oldPod.ObjectMeta, curPod.ObjectMeta)
|
|
for _, rs := range rss {
|
|
rsc.enqueueRS(rs)
|
|
}
|
|
}
|
|
}
|
|
|
|
// When a pod is deleted, enqueue the replica set that manages the pod and update its expectations.
|
|
// obj could be an *v1.Pod, or a DeletionFinalStateUnknown marker item.
|
|
func (rsc *ReplicaSetController) deletePod(obj interface{}) {
|
|
pod, ok := obj.(*v1.Pod)
|
|
|
|
// When a delete is dropped, the relist will notice a pod in the store not
|
|
// in the list, leading to the insertion of a tombstone object which contains
|
|
// the deleted key/value. Note that this value might be stale. If the pod
|
|
// changed labels the new ReplicaSet will not be woken up till the periodic resync.
|
|
if !ok {
|
|
tombstone, ok := obj.(cache.DeletedFinalStateUnknown)
|
|
if !ok {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get object from tombstone %+v", obj))
|
|
return
|
|
}
|
|
pod, ok = tombstone.Obj.(*v1.Pod)
|
|
if !ok {
|
|
utilruntime.HandleError(fmt.Errorf("tombstone contained object that is not a pod %#v", obj))
|
|
return
|
|
}
|
|
}
|
|
|
|
controllerRef := metav1.GetControllerOf(pod)
|
|
if controllerRef == nil {
|
|
// No controller should care about orphans being deleted.
|
|
return
|
|
}
|
|
rs := rsc.resolveControllerRef(pod.Namespace, controllerRef)
|
|
if rs == nil {
|
|
return
|
|
}
|
|
rsKey, err := controller.KeyFunc(rs)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get key for object %#v: %v", rs, err))
|
|
return
|
|
}
|
|
klog.V(4).Infof("Pod %s/%s deleted through %v, timestamp %+v: %#v.", pod.Namespace, pod.Name, utilruntime.GetCaller(), pod.DeletionTimestamp, pod)
|
|
rsc.expectations.DeletionObserved(rsKey, controller.PodKey(pod))
|
|
rsc.queue.Add(rsKey)
|
|
}
|
|
|
|
// worker runs a worker thread that just dequeues items, processes them, and marks them done.
|
|
// It enforces that the syncHandler is never invoked concurrently with the same key.
|
|
func (rsc *ReplicaSetController) worker() {
|
|
for rsc.processNextWorkItem() {
|
|
}
|
|
}
|
|
|
|
func (rsc *ReplicaSetController) processNextWorkItem() bool {
|
|
key, quit := rsc.queue.Get()
|
|
if quit {
|
|
return false
|
|
}
|
|
defer rsc.queue.Done(key)
|
|
|
|
err := rsc.syncHandler(key.(string))
|
|
if err == nil {
|
|
rsc.queue.Forget(key)
|
|
return true
|
|
}
|
|
|
|
utilruntime.HandleError(fmt.Errorf("sync %q failed with %v", key, err))
|
|
rsc.queue.AddRateLimited(key)
|
|
|
|
return true
|
|
}
|
|
|
|
// manageReplicas checks and updates replicas for the given ReplicaSet.
|
|
// Does NOT modify <filteredPods>.
|
|
// It will requeue the replica set in case of an error while creating/deleting pods.
|
|
func (rsc *ReplicaSetController) manageReplicas(filteredPods []*v1.Pod, rs *apps.ReplicaSet) error {
|
|
diff := len(filteredPods) - int(*(rs.Spec.Replicas))
|
|
rsKey, err := controller.KeyFunc(rs)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("couldn't get key for %v %#v: %v", rsc.Kind, rs, err))
|
|
return nil
|
|
}
|
|
if diff < 0 {
|
|
diff *= -1
|
|
if diff > rsc.burstReplicas {
|
|
diff = rsc.burstReplicas
|
|
}
|
|
// TODO: Track UIDs of creates just like deletes. The problem currently
|
|
// is we'd need to wait on the result of a create to record the pod's
|
|
// UID, which would require locking *across* the create, which will turn
|
|
// into a performance bottleneck. We should generate a UID for the pod
|
|
// beforehand and store it via ExpectCreations.
|
|
rsc.expectations.ExpectCreations(rsKey, diff)
|
|
klog.V(2).InfoS("Too few replicas", "replicaSet", klog.KObj(rs), "need", *(rs.Spec.Replicas), "creating", diff)
|
|
// Batch the pod creates. Batch sizes start at SlowStartInitialBatchSize
|
|
// and double with each successful iteration in a kind of "slow start".
|
|
// This handles attempts to start large numbers of pods that would
|
|
// likely all fail with the same error. For example a project with a
|
|
// low quota that attempts to create a large number of pods will be
|
|
// prevented from spamming the API service with the pod create requests
|
|
// after one of its pods fails. Conveniently, this also prevents the
|
|
// event spam that those failures would generate.
|
|
successfulCreations, err := slowStartBatch(diff, controller.SlowStartInitialBatchSize, func() error {
|
|
err := rsc.podControl.CreatePodsWithControllerRef(rs.Namespace, &rs.Spec.Template, rs, metav1.NewControllerRef(rs, rsc.GroupVersionKind))
|
|
if err != nil {
|
|
if errors.HasStatusCause(err, v1.NamespaceTerminatingCause) {
|
|
// if the namespace is being terminated, we don't have to do
|
|
// anything because any creation will fail
|
|
return nil
|
|
}
|
|
}
|
|
return err
|
|
})
|
|
|
|
// Any skipped pods that we never attempted to start shouldn't be expected.
|
|
// The skipped pods will be retried later. The next controller resync will
|
|
// retry the slow start process.
|
|
if skippedPods := diff - successfulCreations; skippedPods > 0 {
|
|
klog.V(2).Infof("Slow-start failure. Skipping creation of %d pods, decrementing expectations for %v %v/%v", skippedPods, rsc.Kind, rs.Namespace, rs.Name)
|
|
for i := 0; i < skippedPods; i++ {
|
|
// Decrement the expected number of creates because the informer won't observe this pod
|
|
rsc.expectations.CreationObserved(rsKey)
|
|
}
|
|
}
|
|
return err
|
|
} else if diff > 0 {
|
|
if diff > rsc.burstReplicas {
|
|
diff = rsc.burstReplicas
|
|
}
|
|
klog.V(2).InfoS("Too many replicas", "replicaSet", klog.KObj(rs), "need", *(rs.Spec.Replicas), "deleting", diff)
|
|
|
|
relatedPods, err := rsc.getIndirectlyRelatedPods(rs)
|
|
utilruntime.HandleError(err)
|
|
|
|
// Choose which Pods to delete, preferring those in earlier phases of startup.
|
|
podsToDelete := getPodsToDelete(filteredPods, relatedPods, diff)
|
|
|
|
// Snapshot the UIDs (ns/name) of the pods we're expecting to see
|
|
// deleted, so we know to record their expectations exactly once either
|
|
// when we see it as an update of the deletion timestamp, or as a delete.
|
|
// Note that if the labels on a pod/rs change in a way that the pod gets
|
|
// orphaned, the rs will only wake up after the expectations have
|
|
// expired even if other pods are deleted.
|
|
rsc.expectations.ExpectDeletions(rsKey, getPodKeys(podsToDelete))
|
|
|
|
errCh := make(chan error, diff)
|
|
var wg sync.WaitGroup
|
|
wg.Add(diff)
|
|
for _, pod := range podsToDelete {
|
|
go func(targetPod *v1.Pod) {
|
|
defer wg.Done()
|
|
if err := rsc.podControl.DeletePod(rs.Namespace, targetPod.Name, rs); err != nil {
|
|
// Decrement the expected number of deletes because the informer won't observe this deletion
|
|
podKey := controller.PodKey(targetPod)
|
|
rsc.expectations.DeletionObserved(rsKey, podKey)
|
|
if !apierrors.IsNotFound(err) {
|
|
klog.V(2).Infof("Failed to delete %v, decremented expectations for %v %s/%s", podKey, rsc.Kind, rs.Namespace, rs.Name)
|
|
errCh <- err
|
|
}
|
|
}
|
|
}(pod)
|
|
}
|
|
wg.Wait()
|
|
|
|
select {
|
|
case err := <-errCh:
|
|
// all errors have been reported before and they're likely to be the same, so we'll only return the first one we hit.
|
|
if err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// syncReplicaSet will sync the ReplicaSet with the given key if it has had its expectations fulfilled,
|
|
// meaning it did not expect to see any more of its pods created or deleted. This function is not meant to be
|
|
// invoked concurrently with the same key.
|
|
func (rsc *ReplicaSetController) syncReplicaSet(key string) error {
|
|
startTime := time.Now()
|
|
defer func() {
|
|
klog.V(4).Infof("Finished syncing %v %q (%v)", rsc.Kind, key, time.Since(startTime))
|
|
}()
|
|
|
|
namespace, name, err := cache.SplitMetaNamespaceKey(key)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
rs, err := rsc.rsLister.ReplicaSets(namespace).Get(name)
|
|
if errors.IsNotFound(err) {
|
|
klog.V(4).Infof("%v %v has been deleted", rsc.Kind, key)
|
|
rsc.expectations.DeleteExpectations(key)
|
|
return nil
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rsNeedsSync := rsc.expectations.SatisfiedExpectations(key)
|
|
selector, err := metav1.LabelSelectorAsSelector(rs.Spec.Selector)
|
|
if err != nil {
|
|
utilruntime.HandleError(fmt.Errorf("error converting pod selector to selector: %v", err))
|
|
return nil
|
|
}
|
|
|
|
// list all pods to include the pods that don't match the rs`s selector
|
|
// anymore but has the stale controller ref.
|
|
// TODO: Do the List and Filter in a single pass, or use an index.
|
|
allPods, err := rsc.podLister.Pods(rs.Namespace).List(labels.Everything())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Ignore inactive pods.
|
|
filteredPods := controller.FilterActivePods(allPods)
|
|
|
|
// NOTE: filteredPods are pointing to objects from cache - if you need to
|
|
// modify them, you need to copy it first.
|
|
filteredPods, err = rsc.claimPods(rs, selector, filteredPods)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var manageReplicasErr error
|
|
if rsNeedsSync && rs.DeletionTimestamp == nil {
|
|
manageReplicasErr = rsc.manageReplicas(filteredPods, rs)
|
|
}
|
|
rs = rs.DeepCopy()
|
|
newStatus := calculateStatus(rs, filteredPods, manageReplicasErr)
|
|
|
|
// Always updates status as pods come up or die.
|
|
updatedRS, err := updateReplicaSetStatus(rsc.kubeClient.AppsV1().ReplicaSets(rs.Namespace), rs, newStatus)
|
|
if err != nil {
|
|
// Multiple things could lead to this update failing. Requeuing the replica set ensures
|
|
// Returning an error causes a requeue without forcing a hotloop
|
|
return err
|
|
}
|
|
// Resync the ReplicaSet after MinReadySeconds as a last line of defense to guard against clock-skew.
|
|
if manageReplicasErr == nil && updatedRS.Spec.MinReadySeconds > 0 &&
|
|
updatedRS.Status.ReadyReplicas == *(updatedRS.Spec.Replicas) &&
|
|
updatedRS.Status.AvailableReplicas != *(updatedRS.Spec.Replicas) {
|
|
rsc.queue.AddAfter(key, time.Duration(updatedRS.Spec.MinReadySeconds)*time.Second)
|
|
}
|
|
return manageReplicasErr
|
|
}
|
|
|
|
func (rsc *ReplicaSetController) claimPods(rs *apps.ReplicaSet, selector labels.Selector, filteredPods []*v1.Pod) ([]*v1.Pod, error) {
|
|
// If any adoptions are attempted, we should first recheck for deletion with
|
|
// an uncached quorum read sometime after listing Pods (see #42639).
|
|
canAdoptFunc := controller.RecheckDeletionTimestamp(func() (metav1.Object, error) {
|
|
fresh, err := rsc.kubeClient.AppsV1().ReplicaSets(rs.Namespace).Get(context.TODO(), rs.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if fresh.UID != rs.UID {
|
|
return nil, fmt.Errorf("original %v %v/%v is gone: got uid %v, wanted %v", rsc.Kind, rs.Namespace, rs.Name, fresh.UID, rs.UID)
|
|
}
|
|
return fresh, nil
|
|
})
|
|
cm := controller.NewPodControllerRefManager(rsc.podControl, rs, selector, rsc.GroupVersionKind, canAdoptFunc)
|
|
return cm.ClaimPods(filteredPods)
|
|
}
|
|
|
|
// slowStartBatch tries to call the provided function a total of 'count' times,
|
|
// starting slow to check for errors, then speeding up if calls succeed.
|
|
//
|
|
// It groups the calls into batches, starting with a group of initialBatchSize.
|
|
// Within each batch, it may call the function multiple times concurrently.
|
|
//
|
|
// If a whole batch succeeds, the next batch may get exponentially larger.
|
|
// If there are any failures in a batch, all remaining batches are skipped
|
|
// after waiting for the current batch to complete.
|
|
//
|
|
// It returns the number of successful calls to the function.
|
|
func slowStartBatch(count int, initialBatchSize int, fn func() error) (int, error) {
|
|
remaining := count
|
|
successes := 0
|
|
for batchSize := integer.IntMin(remaining, initialBatchSize); batchSize > 0; batchSize = integer.IntMin(2*batchSize, remaining) {
|
|
errCh := make(chan error, batchSize)
|
|
var wg sync.WaitGroup
|
|
wg.Add(batchSize)
|
|
for i := 0; i < batchSize; i++ {
|
|
go func() {
|
|
defer wg.Done()
|
|
if err := fn(); err != nil {
|
|
errCh <- err
|
|
}
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
curSuccesses := batchSize - len(errCh)
|
|
successes += curSuccesses
|
|
if len(errCh) > 0 {
|
|
return successes, <-errCh
|
|
}
|
|
remaining -= batchSize
|
|
}
|
|
return successes, nil
|
|
}
|
|
|
|
// getIndirectlyRelatedPods returns all pods that are owned by any ReplicaSet
|
|
// that is owned by the given ReplicaSet's owner.
|
|
func (rsc *ReplicaSetController) getIndirectlyRelatedPods(rs *apps.ReplicaSet) ([]*v1.Pod, error) {
|
|
var relatedPods []*v1.Pod
|
|
seen := make(map[types.UID]*apps.ReplicaSet)
|
|
for _, relatedRS := range rsc.getReplicaSetsWithSameController(rs) {
|
|
selector, err := metav1.LabelSelectorAsSelector(relatedRS.Spec.Selector)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
pods, err := rsc.podLister.Pods(relatedRS.Namespace).List(selector)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, pod := range pods {
|
|
if otherRS, found := seen[pod.UID]; found {
|
|
klog.V(5).Infof("Pod %s/%s is owned by both %v %s/%s and %v %s/%s", pod.Namespace, pod.Name, rsc.Kind, otherRS.Namespace, otherRS.Name, rsc.Kind, relatedRS.Namespace, relatedRS.Name)
|
|
continue
|
|
}
|
|
seen[pod.UID] = relatedRS
|
|
relatedPods = append(relatedPods, pod)
|
|
}
|
|
}
|
|
if klog.V(4).Enabled() {
|
|
var relatedNames []string
|
|
for _, related := range relatedPods {
|
|
relatedNames = append(relatedNames, related.Name)
|
|
}
|
|
klog.Infof("Found %d related pods for %v %s/%s: %v", len(relatedPods), rsc.Kind, rs.Namespace, rs.Name, strings.Join(relatedNames, ", "))
|
|
}
|
|
return relatedPods, nil
|
|
}
|
|
|
|
func getPodsToDelete(filteredPods, relatedPods []*v1.Pod, diff int) []*v1.Pod {
|
|
// No need to sort pods if we are about to delete all of them.
|
|
// diff will always be <= len(filteredPods), so not need to handle > case.
|
|
if diff < len(filteredPods) {
|
|
podsWithRanks := getPodsRankedByRelatedPodsOnSameNode(filteredPods, relatedPods)
|
|
sort.Sort(podsWithRanks)
|
|
}
|
|
return filteredPods[:diff]
|
|
}
|
|
|
|
// getPodsRankedByRelatedPodsOnSameNode returns an ActivePodsWithRanks value
|
|
// that wraps podsToRank and assigns each pod a rank equal to the number of
|
|
// active pods in relatedPods that are colocated on the same node with the pod.
|
|
// relatedPods generally should be a superset of podsToRank.
|
|
func getPodsRankedByRelatedPodsOnSameNode(podsToRank, relatedPods []*v1.Pod) controller.ActivePodsWithRanks {
|
|
podsOnNode := make(map[string]int)
|
|
for _, pod := range relatedPods {
|
|
if controller.IsPodActive(pod) {
|
|
podsOnNode[pod.Spec.NodeName]++
|
|
}
|
|
}
|
|
ranks := make([]int, len(podsToRank))
|
|
for i, pod := range podsToRank {
|
|
ranks[i] = podsOnNode[pod.Spec.NodeName]
|
|
}
|
|
return controller.ActivePodsWithRanks{Pods: podsToRank, Rank: ranks}
|
|
}
|
|
|
|
func getPodKeys(pods []*v1.Pod) []string {
|
|
podKeys := make([]string, 0, len(pods))
|
|
for _, pod := range pods {
|
|
podKeys = append(podKeys, controller.PodKey(pod))
|
|
}
|
|
return podKeys
|
|
}
|