From 6eefba812ede29549af3633c490f2e85a4805524 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期三, 31 三月 2021 11:24:20 +0800 Subject: [PATCH] format code style. --- src/msg.h | 79 ++++++++++++++++++++++++--------------- 1 files changed, 49 insertions(+), 30 deletions(-) diff --git a/src/msg.h b/src/msg.h index f3fe726..2154eba 100644 --- a/src/msg.h +++ b/src/msg.h @@ -18,15 +18,16 @@ #ifndef MSG_5BILLZET #define MSG_5BILLZET -#include <stdint.h> +#include "bhome_msg.pb.h" #include "shm.h" #include <boost/interprocess/offset_ptr.hpp> #include <boost/uuid/uuid_generators.hpp> -#include "bhome_msg.pb.h" +#include <stdint.h> -namespace bhome_msg { - using namespace bhome_shm; - using namespace bhome::msg; // for serialized data in MsgI +namespace bhome_msg +{ +using namespace bhome_shm; +using namespace bhome::msg; // for serialized data in MsgI // MsgI is safe to be stored in shared memory, so POD data or offset_ptr is required. // message format: header(meta) + body(data). @@ -37,49 +38,67 @@ class RefCount : private boost::noncopyable { public: - int Inc() { Guard lk(mutex_); return ++num_; } - int Dec() { Guard lk(mutex_); return --num_; } - int Get() { Guard lk(mutex_); return num_; } + int Inc() + { + Guard lk(mutex_); + return ++num_; + } + int Dec() + { + Guard lk(mutex_); + return --num_; + } + int Get() + { + Guard lk(mutex_); + return num_; + } + private: - Mutex mutex_; - int num_ = 1; + Mutex mutex_; + int num_ = 1; }; BHMsg MakeRequest(const MQId &src_id, const void *data, const size_t size); BHMsg MakeReply(const void *data, const size_t size); -BHMsg MakeSub(const MQId &client, const std::vector<std::string> &topics); -BHMsg MakeUnsub(const MQId &client, const std::vector<std::string> &topics); +BHMsg MakeSub(const MQId &client, const std::vector<std::string> &topics); +BHMsg MakeUnsub(const MQId &client, const std::vector<std::string> &topics); BHMsg MakePub(const std::string &topic, const void *data, const size_t size); -class MsgI { +class MsgI +{ private: - offset_ptr<void> ptr_; - offset_ptr<RefCount> count_; + offset_ptr<void> ptr_; + offset_ptr<RefCount> count_; - bool BuildSubOrUnsub(SharedMemory &shm, const std::vector<std::string> &topics, const MsgType sub_unsub); + bool BuildSubOrUnsub(SharedMemory &shm, const std::vector<std::string> &topics, const MsgType sub_unsub); + public: - MsgI(void *p=0, RefCount *c=0):ptr_(p), count_(c) {} - void swap(MsgI &a) { std::swap(ptr_, a.ptr_); std::swap(count_, a.count_); } - template <class T = void> T *get() { return static_cast<T*>(ptr_.get()); } + MsgI(void *p = 0, RefCount *c = 0) : + ptr_(p), count_(c) {} - // AddRef and Release works for both counted and not counted msg. - int AddRef() const { return IsCounted() ? count_->Inc() : 1; } - int Release(SharedMemory &shm); + void swap(MsgI &a) + { + std::swap(ptr_, a.ptr_); + std::swap(count_, a.count_); + } + template <class T = void> + T *get() { return static_cast<T *>(ptr_.get()); } - int Count() const{ return IsCounted() ? count_->Get() : 1; } + // AddRef and Release works for both counted and not counted msg. + int AddRef() const { return IsCounted() ? count_->Inc() : 1; } + int Release(SharedMemory &shm); - bool IsCounted() const { return static_cast<bool>(count_); } + int Count() const { return IsCounted() ? count_->Get() : 1; } + bool IsCounted() const { return static_cast<bool>(count_); } - bool Make(SharedMemory &shm, const BHMsg &msg); - bool MakeRC(SharedMemory &shm, const BHMsg &msg); - bool Unpack(BHMsg &msg) const; + bool Make(SharedMemory &shm, const BHMsg &msg); + bool MakeRC(SharedMemory &shm, const BHMsg &msg); + bool Unpack(BHMsg &msg) const; }; inline void swap(MsgI &m1, MsgI &m2) { m1.swap(m2); } - } // namespace bhome_msg - - #endif // end of include guard: MSG_5BILLZET -- Gitblit v1.8.0