Go语言：跨平台获取系统磁盘空间详解

package main

import (
    "fmt"
    "os"

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

func main() {
    var stat unix.Statfs_t

    // 获取当前工作目录
    wd, err := os.Getwd()
    if err != nil {
        fmt.Printf("获取当前工作目录失败: %v
", err)
        return
    }

    // 调用Statfs获取文件系统统计信息
    err = unix.Statfs(wd, &stat)
    if err != nil {
        fmt.Printf("调用Statfs失败: %v
", err)
        return
    }

    // 计算可用空间字节数：可用块数量 * 每个块的字节大小
    availableBytes := stat.Bavail * uint64(stat.Bsize)
    // 计算总空间字节数：总块数量 * 每个块的字节大小
    totalBytes := stat.Blocks * uint64(stat.Bsize)

    fmt.Printf("目录 '%s' 所在文件系统信息:
", wd)
    fmt.Printf("可用空间: %d 字节 (%.2f GB)
", availableBytes, float64(availableBytes)/(1024*1024*1024))
    fmt.Printf("总空间: %d 字节 (%.2f GB)
", totalBytes, float64(totalBytes)/(1024*1024*1024))
}

package main

import (
    "fmt"
    "syscall"

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

func main() {
    var freeBytesAvailable uint64      // 可用字节数（针对调用进程）
    var totalNumberOfBytes uint64      // 磁盘总字节数
    var totalNumberOfFreeBytes uint64  // 总空闲字节数（所有用户）

    // 指定要查询的驱动器路径，例如 "C:" 或 "C:"
    // 注意：Windows API通常期望路径以反斜杠结尾，但Go的windows.StringToUTF16Ptr可以处理"C:"
    drivePath := "C:\" 

    // 调用GetDiskFreeSpaceEx函数
    err := windows.GetDiskFreeSpaceEx(windows.StringToUTF16Ptr(drivePath),
        &freeBytesAvailable, &totalNumberOfBytes, &totalNumberOfFreeBytes)

    if err != nil {
        // 检查错误是否为 ERROR_FILE_NOT_FOUND (2) 或 ERROR_PATH_NOT_FOUND (3)
        // 这通常意味着驱动器不存在或路径无效
        if err == syscall.Errno(2) || err == syscall.Errno(3) {
            fmt.Printf("驱动器 '%s' 不存在或路径无效。
", drivePath)
        } else {
            fmt.Printf("调用GetDiskFreeSpaceEx失败: %v
", err)
        }
        return
    }

    fmt.Printf("驱动器 '%s' 信息:
", drivePath)
    fmt.Printf("可用空间 (当前用户): %d 字节 (%.2f GB)
", freeBytesAvailable, float64(freeBytesAvailable)/(1024*1024*1024))
    fmt.Printf("总空闲空间 (所有用户): %d 字节 (%.2f GB)
", totalNumberOfFreeBytes, float64(totalNumberOfFreeBytes)/(1024*1024*1024))
    fmt.Printf("总容量: %d 字节 (%.2f GB)
", totalNumberOfBytes, float64(totalNumberOfBytes)/(1024*1024*1024))
}

your_project/
├── main.go
├── disk_info.go         // 定义通用接口/结构体
├── disk_posix.go        // POSIX系统实现 (Linux, macOS)
└── disk_windows.go      // Windows系统实现

// +build linux darwin freebsd

package your_package

import (
    "os"
    "golang.org/x/sys/unix"
)

// 定义一个结构体来存储磁盘信息
type DiskUsage struct {
    Total      uint64
    Free       uint64
    Available  uint64
}

// GetDiskUsage 获取指定路径的磁盘使用情况
func GetDiskUsage(path string) (*DiskUsage, error) {
    var stat unix.Statfs_t
    err := unix.Statfs(path, &stat)
    if err != nil {
        return nil, err
    }

    return &DiskUsage{
        Total:     stat.Blocks * uint64(stat.Bsize),
        Free:      stat.Bfree * uint64(stat.Bsize), // Bfree: 文件系统中的空闲块数量
        Available: stat.Bavail * uint64(stat.Bsize), // Bavail: 非超级用户可用的空闲块数量
    }, nil
}

// +build windows

package your_package

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

// 定义一个结构体来存储磁盘信息 (与posix版本相同，保持接口一致性)
type DiskUsage struct {
    Total      uint64
    Free       uint64
    Available  uint64
}

// GetDiskUsage 获取指定路径的磁盘使用情况
func GetDiskUsage(path string) (*DiskUsage, error) {
    var freeBytesAvailable uint64
    var totalNumberOfBytes uint64
    var totalNumberOfFreeBytes uint64

    // GetDiskFreeSpaceEx 期望路径以反斜杠结尾，如果不是，Go会自动处理
    err := windows.GetDiskFreeSpaceEx(windows.StringToUTF16Ptr(path),
        &freeBytesAvailable, &totalNumberOfBytes, &totalNumberOfFreeBytes)
    if err != nil {
        return nil, err
    }

    return &DiskUsage{
        Total:     totalNumberOfBytes,
        Free:      totalNumberOfFreeBytes, // 通常指所有用户可用的总空闲空间
        Available: freeBytesAvailable,     // 通常指当前用户可用的空闲空间
    }, nil
}