Feng
5 years ago
4 changed files with 2176 additions and 2176 deletions
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,187 @@ |
|||||||
|
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.getRawObjectMultiBulkReply(); |
||||||
|
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; |
||||||
|
} |
||||||
|
} |
File diff suppressed because it is too large
Load Diff
@ -1,187 +0,0 @@ |
|||||||
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.getRawObjectMultiBulkReply(); |
|
||||||
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; |
|
||||||
} |
|
||||||
} |
|
Loading…
Reference in new issue