Added plugins enabled at each extension point

This commit is contained in:
hub-Prateek 2023-11-01 23:03:13 +05:30
parent 32db5774a3
commit 7b60e7e2a3
3 changed files with 28 additions and 8 deletions

View File

@ -90,12 +90,6 @@ func LogOrWriteConfig(logger klog.Logger, fileName string, cfg *config.KubeSched
return err return err
} }
defaultPluginString := "Plugins Enabled By Default:"
for _, v := range configv1.GetDefaultPlugins.MultiPoint.Enabled {
name := v.Name
defaultPluginString += "\n" + "\t" + name
}
buf.WriteString(defaultPluginString)
if loggerV.Enabled() { if loggerV.Enabled() {
loggerV.Info("Using component config", "config", buf.String()) loggerV.Info("Using component config", "config", buf.String())
} }

View File

@ -26,8 +26,6 @@ import (
"k8s.io/utils/ptr" "k8s.io/utils/ptr"
) )
var GetDefaultPlugins = getDefaultPlugins()
// getDefaultPlugins returns the default set of plugins. // getDefaultPlugins returns the default set of plugins.
func getDefaultPlugins() *v1.Plugins { func getDefaultPlugins() *v1.Plugins {
plugins := &v1.Plugins{ plugins := &v1.Plugins{

View File

@ -355,6 +355,34 @@ func NewFramework(ctx context.Context, r Registry, profile *config.KubeScheduler
options.captureProfile(outputProfile) options.captureProfile(outputProfile)
} }
for _, e := range f.getExtensionPoints(profile.Plugins) {
plugins := reflect.ValueOf(e.slicePtr).Elem()
pluginType := plugins.Type().Elem()
loggerV := logger.V(2)
enabledSet := newOrderedSet()
for _, plugin := range e.plugins.Enabled {
enabledSet.insert(plugin.Name)
}
disabledSet := sets.New[string]()
for _, disabledPlugin := range e.plugins.Disabled {
disabledSet.Insert(disabledPlugin.Name)
}
if disabledSet.Has("*") {
continue
}
for _, ep := range profile.Plugins.MultiPoint.Enabled {
if !enabledSet.has(ep.Name) {
enabledSet.insert(ep.Name)
}
}
loggerV.Info("Plugins enabled for", "extension", pluginType, "plugins", enabledSet.list)
}
f.setInstrumentedPlugins() f.setInstrumentedPlugins()
return f, nil return f, nil
} }