removes extra upgrade aware proxy logging; returns tunneling connection close error

This commit is contained in:
Sean Sullivan 2024-03-02 16:10:39 -08:00
parent 2443b3fa69
commit e8bbb221d3
2 changed files with 9 additions and 16 deletions

View File

@ -422,7 +422,7 @@ func (h *UpgradeAwareHandler) tryUpgrade(w http.ResponseWriter, req *http.Reques
} else { } else {
writer = backendConn writer = backendConn
} }
_, err := io.Copy(writer, &loggingReader{name: "client->backend", delegate: requestHijackedConn}) _, err := io.Copy(writer, requestHijackedConn)
if err != nil && !strings.Contains(err.Error(), "use of closed network connection") { if err != nil && !strings.Contains(err.Error(), "use of closed network connection") {
klog.Errorf("Error proxying data from client to backend: %v", err) klog.Errorf("Error proxying data from client to backend: %v", err)
} }
@ -436,7 +436,7 @@ func (h *UpgradeAwareHandler) tryUpgrade(w http.ResponseWriter, req *http.Reques
} else { } else {
reader = backendConn reader = backendConn
} }
_, err := io.Copy(requestHijackedConn, &loggingReader{name: "backend->client", delegate: reader}) _, err := io.Copy(requestHijackedConn, reader)
if err != nil && !strings.Contains(err.Error(), "use of closed network connection") { if err != nil && !strings.Contains(err.Error(), "use of closed network connection") {
klog.Errorf("Error proxying data from backend to client: %v", err) klog.Errorf("Error proxying data from backend to client: %v", err)
} }
@ -454,18 +454,6 @@ func (h *UpgradeAwareHandler) tryUpgrade(w http.ResponseWriter, req *http.Reques
return true return true
} }
// loggingReader logs the bytes read from the "delegate" with a "name" prefix.
type loggingReader struct {
name string
delegate io.Reader
}
func (l *loggingReader) Read(p []byte) (int, error) {
n, err := l.delegate.Read(p)
klog.V(8).Infof("%s: %d bytes, err=%v, bytes=% X", l.name, n, err, p[:n])
return n, err
}
// FIXME: Taken from net/http/httputil/reverseproxy.go as singleJoiningSlash is not exported to be re-used. // FIXME: Taken from net/http/httputil/reverseproxy.go as singleJoiningSlash is not exported to be re-used.
// See-also: https://github.com/golang/go/issues/44290 // See-also: https://github.com/golang/go/issues/44290
func singleJoiningSlash(a, b string) string { func singleJoiningSlash(a, b string) string {

View File

@ -114,8 +114,13 @@ func (c *TunnelingConnection) Close() error {
klog.V(7).Infof("%s: tunneling connection Close()...", c.name) klog.V(7).Infof("%s: tunneling connection Close()...", c.name)
// Signal other endpoint that websocket connection is closing; ignore error. // Signal other endpoint that websocket connection is closing; ignore error.
normalCloseMsg := gwebsocket.FormatCloseMessage(gwebsocket.CloseNormalClosure, "") normalCloseMsg := gwebsocket.FormatCloseMessage(gwebsocket.CloseNormalClosure, "")
c.conn.WriteControl(gwebsocket.CloseMessage, normalCloseMsg, time.Now().Add(time.Second)) //nolint:errcheck writeControlErr := c.conn.WriteControl(gwebsocket.CloseMessage, normalCloseMsg, time.Now().Add(time.Second))
err = c.conn.Close() closeErr := c.conn.Close()
if closeErr != nil {
err = closeErr
} else if writeControlErr != nil {
err = writeControlErr
}
}) })
return err return err
} }