当一个客户端执行 SUBSCRIBE 命令, 订阅某个或某些频道的时候, 这个客户端与被订阅频道之间就建立起了一种订阅关系。
Redis 将所有频道的订阅关系都保存在服务器状态的 pubsub_channels
字典里面, 这个字典的键是某个被订阅的频道, 而键的值则是一个链表, 链表里面记录了所有订阅这个频道的客户端:
struct redisServer {
// ...
// 保存所有频道的订阅关系
dict *pubsub_channels;
// ...
};
每当客户端执行 SUBSCRIBE 命令, 订阅某个或某些频道的时候, 服务器都会将客户端与被订阅的频道在 pubsub_channels
字典中进行关联。
根据频道是否已经有其他订阅者, 关联操作分为两种情况执行:
pubsub_channels
字典中必然有相应的订阅者链表, 程序唯一要做的就是将客户端添加到订阅者链表的末尾。pubsub_channels
字典, 程序首先要在 pubsub_channels
字典中为频道创建一个键, 并将这个键的值设置为空链表, 然后再将客户端添加到链表, 成为链表的第一个元素。SUBSCRIBE 命令的实现可以用以下伪代码来描述:
def subscribe(*all_input_channels):
# 遍历输入的所有频道
for channel in all_input_channels:
# 如果 channel 不存在于 pubsub_channels 字典(没有任何订阅者)
# 那么在字典中添加 channel 键,并设置它的值为空链表
if channel not in server.pubsub_channels:
server.pubsub_channels[channel] = []
# 将订阅者添加到频道所对应的链表的末尾
server.pubsub_channels[channel].append(client)
UNSUBSCRIBE 命令的行为和 SUBSCRIBE 命令的行为正好相反 —— 当一个客户端退订某个或某些频道的时候, 服务器将从 pubsub_channels
中解除客户端与被退订频道之间的关联:
pubsub_channels
字典中找到频道对应的订阅者链表, 然后从订阅者链表中删除退订客户端的信息。pubsub_channels
字典中删除频道对应的键。UNSUBSCRIBE 命令的实现可以用以下伪代码来描述:
def unsubscribe(*all_input_channels):
# 遍历要退订的所有频道
for channel in all_input_channels:
# 在订阅者链表中删除退订的客户端
server.pubsub_channels[channel].remove(client)
# 如果频道已经没有任何订阅者了(订阅者链表为空)
# 那么将频道从字典中删除
if len(server.pubsub_channels[channel]) == 0:
server.pubsub_channels.remove(channel)
前面说过,服务器将所有频道的订阅关系保存起来,与此类似,服务器也将所有模式的订阅关系存在了pubsub_Patterns属性里。
struct redisServer {
// ...
// 保存所有频道的订阅关系
list *pubsub_patterns;
// ...
};
pubsub_Patterns属性是一个链表,每个结点是被订阅的模式,节点内记录了模式,节点内的client属性记录了订阅模式的客户端。
typedef struct pubsubPattern{
//订阅模式的客户端
redisClient *client;
//被订阅的模式
robj *pattern;
}pubsubPattern;
每当客户端执行PSUBSCRIBE这个命令来订阅某个或某些模式时,服务器会对每个被订阅的模式执行下面的操作:
1)新建一个pubsubPattern结构,设置好两个属性
2)将新节点加到pubsub_patterns尾部
伪代码实现:
def osubscribe(*all_input_patterns):
#遍历所有输入的模式
#记录被订阅的模式和对应的客户端
pubsubPattern=create()
pubsubPattern.client=client
pubsubPattern.pattern=pattern
#插入链表末尾
server.pub_patterns.append(pubsubPattern)
模式退订命令PUNSUBSCRIBE是PSUBSCRIBE的反操作
服务器将找到并删除那些被退订的模式
伪代码如下:(我想吐槽一下这样时间复杂度。。。没有更好的办法吗?)
def osubscribe(*all_input_patterns):
#遍历所有退订的模式
for pattern in all_input_patterns:
#遍历每一个节点
for pubsubPattern in server.pubsub_patterns:
#如果客户端和模式都相同
if client==pubsubPattern.client:
if pattern==pubsubPattern.pattern:
#删除
server.pub_patterns.remove(pubsubPattern)
当一个客户端执行PUBLISH
1)把消息发送给所有本频道的订阅者
具体做法就是去pubsub_channels字典找到本频道的链表,也就是订阅名单,然后发消息
2)将消息发给,包含本频道的所有模式中的所有订阅者
具体做法就是去pubsub_patterns查找包含本频道的模式,并且把消息发送给订阅它们的客户端。
redis2.8新增三个命令,用来查看频道和模式的相关信息。
PUBLISH CHANNELS[pattern]用于返回服务器当前被订阅的频道,pattern可写可不写,不写就查看所有,否则查看与pattern匹配的对应频道
这个子命令是通过遍历pubsub_channels字典实现的。
PUBLISH NUMSUB[CHANNEL-1 CHANNEL-2.....]返回这些频道的订阅者数量
这个子命令是通过遍历pubsub_channels字典,查看对应链表长度实现的。
PUBLISH NUMPAT返回被订阅模式数量
这个子命令是通过返回pubsub_patterns的长度实现的。
总而言之,PUBSUB 命令的三个子命令都是通过读取 pubsub_channels
字典和 pubsub_patterns
链表中的信息来实现的。