见博客 详解Redis —— 发布与订阅(上)
当一个Redis客户端执行PUBLISH 命令将消息message发送给频道channel的时候,服务器需要执行以下两个动作:
1 )将消息message发送给channel频道的所有订阅者。
2)如果有一个或多个模式pattern与频道channel相匹配,那么将消息message 发送给pattern模式的订阅者。
接下来的两个小节将分别介绍这两个动作的实现方式。
因为服务器状态中的pubsub_channels字典记录了所有频道的订阅关系,所以为了将消息发送给channel频道的所有订阅者,PUBLISH命令要做的就是在pubsub_channels字典里找到频道channel的订阅者名单(一个链表), 然后将消息发送给名单上的所有客户端。伪代码如下图所示:
def channel_publish (channel, message) : #如果channel键不存在于pubsub_channels字典中 #那么说明channel频道没有任何订阅者 #程序不做发送动作,直接返回 if channel not in server.pubsub_channels: return #运行到这里,说明channel频道至少有一个订阅者 #程序遍历channel频道的订阅者链表 #将消息发送给所有订阅者 for subscriber in server.pubsub_channels[channel] : send_message(subscriber, message)
因为服务器状态中的pubsub_patterns 链表记录了所有模式的订阅关系,所以为了将消息发送给所有与channel频道相匹配的模式的订阅者,PUBLISH命令要做的就是遍历整个pubsub_patterns 链表,查找那些与channel频道相匹配的模式,并将消息发送给订阅了这些模式的客户端。
PUBLISH命令将消息发送给模式订阅者的方法可以用以下伪代码来描述:
def pattern_publish (channel, message) : #遍历所有模式订阅消息 for pubsubPattern in server.pubsub_patterns: #如果频道和模式相匹配 if match (channel, pubsubPattern.pattern) : #那么将消息发送给订阅该模式的客户端 send_message(pubsubPattern.client, message)
最后,PUBLISH命令的实现可以用以下伪代码来描述:
def publish (channel, message) : #将消息发送给channel频道的所有订阅者 channel_publish (channel, message) #将消息发送给所有和channel频道相匹配的模式的订阅者 pattern_publish (channel, message)
PUBSUB命令是Redis2.8新增加的命令之–,客户端可以通过这个命令来查看频道或者模式的相关信息,比如某个频道目前有多少订阅者,又或者某个模式目前有多少订阅者,诸如此类。
以下三个小节将分别介绍PUBSUB命令的三个子命令,以及这些子命令的实现原理。
PUBSUB CHANNELS [pattern]子命令用于返回服务器当前被订阅的频道,其中pattern参数是可选的:
这个子命令是通过遍历服务器pubsub_channels字典的所有键(每个键都是一个被订阅的频道),然后记录并返回所有符合条件的频道来实现的,这个过程可以用以下伪代码来描述:
def pubsub_channels (pattern=None) : #一个列表,用于记录所有符合条件的频道 channel_list = [] #遍历服务器中的所有频道 # (也即是pubsub_channels字典的所有键) for channel in server.pubsub_channels: #当以下两个条件的任意一个满足时,将频道添加到链表里面: #1 )用户没有指定pattern参数 #2 )用户指定了pattern参数,并且channel和pattern匹配 if (pattern is None) or match(channel, pattern) : channel_list.append(channel) #向客户端返回频道列表 return channel_list
PUBSUB NUMSUB [channel-1 channel-2 … channel-n]子命令接受任意多个频道作为输入,并返回这些频道的订阅者数量。
这个子命令是通过在pubsub_ channels 字典中找到频道对应的订阅者链表,然后返回订阅者链表的长度来实现的(订阅者链表的长度就是频道订阅者的数量),这个过程可以用以下伪代码来描述:
def pubsub_numsub (*all_input_channels) : #遍历输入的所有频道 for channel in all_input_channels: #如果pubsub_channels 字典中没有channel这个键 #那么说明channel频道没有任何订阅者 if channel not in server.pubsub_channels: #返回频道名 reply_channel_name(channel) #订阅者数量为0 reply_subscribe_count (0) #如果pubsub_channels 字典中存在channel键 #那么说明channel频道至少有一个订阅者 else: #返回频道名 reply_channel_name(channel) #订阅者链表的长度就是订阅者数量 reply_subscribe_count(len(server.pubsub_channels[channel]) )
PUBSUB NUMPAT 子命令用于返回服务器当前被订阅模式的数量。
这个子命令是通过返回pubsub_ patterns 链表的长度来实现的,因为这个链表的长度就是服务器被订阅模式的数量,这个过程可以用以下伪代码来描述:
def pubsub_numpat () : # pubsub_ patterns 链表的长度就是被订阅模式的数量 reply_pattern_count(len(server.pubsub_patterns) )