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

taosdata / TDengine / #4881

14 Dec 2025 03:48AM UTC coverage: 60.617% (+0.5%) from 60.092%
#4881

push

travis-ci

web-flow
test: update coverage workflow time (#33918)

156854 of 258761 relevant lines covered (60.62%)

75258957.81 hits per line

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

78.21
/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) {
2,488,041✔
34
  pRequest->code = code;
2,488,041✔
35
  terrno = code;
2,488,041✔
36
}
2,488,041✔
37

38
int32_t genericRspCallback(void* param, SDataBuf* pMsg, int32_t code) {
1,734,933✔
39
  SRequestObj* pRequest = param;
1,734,933✔
40
  setErrno(pRequest, code);
1,734,933✔
41

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

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

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

66
  if (code != TSDB_CODE_SUCCESS) {
1,403,400✔
67
    goto End;
5,914✔
68
  }
69

70
  STscObj* pTscObj = pRequest->pTscObj;
1,397,486✔
71

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

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

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

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

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

106
  int    updateEpSet = 1;
1,396,957✔
107
  SEpSet srcEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
1,396,957✔
108
  if (connectRsp.dnodeNum == 1) {
1,396,569✔
109
    SEpSet dstEpSet = connectRsp.epSet;
1,357,581✔
110
    if (srcEpSet.numOfEps == 1) {
1,357,581✔
111
      if (rpcSetDefaultAddr(pTscObj->pAppInfo->pTransporter, srcEpSet.eps[srcEpSet.inUse].fqdn,
827,262✔
112
                            dstEpSet.eps[dstEpSet.inUse].fqdn) != 0) {
827,286✔
113
        tscError("failed to set default addr for rpc");
×
114
      }
115
      updateEpSet = 0;
827,801✔
116
    }
117
  }
118
  if (updateEpSet == 1 && !isEpsetEqual(&srcEpSet, &connectRsp.epSet)) {
1,396,776✔
119
    SEpSet corEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
534,404✔
120

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

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

135
  pTscObj->sysInfo = connectRsp.sysInfo;
1,397,201✔
136
  pTscObj->connId = connectRsp.connId;
1,397,304✔
137
  pTscObj->acctId = connectRsp.acctId;
1,397,295✔
138
  tstrncpy(pTscObj->sVer, connectRsp.sVer, tListLen(pTscObj->sVer));
1,397,440✔
139
  tstrncpy(pTscObj->sDetailVer, connectRsp.sDetailVer, tListLen(pTscObj->sDetailVer));
1,397,295✔
140

141
  // update the appInstInfo
142
  pTscObj->pAppInfo->clusterId = connectRsp.clusterId;
1,397,155✔
143
  pTscObj->pAppInfo->serverCfg.monitorParas = connectRsp.monitorParas;
1,397,043✔
144
  pTscObj->pAppInfo->serverCfg.enableAuditDelete = connectRsp.enableAuditDelete;
1,397,201✔
145
  tscDebug("monitor paras from connect rsp, clusterId:0x%" PRIx64 ", threshold:%d scope:%d",
1,397,012✔
146
           connectRsp.clusterId, connectRsp.monitorParas.tsSlowLogThreshold, connectRsp.monitorParas.tsSlowLogScope);
147
  lastClusterId = connectRsp.clusterId;
1,397,012✔
148

149
  pTscObj->connType = connectRsp.connType;
1,397,012✔
150
  pTscObj->passInfo.ver = connectRsp.passVer;
1,396,422✔
151
  pTscObj->authVer = connectRsp.authVer;
1,396,864✔
152
  pTscObj->whiteListInfo.ver = connectRsp.whiteListVer;
1,396,584✔
153

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

170
  (void)taosThreadMutexLock(&clientHbMgr.lock);
1,397,335✔
171
  SAppHbMgr* pAppHbMgr = taosArrayGetP(clientHbMgr.appHbMgrs, pTscObj->appHbMgrIdx);
1,397,486✔
172
  if (pAppHbMgr) {
1,397,486✔
173
    if (hbRegisterConn(pAppHbMgr, pTscObj->id, connectRsp.clusterId, connectRsp.connType) != 0) {
1,397,486✔
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);
1,397,486✔
182

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

186
End:
1,402,736✔
187
  if (code != 0) {
1,403,400✔
188
    setErrno(pRequest, code);
5,914✔
189
  }
190
  if (tsem_post(&pRequest->body.rspSem) != 0) {
1,403,400✔
191
    tscError("failed to post semaphore");
×
192
  }
193

194
  if (pRequest) {
1,403,400✔
195
    (void)releaseRequest(pRequest->self);
1,403,400✔
196
  }
197

198
EXIT:
1,383,196✔
199
  taosMemoryFree(param);
1,403,400✔
200
  taosMemoryFree(pMsg->pEpSet);
1,403,400✔
201
  taosMemoryFree(pMsg->pData);
1,403,400✔
202
  return code;
1,403,400✔
203
}
204

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

214
  pMsgSendInfo->msgInfo = pRequest->body.requestMsg;
8,974,937✔
215
  pMsgSendInfo->fp = getMsgRspHandle(pRequest->type);
8,975,090✔
216
  return pMsgSendInfo;
8,974,937✔
217
}
218

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

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

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

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

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

284
  if (code != TSDB_CODE_SUCCESS) {
1,017,136✔
285
    taosMemoryFree(pMsg->pData);
2,609✔
286
    taosMemoryFree(pMsg->pEpSet);
2,609✔
287
    setErrno(pRequest, code);
2,609✔
288

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

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

298
    return code;
2,609✔
299
  }
300

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

306
  if (strlen(usedbRsp.db) == 0) {
1,014,505✔
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);
1,014,505✔
318
  for (int32_t i = 0; i < usedbRsp.vgNum; ++i) {
2,324,688✔
319
    SVgroupInfo* pInfo = taosArrayGet(usedbRsp.pVgroupInfos, i);
1,310,183✔
320
    if (pInfo == NULL) {
1,310,183✔
321
      continue;
×
322
    }
323
    tscTrace("vgId:%d, numOfEps:%d inUse:%d ", pInfo->vgId, pInfo->epSet.numOfEps, pInfo->epSet.inUse);
1,310,183✔
324
    for (int32_t j = 0; j < pInfo->epSet.numOfEps; ++j) {
2,688,491✔
325
      tscTrace("vgId:%d, index:%d epset:%s:%u", pInfo->vgId, j, pInfo->epSet.eps[j].fqdn, pInfo->epSet.eps[j].port);
1,378,308✔
326
    }
327
  }
328

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

334
  SUseDbOutput output = {0};
1,014,505✔
335
  code = queryBuildUseDbOutput(&output, &usedbRsp);
1,014,505✔
336
  if (code != 0) {
1,014,527✔
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) {
1,014,527✔
342
    struct SCatalog* pCatalog = NULL;
604,128✔
343

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

357
  taosMemoryFreeClear(output.dbVgroup);
1,014,505✔
358
  tFreeSUsedbRsp(&usedbRsp);
1,014,505✔
359

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

365
  setConnectionDB(pRequest->pTscObj, db);
1,014,344✔
366

367
  taosMemoryFree(pMsg->pData);
1,014,527✔
368
  taosMemoryFree(pMsg->pEpSet);
1,014,527✔
369

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

380
int32_t processCreateSTableRsp(void* param, SDataBuf* pMsg, int32_t code) {
867,540✔
381
  if (pMsg == NULL) {
867,540✔
382
    tscError("processCreateSTableRsp: invalid input param, pMsg is NULL");
×
383
    return TSDB_CODE_TSC_INVALID_INPUT;
×
384
  }
385
  if (param == NULL) {
867,540✔
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;
867,540✔
393

394
  if (code != TSDB_CODE_SUCCESS) {
867,540✔
395
    setErrno(pRequest, code);
6,136✔
396
  } else {
397
    SMCreateStbRsp createRsp = {0};
861,404✔
398
    SDecoder       coder = {0};
861,404✔
399
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
861,404✔
400
    if (pMsg->len > 0) {
861,404✔
401
      code = tDecodeSMCreateStbRsp(&coder, &createRsp);  // pMsg->len == 0
857,142✔
402
      if (code != TSDB_CODE_SUCCESS) {
857,142✔
403
        setErrno(pRequest, code);
×
404
      }
405
    }
406
    tDecoderClear(&coder);
861,404✔
407

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

412
  taosMemoryFree(pMsg->pEpSet);
867,540✔
413
  taosMemoryFree(pMsg->pData);
867,540✔
414

415
  if (pRequest->body.queryFp != NULL) {
867,540✔
416
    SExecResult* pRes = &pRequest->body.resInfo.execRes;
659,008✔
417

418
    if (code == TSDB_CODE_SUCCESS) {
659,008✔
419
      SCatalog* pCatalog = NULL;
656,048✔
420
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
656,048✔
421
      if (pRes->res != NULL) {
656,048✔
422
        ret = handleCreateTbExecRes(pRes->res, pCatalog);
652,104✔
423
      }
424

425
      if (ret != TSDB_CODE_SUCCESS) {
656,048✔
426
        code = ret;
×
427
      }
428
    }
429

430
    doRequestCallback(pRequest, code);
659,008✔
431
  } else {
432
    if (tsem_post(&pRequest->body.rspSem) != 0) {
208,532✔
433
      tscError("failed to post semaphore");
×
434
    }
435
  }
436
  return code;
867,540✔
437
}
438

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

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

472
  taosMemoryFree(pMsg->pData);
675,054✔
473
  taosMemoryFree(pMsg->pEpSet);
675,054✔
474

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

485
int32_t processAlterStbRsp(void* param, SDataBuf* pMsg, int32_t code) {
3,923,139✔
486
  SRequestObj* pRequest = param;
3,923,139✔
487
  if (code != TSDB_CODE_SUCCESS) {
3,923,139✔
488
    setErrno(pRequest, code);
710,876✔
489
  } else {
490
    SMAlterStbRsp alterRsp = {0};
3,212,263✔
491
    SDecoder      coder = {0};
3,212,263✔
492
    tDecoderInit(&coder, pMsg->pData, pMsg->len);
3,212,263✔
493
    if (pMsg->len > 0) {
3,212,263✔
494
      code = tDecodeSMAlterStbRsp(&coder, &alterRsp);  // pMsg->len == 0
3,197,441✔
495
      if (code != TSDB_CODE_SUCCESS) {
3,197,441✔
496
        setErrno(pRequest, code);
×
497
      }
498
    }
499
    tDecoderClear(&coder);
3,212,263✔
500

501
    pRequest->body.resInfo.execRes.msgType = TDMT_MND_ALTER_STB;
3,212,263✔
502
    pRequest->body.resInfo.execRes.res = alterRsp.pMeta;
3,212,263✔
503
  }
504

505
  taosMemoryFree(pMsg->pData);
3,923,139✔
506
  taosMemoryFree(pMsg->pEpSet);
3,923,139✔
507

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

511
    if (code == TSDB_CODE_SUCCESS) {
3,923,139✔
512
      SCatalog* pCatalog = NULL;
3,212,263✔
513
      int32_t   ret = catalogGetHandle(pRequest->pTscObj->pAppInfo->clusterId, &pCatalog);
3,212,263✔
514
      if (pRes->res != NULL) {
3,212,263✔
515
        ret = handleAlterTbExecRes(pRes->res, pCatalog);
3,197,441✔
516
      }
517

518
      if (ret != TSDB_CODE_SUCCESS) {
3,212,263✔
519
        code = ret;
×
520
      }
521
    }
522

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

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

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

546
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
24,474✔
547
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD2_LEN;
24,474✔
548
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
48,948✔
549

550
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
24,474✔
551
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD3_LEN;
24,474✔
552
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
48,948✔
553

554
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
24,474✔
555
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD4_LEN;
24,474✔
556
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
48,948✔
557

558
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
24,474✔
559
  infoData.info.bytes = SHOW_VARIABLES_RESULT_FIELD5_LEN;
24,474✔
560
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
48,948✔
561

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

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

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

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

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

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

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

606
  pBlock->info.rows = numOfCfg;
24,474✔
607

608
  *block = pBlock;
24,474✔
609
  return code;
24,474✔
610

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

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

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

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

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

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

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

654
    if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
24,110✔
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);
24,474✔
663
  pBlock = NULL;
24,474✔
664

665
  return TSDB_CODE_SUCCESS;
24,474✔
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) {
24,474✔
679
  SRequestObj* pRequest = param;
24,474✔
680
  if (code != TSDB_CODE_SUCCESS) {
24,474✔
681
    setErrno(pRequest, code);
×
682
  } else {
683
    SShowVariablesRsp  rsp = {0};
24,474✔
684
    SRetrieveTableRsp* pRes = NULL;
24,474✔
685
    code = tDeserializeSShowVariablesRsp(pMsg->pData, pMsg->len, &rsp);
24,474✔
686
    if (TSDB_CODE_SUCCESS == code) {
24,474✔
687
      code = buildShowVariablesRsp(rsp.variables, &pRes);
24,474✔
688
    }
689
    if (TSDB_CODE_SUCCESS == code) {
24,474✔
690
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
24,474✔
691
    }
692

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

700
  taosMemoryFree(pMsg->pData);
24,474✔
701
  taosMemoryFree(pMsg->pEpSet);
24,474✔
702

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

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

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

727
  infoData.info.type = TSDB_DATA_TYPE_INT;
30,507✔
728
  infoData.info.bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;
30,507✔
729
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
61,014✔
730

731
  infoData.info.type = TSDB_DATA_TYPE_VARCHAR;
30,507✔
732
  infoData.info.bytes = COMPACT_DB_RESULT_FIELD3_LEN;
30,507✔
733
  TSDB_CHECK_NULL(taosArrayPush(pBlock->pDataBlock, &infoData), code, line, END, terrno);
61,014✔
734

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

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

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

767
  *block = pBlock;
30,507✔
768

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

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

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

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

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

808
  SET_PAYLOAD_LEN((*pRsp)->data, len, len);
30,507✔
809

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

814
  if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
30,507✔
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;
30,507✔
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) {
20,632✔
835
  SRequestObj* pRequest = param;
20,632✔
836
  if (code != TSDB_CODE_SUCCESS) {
20,632✔
837
    setErrno(pRequest, code);
1,078✔
838
  } else {
839
    SCompactDbRsp      rsp = {0};
19,554✔
840
    SRetrieveTableRsp* pRes = NULL;
19,554✔
841
    code = tDeserializeSCompactDbRsp(pMsg->pData, pMsg->len, &rsp);
19,554✔
842
    if (TSDB_CODE_SUCCESS == code) {
19,554✔
843
      code = buildRetriveTableRspForCompactDb(&rsp, &pRes);
19,554✔
844
    }
845
    if (TSDB_CODE_SUCCESS == code) {
19,554✔
846
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
19,554✔
847
    }
848

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

855
  taosMemoryFree(pMsg->pData);
20,632✔
856
  taosMemoryFree(pMsg->pEpSet);
20,632✔
857

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

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

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

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

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

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

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

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

922
  *block = pBlock;
52✔
923

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

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

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

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

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

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

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

969
  if (payloadLen != rspSize - sizeof(SRetrieveTableRsp)) {
52✔
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;
52✔
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) {
76✔
990
  SRequestObj* pRequest = param;
76✔
991
  if (code != TSDB_CODE_SUCCESS) {
76✔
992
    setErrno(pRequest, code);
24✔
993
  } else {
994
    SScanDbRsp         rsp = {0};
52✔
995
    SRetrieveTableRsp* pRes = NULL;
52✔
996
    code = tDeserializeSScanDbRsp(pMsg->pData, pMsg->len, &rsp);
52✔
997
    if (TSDB_CODE_SUCCESS == code) {
52✔
998
      code = buildRetriveTableRspForScanDb(&rsp, &pRes);
52✔
999
    }
1000
    if (TSDB_CODE_SUCCESS == code) {
52✔
1001
      code = setQueryResultFromRsp(&pRequest->body.resInfo, pRes, false, pRequest->stmtBindVersion > 0);
52✔
1002
    }
1003

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

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

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

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

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

1044
  taosMemoryFree(pMsg->pData);
15,813✔
1045
  taosMemoryFree(pMsg->pEpSet);
15,813✔
1046

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

1057
__async_send_cb_fn_t getMsgRspHandle(int32_t msgType) {
10,377,581✔
1058
  switch (msgType) {
10,377,581✔
1059
    case TDMT_MND_CONNECT:
1,403,400✔
1060
      return processConnectRsp;
1,403,400✔
1061
    case TDMT_MND_CREATE_DB:
696,675✔
1062
      return processCreateDbRsp;
696,675✔
1063
    case TDMT_MND_USE_DB:
1,016,219✔
1064
      return processUseDbRsp;
1,016,219✔
1065
    case TDMT_MND_CREATE_STB:
867,540✔
1066
      return processCreateSTableRsp;
867,540✔
1067
    case TDMT_MND_DROP_DB:
675,054✔
1068
      return processDropDbRsp;
675,054✔
1069
    case TDMT_MND_ALTER_STB:
3,923,139✔
1070
      return processAlterStbRsp;
3,923,139✔
1071
    case TDMT_MND_SHOW_VARIABLES:
24,474✔
1072
      return processShowVariablesRsp;
24,474✔
1073
    case TDMT_MND_COMPACT_DB:
20,632✔
1074
      return processCompactDbRsp;
20,632✔
1075
    case TDMT_MND_TRIM_DB:
15,813✔
1076
      return processTrimDbRsp;
15,813✔
1077
    case TDMT_MND_SCAN_DB:
76✔
1078
      return processScanDbRsp;
76✔
1079
    default:
1,734,559✔
1080
      return genericRspCallback;
1,734,559✔
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