Fix slice grow in readN.

GetStream · Mar 14, 2016 · 672fb9b · 672fb9b
1 parent 4665ad8
commit 672fb9b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/command_test.go b/command_test.go
@@ -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))
 	})
 

diff --git a/parser.go b/parser.go
@@ -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]