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

taosdata / TDengine / #4888

17 Dec 2025 07:43AM UTC coverage: 65.281% (-0.008%) from 65.289%
#4888

push

travis-ci

web-flow
test: update ci env (#33959)

178960 of 274136 relevant lines covered (65.28%)

101703773.12 hits per line

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

78.16
/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) {
4,959,830✔
34
  pRequest->code = code;
4,959,830✔
35
  terrno = code;
4,959,830✔
36
}
4,959,830✔
37

38
int32_t genericRspCallback(void* param, SDataBuf* pMsg, int32_t code) {
3,840,554✔
39
  SRequestObj* pRequest = param;
3,840,554✔
40
  setErrno(pRequest, code);
3,840,554✔
41

42
  if (NEED_CLIENT_RM_TBLMETA_REQ(pRequest->type)) {
3,840,554✔
43
    if (removeMeta(pRequest->pTscObj, pRequest->targetTableList, IS_VIEW_REQUEST(pRequest->type)) != 0) {
1,235,739✔
44
      tscError("failed to remove meta data for table");
×
45
    }
46
  }
47

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

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

66
  if (code != TSDB_CODE_SUCCESS) {
2,702,475✔
67
    goto End;
13,247✔
68
  }
69

70
  STscObj* pTscObj = pRequest->pTscObj;
2,689,228✔
71

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

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

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

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

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

106
  int    updateEpSet = 1;
2,688,978✔
107
  SEpSet srcEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
2,688,978✔
108
  if (connectRsp.dnodeNum == 1) {
2,688,719✔
109
    SEpSet dstEpSet = connectRsp.epSet;
2,555,789✔
110
    if (srcEpSet.numOfEps == 1) {
2,555,789✔
111
      if (rpcSetDefaultAddr(pTscObj->pAppInfo->pTransporter, srcEpSet.eps[srcEpSet.inUse].fqdn,
1,713,897✔
112
                            dstEpSet.eps[dstEpSet.inUse].fqdn) != 0) {
1,712,641✔
113
        tscError("failed to set default addr for rpc");
×
114
      }
115
      updateEpSet = 0;
1,713,918✔
116
    }
117
  }
118
  if (updateEpSet == 1 && !isEpsetEqual(&srcEpSet, &connectRsp.epSet)) {
2,689,228✔
119
    SEpSet corEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
866,529✔
120

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

130
  for (int32_t i = 0; i < connectRsp.epSet.numOfEps; ++i) {
5,484,486✔
131
    tscDebug("QID:0x%" PRIx64 ", epSet.fqdn[%d]:%s port:%d, conn:0x%" PRIx64, pRequest->requestId, i,
2,797,604✔
132
             connectRsp.epSet.eps[i].fqdn, connectRsp.epSet.eps[i].port, pTscObj->id);
133
  }
134

135
  pTscObj->sysInfo = connectRsp.sysInfo;
2,686,882✔
136
  pTscObj->connId = connectRsp.connId;
2,688,514✔
137
  pTscObj->acctId = connectRsp.acctId;
2,688,100✔
138
  tstrncpy(pTscObj->sVer, connectRsp.sVer, tListLen(pTscObj->sVer));
2,687,493✔
139
  tstrncpy(pTscObj->sDetailVer, connectRsp.sDetailVer, tListLen(pTscObj->sDetailVer));
2,688,047✔
140

141
  // update the appInstInfo
142
  pTscObj->pAppInfo->clusterId = connectRsp.clusterId;
2,688,978✔
143
  pTscObj->pAppInfo->serverCfg.monitorParas = connectRsp.monitorParas;
2,687,011✔
144
  pTscObj->pAppInfo->serverCfg.enableAuditDelete = connectRsp.enableAuditDelete;
2,687,862✔
145
  pTscObj->pAppInfo->serverCfg.enableAuditSelect = connectRsp.enableAuditSelect;
2,688,109✔
146
  pTscObj->pAppInfo->serverCfg.enableAuditInsert = connectRsp.enableAuditInsert;
2,687,017✔
147
  pTscObj->pAppInfo->serverCfg.auditLevel = connectRsp.auditLevel;
2,686,244✔
148
  tscDebug("monitor paras from connect rsp, clusterId:0x%" PRIx64 ", threshold:%d scope:%d",
2,686,686✔
149
           connectRsp.clusterId, connectRsp.monitorParas.tsSlowLogThreshold, connectRsp.monitorParas.tsSlowLogScope);
150
  lastClusterId = connectRsp.clusterId;
2,686,686✔
151

152
  pTscObj->connType = connectRsp.connType;
2,686,686✔
153
  pTscObj->passInfo.ver = connectRsp.passVer;
2,685,765✔
154
  pTscObj->authVer = connectRsp.authVer;
2,687,858✔
155
  pTscObj->whiteListInfo.ver = connectRsp.whiteListVer;
2,686,971✔
156

157
  if (taosHashGet(appInfo.pInstMapByClusterId, &connectRsp.clusterId, LONG_BYTES) == NULL) {
2,687,767✔
158
    if (taosHashPut(appInfo.pInstMapByClusterId, &connectRsp.clusterId, LONG_BYTES, &pTscObj->pAppInfo,
1,186,943✔
159
                    POINTER_BYTES) != 0) {
160
      tscError("failed to put appInfo into appInfo.pInstMapByClusterId");
×
161
    } else {
162
#ifdef USE_MONITOR
163
      MonitorSlowLogData data = {0};
1,186,943✔
164
      data.clusterId = pTscObj->pAppInfo->clusterId;
1,186,943✔
165
      data.type = SLOW_LOG_READ_BEGINNIG;
1,186,943✔
166
      (void)monitorPutData2MonitorQueue(data);  // ignore
1,186,943✔
167
      monitorClientSlowQueryInit(connectRsp.clusterId);
1,186,943✔
168
      monitorClientSQLReqInit(connectRsp.clusterId);
1,186,943✔
169
#endif
170
    }
171
  }
172

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

186
  tscDebug("QID:0x%" PRIx64 ", clusterId:0x%" PRIx64 ", totalConn:%" PRId64, pRequest->requestId, connectRsp.clusterId,
2,689,228✔
187
           pTscObj->pAppInfo->numOfConns);
188

189
End:
2,680,056✔
190
  if (code != 0) {
2,702,475✔
191
    setErrno(pRequest, code);
13,247✔
192
  }
193
  if (tsem_post(&pRequest->body.rspSem) != 0) {
2,702,475✔
194
    tscError("failed to post semaphore");
×
195
  }
196

197
  if (pRequest) {
2,702,475✔
198
    (void)releaseRequest(pRequest->self);
2,702,475✔
199
  }
200

201
EXIT:
2,636,275✔
202
  taosMemoryFree(param);
2,702,475✔
203
  taosMemoryFree(pMsg->pEpSet);
2,702,475✔
204
  taosMemoryFree(pMsg->pData);
2,702,475✔
205
  return code;
2,702,475✔
206
}
207

208
SMsgSendInfo* buildMsgInfoImpl(SRequestObj* pRequest) {
16,891,684✔
209
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
16,891,684✔
210
  if (pMsgSendInfo == NULL) return pMsgSendInfo;
16,891,104✔
211
  pMsgSendInfo->requestObjRefId = pRequest->self;
16,891,104✔
212
  pMsgSendInfo->requestId = pRequest->requestId;
16,891,104✔
213
  pMsgSendInfo->param = pRequest;
16,891,684✔
214
  pMsgSendInfo->msgType = pRequest->type;
16,891,394✔
215
  pMsgSendInfo->target.type = TARGET_TYPE_MNODE;
16,890,596✔
216

217
  pMsgSendInfo->msgInfo = pRequest->body.requestMsg;
16,891,222✔
218
  pMsgSendInfo->fp = getMsgRspHandle(pRequest->type);
16,891,684✔
219
  return pMsgSendInfo;
16,891,666✔
220
}
221

222
int32_t processCreateDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
1,315,468✔
223
  // todo rsp with the vnode id list
224
  SRequestObj* pRequest = param;
1,315,468✔
225
  taosMemoryFree(pMsg->pData);
1,315,468✔
226
  taosMemoryFree(pMsg->pEpSet);
1,315,468✔
227
  if (code != TSDB_CODE_SUCCESS) {
1,315,468✔
228
    setErrno(pRequest, code);
30,027✔
229
  } else {
230
    struct SCatalog* pCatalog = NULL;
1,285,441✔
231
    code = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,285,441✔
232
    if (TSDB_CODE_SUCCESS == code) {
1,285,441✔
233
      STscObj* pTscObj = pRequest->pTscObj;
1,285,441✔
234

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

251
  if (pRequest->body.queryFp) {
1,315,468✔
252
    doRequestCallback(pRequest, code);
1,315,468✔
253
  } else {
254
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
255
      tscError("failed to post semaphore");
×
256
    }
257
  }
258
  return code;
1,315,468✔
259
}
260

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

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

287
  if (code != TSDB_CODE_SUCCESS) {
2,464,150✔
288
    taosMemoryFree(pMsg->pData);
4,299✔
289
    taosMemoryFree(pMsg->pEpSet);
4,299✔
290
    setErrno(pRequest, code);
4,299✔
291

292
    if (pRequest->body.queryFp != NULL) {
4,299✔
293
      doRequestCallback(pRequest, pRequest->code);
4,299✔
294

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

301
    return code;
4,299✔
302
  }
303

304
  SUseDbRsp usedbRsp = {0};
2,459,851✔
305
  if (tDeserializeSUseDbRsp(pMsg->pData, pMsg->len, &usedbRsp) != 0) {
2,459,851✔
306
    tscError("QID:0x%" PRIx64 ", deserialize SUseDbRsp failed", pRequest->requestId);
×
307
  }
308

309
  if (strlen(usedbRsp.db) == 0) {
2,459,851✔
310
    taosMemoryFree(pMsg->pData);
×
311
    taosMemoryFree(pMsg->pEpSet);
×
312

313
    if (usedbRsp.errCode != 0) {
×
314
      return usedbRsp.errCode;
×
315
    } else {
316
      return TSDB_CODE_APP_ERROR;
×
317
    }
318
  }
319

320
  tscTrace("db:%s, usedbRsp received, numOfVgroups:%d", usedbRsp.db, usedbRsp.vgNum);
2,459,639✔
321
  for (int32_t i = 0; i < usedbRsp.vgNum; ++i) {
4,858,410✔
322
    SVgroupInfo* pInfo = taosArrayGet(usedbRsp.pVgroupInfos, i);
2,398,771✔
323
    if (pInfo == NULL) {
2,398,771✔
324
      continue;
×
325
    }
326
    tscTrace("vgId:%d, numOfEps:%d inUse:%d ", pInfo->vgId, pInfo->epSet.numOfEps, pInfo->epSet.inUse);
2,398,771✔
327
    for (int32_t j = 0; j < pInfo->epSet.numOfEps; ++j) {
5,042,113✔
328
      tscTrace("vgId:%d, index:%d epset:%s:%u", pInfo->vgId, j, pInfo->epSet.eps[j].fqdn, pInfo->epSet.eps[j].port);
2,643,342✔
329
    }
330
  }
331

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

337
  SUseDbOutput output = {0};
2,459,639✔
338
  code = queryBuildUseDbOutput(&output, &usedbRsp);
2,459,639✔
339
  if (code != 0) {
2,459,639✔
340
    terrno = code;
×
341
    if (output.dbVgroup) taosHashCleanup(output.dbVgroup->vgHash);
×
342

343
    tscError("QID:0x%" PRIx64 ", failed to build use db output since %s", pRequest->requestId, terrstr());
×
344
  } else if (output.dbVgroup && output.dbVgroup->vgHash) {
2,459,639✔
345
    struct SCatalog* pCatalog = NULL;
1,092,271✔
346

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

360
  taosMemoryFreeClear(output.dbVgroup);
2,459,431✔
361
  tFreeSUsedbRsp(&usedbRsp);
2,459,431✔
362

363
  char db[TSDB_DB_NAME_LEN] = {0};
2,459,431✔
364
  if (tNameGetDbName(&name, db) != TSDB_CODE_SUCCESS) {
2,459,639✔
365
    tscError("QID:0x%" PRIx64 ", failed to get db name since %s", pRequest->requestId, tstrerror(code));
×
366
  }
367

368
  setConnectionDB(pRequest->pTscObj, db);
2,459,643✔
369

370
  taosMemoryFree(pMsg->pData);
2,459,639✔
371
  taosMemoryFree(pMsg->pEpSet);
2,459,431✔
372

373
  if (pRequest->body.queryFp != NULL) {
2,459,643✔
374
    doRequestCallback(pRequest, pRequest->code);
2,459,851✔
375
  } else {
376
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
377
      tscError("failed to post semaphore");
×
378
    }
379
  }
380
  return 0;
2,459,639✔
381
}
382

383
int32_t processCreateSTableRsp(void* param, SDataBuf* pMsg, int32_t code) {
1,902,458✔
384
  if (pMsg == NULL) {
1,902,458✔
385
    tscError("processCreateSTableRsp: invalid input param, pMsg is NULL");
×
386
    return TSDB_CODE_TSC_INVALID_INPUT;
×
387
  }
388
  if (param == NULL) {
1,902,458✔
389
    taosMemoryFree(pMsg->pEpSet);
×
390
    taosMemoryFree(pMsg->pData);
×
391
    tscError("processCreateSTableRsp: invalid input param, param is NULL");
×
392
    return TSDB_CODE_TSC_INVALID_INPUT;
×
393
  }
394

395
  SRequestObj* pRequest = param;
1,902,458✔
396

397
  if (code != TSDB_CODE_SUCCESS) {
1,902,458✔
398
    setErrno(pRequest, code);
10,264✔
399
  } else {
400
    SMCreateStbRsp createRsp = {0};
1,892,194✔
401
    SDecoder       coder = {0};
1,892,194✔
402
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
1,892,194✔
403
    if (pMsg->len > 0) {
1,892,194✔
404
      code = tDecodeSMCreateStbRsp(&coder, &createRsp);  // pMsg->len == 0
1,883,114✔
405
      if (code != TSDB_CODE_SUCCESS) {
1,883,114✔
406
        setErrno(pRequest, code);
×
407
      }
408
    }
409
    tDecoderClear(&coder);
1,892,194✔
410

411
    pRequest->body.resInfo.execRes.msgType = TDMT_MND_CREATE_STB;
1,892,194✔
412
    pRequest->body.resInfo.execRes.res = createRsp.pMeta;
1,892,194✔
413
  }
414

415
  taosMemoryFree(pMsg->pEpSet);
1,902,458✔
416
  taosMemoryFree(pMsg->pData);
1,902,458✔
417

418
  if (pRequest->body.queryFp != NULL) {
1,902,458✔
419
    SExecResult* pRes = &pRequest->body.resInfo.execRes;
1,562,168✔
420

421
    if (code == TSDB_CODE_SUCCESS) {
1,562,168✔
422
      SCatalog* pCatalog = NULL;
1,556,856✔
423
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
1,556,856✔
424
      if (pRes->res != NULL) {
1,556,856✔
425
        ret = handleCreateTbExecRes(pRes->res, pCatalog);
1,550,052✔
426
      }
427

428
      if (ret != TSDB_CODE_SUCCESS) {
1,556,856✔
429
        code = ret;
×
430
      }
431
    }
432

433
    doRequestCallback(pRequest, code);
1,562,168✔
434
  } else {
435
    if (tsem_post(&pRequest->body.rspSem) != 0) {
340,290✔
436
      tscError("failed to post semaphore");
×
437
    }
438
  }
439
  return code;
1,902,458✔
440
}
441

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

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

475
  taosMemoryFree(pMsg->pData);
1,188,760✔
476
  taosMemoryFree(pMsg->pEpSet);
1,188,760✔
477

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

488
int32_t processAlterStbRsp(void* param, SDataBuf* pMsg, int32_t code) {
6,079,855✔
489
  SRequestObj* pRequest = param;
6,079,855✔
490
  if (code != TSDB_CODE_SUCCESS) {
6,079,855✔
491
    setErrno(pRequest, code);
1,041,087✔
492
  } else {
493
    SMAlterStbRsp alterRsp = {0};
5,038,768✔
494
    SDecoder      coder = {0};
5,038,768✔
495
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
5,038,768✔
496
    if (pMsg->len > 0) {
5,038,768✔
497
      code = tDecodeSMAlterStbRsp(&coder, &alterRsp);  // pMsg->len == 0
5,016,107✔
498
      if (code != TSDB_CODE_SUCCESS) {
5,016,107✔
499
        setErrno(pRequest, code);
×
500
      }
501
    }
502
    tDecoderClear(&coder);
5,038,768✔
503

504
    pRequest->body.resInfo.execRes.msgType = TDMT_MND_ALTER_STB;
5,038,768✔
505
    pRequest->body.resInfo.execRes.res = alterRsp.pMeta;
5,038,768✔
506
  }
507

508
  taosMemoryFree(pMsg->pData);
6,079,855✔
509
  taosMemoryFree(pMsg->pEpSet);
6,079,855✔
510

511
  if (pRequest->body.queryFp != NULL) {
6,079,855✔
512
    SExecResult* pRes = &pRequest->body.resInfo.execRes;
6,079,855✔
513

514
    if (code == TSDB_CODE_SUCCESS) {
6,079,855✔
515
      SCatalog* pCatalog = NULL;
5,038,768✔
516
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
5,038,768✔
517
      if (pRes->res != NULL) {
5,038,768✔
518
        ret = handleAlterTbExecRes(pRes->res, pCatalog);
5,016,107✔
519
      }
520

521
      if (ret != TSDB_CODE_SUCCESS) {
5,038,768✔
522
        code = ret;
×
523
      }
524
    }
525

526
    doRequestCallback(pRequest, code);
6,079,855✔
527
  } else {
528
    if (tsem_post(&pRequest->body.rspSem) != 0) {
×
529
      tscError("failed to post semaphore");
×
530
    }
531
  }
532
  return code;
6,079,855✔
533
}
534

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

542
  pBlock->pDataBlock = taosArrayInit(SHOW_VARIABLES_RESULT_COLS, sizeof(SColumnInfoData));
42,822✔
543
  TSDB_CHECK_NULL(pBlock->pDataBlock, code, line, END, terrno);
42,822✔
544
  SColumnInfoData infoData = {0};
42,822✔
545
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,822✔
546
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD1_LEN;
42,822✔
547
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,644✔
548

549
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,822✔
550
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD2_LEN;
42,822✔
551
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,644✔
552

553
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,822✔
554
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD3_LEN;
42,822✔
555
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,644✔
556

557
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,822✔
558
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD4_LEN;
42,822✔
559
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,644✔
560

561
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
42,822✔
562
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD5_LEN;
42,822✔
563
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
85,644✔
564

565
  int32_t numOfCfg = taosArrayGetSize(pVars);
42,822✔
566
  code = blockDataEnsureCapacity(pBlock, numOfCfg);
42,822✔
567
  TSDB_CHECK_CODE(code, line, END);
42,822✔
568

569
  for (int32_t i = 0, c = 0; i < numOfCfg; ++i, c = 0) {
4,559,430✔
570
    SVariablesInfo* pInfo = taosArrayGet(pVars, i);
4,516,608✔
571
    TSDB_CHECK_NULL(pInfo, code, line, END, terrno);
4,516,608✔
572

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

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

587
    char scope[TSDB_CONFIG_SCOPE_LEN + VARSTR_HEADER_SIZE] = {0};
4,516,608✔
588
    STR_WITH_MAXSIZE_TO_VARSTR(scope, pInfo->scope, TSDB_CONFIG_SCOPE_LEN + VARSTR_HEADER_SIZE);
4,516,608✔
589
    pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
4,516,608✔
590
    TSDB_CHECK_NULL(pColInfo, code, line, END, terrno);
4,516,608✔
591
    code = colDataSetVal(pColInfo, i, scope, false);
4,516,608✔
592
    TSDB_CHECK_CODE(code, line, END);
4,516,608✔
593

594
    char category[TSDB_CONFIG_CATEGORY_LEN + VARSTR_HEADER_SIZE] = {0};
4,516,608✔
595
    STR_WITH_MAXSIZE_TO_VARSTR(category, pInfo->category, TSDB_CONFIG_CATEGORY_LEN + VARSTR_HEADER_SIZE);
4,516,608✔
596
    pColInfo = taosArrayGet(pBlock->pDataBlock, c++);
4,516,608✔
597
    TSDB_CHECK_NULL(pColInfo, code, line, END, terrno);
4,516,608✔
598
    code = colDataSetVal(pColInfo, i, category, false);
4,516,608✔
599
    TSDB_CHECK_CODE(code, line, END);
4,516,608✔
600

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

609
  pBlock->info.rows = numOfCfg;
42,822✔
610

611
  *block = pBlock;
42,822✔
612
  return code;
42,822✔
613

614
END:
×
615
  taosArrayDestroy(pBlock->pDataBlock);
×
616
  taosMemoryFree(pBlock);
×
617
  return code;
×
618
}
619

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

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

635
  (*pRsp)->useconds = 0;
42,822✔
636
  (*pRsp)->completed = 1;
42,822✔
637
  (*pRsp)->precision = 0;
42,822✔
638
  (*pRsp)->compressed = 0;
42,822✔
639

640
  (*pRsp)->numOfRows = htobe64((int64_t)pBlock->info.rows);
42,822✔
641
  (*pRsp)->numOfCols = htonl(SHOW_VARIABLES_RESULT_COLS);
42,822✔
642

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

653
    int32_t payloadLen = len + PAYLOAD_PREFIX_LEN;
42,058✔
654
    (*pRsp)->payloadLen = htonl(payloadLen);
42,058✔
655
    (*pRsp)->compLen = htonl(payloadLen);
42,058✔
656

657
    if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
42,058✔
658
      uError("buildShowVariablesRsp error, len:%d != rspSize - sizeof(SRetrieveTableRsp):%" PRIu64, len,
×
659
             (uint64_t)(rspSize - sizeof(SRetrieveTableRsp)));
660
      code = TSDB_CODE_TSC_INVALID_INPUT;
×
661
      goto _exit;
×
662
    }
