使用互斥鎖和條件變量實(shí)現(xiàn)讀寫(xiě)鎖
代碼
class rw_lock {
private:
int stat; //0 : 無(wú)鎖差购，-1 : 寫(xiě)鎖， >0 : 讀鎖的數(shù)量
pthread_mutex_t mutex;
pthread_cond_t cond;
public:
rw_lock() : stat(0) {
pthread_mutex_init(&mutex, NULL);
pthread_cond_init(&cond, NULL);
}
~rw_lock() {}
void rdlock() {
pthread_mutex_lock(&mutex);
while (stat < 0) {
pthread_cond_wait(&cond, &mutex);
}
++stat;
pthread_mutex_unlock(&mutex);
}
void rwunlock() {
pthread_mutex_lock(&mutex);
--stat;
if (stat == 0)
pthread_cond_signal(&cond);
pthread_mutex_unlock(&mutex);
}
void wrlock() {
pthread_mutex_lock(&mutex);
while (stat > 0) {
pthread_cond_wait(&cond, &mutex);
}
--stat;
pthread_mutex_unlock(&mutex);
}
void wrunlock() {
pthread_mutex_lock(&mutex);
++stat;
pthread_cond_signal(&cond);
pthread_mutex_unlock(&mutex);
}
};
