|
|
|
package com.fr.third.redis.clients.jedis;
|
|
|
|
|
|
|
|
import static com.fr.third.redis.clients.jedis.Protocol.Keyword.MESSAGE;
|
|
|
|
import static com.fr.third.redis.clients.jedis.Protocol.Keyword.PMESSAGE;
|
|
|
|
import static com.fr.third.redis.clients.jedis.Protocol.Keyword.PSUBSCRIBE;
|
|
|
|
import static com.fr.third.redis.clients.jedis.Protocol.Keyword.PUNSUBSCRIBE;
|
|
|
|
import static com.fr.third.redis.clients.jedis.Protocol.Keyword.SUBSCRIBE;
|
|
|
|
import static com.fr.third.redis.clients.jedis.Protocol.Keyword.UNSUBSCRIBE;
|
|
|
|
import static com.fr.third.redis.clients.jedis.Protocol.Keyword.PONG;
|
|
|
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
import java.util.List;
|
|
|
|
|
|
|
|
import com.fr.third.redis.clients.jedis.exceptions.JedisConnectionException;
|
|
|
|
import com.fr.third.redis.clients.jedis.exceptions.JedisException;
|
|
|
|
import com.fr.third.redis.clients.jedis.util.SafeEncoder;
|
|
|
|
|
|
|
|
public abstract class JedisPubSub {
|
|
|
|
|
|
|
|
private static final String JEDIS_SUBSCRIPTION_MESSAGE = "JedisPubSub is not subscribed to a Jedis instance.";
|
|
|
|
private int subscribedChannels = 0;
|
|
|
|
private volatile Client client;
|
|
|
|
|
|
|
|
public JedisPubSub() {
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
public JedisPubSub(Client client) {
|
|
|
|
this.client = client;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void onMessage(String channel, String message) {
|
|
|
|
}
|
|
|
|
|
|
|
|
public void onPMessage(String pattern, String channel, String message) {
|
|
|
|
}
|
|
|
|
|
|
|
|
public void onSubscribe(String channel, int subscribedChannels) {
|
|
|
|
}
|
|
|
|
|
|
|
|
public void onUnsubscribe(String channel, int subscribedChannels) {
|
|
|
|
}
|
|
|
|
|
|
|
|
public void onPUnsubscribe(String pattern, int subscribedChannels) {
|
|
|
|
}
|
|
|
|
|
|
|
|
public void onPSubscribe(String pattern, int subscribedChannels) {
|
|
|
|
}
|
|
|
|
|
|
|
|
public void onPong(String pattern) {
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
public void unsubscribe() {
|
|
|
|
if (client == null) {
|
|
|
|
throw new JedisConnectionException(JEDIS_SUBSCRIPTION_MESSAGE);
|
|
|
|
}
|
|
|
|
client.unsubscribe();
|
|
|
|
client.flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void unsubscribe(String... channels) {
|
|
|
|
if (client == null) {
|
|
|
|
throw new JedisConnectionException(JEDIS_SUBSCRIPTION_MESSAGE);
|
|
|
|
}
|
|
|
|
client.unsubscribe(channels);
|
|
|
|
client.flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void subscribe(String... channels) {
|
|
|
|
if (client == null) {
|
|
|
|
throw new JedisConnectionException(JEDIS_SUBSCRIPTION_MESSAGE);
|
|
|
|
}
|
|
|
|
client.subscribe(channels);
|
|
|
|
client.flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void psubscribe(String... patterns) {
|
|
|
|
if (client == null) {
|
|
|
|
throw new JedisConnectionException(JEDIS_SUBSCRIPTION_MESSAGE);
|
|
|
|
}
|
|
|
|
client.psubscribe(patterns);
|
|
|
|
client.flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void punsubscribe() {
|
|
|
|
if (client == null) {
|
|
|
|
throw new JedisConnectionException(JEDIS_SUBSCRIPTION_MESSAGE);
|
|
|
|
}
|
|
|
|
client.punsubscribe();
|
|
|
|
client.flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void punsubscribe(String... patterns) {
|
|
|
|
if (client == null) {
|
|
|
|
throw new JedisConnectionException(JEDIS_SUBSCRIPTION_MESSAGE);
|
|
|
|
}
|
|
|
|
client.punsubscribe(patterns);
|
|
|
|
client.flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
public void ping() {
|
|
|
|
if (client == null) {
|
|
|
|
throw new JedisConnectionException(JEDIS_SUBSCRIPTION_MESSAGE);
|
|
|
|
}
|
|
|
|
client.ping();
|
|
|
|
client.flush();
|
|
|
|
}
|
|
|
|
|
|
|
|
public boolean isSubscribed() {
|
|
|
|
return subscribedChannels > 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void proceedWithPatterns(Client client, String... patterns) {
|
|
|
|
this.client = client;
|
|
|
|
client.psubscribe(patterns);
|
|
|
|
client.flush();
|
|
|
|
process(client);
|
|
|
|
}
|
|
|
|
|
|
|
|
public void proceed(Client client, String... channels) {
|
|
|
|
this.client = client;
|
|
|
|
client.subscribe(channels);
|
|
|
|
client.flush();
|
|
|
|
process(client);
|
|
|
|
}
|
|
|
|
|
|
|
|
private void process(Client client) {
|
|
|
|
|
|
|
|
do {
|
|
|
|
List<Object> reply = client.getUnflushedObjectMultiBulkReply();
|
|
|
|
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);
|
|
|
|
final String strchannel = (bchannel == null) ? null : SafeEncoder.encode(bchannel);
|
|
|
|
onSubscribe(strchannel, subscribedChannels);
|
|
|
|
} else if (Arrays.equals(UNSUBSCRIBE.raw, resp)) {
|
|
|
|
subscribedChannels = ((Long) reply.get(2)).intValue();
|
|
|
|
final byte[] bchannel = (byte[]) reply.get(1);
|
|
|
|
final String strchannel = (bchannel == null) ? null : SafeEncoder.encode(bchannel);
|
|
|
|
onUnsubscribe(strchannel, subscribedChannels);
|
|
|
|
} else if (Arrays.equals(MESSAGE.raw, resp)) {
|
|
|
|
final byte[] bchannel = (byte[]) reply.get(1);
|
|
|
|
final byte[] bmesg = (byte[]) reply.get(2);
|
|
|
|
final String strchannel = (bchannel == null) ? null : SafeEncoder.encode(bchannel);
|
|
|
|
final String strmesg = (bmesg == null) ? null : SafeEncoder.encode(bmesg);
|
|
|
|
onMessage(strchannel, strmesg);
|
|
|
|
} 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);
|
|
|
|
final String strpattern = (bpattern == null) ? null : SafeEncoder.encode(bpattern);
|
|
|
|
final String strchannel = (bchannel == null) ? null : SafeEncoder.encode(bchannel);
|
|
|
|
final String strmesg = (bmesg == null) ? null : SafeEncoder.encode(bmesg);
|
|
|
|
onPMessage(strpattern, strchannel, strmesg);
|
|
|
|
} else if (Arrays.equals(PSUBSCRIBE.raw, resp)) {
|
|
|
|
subscribedChannels = ((Long) reply.get(2)).intValue();
|
|
|
|
final byte[] bpattern = (byte[]) reply.get(1);
|
|
|
|
final String strpattern = (bpattern == null) ? null : SafeEncoder.encode(bpattern);
|
|
|
|
onPSubscribe(strpattern, subscribedChannels);
|
|
|
|
} else if (Arrays.equals(PUNSUBSCRIBE.raw, resp)) {
|
|
|
|
subscribedChannels = ((Long) reply.get(2)).intValue();
|
|
|
|
final byte[] bpattern = (byte[]) reply.get(1);
|
|
|
|
final String strpattern = (bpattern == null) ? null : SafeEncoder.encode(bpattern);
|
|
|
|
onPUnsubscribe(strpattern, subscribedChannels);
|
|
|
|
} else if (Arrays.equals(PONG.raw, resp)) {
|
|
|
|
final byte[] bpattern = (byte[]) reply.get(1);
|
|
|
|
final String strpattern = (bpattern == null) ? null : SafeEncoder.encode(bpattern);
|
|
|
|
onPong(strpattern);
|
|
|
|
} else {
|
|
|
|
throw new JedisException("Unknown message type: " + firstObj);
|
|
|
|
}
|
|
|
|
} while (isSubscribed());
|
|
|
|
|
|
|
|
/* Invalidate instance since this thread is no longer listening */
|
|
|
|
this.client = null;
|
|
|
|
}
|
|
|
|
|
|
|
|
public int getSubscribedChannels() {
|
|
|
|
return subscribedChannels;
|
|
|
|
}
|
|
|
|
}
|