fix CloseNotify called after ServeHTTP finished bug.

yunify · Nov 2, 2016 · 7bfa598 · 7bfa598
1 parent 600c5aa
commit 7bfa598
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/metad.go b/metad.go
@@ -550,9 +550,10 @@ func (m *Metad) handleWrapper(handler handleFunc) func(w http.ResponseWriter, re
 		ctx := context.WithValue(req.Context(), "requestID", requestID)
 		cancelCtx, cancelFun := context.WithCancel(ctx)
 		if x, ok := w.(http.CloseNotifier); ok {
+			closeNotify := x.CloseNotify()
 			go func() {
 				select {
-				case <-x.CloseNotify():
+				case <-closeNotify:
 					cancelFun()
 				}
 			}()