如何使用 Go 语言并发创建文件夹并写入文件？

并发创建文件夹并写入文件

在 go 中并发的创建多个文件夹并向这些文件夹写入文件需要两个步骤：

1. 使用协程并发创建文件夹。
2. 使用 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
", dir, err)
                return
            }
            fmt.Printf("成功创建文件夹 %s
", 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
", dir, err)
            continue
        }
        fmt.Printf("成功写入文件 %s
", dir)
    }
    time.Sleep(time.Second) // 延时，方便查看输出结果
}

在上面的示例中，我们使用协程并发创建文件夹，然后使用 channel 来同步协程和主 goroutine，确保所有文件夹创建完成后再写入文件。通过这种方式，我们可以提高创建文件夹和写入文件的效率。

以上就是如何使用 Go 语言并发创建文件夹并写入文件？的详细内容，更多请关注硕下网其它相关文章！