test

kubernetes-sigs · Nov 27, 2024 · 10d0cf7 · 10d0cf7
1 parent 6c30299
commit 10d0cf7
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 0 deletions.
diff --git a/pkg/kwok/controllers/node_controller.go b/pkg/kwok/controllers/node_controller.go
@@ -327,6 +327,9 @@ func (c *NodeController) playStageWorker(ctx context.Context) {
 		if !ok {
 			return
 		}
+		if c.delayQueue.Len() > 10 {
+			go c.playStageWorker(ctx)
+		}
 		c.delayQueueMapping.Delete(node.Key)
 		needRetry, err := c.playStage(ctx, node.Resource, node.Stage)
 		if err != nil {

diff --git a/pkg/kwok/controllers/node_lease_controller.go b/pkg/kwok/controllers/node_lease_controller.go
@@ -112,6 +112,9 @@ func (c *NodeLeaseController) syncWorker(ctx context.Context) {
 		if !ok {
 			return
 		}
+		if c.delayQueue.Len() > 10 {
+			go c.syncWorker(ctx)
+		}
 		first, ok := c.holdLeaseSet.Load(nodeName)
 		if !ok {
 			continue

diff --git a/pkg/kwok/controllers/pod_controller.go b/pkg/kwok/controllers/pod_controller.go
@@ -262,6 +262,9 @@ func (c *PodController) playStageWorker(ctx context.Context) {
 		if !ok {
 			return
 		}
+		if c.delayQueue.Len() > 10 {
+			go c.playStageWorker(ctx)
+		}
 		c.delayQueueMapping.Delete(pod.Key)
 		needRetry, err := c.playStage(ctx, pod.Resource, pod.Stage)
 		if err != nil {

diff --git a/pkg/kwok/controllers/stage_controller.go b/pkg/kwok/controllers/stage_controller.go
@@ -240,6 +240,9 @@ func (c *StageController) playStageWorker(ctx context.Context) {
 		if !ok {
 			return
 		}
+		if c.delayQueue.Len() > 10 {
+			go c.playStageWorker(ctx)
+		}
 		c.delayQueueMapping.Delete(resource.Key)
 		needRetry, err := c.playStage(ctx, resource.Resource, resource.Stage)
 		if err != nil {