Redis SortedSet实现原理

Redis中支持的数据结构比Memcached要多的多啦,如基本的字符串哈希表列表集合可排序集,在这些基本数据结构上也提供了针对该数据结构的各种操作,这也是Redis所以流行起来的一个重要原因,当然Redis能够流行起来的原因,远远不只这一个,如支持高并发的读写、数据的持久化、高效的内存管理及淘汰机制...

Redisgit提交历史中,可以查到,2009/10/241.050版本,Redis开始支持可排序集,在该版本中,只提供了一条命令zadd,宏定义如下所示:

1     {"zadd",zaddCommand,4,REDIS_CMD_BULK|REDIS_CMD_DENYOOM},

那么什么是可排序集呢? 从Redis 1.0开始就给我们提供了集合(Set)这种数据结构,集合就跟数学上的集合概念是一个道理【无序性,确定性,互异性】,集合里的元素无法保证元素的顺序,而业务上的需求,可能不止是一个集合,而且还要求能够快速地对集合元素进行排序,于是乎,Redis中提供了可排序集这么一种数据结构,似乎也是合情合理,无非就是在集合的基础上增加了排序功能,也许有人会问,Redis中不是有Sort命令嘛,下面的操作不也是同样可以达到对无序集的排序功能嘛,是的,是可以,但是在这里我们一直强调的是快速这两个字,而Sort命令的时间复杂度为O(N+M*Log(M)),可排序集获取一定范围内元素的时间复杂度为O(log(N) + M)

复制代码
root@bjpengpeng-VirtualBox:/home/bjpengpeng/redis-3.0.1/src# ./redis-cli 
127.0.0.1:6379> sort set
1) "1"
2) "2"
3) "3"
4) "5"
127.0.0.1:6379> sort set desc
1) "5"
2) "3"
3) "2"
4) "1"
127.0.0.1:6379>
复制代码

在了解可排序集是如何实现之前,需要了解一种数据结构跳表(Skip List),跳表与AVL、红黑树...等相比,数据结构简单,算法易懂,但查询的时间复杂度与平衡二叉树/红黑树相当,跳表的基本结构如下图所示

上图中整个跳表结构存放了4个元素5->10->20->30,图中的红色线表示查找元素30时,走的查找路线,从Head指针数组里最顶层的指针所指的20开始比较,与普通的链表查找相比,跳表的查询可以跳跃元素,上图中查询30,发现3020大,则查找就是20开始,而普通链表的查询必须一个元素一个元素的比较,时间复杂度为O(n)

有了上图所示的跳表基本结构,再看看如何向跳表中插入元素,向跳表中插入元素,由于元素所在层级的随机性,平均起来也是O(logn),说白了,就是查找元素应该插入在什么位置,然后就是普通的移动指针问题,再想想往有序单链表的插入操作吧,时间复杂度是不是也是O(n),下图所示是往跳表中插入元素28的过程,图中红色线表示查找插入位置的过程,绿色线表示进行指针的移动,将该元素插入

 

有了跳表的查找及插入那么就看看在跳表中如何删除元素吧,跳表中删除元素的个程,查找要删除的元素,找到后,进行指针的移动,过程如下图所示,删除元素30

有了上面的跳表基本结构图及原理,自已设计及实现跳表吧,这样当看到Redis里面的跳表结构时我们会更加熟悉,更容易理解些,【下面是对Redis中的跳表数据结构及相关代码进行精减后形成的可运行代码】,首先定义跳表的基本数据结构如下所示

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include<stdio.h>
#include<stdlib.h>
 
#define ZSKIPLIST_MAXLEVEL 32
#define ZSKIPLIST_P 0.25
#include <math.h>
 
//跳表节点
typedef  struct  zskiplistNode {
     int  key;
     int  value;
     struct  zskiplistLevel {
         struct  zskiplistNode *forward;
     } level[1];
} zskiplistNode;
 
//跳表
typedef  struct  zskiplist {
     struct  zskiplistNode *header;
     int  level;
} zskiplist;

 

在代码中我们定义了跳表结构中保存的数据为Key->Value这种形式的键值对,注意的是skiplistNode里面内含了一个结构体,代表的是层级,并且定义了跳表的最大层级为32级,下面的代码是创建空跳表,以及层级的获取方式

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
//创建跳表的节点
zskiplistNode *zslCreateNode( int  level,  int  key,  int  value) {
     zskiplistNode *zn = (zskiplistNode *) malloc ( sizeof (*zn)+level* sizeof (zn->level));
     zn->key = key;
     zn->value = value;
     return  zn;
}
 
