Precheck score plugins' weight when initializing

This commit is contained in:
Jun Gong 2019-07-18 21:19:21 +08:00
parent b5dfcf357a
commit 66347b516c
2 changed files with 8 additions and 9 deletions

View File

@ -93,9 +93,8 @@ func NewFramework(r Registry, plugins *config.Plugins, args []config.PluginConfi
// A weight of zero is not permitted, plugins can be disabled explicitly
// when configured.
f.pluginNameToWeightMap[name] = int(pg[name].Weight)
if f.pluginNameToWeightMap[name] == 0 {
f.pluginNameToWeightMap[name] = 1
if pg[name].Weight != 0 {
f.pluginNameToWeightMap[name] = int(pg[name].Weight)
}
}
@ -121,6 +120,9 @@ func NewFramework(r Registry, plugins *config.Plugins, args []config.PluginConfi
return nil, fmt.Errorf("plugin %v does not extend score plugin", sc.Name)
}
f.scorePlugins = append(f.scorePlugins, p)
if _, exists := f.pluginNameToWeightMap[p.Name()]; !exists {
return nil, fmt.Errorf("score plugin %v is not configured with weight", sc.Name)
}
} else {
return nil, fmt.Errorf("score plugin %v does not exist", sc.Name)
}
@ -276,12 +278,8 @@ func (f *framework) RunScorePlugins(pc *PluginContext, pod *v1.Pod, nodes []*v1.
errCh := schedutil.NewErrorChannel()
workqueue.ParallelizeUntil(ctx, 16, len(nodes), func(index int) {
for _, pl := range f.scorePlugins {
weight, weightExists := f.pluginNameToWeightMap[pl.Name()]
if !weightExists {
err := fmt.Errorf("weight does not exist for plugin %v", pl.Name())
errCh.SendErrorWithCancel(err, cancel)
return
}
// Score plugins' weight has been checked when they are initialized.
weight, _ := f.pluginNameToWeightMap[pl.Name()]
score, status := pl.Score(pc, pod, nodes[index].Name)
if !status.IsSuccess() {
errCh.SendErrorWithCancel(fmt.Errorf(status.Message()), cancel)

View File

@ -456,6 +456,7 @@ func TestScorePlugin(t *testing.T) {
Enabled: []schedulerconfig.Plugin{
{
Name: scorePluginName,
Weight: 1,
},
},
},