• Home
  • Features
  • Pricing
  • Docs
  • Announcements
  • Sign In

taosdata / TDengine / #4880

11 Dec 2025 02:43AM UTC coverage: 64.544%. Remained the same
#4880

push

travis-ci

guanshengliang
feat(TS-7270): internal dependence

307 of 617 new or added lines in 24 files covered. (49.76%)

3380 existing lines in 25 files now uncovered.

163565 of 253417 relevant lines covered (64.54%)

105600506.39 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

78.35
/source/client/src/clientMsgHandler.c
1
/*
2
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3
 *
4
 * This program is free software: you can use, redistribute, and/or modify
5
 * it under the terms of the GNU Affero General Public License, version 3
6
 * or later ("AGPL"), as published by the Free Software Foundation.
7
 *
8
 * This program is distributed in the hope that it will be useful, but WITHOUT
9
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10
 * FITNESS FOR A PARTICULAR PURPOSE.
11
 *
12
 * You should have received a copy of the GNU Affero General Public License
13
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
14
 */
15

16
#include "catalog.h"
17
#include "clientInt.h"
18
#include "clientLog.h"
19
#include "clientMonitor.h"
20
#include "cmdnodes.h"
21
#include "command.h"
22
#include "os.h"
23
#include "query.h"
24
#include "systable.h"
25
#include "tdatablock.h"
26
#include "tdef.h"
27
#include "tglobal.h"
28
#include "tname.h"
29
#include "tversion.h"
30

31
extern SClientHbMgr clientHbMgr;
32

33
static void setErrno(SRequestObj* pRequest, int32_t code) {
5,632,140✔
34
  pRequest->code = code;
5,632,140✔
35
  terrno = code;
5,632,140✔
36
}
5,632,140✔
37

38
int32_t genericRspCallback(void* param, SDataBuf* pMsg, int32_t code) {
4,406,690✔
39
  SRequestObj* pRequest = param;
4,406,690✔
40
  setErrno(pRequest, code);
4,406,690✔
41

42
  if (NEED_CLIENT_RM_TBLMETA_REQ(pRequest->type)) {
4,406,690✔
43
    if (removeMeta(pRequest->pTscObj, pRequest->targetTableList, IS_VIEW_REQUEST(pRequest->type)) != 0) {
1,281,892✔
44
      tscError("failed to remove meta data for table");
×
45
    }
46
  }
47

48
  taosMemoryFree(pMsg->pEpSet);
4,406,690✔
49
  taosMemoryFree(pMsg->pData);
4,406,690✔
50
  if (pRequest->body.queryFp != NULL) {
4,406,690✔
51
    doRequestCallback(pRequest, code);
4,406,386✔
52
  } else {
53
    if (tsem_post(&pRequest->body.rspSem) != 0) {
304✔
54
      tscError("failed to post semaphore");
×
55
    }
56
  }
57
  return code;
4,406,690✔
58
}
59

60
int32_t processConnectRsp(void* param, SDataBuf* pMsg, int32_t code) {
3,402,148✔
61
  SRequestObj* pRequest = acquireRequest(*(int64_t*)param);
3,402,148✔
62
  if (NULL == pRequest) {
3,402,148✔
63
    goto EXIT;
×
64
  }
65

66
  if (code != TSDB_CODE_SUCCESS) {
3,402,148✔
67
    goto End;
17,003✔
68
  }
69

70
  STscObj* pTscObj = pRequest->pTscObj;
3,385,145✔
71

72
  if (NULL == pTscObj->pAppInfo) {
3,385,145✔
73
    code = TSDB_CODE_TSC_DISCONNECTED;
×
74
    goto End;
×
75
  }
76

77
  if (pTscObj->connType == CONN_TYPE__AUTH_TEST) {
3,385,145✔
78
    // auth test connection, no need to process connect rsp
NEW
79
    goto End;
×
80
  }
81

82
  SConnectRsp connectRsp = {0};
3,385,145✔
83
  if (tDeserializeSConnectRsp(pMsg->pData, pMsg->len, &connectRsp) != 0) {
3,385,145✔
84
    code = TSDB_CODE_TSC_INVALID_VERSION;
×
UNCOV
85
    goto End;
×
86
  }
87

88
  if ((code = taosCheckVersionCompatibleFromStr(td_version, connectRsp.sVer, 3)) != 0) {
3,383,714✔
89
    tscError("version not compatible. client version:%s, server version:%s", td_version, connectRsp.sVer);
×
90
    goto End;
×
91
  }
92

93
  int32_t now = taosGetTimestampSec();
3,383,830✔
94
  int32_t delta = abs(now - connectRsp.svrTimestamp);
3,383,884✔
95
  if (delta > tsTimestampDeltaLimit) {
3,383,884✔
96
    code = TSDB_CODE_TIME_UNSYNCED;
171✔
97
    tscError("time diff:%ds is too big", delta);
171✔
UNCOV
98
    goto End;
×
99
  }
100

101
  if (connectRsp.epSet.numOfEps == 0) {
3,383,814✔
UNCOV
102
    code = TSDB_CODE_APP_ERROR;
×
103
    goto End;
×
104
  }
105

106
  int    updateEpSet = 1;
3,383,814✔
107
  SEpSet srcEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
3,383,814✔
108
  if (connectRsp.dnodeNum == 1) {
3,384,072✔
109
    SEpSet dstEpSet = connectRsp.epSet;
3,253,436✔
110
    if (srcEpSet.numOfEps == 1) {
3,253,436✔
111
      if (rpcSetDefaultAddr(pTscObj->pAppInfo->pTransporter, srcEpSet.eps[srcEpSet.inUse].fqdn,
2,320,957✔
112
                            dstEpSet.eps[dstEpSet.inUse].fqdn) != 0) {
2,319,631✔
UNCOV
113
        tscError("failed to set default addr for rpc");
×
114
      }
115
      updateEpSet = 0;
2,320,970✔
116
    }
117
  }
118
  if (updateEpSet == 1 && !isEpsetEqual(&srcEpSet, &connectRsp.epSet)) {
3,384,825✔
119
    SEpSet corEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
959,750✔
120

121
    SEpSet* pOrig = &corEpSet;
959,349✔
122
    SEp*    pOrigEp = &pOrig->eps[pOrig->inUse];
959,349✔
123
    SEp*    pNewEp = &connectRsp.epSet.eps[connectRsp.epSet.inUse];
959,349✔
124
    tscDebug("mnode epset updated from %d/%d=>%s:%d to %d/%d=>%s:%d in connRsp", pOrig->inUse, pOrig->numOfEps,
959,349✔
125
             pOrigEp->fqdn, pOrigEp->port, connectRsp.epSet.inUse, connectRsp.epSet.numOfEps, pNewEp->fqdn,
126
             pNewEp->port);
127
    updateEpSet_s(&pTscObj->pAppInfo->mgmtEp, &connectRsp.epSet);
959,349✔
128
  }
129

130
  for (int32_t i = 0; i < connectRsp.epSet.numOfEps; ++i) {
6,873,740✔
131
    tscDebug("QID:0x%" PRIx64 ", epSet.fqdn[%d]:%s port:%d, conn:0x%" PRIx64, pRequest->requestId, i,
3,489,455✔
132
             connectRsp.epSet.eps[i].fqdn, connectRsp.epSet.eps[i].port, pTscObj->id);
133
  }
134

135
  pTscObj->sysInfo = connectRsp.sysInfo;
3,384,285✔
136
  pTscObj->connId = connectRsp.connId;
3,383,857✔
137
  pTscObj->acctId = connectRsp.acctId;
3,384,190✔
138
  tstrncpy(pTscObj->sVer, connectRsp.sVer, tListLen(pTscObj->sVer));
3,383,313✔
139
  tstrncpy(pTscObj->sDetailVer, connectRsp.sDetailVer, tListLen(pTscObj->sDetailVer));
3,383,902✔
140

141
  // update the appInstInfo
142
  pTscObj->pAppInfo->clusterId = connectRsp.clusterId;
3,384,237✔
143
  pTscObj->pAppInfo->serverCfg.monitorParas = connectRsp.monitorParas;
3,383,093✔
144
  pTscObj->pAppInfo->serverCfg.enableAuditDelete = connectRsp.enableAuditDelete;
3,382,658✔
145
  tscDebug("monitor paras from connect rsp, clusterId:0x%" PRIx64 ", threshold:%d scope:%d",
3,383,914✔
146
           connectRsp.clusterId, connectRsp.monitorParas.tsSlowLogThreshold, connectRsp.monitorParas.tsSlowLogScope);
147
  lastClusterId = connectRsp.clusterId;
3,383,914✔
148

149
  pTscObj->connType = connectRsp.connType;
3,383,914✔
150
  pTscObj->passInfo.ver = connectRsp.passVer;
3,380,908✔
151
  pTscObj->authVer = connectRsp.authVer;
3,382,278✔
152
  pTscObj->whiteListInfo.ver = connectRsp.whiteListVer;
3,381,689✔
153

154
  if (taosHashGet(appInfo.pInstMapByClusterId, &connectRsp.clusterId, LONG_BYTES) == NULL) {
3,382,628✔
155
    if (taosHashPut(appInfo.pInstMapByClusterId, &connectRsp.clusterId, LONG_BYTES, &pTscObj->pAppInfo,
1,330,204✔
156
                    POINTER_BYTES) != 0) {
UNCOV
157
      tscError("failed to put appInfo into appInfo.pInstMapByClusterId");
×
158
    } else {
159
#ifdef USE_MONITOR
160
      MonitorSlowLogData data = {0};
1,330,204✔
161
      data.clusterId = pTscObj->pAppInfo->clusterId;
1,330,204✔
162
      data.type = SLOW_LOG_READ_BEGINNIG;
1,330,204✔
163
      (void)monitorPutData2MonitorQueue(data);  // ignore
1,330,204✔
164
      monitorClientSlowQueryInit(connectRsp.clusterId);
1,330,204✔
165
      monitorClientSQLReqInit(connectRsp.clusterId);
1,330,204✔
166
#endif
167
    }
168
  }
169

170
  (void)taosThreadMutexLock(&clientHbMgr.lock);
3,383,990✔
171
  SAppHbMgr* pAppHbMgr = taosArrayGetP(clientHbMgr.appHbMgrs, pTscObj->appHbMgrIdx);
3,385,145✔
172
  if (pAppHbMgr) {
3,385,145✔
173
    if (hbRegisterConn(pAppHbMgr, pTscObj->id, connectRsp.clusterId, connectRsp.connType) != 0) {
3,385,145✔
UNCOV
174
      tscError("QID:0x%" PRIx64 ", failed to register conn to hbMgr", pRequest->requestId);
×
175
    }
176
  } else {
UNCOV
177
    (void)taosThreadMutexUnlock(&clientHbMgr.lock);
×
UNCOV
178
    code = TSDB_CODE_TSC_DISCONNECTED;
×
179
    goto End;
×
180
  }
181
  (void)taosThreadMutexUnlock(&clientHbMgr.lock);
3,385,145✔
182

183
  tscDebug("QID:0x%" PRIx64 ", clusterId:0x%" PRIx64 ", totalConn:%" PRId64, pRequest->requestId, connectRsp.clusterId,
3,385,145✔
184
           pTscObj->pAppInfo->numOfConns);
185

186
End:
3,264,236✔
187
  if (code != 0) {
3,402,148✔
188
    setErrno(pRequest, code);
17,003✔
189
  }
190
  if (tsem_post(&pRequest->body.rspSem) != 0) {
3,402,148✔
UNCOV
191
    tscError("failed to post semaphore");
×
192
  }
193

194
  if (pRequest) {
3,402,148✔
195
    (void)releaseRequest(pRequest->self);
3,402,148✔
196
  }
197

198
EXIT:
2,842,457✔
199
  taosMemoryFree(param);
3,402,148✔
200
  taosMemoryFree(pMsg->pEpSet);
3,402,148✔
201
  taosMemoryFree(pMsg->pData);
3,402,148✔
202
  return code;
3,402,148✔
203
}
204

