使用Context控制goroutine生命周期,通过context.WithCancel创建可取消的上下文,调用cancel函数通知goroutine退出,示例中在循环内检查ctx.Done()并执行清理后退出。

在Go语言中,实现并发任务的优雅停止是构建健壮服务的关键。当程序需要退出时,不能直接终止正在运行的goroutine,否则可能导致资源泄漏、数据不一致或文件写入不完整等问题。以下是几种常见的golang并发任务优雅停止的方法。
使用Context控制goroutine生命周期
Go的context.Context是管理并发任务生命周期的标准方式,尤其适合处理http请求链路或超时控制。
说明:通过context.WithCancel创建可取消的上下文,在需要停止时调用cancel函数,通知所有监听该context的goroutine退出。
- 启动goroutine时传入context
- 在循环中定期检查ctx.Done()
- 收到信号后执行清理操作再退出
示例代码:
立即学习“go语言免费学习笔记(深入)”;
ctx, cancel := context.WithCancel(context.Background()) go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println("任务收到退出信号") // 执行清理工作 return default: // 正常任务逻辑 time.Sleep(100 * time.Millisecond) } } }(ctx) <p>// 外部触发停止 cancel()
通过channel发送停止信号
使用布尔型或空结构体channel作为通知机制,是最基础且清晰的停止方式。
说明:定义一个stopCh chan Struct{},goroutine监听此channel,主程序关闭或发送值到channel即可触发退出。
- 推荐使用struct{}{}作为空信号,节省内存
- 可用close(stopCh)触发广播(适用于多个监听者)
- 结合select使用避免阻塞
示例代码:
立即学习“go语言免费学习笔记(深入)”;
stopCh := make(chan struct{}) go func() { for { select { case <-stopCh: fmt.Println("接收到停止指令") return default: // 执行任务 time.Sleep(50 * time.Millisecond) } } }() <p>// 触发停止 close(stopCh)
结合WaitGroup等待任务完成
当需要确保所有并发任务在退出前完成当前工作时,应配合sync.WaitGroup使用。
说明:在每个goroutine启动前wg.Add(1),退出前调用wg.Done(),主协程调用wg.Wait()等待全部结束。
- 防止main函数提前退出导致goroutine被强制中断
- 与context或channel搭配使用效果更佳
- 注意Add应在goroutine外调用,避免竞态
典型场景:批量处理任务,需等所有任务处理完才退出。
捕获系统信号实现优雅退出
真实服务常需响应SIGTERM或SIGINT(Ctrl+C)来安全关闭。
方法:使用signal.Notify监听信号,触发cancel或关闭stop channel。
- 创建专门的信号监听channel
- 收到信号后启动关闭流程
- 可设置超时,防止长时间等待
示例片段:
sigCh := make(chan os.Signal, 1) signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM) go func() { <-sigCh fmt.Println("收到退出信号,开始关闭...") cancel() // 或 close(stopCh) }()
基本上就这些。选择哪种方式取决于具体场景:简单任务用channel,复杂调用链用context,生产服务建议组合使用context、channel和WaitGroup,并接入信号处理。关键是让每个goroutine有机会清理资源,做到真正“优雅”退出。


