resource_st.go

package boomer

import (
	"fmt"
	"strconv"
	"time"

	"github.com/shirou/gopsutil/mem"
	"github.com/shirou/gopsutil/v3/cpu"
)

type Storage struct {
	Name       string
	FileSystem string
	Total      uint64
	Free       uint64
}

type storageInfo struct {
	Name       string
	Size       uint64
	FreeSpace  uint64
	FileSystem string
}

//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)
//}

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 {
		fmt.Println(err.Error())
		return -1
	}
	result, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", percent[0]), 64)
	return result
}

// GetMemPercent 获取内存使用率
func GetMemPercent() float64 {
	memInfo, err := mem.VirtualMemory()
	if err != nil {
		fmt.Println(err.Error())
		return -1
	}
	result, _ := strconv.ParseFloat(fmt.Sprintf("%.2f", memInfo.UsedPercent), 64)
	return result
}

func GetCpuMem() ComputerMonitor {
	var res ComputerMonitor
	res.CPU = GetCPUPercent()
	res.Mem = GetMemPercent()
	return res
}