使用 Go 语言获取跨平台磁盘空间信息教程

package main

import (
    "fmt"
    "os"

    "golang.org/x/sys/unix"
)

func getDiskUsagePOSIX(path string) (total uint64, free uint64, err error) {
    var stat unix.Statfs_t
    err = unix.Statfs(path, &stat)
    if err != nil {
        return 0, 0, fmt.Errorf("无法获取 %s 的文件系统状态: %w", path, err)
    }

    // 总字节数 = 总块数 * 每块大小
    total = stat.Blocks * uint64(stat.Bsize)
    // 可用字节数 = 可用块数 * 每块大小
    free = stat.Bavail * uint64(stat.Bsize)

    return total, free, nil
}

func main() {
    // 获取当前工作目录的磁盘使用情况
    wd, err := os.Getwd()
    if err != nil {
        fmt.Printf("获取当前工作目录失败: %v
", err)
        return
    }

    total, free, err := getDiskUsagePOSIX(wd)
    if err != nil {
        fmt.Println(err)
        return
    }

    fmt.Printf("POSIX 系统 (%s):
", wd)
    fmt.Printf("  总空间: %d 字节 (%.2f GB)
", total, float64(total)/(1024*1024*1024))
    fmt.Printf("  可用空间: %d 字节 (%.2f GB)
", free, float64(free)/(1024*1024*1024))
}

package main

import (
    "fmt"
    "syscall"

    "golang.org/x/sys/windows"
)

func getDiskUsageWindows(drivePath string) (total uint64, free uint64, err error) {
    var freeBytesAvailable uint64
    var totalNumberOfBytes uint64
    var totalNumberOfFreeBytes uint64

    // GetDiskFreeSpaceEx 需要以 UTF16 编码的路径
    drivePathPtr, err := syscall.UTF16PtrFromString(drivePath)
    if err != nil {
        return 0, 0, fmt.Errorf("转换路径 %s 失败: %w", drivePath, err)
    }

    err = windows.GetDiskFreeSpaceEx(
        drivePathPtr,
        &freeBytesAvailable,
        &totalNumberOfBytes,
        &totalNumberOfFreeBytes,
    )
    if err != nil {
        return 0, 0, fmt.Errorf("获取 %s 磁盘空间失败: %w", drivePath, err)
    }

    // 返回总字节数和调用进程可用的字节数
    return totalNumberOfBytes, freeBytesAvailable, nil
}

func main() {
    // 获取 C 盘的磁盘使用情况
    drive := "C:\"
    total, free, err := getDiskUsageWindows(drive)
    if err != nil {
        fmt.Println(err)
        return
    }

    fmt.Printf("Windows 系统 (%s):
", drive)
    fmt.Printf("  总空间: %d 字节 (%.2f GB)
", total, float64(total)/(1024*1024*1024))
    fmt.Printf("  可用空间: %d 字节 (%.2f GB)
", free, float64(free)/(1024*1024*1024))
}