1
0
mirror of https://github.com/redis/go-redis.git synced 2025-07-31 05:04:23 +03:00

Add test for reading many keys.

This commit is contained in:
Vladimir Mihailenco
2013-03-21 12:28:19 +02:00
parent ef0cc25b9b
commit f14cf3644b
2 changed files with 15 additions and 6 deletions

View File

@ -74,14 +74,14 @@ func readLine(rd reader) ([]byte, error) {
}
func readN(rd reader, n int) ([]byte, error) {
buf, err := rd.ReadN(n)
b, err := rd.ReadN(n)
if err == bufio.ErrBufferFull {
newBuf := make([]byte, n)
r := copy(newBuf, buf)
buf = newBuf
newB := make([]byte, n)
r := copy(newB, b)
b = newB
for {
nn, err := rd.Read(buf[r:])
nn, err := rd.Read(b[r:])
r += nn
if r >= n {
// Ignore error if we read enough.
@ -94,7 +94,7 @@ func readN(rd reader, n int) ([]byte, error) {
} else if err != nil {
return nil, err
}
return buf, nil
return b, nil
}
//------------------------------------------------------------------------------