scottk/kubernetes
1
0
Fork 0
mirror of https://github.com/optim-enterprises-bv/kubernetes.git synced 2025-11-04 04:08:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

using wait.PollUntilContextTimeout instead of deprecated wait.Poll/PollWithContext/PollImmediate/PollImmediateWithContext methods for scheduler

Browse Source
This commit is contained in:
SataQiu
2023-08-24 18:35:59 +08:00
parent 02e51b27a9
commit 5524f1651a
5 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/kube-scheduler/app/testing/testserver.go
View File

@@ -132,14 +132,14 @@ func StartTestServer(ctx context.Context, customFlags []string) (result TestServ
if err != nil { if err != nil {
return result, fmt.Errorf("failed to create a client: %v", err) return result, fmt.Errorf("failed to create a client: %v", err)
} }
err = wait.Poll(100*time.Millisecond, 30*time.Second, func() (bool, error) { err = wait.PollUntilContextTimeout(ctx, 100*time.Millisecond, 30*time.Second, false, func(ctx context.Context) (bool, error) {
select { select {
case err := <-errCh: case err := <-errCh:
return false, err return false, err
default: default:
} }
result := client.CoreV1().RESTClient().Get().AbsPath("/healthz").Do(context.TODO()) result := client.CoreV1().RESTClient().Get().AbsPath("/healthz").Do(ctx)
status := 0 status := 0
result.StatusCode(&status) result.StatusCode(&status)
if status == 200 { if status == 200 {

2
test/e2e/scheduling/events.go
View File

@@ -109,7 +109,7 @@ func observeEventAfterAction(ctx context.Context, c clientset.Interface, ns stri
// Wait up 2 minutes polling every second. // Wait up 2 minutes polling every second.
timeout := 2 * time.Minute timeout := 2 * time.Minute
interval := 1 * time.Second interval := 1 * time.Second
err = wait.PollWithContext(ctx, interval, timeout, func(ctx context.Context) (bool, error) { err = wait.PollUntilContextTimeout(ctx, interval, timeout, false, func(ctx context.Context) (bool, error) {
return observedMatchingEvent, nil return observedMatchingEvent, nil
}) })
return err == nil, err return err == nil, err

2
test/e2e/scheduling/limit_range.go
View File

@@ -326,7 +326,7 @@ var _ = SIGDescribe("LimitRange", func() {
framework.ExpectNoError(err, "failed to delete the LimitRange by Collection") framework.ExpectNoError(err, "failed to delete the LimitRange by Collection")
ginkgo.By(fmt.Sprintf("Confirm that the limitRange %q has been deleted", lrName)) ginkgo.By(fmt.Sprintf("Confirm that the limitRange %q has been deleted", lrName))
err = wait.PollImmediateWithContext(ctx, 1*time.Second, 10*time.Second, checkLimitRangeListQuantity(f, patchedLabelSelector, 0)) err = wait.PollUntilContextTimeout(ctx, 1*time.Second, 10*time.Second, true, checkLimitRangeListQuantity(f, patchedLabelSelector, 0))
framework.ExpectNoError(err, "failed to count the required limitRanges") framework.ExpectNoError(err, "failed to count the required limitRanges")
framework.Logf("LimitRange %q has been deleted.", lrName) framework.Logf("LimitRange %q has been deleted.", lrName)

2
test/e2e/scheduling/priorities.go
View File

@@ -364,7 +364,7 @@ func createBalancedPodForNodes(ctx context.Context, f *framework.Framework, cs c
if err != nil { if err != nil {
framework.Logf("Failed to delete memory balanced pods: %v.", err) framework.Logf("Failed to delete memory balanced pods: %v.", err)
} else { } else {
err := wait.PollImmediateWithContext(ctx, 2*time.Second, time.Minute, func(ctx context.Context) (bool, error) { err := wait.PollUntilContextTimeout(ctx, 2*time.Second, time.Minute, true, func(ctx context.Context) (bool, error) {
podList, err := cs.CoreV1().Pods(ns).List(ctx, metav1.ListOptions{ podList, err := cs.CoreV1().Pods(ns).List(ctx, metav1.ListOptions{
LabelSelector: labels.SelectorFromSet(labels.Set(balancePodLabel)).String(), LabelSelector: labels.SelectorFromSet(labels.Set(balancePodLabel)).String(),
}) })

2
test/integration/scheduler_perf/scheduler_perf_test.go
View File

@@ -1312,7 +1312,7 @@ func createPods(ctx context.Context, tb testing.TB, namespace string, cpo *creat
// namespace are scheduled. Times out after 10 minutes because even at the // namespace are scheduled. Times out after 10 minutes because even at the
// lowest observed QPS of ~10 pods/sec, a 5000-node test should complete. // lowest observed QPS of ~10 pods/sec, a 5000-node test should complete.
func waitUntilPodsScheduledInNamespace(ctx context.Context, tb testing.TB, podInformer coreinformers.PodInformer, namespace string, wantCount int) error { func waitUntilPodsScheduledInNamespace(ctx context.Context, tb testing.TB, podInformer coreinformers.PodInformer, namespace string, wantCount int) error {
return wait.PollImmediate(1*time.Second, 10*time.Minute, func() (bool, error) { return wait.PollUntilContextTimeout(ctx, 1*time.Second, 10*time.Minute, true, func(ctx context.Context) (bool, error) {
select { select {
case <-ctx.Done(): case <-ctx.Done():
return true, ctx.Err() return true, ctx.Err()