205
SMsgSendInfo* buildMsgInfoImpl(SRequestObj* pRequest) {
18,656,379✔
206
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
18,656,379✔
207
  if (pMsgSendInfo == NULL) return pMsgSendInfo;
18,655,975✔
208
  pMsgSendInfo->requestObjRefId = pRequest->self;
18,655,975✔
209
  pMsgSendInfo->requestId = pRequest->requestId;
18,655,975✔
210
  pMsgSendInfo->param = pRequest;
18,656,448✔
211
  pMsgSendInfo->msgType = pRequest->type;
18,655,975✔
212
  pMsgSendInfo->target.type = TARGET_TYPE_MNODE;
18,656,448✔
213

214
  pMsgSendInfo->msgInfo = pRequest->body.requestMsg;
18,656,448✔
215
  pMsgSendInfo->fp = getMsgRspHandle(pRequest->type);
18,656,448✔
216
  return pMsgSendInfo;
18,655,774✔
217
}
218

219
int32_t processCreateDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
1,482,613✔
220
  // todo rsp with the vnode id list
221
  SRequestObj* pRequest = param;
1,482,613✔
222
  taosMemoryFree(pMsg->pData);
1,482,613✔
223
  taosMemoryFree(pMsg->pEpSet);
1,482,613✔
224
  if (code != TSDB_CODE_SUCCESS) {
1,482,613✔
225
    setErrno(pRequest, code);
36,359✔
226
  } else {
227
    struct SCatalog* pCatalog = NULL;
1,446,254✔
228
    code = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,446,254✔
229
    if (TSDB_CODE_SUCCESS == code) {
1,446,254✔
230
      STscObj* pTscObj = pRequest->pTscObj;
1,446,254✔
231

232
      SRequestConnInfo conn = {.pTrans = pTscObj->pAppInfo->pTransporter,
1,446,254✔
233
                               .requestId = pRequest->requestId,
1,446,254✔
234
                               .requestObjRefId = pRequest->self,
1,446,254✔
235
                               .mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp)};
1,446,254✔
236
      char             dbFName[TSDB_DB_FNAME_LEN];
1,314,907✔
237
      (void)snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_INFORMATION_SCHEMA_DB);
1,446,254✔
238
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != 0) {
1,446,254✔
UNCOV
239
        tscError("QID:0x%" PRIx64 ", failed to refresh db vg info", pRequest->requestId);
×
240
      }
241
      (void)snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_PERFORMANCE_SCHEMA_DB);
1,446,254✔
242
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != 0) {
1,446,254✔
UNCOV
243
        tscError("QID:0x%" PRIx64 ", failed to refresh db vg info", pRequest->requestId);
×
244
      }
245
    }
246
  }
247

248
  if (pRequest->body.queryFp) {
1,482,613✔
249
    doRequestCallback(pRequest, code);
1,482,613✔
250
  } else {
UNCOV
251
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
252
      tscError("failed to post semaphore");
×
253
    }
254
  }
255
  return code;
1,482,613✔
256
}
257

258
int32_t processUseDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
3,090,895✔
259
  SRequestObj* pRequest = param;
