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 | |||||
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 11 | #include <atomic> |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 12 | |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 13 | class SkSpinlock { |
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 14 | public: |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 15 | void acquire() { |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 16 | // To act as a mutex, we need an acquire barrier when we acquire the lock. |
17 | if (fLocked.exchange(true, std::memory_order_acquire)) { | ||||
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 18 | // Lock was contended. Fall back to an out-of-line spin loop. |
19 | this->contendedAcquire(); | ||||
20 | } | ||||
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 21 | } |
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 22 | |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 23 | void release() { |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 24 | // To act as a mutex, we need a release barrier when we release the lock. |
25 | fLocked.store(false, std::memory_order_release); | ||||
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 26 | } |
27 | |||||
mtklein | 828877d | 2015-07-09 10:51:36 -0700 | [diff] [blame] | 28 | private: |
29 | void contendedAcquire(); | ||||
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 30 | |
mtklein | 15923c9 | 2016-02-29 10:14:38 -0800 | [diff] [blame] | 31 | std::atomic<bool> fLocked{false}; |
mtklein | 50ffd99 | 2015-03-30 08:13:33 -0700 | [diff] [blame] | 32 | }; |
33 | |||||
34 | #endif//SkSpinlock_DEFINED |