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
    }

    // 计算可用空间（以字节为单位）
    // 可用块数 * 每块大小 = 可用空间字节数
    availableSpaceBytes := stat.Bavail * uint64(stat.Bsize)
    totalSpaceBytes := stat.Blocks * uint64(stat.Bsize)

    fmt.Printf("路径: %s
", wd)
    fmt.Printf("文件系统块大小: %d 字节
", stat.Bsize)
    fmt.Printf("总块数: %d
", stat.Blocks)
    fmt.Printf("可用块数 (非特权用户): %d
", stat.Bavail)
    fmt.Printf("总空间: %d 字节 (%.2f GB)
", totalSpaceBytes, float64(totalSpaceBytes)/(1024*1024*1024))
    fmt.Printf("可用空间: %d 字节 (%.2f GB)
", availableSpaceBytes, float64(availableSpaceBytes)/(1024*1024*1024))
}

package main

import (
    "fmt"
    "golang.org/x/sys/windows"
    "syscall" // 用于错误码
)

func main() {
    var freeBytesAvailable uint64
    var totalNumberOfBytes uint64
    var totalNumberOfFreeBytes uint64

    // 指定要查询的驱动器根路径，注意需要转换为UTF16Ptr
    drive := "C:\" // 或 "C:"
    drivePtr := windows.StringToUTF16Ptr(drive)

    err := windows.GetDiskFreeSpaceEx(
        drivePtr,
        &freeBytesAvailable,
        &totalNumberOfBytes,
        &totalNumberOfFreeBytes,
    )

    if err != nil {
        // 检查特定的错误码，例如 ERROR_FILE_NOT_FOUND (2) 表示驱动器不存在
        if errno, ok := err.(syscall.Errno); ok && errno == windows.ERROR_FILE_NOT_FOUND {
            fmt.Printf("驱动器 '%s' 不存在或无法访问。
", drive)
        } else {
            fmt.Printf("调用GetDiskFreeSpaceEx失败: %v
", err)
        }
        return
    }

    fmt.Printf("驱动器: %s
", drive)
    fmt.Printf("总空间: %d 字节 (%.2f GB)
", totalNumberOfBytes, float64(totalNumberOfBytes)/(1024*1024*1024))
    fmt.Printf("调用进程可用空间: %d 字节 (%.2f GB)
", freeBytesAvailable, float64(freeBytesAvailable)/(1024*1024*1024))
    fmt.Printf("总空闲空间: %d 字节 (%.2f GB)
", totalNumberOfFreeBytes, float64(totalNumberOfFreeBytes)/(1024*1024*1024))
}

package disk

import "errors"

// DiskUsage 结构体存储磁盘使用情况
type DiskUsage struct {
    Total       uint64 // 总空间 (字节)
    Free        uint64 // 可用空间 (字节)
    Available   uint64 // 实际可用空间 (字节，可能与Free不同，取决于权限等)
}

// GetDiskUsage 获取指定路径的磁盘使用情况
func GetDiskUsage(path string) (*DiskUsage, error) {
    // 实际实现将由平台特定文件提供
    return getDiskUsage(path)
}

// getDiskUsage 函数的实际实现由 disk_unix.go 或 disk_windows.go 提供
func getDiskUsage(path string) (*DiskUsage, error) {
    return nil, errors.New("platform-specific implementation not found")
}

// +build unix

package disk

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

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 disk

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

func getDiskUsage(path string) (*DiskUsage, error) {
    var freeBytesAvailable uint64
    var totalNumberOfBytes uint64
    var totalNumberOfFreeBytes uint64

    // Windows需要驱动器根路径，如果path不是，需要提取
    rootPath := filepath.VolumeName(path)
    if rootPath == "" {
        // 可能是相对路径，或者没有卷名，尝试使用path本身
        rootPath = path
    }
    // 确保以斜杠结尾，如果不是根目录
    if len(rootPath) == 2 && rootPath[1] == ':' { // C: -> C:
        rootPath += "\"
    } else if len(rootPath) > 0 && rootPath[len(rootPath)-1] != '\' {
        // 对于非根目录路径，GetDiskFreeSpaceEx 仍然需要一个有效的路径，
        // 但通常我们查询的是整个卷。此处简化处理，直接使用path
        // 更健壮的实现可能需要获取路径所在的卷根目录。
        // 对于GetDiskFreeSpaceEx，如果path是文件路径，它会查询文件所在的卷。
    }


    drivePtr := windows.StringToUTF16Ptr(rootPath)

    err := windows.GetDiskFreeSpaceEx(
        drivePtr,
        &freeBytesAvailable,
        &totalNumberOfBytes,
        &totalNumberOfFreeBytes,
    )
    if err != nil {
        return nil, err
    }

    return &DiskUsage{
        Total:     totalNumberOfBytes,
        Free:      totalNumberOfFreeBytes,
        Available: freeBytesAvailable, // 进程可用空间
    }, nil
}

package main

import (
    "fmt"
    "github.com/ricochet2200/go-disk-usage/du"
)

func main() {
    // 获取指定路径的磁盘使用情况
    // 例如，获取当前目录所在分区的磁盘使用情况
    usage, err := du.NewDiskUsage("/") // Linux/macOS 通常是 "/"，Windows 可以是 "C:"
    if err != nil {
        fmt.Printf("获取磁盘使用情况失败: %v
", err)
        return
    }

    fmt.Printf("路径: /
")
    fmt.Printf("总空间: %.2f GB
", float64(usage.Size())/(1024*1024*1024))
    fmt.Printf("已用空间: %.2f GB
", float64(usage.Used())/(1024*1024*1024))
    fmt.Printf("可用空间: %.2f GB
", float64(usage.Available())/(1024*1024*1024))
    fmt.Printf("使用率: %.2f%%
", usage.Usage()*100)
}