update

src/resource_pool.go

@@ -2,12 +2,69 @@ package main
 
 import (
 	"sync"
-	)
+	"time"
+	"strconv"
+	"fmt"
+)
 
 type ResourcePool struct {
-	mu sync.Mutex
-
+	mu    sync.Mutex
 	nodes map[string]NodeStatus
+
+	history []map[string]string
+}
+
+func (pool *ResourcePool) start() {
+	go func() {
+		for {
+			summary := map[string]string{}
+
+			UtilCPU := 0.0
+			TotalCPU := 0
+			TotalMem := 0
+			AvailableMem := 0
+
+			TotalGPU := 0
+			UtilGPU := 0
+			TotalMemGPU := 0
+			AvailableMemGPU := 0
+			for _, node := range pool.nodes {
+				if i, err := strconv.ParseFloat(node.UtilCPU, 64); err != nil {
+					UtilCPU += i
+				}
+				TotalCPU += node.NumCPU
+				TotalMem += str2int(node.MemTotal, 0)
+				AvailableMem += str2int(node.MemAvailable, 0)
+
+				for _, GPU := range node.Status {
+					UtilGPU += GPU.UtilizationGPU
+					TotalGPU ++
+					TotalMemGPU += GPU.MemoryTotal
+					AvailableMemGPU += GPU.MemoryFree
+				}
+			}
+			summary["ts"] = time.Now().Format("2006-01-02 15:04:05")
+			summary["cpu_util"] = fmt.Sprintf("%.2f", UtilCPU/(float64(len(pool.nodes))+0.001))
+			summary["cpu_total"] = strconv.Itoa(TotalCPU)
+			summary["mem_total"] = strconv.Itoa(TotalMem)
+			summary["mem_available"] = strconv.Itoa(AvailableMem)
+			summary["gpu_total"] = strconv.Itoa(TotalGPU)
+			if TotalGPU == 0 {
+				summary["gpu_util"] = "0"
+			} else {
+				summary["gpu_util"] = fmt.Sprintf("%2d", UtilGPU/TotalGPU)
+			}
+			summary["gpu_mem_total"] = strconv.Itoa(TotalMemGPU)
+			summary["gpu_mem_available"] = strconv.Itoa(AvailableMemGPU)
+
+			pool.history = append(pool.history, summary)
+
+			if len(pool.history) > 60 {
+				pool.history = pool.history[0:60]
+			}
+			time.Sleep(time.Second * 60)
+		}
+	}()
 }
 
 func (pool *ResourcePool) update(node NodeStatus) {
@@ -37,3 +94,11 @@ func (pool *ResourcePool) getByID(id string) NodeStatus {
 	}
 	return NodeStatus{}
 }
+
+func (pool *ResourcePool) list() MsgResource {
+	return MsgResource{Code: 0, Resource: pool.nodes}
+}
+
+func (pool *ResourcePool) statusHistory() MsgPoolStatusHistory {
+	return MsgPoolStatusHistory{Code: 0, Data: pool.history}
+}