83 lines
3.2 KiB
Go
83 lines
3.2 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package antiaffinity
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/golang/glog"
|
|
"k8s.io/kubernetes/pkg/admission"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
apierrors "k8s.io/kubernetes/pkg/api/errors"
|
|
"k8s.io/kubernetes/pkg/api/unversioned"
|
|
clientset "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset"
|
|
)
|
|
|
|
func init() {
|
|
admission.RegisterPlugin("LimitPodHardAntiAffinityTopology", func(client clientset.Interface, config io.Reader) (admission.Interface, error) {
|
|
return NewInterPodAntiAffinity(client), nil
|
|
})
|
|
}
|
|
|
|
// plugin contains the client used by the admission controller
|
|
type plugin struct {
|
|
*admission.Handler
|
|
client clientset.Interface
|
|
}
|
|
|
|
// NewInterPodAntiAffinity creates a new instance of the LimitPodHardAntiAffinityTopology admission controller
|
|
func NewInterPodAntiAffinity(client clientset.Interface) admission.Interface {
|
|
return &plugin{
|
|
Handler: admission.NewHandler(admission.Create, admission.Update),
|
|
client: client,
|
|
}
|
|
}
|
|
|
|
// Admit will deny any pod that defines AntiAffinity topology key other than unversioned.LabelHostname i.e. "kubernetes.io/hostname"
|
|
// in requiredDuringSchedulingRequiredDuringExecution and requiredDuringSchedulingIgnoredDuringExecution.
|
|
func (p *plugin) Admit(attributes admission.Attributes) (err error) {
|
|
if attributes.GetResource().GroupResource() != api.Resource("pods") {
|
|
return nil
|
|
}
|
|
pod, ok := attributes.GetObject().(*api.Pod)
|
|
if !ok {
|
|
return apierrors.NewBadRequest("Resource was marked with kind Pod but was unable to be converted")
|
|
}
|
|
affinity, err := api.GetAffinityFromPodAnnotations(pod.Annotations)
|
|
if err != nil {
|
|
glog.V(5).Infof("Invalid Affinity detected, but we will leave handling of this to validation phase")
|
|
return nil
|
|
}
|
|
if affinity != nil && affinity.PodAntiAffinity != nil {
|
|
var podAntiAffinityTerms []api.PodAffinityTerm
|
|
if len(affinity.PodAntiAffinity.RequiredDuringSchedulingIgnoredDuringExecution) != 0 {
|
|
podAntiAffinityTerms = affinity.PodAntiAffinity.RequiredDuringSchedulingIgnoredDuringExecution
|
|
}
|
|
// TODO: Uncomment this block when implement RequiredDuringSchedulingRequiredDuringExecution.
|
|
//if len(affinity.PodAntiAffinity.RequiredDuringSchedulingRequiredDuringExecution) != 0 {
|
|
// podAntiAffinityTerms = append(podAntiAffinityTerms, affinity.PodAntiAffinity.RequiredDuringSchedulingRequiredDuringExecution...)
|
|
//}
|
|
for _, v := range podAntiAffinityTerms {
|
|
if v.TopologyKey != unversioned.LabelHostname {
|
|
return apierrors.NewForbidden(attributes.GetResource().GroupResource(), pod.Name, fmt.Errorf("affinity.PodAntiAffinity.RequiredDuringScheduling has TopologyKey %v but only key %v is allowed", v.TopologyKey, unversioned.LabelHostname))
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|