63 lines
1.0 KiB
C++
63 lines
1.0 KiB
C++
|
#pragma once
|
||
|
|
|
||
|
|
#include <mutex>
|
||
|
|
|
||
|
|
#include "common.hpp"
|
||
|
|
#include "utils/mixins.hpp"
|
||
|
|
|
||
|
|
LEMONBUDDY_NS
|
||
|
|
|
||
|
|
namespace threading_util {
|
||
|
|
namespace locking_strategy {
|
||
|
|
struct no_backoff {
|
||
|
|
bool operator()() {
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
struct yield_backoff {
|
||
|
|
bool operator()() {
|
||
|
|
this_thread::yield();
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
class spin_lock : public non_copyable_mixin<spin_lock> {
|
||
|
|
public:
|
||
|
|
/**
|
||
|
|
* Construct spin_lock
|
||
|
|
*/
|
||
|
|
spin_lock() = default;
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Lock using custom strategy
|
||
|
|
*/
|
||
|
|
template <typename Backoff>
|
||
|
|
void lock(Backoff backoff) noexcept {
|
||
|
|
while (m_locked.test_and_set(std::memory_order_acquire)) {
|
||
|
|
backoff();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Lock using default strategy
|
||
|
|
*/
|
||
|
|
void lock() noexcept {
|
||
|
|
lock(locking_strategy::no_backoff{});
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Unlock
|
||
|
|
*/
|
||
|
|
void unlock() noexcept {
|
||
|
|
m_locked.clear(std::memory_order_release);
|
||
|
|
}
|
||
|
|
|
||
|
|
protected:
|
||
|
|
std::atomic_flag m_locked{false};
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
LEMONBUDDY_NS_END
|