From 1d6c040dcb9a01648edc66d8c0006c8c9294a705 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期四, 22 四月 2021 18:28:30 +0800
Subject: [PATCH] add mutex timeout limit; use atomic as refcount.
---
src/shm.cpp | 26 ++++++++++++++++++++++++++
1 files changed, 26 insertions(+), 0 deletions(-)
diff --git a/src/shm.cpp b/src/shm.cpp
index 1658900..d499b16 100644
--- a/src/shm.cpp
+++ b/src/shm.cpp
@@ -21,6 +21,32 @@
namespace bhome_shm
{
+bool MutexWithTimeLimit::try_lock()
+{
+ if (mutex_.try_lock()) {
+ auto old_time = last_lock_time_.load();
+ if (Now() - old_time > limit_) {
+ return last_lock_time_.compare_exchange_strong(old_time, Now());
+ } else {
+ last_lock_time_.store(Now());
+ return true;
+ }
+ } else {
+ auto old_time = last_lock_time_.load();
+ if (Now() - old_time > limit_) {
+ return last_lock_time_.compare_exchange_strong(old_time, Now());
+ } else {
+ return false;
+ }
+ }
+}
+void MutexWithTimeLimit::lock()
+{
+ while (!try_lock()) {
+ std::this_thread::yield();
+ }
+}
+
SharedMemory::SharedMemory(const std::string &name, const uint64_t size) :
mshm_t(open_or_create, name.c_str(), size, 0, AllowAll()),
name_(name)
--
Gitblit v1.8.0