106 lines
2.7 KiB
Go
106 lines
2.7 KiB
Go
//go:build windows
|
||
|
||
package sysinfo
|
||
|
||
import (
|
||
"os"
|
||
"syscall"
|
||
"unsafe"
|
||
)
|
||
|
||
var (
|
||
kernel32 = syscall.NewLazyDLL("kernel32.dll")
|
||
procGetSystemTimes = kernel32.NewProc("GetSystemTimes")
|
||
procGlobalMemoryStatusEx = kernel32.NewProc("GlobalMemoryStatusEx")
|
||
procGetDiskFreeSpaceExW = kernel32.NewProc("GetDiskFreeSpaceExW")
|
||
procGetTickCount64 = kernel32.NewProc("GetTickCount64")
|
||
)
|
||
|
||
type memoryStatusEx struct {
|
||
Length uint32
|
||
MemoryLoad uint32
|
||
TotalPhys uint64
|
||
AvailPhys uint64
|
||
TotalPageFile uint64
|
||
AvailPageFile uint64
|
||
TotalVirtual uint64
|
||
AvailVirtual uint64
|
||
AvailExtendedVirtual uint64
|
||
}
|
||
|
||
func readPlatform() (Stats, CPUSample) {
|
||
var (
|
||
stats Stats
|
||
sample CPUSample
|
||
)
|
||
|
||
var idle, kernel, user syscall.Filetime
|
||
if ret, _, _ := procGetSystemTimes.Call(
|
||
uintptr(unsafe.Pointer(&idle)),
|
||
uintptr(unsafe.Pointer(&kernel)),
|
||
uintptr(unsafe.Pointer(&user)),
|
||
); ret != 0 {
|
||
// kernel 时间包含 idle，因此总时间 = kernel + user
|
||
sample = CPUSample{
|
||
Total: filetimeTicks(kernel) + filetimeTicks(user),
|
||
Idle: filetimeTicks(idle),
|
||
}
|
||
}
|
||
|
||
var mem memoryStatusEx
|
||
mem.Length = uint32(unsafe.Sizeof(mem))
|
||
if ret, _, _ := procGlobalMemoryStatusEx.Call(uintptr(unsafe.Pointer(&mem))); ret != 0 {
|
||
stats.MemTotalBytes = mem.TotalPhys
|
||
stats.MemUsedBytes = mem.TotalPhys - mem.AvailPhys
|
||
}
|
||
|
||
stats.DiskTotalBytes, stats.DiskUsedBytes = diskUsage(volumeRoot())
|
||
return stats, sample
|
||
}
|
||
|
||
// volumeRoot 取当前工作目录所在盘符（Havline 的数据目录与工作目录同盘），失败时回落到 C:\
|
||
func volumeRoot() string {
|
||
wd, err := os.Getwd()
|
||
if err != nil || len(wd) < 2 || wd[1] != ':' {
|
||
return `C:\`
|
||
}
|
||
return wd[:3]
|
||
}
|
||
|
||
func diskUsage(root string) (total, used uint64) {
|
||
rootPtr, err := syscall.UTF16PtrFromString(root)
|
||
if err != nil {
|
||
return 0, 0
|
||
}
|
||
var freeToCaller, totalBytes, totalFree uint64
|
||
ret, _, _ := procGetDiskFreeSpaceExW.Call(
|
||
uintptr(unsafe.Pointer(rootPtr)),
|
||
uintptr(unsafe.Pointer(&freeToCaller)),
|
||
uintptr(unsafe.Pointer(&totalBytes)),
|
||
uintptr(unsafe.Pointer(&totalFree)),
|
||
)
|
||
if ret == 0 || totalBytes == 0 {
|
||
return 0, 0
|
||
}
|
||
if totalFree > totalBytes {
|
||
return totalBytes, 0
|
||
}
|
||
return totalBytes, totalBytes - totalFree
|
||
}
|
||
|
||
// LoadAvg Windows 没有平均负载概念，固定返回 nil
|
||
func LoadAvg() []float64 {
|
||
return nil
|
||
}
|
||
|
||
// UptimeSeconds 用 GetTickCount64 取系统运行秒数
|
||
func UptimeSeconds() uint64 {
|
||
ret, _, _ := procGetTickCount64.Call()
|
||
return uint64(ret) / 1000
|
||
}
|
||
|
||
// filetimeTicks 把 FILETIME（100ns 单位）折算为 10ms 刻度，避免累加时溢出
|
||
func filetimeTicks(ft syscall.Filetime) uint64 {
|
||
return (uint64(ft.HighDateTime)<<32 | uint64(ft.LowDateTime)) / 10000
|
||
}
|