fix(node): 发现节点过滤问题

main
youys 1 week ago
parent a25f046140
commit 37f0c4ae58

@ -106,7 +106,7 @@ func (r *nodeRepo) updateLocalNodes() {
var all []*biz.Node var all []*biz.Node
allNodes, _ := r.nodeLister.List(labels.Set{"gpu": "on"}.AsSelector()) allNodes, _ := r.nodeLister.List(labels.Everything())
for _, node := range allNodes { for _, node := range allNodes {
bizNode := r.fetchNodeInfo(node) bizNode := r.fetchNodeInfo(node)
gpuNode := n[k8stypes.UID(bizNode.Uid)] gpuNode := n[k8stypes.UID(bizNode.Uid)]

@ -207,6 +207,9 @@ func (s *ResourcePoolService) getK8sNodes(ctx context.Context) map[string]*biz.N
nodes, _ := s.uc.ListAllNodesV2(ctx) nodes, _ := s.uc.ListAllNodesV2(ctx)
m := make(map[string]*biz.Node) m := make(map[string]*biz.Node)
for _, node := range nodes { for _, node := range nodes {
if node.Lables["gpu"] != "on" {
continue
}
node.GPUCount = int64(len(node.Devices)) node.GPUCount = int64(len(node.Devices))
m[node.Name] = node m[node.Name] = node
} }

Loading…
Cancel
Save