获取磁盘目录使用情况的 Golang 实现

在IT行业这个发展更新速度很快的行业，只有不停止的学习，才不会被行业所淘汰。如果你是Golang学习者，那么本文《获取磁盘目录使用情况的 Golang 实现》就很适合你！本篇内容主要包括##content_title##，希望对大家的知识积累有所帮助，助力实战开发！

我尝试使用 golang 了解目录的磁盘使用情况，但下面的应用程序计算不正确。我不想使用 filepath.walk。我尝试了 os.stat() 但没有获得实际的目录大小。我在下面分享了相关的代码语句，正确的目录信息和应用程序的结果。

const (
    b  = 1
    kb = 1024 * b
    mb = 1024 * kb
    gb = 1024 * mb
)

type diskstatus struct {
    all  uint64
    free uint64
    used uint64
}

func diskusage(path string) (ds diskstatus) {
    fs := syscall.statfs_t{}
    err := syscall.statfs(path, &fs)
    if err != nil {
        return
    }
    ds.all = fs.blocks * uint64(fs.bsize)
    ds.free = fs.bfree * uint64(fs.bsize)
    ds.used = ds.all - ds.free
    fmt.println(path, ds.used)
    return ds
}

func getdir(destination_directory string) *[]model.directory {
    var dir []model.directory
    items, err := ioutil.readdir(destination_directory)

    if err != nil {
        log.fatalln(err)
    }

    for _, item := range items {
        size := diskusage(destination_directory+item.name()).used / gb
        item := model.directory{path: item.name(), totalsize: size}
        dir = append(dir, item)
    }
    return &dir
}

正确的目录信息：

$du -sh *
8.0k containers
2.0g testfolder
3.2m workspace

按应用划分的结果：

containers --> 90
testfolder --> 90
workspace --> 90

正确答案

我相信您滥用了 syscall.Statfs function：它返回 file system statistics，而不是单个目录或文件的大小。因此，您看到的输出不是各个目录的大小，而是这些目录所在的文件系统的总体磁盘使用情况。

如果您不想使用 filepath.Walk（这实际上是通过递归遍历所有文件来计算目录大小的好方法），您可以编写 your own recursive function。

package main

import (
    "os"
    "log"
    "fmt"
)

const (
    b  = 1
    kb = 1024 * b
    mb = 1024 * kb
    gb = 1024 * mb
)

func dirsize(path string) (int64, error) {
    var size int64
    entries, err := os.readdir(path)
    if err != nil {
        return size, err
    }
    for _, entry := range entries {
        if entry.isdir() {
            subdirsize, err := dirsize(path + "/" + entry.name())
            if err != nil {
                log.printf("failed to calculate size of directory %s: %v\n", entry.name(), err)
                continue
            }
            size += subdirsize
        } else {
            fileinfo, err := entry.info()
            if err != nil {
                log.printf("failed to get info of file %s: %v\n", entry.name(), err)
                continue
            }
            size += fileinfo.size()
        }
    }
    return size, nil
}

func main() {
    dirs := []string{"./containers", "./testfolder", "./workspace"}

    for _, dir := range dirs {
        size, err := dirsize(dir)
        if err != nil {
            log.printf("failed to calculate size of directory %s: %v\n", dir, err)
            continue
        }
        fmt.printf("%s --> %.2f gb\n", dir, float64(size)/float64(gb))
    }
}

这将遍历给定目录下的每个目录和文件并总结大小。
如果条目是目录，则会递归计算目录的大小。
虽然...此代码不会处理符号链接或其他特殊文件。

另一个选项是dirsize函数使用filepath.walk遍历指定目录下的所有文件，包括子目录。
filepath.walk 函数有两个参数：根目录和将为根目录树中的每个文件或目录调用的函数。

在这里，对于遇到的每个文件，它将文件的大小添加到总大小中（忽略目录）。
由于 filepath.walk 自动为您处理递归遍历，因此它通常比手动编写递归函数更简单、更直接。
（playground）

package main

import (
    "os"
    "log"
    "fmt"
    "path/filepath"
)

const (
    B  = 1
    KB = 1024 * B
    MB = 1024 * KB
    GB = 1024 * MB
)

func DirSize(path string) (int64, error) {
    var size int64
    err := filepath.Walk(path, func(_ string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }
        if !info.IsDir() {
            size += info.Size()
        }
        return err
    })
    return size, err
}

func main() {
    dirs := []string{"./containers", "./testfolder", "./workspace"}

    for _, dir := range dirs {
        size, err := DirSize(dir)
        if err != nil {
            log.Printf("failed to calculate size of directory %s: %v\n", dir, err)
            continue
        }
        fmt.Printf("%s --> %.2f GB\n", dir, float64(size)/float64(GB))
    }
}

今天关于《获取磁盘目录使用情况的 Golang 实现》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！