From c64c54d8e75b9354dc49a7b6b2d326e7dd59eb37 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期四, 15 四月 2021 19:32:16 +0800
Subject: [PATCH] add api; fix send, socknode mem leak.
---
src/topic_node.h | 47 +++++++++++++++++++++++------------------------
1 files changed, 23 insertions(+), 24 deletions(-)
diff --git a/src/topic_node.h b/src/topic_node.h
index 60497ad..8c3c48e 100644
--- a/src/topic_node.h
+++ b/src/topic_node.h
@@ -29,51 +29,48 @@
class TopicNode
{
SharedMemory &shm_;
- MsgRegister info_;
+ ProcInfo info_;
SharedMemory &shm() { return shm_; }
public:
+ typedef std::function<void(std::string &proc_id, const void *data, const int len)> DataCB;
TopicNode(SharedMemory &shm);
~TopicNode();
- void StopAll();
// topic node
- bool Register(ProcInfo &body, MsgCommonReply &reply, const int timeout_ms);
+ bool Register(ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms);
+ bool Heartbeat(ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms);
+ bool Heartbeat(const int timeout_ms);
// topic rpc server
- typedef std::function<bool(const std::string &topic, const std::string &data, std::string &reply)> OnRequest;
- bool ServerStart(OnRequest const &cb, const int nworker = 2);
+ typedef std::function<bool(const std::string &client_proc_id, const MsgRequestTopic &request, MsgRequestTopicReply &reply)> ServerCB;
+ bool ServerStart(ServerCB const &cb, const int nworker = 2);
bool ServerRegisterRPC(MsgTopicList &topics, MsgCommonReply &reply, const int timeout_ms);
- bool ServerRecvRequest(void *&src_info, std::string &topic, std::string &data, const int timeout_ms);
- bool ServerSendReply(void *src_info, const std::string &data, const int timeout_ms);
+ bool ServerRecvRequest(void *&src_info, std::string &proc_id, MsgRequestTopic &request, const int timeout_ms);
+ bool ServerSendReply(void *src_info, const MsgRequestTopicReply &reply);
// topic client
- typedef std::function<void(const std::string &data)> RequestResultCB;
+ typedef std::function<void(const BHMsgHead &head, const MsgRequestTopicReply &reply)> RequestResultCB;
bool ClientStartWorker(RequestResultCB const &cb, const int nworker = 2);
- bool ClientAsyncRequest(const Topic &topic, const void *data, const size_t size, const int timeout_ms, const RequestResultCB &rrcb = RequestResultCB());
- bool ClientAsyncRequest(const Topic &topic, const std::string &data, const int timeout_ms, const RequestResultCB &rrcb = RequestResultCB())
- {
- return ClientAsyncRequest(topic, data.data(), data.size(), timeout_ms, rrcb);
- }
- bool ClientSyncRequest(const Topic &topic, const void *data, const size_t size, std::string &out, const int timeout_ms);
- bool ClientSyncRequest(const Topic &topic, const std::string &data, std::string &out, const int timeout_ms)
- {
- return ClientSyncRequest(topic, data.data(), data.size(), out, timeout_ms);
- }
+ bool ClientAsyncRequest(const MsgRequestTopic &request, std::string &msg_id, const RequestResultCB &rrcb = RequestResultCB());
+ bool ClientSyncRequest(const MsgRequestTopic &request, std::string &proc_id, MsgRequestTopicReply &reply, const int timeout_ms);
// publish
- bool Publish(const Topic &topic, const void *data, const size_t size, const int timeout_ms);
+ bool Publish(const MsgPublish &pub, const int timeout_ms);
// subscribe
- typedef std::function<void(const std::string &proc_id, const Topic &topic, const std::string &data)> TopicDataCB;
- bool SubscribeStartWorker(const TopicDataCB &tdcb, int nworker = 2);
- bool Subscribe(MsgTopicList &topics, const int timeout_ms);
- bool RecvSub(std::string &proc_id, Topic &topic, std::string &data, const int timeout_ms);
+ typedef std::function<void(const std::string &proc_id, const MsgPublish &data)> SubDataCB;
+ bool SubscribeStartWorker(const SubDataCB &tdcb, int nworker = 2);
+ bool Subscribe(MsgTopicList &topics, MsgCommonReply &reply_body, const int timeout_ms);
+ bool RecvSub(std::string &proc_id, MsgPublish &pub, const int timeout_ms);
+
+ void Start(ServerCB const &server_cb, SubDataCB const &sub_cb, RequestResultCB &client_cb, int nworker = 2);
+ void Stop();
private:
bool ClientQueryRPCTopic(const Topic &topic, bhome::msg::BHAddress &addr, const int timeout_ms);
- const std::string &proc_id() { return info_.proc().proc_id(); }
+ const std::string &proc_id() { return info_.proc_id(); }
typedef bhome_msg::BHAddress Address;
class TopicQueryCache
@@ -121,7 +118,9 @@
auto &SockClient() { return SockRequest(); }
auto &SockReply() { return sock_reply_; }
auto &SockServer() { return SockReply(); }
+ bool IsRegistered() const { return registered_.load(); }
+ std::atomic<bool> registered_;
ShmSocket sock_node_;
ShmSocket sock_request_;
ShmSocket sock_reply_;
--
Gitblit v1.8.0