rik/jlibredis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

a bit faster implementation of output stream for Redis

Browse Source
This commit is contained in:
Alex Tkachman
2010-09-11 21:15:38 +03:00
parent d4d133291e
commit a521841ff5
4 changed files with 147 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/redis/clients/jedis/Connection.java
View File

@@ -1,9 +1,8 @@
package redis.clients.jedis;
import java.io.BufferedInputStream;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import redis.clients.util.RedisOutputStream;
import java.io.*;
import java.net.Socket;
import java.net.SocketException;
import java.net.UnknownHostException;
@@ -15,7 +14,7 @@ public class Connection {
private int port = Protocol.DEFAULT_PORT;
private Socket socket;
private Protocol protocol = new Protocol();
private DataOutputStream outputStream;
private RedisOutputStream outputStream;
private DataInputStream inputStream;
private int pipelinedCommands = 0;
private int timeout = 2000;
@@ -91,9 +90,8 @@ public class Connection {
if (!isConnected()) {
socket = new Socket(host, port);
socket.setSoTimeout(timeout);
outputStream = new DataOutputStream(socket.getOutputStream());
inputStream = new DataInputStream(new BufferedInputStream(socket
.getInputStream()));
outputStream = new RedisOutputStream(socket.getOutputStream());
inputStream = new DataInputStream(new BufferedInputStream(socket.getInputStream()));
}
}

67
src/main/java/redis/clients/jedis/Protocol.java
View File

@@ -1,15 +1,23 @@
package redis.clients.jedis;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import redis.clients.util.RedisOutputStream;
import java.io.*;
import java.nio.charset.Charset;
import java.nio.charset.CharsetEncoder;
import java.util.ArrayList;
import java.util.List;
public class Protocol {
public static final Charset CHARSET = Charset.forName("UTF-8");
private static final ThreadLocal<CharsetEncoder> CHARSET_ENCODER = new ThreadLocal<CharsetEncoder> (){
@Override
protected CharsetEncoder initialValue() {
return CHARSET.newEncoder();
}
};
public static final String DOLLAR = "$";
public static final String ASTERISK = "*";
public static final String PLUS = "+";
@@ -26,29 +34,36 @@ public class Protocol {
public static final byte MINUS_BYTE = MINUS.getBytes(CHARSET)[0];
public static final byte COLON_BYTE = COLON.getBytes(CHARSET)[0];
public void sendCommand(DataOutputStream os, String name, String... args) {
StringBuilder sb = new StringBuilder();
sb.append(ASTERISK);
sb.append((new Integer(args.length + 1)).toString());
sb.append(COMMAND_DELIMITER);
sb.append(DOLLAR);
sb.append((new Integer(name.length())).toString());
sb.append(COMMAND_DELIMITER);
sb.append(name);
sb.append(COMMAND_DELIMITER);
public void sendCommand(RedisOutputStream os, String name, String... args) {
try {
final CharsetEncoder encoder = CHARSET_ENCODER.get();
for (String arg : args) {
int size = arg.getBytes(CHARSET).length;
os.write(ASTERISK_BYTE);
os.write(String.valueOf(args.length + 1), encoder);
os.write(COMMAND_DELIMITER_BYTES);
os.write(DOLLAR_BYTE);
os.write(String.valueOf(name.length()), encoder);
os.write(COMMAND_DELIMITER_BYTES);
os.write(name, encoder);
os.write(COMMAND_DELIMITER_BYTES);
sb.append(DOLLAR);
sb.append((new Integer(size)).toString());
sb.append(COMMAND_DELIMITER);
sb.append(arg);
sb.append(COMMAND_DELIMITER);
}
for (String arg : args) {
final byte[] bytes = arg.getBytes(CHARSET);
int size = bytes.length;
try {
os.write(sb.toString().getBytes(CHARSET));
os.write(DOLLAR_BYTE);
os.write(String.valueOf(size), encoder);
os.write(COMMAND_DELIMITER_BYTES);
os.write(bytes);
os.write(COMMAND_DELIMITER_BYTES);
}
os.flush ();
} catch (IOException e) {
throw new JedisException(e);
}
// try {
// os.write(os.toByteArray());
/*
os.write(ASTERISK_BYTE);
os.write((new Integer(args.length + 1)).toString()
@@ -71,9 +86,9 @@ public class Protocol {
os.write(COMMAND_DELIMITER_BYTES);
}
*/
} catch (IOException e) {
throw new JedisException(e);
}
// } catch (IOException e) {
// throw new JedisException(e);
// }
}
public void processError(DataInputStream is) {