Browse Source

Support Tee() in the ReadFrom case as well

Carl Jackson 11 years ago
parent
commit
8b5a2b9162
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      web/util/writer_proxy.go

+ 3
- 0
web/util/writer_proxy.go View File

@ -104,6 +104,9 @@ func (f *fancyWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
return hj.Hijack() return hj.Hijack()
} }
func (f *fancyWriter) ReadFrom(r io.Reader) (int64, error) { 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) rf := f.basicWriter.ResponseWriter.(io.ReaderFrom)
f.basicWriter.maybeWriteHeader() f.basicWriter.maybeWriteHeader()
return rf.ReadFrom(r) return rf.ReadFrom(r)


Loading…
Cancel
Save