Go 的 select 实战:超时、非阻塞收发与优雅退出的三个套路

Go 的 select 实战:超时、非阻塞收发与优雅退出的三个套路很多人学 Go 并发,select只会写「从多个 channel 里随便读一个」。但实战里select真正好用的是三个套路:给操作加超时、非阻塞地试探 channel、以及优雅关闭 goroutine。这三个不掌握,写出来的并发代码要么卡死,要么泄漏 goroutine。这篇一个一个过。套路一:给 channel 操作加超时朴素写法:直接从 channel 读,读不到就永远阻塞。// 危险:worker 出问题没往 ch 写,这里会永久卡住funcwaitResult(ch-chanint)int{return-ch}生产环境这种代码就是定时炸弹——上游一旦不响应,这个 goroutine 就永远挂在这里,连带上游的资源也释放不掉。正确做法是用select配一个time.After:importtimefuncwaitResult(ch-chanint)(int,error){select{casev:-ch:returnv,nilcase-time.After(2*time.Second):// 超过 2 秒没结果就返回错误,不再干等return0,fmt.Errorf(timeout waiting for result)}}time.After(d)返回一个 channel,d时间后会往里塞一个值。select会同时监听结果 channel 和这个定时 channel,哪个先来走哪个。一个坑:time.After在超时前不会被 GC 回收。如果你在一个高频循环里用它,会短暂堆积大量 Timer。循环内的超时应改用time.NewTimer并手动Stop:funcwaitResultLoop(ch-chanint){timer:time.NewTimer(2*time.Second)defertimer.Stop()// 用完及时释放,避免 Timer 堆积select{casev:-ch:fmt.Println(v)case-timer.C:fmt.Println(timeout)}}套路二:非阻塞收发有时你不想等,只想「试一下,能读就读,不能读就算了」。关键是给select加一个default分支——只要有default,select就永远不会阻塞。// 非阻塞读:channel 里有数据就拿,没有就走 defaultfunctryRead(ch-chanint){select{casev:-ch:fmt.Println(got:,v)default:fmt.Println(nothing ready, skip)}}// 非阻塞写:channel 满了就丢弃,不阻塞生产者functrySend(chchan-int,vint)bool{select{casech-v:returntruedefault:returnfalse// 缓冲满了,主动丢弃(比如做限流/降级)}}非阻塞写在实战里特别有用:比如做一个「有界的事件上报队列」,队列满了宁可丢事件也不能阻塞主流程,trySend就是标准写法。套路三:优雅退出 goroutine启动了 goroutine 却没法通知它退出,是 Go 里最常见的泄漏源。标准解法是用一个donechannel(或context)配select:funcworker(jobs-chanint,done-chanstruct{}){for{select{casejob:-jobs:process(job)case-done:// 收到退出信号,清理后返回,goroutine 正常结束fmt.Println(worker exiting)return}}}funcmain(){jobs:make(chanint)done:make(chanstruct{})goworker(jobs,done)jobs-1jobs-2close(done)// 关闭 done,所有监听它的 select 都会立刻收到零值time.Sleep(100*time.Millisecond)// 等 worker 打印退出(实际用 WaitGroup)}funcprocess(nint){fmt.Println(processing,n)}这里的精髓是close(done)而不是往里发值:关闭 channel 后,所有从它读取的select分支都会立刻返回(读到零值),所以能一次性通知任意数量的 goroutine 退出,不用给每个 goroutine 单独发信号。实际项目里更推荐直接用context.Context,它就是把这套donechannel 封装好了,还能级联取消:funcworker(ctx context.Context,jobs-chanint){for{select{casejob:-jobs:process(job)case-ctx.Done():return// ctx 被 cancel 或超时,退出}}}小结超时:selecttime.After,别裸读 channel;循环里改用time.NewTimerStop避免 Timer 堆积。非阻塞:加default分支,select就永不阻塞,适合做「能读就读、队满就丢」的降级逻辑。优雅退出:select监听done/ctx.Done(),用close(done)一次性通知所有 goroutine 退出,避免泄漏。一句话记忆:select不只是「多路复用」,time.After、default、done三个搭配才是它在生产环境的真正价值——分别对应「不干等」「不阻塞」「能收场」。