github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-24 20:24:09 +00:00
Code Issues Projects Releases Wiki Activity

scheduler: tweak scheduler factory mutex

Browse Source
This commit is contained in:
Wei Huang 2019-04-04 10:32:27 -07:00
parent 9600555969
commit 8f87f580e5
No known key found for this signature in database
GPG Key ID: BE5E9752F8B6E005
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/scheduler/factory/plugins.go
View File

@ -77,7 +77,7 @@ type PriorityConfigFactory struct {
} }
var ( var (
schedulerFactoryMutex sync.Mutex schedulerFactoryMutex sync.RWMutex
// maps that hold registered algorithm types // maps that hold registered algorithm types
fitPredicateMap = make(map[string]FitPredicateFactory) fitPredicateMap = make(map[string]FitPredicateFactory)
@ -139,7 +139,6 @@ func RemovePredicateKeyFromAlgorithmProviderMap(key string) {
for _, provider := range algorithmProviderMap { for _, provider := range algorithmProviderMap {
provider.FitPredicateKeys.Delete(key) provider.FitPredicateKeys.Delete(key)
} }
return
} }
// InsertPredicateKeyToAlgoProvider insert a fit predicate key to algorithmProvider. // InsertPredicateKeyToAlgoProvider insert a fit predicate key to algorithmProvider.
@ -376,8 +375,8 @@ func buildScoringFunctionShapeFromRequestedToCapacityRatioArguments(arguments *s
// IsPriorityFunctionRegistered is useful for testing providers. // IsPriorityFunctionRegistered is useful for testing providers.
func IsPriorityFunctionRegistered(name string) bool { func IsPriorityFunctionRegistered(name string) bool {
schedulerFactoryMutex.Lock() schedulerFactoryMutex.RLock()
defer schedulerFactoryMutex.Unlock() defer schedulerFactoryMutex.RUnlock()
_, ok := priorityFunctionMap[name] _, ok := priorityFunctionMap[name]
return ok return ok
} }
@ -397,8 +396,8 @@ func RegisterAlgorithmProvider(name string, predicateKeys, priorityKeys sets.Str
// GetAlgorithmProvider should not be used to modify providers. It is publicly visible for testing. // GetAlgorithmProvider should not be used to modify providers. It is publicly visible for testing.
func GetAlgorithmProvider(name string) (*AlgorithmProviderConfig, error) { func GetAlgorithmProvider(name string) (*AlgorithmProviderConfig, error) {
schedulerFactoryMutex.Lock() schedulerFactoryMutex.RLock()
defer schedulerFactoryMutex.Unlock() defer schedulerFactoryMutex.RUnlock()
provider, ok := algorithmProviderMap[name] provider, ok := algorithmProviderMap[name]
if !ok { if !ok {