日日操夜夜添-日日操影院-日日草夜夜操-日日干干-精品一区二区三区波多野结衣-精品一区二区三区高清免费不卡

php小編蘋果為您介紹一種實(shí)用的方法來執(zhí)行所有的goroutine – sync.WaitGroup。在Go語言中，goroutine是輕量級的線程，可以同時(shí)執(zhí)行多個(gè)任務(wù)。然而，有時(shí)我們需要等待所有的goroutine執(zhí)行完畢后再繼續(xù)下一步操作。這時(shí)，sync.WaitGroup就派上用場了。它提供了一種簡單而有效的方式來等待所有的goroutine完成，確保程序的正確執(zhí)行順序。接下來，我們將詳細(xì)介紹如何使用sync.WaitGroup來實(shí)現(xiàn)這個(gè)功能。

問題內(nèi)容

目前我正在將所有值推送到通道并從中讀取并計(jì)算其平方。
我想避免使用 time.sleep(2000 * time.millisecond) 因?yàn)樗鼤柚箞?zhí)行 2 秒，而是我希望每個(gè) goroutine 都處理并等待其執(zhí)行，然后退出程序。我剛剛脫離了 golang，所以現(xiàn)在問這個(gè)基本問題:(。有人能幫我解決這個(gè)問題嗎？

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)
}

登錄后復(fù)制

解決方法

只需將 qui??t <- true 移至第一個(gè) goroutine 的末尾即可

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()
}

登錄后復(fù)制

這是 close(ch) 表示沒有更多數(shù)據(jù)的另一種方式

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
}

登錄后復(fù)制