add binary support for pubsub

2011-01-24 13:24:08 -03:00
parent 7150c5feb3
commit 4d5e5a7c5d
6 changed files with 440 additions and 31 deletions
--- a/src/main/java/redis/clients/jedis/BinaryJedisPubSub.java
+++ b/src/main/java/redis/clients/jedis/BinaryJedisPubSub.java
@@ -0,0 +1,112 @@
+package redis.clients.jedis;
+
+import static redis.clients.jedis.Protocol.Keyword.MESSAGE;
+import static redis.clients.jedis.Protocol.Keyword.PMESSAGE;
+import static redis.clients.jedis.Protocol.Keyword.PSUBSCRIBE;
+import static redis.clients.jedis.Protocol.Keyword.PUNSUBSCRIBE;
+import static redis.clients.jedis.Protocol.Keyword.SUBSCRIBE;
+import static redis.clients.jedis.Protocol.Keyword.UNSUBSCRIBE;
+
+import java.util.Arrays;
+import java.util.List;
+
+public abstract class BinaryJedisPubSub {
+    private int subscribedChannels = 0;
+    private Client client;
+
+    public abstract void onMessage(byte[] channel, byte[] message);
+
+    public abstract void onPMessage(byte[] pattern, byte[] channel,
+            byte[] message);
+
+    public abstract void onSubscribe(byte[] channel, int subscribedChannels);
+
+    public abstract void onUnsubscribe(byte[] channel, int subscribedChannels);
+
+    public abstract void onPUnsubscribe(byte[] pattern, int subscribedChannels);
+
+    public abstract void onPSubscribe(byte[] pattern, int subscribedChannels);
+
+    public void unsubscribe() {
+        client.unsubscribe();
+    }
+
+    public void unsubscribe(byte[]... channels) {
+        client.unsubscribe(channels);
+    }
+
+    public void subscribe(byte[]... channels) {
+        client.subscribe(channels);
+    }
+
+    public void psubscribe(byte[]... patterns) {
+        client.psubscribe(patterns);
+    }
+
+    public void punsubscribe() {
+        client.punsubscribe();
+    }
+
+    public void punsubscribe(byte[]... patterns) {
+        client.punsubscribe(patterns);
+    }
+
+    public boolean isSubscribed() {
+        return subscribedChannels > 0;
+    }
+
+    public void proceedWithPatterns(Client client, byte[]... patterns) {
+        this.client = client;
+        client.psubscribe(patterns);
+        process(client);
+    }
+
+    public void proceed(Client client, byte[]... channels) {
+        this.client = client;
+        client.subscribe(channels);
+        process(client);
+    }
+
+    private void process(Client client) {
+        do {
+            List<Object> reply = client.getObjectMultiBulkReply();
+            final Object firstObj = reply.get(0);
+            if (!(firstObj instanceof byte[])) {
+                throw new JedisException("Unknown message type: " + firstObj);
+            }
+            final byte[] resp = (byte[]) firstObj;
+            if (Arrays.equals(SUBSCRIBE.raw, resp)) {
+                subscribedChannels = ((Long) reply.get(2)).intValue();
+                final byte[] bchannel = (byte[]) reply.get(1);
+                onSubscribe(bchannel, subscribedChannels);
+            } else if (Arrays.equals(UNSUBSCRIBE.raw, resp)) {
+                subscribedChannels = ((Long) reply.get(2)).intValue();
+                final byte[] bchannel = (byte[]) reply.get(1);
+                onUnsubscribe(bchannel, subscribedChannels);
+            } else if (Arrays.equals(MESSAGE.raw, resp)) {
+                final byte[] bchannel = (byte[]) reply.get(1);
+                final byte[] bmesg = (byte[]) reply.get(2);
+                onMessage(bchannel, bmesg);
+            } else if (Arrays.equals(PMESSAGE.raw, resp)) {
+                final byte[] bpattern = (byte[]) reply.get(1);
+                final byte[] bchannel = (byte[]) reply.get(2);
+                final byte[] bmesg = (byte[]) reply.get(3);
+                onPMessage(bpattern, bchannel, bmesg);
+            } else if (Arrays.equals(PSUBSCRIBE.raw, resp)) {
+                subscribedChannels = ((Long) reply.get(2)).intValue();
+                final byte[] bpattern = (byte[]) reply.get(1);
+                onPSubscribe(bpattern, subscribedChannels);
+            } else if (Arrays.equals(PUNSUBSCRIBE.raw, resp)) {
+                subscribedChannels = ((Long) reply.get(2)).intValue();
+                final byte[] bpattern = (byte[]) reply.get(1);
+                onPUnsubscribe(bpattern, subscribedChannels);
+            } else {
+                throw new JedisException("Unknown message type: " + firstObj);
+            }
+        } while (isSubscribed());
+    }
+
+    public int getSubscribedChannels() {
+        return subscribedChannels;
+    }
+}