mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-10 18:05:48 +02:00
cb4977e5ab
* refactor: nodes analytics * feat(debug): add pprof in debug mode * refactor: websocket error handler
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package analytic
|
|
|
|
import (
|
|
"github.com/shirou/gopsutil/v4/cpu"
|
|
"github.com/shirou/gopsutil/v4/load"
|
|
"github.com/shirou/gopsutil/v4/net"
|
|
"github.com/uozi-tech/cosy/logger"
|
|
"math"
|
|
"runtime"
|
|
"time"
|
|
)
|
|
|
|
func GetNodeStat() (data NodeStat) {
|
|
memory, err := GetMemoryStat()
|
|
|
|
if err != nil {
|
|
logger.Error(err)
|
|
return
|
|
}
|
|
|
|
cpuTimesBefore, _ := cpu.Times(false)
|
|
time.Sleep(1000 * time.Millisecond)
|
|
cpuTimesAfter, _ := cpu.Times(false)
|
|
threadNum := runtime.GOMAXPROCS(0)
|
|
cpuUserUsage := (cpuTimesAfter[0].User - cpuTimesBefore[0].User) / (float64(1000*threadNum) / 1000)
|
|
cpuSystemUsage := (cpuTimesAfter[0].System - cpuTimesBefore[0].System) / (float64(1000*threadNum) / 1000)
|
|
|
|
loadAvg, err := load.Avg()
|
|
if err != nil {
|
|
logger.Error(err)
|
|
return
|
|
}
|
|
|
|
diskStat, err := GetDiskStat()
|
|
if err != nil {
|
|
logger.Error(err)
|
|
return
|
|
}
|
|
|
|
netIO, err := net.IOCounters(false)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
return
|
|
}
|
|
|
|
var network net.IOCountersStat
|
|
if len(netIO) > 0 {
|
|
network = netIO[0]
|
|
}
|
|
|
|
return NodeStat{
|
|
AvgLoad: loadAvg,
|
|
CPUPercent: math.Min((cpuUserUsage+cpuSystemUsage)*100, 100),
|
|
MemoryPercent: memory.Pressure,
|
|
DiskPercent: diskStat.Percentage,
|
|
Network: network,
|
|
}
|
|
}
|