3,090,895✔
260
  if (TSDB_CODE_MND_DB_NOT_EXIST == code || TSDB_CODE_MND_DB_IN_CREATING == code ||
3,090,895✔
261
      TSDB_CODE_MND_DB_IN_DROPPING == code) {
262
    SUseDbRsp usedbRsp = {0};
1,898✔
263
    if (tDeserializeSUseDbRsp(pMsg->pData, pMsg->len, &usedbRsp) != 0) {
1,898✔
264
      tscError("QID:0x%" PRIx64 ", deserialize SUseDbRsp failed", pRequest->requestId);
1,898✔
265
    }
266
    struct SCatalog* pCatalog = NULL;
1,898✔
267

268
    if (usedbRsp.vgVersion >= 0) {  // cached in local
1,898✔
269
      int64_t clusterId = pRequest->pTscObj->pAppInfo->clusterId;
1,898✔
270
      int32_t code1 = catalogGetHandle(clusterId, &pCatalog);
1,898✔
271
      if (code1 != TSDB_CODE_SUCCESS) {
1,898✔
UNCOV
272
        tscWarn("QID:0x%" PRIx64 ", catalogGetHandle failed, clusterId:0x%" PRIx64 ", error:%s", pRequest->requestId, clusterId,
×
273
                tstrerror(code1));
274
      } else {
275
        if (catalogRemoveDB(pCatalog, usedbRsp.db, usedbRsp.uid) != 0) {
1,898✔
UNCOV
276
          tscError("QID:0x%" PRIx64 ", catalogRemoveDB failed, db:%s, uid:%" PRId64, pRequest->requestId, usedbRsp.db,
×
277
                   usedbRsp.uid);
278
        }
279
      }
280
    }
281
    tFreeSUsedbRsp(&usedbRsp);
1,898✔
282
  }
283

284
  if (code != TSDB_CODE_SUCCESS) {
3,090,895✔
285
    taosMemoryFree(pMsg->pData);
4,400✔
286
    taosMemoryFree(pMsg->pEpSet);
4,400✔
287
    setErrno(pRequest, code);
4,400✔
288

289
    if (pRequest->body.queryFp != NULL) {
4,400✔
290
      doRequestCallback(pRequest, pRequest->code);
4,400✔
291

292
    } else {
UNCOV
293
      if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
294
        tscError("failed to post semaphore");
×
295
      }
296
    }
297

298
    return code;
4,400✔
299
  }
300

301
  SUseDbRsp usedbRsp = {0};
3,086,495✔
302
  if (tDeserializeSUseDbRsp(pMsg->pData, pMsg->len, &usedbRsp) != 0) {
3,086,495✔
UNCOV
303
    tscError("QID:0x%" PRIx64 ", deserialize SUseDbRsp failed", pRequest->requestId);
×
304
  }
305

306
  if (strlen(usedbRsp.db) == 0) {
3,086,495✔
UNCOV
307
    taosMemoryFree(pMsg->pData);
×
308
    taosMemoryFree(pMsg->pEpSet);
×
309

UNCOV
310
    if (usedbRsp.errCode != 0) {
×
UNCOV
311
      return usedbRsp.errCode;
×
312
    } else {
313
      return TSDB_CODE_APP_ERROR;
×
314
    }
315
  }
316

317
  tscTrace("db:%s, usedbRsp received, numOfVgroups:%d", usedbRsp.db, usedbRsp.vgNum);
3,086,495✔
318
  for (int32_t i = 0; i < usedbRsp.vgNum; ++i) {
5,700,377✔
319
    SVgroupInfo* pInfo = taosArrayGet(usedbRsp.pVgroupInfos, i);
2,613,882✔
320
    if (pInfo == NULL) {
2,613,882✔
UNCOV
321
      continue;
×
322
    }
323
    tscTrace("vgId:%d, numOfEps:%d inUse:%d ", pInfo->vgId, pInfo->epSet.numOfEps, pInfo->epSet.inUse);
2,613,882✔
324
    for (int32_t j = 0; j < pInfo->epSet.numOfEps; ++j) {
5,469,117✔
325
      tscTrace("vgId:%d, index:%d epset:%s:%u", pInfo->vgId, j, pInfo->epSet.eps[j].fqdn, pInfo->epSet.eps[j].port);
2,855,235✔
326
    }
327
  }
328

329
  SName name = {0};
3,086,495✔
330
  if (tNameFromString(&name, usedbRsp.db, T_NAME_ACCT | T_NAME_DB) != TSDB_CODE_SUCCESS) {
3,086,495✔
UNCOV
331
    tscError("QID:0x%" PRIx64 ", failed to parse db name:%s", pRequest->requestId, usedbRsp.db);
×
332
  }
333

334
  SUseDbOutput output = {0};
3,086,495✔
335
  code = queryBuildUseDbOutput(&output, &usedbRsp);
3,086,495✔
336
  if (code != 0) {
3,086,347✔
UNCOV
337
    terrno = code;
×
UNCOV
338
    if (output.dbVgroup) taosHashCleanup(output.dbVgroup->vgHash);
×
339

UNCOV
340
    tscError("QID:0x%" PRIx64 ", failed to build use db output since %s", pRequest->requestId, terrstr());
×
341
  } else if (output.dbVgroup && output.dbVgroup->vgHash) {
3,086,347✔
342
    struct SCatalog* pCatalog = NULL;
1,214,044✔
343

344
    int32_t code1 = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,214,044✔
345
    if (code1 != TSDB_CODE_SUCCESS) {
1,214,044✔
UNCOV
346
      tscWarn("catalogGetHandle failed, clusterId:0x%" PRIx64 ", error:%s", pRequest->pTscObj->pAppInfo->clusterId,
×
347
              tstrerror(code1));
348
    } else {
349
      if (catalogUpdateDBVgInfo(pCatalog, output.db, output.dbId, output.dbVgroup) != 0) {
1,214,044✔
UNCOV
350
        tscError("QID:0x%" PRIx64 ", failed to update db vg info, db:%s, dbId:%" PRId64, pRequest->requestId, output.db,
×
351
                 output.dbId);
352
      }
353
      output.dbVgroup = NULL;
1,214,044✔
354
    }
355
  }
356

357
  taosMemoryFreeClear(output.dbVgroup);
3,086,495✔
358
  tFreeSUsedbRsp(&usedbRsp);
3,086,495✔
359

360
  char db[TSDB_DB_NAME_LEN] = {0};
3,086,495✔
361
  if (tNameGetDbName(&name, db) != TSDB_CODE_SUCCESS) {
3,086,495✔
UNCOV
362
    tscError("QID:0x%" PRIx64 ", failed to get db name since %s", pRequest->requestId, tstrerror(code));
×
363
  }
364

365
  setConnectionDB(pRequest->pTscObj, db);
3,086,337✔
366

367
  taosMemoryFree(pMsg->pData);
3,086,495✔
368
  taosMemoryFree(pMsg->pEpSet);
3,086,220✔
369

370
  if (pRequest->body.queryFp != NULL) {
3,086,220✔
371
    doRequestCallback(pRequest, pRequest->code);
3,085,840✔
372
  } else {
UNCOV
373
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
374
      tscError("failed to post semaphore");
×
375
    }
376
  }
377
  return 0;
3,086,495✔
378
}
379

