b975573385
Co-Authored-By: Weibin Lin <linweibin1@huawei.com> Change-Id: I163b2f2833e6b8767f72e2c815dcacd0f4e504ea
92 lines
2.7 KiB
Go
92 lines
2.7 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 apis
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
)
|
|
|
|
const (
|
|
// GA versions of the legacy beta labels.
|
|
// TODO: update kubelet and controllers to set both beta and GA labels, then export these constants
|
|
labelZoneFailureDomainGA = "failure-domain.kubernetes.io/zone"
|
|
labelZoneRegionGA = "failure-domain.kubernetes.io/region"
|
|
labelInstanceTypeGA = "kubernetes.io/instance-type"
|
|
)
|
|
|
|
// When the --failure-domains scheduler flag is not specified,
|
|
// DefaultFailureDomains defines the set of label keys used when TopologyKey is empty in PreferredDuringScheduling anti-affinity.
|
|
var DefaultFailureDomains string = v1.LabelHostname + "," + v1.LabelZoneFailureDomain + "," + v1.LabelZoneRegion
|
|
|
|
var kubeletLabels = sets.NewString(
|
|
v1.LabelHostname,
|
|
v1.LabelZoneFailureDomain,
|
|
v1.LabelZoneRegion,
|
|
v1.LabelInstanceType,
|
|
v1.LabelOS,
|
|
v1.LabelArch,
|
|
|
|
v1.LegacyLabelOS,
|
|
v1.LegacyLabelArch,
|
|
|
|
labelZoneFailureDomainGA,
|
|
labelZoneRegionGA,
|
|
labelInstanceTypeGA,
|
|
)
|
|
|
|
var kubeletLabelNamespaces = sets.NewString(
|
|
v1.LabelNamespaceSuffixKubelet,
|
|
v1.LabelNamespaceSuffixNode,
|
|
)
|
|
|
|
// KubeletLabels returns the list of label keys kubelets are allowed to set on their own Node objects
|
|
func KubeletLabels() []string {
|
|
return kubeletLabels.List()
|
|
}
|
|
|
|
// KubeletLabelNamespaces returns the list of label key namespaces kubelets are allowed to set on their own Node objects
|
|
func KubeletLabelNamespaces() []string {
|
|
return kubeletLabelNamespaces.List()
|
|
}
|
|
|
|
// IsKubeletLabel returns true if the label key is one that kubelets are allowed to set on their own Node object.
|
|
// This checks if the key is in the KubeletLabels() list, or has a namespace in the KubeletLabelNamespaces() list.
|
|
func IsKubeletLabel(key string) bool {
|
|
if kubeletLabels.Has(key) {
|
|
return true
|
|
}
|
|
|
|
namespace := getLabelNamespace(key)
|
|
for allowedNamespace := range kubeletLabelNamespaces {
|
|
if namespace == allowedNamespace || strings.HasSuffix(namespace, "."+allowedNamespace) {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func getLabelNamespace(key string) string {
|
|
if parts := strings.SplitN(key, "/", 2); len(parts) == 2 {
|
|
return parts[0]
|
|
}
|
|
return ""
|
|
}
|