mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-28 05:57:25 +00:00
Add possibility to delete pods at specified frequency in scheduler_perf tests
This commit is contained in:
parent
5df8e15a84
commit
c09440c691
@ -19,6 +19,7 @@ package benchmark
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"flag"
|
||||
"fmt"
|
||||
"io"
|
||||
@ -37,6 +38,7 @@ import (
|
||||
"k8s.io/apimachinery/pkg/api/meta"
|
||||
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
||||
"k8s.io/apimachinery/pkg/labels"
|
||||
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||
"k8s.io/apimachinery/pkg/util/wait"
|
||||
utilfeature "k8s.io/apiserver/pkg/util/feature"
|
||||
@ -464,6 +466,9 @@ type createPodsOp struct {
|
||||
// Optional
|
||||
PersistentVolumeTemplatePath *string
|
||||
PersistentVolumeClaimTemplatePath *string
|
||||
// Number of pods to be deleted per second after they were scheduled. If set to 0, pods are not deleted.
|
||||
// Optional
|
||||
DeletePodsPerSecond int
|
||||
}
|
||||
|
||||
func (cpo *createPodsOp) isValid(allowParameterization bool) error {
|
||||
@ -479,6 +484,9 @@ func (cpo *createPodsOp) isValid(allowParameterization bool) error {
|
||||
// use-cases right now.
|
||||
return fmt.Errorf("collectMetrics and skipWaitToCompletion cannot be true at the same time")
|
||||
}
|
||||
if cpo.DeletePodsPerSecond < 0 {
|
||||
return fmt.Errorf("invalid DeletePodsPerSecond=%d; should be non-negative", cpo.DeletePodsPerSecond)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
@ -1030,6 +1038,34 @@ func runWorkload(tCtx ktesting.TContext, tc *testCase, w *workload, informerFact
|
||||
mu.Unlock()
|
||||
}
|
||||
|
||||
if concreteOp.DeletePodsPerSecond > 0 {
|
||||
pods, err := podInformer.Lister().Pods(namespace).List(labels.Everything())
|
||||
if err != nil {
|
||||
tCtx.Fatalf("op %d: error in listing scheduled pods in the namespace: %v", opIndex, err)
|
||||
}
|
||||
|
||||
ticker := time.NewTicker(time.Second / time.Duration(concreteOp.DeletePodsPerSecond))
|
||||
defer ticker.Stop()
|
||||
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
for i := 0; i < len(pods); i++ {
|
||||
select {
|
||||
case <-ticker.C:
|
||||
if err := tCtx.Client().CoreV1().Pods(namespace).Delete(tCtx, pods[i].Name, metav1.DeleteOptions{}); err != nil {
|
||||
if errors.Is(err, context.Canceled) {
|
||||
return
|
||||
}
|
||||
tCtx.Errorf("op %d: unable to delete pod %v: %v", opIndex, pods[i].Name, err)
|
||||
}
|
||||
case <-tCtx.Done():
|
||||
return
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
if !concreteOp.SkipWaitToCompletion {
|
||||
// SkipWaitToCompletion=false indicates this step has waited for the Pods to be scheduled.
|
||||
// So we reset the metrics in global registry; otherwise metrics gathered in this step
|
||||
|
Loading…
Reference in New Issue
Block a user