380
int32_t processCreateSTableRsp(void* param, SDataBuf* pMsg, int32_t code) {
2,115,997✔
381
  if (pMsg == NULL) {
2,115,997✔
UNCOV
382
    tscError("processCreateSTableRsp: invalid input param, pMsg is NULL");
×
UNCOV
383
    return TSDB_CODE_TSC_INVALID_INPUT;
×
384
  }
385
  if (param == NULL) {
2,115,997✔
UNCOV
386
    taosMemoryFree(pMsg->pEpSet);
×
387
    taosMemoryFree(pMsg->pData);
×
388
    tscError("processCreateSTableRsp: invalid input param, param is NULL");
×
UNCOV
389
    return TSDB_CODE_TSC_INVALID_INPUT;
×
390
  }
391

392
  SRequestObj* pRequest = param;
2,115,997✔
393

394
  if (code != TSDB_CODE_SUCCESS) {
2,115,997✔
395
    setErrno(pRequest, code);
34,146✔
396
  } else {
397
    SMCreateStbRsp createRsp = {0};
2,081,851✔
398
    SDecoder       coder = {0};
2,081,851✔
399
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
2,081,851✔
400
    if (pMsg->len > 0) {
2,081,851✔
401
      code = tDecodeSMCreateStbRsp(&coder, &createRsp);  // pMsg->len == 0
2,067,104✔
402
      if (code != TSDB_CODE_SUCCESS) {
2,067,104✔
UNCOV
403
        setErrno(pRequest, code);
×
404
      }
405
    }
406
    tDecoderClear(&coder);
2,081,851✔
407

408
    pRequest->body.resInfo.execRes.msgType = TDMT_MND_CREATE_STB;
2,081,851✔
409
    pRequest->body.resInfo.execRes.res = createRsp.pMeta;
2,081,851✔
410
  }
411

412
  taosMemoryFree(pMsg->pEpSet);
2,115,997✔
413
  taosMemoryFree(pMsg->pData);
2,115,997✔
414

415
  if (pRequest->body.queryFp != NULL) {
2,115,997✔
416
    SExecResult* pRes = &pRequest->body.resInfo.execRes;
1,710,326✔
417

418
    if (code == TSDB_CODE_SUCCESS) {
1,710,326✔
419
      SCatalog* pCatalog = NULL;
1,705,976✔
420
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,705,976✔
421
      if (pRes->res != NULL) {
1,705,976✔
422
        ret = handleCreateTbExecRes(pRes->res, pCatalog);
1,697,597✔
423
      }
424

425
      if (ret != TSDB_CODE_SUCCESS) {
1,705,976✔
UNCOV
426
        code = ret;
×
427
      }
428
    }
429

430
    doRequestCallback(pRequest, code);
1,710,326✔
431
  } else {
432
    if (tsem_post(&pRequest->body.rspSem) != 0) {
405,671✔
UNCOV
433
      tscError("failed to post semaphore");
×
434
    }
435
  }
436
  return code;
2,115,997✔
437
}
438

439
int32_t processDropDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
1,322,044✔
440
  SRequestObj* pRequest = param;
1,322,044✔
441
  if (code != TSDB_CODE_SUCCESS) {
1,322,044✔
442
    setErrno(pRequest, code);
12,146✔
443
  } else {
444
    SDropDbRsp dropdbRsp = {0};
1,309,898✔
445
    if (tDeserializeSDropDbRsp(pMsg->pData, pMsg->len, &dropdbRsp) != 0) {
1,309,898✔
UNCOV
446
      tscError("QID:0x%" PRIx64 ", deserialize SDropDbRsp failed", pRequest->requestId);
×
447
    }
448
    struct SCatalog* pCatalog = NULL;
1,309,898✔
449
    code = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,309,898✔
450
    if (TSDB_CODE_SUCCESS == code) {
1,309,898✔
451
      if (catalogRemoveDB(pCatalog, dropdbRsp.db, dropdbRsp.uid) != 0) {
1,309,898✔
UNCOV
452
        tscError("QID:0x%" PRIx64 ", failed to remove db:%s", pRequest->requestId, dropdbRsp.db);
×
453
      }
454
      STscObj* pTscObj = pRequest->pTscObj;
1,309,898✔
455

456
      SRequestConnInfo conn = {.pTrans = pTscObj->pAppInfo->pTransporter,
1,309,898✔
457
                               .requestId = pRequest->requestId,
1,309,898✔
458
                               .requestObjRefId = pRequest->self,
1,309,898✔
459
                               .mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp)};
1,309,898✔
460
      char             dbFName[TSDB_DB_FNAME_LEN] = {0};
1,309,898✔
461
      (void)snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_INFORMATION_SCHEMA_DB);
1,309,898✔
462
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != TSDB_CODE_SUCCESS) {
1,309,898✔
UNCOV
463
        tscError("QID:0x%" PRIx64 ", failed to refresh db vg info, db:%s", pRequest->requestId, dbFName);
×
464
      }
465
      (void)snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_PERFORMANCE_SCHEMA_DB);
1,309,898✔
466
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != 0) {
1,309,898✔
UNCOV
467
        tscError("QID:0x%" PRIx64 ", failed to refresh db vg info, db:%s", pRequest->requestId, dbFName);
×
468
      }
469
    }
470
  }
471

472
  taosMemoryFree(pMsg->pData);
1,322,044✔
473
  taosMemoryFree(pMsg->pEpSet);
1,322,044✔
474

475
  if (pRequest->body.queryFp != NULL) {
1,322,044✔
476
    doRequestCallback(pRequest, code);
1,322,044✔
477
  } else {
UNCOV
478
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
479
      tscError("failed to post semaphore");
×
480
    }
481
  }
482
  return code;
1,322,044✔
483
}
484

485
int32_t processAlterStbRsp(void* param, SDataBuf* pMsg, int32_t code) {
6,114,467✔
486
  SRequestObj* pRequest = param;
6,114,467✔
487
  if (code != TSDB_CODE_SUCCESS) {
6,114,467✔
488
    setErrno(pRequest, code);
1,104,224✔
489
  } else {
490
    SMAlterStbRsp alterRsp = {0};
5,010,243✔
491
    SDecoder      coder = {0};
5,010,243✔
492
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
5,010,243✔
493
    if (pMsg->len > 0) {
5,010,243✔
494
      code = tDecodeSMAlterStbRsp(&coder, &alterRsp);  // pMsg->len == 0
4,987,173✔
495
      if (code != TSDB_CODE_SUCCESS) {
4,987,173✔
UNCOV
496
        setErrno(pRequest, code);
×
497
      }
498
    }
499
    tDecoderClear(&coder);
5,010,243✔
500

501
    pRequest->body.resInfo.execRes.msgType = TDMT_MND_ALTER_STB;
5,010,243✔
502
    pRequest->body.resInfo.execRes.res = alterRsp.pMeta;
5,010,243✔
503
  }
504

505
  taosMemoryFree(pMsg->pData);
6,114,467✔
506
  taosMemoryFree(pMsg->pEpSet);
6,114,467✔
507

508
  if (pRequest->body.queryFp != NULL) {
6,114,467✔
509
    SExecResult* pRes = &pRequest->body.resInfo.execRes;
6,114,467✔
510

511
    if (code == TSDB_CODE_SUCCESS) {
6,114,467✔
512
      SCatalog* pCatalog = NULL;
5,010,243✔
513
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
5,010,243✔
514
      if (pRes->res != NULL) {
5,010,243✔
515
        ret = handleAlterTbExecRes(pRes->res, pCatalog);
4,987,173✔
516
      }
517

518
      if (ret != TSDB_CODE_SUCCESS) {
5,010,243✔
UNCOV
519
        code = ret;
×
520
      }
521
    }
522

523
    doRequestCallback(pRequest, code);
6,114,467✔
524
  } else {
UNCOV
525
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
526
      tscError("failed to post semaphore");
×
527
    }
528
  }
529
  return code;
6,114,467✔
530
}
531

