Added pubsub support

This commit is contained in:
Jonathan Leibiusky
2010-08-07 17:54:14 -03:00
parent 4c5e8d9ff8
commit 2a1ca391e0
5 changed files with 315 additions and 0 deletions

View File

@@ -407,4 +407,32 @@ public class Client extends Connection {
public void auth(String password) {
sendCommand("AUTH", password);
}
public void subscribe(String... channels) {
sendCommand("SUBSCRIBE", channels);
}
public void publish(String channel, String message) {
sendCommand("PUBLISH", channel, message);
}
public void unsubscribe() {
sendCommand("UNSUBSCRIBE");
}
public void unsubscribe(String... channels) {
sendCommand("UNSUBSCRIBE", channels);
}
public void psubscribe(String[] patterns) {
sendCommand("PSUBSCRIBE", patterns);
}
public void punsubscribe() {
sendCommand("PUNSUBSCRIBE");
}
public void punsubscribe(String... patterns) {
sendCommand("PUNSUBSCRIBE", patterns);
}
}

View File

@@ -533,4 +533,16 @@ public class Jedis {
jedisPipeline.execute();
return client.getAll();
}
public void subscribe(JedisPubSub jedisPubSub, String... channels) {
jedisPubSub.proceed(client, channels);
}
public void publish(String channel, String message) {
client.publish(channel, message);
}
public void psubscribe(JedisPubSub jedisPubSub, String... patterns) {
jedisPubSub.proceedWithPatterns(client, patterns);
}
}

View File

@@ -0,0 +1,84 @@
package redis.clients.jedis;
import java.util.List;
public abstract class JedisPubSub {
private int subscribedChannels = 0;
private Client client;
public abstract void onMessage(String channel, String message);
public abstract void onPMessage(String pattern, String channel,
String message);
public abstract void onSubscribe(String channel, int subscribedChannels);
public abstract void onUnsubscribe(String channel, int subscribedChannels);
public abstract void onPUnsubscribe(String pattern, int subscribedChannels);
public abstract void onPSubscribe(String pattern, int subscribedChannels);
protected void unsubscribe() {
client.unsubscribe();
}
protected void unsubscribe(String... channels) {
client.unsubscribe(channels);
}
protected void subscribe(String... channels) {
client.subscribe(channels);
}
public boolean isSubscribed() {
return subscribedChannels > 0;
}
public void proceedWithPatterns(Client client, String... patterns) {
this.client = client;
client.psubscribe(patterns);
process(client);
}
public void proceed(Client client, String... channels) {
this.client = client;
client.subscribe(channels);
process(client);
}
private void process(Client client) {
do {
List<Object> reply = client.getObjectMultiBulkReply();
if (reply.get(0).equals("subscribe")) {
subscribedChannels = (Integer) reply.get(2);
onSubscribe((String) reply.get(1), subscribedChannels);
} else if (reply.get(0).equals("unsubscribe")) {
subscribedChannels = (Integer) reply.get(2);
onUnsubscribe((String) reply.get(1), subscribedChannels);
} else if (reply.get(0).equals("message")) {
onMessage((String) reply.get(1), (String) reply.get(2));
} else if (reply.get(0).equals("pmessage")) {
onPMessage((String) reply.get(1), (String) reply.get(2),
(String) reply.get(3));
} else if (reply.get(0).equals("psubscribe")) {
subscribedChannels = (Integer) reply.get(2);
onPSubscribe((String) reply.get(1), subscribedChannels);
} else if (reply.get(0).equals("punsubscribe")) {
subscribedChannels = (Integer) reply.get(2);
onPUnsubscribe((String) reply.get(1), subscribedChannels);
} else {
throw new JedisException("Unknown message type: "
+ reply.get(0));
}
} while (isSubscribed());
}
protected void punsubscribe() {
client.punsubscribe();
}
protected void punsubscribe(String... patterns) {
client.punsubscribe(patterns);
}
}