使用context关闭协程以及协程中的协程

Posted 2021-01-14 csir

package main

import (
    "sync"
    "context"
    "fmt"
    "time"
)

var wg sync.WaitGroup

func worker2(ctx context.Context) {
    LOOP:
        for {
            fmt.Printf("worker2
")
            time.Sleep(time.Second)
            select {
            case <- ctx.Done():
                break LOOP
            default:
            }
        }
}


func worker(ctx context.Context) {
    go worker2(ctx)
    LOOP:
        for {
            fmt.Printf("worker
")
            time.Sleep(time.Second)
            select {
            case <- ctx.Done():
                break LOOP
            default:

            }
        }
    wg.Done()
}

func main() {
    ctx := context.Background()
    ctx, cancel := context.WithCancel(ctx)
    wg.Add(1)
    go worker(ctx)
    //time.Sleep(time.Second*3)
    cancel()
    wg.Wait()
}