Golang 目录文件数量和大小统计
发布时间:2020-12-16 18:23:30 所属栏目:大数据 来源:网络整理
导读:《The Go Programming Language》笔记 import ( "flag" "fmt" "io/ioutil" "os" "path/filepath" "sync" "time" ) //获取目录dir下的文件大小 func walkDir(dir string ,wg *sync.WaitGroup,fileSizes chan - int64 ) { defer wg.Done() for _,entry := rang
《The Go Programming Language》笔记import (
"flag"
"fmt"
"io/ioutil"
"os"
"path/filepath"
"sync"
"time"
)
//获取目录dir下的文件大小
func walkDir(dir string,wg *sync.WaitGroup,fileSizes chan<- int64) {
defer wg.Done()
for _,entry := range dirents(dir) {
if entry.IsDir() {//目录
wg.Add(1)
subDir := filepath.Join(dir,entry.Name())
go walkDir(subDir,wg,fileSizes)
} else {
fileSizes <- entry.Size()
}
}
}
//sema is a counting semaphore for limiting concurrency in dirents
var sema = make(chan struct{}, 20)
//读取目录dir下的文件信息
func dirents(dir string) []os.FileInfo {
sema <- struct{}{}
defer func() { <-sema }()
entries,err := ioutil.ReadDir(dir)
if err != nil {
fmt.Fprintf(os.Stderr,"du: %vn",err)
return nil
}
return entries
}
//输出文件数量的大小
func printDiskUsage(nfiles,nbytes int64) {
fmt.Printf("%d files %.1f GBn",nfiles,float64(nbytes)/1e9)
}
//提供-v 参数会显示程序进度信息
var verbose = flag.Bool("v",false,"show verbose progress messages")
func Start() {
flag.Parse()
roots := flag.Args()//需要统计的目录
if len(roots) == 0 {
roots = []string{"."}
}
fileSizes := make(chan int64)
var wg sync.WaitGroup
for _,root := range roots {
wg.Add(1)
go walkDir(root,&wg,fileSizes)
}
go func() {
wg.Wait() //等待goroutine结束
close(fileSizes)
}()
var tick <-chan time.Time
if *verbose {
tick = time.Tick(100 * time.Millisecond) //输出时间间隔
}
var nfiles,nbytes int64
loop:
for {
select {
case size,ok := <-fileSizes:
if !ok {
break loop
}
nfiles++
nbytes += size
case <-tick:
printDiskUsage(nfiles,nbytes)
}
}
printDiskUsage(nfiles,nbytes)
}
(编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |