| 1 | // |
| 2 | // Mutex.h |
| 3 | // |
| 4 | // Library: Foundation |
| 5 | // Package: Threading |
| 6 | // Module: Mutex |
| 7 | // |
| 8 | // Definition of the Mutex and FastMutex classes. |
| 9 | // |
| 10 | // Copyright (c) 2004-2008, Applied Informatics Software Engineering GmbH. |
| 11 | // and Contributors. |
| 12 | // |
| 13 | // SPDX-License-Identifier: BSL-1.0 |
| 14 | // |
| 15 | |
| 16 | |
| 17 | #ifndef Foundation_Mutex_INCLUDED |
| 18 | #define Foundation_Mutex_INCLUDED |
| 19 | |
| 20 | |
| 21 | #include "Poco/Foundation.h" |
| 22 | #include "Poco/Timestamp.h" |
| 23 | #include "Poco/Exception.h" |
| 24 | #include "Poco/ScopedLock.h" |
| 25 | #include <atomic> |
| 26 | |
| 27 | |
| 28 | #if (POCO_OS == POCO_OS_CYGWIN || POCO_OS == POCO_OS_ANDROID) |
| 29 | #include "Poco/Mutex_POSIX.h" |
| 30 | #else |
| 31 | #include "Poco/Mutex_STD.h" |
| 32 | #endif |
| 33 | |
| 34 | namespace Poco { |
| 35 | |
| 36 | |
| 37 | class Foundation_API Mutex: private MutexImpl |
| 38 | /// A Mutex (mutual exclusion) is a synchronization |
| 39 | /// mechanism used to control access to a shared resource |
| 40 | /// in a concurrent (multithreaded) scenario. |
| 41 | /// Using the ScopedLock class is the preferred way to automatically |
| 42 | /// lock and unlock a mutex. |
| 43 | { |
| 44 | public: |
| 45 | enum MutexType |
| 46 | /// The type of a mutex. |
| 47 | { |
| 48 | MUTEX_RECURSIVE = MUTEX_RECURSIVE_IMPL, /// A recursive mutex |
| 49 | MUTEX_NONRECURSIVE = MUTEX_NONRECURSIVE_IMPL /// A non-recursive mutex |
| 50 | }; |
| 51 | |
| 52 | typedef Poco::ScopedLock<Mutex> ScopedLock; |
| 53 | |
| 54 | explicit Mutex(MutexType type = MUTEX_RECURSIVE); |
| 55 | /// Creates the Mutex. |
| 56 | |
| 57 | ~Mutex(); |
| 58 | /// Destroys the Mutex. |
| 59 | |
| 60 | void lock(); |
| 61 | /// Locks the mutex. Blocks if the mutex |
| 62 | /// is held by another thread. |
| 63 | |
| 64 | void lock(long milliseconds); |
| 65 | /// Locks the mutex. Blocks up to the given number of milliseconds |
| 66 | /// if the mutex is held by another thread. Throws a TimeoutException |
| 67 | /// if the mutex can not be locked within the given timeout. |
| 68 | /// |
| 69 | /// Performance Note: On most platforms (including Windows), this member function is |
| 70 | /// implemented using a loop calling (the equivalent of) tryLock() and Thread::sleep(). |
| 71 | /// On POSIX platforms that support pthread_mutex_timedlock(), this is used. |
| 72 | |
| 73 | bool tryLock(); |
| 74 | /// Tries to lock the mutex. Returns false immediately |
| 75 | /// if the mutex is already held by another thread. |
| 76 | /// Returns true if the mutex was successfully locked. |
| 77 | |
| 78 | bool tryLock(long milliseconds); |
| 79 | /// Locks the mutex. Blocks up to the given number of milliseconds |
| 80 | /// if the mutex is held by another thread. |
| 81 | /// Returns true if the mutex was successfully locked. |
| 82 | /// |
| 83 | /// Performance Note: On most platforms (including Windows), this member function is |
| 84 | /// implemented using a loop calling (the equivalent of) tryLock() and Thread::sleep(). |
| 85 | /// On POSIX platforms that support pthread_mutex_timedlock(), this is used. |
| 86 | |
| 87 | void unlock(); |
| 88 | /// Unlocks the mutex so that it can be acquired by |
| 89 | /// other threads. |
| 90 | |
| 91 | private: |
| 92 | Mutex(const Mutex&); |
| 93 | Mutex& operator = (const Mutex&); |
| 94 | }; |
| 95 | |
| 96 | |
| 97 | class Foundation_API FastMutex: private FastMutexImpl |
| 98 | /// A FastMutex (mutual exclusion) is similar to a Mutex. |
| 99 | /// Locking a FastMutex is guaranteed to be at least as |
| 100 | /// fast as locking a Mutex. However, a FastMutex is not |
| 101 | /// guaranteed to be either recursive or non-recursive. |
| 102 | /// It is best suited to thread safe components like pools, |
| 103 | /// caches and queues where locking is internal to the component. |
| 104 | /// Using the ScopedLock class is the preferred way to automatically |
| 105 | /// lock and unlock a mutex. |
| 106 | { |
| 107 | public: |
| 108 | typedef Poco::ScopedLock<FastMutex> ScopedLock; |
| 109 | |
| 110 | FastMutex(); |
| 111 | /// creates the Mutex. |
| 112 | |
| 113 | ~FastMutex(); |
| 114 | /// destroys the Mutex. |
| 115 | |
| 116 | void lock(); |
| 117 | /// Locks the mutex. Blocks if the mutex |
| 118 | /// is held by another thread. |
| 119 | |
| 120 | void lock(long milliseconds); |
| 121 | /// Locks the mutex. Blocks up to the given number of milliseconds |
| 122 | /// if the mutex is held by another thread. Throws a TimeoutException |
| 123 | /// if the mutex can not be locked within the given timeout. |
| 124 | /// |
| 125 | /// Performance Note: On most platforms (including Windows), this member function is |
| 126 | /// implemented using a loop calling (the equivalent of) tryLock() and Thread::sleep(). |
| 127 | /// On POSIX platforms that support pthread_mutex_timedlock(), this is used. |
| 128 | |
| 129 | bool tryLock(); |
| 130 | /// Tries to lock the mutex. Returns false immediately |
| 131 | /// if the mutex is already held by another thread. |
| 132 | /// Returns true if the mutex was successfully locked. |
| 133 | |
| 134 | bool tryLock(long milliseconds); |
| 135 | /// Locks the mutex. Blocks up to the given number of milliseconds |
| 136 | /// if the mutex is held by another thread. |
| 137 | /// Returns true if the mutex was successfully locked. |
| 138 | /// |
| 139 | /// Performance Note: On most platforms (including Windows), this member function is |
| 140 | /// implemented using a loop calling (the equivalent of) tryLock() and Thread::sleep(). |
| 141 | /// On POSIX platforms that support pthread_mutex_timedlock(), this is used. |
| 142 | |
| 143 | void unlock(); |
| 144 | /// Unlocks the mutex so that it can be acquired by |
| 145 | /// other threads. |
| 146 | |
| 147 | private: |
| 148 | FastMutex(const FastMutex&); |
| 149 | FastMutex& operator = (const FastMutex&); |
| 150 | }; |
| 151 | |
| 152 | |
| 153 | class Foundation_API SpinlockMutex |
| 154 | /// A SpinlockMutex, implemented in terms of std::atomic_flag, as |
| 155 | /// busy-wait mutual exclusion. |
| 156 | /// |
| 157 | /// While in some cases (eg. locking small blocks of code) |
| 158 | /// busy-waiting may be an optimal solution, in many scenarios |
| 159 | /// spinlock may not be the right choice - it is up to the user to |
| 160 | /// choose the proper mutex type for their particular case. |
| 161 | /// |
| 162 | /// Works with the ScopedLock class. |
| 163 | { |
| 164 | public: |
| 165 | typedef Poco::ScopedLock<SpinlockMutex> ScopedLock; |
| 166 | |
| 167 | SpinlockMutex(); |
| 168 | /// Creates the SpinlockMutex. |
| 169 | |
| 170 | ~SpinlockMutex(); |
| 171 | /// Destroys the SpinlockMutex. |
| 172 | |
| 173 | void lock(); |
| 174 | /// Locks the mutex. Blocks if the mutex |
| 175 | /// is held by another thread. |
| 176 | |
| 177 | void lock(long milliseconds); |
| 178 | /// Locks the mutex. Blocks up to the given number of milliseconds |
| 179 | /// if the mutex is held by another thread. Throws a TimeoutException |
| 180 | /// if the mutex can not be locked within the given timeout. |
| 181 | |
| 182 | bool tryLock(); |
| 183 | /// Tries to lock the mutex. Returns immediately, false |
| 184 | /// if the mutex is already held by another thread, true |
| 185 | /// if the mutex was successfully locked. |
| 186 | |
| 187 | bool tryLock(long milliseconds); |
| 188 | /// Locks the mutex. Blocks up to the given number of milliseconds |
| 189 | /// if the mutex is held by another thread. |
| 190 | /// Returns true if the mutex was successfully locked. |
| 191 | |
| 192 | void unlock(); |
| 193 | /// Unlocks the mutex so that it can be acquired by |
| 194 | /// other threads. |
| 195 | |
| 196 | private: |
| 197 | std::atomic_flag _flag = ATOMIC_FLAG_INIT; |
| 198 | }; |
| 199 | |
| 200 | |
| 201 | class Foundation_API NullMutex |
| 202 | /// A NullMutex is an empty mutex implementation |
| 203 | /// which performs no locking at all. Useful in policy driven design |
| 204 | /// where the type of mutex used can be now a template parameter allowing the user to switch |
| 205 | /// between thread-safe and not thread-safe depending on his need |
| 206 | /// Works with the ScopedLock class |
| 207 | { |
| 208 | public: |
| 209 | typedef Poco::ScopedLock<NullMutex> ScopedLock; |
| 210 | |
| 211 | NullMutex() |
| 212 | /// Creates the NullMutex. |
| 213 | { |
| 214 | } |
| 215 | |
| 216 | ~NullMutex() |
| 217 | /// Destroys the NullMutex. |
| 218 | { |
| 219 | } |
| 220 | |
| 221 | void lock() |
| 222 | /// Does nothing. |
| 223 | { |
| 224 | } |
| 225 | |
| 226 | void lock(long) |
| 227 | /// Does nothing. |
| 228 | { |
| 229 | } |
| 230 | |
| 231 | bool tryLock() |
| 232 | /// Does nothing and always returns true. |
| 233 | { |
| 234 | return true; |
| 235 | } |
| 236 | |
| 237 | bool tryLock(long) |
| 238 | /// Does nothing and always returns true. |
| 239 | { |
| 240 | return true; |
| 241 | } |
| 242 | |
| 243 | void unlock() |
| 244 | /// Does nothing. |
| 245 | { |
| 246 | } |
| 247 | }; |
| 248 | |
| 249 | |
| 250 | // |
| 251 | // inlines |
| 252 | // |
| 253 | |
| 254 | // |
| 255 | // Mutex |
| 256 | // |
| 257 | |
| 258 | inline void Mutex::lock() |
| 259 | { |
| 260 | lockImpl(); |
| 261 | } |
| 262 | |
| 263 | |
| 264 | inline void Mutex::lock(long milliseconds) |
| 265 | { |
| 266 | if (!tryLockImpl(milliseconds)) |
| 267 | throw TimeoutException(); |
| 268 | } |
| 269 | |
| 270 | |
| 271 | inline bool Mutex::tryLock() |
| 272 | { |
| 273 | return tryLockImpl(); |
| 274 | } |
| 275 | |
| 276 | |
| 277 | inline bool Mutex::tryLock(long milliseconds) |
| 278 | { |
| 279 | return tryLockImpl(milliseconds); |
| 280 | } |
| 281 | |
| 282 | |
| 283 | inline void Mutex::unlock() |
| 284 | { |
| 285 | unlockImpl(); |
| 286 | } |
| 287 | |
| 288 | |
| 289 | // |
| 290 | // FastMutex |
| 291 | // |
| 292 | |
| 293 | inline void FastMutex::lock() |
| 294 | { |
| 295 | lockImpl(); |
| 296 | } |
| 297 | |
| 298 | |
| 299 | inline void FastMutex::lock(long milliseconds) |
| 300 | { |
| 301 | if (!tryLockImpl(milliseconds)) |
| 302 | throw TimeoutException(); |
| 303 | } |
| 304 | |
| 305 | |
| 306 | inline bool FastMutex::tryLock() |
| 307 | { |
| 308 | return tryLockImpl(); |
| 309 | } |
| 310 | |
| 311 | |
| 312 | inline bool FastMutex::tryLock(long milliseconds) |
| 313 | { |
| 314 | return tryLockImpl(milliseconds); |
| 315 | } |
| 316 | |
| 317 | |
| 318 | inline void FastMutex::unlock() |
| 319 | { |
| 320 | unlockImpl(); |
| 321 | } |
| 322 | |
| 323 | |
| 324 | // |
| 325 | // SpinlockMutex |
| 326 | // |
| 327 | |
| 328 | inline void SpinlockMutex::lock() |
| 329 | { |
| 330 | while (_flag.test_and_set(std::memory_order_acquire)); |
| 331 | } |
| 332 | |
| 333 | |
| 334 | inline void SpinlockMutex::lock(long milliseconds) |
| 335 | { |
| 336 | Timestamp now; |
| 337 | Timestamp::TimeDiff diff(Timestamp::TimeDiff(milliseconds)*1000); |
| 338 | while (_flag.test_and_set(std::memory_order_acquire)) |
| 339 | { |
| 340 | if (now.isElapsed(diff)) throw TimeoutException(); |
| 341 | } |
| 342 | } |
| 343 | |
| 344 | |
| 345 | inline bool SpinlockMutex::tryLock() |
| 346 | { |
| 347 | return !_flag.test_and_set(std::memory_order_acquire); |
| 348 | } |
| 349 | |
| 350 | |
| 351 | inline bool SpinlockMutex::tryLock(long milliseconds) |
| 352 | { |
| 353 | Timestamp now; |
| 354 | Timestamp::TimeDiff diff(Timestamp::TimeDiff(milliseconds)*1000); |
| 355 | while (_flag.test_and_set(std::memory_order_acquire)) |
| 356 | { |
| 357 | if (now.isElapsed(diff)) return false; |
| 358 | } |
| 359 | return true; |
| 360 | } |
| 361 | |
| 362 | |
| 363 | inline void SpinlockMutex::unlock() |
| 364 | { |
| 365 | _flag.clear(std::memory_order_release); |
| 366 | } |
| 367 | |
| 368 | |
| 369 | } // namespace Poco |
| 370 | |
| 371 | |
| 372 | #endif // Foundation_Mutex_INCLUDED |
| 373 | |