多 goroutine 问题

[seacoastboy]

package main
import (
        "fmt"
)

func main(){
        var str = []string {"1", "2", "1","1", "2","1", "2", "1","1", "2","1", "2", "1","1", "2","1", "2", "1","1", "2", "0", "6","9","4"}
        for _, t := range str {
                switch t {
                case "1":
                        go go1(t)
                case "2":
                        go go2(t)
                default:
                        //
                }
        }

        fmt.Println("hello")


}


func go1(st string) {
        //逻辑处理
        fmt.Println(st)
}
func go2(st string){
        //逻辑处理
        fmt.Println(st)
}

我开启 2 个 goroutine ， 但是我不知道如何做堵塞，确保 2 个 goroutine 都运行完后在退出主 goroutine 。开始的时候我用 channel 加 计数器去实现，这样的话就需要给函数添加一个参数:

func go1(st string, ch chan bool){
    //逻辑处理
}
func go2(st string, ch chan bool){
    //逻辑处理
}
func main(){
    //省略一些代码
    for {
        <- ch
        t ++
        if t >= len(str) {
            break
        }
    }
}

我想问问， 能不能有更简介明了的办法去实现。

[mkeith]

go 语言 WaitGroup 用法
http://www.baiyuxiong.com/?p=913

[darasion]

sync 包有个 WaitGroup 可以满足你的需要。

[zts1993]

看上去你这个好像不止会产生两个 goroutine 哦。。。

[seacoastboy]

@darasion 对的使用 WaitGroup 可以实现。
@zts1993 有 for 产生很多多 goroutine 的，

我想使用一个全局的 channel 来实现。

[darasion]

@seacoastboy https://golang.org/doc%2Fcodewalk%2Furlpoll.go
这里貌似有个 Poller 函数，是不是你想要的？
想停下 goroutine 的话，发送端 close 就好。

[cobopanda]

看看 WaitGroup 的用法

[cobopanda]

以下代码仅供参考
package main

import (
"fmt"
"runtime"
)

func main() {
data := []int{1, 8, 6, 7, 20}
NumCPU := runtime.NumCPU()
fmt.Println("NumCPU:", NumCPU)
sem := make(chan int, NumCPU)
for _, value := range data {
if value%2 == 0 {
go func(param int) {
sem <- param
fmt.Println("oushu:", param)
}(value)
} else {
go func(param int) {
sem <- param
fmt.Println("jishu:", param)
}(value)
}
}

for i := 0; i < len(data); i++ {
<-sem
}
}