mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-10-31 13:50:01 +00:00
9d8d6ad16c
Where possible, switch the scheduler to use generated listers and informers. There are still some places where it probably makes more sense to use one-off reflectors/informers (listing/watching just a single node, listing/watching scheduled & unscheduled pods using a field selector).
108 lines
3.6 KiB
Go
108 lines
3.6 KiB
Go
/*
|
|
Copyright 2015 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package benchmark
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/kubernetes/test/integration/framework"
|
|
testutils "k8s.io/kubernetes/test/utils"
|
|
|
|
"github.com/golang/glog"
|
|
)
|
|
|
|
// BenchmarkScheduling100Nodes0Pods benchmarks the scheduling rate
|
|
// when the cluster has 100 nodes and 0 scheduled pods
|
|
func BenchmarkScheduling100Nodes0Pods(b *testing.B) {
|
|
benchmarkScheduling(100, 0, b)
|
|
}
|
|
|
|
// BenchmarkScheduling100Nodes1000Pods benchmarks the scheduling rate
|
|
// when the cluster has 100 nodes and 1000 scheduled pods
|
|
func BenchmarkScheduling100Nodes1000Pods(b *testing.B) {
|
|
benchmarkScheduling(100, 1000, b)
|
|
}
|
|
|
|
// BenchmarkScheduling1000Nodes0Pods benchmarks the scheduling rate
|
|
// when the cluster has 1000 nodes and 0 scheduled pods
|
|
func BenchmarkScheduling1000Nodes0Pods(b *testing.B) {
|
|
benchmarkScheduling(1000, 0, b)
|
|
}
|
|
|
|
// BenchmarkScheduling1000Nodes1000Pods benchmarks the scheduling rate
|
|
// when the cluster has 1000 nodes and 1000 scheduled pods
|
|
func BenchmarkScheduling1000Nodes1000Pods(b *testing.B) {
|
|
benchmarkScheduling(1000, 1000, b)
|
|
}
|
|
|
|
// benchmarkScheduling benchmarks scheduling rate with specific number of nodes
|
|
// and specific number of pods already scheduled. Since an operation takes relatively
|
|
// long time, b.N should be small: 10 - 100.
|
|
func benchmarkScheduling(numNodes, numScheduledPods int, b *testing.B) {
|
|
schedulerConfigFactory, finalFunc := mustSetupScheduler()
|
|
defer finalFunc()
|
|
c := schedulerConfigFactory.GetClient()
|
|
|
|
nodePreparer := framework.NewIntegrationTestNodePreparer(
|
|
c,
|
|
[]testutils.CountToStrategy{{Count: numNodes, Strategy: &testutils.TrivialNodePrepareStrategy{}}},
|
|
"scheduler-perf-",
|
|
)
|
|
if err := nodePreparer.PrepareNodes(); err != nil {
|
|
glog.Fatalf("%v", err)
|
|
}
|
|
defer nodePreparer.CleanupNodes()
|
|
|
|
config := testutils.NewTestPodCreatorConfig()
|
|
config.AddStrategy("sched-test", numScheduledPods, testutils.NewSimpleWithControllerCreatePodStrategy("rc1"))
|
|
podCreator := testutils.NewTestPodCreator(c, config)
|
|
podCreator.CreatePods()
|
|
|
|
for {
|
|
scheduled, err := schedulerConfigFactory.GetScheduledPodLister().List(labels.Everything())
|
|
if err != nil {
|
|
glog.Fatalf("%v", err)
|
|
}
|
|
if len(scheduled) >= numScheduledPods {
|
|
break
|
|
}
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
// start benchmark
|
|
b.ResetTimer()
|
|
config = testutils.NewTestPodCreatorConfig()
|
|
config.AddStrategy("sched-test", b.N, testutils.NewSimpleWithControllerCreatePodStrategy("rc2"))
|
|
podCreator = testutils.NewTestPodCreator(c, config)
|
|
podCreator.CreatePods()
|
|
for {
|
|
// This can potentially affect performance of scheduler, since List() is done under mutex.
|
|
// TODO: Setup watch on apiserver and wait until all pods scheduled.
|
|
scheduled, err := schedulerConfigFactory.GetScheduledPodLister().List(labels.Everything())
|
|
if err != nil {
|
|
glog.Fatalf("%v", err)
|
|
}
|
|
if len(scheduled) >= numScheduledPods+b.N {
|
|
break
|
|
}
|
|
// Note: This might introduce slight deviation in accuracy of benchmark results.
|
|
// Since the total amount of time is relatively large, it might not be a concern.
|
|
time.Sleep(100 * time.Millisecond)
|
|
}
|
|
}
|