@@ -40,26 +40,26 @@ import (
40
40
"k8s.io/klog"
41
41
)
42
42
43
- // PluginFactoryArgs are passed to all plugin factory functions.
44
- type PluginFactoryArgs struct {
43
+ // AlgorithmFactoryArgs are passed to all factory functions.
44
+ type AlgorithmFactoryArgs struct {
45
45
SharedLister schedulerlisters.SharedLister
46
46
InformerFactory informers.SharedInformerFactory
47
47
VolumeBinder * volumebinder.VolumeBinder
48
48
HardPodAffinitySymmetricWeight int32
49
49
}
50
50
51
51
// PriorityMetadataProducerFactory produces MetadataProducer from the given args.
52
- type PriorityMetadataProducerFactory func (PluginFactoryArgs ) priorities.MetadataProducer
52
+ type PriorityMetadataProducerFactory func (AlgorithmFactoryArgs ) priorities.MetadataProducer
53
53
54
54
// PredicateMetadataProducerFactory produces MetadataProducer from the given args.
55
- type PredicateMetadataProducerFactory func (PluginFactoryArgs ) predicates.MetadataProducer
55
+ type PredicateMetadataProducerFactory func (AlgorithmFactoryArgs ) predicates.MetadataProducer
56
56
57
57
// FitPredicateFactory produces a FitPredicate from the given args.
58
- type FitPredicateFactory func (PluginFactoryArgs ) predicates.FitPredicate
58
+ type FitPredicateFactory func (AlgorithmFactoryArgs ) predicates.FitPredicate
59
59
60
60
// PriorityFunctionFactory produces map & reduce priority functions
61
61
// from a given args.
62
- type PriorityFunctionFactory func (PluginFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction )
62
+ type PriorityFunctionFactory func (AlgorithmFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction )
63
63
64
64
// PriorityConfigFactory produces a PriorityConfig from the given function and weight
65
65
type PriorityConfigFactory struct {
@@ -149,7 +149,7 @@ func ApplyPredicatesAndPriorities(s *Snapshot) {
149
149
// RegisterFitPredicate registers a fit predicate with the algorithm
150
150
// registry. Returns the name with which the predicate was registered.
151
151
func RegisterFitPredicate (name string , predicate predicates.FitPredicate ) string {
152
- return RegisterFitPredicateFactory (name , func (PluginFactoryArgs ) predicates.FitPredicate { return predicate })
152
+ return RegisterFitPredicateFactory (name , func (AlgorithmFactoryArgs ) predicates.FitPredicate { return predicate })
153
153
}
154
154
155
155
// RemoveFitPredicate removes a fit predicate from factory.
@@ -170,7 +170,7 @@ func RemovePredicateKeyFromAlgoProvider(providerName, key string) error {
170
170
validateAlgorithmNameOrDie (providerName )
171
171
provider , ok := algorithmProviderMap [providerName ]
172
172
if ! ok {
173
- return fmt .Errorf ("plugin %v has not been registered" , providerName )
173
+ return fmt .Errorf ("provider %v is not registered" , providerName )
174
174
}
175
175
provider .FitPredicateKeys .Delete (key )
176
176
return nil
@@ -194,7 +194,7 @@ func InsertPredicateKeyToAlgoProvider(providerName, key string) error {
194
194
validateAlgorithmNameOrDie (providerName )
195
195
provider , ok := algorithmProviderMap [providerName ]
196
196
if ! ok {
197
- return fmt .Errorf ("plugin %v has not been registered" , providerName )
197
+ return fmt .Errorf ("provider %v is not registered" , providerName )
198
198
}
199
199
provider .FitPredicateKeys .Insert (key )
200
200
return nil
@@ -227,7 +227,7 @@ func RegisterMandatoryFitPredicate(name string, predicate predicates.FitPredicat
227
227
schedulerFactoryMutex .Lock ()
228
228
defer schedulerFactoryMutex .Unlock ()
229
229
validateAlgorithmNameOrDie (name )
230
- fitPredicateMap [name ] = func (PluginFactoryArgs ) predicates.FitPredicate { return predicate }
230
+ fitPredicateMap [name ] = func (AlgorithmFactoryArgs ) predicates.FitPredicate { return predicate }
231
231
mandatoryFitPredicates .Insert (name )
232
232
return name
233
233
}
@@ -254,7 +254,7 @@ func RegisterCustomFitPredicate(policy schedulerapi.PredicatePolicy, pluginArgs
254
254
// generate the predicate function, if a custom type is requested
255
255
if policy .Argument != nil {
256
256
if policy .Argument .ServiceAffinity != nil {
257
- // We use the ServiceAffinity plugin name for all ServiceAffinity custom priorities .
257
+ // We use the ServiceAffinity predicate name for all ServiceAffinity custom predicates .
258
258
// It may get called multiple times but we essentially only register one instance of ServiceAffinity predicate.
259
259
// This name is then used to find the registered plugin and run the plugin instead of the predicate.
260
260
policyName = predicates .CheckServiceAffinityPred
@@ -266,7 +266,7 @@ func RegisterCustomFitPredicate(policy schedulerapi.PredicatePolicy, pluginArgs
266
266
267
267
pluginArgs .ServiceAffinityArgs .AffinityLabels = append (pluginArgs .ServiceAffinityArgs .AffinityLabels , policy .Argument .ServiceAffinity .Labels ... )
268
268
269
- predicateFactory = func (args PluginFactoryArgs ) predicates.FitPredicate {
269
+ predicateFactory = func (args AlgorithmFactoryArgs ) predicates.FitPredicate {
270
270
predicate , precomputationFunction := predicates .NewServiceAffinityPredicate (
271
271
args .SharedLister .NodeInfos (),
272
272
args .SharedLister .Pods (),
@@ -279,12 +279,12 @@ func RegisterCustomFitPredicate(policy schedulerapi.PredicatePolicy, pluginArgs
279
279
return predicate
280
280
}
281
281
} else if policy .Argument .LabelsPresence != nil {
282
- // We use the NodeLabel plugin name for all NodeLabel custom priorities .
282
+ // We use the CheckNodeLabelPresencePred predicate name for all kNodeLabel custom predicates .
283
283
// It may get called multiple times but we essentially only register one instance of NodeLabel predicate.
284
284
// This name is then used to find the registered plugin and run the plugin instead of the predicate.
285
285
policyName = predicates .CheckNodeLabelPresencePred
286
286
287
- // map LabelPresence policy to ConfigProducerArgs that's used to configure the NodeLabel plugin.
287
+ // Map LabelPresence policy to ConfigProducerArgs that's used to configure the NodeLabel plugin.
288
288
if pluginArgs .NodeLabelArgs == nil {
289
289
pluginArgs .NodeLabelArgs = & nodelabel.Args {}
290
290
}
@@ -293,7 +293,7 @@ func RegisterCustomFitPredicate(policy schedulerapi.PredicatePolicy, pluginArgs
293
293
} else {
294
294
pluginArgs .NodeLabelArgs .AbsentLabels = append (pluginArgs .NodeLabelArgs .AbsentLabels , policy .Argument .LabelsPresence .Labels ... )
295
295
}
296
- predicateFactory = func (_ PluginFactoryArgs ) predicates.FitPredicate {
296
+ predicateFactory = func (_ AlgorithmFactoryArgs ) predicates.FitPredicate {
297
297
return predicates .NewNodeLabelPredicate (
298
298
pluginArgs .NodeLabelArgs .PresentLabels ,
299
299
pluginArgs .NodeLabelArgs .AbsentLabels ,
@@ -343,7 +343,7 @@ func RegisterPriorityMapReduceFunction(
343
343
reduceFunction priorities.PriorityReduceFunction ,
344
344
weight int ) string {
345
345
return RegisterPriorityConfigFactory (name , PriorityConfigFactory {
346
- MapReduceFunction : func (PluginFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
346
+ MapReduceFunction : func (AlgorithmFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
347
347
return mapFunction , reduceFunction
348
348
},
349
349
Weight : int64 (weight ),
@@ -390,7 +390,7 @@ func RegisterCustomPriorityFunction(policy schedulerapi.PriorityPolicy, configPr
390
390
}
391
391
schedulerFactoryMutex .RUnlock ()
392
392
pcf = & PriorityConfigFactory {
393
- MapReduceFunction : func (args PluginFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
393
+ MapReduceFunction : func (args AlgorithmFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
394
394
return priorities .NewServiceAntiAffinityPriority (
395
395
args .SharedLister .Pods (),
396
396
args .InformerFactory .Core ().V1 ().Services ().Lister (),
@@ -421,7 +421,7 @@ func RegisterCustomPriorityFunction(policy schedulerapi.PriorityPolicy, configPr
421
421
}
422
422
schedulerFactoryMutex .RUnlock ()
423
423
pcf = & PriorityConfigFactory {
424
- MapReduceFunction : func (_ PluginFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
424
+ MapReduceFunction : func (_ AlgorithmFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
425
425
return priorities .NewNodeLabelPriority (
426
426
configProducerArgs .NodeLabelArgs .PresentLabelsPreference ,
427
427
configProducerArgs .NodeLabelArgs .AbsentLabelsPreference ,
@@ -436,7 +436,7 @@ func RegisterCustomPriorityFunction(policy schedulerapi.PriorityPolicy, configPr
436
436
ResourceToWeightMap : resources ,
437
437
}
438
438
pcf = & PriorityConfigFactory {
439
- MapReduceFunction : func (args PluginFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
439
+ MapReduceFunction : func (args AlgorithmFactoryArgs ) (priorities.PriorityMapFunction , priorities.PriorityReduceFunction ) {
440
440
p := priorities .RequestedToCapacityRatioResourceAllocationPriority (scoringFunctionShape , resources )
441
441
return p .PriorityMap , nil
442
442
},
@@ -499,8 +499,7 @@ func IsPriorityFunctionRegistered(name string) bool {
499
499
return ok
500
500
}
501
501
502
- // RegisterAlgorithmProvider registers a new algorithm provider with the algorithm registry. This should
503
- // be called from the init function in a provider plugin.
502
+ // RegisterAlgorithmProvider registers a new algorithm provider with the algorithm registry.
504
503
func RegisterAlgorithmProvider (name string , predicateKeys , priorityKeys sets.String ) string {
505
504
schedulerFactoryMutex .Lock ()
506
505
defer schedulerFactoryMutex .Unlock ()
@@ -519,13 +518,13 @@ func GetAlgorithmProvider(name string) (*AlgorithmProviderConfig, error) {
519
518
520
519
provider , ok := algorithmProviderMap [name ]
521
520
if ! ok {
522
- return nil , fmt .Errorf ("plugin %q has not been registered" , name )
521
+ return nil , fmt .Errorf ("provider %q is not registered" , name )
523
522
}
524
523
525
524
return & provider , nil
526
525
}
527
526
528
- func getFitPredicateFunctions (names sets.String , args PluginFactoryArgs ) (map [string ]predicates.FitPredicate , error ) {
527
+ func getFitPredicateFunctions (names sets.String , args AlgorithmFactoryArgs ) (map [string ]predicates.FitPredicate , error ) {
529
528
schedulerFactoryMutex .RLock ()
530
529
defer schedulerFactoryMutex .RUnlock ()
531
530
@@ -548,7 +547,7 @@ func getFitPredicateFunctions(names sets.String, args PluginFactoryArgs) (map[st
548
547
return fitPredicates , nil
549
548
}
550
549
551
- func getPriorityMetadataProducer (args PluginFactoryArgs ) (priorities.MetadataProducer , error ) {
550
+ func getPriorityMetadataProducer (args AlgorithmFactoryArgs ) (priorities.MetadataProducer , error ) {
552
551
schedulerFactoryMutex .Lock ()
553
552
defer schedulerFactoryMutex .Unlock ()
554
553
@@ -558,7 +557,7 @@ func getPriorityMetadataProducer(args PluginFactoryArgs) (priorities.MetadataPro
558
557
return priorityMetadataProducerFactory (args ), nil
559
558
}
560
559
561
- func getPredicateMetadataProducer (args PluginFactoryArgs ) (predicates.MetadataProducer , error ) {
560
+ func getPredicateMetadataProducer (args AlgorithmFactoryArgs ) (predicates.MetadataProducer , error ) {
562
561
schedulerFactoryMutex .Lock ()
563
562
defer schedulerFactoryMutex .Unlock ()
564
563
@@ -568,7 +567,7 @@ func getPredicateMetadataProducer(args PluginFactoryArgs) (predicates.MetadataPr
568
567
return predicateMetadataProducerFactory (args ), nil
569
568
}
570
569
571
- func getPriorityFunctionConfigs (names sets.String , args PluginFactoryArgs ) ([]priorities.PriorityConfig , error ) {
570
+ func getPriorityFunctionConfigs (names sets.String , args AlgorithmFactoryArgs ) ([]priorities.PriorityConfig , error ) {
572
571
schedulerFactoryMutex .RLock ()
573
572
defer schedulerFactoryMutex .RUnlock ()
574
573
0 commit comments