feat(profile): optimize profile metrics for Agent decision-making (add disk_free and load_1m, replace swap with load)

This commit is contained in:
kory
2026-09-28 10:15:14 +08:00
parent 0381fe3f2f
commit 886cb2af27
3 changed files with 47 additions and 44 deletions
+39 -30
View File
@@ -12,13 +12,14 @@ import (
) )
type Profile struct { type Profile struct {
Role string `json:"role"` Role string `json:"role,omitempty"`
CPUCores int `json:"cpu_cores"` CPUCores int `json:"cpu_cores"`
CPUUsagePercent int `json:"cpu_usage_percent"` CPUUsagePercent int `json:"cpu_usage_percent"`
Load1M float64 `json:"load_1m"`
MemPercent int `json:"mem_percent"` MemPercent int `json:"mem_percent"`
MemAvailable string `json:"mem_available"` MemAvailable string `json:"mem_available"`
SwapPercent int `json:"swap_percent"`
DiskPercent int `json:"disk_percent"` DiskPercent int `json:"disk_percent"`
DiskFree string `json:"disk_free"`
IOPSI float64 `json:"io_psi"` IOPSI float64 `json:"io_psi"`
Docker bool `json:"docker"` Docker bool `json:"docker"`
UpdatedAt int64 `json:"updated_at"` UpdatedAt int64 `json:"updated_at"`
@@ -35,8 +36,9 @@ func Collect(defaultRole string) Profile {
cores := runtime.NumCPU() cores := runtime.NumCPU()
cpuPercent := getCPUUsage() cpuPercent := getCPUUsage()
memPct, memAvl, swapPct := getMemAndSwap() load1m := getLoad1M()
diskPct := getDiskPercent() memPct, memAvl := getMemInfo()
diskPct, diskFree := getDiskInfo()
psi := getIOPSI() psi := getIOPSI()
docker := hasDocker() docker := hasDocker()
@@ -44,23 +46,34 @@ func Collect(defaultRole string) Profile {
Role: role, Role: role,
CPUCores: cores, CPUCores: cores,
CPUUsagePercent: cpuPercent, CPUUsagePercent: cpuPercent,
Load1M: load1m,
MemPercent: memPct, MemPercent: memPct,
MemAvailable: memAvl, MemAvailable: memAvl,
SwapPercent: swapPct,
DiskPercent: diskPct, DiskPercent: diskPct,
DiskFree: diskFree,
IOPSI: psi, IOPSI: psi,
Docker: docker, Docker: docker,
UpdatedAt: time.Now().Unix(), UpdatedAt: time.Now().Unix(),
} }
} }
func getLoad1M() float64 {
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 getCPUUsage() int { func getCPUUsage() int {
// 尝试从 /proc/stat 快速采样或 /proc/loadavg
pct := sampleCPUStat() pct := sampleCPUStat()
if pct >= 0 { if pct >= 0 {
return pct return pct
} }
// 回退到 loadavg / cores * 100
if data, err := os.ReadFile("/proc/loadavg"); err == nil { if data, err := os.ReadFile("/proc/loadavg"); err == nil {
fields := strings.Fields(string(data)) fields := strings.Fields(string(data))
if len(fields) > 0 { if len(fields) > 0 {
@@ -127,14 +140,14 @@ func sampleCPUStat() int {
return int(busy) return int(busy)
} }
func getMemAndSwap() (memPct int, memAvl string, swapPct int) { func getMemInfo() (memPct int, memAvl string) {
f, err := os.Open("/proc/meminfo") f, err := os.Open("/proc/meminfo")
if err != nil { if err != nil {
return 0, "N/A", 0 return 0, "N/A"
} }
defer f.Close() defer f.Close()
var total, free, avail, buffers, cached, swapTotal, swapFree uint64 var total, free, avail, buffers, cached uint64
scanner := bufio.NewScanner(f) scanner := bufio.NewScanner(f)
for scanner.Scan() { for scanner.Scan() {
line := scanner.Text() line := scanner.Text()
@@ -159,10 +172,6 @@ func getMemAndSwap() (memPct int, memAvl string, swapPct int) {
buffers = v buffers = v
case "Cached": case "Cached":
cached = v cached = v
case "SwapTotal":
swapTotal = v
case "SwapFree":
swapFree = v
} }
} }
@@ -182,34 +191,35 @@ func getMemAndSwap() (memPct int, memAvl string, swapPct int) {
memAvl = "N/A" memAvl = "N/A"
} }
if swapTotal > 0 { return memPct, memAvl
swapUsed := swapTotal - swapFree
swapPct = int(float64(swapUsed) / float64(swapTotal) * 100)
}
return memPct, memAvl, swapPct
} }
func getDiskPercent() int { func getDiskInfo() (diskPct int, diskFree string) {
var stat syscall.Statfs_t var stat syscall.Statfs_t
if err := syscall.Statfs("/", &stat); err != nil { if err := syscall.Statfs("/", &stat); err != nil {
return 0 return 0, "N/A"
} }
total := stat.Blocks * uint64(stat.Bsize) total := stat.Blocks * uint64(stat.Bsize)
free := stat.Bfree * uint64(stat.Bsize) free := stat.Bavail * uint64(stat.Bsize)
if total == 0 { if total == 0 {
return 0 return 0, "N/A"
} }
used := total - free used := total - free
pct := int(float64(used) / float64(total) * 100) diskPct = int(float64(used) / float64(total) * 100)
if pct > 100 { if diskPct > 100 {
pct = 100 diskPct = 100
} }
return pct
if free >= 1024*1024*1024 {
diskFree = fmt.Sprintf("%.1fG", float64(free)/(1024*1024*1024))
} else {
diskFree = fmt.Sprintf("%dM", free/(1024*1024))
}
return diskPct, diskFree
} }
func getIOPSI() float64 { func getIOPSI() float64 {
// 读取 /proc/pressure/io
data, err := os.ReadFile("/proc/pressure/io") data, err := os.ReadFile("/proc/pressure/io")
if err == nil { if err == nil {
lines := strings.Split(string(data), "\n") lines := strings.Split(string(data), "\n")
@@ -227,7 +237,6 @@ func getIOPSI() float64 {
} }
} }
} }
// 回退:/proc/loadavg 的 1min load
if data, err := os.ReadFile("/proc/loadavg"); err == nil { if data, err := os.ReadFile("/proc/loadavg"); err == nil {
fields := strings.Fields(string(data)) fields := strings.Fields(string(data))
if len(fields) > 0 { if len(fields) > 0 {
+4 -7
View File
@@ -39,16 +39,13 @@ async function loadDevices(){
const role = p.role || "APPLICATION"; const role = p.role || "APPLICATION";
const isOnline = d.status === "online"; const isOnline = d.status === "online";
const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-'; const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-';
const memStr = p.mem_percent !== undefined ? `${p.mem_percent}%` + (p.mem_available ? ` (${p.mem_available})` : '') : '-'; const loadStr = p.load_1m !== undefined ? p.load_1m.toFixed(2) : '-';
const swapStr = (p.swap_percent !== undefined && p.swap_percent !== null) ? `${p.swap_percent}%` : '-'; const memStr = p.mem_available ? `${p.mem_available} (${p.mem_percent}%)` : (p.mem_percent !== undefined ? `${p.mem_percent}%` : '-');
const diskStr = p.disk_percent !== undefined ? `${p.disk_percent}%` : '-'; const diskStr = p.disk_free ? `${p.disk_free} (${p.disk_percent}%)` : (p.disk_percent !== undefined ? `${p.disk_percent}%` : '-');
const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-'; const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-';
const dockerVal = p.docker ? 'YES' : 'NO'; const dockerVal = p.docker ? 'YES' : 'NO';
const dockerClass = p.docker ? 'yes' : 'no'; const dockerClass = p.docker ? 'yes' : 'no';
const statusText = isOnline ? "在线" : "离线";
const archOs = (d.arch || d.os) ? `${esc(d.arch||"")}/${esc(d.os||"")} · ` : "";
return ` return `
<div class="device-card ${isOnline ? "online" : "offline"}"> <div class="device-card ${isOnline ? "online" : "offline"}">
<div class="v1-head"> <div class="v1-head">
@@ -63,9 +60,9 @@ async function loadDevices(){
</div> </div>
<div class="profile-grid"> <div class="profile-grid">
<div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div> <div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div>
<div class="p-item"><span class="p-label">LOAD</span><span class="p-val">${loadStr}</span></div>
<div class="p-item"><span class="p-label">MEM</span><span class="p-val">${memStr}</span></div> <div class="p-item"><span class="p-label">MEM</span><span class="p-val">${memStr}</span></div>
<div class="p-item"><span class="p-label">DISK</span><span class="p-val">${diskStr}</span></div> <div class="p-item"><span class="p-label">DISK</span><span class="p-val">${diskStr}</span></div>
<div class="p-item"><span class="p-label">SWAP</span><span class="p-val">${swapStr}</span></div>
<div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div> <div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div>
<div class="p-item"><span class="p-label">DOCKER</span><span class="p-val badge-docker-text ${dockerClass}">${dockerVal}</span></div> <div class="p-item"><span class="p-label">DOCKER</span><span class="p-val badge-docker-text ${dockerClass}">${dockerVal}</span></div>
</div> </div>
+4 -7
View File
@@ -39,16 +39,13 @@ async function loadDevices(){
const role = p.role || "APPLICATION"; const role = p.role || "APPLICATION";
const isOnline = d.status === "online"; const isOnline = d.status === "online";
const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-'; const cpuStr = p.cpu_cores ? `${p.cpu_cores}C · ${p.cpu_usage_percent ?? 0}%` : '-';
const memStr = p.mem_percent !== undefined ? `${p.mem_percent}%` + (p.mem_available ? ` (${p.mem_available})` : '') : '-'; const loadStr = p.load_1m !== undefined ? p.load_1m.toFixed(2) : '-';
const swapStr = (p.swap_percent !== undefined && p.swap_percent !== null) ? `${p.swap_percent}%` : '-'; const memStr = p.mem_available ? `${p.mem_available} (${p.mem_percent}%)` : (p.mem_percent !== undefined ? `${p.mem_percent}%` : '-');
const diskStr = p.disk_percent !== undefined ? `${p.disk_percent}%` : '-'; const diskStr = p.disk_free ? `${p.disk_free} (${p.disk_percent}%)` : (p.disk_percent !== undefined ? `${p.disk_percent}%` : '-');
const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-'; const psiStr = p.io_psi !== undefined ? p.io_psi.toFixed(2) : '-';
const dockerVal = p.docker ? 'YES' : 'NO'; const dockerVal = p.docker ? 'YES' : 'NO';
const dockerClass = p.docker ? 'yes' : 'no'; const dockerClass = p.docker ? 'yes' : 'no';
const statusText = isOnline ? "在线" : "离线";
const archOs = (d.arch || d.os) ? `${esc(d.arch||"")}/${esc(d.os||"")} · ` : "";
return ` return `
<div class="device-card ${isOnline ? "online" : "offline"}"> <div class="device-card ${isOnline ? "online" : "offline"}">
<div class="v1-head"> <div class="v1-head">
@@ -63,9 +60,9 @@ async function loadDevices(){
</div> </div>
<div class="profile-grid"> <div class="profile-grid">
<div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div> <div class="p-item"><span class="p-label">CPU</span><span class="p-val">${cpuStr}</span></div>
<div class="p-item"><span class="p-label">LOAD</span><span class="p-val">${loadStr}</span></div>
<div class="p-item"><span class="p-label">MEM</span><span class="p-val">${memStr}</span></div> <div class="p-item"><span class="p-label">MEM</span><span class="p-val">${memStr}</span></div>
<div class="p-item"><span class="p-label">DISK</span><span class="p-val">${diskStr}</span></div> <div class="p-item"><span class="p-label">DISK</span><span class="p-val">${diskStr}</span></div>
<div class="p-item"><span class="p-label">SWAP</span><span class="p-val">${swapStr}</span></div>
<div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div> <div class="p-item"><span class="p-label">IO PSI</span><span class="p-val">${psiStr}</span></div>
<div class="p-item"><span class="p-label">DOCKER</span><span class="p-val badge-docker-text ${dockerClass}">${dockerVal}</span></div> <div class="p-item"><span class="p-label">DOCKER</span><span class="p-val badge-docker-text ${dockerClass}">${dockerVal}</span></div>
</div> </div>