532
static int32_t buildShowVariablesBlock(SArray* pVars, SSDataBlock** block) {
45,400✔
533
  int32_t      code = 0;
45,400✔
534
  int32_t      line = 0;
45,400✔
535
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
45,400✔
536
  TSDB_CHECK_NULL(pBlock, code, line, END, terrno);
45,400✔
537
  pBlock->info.hasVarCol = true;
45,400✔
538

539
  pBlock->pDataBlock = taosArrayInit(SHOW_VARIABLES_RESULT_COLS, sizeof(SColumnInfoData));
45,400✔
540
  TSDB_CHECK_NULL(pBlock->pDataBlock, code, line, END, terrno);
45,400✔
541
  SColumnInfoData infoData = {0};
45,400✔
542
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
45,400✔
543
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD1_LEN;
45,400✔
544
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
90,800✔
545

546
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
45,400✔
547
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD2_LEN;
45,400✔
548
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
90,800✔
549

550
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
45,400✔
551
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD3_LEN;
45,400✔
552
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
90,800✔
553

554
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
45,400✔
555
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD4_LEN;
45,400✔
556
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
90,800✔
557

558
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
45,400✔
559
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD5_LEN;
45,400✔
560
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
90,800✔
561

562
  int32_t numOfCfg = taosArrayGetSize(pVars);
45,400✔
563
  code = blockDataEnsureCapacity(pBlock, numOfCfg);
45,400✔
564
  TSDB_CHECK_CODE(code, line, END);
45,400✔
565

566
  for (int32_t i = 0, c = 0; i < numOfCfg; ++i, c = 0) {
4,614,661✔
567
    SVariablesInfo* pInfo = taosArrayGet(pVars, i);
4,569,261✔
568
    TSDB_CHECK_NULL(pInfo, code, line, END, terrno);
4,569,261✔
569

570
    char name[TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE] = {0};
4,569,261✔
571
    STR_WITH_MAXSIZE_TO_VARSTR(name, pInfo->name, TSDB_CONFIG_OPTION_LEN + VARSTR_HEADER_SIZE);
4,569,261✔
572
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
4,569,261✔
573
    TSDB_CHECK_NULL(pColInfo, code, line, END, terrno);
4,569,261✔
574
    code = colDataSetVal(pColInfo, i, name, false);
4,569,261✔
575
    TSDB_CHECK_CODE(code, line, END);
4,569,261✔
576

577
    char value[TSDB_CONFIG_VALUE_LEN + VARSTR_HEADER_SIZE] = {0};
4,569,261✔
578
    STR_WITH_MAXSIZE_TO_VARSTR(value, pInfo->value, TSDB_CONFIG_VALUE_LEN + VARSTR_HEADER_SIZE);
4,569,261✔
579
    pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
4,569,261✔
580
    TSDB_CHECK_NULL(pColInfo, code, line, END, terrno);
4,569,261✔
581
    code = colDataSetVal(pColInfo, i, value, false);
4,569,261✔
582
    TSDB_CHECK_CODE(code, line, END);
4,569,261✔
583

584
    char scope[TSDB_CONFIG_SCOPE_LEN + VARSTR_HEADER_SIZE] = {0};
4,569,261✔
585
    STR_WITH_MAXSIZE_TO_VARSTR(scope, pInfo->scope, TSDB_CONFIG_SCOPE_LEN + VARSTR_HEADER_SIZE);
4,569,261✔
586
    pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
4,569,261✔
587
    TSDB_CHECK_NULL(pColInfo, code, line, END, terrno);
4,569,261✔
588
    code = colDataSetVal(pColInfo, i, scope, false);
4,569,261✔
589
    TSDB_CHECK_CODE(code, line, END);
4,569,261✔
590

591
    char category[TSDB_CONFIG_CATEGORY_LEN + VARSTR_HEADER_SIZE] = {0};
4,569,261✔
592
    STR_WITH_MAXSIZE_TO_VARSTR(category, pInfo->category, TSDB_CONFIG_CATEGORY_LEN + VARSTR_HEADER_SIZE);
4,569,261✔
593
    pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
4,569,261✔
594
    TSDB_CHECK_NULL(pColInfo, code, line, END, terrno);
4,569,261✔
595
    code = colDataSetVal(pColInfo, i, category, false);
4,569,261✔
596
    TSDB_CHECK_CODE(code, line, END);
4,569,261✔
597

598
    char info[TSDB_CONFIG_INFO_LEN + VARSTR_HEADER_SIZE] = {0};
4,569,261✔
599
    STR_WITH_MAXSIZE_TO_VARSTR(info, pInfo->info, TSDB_CONFIG_INFO_LEN + VARSTR_HEADER_SIZE);
4,569,261✔
600
    pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
4,569,261✔
601
    TSDB_CHECK_NULL(pColInfo, code, line, END, terrno);
4,569,261✔
602
    code = colDataSetVal(pColInfo, i, info, false);
4,569,261✔
603
    TSDB_CHECK_CODE(code, line, END);
4,569,261✔
604
  }
605

606
  pBlock->info.rows = numOfCfg;
45,400✔
607

608
  *block = pBlock;
45,400✔
609
  return code;
45,400✔
610

UNCOV
611
END:
×
UNCOV
612
  taosArrayDestroy(pBlock->pDataBlock);
×
UNCOV
613
  taosMemoryFree(pBlock);
×
UNCOV
614
  return code;
×
615
}
616

617
static int32_t buildShowVariablesRsp(SArray* pVars, SRetrieveTableRsp** pRsp) {
45,400✔
618
  SSDataBlock* pBlock = NULL;
45,400✔
619
  int32_t      code = buildShowVariablesBlock(pVars, &pBlock);
45,400✔
620
  if (code) {
45,400✔
UNCOV
621
    return code;
×
622
  }
623

624
  size_t dataEncodeBufSize = blockGetEncodeSize(pBlock);
45,400✔
625
  size_t rspSize = sizeof(SRetrieveTableRsp) + dataEncodeBufSize + PAYLOAD_PREFIX_LEN;
45,400✔
626
  *pRsp = taosMemoryCalloc(1, rspSize);
45,400✔
627
  if (NULL == *pRsp) {
45,400✔
UNCOV
628
    code = terrno;
×
UNCOV
629
    goto _exit;
×
630
  }
631

632
  (*pRsp)->useconds = 0;
45,400✔
633
  (*pRsp)->completed = 1;
45,400✔
634
  (*pRsp)->precision = 0;
45,400✔
635
  (*pRsp)->compressed = 0;
45,400✔
636

637
  (*pRsp)->numOfRows = htobe64((int64_t)pBlock->info.rows);
45,400✔
638
  (*pRsp)->numOfCols = htonl(SHOW_VARIABLES_RESULT_COLS);
45,400✔
639

640
  int32_t len = 0;
45,400✔
641
  if ((*pRsp)->numOfRows > 0) {
45,400✔
642
    len = blockEncode(pBlock, (*pRsp)->data + PAYLOAD_PREFIX_LEN, dataEncodeBufSize, SHOW_VARIABLES_RESULT_COLS);
44,576✔
643
    if (len < 0) {
44,576✔
UNCOV
644
      uError("buildShowVariablesRsp error, len:%d", len);
×
UNCOV
645
      code = terrno;
×
UNCOV
646
      goto _exit;
×
647
    }
648
    SET_PAYLOAD_LEN((*pRsp)->data, len, len);
44,576✔
649

650
    int32_t payloadLen = len + PAYLOAD_PREFIX_LEN;
44,576✔
651
    (*pRsp)->payloadLen = htonl(payloadLen);
44,576✔
652
    (*pRsp)->compLen = htonl(payloadLen);
44,576✔
653

654
    if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
44,576✔
UNCOV
655
      uError("buildShowVariablesRsp error, len:%d != rspSize - sizeof(SRetrieveTableRsp):%" PRIu64, len,
×
656
             (uint64_t)(rspSize - sizeof(SRetrieveTableRsp)));
UNCOV
657
      code = TSDB_CODE_TSC_INVALID_INPUT;
×
UNCOV
658
      goto _exit;
×
659
    }
660
  }
