mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 1 | /* |
2 | * Copyright 2015 Google Inc. | ||||
3 | * | ||||
4 | * Use of this source code is governed by a BSD-style license that can be | ||||
5 | * found in the LICENSE file. | ||||
6 | */ | ||||
7 | |||||
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 8 | #ifndef SkSpinlock_DEFINED |
9 | #define SkSpinlock_DEFINED | ||||
10 | |||||
halcanary | 7b0b2ca | 2016-03-04 08:30:05 -0800 | [diff] [blame^] | 11 | #include "SkTypes.h" |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 12 | #include <atomic> |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 13 | |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 14 | class SkSpinlock { |
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 15 | public: |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 16 | void acquire() { |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 17 | // To act as a mutex, we need an acquire barrier when we acquire the lock. |
18 | if (fLocked.exchange(true, std::memory_order_acquire)) { | ||||
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 19 | // Lock was contended. Fall back to an out-of-line spin loop. |
20 | this->contendedAcquire(); | ||||
21 | } | ||||
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 22 | } |
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 23 | |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 24 | void release() { |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 25 | // To act as a mutex, we need a release barrier when we release the lock. |
26 | fLocked.store(false, std::memory_order_release); | ||||
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 27 | } |
28 | |||||
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 29 | private: |
halcanary | 7b0b2ca | 2016-03-04 08:30:05 -0800 | [diff] [blame^] | 30 | SK_API void contendedAcquire(); |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 31 | |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 32 | std::atomic<bool> fLocked{false}; |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 33 | }; |
34 | |||||
35 | #endif//SkSpinlock_DEFINED |