From bcd780993c176b93f7393607f8003adf66e6676a Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期一, 26 四月 2021 10:27:14 +0800
Subject: [PATCH] fix node default ignore msg.
---
src/shm.cpp | 42 +++++++++++++++++++++++++++++++++++++++---
1 files changed, 39 insertions(+), 3 deletions(-)
diff --git a/src/shm.cpp b/src/shm.cpp
index d628b56..6d7dccd 100644
--- a/src/shm.cpp
+++ b/src/shm.cpp
@@ -21,9 +21,45 @@
namespace bhome_shm
{
-SharedMemory::SharedMemory(const std::string &name, const uint64_t size)
- : mshm_t(open_or_create, name.c_str(), size, 0, AllowAll()),
- name_(name)
+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();
+ }
+}
+void MutexWithTimeLimit::unlock()
+{
+ auto old_time = last_lock_time_.load();
+ if (Now() - old_time > limit_) {
+ } else {
+ if (last_lock_time_.compare_exchange_strong(old_time, Now())) {
+ mutex_.unlock();
+ }
+ }
+}
+
+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