663
  }
664

665
  blockDataDestroy(pBlock);
42,822✔
666
  pBlock = NULL;
42,822✔
667

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

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

696
    if (code != 0) {
42,822✔
697
      pRequest->body.resInfo.pRspMsg = NULL;
×
698
      taosMemoryFree(pRes);
×
699
    }
700
    tFreeSShowVariablesRsp(&rsp);
42,822✔
701
  }
702

703
  taosMemoryFree(pMsg->pData);
42,822✔
704
  taosMemoryFree(pMsg->pEpSet);
42,822✔
705

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

716
static int32_t buildCompactDbBlock(SCompactDbRsp* pRsp, SSDataBlock** block) {
47,356✔
717
  int32_t      code = 0;
47,356✔
718
  int32_t      line = 0;
47,356✔
719
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
47,356✔
720
  TSDB_CHECK_NULL(pBlock, code, line, END, terrno);
47,356✔
721
  pBlock->info.hasVarCol = true;
47,356✔
722

723
  pBlock->pDataBlock = taosArrayInit(COMPACT_DB_RESULT_COLS, sizeof(SColumnInfoData));
47,356✔
724
  TSDB_CHECK_NULL(pBlock->pDataBlock, code, line, END, terrno);
47,356✔
725
  SColumnInfoData infoData = {0};
47,356✔
726
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
47,356✔
727
  infoData.info.bytes = COMPACT_DB_RESULT_FIELD1_LEN;
47,356✔
728
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
94,712✔
729

730
  infoData.info.type = TSDB_DATA_TYPE_INT;
47,356✔
731
  infoData.info.bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;
47,356✔
732
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
94,712✔
733

734
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
47,356✔
735
  infoData.info.bytes = COMPACT_DB_RESULT_FIELD3_LEN;
47,356✔
736
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
94,712✔
737

738
  code = blockDataEnsureCapacity(pBlock, 1);
47,356✔
739
  TSDB_CHECK_CODE(code, line, END);
47,356✔
740

741
  SColumnInfoData* pResultCol = taosArrayGet(pBlock->pDataBlock, 0);
47,356✔
742
  TSDB_CHECK_NULL(pResultCol, code, line, END, terrno);
47,356✔
743
  SColumnInfoData* pIdCol = taosArrayGet(pBlock->pDataBlock, 1);
47,356✔
744
  TSDB_CHECK_NULL(pIdCol, code, line, END, terrno);
47,356✔
745
  SColumnInfoData* pReasonCol = taosArrayGet(pBlock->pDataBlock, 2);
47,356✔
746
  TSDB_CHECK_NULL(pReasonCol, code, line, END, terrno);
47,356✔
747

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

770
  *block = pBlock;
47,356✔
771

772
  return TSDB_CODE_SUCCESS;
47,356✔
773
END:
×
774
  taosMemoryFree(pBlock);
×
775
  taosArrayDestroy(pBlock->pDataBlock);
×
776
  return code;
×
777
}
778

