使用 done 通道可实现 Go 函数中的上下文取消:创建一个无缓冲 done 通道,并由调用方传递给函数。函数中启动一个 goroutine 监听 done 通道。当需要取消函数时,调用方向 done 通道发送一个值,触发函数 goroutine 退出。
Go 语言函数:使用 done 通道进行上下文取消
done
通道是一种用于在 Go 语言函数中取消上下文的机制。当需要在特定条件下终止函数执行时,它非常有用,例如当接收到信号或超时时。
概念
done
通道是一个无缓冲通道,通常由调用方创建并传递给函数。函数启动一个 goroutine 来监听 done
通道。当调用方决定取消函数时,它会向 done
通道发送一个值,从而导致 goroutine 退出。
代码示例
考虑以下函数,该函数无限循环地打印数字:
func printNumbers(done <-chan struct{}) { for { select { case <-done: return default: fmt.Println("Printing number...") time.Sleep(1 * time.Second) } } }
我们可以使用 done
通道来取消此函数:
func main() { // 创建一个 done 通道 done := make(chan struct{}) // 启动打印数字的 goroutine go printNumbers(done) // 等待 5 秒钟 time.Sleep(5 * time.Second) // 取消函数 close(done) }
当 main
函数关闭 done
通道时,printNumbers
函数中的 goroutine 会收到信号并退出循环。
实战案例
done
通道在需要以可取消的方式执行任务的场景中特别有用。例如:
done
通道优雅地关闭服务器。done
通道取消长期运行的任务。done
通道在超时后取消操作,防止资源浪费。