列表

REDIS_LIST (列表)是 LPUSHLRANGE 等命令的操作对象, 它使用 REDIS_ENCODING_ZIPLISTREDIS_ENCODING_LINKEDLIST 这两种方式编码:

digraph redis_list {

    node[shape=plaintext, style = filled];

    edge [style = bold];

    // type

    REDIS_LIST [label="列表\nREDIS_LIST", fillcolor = "#95BBE3"];

    // encoding

    REDIS_ENCODING_ZIPLIST [label="压缩列表\nREDIS_ENCODING_ZIPLIST", fillcolor = "#FADCAD"];
    REDIS_ENCODING_LINKEDLIST [label="双端链表\nREDIS_ENCODING_LINKEDLIST", fillcolor = "#FADCAD"];

    // edge

    REDIS_LIST -> REDIS_ENCODING_LINKEDLIST;
    REDIS_LIST -> REDIS_ENCODING_ZIPLIST;

    REDIS_ENCODING_LINKEDLIST -> list;
    REDIS_ENCODING_ZIPLIST -> ziplist;

    // datastruct 1
    
    list [label="adlist.h/list"];

    // datastruct 2

    ziplist [label="ziplist"];

}

编码的选择

创建新列表时 Redis 默认使用 REDIS_ENCODING_ZIPLIST 编码, 当以下任意一个条件被满足时, 列表会被转换成 REDIS_ENCODING_LINKEDLIST 编码:

  • 试图往列表新添加一个字符串值,且这个字符串的长度超过 server.list_max_ziplist_value (默认值为 64 )。

  • ziplist 包含的节点超过 server.list_max_ziplist_entries (默认值为 512 )。

列表命令的实现

因为两种底层实现的抽象方式和列表的抽象方式非常接近, 所以列表命令几乎就是通过一对一地映射到底层数据结构的操作来实现的。

既然这些映射都非常直观, 这里就不做赘述了, 在以下的内容中, 我们将焦点放在 BLPOPBRPOPBRPOPLPUSH 这个几个阻塞命令的实现原理上。

阻塞的条件

BLPOPBRPOPBRPOPLPUSH 三个命令都可能造成客户端被阻塞, 以下将这些命令统称为列表的阻塞原语。

阻塞原语并不是一定会造成客户端阻塞:

  • 只有当这些命令被用于空列表时, 它们才会阻塞客户端。

  • 如果被处理的列表不为空的话, 它们就执行无阻塞版本的 LPOPRPOPRPOPLPUSH 命令。

作为例子,以下流程图展示了 BLPOP 决定是否对客户端进行阻塞过程:

digraph blpop_decide_block_or_not {

    node [shape=plaintext, style = filled];

    edge [style = bold];

    //

    call_blpop [label = "BLPOP key", fillcolor = "#A8E270"];

    wrong_type_or_not [label = "key 非空且不是列表?", shape = diamond, fillcolor = "#95BBE3"];

    return_wrong_type [label = "返回类型错误"];

    key_empty_or_not [label = "key 是否为空?", shape = diamond, fillcolor = "#95BBE3"];

    block_client [label = "阻塞客户端"];

    lpop [label = "执行 LPOP key 命令", fillcolor = "#A8E270"];

    //

    call_blpop -> wrong_type_or_not;

    wrong_type_or_not -> return_wrong_type [label = "是"];

    wrong_type_or_not -> key_empty_or_not [label = "否"];

    key_empty_or_not -> block_client [label = "是"];

    key_empty_or_not -> lpop [label = "否"];

}

阻塞

当一个阻塞原语的处理目标为空键时, 执行该阻塞原语的客户端就会被阻塞。

阻塞一个客户端需要执行以下步骤:

  1. 将客户端的状态设为“正在阻塞”,并记录阻塞这个客户端的各个键,以及阻塞的最长时限(timeout)等数据。

  2. 将客户端的信息记录到 server.db[i]->blocking_keys 中(其中 i 为客户端所使用的数据库号码)。

  3. 继续维持客户端和服务器之间的网络连接,但不再向客户端传送任何信息,造成客户端阻塞。

步骤 2 是将来解除阻塞的关键, server.db[i]->blocking_keys 是一个字典, 字典的键是那些造成客户端阻塞的键, 而字典的值是一个链表, 链表里保存了所有因为这个键而被阻塞的客户端 (被同一个键所阻塞的客户端可能不止一个):

