Merge pull request #129 from CoderTH/fix-metrics-bug

fix: loss of metrics after vdeivce restart
4paradigm · Jan 19, 2024 · 2d4e2d1 · 2d4e2d1
2 parents e5a7647 + 8a84972
commit 2d4e2d1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/scheduler/scheduler.go b/pkg/scheduler/scheduler.go
@@ -141,7 +141,9 @@ func (s *Scheduler) RegisterFromNodeAnnotatons() error {
 			klog.Errorln("nodes list failed", err.Error())
 			return err
 		}
+		nodeNames := []string{}
 		for _, val := range nodes {
+			nodeNames = append(nodeNames, val.Name)
 			for devhandsk, devreg := range device.KnownDevice {
 				_, ok := val.Annotations[devreg]
 				if !ok {
@@ -229,6 +231,11 @@ func (s *Scheduler) RegisterFromNodeAnnotatons() error {
 				}
 			}
 		}
+		_, _, err = s.getNodesUsage(&nodeNames, nil)
+		if err != nil {
+			klog.Errorln("get node usage failed", err.Error())
+			return err
+		}
 		time.Sleep(time.Second * 15)
 	}
 }