tyiu/damus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nostrdb: queue: switch to prot_queue_try_pop_all

Browse Source 
This allows you to `try pop` multiple items instead of 1

Signed-off-by: William Casarin <jb55@jb55.com>
This commit is contained in:
William Casarin
2024-02-05 16:47:10 -08:00
committed by Daniel D’Aquino
parent b8bef86ea1
commit c8d88058d4
1 changed files with 10 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
nostrdb/src/protected_queue.h
View File

@@ -154,7 +154,9 @@ static int prot_queue_push_all(struct prot_queue* q, void *data, int count)
* data - Pointer to where the popped data will be stored.
* Returns 1 if successful, 0 if the queue is empty.
*/
static inline int prot_queue_try_pop(struct prot_queue *q, void *data) {
static inline int prot_queue_try_pop_all(struct prot_queue *q, void *data, int max_items) {
int items_to_pop, items_until_end;
pthread_mutex_lock(&q->mutex);
if (q->count == 0) {
@@ -162,9 +164,13 @@ static inline int prot_queue_try_pop(struct prot_queue *q, void *data) {
return 0;
}
memcpy(data, &q->buf[q->head * q->elem_size], q->elem_size);
q->head = (q->head + 1) % prot_queue_capacity(q);
q->count--;
items_until_end = (q->buflen - q->head * q->elem_size) / q->elem_size;
items_to_pop = min(q->count, max_items);
items_to_pop = min(items_to_pop, items_until_end);
memcpy(data, &q->buf[q->head * q->elem_size], items_to_pop * q->elem_size);
q->head = (q->head + items_to_pop) % prot_queue_capacity(q);
q->count -= items_to_pop;
pthread_mutex_unlock(&q->mutex);
return 1;