Redis是最常見的緩存服務(wù)中間件，在java開發(fā)中，一般使用 jedis 來實(shí)現(xiàn)。
如果不想依賴第三方組件，自己實(shí)現(xiàn)一個(gè)簡單的redis客戶端工具，該如何實(shí)現(xiàn)呢？本文就是介紹這樣一種方法。
Redis的協(xié)議非常簡單，而且輸入數(shù)據(jù)和輸出數(shù)據(jù)都遵循統(tǒng)一的協(xié)議，具體規(guī)則參考這里：
http://redisdoc.com/topic/protocol.html
Redis的命令協(xié)議：
$參數(shù)數(shù)量n
$參數(shù)1的值的字節(jié)數(shù)組長度
$參數(shù)1的值的字符串表示
$參數(shù)2的值的字節(jié)數(shù)組長度
$參數(shù)2的值的字符串表示
...
$參數(shù)n的值的字節(jié)數(shù)組長度
$參數(shù)n的值的字符串表示
Redis的返回協(xié)議：
1、狀態(tài)回復(fù)（status reply）的第一個(gè)字節(jié)是 "+"，單行字符串；
2、錯(cuò)誤回復(fù)（error reply）的第一個(gè)字節(jié)是 "-"；
3、整數(shù)回復(fù)（integer reply）的第一個(gè)字節(jié)是 ":"；
4、批量回復(fù)（bulk reply）的第一個(gè)字節(jié)是 "$"；
5、多條批量回復(fù)（multi bulk reply）的第一個(gè)字節(jié)是 "*"；
6、所有的命令都是以 \r\n 結(jié)尾。
Java代碼說明
針對上述規(guī)則，我們用兩個(gè)類來實(shí)現(xiàn)：
1、SimpleRedisClient類，主要用于發(fā)送請求，并讀取響應(yīng)結(jié)果（字符串）；
整體比較簡單，稍微復(fù)雜點(diǎn)的地方就是讀取流數(shù)據(jù)，遇到兩種情況就該結(jié)束循環(huán)，一是返回長度為-1，二是返回字符串以 \r\n 結(jié)尾。
如果處理不當(dāng)，可能會(huì)導(dǎo)致 read 阻塞，Socket卡住。
2、SimpleRedisData類，用于解析響應(yīng)結(jié)果，把redis統(tǒng)一協(xié)議的字符串，解析為具體的對象。
這部分代碼完全是按照協(xié)議規(guī)則來實(shí)現(xiàn)的，通過一個(gè)游標(biāo) pos 來向前移動(dòng)，在移動(dòng)過程中識別不同格式的數(shù)據(jù)。
最復(fù)雜的是 list 類型的數(shù)據(jù)，以 * 開頭，后面跟著一個(gè)整數(shù)，表示列表中所有元素的數(shù)量，然后就是每一個(gè)列表元素的值，循環(huán)解析即可。
package demo;
import java.io.Closeable;
import java.io.IOException;
import java.net.Socket;
import java.util.List;
public class SimpleRedisClient implements Closeable {
private String host;
private int port;
private String auth;
private Socket socket = null;
public SimpleRedisClient(String host, int port, String auth) {
this.host = host;
this.port = port;
this.auth = auth;
try {
socket = new Socket(this.host, this.port);
socket.setSoTimeout(8 * 1000);//8秒
} catch (Exception ex) {
socket = null;
ex.printStackTrace();
}
}
public boolean connect() throws IOException {
if (socket == null || auth == null || auth.length() = 0) {
return false;
}
String response = execute("AUTH", auth);
if (response == null || response.length() = 0) {
return false;
}
String res = new SimpleRedisData(response).getString();
return "OK".compareTo(res) == 0;
}
@Override
public void close() {
try {
if (socket != null) {
socket.shutdownOutput();
socket.close();
}
//System.out.println("closed");
} catch (Exception ex) {
ex.printStackTrace();
}
}
public String getString(String key) {
if (socket == null || key == null || key.isEmpty()) {
return null;
}
try {
String response = execute("GET", key);
return new SimpleRedisData(response).getString();
} catch (Exception ex) {
ex.printStackTrace();
return null;
}
}
public String setString(String key, String value) {
if (socket == null || key == null || key.isEmpty()) {
return null;
}
try {
String response = execute("SET", key, value);
return new SimpleRedisData(response).getString();
} catch (Exception ex) {
ex.printStackTrace();
return null;
}
}
public String deleteKey(String key) throws IOException {
if (socket == null || key == null || key.isEmpty()) {
return null;
}
String response = execute("DEL", key);
return new SimpleRedisData(response).getString();
}
public ListString> getKeys(String pattern) throws IOException {
if (socket == null || pattern == null || pattern.isEmpty()) {
return null;
}
String response = execute("KEYS", pattern);
return new SimpleRedisData(response).getStringList();
}
public String execute(String... args) throws IOException {
if (socket == null || args == null || args.length = 0) {
return null;
}
//System.out.println(StringUtil.join(args, " "));
StringBuilder request = new StringBuilder();
request.append("*" + args.length).append("\r\n");//參數(shù)的數(shù)量
for (int i = 0; i args.length; i++) {
request.append("$" + args[i].getBytes("utf8").length).append("\r\n");//參數(shù)的長度
request.append(args[i]).append("\r\n");//參數(shù)的內(nèi)容
}
socket.getOutputStream().write(request.toString().getBytes());
socket.getOutputStream().flush();
StringBuilder reply = new StringBuilder();
int bufSize = 1024;
while (true) {
byte[] buf = new byte[bufSize];
int len = socket.getInputStream().read(buf);
if (len 0) {
break;
}
String str = new String(buf, 0, len);
reply.append(str);
if (str.endsWith("\r\n")) {
break;
}
}
String response = reply.toString();
//System.out.println("response: " + response);
return response;
}
}
package demo;
import java.util.ArrayList;
import java.util.List;
public class SimpleRedisData {
public SimpleRedisData(String rawData) {
this.rawData = rawData;
//System.out.println(rawData);
}
private int pos;
private String rawData;
public String getString() {
if (rawData == null || rawData.length() = 0) {
return null;
}
int i = rawData.indexOf("\r\n", pos);
if (i = 0) {
return null;
}
char c = rawData.charAt(pos);
if (c == '+') {
int from = pos + 1;
int to = i;
String v = rawData.substring(from, to);
pos = to + 2;
return v;
} else if (c == '-') {
int from = pos + 1;
int to = i;
String v = rawData.substring(from, to);
pos = to + 2;
return v;
} else if (c == ':') {
int from = pos + 1;
int to = i;
String v = rawData.substring(from, to);
pos = to + 2;
return v;
} else if (c == '$') {
int from = pos + 1;
int to = i;
int bulkSize = Integer.parseInt(rawData.substring(from, to));
pos = to + 2;
from = pos;
to = pos + bulkSize;
try {
//$符號后面的數(shù)值是指內(nèi)容的字節(jié)長度，而不是字符數(shù)量，所以要轉(zhuǎn)換為二進(jìn)制字節(jié)數(shù)組，再取指定長度的數(shù)據(jù)
byte[] buf = rawData.substring(from).getBytes("utf-8");
String v = new String(buf, 0, bulkSize);
pos = to + 2;
return v;
} catch (Exception ex) {
ex.printStackTrace();
return null;
}
} else {
return null;
}
}
public ListString> getStringList() {
if (rawData == null || rawData.length() = 0) {
return null;
}
int i = rawData.indexOf("\r\n", pos);
if (i = 0) {
return null;
}
char c = rawData.charAt(pos);
if (c == '*') {
ListString> values = new ArrayList>();
int from = pos + 1;
int to = i;
int multSize = Integer.parseInt(rawData.substring(from, to));
pos = to + 2;
for (int index = 0; index multSize; index++) {
values.add(getString());
}
return values;
} else {
return null;
}
}
}
package demo;
import org.junit.jupiter.api.Test;
import java.util.List;
public class RedisTest {
@Test
public void test() {
SimpleRedisClient client = null;
try {
client = new SimpleRedisClient("127.0.0.1", 6379, "123456");
System.out.println("connected: " + client.connect());
ListString> keyList = client.getKeys("api_*");
for (int i = 0; i keyList.size(); i++) {
System.out.println((i + 1) + "\t" + keyList.get(i));
}
System.out.println("keys: " + keyList != null ? keyList.size() : "null");
System.out.println(client.getString("api_getCustomerName"));
} catch (Exception ex) {
ex.printStackTrace();
} finally {
if (client != null) {
client.close();
}
}
}
}
優(yōu)點(diǎn)：
1、不依賴任何第三方組件，可以順利編譯通過；
2、代碼極其簡單。
不足之處：
1、未考慮并發(fā)訪問；
2、未提供更多的數(shù)據(jù)類型，以及讀寫方法，大家可以在此基礎(chǔ)上包裝一下。
以上就是如何用Java Socket實(shí)現(xiàn)一個(gè)簡單的Redis客戶端的詳細(xì)內(nèi)容，更多關(guān)于Java Socket Redis客戶端的資料請關(guān)注腳本之家其它相關(guān)文章！
您可能感興趣的文章:- Java Socket+多線程實(shí)現(xiàn)多人聊天室功能
- Java Socket實(shí)現(xiàn)多人聊天系統(tǒng)
- Java Socket模擬實(shí)現(xiàn)聊天室
- java Nio使用NioSocket客戶端與服務(wù)端交互實(shí)現(xiàn)方式
- 淺談java socket的正確關(guān)閉姿勢
- 解決java.net.SocketTimeoutException: Read timed out的問題
- 詳解Java Socket通信封裝MIna框架
- Java使用Socket簡單通訊詳解
