
本文介绍在 go 中如何通过通道(channel)协调 goroutine 生命周期,特别是当某个 goroutine 因错误或条件满足而退出时,安全、及时地终止仍在阻塞等待标准输入(如 fmt.Scan)的其他 goroutine。
本文介绍在 go 中如何通过通道(channel)协调 goroutine 生命周期,特别是当某个 goroutine 因错误或条件满足而退出时,安全、及时地终止仍在阻塞等待标准输入(如 `fmt.scan`)的其他 goroutine。
在 Go 并发编程中,一个常见却容易被忽视的问题是:goroutine 因同步 I/O(如 fmt.Scan、os.Stdin.Read)而永久阻塞,无法响应外部终止信号。原始代码中使用全局布尔变量 stop 试图通知 sender 退出,但 fmt.Scanf 是同步阻塞调用——它不会定期检查 stop 状态,导致 sender 卡死在输入等待阶段,违背了“一个组件失败,应协同清理相关协程”的设计原则。
根本解法不是轮询或强制杀掉 goroutine(Go 不支持强行终止),而是将阻塞 I/O 转为可中断的协作式模型。核心思路是:
✅ 将用户输入采集与业务逻辑分离;
✅ 使用无缓冲 channel 传递输入数据,实现 sender 与 receiver 的解耦;
✅ 用专用 stop channel 发送退出信号,receiver 主动关闭后,sender 可通过 select 配合 default 或 context 实现非阻塞检测。
以下是改进后的完整示例(采用更健壮的 context 方案,兼容超时与取消):
package main import ( "bufio" "context" "fmt" "os" "time" ) func listen(ctx context.Context, ch chan<- string) { scanner := bufio.NewScanner(os.Stdin) for { select { case <-ctx.Done(): fmt.Println("Listener exiting: context cancelled") return default: if scanner.Scan() { ch <- scanner.Text() } else { fmt.Println("Input scan failed:", scanner.Err()) return } } } } func write(ctx context.Context, ch <-chan string) { for i := 0; i < 5; i++ { select { case <-ctx.Done(): fmt.Println("Writer exiting early due to context cancellation") return case msg := <-ch: fmt.Printf("Received [%d]: %sn", i+1, msg) } } fmt.Println("Writer completed 5 messages, signalling shutdown") } func main() { ctx, cancel := context.WithCancel(context.Background()) defer cancel() // 确保退出前触发 cancel ch := make(chan string, 1) // 缓冲区为 1,避免 sender 因 receiver 暂未读而阻塞 go listen(ctx, ch) go write(ctx, ch) // 等待 writer 完成或超时(模拟 receiver 异常退出) done := make(chan struct{}) go func() { time.Sleep(6 * time.Second) // 超时兜底 cancel() close(done) }() <-done fmt.Println("All goroutines exited gracefully.") }
? 关键要点说明:
- context.WithCancel 是标准实践:它提供线程安全的取消传播机制,比全局变量或自定义 stop channel 更可靠、可组合;
- bufio.Scanner 替代 fmt.Scan:更安全(防止输入过长 panic)、支持错误检查,且配合 select 可实现非阻塞轮询;
- channel 缓冲区设为 1:避免 sender 在 receiver 退出后因发送阻塞而卡死(若用无缓冲 channel,receiver 退出后 sender 第一次发送即阻塞);
- select + default 或 context.Done() 是非阻塞检测的核心:永远不要在 goroutine 中直接调用阻塞 I/O 而不提供退出路径。
⚠️ 注意事项:
- 标准输入(os.Stdin)本身不支持原生取消,因此必须将读取逻辑包裹在 select 控制流中,借助 context 触发退出;
- 若需支持更复杂的中断(如 Ctrl+C),可监听 os.Interrupt 信号并调用 cancel();
- 生产环境建议统一使用 context 管理所有 goroutine 生命周期,并通过 errgroup.Group 简化错误传播与等待。
通过以上设计,当 receiver 因完成任务或异常退出时,cancel() 调用会立即通知 listener 停止扫描并退出,整个系统实现确定性、可预测、无资源泄漏的优雅终止。