-
Notifications
You must be signed in to change notification settings - Fork 1
/
queue.c
66 lines (53 loc) · 1.31 KB
/
queue.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#include "queue.h"
uint8_t queue_init(queue_t *queue, MemPool *mem_pool)
{
queue_node_t *head_node = NULL;
head_node = (queue_node_t *)mem_pool_alloc(mem_pool, queue_node_size
#if MEM_POOL_DEBUG
, __FUNCTION__
#endif
);
if(!head_node) return QUEUE_ERR;
head_node->next = head_node->prev = NULL;
queue->length = 0;
queue->head = queue->tail = head_node;
splock_init(&queue->head_lock);
splock_init(&queue->tail_lock);
return QUEUE_OK;
}
void enqueue(queue_t *queue, queue_node_t *node)
{
splock_lock(&queue->tail_lock);
node->prev = queue->tail;
node->next = NULL;
queue->tail->next = node;
queue->tail = node;
++queue->length;
splock_unlock(&queue->tail_lock);
}
queue_node_t *dequeue(queue_t *queue)
{
queue_node_t *node = NULL;
if(splock_try_lock(&queue->head_lock)) {
return NULL; //Ëøʧ°Ü£¬·µ»ØNULL
}
node = queue->head->next;
if(node) {
queue->head = node;
}
--queue->length;
splock_unlock(&queue->head_lock);
return node;
}
queue_node_t *dequeue_blocking(queue_t *queue)
{
queue_node_t *node = NULL;
splock_lock(&queue->head_lock);
node = queue->head->next;
if(node) {
queue->head = node;
}
--queue->length;
splock_unlock(&queue->head_lock);
return node;
}