go协程的2+1

go协程使用心得-go协程的2+1

go协程只做两件事：执行逻辑函数、监听执行状态并且控制流程

go协程一次只做一件事：执行逻辑函数或监听执行状态并且控制流程

func TestCtx(t *testing.T) {
    ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
    defer cancel()

    res := make([]string, 0)
    mutex := sync.Mutex{}
    go func() { // 监听状态，控制流程
        wg := sync.WaitGroup{}
        for i := 0; i < 100000; i++ {
            wg.Add(1)
            go func() { // 监听状态，控制流程
                defer wg.Done()

                cc, cCancel := context.WithCancel(ctx)
                defer cCancel()

                ch := make(chan string, 1)
                go func() { // 执行逻辑代码
                    // todo logic code...
                    time.Sleep(2 * time.Second)
                    ch <- ""
                }()

                select {
                case <-cc.Done(): // 超时直接退出
                    return
                case a := <-ch:
                    mutex.Lock()
                    defer mutex.Unlock()
                    res = append(res, a) // 完成任务塞入slice
                }
            }()
        }
        wg.Wait() // 正常或非正常完成，退出主context
        cancel()
    }()

    select {
    case <-ctx.Done():
        fmt.Println(len(res))
        return
    }
}