661

662
  blockDataDestroy(pBlock);
45,400✔
663
  pBlock = NULL;
45,400✔
664

665
  return TSDB_CODE_SUCCESS;
45,400✔
UNCOV
666
_exit:
×
UNCOV
667
  if (*pRsp) {
×
UNCOV
668
    taosMemoryFree(*pRsp);
×
UNCOV
669
    *pRsp = NULL;
×
670
  }
671
  if (pBlock) {
×
672
    blockDataDestroy(pBlock);
×
673
    pBlock = NULL;
×
674
  }
UNCOV
675
  return code;
×
676
}
677

678
int32_t processShowVariablesRsp(void* param, SDataBuf* pMsg, int32_t code) {
45,400✔
679
  SRequestObj* pRequest = param;
45,400✔
680
  if (code != TSDB_CODE_SUCCESS) {
45,400✔
UNCOV
681
    setErrno(pRequest, code);
×
682
  } else {
683
    SShowVariablesRsp  rsp = {0};
45,400✔
684
    SRetrieveTableRsp* pRes = NULL;
45,400✔
685
    code = tDeserializeSShowVariablesRsp(pMsg->pData, pMsg->len, &rsp);
45,400✔
686
    if (TSDB_CODE_SUCCESS == code) {
45,400✔
687
      code = buildShowVariablesRsp(rsp.variables, &pRes);
45,400✔
688
    }
689
    if (TSDB_CODE_SUCCESS == code) {
45,400✔
690
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
45,400✔
691
    }
692

693
    if (code != 0) {
45,400✔
UNCOV
694
      pRequest->body.resInfo.pRspMsg = NULL;
×
UNCOV
695
      taosMemoryFree(pRes);
×
696
    }
697
    tFreeSShowVariablesRsp(&rsp);
45,400✔
698
  }
699

700
  taosMemoryFree(pMsg->pData);
45,400✔
701
  taosMemoryFree(pMsg->pEpSet);
45,400✔
702

703
  if (pRequest->body.queryFp != NULL) {
45,400✔
704
    doRequestCallback(pRequest, code);
45,400✔
705
  } else {
UNCOV
706
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
707
      tscError("failed to post semaphore");
×
708
    }
709
  }
710
  return code;
45,400✔
711
}
712

713
static int32_t buildCompactDbBlock(SCompactDbRsp* pRsp, SSDataBlock** block) {
60,940✔
714
  int32_t      code = 0;
60,940✔
715
  int32_t      line = 0;
60,940✔
716
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
60,940✔
717
  TSDB_CHECK_NULL(pBlock, code, line, END, terrno);
60,940✔
718
  pBlock->info.hasVarCol = true;
60,940✔
719

720
  pBlock->pDataBlock = taosArrayInit(COMPACT_DB_RESULT_COLS, sizeof(SColumnInfoData));
60,940✔
721
  TSDB_CHECK_NULL(pBlock->pDataBlock, code, line, END, terrno);
60,940✔
722
  SColumnInfoData infoData = {0};
60,940✔
723
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
60,940✔
724
  infoData.info.bytes = COMPACT_DB_RESULT_FIELD1_LEN;
60,940✔
725
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
121,880✔
726

727
  infoData.info.type = TSDB_DATA_TYPE_INT;
60,940✔
728
  infoData.info.bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;
60,940✔
729
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
121,880✔
730

731
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
60,940✔
732
  infoData.info.bytes = COMPACT_DB_RESULT_FIELD3_LEN;
60,940✔
733
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
121,880✔
734

735
  code = blockDataEnsureCapacity(pBlock, 1);
60,940✔
736
  TSDB_CHECK_CODE(code, line, END);
60,940✔
737

738
  SColumnInfoData* pResultCol = taosArrayGet(pBlock->pDataBlock, 0);
60,940✔
739
  TSDB_CHECK_NULL(pResultCol, code, line, END, terrno);
60,940✔
740
  SColumnInfoData* pIdCol = taosArrayGet(pBlock->pDataBlock, 1);
60,940✔
741
  TSDB_CHECK_NULL(pIdCol, code, line, END, terrno);
60,940✔
742
  SColumnInfoData* pReasonCol = taosArrayGet(pBlock->pDataBlock, 2);
60,940✔
743
  TSDB_CHECK_NULL(pReasonCol, code, line, END, terrno);
60,940✔
744

745
  char result[COMPACT_DB_RESULT_FIELD1_LEN] = {0};
60,940✔
746
  char reason[COMPACT_DB_RESULT_FIELD3_LEN] = {0};
60,940✔
747
  if (pRsp->bAccepted) {
60,940✔
748
    STR_TO_VARSTR(result, "accepted");
60,940✔
749
    code = colDataSetVal(pResultCol, 0, result, false);
60,940✔
750
    TSDB_CHECK_CODE(code, line, END);
60,940✔
751
    code = colDataSetVal(pIdCol, 0, (void*)&pRsp->compactId, false);
60,940✔
752
    TSDB_CHECK_CODE(code, line, END);
60,940✔
753
    STR_TO_VARSTR(reason, "success");
60,940✔
754
    code = colDataSetVal(pReasonCol, 0, reason, false);
60,940✔
755
    TSDB_CHECK_CODE(code, line, END);
60,940✔
756
  } else {
UNCOV
757
    STR_TO_VARSTR(result, "rejected");
×
UNCOV
758
    code = colDataSetVal(pResultCol, 0, result, false);
×
UNCOV
759
    TSDB_CHECK_CODE(code, line, END);
×
760
    colDataSetNULL(pIdCol, 0);
UNCOV
761
    STR_TO_VARSTR(reason, "compaction is ongoing");
×
762
    code = colDataSetVal(pReasonCol, 0, reason, false);
×
763
    TSDB_CHECK_CODE(code, line, END);
×
764
  }
765
  pBlock->info.rows = 1;
60,940✔
766

767
  *block = pBlock;
60,940✔
768

769
  return TSDB_CODE_SUCCESS;
60,940✔
UNCOV
770
END:
×
UNCOV
771
  taosMemoryFree(pBlock);
×
UNCOV
772
  taosArrayDestroy(pBlock->pDataBlock);
×
UNCOV
773
  return code;
×
774
}
775

776
static int32_t buildRetriveTableRspForCompactDb(SCompactDbRsp* pCompactDb, SRetrieveTableRsp** pRsp) {
60,940✔
777
  SSDataBlock* pBlock = NULL;
60,940✔
778
  int32_t      code = buildCompactDbBlock(pCompactDb, &pBlock);
60,940✔
779
  if (code) {
60,940✔
UNCOV
780
    return code;
×
781
  }
782

783
  size_t dataEncodeBufSize = blockGetEncodeSize(pBlock);
60,940✔
784
  size_t rspSize = sizeof(SRetrieveTableRsp) + dataEncodeBufSize + PAYLOAD_PREFIX_LEN;
60,940✔
785
  *pRsp = taosMemoryCalloc(1, rspSize);
60,940✔
786
  if (NULL == *pRsp) {
60,940✔
UNCOV
787
    code = terrno;
×
UNCOV
788
    goto _exit;
×
789
  }
790

791
  (*pRsp)->useconds = 0;
60,940✔
792
  (*pRsp)->completed = 1;
60,940✔
793
  (*pRsp)->precision = 0;
60,940✔
794
  (*pRsp)->compressed = 0;
60,940✔
795
  (*pRsp)->compLen = 0;
60,940✔
796
  (*pRsp)->payloadLen = 0;
60,940✔
797
  (*pRsp)->numOfRows = htobe64((int64_t)pBlock->info.rows);
60,940✔
798
  (*pRsp)->numOfCols = htonl(COMPACT_DB_RESULT_COLS);
60,940✔
799

800
  int32_t len = blockEncode(pBlock, (*pRsp)->data + PAYLOAD_PREFIX_LEN, dataEncodeBufSize, COMPACT_DB_RESULT_COLS);
60,940✔
801
  if (len < 0) {
60,940✔
UNCOV
802
    uError("buildRetriveTableRspForCompactDb error, len:%d", len);
×
UNCOV
803
    code = terrno;
×
UNCOV
804
    goto _exit;
×
805
  }
806
  blockDataDestroy(pBlock);
60,940✔
807

808
  SET_PAYLOAD_LEN((*pRsp)->data, len, len);
60,940✔
809

810
  int32_t payloadLen = len + PAYLOAD_PREFIX_LEN;
60,940✔
811
  (*pRsp)->payloadLen = htonl(payloadLen);
60,940✔
812
  (*pRsp)->compLen = htonl(payloadLen);
60,940✔
813

814
  if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
60,940✔
UNCOV
815
    uError("buildRetriveTableRspForCompactDb error, len:%d != rspSize - sizeof(SRetrieveTableRsp):%" PRIu64, len,
×
816
           (uint64_t)(rspSize - sizeof(SRetrieveTableRsp)));
UNCOV
817
    code = TSDB_CODE_TSC_INVALID_INPUT;
×
UNCOV
818
    goto _exit;
×
819
  }
