package main import ( "bufio" "bytes" "flag" "fmt" "io" "os" "path/filepath" "sync/atomic" ) var re []byte var count int32 var c, f, d bool var i, p string func main() { flag.StringVar(&p, "p", "./", "-p=finddir") //搜索的文件夾。默認是當前文件夾。path flag.StringVar(&i, "r", "", "-r=matchstring") //keyword。index flag.BoolVar(&c, "c", false, "-c=true") //是否在文件內容搜索keyword,content flag.BoolVar(&f, "f", false, "-f=true") //是否依據keyword查找文件,file flag.BoolVar(&d, "d", false, "-d=true") //是否依據keyword查找文件夾,directory flag.Parse() if len(i) == 0 { fmt.Println("keyword不能爲空") return } re = []byte(i) if err := filepath.Walk(p, walk); err != nil { fmt.Println("Walk:", err) } if count == 0 { fmt.Println("搜索路徑:", p) fmt.Println("keyword:", i) if d { fmt.Println("是否搜索文件夾:", d) } if f { fmt.Println("是否搜索文件:", f) } if c { fmt.Println("是否搜索內容:", c) } } fmt.Printf("共發現 : %d\n", count) } func walk(path string, info os.FileInfo, err error) error { if err != nil { return err } if info.IsDir() { if d && bytes.Contains([]byte(info.Name()), re) { atomic.AddInt32(&count, 1) fmt.Println("匹配到文件夾:", path) } return nil } if f { if bytes.Contains([]byte(info.Name()), re) { atomic.AddInt32(&count, 1) fmt.Println("匹配到文件:", path) return nil } } if c { return read(path) } return nil } //在文件裏查找keyword func read(path string) error { File, err := os.Open(path) defer File.Close() if err != nil { fmt.Println(path, ":", err) return err } r := bufio.NewReader(File) i := 0 for { i++ b, _, err := r.ReadLine() if bytes.Contains(b, []byte{0}) { //排除二進制文件 return nil } if err != nil { if err == io.EOF { if bytes.Contains(b, re) { atomic.AddInt32(&count, 1) fmt.Printf("在文件:%s第 %d行\n", path, i) fmt.Println(string(bytes.TrimSpace(b))) } return nil } return err } if bytes.Contains(b, re) { atomic.AddInt32(&count, 1) fmt.Printf("在文件:%s第 %d行\n", path, i) fmt.Println(string(bytes.TrimSpace(b))) } } return nil }