From 5bb28835d06e27dbd960916c9fb11a555fc5a9bc Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期一, 26 四月 2021 14:24:29 +0800 Subject: [PATCH] add shared lib target. --- src/shm.cpp | 98 ++++++++++++++++++++---------------------------- 1 files changed, 41 insertions(+), 57 deletions(-) diff --git a/src/shm.cpp b/src/shm.cpp index b89acef..6d7dccd 100644 --- a/src/shm.cpp +++ b/src/shm.cpp @@ -10,76 +10,60 @@ * Revision: none * Compiler: gcc * - * Author: YOUR NAME (), + * Author: Li Chao (), * Organization: * * ===================================================================================== */ + #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(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(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(size, std::nothrow); - bool r = false; - if (p) { - Msg buf = { Id(), p, size}; - memcpy(p, data, size); - if (remote->Write(buf, 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 buf; - if (Read(buf, timeout_ms) && buf.size_ > 0) { - DEFER1(shm().deallocate(buf.data_.get());); - source_id = buf.src_; - size = buf.size_; - if (data = malloc(size)) { - memcpy(data, buf.data_.get(), 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