digraph db_blocking_keys {

    rankdir = LR;

    node [shape = record, style = filled];

    edge [style = bold];

    // keys

    blocking_keys [label = "blocking_keys |<key1> key1 |<key2> key2 |<key3> key3 | ... |<keyN> keyN", fillcolor = "#A8E270"];

    // clients blocking for key1
    client1 [label = "client1", fillcolor = "#95BBE3"];
    client5 [label = "client5", fillcolor = "#95BBE3"];
    client2 [label = "client2", fillcolor = "#95BBE3"];
    null_1 [label = "NULL", shape=plaintext];
    
    blocking_keys:key1 -> client2;
    client2 -> client5;
    client5 -> client1;
    client1 -> null_1;

    // clients blocking for key2
    client7 [label = "client7", fillcolor = "#95BBE3"];
    null_2 [label = "NULL", shape=plaintext];

    blocking_keys:key2 -> client7;
    client7 -> null_2;

    // key3

    client3 [label = "client3", fillcolor = "#95BBE3"];
    client4 [label = "client4", fillcolor = "#95BBE3"];
    client6 [label = "client6", fillcolor = "#95BBE3"];
    null_3 [label = "NULL", shape=plaintext];

    blocking_keys:key3 -> client3;
    client3 -> client4;
    client4 -> client6;
    client6 -> null_3;
}

在上图展示的 blocking_keys 例子中, client2client5client1 三个客户端就正被 key1 阻塞, 而其他几个客户端也正在被别的两个 key 阻塞。

当客户端被阻塞之后,脱离阻塞状态有以下三种方法:

  1. 被动脱离:有其他客户端为造成阻塞的键推入了新元素。

  2. 主动脱离:到达执行阻塞原语时设定的最大阻塞时间。

  3. 强制脱离:客户端强制终止和服务器的连接,或者服务器停机。

以下内容将分别介绍被动脱离和主动脱离的实现方式。

阻塞因 LPUSH 、 RPUSH 、 LINSERT 等添加命令而被取消

通过将新元素推入造成客户端阻塞的某个键中, 可以让相应的客户端从阻塞状态中脱离出来 (取消阻塞的客户端数量取决于推入元素的数量)。

LPUSHRPUSHLINSERT 这三个添加新元素到列表的命令, 在底层都由一个 pushGenericCommand 的函数实现, 这个函数的运作流程如下图:

digraph push_generic_command {
    
    node [shape = plaintext, style = filled];

    edge [style = bold];

    /*
    lpush [label = "LPUSH key value [value ...]"];
    rpush [label = "RPUSH key value [value ...]"];
    linsert [label = "LINSERT key BEFORE\|AFTER pivot value"];
    */

    pushGenericCommand [label = "pushGenericCommand", fillcolor = "#A8E270"];

    key_wrong_type_or_not [label = "key 非空且不是列表?", shape = diamond, fillcolor = "#95BBE3"];

    return_wrong_type_error [label = "返回类型错误"];

    key_empty_or_not [label = "key 为空?", shape = diamond, fillcolor = "#95BBE3"];

    // call_signal_list_as_ready [label = "调用 signalListAsReady"];

    add_key_to_ready_list_if_need [label = "如果 key 存在于 server.db[i]-\>blocking_keys\n那么为 key 创建一个 readyList 结构\n并将它添加到 server.ready_keys 链表中"];

    push_value_to_list [label = "将输入值推入列表"];

    /*
    lpush -> pushGenericCommand;
    rpush -> pushGenericCommand;
    linsert -> pushGenericCommand;
    */

    pushGenericCommand -> key_wrong_type_or_not;

    key_wrong_type_or_not -> return_wrong_type_error [label = "是"];

    key_wrong_type_or_not -> key_empty_or_not [label = "否"];

    // key_empty_or_not -> call_signal_list_as_ready [label = "是"];

    // call_signal_list_as_ready -> add_key_to_ready_list_if_need;

    key_empty_or_not -> add_key_to_ready_list_if_need [label = "是"];

    key_empty_or_not -> push_value_to_list [label = "否"];
    add_key_to_ready_list_if_need -> push_value_to_list;
}

当向一个空键推入新元素时, pushGenericCommand 函数执行以下两件事:

  1. 检查这个键是否存在于前面提到的 server.db[i]->blocking_keys 字典里, 如果是的话, 那么说明有至少一个客户端因为这个 key 而被阻塞,程序会为这个键创建一个 redis.h/readyList 结构, 并将它添加到 server.ready_keys 链表中。

  2. 将给定的值添加到列表键中。

