走进KeyDB

KeyDB项目是从redis fork出来的分支。众所周知redis是一个单线程的kv内存存储系统，而KeyDB在100%兼容redis API的状况下将redis改形成多线程。

项目git地址：https://github.com/JohnSully/KeyDB

网上公开的技术细节比较少，本文基本是经过阅读源码总结出来的，若有错漏之处欢迎指正。

多线程架构

线程模型

KeyDB将redis原来的主线程拆分红了主线程和worker线程。每一个worker线程都是io线程，负责监听端口，accept请求，读取数据和解析协议。如图所示：

KeyDB使用了SO_REUSEPORT特性，多个线程能够绑定监听同个端口。
每一个worker线程作了cpu绑核，读取数据也使用了SO_INCOMING_CPU特性，指定cpu接收数据。
解析协议以后每一个线程都会去操做内存中的数据，由一把全局锁来控制多线程访问内存数据。
主线程其实也是一个worker线程，包括了worker线程的工做内容，同时也包括只有主线程才能够完成的工做内容。在worker线程数组中下标为0的就是主线程。
主线程的主要工做在实现serverCron，包括：

• 处理统计
• 客户端连接管理
• db数据的resize和reshard
• 处理aof
• replication主备同步
• cluster模式下的任务

连接管理

在redis中全部连接管理都是在一个线程中完成的。在KeyDB的设计中，每一个worker线程负责一组连接，全部的连接插入到本线程的连接列表中维护。连接的产生、工做、销毁必须在同个线程中。每一个连接新增一个字段
int iel; /* the event loop index we're registered with */
用来表示连接属于哪一个线程接管。
KeyDB维护了三个关键的数据结构作连接管理：

• clients_pending_write：线程专属的链表，维护同步给客户连接发送数据的队列
• clients_pending_asyncwrite：线程专属的链表，维护异步给客户连接发送数据的队列
• clients_to_close：全局链表，维护须要异步关闭的客户连接

分红同步和异步两个队列，是由于redis有些联动api，好比pub/sub，pub以后须要给sub的客户端发送消息，pub执行的线程和sub的客户端所在线程不是同一个线程，为了处理这种状况，KeyDB将须要给非本线程的客户端发送数据维护在异步队列中。
同步发送的逻辑比较简单，都是在本线程中完成，如下图来讲明如何同步给客户端发送数据：

如上文所提到的，一个连接的建立、接收数据、发送数据、释放连接都必须在同个线程执行。异步发送涉及到两个线程之间的交互。KeyDB经过管道在两个线程中传递消息：

int fdCmdWrite; //写管道
int fdCmdRead; //读管道

本地线程须要异步发送数据时，先检查client是否属于本地线程，非本地线程获取到client专属的线程ID，以后给专属的线程管到发送AE_ASYNC_OP::CreateFileEvent的操做，要求添加写socket事件。专属线程在处理管道消息时将对应的请求添加到写事件中，如图所示：

redis有些关闭客户端的请求并不是彻底是在连接所在的线程执行关闭，因此在这里维护了一个全局的异步关闭链表。

锁机制

KeyDB实现了一套相似spinlock的锁机制，称之为fastlock。
fastlock的主要数据结构有：

struct ticket
{
    uint16_t m_active;  //解锁+1
    uint16_t m_avail;  //加锁+1
};
struct fastlock
{
    volatile struct ticket m_ticket;

    volatile int m_pidOwner; //当前解锁的线程id
    volatile int m_depth; //当前线程重复加锁的次数
};

使用原子操做__atomic_load_2，__atomic_fetch_add，__atomic_compare_exchange来经过比较m_active=m_avail判断是否能够获取锁。
fastlock提供了两种获取锁的方式：

• try_lock：一次获取失败，直接返回
• lock：忙等，每1024 * 1024次忙等后使用sched_yield 主动交出cpu，挪到cpu的任务末尾等待执行。

在KeyDB中将try_lock和事件结合起来，来避免忙等的状况发生。每一个客户端有一个专属的lock，在读取客户端数据以前会先尝试加锁，若是失败，则退出，由于数据还未读取，因此在下个epoll_wait处理事件循环中能够再次处理。

Active-Replica

KeyDB实现了多活的机制，每一个replica可设置成可写非只读，replica之间互相同步数据。主要特性有：

• 每一个replica有个uuid标志，用来去除环形复制
• 新增长rreplay API，将增量命令打包成rreplay命令，带上本地的uuid
• key，value加上时间戳版本号，做为冲突校验，若是本地有相同的key且时间戳版本号大于同步过来的数据，新写入失败。采用当前时间戳向左移20位，再加上后44位自增的方式来获取key的时间戳版本号。

结束语

云数据库Redis版（ApsaraDB for Redis）是一种稳定可靠、性能卓越、可弹性伸缩的数据库服务。基于飞天分布式系统和全SSD盘高性能存储，支持主备版和集群版两套高可用架构。提供了全套的容灾切换、故障迁移、在线扩容、性能优化的数据库解决方案。

---

本文做者：羽洵

阅读原文

本文为云栖社区原创内容，未经容许不得转载。