Merge pull request #45 from everpeace/enqueue-extention

register ClusterEvents for waking up pods in unschedulableQ quickly.

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

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 {