如何使用 Go 语言并发创建文件夹并写入文件？-Golang-PHP中文网

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

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

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

使用协程并发创建文件夹。
使用 channel 同步协程和主 goroutine，以确保所有文件夹创建后再写入文件。

以下是具体实现：

云模块网站管理系统3.1.03

云模块_YunMOK网站管理系统采用PHP+MYSQL为编程语言，搭载自主研发的模块化引擎驱动技术，实现可视化拖拽无技术创建并管理网站！如你所想，无限可能，支持创建任何网站：企业、商城、O2O、门户、论坛、人才等一块儿搞定！永久免费授权，包括商业用途；默认内置三套免费模板。PC网站+手机网站+适配微信+文章管理+产品管理+SEO优化+组件扩展+NEW Login界面.....目测已经遥遥领先..

下载

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，确保所有文件夹创建完成后再写入文件。通过这种方式，我们可以提高创建文件夹和写入文件的效率。