readyList 结构的定义如下:

typedef struct readyList {
    redisDb *db;
    robj *key;
} readyList;

readyList 结构的 key 属性指向造成阻塞的键,而 db 则指向该键所在的数据库。

举个例子, 假设某个非阻塞客户端正在使用 0 号数据库, 而这个数据库当前的 blocking_keys 属性的值如下:

digraph db_blocking_keys {

    rankdir = LR;

    node [shape = record, style = filled];

    edge [style = bold];

    // keys

    blocking_keys [label = "blocking_keys |<key1> key1 |<key2> key2 |<key3> key3 | ... |<keyN> keyN", fillcolor = "#A8E270"];

    // clients blocking for key1
    client1 [label = "client1", fillcolor = "#95BBE3"];
    client5 [label = "client5", fillcolor = "#95BBE3"];
    client2 [label = "client2", fillcolor = "#95BBE3"];
    null_1 [label = "NULL", shape=plaintext];
    
    blocking_keys:key1 -> client2;
    client2 -> client5;
    client5 -> client1;
    client1 -> null_1;

    // clients blocking for key2
    client7 [label = "client7", fillcolor = "#95BBE3"];
    null_2 [label = "NULL", shape=plaintext];

    blocking_keys:key2 -> client7;
    client7 -> null_2;

    // key3

    client3 [label = "client3", fillcolor = "#95BBE3"];
    client4 [label = "client4", fillcolor = "#95BBE3"];
    client6 [label = "client6", fillcolor = "#95BBE3"];
    null_3 [label = "NULL", shape=plaintext];

    blocking_keys:key3 -> client3;
    client3 -> client4;
    client4 -> client6;
    client6 -> null_3;
}

如果这时客户端对该数据库执行 PUSH key3 value , 那么 pushGenericCommand 将创建一个 db 属性指向 0 号数据库、 key 属性指向 key3 键对象的 readyList 结构 , 并将它添加到服务器 server.ready_keys 属性的链表中:

digraph update_ready_keys {

    rankdir = LR;

    node [shape = record, style = filled];

    edge [style = bold];

    redisServer [label = "redisServer | ... |<ready_keys> ready_keys | ...", fillcolor = "#A8E270"];

    readyList [label = "<head>readyList |<db> db |<key> key", fillcolor = "#95BBE3"];

    listNode [label = "<head>listNode |{<prev> prev |<next> next |<value> value} ", fillcolor = "#FADCAD"];

    null [label = "NULL", shape = plaintext];

    redisServer:ready_keys -> listNode:head [label = "list"];
    listNode:next -> null;
    listNode:prev -> null;
    listNode:value -> readyList:head;
    
    redisDb [label = "<head> redisDb | ... |<dict> dict | ...", fillcolor = "#FFC1C1"];
    readyList:db -> redisDb:head;

    dict [label = "<head>dict\n(key space of DB) | ... |<key3> key3 | ...", fillcolor = "#F2F2F2"];
    redisDb:dict -> dict:head;

    readyList:key -> dict:key3;
}

在我们这个例子中, 到目前为止, pushGenericCommand 函数完成了以下两件事:

  1. readyList 添加到服务器。

  2. 将新元素 value 添加到键 key3

虽然 key3 已经不再是空键, 但到目前为止, 被 key3 阻塞的客户端还没有任何一个被解除阻塞状态。

为了做到这一点, Redis 的主进程在执行完 pushGenericCommand 函数之后, 会继续调用 handleClientsBlockedOnLists 函数, 这个函数执行以下操作:

  1. 如果 server.ready_keys 不为空,那么弹出该链表的表头元素,并取出元素中的 readyList 值。

  2. 根据 readyList 值所保存的 keydb ,在 server.blocking_keys 中查找所有因为 key 而被阻塞的客户端(以链表的形式保存)。

  3. 如果 key 不为空,那么从 key 中弹出一个元素,并弹出客户端链表的第一个客户端,然后将被弹出元素返回给被弹出客户端作为阻塞原语的返回值。

  4. 根据 readyList 结构的属性,删除 server.blocking_keys 中相应的客户端数据,取消客户端的阻塞状态。

  5. 继续执行步骤 3 和 4 ,直到 key 没有元素可弹出,或者所有因为 key 而阻塞的客户端都取消阻塞为止。

  6. 继续执行步骤 1 ,直到 ready_keys 链表里的所有 readyList 结构都被处理完为止。

