Addresses issues #779 and #775.

Conflicts: src/main/java/redis/clients/jedis/Protocol.java
2014-10-21 18:45:50 -07:00
parent 5d0f75a32f
commit 59a13121ab
4 changed files with 299 additions and 98 deletions
--- a/src/test/java/redis/clients/jedis/tests/benchmark/ProtocolBenchmark.java
+++ b/src/test/java/redis/clients/jedis/tests/benchmark/ProtocolBenchmark.java
@@ -0,0 +1,105 @@
+package redis.clients.jedis.tests.benchmark;
+
+import redis.clients.jedis.Protocol;
+import redis.clients.util.RedisInputStream;
+import redis.clients.util.RedisOutputStream;
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * Copyright (c) 2014
+ */
+public class ProtocolBenchmark {
+    private static final int TOTAL_OPERATIONS = 500000;
+
+    public static void main(String[] args) throws Exception,
+	IOException {
+	long total = 0;
+	for (int at = 0; at != 10; ++at) {
+	    long elapsed = measureInputMulti();
+	    long ops = ((1000 * 2 * TOTAL_OPERATIONS) / TimeUnit.NANOSECONDS.toMillis(elapsed));
+	    if (at >= 5) {
+		total += ops;
+	    }
+	}
+	System.out.println((total / 5) + " avg");
+
+	total = 0;
+	for (int at = 0; at != 10; ++at) {
+	    long elapsed = measureInputStatus();
+	    long ops = ((1000 * 2 * TOTAL_OPERATIONS) / TimeUnit.NANOSECONDS.toMillis(elapsed));
+	    if (at >= 5) {
+		total += ops;
+	    }
+	}
+
+	System.out.println((total / 5) + " avg");
+
+	total = 0;
+	for (int at = 0; at != 10; ++at) {
+	    long elapsed = measureCommand();
+	    long ops = ((1000 * 2 * TOTAL_OPERATIONS) / TimeUnit.NANOSECONDS.toMillis(elapsed));
+	    if (at >= 5) {
+		total += ops;
+	    }
+	}
+
+	System.out.println((total / 5) + " avg");
+    }
+
+    private static long measureInputMulti() throws Exception {
+	long duration = 0;
+
+	InputStream is = new ByteArrayInputStream(
+	    "*4\r\n$3\r\nfoo\r\n$13\r\nbarbarbarfooz\r\n$5\r\nHello\r\n$5\r\nWorld\r\n"
+		.getBytes());
+
+	RedisInputStream in = new RedisInputStream(is);
+	for (int n = 0; n <= TOTAL_OPERATIONS; n++) {
+	    long start = System.nanoTime();
+	    Protocol.read(in);
+	    duration += (System.nanoTime() - start);
+	    in.reset();
+	}
+
+	return duration;
+    }
+
+    private static long measureInputStatus() throws Exception {
+	long duration = 0;
+
+	InputStream is = new ByteArrayInputStream(
+	    "+OK\r\n"
+		.getBytes());
+
+	RedisInputStream in = new RedisInputStream(is);
+	for (int n = 0; n <= TOTAL_OPERATIONS; n++) {
+	    long start = System.nanoTime();
+	    Protocol.read(in);
+	    duration += (System.nanoTime() - start);
+	    in.reset();
+	}
+
+	return duration;
+    }
+
+    private static long measureCommand() throws Exception {
+	long duration = 0;
+
+	byte[] KEY = "123456789".getBytes();
+	byte[] VAL = "FooBar".getBytes();
+
+	for (int n = 0; n <= TOTAL_OPERATIONS; n++) {
+	    RedisOutputStream out = new RedisOutputStream(new ByteArrayOutputStream(8192));
+	    long start = System.nanoTime();
+	    Protocol.sendCommand(out, Protocol.Command.SET, KEY, VAL);
+	    duration += (System.nanoTime() - start);
+	}
+
+	return duration;
+    }
+}