Redis源码学习（17），t_zset.c 学习（二），zrem、zrange 命令学习

1 zremCommand

1.1 方法说明

  从有序集合中移出一些成员，成功返回移出成员的数量。

1.2 命令实践

1.3 方法源代码

void zremCommand(redisClient *c) {//获取键值robj *key = c->argv[1];robj *zobj;int deleted = 0, keyremoved = 0, j;//获取有序集合对象，并判断对象类型if ((zobj = lookupKeyWriteOrReply(c,key,shared.czero)) == NULL ||checkType(c,zobj,REDIS_ZSET)) return;//如果数据结构是压缩表if (zobj->encoding == REDIS_ENCODING_ZIPLIST) {unsigned char *eptr;//遍历需要移出的元素for (j = 2; j < c->argc; j++) {if ((eptr = zzlFind(zobj->ptr,c->argv[j],NULL)) != NULL) {deleted++;//从压缩表删除成员zobj->ptr = zzlDelete(zobj->ptr,eptr);//如果成员数量为0，则删除这个有序集合if (zzlLength(zobj->ptr) == 0) {dbDelete(c->db,key);keyremoved = 1;break;}}}}//如果数据结构是跳跃表 else if (zobj->encoding == REDIS_ENCODING_SKIPLIST) {zset *zs = zobj->ptr;dictEntry *de;double score;//遍历需要移出的成员for (j = 2; j < c->argc; j++) {de = dictFind(zs->dict,c->argv[j]);if (de != NULL) {deleted++;/* Delete from the skiplist */score = *(double*)dictGetVal(de);//从跳跃表中redisAssertWithInfo(c,c->argv[j],zslDelete(zs->zsl,score,c->argv[j]));/* Delete from the hash table *///从哈希表中删除这个元素dictDelete(zs->dict,c->argv[j]);if (htNeedsResize(zs->dict)) dictResize(zs->dict);//如果成员数量为0，则删除这个有序集合if (dictSize(zs->dict) == 0) {dbDelete(c->db,key);keyremoved = 1;break;}}}} else {redisPanic("Unknown sorted set encoding");}//如果删除了成员，则触发zrem和del通知事件。//标记键被修改if (deleted) {notifyKeyspaceEvent(REDIS_NOTIFY_ZSET,"zrem",key,c->db->id);if (keyremoved)notifyKeyspaceEvent(REDIS_NOTIFY_GENERIC,"del",key,c->db->id);signalModifiedKey(c->db,key);server.dirty += deleted;}	//响应删除成员的数量addReplyLongLong(c,deleted);
}

1.4 代码理解

1. 获取有序集合对象，并判断对象类型。
2. 判断数据结构，如果是压缩表的话，则调用压缩表相关方法移除成员。
3. 如果是跳跃表的话，则调用跳跃表相关方法移除成员，并且从字典中移除成员。
4. 如果所有成员都被移除了，则有序集合本身也要被删除掉。
5. 如果删除了成员，则触发zrem通知事件。
6. 如果有序集合被删除了，则触发 del 通知事件。
7. 返回删除成员的数量。

2 zrangeCommand

2.1 方法说明

  遍历有序集合某位置到某位置的成员，可选参数 withscores 显示分数。

2.2 命令实践

2.3 方法源代码

