mirror of
https://github.com/tw93/Mole.git
synced 2026-02-04 19:44:44 +00:00
53 lines
965 B
Go
53 lines
965 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/shirou/gopsutil/v3/mem"
|
|
)
|
|
|
|
func collectMemory() (MemoryStatus, error) {
|
|
vm, err := mem.VirtualMemory()
|
|
if err != nil {
|
|
return MemoryStatus{}, err
|
|
}
|
|
|
|
swap, _ := mem.SwapMemory()
|
|
pressure := getMemoryPressure()
|
|
|
|
return MemoryStatus{
|
|
Used: vm.Used,
|
|
Total: vm.Total,
|
|
UsedPercent: vm.UsedPercent,
|
|
SwapUsed: swap.Used,
|
|
SwapTotal: swap.Total,
|
|
Pressure: pressure,
|
|
}, nil
|
|
}
|
|
|
|
func getMemoryPressure() string {
|
|
if runtime.GOOS != "darwin" {
|
|
return ""
|
|
}
|
|
ctx, cancel := context.WithTimeout(context.Background(), 500*time.Millisecond)
|
|
defer cancel()
|
|
out, err := runCmd(ctx, "memory_pressure")
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
lower := strings.ToLower(out)
|
|
if strings.Contains(lower, "critical") {
|
|
return "critical"
|
|
}
|
|
if strings.Contains(lower, "warn") {
|
|
return "warn"
|
|
}
|
|
if strings.Contains(lower, "normal") {
|
|
return "normal"
|
|
}
|
|
return ""
|
|
}
|