779
static int32_t buildRetriveTableRspForCompactDb(SCompactDbRsp* pCompactDb, SRetrieveTableRsp** pRsp) {
47,356✔
780
  SSDataBlock* pBlock = NULL;
47,356✔
781
  int32_t      code = buildCompactDbBlock(pCompactDb, &pBlock);
47,356✔
782
  if (code) {
47,356✔
783
    return code;
×
784
  }
785

786
  size_t dataEncodeBufSize = blockGetEncodeSize(pBlock);
47,356✔
787
  size_t rspSize = sizeof(SRetrieveTableRsp) + dataEncodeBufSize + PAYLOAD_PREFIX_LEN;
47,356✔
788
  *pRsp = taosMemoryCalloc(1, rspSize);
47,356✔
789
  if (NULL == *pRsp) {
47,356✔
790
    code = terrno;
×
791
    goto _exit;
×
792
  }
793

794
  (*pRsp)->useconds = 0;
47,356✔
795
  (*pRsp)->completed = 1;
47,356✔
796
  (*pRsp)->precision = 0;
47,356✔
797
  (*pRsp)->compressed = 0;
47,356✔
798
  (*pRsp)->compLen = 0;
47,356✔
799
  (*pRsp)->payloadLen = 0;
47,356✔
800
  (*pRsp)->numOfRows = htobe64((int64_t)pBlock->info.rows);
47,356✔
801
  (*pRsp)->numOfCols = htonl(COMPACT_DB_RESULT_COLS);
47,356✔
802

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

811
  SET_PAYLOAD_LEN((*pRsp)->data, len, len);
47,356✔
812

813
  int32_t payloadLen = len + PAYLOAD_PREFIX_LEN;
47,356✔
814
  (*pRsp)->payloadLen = htonl(payloadLen);
47,356✔
815
  (*pRsp)->compLen = htonl(payloadLen);
47,356✔
816

817
  if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
47,356✔
818
    uError("buildRetriveTableRspForCompactDb error, len:%d != rspSize - sizeof(SRetrieveTableRsp):%" PRIu64, len,
×
819
           (uint64_t)(rspSize - sizeof(SRetrieveTableRsp)));
820
    code = TSDB_CODE_TSC_INVALID_INPUT;
×
821
    goto _exit;
×
822
  }
