query(fix): return new error code except connect msg

This commit is contained in:
Alex Duan 2022-12-23 16:02:08 +08:00
parent cd1b87031a
commit 99ef55c7d4
1 changed files with 12 additions and 18 deletions

View File

@ -1425,24 +1425,18 @@ void processMsgFromServer(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet) {
}
// pMsg is response msg
switch (pMsg->msgType) {
case TDMT_VND_BATCH_META + 1:
case TDMT_VND_SUBMIT + 1:
case TDMT_SCH_QUERY + 1:
case TDMT_SCH_MERGE_QUERY + 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;
}
break;
default:
// 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;
}
break;
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 {
// 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;
}
}
AsyncArg* arg = taosMemoryCalloc(1, sizeof(AsyncArg));