Golang利用channel协调协程的方法详解

// 构建一个缓存长度为8 的管道
ch := make(chan int ,8)
// 写入
ch <- 10
// 取出
number := <-ch
// 关闭
close(ch)

func main() {  
    ch := make(chan int, 100)  
    wg := sync.WaitGroup{}  
    wg.Add(2)  
    // 生产者
    go func() {  
        defer wg.Done()  
        // 写入数据  
        for i := 0; i < 10; i++ {  
            ch <- i  
        }  
    }()  
    // 生产者
    go func() {  
        defer wg.Done()  
        // 写入数据  
        for i := 0; i < 10; i++ {  
            ch <- i  
        }  
    }()  
    wg2 := sync.WaitGroup{}  
    wg2.Add(1)  
    // 消费者
    go func() {  
        sum := 0  
        fmt.Printf("sum %d \n", sum)  
        for {  
            // 这里会等待  
            temp, ok := <-ch  
            // close 并且 管道为空，ok = false  
            if !ok {    
                break  
            } else {  
                sum += temp  
            }  
        }  
        fmt.Printf("sum %d \n", sum)  
        wg2.Done()  
    }()  
    // 等待俩生产者结束  
    wg.Wait()  
    // 生产数据之后，消费者也并行读完了，此时可以关闭 管道 来 跳出for循环了 
    close(ch)  
    // 等待消费者协程结束  
    wg2.Wait()  
}

func main() {  
    //... 
    //...
    ch2 := make(chan struct{}, 0)  
    go func() {  
        sum := 0  
        fmt.Printf("sum %d \n", sum)  
        for {  
            // 这里会等待  
            temp, ok := <- ch  
            // close 并且 管道为空，ok = false  
            if !ok {  
                break  
            } else {  
                sum += temp  
            }  
        }  
        fmt.Printf("sum %d \n", sum)  
        ch2 <- struct{}{}  
    }()  
    // 等待俩生产者结束  
    wg.Wait()  
    // 关闭管道
    close(ch)  
    // 等待消费者协程结束  
    <-ch2
}

func main() {  
    letter, number := make(chan bool), make(chan bool)  
    wait := sync.WaitGroup{}  
    go func() {  
        i := 1  
        for {  
            if <-number {  
                fmt.Print(i)  
                i++  
                fmt.Print(i)  
                i++  
                letter <- true  
            }  
        }  
    }()  
    wait.Add(1)  
    go func() {  
        // 获得ASCII码  
        i := 'A'  
        for {  
            if <-letter {  
                // 当前已经超过Z时，无需再打印
                if i > 'Z' {  
                    // 停止等待，并且跳出循环
                    wait.Done()  
                    break  
                }  
                // 将ASCII码强转成字母输出  
                fmt.Print(string(i))  
                i++  
                fmt.Print(string(i))  
                i++  
                number <- true  
            }  
        }  
    }()  
    // 放行数字打印的阻塞 
    number <- true  
    // 等待关闭主线程
    wait.Wait()  
}

func main() {  
    letter, number := make(chan bool), make(chan bool)  
    // 再定义一个管道来等待，替代waitGroup的作用
    wait := make(chan bool)
    // 打印数字
    go func() {  
        i := 1  
        for {  
            if <-number {  
                fmt.Print(i)  
                i++  
                fmt.Print(i)  
                i++  
                letter <- true  
            }  
        }  
    }()  
    // 打印字母
    go func() {  
        // 获得ASCII码  
        i := 'A'  
        for {  
            if <-letter {  
                if i > 'Z' {  
                    wait <- true  
                    break  
                }  
                // 将ASCII码强转成字母输出  
                fmt.Print(string(i))  
                i++  
                fmt.Print(string(i))  
                i++  
                number <- true  
            }  
        }  
    }()  
    number <- true  
    // 等待管道取值
    <- wait 
}