694 lines
27 KiB
Go
694 lines
27 KiB
Go
|
/*
|
||
|
Copyright 2014 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.
|
||
|
*/
|
||
|
|
||
|
// If you make changes to this file, you should also make the corresponding change in ReplicaSet.
|
||
|
|
||
|
package replication
|
||
|
|
||
|
import (
|
||
|
"reflect"
|
||
|
"sort"
|
||
|
"sync"
|
||
|
"time"
|
||
|
|
||
|
"github.com/golang/glog"
|
||
|
"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"
|
||
|
utilerrors "k8s.io/apimachinery/pkg/util/errors"
|
||
|
utilruntime "k8s.io/apimachinery/pkg/util/runtime"
|
||
|
"k8s.io/apimachinery/pkg/util/wait"
|
||
|
"k8s.io/kubernetes/pkg/api/v1"
|
||
|
"k8s.io/kubernetes/pkg/client/cache"
|
||
|
"k8s.io/kubernetes/pkg/client/clientset_generated/clientset"
|
||
|
v1core "k8s.io/kubernetes/pkg/client/clientset_generated/clientset/typed/core/v1"
|
||
|
"k8s.io/kubernetes/pkg/client/record"
|
||
|
"k8s.io/kubernetes/pkg/controller"
|
||
|
"k8s.io/kubernetes/pkg/controller/informers"
|
||
|
"k8s.io/kubernetes/pkg/util"
|
||
|
"k8s.io/kubernetes/pkg/util/metrics"
|
||
|
"k8s.io/kubernetes/pkg/util/workqueue"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
// Realistic value of the burstReplica field for the replication manager based off
|
||
|
// performance requirements for kubernetes 1.0.
|
||
|
BurstReplicas = 500
|
||
|
|
||
|
// The number of times we retry updating a replication controller's status.
|
||
|
statusUpdateRetries = 1
|
||
|
)
|
||
|
|
||
|
func getRCKind() schema.GroupVersionKind {
|
||
|
return v1.SchemeGroupVersion.WithKind("ReplicationController")
|
||
|
}
|
||
|
|
||
|
// ReplicationManager is responsible for synchronizing ReplicationController objects stored
|
||
|
// in the system with actual running pods.
|
||
|
// TODO: this really should be called ReplicationController. The only reason why it's a Manager
|
||
|
// is to distinguish this type from API object "ReplicationController". We should fix this.
|
||
|
type ReplicationManager struct {
|
||
|
kubeClient clientset.Interface
|
||
|
podControl controller.PodControlInterface
|
||
|
|
||
|
// internalPodInformer is used to hold a personal informer. If we're using
|
||
|
// a normal shared informer, then the informer will be started for us. If
|
||
|
// we have a personal informer, we must start it ourselves. If you start
|
||
|
// the controller using NewReplicationManager(passing SharedInformer), this
|
||
|
// will be null
|
||
|
internalPodInformer cache.SharedIndexInformer
|
||
|
|
||
|
// An rc 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 syncReplicationController for testing.
|
||
|
syncHandler func(rcKey string) error
|
||
|
|
||
|
// A TTLCache of pod creates/deletes each rc expects to see.
|
||
|
expectations *controller.UIDTrackingControllerExpectations
|
||
|
|
||
|
// A store of replication controllers, populated by the rcController
|
||
|
rcLister cache.StoreToReplicationControllerLister
|
||
|
// A store of pods, populated by the podController
|
||
|
podLister cache.StoreToPodLister
|
||
|
// Watches changes to all pods
|
||
|
podController cache.Controller
|
||
|
// 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 func() bool
|
||
|
|
||
|
lookupCache *controller.MatchingCache
|
||
|
|
||
|
// Controllers that need to be synced
|
||
|
queue workqueue.RateLimitingInterface
|
||
|
|
||
|
// garbageCollectorEnabled denotes if the garbage collector is enabled. RC
|
||
|
// manager behaves differently if GC is enabled.
|
||
|
garbageCollectorEnabled bool
|
||
|
}
|
||
|
|
||
|
// NewReplicationManager configures a replication manager with the specified event recorder
|
||
|
func NewReplicationManager(podInformer, rcInformer cache.SharedIndexInformer, kubeClient clientset.Interface, burstReplicas int, lookupCacheSize int, garbageCollectorEnabled bool) *ReplicationManager {
|
||
|
if kubeClient != nil && kubeClient.Core().RESTClient().GetRateLimiter() != nil {
|
||
|
metrics.RegisterMetricAndTrackRateLimiterUsage("replication_controller", kubeClient.Core().RESTClient().GetRateLimiter())
|
||
|
}
|
||
|
|
||
|
eventBroadcaster := record.NewBroadcaster()
|
||
|
eventBroadcaster.StartLogging(glog.Infof)
|
||
|
eventBroadcaster.StartRecordingToSink(&v1core.EventSinkImpl{Interface: kubeClient.Core().Events("")})
|
||
|
|
||
|
rm := &ReplicationManager{
|
||
|
kubeClient: kubeClient,
|
||
|
podControl: controller.RealPodControl{
|
||
|
KubeClient: kubeClient,
|
||
|
Recorder: eventBroadcaster.NewRecorder(v1.EventSource{Component: "replication-controller"}),
|
||
|
},
|
||
|
burstReplicas: burstReplicas,
|
||
|
expectations: controller.NewUIDTrackingControllerExpectations(controller.NewControllerExpectations()),
|
||
|
queue: workqueue.NewNamedRateLimitingQueue(workqueue.DefaultControllerRateLimiter(), "replicationmanager"),
|
||
|
garbageCollectorEnabled: garbageCollectorEnabled,
|
||
|
}
|
||
|
|
||
|
rcInformer.AddEventHandler(cache.ResourceEventHandlerFuncs{
|
||
|
AddFunc: rm.enqueueController,
|
||
|
UpdateFunc: rm.updateRC,
|
||
|
// This will enter the sync loop and no-op, because the controller has been deleted from the store.
|
||
|
// Note that deleting a controller immediately after scaling it to 0 will not work. The recommended
|
||
|
// way of achieving this is by performing a `stop` operation on the controller.
|
||
|
DeleteFunc: rm.enqueueController,
|
||
|
})
|
||
|
podInformer.AddEventHandler(cache.ResourceEventHandlerFuncs{
|
||
|
AddFunc: rm.addPod,
|
||
|
// This invokes the rc for every pod change, eg: host assignment. Though this might seem like overkill
|
||
|
// the most frequent pod update is status, and the associated rc will only list from local storage, so
|
||
|
// it should be ok.
|
||
|
UpdateFunc: rm.updatePod,
|
||
|
DeleteFunc: rm.deletePod,
|
||
|
})
|
||
|
|
||
|
rm.syncHandler = rm.syncReplicationController
|
||
|
rm.rcLister.Indexer = rcInformer.GetIndexer()
|
||
|
rm.podLister.Indexer = podInformer.GetIndexer()
|
||
|
rm.podListerSynced = podInformer.HasSynced
|
||
|
rm.lookupCache = controller.NewMatchingCache(lookupCacheSize)
|
||
|
return rm
|
||
|
}
|
||
|
|
||
|
// NewReplicationManagerFromClient creates a new ReplicationManager that runs its own informer.
|
||
|
func NewReplicationManagerFromClient(kubeClient clientset.Interface, resyncPeriod controller.ResyncPeriodFunc, burstReplicas int, lookupCacheSize int) *ReplicationManager {
|
||
|
podInformer := informers.NewPodInformer(kubeClient, resyncPeriod())
|
||
|
rcInformer := informers.NewReplicationControllerInformer(kubeClient, resyncPeriod())
|
||
|
rm := NewReplicationManager(podInformer, rcInformer, kubeClient, burstReplicas, lookupCacheSize, false)
|
||
|
rm.internalPodInformer = podInformer
|
||
|
return rm
|
||
|
}
|
||
|
|
||
|
// SetEventRecorder replaces the event recorder used by the replication manager
|
||
|
// with the given recorder. Only used for testing.
|
||
|
func (rm *ReplicationManager) SetEventRecorder(recorder record.EventRecorder) {
|
||
|
// TODO: Hack. We can't cleanly shutdown the event recorder, so benchmarks
|
||
|
// need to pass in a fake.
|
||
|
rm.podControl = controller.RealPodControl{KubeClient: rm.kubeClient, Recorder: recorder}
|
||
|
}
|
||
|
|
||
|
// Run begins watching and syncing.
|
||
|
func (rm *ReplicationManager) Run(workers int, stopCh <-chan struct{}) {
|
||
|
defer utilruntime.HandleCrash()
|
||
|
defer rm.queue.ShutDown()
|
||
|
|
||
|
glog.Infof("Starting RC Manager")
|
||
|
if rm.internalPodInformer != nil {
|
||
|
go rm.internalPodInformer.Run(stopCh)
|
||
|
}
|
||
|
|
||
|
if !cache.WaitForCacheSync(stopCh, rm.podListerSynced) {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
for i := 0; i < workers; i++ {
|
||
|
go wait.Until(rm.worker, time.Second, stopCh)
|
||
|
}
|
||
|
|
||
|
<-stopCh
|
||
|
glog.Infof("Shutting down RC Manager")
|
||
|
}
|
||
|
|
||
|
// getPodController returns the controller managing the given pod.
|
||
|
// TODO: Surface that we are ignoring multiple controllers for a single pod.
|
||
|
// TODO: use ownerReference.Controller to determine if the rc controls the pod.
|
||
|
func (rm *ReplicationManager) getPodController(pod *v1.Pod) *v1.ReplicationController {
|
||
|
// look up in the cache, if cached and the cache is valid, just return cached value
|
||
|
if obj, cached := rm.lookupCache.GetMatchingObject(pod); cached {
|
||
|
controller, ok := obj.(*v1.ReplicationController)
|
||
|
if !ok {
|
||
|
// This should not happen
|
||
|
glog.Errorf("lookup cache does not return a ReplicationController object")
|
||
|
return nil
|
||
|
}
|
||
|
if cached && rm.isCacheValid(pod, controller) {
|
||
|
return controller
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// if not cached or cached value is invalid, search all the rc to find the matching one, and update cache
|
||
|
controllers, err := rm.rcLister.GetPodControllers(pod)
|
||
|
if err != nil {
|
||
|
glog.V(4).Infof("No controllers found for pod %v, replication manager will avoid syncing", pod.Name)
|
||
|
return nil
|
||
|
}
|
||
|
// In theory, overlapping controllers is user error. This sorting will not prevent
|
||
|
// oscillation of replicas in all cases, eg:
|
||
|
// rc1 (older rc): [(k1=v1)], replicas=1 rc2: [(k2=v2)], replicas=2
|
||
|
// pod: [(k1:v1), (k2:v2)] will wake both rc1 and rc2, and we will sync rc1.
|
||
|
// pod: [(k2:v2)] will wake rc2 which creates a new replica.
|
||
|
if len(controllers) > 1 {
|
||
|
// More than two items in this list indicates user error. If two replication-controller
|
||
|
// overlap, sort by creation timestamp, subsort by name, then pick
|
||
|
// the first.
|
||
|
glog.Errorf("user error! more than one replication controller is selecting pods with labels: %+v", pod.Labels)
|
||
|
sort.Sort(OverlappingControllers(controllers))
|
||
|
}
|
||
|
|
||
|
// update lookup cache
|
||
|
rm.lookupCache.Update(pod, controllers[0])
|
||
|
|
||
|
return controllers[0]
|
||
|
}
|
||
|
|
||
|
// isCacheValid check if the cache is valid
|
||
|
func (rm *ReplicationManager) isCacheValid(pod *v1.Pod, cachedRC *v1.ReplicationController) bool {
|
||
|
_, err := rm.rcLister.ReplicationControllers(cachedRC.Namespace).Get(cachedRC.Name)
|
||
|
// rc has been deleted or updated, cache is invalid
|
||
|
if err != nil || !isControllerMatch(pod, cachedRC) {
|
||
|
return false
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
// isControllerMatch take a Pod and ReplicationController, return whether the Pod and ReplicationController are matching
|
||
|
// TODO(mqliang): This logic is a copy from GetPodControllers(), remove the duplication
|
||
|
func isControllerMatch(pod *v1.Pod, rc *v1.ReplicationController) bool {
|
||
|
if rc.Namespace != pod.Namespace {
|
||
|
return false
|
||
|
}
|
||
|
selector := labels.Set(rc.Spec.Selector).AsSelectorPreValidated()
|
||
|
|
||
|
// If an rc with a nil or empty selector creeps in, it should match nothing, not everything.
|
||
|
if selector.Empty() || !selector.Matches(labels.Set(pod.Labels)) {
|
||
|
return false
|
||
|
}
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
// callback when RC is updated
|
||
|
func (rm *ReplicationManager) updateRC(old, cur interface{}) {
|
||
|
oldRC := old.(*v1.ReplicationController)
|
||
|
curRC := cur.(*v1.ReplicationController)
|
||
|
|
||
|
// We should invalidate the whole lookup cache if a RC's selector has been updated.
|
||
|
//
|
||
|
// Imagine that you have two RCs:
|
||
|
// * old RC1
|
||
|
// * new RC2
|
||
|
// You also have a pod that is attached to RC2 (because it doesn't match RC1 selector).
|
||
|
// Now imagine that you are changing RC1 selector so that it is now matching that pod,
|
||
|
// in such case, we must invalidate the whole cache so that pod could be adopted by RC1
|
||
|
//
|
||
|
// This makes the lookup cache less helpful, but selector update does not happen often,
|
||
|
// so it's not a big problem
|
||
|
if !reflect.DeepEqual(oldRC.Spec.Selector, curRC.Spec.Selector) {
|
||
|
rm.lookupCache.InvalidateAll()
|
||
|
}
|
||
|
// TODO: Remove when #31981 is resolved!
|
||
|
glog.Infof("Observed updated replication controller %v. Desired pod count change: %d->%d", curRC.Name, *(oldRC.Spec.Replicas), *(curRC.Spec.Replicas))
|
||
|
|
||
|
// You might imagine that we only really need to enqueue the
|
||
|
// controller when Spec changes, but it is safer to sync any
|
||
|
// time this function is triggered. That way a full informer
|
||
|
// resync can requeue any controllers 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
|
||
|
// controllers 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 rcs that haven't met expectations yet won't
|
||
|
// sync, and all the listing is done using local stores.
|
||
|
if oldRC.Status.Replicas != curRC.Status.Replicas {
|
||
|
// TODO: Should we log status or spec?
|
||
|
glog.V(4).Infof("Observed updated replica count for rc: %v, %d->%d", curRC.Name, oldRC.Status.Replicas, curRC.Status.Replicas)
|
||
|
}
|
||
|
rm.enqueueController(cur)
|
||
|
}
|
||
|
|
||
|
// When a pod is created, enqueue the controller that manages it and update it's expectations.
|
||
|
func (rm *ReplicationManager) addPod(obj interface{}) {
|
||
|
pod := obj.(*v1.Pod)
|
||
|
|
||
|
rc := rm.getPodController(pod)
|
||
|
if rc == nil {
|
||
|
return
|
||
|
}
|
||
|
rcKey, err := controller.KeyFunc(rc)
|
||
|
if err != nil {
|
||
|
glog.Errorf("Couldn't get key for replication controller %#v: %v", rc, err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
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.
|
||
|
rm.deletePod(pod)
|
||
|
return
|
||
|
}
|
||
|
rm.expectations.CreationObserved(rcKey)
|
||
|
rm.enqueueController(rc)
|
||
|
}
|
||
|
|
||
|
// When a pod is updated, figure out what controller/s manage it and wake them
|
||
|
// up. If the labels of the pod have changed we need to awaken both the old
|
||
|
// and new controller. old and cur must be *v1.Pod types.
|
||
|
func (rm *ReplicationManager) 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
|
||
|
}
|
||
|
glog.V(4).Infof("Pod %s updated, objectMeta %+v -> %+v.", curPod.Name, oldPod.ObjectMeta, curPod.ObjectMeta)
|
||
|
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 rc 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 rc never initiates a phase change, and so is never asleep waiting for the same.
|
||
|
rm.deletePod(curPod)
|
||
|
if labelChanged {
|
||
|
// we don't need to check the oldPod.DeletionTimestamp because DeletionTimestamp cannot be unset.
|
||
|
rm.deletePod(oldPod)
|
||
|
}
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// Only need to get the old controller if the labels changed.
|
||
|
// Enqueue the oldRC before the curRC to give curRC a chance to adopt the oldPod.
|
||
|
if labelChanged {
|
||
|
// If the old and new rc are the same, the first one that syncs
|
||
|
// will set expectations preventing any damage from the second.
|
||
|
if oldRC := rm.getPodController(oldPod); oldRC != nil {
|
||
|
rm.enqueueController(oldRC)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
changedToReady := !v1.IsPodReady(oldPod) && v1.IsPodReady(curPod)
|
||
|
if curRC := rm.getPodController(curPod); curRC != nil {
|
||
|
rm.enqueueController(curRC)
|
||
|
// 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 replication controller
|
||
|
// 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 replication controller manager).
|
||
|
if changedToReady && curRC.Spec.MinReadySeconds > 0 {
|
||
|
glog.V(2).Infof("ReplicationController %q will be enqueued after %ds for availability check", curRC.Name, curRC.Spec.MinReadySeconds)
|
||
|
rm.enqueueControllerAfter(curRC, time.Duration(curRC.Spec.MinReadySeconds)*time.Second)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// When a pod is deleted, enqueue the controller that manages the pod and update its expectations.
|
||
|
// obj could be an *v1.Pod, or a DeletionFinalStateUnknown marker item.
|
||
|
func (rm *ReplicationManager) 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 rc will not be woken up till the periodic resync.
|
||
|
if !ok {
|
||
|
tombstone, ok := obj.(cache.DeletedFinalStateUnknown)
|
||
|
if !ok {
|
||
|
glog.Errorf("Couldn't get object from tombstone %#v", obj)
|
||
|
return
|
||
|
}
|
||
|
pod, ok = tombstone.Obj.(*v1.Pod)
|
||
|
if !ok {
|
||
|
glog.Errorf("Tombstone contained object that is not a pod %#v", obj)
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
glog.V(4).Infof("Pod %s/%s deleted through %v, timestamp %+v, labels %+v.", pod.Namespace, pod.Name, utilruntime.GetCaller(), pod.DeletionTimestamp, pod.Labels)
|
||
|
if rc := rm.getPodController(pod); rc != nil {
|
||
|
rcKey, err := controller.KeyFunc(rc)
|
||
|
if err != nil {
|
||
|
glog.Errorf("Couldn't get key for replication controller %#v: %v", rc, err)
|
||
|
return
|
||
|
}
|
||
|
rm.expectations.DeletionObserved(rcKey, controller.PodKey(pod))
|
||
|
rm.enqueueController(rc)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// obj could be an *v1.ReplicationController, or a DeletionFinalStateUnknown marker item.
|
||
|
func (rm *ReplicationManager) enqueueController(obj interface{}) {
|
||
|
key, err := controller.KeyFunc(obj)
|
||
|
if err != nil {
|
||
|
glog.Errorf("Couldn't get key for object %+v: %v", obj, err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// TODO: Handle overlapping controllers better. Either disallow them at admission time or
|
||
|
// deterministically avoid syncing controllers that fight over pods. Currently, we only
|
||
|
// ensure that the same controller is synced for a given pod. When we periodically relist
|
||
|
// all controllers there will still be some replica instability. One way to handle this is
|
||
|
// by querying the store for all controllers that this rc overlaps, as well as all
|
||
|
// controllers that overlap this rc, and sorting them.
|
||
|
rm.queue.Add(key)
|
||
|
}
|
||
|
|
||
|
// obj could be an *v1.ReplicationController, or a DeletionFinalStateUnknown marker item.
|
||
|
func (rm *ReplicationManager) enqueueControllerAfter(obj interface{}, after time.Duration) {
|
||
|
key, err := controller.KeyFunc(obj)
|
||
|
if err != nil {
|
||
|
glog.Errorf("Couldn't get key for object %+v: %v", obj, err)
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// TODO: Handle overlapping controllers better. Either disallow them at admission time or
|
||
|
// deterministically avoid syncing controllers that fight over pods. Currently, we only
|
||
|
// ensure that the same controller is synced for a given pod. When we periodically relist
|
||
|
// all controllers there will still be some replica instability. One way to handle this is
|
||
|
// by querying the store for all controllers that this rc overlaps, as well as all
|
||
|
// controllers that overlap this rc, and sorting them.
|
||
|
rm.queue.AddAfter(key, after)
|
||
|
}
|
||
|
|
||
|
// 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 (rm *ReplicationManager) worker() {
|
||
|
workFunc := func() bool {
|
||
|
key, quit := rm.queue.Get()
|
||
|
if quit {
|
||
|
return true
|
||
|
}
|
||
|
defer rm.queue.Done(key)
|
||
|
|
||
|
err := rm.syncHandler(key.(string))
|
||
|
if err == nil {
|
||
|
rm.queue.Forget(key)
|
||
|
return false
|
||
|
}
|
||
|
|
||
|
rm.queue.AddRateLimited(key)
|
||
|
utilruntime.HandleError(err)
|
||
|
return false
|
||
|
}
|
||
|
for {
|
||
|
if quit := workFunc(); quit {
|
||
|
glog.Infof("replication controller worker shutting down")
|
||
|
return
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// manageReplicas checks and updates replicas for the given replication controller.
|
||
|
// Does NOT modify <filteredPods>.
|
||
|
func (rm *ReplicationManager) manageReplicas(filteredPods []*v1.Pod, rc *v1.ReplicationController) error {
|
||
|
diff := len(filteredPods) - int(*(rc.Spec.Replicas))
|
||
|
rcKey, err := controller.KeyFunc(rc)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if diff == 0 {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
if diff < 0 {
|
||
|
diff *= -1
|
||
|
if diff > rm.burstReplicas {
|
||
|
diff = rm.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.
|
||
|
errCh := make(chan error, diff)
|
||
|
rm.expectations.ExpectCreations(rcKey, diff)
|
||
|
var wg sync.WaitGroup
|
||
|
wg.Add(diff)
|
||
|
glog.V(2).Infof("Too few %q/%q replicas, need %d, creating %d", rc.Namespace, rc.Name, *(rc.Spec.Replicas), diff)
|
||
|
for i := 0; i < diff; i++ {
|
||
|
go func() {
|
||
|
defer wg.Done()
|
||
|
var err error
|
||
|
if rm.garbageCollectorEnabled {
|
||
|
var trueVar = true
|
||
|
controllerRef := &metav1.OwnerReference{
|
||
|
APIVersion: getRCKind().GroupVersion().String(),
|
||
|
Kind: getRCKind().Kind,
|
||
|
Name: rc.Name,
|
||
|
UID: rc.UID,
|
||
|
Controller: &trueVar,
|
||
|
}
|
||
|
err = rm.podControl.CreatePodsWithControllerRef(rc.Namespace, rc.Spec.Template, rc, controllerRef)
|
||
|
} else {
|
||
|
err = rm.podControl.CreatePods(rc.Namespace, rc.Spec.Template, rc)
|
||
|
}
|
||
|
if err != nil {
|
||
|
// Decrement the expected number of creates because the informer won't observe this pod
|
||
|
glog.V(2).Infof("Failed creation, decrementing expectations for controller %q/%q", rc.Namespace, rc.Name)
|
||
|
rm.expectations.CreationObserved(rcKey)
|
||
|
errCh <- err
|
||
|
utilruntime.HandleError(err)
|
||
|
}
|
||
|
}()
|
||
|
}
|
||
|
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
|
||
|
}
|
||
|
|
||
|
if diff > rm.burstReplicas {
|
||
|
diff = rm.burstReplicas
|
||
|
}
|
||
|
glog.V(2).Infof("Too many %q/%q replicas, need %d, deleting %d", rc.Namespace, rc.Name, *(rc.Spec.Replicas), diff)
|
||
|
// No need to sort pods if we are about to delete all of them
|
||
|
if *(rc.Spec.Replicas) != 0 {
|
||
|
// Sort the pods in the order such that not-ready < ready, unscheduled
|
||
|
// < scheduled, and pending < running. This ensures that we delete pods
|
||
|
// in the earlier stages whenever possible.
|
||
|
sort.Sort(controller.ActivePods(filteredPods))
|
||
|
}
|
||
|
// 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/rc 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.
|
||
|
deletedPodKeys := []string{}
|
||
|
for i := 0; i < diff; i++ {
|
||
|
deletedPodKeys = append(deletedPodKeys, controller.PodKey(filteredPods[i]))
|
||
|
}
|
||
|
// We use pod namespace/name as a UID to wait for deletions, so if the
|
||
|
// labels on a pod/rc change in a way that the pod gets orphaned, the
|
||
|
// rc will only wake up after the expectation has expired.
|
||
|
errCh := make(chan error, diff)
|
||
|
rm.expectations.ExpectDeletions(rcKey, deletedPodKeys)
|
||
|
var wg sync.WaitGroup
|
||
|
wg.Add(diff)
|
||
|
for i := 0; i < diff; i++ {
|
||
|
go func(ix int) {
|
||
|
defer wg.Done()
|
||
|
if err := rm.podControl.DeletePod(rc.Namespace, filteredPods[ix].Name, rc); err != nil {
|
||
|
// Decrement the expected number of deletes because the informer won't observe this deletion
|
||
|
podKey := controller.PodKey(filteredPods[ix])
|
||
|
glog.V(2).Infof("Failed to delete %v due to %v, decrementing expectations for controller %q/%q", podKey, err, rc.Namespace, rc.Name)
|
||
|
rm.expectations.DeletionObserved(rcKey, podKey)
|
||
|
errCh <- err
|
||
|
utilruntime.HandleError(err)
|
||
|
}
|
||
|
}(i)
|
||
|
}
|
||
|
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
|
||
|
|
||
|
}
|
||
|
|
||
|
// syncReplicationController will sync the rc 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 (rm *ReplicationManager) syncReplicationController(key string) error {
|
||
|
trace := util.NewTrace("syncReplicationController: " + key)
|
||
|
defer trace.LogIfLong(250 * time.Millisecond)
|
||
|
|
||
|
startTime := time.Now()
|
||
|
defer func() {
|
||
|
glog.V(4).Infof("Finished syncing controller %q (%v)", key, time.Now().Sub(startTime))
|
||
|
}()
|
||
|
|
||
|
obj, exists, err := rm.rcLister.Indexer.GetByKey(key)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if !exists {
|
||
|
glog.Infof("Replication Controller has been deleted %v", key)
|
||
|
rm.expectations.DeleteExpectations(key)
|
||
|
return nil
|
||
|
}
|
||
|
rc := *obj.(*v1.ReplicationController)
|
||
|
|
||
|
trace.Step("ReplicationController restored")
|
||
|
rcNeedsSync := rm.expectations.SatisfiedExpectations(key)
|
||
|
trace.Step("Expectations restored")
|
||
|
|
||
|
// NOTE: filteredPods are pointing to objects from cache - if you need to
|
||
|
// modify them, you need to copy it first.
|
||
|
// TODO: Do the List and Filter in a single pass, or use an index.
|
||
|
var filteredPods []*v1.Pod
|
||
|
if rm.garbageCollectorEnabled {
|
||
|
// list all pods to include the pods that don't match the rc's selector
|
||
|
// anymore but has the stale controller ref.
|
||
|
pods, err := rm.podLister.Pods(rc.Namespace).List(labels.Everything())
|
||
|
if err != nil {
|
||
|
glog.Errorf("Error getting pods for rc %q: %v", key, err)
|
||
|
rm.queue.Add(key)
|
||
|
return err
|
||
|
}
|
||
|
cm := controller.NewPodControllerRefManager(rm.podControl, rc.ObjectMeta, labels.Set(rc.Spec.Selector).AsSelectorPreValidated(), getRCKind())
|
||
|
matchesAndControlled, matchesNeedsController, controlledDoesNotMatch := cm.Classify(pods)
|
||
|
// Adopt pods only if this replication controller is not going to be deleted.
|
||
|
if rc.DeletionTimestamp == nil {
|
||
|
for _, pod := range matchesNeedsController {
|
||
|
err := cm.AdoptPod(pod)
|
||
|
// continue to next pod if adoption fails.
|
||
|
if err != nil {
|
||
|
// If the pod no longer exists, don't even log the error.
|
||
|
if !errors.IsNotFound(err) {
|
||
|
utilruntime.HandleError(err)
|
||
|
}
|
||
|
} else {
|
||
|
matchesAndControlled = append(matchesAndControlled, pod)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
filteredPods = matchesAndControlled
|
||
|
// remove the controllerRef for the pods that no longer have matching labels
|
||
|
var errlist []error
|
||
|
for _, pod := range controlledDoesNotMatch {
|
||
|
err := cm.ReleasePod(pod)
|
||
|
if err != nil {
|
||
|
errlist = append(errlist, err)
|
||
|
}
|
||
|
}
|
||
|
if len(errlist) != 0 {
|
||
|
aggregate := utilerrors.NewAggregate(errlist)
|
||
|
// push the RC into work queue again. We need to try to free the
|
||
|
// pods again otherwise they will stuck with the stale
|
||
|
// controllerRef.
|
||
|
rm.queue.Add(key)
|
||
|
return aggregate
|
||
|
}
|
||
|
} else {
|
||
|
pods, err := rm.podLister.Pods(rc.Namespace).List(labels.Set(rc.Spec.Selector).AsSelectorPreValidated())
|
||
|
if err != nil {
|
||
|
glog.Errorf("Error getting pods for rc %q: %v", key, err)
|
||
|
rm.queue.Add(key)
|
||
|
return err
|
||
|
}
|
||
|
filteredPods = controller.FilterActivePods(pods)
|
||
|
}
|
||
|
|
||
|
var manageReplicasErr error
|
||
|
if rcNeedsSync && rc.DeletionTimestamp == nil {
|
||
|
manageReplicasErr = rm.manageReplicas(filteredPods, &rc)
|
||
|
}
|
||
|
trace.Step("manageReplicas done")
|
||
|
|
||
|
newStatus := calculateStatus(rc, filteredPods, manageReplicasErr)
|
||
|
|
||
|
// Always updates status as pods come up or die.
|
||
|
if err := updateReplicationControllerStatus(rm.kubeClient.Core().ReplicationControllers(rc.Namespace), rc, newStatus); err != nil {
|
||
|
// Multiple things could lead to this update failing. Returning an error causes a requeue without forcing a hotloop
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return manageReplicasErr
|
||
|
}
|