并发创建文件夹并写入文件
在 go 中并发的创建多个文件夹并向这些文件夹写入文件需要两个步骤：
- 使用协程并发创建文件夹。
- 使用 channel 同步协程和主 goroutine，以确保所有文件夹创建后再写入文件。
以下是具体实现：
package main
import (
"fmt"
"io/ioutil"
"os"
"sync"
"time"
)
func main() {
// 创建一个 channel 来同步协程和主 goroutine
done := make(chan struct{})
// 要创建的文件夹列表
dirs := []string{"dir1", "dir2", "dir3"}
// 使用协程并发创建文件夹
var wg sync.WaitGroup
wg.Add(len(dirs))
for _, dir := range dirs {
go func(dir string) {
defer wg.Done()
err := os.Mkdir(dir, 0755)
if err != nil {
fmt.Printf("创建文件夹 %s 出错：%v\n", dir, err)
return
}
fmt.Printf("成功创建文件夹 %s\n", dir)
done <- struct{}{} // 向 channel 发送一个空 struct 来通知已完成
}(dir)
}
// 等待所有文件夹创建完成
for i := 0; i < len(dirs); i++ {
<-done // 从 channel 接收空 struct
}
// 等待所有协程结束
wg.Wait()
// 向文件夹写入文件
for _, dir := range dirs {
err := ioutil.WriteFile(fmt.Sprintf("%s/file.txt", dir), []byte("测试内容"), 0644)
if err != nil {
fmt.Printf("写入文件 %s 出错：%v\n", dir, err)
continue
}
fmt.Printf("成功写入文件 %s\n", dir)
}
time.Sleep(time.Second) // 延时，方便查看输出结果
}在上面的示例中，我们使用协程并发创建文件夹，然后使用 channel 来同步协程和主 goroutine，确保所有文件夹创建完成后再写入文件。通过这种方式，我们可以提高创建文件夹和写入文件的效率。
