package mainimport (	"bufio"	"bytes"	"flag"	"fmt"	"io"	"os"	"path/filepath"	"sync/atomic")var re []bytevar count int32var c, f, d boolvar i, p stringfunc 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}//在文件里查找keywordfunc 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}