Преглед изворни кода

warn user when http proxy is not finished properly

Darien Raymond пре 8 година
родитељ
комит
b7d022702e
1 измењених фајлова са 5 додато и 2 уклоњено
  1. 5 2
      proxy/http/server.go

+ 5 - 2
proxy/http/server.go

@@ -274,7 +274,10 @@ func (s *Server) handlePlainHTTP(ctx context.Context, request *http.Request, wri
 
 		requestWriter := buf.NewBufferedWriter(ray.InboundInput())
 		common.Must(requestWriter.SetBuffered(false))
-		return request.Write(requestWriter)
+		if err := request.Write(requestWriter); err != nil {
+			return newError("failed to write whole request").Base(err).AtWarning()
+		}
+		return nil
 	})
 
 	responseDone := signal.ExecuteAsync(func() error {
@@ -308,7 +311,7 @@ func (s *Server) handlePlainHTTP(ctx context.Context, request *http.Request, wri
 			response.Header.Set("Proxy-Connection", "close")
 		}
 		if err := response.Write(writer); err != nil {
-			return newError("failed to write response").Base(err)
+			return newError("failed to write response").Base(err).AtWarning()
 		}
 		return nil
 	})