框架over

secretflow · Dec 1, 2024 · 835837a · 835837a
1 parent 1a01a86
commit 835837a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pkg/ktexporter/fetchmetrics/fetchmetrics.go b/pkg/ktexporter/fetchmetrics/fetchmetrics.go
@@ -95,7 +95,7 @@ type ContainerStats struct {
 }
 
 // GetContainerStats fetches the container stats using crictl stats command
-func GetContainerStats() (map[string]ContainerStats, error) {
+func GetContainerStats(podLister listers.PodLister) (map[string]ContainerStats, error) {
 	// Execute the crictl stats command
 	cmd := exec.Command("crictl", "stats")
 	var out bytes.Buffer

diff --git a/pkg/ktexporter/ktmetrics/ktmetrics.go b/pkg/ktexporter/ktmetrics/ktmetrics.go
@@ -150,7 +150,11 @@ func GetStatisticFromKt() ([]map[string]string, error) {
 		ktMetrics[parse.MetricRecvBw] = fmt.Sprintf("%.2f", recvBW)
 		ktMetrics[parse.MetricXmitBw] = fmt.Sprintf("%.2f", xmitBW)
 		ktMetrics["taskid"] = kusciaTaskID
-		containerStats, err := fetchmetrics.GetContainerStats()
+
+		podLister := informers.NewSharedInformerFactoryWithOptions(kubernetes.NewForConfigOrDie(nil), 0).Core().V1().Pods().Lister()
+
+		containerStats, err := fetchmetrics.GetContainerStats(podLister)
+
 		if err != nil {
 			nlog.Warn("Fail to get the stats of containers")
 			continue