Merge pull request #60 from timothegenzmer/fix-large-input

scan stdin by byte instead of line
mercari · Mar 10, 2020 · f863e19 · f863e19
2 parents c0d0933 + 44c9603
commit f863e19
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tee.go b/tee.go
@@ -3,7 +3,6 @@ package main
 import (
 	"bufio"
 	"bytes"
-	"fmt"
 	"io"
 
 	"github.com/mattn/go-colorable"
@@ -15,8 +14,9 @@ func tee(stdin io.Reader, stdout io.Writer) string {
 
 	tee := io.TeeReader(stdin, &b1)
 	s := bufio.NewScanner(tee)
+	s.Split(bufio.ScanBytes)
 	for s.Scan() {
-		fmt.Fprintln(stdout, s.Text())
+		stdout.Write(s.Bytes())
 	}
 
 	uncolorize := colorable.NewNonColorable(&b2)