93 lines
2.1 KiB
Go
93 lines
2.1 KiB
Go
package computerInfo
|
||
|
||
import (
|
||
"log"
|
||
"time"
|
||
|
||
"github.com/shirou/gopsutil/mem"
|
||
"github.com/shirou/gopsutil/v3/cpu"
|
||
"gitlab.com/tingshuo/go-diskstate/diskstate"
|
||
)
|
||
|
||
type Storage struct {
|
||
Name string
|
||
FileSystem string
|
||
Total uint64
|
||
Free uint64
|
||
}
|
||
|
||
type storageInfo struct {
|
||
Name string
|
||
Size uint64
|
||
FreeSpace uint64
|
||
FileSystem string
|
||
Used uint64
|
||
}
|
||
|
||
// func GetStorageInfo() {
|
||
// var storageinfo []storageInfo
|
||
// var loaclStorages []Storage
|
||
// err := wmi.Query("Select * from Win32_LogicalDisk", &storageinfo)
|
||
// if err != nil {
|
||
// return
|
||
// }
|
||
|
||
// for _, storage := range storageinfo {
|
||
// info := Storage{
|
||
// Name: storage.Name,
|
||
// FileSystem: storage.FileSystem,
|
||
// Total: storage.Size / 1024 / 1024 / 1024,
|
||
// Free: storage.FreeSpace / 1024 / 1024 / 1024,
|
||
// }
|
||
// if info.Total >= 1 {
|
||
// fmt.Printf("%s总大小%dG，可用%dG\n", info.Name, info.Total, info.Free)
|
||
// loaclStorages = append(loaclStorages, info)
|
||
// }
|
||
// }
|
||
// fmt.Printf("localStorages:= %v\n", loaclStorages)
|
||
// }
|
||
|
||
func GetCurrentStorageInfo(path string) storageInfo {
|
||
state := diskstate.DiskUsage(path)
|
||
info := storageInfo{}
|
||
info.Size = uint64(state.All / diskstate.B)
|
||
info.FreeSpace = uint64(state.Free / diskstate.B)
|
||
info.Used = uint64(state.Used / diskstate.B)
|
||
|
||
// fmt.Printf("All=%dM, Free=%dM, Available=%dM, Used=%dM, Usage=%d%%",
|
||
// state.All/diskstate.B, state.Free/diskstate.MB, state.Available/diskstate.MB, state.Used/diskstate.MB, 100*state.Used/state.All)
|
||
return info
|
||
}
|
||
|
||
type ComputerMonitor struct {
|
||
CPU float64 `json:"cpu"`
|
||
Mem float64 `json:"mem"`
|
||
}
|
||
|
||
// GetCPUPercent 获取CPU使用率
|
||
func GetCPUPercent() float64 {
|
||
percent, err := cpu.Percent(time.Second, false)
|
||
if err != nil {
|
||
log.Fatalln(err.Error())
|
||
return -1
|
||
}
|
||
return percent[0]
|
||
}
|
||
|
||
// GetMemPercent 获取内存使用率
|
||
func GetMemPercent() float64 {
|
||
memInfo, err := mem.VirtualMemory()
|
||
if err != nil {
|
||
log.Fatalln(err.Error())
|
||
return -1
|
||
}
|
||
return memInfo.UsedPercent
|
||
}
|
||
|
||
func GetCpuMem() ComputerMonitor {
|
||
var res ComputerMonitor
|
||
res.CPU = GetCPUPercent()
|
||
res.Mem = GetMemPercent()
|
||
return res
|
||
}
|