]> git.saurik.com Git - redis.git/blame - src/t_list.c
dict.c: added two lower level methods for directly manipulating hash entries. This...
[redis.git] / src / t_list.c
CommitLineData
e2641e09 1#include "redis.h"
2
3/*-----------------------------------------------------------------------------
4 * List API
5 *----------------------------------------------------------------------------*/
6
7/* Check the argument length to see if it requires us to convert the ziplist
8 * to a real list. Only check raw-encoded objects because integer encoded
9 * objects are never too long. */
10void listTypeTryConversion(robj *subject, robj *value) {
11 if (subject->encoding != REDIS_ENCODING_ZIPLIST) return;
12 if (value->encoding == REDIS_ENCODING_RAW &&
13 sdslen(value->ptr) > server.list_max_ziplist_value)
14 listTypeConvert(subject,REDIS_ENCODING_LINKEDLIST);
15}
16
17void listTypePush(robj *subject, robj *value, int where) {
18 /* Check if we need to convert the ziplist */
19 listTypeTryConversion(subject,value);
20 if (subject->encoding == REDIS_ENCODING_ZIPLIST &&
21 ziplistLen(subject->ptr) >= server.list_max_ziplist_entries)
22 listTypeConvert(subject,REDIS_ENCODING_LINKEDLIST);
23
24 if (subject->encoding == REDIS_ENCODING_ZIPLIST) {
25 int pos = (where == REDIS_HEAD) ? ZIPLIST_HEAD : ZIPLIST_TAIL;
26 value = getDecodedObject(value);
27 subject->ptr = ziplistPush(subject->ptr,value->ptr,sdslen(value->ptr),pos);
28 decrRefCount(value);
29 } else if (subject->encoding == REDIS_ENCODING_LINKEDLIST) {
30 if (where == REDIS_HEAD) {
31 listAddNodeHead(subject->ptr,value);
32 } else {
33 listAddNodeTail(subject->ptr,value);
34 }
35 incrRefCount(value);
36 } else {
37 redisPanic("Unknown list encoding");
38 }
39}
40
41robj *listTypePop(robj *subject, int where) {
42 robj *value = NULL;
43 if (subject->encoding == REDIS_ENCODING_ZIPLIST) {
44 unsigned char *p;
45 unsigned char *vstr;
46 unsigned int vlen;
47 long long vlong;
48 int pos = (where == REDIS_HEAD) ? 0 : -1;
49 p = ziplistIndex(subject->ptr,pos);
50 if (ziplistGet(p,&vstr,&vlen,&vlong)) {
51 if (vstr) {
52 value = createStringObject((char*)vstr,vlen);
53 } else {
54 value = createStringObjectFromLongLong(vlong);
55 }
56 /* We only need to delete an element when it exists */
57 subject->ptr = ziplistDelete(subject->ptr,&p);
58 }
59 } else if (subject->encoding == REDIS_ENCODING_LINKEDLIST) {
60 list *list = subject->ptr;
61 listNode *ln;
62 if (where == REDIS_HEAD) {
63 ln = listFirst(list);
64 } else {
65 ln = listLast(list);
66 }
67 if (ln != NULL) {
68 value = listNodeValue(ln);
69 incrRefCount(value);
70 listDelNode(list,ln);
71 }
72 } else {
73 redisPanic("Unknown list encoding");
74 }
75 return value;
76}
77
78unsigned long listTypeLength(robj *subject) {
79 if (subject->encoding == REDIS_ENCODING_ZIPLIST) {
80 return ziplistLen(subject->ptr);
81 } else if (subject->encoding == REDIS_ENCODING_LINKEDLIST) {
82 return listLength((list*)subject->ptr);
83 } else {
84 redisPanic("Unknown list encoding");
85 }
86}
87
88/* Initialize an iterator at the specified index. */
89listTypeIterator *listTypeInitIterator(robj *subject, int index, unsigned char direction) {
90 listTypeIterator *li = zmalloc(sizeof(listTypeIterator));
91 li->subject = subject;
92 li->encoding = subject->encoding;
93 li->direction = direction;
94 if (li->encoding == REDIS_ENCODING_ZIPLIST) {
95 li->zi = ziplistIndex(subject->ptr,index);
96 } else if (li->encoding == REDIS_ENCODING_LINKEDLIST) {
97 li->ln = listIndex(subject->ptr,index);
98 } else {
99 redisPanic("Unknown list encoding");
100 }
101 return li;
102}
103
104/* Clean up the iterator. */
105void listTypeReleaseIterator(listTypeIterator *li) {
106 zfree(li);
107}
108
109/* Stores pointer to current the entry in the provided entry structure
110 * and advances the position of the iterator. Returns 1 when the current
111 * entry is in fact an entry, 0 otherwise. */
112int listTypeNext(listTypeIterator *li, listTypeEntry *entry) {
113 /* Protect from converting when iterating */
114 redisAssert(li->subject->encoding == li->encoding);
115
116 entry->li = li;
117 if (li->encoding == REDIS_ENCODING_ZIPLIST) {
118 entry->zi = li->zi;
119 if (entry->zi != NULL) {
120 if (li->direction == REDIS_TAIL)
121 li->zi = ziplistNext(li->subject->ptr,li->zi);
122 else
123 li->zi = ziplistPrev(li->subject->ptr,li->zi);
124 return 1;
125 }
126 } else if (li->encoding == REDIS_ENCODING_LINKEDLIST) {
127 entry->ln = li->ln;
128 if (entry->ln != NULL) {
129 if (li->direction == REDIS_TAIL)
130 li->ln = li->ln->next;
131 else
132 li->ln = li->ln->prev;
133 return 1;
134 }
135 } else {
136 redisPanic("Unknown list encoding");
137 }
138 return 0;
139}
140
141/* Return entry or NULL at the current position of the iterator. */
142robj *listTypeGet(listTypeEntry *entry) {
143 listTypeIterator *li = entry->li;
144 robj *value = NULL;
145 if (li->encoding == REDIS_ENCODING_ZIPLIST) {
146 unsigned char *vstr;
147 unsigned int vlen;
148 long long vlong;
149 redisAssert(entry->zi != NULL);
150 if (ziplistGet(entry->zi,&vstr,&vlen,&vlong)) {
151 if (vstr) {
152 value = createStringObject((char*)vstr,vlen);
153 } else {
154 value = createStringObjectFromLongLong(vlong);
155 }
156 }
157 } else if (li->encoding == REDIS_ENCODING_LINKEDLIST) {
158 redisAssert(entry->ln != NULL);
159 value = listNodeValue(entry->ln);
160 incrRefCount(value);
161 } else {
162 redisPanic("Unknown list encoding");
163 }
164 return value;
165}
166
167void listTypeInsert(listTypeEntry *entry, robj *value, int where) {
168 robj *subject = entry->li->subject;
169 if (entry->li->encoding == REDIS_ENCODING_ZIPLIST) {
170 value = getDecodedObject(value);
171 if (where == REDIS_TAIL) {
172 unsigned char *next = ziplistNext(subject->ptr,entry->zi);
173
174 /* When we insert after the current element, but the current element
175 * is the tail of the list, we need to do a push. */
176 if (next == NULL) {
177 subject->ptr = ziplistPush(subject->ptr,value->ptr,sdslen(value->ptr),REDIS_TAIL);
178 } else {
179 subject->ptr = ziplistInsert(subject->ptr,next,value->ptr,sdslen(value->ptr));
180 }
181 } else {
182 subject->ptr = ziplistInsert(subject->ptr,entry->zi,value->ptr,sdslen(value->ptr));
183 }
184 decrRefCount(value);
185 } else if (entry->li->encoding == REDIS_ENCODING_LINKEDLIST) {
186 if (where == REDIS_TAIL) {
187 listInsertNode(subject->ptr,entry->ln,value,AL_START_TAIL);
188 } else {
189 listInsertNode(subject->ptr,entry->ln,value,AL_START_HEAD);
190 }
191 incrRefCount(value);
192 } else {
193 redisPanic("Unknown list encoding");
194 }
195}
196
197/* Compare the given object with the entry at the current position. */
198int listTypeEqual(listTypeEntry *entry, robj *o) {
199 listTypeIterator *li = entry->li;
200 if (li->encoding == REDIS_ENCODING_ZIPLIST) {
eab0e26e 201 redisAssertWithInfo(NULL,o,o->encoding == REDIS_ENCODING_RAW);
e2641e09 202 return ziplistCompare(entry->zi,o->ptr,sdslen(o->ptr));
203 } else if (li->encoding == REDIS_ENCODING_LINKEDLIST) {
204 return equalStringObjects(o,listNodeValue(entry->ln));
205 } else {
206 redisPanic("Unknown list encoding");
207 }
208}
209
210/* Delete the element pointed to. */
211void listTypeDelete(listTypeEntry *entry) {
212 listTypeIterator *li = entry->li;
213 if (li->encoding == REDIS_ENCODING_ZIPLIST) {
214 unsigned char *p = entry->zi;
215 li->subject->ptr = ziplistDelete(li->subject->ptr,&p);
216
217 /* Update position of the iterator depending on the direction */
218 if (li->direction == REDIS_TAIL)
219 li->zi = p;
220 else
221 li->zi = ziplistPrev(li->subject->ptr,p);
222 } else if (entry->li->encoding == REDIS_ENCODING_LINKEDLIST) {
223 listNode *next;
224 if (li->direction == REDIS_TAIL)
225 next = entry->ln->next;
226 else
227 next = entry->ln->prev;
228 listDelNode(li->subject->ptr,entry->ln);
229 li->ln = next;
230 } else {
231 redisPanic("Unknown list encoding");
232 }
233}
234
235void listTypeConvert(robj *subject, int enc) {
236 listTypeIterator *li;
237 listTypeEntry entry;
eab0e26e 238 redisAssertWithInfo(NULL,subject,subject->type == REDIS_LIST);
e2641e09 239
240 if (enc == REDIS_ENCODING_LINKEDLIST) {
241 list *l = listCreate();
242 listSetFreeMethod(l,decrRefCount);
243
244 /* listTypeGet returns a robj with incremented refcount */
245 li = listTypeInitIterator(subject,0,REDIS_TAIL);
246 while (listTypeNext(li,&entry)) listAddNodeTail(l,listTypeGet(&entry));
247 listTypeReleaseIterator(li);
248
249 subject->encoding = REDIS_ENCODING_LINKEDLIST;
250 zfree(subject->ptr);
251 subject->ptr = l;
252 } else {
253 redisPanic("Unsupported list conversion");
254 }
255}
256
257/*-----------------------------------------------------------------------------
258 * List Commands
259 *----------------------------------------------------------------------------*/
260
261void pushGenericCommand(redisClient *c, int where) {
fb2feae5 262 int j, addlen = 0, pushed = 0;
e2641e09 263 robj *lobj = lookupKeyWrite(c->db,c->argv[1]);
fb2feae5 264 int may_have_waiting_clients = (lobj == NULL);
265
266 if (lobj && lobj->type != REDIS_LIST) {
267 addReply(c,shared.wrongtypeerr);
268 return;
269 }
270
271 for (j = 2; j < c->argc; j++) {
272 c->argv[j] = tryObjectEncoding(c->argv[j]);
273 if (may_have_waiting_clients) {
274 if (handleClientsWaitingListPush(c,c->argv[1],c->argv[j])) {
275 addlen++;
276 continue;
277 } else {
278 may_have_waiting_clients = 0;
279 }
e2641e09 280 }
fb2feae5 281 if (!lobj) {
282 lobj = createZiplistObject();
283 dbAdd(c->db,c->argv[1],lobj);
e2641e09 284 }
fb2feae5 285 listTypePush(lobj,c->argv[j],where);
286 pushed++;
e2641e09 287 }
fb2feae5 288 addReplyLongLong(c,addlen + (lobj ? listTypeLength(lobj) : 0));
289 if (pushed) signalModifiedKey(c->db,c->argv[1]);
290 server.dirty += pushed;
e2641e09 291}
292
293void lpushCommand(redisClient *c) {
294 pushGenericCommand(c,REDIS_HEAD);
295}
296
297void rpushCommand(redisClient *c) {
298 pushGenericCommand(c,REDIS_TAIL);
299}
300
301void pushxGenericCommand(redisClient *c, robj *refval, robj *val, int where) {
302 robj *subject;
303 listTypeIterator *iter;
304 listTypeEntry entry;
305 int inserted = 0;
306
307 if ((subject = lookupKeyReadOrReply(c,c->argv[1],shared.czero)) == NULL ||
308 checkType(c,subject,REDIS_LIST)) return;
309
310 if (refval != NULL) {
311 /* Note: we expect refval to be string-encoded because it is *not* the
312 * last argument of the multi-bulk LINSERT. */
eab0e26e 313 redisAssertWithInfo(c,refval,refval->encoding == REDIS_ENCODING_RAW);
e2641e09 314
315 /* We're not sure if this value can be inserted yet, but we cannot
316 * convert the list inside the iterator. We don't want to loop over
317 * the list twice (once to see if the value can be inserted and once
318 * to do the actual insert), so we assume this value can be inserted
319 * and convert the ziplist to a regular list if necessary. */
320 listTypeTryConversion(subject,val);
321
322 /* Seek refval from head to tail */
323 iter = listTypeInitIterator(subject,0,REDIS_TAIL);
324 while (listTypeNext(iter,&entry)) {
325 if (listTypeEqual(&entry,refval)) {
326 listTypeInsert(&entry,val,where);
327 inserted = 1;
328 break;
329 }
330 }
331 listTypeReleaseIterator(iter);
332
333 if (inserted) {
334 /* Check if the length exceeds the ziplist length threshold. */
335 if (subject->encoding == REDIS_ENCODING_ZIPLIST &&
336 ziplistLen(subject->ptr) > server.list_max_ziplist_entries)
337 listTypeConvert(subject,REDIS_ENCODING_LINKEDLIST);
cea8c5cd 338 signalModifiedKey(c->db,c->argv[1]);
e2641e09 339 server.dirty++;
340 } else {
341 /* Notify client of a failed insert */
342 addReply(c,shared.cnegone);
343 return;
344 }
345 } else {
346 listTypePush(subject,val,where);
cea8c5cd 347 signalModifiedKey(c->db,c->argv[1]);
e2641e09 348 server.dirty++;
349 }
350
b70d3555 351 addReplyLongLong(c,listTypeLength(subject));
e2641e09 352}
353
354void lpushxCommand(redisClient *c) {
75b41de8 355 c->argv[2] = tryObjectEncoding(c->argv[2]);
e2641e09 356 pushxGenericCommand(c,NULL,c->argv[2],REDIS_HEAD);
357}
358
359void rpushxCommand(redisClient *c) {
75b41de8 360 c->argv[2] = tryObjectEncoding(c->argv[2]);
e2641e09 361 pushxGenericCommand(c,NULL,c->argv[2],REDIS_TAIL);
362}
363
364void linsertCommand(redisClient *c) {
75b41de8 365 c->argv[4] = tryObjectEncoding(c->argv[4]);
e2641e09 366 if (strcasecmp(c->argv[2]->ptr,"after") == 0) {
367 pushxGenericCommand(c,c->argv[3],c->argv[4],REDIS_TAIL);
368 } else if (strcasecmp(c->argv[2]->ptr,"before") == 0) {
369 pushxGenericCommand(c,c->argv[3],c->argv[4],REDIS_HEAD);
370 } else {
371 addReply(c,shared.syntaxerr);
372 }
373}
374
375void llenCommand(redisClient *c) {
376 robj *o = lookupKeyReadOrReply(c,c->argv[1],shared.czero);
377 if (o == NULL || checkType(c,o,REDIS_LIST)) return;
b70d3555 378 addReplyLongLong(c,listTypeLength(o));
e2641e09 379}
380
381void lindexCommand(redisClient *c) {
382 robj *o = lookupKeyReadOrReply(c,c->argv[1],shared.nullbulk);
383 if (o == NULL || checkType(c,o,REDIS_LIST)) return;
384 int index = atoi(c->argv[2]->ptr);
385 robj *value = NULL;
386
387 if (o->encoding == REDIS_ENCODING_ZIPLIST) {
388 unsigned char *p;
389 unsigned char *vstr;
390 unsigned int vlen;
391 long long vlong;
392 p = ziplistIndex(o->ptr,index);
393 if (ziplistGet(p,&vstr,&vlen,&vlong)) {
394 if (vstr) {
395 value = createStringObject((char*)vstr,vlen);
396 } else {
397 value = createStringObjectFromLongLong(vlong);
398 }
399 addReplyBulk(c,value);
400 decrRefCount(value);
401 } else {
402 addReply(c,shared.nullbulk);
403 }
404 } else if (o->encoding == REDIS_ENCODING_LINKEDLIST) {
405 listNode *ln = listIndex(o->ptr,index);
406 if (ln != NULL) {
407 value = listNodeValue(ln);
408 addReplyBulk(c,value);
409 } else {
410 addReply(c,shared.nullbulk);
411 }
412 } else {
413 redisPanic("Unknown list encoding");
414 }
415}
416
417void lsetCommand(redisClient *c) {
418 robj *o = lookupKeyWriteOrReply(c,c->argv[1],shared.nokeyerr);
419 if (o == NULL || checkType(c,o,REDIS_LIST)) return;
420 int index = atoi(c->argv[2]->ptr);
75b41de8 421 robj *value = (c->argv[3] = tryObjectEncoding(c->argv[3]));
e2641e09 422
423 listTypeTryConversion(o,value);
424 if (o->encoding == REDIS_ENCODING_ZIPLIST) {
425 unsigned char *p, *zl = o->ptr;
426 p = ziplistIndex(zl,index);
427 if (p == NULL) {
428 addReply(c,shared.outofrangeerr);
429 } else {
430 o->ptr = ziplistDelete(o->ptr,&p);
431 value = getDecodedObject(value);
432 o->ptr = ziplistInsert(o->ptr,p,value->ptr,sdslen(value->ptr));
433 decrRefCount(value);
434 addReply(c,shared.ok);
cea8c5cd 435 signalModifiedKey(c->db,c->argv[1]);
e2641e09 436 server.dirty++;
437 }
438 } else if (o->encoding == REDIS_ENCODING_LINKEDLIST) {
439 listNode *ln = listIndex(o->ptr,index);
440 if (ln == NULL) {
441 addReply(c,shared.outofrangeerr);
442 } else {
443 decrRefCount((robj*)listNodeValue(ln));
444 listNodeValue(ln) = value;
445 incrRefCount(value);
446 addReply(c,shared.ok);
cea8c5cd 447 signalModifiedKey(c->db,c->argv[1]);
e2641e09 448 server.dirty++;
449 }
450 } else {
451 redisPanic("Unknown list encoding");
452 }
453}
454
455void popGenericCommand(redisClient *c, int where) {
456 robj *o = lookupKeyWriteOrReply(c,c->argv[1],shared.nullbulk);
457 if (o == NULL || checkType(c,o,REDIS_LIST)) return;
458
459 robj *value = listTypePop(o,where);
460 if (value == NULL) {
461 addReply(c,shared.nullbulk);
462 } else {
463 addReplyBulk(c,value);
464 decrRefCount(value);
465 if (listTypeLength(o) == 0) dbDelete(c->db,c->argv[1]);
cea8c5cd 466 signalModifiedKey(c->db,c->argv[1]);
e2641e09 467 server.dirty++;
468 }
469}
470
471void lpopCommand(redisClient *c) {
472 popGenericCommand(c,REDIS_HEAD);
473}
474
475void rpopCommand(redisClient *c) {
476 popGenericCommand(c,REDIS_TAIL);
477}
478
479void lrangeCommand(redisClient *c) {
d51ebef5 480 robj *o;
e2641e09 481 int start = atoi(c->argv[2]->ptr);
482 int end = atoi(c->argv[3]->ptr);
483 int llen;
d51ebef5 484 int rangelen;
e2641e09 485
486 if ((o = lookupKeyReadOrReply(c,c->argv[1],shared.emptymultibulk)) == NULL
487 || checkType(c,o,REDIS_LIST)) return;
488 llen = listTypeLength(o);
489
490 /* convert negative indexes */
491 if (start < 0) start = llen+start;
492 if (end < 0) end = llen+end;
493 if (start < 0) start = 0;
e2641e09 494
d0a4e24e
PN
495 /* Invariant: start >= 0, so this test will be true when end < 0.
496 * The range is empty when start > end or start >= length. */
e2641e09 497 if (start > end || start >= llen) {
e2641e09 498 addReply(c,shared.emptymultibulk);
499 return;
500 }
501 if (end >= llen) end = llen-1;
502 rangelen = (end-start)+1;
503
504 /* Return the result in form of a multi-bulk reply */
0537e7bf 505 addReplyMultiBulkLen(c,rangelen);
d51ebef5 506 if (o->encoding == REDIS_ENCODING_ZIPLIST) {
507 unsigned char *p = ziplistIndex(o->ptr,start);
508 unsigned char *vstr;
509 unsigned int vlen;
510 long long vlong;
511
512 while(rangelen--) {
513 ziplistGet(p,&vstr,&vlen,&vlong);
514 if (vstr) {
515 addReplyBulkCBuffer(c,vstr,vlen);
516 } else {
517 addReplyBulkLongLong(c,vlong);
518 }
519 p = ziplistNext(o->ptr,p);
520 }
521 } else if (o->encoding == REDIS_ENCODING_LINKEDLIST) {
7cfeb8cc 522 listNode *ln;
523
524 /* If we are nearest to the end of the list, reach the element
525 * starting from tail and going backward, as it is faster. */
526 if (start > llen/2) start -= llen;
527 ln = listIndex(o->ptr,start);
d51ebef5 528
529 while(rangelen--) {
530 addReplyBulk(c,ln->value);
531 ln = ln->next;
532 }
533 } else {
534 redisPanic("List encoding is not LINKEDLIST nor ZIPLIST!");
e2641e09 535 }
e2641e09 536}
537
538void ltrimCommand(redisClient *c) {
539 robj *o;
540 int start = atoi(c->argv[2]->ptr);
541 int end = atoi(c->argv[3]->ptr);
542 int llen;
543 int j, ltrim, rtrim;
544 list *list;
545 listNode *ln;
546
547 if ((o = lookupKeyWriteOrReply(c,c->argv[1],shared.ok)) == NULL ||
548 checkType(c,o,REDIS_LIST)) return;
549 llen = listTypeLength(o);
550
551 /* convert negative indexes */
552 if (start < 0) start = llen+start;
553 if (end < 0) end = llen+end;
554 if (start < 0) start = 0;
e2641e09 555
d0a4e24e
PN
556 /* Invariant: start >= 0, so this test will be true when end < 0.
557 * The range is empty when start > end or start >= length. */
e2641e09 558 if (start > end || start >= llen) {
559 /* Out of range start or start > end result in empty list */
560 ltrim = llen;
561 rtrim = 0;
562 } else {
563 if (end >= llen) end = llen-1;
564 ltrim = start;
565 rtrim = llen-end-1;
566 }
567
568 /* Remove list elements to perform the trim */
569 if (o->encoding == REDIS_ENCODING_ZIPLIST) {
570 o->ptr = ziplistDeleteRange(o->ptr,0,ltrim);
571 o->ptr = ziplistDeleteRange(o->ptr,-rtrim,rtrim);
572 } else if (o->encoding == REDIS_ENCODING_LINKEDLIST) {
573 list = o->ptr;
574 for (j = 0; j < ltrim; j++) {
575 ln = listFirst(list);
576 listDelNode(list,ln);
577 }
578 for (j = 0; j < rtrim; j++) {
579 ln = listLast(list);
580 listDelNode(list,ln);
581 }
582 } else {
583 redisPanic("Unknown list encoding");
584 }
585 if (listTypeLength(o) == 0) dbDelete(c->db,c->argv[1]);
cea8c5cd 586 signalModifiedKey(c->db,c->argv[1]);
e2641e09 587 server.dirty++;
588 addReply(c,shared.ok);
589}
590
591void lremCommand(redisClient *c) {
75b41de8
PN
592 robj *subject, *obj;
593 obj = c->argv[3] = tryObjectEncoding(c->argv[3]);
e2641e09 594 int toremove = atoi(c->argv[2]->ptr);
595 int removed = 0;
596 listTypeEntry entry;
597
598 subject = lookupKeyWriteOrReply(c,c->argv[1],shared.czero);
599 if (subject == NULL || checkType(c,subject,REDIS_LIST)) return;
600
601 /* Make sure obj is raw when we're dealing with a ziplist */
602 if (subject->encoding == REDIS_ENCODING_ZIPLIST)
603 obj = getDecodedObject(obj);
604
605 listTypeIterator *li;
606 if (toremove < 0) {
607 toremove = -toremove;
608 li = listTypeInitIterator(subject,-1,REDIS_HEAD);
609 } else {
610 li = listTypeInitIterator(subject,0,REDIS_TAIL);
611 }
612
613 while (listTypeNext(li,&entry)) {
614 if (listTypeEqual(&entry,obj)) {
615 listTypeDelete(&entry);
616 server.dirty++;
617 removed++;
618 if (toremove && removed == toremove) break;
619 }
620 }
621 listTypeReleaseIterator(li);
622
623 /* Clean up raw encoded object */
624 if (subject->encoding == REDIS_ENCODING_ZIPLIST)
625 decrRefCount(obj);
626
627 if (listTypeLength(subject) == 0) dbDelete(c->db,c->argv[1]);
b70d3555 628 addReplyLongLong(c,removed);
cea8c5cd 629 if (removed) signalModifiedKey(c->db,c->argv[1]);
e2641e09 630}
631
632/* This is the semantic of this command:
633 * RPOPLPUSH srclist dstlist:
ac06fc01
PN
634 * IF LLEN(srclist) > 0
635 * element = RPOP srclist
636 * LPUSH dstlist element
637 * RETURN element
638 * ELSE
639 * RETURN nil
640 * END
e2641e09 641 * END
642 *
643 * The idea is to be able to get an element from a list in a reliable way
644 * since the element is not just returned but pushed against another list
645 * as well. This command was originally proposed by Ezra Zygmuntowicz.
646 */
ac06fc01 647
c1c9d551 648void rpoplpushHandlePush(redisClient *origclient, redisClient *c, robj *dstkey, robj *dstobj, robj *value) {
649 robj *aux;
650
c47d152c 651 if (!handleClientsWaitingListPush(origclient,dstkey,value)) {
ac06fc01
PN
652 /* Create the list if the key does not exist */
653 if (!dstobj) {
654 dstobj = createZiplistObject();
655 dbAdd(c->db,dstkey,dstobj);
656 } else {
cea8c5cd 657 signalModifiedKey(c->db,dstkey);
ac06fc01
PN
658 }
659 listTypePush(dstobj,value,REDIS_HEAD);
c1c9d551 660 /* If we are pushing as a result of LPUSH against a key
48082cc0 661 * watched by BRPOPLPUSH, we need to rewrite the command vector
662 * as an LPUSH.
663 *
664 * If this is called directly by RPOPLPUSH (either directly
c1c9d551 665 * or via a BRPOPLPUSH where the popped list exists)
48082cc0 666 * we should replicate the RPOPLPUSH command itself. */
c1c9d551 667 if (c != origclient) {
668 aux = createStringObject("LPUSH",5);
669 rewriteClientCommandVector(origclient,3,aux,dstkey,value);
670 decrRefCount(aux);
671 } else {
672 /* Make sure to always use RPOPLPUSH in the replication / AOF,
673 * even if the original command was BRPOPLPUSH. */
674 aux = createStringObject("RPOPLPUSH",9);
675 rewriteClientCommandVector(origclient,3,aux,c->argv[1],c->argv[2]);
676 decrRefCount(aux);
677 }
678 server.dirty++;
ac06fc01
PN
679 }
680
681 /* Always send the pushed value to the client. */
682 addReplyBulk(c,value);
683}
684
8a979f03 685void rpoplpushCommand(redisClient *c) {
e2641e09 686 robj *sobj, *value;
687 if ((sobj = lookupKeyWriteOrReply(c,c->argv[1],shared.nullbulk)) == NULL ||
688 checkType(c,sobj,REDIS_LIST)) return;
689
690 if (listTypeLength(sobj) == 0) {
691 addReply(c,shared.nullbulk);
692 } else {
693 robj *dobj = lookupKeyWrite(c->db,c->argv[2]);
c1c9d551 694 robj *touchedkey = c->argv[1];
695
e2641e09 696 if (dobj && checkType(c,dobj,REDIS_LIST)) return;
697 value = listTypePop(sobj,REDIS_TAIL);
c1c9d551 698 /* We saved touched key, and protect it, since rpoplpushHandlePush
699 * may change the client command argument vector. */
700 incrRefCount(touchedkey);
701 rpoplpushHandlePush(c,c,c->argv[2],dobj,value);
e2641e09 702
703 /* listTypePop returns an object with its refcount incremented */
704 decrRefCount(value);
705
706 /* Delete the source list when it is empty */
c1c9d551 707 if (listTypeLength(sobj) == 0) dbDelete(c->db,touchedkey);
708 signalModifiedKey(c->db,touchedkey);
709 decrRefCount(touchedkey);
e2641e09 710 server.dirty++;
711 }
712}
713
714/*-----------------------------------------------------------------------------
715 * Blocking POP operations
716 *----------------------------------------------------------------------------*/
717
718/* Currently Redis blocking operations support is limited to list POP ops,
719 * so the current implementation is not fully generic, but it is also not
720 * completely specific so it will not require a rewrite to support new
721 * kind of blocking operations in the future.
722 *
723 * Still it's important to note that list blocking operations can be already
724 * used as a notification mechanism in order to implement other blocking
725 * operations at application level, so there must be a very strong evidence
726 * of usefulness and generality before new blocking operations are implemented.
727 *
728 * This is how the current blocking POP works, we use BLPOP as example:
729 * - If the user calls BLPOP and the key exists and contains a non empty list
730 * then LPOP is called instead. So BLPOP is semantically the same as LPOP
731 * if there is not to block.
732 * - If instead BLPOP is called and the key does not exists or the list is
733 * empty we need to block. In order to do so we remove the notification for
734 * new data to read in the client socket (so that we'll not serve new
735 * requests if the blocking request is not served). Also we put the client
736 * in a dictionary (db->blocking_keys) mapping keys to a list of clients
737 * blocking for this keys.
738 * - If a PUSH operation against a key with blocked clients waiting is
739 * performed, we serve the first in the list: basically instead to push
740 * the new element inside the list we return it to the (first / oldest)
741 * blocking client, unblock the client, and remove it form the list.
742 *
743 * The above comment and the source code should be enough in order to understand
744 * the implementation and modify / fix it later.
745 */
746
747/* Set a client in blocking mode for the specified key, with the specified
748 * timeout */
ba3b4741 749void blockForKeys(redisClient *c, robj **keys, int numkeys, time_t timeout, robj *target) {
e2641e09 750 dictEntry *de;
751 list *l;
752 int j;
753
e3c51c4b
DJMM
754 c->bpop.keys = zmalloc(sizeof(robj*)*numkeys);
755 c->bpop.count = numkeys;
756 c->bpop.timeout = timeout;
757 c->bpop.target = target;
ba3b4741
DJMM
758
759 if (target != NULL) {
ecf94014 760 incrRefCount(target);
ba3b4741
DJMM
761 }
762
e2641e09 763 for (j = 0; j < numkeys; j++) {
764 /* Add the key in the client structure, to map clients -> keys */
e3c51c4b 765 c->bpop.keys[j] = keys[j];
e2641e09 766 incrRefCount(keys[j]);
767
768 /* And in the other "side", to map keys -> clients */
769 de = dictFind(c->db->blocking_keys,keys[j]);
770 if (de == NULL) {
771 int retval;
772
773 /* For every key we take a list of clients blocked for it */
774 l = listCreate();
775 retval = dictAdd(c->db->blocking_keys,keys[j],l);
776 incrRefCount(keys[j]);
eab0e26e 777 redisAssertWithInfo(c,keys[j],retval == DICT_OK);
e2641e09 778 } else {
779 l = dictGetEntryVal(de);
780 }
781 listAddNodeTail(l,c);
782 }
783 /* Mark the client as a blocked client */
784 c->flags |= REDIS_BLOCKED;
5fa95ad7 785 server.bpop_blocked_clients++;
e2641e09 786}
787
788/* Unblock a client that's waiting in a blocking operation such as BLPOP */
789void unblockClientWaitingData(redisClient *c) {
790 dictEntry *de;
791 list *l;
792 int j;
793
eab0e26e 794 redisAssertWithInfo(c,NULL,c->bpop.keys != NULL);
e2641e09 795 /* The client may wait for multiple keys, so unblock it for every key. */
e3c51c4b 796 for (j = 0; j < c->bpop.count; j++) {
e2641e09 797 /* Remove this client from the list of clients waiting for this key. */
e3c51c4b 798 de = dictFind(c->db->blocking_keys,c->bpop.keys[j]);
eab0e26e 799 redisAssertWithInfo(c,c->bpop.keys[j],de != NULL);
e2641e09 800 l = dictGetEntryVal(de);
801 listDelNode(l,listSearchKey(l,c));
802 /* If the list is empty we need to remove it to avoid wasting memory */
803 if (listLength(l) == 0)
e3c51c4b
DJMM
804 dictDelete(c->db->blocking_keys,c->bpop.keys[j]);
805 decrRefCount(c->bpop.keys[j]);
e2641e09 806 }
ba3b4741 807
e2641e09 808 /* Cleanup the client structure */
e3c51c4b
DJMM
809 zfree(c->bpop.keys);
810 c->bpop.keys = NULL;
c1c9d551 811 if (c->bpop.target) decrRefCount(c->bpop.target);
e3c51c4b 812 c->bpop.target = NULL;
3bcffcbe
PN
813 c->flags &= ~REDIS_BLOCKED;
814 c->flags |= REDIS_UNBLOCKED;
5fa95ad7 815 server.bpop_blocked_clients--;
a4ce7581 816 listAddNodeTail(server.unblocked_clients,c);
e2641e09 817}
818
819/* This should be called from any function PUSHing into lists.
820 * 'c' is the "pushing client", 'key' is the key it is pushing data against,
821 * 'ele' is the element pushed.
822 *
823 * If the function returns 0 there was no client waiting for a list push
824 * against this key.
825 *
826 * If the function returns 1 there was a client waiting for a list push
827 * against this key, the element was passed to this client thus it's not
828 * needed to actually add it to the list and the caller should return asap. */
829int handleClientsWaitingListPush(redisClient *c, robj *key, robj *ele) {
830 struct dictEntry *de;
831 redisClient *receiver;
8a88c368
PN
832 int numclients;
833 list *clients;
e2641e09 834 listNode *ln;
8a88c368 835 robj *dstkey, *dstobj;
e2641e09 836
837 de = dictFind(c->db->blocking_keys,key);
838 if (de == NULL) return 0;
8a88c368
PN
839 clients = dictGetEntryVal(de);
840 numclients = listLength(clients);
841
842 /* Try to handle the push as long as there are clients waiting for a push.
843 * Note that "numclients" is used because the list of clients waiting for a
844 * push on "key" is deleted by unblockClient() when empty.
845 *
846 * This loop will have more than 1 iteration when there is a BRPOPLPUSH
847 * that cannot push the target list because it does not contain a list. If
848 * this happens, it simply tries the next client waiting for a push. */
849 while (numclients--) {
850 ln = listFirst(clients);
eab0e26e 851 redisAssertWithInfo(c,key,ln != NULL);
8a88c368
PN
852 receiver = ln->value;
853 dstkey = receiver->bpop.target;
854
c1c9d551 855 /* Protect receiver->bpop.target, that will be freed by
856 * the next unblockClientWaitingData() call. */
857 if (dstkey) incrRefCount(dstkey);
858
8a88c368
PN
859 /* This should remove the first element of the "clients" list. */
860 unblockClientWaitingData(receiver);
8a88c368
PN
861
862 if (dstkey == NULL) {
863 /* BRPOP/BLPOP */
864 addReplyMultiBulkLen(receiver,2);
865 addReplyBulk(receiver,key);
866 addReplyBulk(receiver,ele);
c1c9d551 867 return 1; /* Serve just the first client as in B[RL]POP semantics */
8a88c368 868 } else {
554a5dd2 869 /* BRPOPLPUSH, note that receiver->db is always equal to c->db. */
8a88c368 870 dstobj = lookupKeyWrite(receiver->db,dstkey);
c1c9d551 871 if (!(dstobj && checkType(receiver,dstobj,REDIS_LIST))) {
872 rpoplpushHandlePush(c,receiver,dstkey,dstobj,ele);
8a88c368
PN
873 decrRefCount(dstkey);
874 return 1;
875 }
c1c9d551 876 decrRefCount(dstkey);
8a88c368 877 }
b2a7fd0c 878 }
e2641e09 879
8a88c368 880 return 0;
e2641e09 881}
882
c8a0070a
PN
883int getTimeoutFromObjectOrReply(redisClient *c, robj *object, time_t *timeout) {
884 long tval;
59bd44d1 885
c8a0070a
PN
886 if (getLongFromObjectOrReply(c,object,&tval,
887 "timeout is not an integer or out of range") != REDIS_OK)
59bd44d1 888 return REDIS_ERR;
59bd44d1 889
c8a0070a
PN
890 if (tval < 0) {
891 addReplyError(c,"timeout is negative");
59bd44d1
DJMM
892 return REDIS_ERR;
893 }
894
c8a0070a
PN
895 if (tval > 0) tval += time(NULL);
896 *timeout = tval;
59bd44d1
DJMM
897
898 return REDIS_OK;
e2641e09 899}
900
901/* Blocking RPOP/LPOP */
902void blockingPopGenericCommand(redisClient *c, int where) {
903 robj *o;
904 time_t timeout;
905 int j;
906
c8a0070a 907 if (getTimeoutFromObjectOrReply(c,c->argv[c->argc-1],&timeout) != REDIS_OK)
94364d53 908 return;
94364d53 909
e2641e09 910 for (j = 1; j < c->argc-1; j++) {
911 o = lookupKeyWrite(c->db,c->argv[j]);
912 if (o != NULL) {
913 if (o->type != REDIS_LIST) {
914 addReply(c,shared.wrongtypeerr);
915 return;
916 } else {
917 if (listTypeLength(o) != 0) {
918 /* If the list contains elements fall back to the usual
919 * non-blocking POP operation */
09e2d9ee 920 struct redisCommand *orig_cmd;
e2641e09 921 robj *argv[2], **orig_argv;
922 int orig_argc;
923
924 /* We need to alter the command arguments before to call
925 * popGenericCommand() as the command takes a single key. */
926 orig_argv = c->argv;
927 orig_argc = c->argc;
09e2d9ee 928 orig_cmd = c->cmd;
e2641e09 929 argv[1] = c->argv[j];
930 c->argv = argv;
931 c->argc = 2;
932
933 /* Also the return value is different, we need to output
934 * the multi bulk reply header and the key name. The
935 * "real" command will add the last element (the value)
936 * for us. If this souds like an hack to you it's just
937 * because it is... */
0537e7bf 938 addReplyMultiBulkLen(c,2);
e2641e09 939 addReplyBulk(c,argv[1]);
ba3b4741 940
e2641e09 941 popGenericCommand(c,where);
942
943 /* Fix the client structure with the original stuff */
944 c->argv = orig_argv;
945 c->argc = orig_argc;
09e2d9ee 946 c->cmd = orig_cmd;
b2a7fd0c 947
e2641e09 948 return;
949 }
950 }
951 }
952 }
94364d53 953
fb92ecec 954 /* If we are inside a MULTI/EXEC and the list is empty the only thing
955 * we can do is treating it as a timeout (even with timeout 0). */
956 if (c->flags & REDIS_MULTI) {
957 addReply(c,shared.nullmultibulk);
958 return;
959 }
960
e2641e09 961 /* If the list is empty or the key does not exists we must block */
ba3b4741 962 blockForKeys(c, c->argv + 1, c->argc - 2, timeout, NULL);
e2641e09 963}
964
965void blpopCommand(redisClient *c) {
966 blockingPopGenericCommand(c,REDIS_HEAD);
967}
968
969void brpopCommand(redisClient *c) {
970 blockingPopGenericCommand(c,REDIS_TAIL);
971}
b2a7fd0c
DJMM
972
973void brpoplpushCommand(redisClient *c) {
ba3b4741 974 time_t timeout;
b2a7fd0c 975
c8a0070a 976 if (getTimeoutFromObjectOrReply(c,c->argv[3],&timeout) != REDIS_OK)
ba3b4741 977 return;
ba3b4741
DJMM
978
979 robj *key = lookupKeyWrite(c->db, c->argv[1]);
980
ba3b4741 981 if (key == NULL) {
ba3b4741 982 if (c->flags & REDIS_MULTI) {
7c25a43a
DJMM
983
984 /* Blocking against an empty list in a multi state
985 * returns immediately. */
d5870d7a 986 addReply(c, shared.nullbulk);
ba3b4741 987 } else {
7c25a43a 988 /* The list is empty and the client blocks. */
ba3b4741
DJMM
989 blockForKeys(c, c->argv + 1, 1, timeout, c->argv[2]);
990 }
ba3b4741 991 } else {
7c25a43a
DJMM
992 if (key->type != REDIS_LIST) {
993 addReply(c, shared.wrongtypeerr);
994 } else {
995
996 /* The list exists and has elements, so
997 * the regular rpoplpushCommand is executed. */
eab0e26e 998 redisAssertWithInfo(c,key,listTypeLength(key) > 0);
7c25a43a
DJMM
999 rpoplpushCommand(c);
1000 }
ba3b4741 1001 }
b2a7fd0c 1002}