mirror of https://github.com/k3s-io/k3s
76 lines
2.6 KiB
Go
76 lines
2.6 KiB
Go
/*
|
|
Copyright 2016 The Kubernetes Authors All rights reserved.
|
|
|
|
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 core
|
|
|
|
import (
|
|
"k8s.io/kubernetes/pkg/admission"
|
|
"k8s.io/kubernetes/pkg/api"
|
|
"k8s.io/kubernetes/pkg/api/resource"
|
|
clientset "k8s.io/kubernetes/pkg/client/clientset_generated/internalclientset"
|
|
"k8s.io/kubernetes/pkg/quota"
|
|
"k8s.io/kubernetes/pkg/quota/generic"
|
|
"k8s.io/kubernetes/pkg/runtime"
|
|
)
|
|
|
|
// NewServiceEvaluator returns an evaluator that can evaluate service quotas
|
|
func NewServiceEvaluator(kubeClient clientset.Interface) quota.Evaluator {
|
|
allResources := []api.ResourceName{
|
|
api.ResourceServices,
|
|
api.ResourceServicesNodePorts,
|
|
api.ResourceServicesLoadBalancers,
|
|
}
|
|
return &generic.GenericEvaluator{
|
|
Name: "Evaluator.Service",
|
|
InternalGroupKind: api.Kind("Service"),
|
|
InternalOperationResources: map[admission.Operation][]api.ResourceName{
|
|
admission.Create: allResources,
|
|
admission.Update: allResources,
|
|
},
|
|
MatchedResourceNames: allResources,
|
|
MatchesScopeFunc: generic.MatchesNoScopeFunc,
|
|
ConstraintsFunc: generic.ObjectCountConstraintsFunc(api.ResourceServices),
|
|
UsageFunc: ServiceUsageFunc,
|
|
ListFuncByNamespace: func(namespace string, options api.ListOptions) (runtime.Object, error) {
|
|
return kubeClient.Core().Services(namespace).List(options)
|
|
},
|
|
}
|
|
}
|
|
|
|
// ServiceUsageFunc knows how to measure usage associated with services
|
|
func ServiceUsageFunc(object runtime.Object) api.ResourceList {
|
|
result := api.ResourceList{}
|
|
if service, ok := object.(*api.Service); ok {
|
|
result[api.ResourceServices] = resource.MustParse("1")
|
|
switch service.Spec.Type {
|
|
case api.ServiceTypeNodePort:
|
|
result[api.ResourceServicesNodePorts] = resource.MustParse("1")
|
|
case api.ServiceTypeLoadBalancer:
|
|
result[api.ResourceServicesLoadBalancers] = resource.MustParse("1")
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
// QuotaServiceType returns true if the service type is eligible to track against a quota
|
|
func QuotaServiceType(service *api.Service) bool {
|
|
switch service.Spec.Type {
|
|
case api.ServiceTypeNodePort, api.ServiceTypeLoadBalancer:
|
|
return true
|
|
}
|
|
return false
|
|
}
|