mirror of https://github.com/k3s-io/k3s
205 lines
7.3 KiB
Go
205 lines
7.3 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"
|
|
"strconv"
|
|
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/fields"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/util/validation/field"
|
|
"k8s.io/apiserver/pkg/registry/generic"
|
|
"k8s.io/apiserver/pkg/registry/rest"
|
|
"k8s.io/apiserver/pkg/storage"
|
|
"k8s.io/apiserver/pkg/storage/names"
|
|
"k8s.io/kubernetes/pkg/api/legacyscheme"
|
|
"k8s.io/kubernetes/pkg/api/pod"
|
|
"k8s.io/kubernetes/pkg/apis/batch"
|
|
"k8s.io/kubernetes/pkg/apis/batch/validation"
|
|
)
|
|
|
|
// jobStrategy implements verification logic for Replication Controllers.
|
|
type jobStrategy struct {
|
|
runtime.ObjectTyper
|
|
names.NameGenerator
|
|
}
|
|
|
|
// Strategy is the default logic that applies when creating and updating Replication Controller objects.
|
|
var Strategy = jobStrategy{legacyscheme.Scheme, names.SimpleNameGenerator}
|
|
|
|
// DefaultGarbageCollectionPolicy returns Orphan because that was the default
|
|
// behavior before the server-side garbage collection was implemented.
|
|
func (jobStrategy) DefaultGarbageCollectionPolicy(ctx context.Context) rest.GarbageCollectionPolicy {
|
|
return rest.OrphanDependents
|
|
}
|
|
|
|
// NamespaceScoped returns true because all jobs need to be within a namespace.
|
|
func (jobStrategy) NamespaceScoped() bool {
|
|
return true
|
|
}
|
|
|
|
// PrepareForCreate clears the status of a job before creation.
|
|
func (jobStrategy) PrepareForCreate(ctx context.Context, obj runtime.Object) {
|
|
job := obj.(*batch.Job)
|
|
job.Status = batch.JobStatus{}
|
|
job.Spec.TTLSecondsAfterFinished = nil
|
|
|
|
pod.DropDisabledAlphaFields(&job.Spec.Template.Spec)
|
|
}
|
|
|
|
// PrepareForUpdate clears fields that are not allowed to be set by end users on update.
|
|
func (jobStrategy) PrepareForUpdate(ctx context.Context, obj, old runtime.Object) {
|
|
newJob := obj.(*batch.Job)
|
|
oldJob := old.(*batch.Job)
|
|
newJob.Status = oldJob.Status
|
|
|
|
newJob.Spec.TTLSecondsAfterFinished = nil
|
|
oldJob.Spec.TTLSecondsAfterFinished = nil
|
|
|
|
pod.DropDisabledAlphaFields(&newJob.Spec.Template.Spec)
|
|
pod.DropDisabledAlphaFields(&oldJob.Spec.Template.Spec)
|
|
}
|
|
|
|
// Validate validates a new job.
|
|
func (jobStrategy) Validate(ctx context.Context, obj runtime.Object) field.ErrorList {
|
|
job := obj.(*batch.Job)
|
|
// TODO: move UID generation earlier and do this in defaulting logic?
|
|
if job.Spec.ManualSelector == nil || *job.Spec.ManualSelector == false {
|
|
generateSelector(job)
|
|
}
|
|
return validation.ValidateJob(job)
|
|
}
|
|
|
|
// generateSelector adds a selector to a job and labels to its template
|
|
// which can be used to uniquely identify the pods created by that job,
|
|
// if the user has requested this behavior.
|
|
func generateSelector(obj *batch.Job) {
|
|
if obj.Spec.Template.Labels == nil {
|
|
obj.Spec.Template.Labels = make(map[string]string)
|
|
}
|
|
// The job-name label is unique except in cases that are expected to be
|
|
// quite uncommon, and is more user friendly than uid. So, we add it as
|
|
// a label.
|
|
_, found := obj.Spec.Template.Labels["job-name"]
|
|
if found {
|
|
// User asked us to not automatically generate a selector and labels,
|
|
// but set a possibly conflicting value. If there is a conflict,
|
|
// we will reject in validation.
|
|
} else {
|
|
obj.Spec.Template.Labels["job-name"] = string(obj.ObjectMeta.Name)
|
|
}
|
|
// The controller-uid label makes the pods that belong to this job
|
|
// only match this job.
|
|
_, found = obj.Spec.Template.Labels["controller-uid"]
|
|
if found {
|
|
// User asked us to automatically generate a selector and labels,
|
|
// but set a possibly conflicting value. If there is a conflict,
|
|
// we will reject in validation.
|
|
} else {
|
|
obj.Spec.Template.Labels["controller-uid"] = string(obj.ObjectMeta.UID)
|
|
}
|
|
// Select the controller-uid label. This is sufficient for uniqueness.
|
|
if obj.Spec.Selector == nil {
|
|
obj.Spec.Selector = &metav1.LabelSelector{}
|
|
}
|
|
if obj.Spec.Selector.MatchLabels == nil {
|
|
obj.Spec.Selector.MatchLabels = make(map[string]string)
|
|
}
|
|
if _, found := obj.Spec.Selector.MatchLabels["controller-uid"]; !found {
|
|
obj.Spec.Selector.MatchLabels["controller-uid"] = string(obj.ObjectMeta.UID)
|
|
}
|
|
// If the user specified matchLabel controller-uid=$WRONGUID, then it should fail
|
|
// in validation, either because the selector does not match the pod template
|
|
// (controller-uid=$WRONGUID does not match controller-uid=$UID, which we applied
|
|
// above, or we will reject in validation because the template has the wrong
|
|
// labels.
|
|
}
|
|
|
|
// TODO: generalize generateSelector so it can work for other controller
|
|
// objects such as ReplicaSet. Can use pkg/api/meta to generically get the
|
|
// UID, but need some way to generically access the selector and pod labels
|
|
// fields.
|
|
|
|
// Canonicalize normalizes the object after validation.
|
|
func (jobStrategy) Canonicalize(obj runtime.Object) {
|
|
}
|
|
|
|
func (jobStrategy) AllowUnconditionalUpdate() bool {
|
|
return true
|
|
}
|
|
|
|
// AllowCreateOnUpdate is false for jobs; this means a POST is needed to create one.
|
|
func (jobStrategy) AllowCreateOnUpdate() bool {
|
|
return false
|
|
}
|
|
|
|
// ValidateUpdate is the default update validation for an end user.
|
|
func (jobStrategy) ValidateUpdate(ctx context.Context, obj, old runtime.Object) field.ErrorList {
|
|
validationErrorList := validation.ValidateJob(obj.(*batch.Job))
|
|
updateErrorList := validation.ValidateJobUpdate(obj.(*batch.Job), old.(*batch.Job))
|
|
return append(validationErrorList, updateErrorList...)
|
|
}
|
|
|
|
type jobStatusStrategy struct {
|
|
jobStrategy
|
|
}
|
|
|
|
var StatusStrategy = jobStatusStrategy{Strategy}
|
|
|
|
func (jobStatusStrategy) PrepareForUpdate(ctx context.Context, obj, old runtime.Object) {
|
|
newJob := obj.(*batch.Job)
|
|
oldJob := old.(*batch.Job)
|
|
newJob.Spec = oldJob.Spec
|
|
}
|
|
|
|
func (jobStatusStrategy) ValidateUpdate(ctx context.Context, obj, old runtime.Object) field.ErrorList {
|
|
return validation.ValidateJobUpdateStatus(obj.(*batch.Job), old.(*batch.Job))
|
|
}
|
|
|
|
// JobSelectableFields returns a field set that represents the object for matching purposes.
|
|
func JobToSelectableFields(job *batch.Job) fields.Set {
|
|
objectMetaFieldsSet := generic.ObjectMetaFieldsSet(&job.ObjectMeta, true)
|
|
specificFieldsSet := fields.Set{
|
|
"status.successful": strconv.Itoa(int(job.Status.Succeeded)),
|
|
}
|
|
return generic.MergeFieldsSets(objectMetaFieldsSet, specificFieldsSet)
|
|
}
|
|
|
|
// GetAttrs returns labels and fields of a given object for filtering purposes.
|
|
func GetAttrs(obj runtime.Object) (labels.Set, fields.Set, bool, error) {
|
|
job, ok := obj.(*batch.Job)
|
|
if !ok {
|
|
return nil, nil, false, fmt.Errorf("given object is not a job.")
|
|
}
|
|
return labels.Set(job.ObjectMeta.Labels), JobToSelectableFields(job), job.Initializers != nil, nil
|
|
}
|
|
|
|
// MatchJob is the filter used by the generic etcd backend to route
|
|
// watch events from etcd to clients of the apiserver only interested in specific
|
|
// labels/fields.
|
|
func MatchJob(label labels.Selector, field fields.Selector) storage.SelectionPredicate {
|
|
return storage.SelectionPredicate{
|
|
Label: label,
|
|
Field: field,
|
|
GetAttrs: GetAttrs,
|
|
}
|
|
}
|