package main import ( "bytes" "fmt" "os" "os/exec" "path/filepath" "sync" ) type jieguo struct { go_num int Lock sync.Mutex } var searchName string = os.Args[1] var search_result chan string = make(chan string, 1) var nums chan int = make(chan int, 1) var x jieguo = jieguo{go_num: 0} func main() { list := []string{"D:","E:","F"} for _, i := range list { fmt.Println(i, searchName) go pan(i) } for { select { case x := <-search_result: fmt.Println(x) case y := <-nums: if y == 0 { os.Exit(0) } } } } func pan(path string) { a := exec.Command("ls", path) result, e := a.Output() if e != nil { fmt.Println(e) } z := bytes.Split(result, []byte{10}) for _, i := range z[:len(z)-1] { s := path + "\\" + string(i) fmt.Println(s) e = os.Chdir(s) if e == nil { x.Lock.Lock() x.go_num = x.go_num + 1 x.Lock.Unlock() go func() { filepath.Walk(s, walk) x.Lock.Lock() x.go_num = x.go_num - 1 nums <- x.go_num x.Lock.Unlock() }() } else { fmt.Println(e) if string(i) == searchName { fmt.Println(s) } } } } func walk(path string, info os.FileInfo, err error) error { if err != nil { return err } if !info.IsDir() && info.Name() == searchName { search_result <- path return nil } return nil }
时间: 2024-10-09 05:28:07