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

avatar
作者
筋斗云
阅读量:1

在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的数量和实现具体的文件处理逻辑。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!