warn user when http proxy is not finished properly

pull/787/head
Darien Raymond 2017-11-26 15:36:05 +01:00
parent e8901368d7
commit b7d022702e
No known key found for this signature in database
GPG Key ID: 7251FFA14BB18169
1 changed files with 5 additions and 2 deletions

View File

@ -274,7 +274,10 @@ func (s *Server) handlePlainHTTP(ctx context.Context, request *http.Request, wri
requestWriter := buf.NewBufferedWriter(ray.InboundInput()) requestWriter := buf.NewBufferedWriter(ray.InboundInput())
common.Must(requestWriter.SetBuffered(false)) 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 { 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") response.Header.Set("Proxy-Connection", "close")
} }
if err := response.Write(writer); err != nil { 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 return nil
}) })