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
|
#ifndef __WORK_H__
#define __WORK_H__
#include <stdbool.h>
#include "list.h"
#include "util.h"
struct work;
typedef void (*work_func_t)(struct work *);
struct work {
struct list_node w_list;
work_func_t fn;
work_func_t done;
};
struct work_queue {
int wq_state;
struct list_head pending_list;
};
enum wq_thread_control {
WQ_ORDERED, /* Only 1 thread created for work queue */
WQ_DYNAMIC, /* # of threads proportional to nr_nodes created */
WQ_UNLIMITED, /* Unlimited # of threads created */
};
static inline bool is_main_thread(void)
{
return gettid() == getpid();
}
static inline bool is_worker_thread(void)
{
return !is_main_thread();
}
/*
* Helper macros to guard variables from being accessed out of the
* main thread. Note that we can use these only for pointers.
*/
#define main_thread(type) struct { type __val; }
#define main_thread_get(var) \
({ \
assert(is_main_thread()); \
(var).__val; \
})
#define main_thread_set(var, val) \
({ \
assert(is_main_thread()); \
(var).__val = (val); \
})
/*
* 'get_nr_nodes' is the function to get the current number of nodes and used
* for dynamic work queues. 'create_cb' will be called when worker threads are
* created and 'destroy_cb' will be called when worker threads are destroyed.
*/
int init_work_queue(size_t (*get_nr_nodes)(void));
struct work_queue *create_work_queue(const char *name, enum wq_thread_control);
struct work_queue *create_ordered_work_queue(const char *name);
void queue_work(struct work_queue *q, struct work *work);
bool work_queue_empty(struct work_queue *q);
#ifdef HAVE_TRACE
void suspend_worker_threads(void);
void resume_worker_threads(void);
#endif /* HAVE_TRACE */
#endif
|
