95 lines
2.8 KiB
Go
95 lines
2.8 KiB
Go
/*
|
|
Copyright 2019 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 nodevolumelimits
|
|
|
|
import (
|
|
"strings"
|
|
|
|
v1 "k8s.io/api/core/v1"
|
|
storagev1 "k8s.io/api/storage/v1"
|
|
"k8s.io/apimachinery/pkg/util/sets"
|
|
utilfeature "k8s.io/apiserver/pkg/util/feature"
|
|
csilibplugins "k8s.io/csi-translation-lib/plugins"
|
|
v1helper "k8s.io/kubernetes/pkg/apis/core/v1/helper"
|
|
"k8s.io/kubernetes/pkg/features"
|
|
"k8s.io/kubernetes/pkg/scheduler/framework"
|
|
)
|
|
|
|
// isCSIMigrationOn returns a boolean value indicating whether
|
|
// the CSI migration has been enabled for a particular storage plugin.
|
|
func isCSIMigrationOn(csiNode *storagev1.CSINode, pluginName string) bool {
|
|
if csiNode == nil || len(pluginName) == 0 {
|
|
return false
|
|
}
|
|
|
|
// In-tree storage to CSI driver migration feature should be enabled,
|
|
// along with the plugin-specific one
|
|
switch pluginName {
|
|
case csilibplugins.AWSEBSInTreePluginName:
|
|
if !utilfeature.DefaultFeatureGate.Enabled(features.CSIMigrationAWS) {
|
|
return false
|
|
}
|
|
case csilibplugins.PortworxVolumePluginName:
|
|
if !utilfeature.DefaultFeatureGate.Enabled(features.CSIMigrationPortworx) {
|
|
return false
|
|
}
|
|
case csilibplugins.GCEPDInTreePluginName:
|
|
if !utilfeature.DefaultFeatureGate.Enabled(features.CSIMigrationGCE) {
|
|
return false
|
|
}
|
|
case csilibplugins.AzureDiskInTreePluginName:
|
|
if !utilfeature.DefaultFeatureGate.Enabled(features.CSIMigrationAzureDisk) {
|
|
return false
|
|
}
|
|
case csilibplugins.RBDVolumePluginName:
|
|
if !utilfeature.DefaultFeatureGate.Enabled(features.CSIMigrationRBD) {
|
|
return false
|
|
}
|
|
default:
|
|
return false
|
|
}
|
|
|
|
// The plugin name should be listed in the CSINode object annotation.
|
|
// This indicates that the plugin has been migrated to a CSI driver in the node.
|
|
csiNodeAnn := csiNode.GetAnnotations()
|
|
if csiNodeAnn == nil {
|
|
return false
|
|
}
|
|
|
|
var mpaSet sets.String
|
|
mpa := csiNodeAnn[v1.MigratedPluginsAnnotationKey]
|
|
if len(mpa) == 0 {
|
|
mpaSet = sets.NewString()
|
|
} else {
|
|
tok := strings.Split(mpa, ",")
|
|
mpaSet = sets.NewString(tok...)
|
|
}
|
|
|
|
return mpaSet.Has(pluginName)
|
|
}
|
|
|
|
// volumeLimits returns volume limits associated with the node.
|
|
func volumeLimits(n *framework.NodeInfo) map[v1.ResourceName]int64 {
|
|
volumeLimits := map[v1.ResourceName]int64{}
|
|
for k, v := range n.Allocatable.ScalarResources {
|
|
if v1helper.IsAttachableVolumeResourceName(k) {
|
|
volumeLimits[k] = v
|
|
}
|
|
}
|
|
return volumeLimits
|
|
}
|