823

824
  return TSDB_CODE_SUCCESS;
47,356✔
825
_exit:
×
826
  if (*pRsp) {
×
827
    taosMemoryFree(*pRsp);
×
828
    *pRsp = NULL;
×
829
  }
830
  if (pBlock) {
×
831
    blockDataDestroy(pBlock);
×
832
    pBlock = NULL;
×
833
  }
834
  return code;
×
835
}
836

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

852
    if (code != 0) {
30,928✔
853
      pRequest->body.resInfo.pRspMsg = NULL;
×
854
      taosMemoryFree(pRes);
×
855
    }
856
  }
857

858
  taosMemoryFree(pMsg->pData);
33,538✔
859
  taosMemoryFree(pMsg->pEpSet);
33,538✔
860

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

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

878
  pBlock->pDataBlock = taosArrayInit(SCAN_DB_RESULT_COLS, sizeof(SColumnInfoData));
464✔
879
  TSDB_CHECK_NULL(pBlock->pDataBlock, code, line, END, terrno);
464✔
880
  SColumnInfoData infoData = {0};
464✔
881
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
464✔
882
  infoData.info.bytes = SCAN_DB_RESULT_FIELD1_LEN;
464✔
883
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
928✔
884

885
  infoData.info.type = TSDB_DATA_TYPE_INT;
