From 501f7dda97657ec36c9085184ed24c083d9ce9ec Mon Sep 17 00:00:00 2001
From: Andy Wang <cbeuw.andy@gmail.com>
Date: Wed, 27 May 2020 13:41:11 +0100
Subject: [PATCH] Fix WriteTo receive loop

---
 internal/multiplex/bufferedPipe.go   | 3 +--
 internal/multiplex/datagramBuffer.go | 3 +--
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/internal/multiplex/bufferedPipe.go b/internal/multiplex/bufferedPipe.go
index ea3f87e..7652cc9 100644
--- a/internal/multiplex/bufferedPipe.go
+++ b/internal/multiplex/bufferedPipe.go
@@ -92,9 +92,8 @@ func (p *bufferedPipe) WriteTo(w io.Writer) (n int64, err error) {
 				return n, er
 			}
 			p.rwCond.Broadcast()
-		} else {
-			p.rwCond.Wait()
 		}
+		p.rwCond.Wait()
 	}
 }
 
diff --git a/internal/multiplex/datagramBuffer.go b/internal/multiplex/datagramBuffer.go
index e802e0a..c3ea771 100644
--- a/internal/multiplex/datagramBuffer.go
+++ b/internal/multiplex/datagramBuffer.go
@@ -98,9 +98,8 @@ func (d *datagramBuffer) WriteTo(w io.Writer) (n int64, err error) {
 				return n, er
 			}
 			d.rwCond.Broadcast()
-		} else {
-			d.rwCond.Wait()
 		}
+		d.rwCond.Wait()
 	}
 }