From d9fa79b73ca25afcdb89df52aac36ed25f9eab52 Mon Sep 17 00:00:00 2001 From: unclejack Date: Wed, 1 Apr 2015 00:23:04 +0300 Subject: [PATCH] pkg/broadcastwriter: reset after 4 KB w/o stream Signed-off-by: Cristian Staretu --- broadcastwriter/broadcastwriter.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/broadcastwriter/broadcastwriter.go b/broadcastwriter/broadcastwriter.go index 248cd8f..bd9b675 100644 --- a/broadcastwriter/broadcastwriter.go +++ b/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 }