Replaced bool map to string set.

This commit is contained in:
Klaus Ma 2017-08-16 14:57:12 +08:00
parent 4a32bde4a5
commit 2da96fc458

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)
}
}