Dong.Ge

左手代码 右手江湖

协程优雅地退出

东哥's Avatar 2019-05-06

  1. 标准库的context.Context用于在协程之间传递信息。它是一个树形的结构,可以从一个Context创建新的Context。当调用一个Context的cancel函数时内部会调用所有儿子的cancel函数

copy from web

标准库的context.Context用于在协程之间传递信息。它是一个树形的结构,可以从一个Context创建新的Context。当调用一个Context的cancel函数时内部会调用所有儿子的cancel函数

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package main

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

func task1(ctx context.Context) {
for {
select {
case <-ctx.Done():
return
default:
}
time.Sleep(time.Second)
}
}

func task2(ctx context.Context) {
for {
select {
case <-ctx.Done():
return
default:
}
time.Sleep(time.Second)
}
}

func main() {
ctx, cancel := context.WithCancel(context.Background())
w := sync.WaitGroup{}
w.Add(1)
go func() {
task1(ctx)
fmt.Println("task1 exit")
w.Done()
}()
w.Add(1)
go func() {
task2(ctx)
fmt.Println("task2 exit")
w.Done()
}()
w.Add(1)
go func() {

time.Sleep(time.Second * 3)
fmt.Println("cancel")
cancel()
w.Done()
}()
w.Wait()
fmt.Println("main exit")
}

本文最后更新于 天前,文中所描述的信息可能已发生改变