//按照分值升序
void zrangeCommand(redisClient *c) {zrangeGenericCommand(c,0);
}//按照分值降序
void zrevrangeCommand(redisClient *c) {zrangeGenericCommand(c,1);
}//zrange通用方法
void zrangeGenericCommand(redisClient *c, int reverse) {//获取键值robj *key = c->argv[1];robj *zobj;int withscores = 0;long start;long end;int llen;int rangelen;	//获取开始位置和结束位置if ((getLongFromObjectOrReply(c, c->argv[2], &start, NULL) != REDIS_OK) ||(getLongFromObjectOrReply(c, c->argv[3], &end, NULL) != REDIS_OK)) return;//校验参是否为5个，并且是否包含可选参数withscores//如果参数超过5个，则报错if (c->argc == 5 && !strcasecmp(c->argv[4]->ptr,"withscores")) {withscores = 1;} else if (c->argc >= 5) {addReply(c,shared.syntaxerr);return;}//获取有序集合对象，并判断对象类型是否有序集合if ((zobj = lookupKeyReadOrReply(c,key,shared.emptymultibulk)) == NULL|| checkType(c,zobj,REDIS_ZSET)) return;/* Sanitize indexes. *///获取有序集合成员数量llen = zsetLength(zobj);//位置负数处理if (start < 0) start = llen+start;if (end < 0) end = llen+end;if (start < 0) start = 0;/* Invariant: start >= 0, so this test will be true when end < 0.* The range is empty when start > end or start >= length. */if (start > end || start >= llen) {addReply(c,shared.emptymultibulk);return;}if (end >= llen) end = llen-1;//计算遍历数量rangelen = (end-start)+1;/* Return the result in form of a multi-bulk reply */addReplyMultiBulkLen(c, withscores ? (rangelen*2) : rangelen);//如果数据结构是压缩表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 (reverse)eptr = ziplistIndex(zl,-2-(2*start));elseeptr = ziplistIndex(zl,2*start);redisAssertWithInfo(c,zobj,eptr != NULL);sptr = ziplistNext(zl,eptr);//按照起始位置遍历while (rangelen--) {redisAssertWithInfo(c,zobj,eptr != NULL && sptr != NULL);redisAssertWithInfo(c,zobj,ziplistGet(eptr,&vstr,&vlen,&vlong));if (vstr == NULL)addReplyBulkLongLong(c,vlong);elseaddReplyBulkCBuffer(c,vstr,vlen);//如果使用了可选分数，则返回分数if (withscores)addReplyDouble(c,zzlGetScore(sptr));if (reverse)zzlPrev(zl,&eptr,&sptr);elsezzlNext(zl,&eptr,&sptr);}}//如果是跳跃表else if (zobj->encoding == REDIS_ENCODING_SKIPLIST) {zset *zs = zobj->ptr;zskiplist *zsl = zs->zsl;zskiplistNode *ln;robj *ele;/* Check if starting point is trivial, before doing log(N) lookup. */if (reverse) {ln = zsl->tail;if (start > 0)ln = zslGetElementByRank(zsl,llen-start);} else {ln = zsl->header->level[0].forward;if (start > 0)ln = zslGetElementByRank(zsl,start+1);}while(rangelen--) {redisAssertWithInfo(c,zobj,ln != NULL);ele = ln->obj;addReplyBulk(c,ele);if (withscores)addReplyDouble(c,ln->score);ln = reverse ? ln->backward : ln->level[0].forward;}} else {redisPanic("Unknown sorted set encoding");}
}

2.4 代码理解

1. 获取开始位置和结束位置，并校验参数类型
2. 校验参是否为5个，并且是否包含可选参数withscores
3. 获取有序集合成员数量
4. 计算需要遍历成员的数量
5. 判断数据结构，如果是压缩表，则调用压缩表相关方法遍历成员。
6. 如果是跳跃表，则调用跳跃表相关方法遍历成员。
7. 如果使用了withsocres，则遍历的时候会显示分数。

  通过代码可以看到，zrangeCommand 和 zrevrangeCommand 两个方法都调用了 zrangeGenericCommand 这个方法，只不过通过传入第二个参数不同来控制遍历的顺序。

  读过 t_list.c 和 t_string.c 源代码的同学可以发现 lrangeCommand 、getrangeCommand 和 现在我们学习的这个 zrangeGenericCommand 遍历成员的时候的逻辑大部分都是相同的，只不过读取不同数据结构的方法不同，但是遍历的思想是一样的，值得我们借鉴和学习。

3 总结

1. zrangeCommand 和 zrevrangeCommand 两个方法都调用了 zrangeGenericCommand 这个方法。
2. getrangeCommand、lrangeCommand、zrangeCommand 三者的遍历逻辑相同。
3. zremCommand 在移出元素的时候，如果使用了跳跃表也会将在哈希表中移出响应的成员。

本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！