From 8b5a2b91620538a67dd8adf5186895fed1a7ae01 Mon Sep 17 00:00:00 2001 From: Carl Jackson Date: Sun, 31 Aug 2014 19:56:16 -0700 Subject: [PATCH] Support Tee() in the ReadFrom case as well --- web/util/writer_proxy.go | 3 +++ 1 file changed, 3 insertions(+) diff --git a/web/util/writer_proxy.go b/web/util/writer_proxy.go index 99e7ce0..3ba08de 100644 --- a/web/util/writer_proxy.go +++ b/web/util/writer_proxy.go @@ -104,6 +104,9 @@ func (f *fancyWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) { return hj.Hijack() } func (f *fancyWriter) ReadFrom(r io.Reader) (int64, error) { + if f.basicWriter.tee != nil { + return io.Copy(&f.basicWriter, r) + } rf := f.basicWriter.ResponseWriter.(io.ReaderFrom) f.basicWriter.maybeWriteHeader() return rf.ReadFrom(r)