温馨提示×

golang多线程遍历目录怎么实现

小亿
117
2023-10-21 15:44:13
栏目: 编程语言

在Go语言中,可以使用goroutine和channel来实现多线程遍历目录。下面是一个简单的示例代码:

package main

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

func main() {
	dir := "path/to/directory" // 目录路径
	fileChan := make(chan string) // 创建一个用于传递文件路径的channel

	// 启动多个goroutine来遍历目录
	for i := 0; i < 5; i++ {
		go func() {
			for file := range fileChan {
				fmt.Println(file)
			}
		}()
	}

	// 遍历目录并将文件路径发送到channel中
	filepath.Walk(dir, func(file string, info os.FileInfo, err error) error {
		if !info.IsDir() {
			fileChan <- file
		}
		return nil
	})

	// 关闭channel
	close(fileChan)
}

在以上代码中,我们使用一个goroutine池来遍历目录,该池中包含了5个goroutine。然后,我们通过filepath.Walk函数遍历目录,并将文件路径发送到fileChan channel中。每个goroutine从fileChan中接收文件路径,并进行处理。最后,我们关闭了fileChan channel。

请根据实际需求调整goroutine的数量和实现具体的文件处理逻辑。

0