blob: f8915ed288e0e25fa53d455fc82fda8e2135e2d2 (
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
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
/*
* Ceph - scalable distributed file system
*
* Copyright (C) 2004-2006 Sage Weil <sage@newdream.net>
*
* This is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License version 2.1, as published by the Free Software
* Foundation. See file COPYING.
*
*/
#ifndef CEPH_RWLock_Posix__H
#define CEPH_RWLock_Posix__H
#include <pthread.h>
#include "lockdep.h"
class RWLock
{
mutable pthread_rwlock_t L;
const char *name;
int id;
public:
RWLock(const RWLock& other);
const RWLock& operator=(const RWLock& other);
RWLock(const char *n) : name(n), id(-1) {
pthread_rwlock_init(&L, NULL);
if (g_lockdep) id = lockdep_register(name);
}
virtual ~RWLock() {
pthread_rwlock_unlock(&L);
pthread_rwlock_destroy(&L);
}
void unlock() {
if (g_lockdep) id = lockdep_will_unlock(name, id);
pthread_rwlock_unlock(&L);
}
// read
void get_read() {
if (g_lockdep) id = lockdep_will_lock(name, id);
pthread_rwlock_rdlock(&L);
if (g_lockdep) id = lockdep_locked(name, id);
}
bool try_get_read() {
if (pthread_rwlock_tryrdlock(&L) == 0) {
if (g_lockdep) id = lockdep_locked(name, id);
return true;
}
return false;
}
void put_read() {
unlock();
}
// write
void get_write() {
if (g_lockdep) id = lockdep_will_lock(name, id);
pthread_rwlock_wrlock(&L);
if (g_lockdep) id = lockdep_locked(name, id);
}
bool try_get_write() {
if (pthread_rwlock_trywrlock(&L) == 0) {
if (g_lockdep) id = lockdep_locked(name, id);
return true;
}
return false;
}
void put_write() {
unlock();
}
};
#endif // !_Mutex_Posix_
|
