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"
"io"
"bufio"
"errors"
)
const defaultBufSize = 65536
@ -54,14 +55,17 @@ func (preader *pipeReader) RecvMessage() []byte {
if preader.scanner.Scan() {
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)
}
return nil
}
func (preader *pipeReader) BufSize(size int) {
preader.scanner.Buffer(make([]byte, size), size)
preader.scanner.Buffer(make([]byte, defaultBufSize), size)
}