fix: scheduler perf test with pod informer

This commit is contained in:
draveness
2019-10-15 12:57:49 +08:00
parent 63bd1d7a5c
commit e5a23f8dda
6 changed files with 17 additions and 20 deletions

View File

@@ -36,7 +36,6 @@ go_test(
tags = ["integration"],
deps = [
"//pkg/features:go_default_library",
"//pkg/scheduler:go_default_library",
"//pkg/volume/util:go_default_library",
"//staging/src/k8s.io/api/core/v1:go_default_library",
"//staging/src/k8s.io/api/storage/v1beta1:go_default_library",

View File

@@ -33,7 +33,6 @@ import (
csilibplugins "k8s.io/csi-translation-lib/plugins"
"k8s.io/klog"
"k8s.io/kubernetes/pkg/features"
"k8s.io/kubernetes/pkg/scheduler"
"k8s.io/kubernetes/pkg/volume/util"
"k8s.io/kubernetes/test/integration/framework"
testutils "k8s.io/kubernetes/test/utils"
@@ -361,10 +360,9 @@ func benchmarkScheduling(numNodes, numExistingPods, minPods int,
if b.N < minPods {
b.N = minPods
}
finalFunc, clientset := mustSetupScheduler()
finalFunc, podInformer, clientset := mustSetupScheduler()
defer finalFunc()
podInformer := scheduler.NewPodInformer(clientset, 0)
nodePreparer := framework.NewIntegrationTestNodePreparer(
clientset,
[]testutils.CountToStrategy{{Count: numNodes, Strategy: nodeStrategy}},
@@ -388,6 +386,7 @@ func benchmarkScheduling(numNodes, numExistingPods, minPods int,
if len(scheduled) >= numExistingPods {
break
}
klog.Infof("got %d existing pods, required: %d", len(scheduled), numExistingPods)
time.Sleep(1 * time.Second)
}

View File

@@ -31,7 +31,6 @@ import (
coreinformers "k8s.io/client-go/informers/core/v1"
clientset "k8s.io/client-go/kubernetes"
"k8s.io/client-go/tools/cache"
"k8s.io/kubernetes/pkg/scheduler"
testutils "k8s.io/kubernetes/test/utils"
"k8s.io/klog"
@@ -116,12 +115,13 @@ type testConfig struct {
// getBaseConfig returns baseConfig after initializing number of nodes and pods.
func getBaseConfig(nodes int, pods int) *testConfig {
destroyFunc, clientset := mustSetupScheduler()
destroyFunc, podInformer, clientset := mustSetupScheduler()
return &testConfig{
clientset: clientset,
destroyFunc: destroyFunc,
numNodes: nodes,
numPods: pods,
podInformer: podInformer,
}
}
@@ -138,11 +138,10 @@ func schedulePods(config *testConfig) int32 {
minQPS := int32(math.MaxInt32)
start := time.Now()
podInformer := scheduler.NewPodInformer(config.clientset, 0)
// Bake in time for the first pod scheduling event.
for {
time.Sleep(50 * time.Millisecond)
scheduled, err := getScheduledPods(podInformer)
scheduled, err := getScheduledPods(config.podInformer)
if err != nil {
klog.Fatalf("%v", err)
}

View File

@@ -33,7 +33,7 @@ import (
// remove resources after finished.
// Notes on rate limiter:
// - client rate limit is set to 5000.
func mustSetupScheduler() (util.ShutdownFunc, clientset.Interface) {
func mustSetupScheduler() (util.ShutdownFunc, coreinformers.PodInformer, clientset.Interface) {
apiURL, apiShutdown := util.StartApiserver()
clientSet := clientset.NewForConfigOrDie(&restclient.Config{
Host: apiURL,
@@ -41,14 +41,14 @@ func mustSetupScheduler() (util.ShutdownFunc, clientset.Interface) {
QPS: 5000.0,
Burst: 5000,
})
_, schedulerShutdown := util.StartScheduler(clientSet)
_, podInformer, schedulerShutdown := util.StartScheduler(clientSet)
shutdownFunc := func() {
schedulerShutdown()
apiShutdown()
}
return shutdownFunc, clientSet
return shutdownFunc, podInformer, clientSet
}
func getScheduledPods(podInformer coreinformers.PodInformer) ([]*v1.Pod, error) {
@@ -56,6 +56,7 @@ func getScheduledPods(podInformer coreinformers.PodInformer) ([]*v1.Pod, error)
if err != nil {
return nil, err
}
scheduled := make([]*v1.Pod, 0, len(pods))
for i := range pods {
pod := pods[i]