From bc60cace70078944ea64699aa99e974a55ee50b6 Mon Sep 17 00:00:00 2001
From: Ivan Krutov <vania-pooh@vania-pooh.com>
Date: Thu, 15 Nov 2018 19:34:17 +0300
Subject: [PATCH] Removed deprecated CloseNotifier

---
 proxy.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/proxy.go b/proxy.go
index 0058333..7221334 100644
--- a/proxy.go
+++ b/proxy.go
@@ -512,7 +512,7 @@ func withCloseNotifier(handler http.HandlerFunc) http.HandlerFunc {
 			cancel()
 		}()
 		select {
-		case <-w.(http.CloseNotifier).CloseNotify():
+		case <-r.Context().Done():
 			cancel()
 		case <-ctx.Done():
 		}