github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-29 06:27:05 +00:00
Code Issues Projects Releases Wiki Activity

Replaced bool map to string set.

Browse Source
This commit is contained in:
Klaus Ma 2017-08-16 14:57:12 +08:00
parent 4a32bde4a5
commit 2da96fc458
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
plugin/pkg/scheduler/factory/plugins.go
View File

@ -73,7 +73,7 @@ var (
// maps that hold registered algorithm types
fitPredicateMap = make(map[string]FitPredicateFactory)
mandatoryFitPredicates = make(map[string]bool)
mandatoryFitPredicates = sets.NewString()
priorityFunctionMap = make(map[string]PriorityConfigFactory)
algorithmProviderMap = make(map[string]AlgorithmProviderConfig)
@ -108,7 +108,7 @@ func RegisterMandatoryFitPredicate(name string, predicate algorithm.FitPredicate
defer schedulerFactoryMutex.Unlock()
validateAlgorithmNameOrDie(name)
fitPredicateMap[name] = func(PluginFactoryArgs) algorithm.FitPredicate { return predicate }
mandatoryFitPredicates[name] = true
mandatoryFitPredicates.Insert(name)
return name
}
@ -323,11 +323,9 @@ func getFitPredicateFunctions(names sets.String, args PluginFactoryArgs) (map[st
}
// Always include mandatory fit predicates.
for name, mandatory := range mandatoryFitPredicates {
if mandatory {
if factory, found := fitPredicateMap[name]; found {
predicates[name] = factory(args)
}
for _, name := range mandatoryFitPredicates.List() {
if factory, found := fitPredicateMap[name]; found {
predicates[name] = factory(args)
}
}