From b7d022702ec0a3cff99768f07df000e0190b237d Mon Sep 17 00:00:00 2001 From: Darien Raymond Date: Sun, 26 Nov 2017 15:36:05 +0100 Subject: [PATCH] warn user when http proxy is not finished properly --- proxy/http/server.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/proxy/http/server.go b/proxy/http/server.go index 520a3a78..9e674950 100644 --- a/proxy/http/server.go +++ b/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 })