820

821
  return TSDB_CODE_SUCCESS;
60,940✔
822
_exit:
×
823
  if (*pRsp) {
×
UNCOV
824
    taosMemoryFree(*pRsp);
×
UNCOV
825
    *pRsp = NULL;
×
826
  }
827
  if (pBlock) {
×
828
    blockDataDestroy(pBlock);
×
829
    pBlock = NULL;
×
830
  }
UNCOV
831
  return code;
×
832
}
833

834
int32_t processCompactDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
34,161✔
835
  SRequestObj* pRequest = param;
34,161✔
836
  if (code != TSDB_CODE_SUCCESS) {
34,161✔
837
    setErrno(pRequest, code);
2,680✔
838
  } else {
839
    SCompactDbRsp      rsp = {0};
31,481✔
840
    SRetrieveTableRsp* pRes = NULL;
31,481✔
841
    code = tDeserializeSCompactDbRsp(pMsg->pData, pMsg->len, &rsp);
31,481✔
842
    if (TSDB_CODE_SUCCESS == code) {
31,481✔
843
      code = buildRetriveTableRspForCompactDb(&rsp, &pRes);
31,481✔
844
    }
845
    if (TSDB_CODE_SUCCESS == code) {
31,481✔
846
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
31,481✔
847
    }
848

849
    if (code != 0) {
31,481✔
UNCOV
850
      pRequest->body.resInfo.pRspMsg = NULL;
×
UNCOV
851
      taosMemoryFree(pRes);
×
852
    }
853
  }
854

855
  taosMemoryFree(pMsg->pData);
34,161✔
856
  taosMemoryFree(pMsg->pEpSet);
34,161✔
857

858
  if (pRequest->body.queryFp != NULL) {
34,161✔
859
    pRequest->body.queryFp(((SSyncQueryParam*)pRequest->body.interParam)->userParam, pRequest, code);
34,161✔
860
  } else {
UNCOV
861
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
862
      tscError("failed to post semaphore");
×
863
    }
864
  }
865
  return code;
34,161✔
866
}
867

868
static int32_t buildScanDbBlock(SScanDbRsp* pRsp, SSDataBlock** block) {
518✔
869
  int32_t      code = 0;
518✔
870
  int32_t      line = 0;
518✔
871
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
518✔
872
  TSDB_CHECK_NULL(pBlock, code, line, END, terrno);
518✔
873
  pBlock->info.hasVarCol = true;
518✔
874

875
  pBlock->pDataBlock = taosArrayInit(SCAN_DB_RESULT_COLS, sizeof(SColumnInfoData));
518✔
876
  TSDB_CHECK_NULL(pBlock->pDataBlock, code, line, END, terrno);
518✔
877
  SColumnInfoData infoData = {0};
518✔
878
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
518✔
879
  infoData.info.bytes = SCAN_DB_RESULT_FIELD1_LEN;
518✔
880
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
1,036✔
881

882
  infoData.info.type = TSDB_DATA_TYPE_INT;
518✔
883
  infoData.info.bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;
518✔
884
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
1,036✔
885

886
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
518✔
887
  infoData.info.bytes = SCAN_DB_RESULT_FIELD3_LEN;
518✔
888
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
1,036✔
889

890
  code = blockDataEnsureCapacity(pBlock, 1);
518✔
891
  TSDB_CHECK_CODE(code, line, END);
518✔
892

893
  SColumnInfoData* pResultCol = taosArrayGet(pBlock->pDataBlock, 0);
518✔
894
  TSDB_CHECK_NULL(pResultCol, code, line, END, terrno);
518✔
895
  SColumnInfoData* pIdCol = taosArrayGet(pBlock->pDataBlock, 1);
518✔
896
  TSDB_CHECK_NULL(pIdCol, code, line, END, terrno);
518✔
897
  SColumnInfoData* pReasonCol = taosArrayGet(pBlock->pDataBlock, 2);
518✔
898
  TSDB_CHECK_NULL(pReasonCol, code, line, END, terrno);
518✔
899

900
  char result[SCAN_DB_RESULT_FIELD1_LEN] = {0};
518✔
901
  char reason[SCAN_DB_RESULT_FIELD3_LEN] = {0};
518✔
902
  if (pRsp->bAccepted) {
518✔
903
    STR_TO_VARSTR(result, "accepted");
518✔
904
    code = colDataSetVal(pResultCol, 0, result, false);
518✔
905
    TSDB_CHECK_CODE(code, line, END);
518✔
906
    code = colDataSetVal(pIdCol, 0, (void*)&pRsp->scanId, false);
518✔
907
    TSDB_CHECK_CODE(code, line, END);
518✔
908
    STR_TO_VARSTR(reason, "success");
518✔
909
    code = colDataSetVal(pReasonCol, 0, reason, false);
518✔
910
    TSDB_CHECK_CODE(code, line, END);
518✔
911
  } else {
UNCOV
912
    STR_TO_VARSTR(result, "rejected");
×
UNCOV
913
    code = colDataSetVal(pResultCol, 0, result, false);
×
UNCOV
914
    TSDB_CHECK_CODE(code, line, END);
×
915
    colDataSetNULL(pIdCol, 0);
UNCOV
916
    STR_TO_VARSTR(reason, "scan is ongoing");
×
917
    code = colDataSetVal(pReasonCol, 0, reason, false);
×
918
    TSDB_CHECK_CODE(code, line, END);
×
919
  }
920
  pBlock->info.rows = 1;
518✔
921

922
  *block = pBlock;
518✔
923

924
  return TSDB_CODE_SUCCESS;
518✔
UNCOV
925
END:
×
UNCOV
926
  taosMemoryFree(pBlock);
×
UNCOV
927
  taosArrayDestroy(pBlock->pDataBlock);
×
UNCOV
928
  return code;
×
929
}
930

