package main import ( "fmt" "os" "path/filepath" "regexp" ) func main() { // 命令行参数 args := os.Args // 检查参数 if len(args) == 1 { fmt.Println("ff is a file find tool. use like bottom") fmt.Println("ff [dir] [regexp]") return } if len(args) < 3 { fmt.Println("args < 3") return } fileName := args[1] pattern := args[2] file, err := os.Open(fileName) if err != nil { fmt.Println(err) return } fi, err := file.Stat() if err != nil { fmt.Println(err) return } if !fi.IsDir() { fmt.Println(fileName, " is not a dir") } reg, err := regexp.Compile(pattern) if err != nil { fmt.Println(err) return } // 遍历目录 filepath.Walk(fileName, func(path string, f os.FileInfo, err error) error { if err != nil { fmt.Println(err) return err } if f.IsDir() { return nil } // 匹配目录 matched := reg.MatchString(f.Name()) if matched { fmt.Println(path) } return nil }) }
有疑问加站长微信联系(非本文作者)