登录
首页 >  Golang >  Go教程

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

时间:2024-11-18 21:43:00 301浏览 收藏

学习知识要善于思考,思考,再思考!今天golang学习网小编就给大家带来《如何使用 Go 语言并发创建文件夹并写入文件?》,以下内容主要包含等知识点,如果你正在学习或准备学习Golang,就都不要错过本文啦~让我们一起来看看吧,能帮助到你就更好了!

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

文中关于的知识介绍,希望对你的学习有所帮助!若是受益匪浅,那就动动鼠标收藏这篇《如何使用 Go 语言并发创建文件夹并写入文件?》文章吧,也可关注golang学习网公众号了解相关技术文章。

相关阅读
更多>
最新阅读
更多>
课程推荐
更多>