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