feat: add node resource profiling and admin device revocation

2026-09-28 09:00:31 +08:00
parent a7b23b760a
commit 3776c310e4
17 changed files with 589 additions and 36 deletions
@@ -0,0 +1,247 @@
+package sysinfo
+
+import (
+	"bufio"
+	"fmt"
+	"os"
+	"runtime"
+	"strconv"
+	"strings"
+	"syscall"
+	"time"
+)
+
+type Profile struct {
+	Role            string  `json:"role"`
+	CPUCores        int     `json:"cpu_cores"`
+	CPUUsagePercent int     `json:"cpu_usage_percent"`
+	MemPercent      int     `json:"mem_percent"`
+	MemAvailable    string  `json:"mem_available"`
+	SwapPercent     int     `json:"swap_percent"`
+	DiskPercent     int     `json:"disk_percent"`
+	IOPSI           float64 `json:"io_psi"`
+	Docker          bool    `json:"docker"`
+	UpdatedAt       int64   `json:"updated_at"`
+}
+
+func Collect(defaultRole string) Profile {
+	role := os.Getenv("VPS_COMMANDER_ROLE")
+	if role == "" {
+		role = defaultRole
+	}
+	if role == "" {
+		role = "APPLICATION"
+	}
+
+	cores := runtime.NumCPU()
+	cpuPercent := getCPUUsage()
+	memPct, memAvl, swapPct := getMemAndSwap()
+	diskPct := getDiskPercent()
+	psi := getIOPSI()
+	docker := hasDocker()
+
+	return Profile{
+		Role:            role,
+		CPUCores:        cores,
+		CPUUsagePercent: cpuPercent,
+		MemPercent:      memPct,
+		MemAvailable:    memAvl,
+		SwapPercent:     swapPct,
+		DiskPercent:     diskPct,
+		IOPSI:           psi,
+		Docker:          docker,
+		UpdatedAt:       time.Now().Unix(),
+	}
+}
+
+func getCPUUsage() int {
+	// 尝试从 /proc/stat 快速采样或 /proc/loadavg
+	pct := sampleCPUStat()
+	if pct >= 0 {
+		return pct
+	}
+	// 回退到 loadavg / cores * 100
+	if data, err := os.ReadFile("/proc/loadavg"); err == nil {
+		fields := strings.Fields(string(data))
+		if len(fields) > 0 {
+			if l1, err := strconv.ParseFloat(fields[0], 64); err == nil {
+				cores := float64(runtime.NumCPU())
+				if cores < 1 {
+					cores = 1
+				}
+				val := int((l1 / cores) * 100)
+				if val > 100 {
+					val = 100
+				}
+				return val
+			}
+		}
+	}
+	return 0
+}
+
+func sampleCPUStat() int {
+	readStat := func() (idle, total uint64, ok bool) {
+		f, err := os.Open("/proc/stat")
+		if err != nil {
+			return 0, 0, false
+		}
+		defer f.Close()
+		scanner := bufio.NewScanner(f)
+		if scanner.Scan() {
+			fields := strings.Fields(scanner.Text())
+			if len(fields) >= 5 && fields[0] == "cpu" {
+				var sum uint64
+				for i := 1; i < len(fields); i++ {
+					v, _ := strconv.ParseUint(fields[i], 10, 64)
+					sum += v
+				}
+				idleVal, _ := strconv.ParseUint(fields[4], 10, 64)
+				return idleVal, sum, true
+			}
+		}
+		return 0, 0, false
+	}
+
+	idle1, total1, ok1 := readStat()
+	if !ok1 {
+		return -1
+	}
+	time.Sleep(50 * time.Millisecond)
+	idle2, total2, ok2 := readStat()
+	if !ok2 || total2 <= total1 {
+		return -1
+	}
+	deltaTotal := total2 - total1
+	deltaIdle := idle2 - idle1
+	if deltaTotal == 0 {
+		return 0
+	}
+	busy := float64(deltaTotal-deltaIdle) / float64(deltaTotal) * 100
+	if busy < 0 {
+		busy = 0
+	}
+	if busy > 100 {
+		busy = 100
+	}
+	return int(busy)
+}
+
+func getMemAndSwap() (memPct int, memAvl string, swapPct int) {
+	f, err := os.Open("/proc/meminfo")
+	if err != nil {
+		return 0, "N/A", 0
+	}
+	defer f.Close()
+
+	var total, free, avail, buffers, cached, swapTotal, swapFree uint64
+	scanner := bufio.NewScanner(f)
+	for scanner.Scan() {
+		line := scanner.Text()
+		parts := strings.SplitN(line, ":", 2)
+		if len(parts) != 2 {
+			continue
+		}
+		key := strings.TrimSpace(parts[0])
+		valFields := strings.Fields(parts[1])
+		if len(valFields) == 0 {
+			continue
+		}
+		v, _ := strconv.ParseUint(valFields[0], 10, 64)
+		switch key {
+		case "MemTotal":
+			total = v
+		case "MemFree":
+			free = v
+		case "MemAvailable":
+			avail = v
+		case "Buffers":
+			buffers = v
+		case "Cached":
+			cached = v
+		case "SwapTotal":
+			swapTotal = v
+		case "SwapFree":
+			swapFree = v
+		}
+	}
+
+	if avail == 0 {
+		avail = free + buffers + cached
+	}
+
+	if total > 0 {
+		used := total - avail
+		memPct = int(float64(used) / float64(total) * 100)
+		if avail >= 1024*1024 {
+			memAvl = fmt.Sprintf("%.1fG", float64(avail)/(1024*1024))
+		} else {
+			memAvl = fmt.Sprintf("%dM", avail/1024)
+		}
+	} else {
+		memAvl = "N/A"
+	}
+
+	if swapTotal > 0 {
+		swapUsed := swapTotal - swapFree
+		swapPct = int(float64(swapUsed) / float64(swapTotal) * 100)
+	}
+
+	return memPct, memAvl, swapPct
+}
+
+func getDiskPercent() int {
+	var stat syscall.Statfs_t
+	if err := syscall.Statfs("/", &stat); err != nil {
+		return 0
+	}
+	total := stat.Blocks * uint64(stat.Bsize)
+	free := stat.Bfree * uint64(stat.Bsize)
+	if total == 0 {
+		return 0
+	}
+	used := total - free
+	pct := int(float64(used) / float64(total) * 100)
+	if pct > 100 {
+		pct = 100
+	}
+	return pct
+}
+
+func getIOPSI() float64 {
+	// 读取 /proc/pressure/io
+	data, err := os.ReadFile("/proc/pressure/io")
+	if err == nil {
+		lines := strings.Split(string(data), "\n")
+		for _, line := range lines {
+			if strings.HasPrefix(line, "some") {
+				fields := strings.Fields(line)
+				for _, f := range fields {
+					if strings.HasPrefix(f, "avg10=") {
+						val, err := strconv.ParseFloat(strings.TrimPrefix(f, "avg10="), 64)
+						if err == nil {
+							return val
+						}
+					}
+				}
+			}
+		}
+	}
+	// 回退：/proc/loadavg 的 1min load
+	if data, err := os.ReadFile("/proc/loadavg"); err == nil {
+		fields := strings.Fields(string(data))
+		if len(fields) > 0 {
+			if val, err := strconv.ParseFloat(fields[0], 64); err == nil {
+				return val
+			}
+		}
+	}
+	return 0.0
+}
+
+func hasDocker() bool {
+	if _, err := os.Stat("/var/run/docker.sock"); err == nil {
+		return true
+	}
+	return false
+}
@@ -0,0 +1,16 @@
+package sysinfo
+
+import (
+	"testing"
+)
+
+func TestCollect(t *testing.T) {
+	p := Collect("CONTROL")
+	if p.Role != "CONTROL" {
+		t.Fatalf("expected CONTROL, got %s", p.Role)
+	}
+	if p.CPUCores <= 0 {
+		t.Fatalf("expected >0 CPU cores, got %d", p.CPUCores)
+	}
+	t.Logf("Profile: %+v", p)
+}