github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2026-01-13 11:25:19 +00:00
Code Issues Projects Releases Wiki Activity

Extract logic related with scheduler nominatedPods to an interface

Browse Source 
- rename `UpdateNominatedPodForNode` to `AddNominatedPod`
- promote `update` to `UpdateNominatedPod`
- anonymous lock in nominatedMap
- pass PodNominator as an option to NewFramework
This commit is contained in:
Wei Huang
2020-05-18 10:29:08 -07:00
parent a3d532a3f7
commit bd184035c4
9 changed files with 128 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/scheduler/factory.go
View File

@@ -99,18 +99,21 @@ type Configurator struct {
frameworkCapturer FrameworkCapturer
}
func (c *Configurator) buildFramework(p schedulerapi.KubeSchedulerProfile) (framework.Framework, error) {
func (c *Configurator) buildFramework(p schedulerapi.KubeSchedulerProfile, opts ...framework.Option) (framework.Framework, error) {
if c.frameworkCapturer != nil {
c.frameworkCapturer(p)
}
return framework.NewFramework(
c.registry,
p.Plugins,
p.PluginConfig,
opts = append([]framework.Option{
framework.WithClientSet(c.client),
framework.WithInformerFactory(c.informerFactory),
framework.WithSnapshotSharedLister(c.nodeInfoSnapshot),
framework.WithRunAllFilters(c.alwaysCheckAllPredicates),
}, opts...)
return framework.NewFramework(
c.registry,
p.Plugins,
p.PluginConfig,
opts...,
)
}
@@ -159,7 +162,10 @@ func (c *Configurator) create() (*Scheduler, error) {
}
}
profiles, err := profile.NewMap(c.profiles, c.buildFramework, c.recorderFactory)
// The nominator will be passed all the way to framework instantiation.
nominator := internalqueue.NewPodNominator()
profiles, err := profile.NewMap(c.profiles, c.buildFramework, c.recorderFactory,
framework.WithPodNominator(nominator))
if err != nil {
return nil, fmt.Errorf("initializing profiles: %v", err)
}
@@ -172,6 +178,7 @@ func (c *Configurator) create() (*Scheduler, error) {
lessFn,
internalqueue.WithPodInitialBackoffDuration(time.Duration(c.podInitialBackoffSeconds)*time.Second),
internalqueue.WithPodMaxBackoffDuration(time.Duration(c.podMaxBackoffSeconds)*time.Second),
internalqueue.WithPodNominator(nominator),
)
// Setup cache debugger.