From dc01e4cbb01e96d19b470a366bbe648d426ed171 Mon Sep 17 00:00:00 2001
From: fujuntang <fujuntang@smartai.com>
Date: 星期六, 11 九月 2021 10:06:15 +0800
Subject: [PATCH] Add topics sub and request support.

---
 src/bh_api.cpp |   32 +++++++++++++++++++++++---------
 1 files changed, 23 insertions(+), 9 deletions(-)

diff --git a/src/bh_api.cpp b/src/bh_api.cpp
index 4f623f8..d25bf8a 100644
--- a/src/bh_api.cpp
+++ b/src/bh_api.cpp
@@ -139,6 +139,12 @@
     gNetmod_socket = net_mod_socket_open();
     hashtable_t *hashtable = mm_get_hashtable();
     key = hashtable_alloc_key(hashtable);
+    count = hashtable_alloc_key(hashtable);
+    rv = hashtable_alloc_key(hashtable);
+    net_mod_socket_int_set(gNetmod_socket, count);
+    net_mod_socket_svr_set(gNetmod_socket, rv);
+    sprintf(pData.int_info, "%d", count);
+    sprintf(pData.svr_info, "%d", rv);
     net_mod_socket_bind(gNetmod_socket, key);
   
     rv = net_mod_socket_reg(gNetmod_socket, &pData, sizeof(ProcInfo), NULL, 0, timeout_ms, PROC_REG);
@@ -932,8 +938,8 @@
   ::bhome_msg::MsgCommonReply mcr;
 	mcr.mutable_errmsg()->set_errcode(::bhome_msg::ErrorCode(rv));
 	mcr.mutable_errmsg()->set_errstring(errString);
-	*reply_len=mcr.ByteSizeLong();
-	*reply=malloc(*reply_len);
+	*reply_len = mcr.ByteSizeLong();
+	*reply = malloc(*reply_len);
 	mcr.SerializePartialToArray(*reply,*reply_len);
 #endif 
 
@@ -1207,6 +1213,7 @@
   int val;
   int len;
   int min;
+  int data;
   int sec, nsec;
   std::string MsgID;
   int timeout_ms = 3000;
@@ -1309,20 +1316,21 @@
         strncpy(topics_buf + strlen(buf_temp) + 1, _input1.data, strlen(_input1.data));
 #endif 
 
+        data = net_mod_socket_svr_get(gNetmod_socket); 
         if (timeout_ms > 0) {
 
           sec = timeout_ms / 1000;
           nsec = (timeout_ms - sec * 1000) * 1000 * 1000;
           
-          rv = net_mod_socket_sendto_timeout(gNetmod_socket, topics_buf, len, val, sec, nsec); 
+          rv = net_mod_socket_sendto_timeout(gNetmod_socket, topics_buf, len, val, sec, nsec, SVR_STR, data); 
           
         } else if (timeout_ms == 0) {
 
-          rv = net_mod_socket_sendto_nowait(gNetmod_socket, topics_buf, len, val); 
+          rv = net_mod_socket_sendto_nowait(gNetmod_socket, topics_buf, len, val, SVR_STR, data); 
 
         } else {
 
-          rv = net_mod_socket_sendto(gNetmod_socket, topics_buf, len, val); 
+          rv = net_mod_socket_sendto(gNetmod_socket, topics_buf, len, val, SVR_STR, data); 
         } 
 
         free(topics_buf);
@@ -1377,6 +1385,7 @@
   int size;
   int val;
   int min, len;
+  int data;
   net_node_t node;
   int node_size;  
   int recv_arr_size;
@@ -1469,6 +1478,7 @@
         len += strlen(_input1.data);
 #endif
 
+        data = net_mod_socket_svr_get(gNetmod_socket);
         topics_buf = (char *)malloc(len);
         if (topics_buf == NULL) {
           
@@ -1620,6 +1630,7 @@
   int key;
   int size;
   int len;
+  int data;
   int sec, nsec;
   char buf_temp[MAX_STR_LEN] = { 0x00 };
   char *topics_buf = NULL;
@@ -1642,20 +1653,21 @@
 		return false;
   }
 
+  data = net_mod_socket_svr_get(gNetmod_socket);
   if (timeout_ms > 0) {
 
     sec = timeout_ms / 1000;
     nsec = (timeout_ms - sec * 1000) * 1000 * 1000;
 
-    rv = net_mod_socket_recvfrom_timeout(gNetmod_socket, &buf, &size, &key, sec, nsec);
+    rv = net_mod_socket_recvfrom_timeout(gNetmod_socket, &buf, &size, &key, sec, nsec, SVR_STR, data);
 
   } else if (timeout_ms == 0) {
 
-    rv = net_mod_socket_recvfrom_nowait(gNetmod_socket, &buf, &size, &key);
+    rv = net_mod_socket_recvfrom_nowait(gNetmod_socket, &buf, &size, &key, SVR_STR, data);
   
   } else {
 
-    rv = net_mod_socket_recvfrom(gNetmod_socket, &buf, &size, &key);
+    rv = net_mod_socket_recvfrom(gNetmod_socket, &buf, &size, &key, SVR_STR, data);
   }
 
   if (rv == 0) {
@@ -1735,6 +1747,7 @@
 int BHSendReply(void *src, const void *reply, const int reply_len)
 {
   int rv;
+  int data;
   const char *_input;
   
 #if defined(PRO_DE_SERIALIZE)
@@ -1777,7 +1790,8 @@
   rv = pthread_mutex_trylock(&mutex);
   if (rv == 0) {
 
-    rv = net_mod_socket_sendto(gNetmod_socket, _input, strlen(_input), *(int *)src);
+    data = net_mod_socket_svr_get(gNetmod_socket);
+    rv = net_mod_socket_sendto(gNetmod_socket, _input, strlen(_input), *(int *)src, SVR_STR, data);
 
     memset(errString, 0x00, sizeof(errString));
     strncpy(errString, bus_strerror(rv), sizeof(errString));

--
Gitblit v1.8.0