Add test for reading many keys.

parser.go

@@ -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)
+		newB := make([]byte, n)
-		r := copy(newBuf, buf)
+		r := copy(newB, b)
-		buf = newBuf
+		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
 }
 
 //------------------------------------------------------------------------------

redis_test.go

@@ -318,6 +318,15 @@ func (t *RedisTest) TestGetBigVal(c *C) {
 	c.Assert(get.Val(), Equals, val)
 }
 
+func (t *RedisTest) TestManyKeys(c *C) {
+	for i := 0; i < 100000; i++ {
+		t.client.Set("keys.key"+strconv.Itoa(i), "hello")
+	}
+	keys := t.client.Keys("keys.*")
+	c.Assert(keys.Err(), IsNil)
+	c.Assert(len(keys.Val()), Equals, 100000)
+}
+
 //------------------------------------------------------------------------------
 
 func (t *RedisTest) TestConnPoolRemovesBrokenConn(c *C) {