android代码统计命令,Golang代码搜集-访Unix du命令的目录文件数量和文件总大小统计程序...

package main

import (

"os"

"io/ioutil"

"fmt"

"path/filepath"

"time"

"flag"

"sync"

)

func walkDir(dir string,wg *sync.WaitGroup, fileSizes chan

//一个goroutine结束 -1

defer wg.Done()

//获取目录下的文件和文件夹列表

for _, entry := range dirents(dir) {

if entry.IsDir() {

//每启动一个新goroutine则 +1

wg.Add(1)

//拼接子目录路径

subdir := filepath.Join(dir, entry.Name())

//递归子目录

go walkDir(subdir, wg,fileSizes)

} else {

//如果是文件，则累加文件大小

fileSizes

}

}

}

// sema是限制dirents并发数量的信号量

var sema = make(chan struct{}, 20)

//获取某目录下的文件和文件夹列表

func dirents(dir string) []os.FileInfo {

sema

defer func() {

entries, err := ioutil.ReadDir(dir)

if err != nil {

fmt.Fprintf(os.Stderr, "du1:%v\n", err)

return nil

}

return entries

}

//输入参数是否包括 -v

var verbose = flag.Bool("v", false, "show verbose progress messages")

func main() {

//读取输入参数

flag.Parse()

roots := flag.Args()

if len(roots) == 0 {

roots = []string{"."}

}

//记录搜索开始时间

start := time.Now()

//goroutine计数

var wg sync.WaitGroup

fsizes := make(chan int64)

for _, root := range roots {

//每启动一个新goroutine则 +1

wg.Add(1)

go walkDir(root,&wg, fsizes)

}

go func(){

//等待所有goroutine结束后关闭fsizes

wg.Wait()

close(fsizes)

}()

//声明变量保存 文件总数量和总大小

var nfiles, nbytes int64

//声明定时器，500ms

var tick

//输入参数包括 -v，则启用定时器

if *verbose{

tick = time.Tick(500 * time.Millisecond)

}

loop:

for {

select {

case size, ok :=

//当fsizes关闭后，跳出for循环

if !ok {

break loop

}

//fsizes有数据，则累加文件数量和总大小

nfiles ++

nbytes += size

case

//定时显示当前进度

printDiskUsage(nfiles, nbytes)

}

}

//显示最终结果

printDiskUsage(nfiles, nbytes)

fmt.Printf("%v", time.Since(start))

}

func printDiskUsage(nfiles, nbytes int64) {

fmt.Printf("%d files %.1f GB\n", nfiles, float64(nbytes)/1e9)

}