464✔
886
  infoData.info.bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;
464✔
887
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
928✔
888

889
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
464✔
890
  infoData.info.bytes = SCAN_DB_RESULT_FIELD3_LEN;
464✔
891
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
928✔
892

893
  code = blockDataEnsureCapacity(pBlock, 1);
464✔
894
  TSDB_CHECK_CODE(code, line, END);
464✔
895

896
  SColumnInfoData* pResultCol = taosArrayGet(pBlock->pDataBlock, 0);
464✔
897
  TSDB_CHECK_NULL(pResultCol, code, line, END, terrno);
464✔
898
  SColumnInfoData* pIdCol = taosArrayGet(pBlock->pDataBlock, 1);
464✔
899
  TSDB_CHECK_NULL(pIdCol, code, line, END, terrno);
464✔
900
  SColumnInfoData* pReasonCol = taosArrayGet(pBlock->pDataBlock, 2);
464✔
901
  TSDB_CHECK_NULL(pReasonCol, code, line, END, terrno);
464✔
902

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

925
  *block = pBlock;
464✔
926

927
  return TSDB_CODE_SUCCESS;
464✔
928
END:
×
929
  taosMemoryFree(pBlock);
×
930
  taosArrayDestroy(pBlock->pDataBlock);
×
931
  return code;
