学习通道的一个有趣的小例子

从书中看到的，自己稍微改了些注释，还是很有趣的

package main

import (
    "fmt"
    "math/rand"
    "sync"
    "time"
)

var wg sync.WaitGroup

// 初始化随机数种子
func init() {
    rand.Seed(time.Now().UnixNano())
}

func main() {
    // 两个协程
    wg.Add(2)

    // 无缓存通道
    court := make(chan int)

    // 接收通道数据让协程阻塞
    go player("张三", court)
    go player("李四", court)

    // 由main做裁判发球
    court <- 1

    wg.Wait()
}

func player(name string, court chan int) {
    defer wg.Done()

    // 打球的过程
    for {
        // 接受通道数据，形成阻塞
        // 由于ball还要给下面使用，所以这里和if语句分开写
        ball, ok := <-court
        if !ok {
            // 通道关闭，我们赢了
            fmt.Printf("%s 赢了\n", name)
            return
        }

        // 利用随机数模拟失误，自己失误了，关闭通道，另外
        // 一个被阻塞的协程就立即获得，从阻塞中恢复过来，并输出自己赢了
        if n := rand.Intn(1000); n%13 == 0 {
            fmt.Printf("%s 没接住 ", name)
            close(court)
            return
        }

        // 否则就是把球打回去了
        fmt.Printf("%s打中了%d\n", name, ball)
        ball++
        // 发送数据到通道，让另外一个协程从阻塞中拿到通道数据
        court <- ball
    }

}

输出结果：

李四打中了1
张三打中了2
李四打中了3
张三打中了4
李四打中了5
张三打中了6
李四 没接住 张三 赢了