parser: Optimize multi bulk request parsing.

req_test.go

@@ -0,0 +1,75 @@
+package redis_test
+
+import (
+	"github.com/vmihailenco/bufio"
+	. "launchpad.net/gocheck"
+
+	"github.com/vmihailenco/redis"
+)
+
+//------------------------------------------------------------------------------
+
+type LineReader struct {
+	line []byte
+}
+
+func NewLineReader(line []byte) *LineReader {
+	return &LineReader{line: line}
+}
+
+func (r *LineReader) Read(buf []byte) (int, error) {
+	return copy(buf, r.line), nil
+}
+
+//------------------------------------------------------------------------------
+
+type RequestTest struct{}
+
+var _ = Suite(&RequestTest{})
+
+//------------------------------------------------------------------------------
+
+func (t *RequestTest) SetUpTest(c *C) {}
+
+func (t *RequestTest) TearDownTest(c *C) {}
+
+//------------------------------------------------------------------------------
+
+func (t *RequestTest) benchmarkReq(c *C, reqString string, req redis.Req, checker Checker, expected interface{}) {
+	c.StopTimer()
+
+	lineReader := NewLineReader([]byte(reqString))
+	rd := bufio.NewReaderSize(lineReader, 1024)
+
+	for i := 0; i < 10; i++ {
+		vIface, err := req.ParseReply(rd)
+		c.Check(err, IsNil)
+		c.Check(vIface, checker, expected)
+		req.SetVal(vIface)
+	}
+
+	c.StartTimer()
+
+	for i := 0; i < c.N; i++ {
+		v, _ := req.ParseReply(rd)
+		req.SetVal(v)
+	}
+}
+
+func (t *RequestTest) BenchmarkStatusReq(c *C) {
+	t.benchmarkReq(c, "+OK\r\n", redis.NewStatusReq(), Equals, "OK")
+}
+
+func (t *RequestTest) BenchmarkStringReq(c *C) {
+	t.benchmarkReq(c, "$5\r\nhello\r\n", redis.NewStringReq(), Equals, "hello")
+}
+
+func (t *RequestTest) BenchmarkStringSliceReq(c *C) {
+	t.benchmarkReq(
+		c,
+		"*2\r\n$5\r\nhello\r\n$5\r\nhello\r\n",
+		redis.NewStringSliceReq(),
+		DeepEquals,
+		[]string{"hello", "hello"},
+	)
+}