mirror of
https://github.com/fluencelabs/redis
synced 2025-03-19 17:10:50 +00:00
Merge branch 'zset-mem' into zrevrangebyscore
This commit is contained in:
commit
192fc3376a
@ -340,7 +340,7 @@ dictType zsetDictType = {
|
|||||||
NULL, /* val dup */
|
NULL, /* val dup */
|
||||||
dictEncObjKeyCompare, /* key compare */
|
dictEncObjKeyCompare, /* key compare */
|
||||||
dictRedisObjectDestructor, /* key destructor */
|
dictRedisObjectDestructor, /* key destructor */
|
||||||
dictVanillaFree /* val destructor of malloc(sizeof(double)) */
|
NULL /* val destructor */
|
||||||
};
|
};
|
||||||
|
|
||||||
/* Db->dict, keys are sds strings, vals are Redis objects. */
|
/* Db->dict, keys are sds strings, vals are Redis objects. */
|
||||||
|
13
src/redis.h
13
src/redis.h
@ -490,13 +490,14 @@ typedef struct _redisSortOperation {
|
|||||||
} redisSortOperation;
|
} redisSortOperation;
|
||||||
|
|
||||||
/* ZSETs use a specialized version of Skiplists */
|
/* ZSETs use a specialized version of Skiplists */
|
||||||
|
|
||||||
typedef struct zskiplistNode {
|
typedef struct zskiplistNode {
|
||||||
struct zskiplistNode **forward;
|
|
||||||
struct zskiplistNode *backward;
|
|
||||||
unsigned int *span;
|
|
||||||
double score;
|
|
||||||
robj *obj;
|
robj *obj;
|
||||||
|
double score;
|
||||||
|
struct zskiplistNode *backward;
|
||||||
|
struct zskiplistLevel {
|
||||||
|
struct zskiplistNode *forward;
|
||||||
|
unsigned int span;
|
||||||
|
} level[];
|
||||||
} zskiplistNode;
|
} zskiplistNode;
|
||||||
|
|
||||||
typedef struct zskiplist {
|
typedef struct zskiplist {
|
||||||
@ -707,7 +708,7 @@ void backgroundRewriteDoneHandler(int statloc);
|
|||||||
/* Sorted sets data type */
|
/* Sorted sets data type */
|
||||||
zskiplist *zslCreate(void);
|
zskiplist *zslCreate(void);
|
||||||
void zslFree(zskiplist *zsl);
|
void zslFree(zskiplist *zsl);
|
||||||
void zslInsert(zskiplist *zsl, double score, robj *obj);
|
zskiplistNode *zslInsert(zskiplist *zsl, double score, robj *obj);
|
||||||
|
|
||||||
/* Core functions */
|
/* Core functions */
|
||||||
void freeMemoryIfNeeded(void);
|
void freeMemoryIfNeeded(void);
|
||||||
|
251
src/t_zset.c
251
src/t_zset.c
@ -24,13 +24,7 @@
|
|||||||
* from tail to head, useful for ZREVRANGE. */
|
* from tail to head, useful for ZREVRANGE. */
|
||||||
|
|
||||||
zskiplistNode *zslCreateNode(int level, double score, robj *obj) {
|
zskiplistNode *zslCreateNode(int level, double score, robj *obj) {
|
||||||
zskiplistNode *zn = zmalloc(sizeof(*zn));
|
zskiplistNode *zn = zmalloc(sizeof(*zn)+level*sizeof(struct zskiplistLevel));
|
||||||
|
|
||||||
zn->forward = zmalloc(sizeof(zskiplistNode*) * level);
|
|
||||||
if (level > 1)
|
|
||||||
zn->span = zmalloc(sizeof(unsigned int) * (level - 1));
|
|
||||||
else
|
|
||||||
zn->span = NULL;
|
|
||||||
zn->score = score;
|
zn->score = score;
|
||||||
zn->obj = obj;
|
zn->obj = obj;
|
||||||
return zn;
|
return zn;
|
||||||
@ -45,11 +39,8 @@ zskiplist *zslCreate(void) {
|
|||||||
zsl->length = 0;
|
zsl->length = 0;
|
||||||
zsl->header = zslCreateNode(ZSKIPLIST_MAXLEVEL,0,NULL);
|
zsl->header = zslCreateNode(ZSKIPLIST_MAXLEVEL,0,NULL);
|
||||||
for (j = 0; j < ZSKIPLIST_MAXLEVEL; j++) {
|
for (j = 0; j < ZSKIPLIST_MAXLEVEL; j++) {
|
||||||
zsl->header->forward[j] = NULL;
|
zsl->header->level[j].forward = NULL;
|
||||||
|
zsl->header->level[j].span = 0;
|
||||||
/* span has space for ZSKIPLIST_MAXLEVEL-1 elements */
|
|
||||||
if (j < ZSKIPLIST_MAXLEVEL-1)
|
|
||||||
zsl->header->span[j] = 0;
|
|
||||||
}
|
}
|
||||||
zsl->header->backward = NULL;
|
zsl->header->backward = NULL;
|
||||||
zsl->tail = NULL;
|
zsl->tail = NULL;
|
||||||
@ -58,19 +49,15 @@ zskiplist *zslCreate(void) {
|
|||||||
|
|
||||||
void zslFreeNode(zskiplistNode *node) {
|
void zslFreeNode(zskiplistNode *node) {
|
||||||
decrRefCount(node->obj);
|
decrRefCount(node->obj);
|
||||||
zfree(node->forward);
|
|
||||||
zfree(node->span);
|
|
||||||
zfree(node);
|
zfree(node);
|
||||||
}
|
}
|
||||||
|
|
||||||
void zslFree(zskiplist *zsl) {
|
void zslFree(zskiplist *zsl) {
|
||||||
zskiplistNode *node = zsl->header->forward[0], *next;
|
zskiplistNode *node = zsl->header->level[0].forward, *next;
|
||||||
|
|
||||||
zfree(zsl->header->forward);
|
|
||||||
zfree(zsl->header->span);
|
|
||||||
zfree(zsl->header);
|
zfree(zsl->header);
|
||||||
while(node) {
|
while(node) {
|
||||||
next = node->forward[0];
|
next = node->level[0].forward;
|
||||||
zslFreeNode(node);
|
zslFreeNode(node);
|
||||||
node = next;
|
node = next;
|
||||||
}
|
}
|
||||||
@ -84,7 +71,7 @@ int zslRandomLevel(void) {
|
|||||||
return (level<ZSKIPLIST_MAXLEVEL) ? level : ZSKIPLIST_MAXLEVEL;
|
return (level<ZSKIPLIST_MAXLEVEL) ? level : ZSKIPLIST_MAXLEVEL;
|
||||||
}
|
}
|
||||||
|
|
||||||
void zslInsert(zskiplist *zsl, double score, robj *obj) {
|
zskiplistNode *zslInsert(zskiplist *zsl, double score, robj *obj) {
|
||||||
zskiplistNode *update[ZSKIPLIST_MAXLEVEL], *x;
|
zskiplistNode *update[ZSKIPLIST_MAXLEVEL], *x;
|
||||||
unsigned int rank[ZSKIPLIST_MAXLEVEL];
|
unsigned int rank[ZSKIPLIST_MAXLEVEL];
|
||||||
int i, level;
|
int i, level;
|
||||||
@ -93,13 +80,12 @@ void zslInsert(zskiplist *zsl, double score, robj *obj) {
|
|||||||
for (i = zsl->level-1; i >= 0; i--) {
|
for (i = zsl->level-1; i >= 0; i--) {
|
||||||
/* store rank that is crossed to reach the insert position */
|
/* store rank that is crossed to reach the insert position */
|
||||||
rank[i] = i == (zsl->level-1) ? 0 : rank[i+1];
|
rank[i] = i == (zsl->level-1) ? 0 : rank[i+1];
|
||||||
|
while (x->level[i].forward &&
|
||||||
while (x->forward[i] &&
|
(x->level[i].forward->score < score ||
|
||||||
(x->forward[i]->score < score ||
|
(x->level[i].forward->score == score &&
|
||||||
(x->forward[i]->score == score &&
|
compareStringObjects(x->level[i].forward->obj,obj) < 0))) {
|
||||||
compareStringObjects(x->forward[i]->obj,obj) < 0))) {
|
rank[i] += x->level[i].span;
|
||||||
rank[i] += i > 0 ? x->span[i-1] : 1;
|
x = x->level[i].forward;
|
||||||
x = x->forward[i];
|
|
||||||
}
|
}
|
||||||
update[i] = x;
|
update[i] = x;
|
||||||
}
|
}
|
||||||
@ -112,56 +98,51 @@ void zslInsert(zskiplist *zsl, double score, robj *obj) {
|
|||||||
for (i = zsl->level; i < level; i++) {
|
for (i = zsl->level; i < level; i++) {
|
||||||
rank[i] = 0;
|
rank[i] = 0;
|
||||||
update[i] = zsl->header;
|
update[i] = zsl->header;
|
||||||
update[i]->span[i-1] = zsl->length;
|
update[i]->level[i].span = zsl->length;
|
||||||
}
|
}
|
||||||
zsl->level = level;
|
zsl->level = level;
|
||||||
}
|
}
|
||||||
x = zslCreateNode(level,score,obj);
|
x = zslCreateNode(level,score,obj);
|
||||||
for (i = 0; i < level; i++) {
|
for (i = 0; i < level; i++) {
|
||||||
x->forward[i] = update[i]->forward[i];
|
x->level[i].forward = update[i]->level[i].forward;
|
||||||
update[i]->forward[i] = x;
|
update[i]->level[i].forward = x;
|
||||||
|
|
||||||
/* update span covered by update[i] as x is inserted here */
|
/* update span covered by update[i] as x is inserted here */
|
||||||
if (i > 0) {
|
x->level[i].span = update[i]->level[i].span - (rank[0] - rank[i]);
|
||||||
x->span[i-1] = update[i]->span[i-1] - (rank[0] - rank[i]);
|
update[i]->level[i].span = (rank[0] - rank[i]) + 1;
|
||||||
update[i]->span[i-1] = (rank[0] - rank[i]) + 1;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* increment span for untouched levels */
|
/* increment span for untouched levels */
|
||||||
for (i = level; i < zsl->level; i++) {
|
for (i = level; i < zsl->level; i++) {
|
||||||
update[i]->span[i-1]++;
|
update[i]->level[i].span++;
|
||||||
}
|
}
|
||||||
|
|
||||||
x->backward = (update[0] == zsl->header) ? NULL : update[0];
|
x->backward = (update[0] == zsl->header) ? NULL : update[0];
|
||||||
if (x->forward[0])
|
if (x->level[0].forward)
|
||||||
x->forward[0]->backward = x;
|
x->level[0].forward->backward = x;
|
||||||
else
|
else
|
||||||
zsl->tail = x;
|
zsl->tail = x;
|
||||||
zsl->length++;
|
zsl->length++;
|
||||||
|
return x;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Internal function used by zslDelete, zslDeleteByScore and zslDeleteByRank */
|
/* Internal function used by zslDelete, zslDeleteByScore and zslDeleteByRank */
|
||||||
void zslDeleteNode(zskiplist *zsl, zskiplistNode *x, zskiplistNode **update) {
|
void zslDeleteNode(zskiplist *zsl, zskiplistNode *x, zskiplistNode **update) {
|
||||||
int i;
|
int i;
|
||||||
for (i = 0; i < zsl->level; i++) {
|
for (i = 0; i < zsl->level; i++) {
|
||||||
if (update[i]->forward[i] == x) {
|
if (update[i]->level[i].forward == x) {
|
||||||
if (i > 0) {
|
update[i]->level[i].span += x->level[i].span - 1;
|
||||||
update[i]->span[i-1] += x->span[i-1] - 1;
|
update[i]->level[i].forward = x->level[i].forward;
|
||||||
}
|
|
||||||
update[i]->forward[i] = x->forward[i];
|
|
||||||
} else {
|
} else {
|
||||||
/* invariant: i > 0, because update[0]->forward[0]
|
update[i]->level[i].span -= 1;
|
||||||
* is always equal to x */
|
|
||||||
update[i]->span[i-1] -= 1;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (x->forward[0]) {
|
if (x->level[0].forward) {
|
||||||
x->forward[0]->backward = x->backward;
|
x->level[0].forward->backward = x->backward;
|
||||||
} else {
|
} else {
|
||||||
zsl->tail = x->backward;
|
zsl->tail = x->backward;
|
||||||
}
|
}
|
||||||
while(zsl->level > 1 && zsl->header->forward[zsl->level-1] == NULL)
|
while(zsl->level > 1 && zsl->header->level[zsl->level-1].forward == NULL)
|
||||||
zsl->level--;
|
zsl->level--;
|
||||||
zsl->length--;
|
zsl->length--;
|
||||||
}
|
}
|
||||||
@ -173,16 +154,16 @@ int zslDelete(zskiplist *zsl, double score, robj *obj) {
|
|||||||
|
|
||||||
x = zsl->header;
|
x = zsl->header;
|
||||||
for (i = zsl->level-1; i >= 0; i--) {
|
for (i = zsl->level-1; i >= 0; i--) {
|
||||||
while (x->forward[i] &&
|
while (x->level[i].forward &&
|
||||||
(x->forward[i]->score < score ||
|
(x->level[i].forward->score < score ||
|
||||||
(x->forward[i]->score == score &&
|
(x->level[i].forward->score == score &&
|
||||||
compareStringObjects(x->forward[i]->obj,obj) < 0)))
|
compareStringObjects(x->level[i].forward->obj,obj) < 0)))
|
||||||
x = x->forward[i];
|
x = x->level[i].forward;
|
||||||
update[i] = x;
|
update[i] = x;
|
||||||
}
|
}
|
||||||
/* We may have multiple elements with the same score, what we need
|
/* We may have multiple elements with the same score, what we need
|
||||||
* is to find the element with both the right score and object. */
|
* is to find the element with both the right score and object. */
|
||||||
x = x->forward[0];
|
x = x->level[0].forward;
|
||||||
if (x && score == x->score && equalStringObjects(x->obj,obj)) {
|
if (x && score == x->score && equalStringObjects(x->obj,obj)) {
|
||||||
zslDeleteNode(zsl, x, update);
|
zslDeleteNode(zsl, x, update);
|
||||||
zslFreeNode(x);
|
zslFreeNode(x);
|
||||||
@ -204,16 +185,16 @@ unsigned long zslDeleteRangeByScore(zskiplist *zsl, double min, double max, dict
|
|||||||
|
|
||||||
x = zsl->header;
|
x = zsl->header;
|
||||||
for (i = zsl->level-1; i >= 0; i--) {
|
for (i = zsl->level-1; i >= 0; i--) {
|
||||||
while (x->forward[i] && x->forward[i]->score < min)
|
while (x->level[i].forward && x->level[i].forward->score < min)
|
||||||
x = x->forward[i];
|
x = x->level[i].forward;
|
||||||
update[i] = x;
|
update[i] = x;
|
||||||
}
|
}
|
||||||
/* We may have multiple elements with the same score, what we need
|
/* We may have multiple elements with the same score, what we need
|
||||||
* is to find the element with both the right score and object. */
|
* is to find the element with both the right score and object. */
|
||||||
x = x->forward[0];
|
x = x->level[0].forward;
|
||||||
while (x && x->score <= max) {
|
while (x && x->score <= max) {
|
||||||
zskiplistNode *next = x->forward[0];
|
zskiplistNode *next = x->level[0].forward;
|
||||||
zslDeleteNode(zsl, x, update);
|
zslDeleteNode(zsl,x,update);
|
||||||
dictDelete(dict,x->obj);
|
dictDelete(dict,x->obj);
|
||||||
zslFreeNode(x);
|
zslFreeNode(x);
|
||||||
removed++;
|
removed++;
|
||||||
@ -231,18 +212,18 @@ unsigned long zslDeleteRangeByRank(zskiplist *zsl, unsigned int start, unsigned
|
|||||||
|
|
||||||
x = zsl->header;
|
x = zsl->header;
|
||||||
for (i = zsl->level-1; i >= 0; i--) {
|
for (i = zsl->level-1; i >= 0; i--) {
|
||||||
while (x->forward[i] && (traversed + (i > 0 ? x->span[i-1] : 1)) < start) {
|
while (x->level[i].forward && (traversed + x->level[i].span) < start) {
|
||||||
traversed += i > 0 ? x->span[i-1] : 1;
|
traversed += x->level[i].span;
|
||||||
x = x->forward[i];
|
x = x->level[i].forward;
|
||||||
}
|
}
|
||||||
update[i] = x;
|
update[i] = x;
|
||||||
}
|
}
|
||||||
|
|
||||||
traversed++;
|
traversed++;
|
||||||
x = x->forward[0];
|
x = x->level[0].forward;
|
||||||
while (x && traversed <= end) {
|
while (x && traversed <= end) {
|
||||||
zskiplistNode *next = x->forward[0];
|
zskiplistNode *next = x->level[0].forward;
|
||||||
zslDeleteNode(zsl, x, update);
|
zslDeleteNode(zsl,x,update);
|
||||||
dictDelete(dict,x->obj);
|
dictDelete(dict,x->obj);
|
||||||
zslFreeNode(x);
|
zslFreeNode(x);
|
||||||
removed++;
|
removed++;
|
||||||
@ -260,12 +241,12 @@ zskiplistNode *zslFirstWithScore(zskiplist *zsl, double score) {
|
|||||||
|
|
||||||
x = zsl->header;
|
x = zsl->header;
|
||||||
for (i = zsl->level-1; i >= 0; i--) {
|
for (i = zsl->level-1; i >= 0; i--) {
|
||||||
while (x->forward[i] && x->forward[i]->score < score)
|
while (x->level[i].forward && x->level[i].forward->score < score)
|
||||||
x = x->forward[i];
|
x = x->level[i].forward;
|
||||||
}
|
}
|
||||||
/* We may have multiple elements with the same score, what we need
|
/* We may have multiple elements with the same score, what we need
|
||||||
* is to find the element with both the right score and object. */
|
* is to find the element with both the right score and object. */
|
||||||
return x->forward[0];
|
return x->level[0].forward;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Find the rank for an element by both score and key.
|
/* Find the rank for an element by both score and key.
|
||||||
@ -279,12 +260,12 @@ unsigned long zslistTypeGetRank(zskiplist *zsl, double score, robj *o) {
|
|||||||
|
|
||||||
x = zsl->header;
|
x = zsl->header;
|
||||||
for (i = zsl->level-1; i >= 0; i--) {
|
for (i = zsl->level-1; i >= 0; i--) {
|
||||||
while (x->forward[i] &&
|
while (x->level[i].forward &&
|
||||||
(x->forward[i]->score < score ||
|
(x->level[i].forward->score < score ||
|
||||||
(x->forward[i]->score == score &&
|
(x->level[i].forward->score == score &&
|
||||||
compareStringObjects(x->forward[i]->obj,o) <= 0))) {
|
compareStringObjects(x->level[i].forward->obj,o) <= 0))) {
|
||||||
rank += i > 0 ? x->span[i-1] : 1;
|
rank += x->level[i].span;
|
||||||
x = x->forward[i];
|
x = x->level[i].forward;
|
||||||
}
|
}
|
||||||
|
|
||||||
/* x might be equal to zsl->header, so test if obj is non-NULL */
|
/* x might be equal to zsl->header, so test if obj is non-NULL */
|
||||||
@ -303,10 +284,10 @@ zskiplistNode* zslistTypeGetElementByRank(zskiplist *zsl, unsigned long rank) {
|
|||||||
|
|
||||||
x = zsl->header;
|
x = zsl->header;
|
||||||
for (i = zsl->level-1; i >= 0; i--) {
|
for (i = zsl->level-1; i >= 0; i--) {
|
||||||
while (x->forward[i] && (traversed + (i>0 ? x->span[i-1] : 1)) <= rank)
|
while (x->level[i].forward && (traversed + x->level[i].span) <= rank)
|
||||||
{
|
{
|
||||||
traversed += i > 0 ? x->span[i-1] : 1;
|
traversed += x->level[i].span;
|
||||||
x = x->forward[i];
|
x = x->level[i].forward;
|
||||||
}
|
}
|
||||||
if (traversed == rank) {
|
if (traversed == rank) {
|
||||||
return x;
|
return x;
|
||||||
@ -319,13 +300,11 @@ zskiplistNode* zslistTypeGetElementByRank(zskiplist *zsl, unsigned long rank) {
|
|||||||
* Sorted set commands
|
* Sorted set commands
|
||||||
*----------------------------------------------------------------------------*/
|
*----------------------------------------------------------------------------*/
|
||||||
|
|
||||||
/* This generic command implements both ZADD and ZINCRBY.
|
/* This generic command implements both ZADD and ZINCRBY. */
|
||||||
* scoreval is the score if the operation is a ZADD (doincrement == 0) or
|
void zaddGenericCommand(redisClient *c, robj *key, robj *ele, double score, int incr) {
|
||||||
* the increment if the operation is a ZINCRBY (doincrement == 1). */
|
|
||||||
void zaddGenericCommand(redisClient *c, robj *key, robj *ele, double scoreval, int doincrement) {
|
|
||||||
robj *zsetobj;
|
robj *zsetobj;
|
||||||
zset *zs;
|
zset *zs;
|
||||||
double *score;
|
zskiplistNode *znode;
|
||||||
|
|
||||||
zsetobj = lookupKeyWrite(c->db,key);
|
zsetobj = lookupKeyWrite(c->db,key);
|
||||||
if (zsetobj == NULL) {
|
if (zsetobj == NULL) {
|
||||||
@ -339,71 +318,72 @@ void zaddGenericCommand(redisClient *c, robj *key, robj *ele, double scoreval, i
|
|||||||
}
|
}
|
||||||
zs = zsetobj->ptr;
|
zs = zsetobj->ptr;
|
||||||
|
|
||||||
/* Ok now since we implement both ZADD and ZINCRBY here the code
|
/* Since both ZADD and ZINCRBY are implemented here, we need to increment
|
||||||
* needs to handle the two different conditions. It's all about setting
|
* the score first by the current score if ZINCRBY is called. */
|
||||||
* '*score', that is, the new score to set, to the right value. */
|
if (incr) {
|
||||||
score = zmalloc(sizeof(double));
|
|
||||||
if (doincrement) {
|
|
||||||
dictEntry *de;
|
|
||||||
|
|
||||||
/* Read the old score. If the element was not present starts from 0 */
|
/* Read the old score. If the element was not present starts from 0 */
|
||||||
de = dictFind(zs->dict,ele);
|
dictEntry *de = dictFind(zs->dict,ele);
|
||||||
if (de) {
|
if (de != NULL)
|
||||||
double *oldscore = dictGetEntryVal(de);
|
score += *(double*)dictGetEntryVal(de);
|
||||||
*score = *oldscore + scoreval;
|
|
||||||
} else {
|
if (isnan(score)) {
|
||||||
*score = scoreval;
|
|
||||||
}
|
|
||||||
if (isnan(*score)) {
|
|
||||||
addReplyError(c,"resulting score is not a number (NaN)");
|
addReplyError(c,"resulting score is not a number (NaN)");
|
||||||
zfree(score);
|
|
||||||
/* Note that we don't need to check if the zset may be empty and
|
/* Note that we don't need to check if the zset may be empty and
|
||||||
* should be removed here, as we can only obtain Nan as score if
|
* should be removed here, as we can only obtain Nan as score if
|
||||||
* there was already an element in the sorted set. */
|
* there was already an element in the sorted set. */
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
} else {
|
|
||||||
*score = scoreval;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/* What follows is a simple remove and re-insert operation that is common
|
/* We need to remove and re-insert the element when it was already present
|
||||||
* to both ZADD and ZINCRBY... */
|
* in the dictionary, to update the skiplist. Note that we delay adding a
|
||||||
if (dictAdd(zs->dict,ele,score) == DICT_OK) {
|
* pointer to the score because we want to reference the score in the
|
||||||
/* case 1: New element */
|
* skiplist node. */
|
||||||
|
if (dictAdd(zs->dict,ele,NULL) == DICT_OK) {
|
||||||
|
dictEntry *de;
|
||||||
|
|
||||||
|
/* New element */
|
||||||
incrRefCount(ele); /* added to hash */
|
incrRefCount(ele); /* added to hash */
|
||||||
zslInsert(zs->zsl,*score,ele);
|
znode = zslInsert(zs->zsl,score,ele);
|
||||||
incrRefCount(ele); /* added to skiplist */
|
incrRefCount(ele); /* added to skiplist */
|
||||||
|
|
||||||
|
/* Update the score in the dict entry */
|
||||||
|
de = dictFind(zs->dict,ele);
|
||||||
|
redisAssert(de != NULL);
|
||||||
|
dictGetEntryVal(de) = &znode->score;
|
||||||
touchWatchedKey(c->db,c->argv[1]);
|
touchWatchedKey(c->db,c->argv[1]);
|
||||||
server.dirty++;
|
server.dirty++;
|
||||||
if (doincrement)
|
if (incr)
|
||||||
addReplyDouble(c,*score);
|
addReplyDouble(c,score);
|
||||||
else
|
else
|
||||||
addReply(c,shared.cone);
|
addReply(c,shared.cone);
|
||||||
} else {
|
} else {
|
||||||
dictEntry *de;
|
dictEntry *de;
|
||||||
double *oldscore;
|
robj *curobj;
|
||||||
|
double *curscore;
|
||||||
|
int deleted;
|
||||||
|
|
||||||
/* case 2: Score update operation */
|
/* Update score */
|
||||||
de = dictFind(zs->dict,ele);
|
de = dictFind(zs->dict,ele);
|
||||||
redisAssert(de != NULL);
|
redisAssert(de != NULL);
|
||||||
oldscore = dictGetEntryVal(de);
|
curobj = dictGetEntryKey(de);
|
||||||
if (*score != *oldscore) {
|
curscore = dictGetEntryVal(de);
|
||||||
int deleted;
|
|
||||||
|
|
||||||
/* Remove and insert the element in the skip list with new score */
|
/* When the score is updated, reuse the existing string object to
|
||||||
deleted = zslDelete(zs->zsl,*oldscore,ele);
|
* prevent extra alloc/dealloc of strings on ZINCRBY. */
|
||||||
|
if (score != *curscore) {
|
||||||
|
deleted = zslDelete(zs->zsl,*curscore,curobj);
|
||||||
redisAssert(deleted != 0);
|
redisAssert(deleted != 0);
|
||||||
zslInsert(zs->zsl,*score,ele);
|
znode = zslInsert(zs->zsl,score,curobj);
|
||||||
incrRefCount(ele);
|
incrRefCount(curobj);
|
||||||
/* Update the score in the hash table */
|
|
||||||
dictReplace(zs->dict,ele,score);
|
/* Update the score in the current dict entry */
|
||||||
|
dictGetEntryVal(de) = &znode->score;
|
||||||
touchWatchedKey(c->db,c->argv[1]);
|
touchWatchedKey(c->db,c->argv[1]);
|
||||||
server.dirty++;
|
server.dirty++;
|
||||||
} else {
|
|
||||||
zfree(score);
|
|
||||||
}
|
}
|
||||||
if (doincrement)
|
if (incr)
|
||||||
addReplyDouble(c,*score);
|
addReplyDouble(c,score);
|
||||||
else
|
else
|
||||||
addReply(c,shared.czero);
|
addReply(c,shared.czero);
|
||||||
}
|
}
|
||||||
@ -425,7 +405,7 @@ void zremCommand(redisClient *c) {
|
|||||||
robj *zsetobj;
|
robj *zsetobj;
|
||||||
zset *zs;
|
zset *zs;
|
||||||
dictEntry *de;
|
dictEntry *de;
|
||||||
double *oldscore;
|
double curscore;
|
||||||
int deleted;
|
int deleted;
|
||||||
|
|
||||||
if ((zsetobj = lookupKeyWriteOrReply(c,c->argv[1],shared.czero)) == NULL ||
|
if ((zsetobj = lookupKeyWriteOrReply(c,c->argv[1],shared.czero)) == NULL ||
|
||||||
@ -438,8 +418,8 @@ void zremCommand(redisClient *c) {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
/* Delete from the skiplist */
|
/* Delete from the skiplist */
|
||||||
oldscore = dictGetEntryVal(de);
|
curscore = *(double*)dictGetEntryVal(de);
|
||||||
deleted = zslDelete(zs->zsl,*oldscore,c->argv[2]);
|
deleted = zslDelete(zs->zsl,curscore,c->argv[2]);
|
||||||
redisAssert(deleted != 0);
|
redisAssert(deleted != 0);
|
||||||
|
|
||||||
/* Delete from the hash table */
|
/* Delete from the hash table */
|
||||||
@ -553,6 +533,7 @@ void zunionInterGenericCommand(redisClient *c, robj *dstkey, int op) {
|
|||||||
zsetopsrc *src;
|
zsetopsrc *src;
|
||||||
robj *dstobj;
|
robj *dstobj;
|
||||||
zset *dstzset;
|
zset *dstzset;
|
||||||
|
zskiplistNode *znode;
|
||||||
dictIterator *di;
|
dictIterator *di;
|
||||||
dictEntry *de;
|
dictEntry *de;
|
||||||
int touched = 0;
|
int touched = 0;
|
||||||
@ -662,10 +643,10 @@ void zunionInterGenericCommand(redisClient *c, robj *dstkey, int op) {
|
|||||||
zfree(score);
|
zfree(score);
|
||||||
} else {
|
} else {
|
||||||
robj *o = dictGetEntryKey(de);
|
robj *o = dictGetEntryKey(de);
|
||||||
dictAdd(dstzset->dict,o,score);
|
znode = zslInsert(dstzset->zsl,*score,o);
|
||||||
incrRefCount(o); /* added to dictionary */
|
|
||||||
zslInsert(dstzset->zsl,*score,o);
|
|
||||||
incrRefCount(o); /* added to skiplist */
|
incrRefCount(o); /* added to skiplist */
|
||||||
|
dictAdd(dstzset->dict,o,&znode->score);
|
||||||
|
incrRefCount(o); /* added to dictionary */
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
dictReleaseIterator(di);
|
dictReleaseIterator(di);
|
||||||
@ -693,10 +674,10 @@ void zunionInterGenericCommand(redisClient *c, robj *dstkey, int op) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
robj *o = dictGetEntryKey(de);
|
robj *o = dictGetEntryKey(de);
|
||||||
dictAdd(dstzset->dict,o,score);
|
znode = zslInsert(dstzset->zsl,*score,o);
|
||||||
incrRefCount(o); /* added to dictionary */
|
|
||||||
zslInsert(dstzset->zsl,*score,o);
|
|
||||||
incrRefCount(o); /* added to skiplist */
|
incrRefCount(o); /* added to skiplist */
|
||||||
|
dictAdd(dstzset->dict,o,&znode->score);
|
||||||
|
incrRefCount(o); /* added to dictionary */
|
||||||
}
|
}
|
||||||
dictReleaseIterator(di);
|
dictReleaseIterator(di);
|
||||||
}
|
}
|
||||||
@ -778,7 +759,7 @@ void zrangeGenericCommand(redisClient *c, int reverse) {
|
|||||||
ln = start == 0 ? zsl->tail : zslistTypeGetElementByRank(zsl, llen-start);
|
ln = start == 0 ? zsl->tail : zslistTypeGetElementByRank(zsl, llen-start);
|
||||||
} else {
|
} else {
|
||||||
ln = start == 0 ?
|
ln = start == 0 ?
|
||||||
zsl->header->forward[0] : zslistTypeGetElementByRank(zsl, start+1);
|
zsl->header->level[0].forward : zslistTypeGetElementByRank(zsl, start+1);
|
||||||
}
|
}
|
||||||
|
|
||||||
/* Return the result in form of a multi-bulk reply */
|
/* Return the result in form of a multi-bulk reply */
|
||||||
@ -788,7 +769,7 @@ void zrangeGenericCommand(redisClient *c, int reverse) {
|
|||||||
addReplyBulk(c,ele);
|
addReplyBulk(c,ele);
|
||||||
if (withscores)
|
if (withscores)
|
||||||
addReplyDouble(c,ln->score);
|
addReplyDouble(c,ln->score);
|
||||||
ln = reverse ? ln->backward : ln->forward[0];
|
ln = reverse ? ln->backward : ln->level[0].forward;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -871,7 +852,7 @@ void genericZrangebyscoreCommand(redisClient *c, int justcount) {
|
|||||||
/* Get the first node with the score >= min, or with
|
/* Get the first node with the score >= min, or with
|
||||||
* score > min if 'minex' is true. */
|
* score > min if 'minex' is true. */
|
||||||
ln = zslFirstWithScore(zsl,min);
|
ln = zslFirstWithScore(zsl,min);
|
||||||
while (minex && ln && ln->score == min) ln = ln->forward[0];
|
while (minex && ln && ln->score == min) ln = ln->level[0].forward;
|
||||||
|
|
||||||
if (ln == NULL) {
|
if (ln == NULL) {
|
||||||
/* No element matching the speciifed interval */
|
/* No element matching the speciifed interval */
|
||||||
@ -889,7 +870,7 @@ void genericZrangebyscoreCommand(redisClient *c, int justcount) {
|
|||||||
while(ln && (maxex ? (ln->score < max) : (ln->score <= max))) {
|
while(ln && (maxex ? (ln->score < max) : (ln->score <= max))) {
|
||||||
if (offset) {
|
if (offset) {
|
||||||
offset--;
|
offset--;
|
||||||
ln = ln->forward[0];
|
ln = ln->level[0].forward;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
if (limit == 0) break;
|
if (limit == 0) break;
|
||||||
@ -899,7 +880,7 @@ void genericZrangebyscoreCommand(redisClient *c, int justcount) {
|
|||||||
if (withscores)
|
if (withscores)
|
||||||
addReplyDouble(c,ln->score);
|
addReplyDouble(c,ln->score);
|
||||||
}
|
}
|
||||||
ln = ln->forward[0];
|
ln = ln->level[0].forward;
|
||||||
rangelen++;
|
rangelen++;
|
||||||
if (limit > 0) limit--;
|
if (limit > 0) limit--;
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user