From 34cd75f77d0ca94dbdba4e6cc9451fe4d33e78b3 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期三, 19 五月 2021 19:14:13 +0800
Subject: [PATCH] add api BHQueryProcs.
---
src/shm_msg_queue.h | 60 +++++++++++++++++++++++++++++++++++++++---------------------
1 files changed, 39 insertions(+), 21 deletions(-)
diff --git a/src/shm_msg_queue.h b/src/shm_msg_queue.h
index 1970803..de60fde 100644
--- a/src/shm_msg_queue.h
+++ b/src/shm_msg_queue.h
@@ -18,53 +18,71 @@
#ifndef SHM_MSG_QUEUE_D847TQXH
#define SHM_MSG_QUEUE_D847TQXH
+#include "defs.h"
#include "msg.h"
#include "shm_queue.h"
using namespace bhome_shm;
using namespace bhome_msg;
+#define BH_USE_ATOMIC_Q
+
class ShmMsgQueue : public StaticDataRef<std::atomic<uint64_t>, ShmMsgQueue>
{
- // typedef ShmObject<SharedQ63<4>> Shmq;
- typedef ShmObject<SharedQueue<int64_t>> Shmq;
- typedef Shmq::ShmType ShmType;
- typedef Shmq::Data Queue;
- typedef std::function<void()> OnSend;
- typedef robust::FMutex Mutex;
- // typedef robust::SemMutex Mutex;
- // typedef robust::NullMutex Mutex;
- typedef robust::Guard<Mutex> Guard;
-
public:
+ typedef int64_t RawData;
+
+#ifdef BH_USE_ATOMIC_Q
+ typedef ShmObject<SharedQ63<0>> Shmq;
+#else
+ typedef ShmObject<SharedQueue<RawData>> Shmq;
+ // typedef robust::FMutex Mutex;
+ // typedef robust::SemMutex Mutex;
+ typedef robust::NullMutex Mutex;
+ typedef robust::Guard<Mutex> Guard;
+#endif
+
+ typedef Shmq::Data Queue;
+ typedef Shmq::ShmType ShmType;
typedef uint64_t MQId;
static MQId NewId();
- ShmMsgQueue(const MQId id, ShmType &segment, const int len);
- ShmMsgQueue(const MQId id, const bool create_or_else_find, ShmType &segment, const int len);
- ShmMsgQueue(ShmType &segment, const int len);
+ ShmMsgQueue(ShmType &segment, const MQId id, const int len);
+ ShmMsgQueue(ShmType &segment, const bool create_or_else_find, const MQId id, const int len);
+ ShmMsgQueue(const int64_t abs_addr, ShmType &segment, const MQId id);
~ShmMsgQueue();
- static bool Remove(SharedMemory &shm, const MQId id);
+ static bool Remove(ShmType &shm, const MQId id);
MQId Id() const { return id_; }
ShmType &shm() const { return queue_.shm(); }
+ int64_t AbsAddr() const { return queue_.offset(); }
- bool Recv(MsgI &msg, const int timeout_ms)
+ bool Recv(RawData &val, const int timeout_ms)
{
+#ifndef BH_USE_ATOMIC_Q
Guard lock(GetMutex(Id()));
- return queue().Read(msg.OffsetRef(), timeout_ms);
+#endif
+ return queue().Read(val, timeout_ms);
}
- bool TryRecv(MsgI &msg)
+
+ bool TryRecv(RawData &val)
{
+#ifndef BH_USE_ATOMIC_Q
Guard lock(GetMutex(Id()));
- return queue().TryRead(msg.OffsetRef());
+#endif
+ return queue().TryRead(val);
}
- static Queue *Find(SharedMemory &shm, const MQId remote_id);
- static bool TrySend(SharedMemory &shm, const MQId remote_id, MsgI msg);
- bool TrySend(const MQId remote_id, const MsgI &msg) { return TrySend(shm(), remote_id, msg); }
+
+ bool Recv(MsgI &msg, const int timeout_ms) { return Recv(msg.OffsetRef(), timeout_ms); }
+ bool TryRecv(MsgI &msg) { return TryRecv(msg.OffsetRef()); }
+ static Queue *Find(ShmType &shm, const MQId remote);
+ static bool TrySend(ShmType &shm, const MQInfo &remote, const RawData val);
+ bool TrySend(const MQInfo &remote, const RawData val) { return TrySend(shm(), remote, val); }
private:
+#ifndef BH_USE_ATOMIC_Q
static Mutex &GetMutex(const MQId id);
+#endif
MQId id_;
Queue &queue() { return *queue_.data(); }
Shmq queue_;
--
Gitblit v1.8.0