Change example

acomagu · Jun 18, 2019 · 8cc31fa · 8cc31fa
1 parent 9ef64a6
commit 8cc31fa
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -21,10 +21,16 @@ Reads block if the internal buffer is empty until the writer is closed.
 
 ```Go
 r, w := bufpipe.New(nil)
-go io.Copy(os.Stdout, r) // The reads block until the writer is closed.
+
+done := make(chan struct{})
+go func() {
+	io.Copy(os.Stdout, r) // The reads block until the writer is closed.
+	done <- struct{}{}
+}()
 
 io.WriteString(w, "abc")
 io.WriteString(w, "def")
 w.Close()
+<-done
 // Output: abcdef
 ```
diff --git a/bufpipe_example_test.go b/bufpipe_example_test.go
@@ -9,9 +9,16 @@ import (
 
 func Example() {
 	r, w := bufpipe.New(nil)
-	go io.Copy(os.Stdout, r) // => abcdef.
+
+	done := make(chan struct{})
+	go func() {
+		io.Copy(os.Stdout, r)
+		done <- struct{}{}
+	}()
 
 	io.WriteString(w, "abc")
 	io.WriteString(w, "def")
 	w.Close()
+	<-done
+	// Output: abcdef
 }