Use fasthttp.PipeConns instead of io.Pipe in StreamReader

This improves StreamReader performance by more than 2x.

1 files changed, 5 insertions, 1 deletions

diff --git a/stream.go b/stream.go
index 349bd5b..24bd553 100644
--- a/stream.go
+++ b/stream.go
@@ -5,6 +5,8 @@ import (
 	"io"
 	"runtime/debug"
 	"sync"
+
+	"github.com/valyala/fasthttp/fasthttputil"
 )
 
 // StreamWriter must write data to w.
@@ -26,7 +28,9 @@ type StreamWriter func(w *bufio.Writer)
 //
 // See also Response.SetBodyStreamWriter.
 func NewStreamReader(sw StreamWriter) io.ReadCloser {
-	pr, pw := io.Pipe()
+	pc := fasthttputil.NewPipeConns()
+	pw := pc.Conn1()
+	pr := pc.Conn2()
 
 	var bw *bufio.Writer
 	v := streamWriterBufPool.Get()