re
/
redis
forked from mirror/redis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix slice grow in readN.

This commit is contained in:
Vladimir Mihailenco 2016-03-14 12:36:18 +02:00
parent 4665ad860f
commit 672fb9bb97
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
command_test.go
View File

@ -15,10 +15,14 @@ import (
var _ = Describe("Command", func() {
var client *redis.Client
BeforeEach(func() {
client = redis.NewClient(&redis.Options{
connect := func() *redis.Client {
return redis.NewClient(&redis.Options{
Addr: redisAddr,
})
}
BeforeEach(func() {
client = connect()
})
AfterEach(func() {
@ -63,8 +67,13 @@ var _ = Describe("Command", func() {
Expect(set.Err()).NotTo(HaveOccurred())
Expect(set.Val()).To(Equal("OK"))
// Reconnect to get new connection.
Expect(client.Close()).To(BeNil())
client = connect()
get := client.Get("key")
Expect(get.Err()).NotTo(HaveOccurred())
Expect(len(get.Val())).To(Equal(len(val)))
Expect(get.Val()).To(Equal(val))
})

1
parser.go
View File

@ -247,6 +247,7 @@ func isNilReply(b []byte) bool {
func readN(cn *pool.Conn, n int) ([]byte, error) {
if d := n - cap(cn.Buf); d > 0 {
cn.Buf = cn.Buf[:cap(cn.Buf)]
cn.Buf = append(cn.Buf, make([]byte, d)...)
} else {
cn.Buf = cn.Buf[:n]