mrtoth/pipe-io-server
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improve buffer management in pipeReader

Browse Source
This commit is contained in:
Kristóf Tóth 2019-04-23 10:26:20 +02:00
parent 7f08e01f39
commit 33d142ca08
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
clients/go/pipeio/pipeio.go
View File

@ -4,6 +4,7 @@ import (
"os" "os"
"io" "io"
"bufio" "bufio"
"errors"
) )
const defaultBufSize = 65536 const defaultBufSize = 65536
@ -54,14 +55,17 @@ func (preader *pipeReader) RecvMessage() []byte {
if preader.scanner.Scan() { if preader.scanner.Scan() {
return preader.scanner.Bytes() return preader.scanner.Bytes()
} }
if err := preader.scanner.Err(); err != io.EOF { if err := preader.scanner.Err(); err != nil && err != io.EOF {
if err == bufio.ErrTooLong {
err = errors.New("pipeReader buffer too small!")
}
panic(err) panic(err)
} }
return nil return nil
} }
func (preader *pipeReader) BufSize(size int) { func (preader *pipeReader) BufSize(size int) {
preader.scanner.Buffer(make([]byte, size), size) preader.scanner.Buffer(make([]byte, defaultBufSize), size)
} }