如何使用sync.WaitGroup来执行所有的goroutine？-Golang-PHP中文网

如何使用sync.WaitGroup来执行所有的goroutine？

PHPz

发布： 2024-02-09 11:40:09

转载

818人浏览过

如何使用sync.waitgroup来执行所有的goroutine？

php小编苹果为您介绍一种实用的方法来执行所有的goroutine - sync.WaitGroup。在Go语言中，goroutine是轻量级的线程，可以同时执行多个任务。然而，有时我们需要等待所有的goroutine执行完毕后再继续下一步操作。这时，sync.WaitGroup就派上用场了。它提供了一种简单而有效的方式来等待所有的goroutine完成，确保程序的正确执行顺序。接下来，我们将详细介绍如何使用sync.WaitGroup来实现这个功能。

问题内容

目前我正在将所有值推送到通道并从中读取并计算其平方。我想避免使用 time.sleep(2000 * time.millisecond) 因为它会阻止执行 2 秒，而是我希望每个 goroutine 都处理并等待其执行，然后退出程序。我刚刚脱离了 golang，所以现在问这个基本问题:(。有人能帮我解决这个问题吗？

package main

import (
    "fmt"
    "sync"
    "time"
)

func doSquare(num int) int {
    return num * num
}

var wg sync.WaitGroup

func main() {

    wg.Add(1)
    st := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
    quit := make(chan bool)
    ch := make(chan int)
    go func() {
        for i := range st {
            ch <- i
        }
    }()

    go func() {
        for {
            select {
            case x := <-ch:
                fmt.Println(doSquare(x))
                // return

            case <-quit:
                wg.Done()
            default:
                // fmt.Println("---")
                // do something
            }
        }
    }()

    quit <- true

    wg.Wait()
    time.Sleep(2000 * time.Millisecond)
}

登录后复制

解决方法

只需将 quit <- true 移至第一个 goroutine 的末尾即可

易森网络企业版

如果您是新用户，请直接将本程序的所有文件上传在任一文件夹下，Rewrite 目录下放置了伪静态规则和筛选器，可将规则添加进IIS，即可正常使用，不用进行任何设置；(可修改图片等)默认的管理员用户名、密码和验证码都是：yeesen系统默认关闭，请上传后登陆后台点击“核心管理”里操作如下:进入“配置管理”中的&ld

查看详情

func main() {
    wg.add(1)
    st := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
    quit := make(chan bool)
    ch := make(chan int)
    go func() {
        for i := range st {
            ch <- i
        }
        quit <- true
    }()

    go func() {
        for {
            select {
            case x := <-ch:
                fmt.println(dosquare(x))
                // return

            case <-quit:
                wg.done()
                return
            default:
                // fmt.println("---")
                // do something
            }
        }
    }()

    wg.wait()
}

登录后复制

这是 close(ch) 表示没有更多数据的另一种方式

func main() {
    st := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
    quit := make(chan bool)
    ch := make(chan int) // improve concurrency by `ch := make(chan int, 100)`
    go func() {
        for i := range st {
            ch <- i
        }
        close(ch)
    }()

    go func() {
        for x := range ch {
            fmt.Println(doSquare(x))
        }
        quit <- true
    }()

    <-quit
}

登录后复制

以上就是如何使用sync.WaitGroup来执行所有的goroutine？的详细内容，更多请关注php中文网其它相关文章！