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

taosdata / TDengine / #4877

11 Dec 2025 02:43AM UTC coverage: 64.586% (-0.05%) from 64.632%
#4877

push

travis-ci

guanshengliang
feat(TS-7270): internal dependence

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

673 existing lines in 130 files now uncovered.

163673 of 253417 relevant lines covered (64.59%)

105540806.95 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,373,292✔
34
  pRequest->code = code;
5,373,292✔
35
  terrno = code;
5,373,292✔
36
}
5,373,292✔
37

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

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

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

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

66
  if (code != TSDB_CODE_SUCCESS) {
3,132,269✔
67
    goto End;
20,816✔
68
  }
69

70
  STscObj* pTscObj = pRequest->pTscObj;
3,111,453✔
71

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

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

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

88
  if ((code = taosCheckVersionCompatibleFromStr(td_version, connectRsp.sVer, 3)) != 0) {
3,111,457✔
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,111,270✔
94
  int32_t delta = abs(now - connectRsp.svrTimestamp);
3,110,989✔
95
  if (delta > tsTimestampDeltaLimit) {
3,110,989✔
96
    code = TSDB_CODE_TIME_UNSYNCED;
129✔
97
    tscError("time diff:%ds is too big", delta);
129✔
98
    goto End;
×
99
  }
100

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

106
  int    updateEpSet = 1;
3,110,930✔
107
  SEpSet srcEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
3,110,930✔
108
  if (connectRsp.dnodeNum == 1) {
3,110,403✔
109
    SEpSet dstEpSet = connectRsp.epSet;
2,988,485✔
110
    if (srcEpSet.numOfEps == 1) {
2,988,485✔
111
      if (rpcSetDefaultAddr(pTscObj->pAppInfo->pTransporter, srcEpSet.eps[srcEpSet.inUse].fqdn,
2,100,534✔
112
                            dstEpSet.eps[dstEpSet.inUse].fqdn) != 0) {
2,099,778✔
113
        tscError("failed to set default addr for rpc");
×
114
      }
115
      updateEpSet = 0;
2,099,887✔
116
    }
117
  }
118
  if (updateEpSet == 1 && !isEpsetEqual(&srcEpSet, &connectRsp.epSet)) {
3,110,770✔
119
    SEpSet corEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
912,072✔
120

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

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

135
  pTscObj->sysInfo = connectRsp.sysInfo;
3,110,033✔
136
  pTscObj->connId = connectRsp.connId;
3,110,274✔
137
  pTscObj->acctId = connectRsp.acctId;
3,110,508✔
138
  tstrncpy(pTscObj->sVer, connectRsp.sVer, tListLen(pTscObj->sVer));
3,110,014✔
139
  tstrncpy(pTscObj->sDetailVer, connectRsp.sDetailVer, tListLen(pTscObj->sDetailVer));
3,111,001✔
140

141
  // update the appInstInfo
142
  pTscObj->pAppInfo->clusterId = connectRsp.clusterId;
3,111,001✔
143
  pTscObj->pAppInfo->serverCfg.monitorParas = connectRsp.monitorParas;
3,110,747✔
144
  pTscObj->pAppInfo->serverCfg.enableAuditDelete = connectRsp.enableAuditDelete;
3,110,277✔
145
  tscDebug("monitor paras from connect rsp, clusterId:0x%" PRIx64 ", threshold:%d scope:%d",
3,110,767✔
146
           connectRsp.clusterId, connectRsp.monitorParas.tsSlowLogThreshold, connectRsp.monitorParas.tsSlowLogScope);
147
  lastClusterId = connectRsp.clusterId;
3,110,767✔
148

149
  pTscObj->connType = connectRsp.connType;
3,110,767✔
150
  pTscObj->passInfo.ver = connectRsp.passVer;
3,109,095✔
151
  pTscObj->authVer = connectRsp.authVer;
3,109,588✔
152
  pTscObj->whiteListInfo.ver = connectRsp.whiteListVer;
3,109,569✔
153

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

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

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

186
End:
3,001,357✔
187
  if (code != 0) {
3,132,273✔
188
    setErrno(pRequest, code);
20,816✔
189
  }
190
  if (tsem_post(&pRequest->body.rspSem) != 0) {
3,132,273✔
191
    tscError("failed to post semaphore");
×
192
  }
193

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

198
EXIT:
2,589,419✔
199
  taosMemoryFree(param);
3,132,273✔
200
  taosMemoryFree(pMsg->pEpSet);
3,132,273✔
201
  taosMemoryFree(pMsg->pData);
3,132,273✔
202
  return code;
3,132,273✔
203
}
204

205
SMsgSendInfo* buildMsgInfoImpl(SRequestObj* pRequest) {
17,894,592✔
206
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
17,894,592✔
207
  if (pMsgSendInfo == NULL) return pMsgSendInfo;
17,893,922✔
208
  pMsgSendInfo->requestObjRefId = pRequest->self;
17,893,922✔
209
  pMsgSendInfo->requestId = pRequest->requestId;
17,893,922✔
210
  pMsgSendInfo->param = pRequest;
17,894,281✔
211
  pMsgSendInfo->msgType = pRequest->type;
17,893,945✔
212
  pMsgSendInfo->target.type = TARGET_TYPE_MNODE;
17,894,513✔
213

214
  pMsgSendInfo->msgInfo = pRequest->body.requestMsg;
17,894,536✔
215
  pMsgSendInfo->fp = getMsgRspHandle(pRequest->type);
17,894,020✔
216
  return pMsgSendInfo;
17,894,455✔
217
}
218

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

232
      SRequestConnInfo conn = {.pTrans = pTscObj->pAppInfo->pTransporter,
1,381,091✔
233
                               .requestId = pRequest->requestId,
1,381,091✔
234
                               .requestObjRefId = pRequest->self,
1,381,091✔
235
                               .mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp)};
