Golang中如何用Request.Context替换http.CloseNotifier实现Server Sent Event

Golang中如何用Request.Context替换http.CloseNotifier实现Server Sent Event 亲爱的Gophers们，

我已经实现了一个服务器发送事件代理，它依赖于http.CloseNotifier CloseNotify来关闭连接。但这个方法已被弃用！

我试图弄清楚如何使用Request.Context而不是http.CloseNotifier来实现断开连接

使用CloseNotifier的代码，在收到通知时直接返回：

// Listen to connection close and un-register messageChan
notify := w.(http.CloseNotifier).CloseNotify()
for {
   select {
      case <-notify:
	 return
    default:
       // On message
       commands := <-messageChan

我能否简单地用Done通道替换notify通道？

done := r.Context().Done()
for {
   select {
      case <-done:
         return
      default:
        ...

我是Go语言的新手，不太确定是否完全理解"http.context"的概念。

来自法国南部的诚挚问候