caleb-brown/redict
1
0
Fork 0
mirror of https://codeberg.org/redict/redict.git synced 2025-01-23 08:38:27 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Streams: fix xreadgroup crash after xgroup SETID is sent.

Browse Source 
For issue #5111.
This commit is contained in:
dejun.xdj 2018-07-10 16:26:13 +08:00
parent 7239e9ca5f
commit 3f8a3efe5f
1 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
src/t_stream.c
View File

@ -905,24 +905,29 @@ size_t streamReplyWithRange(client *c, stream *s, streamID *start, streamID *end
* will not require extra lookups. We'll fix the problem later * will not require extra lookups. We'll fix the problem later
* if we find that there is already a entry for this ID. */ * if we find that there is already a entry for this ID. */
streamNACK *nack = streamCreateNACK(consumer); streamNACK *nack = streamCreateNACK(consumer);
int retval = 0; int retval_group_tryins = 0, retval_consumer_tryins = 0;
retval += raxTryInsert(group->pel,buf,sizeof(buf),nack,NULL); retval_group_tryins = raxTryInsert(group->pel,buf,sizeof(buf),nack,NULL);
retval += raxTryInsert(consumer->pel,buf,sizeof(buf),nack,NULL); retval_consumer_tryins = raxTryInsert(consumer->pel,buf,sizeof(buf),nack,NULL);
/* Now we can check if the entry was already busy, and /* Now we can check if the entry was already busy, and
* in that case reassign the entry to the new consumer. */ * in that case reassign the entry to the new consumer. */
if (retval == 0) { if (retval_group_tryins == 0) {
streamNACK *oldnack = raxFind(group->pel,buf,sizeof(buf));
serverAssert(oldnack != raxNotFound);
if (retval_consumer_tryins == 0) {
/* In this case, only need to update old nack
* delivery_time and delivery_count. */
oldnack->delivery_time = mstime();
oldnack->delivery_count++;
streamFreeNACK(nack); streamFreeNACK(nack);
nack = raxFind(group->pel,buf,sizeof(buf)); } else {
serverAssert(nack != raxNotFound); /* remove the old nack from original consumer */
raxRemove(nack->consumer->pel,buf,sizeof(buf),NULL); raxRemove(oldnack->consumer->pel,buf,sizeof(buf),NULL);
/* Update the consumer and idle time. */ /* Overwrite nack in group pel. */
nack->consumer = consumer; raxInsert(group->pel,buf,sizeof(buf),nack,NULL);
nack->delivery_time = mstime(); streamFreeNACK(oldnack);
nack->delivery_count++; }
/* Add the entry in the new consumer local PEL. */ } else if (retval_group_tryins == 1 && retval_consumer_tryins == 0) {
raxInsert(consumer->pel,buf,sizeof(buf),nack,NULL);
} else if (retval == 1) {
serverPanic("NACK half-created. Should not be possible."); serverPanic("NACK half-created. Should not be possible.");
} }