如何递归遍历文件和目录

huangapple
go评论114阅读模式
英文:

how to recursively traverse files and directories

问题

尝试运行程序时，需要显示所有的 .md 文件，我有一个名为 test 的子文件夹，其中有一个 .md 文件，但是脚本找不到它？

  2. import (
  3. 	"fmt"
  4. 	"log"
  5. 	"strings"
  6. 	"os"
  7. )
  9. func main() {
  11. 	dir, err := os.ReadDir(".")
  13. 	if err != nil {
  14. 		log.Fatal(err)
  15. 	}
  17. 	for _, x := range dir {
  18. 		if strings.HasSuffix(x.Name(), ".md") {
  19. 			fmt.Println(x.Name())
  20. 		}
  21. 	}
  22. }
英文:

When trying to run the program is need to show all .md files, i have a test subfolder with a .md in it, but the scripts dose not find it?

  2. import (
  3. 	"fmt"
  4. 	"log"
  5. 	"strings"
  6. 	"os"
  7. )
  9. func main() {
  11. 	dir, err := os.ReadDir(".")
  13. 	if err != nil {
  14. 		log.Fatal(err)
  15. 	}
  17. 	for _, x := range dir {
  18. 		if strings.HasSuffix(x.Name(), ".md") {
  19. 			fmt.Println(x.Name())
  20. 		}
  21. 	}
  22. }
  25. </details>
  28. # 答案1
  29. **得分**: 1
  31. 你可以使用[Walkdir][1]：
  33. ```go
  34. package main
  36. import (
  37.    "io/fs"
  38.    "path/filepath"
  39. )
  41. func main() {
  42.    filepath.WalkDir(".", func(s string, d fs.DirEntry, e error) error {
  43.       if e != nil { return e }
  44.       if !d.IsDir() {
  45.          println(s)
  46.       }
  47.       return nil
  48.    })
  49. }
英文:

You can use Walkdir:

  1. package main
  3. import (
  4.    &quot;io/fs&quot;
  5.    &quot;path/filepath&quot;
  6. )
  9. func main() {
  10.    filepath.WalkDir(&quot;.&quot;, func(s string, d fs.DirEntry, e error) error {
  11. 	if e != nil { return e }
  12. 	if ! d.IsDir() {
  13. 	   println(s)
  14. 	}
  15. 	return nil
  16.  })
  17. }

答案2

得分: 0

你需要使用filepath.WalkDir来递归检查目录，或者你可以使用在1.16版本中引入的filepath.Walk。os.ReadDir只能在提供的目录内工作。

  1. filepath.Walk(".", func(path string, info fs.FileInfo, err error) error {
  2.     if err != nil {
  3.         fmt.Printf("处理访问路径 %q 失败时防止 panic: %v\n", path, err)
  4.         return err
  5.     }
  7.     if strings.HasSuffix(info.Name(), ".md") {
  8.         fmt.Printf("访问的文件或目录：%q\n", path)
  9.     }
  11.     return nil
  12. })
英文:

You need to use filepath.WalkDir to check directories recursively or you can use introduced in 1.16 filepath.Walk. os.ReadDir will work only within provided directory.

  1. filepath.Walk(&quot;.&quot;, func(path string, info fs.FileInfo, err error) error {
  2. 	if err != nil {
  3. 		fmt.Printf(&quot;prevent panic by handling failure accessing a path %q: %v\n&quot;, path, err)
  4. 		return err
  5. 	}
  7. 	if strings.HasSuffix(info.Name(), &quot;.md&quot;) {
  8. 		fmt.Printf(&quot;visited file or dir: %q\n&quot;, path)
  9. 	}
  11. 	return nil
  12. })

答案3

得分: 0

以下是使用container/ring结构的实现代码：

  1. type (
  2. 	DirectoryGraph struct {
  3. 		RootPath string
  4. 		root     *ring.Ring
  5. 		Node     *ring.Ring
  6. 	}
  7. )
  9. func NewDirectoryGraph(root string) DirectoryGraph {
  10. 	r := ring.New(1)
  11. 	graph := DirectoryGraph{
  12. 		RootPath: root,
  13. 		root:     r,
  14. 		Node:     r,
  15. 	}
  16. 	filepath.WalkDir(graph.RootPath, graph.walk)
  17. 	return graph
  18. }
  20. func (g DirectoryGraph) walk(s string, d fs.DirEntry, e error) error {
  21. 	if e != nil {
  22. 		return e
  23. 	}
  24. 	next := ring.New(1)
  25. 	node := g.serialize(s, d, e)
  26. 	next.Value = node
  27. 	g.root.Link(next).Next()
  28. 	return nil
  29. }
  31. // Serializes a file-node
  32. func (g DirectoryGraph) serialize(s string, d fs.DirEntry, e error) FileNode {
  33. 	n := FileNode{
  34. 		Path: s,
  35. 		Dir:  d,
  36. 		Sys:  SysInfo{},
  37. 	}
  39.     ...
  41. 	return n
  42. }

完整代码请参考这里。

英文:

To add onto the already sufficient responses here is my implementation using the container/ring structure

full code here

  2. type (
  3. 	DirectoryGraph struct {
  4. 		RootPath string
  5. 		root     *ring.Ring
  6. 		Node     *ring.Ring
  7. 	}
  8. )
  10. func NewDirectoryGraph(root string) DirectoryGraph {
  11. 	r := ring.New(1)
  12. 	graph := DirectoryGraph{
  13. 		RootPath: root,
  14. 		root:     r,
  15. 		Node:     r,
  16. 	}
  17. 	filepath.WalkDir(graph.RootPath, graph.walk)
  18. 	return graph
  19. }
  21. func (g DirectoryGraph) walk(s string, d fs.DirEntry, e error) error {
  22. 	if e != nil {
  23. 		return e
  24. 	}
  25. 	next := ring.New(1)
  26. 	node := g.serialize(s, d, e)
  27. 	next.Value = node
  28. 	g.root.Link(next).Next()
  29. 	return nil
  30. }
  32. // Serializes a file-node
  33. func (g DirectoryGraph) serialize(s string, d fs.DirEntry, e error) FileNode {
  34. 	n := FileNode{
  35. 		Path: s,
  36. 		Dir:  d,
  37. 		Sys:  SysInfo{},
  38. 	}
  40.     ...
  42. 	return n
  43. }

huangapple
  • 本文由 发表于 2021年8月12日 15:10:27
  • 转载请务必保留本文链接：https://go.coder-hub.com/68753063.html
  • go
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定