golang 里出现多 goroutine 的场景很常见, 最常用的两种方式就是 WaitGroup 和 Context, 今天我们了解一下 Context 的应用场景
使用场景
场景一: 多goroutine执行超时通知
并发执行的业务中最常见的就是有协程执行超时, 如果不做超时处理就会出现一个僵尸进程, 这累计的多了就会有一阵手忙脚乱了, 所以我们要在源头上就避免它们
看下面这个示例:
package main
import (
"context"
"fmt"
"time"
)
/**
同一个content可以控制多个goroutine, 确保线程可控, 而不是每新建一个goroutine就要有一个chan去通知他关闭
有了他代码更加简洁
*/
func main() {
fmt.Println("run demo nnn")
demo()
}
func demo() {
ctx, cancel := context.WithTimeout(context.Background(), 9*time.Second)
go watch(ctx, "[线程1]")
go watch(ctx, "[线程2]")
go watch(ctx, "[线程3]")
index := 0
for {
index++
fmt.Printf("%d 秒过去了 n", index)
time.Sleep(1 * time.Second)
if index > 10 {
break
}
}
fmt.Println("通知停止监控")
// 其实此时已经超时, 协程已经提前退出
cancel()
// 防止主进程提前退出
time.Sleep(3 * time.Second)
fmt.Println("done")
}
func watch(ctx context.Context, name string) {
for {
select {
case <-ctx.Done():
fmt.Printf("%s 监控退出, 停止了...n", name)
return
default:
fmt.Printf("%s goroutine监控中... n", name)
time.Sleep(2 * time.Second)
}
}
}
使用 context.WithTimeout() 给文本流设置一个时间上限, 结合 for+select 去接收消息. 当执行超时,或手动关闭都会给 <-ctx.Done() 发送消息,而且所有使用同一个 context 都会收到这个通知, 免去了一个一个通知的繁琐代码
场景二: 类似web服务器中的session
比如在php中(没用swoole扩展), 一个请求进来, 从 $_REQUEST $_SERVER 能获取到的是有关这一条请求的所有信息, 哪怕是使用全局变量也是给这一个请求来服务的, 是线程安全的
但是 golang 就不一样了, 因为程序本身就能起一个 web sever, 因此就不能随便使用全局变量了, 不然就是内存泄露警告. 但是实际业务当中需要有一个类似session 的东西来承载单次请求的信息, 举一个具体的例子就是: 给每次请求加一个 uniqueID 该如何处理? 有了这个 uniqueID, 请求的所有日志都能带上它, 这样排查问题的时候方便追踪一次请求发生了什么
如下:
func demo2() {
pCtx, pCancel := context.WithCancel(context.Background())
pCtx = context.WithValue(pCtx, "parentKey", "parentVale")
go watch(pCtx, "[父进程1]")
go watch(pCtx, "[父进程2]")
cCtx, cCancel := context.WithCancel(pCtx)
go watch(cCtx, "[子进程1]")
go watch(cCtx, "[子进程2]")
fmt.Println(pCtx.Value("parentKey"))
fmt.Println(cCtx.Value("parentKey"))
time.Sleep(10 * time.Second)
fmt.Println("子进程关闭")
cCancel()
time.Sleep(5 * time.Second)
fmt.Println("父进程关闭")
pCancel()
time.Sleep(3 * time.Second)
fmt.Println("done")
}
