mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-10-29 21:01:02 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			2001 lines
		
	
	
		
			77 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			2001 lines
		
	
	
		
			77 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| /*
 | |
| Copyright 2015 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.
 | |
| */
 | |
| 
 | |
| package job
 | |
| 
 | |
| import (
 | |
| 	"context"
 | |
| 	"fmt"
 | |
| 	"reflect"
 | |
| 	"sort"
 | |
| 	"sync"
 | |
| 	"sync/atomic"
 | |
| 	"time"
 | |
| 
 | |
| 	batch "k8s.io/api/batch/v1"
 | |
| 	v1 "k8s.io/api/core/v1"
 | |
| 	apierrors "k8s.io/apimachinery/pkg/api/errors"
 | |
| 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 | |
| 	"k8s.io/apimachinery/pkg/labels"
 | |
| 	"k8s.io/apimachinery/pkg/types"
 | |
| 	"k8s.io/apimachinery/pkg/util/json"
 | |
| 	utilruntime "k8s.io/apimachinery/pkg/util/runtime"
 | |
| 	"k8s.io/apimachinery/pkg/util/sets"
 | |
| 	"k8s.io/apimachinery/pkg/util/wait"
 | |
| 	"k8s.io/apiserver/pkg/util/feature"
 | |
| 	batchinformers "k8s.io/client-go/informers/batch/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"
 | |
| 	batchv1listers "k8s.io/client-go/listers/batch/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/klog/v2"
 | |
| 	podutil "k8s.io/kubernetes/pkg/api/v1/pod"
 | |
| 	"k8s.io/kubernetes/pkg/controller"
 | |
| 	"k8s.io/kubernetes/pkg/controller/job/metrics"
 | |
| 	"k8s.io/kubernetes/pkg/controller/job/util"
 | |
| 	"k8s.io/kubernetes/pkg/features"
 | |
| 	"k8s.io/utils/clock"
 | |
| 	"k8s.io/utils/ptr"
 | |
| )
 | |
| 
 | |
| // controllerKind contains the schema.GroupVersionKind for this controller type.
 | |
| var controllerKind = batch.SchemeGroupVersion.WithKind("Job")
 | |
| 
 | |
| var (
 | |
| 	// syncJobBatchPeriod is the batch period for controller sync invocations for a Job.
 | |
| 	syncJobBatchPeriod = time.Second
 | |
| 	// DefaultJobApiBackOff is the default API backoff period. Exported for tests.
 | |
| 	DefaultJobApiBackOff = time.Second
 | |
| 	// MaxJobApiBackOff is the max API backoff period. Exported for tests.
 | |
| 	MaxJobApiBackOff = time.Minute
 | |
| 	// DefaultJobPodFailureBackOff is the default pod failure backoff period. Exported for tests.
 | |
| 	DefaultJobPodFailureBackOff = 10 * time.Second
 | |
| 	// MaxJobPodFailureBackOff is the max  pod failure backoff period. Exported for tests.
 | |
| 	MaxJobPodFailureBackOff = 10 * time.Minute
 | |
| 	// MaxUncountedPods is the maximum size the slices in
 | |
| 	// .status.uncountedTerminatedPods should have to keep their representation
 | |
| 	// roughly below 20 KB. Exported for tests
 | |
| 	MaxUncountedPods = 500
 | |
| 	// MaxPodCreateDeletePerSync is the maximum number of pods that can be
 | |
| 	// created or deleted in a single sync call. Exported for tests.
 | |
| 	MaxPodCreateDeletePerSync = 500
 | |
| )
 | |
| 
 | |
| // Controller ensures that all Job objects have corresponding pods to
 | |
| // run their configured workload.
 | |
| type Controller struct {
 | |
| 	kubeClient clientset.Interface
 | |
| 	podControl controller.PodControlInterface
 | |
| 
 | |
| 	// To allow injection of the following for testing.
 | |
| 	updateStatusHandler func(ctx context.Context, job *batch.Job) (*batch.Job, error)
 | |
| 	patchJobHandler     func(ctx context.Context, job *batch.Job, patch []byte) error
 | |
| 	syncHandler         func(ctx context.Context, jobKey string) error
 | |
| 	// podStoreSynced returns true if the pod store has been synced at least once.
 | |
| 	// Added as a member to the struct to allow injection for testing.
 | |
| 	podStoreSynced cache.InformerSynced
 | |
| 	// jobStoreSynced returns true if the job store has been synced at least once.
 | |
| 	// Added as a member to the struct to allow injection for testing.
 | |
| 	jobStoreSynced cache.InformerSynced
 | |
| 
 | |
| 	// A TTLCache of pod creates/deletes each rc expects to see
 | |
| 	expectations controller.ControllerExpectationsInterface
 | |
| 
 | |
| 	// finalizerExpectations tracks the Pod UIDs for which the controller
 | |
| 	// expects to observe the tracking finalizer removed.
 | |
| 	finalizerExpectations *uidTrackingExpectations
 | |
| 
 | |
| 	// A store of jobs
 | |
| 	jobLister batchv1listers.JobLister
 | |
| 
 | |
| 	// A store of pods, populated by the podController
 | |
| 	podStore corelisters.PodLister
 | |
| 
 | |
| 	// Jobs that need to be updated
 | |
| 	queue workqueue.TypedRateLimitingInterface[string]
 | |
| 
 | |
| 	// Orphan deleted pods that still have a Job tracking finalizer to be removed
 | |
| 	orphanQueue workqueue.TypedRateLimitingInterface[string]
 | |
| 
 | |
| 	broadcaster record.EventBroadcaster
 | |
| 	recorder    record.EventRecorder
 | |
| 
 | |
| 	clock clock.WithTicker
 | |
| 
 | |
| 	// Store with information to compute the expotential backoff delay for pod
 | |
| 	// recreation in case of pod failures.
 | |
| 	podBackoffStore *backoffStore
 | |
| }
 | |
| 
 | |
| type syncJobCtx struct {
 | |
| 	job                             *batch.Job
 | |
| 	pods                            []*v1.Pod
 | |
| 	finishedCondition               *batch.JobCondition
 | |
| 	activePods                      []*v1.Pod
 | |
| 	succeeded                       int32
 | |
| 	failed                          int32
 | |
| 	prevSucceededIndexes            orderedIntervals
 | |
| 	succeededIndexes                orderedIntervals
 | |
| 	failedIndexes                   *orderedIntervals
 | |
| 	newBackoffRecord                backoffRecord
 | |
| 	expectedRmFinalizers            sets.Set[string]
 | |
| 	uncounted                       *uncountedTerminatedPods
 | |
| 	podsWithDelayedDeletionPerIndex map[int]*v1.Pod
 | |
| 	terminating                     *int32
 | |
| }
 | |
| 
 | |
| // NewController creates a new Job controller that keeps the relevant pods
 | |
| // in sync with their corresponding Job objects.
 | |
| func NewController(ctx context.Context, podInformer coreinformers.PodInformer, jobInformer batchinformers.JobInformer, kubeClient clientset.Interface) (*Controller, error) {
 | |
| 	return newControllerWithClock(ctx, podInformer, jobInformer, kubeClient, &clock.RealClock{})
 | |
| }
 | |
| 
 | |
| func newControllerWithClock(ctx context.Context, podInformer coreinformers.PodInformer, jobInformer batchinformers.JobInformer, kubeClient clientset.Interface, clock clock.WithTicker) (*Controller, error) {
 | |
| 	eventBroadcaster := record.NewBroadcaster(record.WithContext(ctx))
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 
 | |
| 	jm := &Controller{
 | |
| 		kubeClient: kubeClient,
 | |
| 		podControl: controller.RealPodControl{
 | |
| 			KubeClient: kubeClient,
 | |
| 			Recorder:   eventBroadcaster.NewRecorder(scheme.Scheme, v1.EventSource{Component: "job-controller"}),
 | |
| 		},
 | |
| 		expectations:          controller.NewControllerExpectations(),
 | |
| 		finalizerExpectations: newUIDTrackingExpectations(),
 | |
| 		queue:                 workqueue.NewTypedRateLimitingQueueWithConfig(workqueue.NewTypedItemExponentialFailureRateLimiter[string](DefaultJobApiBackOff, MaxJobApiBackOff), workqueue.TypedRateLimitingQueueConfig[string]{Name: "job", Clock: clock}),
 | |
| 		orphanQueue:           workqueue.NewTypedRateLimitingQueueWithConfig(workqueue.NewTypedItemExponentialFailureRateLimiter[string](DefaultJobApiBackOff, MaxJobApiBackOff), workqueue.TypedRateLimitingQueueConfig[string]{Name: "job_orphan_pod", Clock: clock}),
 | |
| 		broadcaster:           eventBroadcaster,
 | |
| 		recorder:              eventBroadcaster.NewRecorder(scheme.Scheme, v1.EventSource{Component: "job-controller"}),
 | |
| 		clock:                 clock,
 | |
| 		podBackoffStore:       newBackoffStore(),
 | |
| 	}
 | |
| 
 | |
| 	if _, err := jobInformer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
 | |
| 		AddFunc: func(obj interface{}) {
 | |
| 			jm.addJob(logger, obj)
 | |
| 		},
 | |
| 		UpdateFunc: func(oldObj, newObj interface{}) {
 | |
| 			jm.updateJob(logger, oldObj, newObj)
 | |
| 		},
 | |
| 		DeleteFunc: func(obj interface{}) {
 | |
| 			jm.deleteJob(logger, obj)
 | |
| 		},
 | |
| 	}); err != nil {
 | |
| 		return nil, fmt.Errorf("adding Job event handler: %w", err)
 | |
| 	}
 | |
| 	jm.jobLister = jobInformer.Lister()
 | |
| 	jm.jobStoreSynced = jobInformer.Informer().HasSynced
 | |
| 
 | |
| 	if _, err := podInformer.Informer().AddEventHandler(cache.ResourceEventHandlerFuncs{
 | |
| 		AddFunc: func(obj interface{}) {
 | |
| 			jm.addPod(logger, obj)
 | |
| 		},
 | |
| 		UpdateFunc: func(oldObj, newObj interface{}) {
 | |
| 			jm.updatePod(logger, oldObj, newObj)
 | |
| 		},
 | |
| 		DeleteFunc: func(obj interface{}) {
 | |
| 			jm.deletePod(logger, obj, true)
 | |
| 		},
 | |
| 	}); err != nil {
 | |
| 		return nil, fmt.Errorf("adding Pod event handler: %w", err)
 | |
| 	}
 | |
| 	jm.podStore = podInformer.Lister()
 | |
| 	jm.podStoreSynced = podInformer.Informer().HasSynced
 | |
| 
 | |
| 	jm.updateStatusHandler = jm.updateJobStatus
 | |
| 	jm.patchJobHandler = jm.patchJob
 | |
| 	jm.syncHandler = jm.syncJob
 | |
| 
 | |
| 	metrics.Register()
 | |
| 
 | |
| 	return jm, nil
 | |
| }
 | |
| 
 | |
| // Run the main goroutine responsible for watching and syncing jobs.
 | |
| func (jm *Controller) Run(ctx context.Context, workers int) {
 | |
| 	defer utilruntime.HandleCrash()
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 
 | |
| 	// Start events processing pipeline.
 | |
| 	jm.broadcaster.StartStructuredLogging(3)
 | |
| 	jm.broadcaster.StartRecordingToSink(&v1core.EventSinkImpl{Interface: jm.kubeClient.CoreV1().Events("")})
 | |
| 	defer jm.broadcaster.Shutdown()
 | |
| 
 | |
| 	defer jm.queue.ShutDown()
 | |
| 	defer jm.orphanQueue.ShutDown()
 | |
| 
 | |
| 	logger.Info("Starting job controller")
 | |
| 	defer logger.Info("Shutting down job controller")
 | |
| 
 | |
| 	if !cache.WaitForNamedCacheSync("job", ctx.Done(), jm.podStoreSynced, jm.jobStoreSynced) {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	for i := 0; i < workers; i++ {
 | |
| 		go wait.UntilWithContext(ctx, jm.worker, time.Second)
 | |
| 	}
 | |
| 
 | |
| 	go wait.UntilWithContext(ctx, jm.orphanWorker, time.Second)
 | |
| 
 | |
| 	<-ctx.Done()
 | |
| }
 | |
| 
 | |
| // getPodJobs returns a list of Jobs that potentially match a Pod.
 | |
| func (jm *Controller) getPodJobs(pod *v1.Pod) []*batch.Job {
 | |
| 	jobs, err := jm.jobLister.GetPodJobs(pod)
 | |
| 	if err != nil {
 | |
| 		return nil
 | |
| 	}
 | |
| 	if len(jobs) > 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 job is selecting pods with labels: %+v", pod.Labels))
 | |
| 	}
 | |
| 	ret := make([]*batch.Job, 0, len(jobs))
 | |
| 	for i := range jobs {
 | |
| 		ret = append(ret, &jobs[i])
 | |
| 	}
 | |
| 	return ret
 | |
| }
 | |
| 
 | |
| // 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 (jm *Controller) resolveControllerRef(namespace string, controllerRef *metav1.OwnerReference) *batch.Job {
 | |
| 	// 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 != controllerKind.Kind {
 | |
| 		return nil
 | |
| 	}
 | |
| 	job, err := jm.jobLister.Jobs(namespace).Get(controllerRef.Name)
 | |
