Browse Source

proxy: always set requestClosed flag when client closes the connection prematurely

Wolfgang Ebner 10 years ago
parent
commit
a6e6186477
1 changed files with 3 additions and 2 deletions
  1. 3 2
      proxy/reverse.go

+ 3 - 2
proxy/reverse.go

@@ -98,10 +98,11 @@ func (p *reverseProxy) ServeHTTP(rw http.ResponseWriter, clientreq *http.Request
 		go func() {
 			select {
 			case <-closeNotifier.CloseNotify():
+				atomic.StoreInt32(&requestClosed, 1)
+				log.Printf("proxy: client %v closed request prematurely", clientreq.RemoteAddr)
+
 				tp, ok := p.transport.(*http.Transport)
 				if ok {
-					atomic.StoreInt32(&requestClosed, 1)
-					log.Printf("proxy: request from %v canceled", clientreq.RemoteAddr)
 					tp.CancelRequest(proxyreq)
 				}
 			case <-completeCh: