Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

Latest commit

 

History

History
History
59 lines (50 loc) · 1.39 KB

File metadata and controls

59 lines (50 loc) · 1.39 KB
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
#ifndef __CPPCODE_RECURSIVE_MUTEX_HPP__
#define __CPPCODE_RECURSIVE_MUTEX_HPP__
#include <iostream>
#include <cstdint>
#include <mutex>
#include <thread>
#include <atomic>
namespace cppcode { namespace common {
class rmutex : public std::mutex
{
public:
void lock()
{
std::thread::id defaultId;
const std::thread::id currentThreadId = std::this_thread::get_id();
if (m_ownerThreadId.compare_exchange_strong(defaultId, currentThreadId))
{
const uint32_t oldValue = m_count.fetch_add(1);
if (oldValue == 0)
{
std::mutex::lock();
}
}
else if (defaultId != currentThreadId)
{
std::mutex::lock(); // lock and wait for the other thread that owns the mutex
m_ownerThreadId = currentThreadId;
m_count = 1;
}
else
{
// defaultId == currentThreadId: reentry of mutex owner thread
m_count++;
}
}
void unlock()
{
const uint32_t oldValue = m_count.fetch_sub(1);
if (oldValue == 1)
{
m_ownerThreadId = std::thread::id{};
std::mutex::unlock();
}
}
private:
std::atomic<std::thread::id> m_ownerThreadId{}; // a default-constructed thread id is treated as an invalid thread id
std::atomic<uint32_t> m_count{0};
};
}}
#endif
Morty Proxy This is a proxified and sanitized view of the page, visit original site.