From ae7d4f1f85e6a013ed796500876e6abda573b501 Mon Sep 17 00:00:00 2001
From: Yihao Deng <luomoxyz@126.com>
Date: Tue, 23 Jul 2024 05:46:08 +0000
Subject: [PATCH] refactor transport

---
 source/libs/transport/inc/transComm.h |  6 +++---
 source/libs/transport/src/trans.c     | 12 +++++++-----
 source/libs/transport/src/transComm.c |  8 ++++++--
 3 files changed, 16 insertions(+), 10 deletions(-)

diff --git a/source/libs/transport/inc/transComm.h b/source/libs/transport/inc/transComm.h
index 47eeeee5cb..73c0fb6228 100644
--- a/source/libs/transport/inc/transComm.h
+++ b/source/libs/transport/inc/transComm.h
@@ -435,9 +435,9 @@ bool transEpSetIsEqual2(SEpSet* a, SEpSet* b);
  */
 void transThreadOnce();
 
-void transInit();
-void transCleanup();
-void transPrintEpSet(SEpSet* pEpSet);
+int32_t transInit();
+void    transCleanup();
+void    transPrintEpSet(SEpSet* pEpSet);
 
 void    transFreeMsg(void* msg);
 int32_t transCompressMsg(char* msg, int32_t len);
diff --git a/source/libs/transport/src/trans.c b/source/libs/transport/src/trans.c
index d5a50ccf20..58b1d76d23 100644
--- a/source/libs/transport/src/trans.c
+++ b/source/libs/transport/src/trans.c
@@ -34,10 +34,14 @@ static int32_t transValidLocalFqdn(const char* localFqdn, uint32_t* ip) {
   return 0;
 }
 void* rpcOpen(const SRpcInit* pInit) {
-  rpcInit();
+  int32_t code = rpcInit();
+  if (code != 0) {
+    return NULL;
+  }
 
   SRpcInfo* pRpc = taosMemoryCalloc(1, sizeof(SRpcInfo));
   if (pRpc == NULL) {
+    // return TSDB_CODE_OUT_OF_MEMORY;
     return NULL;
   }
   if (pInit->label) {
@@ -202,10 +206,8 @@ int32_t rpcCvtErrCode(int32_t code) {
   return code;
 }
 
-int32_t rpcInit() {
-  transInit();
-  return 0;
-}
+int32_t rpcInit() { return transInit(); }
+
 void rpcCleanup(void) {
   transCleanup();
   transHttpEnvDestroy();
diff --git a/source/libs/transport/src/transComm.c b/source/libs/transport/src/transComm.c
index 53ee1bbca7..3acdef7f5e 100644
--- a/source/libs/transport/src/transComm.c
+++ b/source/libs/transport/src/transComm.c
@@ -692,9 +692,13 @@ static void transDestroyEnv() {
   transCloseRefMgt(transSyncMsgMgt);
 }
 
-void transInit() {
+int32_t transInit() {
   // init env
-  taosThreadOnce(&transModuleInit, transInitEnv);
+  int32_t code = taosThreadOnce(&transModuleInit, transInitEnv);
+  if (code != 0) {
+    code = TAOS_SYSTEM_ERROR(errno);
+  }
+  return code;
 }
 
 int32_t transGetRefMgt() { return refMgt; }