| 	if err != nil {
 | |
| 		return nil
 | |
| 	}
 | |
| 	if job.UID != controllerRef.UID {
 | |
| 		// The controller we found with this Name is not the same one that the
 | |
| 		// ControllerRef points to.
 | |
| 		return nil
 | |
| 	}
 | |
| 	return job
 | |
| }
 | |
| 
 | |
| // When a pod is created, enqueue the controller that manages it and update its expectations.
 | |
| func (jm *Controller) addPod(logger klog.Logger, obj interface{}) {
 | |
| 	pod := obj.(*v1.Pod)
 | |
| 	recordFinishedPodWithTrackingFinalizer(nil, pod)
 | |
| 	if pod.DeletionTimestamp != nil {
 | |
| 		// on a restart of the controller, it's possible a new pod shows up in a state that
 | |
| 		// is already pending deletion. Prevent the pod from being a creation observation.
 | |
| 		jm.deletePod(logger, pod, false)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	// If it has a ControllerRef, that's all that matters.
 | |
| 	if controllerRef := metav1.GetControllerOf(pod); controllerRef != nil {
 | |
| 		job := jm.resolveControllerRef(pod.Namespace, controllerRef)
 | |
| 		if job == nil {
 | |
| 			return
 | |
| 		}
 | |
| 		jobKey, err := controller.KeyFunc(job)
 | |
| 		if err != nil {
 | |
| 			return
 | |
| 		}
 | |
| 		jm.expectations.CreationObserved(logger, jobKey)
 | |
| 		jm.enqueueSyncJobBatched(logger, job)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	// Otherwise, it's an orphan.
 | |
| 	// Clean the finalizer.
 | |
| 	if hasJobTrackingFinalizer(pod) {
 | |
| 		jm.enqueueOrphanPod(pod)
 | |
| 	}
 | |
| 	// Get a list of all matching controllers and sync
 | |
| 	// them to see if anyone wants to adopt it.
 | |
| 	// DO NOT observe creation because no controller should be waiting for an
 | |
| 	// orphan.
 | |
| 	for _, job := range jm.getPodJobs(pod) {
 | |
| 		jm.enqueueSyncJobBatched(logger, job)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // When a pod is updated, figure out what job/s manage it and wake them up.
 | |
| // If the labels of the pod have changed we need to awaken both the old
 | |
| // and new job. old and cur must be *v1.Pod types.
 | |
| func (jm *Controller) updatePod(logger klog.Logger, old, cur interface{}) {
 | |
| 	curPod := cur.(*v1.Pod)
 | |
| 	oldPod := old.(*v1.Pod)
 | |
| 	recordFinishedPodWithTrackingFinalizer(oldPod, curPod)
 | |
| 	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
 | |
| 	}
 | |
| 	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 job to create more pods asap, not wait
 | |
| 		// until the kubelet actually deletes the pod.
 | |
| 		jm.deletePod(logger, curPod, false)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	// Don't check if oldPod has the finalizer, as during ownership transfer
 | |
| 	// finalizers might be re-added and removed again in behalf of the new owner.
 | |
| 	// If all those Pod updates collapse into a single event, the finalizer
 | |
| 	// might be removed in oldPod and curPod. We want to record the latest
 | |
| 	// state.
 | |
| 	finalizerRemoved := !hasJobTrackingFinalizer(curPod)
 | |
| 	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 job := jm.resolveControllerRef(oldPod.Namespace, oldControllerRef); job != nil {
 | |
| 			if finalizerRemoved {
 | |
| 				key, err := controller.KeyFunc(job)
 | |
| 				if err == nil {
 | |
| 					jm.finalizerExpectations.finalizerRemovalObserved(logger, key, string(curPod.UID))
 | |
| 				}
 | |
| 			}
 | |
| 			jm.enqueueSyncJobBatched(logger, job)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	// If it has a ControllerRef, that's all that matters.
 | |
| 	if curControllerRef != nil {
 | |
| 		job := jm.resolveControllerRef(curPod.Namespace, curControllerRef)
 | |
| 		if job == nil {
 | |
| 			return
 | |
| 		}
 | |
| 		if finalizerRemoved {
 | |
| 			key, err := controller.KeyFunc(job)
 | |
| 			if err == nil {
 | |
| 				jm.finalizerExpectations.finalizerRemovalObserved(logger, key, string(curPod.UID))
 | |
| 			}
 | |
| 		}
 | |
| 		jm.enqueueSyncJobBatched(logger, job)
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	// Otherwise, it's an orphan.
 | |
| 	// Clean the finalizer.
 | |
| 	if hasJobTrackingFinalizer(curPod) {
 | |
| 		jm.enqueueOrphanPod(curPod)
 | |
| 	}
 | |
| 	// If anything changed, sync matching controllers
 | |
| 	// to see if anyone wants to adopt it now.
 | |
| 	labelChanged := !reflect.DeepEqual(curPod.Labels, oldPod.Labels)
 | |
| 	if labelChanged || controllerRefChanged {
 | |
| 		for _, job := range jm.getPodJobs(curPod) {
 | |
| 			jm.enqueueSyncJobBatched(logger, job)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // When a pod is deleted, enqueue the job that manages the pod and update its expectations.
 | |
| // obj could be an *v1.Pod, or a DeleteFinalStateUnknown marker item.
 | |
| func (jm *Controller) deletePod(logger klog.Logger, obj interface{}, final bool) {
 | |
| 	pod, ok := obj.(*v1.Pod)
 | |
| 	if final {
 | |
| 		recordFinishedPodWithTrackingFinalizer(pod, nil)
 | |
| 	}
 | |
| 
 | |
| 	// 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 job 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)
 | |
| 	hasFinalizer := hasJobTrackingFinalizer(pod)
 | |
| 	if controllerRef == nil {
 | |
| 		// No controller should care about orphans being deleted.
 | |
| 		// But this pod might have belonged to a Job and the GC removed the reference.
 | |
| 		if hasFinalizer {
 | |
| 			jm.enqueueOrphanPod(pod)
 | |
| 		}
 | |
| 		return
 | |
| 	}
 | |
| 	job := jm.resolveControllerRef(pod.Namespace, controllerRef)
 | |
| 	if job == nil || util.IsJobFinished(job) {
 | |
| 		// syncJob will not remove this finalizer.
 | |
| 		if hasFinalizer {
 | |
| 			jm.enqueueOrphanPod(pod)
 | |
| 		}
 | |
| 		return
 | |
| 	}
 | |
| 	jobKey, err := controller.KeyFunc(job)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 	jm.expectations.DeletionObserved(logger, jobKey)
 | |
| 
 | |
| 	// Consider the finalizer removed if this is the final delete. Otherwise,
 | |
| 	// it's an update for the deletion timestamp, then check finalizer.
 | |
| 	if final || !hasFinalizer {
 | |
| 		jm.finalizerExpectations.finalizerRemovalObserved(logger, jobKey, string(pod.UID))
 | |
| 	}
 | |
| 
 | |
| 	jm.enqueueSyncJobBatched(logger, job)
 | |
| }
 | |
| 
 | |
| func (jm *Controller) addJob(logger klog.Logger, obj interface{}) {
 | |
| 	jm.enqueueSyncJobImmediately(logger, obj)
 | |
| 	jobObj, ok := obj.(*batch.Job)
 | |
| 	if !ok {
 | |
| 		return
 | |
| 	}
 | |
| 	if controllerName := managedByExternalController(jobObj); controllerName != nil {
 | |
| 		metrics.JobByExternalControllerTotal.WithLabelValues(*controllerName).Inc()
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (jm *Controller) updateJob(logger klog.Logger, old, cur interface{}) {
 | |
| 	oldJob := old.(*batch.Job)
 | |
| 	curJob := cur.(*batch.Job)
 | |
| 
 | |
| 	// never return error
 | |
| 	key, err := controller.KeyFunc(curJob)
 | |
| 	if err != nil {
 | |
| 		return
 | |
| 	}
 | |
| 
 | |
| 	if curJob.Generation == oldJob.Generation {
 | |
| 		// Delay the Job sync when no generation change to batch Job status updates,
 | |
| 		// typically triggered by pod events.
 | |
| 		jm.enqueueSyncJobBatched(logger, curJob)
 | |
| 	} else {
 | |
| 		// Trigger immediate sync when spec is changed.
 | |
| 		jm.enqueueSyncJobImmediately(logger, curJob)
 | |
| 	}
 | |
| 
 | |
| 	// The job shouldn't be marked as finished until all pod finalizers are removed.
 | |
| 	// This is a backup operation in this case.
 | |
| 	if util.IsJobFinished(curJob) {
 | |
| 		jm.cleanupPodFinalizers(curJob)
 | |
| 	}
 | |
| 
 | |
| 	// check if need to add a new rsync for ActiveDeadlineSeconds
 | |
| 	if curJob.Status.StartTime != nil {
 | |
| 		curADS := curJob.Spec.ActiveDeadlineSeconds
 | |
| 		if curADS == nil {
 | |
| 			return
 | |
| 		}
 | |
| 		oldADS := oldJob.Spec.ActiveDeadlineSeconds
 | |
| 		if oldADS == nil || *oldADS != *curADS {
 | |
| 			passed := jm.clock.Since(curJob.Status.StartTime.Time)
 | |
| 			total := time.Duration(*curADS) * time.Second
 | |
| 			// AddAfter will handle total < passed
 | |
| 			jm.queue.AddAfter(key, total-passed)
 | |
| 			logger.V(4).Info("job's ActiveDeadlineSeconds updated, will rsync", "key", key, "interval", total-passed)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // deleteJob enqueues the job and all the pods associated with it that still
 | |
| // have a finalizer.
 | |
| func (jm *Controller) deleteJob(logger klog.Logger, obj interface{}) {
 | |
| 	jm.enqueueSyncJobImmediately(logger, obj)
 | |
| 	jobObj, ok := obj.(*batch.Job)
 | |
| 	if !ok {
 | |
| 		tombstone, ok := obj.(cache.DeletedFinalStateUnknown)
 | |
| 		if !ok {
 | |
| 			utilruntime.HandleError(fmt.Errorf("couldn't get object from tombstone %+v", obj))
 | |
| 			return
 | |
| 		}
 | |
| 		jobObj, ok = tombstone.Obj.(*batch.Job)
 | |
| 		if !ok {
 | |
| 			utilruntime.HandleError(fmt.Errorf("tombstone contained object that is not a job %+v", obj))
 | |
| 			return
 | |
| 		}
 | |
| 	}
 | |
| 	jm.cleanupPodFinalizers(jobObj)
 | |
| }
 | |
| 
 | |
| // enqueueSyncJobImmediately tells the Job controller to invoke syncJob
 | |
| // immediately.
 | |
| // It is only used for Job events (creation, deletion, spec update).
 | |
| // obj could be an *batch.Job, or a DeletionFinalStateUnknown marker item.
 | |
| func (jm *Controller) enqueueSyncJobImmediately(logger klog.Logger, obj interface{}) {
 | |
| 	jm.enqueueSyncJobInternal(logger, obj, 0)
 | |
| }
 | |
| 
 | |
| // enqueueSyncJobBatched tells the controller to invoke syncJob with a
 | |
| // constant batching delay.
 | |
| // It is used for:
 | |
| // - Pod events (creation, deletion, update)
 | |
| // - Job status update
 | |
| // obj could be an *batch.Job, or a DeletionFinalStateUnknown marker item.
 | |
| func (jm *Controller) enqueueSyncJobBatched(logger klog.Logger, obj interface{}) {
 | |
| 	jm.enqueueSyncJobInternal(logger, obj, syncJobBatchPeriod)
 | |
| }
 | |
| 
 | |
| // enqueueSyncJobWithDelay tells the controller to invoke syncJob with a
 | |
| // custom delay, but not smaller than the batching delay.
 | |
| // It is used when pod recreations are delayed due to pod failures.
 | |
| // obj could be an *batch.Job, or a DeletionFinalStateUnknown marker item.
 | |
| func (jm *Controller) enqueueSyncJobWithDelay(logger klog.Logger, obj interface{}, delay time.Duration) {
 | |
| 	if delay < syncJobBatchPeriod {
 | |
| 		delay = syncJobBatchPeriod
 | |
| 	}
 | |
| 	jm.enqueueSyncJobInternal(logger, obj, delay)
 | |
| }
 | |
| 
 | |
| func (jm *Controller) enqueueSyncJobInternal(logger klog.Logger, obj interface{}, delay time.Duration) {
 | |
| 	key, err := controller.KeyFunc(obj)
 | |
| 	if err != nil {
 | |
| 		utilruntime.HandleError(fmt.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.
 | |
| 	logger.Info("enqueueing job", "key", key, "delay", delay)
 | |
| 	jm.queue.AddAfter(key, delay)
 | |
| }
 | |
| 
 | |
| func (jm *Controller) enqueueOrphanPod(obj *v1.Pod) {
 | |
| 	key, err := controller.KeyFunc(obj)
 | |
| 	if err != nil {
 | |
| 		utilruntime.HandleError(fmt.Errorf("couldn't get key for object %+v: %v", obj, err))
 | |
| 		return
 | |
| 	}
 | |
| 	jm.orphanQueue.Add(key)
 | |
| }
 | |
| 
 | |
| // 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 (jm *Controller) worker(ctx context.Context) {
 | |
| 	for jm.processNextWorkItem(ctx) {
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (jm *Controller) processNextWorkItem(ctx context.Context) bool {
 | |
| 	key, quit := jm.queue.Get()
 | |
| 	if quit {
 | |
| 		return false
 | |
| 	}
 | |
| 	defer jm.queue.Done(key)
 | |
| 
 | |
| 	err := jm.syncHandler(ctx, key)
 | |
| 	if err == nil {
 | |
| 		jm.queue.Forget(key)
 | |
| 		return true
 | |
| 	}
 | |
| 
 | |
| 	utilruntime.HandleError(fmt.Errorf("syncing job: %w", err))
 | |
| 	jm.queue.AddRateLimited(key)
 | |
| 
 | |
| 	return true
 | |
| }
 | |
| 
 | |
| func (jm *Controller) orphanWorker(ctx context.Context) {
 | |
| 	for jm.processNextOrphanPod(ctx) {
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (jm *Controller) processNextOrphanPod(ctx context.Context) bool {
 | |
| 	key, quit := jm.orphanQueue.Get()
 | |
| 	if quit {
 | |
| 		return false
 | |
| 	}
 | |
| 	defer jm.orphanQueue.Done(key)
 | |
| 	err := jm.syncOrphanPod(ctx, key)
 | |
| 	if err != nil {
 | |
| 		utilruntime.HandleError(fmt.Errorf("Error syncing orphan pod: %v", err))
 | |
| 		jm.orphanQueue.AddRateLimited(key)
 | |
| 	} else {
 | |
| 		jm.orphanQueue.Forget(key)
 | |
| 	}
 | |
| 
 | |
| 	return true
 | |
| }
 | |
| 
 | |
| // syncOrphanPod removes the tracking finalizer from an orphan pod if found.
 | |
| func (jm *Controller) syncOrphanPod(ctx context.Context, key string) error {
 | |
| 	startTime := jm.clock.Now()
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 	defer func() {
 | |
| 		logger.V(4).Info("Finished syncing orphan pod", "pod", key, "elapsed", jm.clock.Since(startTime))
 | |
| 	}()
 | |
| 
 | |
| 	ns, name, err := cache.SplitMetaNamespaceKey(key)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	sharedPod, err := jm.podStore.Pods(ns).Get(name)
 | |
| 	if err != nil {
 | |
| 		if apierrors.IsNotFound(err) {
 | |
| 			logger.V(4).Info("Orphan pod has been deleted", "pod", key)
 | |
| 			return nil
 | |
| 		}
 | |
| 		return err
 | |
| 	}
 | |
| 	// Make sure the pod is still orphaned.
 | |
| 	if controllerRef := metav1.GetControllerOf(sharedPod); controllerRef != nil {
 | |
| 		if controllerRef.Kind != controllerKind.Kind || controllerRef.APIVersion != batch.SchemeGroupVersion.String() {
 | |
| 			// The pod is controlled by an owner that is not a batch/v1 Job. Do not remove finalizer.
 | |
| 			return nil
 | |
| 		}
 | |
| 		job := jm.resolveControllerRef(sharedPod.Namespace, controllerRef)
 | |
| 		if job != nil {
 | |
| 			// Skip cleanup of finalizers for pods owned by a job managed by an external controller
 | |
| 			if controllerName := managedByExternalController(job); controllerName != nil {
 | |
| 				logger.V(2).Info("Skip cleanup of the job finalizer for a pod owned by a job that is managed by an external controller", "key", key, "podUID", sharedPod.UID, "jobUID", job.UID, "controllerName", controllerName)
 | |
| 				return nil
 | |
| 			}
 | |
| 		}
 | |
| 		if job != nil && !util.IsJobFinished(job) {
 | |
| 			// The pod was adopted. Do not remove finalizer.
 | |
| 			return nil
 | |
| 		}
 | |
| 	}
 | |
| 	if patch := removeTrackingFinalizerPatch(sharedPod); patch != nil {
 | |
| 		if err := jm.podControl.PatchPod(ctx, ns, name, patch); err != nil && !apierrors.IsNotFound(err) {
 | |
| 			return err
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // getPodsForJob returns the set of pods that this Job should manage.
 | |
| // It also reconciles ControllerRef by adopting/orphaning, adding tracking
 | |
| // finalizers.
 | |
| // Note that the returned Pods are pointers into the cache.
 | |
| func (jm *Controller) getPodsForJob(ctx context.Context, j *batch.Job) ([]*v1.Pod, error) {
 | |
| 	selector, err := metav1.LabelSelectorAsSelector(j.Spec.Selector)
 | |
| 	if err != nil {
 | |
| 		return nil, fmt.Errorf("couldn't convert Job selector: %v", err)
 | |
| 	}
 | |
| 	// List all pods to include those that don't match the selector anymore
 | |
| 	// but have a ControllerRef pointing to this controller.
 | |
| 	pods, err := jm.podStore.Pods(j.Namespace).List(labels.Everything())
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	// 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(ctx context.Context) (metav1.Object, error) {
 | |
| 		fresh, err := jm.kubeClient.BatchV1().Jobs(j.Namespace).Get(ctx, j.Name, metav1.GetOptions{})
 | |
| 		if err != nil {
 | |
| 			return nil, err
 | |
| 		}
 | |
| 		if fresh.UID != j.UID {
 | |
| 			return nil, fmt.Errorf("original Job %v/%v is gone: got uid %v, wanted %v", j.Namespace, j.Name, fresh.UID, j.UID)
 | |
| 		}
 | |
| 		return fresh, nil
 | |
| 	})
 | |
| 	cm := controller.NewPodControllerRefManager(jm.podControl, j, selector, controllerKind, canAdoptFunc, batch.JobTrackingFinalizer)
 | |
| 	// When adopting Pods, this operation adds an ownerRef and finalizers.
 | |
| 	pods, err = cm.ClaimPods(ctx, pods)
 | |
| 	if err != nil {
 | |
| 		return pods, err
 | |
| 	}
 | |
| 	// Set finalizer on adopted pods for the remaining calculations.
 | |
| 	for i, p := range pods {
 | |
| 		adopted := true
 | |
| 		for _, r := range p.OwnerReferences {
 | |
| 			if r.UID == j.UID {
 | |
| 				adopted = false
 | |
| 				break
 | |
| 			}
 | |
| 		}
 | |
| 		if adopted && !hasJobTrackingFinalizer(p) {
 | |
| 			pods[i] = p.DeepCopy()
 | |
| 			pods[i].Finalizers = append(p.Finalizers, batch.JobTrackingFinalizer)
 | |
| 		}
 | |
| 	}
 | |
| 	return pods, err
 | |
| }
 | |
| 
 | |
| // syncJob will sync the job 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 (jm *Controller) syncJob(ctx context.Context, key string) (rErr error) {
 | |
| 	startTime := jm.clock.Now()
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 	defer func() {
 | |
| 		logger.V(4).Info("Finished syncing job", "key", key, "elapsed", jm.clock.Since(startTime))
 | |
| 	}()
 | |
| 
 | |
| 	ns, name, err := cache.SplitMetaNamespaceKey(key)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	if len(ns) == 0 || len(name) == 0 {
 | |
| 		return fmt.Errorf("invalid job key %q: either namespace or name is missing", key)
 | |
| 	}
 | |
| 	sharedJob, err := jm.jobLister.Jobs(ns).Get(name)
 | |
| 	if err != nil {
 | |
| 		if apierrors.IsNotFound(err) {
 | |
| 			logger.V(4).Info("Job has been deleted", "key", key)
 | |
| 			jm.expectations.DeleteExpectations(logger, key)
 | |
| 			jm.finalizerExpectations.deleteExpectations(logger, key)
 | |
| 
 | |
| 			err := jm.podBackoffStore.removeBackoffRecord(key)
 | |
| 			if err != nil {
 | |
| 				// re-syncing here as the record has to be removed for finished/deleted jobs
 | |
| 				return fmt.Errorf("error removing backoff record %w", err)
 | |
| 			}
 | |
| 			return nil
 | |
| 		}
 | |
| 		return err
 | |
| 	}
 | |
| 
 | |
| 	// Skip syncing of the job it is managed by another controller.
 | |
| 	// We cannot rely solely on skipping of queueing such jobs for synchronization,
 | |
| 	// because it is possible a synchronization task is queued for a job, without
 | |
| 	// the managedBy field, but the job is quickly replaced by another job with
 | |
| 	// the field. Then, the syncJob might be invoked for a job with the field.
 | |
| 	if controllerName := managedByExternalController(sharedJob); controllerName != nil {
 | |
| 		logger.V(2).Info("Skip syncing the job as it is managed by an external controller", "key", key, "uid", sharedJob.UID, "controllerName", controllerName)
 | |
| 		return nil
 | |
| 	}
 | |
| 
 | |
| 	// make a copy so we don't mutate the shared cache
 | |
| 	job := *sharedJob.DeepCopy()
 | |
| 
 | |
| 	// if job was finished previously, we don't want to redo the termination
 | |
| 	if util.IsJobFinished(&job) {
 | |
| 		err := jm.podBackoffStore.removeBackoffRecord(key)
 | |
| 		if err != nil {
 | |
| 			// re-syncing here as the record has to be removed for finished/deleted jobs
 | |
| 			return fmt.Errorf("error removing backoff record %w", err)
 | |
| 		}
 | |
| 		return nil
 | |
| 	}
 | |
| 
 | |
| 	if job.Spec.CompletionMode != nil && *job.Spec.CompletionMode != batch.NonIndexedCompletion && *job.Spec.CompletionMode != batch.IndexedCompletion {
 | |
| 		jm.recorder.Event(&job, v1.EventTypeWarning, "UnknownCompletionMode", "Skipped Job sync because completion mode is unknown")
 | |
| 		return nil
 | |
| 	}
 | |
| 
 | |
| 	completionMode := getCompletionMode(&job)
 | |
| 	action := metrics.JobSyncActionReconciling
 | |
| 
 | |
| 	defer func() {
 | |
| 		result := "success"
 | |
| 		if rErr != nil {
 | |
| 			result = "error"
 | |
| 		}
 | |
| 
 | |
| 		metrics.JobSyncDurationSeconds.WithLabelValues(completionMode, result, action).Observe(jm.clock.Since(startTime).Seconds())
 | |
| 		metrics.JobSyncNum.WithLabelValues(completionMode, result, action).Inc()
 | |
| 	}()
 | |
| 
 | |
| 	if job.Status.UncountedTerminatedPods == nil {
 | |
| 		job.Status.UncountedTerminatedPods = &batch.UncountedTerminatedPods{}
 | |
| 	}
 | |
| 
 | |
| 	// Check the expectations of the job before counting active pods, otherwise a new pod can sneak in
 | |
| 	// and update the expectations after we've retrieved active pods from the store. If a new pod enters
 | |
| 	// the store after we've checked the expectation, the job sync is just deferred till the next relist.
 | |
| 	satisfiedExpectations := jm.expectations.SatisfiedExpectations(logger, key)
 | |
| 
 | |
| 	pods, err := jm.getPodsForJob(ctx, &job)
 | |
| 	if err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	var terminating *int32
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.JobPodReplacementPolicy) {
 | |
| 		terminating = ptr.To(controller.CountTerminatingPods(pods))
 | |
| 	}
 | |
| 	jobCtx := &syncJobCtx{
 | |
| 		job:                  &job,
 | |
| 		pods:                 pods,
 | |
| 		activePods:           controller.FilterActivePods(logger, pods),
 | |
| 		terminating:          terminating,
 | |
| 		uncounted:            newUncountedTerminatedPods(*job.Status.UncountedTerminatedPods),
 | |
| 		expectedRmFinalizers: jm.finalizerExpectations.getExpectedUIDs(key),
 | |
| 	}
 | |
| 	active := int32(len(jobCtx.activePods))
 | |
| 	newSucceededPods, newFailedPods := getNewFinishedPods(jobCtx)
 | |
| 	jobCtx.succeeded = job.Status.Succeeded + int32(len(newSucceededPods)) + int32(len(jobCtx.uncounted.succeeded))
 | |
| 	jobCtx.failed = job.Status.Failed + int32(nonIgnoredFailedPodsCount(jobCtx, newFailedPods)) + int32(len(jobCtx.uncounted.failed))
 | |
| 	var ready *int32
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.JobReadyPods) {
 | |
| 		ready = ptr.To(countReadyPods(jobCtx.activePods))
 | |
| 	}
 | |
| 
 | |
| 	// Job first start. Set StartTime only if the job is not in the suspended state.
 | |
| 	if job.Status.StartTime == nil && !jobSuspended(&job) {
 | |
| 		now := metav1.NewTime(jm.clock.Now())
 | |
| 		job.Status.StartTime = &now
 | |
| 	}
 | |
| 
 | |
| 	jobCtx.newBackoffRecord = jm.podBackoffStore.newBackoffRecord(key, newSucceededPods, newFailedPods)
 | |
| 
 | |
| 	var manageJobErr error
 | |
| 
 | |
| 	exceedsBackoffLimit := jobCtx.failed > *job.Spec.BackoffLimit
 | |
| 	jobCtx.finishedCondition = hasSuccessCriteriaMetCondition(&job)
 | |
| 
 | |
| 	// Given that the Job already has the SuccessCriteriaMet condition, the termination condition already had confirmed in another cycle.
 | |
| 	// So, the job-controller evaluates the podFailurePolicy only when the Job doesn't have the SuccessCriteriaMet condition.
 | |
| 	if jobCtx.finishedCondition == nil && feature.DefaultFeatureGate.Enabled(features.JobPodFailurePolicy) {
 | |
| 		if failureTargetCondition := findConditionByType(job.Status.Conditions, batch.JobFailureTarget); failureTargetCondition != nil {
 | |
| 			jobCtx.finishedCondition = newFailedConditionForFailureTarget(failureTargetCondition, jm.clock.Now())
 | |
| 		} else if failJobMessage := getFailJobMessage(&job, pods); failJobMessage != nil {
 | |
| 			// Prepare the interim FailureTarget condition to record the failure message before the finalizers (allowing removal of the pods) are removed.
 | |
| 			jobCtx.finishedCondition = newCondition(batch.JobFailureTarget, v1.ConditionTrue, batch.JobReasonPodFailurePolicy, *failJobMessage, jm.clock.Now())
 | |
| 		}
 | |
| 	}
 | |
| 	if jobCtx.finishedCondition == nil {
 | |
| 		if exceedsBackoffLimit || pastBackoffLimitOnFailure(&job, pods) {
 | |
| 			// check if the number of pod restart exceeds backoff (for restart OnFailure only)
 | |
| 			// OR if the number of failed jobs increased since the last syncJob
 | |
| 			jobCtx.finishedCondition = newCondition(batch.JobFailed, v1.ConditionTrue, batch.JobReasonBackoffLimitExceeded, "Job has reached the specified backoff limit", jm.clock.Now())
 | |
| 		} else if jm.pastActiveDeadline(&job) {
 | |
| 			jobCtx.finishedCondition = newCondition(batch.JobFailed, v1.ConditionTrue, batch.JobReasonDeadlineExceeded, "Job was active longer than specified deadline", jm.clock.Now())
 | |
| 		} else if job.Spec.ActiveDeadlineSeconds != nil && !jobSuspended(&job) {
 | |
| 			syncDuration := time.Duration(*job.Spec.ActiveDeadlineSeconds)*time.Second - jm.clock.Since(job.Status.StartTime.Time)
 | |
| 			logger.V(2).Info("Job has activeDeadlineSeconds configuration. Will sync this job again", "key", key, "nextSyncIn", syncDuration)
 | |
| 			jm.queue.AddAfter(key, syncDuration)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	if isIndexedJob(&job) {
 | |
| 		jobCtx.prevSucceededIndexes, jobCtx.succeededIndexes = calculateSucceededIndexes(logger, &job, pods)
 | |
| 		jobCtx.succeeded = int32(jobCtx.succeededIndexes.total())
 | |
| 		if hasBackoffLimitPerIndex(&job) {
 | |
| 			jobCtx.failedIndexes = calculateFailedIndexes(logger, &job, pods)
 | |
| 			if jobCtx.finishedCondition == nil {
 | |
| 				if job.Spec.MaxFailedIndexes != nil && jobCtx.failedIndexes.total() > int(*job.Spec.MaxFailedIndexes) {
 | |
| 					jobCtx.finishedCondition = newCondition(batch.JobFailed, v1.ConditionTrue, batch.JobReasonMaxFailedIndexesExceeded, "Job has exceeded the specified maximal number of failed indexes", jm.clock.Now())
 | |
| 				} else if jobCtx.failedIndexes.total() > 0 && jobCtx.failedIndexes.total()+jobCtx.succeededIndexes.total() >= int(*job.Spec.Completions) {
 | |
| 					jobCtx.finishedCondition = newCondition(batch.JobFailed, v1.ConditionTrue, batch.JobReasonFailedIndexes, "Job has failed indexes", jm.clock.Now())
 | |
| 				}
 | |
| 			}
 | |
| 			jobCtx.podsWithDelayedDeletionPerIndex = getPodsWithDelayedDeletionPerIndex(logger, jobCtx)
 | |
| 		}
 | |
| 		if jobCtx.finishedCondition == nil && hasSuccessCriteriaMetCondition(jobCtx.job) == nil {
 | |
| 			if msg, met := matchSuccessPolicy(logger, job.Spec.SuccessPolicy, *job.Spec.Completions, jobCtx.succeededIndexes); met {
 | |
| 				jobCtx.finishedCondition = newCondition(batch.JobSuccessCriteriaMet, v1.ConditionTrue, batch.JobReasonSuccessPolicy, msg, jm.clock.Now())
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	suspendCondChanged := false
 | |
| 	// Remove active pods if Job failed.
 | |
| 	if jobCtx.finishedCondition != nil {
 | |
| 		deleted, err := jm.deleteActivePods(ctx, &job, jobCtx.activePods)
 | |
| 		if deleted != active || !satisfiedExpectations {
 | |
| 			// Can't declare the Job as finished yet, as there might be remaining
 | |
| 			// pod finalizers or pods that are not in the informer's cache yet.
 | |
| 			jobCtx.finishedCondition = nil
 | |
| 		}
 | |
| 		active -= deleted
 | |
| 		manageJobErr = err
 | |
| 	} else {
 | |
| 		manageJobCalled := false
 | |
| 		if satisfiedExpectations && job.DeletionTimestamp == nil {
 | |
| 			active, action, manageJobErr = jm.manageJob(ctx, &job, jobCtx)
 | |
| 			manageJobCalled = true
 | |
| 		}
 | |
| 		complete := false
 | |
| 		if job.Spec.Completions == nil {
 | |
| 			// This type of job is complete when any pod exits with success.
 | |
| 			// Each pod is capable of
 | |
| 			// determining whether or not the entire Job is done.  Subsequent pods are
 | |
| 			// not expected to fail, but if they do, the failure is ignored.  Once any
 | |
| 			// pod succeeds, the controller waits for remaining pods to finish, and
 | |
| 			// then the job is complete.
 | |
| 			complete = jobCtx.succeeded > 0 && active == 0
 | |
| 		} else {
 | |
| 			// Job specifies a number of completions.  This type of job signals
 | |
| 			// success by having that number of successes.  Since we do not
 | |
| 			// start more pods than there are remaining completions, there should
 | |
| 			// not be any remaining active pods once this count is reached.
 | |
| 			complete = jobCtx.succeeded >= *job.Spec.Completions && active == 0
 | |
| 		}
 | |
| 		if complete {
 | |
| 			jobCtx.finishedCondition = newCondition(batch.JobComplete, v1.ConditionTrue, "", "", jm.clock.Now())
 | |
| 		} else if manageJobCalled {
 | |
| 			// Update the conditions / emit events only if manageJob was called in
 | |
| 			// this syncJob. Otherwise wait for the right syncJob call to make
 | |
| 			// updates.
 | |
| 			if job.Spec.Suspend != nil && *job.Spec.Suspend {
 | |
| 				// Job can be in the suspended state only if it is NOT completed.
 | |
| 				var isUpdated bool
 | |
| 				job.Status.Conditions, isUpdated = ensureJobConditionStatus(job.Status.Conditions, batch.JobSuspended, v1.ConditionTrue, "JobSuspended", "Job suspended", jm.clock.Now())
 | |
| 				if isUpdated {
 | |
| 					suspendCondChanged = true
 | |
| 					jm.recorder.Event(&job, v1.EventTypeNormal, "Suspended", "Job suspended")
 | |
| 				}
 | |
| 			} else {
 | |
| 				// Job not suspended.
 | |
| 				var isUpdated bool
 | |
| 				job.Status.Conditions, isUpdated = ensureJobConditionStatus(job.Status.Conditions, batch.JobSuspended, v1.ConditionFalse, "JobResumed", "Job resumed", jm.clock.Now())
 | |
| 				if isUpdated {
 | |
| 					suspendCondChanged = true
 | |
| 					jm.recorder.Event(&job, v1.EventTypeNormal, "Resumed", "Job resumed")
 | |
| 					// Resumed jobs will always reset StartTime to current time. This is
 | |
| 					// done because the ActiveDeadlineSeconds timer shouldn't go off
 | |
| 					// whilst the Job is still suspended and resetting StartTime is
 | |
| 					// consistent with resuming a Job created in the suspended state.
 | |
| 					// (ActiveDeadlineSeconds is interpreted as the number of seconds a
 | |
| 					// Job is continuously active.)
 | |
| 					now := metav1.NewTime(jm.clock.Now())
 | |
| 					job.Status.StartTime = &now
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	needsStatusUpdate := suspendCondChanged || active != job.Status.Active || !ptr.Equal(ready, job.Status.Ready)
 | |
| 	needsStatusUpdate = needsStatusUpdate || !ptr.Equal(job.Status.Terminating, jobCtx.terminating)
 | |
| 	job.Status.Active = active
 | |
| 	job.Status.Ready = ready
 | |
| 	job.Status.Terminating = jobCtx.terminating
 | |
| 	err = jm.trackJobStatusAndRemoveFinalizers(ctx, jobCtx, needsStatusUpdate)
 | |
| 	if err != nil {
 | |
| 		return fmt.Errorf("tracking status: %w", err)
 | |
| 	}
 | |
| 
 | |
| 	return manageJobErr
 | |
| }
 | |
| 
 | |
| // deleteActivePods issues deletion for active Pods, preserving finalizers.
 | |
| // This is done through DELETE calls that set deletion timestamps.
 | |
| // The method trackJobStatusAndRemoveFinalizers removes the finalizers, after
 | |
| // which the objects can actually be deleted.
 | |
| // Returns number of successfully deletions issued.
 | |
| func (jm *Controller) deleteActivePods(ctx context.Context, job *batch.Job, pods []*v1.Pod) (int32, error) {
 | |
| 	errCh := make(chan error, len(pods))
 | |
| 	successfulDeletes := int32(len(pods))
 | |
| 	wg := sync.WaitGroup{}
 | |
| 	wg.Add(len(pods))
 | |
| 	for i := range pods {
 | |
| 		go func(pod *v1.Pod) {
 | |
| 			defer wg.Done()
 | |
| 			if err := jm.podControl.DeletePod(ctx, job.Namespace, pod.Name, job); err != nil && !apierrors.IsNotFound(err) {
 | |
| 				atomic.AddInt32(&successfulDeletes, -1)
 | |
| 				errCh <- err
 | |
| 				utilruntime.HandleError(err)
 | |
| 			}
 | |
| 		}(pods[i])
 | |
| 	}
 | |
| 	wg.Wait()
 | |
| 	return successfulDeletes, errorFromChannel(errCh)
 | |
| }
 | |
| 
 | |
| func nonIgnoredFailedPodsCount(jobCtx *syncJobCtx, failedPods []*v1.Pod) int {
 | |
| 	result := len(failedPods)
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.JobPodFailurePolicy) && jobCtx.job.Spec.PodFailurePolicy != nil {
 | |
| 		for _, p := range failedPods {
 | |
| 			_, countFailed, _ := matchPodFailurePolicy(jobCtx.job.Spec.PodFailurePolicy, p)
 | |
| 			if !countFailed {
 | |
| 				result--
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	return result
 | |
| }
 | |
| 
 | |
| // deleteJobPods deletes the pods, returns the number of successful removals
 | |
| // and any error.
 | |
| func (jm *Controller) deleteJobPods(ctx context.Context, job *batch.Job, jobKey string, pods []*v1.Pod) (int32, error) {
 | |
| 	errCh := make(chan error, len(pods))
 | |
| 	successfulDeletes := int32(len(pods))
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 
 | |
| 	failDelete := func(pod *v1.Pod, err error) {
 | |
| 		// Decrement the expected number of deletes because the informer won't observe this deletion
 | |
| 		jm.expectations.DeletionObserved(logger, jobKey)
 | |
| 		if !apierrors.IsNotFound(err) {
 | |
| 			logger.V(2).Info("Failed to delete Pod", "job", klog.KObj(job), "pod", klog.KObj(pod), "err", err)
 | |
| 			atomic.AddInt32(&successfulDeletes, -1)
 | |
| 			errCh <- err
 | |
| 			utilruntime.HandleError(err)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	wg := sync.WaitGroup{}
 | |
| 	wg.Add(len(pods))
 | |
| 	for i := range pods {
 | |
| 		go func(pod *v1.Pod) {
 | |
| 			defer wg.Done()
 | |
| 			if patch := removeTrackingFinalizerPatch(pod); patch != nil {
 | |
| 				if err := jm.podControl.PatchPod(ctx, pod.Namespace, pod.Name, patch); err != nil {
 | |
| 					failDelete(pod, fmt.Errorf("removing completion finalizer: %w", err))
 | |
| 					return
 | |
| 				}
 | |
| 			}
 | |
| 			if err := jm.podControl.DeletePod(ctx, job.Namespace, pod.Name, job); err != nil {
 | |
| 				failDelete(pod, err)
 | |
| 			}
 | |
| 		}(pods[i])
 | |
| 	}
 | |
| 	wg.Wait()
 | |
| 	return successfulDeletes, errorFromChannel(errCh)
 | |
| }
 | |
| 
 | |
| // trackJobStatusAndRemoveFinalizers does:
 | |
| //  1. Add finished Pods to .status.uncountedTerminatedPods
 | |
| //  2. Remove the finalizers from the Pods if they completed or were removed
 | |
| //     or the job was removed.
 | |
| //  3. Increment job counters for pods that no longer have a finalizer.
 | |
| //  4. Add Complete condition if satisfied with current counters.
 | |
| //
 | |
| // It does this up to a limited number of Pods so that the size of .status
 | |
| // doesn't grow too much and this sync doesn't starve other Jobs.
 | |
| func (jm *Controller) trackJobStatusAndRemoveFinalizers(ctx context.Context, jobCtx *syncJobCtx, needsFlush bool) error {
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 
 | |
| 	isIndexed := isIndexedJob(jobCtx.job)
 | |
| 	var podsToRemoveFinalizer []*v1.Pod
 | |
| 	uncountedStatus := jobCtx.job.Status.UncountedTerminatedPods
 | |
| 	var newSucceededIndexes []int
 | |
| 	if isIndexed {
 | |
| 		// Sort to introduce completed Indexes in order.
 | |
| 		sort.Sort(byCompletionIndex(jobCtx.pods))
 | |
| 	}
 | |
| 	uidsWithFinalizer := make(sets.Set[string], len(jobCtx.pods))
 | |
| 	for _, p := range jobCtx.pods {
 | |
| 		uid := string(p.UID)
 | |
| 		if hasJobTrackingFinalizer(p) && !jobCtx.expectedRmFinalizers.Has(uid) {
 | |
| 			uidsWithFinalizer.Insert(uid)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	// Shallow copy, as it will only be used to detect changes in the counters.
 | |
| 	oldCounters := jobCtx.job.Status
 | |
| 	if cleanUncountedPodsWithoutFinalizers(&jobCtx.job.Status, uidsWithFinalizer) {
 | |
| 		needsFlush = true
 | |
| 	}
 | |
| 	podFailureCountByPolicyAction := map[string]int{}
 | |
| 	reachedMaxUncountedPods := false
 | |
| 	for _, pod := range jobCtx.pods {
 | |
| 		if !hasJobTrackingFinalizer(pod) || jobCtx.expectedRmFinalizers.Has(string(pod.UID)) {
 | |
| 			// This pod was processed in a previous sync.
 | |
| 			continue
 | |
| 		}
 | |
| 		considerPodFailed := isPodFailed(pod, jobCtx.job)
 | |
| 		if !canRemoveFinalizer(logger, jobCtx, pod, considerPodFailed) {
 | |
| 			continue
 | |
| 		}
 | |
| 		podsToRemoveFinalizer = append(podsToRemoveFinalizer, pod)
 | |
| 		if pod.Status.Phase == v1.PodSucceeded && !jobCtx.uncounted.failed.Has(string(pod.UID)) {
 | |
| 			if isIndexed {
 | |
| 				// The completion index is enough to avoid recounting succeeded pods.
 | |
| 				// No need to track UIDs.
 | |
| 				ix := getCompletionIndex(pod.Annotations)
 | |
| 				if ix != unknownCompletionIndex && ix < int(*jobCtx.job.Spec.Completions) && !jobCtx.prevSucceededIndexes.has(ix) {
 | |
| 					newSucceededIndexes = append(newSucceededIndexes, ix)
 | |
| 					needsFlush = true
 | |
| 				}
 | |
| 			} else if !jobCtx.uncounted.succeeded.Has(string(pod.UID)) {
 | |
| 				needsFlush = true
 | |
| 				uncountedStatus.Succeeded = append(uncountedStatus.Succeeded, pod.UID)
 | |
| 			}
 | |
| 		} else if considerPodFailed || (jobCtx.finishedCondition != nil && !isSuccessCriteriaMetCondition(jobCtx.finishedCondition)) {
 | |
| 			// When the job is considered finished, every non-terminated pod is considered failed.
 | |
| 			ix := getCompletionIndex(pod.Annotations)
 | |
| 			if !jobCtx.uncounted.failed.Has(string(pod.UID)) && (!isIndexed || (ix != unknownCompletionIndex && ix < int(*jobCtx.job.Spec.Completions))) {
 | |
| 				if feature.DefaultFeatureGate.Enabled(features.JobPodFailurePolicy) && jobCtx.job.Spec.PodFailurePolicy != nil {
 | |
| 					_, countFailed, action := matchPodFailurePolicy(jobCtx.job.Spec.PodFailurePolicy, pod)
 | |
| 					if action != nil {
 | |
| 						podFailureCountByPolicyAction[string(*action)] += 1
 | |
| 					}
 | |
| 					if countFailed {
 | |
| 						needsFlush = true
 | |
| 						uncountedStatus.Failed = append(uncountedStatus.Failed, pod.UID)
 | |
| 					}
 | |
| 				} else {
 | |
| 					needsFlush = true
 | |
| 					uncountedStatus.Failed = append(uncountedStatus.Failed, pod.UID)
 | |
| 				}
 | |
| 			}
 | |
| 		}
 | |
| 		if len(newSucceededIndexes)+len(uncountedStatus.Succeeded)+len(uncountedStatus.Failed) >= MaxUncountedPods {
 | |
| 			// The controller added enough Pods already to .status.uncountedTerminatedPods
 | |
| 			// We stop counting pods and removing finalizers here to:
 | |
| 			// 1. Ensure that the UIDs representation are under 20 KB.
 | |
| 			// 2. Cap the number of finalizer removals so that syncing of big Jobs
 | |
| 			//    doesn't starve smaller ones.
 | |
| 			//
 | |
| 			// The job will be synced again because the Job status and Pod updates
 | |
| 			// will put the Job back to the work queue.
 | |
| 			reachedMaxUncountedPods = true
 | |
| 			break
 | |
| 		}
 | |
| 	}
 | |
| 	if isIndexed {
 | |
| 		jobCtx.succeededIndexes = jobCtx.succeededIndexes.withOrderedIndexes(newSucceededIndexes)
 | |
| 		succeededIndexesStr := jobCtx.succeededIndexes.String()
 | |
| 		if succeededIndexesStr != jobCtx.job.Status.CompletedIndexes {
 | |
| 			needsFlush = true
 | |
| 		}
 | |
| 		jobCtx.job.Status.Succeeded = int32(jobCtx.succeededIndexes.total())
 | |
| 		jobCtx.job.Status.CompletedIndexes = succeededIndexesStr
 | |
| 		var failedIndexesStr *string
 | |
| 		if jobCtx.failedIndexes != nil {
 | |
| 			failedIndexesStr = ptr.To(jobCtx.failedIndexes.String())
 | |
| 		}
 | |
| 		if !ptr.Equal(jobCtx.job.Status.FailedIndexes, failedIndexesStr) {
 | |
| 			jobCtx.job.Status.FailedIndexes = failedIndexesStr
 | |
| 			needsFlush = true
 | |
| 		}
 | |
| 	}
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.JobPodFailurePolicy) {
 | |
| 		if jobCtx.finishedCondition != nil && jobCtx.finishedCondition.Type == batch.JobFailureTarget {
 | |
| 
 | |
| 			// Append the interim FailureTarget condition to update the job status with before finalizers are removed.
 | |
| 			jobCtx.job.Status.Conditions = append(jobCtx.job.Status.Conditions, *jobCtx.finishedCondition)
 | |
| 			needsFlush = true
 | |
| 
 | |
| 			// Prepare the final Failed condition to update the job status with after the finalizers are removed.
 | |
| 			// It is also used in the enactJobFinished function for reporting.
 | |
| 			jobCtx.finishedCondition = newFailedConditionForFailureTarget(jobCtx.finishedCondition, jm.clock.Now())
 | |
| 		}
 | |
| 	}
 | |
| 	if isSuccessCriteriaMetCondition(jobCtx.finishedCondition) {
 | |
| 		// Append the interim SuccessCriteriaMet condition to update the job status with before finalizers are removed.
 | |
| 		if hasSuccessCriteriaMetCondition(jobCtx.job) == nil {
 | |
| 			jobCtx.job.Status.Conditions = append(jobCtx.job.Status.Conditions, *jobCtx.finishedCondition)
 | |
| 			needsFlush = true
 | |
| 		}
 | |
| 
 | |
| 		// Prepare the final Complete condition to update the job status with after the finalizers are removed.
 | |
| 		// It is also used in the enactJobFinished function for reporting.
 | |
| 		jobCtx.finishedCondition = newCondition(batch.JobComplete, v1.ConditionTrue, jobCtx.finishedCondition.Reason, jobCtx.finishedCondition.Message, jm.clock.Now())
 | |
| 	}
 | |
| 	var err error
 | |
| 	if jobCtx.job, needsFlush, err = jm.flushUncountedAndRemoveFinalizers(ctx, jobCtx, podsToRemoveFinalizer, uidsWithFinalizer, &oldCounters, podFailureCountByPolicyAction, needsFlush); err != nil {
 | |
| 		return err
 | |
| 	}
 | |
| 	jobFinished := !reachedMaxUncountedPods && jm.enactJobFinished(jobCtx.job, jobCtx.finishedCondition)
 | |
| 	if jobFinished {
 | |
| 		needsFlush = true
 | |
| 	}
 | |
| 	if needsFlush {
 | |
| 		if _, err := jm.updateStatusHandler(ctx, jobCtx.job); err != nil {
 | |
| 			return fmt.Errorf("removing uncounted pods from status: %w", err)
 | |
| 		}
 | |
| 		if jobFinished {
 | |
| 			jm.recordJobFinished(jobCtx.job, jobCtx.finishedCondition)
 | |
| 		}
 | |
| 		recordJobPodFinished(logger, jobCtx.job, oldCounters)
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // canRemoveFinalizer determines if the pod's finalizer can be safely removed.
 | |
| // The finalizer can be removed when:
 | |
| //   - the entire Job is terminating; or
 | |
| //   - the pod's index is succeeded; or
 | |
| //   - the Pod is considered failed, unless it's removal is delayed for the
 | |
| //     purpose of transferring the JobIndexFailureCount annotations to the
 | |
| //     replacement pod. the entire Job is terminating the finalizer can be
 | |
| //     removed unconditionally; or
 | |
| //   - the Job met successPolicy.
 | |
| func canRemoveFinalizer(logger klog.Logger, jobCtx *syncJobCtx, pod *v1.Pod, considerPodFailed bool) bool {
 | |
| 	if jobCtx.job.DeletionTimestamp != nil || jobCtx.finishedCondition != nil || pod.Status.Phase == v1.PodSucceeded {
 | |
| 		return true
 | |
| 	}
 | |
| 	if !considerPodFailed {
 | |
| 		return false
 | |
| 	}
 | |
| 	if hasBackoffLimitPerIndex(jobCtx.job) {
 | |
| 		if index := getCompletionIndex(pod.Annotations); index != unknownCompletionIndex {
 | |
| 			if p, ok := jobCtx.podsWithDelayedDeletionPerIndex[index]; ok && p.UID == pod.UID {
 | |
| 				logger.V(3).Info("Delaying pod finalizer removal to await for pod recreation within the index", "pod", klog.KObj(pod))
 | |
| 				return false
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	return true
 | |
| }
 | |
| 
 | |
| // flushUncountedAndRemoveFinalizers does:
 | |
| //  1. flush the Job status that might include new uncounted Pod UIDs.
 | |
| //     Also flush the interim FailureTarget and SuccessCriteriaMet conditions if present.
 | |
| //  2. perform the removal of finalizers from Pods which are in the uncounted
 | |
| //     lists.
 | |
| //  3. update the counters based on the Pods for which it successfully removed
 | |
| //     the finalizers.
 | |
| //  4. (if not all removals succeeded) flush Job status again.
 | |
| //
 | |
| // Returns whether there are pending changes in the Job status that need to be
 | |
| // flushed in subsequent calls.
 | |
| func (jm *Controller) flushUncountedAndRemoveFinalizers(ctx context.Context, jobCtx *syncJobCtx, podsToRemoveFinalizer []*v1.Pod, uidsWithFinalizer sets.Set[string], oldCounters *batch.JobStatus, podFailureCountByPolicyAction map[string]int, needsFlush bool) (*batch.Job, bool, error) {
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 	var err error
 | |
| 	if needsFlush {
 | |
| 		if jobCtx.job, err = jm.updateStatusHandler(ctx, jobCtx.job); err != nil {
 | |
| 			return jobCtx.job, needsFlush, fmt.Errorf("adding uncounted pods to status: %w", err)
 | |
| 		}
 | |
| 
 | |
| 		err = jm.podBackoffStore.updateBackoffRecord(jobCtx.newBackoffRecord)
 | |
| 
 | |
| 		if err != nil {
 | |
| 			// this error might undercount the backoff.
 | |
| 			// re-syncing from the current state might not help to recover
 | |
| 			// the backoff information
 | |
| 			logger.Error(err, "Backoff update failed")
 | |
| 		}
 | |
| 
 | |
| 		recordJobPodFinished(logger, jobCtx.job, *oldCounters)
 | |
| 		// Shallow copy, as it will only be used to detect changes in the counters.
 | |
| 		*oldCounters = jobCtx.job.Status
 | |
| 		needsFlush = false
 | |
| 	}
 | |
| 	recordJobPodFailurePolicyActions(jobCtx.job, podFailureCountByPolicyAction)
 | |
| 
 | |
| 	jobKey, err := controller.KeyFunc(jobCtx.job)
 | |
| 	if err != nil {
 | |
| 		return jobCtx.job, needsFlush, fmt.Errorf("getting job key: %w", err)
 | |
| 	}
 | |
| 	var rmErr error
 | |
| 	if len(podsToRemoveFinalizer) > 0 {
 | |
| 		var rmSucceded []bool
 | |
| 		rmSucceded, rmErr = jm.removeTrackingFinalizerFromPods(ctx, jobKey, podsToRemoveFinalizer)
 | |
| 		for i, p := range podsToRemoveFinalizer {
 | |
| 			if rmSucceded[i] {
 | |
| 				uidsWithFinalizer.Delete(string(p.UID))
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	// Failed to remove some finalizers. Attempt to update the status with the
 | |
| 	// partial progress.
 | |
| 	if cleanUncountedPodsWithoutFinalizers(&jobCtx.job.Status, uidsWithFinalizer) {
 | |
| 		needsFlush = true
 | |
| 	}
 | |
| 	if rmErr != nil && needsFlush {
 | |
| 		if job, err := jm.updateStatusHandler(ctx, jobCtx.job); err != nil {
 | |
| 			return job, needsFlush, fmt.Errorf("removing uncounted pods from status: %w", err)
 | |
| 		}
 | |
| 	}
 | |
| 	return jobCtx.job, needsFlush, rmErr
 | |
| }
 | |
| 
 | |
| // cleanUncountedPodsWithoutFinalizers removes the Pod UIDs from
 | |
| // .status.uncountedTerminatedPods for which the finalizer was successfully
 | |
| // removed and increments the corresponding status counters.
 | |
| // Returns whether there was any status change.
 | |
| func cleanUncountedPodsWithoutFinalizers(status *batch.JobStatus, uidsWithFinalizer sets.Set[string]) bool {
 | |
| 	updated := false
 | |
| 	uncountedStatus := status.UncountedTerminatedPods
 | |
| 	newUncounted := filterInUncountedUIDs(uncountedStatus.Succeeded, uidsWithFinalizer)
 | |
| 	if len(newUncounted) != len(uncountedStatus.Succeeded) {
 | |
| 		updated = true
 | |
| 		status.Succeeded += int32(len(uncountedStatus.Succeeded) - len(newUncounted))
 | |
| 		uncountedStatus.Succeeded = newUncounted
 | |
| 	}
 | |
| 	newUncounted = filterInUncountedUIDs(uncountedStatus.Failed, uidsWithFinalizer)
 | |
| 	if len(newUncounted) != len(uncountedStatus.Failed) {
 | |
| 		updated = true
 | |
| 		status.Failed += int32(len(uncountedStatus.Failed) - len(newUncounted))
 | |
| 		uncountedStatus.Failed = newUncounted
 | |
| 	}
 | |
| 	return updated
 | |
| }
 | |
| 
 | |
| // removeTrackingFinalizerFromPods removes tracking finalizers from Pods and
 | |
| // returns an array of booleans where the i-th value is true if the finalizer
 | |
| // of the i-th Pod was successfully removed (if the pod was deleted when this
 | |
| // function was called, it's considered as the finalizer was removed successfully).
 | |
| func (jm *Controller) removeTrackingFinalizerFromPods(ctx context.Context, jobKey string, pods []*v1.Pod) ([]bool, error) {
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 	errCh := make(chan error, len(pods))
 | |
| 	succeeded := make([]bool, len(pods))
 | |
| 	uids := make([]string, len(pods))
 | |
| 	for i, p := range pods {
 | |
| 		uids[i] = string(p.UID)
 | |
| 	}
 | |
| 	if jobKey != "" {
 | |
| 		err := jm.finalizerExpectations.expectFinalizersRemoved(logger, jobKey, uids)
 | |
| 		if err != nil {
 | |
| 			return succeeded, fmt.Errorf("setting expected removed finalizers: %w", err)
 | |
| 		}
 | |
| 	}
 | |
| 	wg := sync.WaitGroup{}
 | |
| 	wg.Add(len(pods))
 | |
| 	for i := range pods {
 | |
| 		go func(i int) {
 | |
| 			pod := pods[i]
 | |
| 			defer wg.Done()
 | |
| 			if patch := removeTrackingFinalizerPatch(pod); patch != nil {
 | |
| 				if err := jm.podControl.PatchPod(ctx, pod.Namespace, pod.Name, patch); err != nil {
 | |
| 					// In case of any failure, we don't expect a Pod update for the
 | |
| 					// finalizer removed. Clear expectation now.
 | |
| 					if jobKey != "" {
 | |
| 						jm.finalizerExpectations.finalizerRemovalObserved(logger, jobKey, string(pod.UID))
 | |
| 					}
 | |
| 					if !apierrors.IsNotFound(err) {
 | |
| 						errCh <- err
 | |
| 						utilruntime.HandleError(fmt.Errorf("removing tracking finalizer: %w", err))
 | |
| 						return
 | |
| 					}
 | |
| 				}
 | |
| 				succeeded[i] = true
 | |
| 			}
 | |
| 		}(i)
 | |
| 	}
 | |
| 	wg.Wait()
 | |
| 
 | |
| 	return succeeded, errorFromChannel(errCh)
 | |
| }
 | |
| 
 | |
| // enactJobFinished adds the Complete or Failed condition and records events.
 | |
| // Returns whether the Job was considered finished.
 | |
| func (jm *Controller) enactJobFinished(job *batch.Job, finishedCond *batch.JobCondition) bool {
 | |
| 	if finishedCond == nil {
 | |
| 		return false
 | |
| 	}
 | |
| 	if uncounted := job.Status.UncountedTerminatedPods; uncounted != nil {
 | |
| 		if len(uncounted.Succeeded) > 0 || len(uncounted.Failed) > 0 {
 | |
| 			return false
 | |
| 		}
 | |
| 	}
 | |
| 	job.Status.Conditions, _ = ensureJobConditionStatus(job.Status.Conditions, finishedCond.Type, finishedCond.Status, finishedCond.Reason, finishedCond.Message, jm.clock.Now())
 | |
| 	if finishedCond.Type == batch.JobComplete {
 | |
| 		job.Status.CompletionTime = &finishedCond.LastTransitionTime
 | |
| 	}
 | |
| 	return true
 | |
| }
 | |
| 
 | |
| // recordJobFinished records events and the job_finished_total metric for a finished job.
 | |
| func (jm *Controller) recordJobFinished(job *batch.Job, finishedCond *batch.JobCondition) bool {
 | |
| 	completionMode := getCompletionMode(job)
 | |
| 	if finishedCond.Type == batch.JobComplete {
 | |
| 		if job.Spec.Completions != nil && job.Status.Succeeded > *job.Spec.Completions {
 | |
| 			jm.recorder.Event(job, v1.EventTypeWarning, "TooManySucceededPods", "Too many succeeded pods running after completion count reached")
 | |
| 		}
 | |
| 		jm.recorder.Event(job, v1.EventTypeNormal, "Completed", "Job completed")
 | |
| 		metrics.JobFinishedNum.WithLabelValues(completionMode, "succeeded", "").Inc()
 | |
| 	} else {
 | |
| 		jm.recorder.Event(job, v1.EventTypeWarning, finishedCond.Reason, finishedCond.Message)
 | |
| 		metrics.JobFinishedNum.WithLabelValues(completionMode, "failed", finishedCond.Reason).Inc()
 | |
| 	}
 | |
| 	return true
 | |
| }
 | |
| 
 | |
| func filterInUncountedUIDs(uncounted []types.UID, include sets.Set[string]) []types.UID {
 | |
| 	var newUncounted []types.UID
 | |
| 	for _, uid := range uncounted {
 | |
| 		if include.Has(string(uid)) {
 | |
| 			newUncounted = append(newUncounted, uid)
 | |
| 		}
 | |
| 	}
 | |
| 	return newUncounted
 | |
| }
 | |
| 
 | |
| // newFailedConditionForFailureTarget creates a job Failed condition based on
 | |
| // the interim FailureTarget condition.
 | |
| func newFailedConditionForFailureTarget(condition *batch.JobCondition, now time.Time) *batch.JobCondition {
 | |
| 	return newCondition(batch.JobFailed, v1.ConditionTrue, condition.Reason, condition.Message, now)
 | |
| }
 | |
| 
 | |
| // pastBackoffLimitOnFailure checks if container restartCounts sum exceeds BackoffLimit
 | |
| // this method applies only to pods with restartPolicy == OnFailure
 | |
| func pastBackoffLimitOnFailure(job *batch.Job, pods []*v1.Pod) bool {
 | |
| 	if job.Spec.Template.Spec.RestartPolicy != v1.RestartPolicyOnFailure {
 | |
| 		return false
 | |
| 	}
 | |
| 	result := int32(0)
 | |
| 	for i := range pods {
 | |
| 		po := pods[i]
 | |
| 		if po.Status.Phase == v1.PodRunning || po.Status.Phase == v1.PodPending {
 | |
| 			for j := range po.Status.InitContainerStatuses {
 | |
| 				stat := po.Status.InitContainerStatuses[j]
 | |
| 				result += stat.RestartCount
 | |
| 			}
 | |
| 			for j := range po.Status.ContainerStatuses {
 | |
| 				stat := po.Status.ContainerStatuses[j]
 | |
| 				result += stat.RestartCount
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	if *job.Spec.BackoffLimit == 0 {
 | |
| 		return result > 0
 | |
| 	}
 | |
| 	return result >= *job.Spec.BackoffLimit
 | |
| }
 | |
| 
 | |
| // pastActiveDeadline checks if job has ActiveDeadlineSeconds field set and if
 | |
| // it is exceeded. If the job is currently suspended, the function will always
 | |
| // return false.
 | |
| func (jm *Controller) pastActiveDeadline(job *batch.Job) bool {
 | |
| 	if job.Spec.ActiveDeadlineSeconds == nil || job.Status.StartTime == nil || jobSuspended(job) {
 | |
| 		return false
 | |
| 	}
 | |
| 	duration := jm.clock.Since(job.Status.StartTime.Time)
 | |
| 	allowedDuration := time.Duration(*job.Spec.ActiveDeadlineSeconds) * time.Second
 | |
| 	return duration >= allowedDuration
 | |
| }
 | |
| 
 | |
| func newCondition(conditionType batch.JobConditionType, status v1.ConditionStatus, reason, message string, now time.Time) *batch.JobCondition {
 | |
| 	return &batch.JobCondition{
 | |
| 		Type:               conditionType,
 | |
| 		Status:             status,
 | |
| 		LastProbeTime:      metav1.NewTime(now),
 | |
| 		LastTransitionTime: metav1.NewTime(now),
 | |
| 		Reason:             reason,
 | |
| 		Message:            message,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| // getFailJobMessage returns a job failure message if the job should fail with the current counters
 | |
| func getFailJobMessage(job *batch.Job, pods []*v1.Pod) *string {
 | |
| 	if !feature.DefaultFeatureGate.Enabled(features.JobPodFailurePolicy) || job.Spec.PodFailurePolicy == nil {
 | |
| 		return nil
 | |
| 	}
 | |
| 	for _, p := range pods {
 | |
| 		if isPodFailed(p, job) {
 | |
| 			jobFailureMessage, _, _ := matchPodFailurePolicy(job.Spec.PodFailurePolicy, p)
 | |
| 			if jobFailureMessage != nil {
 | |
| 				return jobFailureMessage
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // getNewFinishedPods returns the list of newly succeeded and failed pods that are not accounted
 | |
| // in the job status. The list of failed pods can be affected by the podFailurePolicy.
 | |
| func getNewFinishedPods(jobCtx *syncJobCtx) (succeededPods, failedPods []*v1.Pod) {
 | |
| 	succeededPods = getValidPodsWithFilter(jobCtx, jobCtx.uncounted.Succeeded(), func(p *v1.Pod) bool {
 | |
| 		return p.Status.Phase == v1.PodSucceeded
 | |
| 	})
 | |
| 	failedPods = getValidPodsWithFilter(jobCtx, jobCtx.uncounted.Failed(), func(p *v1.Pod) bool {
 | |
| 		return isPodFailed(p, jobCtx.job)
 | |
| 	})
 | |
| 	return succeededPods, failedPods
 | |
| }
 | |
| 
 | |
| // jobSuspended returns whether a Job is suspended while taking the feature
 | |
| // gate into account.
 | |
| func jobSuspended(job *batch.Job) bool {
 | |
| 	return job.Spec.Suspend != nil && *job.Spec.Suspend
 | |
| }
 | |
| 
 | |
| // manageJob is the core method responsible for managing the number of running
 | |
| // pods according to what is specified in the job.Spec.
 | |
| // Respects back-off; does not create new pods if the back-off time has not passed
 | |
| // Does NOT modify <activePods>.
 | |
| func (jm *Controller) manageJob(ctx context.Context, job *batch.Job, jobCtx *syncJobCtx) (int32, string, error) {
 | |
| 	logger := klog.FromContext(ctx)
 | |
| 	active := int32(len(jobCtx.activePods))
 | |
| 	parallelism := *job.Spec.Parallelism
 | |
| 	jobKey, err := controller.KeyFunc(job)
 | |
| 	if err != nil {
 | |
| 		utilruntime.HandleError(fmt.Errorf("Couldn't get key for job %#v: %v", job, err))
 | |
| 		return 0, metrics.JobSyncActionTracking, nil
 | |
| 	}
 | |
| 
 | |
| 	if jobSuspended(job) {
 | |
| 		logger.V(4).Info("Deleting all active pods in suspended job", "job", klog.KObj(job), "active", active)
 | |
| 		podsToDelete := activePodsForRemoval(job, jobCtx.activePods, int(active))
 | |
| 		jm.expectations.ExpectDeletions(logger, jobKey, len(podsToDelete))
 | |
| 		removed, err := jm.deleteJobPods(ctx, job, jobKey, podsToDelete)
 | |
| 		active -= removed
 | |
| 		return active, metrics.JobSyncActionPodsDeleted, err
 | |
| 	}
 | |
| 
 | |
| 	var terminating int32 = 0
 | |
| 	if onlyReplaceFailedPods(jobCtx.job) {
 | |
| 		// For PodFailurePolicy specified but PodReplacementPolicy disabled
 | |
| 		// we still need to count terminating pods for replica counts
 | |
| 		// But we will not allow updates to status.
 | |
| 		if jobCtx.terminating == nil {
 | |
| 			terminating = controller.CountTerminatingPods(jobCtx.pods)
 | |
| 		} else {
 | |
| 			terminating = *jobCtx.terminating
 | |
| 		}
 | |
| 	}
 | |
| 	wantActive := int32(0)
 | |
| 	if job.Spec.Completions == nil {
 | |
| 		// Job does not specify a number of completions.  Therefore, number active
 | |
| 		// should be equal to parallelism, unless the job has seen at least
 | |
| 		// once success, in which leave whatever is running, running.
 | |
| 		if jobCtx.succeeded > 0 {
 | |
| 			wantActive = active
 | |
| 		} else {
 | |
| 			wantActive = parallelism
 | |
| 		}
 | |
| 	} else {
 | |
| 		// Job specifies a specific number of completions.  Therefore, number
 | |
| 		// active should not ever exceed number of remaining completions.
 | |
| 		wantActive = *job.Spec.Completions - jobCtx.succeeded
 | |
| 		if wantActive > parallelism {
 | |
| 			wantActive = parallelism
 | |
| 		}
 | |
| 		if wantActive < 0 {
 | |
| 			wantActive = 0
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	rmAtLeast := active - wantActive
 | |
| 	if rmAtLeast < 0 {
 | |
| 		rmAtLeast = 0
 | |
| 	}
 | |
| 	podsToDelete := activePodsForRemoval(job, jobCtx.activePods, int(rmAtLeast))
 | |
| 	if len(podsToDelete) > MaxPodCreateDeletePerSync {
 | |
| 		podsToDelete = podsToDelete[:MaxPodCreateDeletePerSync]
 | |
| 	}
 | |
| 	if len(podsToDelete) > 0 {
 | |
| 		jm.expectations.ExpectDeletions(logger, jobKey, len(podsToDelete))
 | |
| 		logger.V(4).Info("Too many pods running for job", "job", klog.KObj(job), "deleted", len(podsToDelete), "target", wantActive)
 | |
| 		removed, err := jm.deleteJobPods(ctx, job, jobKey, podsToDelete)
 | |
| 		active -= removed
 | |
| 		// While it is possible for a Job to require both pod creations and
 | |
| 		// deletions at the same time (e.g. indexed Jobs with repeated indexes), we
 | |
| 		// restrict ourselves to either just pod deletion or pod creation in any
 | |
| 		// given sync cycle. Of these two, pod deletion takes precedence.
 | |
| 		return active, metrics.JobSyncActionPodsDeleted, err
 | |
| 	}
 | |
| 
 | |
| 	if diff := wantActive - terminating - active; diff > 0 {
 | |
| 		var remainingTime time.Duration
 | |
| 		if !hasBackoffLimitPerIndex(job) {
 | |
| 			// we compute the global remaining time for pod creation when backoffLimitPerIndex is not used
 | |
| 			remainingTime = jobCtx.newBackoffRecord.getRemainingTime(jm.clock, DefaultJobPodFailureBackOff, MaxJobPodFailureBackOff)
 | |
| 		}
 | |
| 		if remainingTime > 0 {
 | |
| 			jm.enqueueSyncJobWithDelay(logger, job, remainingTime)
 | |
| 			return 0, metrics.JobSyncActionPodsCreated, nil
 | |
| 		}
 | |
| 		if diff > int32(MaxPodCreateDeletePerSync) {
 | |
| 			diff = int32(MaxPodCreateDeletePerSync)
 | |
| 		}
 | |
| 
 | |
| 		var indexesToAdd []int
 | |
| 		if isIndexedJob(job) {
 | |
| 			indexesToAdd = firstPendingIndexes(jobCtx, int(diff), int(*job.Spec.Completions))
 | |
| 			if hasBackoffLimitPerIndex(job) {
 | |
| 				indexesToAdd, remainingTime = jm.getPodCreationInfoForIndependentIndexes(logger, indexesToAdd, jobCtx.podsWithDelayedDeletionPerIndex)
 | |
| 				if remainingTime > 0 {
 | |
| 					jm.enqueueSyncJobWithDelay(logger, job, remainingTime)
 | |
| 					return 0, metrics.JobSyncActionPodsCreated, nil
 | |
| 				}
 | |
| 			}
 | |
| 			diff = int32(len(indexesToAdd))
 | |
| 		}
 | |
| 
 | |
| 		jm.expectations.ExpectCreations(logger, jobKey, int(diff))
 | |
| 		errCh := make(chan error, diff)
 | |
| 		logger.V(4).Info("Too few pods running", "key", jobKey, "need", wantActive, "creating", diff)
 | |
| 
 | |
| 		wait := sync.WaitGroup{}
 | |
| 
 | |
| 		active += diff
 | |
| 
 | |
| 		podTemplate := job.Spec.Template.DeepCopy()
 | |
| 		if isIndexedJob(job) {
 | |
| 			addCompletionIndexEnvVariables(podTemplate)
 | |
| 		}
 | |
| 		podTemplate.Finalizers = appendJobCompletionFinalizerIfNotFound(podTemplate.Finalizers)
 | |
| 
 | |
| 		// Counters for pod creation status (used by the job_pods_creation_total metric)
 | |
| 		var creationsSucceeded, creationsFailed int32 = 0, 0
 | |
| 
 | |
| 		// 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.
 | |
| 		for batchSize := min(diff, int32(controller.SlowStartInitialBatchSize)); diff > 0; batchSize = min(2*batchSize, diff) {
 | |
| 			errorCount := len(errCh)
 | |
| 			wait.Add(int(batchSize))
 | |
| 			for i := int32(0); i < batchSize; i++ {
 | |
| 				completionIndex := unknownCompletionIndex
 | |
| 				if len(indexesToAdd) > 0 {
 | |
| 					completionIndex = indexesToAdd[0]
 | |
| 					indexesToAdd = indexesToAdd[1:]
 | |
| 				}
 | |
| 				go func() {
 | |
| 					template := podTemplate
 | |
| 					generateName := ""
 | |
| 					if completionIndex != unknownCompletionIndex {
 | |
| 						template = podTemplate.DeepCopy()
 | |
| 						addCompletionIndexAnnotation(template, completionIndex)
 | |
| 
 | |
| 						if feature.DefaultFeatureGate.Enabled(features.PodIndexLabel) {
 | |
| 							addCompletionIndexLabel(template, completionIndex)
 | |
| 						}
 | |
| 						template.Spec.Hostname = fmt.Sprintf("%s-%d", job.Name, completionIndex)
 | |
| 						generateName = podGenerateNameWithIndex(job.Name, completionIndex)
 | |
| 						if hasBackoffLimitPerIndex(job) {
 | |
| 							addIndexFailureCountAnnotation(logger, template, job, jobCtx.podsWithDelayedDeletionPerIndex[completionIndex])
 | |
| 						}
 | |
| 					}
 | |
| 					defer wait.Done()
 | |
| 					err := jm.podControl.CreatePodsWithGenerateName(ctx, job.Namespace, template, job, metav1.NewControllerRef(job, controllerKind), generateName)
 | |
| 					if err != nil {
 | |
| 						if apierrors.HasStatusCause(err, v1.NamespaceTerminatingCause) {
 | |
| 							// If the namespace is being torn down, we can safely ignore
 | |
| 							// this error since all subsequent creations will fail.
 | |
| 							return
 | |
| 						}
 | |
| 					}
 | |
| 					if err != nil {
 | |
| 						defer utilruntime.HandleError(err)
 | |
| 						// Decrement the expected number of creates because the informer won't observe this pod
 | |
| 						logger.V(2).Info("Failed creation, decrementing expectations", "job", klog.KObj(job))
 | |
| 						jm.expectations.CreationObserved(logger, jobKey)
 | |
| 						atomic.AddInt32(&active, -1)
 | |
| 						errCh <- err
 | |
| 						atomic.AddInt32(&creationsFailed, 1)
 | |
| 					}
 | |
| 					atomic.AddInt32(&creationsSucceeded, 1)
 | |
| 				}()
 | |
| 			}
 | |
| 			wait.Wait()
 | |
| 			// any skipped pods that we never attempted to start shouldn't be expected.
 | |
| 			skippedPods := diff - batchSize
 | |
| 			if errorCount < len(errCh) && skippedPods > 0 {
 | |
| 				logger.V(2).Info("Slow-start failure. Skipping creating pods, decrementing expectations", "skippedCount", skippedPods, "job", klog.KObj(job))
 | |
| 				active -= skippedPods
 | |
| 				for i := int32(0); i < skippedPods; i++ {
 | |
| 					// Decrement the expected number of creates because the informer won't observe this pod
 | |
| 					jm.expectations.CreationObserved(logger, jobKey)
 | |
| 				}
 | |
| 				// The skipped pods will be retried later. The next controller resync will
 | |
| 				// retry the slow start process.
 | |
| 				break
 | |
| 			}
 | |
| 			diff -= batchSize
 | |
| 		}
 | |
| 		recordJobPodsCreationTotal(job, jobCtx, creationsSucceeded, creationsFailed)
 | |
| 		return active, metrics.JobSyncActionPodsCreated, errorFromChannel(errCh)
 | |
| 	}
 | |
| 
 | |
| 	return active, metrics.JobSyncActionTracking, nil
 | |
| }
 | |
| 
 | |
| // getPodCreationInfoForIndependentIndexes returns a sub-list of all indexes
 | |
| // to create that contains those which can be already created. In case no indexes
 | |
| // are ready to create pods, it returns the lowest remaining time to create pods
 | |
| // out of all indexes.
 | |
| func (jm *Controller) getPodCreationInfoForIndependentIndexes(logger klog.Logger, indexesToAdd []int, podsWithDelayedDeletionPerIndex map[int]*v1.Pod) ([]int, time.Duration) {
 | |
| 	var indexesToAddNow []int
 | |
| 	var minRemainingTimePerIndex *time.Duration
 | |
| 	for _, indexToAdd := range indexesToAdd {
 | |
| 		if remainingTimePerIndex := getRemainingTimePerIndex(logger, jm.clock, DefaultJobPodFailureBackOff, MaxJobPodFailureBackOff, podsWithDelayedDeletionPerIndex[indexToAdd]); remainingTimePerIndex == 0 {
 | |
| 			indexesToAddNow = append(indexesToAddNow, indexToAdd)
 | |
| 		} else if minRemainingTimePerIndex == nil || remainingTimePerIndex < *minRemainingTimePerIndex {
 | |
| 			minRemainingTimePerIndex = &remainingTimePerIndex
 | |
| 		}
 | |
| 	}
 | |
| 	if len(indexesToAddNow) > 0 {
 | |
| 		return indexesToAddNow, 0
 | |
| 	}
 | |
| 	return indexesToAddNow, ptr.Deref(minRemainingTimePerIndex, 0)
 | |
| }
 | |
| 
 | |
| // activePodsForRemoval returns Pods that should be removed because there
 | |
| // are too many pods running or, if this is an indexed job, there are repeated
 | |
| // indexes or invalid indexes or some pods don't have indexes.
 | |
| // Sorts candidate 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.
 | |
| func activePodsForRemoval(job *batch.Job, pods []*v1.Pod, rmAtLeast int) []*v1.Pod {
 | |
| 	var rm, left []*v1.Pod
 | |
| 
 | |
| 	if isIndexedJob(job) {
 | |
| 		rm = make([]*v1.Pod, 0, rmAtLeast)
 | |
| 		left = make([]*v1.Pod, 0, len(pods)-rmAtLeast)
 | |
| 		rm, left = appendDuplicatedIndexPodsForRemoval(rm, left, pods, int(*job.Spec.Completions))
 | |
| 	} else {
 | |
| 		left = pods
 | |
| 	}
 | |
| 
 | |
| 	if len(rm) < rmAtLeast {
 | |
| 		sort.Sort(controller.ActivePods(left))
 | |
| 		rm = append(rm, left[:rmAtLeast-len(rm)]...)
 | |
| 	}
 | |
| 	return rm
 | |
| }
 | |
| 
 | |
| // updateJobStatus calls the API to update the job status.
 | |
| func (jm *Controller) updateJobStatus(ctx context.Context, job *batch.Job) (*batch.Job, error) {
 | |
| 	return jm.kubeClient.BatchV1().Jobs(job.Namespace).UpdateStatus(ctx, job, metav1.UpdateOptions{})
 | |
| }
 | |
| 
 | |
| func (jm *Controller) patchJob(ctx context.Context, job *batch.Job, data []byte) error {
 | |
| 	_, err := jm.kubeClient.BatchV1().Jobs(job.Namespace).Patch(
 | |
| 		ctx, job.Name, types.StrategicMergePatchType, data, metav1.PatchOptions{})
 | |
| 	return err
 | |
| }
 | |
| 
 | |
| // getValidPodsWithFilter returns the valid pods that pass the filter.
 | |
| // Pods are valid if they have a finalizer or in uncounted set
 | |
| // and, for Indexed Jobs, a valid completion index.
 | |
| func getValidPodsWithFilter(jobCtx *syncJobCtx, uncounted sets.Set[string], filter func(*v1.Pod) bool) []*v1.Pod {
 | |
| 	var result []*v1.Pod
 | |
| 	for _, p := range jobCtx.pods {
 | |
| 		uid := string(p.UID)
 | |
| 
 | |
| 		// Pods that don't have a completion finalizer are in the uncounted set or
 | |
| 		// have already been accounted for in the Job status.
 | |
| 		if !hasJobTrackingFinalizer(p) || uncounted.Has(uid) || jobCtx.expectedRmFinalizers.Has(uid) {
 | |
| 			continue
 | |
| 		}
 | |
| 		if isIndexedJob(jobCtx.job) {
 | |
| 			idx := getCompletionIndex(p.Annotations)
 | |
| 			if idx == unknownCompletionIndex || idx >= int(*jobCtx.job.Spec.Completions) {
 | |
| 				continue
 | |
| 			}
 | |
| 		}
 | |
| 		if filter(p) {
 | |
| 			result = append(result, p)
 | |
| 		}
 | |
| 	}
 | |
| 	return result
 | |
| }
 | |
| 
 | |
| // getCompletionMode returns string representation of the completion mode. Used as a label value for metrics.
 | |
| func getCompletionMode(job *batch.Job) string {
 | |
| 	if isIndexedJob(job) {
 | |
| 		return string(batch.IndexedCompletion)
 | |
| 	}
 | |
| 	return string(batch.NonIndexedCompletion)
 | |
| }
 | |
| 
 | |
| func appendJobCompletionFinalizerIfNotFound(finalizers []string) []string {
 | |
| 	for _, fin := range finalizers {
 | |
| 		if fin == batch.JobTrackingFinalizer {
 | |
| 			return finalizers
 | |
| 		}
 | |
| 	}
 | |
| 	return append(finalizers, batch.JobTrackingFinalizer)
 | |
| }
 | |
| 
 | |
| func removeTrackingFinalizerPatch(pod *v1.Pod) []byte {
 | |
| 	if !hasJobTrackingFinalizer(pod) {
 | |
| 		return nil
 | |
| 	}
 | |
| 	patch := map[string]interface{}{
 | |
| 		"metadata": map[string]interface{}{
 | |
| 			"$deleteFromPrimitiveList/finalizers": []string{batch.JobTrackingFinalizer},
 | |
| 		},
 | |
| 	}
 | |
| 	patchBytes, _ := json.Marshal(patch)
 | |
| 	return patchBytes
 | |
| }
 | |
| 
 | |
| type uncountedTerminatedPods struct {
 | |
| 	succeeded sets.Set[string]
 | |
| 	failed    sets.Set[string]
 | |
| }
 | |
| 
 | |
| func newUncountedTerminatedPods(in batch.UncountedTerminatedPods) *uncountedTerminatedPods {
 | |
| 	obj := uncountedTerminatedPods{
 | |
| 		succeeded: make(sets.Set[string], len(in.Succeeded)),
 | |
| 		failed:    make(sets.Set[string], len(in.Failed)),
 | |
| 	}
 | |
| 	for _, v := range in.Succeeded {
 | |
| 		obj.succeeded.Insert(string(v))
 | |
| 	}
 | |
| 	for _, v := range in.Failed {
 | |
| 		obj.failed.Insert(string(v))
 | |
| 	}
 | |
| 	return &obj
 | |
| }
 | |
| 
 | |
| func (u *uncountedTerminatedPods) Succeeded() sets.Set[string] {
 | |
| 	if u == nil {
 | |
| 		return nil
 | |
| 	}
 | |
| 	return u.succeeded
 | |
| }
 | |
| 
 | |
| func (u *uncountedTerminatedPods) Failed() sets.Set[string] {
 | |
| 	if u == nil {
 | |
| 		return nil
 | |
| 	}
 | |
| 	return u.failed
 | |
| }
 | |
| 
 | |
| func errorFromChannel(errCh <-chan error) error {
 | |
| 	select {
 | |
| 	case err := <-errCh:
 | |
| 		return err
 | |
| 	default:
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| // ensureJobConditionStatus appends or updates an existing job condition of the
 | |
| // given type with the given status value. Note that this function will not
 | |
| // append to the conditions list if the new condition's status is false
 | |
| // (because going from nothing to false is meaningless); it can, however,
 | |
| // update the status condition to false. The function returns a bool to let the
 | |
| // caller know if the list was changed (either appended or updated).
 | |
| func ensureJobConditionStatus(list []batch.JobCondition, cType batch.JobConditionType, status v1.ConditionStatus, reason, message string, now time.Time) ([]batch.JobCondition, bool) {
 | |
| 	if condition := findConditionByType(list, cType); condition != nil {
 | |
| 		if condition.Status != status || condition.Reason != reason || condition.Message != message {
 | |
| 			*condition = *newCondition(cType, status, reason, message, now)
 | |
| 			return list, true
 | |
| 		}
 | |
| 		return list, false
 | |
| 	}
 | |
| 	// A condition with that type doesn't exist in the list.
 | |
| 	if status != v1.ConditionFalse {
 | |
| 		return append(list, *newCondition(cType, status, reason, message, now)), true
 | |
| 	}
 | |
| 	return list, false
 | |
| }
 | |
| 
 | |
| func isPodFailed(p *v1.Pod, job *batch.Job) bool {
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.PodDisruptionConditions) && feature.DefaultFeatureGate.Enabled(features.JobPodFailurePolicy) && job.Spec.PodFailurePolicy != nil {
 | |
| 		// When PodDisruptionConditions is enabled, orphan Pods and unschedulable
 | |
| 		// terminating Pods are marked as Failed. So we only need to check the phase.
 | |
| 		// TODO(#113855): Stop limiting this behavior to Jobs with podFailurePolicy.
 | |
| 		// For now, we do so to avoid affecting all running Jobs without the
 | |
| 		// availability to opt-out into the old behavior.
 | |
| 		return p.Status.Phase == v1.PodFailed
 | |
| 	}
 | |
| 	if p.Status.Phase == v1.PodFailed {
 | |
| 		return true
 | |
| 	}
 | |
| 	if onlyReplaceFailedPods(job) {
 | |
| 		return p.Status.Phase == v1.PodFailed
 | |
| 	}
 | |
| 	// Count deleted Pods as failures to account for orphan Pods that
 | |
| 	// never have a chance to reach the Failed phase.
 | |
| 	return p.DeletionTimestamp != nil && p.Status.Phase != v1.PodSucceeded
 | |
| }
 | |
| 
 | |
| func findConditionByType(list []batch.JobCondition, cType batch.JobConditionType) *batch.JobCondition {
 | |
| 	for i := range list {
 | |
| 		if list[i].Type == cType {
 | |
| 			return &list[i]
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func recordJobPodFinished(logger klog.Logger, job *batch.Job, oldCounters batch.JobStatus) {
 | |
| 	completionMode := completionModeStr(job)
 | |
| 	var diff int
 | |
| 
 | |
| 	// Updating succeeded metric must be handled differently
 | |
| 	// for Indexed Jobs to handle the case where the job has
 | |
| 	// been scaled down by reducing completions & parallelism
 | |
| 	// in tandem, and now a previously completed index is
 | |
| 	// now out of range (i.e. index >= spec.Completions).
 | |
| 	if isIndexedJob(job) {
 | |
| 		completions := int(*job.Spec.Completions)
 | |
| 		if job.Status.CompletedIndexes != oldCounters.CompletedIndexes {
 | |
| 			diff = indexesCount(logger, &job.Status.CompletedIndexes, completions) - indexesCount(logger, &oldCounters.CompletedIndexes, completions)
 | |
| 		}
 | |
| 		backoffLimitLabel := backoffLimitMetricsLabel(job)
 | |
| 		metrics.JobFinishedIndexesTotal.WithLabelValues(metrics.Succeeded, backoffLimitLabel).Add(float64(diff))
 | |
| 		if hasBackoffLimitPerIndex(job) && job.Status.FailedIndexes != oldCounters.FailedIndexes {
 | |
| 			if failedDiff := indexesCount(logger, job.Status.FailedIndexes, completions) - indexesCount(logger, oldCounters.FailedIndexes, completions); failedDiff > 0 {
 | |
| 				metrics.JobFinishedIndexesTotal.WithLabelValues(metrics.Failed, backoffLimitLabel).Add(float64(failedDiff))
 | |
| 			}
 | |
| 		}
 | |
| 	} else {
 | |
| 		diff = int(job.Status.Succeeded) - int(oldCounters.Succeeded)
 | |
| 	}
 | |
| 	metrics.JobPodsFinished.WithLabelValues(completionMode, metrics.Succeeded).Add(float64(diff))
 | |
| 
 | |
| 	// Update failed metric.
 | |
| 	diff = int(job.Status.Failed - oldCounters.Failed)
 | |
| 	metrics.JobPodsFinished.WithLabelValues(completionMode, metrics.Failed).Add(float64(diff))
 | |
| }
 | |
| 
 | |
| func indexesCount(logger klog.Logger, indexesStr *string, completions int) int {
 | |
| 	if indexesStr == nil {
 | |
| 		return 0
 | |
| 	}
 | |
| 	return parseIndexesFromString(logger, *indexesStr, completions).total()
 | |
| }
 | |
| 
 | |
| func backoffLimitMetricsLabel(job *batch.Job) string {
 | |
| 	if hasBackoffLimitPerIndex(job) {
 | |
| 		return "perIndex"
 | |
| 	}
 | |
| 	return "global"
 | |
| }
 | |
| 
 | |
| func recordJobPodFailurePolicyActions(job *batch.Job, podFailureCountByPolicyAction map[string]int) {
 | |
| 	for action, count := range podFailureCountByPolicyAction {
 | |
| 		metrics.PodFailuresHandledByFailurePolicy.WithLabelValues(action).Add(float64(count))
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func countReadyPods(pods []*v1.Pod) int32 {
 | |
| 	cnt := int32(0)
 | |
| 	for _, p := range pods {
 | |
| 		if podutil.IsPodReady(p) {
 | |
| 			cnt++
 | |
| 		}
 | |
| 	}
 | |
| 	return cnt
 | |
| }
 | |
| 
 | |
| // This checks if we should apply PodReplacementPolicy.
 | |
| // PodReplacementPolicy controls when we recreate pods if they are marked as terminating
 | |
| // Failed means that we recreate only once the pod has terminated.
 | |
| func onlyReplaceFailedPods(job *batch.Job) bool {
 | |
| 	// We check both PodReplacementPolicy for nil and failed
 | |
| 	// because it is possible that  `PodReplacementPolicy` is not defaulted,
 | |
| 	// when the `JobPodReplacementPolicy` feature gate is disabled for API server.
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.JobPodReplacementPolicy) && job.Spec.PodReplacementPolicy != nil && *job.Spec.PodReplacementPolicy == batch.Failed {
 | |
| 		return true
 | |
| 	}
 | |
| 	return feature.DefaultFeatureGate.Enabled(features.JobPodFailurePolicy) && job.Spec.PodFailurePolicy != nil
 | |
| }
 | |
| 
 | |
| func (jm *Controller) cleanupPodFinalizers(job *batch.Job) {
 | |
| 	// Listing pods shouldn't really fail, as we are just querying the informer cache.
 | |
| 	selector, err := metav1.LabelSelectorAsSelector(job.Spec.Selector)
 | |
| 	if err != nil {
 | |
| 		utilruntime.HandleError(fmt.Errorf("parsing deleted job selector: %v", err))
 | |
| 		return
 | |
| 	}
 | |
| 	pods, _ := jm.podStore.Pods(job.Namespace).List(selector)
 | |
| 	for _, pod := range pods {
 | |
| 		if metav1.IsControlledBy(pod, job) && hasJobTrackingFinalizer(pod) {
 | |
| 			jm.enqueueOrphanPod(pod)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func recordJobPodsCreationTotal(job *batch.Job, jobCtx *syncJobCtx, succeeded, failed int32) {
 | |
| 	reason := metrics.PodCreateNew
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.JobPodReplacementPolicy) {
 | |
| 		if ptr.Deref(job.Spec.PodReplacementPolicy, batch.TerminatingOrFailed) == batch.Failed && jobCtx.failed > 0 {
 | |
| 			reason = metrics.PodRecreateFailed
 | |
| 		} else if jobCtx.failed > 0 || ptr.Deref(jobCtx.terminating, 0) > 0 {
 | |
| 			reason = metrics.PodRecreateTerminatingOrFailed
 | |
| 		}
 | |
| 	}
 | |
| 	if succeeded > 0 {
 | |
| 		metrics.JobPodsCreationTotal.WithLabelValues(reason, metrics.Succeeded).Add(float64(succeeded))
 | |
| 	}
 | |
| 	if failed > 0 {
 | |
| 		metrics.JobPodsCreationTotal.WithLabelValues(reason, metrics.Failed).Add(float64(failed))
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func managedByExternalController(jobObj *batch.Job) *string {
 | |
| 	if feature.DefaultFeatureGate.Enabled(features.JobManagedBy) {
 | |
| 		if controllerName := jobObj.Spec.ManagedBy; controllerName != nil && *controllerName != batch.JobControllerName {
 | |
| 			return controllerName
 | |
| 		}
 | |
| 	}
 | |
| 	return nil
 | |
| }
 |