Go中groutine通信与context控制实例详解

var ch1 chan int
var ch2 chan int
<br>// 任务A， 通过最外层的for来控制定期执行
func TestMe(t *testing.T) {
    ch1 = make(chan int, 0)
    ch2 = make(chan int, 0)
    count := 0
    for {
        count ++
        ctx, cancel := context.WithTimeout(context.Background(), time.Second * 2)<br>                // 任务A的逻辑部分，开启子任务B, C。<br>                // B,C通过ch1，ch2和A通信。<br>                // 同时监听ctx.Done，如果超时了立即结束本次任务不继续执行
        go func(ctx context.Context) {
            go g1(ctx, count)
            go g2(ctx, count)
            v1, v2 := -1, -1
            for v1 == -1 || v2 == -1 {
                select {
                case <- ctx.Done():
                    cancel()
                    fmt.Println("父级2超时退出，当前count值为", count, "当前时间：", time.Now())
                    return
                case v1 = <- ch1:
                case v2 = <- ch2:
                }
            }
            fmt.Println("正常执行完成退出, 开启下次循环，当前count值为：", count, "当前 v1: ", v1, "当前 v2: ", v2)
        }(ctx)<br>                // 任务A监控ctx是否到达timeOUT，timeout就终止本次执行
        select {
        case <- ctx.Done():
            fmt.Println("父级1超时退出，当前count值为", count, "当前时间：", time.Now())
        }
        time.Sleep(time.Second * 3)
    }
}
<br>// 改进后的任务B，即使计算出了结果，也不会再向ch1写数据了，不会造成脏数据
func g1 (ctx context.Context, num int) {
    fmt.Println("g1 num", num, "time", time.Now())
    select {
    case <-ctx.Done():
        fmt.Println("子级 g1关闭, 不向channel中写数据")
        return
    default:
        ch1 <- num
    }
}
<br>// 改进前的任务C
func g2 (ctx context.Context, num int) {
    fmt.Println("g2 num", num, "time", time.Now())
    ch2 <- num

// 在第3次任务重等待3s, 使得它超时<br>func g2 (ctx context.Context, num int) {
    if num == 3 {
        time.Sleep(time.Second * 3)
    }
    fmt.Println("g2 num", num, "time", time.Now())
    ch2 <- num
}

func g2 (ctx context.Context, num int) {
    if num == 3 {
        time.Sleep(time.Second * 10)
        fmt.Println("这次g2 超时，应当g1, g2都不返回")
    }
    fmt.Println("g2 num", num, "time", time.Now())
    select {
    case <-ctx.Done():
        fmt.Println("子级 g2关闭, 不向channel中写数据")
        return
    default:
        ch2 <- num
    }
}