1,381,091✔
236
      char             dbFName[TSDB_DB_FNAME_LEN];
1,257,457✔
237
      (void)snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_INFORMATION_SCHEMA_DB);
1,381,091✔
238
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != 0) {
1,381,091✔
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,381,091✔
242
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != 0) {
1,381,091✔
243
        tscError("QID:0x%" PRIx64 ", failed to refresh db vg info", pRequest->requestId);
×
244
      }
245
    }
246
  }
247

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

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

268
    if (usedbRsp.vgVersion >= 0) {  // cached in local
1,850✔
269
      int64_t clusterId = pRequest->pTscObj->pAppInfo->clusterId;
1,850✔
270
      int32_t code1 = catalogGetHandle(clusterId, &pCatalog);
1,850✔
271
      if (code1 != TSDB_CODE_SUCCESS) {
1,850✔
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,850✔
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,850✔
282
  }
283

284
  if (code != TSDB_CODE_SUCCESS) {
2,950,914✔
285
    taosMemoryFree(pMsg->pData);
4,208✔
286
    taosMemoryFree(pMsg->pEpSet);
4,208✔
287
    setErrno(pRequest, code);
4,208✔
288

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

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

298
    return code;
4,208✔
299
  }
300

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

306
  if (strlen(usedbRsp.db) == 0) {
2,946,706✔
307
    taosMemoryFree(pMsg->pData);
×
308
    taosMemoryFree(pMsg->pEpSet);
×
309

310
    if (usedbRsp.errCode != 0) {
×
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);
2,946,706✔
318
  for (int32_t i = 0; i < usedbRsp.vgNum; ++i) {
5,478,803✔
319
    SVgroupInfo* pInfo = taosArrayGet(usedbRsp.pVgroupInfos, i);
2,532,097✔
320
    if (pInfo == NULL) {
2,532,097✔
321
      continue;
×
322
    }
323
    tscTrace("vgId:%d, numOfEps:%d inUse:%d ", pInfo->vgId, pInfo->epSet.numOfEps, pInfo->epSet.inUse);
2,532,097✔
324
    for (int32_t j = 0; j < pInfo->epSet.numOfEps; ++j) {
5,279,586✔
325
      tscTrace("vgId:%d, index:%d epset:%s:%u", pInfo->vgId, j, pInfo->epSet.eps[j].fqdn, pInfo->epSet.eps[j].port);
2,747,489✔
326
    }
327
  }
328

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

334
  SUseDbOutput output = {0};
2,946,706✔
335
  code = queryBuildUseDbOutput(&output, &usedbRsp);
2,946,706✔
336
  if (code != 0) {
2,946,706✔
337
    terrno = code;
×
338
    if (output.dbVgroup) taosHashCleanup(output.dbVgroup->vgHash);
×
339

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

344
    int32_t code1 = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,180,370✔
345
    if (code1 != TSDB_CODE_SUCCESS) {
1,180,370✔
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,180,370✔
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,180,370✔
354
    }
355
  }
356

357
  taosMemoryFreeClear(output.dbVgroup);
2,946,706✔
358
  tFreeSUsedbRsp(&usedbRsp);
2,946,706✔
359

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

365
  setConnectionDB(pRequest->pTscObj, db);
2,946,706✔
366

367
  taosMemoryFree(pMsg->pData);
2,946,706✔
368
  taosMemoryFree(pMsg->pEpSet);
2,946,706✔
369

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

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

392
  SRequestObj* pRequest = param;
2,004,185✔
393

394
  if (code != TSDB_CODE_SUCCESS) {
2,004,185✔
395
    setErrno(pRequest, code);
30,225✔
396
  } else {
397
    SMCreateStbRsp createRsp = {0};
1,973,960✔
398
    SDecoder       coder = {0};
1,973,960✔
399
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
1,973,960✔
400
    if (pMsg->len > 0) {
1,973,960✔
401
      code = tDecodeSMCreateStbRsp(&coder, &createRsp);  // pMsg->len == 0
1,960,818✔
402
      if (code != TSDB_CODE_SUCCESS) {
1,960,818✔
403
        setErrno(pRequest, code);
×
404
      }
405
    }
406
    tDecoderClear(&coder);
1,973,960✔
407

408
    pRequest->body.resInfo.execRes.msgType = TDMT_MND_CREATE_STB;
1,973,960✔
409
    pRequest->body.resInfo.execRes.res = createRsp.pMeta;
1,973,960✔
410
  }
411

412
  taosMemoryFree(pMsg->pEpSet);
2,004,185✔
413
  taosMemoryFree(pMsg->pData);
2,004,185✔
414

415
  if (pRequest->body.queryFp != NULL) {
2,004,185✔
416
    SExecResult* pRes = &pRequest->body.resInfo.execRes;
1,624,645✔
417

418
    if (code == TSDB_CODE_SUCCESS) {
1,624,645✔
419
      SCatalog* pCatalog = NULL;
1,620,409✔
420
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,620,409✔
421
      if (pRes->res != NULL) {
1,620,409✔
422
        ret = handleCreateTbExecRes(pRes->res, pCatalog);
1,612,771✔
423
      }
424

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

430
    doRequestCallback(pRequest, code);
1,624,645✔
431
  } else {
432
    if (tsem_post(&pRequest->body.rspSem) != 0) {
379,540✔
433
      tscError("failed to post semaphore");
×
434
    }
435
  }
436
  return code;
2,004,185✔
437
}
438

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

456
      SRequestConnInfo conn = {.pTrans = pTscObj->pAppInfo->pTransporter,
1,255,040✔
457
                               .requestId = pRequest->requestId,
1,255,320✔
458
                               .requestObjRefId = pRequest->self,
1,255,320✔
459
                               .mgmtEps = getEpSet_s(&pTscObj->pAppInfo->mgmtEp)};
1,255,320✔
460
      char             dbFName[TSDB_DB_FNAME_LEN] = {0};
1,255,320✔
461
      (void)snprintf(dbFName, sizeof(dbFName) - 1, "%d.%s", pTscObj->acctId, TSDB_INFORMATION_SCHEMA_DB);
1,255,320✔
462
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != TSDB_CODE_SUCCESS) {
1,255,320✔
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,255,320✔
466
      if (catalogRefreshDBVgInfo(pCatalog, &conn, dbFName) != 0) {
1,255,320✔
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,266,385✔
473
  taosMemoryFree(pMsg->pEpSet);
1,266,385✔
474

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

485
int32_t processAlterStbRsp(void* param, SDataBuf* pMsg, int32_t code) {
5,917,750✔
486
  SRequestObj* pRequest = param;
5,917,750✔
487
  if (code != TSDB_CODE_SUCCESS) {
5,917,750✔
488
    setErrno(pRequest, code);
1,034,811✔
489
  } else {
490
    SMAlterStbRsp alterRsp = {0};
4,882,939✔
491
    SDecoder      coder = {0};
4,882,939✔
492
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
4,882,939✔
493
    if (pMsg->len > 0) {
4,882,939✔
494
      code = tDecodeSMAlterStbRsp(&coder, &alterRsp);  // pMsg->len == 0
4,860,629✔
495
      if (code != TSDB_CODE_SUCCESS) {
4,860,629✔
496
        setErrno(pRequest, code);
×
497
      }
498
    }
499
    tDecoderClear(&coder);
4,882,939✔
500

501
    pRequest->body.resInfo.execRes.msgType = TDMT_MND_ALTER_STB;
4,882,939✔
502
    pRequest->body.resInfo.execRes.res = alterRsp.pMeta;
4,882,939✔
503
  }
504

505
  taosMemoryFree(pMsg->pData);
5,917,750✔
506
  taosMemoryFree(pMsg->pEpSet);
5,917,750✔
507

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

511
    if (code == TSDB_CODE_SUCCESS) {
5,917,750✔
512
      SCatalog* pCatalog = NULL;
4,882,939✔
513
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
4,882,939✔
514
      if (pRes->res != NULL) {
4,882,939✔
515
        ret = handleAlterTbExecRes(pRes->res, pCatalog);
4,860,629✔
516
      }
517

518
      if (ret != TSDB_CODE_SUCCESS) {
4,882,939✔
519
        code = ret;
×
520
      }
521
    }
522

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

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

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

546
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,623✔
547
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD2_LEN;
42,623✔
548
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,246✔
549

550
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,623✔
551
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD3_LEN;
42,623✔
552
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,246✔
553

554
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,623✔
555
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD4_LEN;
42,623✔
556
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,246✔
557

558
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,623✔
559
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD5_LEN;
42,623✔
560
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,246✔
561

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

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

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

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

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

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

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

606
  pBlock->info.rows = numOfCfg;
42,623✔
607

608
  *block = pBlock;
42,623✔
609
  return code;
42,623✔
610

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

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

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

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

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

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

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

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

662
  blockDataDestroy(pBlock);
42,623✔
663
  pBlock = NULL;
42,623✔
664

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

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

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

700
  taosMemoryFree(pMsg->pData);
42,623✔
701
  taosMemoryFree(pMsg->pEpSet);
42,623✔
702

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

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

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

727
  infoData.info.type = TSDB_DATA_TYPE_INT;
59,129✔
728
  infoData.info.bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;
59,129✔
729
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
118,258✔
730

731
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
59,129✔
732
  infoData.info.bytes = COMPACT_DB_RESULT_FIELD3_LEN;
59,129✔
733
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
118,258✔
734

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

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

745
  char result[COMPACT_DB_RESULT_FIELD1_LEN] = {0};
59,129✔
746
  char reason[COMPACT_DB_RESULT_FIELD3_LEN] = {0};
59,129✔
747
  if (pRsp->bAccepted) {
59,129✔
748
    STR_TO_VARSTR(result, "accepted");
59,129✔
749
    code = colDataSetVal(pResultCol, 0, result, false);
59,129✔
750
    TSDB_CHECK_CODE(code, line, END);
59,129✔
751
    code = colDataSetVal(pIdCol, 0, (void*)&pRsp->compactId, false);
59,129✔
752
    TSDB_CHECK_CODE(code, line, END);
59,129✔
753
    STR_TO_VARSTR(reason, "success");
59,129✔
754
    code = colDataSetVal(pReasonCol, 0, reason, false);
59,129✔
755
    TSDB_CHECK_CODE(code, line, END);
59,129✔
756
  } else {
757
    STR_TO_VARSTR(result, "rejected");
×
758
    code = colDataSetVal(pResultCol, 0, result, false);
×
759
    TSDB_CHECK_CODE(code, line, END);
×
760
    colDataSetNULL(pIdCol, 0);
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;
59,129✔
766

767
  *block = pBlock;
59,129✔
768

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

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

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

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

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

808
  SET_PAYLOAD_LEN((*pRsp)->data, len, len);
59,129✔
809

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

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

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

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

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

855
  taosMemoryFree(pMsg->pData);
32,958✔
856
  taosMemoryFree(pMsg->pEpSet);
32,958✔
857

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

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

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

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

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

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

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

900
  char result[SCAN_DB_RESULT_FIELD1_LEN] = {0};
452✔
901
  char reason[SCAN_DB_RESULT_FIELD3_LEN] = {0};
452✔
902
  if (pRsp->bAccepted) {
452✔
903
    STR_TO_VARSTR(result, "accepted");
452✔
904
    code = colDataSetVal(pResultCol, 0, result, false);
452✔
905
    TSDB_CHECK_CODE(code, line, END);
452✔
906
    code = colDataSetVal(pIdCol, 0, (void*)&pRsp->scanId, false);
452✔
907
    TSDB_CHECK_CODE(code, line, END);
452✔
908
    STR_TO_VARSTR(reason, "success");
452✔
909
    code = colDataSetVal(pReasonCol, 0, reason, false);
452✔
910
    TSDB_CHECK_CODE(code, line, END);
452✔
911
  } else {
912
    STR_TO_VARSTR(result, "rejected");
×
913
    code = colDataSetVal(pResultCol, 0, result, false);
×
914
    TSDB_CHECK_CODE(code, line, END);
×
915
    colDataSetNULL(pIdCol, 0);
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;
452✔
921

922
  *block = pBlock;
452✔
923

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1044
  taosMemoryFree(pMsg->pData);
42,609✔
1045
  taosMemoryFree(pMsg->pEpSet);
42,609✔
1046

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

1057
__async_send_cb_fn_t getMsgRspHandle(int32_t msgType) {
21,025,778✔
1058
  switch (msgType) {
21,025,778✔
1059
    case TDMT_MND_CONNECT:
3,132,196✔
1060
      return processConnectRsp;
3,132,196✔
1061
    case TDMT_MND_CREATE_DB:
1,409,514✔
1062
      return processCreateDbRsp;
1,409,514✔
1063
    case TDMT_MND_USE_DB:
2,950,891✔
1064
      return processUseDbRsp;
2,950,891✔
1065
    case TDMT_MND_CREATE_STB:
2,004,104✔
1066
      return processCreateSTableRsp;
2,004,104✔
1067
    case TDMT_MND_DROP_DB:
1,266,385✔
1068
      return processDropDbRsp;
1,266,385✔
1069
    case TDMT_MND_ALTER_STB:
5,917,750✔
1070
      return processAlterStbRsp;
5,917,750✔
1071
    case TDMT_MND_SHOW_VARIABLES:
42,623✔
1072
      return processShowVariablesRsp;
42,623✔
1073
    case TDMT_MND_COMPACT_DB:
32,958✔
1074
      return processCompactDbRsp;
32,958✔
1075
    case TDMT_MND_TRIM_DB:
42,609✔
1076
      return processTrimDbRsp;
42,609✔
1077
    case TDMT_MND_SCAN_DB:
674✔
1078
      return processScanDbRsp;
674✔
1079
    default:
4,226,074✔
1080
      return genericRspCallback;
4,226,074✔
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