pkg/broadcastwriter: reset after 4 KB w/o stream

Signed-off-by: Cristian Staretu <cristian.staretu@gmail.com>
2015-04-01 00:23:04 +03:00 · 2015-04-01 00:23:04 +03:00 · 153f98bad5
commit 153f98bad5
parent d15d1674c3
1 changed files with 5 additions and 2 deletions
--- a/pkg/broadcastwriter/broadcastwriter.go
+++ b/pkg/broadcastwriter/broadcastwriter.go
@ -42,9 +42,12 @@ func (w *BroadcastWriter) Write(p []byte) (n int, err error) {
 				delete(writers, sw)
 			}
 		}
-		// exit if there is no more writers
 		if len(w.streams) == 1 {
-			w.buf.Reset()
+			if w.buf.Len() >= 4096 {
+				w.buf.Reset()
+			} else {
+				w.buf.Write(p)
+			}
 			w.Unlock()
 			return len(p), nil
 		}