×
932
}
933

934
static int32_t buildRetriveTableRspForScanDb(SScanDbRsp* pScanDb, SRetrieveTableRsp** pRsp) {
464✔
935
  SSDataBlock* pBlock = NULL;
464✔
936
  int32_t      code = buildScanDbBlock(pScanDb, &pBlock);
464✔
937
  if (code) {
464✔
938
    return code;
×
939
  }
940

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

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

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

966
  SET_PAYLOAD_LEN((*pRsp)->data, len, len);
464✔
967

968
  int32_t payloadLen = len + PAYLOAD_PREFIX_LEN;
464✔
969
  (*pRsp)->payloadLen = htonl(payloadLen);
464✔
970
  (*pRsp)->compLen = htonl(payloadLen);
464✔
971

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

979
  return TSDB_CODE_SUCCESS;
464✔
980
_exit:
×
981
  if (*pRsp) {
×
982
    taosMemoryFree(*pRsp);
×
983
    *pRsp = NULL;
×
984
  }
985
  if (pBlock) {
×
986
    blockDataDestroy(pBlock);
×
987
    pBlock = NULL;
×
988
  }
989
  return code;
×
990
}
991

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

1007
    if (code != 0) {
464✔
1008
      pRequest->body.resInfo.pRspMsg = NULL;
×
1009
      taosMemoryFree(pRes);
×
1010
    }
