k8s.io/kubernetes@v1.29.3/test/e2e/framework/perf/perf.go (about)

     1  /*
     2  Copyright 2016 The Kubernetes Authors.
     3  
     4  Licensed under the Apache License, Version 2.0 (the "License");
     5  you may not use this file except in compliance with the License.
     6  You may obtain a copy of the License at
     7  
     8      http://www.apache.org/licenses/LICENSE-2.0
     9  
    10  Unless required by applicable law or agreed to in writing, software
    11  distributed under the License is distributed on an "AS IS" BASIS,
    12  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13  See the License for the specific language governing permissions and
    14  limitations under the License.
    15  */
    16  
    17  package perf
    18  
    19  import (
    20  	"fmt"
    21  
    22  	e2ekubelet "k8s.io/kubernetes/test/e2e/framework/kubelet"
    23  	"k8s.io/kubernetes/test/e2e/perftype"
    24  )
    25  
    26  // CurrentKubeletPerfMetricsVersion is the current kubelet performance metrics
    27  // version. This is used by multiple perf related data structures. We should
    28  // bump up the version each time we make an incompatible change to the metrics.
    29  const CurrentKubeletPerfMetricsVersion = "v2"
    30  
    31  // ResourceUsageToPerfData transforms ResourceUsagePerNode to PerfData. Notice that this function
    32  // only cares about memory usage, because cpu usage information will be extracted from NodesCPUSummary.
    33  func ResourceUsageToPerfData(usagePerNode e2ekubelet.ResourceUsagePerNode) *perftype.PerfData {
    34  	return ResourceUsageToPerfDataWithLabels(usagePerNode, nil)
    35  }
    36  
    37  // CPUUsageToPerfData transforms NodesCPUSummary to PerfData.
    38  func CPUUsageToPerfData(usagePerNode e2ekubelet.NodesCPUSummary) *perftype.PerfData {
    39  	return CPUUsageToPerfDataWithLabels(usagePerNode, nil)
    40  }
    41  
    42  // ResourceUsageToPerfDataWithLabels transforms ResourceUsagePerNode to PerfData with additional labels.
    43  // Notice that this function only cares about memory usage, because cpu usage information will be extracted from NodesCPUSummary.
    44  func ResourceUsageToPerfDataWithLabels(usagePerNode e2ekubelet.ResourceUsagePerNode, labels map[string]string) *perftype.PerfData {
    45  	items := []perftype.DataItem{}
    46  	for node, usages := range usagePerNode {
    47  		for c, usage := range usages {
    48  			item := perftype.DataItem{
    49  				Data: map[string]float64{
    50  					"memory":     float64(usage.MemoryUsageInBytes) / (1024 * 1024),
    51  					"workingset": float64(usage.MemoryWorkingSetInBytes) / (1024 * 1024),
    52  					"rss":        float64(usage.MemoryRSSInBytes) / (1024 * 1024),
    53  				},
    54  				Unit: "MB",
    55  				Labels: map[string]string{
    56  					"node":      node,
    57  					"container": c,
    58  					"datatype":  "resource",
    59  					"resource":  "memory",
    60  				},
    61  			}
    62  			items = append(items, item)
    63  		}
    64  	}
    65  	return &perftype.PerfData{
    66  		Version:   CurrentKubeletPerfMetricsVersion,
    67  		DataItems: items,
    68  		Labels:    labels,
    69  	}
    70  }
    71  
    72  // CPUUsageToPerfDataWithLabels transforms NodesCPUSummary to PerfData with additional labels.
    73  func CPUUsageToPerfDataWithLabels(usagePerNode e2ekubelet.NodesCPUSummary, labels map[string]string) *perftype.PerfData {
    74  	items := []perftype.DataItem{}
    75  	for node, usages := range usagePerNode {
    76  		for c, usage := range usages {
    77  			data := map[string]float64{}
    78  			for perc, value := range usage {
    79  				data[fmt.Sprintf("Perc%02.0f", perc*100)] = value * 1000
    80  			}
    81  
    82  			item := perftype.DataItem{
    83  				Data: data,
    84  				Unit: "mCPU",
    85  				Labels: map[string]string{
    86  					"node":      node,
    87  					"container": c,
    88  					"datatype":  "resource",
    89  					"resource":  "cpu",
    90  				},
    91  			}
    92  			items = append(items, item)
    93  		}
    94  	}
    95  	return &perftype.PerfData{
    96  		Version:   CurrentKubeletPerfMetricsVersion,
    97  		DataItems: items,
    98  		Labels:    labels,
    99  	}
   100  }