/tmp/bitcoin/src/semaphore_grant.h
Line | Count | Source |
1 | | // Copyright (c) 2009-2010 Satoshi Nakamoto |
2 | | // Copyright (c) 2009-present The Bitcoin Core developers |
3 | | // Distributed under the MIT software license, see the accompanying |
4 | | // file COPYING or http://www.opensource.org/licenses/mit-license.php. |
5 | | |
6 | | #ifndef BITCOIN_SEMAPHORE_GRANT_H |
7 | | #define BITCOIN_SEMAPHORE_GRANT_H |
8 | | |
9 | | #include <semaphore> |
10 | | |
11 | | /** RAII-style semaphore lock */ |
12 | | template <std::ptrdiff_t LeastMaxValue = std::counting_semaphore<>::max()> |
13 | | class CountingSemaphoreGrant |
14 | | { |
15 | | private: |
16 | | std::counting_semaphore<LeastMaxValue>* sem; |
17 | | bool fHaveGrant; |
18 | | |
19 | | public: |
20 | | void Acquire() noexcept |
21 | 5.04k | { |
22 | 5.04k | if (fHaveGrant) { |
23 | 0 | return; |
24 | 0 | } |
25 | 5.04k | sem->acquire(); |
26 | 5.04k | fHaveGrant = true; |
27 | 5.04k | } |
28 | | |
29 | | void Release() noexcept |
30 | 8.71k | { |
31 | 8.71k | if (!fHaveGrant) { |
32 | 3.50k | return; |
33 | 3.50k | } |
34 | 5.20k | sem->release(); |
35 | 5.20k | fHaveGrant = false; |
36 | 5.20k | } |
37 | | |
38 | | bool TryAcquire() noexcept |
39 | 156 | { |
40 | 156 | if (!fHaveGrant && sem->try_acquire()) { |
41 | 156 | fHaveGrant = true; |
42 | 156 | } |
43 | 156 | return fHaveGrant; |
44 | 156 | } |
45 | | |
46 | | // Disallow copy. |
47 | | CountingSemaphoreGrant(const CountingSemaphoreGrant&) = delete; |
48 | | CountingSemaphoreGrant& operator=(const CountingSemaphoreGrant&) = delete; |
49 | | |
50 | | // Allow move. |
51 | | CountingSemaphoreGrant(CountingSemaphoreGrant&& other) noexcept |
52 | 6 | { |
53 | 6 | sem = other.sem; |
54 | 6 | fHaveGrant = other.fHaveGrant; |
55 | 6 | other.fHaveGrant = false; |
56 | 6 | other.sem = nullptr; |
57 | 6 | } |
58 | | |
59 | | CountingSemaphoreGrant& operator=(CountingSemaphoreGrant&& other) noexcept |
60 | 590 | { |
61 | 590 | Release(); |
62 | 590 | sem = other.sem; |
63 | 590 | fHaveGrant = other.fHaveGrant; |
64 | 590 | other.fHaveGrant = false; |
65 | 590 | other.sem = nullptr; |
66 | 590 | return *this; |
67 | 590 | } |
68 | | |
69 | 2.05k | CountingSemaphoreGrant() noexcept : sem(nullptr), fHaveGrant(false) {} |
70 | | |
71 | 5.20k | explicit CountingSemaphoreGrant(std::counting_semaphore<LeastMaxValue>& sema, bool fTry = false) noexcept : sem(&sema), fHaveGrant(false) |
72 | 5.20k | { |
73 | 5.20k | if (fTry) { |
74 | 156 | TryAcquire(); |
75 | 5.04k | } else { |
76 | 5.04k | Acquire(); |
77 | 5.04k | } |
78 | 5.20k | } |
79 | | |
80 | | ~CountingSemaphoreGrant() |
81 | 7.27k | { |
82 | 7.27k | Release(); |
83 | 7.27k | } |
84 | | |
85 | | explicit operator bool() const noexcept |
86 | 157 | { |
87 | 157 | return fHaveGrant; |
88 | 157 | } |
89 | | }; |
90 | | |
91 | | using BinarySemaphoreGrant = CountingSemaphoreGrant<1>; |
92 | | |
93 | | #endif // BITCOIN_SEMAPHORE_GRANT_H |