1011
  }
1012

1013
  taosMemoryFree(pMsg->pData);
694✔
1014
  taosMemoryFree(pMsg->pEpSet);
694✔
1015

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

1026
int32_t processTrimDbRsp(void* param, SDataBuf* pMsg, int32_t code) {
23,385✔
1027
  SRequestObj* pRequest = param;
23,385✔
1028
  if (code != TSDB_CODE_SUCCESS) {
23,385✔
1029
    setErrno(pRequest, code);
6,957✔
1030
  } else {
1031
    STrimDbRsp         rsp = {0};
16,428✔
1032
    SRetrieveTableRsp* pRes = NULL;
16,428✔
1033
    code = tDeserializeSCompactDbRsp(pMsg->pData, pMsg->len, (SCompactDbRsp*)&rsp);
16,428✔
1034
    if (TSDB_CODE_SUCCESS == code) {
16,428✔
1035
      code = buildRetriveTableRspForCompactDb(&rsp, &pRes);
16,428✔
1036
    }
1037
    if (TSDB_CODE_SUCCESS == code) {
16,428✔
1038
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
16,428✔
1039
    }
1040

1041
    if (code != 0) {
16,428✔
1042
      pRequest->body.resInfo.pRspMsg = NULL;
×
1043
      taosMemoryFree(pRes);
×
1044
    }
1045
  }
1046

1047
  taosMemoryFree(pMsg->pData);
23,385✔
1048
  taosMemoryFree(pMsg->pEpSet);
23,385✔
1049

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

1060
__async_send_cb_fn_t getMsgRspHandle(int32_t msgType) {
19,594,159✔
1061
  switch (msgType) {
19,594,159✔
1062
    case TDMT_MND_CONNECT:
2,702,475✔
1063
      return processConnectRsp;
2,702,475✔
1064
    case TDMT_MND_CREATE_DB:
1,315,178✔
1065
      return processCreateDbRsp;
1,315,178✔
1066
    case TDMT_MND_USE_DB:
2,464,132✔
1067
      return processUseDbRsp;
2,464,132✔
1068
    case TDMT_MND_CREATE_STB:
1,902,458✔
1069
      return processCreateSTableRsp;
1,902,458✔
1070
    case TDMT_MND_DROP_DB:
1,188,760✔
1071
      return processDropDbRsp;
1,188,760✔
1072
    case TDMT_MND_ALTER_STB:
6,079,855✔
1073
      return processAlterStbRsp;
6,079,855✔
1074
    case TDMT_MND_SHOW_VARIABLES:
42,822✔
1075
      return processShowVariablesRsp;
42,822✔
1076
    case TDMT_MND_COMPACT_DB:
33,538✔
1077
      return processCompactDbRsp;
33,538✔
1078
    case TDMT_MND_TRIM_DB:
23,385✔
1079
      return processTrimDbRsp;
23,385✔
1080
    case TDMT_MND_SCAN_DB:
694✔
1081
      return processScanDbRsp;
694✔
1082
    default:
3,840,862✔
1083
      return genericRspCallback;
3,840,862✔
1084
  }
1085
}
1086

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