用一段伪代码描述以上操作可能会更直观一些:

def handleClientsBlockedOnLists():

    # 执行直到 ready_keys 为空
    while server.ready_keys != NULL:

        # 弹出链表中的第一个 readyList
        rl = server.ready_keys.pop_first_node()

        # 遍历所有因为这个键而被阻塞的客户端
        for client in all_client_blocking_by_key(rl.key, rl.db):

            # 只要还有客户端被这个键阻塞,就一直从键中弹出元素
            # 如果被阻塞客户端执行的是 BLPOP ,那么对键执行 LPOP
            # 如果执行的是 BRPOP ,那么对键执行 RPOP
            element = rl.key.pop_element()

            if element == NULL:
                # 键为空,跳出 for 循环
                # 余下的未解除阻塞的客户端只能等待下次新元素的进入了
                break
            else:
                # 清除客户端的阻塞信息
                server.blocking_keys.remove_blocking_info(client)
                # 将元素返回给客户端,脱离阻塞状态
                client.reply_list_item(element)

先阻塞先服务(FBFS)策略

值得一提的是, 当程序添加一个新的被阻塞客户端到 server.blocking_keys 字典的链表中时, 它将该客户端放在链表的最后, 而当 handleClientsBlockedOnLists 取消客户端的阻塞时, 它从链表的最前面开始取消阻塞: 这个链表形成了一个 FIFO 队列, 最先被阻塞的客户端总是最先脱离阻塞状态, Redis 文档称这种模式为先阻塞先服务(FBFS,first-block-first-serve)。

举个例子,在下图所示的阻塞状况中, 如果客户端对数据库执行 PUSH key3 value , 那么只有 client3 会被取消阻塞, client6client4 仍然阻塞; 如果客户端对数据库执行 PUSH key3 value1 value2 , 那么 client3client4 的阻塞都会被取消, 而客户端 client6 依然处于阻塞状态:

digraph db_blocking_keys {

    rankdir = LR;

    node [shape = record, style = filled];

    edge [style = bold];

    // keys

    blocking_keys [label = "blocking_keys |<key1> key1 |<key2> key2 |<key3> key3 | ... |<keyN> keyN", fillcolor = "#A8E270"];

    // clients blocking for key1
    client1 [label = "client1", fillcolor = "#95BBE3"];
    client5 [label = "client5", fillcolor = "#95BBE3"];
    client2 [label = "client2", fillcolor = "#95BBE3"];
    null_1 [label = "NULL", shape=plaintext];
    
    blocking_keys:key1 -> client2;
    client2 -> client5;
    client5 -> client1;
    client1 -> null_1;

    // clients blocking for key2
    client7 [label = "client7", fillcolor = "#95BBE3"];
    null_2 [label = "NULL", shape=plaintext];

    blocking_keys:key2 -> client7;
    client7 -> null_2;

    // key3

    client3 [label = "client3", fillcolor = "#95BBE3"];
    client4 [label = "client4", fillcolor = "#95BBE3"];
    client6 [label = "client6", fillcolor = "#95BBE3"];
    null_3 [label = "NULL", shape=plaintext];

    blocking_keys:key3 -> client3;
    client3 -> client4;
    client4 -> client6;
    client6 -> null_3;
}

阻塞因超过最大等待时间而被取消

前面提到过, 当客户端被阻塞时, 所有造成它阻塞的键, 以及阻塞的最长时限会被记录在客户端里面, 并且该客户端的状态会被设置为“正在阻塞”。

每次 Redis 服务器常规操作函数(server cron job)执行时, 程序都会检查所有连接到服务器的客户端, 查看那些处于“正在阻塞”状态的客户端的最大阻塞时限是否已经过期, 如果是的话, 就给客户端返回一个空白回复, 然后撤销对客户端的阻塞。

可以用一段伪代码来描述这个过程:

def server_cron_job():

    # 其他操作 ...

    # 遍历所有已连接客户端
    for client in server.all_connected_client:

        # 如果客户端状态为“正在阻塞”,并且最大阻塞时限已到达
        if client.state == BLOCKING and \
           client.max_blocking_timestamp < current_timestamp():

            # 那么给客户端发送空回复,脱离阻塞状态
            client.send_empty_reply()

            # 并清除客户端在服务器上的阻塞信息
            server.blocking_keys.remove_blocking_info(client)

    # 其他操作 ...