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

Threaded IO: 3rd version: use the mutex only to stop the thread.

Browse Source
This commit is contained in:
antirez 2019-03-25 16:33:23 +01:00
parent 6f4f36c0fb
commit ea35a81c42
1 changed files with 33 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
src/networking.c
View File

@ -2491,24 +2491,34 @@ void *IOThreadMain(void *myid) {
long id = (unsigned long)myid; long id = (unsigned long)myid;
while(1) { while(1) {
/* ... Wait for start ... */ /* Wait for start */
pthread_mutex_lock(&io_threads_mutex[id]); for (int j = 0; j < 1000000; j++) {
if (io_threads_pending[id]) { if (io_threads_pending[id] != 0) break;
if (tio_debug) printf("[%ld] %d to handle\n", id, (int)listLength(io_threads_list[id]));
/* ... Process ... */
listIter li;
listNode *ln;
listRewind(io_threads_list[id],&li);
while((ln = listNext(&li))) {
client *c = listNodeValue(ln);
writeToClient(c->fd,c,0);
io_threads_pending[id]--;
}
listEmpty(io_threads_list[id]);
} }
pthread_mutex_unlock(&io_threads_mutex[id]); /* Give the main thread a chance to stop this thread. */
if (io_threads_pending[id] == 0) {
pthread_mutex_lock(&io_threads_mutex[id]);
pthread_mutex_unlock(&io_threads_mutex[id]);
continue;
}
serverAssert(io_threads_pending[id] != 0);
if (tio_debug) printf("[%ld] %d to handle\n", id, (int)listLength(io_threads_list[id]));
/* Process: note that the main thread will never touch our list
* before we drop the pending count to 0. */
listIter li;
listNode *ln;
listRewind(io_threads_list[id],&li);
while((ln = listNext(&li))) {
client *c = listNodeValue(ln);
writeToClient(c->fd,c,0);
}
listEmpty(io_threads_list[id]);
io_threads_pending[id] = 0;
if (tio_debug) printf("[%ld] Done\n", id); if (tio_debug) printf("[%ld] Done\n", id);
} }
} }
@ -2572,13 +2582,17 @@ int handleClientsWithPendingWritesUsingThreads(void) {
client *c = listNodeValue(ln); client *c = listNodeValue(ln);
c->flags &= ~CLIENT_PENDING_WRITE; c->flags &= ~CLIENT_PENDING_WRITE;
int target_id = item_id % server.io_threads_num; int target_id = item_id % server.io_threads_num;
pthread_mutex_lock(&io_threads_mutex[target_id]);
listAddNodeTail(io_threads_list[target_id],c); listAddNodeTail(io_threads_list[target_id],c);
io_threads_pending[target_id]++;
pthread_mutex_unlock(&io_threads_mutex[target_id]);
item_id++; item_id++;
} }
/* Give the start condition to the waiting threads, by setting the
* start condition atomic var. */
for (int j = 0; j < server.io_threads_num; j++) {
int count = listLength(io_threads_list[j]);
io_threads_pending[j] = count;
}
/* Wait for all threads to end their work. */ /* Wait for all threads to end their work. */
while(1) { while(1) {
unsigned long pending = 0; unsigned long pending = 0;