redis对象编码源码阅读——有序集合编码和转码

redis对象编码源码阅读——有序集合编码和转码

1. 有序集合对象的编码类型

类型	编码	对象
REDIS_ZSET	REDIS_ENCODING_ZIPLIST	使用压缩列表对象实现的有序集合对象
REDIS_ZSET	REDIS_ENCODING_SKIPLIST	使用跳跃表和字典实现的有序集合对象

2. 两种编码的转换

void zsetConvert(robj *zobj, int encoding) {
   
    zset *zs;
    zskiplistNode *node, *next;
    robj *ele;
    double score;

    if (zobj->encoding == encoding) return;

2.1 压缩列表转跳表

    if (zobj->encoding == REDIS_ENCODING_ZIPLIST) {
   
        unsigned char *zl = zobj->ptr;
        unsigned char *eptr, *sptr;
        unsigned char *vstr;
        unsigned int vlen;
        long long vlong;

        if (encoding != REDIS_ENCODING_SKIPLIST)
            redisPanic("Unknown target encoding");

1. 重新申请空间

        zs = zmalloc(sizeof(*zs));
        zs->dict = dictCreate(&zsetDictType,NULL);
        zs->zsl = zslCreate();

2. 得到 eptr = ziplist[0] 和 sptr = ziplist[1] = ziplist[0]->next 指针

        eptr = ziplistIndex(zl,0);
        redisAssertWithInfo(NULL,zobj,eptr != NULL);
        sptr = ziplistNext(zl,eptr);
        redisAssertWithInfo(NULL,zobj,sptr != NULL);

• ziplistIndex: 通过下标index，得到ziplist[index]

/* Returns an offset to use for iterating with ziplistNext. When the given
 * index is negative, the list is traversed back to front. When the list
 * doesn't contain an element at the provided index, NULL is returned. */
unsigned char *ziplistIndex(unsigned char *zl, int index) {
   
    unsigned char *p;
    unsigned int prevlensize, prevlen = 0;
    // 当 index < 0 的时候，反向遍历找到对应位置
    // 比如 index = -2， 就是倒数第二个节点
    // 注意：倒数第一个节点是 ZIP_END
    if (index < 0) {
   
        index = (-index)-1;
        // 得到压缩列表的尾节点
        p = ZIPLIST_ENTRY_TAIL(zl);
        if (p[0] != ZIP_END) {
   
            // 得到前一个节点的长度
            ZIP_DECODE_PREVLEN(p, prevlensize, prevlen);
            while (prevlen > 0 && index--) {
   
                p -= prevlen;
                ZIP_DECODE_PREVLEN(p, prevlensize, prevlen);
            }
        }
    // 正向遍历
    } else