From b6e65d498dd66e3919b2eb74364094c804f13567 Mon Sep 17 00:00:00 2001
From: Josh Baker <joshbaker77@gmail.com>
Date: Thu, 9 Nov 2017 05:50:32 -0700
Subject: [PATCH] lil perf boost

---
 append.go | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/append.go b/append.go
index 2c4ea71..422f24d 100644
--- a/append.go
+++ b/append.go
@@ -229,6 +229,16 @@ func readTelnetCommand(packet []byte, argsbuf [][]byte) (
 	return false, args[:0], Telnet, packet, nil
 }
 
+// appendPrefix will append a "$3\r\n" style redis prefix for a message.
+func appendPrefix(b []byte, c byte, n int64) []byte {
+	if n >= 0 && n <= 9 {
+		return append(b, c, byte('0'+n), '\r', '\n')
+	}
+	b = append(b, c)
+	b = strconv.AppendInt(b, n, 10)
+	return append(b, '\r', '\n')
+}
+
 // AppendUint appends a Redis protocol uint64 to the input bytes.
 func AppendUint(b []byte, n uint64) []byte {
 	b = append(b, ':')
@@ -238,32 +248,24 @@ func AppendUint(b []byte, n uint64) []byte {
 
 // AppendInt appends a Redis protocol int64 to the input bytes.
 func AppendInt(b []byte, n int64) []byte {
-	b = append(b, ':')
-	b = strconv.AppendInt(b, n, 10)
-	return append(b, '\r', '\n')
+	return appendPrefix(b, ':', n)
 }
 
 // AppendArray appends a Redis protocol array to the input bytes.
 func AppendArray(b []byte, n int) []byte {
-	b = append(b, '*')
-	b = strconv.AppendInt(b, int64(n), 10)
-	return append(b, '\r', '\n')
+	return appendPrefix(b, '*', int64(n))
 }
 
 // AppendBulk appends a Redis protocol bulk byte slice to the input bytes.
 func AppendBulk(b []byte, bulk []byte) []byte {
-	b = append(b, '$')
-	b = strconv.AppendInt(b, int64(len(bulk)), 10)
-	b = append(b, '\r', '\n')
+	b = appendPrefix(b, '$', int64(len(bulk)))
 	b = append(b, bulk...)
 	return append(b, '\r', '\n')
 }
 
 // AppendBulkString appends a Redis protocol bulk string to the input bytes.
 func AppendBulkString(b []byte, bulk string) []byte {
-	b = append(b, '$')
-	b = strconv.AppendInt(b, int64(len(bulk)), 10)
-	b = append(b, '\r', '\n')
+	b = appendPrefix(b, '$', int64(len(bulk)))
 	b = append(b, bulk...)
 	return append(b, '\r', '\n')
 }