//初始化跳表
zskiplist *zslCreate( void ) {
     int  j;
     zskiplist *zsl;
     zsl = (zskiplist *)  malloc ( sizeof (*zsl));
     zsl->level = 1; //将层级设置为1
     zsl->header = zslCreateNode(ZSKIPLIST_MAXLEVEL,NULL,NULL);
     for  (j = 0; j < ZSKIPLIST_MAXLEVEL; j++) {
         zsl->header->level[j].forward = NULL;
     }
     return  zsl;
}
 
//向跳表中插入元素时,随机一个层级,表示插入在哪一层
int  zslRandomLevel( void ) {
     int  level = 1;
     while  (( rand ()&0xFFFF) < (ZSKIPLIST_P * 0xFFFF))
         level += 1;
     return  (level<ZSKIPLIST_MAXLEVEL) ? level : ZSKIPLIST_MAXLEVEL;
}

在这段代码中,使用了随机函数获取过元素所在的层级,下面就是重点,向跳表中插入元素,插入元素之前先查找插入的位置,代码如下所示,代码中注意update[i]

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
//向跳表中插入元素
zskiplistNode *zslInsert(zskiplist *zsl,  int  key,  int  value) {
     zskiplistNode *update[ZSKIPLIST_MAXLEVEL], *x;
     int  i, level;
     x = zsl->header;
     //在跳表中寻找合适的位置并插入元素
     for  (i = zsl->level-1; i >= 0; i--) {
         while  (x->level[i].forward &&
             (x->level[i].forward->key < key ||
                 (x->level[i].forward->key == key &&
                 x->level[i].forward->value < value))) {
             x = x->level[i].forward;
         }
         update[i] = x;
     }
     //获取元素所在的随机层数
     level = zslRandomLevel();
     if  (level > zsl->level) {
         for  (i = zsl->level; i < level; i++) {
             update[i] = zsl->header;
         }
         zsl->level = level;
     }
     //为新创建的元素创建数据节点
     x = zslCreateNode(level,key,value);
     for  (i = 0; i < level; i++) {
         x->level[i].forward = update[i]->level[i].forward;
         update[i]->level[i].forward = x;
     }
     return  x;
}

下面是代码中删除节点的操作,和插入节点类似

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
//跳表中删除节点的操作
void  zslDeleteNode(zskiplist *zsl, zskiplistNode *x, zskiplistNode **update) {
     int  i;
     for  (i = 0; i < zsl->level; i++) {
         if  (update[i]->level[i].forward == x) {
             update[i]->level[i].forward = x->level[i].forward;
         }
     }
     //如果层数变了,相应的将层数进行减1操作
     while (zsl->level > 1 && zsl->header->level[zsl->level-1].forward == NULL)
         zsl->level--;
}
 
//从跳表中删除元素
int  zslDelete(zskiplist *zsl,  int  key,  int  value) {
     zskiplistNode *update[ZSKIPLIST_MAXLEVEL], *x;
     int  i;
     x = zsl->header;
     //寻找待删除元素
     for  (i = zsl->level-1; i >= 0; i--) {
         while  (x->level[i].forward &&
             (x->level[i].forward->key < key ||
                 (x->level[i].forward->key == key &&
                 x->level[i].forward->value < value))) {
             x = x->level[i].forward;
         }
         update[i] = x;
     }
     x = x->level[0].forward;
     if  (x && key == x->key && x->value == value) {
         zslDeleteNode(zsl, x, update);
         //别忘了释放节点所占用的存储空间
         free (x);
         return  1;
     else  {
         //未找到相应的元素
         return  0;
     }
     return  0;
}

最后,附上一个不优雅的测试样例

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
//将链表中的元素打印出来
void  printZslList(zskiplist *zsl) {
     zskiplistNode  *x;
     x = zsl->header;
     for  ( int  i = zsl->level-1; i >= 0; i--) {
         zskiplistNode *p = x->level[i].forward;
         while  (p) {
             printf ( " %d|%d " ,p->key,p->value);
             p = p->level[i].forward;
         }
         printf ( "\n" );
     }
}
 
int  main() {
     zskiplist *list = zslCreate();
     zslInsert(list,1,2);
     zslInsert(list,4,5);
     zslInsert(list,2,2);
     zslInsert(list,7,2);
     zslInsert(list,7,3);
     zslInsert(list,7,3);
     printZslList(list);
     //zslDelete(list,7,2);
     printZslList(list);
}

有了上面的跳表理论基础,理解Redis中跳表的实现就不是那么难了,先分析到这,下回续写,【代码以Redis 2.9为例】~


转自:http://www.cnblogs.com/WJ5888/p/4516782.html

  • 5
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值