-
Notifications
You must be signed in to change notification settings - Fork 1
/
queue.c
118 lines (95 loc) · 2.74 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/*
* queue.c - Queued workitem system
*
* Copyright 2008 Paul Betts <paul.betts@gmail.com>
*
*
* License:
*
* This package is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This package is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this package; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
#include "stdafx.h"
#include "stats.h"
#include "config.h"
struct WorkitemQueue {
GAsyncQueue* to_process;
GThread* thread;
gboolean should_quit;
};
struct Workitem {
GFunc func;
gpointer data;
gpointer context;
};
static gpointer worker_thread_proc(gpointer data)
{
struct WorkitemQueue* this = data;
g_async_queue_ref(this->to_process);
while(!this->should_quit) {
struct Workitem* item = g_async_queue_pop(this->to_process);
if (item->func)
(item->func)(item->data, item->context);
g_free(item);
}
g_async_queue_unref(this->to_process);
return 0;
}
struct WorkitemQueue* workitem_queue_new(void)
{
struct WorkitemQueue* ret = g_new0(struct WorkitemQueue, 1);
if (!ret)
goto failed;
ret->to_process = g_async_queue_new_full(g_free);
if (!ret->to_process)
goto failed;
if (!(ret->thread = g_thread_create(worker_thread_proc, ret, TRUE, NULL)))
goto failed;
return ret;
failed:
if (ret) {
if (ret->to_process)
g_async_queue_unref(ret->to_process);
g_free(ret);
}
return NULL;
}
void workitem_queue_free(struct WorkitemQueue* queue)
{
if (!queue)
return;
queue->should_quit = TRUE;
/* Clear out the action queue */
struct Workitem* to_free;
g_async_queue_lock(queue->to_process);
while( !(to_free = g_async_queue_try_pop_unlocked(queue->to_process)) ) {
g_free(to_free);
}
/* Make sure we've got one extra dummy item so that the worker thread
* doesn't deadlock */
g_async_queue_push_unlocked(queue->to_process, g_new0(struct Workitem*, 1));
g_async_queue_unlock(queue->to_process);
g_thread_join(queue->thread);
g_async_queue_unref(queue->to_process);
g_free(queue);
}
gboolean workitem_queue_insert(struct WorkitemQueue* queue, GFunc func, gpointer data, gpointer context)
{
if (!queue)
return FALSE;
struct Workitem* obj = g_new(struct Workitem, 1);
obj->func = func; obj->data = data; obj->context = context;
g_async_queue_push(queue->to_process, obj);
return TRUE;
}