add error check to scanner

bcicen · Dec 6, 2017 · 336252b · 336252b
1 parent d2b7a8c
commit 336252b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -36,6 +36,7 @@ func (s *StdinScanner) StreamBytes() chan []byte {
 				fmt.Printf("%s", b)
 			}
 		}
+		failOnError(s.Err(), "error reading input")
 		close(ch)
 	}()
 	return ch
@@ -51,6 +52,7 @@ func (s *StdinScanner) StreamLines() chan string {
 				fmt.Println(s.Text())
 			}
 		}
+		failOnError(s.Err(), "error reading input")
 		close(ch)
 	}()
 	return ch