From ca8a0f5f1be9ca615297a7fe682d0c5aa3cfb4bc Mon Sep 17 00:00:00 2001 From: YamasouA Date: Thu, 13 Feb 2025 23:46:43 +0900 Subject: [PATCH] separete sleep func --- test/integration/scheduler_perf/scheduler_perf.go | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/test/integration/scheduler_perf/scheduler_perf.go b/test/integration/scheduler_perf/scheduler_perf.go index 318d181c0a5..312184fe7be 100644 --- a/test/integration/scheduler_perf/scheduler_perf.go +++ b/test/integration/scheduler_perf/scheduler_perf.go @@ -1553,10 +1553,7 @@ func runWorkload(tCtx ktesting.TContext, tc *testCase, w *workload, informerFact case *barrierOp: executor.runBarrierOp(opIndex, concreteOp) case *sleepOp: - select { - case <-tCtx.Done(): - case <-time.After(concreteOp.Duration.Duration): - } + executor.runSleepOp(concreteOp) case *startCollectingMetricsOp: executor.runStartCollectingMetricsOp(opIndex, concreteOp) defer (*executor.collectorCtx).Cancel("cleaning up") @@ -1640,6 +1637,13 @@ func (e *WorkloadExecutor) runBarrierOp(opIndex int, op *barrierOp) { } } +func (e *WorkloadExecutor) runSleepOp(op *sleepOp) { + select { + case <-(*e.tCtx).Done(): + case <-time.After(op.Duration.Duration): + } +} + func (e *WorkloadExecutor) runStopCollectingMetrics(opIndex int) { items := stopCollectingMetrics(*e.tCtx, *e.collectorCtx, e.collectorWG, e.workload.Threshold, *e.workload.ThresholdMetricSelector, opIndex, e.collectors) e.dataItems = append(e.dataItems, items...)