synchronization

gpushare-scheduler-extender/pkg/scheduler/inspect.go

@@ -0,0 +1,69 @@
+package scheduler
+
+import (
+	"github.com/AliyunContainerService/gpushare-scheduler-extender/pkg/cache"
+	"github.com/AliyunContainerService/gpushare-scheduler-extender/pkg/utils"
+)
+
+func (in Inspect) Handler(name string) *Result {
+	nodes := []*Node{}
+	errMsg := ""
+	if len(name) == 0 {
+		nodeInfos := in.cache.GetNodeinfos()
+		for _, info := range nodeInfos {
+			nodes = append(nodes, buildNode(info))
+		}
+
+	} else {
+		node, err := in.cache.GetNodeInfo(name)
+		if err != nil {
+			errMsg = err.Error()
+		}
+		// nodeInfos = append(nodeInfos, node)
+		nodes = append(nodes, buildNode(node))
+	}
+
+	return &Result{
+		Nodes: nodes,
+		Error: errMsg,
+	}
+}
+
+func buildNode(info *cache.NodeInfo) *Node {
+
+	devInfos := info.GetDevs()
+	devs := []*Device{}
+	var usedGPU uint
+
+	for i, devInfo := range devInfos {
+		dev := &Device{
+			ID:       i,
+			TotalGPU: devInfo.GetTotalGPUMemory(),
+			UsedGPU:  devInfo.GetUsedGPUMemory(),
+		}
+
+		podInfos := devInfo.GetPods()
+		pods := []*Pod{}
+		for _, podInfo := range podInfos {
+			if utils.AssignedNonTerminatedPod(podInfo) {
+				pod := &Pod{
+					Namespace: podInfo.Namespace,
+					Name:      podInfo.Name,
+					UsedGPU:   utils.GetGPUMemoryFromPodResource(podInfo),
+				}
+				pods = append(pods, pod)
+			}
+		}
+		dev.Pods = pods
+		devs = append(devs, dev)
+		usedGPU += devInfo.GetUsedGPUMemory()
+	}
+
+	return &Node{
+		Name:     info.GetName(),
+		TotalGPU: uint(info.GetTotalGPUMemory()),
+		UsedGPU:  usedGPU,
+		Devices:  devs,
+	}
+
+}