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 | 103 +++++++++++++++++++--------------------------------
1 files changed, 39 insertions(+), 64 deletions(-)
diff --git a/src/shm.cpp b/src/shm.cpp
index d271ef5..6d7dccd 100644
--- a/src/shm.cpp
+++ b/src/shm.cpp
@@ -17,78 +17,53 @@
*/
#include "shm.h"
-#include "bh_util.h"
-#include <mutex>
-#include <boost/uuid/uuid_io.hpp>
-#include <boost/uuid/uuid_generators.hpp>
-namespace bhome_shm {
-using namespace boost::interprocess;
-using namespace boost::uuids;
-
-namespace {
-std::string MsgQIdToName(const MQId& id) { return "shmq" + to_string(id); }
-MQId EmptyId() { return nil_uuid(); }
-MQId NewId() { return random_generator()(); }
-}
-
-ShmMsgQueue::ShmMsgQueue(const MQId &id, ShmType &segment, const uint32_t len):
-SharedQueue(segment, MsgQIdToName(id), id, len, segment.get_segment_manager())
+namespace bhome_shm
{
- printf("queue size: %ld cap: %ld\n", data()->size(), data()->capacity());
-}
-ShmMsgQueue::ShmMsgQueue(ShmType &segment, const uint32_t len):ShmMsgQueue(NewId(), segment, len)
-{}
-
-ShmMsgQueue::~ShmMsgQueue()
+bool MutexWithTimeLimit::try_lock()
{
- Remove();
+ 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;
+ }
+ }
}
-
-bool ShmMsgQueue::Send(const MQId &remote_id, const void *data, const size_t size, const int timeout_ms)
+void MutexWithTimeLimit::lock()
{
- if (data && size) {
- Queue *remote = find(MsgQIdToName(remote_id));
- if (remote) {
- void *p = shm().allocate(sizeof(MsgMetaV1) + size, std::nothrow);
- bool r = false;
- if (p) {
- MsgMetaV1 meta;
- meta.data_size_ = size;
- memcpy(meta.src_id_, &Id(), sizeof(MQId));
- meta.Pack(p);
-
- memcpy(static_cast<char*>(p) + sizeof(meta), data, size);
- if (remote->Write(p, timeout_ms)) {
- return true;
- } else {
- shm().deallocate(p);
- }
- }
- }
- }
- return false;
+ while (!try_lock()) {
+ std::this_thread::yield();
+ }
}
-
-bool ShmMsgQueue::Recv(MQId &source_id, void *&data, size_t &size, const int timeout_ms)
+void MutexWithTimeLimit::unlock()
{
- Msg msg;
- if (Read(msg, timeout_ms) && msg) {
- DEFER1(shm().deallocate(msg.get()););
- MsgMetaV1 meta;
- meta.Parse(msg.get());
- memcpy(&source_id, meta.src_id_, sizeof(MQId));
- size = meta.data_size_;
- if (data = malloc(size)) {
- memcpy(data, static_cast<char*>(msg.get()) + meta.self_size_, size);
- return true;
- }
- }
- source_id = EmptyId();
- data = 0;
- size = 0;
- return false;
+ 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)
+{
+}
+
+SharedMemory::~SharedMemory()
+{
+}
} // namespace bhome_shm
--
Gitblit v1.8.0