blob: 67143ca155580bd2163fb522feb9dcacf26b1af8 (
plain)
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
|
#pragma once
#include <AtStartExit.h>
#include <pthread.h>
namespace Balau {
template<class T>
class Queue;
class Lock {
public:
Lock();
~Lock() { pthread_mutex_destroy(&m_lock); }
void enter() { pthread_mutex_lock(&m_lock); }
void leave() { pthread_mutex_unlock(&m_lock); }
private:
pthread_mutex_t m_lock;
template<class T>
friend class Queue;
};
class RWLock {
public:
RWLock();
~RWLock() { pthread_rwlock_destroy(&m_lock); }
void enterR() { pthread_rwlock_rdlock(&m_lock); }
void enterW() { pthread_rwlock_wrlock(&m_lock); }
void leave() { pthread_rwlock_unlock(&m_lock); }
private:
pthread_rwlock_t m_lock;
};
class ThreadHelper;
class Thread {
public:
virtual ~Thread();
void threadStart();
void * join();
protected:
Thread() : m_joined(false) { }
virtual void * proc() = 0;
virtual void threadExit() { };
private:
pthread_t m_thread;
volatile bool m_joined;
friend class ThreadHelper;
};
class GlobalThread : public Thread, public AtStart, public AtExit {
protected:
GlobalThread(int startOrder = 10) : AtStart(startOrder), AtExit(1) { }
private:
virtual void doStart() { threadStart(); }
virtual void doExit() { join(); }
};
template<class T>
class Queue {
public:
Queue() : m_front(NULL), m_back(NULL) { pthread_cond_init(&m_cond, NULL); }
~Queue() { while (!isEmpty()) pop(); pthread_cond_destroy(&m_cond); }
void push(T * t) {
m_lock.enter();
Cell * c = new Cell(t);
c->m_prev = m_back;
if (m_back)
m_back->m_next = c;
else
m_front = c;
m_back = c;
pthread_cond_signal(&m_cond);
m_lock.leave();
}
T * pop() {
m_lock.enter();
while (!m_front)
pthread_cond_wait(&m_cond, &m_lock.m_lock);
Cell * c = m_front;
m_front = c->m_next;
if (m_front)
m_front->m_prev = NULL;
else
m_back = NULL;
T * t = c->m_elem;
delete c;
m_lock.leave();
return t;
}
bool isEmpty() {
bool r;
m_lock.enter();
r = !m_front;
m_lock.leave();
return r;
}
private:
Lock m_lock;
class Cell {
public:
Cell(T * elem) : m_next(NULL), m_prev(NULL), m_elem(elem) { }
Cell * m_next, * m_prev;
T * m_elem;
};
Cell * volatile m_front;
Cell * volatile m_back;
pthread_cond_t m_cond;
};
};
|