在go语言中使用filepath.Walk 查找文件大小（检测磁盘使用情况）示例

Go  /  管理员 发布于 1年前   588

本文中的这个Go语言小程序的代码示例，目的是找出一个文件在给定目标目录下占用了多少空间。

如果你需要确定哪个文件占用了所有磁盘空间？也许这个Go小程序对你有用。

示例代码：

package main

 import (
         "fmt"
         "os"
 )
 
 // 函数来返回磁盘使用信息
 func diskUsage(currentPath string, info os.FileInfo) int64 {
         size := info.Size()
         if !info.IsDir() {
                 return size
         }
         dir, err := os.Open(currentPath)
         if err != nil {
                 fmt.Println(err)
                 return size
         }
         defer dir.Close()
         files, err := dir.Readdir(-1)
         if err != nil {
                 fmt.Println(err)
                 os.Exit(1)
         }
         for _, file := range files {
                 if file.Name() == "." || file.Name() == ".." {
                         continue
                 }
                 size += diskUsage(currentPath+"/"+file.Name(), file)
         }
         fmt.Printf("Size in bytes : [%d] : [%s]\n", size, currentPath)
         return size
 }
 
 func main() {
         if len(os.Args) != 2 {
            fmt.Printf("USAGE : %s <target_directory> \n", os.Args[0])
            os.Exit(0)
         }
         dir := os.Args[1] // 获得目标目录
         info, err := os.Lstat(dir)
         if err != nil {
                 fmt.Println(err)
                 os.Exit(1)
         }
         diskUsage(dir, info)
 }

输出 :

$ ./diskusage /Users/ | more
Size in bytes : [68] : [/Users//Shared/SC Info]
Size in bytes : [204] : [/Users//Shared]
Size in bytes : [139] : [/Users//sweetlogic/.atom/.apm]
Size in bytes : [402] : [/Users//sweetlogic/.atom/compile-cache/coffee]
Size in bytes : [895] : [/Users//sweetlogic/.atom/compile-cache/cson]