From ae3f97df1e264f955d9a258b3f2b3cb3987096b0 Mon Sep 17 00:00:00 2001
From: Andy Wang <cbeuw.andy@gmail.com>
Date: Sat, 23 May 2020 22:22:58 +0100
Subject: [PATCH] Rearrange if statement

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

diff --git a/internal/multiplex/bufferedPipe.go b/internal/multiplex/bufferedPipe.go
index 7652cc9..ea3f87e 100644
--- a/internal/multiplex/bufferedPipe.go
+++ b/internal/multiplex/bufferedPipe.go
@@ -92,8 +92,9 @@ 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 630b2c9..e802e0a 100644
--- a/internal/multiplex/datagramBuffer.go
+++ b/internal/multiplex/datagramBuffer.go
@@ -98,9 +98,9 @@ func (d *datagramBuffer) WriteTo(w io.Writer) (n int64, err error) {
 				return n, er
 			}
 			d.rwCond.Broadcast()
-			continue
+		} else {
+			d.rwCond.Wait()
 		}
-		d.rwCond.Wait()
 	}
 }