golang linux walk

引言

Golang是一种并发、安全、快速的编程语言，适用于系统级编程和网络编程。它在Linux平台上特别受欢迎，因为它提供了许多强大的工具和库，以便我们可以轻松地在Linux上进行开发。

文件遍历

Golang提供了一个很方便的包--filepath/filepath来实现文件系统的遍历操作。通过使用filepath.Walk函数，我们可以轻松遍历整个文件树，并对每个文件或目录执行自定义操作。

遍历目录

下面的代码展示了如何使用filepath.Walk函数来遍历目录及其所有的子目录和文件：

package main

import (
	"fmt"
	"os"
	"path/filepath"
)

func walkFunction(path string, info os.FileInfo, err error) error {
	if err != nil {
        // 错误处理
		return err
	}
    // 根据需求处理文件或者目录
	fmt.Println(path)
	return nil
}

func main() {
	root := "/path/to/directory"
	err := filepath.Walk(root, walkFunction)
	if err != nil {
        // 错误处理
		fmt.Printf("Error walking the path %s: %v\n", root, err)
	}
}

上面的代码中，filepath.Walk函数接受一个路径和一个函数。该函数会在遍历期间被回调，并传递每个文件/目录的路径、文件信息和错误（如果有）。在我们的walkFunction函数中，我们只是简单地将路径打印出来，并返回空。

过滤文件

有时候我们只关心某一类型的文件，可以使用文件扩展名或者正则表达式进行过滤。

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"regexp"
)

func walkFunction(path string, info os.FileInfo, err error) error {
	if err != nil {
        // 错误处理
		return err
	}
	if !info.IsDir() && regexp.MustCompile(".txt$").MatchString(info.Name()) {
        // 处理指定扩展名的文件
		fmt.Println(path)
	}
	return nil
}

func main() {
	root := "/path/to/directory"
	err := filepath.Walk(root, walkFunction)
	if err != nil {
        // 错误处理
		fmt.Printf("Error walking the path %s: %v\n", root, err)
	}
}

在上面的代码中，我们使用了regexp包来创建一个正则表达式对象，该对象用于匹配以.txt结尾的文件。如果一个文件匹配该表达式，我们就对它进行处理。通过这种方式，我们可以根据需要自定义过滤条件。

递归遍历

filepath.Walk函数会自动递归遍历目录及其子目录，但有时我们希望控制递归的深度。在这种情况下，我们可以使用filepath.WalkDir代替filepath.Walk来实现更灵活的遍历。

package main

import (
	"fmt"
	"os"
	"path/filepath"
)

func walkFunction(path string, d os.DirEntry, err error) error {
	if err != nil {
        // 错误处理
		return err
	}
	if d.IsDir() {
		fmt.Println("[D]", path)
	} else {
		fmt.Println("[F]", path)
	}
	return nil
}

func main() {
	root := "/path/to/directory"
	err := filepath.WalkDir(root, walkFunction)
	if err != nil {
        // 错误处理
		fmt.Printf("Error walking the path %s: %v\n", root, err)
	}
}

上面的代码中，我们使用filepath.WalkDir函数来代替filepath.Walk函数。walkFunction函数带有三个参数，分别是路径、目录信息和错误。我们通过判断目录信息对象的IsDir方法来区分目录和文件，并进行不同的处理。通过这种方式，我们可以自由地控制递归遍历的深度。

结论

通过使用Golang的filepath包，我们可以轻松地遍历Linux上的文件系统。我们可以编写自定义函数来处理每个文件或目录，并且还可以根据需要过滤特定类型的文件。这使得我们能够更好地管理我们的文件和目录，并可以快速找到我们想要的内容。