go 中的 Context 用法

context 可以用来告诉子 grountine 什么时候结束。

func worker(ctx context.Context) {
	defer wg.Done()
LABEL:

	for {
		fmt.Println("worker......")
		time.Sleep(time.Second)
		select {
		case <-ctx.Done(): //对于没有缓冲区的 chan，如果后面没有人接，那么就要直接跳过。
			break LABEL
		default:
		}
	}
}

var wg sync.WaitGroup

func main() {
	ctx, cancel := context.WithCancel(context.Background())
	wg.Add(1)
	go worker(ctx)
	time.Sleep(time.Second * 5)
	cancel()  //关闭
	wg.Wait() // 等待
	fmt.Println("over")
}

context控制子程序和子子程序

var wg sync.WaitGroup

func work1(ctx context.Context) {
	defer wg.Done()
	go work2(ctx)
LABEL: //只能写在for循环的上面，直接可以调到这里，且下面的for循环不会再执行
	for {
		select {
		case <-ctx.Done():
			break LABEL
		default:
			time.Sleep(time.Millisecond * 500)
			fmt.Println("work01在工作******")
		}
	}
}

func work2(ctx context.Context) { //串联着把子程序和子子程序都能停下来
	defer wg.Done()
LABEL: 
	for {
		select {
		case <-ctx.Done():
			break LABEL
		default:
			time.Sleep(time.Millisecond * 500)
			fmt.Println("work02在工作******")
		}
	}
}

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