mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-02 22:10:26 +08:00
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
package nodes
|
|
|
|
import (
|
|
"runtime"
|
|
"runtime/debug"
|
|
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
|
|
"github.com/shirou/gopsutil/v3/load"
|
|
"github.com/shirou/gopsutil/v3/mem"
|
|
)
|
|
|
|
// 更新内存
|
|
func (this *NodeStatusExecutor) updateMem(status *nodeconfigs.NodeStatus) {
|
|
stat, err := mem.VirtualMemory()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
// 重新计算内存
|
|
if stat.Total > 0 {
|
|
stat.Used = stat.Total - stat.Free - stat.Buffers - stat.Cached
|
|
status.MemoryUsage = float64(stat.Used) / float64(stat.Total)
|
|
}
|
|
|
|
status.MemoryTotal = stat.Total
|
|
|
|
// 内存严重不足时自动释放内存
|
|
if stat.Total > 0 {
|
|
var minFreeMemory = stat.Total / 8
|
|
if minFreeMemory > 1<<30 {
|
|
minFreeMemory = 1 << 30
|
|
}
|
|
if stat.Available > 0 && stat.Available < minFreeMemory {
|
|
runtime.GC()
|
|
debug.FreeOSMemory()
|
|
}
|
|
}
|
|
}
|
|
|
|
// 更新负载
|
|
func (this *NodeStatusExecutor) updateLoad(status *nodeconfigs.NodeStatus) {
|
|
stat, err := load.Avg()
|
|
if err != nil {
|
|
status.Error = err.Error()
|
|
return
|
|
}
|
|
if stat == nil {
|
|
status.Error = "load is nil"
|
|
return
|
|
}
|
|
status.Load1m = stat.Load1
|
|
status.Load5m = stat.Load5
|
|
status.Load15m = stat.Load15
|
|
}
|