931
static int32_t buildRetriveTableRspForScanDb(SScanDbRsp* pScanDb, SRetrieveTableRsp** pRsp) {
518✔
932
  SSDataBlock* pBlock = NULL;
518✔
933
  int32_t      code = buildScanDbBlock(pScanDb, &pBlock);
518✔
934
  if (code) {
518✔
UNCOV
935
    return code;
×
936
  }
937

938
  size_t dataEncodeBufSize = blockGetEncodeSize(pBlock);
518✔
939
  size_t rspSize = sizeof(SRetrieveTableRsp) + dataEncodeBufSize + PAYLOAD_PREFIX_LEN;
518✔
940
  *pRsp = taosMemoryCalloc(1, rspSize);
518✔
941
  if (NULL == *pRsp) {
518✔
UNCOV
942
    code = terrno;
×
UNCOV
943
    goto _exit;
×
944
  }
945

946
  (*pRsp)->useconds = 0;
518✔
947
  (*pRsp)->completed = 1;
518✔
948
  (*pRsp)->precision = 0;
518✔
949
  (*pRsp)->compressed = 0;
518✔
950
  (*pRsp)->compLen = 0;
518✔
951
  (*pRsp)->payloadLen = 0;
518✔
952
  (*pRsp)->numOfRows = htobe64((int64_t)pBlock->info.rows);
518✔
953
  (*pRsp)->numOfCols = htonl(SCAN_DB_RESULT_COLS);
518✔
954

955
  int32_t len = blockEncode(pBlock, (*pRsp)->data + PAYLOAD_PREFIX_LEN, dataEncodeBufSize, SCAN_DB_RESULT_COLS);
518✔
956
  if (len < 0) {
518✔
UNCOV
957
    uError("%s error, len:%d", __func__, len);
×
UNCOV
958
    code = terrno;
×
UNCOV
959
    goto _exit;
×
960
  }
961
  blockDataDestroy(pBlock);
518✔
962

963
  SET_PAYLOAD_LEN((*pRsp)->data, len, len);
518✔
964

965
  int32_t payloadLen = len + PAYLOAD_PREFIX_LEN;
518✔
966
  (*pRsp)->payloadLen = htonl(payloadLen);
518✔
967
  (*pRsp)->compLen = htonl(payloadLen);
518✔
968

969
  if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
518✔
UNCOV
970
    uError("%s error, len:%d != rspSize - sizeof(SRetrieveTableRsp):%" PRIu64, __func__, len,
×
971
           (uint64_t)(rspSize - sizeof(SRetrieveTableRsp)));
UNCOV
972
    code = TSDB_CODE_TSC_INVALID_INPUT;
×
UNCOV
973
    goto _exit;
×
974
  }
975

976
  return TSDB_CODE_SUCCESS;
518✔
977
_exit:
×
978
  if (*pRsp) {
×
UNCOV
979
    taosMemoryFree(*pRsp);
×
UNCOV
980
    *pRsp = NULL;
×
981
  }
982
  if (pBlock) {
×
983
    blockDataDestroy(pBlock);
×
984
    pBlock = NULL;
×
985
  }
UNCOV
986
  return code;
×
987
}
988

989
static int32_t processScanDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
772✔
990
  SRequestObj* pRequest = param;
772✔
991
  if (code != TSDB_CODE_SUCCESS) {
772✔
992
    setErrno(pRequest, code);
254✔
993
  } else {
994
    SScanDbRsp         rsp = {0};
518✔
995
    SRetrieveTableRsp* pRes = NULL;
518✔
996
    code = tDeserializeSScanDbRsp(pMsg->pData, pMsg->len, &rsp);
518✔
997
    if (TSDB_CODE_SUCCESS == code) {
518✔
998
      code = buildRetriveTableRspForScanDb(&rsp, &pRes);
518✔
999
    }
1000
    if (TSDB_CODE_SUCCESS == code) {
518✔
1001
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
518✔
1002
    }
1003

1004
    if (code != 0) {
518✔
UNCOV
1005
      pRequest->body.resInfo.pRspMsg = NULL;
×
UNCOV
1006
      taosMemoryFree(pRes);
×
1007
    }
1008
  }
1009

1010
  taosMemoryFree(pMsg->pData);
772✔
1011
  taosMemoryFree(pMsg->pEpSet);
772✔
1012

1013
  if (pRequest->body.queryFp != NULL) {
772✔
1014
    pRequest->body.queryFp(((SSyncQueryParam*)pRequest->body.interParam)->userParam, pRequest, code);
772✔
1015
  } else {
UNCOV
1016
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
1017
      tscError("failed to post semaphore");
×
1018
    }
1019
  }
1020
  return code;
772✔
1021
}
1022

1023
int32_t processTrimDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
43,697✔
1024
  SRequestObj* pRequest = param;
43,697✔
1025
  if (code != TSDB_CODE_SUCCESS) {
43,697✔
1026
    setErrno(pRequest, code);
14,238✔
1027
  } else {
1028
    STrimDbRsp         rsp = {0};
29,459✔
1029
    SRetrieveTableRsp* pRes = NULL;
29,459✔
1030
    code = tDeserializeSCompactDbRsp(pMsg->pData, pMsg->len, (SCompactDbRsp*)&rsp);
29,459✔
1031
    if (TSDB_CODE_SUCCESS == code) {
29,459✔
1032
      code = buildRetriveTableRspForCompactDb(&rsp, &pRes);
29,459✔
1033
    }
1034
    if (TSDB_CODE_SUCCESS == code) {
29,459✔
1035
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
29,459✔
1036
    }
1037

1038
    if (code != 0) {
29,459✔
UNCOV
1039
      pRequest->body.resInfo.pRspMsg = NULL;
×
UNCOV
1040
      taosMemoryFree(pRes);
×
1041
    }
1042
  }
1043

1044
  taosMemoryFree(pMsg->pData);
43,697✔
1045
  taosMemoryFree(pMsg->pEpSet);
43,697✔
1046

1047
  if (pRequest->body.queryFp != NULL) {
43,697✔
1048
    pRequest->body.queryFp(((SSyncQueryParam*)pRequest->body.interParam)->userParam, pRequest, code);
43,697✔
1049
  } else {
UNCOV
1050
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
UNCOV
1051
      tscError("failed to post semaphore");
×
1052
    }
1053
  }
1054
  return code;
43,697✔
1055
}
1056

1057
__async_send_cb_fn_t getMsgRspHandle(int32_t msgType) {
22,055,408✔
1058
  switch (msgType) {
22,055,408✔
1059
    case TDMT_MND_CONNECT:
3,402,148✔
1060
      return processConnectRsp;
3,402,148✔
1061
    case TDMT_MND_CREATE_DB:
1,482,613✔
1062
      return processCreateDbRsp;
1,482,613✔
1063
    case TDMT_MND_USE_DB:
3,090,639✔
1064
      return processUseDbRsp;
3,090,639✔
1065
    case TDMT_MND_CREATE_STB:
2,115,793✔
1066
      return processCreateSTableRsp;
2,115,793✔
1067
    case TDMT_MND_DROP_DB:
1,322,044✔
1068
      return processDropDbRsp;
1,322,044✔
1069
    case TDMT_MND_ALTER_STB:
6,114,467✔
1070
      return processAlterStbRsp;
6,114,467✔
1071
    case TDMT_MND_SHOW_VARIABLES:
45,400✔
1072
      return processShowVariablesRsp;
45,400✔
1073
    case TDMT_MND_COMPACT_DB:
34,161✔
1074
      return processCompactDbRsp;
34,161✔
1075
    case TDMT_MND_TRIM_DB:
43,697✔
1076
      return processTrimDbRsp;
43,697✔
1077
    case TDMT_MND_SCAN_DB:
772✔
1078
      return processScanDbRsp;
772✔
1079
    default:
4,403,674✔
1080
      return genericRspCallback;
4,403,674✔
1081
  }
1082
}
1083

STATUS · Troubleshooting · Open an Issue · Sales · Support · CAREERS · ENTERPRISE · START FREE · SCHEDULE DEMO
ANNOUNCEMENTS · TWITTER · TOS & SLA · Supported CI Services · What's a CI service? · Automated Testing

© 2026 Coveralls, Inc