golang 获取磁盘内存占用

╰+攻爆jí腚メ 2021-09-02 07:44 822阅读 0赞

获取磁盘占用情况 linux mac下有效

import (
        "syscall"
    )
     
    type DiskStatus struct {
        All  uint64 `json:"all"`
        Used uint64 `json:"used"`
        Free uint64 `json:"free"`
    }
     
    // disk usage of path/disk
    func DiskUsage(path string) (disk DiskStatus) {
        fs := syscall.Statfs_t{}
        err := syscall.Statfs(path, &fs)
        if err != nil {
            return
        }
        disk.All = fs.Blocks * uint64(fs.Bsize)
        disk.Free = fs.Bfree * uint64(fs.Bsize)
        disk.Used = disk.All - disk.Free
        return
    }

获取内存占用

import (
        "runtime"
    )
     
    type MemStatus struct {
        All  uint32 `json:"all"`
        Used uint32 `json:"used"`
        Free uint32 `json:"free"`
        Self uint64 `json:"self"`
    }
     
    func MemStat() MemStatus {
        //自身占用
        memStat := new(runtime.MemStats)
        runtime.ReadMemStats(memStat)
        mem := MemStatus{}
        mem.Self = memStat.Alloc
     
        //系统占用,仅linux/mac下有效
        //system memory usage
        sysInfo := new(syscall.Sysinfo_t)
        err := syscall.Sysinfo(sysInfo)
        if err == nil {
            mem.All = sysInfo.Totalram * uint32(syscall.Getpagesize())
            mem.Free = sysInfo.Freeram * uint32(syscall.Getpagesize())
            mem.Used = mem.All - mem.Free
        }
        return mem
    }

通过调用win32 api windows下获取磁盘空间

func main() {
        kernel32, err := syscall.LoadLibrary("Kernel32.dll")
        if err != nil {
            log.Panic(err)
        }
        defer syscall.FreeLibrary(kernel32)
        GetDiskFreeSpaceEx, err := syscall.GetProcAddress(syscall.Handle(kernel32), "GetDiskFreeSpaceExW")
     
        if err != nil {
            log.Panic(err)
        }
     
        lpFreeBytesAvailable := int64(0)
        lpTotalNumberOfBytes := int64(0)
        lpTotalNumberOfFreeBytes := int64(0)
        r, a, b := syscall.Syscall6(uintptr(GetDiskFreeSpaceEx), 4,
            uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr("C:"))),
            uintptr(unsafe.Pointer(&lpFreeBytesAvailable)),
            uintptr(unsafe.Pointer(&lpTotalNumberOfBytes)),
            uintptr(unsafe.Pointer(&lpTotalNumberOfFreeBytes)), 0, 0)
     
        log.Printf("Available  %dmb", lpFreeBytesAvailable/1024/1024.0)
        log.Printf("Total      %dmb", lpTotalNumberOfBytes/1024/1024.0)
        log.Printf("Free       %dmb", lpTotalNumberOfFreeBytes/1024/1024.0)
    }

更简单的版本

func main() {
        h := syscall.MustLoadDLL("kernel32.dll")
        c := h.MustFindProc("GetDiskFreeSpaceExW")
        lpFreeBytesAvailable := int64(0)
        lpTotalNumberOfBytes := int64(0)
        lpTotalNumberOfFreeBytes := int64(0)
        r2, _, err := c.Call(uintptr(unsafe.Pointer(syscall.StringToUTF16Ptr("F:"))),
            uintptr(unsafe.Pointer(&lpFreeBytesAvailable)),
            uintptr(unsafe.Pointer(&lpTotalNumberOfBytes)),
            uintptr(unsafe.Pointer(&lpTotalNumberOfFreeBytes)))
    }