mirror of
https://github.com/k3s-io/kubernetes.git
synced 2025-07-27 21:47:07 +00:00
Refactor code and annotations for readability
Signed-off-by: kerthcet <kerthcet@gmail.com>
This commit is contained in:
parent
21e8a69a22
commit
cfc53ee524
@ -67,10 +67,10 @@ const (
|
|||||||
const (
|
const (
|
||||||
// Two modes supported in "churn" operator.
|
// Two modes supported in "churn" operator.
|
||||||
|
|
||||||
// Recreate creates a number of API objects and then delete them, and repeat the iteration.
|
|
||||||
Recreate = "recreate"
|
|
||||||
// Create continuously create API objects without deleting them.
|
// Create continuously create API objects without deleting them.
|
||||||
Create = "create"
|
Create = "create"
|
||||||
|
// Recreate creates a number of API objects and then delete them, and repeat the iteration.
|
||||||
|
Recreate = "recreate"
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@ -91,16 +91,18 @@ var (
|
|||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
// testCase defines a set of test cases that intend to test the performance of
|
// testCase defines a set of test cases that intends to test the performance of
|
||||||
// similar workloads of varying sizes with shared overall settings such as
|
// similar workloads of varying sizes with shared overall settings such as
|
||||||
// feature gates and metrics collected.
|
// feature gates and metrics collected.
|
||||||
type testCase struct {
|
type testCase struct {
|
||||||
// Name of the testCase.
|
// Name of the testCase.
|
||||||
Name string
|
Name string
|
||||||
// Feature gates to set before running the test. Optional.
|
// Feature gates to set before running the test.
|
||||||
|
// Optional
|
||||||
FeatureGates map[featuregate.Feature]bool
|
FeatureGates map[featuregate.Feature]bool
|
||||||
// List of metrics to collect. Optional, defaults to
|
// List of metrics to collect. Defaults to
|
||||||
// defaultMetricsCollectorConfig if unspecified.
|
// defaultMetricsCollectorConfig if unspecified.
|
||||||
|
// Optional
|
||||||
MetricsCollectorConfig *metricsCollectorConfig
|
MetricsCollectorConfig *metricsCollectorConfig
|
||||||
// Template for sequence of ops that each workload must follow. Each op will
|
// Template for sequence of ops that each workload must follow. Each op will
|
||||||
// be executed serially one after another. Each element of the list must be
|
// be executed serially one after another. Each element of the list must be
|
||||||
@ -110,8 +112,9 @@ type testCase struct {
|
|||||||
Workloads []*workload
|
Workloads []*workload
|
||||||
// SchedulerConfigFile is the path of scheduler configuration
|
// SchedulerConfigFile is the path of scheduler configuration
|
||||||
SchedulerConfigFile string
|
SchedulerConfigFile string
|
||||||
// Default path to spec file describing the pods to create. Optional.
|
// Default path to spec file describing the pods to create.
|
||||||
// This path can be overridden in createPodsOp by setting PodTemplatePath .
|
// This path can be overridden in createPodsOp by setting PodTemplatePath .
|
||||||
|
// Optional
|
||||||
DefaultPodTemplatePath *string
|
DefaultPodTemplatePath *string
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -265,10 +268,12 @@ type createNodesOp struct {
|
|||||||
Count int
|
Count int
|
||||||
// Template parameter for Count.
|
// Template parameter for Count.
|
||||||
CountParam string
|
CountParam string
|
||||||
// Path to spec file describing the nodes to create. Optional.
|
// Path to spec file describing the nodes to create.
|
||||||
|
// Optional
|
||||||
NodeTemplatePath *string
|
NodeTemplatePath *string
|
||||||
// At most one of the following strategies can be defined. Optional, defaults
|
// At most one of the following strategies can be defined. Defaults
|
||||||
// to TrivialNodePrepareStrategy if unspecified.
|
// to TrivialNodePrepareStrategy if unspecified.
|
||||||
|
// Optional
|
||||||
NodeAllocatableStrategy *testutils.NodeAllocatableStrategy
|
NodeAllocatableStrategy *testutils.NodeAllocatableStrategy
|
||||||
LabelNodePrepareStrategy *testutils.LabelNodePrepareStrategy
|
LabelNodePrepareStrategy *testutils.LabelNodePrepareStrategy
|
||||||
UniqueNodeLabelStrategy *testutils.UniqueNodeLabelStrategy
|
UniqueNodeLabelStrategy *testutils.UniqueNodeLabelStrategy
|
||||||
@ -312,7 +317,8 @@ type createNamespacesOp struct {
|
|||||||
Count int
|
Count int
|
||||||
// Template parameter for Count. Takes precedence over Count if both set.
|
// Template parameter for Count. Takes precedence over Count if both set.
|
||||||
CountParam string
|
CountParam string
|
||||||
// Path to spec file describing the Namespaces to create. Optional.
|
// Path to spec file describing the Namespaces to create.
|
||||||
|
// Optional
|
||||||
NamespaceTemplatePath *string
|
NamespaceTemplatePath *string
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -357,16 +363,20 @@ type createPodsOp struct {
|
|||||||
// Optional. Both CollectMetrics and SkipWaitToCompletion cannot be true at
|
// Optional. Both CollectMetrics and SkipWaitToCompletion cannot be true at
|
||||||
// the same time for a particular createPodsOp.
|
// the same time for a particular createPodsOp.
|
||||||
CollectMetrics bool
|
CollectMetrics bool
|
||||||
// Namespace the pods should be created in. Optional, defaults to a unique
|
// Namespace the pods should be created in. Defaults to a unique
|
||||||
// namespace of the format "namespace-<number>".
|
// namespace of the format "namespace-<number>".
|
||||||
|
// Optional
|
||||||
Namespace *string
|
Namespace *string
|
||||||
// Path to spec file describing the pods to schedule. Optional.
|
// Path to spec file describing the pods to schedule.
|
||||||
// If nil, DefaultPodTemplatePath will be used.
|
// If nil, DefaultPodTemplatePath will be used.
|
||||||
|
// Optional
|
||||||
PodTemplatePath *string
|
PodTemplatePath *string
|
||||||
// Whether or not to wait for all pods in this op to get scheduled. Optional,
|
// Whether or not to wait for all pods in this op to get scheduled.
|
||||||
// defaults to false.
|
// Defaults to false if not specified.
|
||||||
|
// Optional
|
||||||
SkipWaitToCompletion bool
|
SkipWaitToCompletion bool
|
||||||
// Persistent volume settings for the pods to be scheduled. Optional.
|
// Persistent volume settings for the pods to be scheduled.
|
||||||
|
// Optional
|
||||||
PersistentVolumeTemplatePath *string
|
PersistentVolumeTemplatePath *string
|
||||||
PersistentVolumeClaimTemplatePath *string
|
PersistentVolumeClaimTemplatePath *string
|
||||||
}
|
}
|
||||||
@ -404,7 +414,7 @@ func (cpo createPodsOp) patchParams(w *workload) (realOp, error) {
|
|||||||
return &cpo, (&cpo).isValid(false)
|
return &cpo, (&cpo).isValid(false)
|
||||||
}
|
}
|
||||||
|
|
||||||
// createPodSetsOp defines an op where a set of createPodsOp is created each in a unique namespace.
|
// createPodSetsOp defines an op where a set of createPodsOps is created in each unique namespace.
|
||||||
type createPodSetsOp struct {
|
type createPodSetsOp struct {
|
||||||
// Must be "createPodSets".
|
// Must be "createPodSets".
|
||||||
Opcode operationCode
|
Opcode operationCode
|
||||||
@ -461,8 +471,9 @@ type churnOp struct {
|
|||||||
Number int
|
Number int
|
||||||
// Intervals of churning. Defaults to 500 millisecond.
|
// Intervals of churning. Defaults to 500 millisecond.
|
||||||
IntervalMilliseconds int64
|
IntervalMilliseconds int64
|
||||||
// Namespace the churning objects should be created in. Optional, defaults to a unique
|
// Namespace the churning objects should be created in. Defaults to a unique
|
||||||
// namespace of the format "namespace-<number>".
|
// namespace of the format "namespace-<number>".
|
||||||
|
// Optional
|
||||||
Namespace *string
|
Namespace *string
|
||||||
// Path of API spec files.
|
// Path of API spec files.
|
||||||
TemplatePaths []string
|
TemplatePaths []string
|
||||||
@ -822,7 +833,26 @@ func runWorkload(b *testing.B, tc *testCase, w *workload) []DataItem {
|
|||||||
ticker := time.NewTicker(time.Duration(interval) * time.Millisecond)
|
ticker := time.NewTicker(time.Duration(interval) * time.Millisecond)
|
||||||
defer ticker.Stop()
|
defer ticker.Stop()
|
||||||
|
|
||||||
if concreteOp.Mode == Recreate {
|
switch concreteOp.Mode {
|
||||||
|
case Create:
|
||||||
|
go func() {
|
||||||
|
count, threshold := 0, concreteOp.Number
|
||||||
|
if threshold == 0 {
|
||||||
|
threshold = math.MaxInt32
|
||||||
|
}
|
||||||
|
for count < threshold {
|
||||||
|
select {
|
||||||
|
case <-ticker.C:
|
||||||
|
for i := range churnFns {
|
||||||
|
churnFns[i]("")
|
||||||
|
}
|
||||||
|
count++
|
||||||
|
case <-ctx.Done():
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
case Recreate:
|
||||||
go func() {
|
go func() {
|
||||||
retVals := make([][]string, len(churnFns))
|
retVals := make([][]string, len(churnFns))
|
||||||
// For each churn function, instantiate a slice of strings with length "concreteOp.Number".
|
// For each churn function, instantiate a slice of strings with length "concreteOp.Number".
|
||||||
@ -843,24 +873,6 @@ func runWorkload(b *testing.B, tc *testCase, w *workload) []DataItem {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}()
|
}()
|
||||||
} else if concreteOp.Mode == Create {
|
|
||||||
go func() {
|
|
||||||
count, threshold := 0, concreteOp.Number
|
|
||||||
if threshold == 0 {
|
|
||||||
threshold = math.MaxInt32
|
|
||||||
}
|
|
||||||
for count < threshold {
|
|
||||||
select {
|
|
||||||
case <-ticker.C:
|
|
||||||
for i := range churnFns {
|
|
||||||
churnFns[i]("")
|
|
||||||
}
|
|
||||||
count++
|
|
||||||
case <-ctx.Done():
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
}
|
}
|
||||||
|
|
||||||
case *barrierOp:
|
case *barrierOp:
|
||||||
@ -1039,7 +1051,7 @@ func getUnstructuredFromFile(path string) (*unstructured.Unstructured, *schema.G
|
|||||||
func getTestCases(path string) ([]*testCase, error) {
|
func getTestCases(path string) ([]*testCase, error) {
|
||||||
testCases := make([]*testCase, 0)
|
testCases := make([]*testCase, 0)
|
||||||
if err := getSpecFromFile(&path, &testCases); err != nil {
|
if err := getSpecFromFile(&path, &testCases); err != nil {
|
||||||
return nil, fmt.Errorf("parsing test cases: %w", err)
|
return nil, fmt.Errorf("parsing test cases error: %w", err)
|
||||||
}
|
}
|
||||||
return testCases, nil
|
return testCases, nil
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user