Go语言实现跨平台磁盘空间查询教程

package main

import (
    "fmt"
    "os"
    "golang.org/x/sys/unix" // 导入unix系统调用包
)

func getDiskUsagePosix(path string) (total uint64, free uint64, err error) {
    var stat unix.Statfs_t
    // 调用Statfs获取文件系统状态
    err = unix.Statfs(path, &stat)
    if err != nil {
        return 0, 0, fmt.Errorf("无法获取文件系统状态: %v", 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.Printf("获取磁盘空间失败: %v
", 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" // 导入syscall包，用于字符串转换
    "golang.org/x/sys/windows" // 导入windows系统调用包
)

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

    // GetDiskFreeSpaceEx需要UTF16编码的路径字符串
    // windows.StringToUTF16Ptr 辅助函数可以将Go字符串转换为*uint16
    err = windows.GetDiskFreeSpaceEx(windows.StringToUTF16Ptr(path),
        &freeBytesAvailable, &totalNumberOfBytes, &totalNumberOfFreeBytes)
    if err != nil {
        return 0, 0, fmt.Errorf("无法获取磁盘空间信息: %v", err)
    }

    // freeBytesAvailable 是当前用户可用的空间
    // totalNumberOfBytes 是总空间
    return totalNumberOfBytes, freeBytesAvailable, nil
}

func main() {
    // 获取C盘的磁盘空间信息
    path := "C:\"
    total, free, err := getDiskUsageWindows(path)
    if err != nil {
        fmt.Printf("获取磁盘空间失败: %v
", err)
        return
    }

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

diskusage/
├── diskusage.go      // 统一接口定义
├── diskusage_windows.go // Windows平台实现
├── diskusage_unix.go    // POSIX平台实现

package diskusage

import "fmt"

// DiskUsage 结构体包含磁盘的总空间和可用空间
type DiskUsage struct {
    Total uint64 // 总空间 (字节)
    Free  uint64 // 可用空间 (字节)
}

// GetDiskUsage 获取指定路径的磁盘使用情况
// 该函数将在编译时根据操作系统选择对应的实现
func GetDiskUsage(path string) (*DiskUsage, error) {
    // 实际的实现将在 diskusage_windows.go 或 diskusage_unix.go 中定义
    // 这里只是一个占位符或默认错误
    return nil, fmt.Errorf("未找到当前平台的磁盘使用情况实现")
}

// +build linux darwin freebsd netbsd openbsd

package diskusage

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

func init() {
    // 在Unix系统上，将GetDiskUsage的实现指向unixGetDiskUsage
    GetDiskUsage = unixGetDiskUsage
}

func unixGetDiskUsage(path string) (*DiskUsage, error) {
    var stat unix.Statfs_t
    err := unix.Statfs(path, &stat)
    if err != nil {
        return nil, fmt.Errorf("无法获取文件系统状态 (%s): %v", path, err)
    }

    total := stat.Blocks * uint64(stat.Bsize)
    free := stat.Bavail * uint64(stat.Bsize) // 普通用户可用空间

    return &DiskUsage{Total: total, Free: free}, nil
}

// +build windows

package diskusage

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

func init() {
    // 在Windows系统上，将GetDiskUsage的实现指向windowsGetDiskUsage
    GetDiskUsage = windowsGetDiskUsage
}

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

    err := windows.GetDiskFreeSpaceEx(windows.StringToUTF16Ptr(path),
        &freeBytesAvailable, &totalNumberOfBytes, &totalNumberOfFreeBytes)
    if err != nil {
        return nil, fmt.Errorf("无法获取磁盘空间信息 (%s): %v", path, err)
    }

    return &DiskUsage{Total: totalNumberOfBytes, Free: freeBytesAvailable}, nil
}