http: ReponseWriter prefer ReadFrom if available (#6565)

Co-authored-by: Matt Holt <mholt@users.noreply.github.com>

WeidiDeng committed Oct 2, 2024 at 01:19 UTC f3aead0e4df57716d525f3a25fb11f70bc72b20a

1 file changed +6 -2

modules/caddyhttp/responsewriter.go

+6 -2

@@ -42,9 +42,13 @@ func (rww *ResponseWriterWrapper) Push(target string, opts *http.PushOptions) er

42 return ErrNotImplemented

43 }

45 -// ReadFrom implements io.ReaderFrom. It simply calls io.Copy,

46 -// which uses io.ReaderFrom if available.

45 +// ReadFrom implements io.ReaderFrom. It retries to use io.ReaderFrom if available,

46 +// then fallback to io.Copy.

47 +// see: https://github.com/caddyserver/caddy/issues/6546

48 func (rww *ResponseWriterWrapper) ReadFrom(r io.Reader) (n int64, err error) {

49 + if rf, ok := rww.ResponseWriter.(io.ReaderFrom); ok {

50 + return rf.ReadFrom(r)

51 + }

52 return io.Copy(rww.ResponseWriter, r)

53 }