OLD | NEW |
1 /* | 1 /* |
2 * Copyright (C) 2013 Google Inc. All rights reserved. | 2 * Copyright (C) 2013 Google Inc. All rights reserved. |
3 * | 3 * |
4 * Redistribution and use in source and binary forms, with or without | 4 * Redistribution and use in source and binary forms, with or without |
5 * modification, are permitted provided that the following conditions are | 5 * modification, are permitted provided that the following conditions are |
6 * met: | 6 * met: |
7 * | 7 * |
8 * * Redistributions of source code must retain the above copyright | 8 * * Redistributions of source code must retain the above copyright |
9 * notice, this list of conditions and the following disclaimer. | 9 * notice, this list of conditions and the following disclaimer. |
10 * * Redistributions in binary form must reproduce the above | 10 * * Redistributions in binary form must reproduce the above |
(...skipping 13 matching lines...) Expand all Loading... |
24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, | 24 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY | 25 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | 26 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | 27 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | 28 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
29 */ | 29 */ |
30 | 30 |
31 #ifndef WTF_SpinLock_h | 31 #ifndef WTF_SpinLock_h |
32 #define WTF_SpinLock_h | 32 #define WTF_SpinLock_h |
33 | 33 |
34 #include "wtf/Compiler.h" | 34 // DESCRIPTION |
35 #include "wtf/WTFExport.h" | 35 // spinLockLock() and spinLockUnlock() are simple spinlock primitives based on |
36 #include <atomic> | 36 // the standard CPU primitive of atomic increment and decrement of an int at |
37 #include <memory> | 37 // a given memory address. These are intended only for very short duration locks |
38 #include <mutex> | 38 // and assume a system with multiple cores. For any potentially longer wait you |
| 39 // should be using a real lock. |
39 | 40 |
40 // DESCRIPTION | 41 #include "wtf/Atomics.h" |
41 // Spinlock is a simple spinlock class based on the standard CPU primitive of | |
42 // atomic increment and decrement of an int at a given memory address. These are | |
43 // intended only for very short duration locks and assume a system with multiple | |
44 // cores. For any potentially longer wait you should be using a real lock. | |
45 | 42 |
46 namespace WTF { | 43 namespace WTF { |
47 | 44 |
48 class SpinLock { | 45 // This is called if the initial attempt to acquire the lock fails. It's a bit |
49 public: | 46 // slower, but has a much better scheduling and power consumption behavior. |
50 using Guard = std::lock_guard<SpinLock>; | 47 WTF_EXPORT void slowSpinLockLock(int volatile* lock); |
51 | 48 |
52 ALWAYS_INLINE void lock() | 49 ALWAYS_INLINE void spinLockLock(int volatile* lock) |
53 { | 50 { |
54 static_assert(sizeof(m_lock) == sizeof(int), "int and m_lock are differe
nt sizes"); | 51 if (LIKELY(!atomicTestAndSetToOne(lock))) |
55 if (LIKELY(!m_lock.exchange(true, std::memory_order_acq_rel))) | 52 return; |
56 return; | 53 slowSpinLockLock(lock); |
57 lockSlow(); | 54 } |
58 } | |
59 | 55 |
60 ALWAYS_INLINE void unlock() | 56 ALWAYS_INLINE void spinLockUnlock(int volatile* lock) |
61 { | 57 { |
62 m_lock.store(false, std::memory_order_release); | 58 atomicSetOneToZero(lock); |
63 } | 59 } |
64 | |
65 private: | |
66 // This is called if the initial attempt to acquire the lock fails. It's | |
67 // slower, but has a much better scheduling and power consumption behavior. | |
68 WTF_EXPORT void lockSlow(); | |
69 | |
70 std::atomic<int> m_lock; | |
71 }; | |
72 | |
73 | 60 |
74 } // namespace WTF | 61 } // namespace WTF |
75 | 62 |
76 using WTF::SpinLock; | 63 using WTF::spinLockLock; |
| 64 using WTF::spinLockUnlock; |
77 | 65 |
78 #endif // WTF_SpinLock_h | 66 #endif // WTF_SpinLock_h |
OLD | NEW |