dongshan2680 2016-04-19 08:48
浏览 363
已采纳

如何在Golang Walk中提前终止?

What's the idiomatic way, if possible, of having Go's filepath.Walk return early?

I'm writing a function to find a nested directory of a given name. Using filepath.Walk I can't see a way of terminating the tree-walking as soon as I find the first match.

func (*RecursiveFinder) Find(needle string, haystack string) (result string, err error) {
    filepath.Walk(haystack, func(path string, fi os.FileInfo, errIn error) (errOut error) {
        fmt.Println(path)

        if fi.Name() == needle {
            fmt.Println("Found " + path)
            result = path
            return nil
        }

        return
    })

    return
}
  • 写回答

1条回答 默认 最新

  • dongtou5557 2016-04-19 09:07
    关注

    You should return an error from your walkfunc. To make sure no real error has been returned, you can just use a known error, like io.EOF.

    func Find(needle string, haystack string) (result string, err error) {
    
        err = filepath.Walk(haystack, 
          filepath.WalkFunc(func(path string, fi os.FileInfo, errIn error) error {
            fmt.Println(path)
    
            if fi.Name() == needle {
                fmt.Println("Found " + path)
                result = path
                return io.EOF
            }
    
            return nil
        }))
    
        if err == io.EOF {
            err = nil
        }
    
        return
    }
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?