diff --git a/Makefile b/Makefile index db8f560..d04f15d 100644 --- a/Makefile +++ b/Makefile @@ -153,7 +153,7 @@ dev-run-debug: dev-scheduler-conf # # E2E test # -export E2E_GOMEGA_DEFAULT_EVENTUALLY_TIMEOUT=90s +export E2E_GOMEGA_DEFAULT_EVENTUALLY_TIMEOUT=180s export E2E_GOMEGA_DEFAULT_CONSISTENTLY_DURATION=2s E2E_PAUSE_IMAGE=k8s.gcr.io/pause:3.2 E2E_KIND_KUBECNOFIG = $(DEV_TOOL_PREFIX)/.kubeconfig diff --git a/pkg/scheduler_plugin/plugin.go b/pkg/scheduler_plugin/plugin.go index c181951..07a4080 100644 --- a/pkg/scheduler_plugin/plugin.go +++ b/pkg/scheduler_plugin/plugin.go @@ -52,6 +52,7 @@ type KubeThrottler struct { } var _ framework.PreFilterPlugin = &KubeThrottler{} +var _ framework.EnqueueExtensions = &KubeThrottler{} var _ framework.ReservePlugin = &KubeThrottler{} func (p *KubeThrottler) Name() string { @@ -261,6 +262,24 @@ func (p *KubeThrottler) PreFilterExtensions() framework.PreFilterExtensions { return nil } +func (p *KubeThrottler) EventsToRegister() []framework.ClusterEvent { + throttlesGVK := framework.GVK(fmt.Sprintf("throttles.%v.%v", schedulev1alpha1.SchemeGroupVersion.Version, schedulev1alpha1.SchemeGroupVersion.Group)) + clusterthrottlesGVK := framework.GVK(fmt.Sprintf("clusterthrottles.%v.%v", schedulev1alpha1.SchemeGroupVersion.Version, schedulev1alpha1.SchemeGroupVersion.Group)) + return []framework.ClusterEvent{{ + Resource: framework.Node, + ActionType: framework.All, + }, { + Resource: framework.Pod, + ActionType: framework.All, + }, { + Resource: throttlesGVK, + ActionType: framework.All, + }, { + Resource: clusterthrottlesGVK, + ActionType: framework.All, + }} +} + func throttleNames(objs []schedulev1alpha1.Throttle) []string { names := make([]string, len(objs)) for i, obj := range objs {