From d0f789744b6521f0ce3bf82cf870f47fe6bf9aac Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Fri, 23 Dec 2022 16:04:48 +0800
Subject: [PATCH] query(fix): return new error code except connect msg

---
 source/client/src/clientImpl.c | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/source/client/src/clientImpl.c b/source/client/src/clientImpl.c
index 4b4a1c368f..456a059191 100644
--- a/source/client/src/clientImpl.c
+++ b/source/client/src/clientImpl.c
@@ -1425,18 +1425,18 @@ void processMsgFromServer(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet) {
   }
 
   // pMsg is response msg
-  if (pMsg->msgType != TDMT_MND_CONNECT + 1) {
-    // uniform to one error code: TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED
-    if (pMsg->code == TSDB_CODE_RPC_SOMENODE_BROKEN_LINK) {
-      pMsg->code = TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED;
-    }
-  } else {
+  if (pMsg->msgType == TDMT_MND_CONNECT + 1) {
     // restore origin code
     if (pMsg->code == TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED) {
       pMsg->code = TSDB_CODE_RPC_NETWORK_UNAVAIL;
     } else if (pMsg->code == TSDB_CODE_RPC_SOMENODE_BROKEN_LINK) {
       pMsg->code = TSDB_CODE_RPC_BROKEN_LINK;
     }
+  } else {
+    // uniform to one error code: TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED
+    if (pMsg->code == TSDB_CODE_RPC_SOMENODE_BROKEN_LINK) {
+      pMsg->code = TSDB_CODE_RPC_SOMENODE_NOT_CONNECTED;
+    }
   }
 
   AsyncArg* arg = taosMemoryCalloc(1, sizeof(AsyncArg));