From 0a0831ff0d1938bf040a21626a0b19ea5aa6c659 Mon Sep 17 00:00:00 2001 From: Aaron Lehmann Date: Thu, 10 Sep 2015 14:58:06 -0700 Subject: [PATCH] Broadcaster writer goroutines must only finish when they are caught up Without this change, there was a narrow race condition that would allow writers to finish when there was still data left to write. This is likely to be what was causing some integration tests to fail with truncated pull output. Signed-off-by: Aaron Lehmann --- progressreader/broadcaster.go | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/progressreader/broadcaster.go b/progressreader/broadcaster.go index 429b1d0..5118e9e 100644 --- a/progressreader/broadcaster.go +++ b/progressreader/broadcaster.go @@ -77,6 +77,12 @@ func (broadcaster *Broadcaster) receiveWrites(observer io.Writer) { broadcaster.Lock() + // If we are behind, we need to catch up instead of waiting + // or handling a closure. + if len(broadcaster.history) != n { + continue + } + // detect closure of the broadcast writer if broadcaster.closed() { broadcaster.Unlock() @@ -84,9 +90,7 @@ func (broadcaster *Broadcaster) receiveWrites(observer io.Writer) { return } - if len(broadcaster.history) == n { - broadcaster.cond.Wait() - } + broadcaster.cond.Wait() // Mutex is still locked as the loop continues }