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

unclejack · unclejack · commit 153f98bad51e · 2015-04-01T01:26:19.000+03:00
Signed-off-by: Cristian Staretu &lt;cristian.staretu@gmail.com&gt;
diff --git 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
 		}

Original file line number	Diff line number	Diff line change
`@@ -42,9 +42,12 @@ func (w *BroadcastWriter) Write(p []byte) (n int, err error) {`
`42`	`42`	`delete(writers, sw)`
`43`	`43`	`}`
`44`	`44`	`}`
`45`		`- // exit if there is no more writers`
`46`	`45`	`if len(w.streams) == 1 {`
`47`		`- w.buf.Reset()`
	`46`	`+ if w.buf.Len() >= 4096 {`
	`47`	`+ w.buf.Reset()`
	`48`	`+ } else {`
	`49`	`+ w.buf.Write(p)`
	`50`	`+ }`
`48`	`51`	`w.Unlock()`
`49`	`52`	`return len(p), nil`
`50`	`53`	`}`