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

taosdata / TDengine / #4999

22 Mar 2026 10:21AM UTC coverage: 72.31% (-0.03%) from 72.335%
#4999

push

travis-ci

web-flow
feat(subq/some): some/any/exists for stream subq (#34860)

50 of 68 new or added lines in 1 file covered. (73.53%)

701 existing lines in 132 files now uncovered.

253472 of 350536 relevant lines covered (72.31%)

131775367.32 hits per line

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

76.66
/source/client/src/clientStmt2.c
1
#include "clientInt.h"
2
#include "clientLog.h"
3
#include "tdef.h"
4
#include "tglobal.h"
5

6
#include "clientStmt.h"
7
#include "clientStmt2.h"
8

9
char* gStmt2StatusStr[] = {"unknown",     "init", "prepare", "settbname", "settags",
10
                           "fetchFields", "bind", "bindCol", "addBatch",  "exec"};
11

12

13
static FORCE_INLINE int32_t stmtAllocQNodeFromBuf(STableBufInfo* pTblBuf, void** pBuf) {
14
  if (pTblBuf->buffOffset < pTblBuf->buffSize) {
1,636,158✔
15
    *pBuf = (char*)pTblBuf->pCurBuff + pTblBuf->buffOffset;
1,636,273✔
16
    pTblBuf->buffOffset += pTblBuf->buffUnit;
1,636,024✔
17
  } else if (pTblBuf->buffIdx < taosArrayGetSize(pTblBuf->pBufList)) {
×
18
    pTblBuf->pCurBuff = taosArrayGetP(pTblBuf->pBufList, pTblBuf->buffIdx++);
×
19
    if (NULL == pTblBuf->pCurBuff) {
×
20
      return TAOS_GET_TERRNO(TSDB_CODE_OUT_OF_MEMORY);
×
21
    }
22
    *pBuf = pTblBuf->pCurBuff;
×
23
    pTblBuf->buffOffset = pTblBuf->buffUnit;
×
24
  } else {
25
    void* buff = taosMemoryMalloc(pTblBuf->buffSize);
×
26
    if (NULL == buff) {
×
27
      return terrno;
×
28
    }
29

30
    if (taosArrayPush(pTblBuf->pBufList, &buff) == NULL) {
×
31
      return terrno;
×
32
    }
33

34
    pTblBuf->buffIdx++;
×
35
    pTblBuf->pCurBuff = buff;
×
36
    *pBuf = buff;
×
37
    pTblBuf->buffOffset = pTblBuf->buffUnit;
×
38
  }
39

40
  return TSDB_CODE_SUCCESS;
1,635,459✔
41
}
42

43
static bool stmtDequeue(STscStmt2* pStmt, SStmtQNode** param) {
1,636,244✔
44
  int i = 0;
1,636,244✔
45
  while (0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) {
7,707,037✔
46
    if (pStmt->queue.stopQueue) {
6,196,203✔
47
      return false;
124,630✔
48
    }
49
    if (i < 10) {
6,071,573✔
50
      taosUsleep(1);
5,676,614✔
51
      i++;
5,675,930✔
52
    } else {
53
      (void)taosThreadMutexLock(&pStmt->queue.mutex);
394,959✔
54
      if (pStmt->queue.stopQueue) {
395,238✔
55
        (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
×
56
        return false;
×
57
      }
58
      if (0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) {
395,238✔
59
        (void)taosThreadCondWait(&pStmt->queue.waitCond, &pStmt->queue.mutex);
395,138✔
60
      }
61
      (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
394,851✔
62
    }
63
  }
64

65
  if (pStmt->queue.stopQueue && 0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) {
1,511,023✔
66
    return false;
×
67
  }
68

69
  (void)taosThreadMutexLock(&pStmt->queue.mutex);
1,511,023✔
70
  if (pStmt->queue.head == pStmt->queue.tail) {
1,511,577✔
71
    pStmt->queue.qRemainNum = 0;
×
72
    (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
×
73
    STMT2_ELOG_E("interlace queue is empty, cannot dequeue");
×
74
    return false;
×
75
  }
76

77
  SStmtQNode* node = pStmt->queue.head->next;
1,511,577✔
78
  pStmt->queue.head->next = node->next;
1,511,499✔
79
  if (pStmt->queue.tail == node) {
1,511,536✔
80
    pStmt->queue.tail = pStmt->queue.head;
830,726✔
81
  }
82
  node->next = NULL;
1,511,536✔
83
  *param = node;
1,511,536✔
84

85
  (void)atomic_sub_fetch_64((int64_t*)&pStmt->queue.qRemainNum, 1);
1,511,499✔
86
  (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
1,511,598✔
87

88
  STMT2_TLOG("dequeue success, node:%p, remainNum:%" PRId64, node, pStmt->queue.qRemainNum);
1,511,516✔
89

90
  return true;
1,511,480✔
91
}
92

93
static void stmtEnqueue(STscStmt2* pStmt, SStmtQNode* param) {
1,510,948✔
94
  if (param == NULL) {
1,510,948✔
95
    STMT2_ELOG_E("enqueue param is NULL");
×
96
    return;
×
97
  }
98

99
  param->next = NULL;
1,510,948✔
100

101
  (void)taosThreadMutexLock(&pStmt->queue.mutex);
1,511,104✔
102

103
  pStmt->queue.tail->next = param;
1,511,498✔
104
  pStmt->queue.tail = param;
1,511,387✔
105
  pStmt->stat.bindDataNum++;
1,511,387✔
106

107
  (void)atomic_add_fetch_64(&pStmt->queue.qRemainNum, 1);
1,511,387✔
108
  (void)taosThreadCondSignal(&(pStmt->queue.waitCond));
1,511,466✔
109

110
  (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
1,511,219✔
111

112
  STMT2_TLOG("enqueue param:%p, remainNum:%" PRId64 ", restoreTbCols:%d", param, pStmt->queue.qRemainNum,
1,511,383✔
113
             param->restoreTbCols);
114
}
115

116
static int32_t stmtCreateRequest(STscStmt2* pStmt) {
45,683,468✔
117
  int32_t code = 0;
45,683,468✔
118

119
  if (pStmt->exec.pRequest == NULL) {
45,683,468✔
120
    code = buildRequest(pStmt->taos->id, pStmt->sql.sqlStr, pStmt->sql.sqlLen, NULL, false, &pStmt->exec.pRequest,
150,837✔
121
                        pStmt->reqid);
122
    if (pStmt->reqid != 0) {
150,590✔
123
      pStmt->reqid++;
8✔
124
    }
125
    pStmt->exec.pRequest->type = RES_TYPE__QUERY;
150,806✔
126
    if (pStmt->db != NULL) {
150,590✔
127
      taosMemoryFreeClear(pStmt->exec.pRequest->pDb);
140,502✔
128
      pStmt->exec.pRequest->pDb = taosStrdup(pStmt->db);
140,474✔
129
    }
130
    if (TSDB_CODE_SUCCESS == code) {
150,590✔
131
      pStmt->exec.pRequest->syncQuery = true;
150,621✔
132
      pStmt->exec.pRequest->stmtBindVersion = 2;
150,806✔
133
    }
134
    STMT2_DLOG("create request:0x%" PRIx64 ", QID:0x%" PRIx64, pStmt->exec.pRequest->self,
150,562✔
135
               pStmt->exec.pRequest->requestId);
136
  }
137

138
  return code;
45,392,283✔
139
}
140

141
static int32_t stmtSwitchStatus(STscStmt2* pStmt, STMT_STATUS newStatus) {
88,328,603✔
142
  int32_t code = 0;
88,328,603✔
143

144
  if (newStatus >= STMT_INIT && newStatus < STMT_MAX) {
88,328,603✔
145
    STMT2_LOG_SEQ(newStatus);
88,969,385✔
146
  }
147

148
  if (pStmt->errCode && newStatus != STMT_PREPARE) {
89,111,468✔
149
    STMT2_ELOG("stmt already failed with err:%s, please use stmt prepare", tstrerror(pStmt->errCode));
×
150
    return pStmt->errCode;
×
151
  }
152

153
  switch (newStatus) {
88,869,759✔
154
    case STMT_PREPARE:
141,535✔
155
      pStmt->errCode = 0;
141,535✔
156
      break;
136,111✔
157
    case STMT_SETTBNAME:
1,045,991✔
158
      if (STMT_STATUS_EQ(INIT)) {
1,045,991✔
159
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
160
      }
161
      if (!pStmt->sql.stbInterlaceMode && (STMT_STATUS_EQ(BIND) || STMT_STATUS_EQ(BIND_COL))) {
1,046,262✔
162
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
163
      }
164
      break;
1,046,040✔
165
    case STMT_SETTAGS:
635,156✔
166
      if (STMT_STATUS_EQ(INIT)) {
635,156✔
167
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
168
      }
169
      break;
635,156✔
170
    case STMT_FETCH_FIELDS:
17,479✔
171
      if (STMT_STATUS_EQ(INIT)) {
17,479✔
172
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
173
      }
174
      break;
17,479✔
175
    case STMT_BIND:
43,522,821✔
176
      if (STMT_STATUS_EQ(INIT) || STMT_STATUS_EQ(BIND_COL)) {
43,522,821✔
177
        code = TSDB_CODE_TSC_STMT_API_ERROR;
194✔
178
      }
179
      /*
180
            if ((pStmt->sql.type == STMT_TYPE_MULTI_INSERT) && ()) {
181
              code = TSDB_CODE_TSC_STMT_API_ERROR;
182
            }
183
      */
184
      break;
43,872,321✔
185
    case STMT_BIND_COL:
×
186
      if (STMT_STATUS_EQ(INIT) || STMT_STATUS_EQ(BIND)) {
×
187
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
188
      }
189
      break;
×
190
    case STMT_ADD_BATCH:
42,887,889✔
191
      if (STMT_STATUS_NE(BIND) && STMT_STATUS_NE(BIND_COL) && STMT_STATUS_NE(FETCH_FIELDS)) {
42,887,889✔
192
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
193
      }
194
      break;
42,625,870✔
195
    case STMT_EXECUTE:
618,888✔
196
      if (STMT_TYPE_QUERY == pStmt->sql.type) {
618,888✔
197
        if (STMT_STATUS_NE(ADD_BATCH) && STMT_STATUS_NE(FETCH_FIELDS) && STMT_STATUS_NE(BIND) &&
7,380✔
198
            STMT_STATUS_NE(BIND_COL)) {
×
199
          code = TSDB_CODE_TSC_STMT_API_ERROR;
×
200
        }
201
      } else {
202
        if (STMT_STATUS_NE(ADD_BATCH) && STMT_STATUS_NE(FETCH_FIELDS)) {
611,224✔
203
          code = TSDB_CODE_TSC_STMT_API_ERROR;
×
204
        }
205
      }
206
      break;
618,567✔
207
    default:
×
208
      code = TSDB_CODE_APP_ERROR;
×
209
      break;
×
210
  }
211

212
  STMT_ERR_RET(code);
88,645,494✔
213

214
  pStmt->sql.status = newStatus;
88,645,300✔
215

216
  return TSDB_CODE_SUCCESS;
89,390,338✔
217
}
218

219
static int32_t stmtGetTbName(TAOS_STMT2* stmt, char** tbName) {
27,532✔
220
  STscStmt2* pStmt = (STscStmt2*)stmt;
27,532✔
221

222
  pStmt->sql.type = STMT_TYPE_MULTI_INSERT;
27,532✔
223

224
  if ('\0' == pStmt->bInfo.tbName[0]) {
27,532✔
225
    tscWarn("no table name set, OK if it is a stmt get fields");
8,555✔
226
    STMT_ERR_RET(TSDB_CODE_TSC_STMT_TBNAME_ERROR);
8,555✔
227
  }
228

229
  *tbName = pStmt->bInfo.tbName;
18,977✔
230

231
  return TSDB_CODE_SUCCESS;
18,977✔
232
}
233

234
static int32_t stmtUpdateBindInfo(TAOS_STMT2* stmt, STableMeta* pTableMeta, void* tags, SSHashObj** cols, SName* tbName,
133,306✔
235
                                  const char* sTableName, bool autoCreateTbl, int8_t tbNameFlag) {
236
  STscStmt2* pStmt = (STscStmt2*)stmt;
133,306✔
237
  char       tbFName[TSDB_TABLE_FNAME_LEN];
96,814✔
238
  int32_t    code = tNameExtractFullName(tbName, tbFName);
133,337✔
239
  if (code != 0) {
133,584✔
240
    return code;
×
241
  }
242

243
  if ((tags != NULL && ((SBoundColInfo*)tags)->numOfCols == 0) || !autoCreateTbl) {
133,584✔
244
    pStmt->sql.autoCreateTbl = false;
112,152✔
245
  }
246

247
  (void)memcpy(&pStmt->bInfo.sname, tbName, sizeof(*tbName));
133,553✔
248
  tstrncpy(pStmt->bInfo.tbFName, tbFName, sizeof(pStmt->bInfo.tbFName));
133,584✔
249
  pStmt->bInfo.tbFName[sizeof(pStmt->bInfo.tbFName) - 1] = 0;
133,152✔
250

251
  pStmt->bInfo.tbUid = autoCreateTbl ? 0 : pTableMeta->uid;
133,090✔
252
  pStmt->bInfo.tbSuid = pTableMeta->suid;
133,275✔
253
  pStmt->bInfo.tbVgId = pTableMeta->vgId;
132,843✔
254
  pStmt->bInfo.tbType = pTableMeta->tableType;
133,053✔
255

256
  if (!pStmt->bInfo.tagsCached) {
133,121✔
257
    qDestroyBoundColInfo(pStmt->bInfo.boundTags);
132,105✔
258
    taosMemoryFreeClear(pStmt->bInfo.boundTags);
132,383✔
259
  }
260

261
  // transfer ownership of cols to stmt
262
  if (cols) {
132,973✔
263
    pStmt->bInfo.fixedValueCols = *cols;
133,306✔
264
    *cols = NULL;
133,275✔
265
  }
266

267
  pStmt->bInfo.boundTags = tags;
132,041✔
268
  pStmt->bInfo.tagsCached = false;
132,479✔
269
  pStmt->bInfo.tbNameFlag = tbNameFlag;
132,442✔
270
  tstrncpy(pStmt->bInfo.stbFName, sTableName, sizeof(pStmt->bInfo.stbFName));
132,726✔
271

272
  if (pTableMeta->tableType != TSDB_CHILD_TABLE && pTableMeta->tableType != TSDB_SUPER_TABLE) {
132,973✔
273
    pStmt->sql.stbInterlaceMode = false;
4,186✔
274
  }
275

276
  return TSDB_CODE_SUCCESS;
133,155✔
277
}
278

279
static int32_t stmtUpdateExecInfo(TAOS_STMT2* stmt, SHashObj* pVgHash, SHashObj* pBlockHash) {
132,979✔
280
  STscStmt2* pStmt = (STscStmt2*)stmt;
132,979✔
281

282
  pStmt->sql.pVgHash = pVgHash;
132,979✔
283
  pStmt->exec.pBlockHash = pBlockHash;
133,627✔
284

285
  return TSDB_CODE_SUCCESS;
133,155✔
286
}
287

288
static int32_t stmtUpdateInfo(TAOS_STMT2* stmt, STableMeta* pTableMeta, void* tags, SSHashObj** cols, SName* tbName,
133,306✔
289
                              bool autoCreateTbl, SHashObj* pVgHash, SHashObj* pBlockHash, const char* sTableName,
290
                              uint8_t tbNameFlag) {
291
  STscStmt2* pStmt = (STscStmt2*)stmt;
133,306✔
292

293
  STMT_ERR_RET(stmtUpdateBindInfo(stmt, pTableMeta, tags, cols, tbName, sTableName, autoCreateTbl, tbNameFlag));
133,306✔
294
  STMT_ERR_RET(stmtUpdateExecInfo(stmt, pVgHash, pBlockHash));
132,976✔
295

296
  pStmt->sql.autoCreateTbl = autoCreateTbl;
132,544✔
297

298
  return TSDB_CODE_SUCCESS;
132,763✔
299
}
300

301
static int32_t stmtGetExecInfo(TAOS_STMT2* stmt, SHashObj** pVgHash, SHashObj** pBlockHash) {
3,511✔
302
  STscStmt2* pStmt = (STscStmt2*)stmt;
3,511✔
303

304
  *pVgHash = pStmt->sql.pVgHash;
3,511✔
305
  pStmt->sql.pVgHash = NULL;
3,511✔
306

307
  *pBlockHash = pStmt->exec.pBlockHash;
3,511✔
308
  pStmt->exec.pBlockHash = NULL;
3,511✔
309

310
  return TSDB_CODE_SUCCESS;
3,511✔
311
}
312

313
static int32_t stmtParseSql(STscStmt2* pStmt) {
142,792✔
314
  pStmt->exec.pCurrBlock = NULL;
142,792✔
315

316
  SStmtCallback stmtCb = {
142,926✔
317
      .pStmt = pStmt,
318
      .getTbNameFn = stmtGetTbName,
319
      .setInfoFn = stmtUpdateInfo,
320
      .getExecInfoFn = stmtGetExecInfo,
321
  };
322

323
  STMT_ERR_RET(stmtCreateRequest(pStmt));
142,926✔
324
  pStmt->exec.pRequest->stmtBindVersion = 2;
142,926✔
325

326
  pStmt->stat.parseSqlNum++;
142,895✔
327

328
  STMT2_DLOG("start to parse, QID:0x%" PRIx64, pStmt->exec.pRequest->requestId);
142,926✔
329
  STMT_ERR_RET(parseSql(pStmt->exec.pRequest, false, &pStmt->sql.pQuery, &stmtCb));
142,926✔
330

331
  pStmt->sql.siInfo.pQuery = pStmt->sql.pQuery;
138,934✔
332

333
  pStmt->bInfo.needParse = false;
138,965✔
334

335
  if (pStmt->sql.type == 0) {
138,721✔
336
    if (pStmt->sql.pQuery->pRoot && LEGAL_INSERT(nodeType(pStmt->sql.pQuery->pRoot))) {
112,134✔
337
      pStmt->sql.type = STMT_TYPE_INSERT;
106,903✔
338
      pStmt->sql.stbInterlaceMode = false;
106,901✔
339
    } else if (pStmt->sql.pQuery->pPrepareRoot && LEGAL_SELECT(nodeType(pStmt->sql.pQuery->pPrepareRoot))) {
6,066✔
340
      pStmt->sql.type = STMT_TYPE_QUERY;
5,634✔
341
      pStmt->sql.stbInterlaceMode = false;
5,634✔
342

343
      return TSDB_CODE_SUCCESS;
5,634✔
344
    } else {
UNCOV
345
      STMT2_ELOG_E("only support select or insert sql");
×
346
      if (pStmt->exec.pRequest->msgBuf) {
×
347
        tstrncpy(pStmt->exec.pRequest->msgBuf, "stmt only support select or insert", pStmt->exec.pRequest->msgBufLen);
×
348
      }
349
      return TSDB_CODE_PAR_SYNTAX_ERROR;
×
350
    }
351
  } else if (pStmt->sql.type == STMT_TYPE_QUERY) {
26,368✔
352
    pStmt->sql.stbInterlaceMode = false;
×
353
    return TSDB_CODE_SUCCESS;
×
354
  } else if (pStmt->sql.type == STMT_TYPE_INSERT) {
26,368✔
355
    pStmt->sql.stbInterlaceMode = false;
×
356
  }
357

358
  STableDataCxt** pSrc =
359
      (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
133,300✔
360
  if (NULL == pSrc || NULL == *pSrc) {
133,584✔
361
    STMT2_ELOG("fail to get exec.pBlockHash, maybe parse failed, tbFName:%s", pStmt->bInfo.tbFName);
×
362
    STMT_ERR_RET(TSDB_CODE_TSC_STMT_CACHE_ERROR);
×
363
  }
364

365
  STableDataCxt* pTableCtx = *pSrc;
133,584✔
366
  if (pStmt->sql.stbInterlaceMode && pTableCtx->pData->pCreateTbReq && (pStmt->bInfo.tbNameFlag & USING_CLAUSE) == 0) {
133,306✔
367
    STMT2_TLOG("destroy pCreateTbReq for no-using insert, tbFName:%s", pStmt->bInfo.tbFName);
3,492✔
368
    tdDestroySVCreateTbReq(pTableCtx->pData->pCreateTbReq);
3,492✔
369
    taosMemoryFreeClear(pTableCtx->pData->pCreateTbReq);
3,492✔
370
    pTableCtx->pData->pCreateTbReq = NULL;
3,492✔
371
  }
372
  // if (pStmt->sql.stbInterlaceMode) {
373
  //   int16_t lastIdx = -1;
374

375
  //   for (int32_t i = 0; i < pTableCtx->boundColsInfo.numOfBound; ++i) {
376
  //     if (pTableCtx->boundColsInfo.pColIndex[i] < lastIdx) {
377
  //       pStmt->sql.stbInterlaceMode = false;
378
  //       break;
379
  //     }
380

381
  //     lastIdx = pTableCtx->boundColsInfo.pColIndex[i];
382
  //   }
383
  // }
384

385
  if (NULL == pStmt->sql.pBindInfo) {
133,553✔
386
    pStmt->sql.pBindInfo = taosMemoryMalloc(pTableCtx->boundColsInfo.numOfBound * sizeof(*pStmt->sql.pBindInfo));
129,820✔
387
    if (NULL == pStmt->sql.pBindInfo) {
129,931✔
388
      STMT2_ELOG_E("fail to malloc pBindInfo");
×
389
      return terrno;
×
390
    }
391
  }
392

393
  return TSDB_CODE_SUCCESS;
133,510✔
394
}
395

396
static int32_t stmtPrintBindv(TAOS_STMT2* stmt, TAOS_STMT2_BIND* bindv, int32_t col_idx, bool isTags) {
×
397
  STscStmt2* pStmt = (STscStmt2*)stmt;
×
398
  int32_t    count = 0;
×
399
  int32_t    code = 0;
×
400

401
  if (bindv == NULL) {
×
402
    STMT2_TLOG("bindv is NULL, col_idx:%d, isTags:%d", col_idx, isTags);
×
403
    return TSDB_CODE_SUCCESS;
×
404
  }
405

406
  if (col_idx >= 0) {
×
407
    count = 1;
×
408
    STMT2_TLOG("single col bind, col_idx:%d", col_idx);
×
409
  } else {
410
    if (STMT_TYPE_INSERT == pStmt->sql.type || STMT_TYPE_MULTI_INSERT == pStmt->sql.type ||
×
411
        (pStmt->sql.type == 0 && stmt2IsInsert(stmt))) {
×
412
      if (pStmt->sql.placeholderOfTags == 0 && pStmt->sql.placeholderOfCols == 0) {
×
413
        code = stmtGetStbColFields2(pStmt, NULL, NULL);
×
414
        if (code != TSDB_CODE_SUCCESS) {
×
415
          return code;
×
416
        }
417
      }
418
      if (isTags) {
×
419
        count = pStmt->sql.placeholderOfTags;
×
420
        STMT2_TLOG("print tags bindv, cols:%d", count);
×
421
      } else {
422
        count = pStmt->sql.placeholderOfCols;
×
423
        STMT2_TLOG("print cols bindv, cols:%d", count);
×
424
      }
425
    } else if (STMT_TYPE_QUERY == pStmt->sql.type || (pStmt->sql.type == 0 && stmt2IsSelect(stmt))) {
×
426
      count = taosArrayGetSize(pStmt->sql.pQuery->pPlaceholderValues);
×
427
      STMT2_TLOG("print query bindv, cols:%d", count);
×
428
    }
429
  }
430

431
  if (code != TSDB_CODE_SUCCESS) {
×
432
    STMT2_ELOG("failed to get param count, code:%d", code);
×
433
    return code;
×
434
  }
435

436
  for (int i = 0; i < count; i++) {
×
437
    int32_t type = bindv[i].buffer_type;
×
438
    int32_t num = bindv[i].num;
×
439
    char*   current_buf = (char*)bindv[i].buffer;
×
440

441
    for (int j = 0; j < num; j++) {
×
442
      char    buf[256] = {0};
×
443
      int32_t len = 0;
×
444
      bool    isNull = (bindv[i].is_null && bindv[i].is_null[j]);
×
445

446
      if (IS_VAR_DATA_TYPE(type) && bindv[i].length) {
×
447
        len = bindv[i].length[j];
×
448
      } else {
449
        len = tDataTypes[type].bytes;
×
450
      }
451

452
      if (isNull) {
×
453
        snprintf(buf, sizeof(buf), "NULL");
×
454
      } else {
455
        if (current_buf == NULL) {
×
456
          snprintf(buf, sizeof(buf), "NULL(Buf)");
×
457
        } else {
458
          switch (type) {
×
459
            case TSDB_DATA_TYPE_BOOL:
×
460
              snprintf(buf, sizeof(buf), "%d", *(int8_t*)current_buf);
×
461
              break;
×
462
            case TSDB_DATA_TYPE_TINYINT:
×
463
              snprintf(buf, sizeof(buf), "%d", *(int8_t*)current_buf);
×
464
              break;
×
465
            case TSDB_DATA_TYPE_SMALLINT:
×
466
              snprintf(buf, sizeof(buf), "%d", *(int16_t*)current_buf);
×
467
              break;
×
468
            case TSDB_DATA_TYPE_INT:
×
469
              snprintf(buf, sizeof(buf), "%d", *(int32_t*)current_buf);
×
470
              break;
×
471
            case TSDB_DATA_TYPE_BIGINT:
×
472
              snprintf(buf, sizeof(buf), "%" PRId64, *(int64_t*)current_buf);
×
473
              break;
×
474
            case TSDB_DATA_TYPE_FLOAT:
×
475
              snprintf(buf, sizeof(buf), "%f", *(float*)current_buf);
×
476
              break;
×
477
            case TSDB_DATA_TYPE_DOUBLE:
×
478
              snprintf(buf, sizeof(buf), "%f", *(double*)current_buf);
×
479
              break;
×
480
            case TSDB_DATA_TYPE_BINARY:
×
481
            case TSDB_DATA_TYPE_NCHAR:
482
            case TSDB_DATA_TYPE_GEOMETRY:
483
            case TSDB_DATA_TYPE_VARBINARY:
484
              snprintf(buf, sizeof(buf), "len:%d, val:%.*s", len, len, current_buf);
×
485
              break;
×
486
            case TSDB_DATA_TYPE_TIMESTAMP:
×
487
              snprintf(buf, sizeof(buf), "%" PRId64, *(int64_t*)current_buf);
×
488
              break;
×
489
            case TSDB_DATA_TYPE_UTINYINT:
×
490
              snprintf(buf, sizeof(buf), "%u", *(uint8_t*)current_buf);
×
491
              break;
×
492
            case TSDB_DATA_TYPE_USMALLINT:
×
493
              snprintf(buf, sizeof(buf), "%u", *(uint16_t*)current_buf);
×
494
              break;
×
495
            case TSDB_DATA_TYPE_UINT:
×
496
              snprintf(buf, sizeof(buf), "%u", *(uint32_t*)current_buf);
×
497
              break;
×
498
            case TSDB_DATA_TYPE_UBIGINT:
×
499
              snprintf(buf, sizeof(buf), "%" PRIu64, *(uint64_t*)current_buf);
×
500
              break;
×
501
            default:
×
502
              snprintf(buf, sizeof(buf), "UnknownType:%d", type);
×
503
              break;
×
504
          }
505
        }
506
      }
507

508
      STMT2_TLOG("bindv[%d] row[%d]: type:%s, val:%s", i, j, tDataTypes[type].name, buf);
×
509

510
      if (!isNull && current_buf) {
×
511
        current_buf += len;
×
512
      }
513
    }
514
  }
515

516
  return TSDB_CODE_SUCCESS;
×
517
}
518

519
static void resetRequest(STscStmt2* pStmt) {
259,430✔
520
  if (pStmt->exec.pRequest) {
259,430✔
521
    taos_free_result(pStmt->exec.pRequest);
145,577✔
522
    pStmt->exec.pRequest = NULL;
145,548✔
523
  }
524
  pStmt->asyncResultAvailable = false;
259,646✔
525
}
259,430✔
526

527
static int32_t stmtCleanBindInfo(STscStmt2* pStmt) {
784,633✔
528
  pStmt->bInfo.tbUid = 0;
784,633✔
529
  pStmt->bInfo.tbVgId = -1;
784,707✔
530
  pStmt->bInfo.tbType = 0;
784,573✔
531
  pStmt->bInfo.needParse = true;
784,610✔
532
  pStmt->bInfo.inExecCache = false;
784,596✔
533

534
  pStmt->bInfo.tbName[0] = 0;
784,596✔
535
  pStmt->bInfo.tbFName[0] = 0;
784,633✔
536
  if (!pStmt->bInfo.tagsCached) {
784,499✔
537
    qDestroyBoundColInfo(pStmt->bInfo.boundTags);
562,214✔
538
    taosMemoryFreeClear(pStmt->bInfo.boundTags);
562,158✔
539
    pStmt->bInfo.boundTags = NULL;
562,262✔
540
  }
541

542
  if (!pStmt->bInfo.boundColsCached) {
784,432✔
543
    tSimpleHashCleanup(pStmt->bInfo.fixedValueCols);
290,595✔
544
    pStmt->bInfo.fixedValueCols = NULL;
290,618✔
545
  }
546

547
  if (!pStmt->sql.autoCreateTbl) {
784,552✔
548
    pStmt->bInfo.stbFName[0] = 0;
544,681✔
549
    pStmt->bInfo.tbSuid = 0;
544,741✔
550
  }
551

552
  STMT2_TLOG("finish clean bind info, tagsCached:%d, autoCreateTbl:%d", pStmt->bInfo.tagsCached,
784,251✔
553
             pStmt->sql.autoCreateTbl);
554

555
  return TSDB_CODE_SUCCESS;
784,195✔
556
}
557

558
static void stmtFreeTableBlkList(STableColsData* pTb) {
×
559
  (void)qResetStmtColumns(pTb->aCol, true);
×
560
  taosArrayDestroy(pTb->aCol);
×
561
}
×
562

563
static void stmtResetQueueTableBuf(STableBufInfo* pTblBuf, SStmtQueue* pQueue) {
493,801✔
564
  pTblBuf->pCurBuff = taosArrayGetP(pTblBuf->pBufList, 0);
493,801✔
565
  if (NULL == pTblBuf->pCurBuff) {
494,055✔
566
    tscError("QInfo:%p, fail to get buffer from list", pTblBuf);
96✔
567
    return;
×
568
  }
569
  pTblBuf->buffIdx = 1;
493,959✔
570
  pTblBuf->buffOffset = sizeof(*pQueue->head);
493,959✔
571

572
  pQueue->head = pQueue->tail = pTblBuf->pCurBuff;
493,959✔
573
  pQueue->qRemainNum = 0;
493,885✔
574
  pQueue->head->next = NULL;
493,885✔
575
}
576

577
static int32_t stmtCleanExecInfo(STscStmt2* pStmt, bool keepTable, bool deepClean) {
758,471✔
578
  if (pStmt->sql.stbInterlaceMode) {
758,471✔
579
    if (deepClean) {
507,377✔
580
      taosHashCleanup(pStmt->exec.pBlockHash);
13,512✔
581
      pStmt->exec.pBlockHash = NULL;
13,512✔
582

583
      if (NULL != pStmt->exec.pCurrBlock) {
13,512✔
584
        taosMemoryFreeClear(pStmt->exec.pCurrBlock->boundColsInfo.pColIndex);
11,780✔
585
        taosMemoryFreeClear(pStmt->exec.pCurrBlock->pData);
11,780✔
586
        qDestroyStmtDataBlock(pStmt->exec.pCurrBlock);
11,780✔
587
        pStmt->exec.pCurrBlock = NULL;
11,780✔
588
      }
589
      if (STMT_TYPE_QUERY != pStmt->sql.type) {
13,512✔
590
        resetRequest(pStmt);
13,512✔
591
      }
592
    } else {
593
      pStmt->sql.siInfo.pTableColsIdx = 0;
493,865✔
594
      stmtResetQueueTableBuf(&pStmt->sql.siInfo.tbBuf, &pStmt->queue);
493,902✔
595
      tSimpleHashClear(pStmt->sql.siInfo.pTableRowDataHash);
493,798✔
596
    }
597
    if (NULL != pStmt->exec.pRequest) {
507,665✔
598
      pStmt->exec.pRequest->body.resInfo.numOfRows = 0;
494,153✔
599
    }
600
  } else {
601
    if (STMT_TYPE_QUERY != pStmt->sql.type || deepClean) {
251,273✔
602
      resetRequest(pStmt);
244,109✔
603
    }
604

605
    size_t keyLen = 0;
251,281✔
606
    void*  pIter = taosHashIterate(pStmt->exec.pBlockHash, NULL);
251,281✔
607
    while (pIter) {
496,069✔
608
      STableDataCxt* pBlocks = *(STableDataCxt**)pIter;
244,301✔
609
      char*          key = taosHashGetKey(pIter, &keyLen);
244,301✔
610
      STableMeta*    pMeta = qGetTableMetaInDataBlock(pBlocks);
244,301✔
611

612
      if (keepTable && pBlocks == pStmt->exec.pCurrBlock) {
244,267✔
613
        TSWAP(pBlocks->pData, pStmt->exec.pCurrTbData);
116,753✔
614
        STMT_ERR_RET(qResetStmtDataBlock(pBlocks, false));
144,977✔
615

616
        pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
116,321✔
617
        continue;
116,753✔
618
      }
619

620
      qDestroyStmtDataBlock(pBlocks);
127,548✔
621
      STMT_ERR_RET(taosHashRemove(pStmt->exec.pBlockHash, key, keyLen));
127,548✔
622

623
      pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
127,548✔
624
    }
625

626
    if (keepTable) {
251,768✔
627
      STMT2_TLOG("finish clean exec info, stbInterlaceMode:%d, keepTable:%d, deepClean:%d", pStmt->sql.stbInterlaceMode,
124,133✔
628
                 keepTable, deepClean);
629
      return TSDB_CODE_SUCCESS;
124,133✔
630
    }
631

632
    taosHashCleanup(pStmt->exec.pBlockHash);
127,635✔
633
    pStmt->exec.pBlockHash = NULL;
127,635✔
634

635
    tDestroySubmitTbData(pStmt->exec.pCurrTbData, TSDB_MSG_FLG_ENCODE);
127,635✔
636
    taosMemoryFreeClear(pStmt->exec.pCurrTbData);
127,635✔
637
  }
638

639
  STMT_ERR_RET(stmtCleanBindInfo(pStmt));
635,300✔
640
  STMT2_TLOG("finish clean exec info, stbInterlaceMode:%d, keepTable:%d, deepClean:%d", pStmt->sql.stbInterlaceMode,
634,724✔
641
             keepTable, deepClean);
642

643
  return TSDB_CODE_SUCCESS;
634,664✔
644
}
645

646
static void stmtFreeTbBuf(void* buf) {
124,630✔
647
  void* pBuf = *(void**)buf;
124,630✔
648
  taosMemoryFree(pBuf);
124,630✔
649
}
124,607✔
650

651
static void stmtFreeTbCols(void* buf) {
11,780,000✔
652
  SArray* pCols = *(SArray**)buf;
11,780,000✔
653
  taosArrayDestroy(pCols);
11,780,000✔
654
}
11,780,000✔
655

656
static int32_t stmtCleanSQLInfo(STscStmt2* pStmt) {
141,147✔
657
  STMT2_TLOG_E("start to free SQL info");
141,147✔
658

659
  taosMemoryFree(pStmt->sql.pBindInfo);
141,147✔
660
  taosMemoryFree(pStmt->sql.queryRes.fields);
141,147✔
661
  taosMemoryFree(pStmt->sql.queryRes.userFields);
141,147✔
662
  taosMemoryFree(pStmt->sql.sqlStr);
141,147✔
663
  qDestroyQuery(pStmt->sql.pQuery);
141,147✔
664
  taosArrayDestroy(pStmt->sql.nodeList);
141,147✔
665
  taosHashCleanup(pStmt->sql.pVgHash);
141,147✔
666
  pStmt->sql.pVgHash = NULL;
141,147✔
667
  if (pStmt->sql.fixValueTags) {
141,147✔
668
    pStmt->sql.fixValueTags = false;
3,278✔
669
    tdDestroySVCreateTbReq(pStmt->sql.fixValueTbReq);
3,278✔
670
    taosMemoryFreeClear(pStmt->sql.fixValueTbReq);
3,278✔
671
    pStmt->sql.fixValueTbReq = NULL;
3,278✔
672
  }
673

674
  void* pIter = taosHashIterate(pStmt->sql.pTableCache, NULL);
141,147✔
675
  while (pIter) {
146,986✔
676
    SStmtTableCache* pCache = (SStmtTableCache*)pIter;
5,839✔
677

678
    qDestroyStmtDataBlock(pCache->pDataCtx);
5,839✔
679
    qDestroyBoundColInfo(pCache->boundTags);
5,839✔
680
    taosMemoryFreeClear(pCache->boundTags);
5,839✔
681

682
    pIter = taosHashIterate(pStmt->sql.pTableCache, pIter);
5,839✔
683
  }
684
  taosHashCleanup(pStmt->sql.pTableCache);
141,147✔
685
  pStmt->sql.pTableCache = NULL;
141,147✔
686

687
  STMT_ERR_RET(stmtCleanExecInfo(pStmt, false, true));
141,147✔
688
  STMT_ERR_RET(stmtCleanBindInfo(pStmt));
141,147✔
689

690
  taos_free_result(pStmt->sql.siInfo.pRequest);
141,147✔
691
  taosHashCleanup(pStmt->sql.siInfo.pVgroupHash);
141,147✔
692
  tSimpleHashCleanup(pStmt->sql.siInfo.pTableHash);
141,147✔
693
  tSimpleHashCleanup(pStmt->sql.siInfo.pTableRowDataHash);
141,147✔
694
  taosArrayDestroyEx(pStmt->sql.siInfo.tbBuf.pBufList, stmtFreeTbBuf);
141,147✔
695
  taosMemoryFree(pStmt->sql.siInfo.pTSchema);
141,147✔
696
  qDestroyStmtDataBlock(pStmt->sql.siInfo.pDataCtx);
141,124✔
697
  taosArrayDestroyEx(pStmt->sql.siInfo.pTableCols, stmtFreeTbCols);
141,147✔
698
  pStmt->sql.siInfo.pTableCols = NULL;
141,147✔
699

700
  (void)memset(&pStmt->sql, 0, sizeof(pStmt->sql));
141,147✔
701
  pStmt->sql.siInfo.tableColsReady = true;
141,147✔
702

703
  STMT2_TLOG_E("end to free SQL info");
141,124✔
704

705
  return TSDB_CODE_SUCCESS;
141,147✔
706
}
707

708
static int32_t stmtTryAddTableVgroupInfo(STscStmt2* pStmt, int32_t* vgId) {
622,750✔
709
  if (*vgId >= 0 && taosHashGet(pStmt->sql.pVgHash, (const char*)vgId, sizeof(*vgId))) {
622,750✔
710
    return TSDB_CODE_SUCCESS;
1,940✔
711
  }
712

713
  SVgroupInfo      vgInfo = {0};
620,810✔
714
  SRequestConnInfo conn = {.pTrans = pStmt->taos->pAppInfo->pTransporter,
620,810✔
715
                           .requestId = pStmt->exec.pRequest->requestId,
620,810✔
716
                           .requestObjRefId = pStmt->exec.pRequest->self,
620,810✔
717
                           .mgmtEps = getEpSet_s(&pStmt->taos->pAppInfo->mgmtEp)};
620,810✔
718

719
  int32_t code = catalogGetTableHashVgroup(pStmt->pCatalog, &conn, &pStmt->bInfo.sname, &vgInfo);
620,983✔
720
  if (TSDB_CODE_SUCCESS != code) {
620,946✔
721
    STMT2_ELOG("fail to get vgroup info from catalog, code:%d", code);
×
722
    return code;
×
723
  }
724

725
  code =
726
      taosHashPut(pStmt->sql.pVgHash, (const char*)&vgInfo.vgId, sizeof(vgInfo.vgId), (char*)&vgInfo, sizeof(vgInfo));
620,946✔
727
  if (TSDB_CODE_SUCCESS != code) {
620,934✔
728
    STMT2_ELOG("fail to put vgroup info, code:%d", code);
74✔
729
    return code;
×
730
  }
731

732
  *vgId = vgInfo.vgId;
620,860✔
733

734
  return TSDB_CODE_SUCCESS;
620,860✔
735
}
736

737
int32_t stmtGetTableMetaAndValidate(STscStmt2* pStmt, uint64_t* uid, uint64_t* suid, int32_t* vgId, int8_t* tableType) {
8,653✔
738
  STableMeta*      pTableMeta = NULL;
8,653✔
739
  SRequestConnInfo conn = {.pTrans = pStmt->taos->pAppInfo->pTransporter,
8,653✔
740
                           .requestId = pStmt->exec.pRequest->requestId,
8,653✔
741
                           .requestObjRefId = pStmt->exec.pRequest->self,
8,653✔
742
                           .mgmtEps = getEpSet_s(&pStmt->taos->pAppInfo->mgmtEp)};
8,653✔
743
  int32_t          code = catalogGetTableMeta(pStmt->pCatalog, &conn, &pStmt->bInfo.sname, &pTableMeta);
8,653✔
744

745
  pStmt->stat.ctgGetTbMetaNum++;
8,653✔
746

747
  if (TSDB_CODE_PAR_TABLE_NOT_EXIST == code) {
8,653✔
748
    STMT2_ELOG("tb %s not exist", pStmt->bInfo.tbFName);
970✔
749
    (void)stmtCleanBindInfo(pStmt);
970✔
750

751
    if (!pStmt->sql.autoCreateTbl) {
970✔
752
      STMT2_ELOG("table %s does not exist and autoCreateTbl is disabled", pStmt->bInfo.tbFName);
970✔
753
      STMT_ERR_RET(TSDB_CODE_PAR_TABLE_NOT_EXIST);
970✔
754
    }
755

756
    STMT_ERR_RET(code);
×
757
  }
758

759
  STMT_ERR_RET(code);
7,683✔
760

761
  *uid = pTableMeta->uid;
7,683✔
762
  *suid = pTableMeta->suid;
7,683✔
763
  *tableType = pTableMeta->tableType;
7,683✔
764
  pStmt->bInfo.tbVgId = pTableMeta->vgId;
7,683✔
765
  *vgId = pTableMeta->vgId;
7,683✔
766

767
  taosMemoryFree(pTableMeta);
7,683✔
768

769
  return TSDB_CODE_SUCCESS;
7,683✔
770
}
771

772
static int32_t stmtRebuildDataBlock(STscStmt2* pStmt, STableDataCxt* pDataBlock, STableDataCxt** newBlock, uint64_t uid,
14,083✔
773
                                    uint64_t suid, int32_t vgId) {
774
  STMT_ERR_RET(stmtTryAddTableVgroupInfo(pStmt, &vgId));
14,083✔
775
  STMT_ERR_RET(qRebuildStmtDataBlock(newBlock, pDataBlock, uid, suid, vgId, pStmt->sql.autoCreateTbl));
14,083✔
776

777
  STMT2_DLOG("uid:%" PRId64 ", rebuild table data context, vgId:%d", uid, vgId);
14,083✔
778

779
  return TSDB_CODE_SUCCESS;
14,083✔
780
}
781

782
static int32_t stmtGetFromCache(STscStmt2* pStmt) {
39,809✔
783
  if (pStmt->sql.stbInterlaceMode && pStmt->sql.siInfo.pDataCtx) {
39,809✔
784
    pStmt->bInfo.needParse = false;
×
785
    pStmt->bInfo.inExecCache = false;
×
786
    return TSDB_CODE_SUCCESS;
×
787
  }
788

789
  pStmt->bInfo.needParse = true;
39,850✔
790
  pStmt->bInfo.inExecCache = false;
39,850✔
791

792
  STableDataCxt** pCxtInExec = taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
39,850✔
793
  if (pCxtInExec) {
39,850✔
794
    pStmt->bInfo.needParse = false;
6,596✔
795
    pStmt->bInfo.inExecCache = true;
6,596✔
796

797
    pStmt->exec.pCurrBlock = *pCxtInExec;
6,596✔
798

799
    if (pStmt->sql.autoCreateTbl) {
6,596✔
800
      STMT2_DLOG("reuse stmt block for tb %s in execBlock", pStmt->bInfo.tbFName);
5,820✔
801
      return TSDB_CODE_SUCCESS;
5,820✔
802
    }
803
  }
804

805
  if (NULL == pStmt->pCatalog) {
34,030✔
806
    STMT_ERR_RET(catalogGetHandle(pStmt->taos->pAppInfo->clusterId, &pStmt->pCatalog));
14,143✔
807
    pStmt->sql.siInfo.pCatalog = pStmt->pCatalog;
14,143✔
808
  }
809

810
  if (NULL == pStmt->sql.pTableCache || taosHashGetSize(pStmt->sql.pTableCache) <= 0) {
34,030✔
811
    if (pStmt->bInfo.inExecCache) {
18,977✔
812
      pStmt->bInfo.needParse = false;
×
813
      STMT2_DLOG("reuse stmt block for tb %s in execBlock", pStmt->bInfo.tbFName);
×
814
      return TSDB_CODE_SUCCESS;
×
815
    }
816

817
    STMT2_DLOG("no stmt block cache for tb %s", pStmt->bInfo.tbFName);
18,977✔
818

819
    return TSDB_CODE_SUCCESS;
18,936✔
820
  }
821

822
  if (pStmt->sql.autoCreateTbl) {
15,053✔
823
    SStmtTableCache* pCache = taosHashGet(pStmt->sql.pTableCache, &pStmt->bInfo.tbSuid, sizeof(pStmt->bInfo.tbSuid));
11,755✔
824
    if (pCache) {
11,755✔
825
      pStmt->bInfo.needParse = false;
11,755✔
826
      pStmt->bInfo.tbUid = 0;
11,755✔
827

828
      STableDataCxt* pNewBlock = NULL;
11,755✔
829
      STMT_ERR_RET(stmtRebuildDataBlock(pStmt, pCache->pDataCtx, &pNewBlock, 0, pStmt->bInfo.tbSuid, -1));
11,755✔
830

831
      if (taosHashPut(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName), &pNewBlock,
11,755✔
832
                      POINTER_BYTES)) {
833
        STMT_ERR_RET(terrno);
×
834
      }
835

836
      pStmt->exec.pCurrBlock = pNewBlock;
11,755✔
837

838
      STMT2_DLOG("reuse stmt block for tb %s in sqlBlock, suid:0x%" PRIx64, pStmt->bInfo.tbFName, pStmt->bInfo.tbSuid);
11,755✔
839

840
      return TSDB_CODE_SUCCESS;
11,755✔
841
    }
842

843
    STMT_RET(stmtCleanBindInfo(pStmt));
×
844
  }
845

846
  uint64_t uid, suid;
×
847
  int32_t  vgId;
×
848
  int8_t   tableType;
×
849

850
  STMT_ERR_RET(stmtGetTableMetaAndValidate(pStmt, &uid, &suid, &vgId, &tableType));
3,298✔
851

852
  uint64_t cacheUid = (TSDB_CHILD_TABLE == tableType) ? suid : uid;
3,104✔
853

854
  if (uid == pStmt->bInfo.tbUid) {
3,104✔
855
    pStmt->bInfo.needParse = false;
×
856

857
    STMT2_DLOG("tb %s is current table", pStmt->bInfo.tbFName);
×
858

859
    return TSDB_CODE_SUCCESS;
×
860
  }
861

862
  if (pStmt->bInfo.inExecCache) {
3,104✔
863
    SStmtTableCache* pCache = taosHashGet(pStmt->sql.pTableCache, &cacheUid, sizeof(cacheUid));
776✔
864
    if (NULL == pCache) {
776✔
865
      STMT2_ELOG("table [%s, %" PRIx64 ", %" PRIx64 "] found in exec blockHash, but not in sql blockHash",
×
866
                 pStmt->bInfo.tbFName, uid, cacheUid);
867

868
      STMT_ERR_RET(TSDB_CODE_APP_ERROR);
×
869
    }
870

871
    pStmt->bInfo.needParse = false;
776✔
872

873
    pStmt->bInfo.tbUid = uid;
776✔
874
    pStmt->bInfo.tbSuid = suid;
776✔
875
    pStmt->bInfo.tbType = tableType;
776✔
876
    pStmt->bInfo.boundTags = pCache->boundTags;
776✔
877
    pStmt->bInfo.tagsCached = true;
776✔
878

879
    STMT2_DLOG("tb %s in execBlock list, set to current", pStmt->bInfo.tbFName);
776✔
880

881
    return TSDB_CODE_SUCCESS;
776✔
882
  }
883

884
  SStmtTableCache* pCache = taosHashGet(pStmt->sql.pTableCache, &cacheUid, sizeof(cacheUid));
2,328✔
885
  if (pCache) {
2,328✔
886
    pStmt->bInfo.needParse = false;
2,328✔
887

888
    pStmt->bInfo.tbUid = uid;
2,328✔
889
    pStmt->bInfo.tbSuid = suid;
2,328✔
890
    pStmt->bInfo.tbType = tableType;
2,328✔
891
    pStmt->bInfo.boundTags = pCache->boundTags;
2,328✔
892
    pStmt->bInfo.tagsCached = true;
2,328✔
893

894
    STableDataCxt* pNewBlock = NULL;
2,328✔
895
    STMT_ERR_RET(stmtRebuildDataBlock(pStmt, pCache->pDataCtx, &pNewBlock, uid, suid, vgId));
2,328✔
896

897
    if (taosHashPut(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName), &pNewBlock,
2,328✔
898
                    POINTER_BYTES)) {
899
      STMT_ERR_RET(terrno);
×
900
    }
901

902
    pStmt->exec.pCurrBlock = pNewBlock;
2,328✔
903

904
    tscDebug("tb %s in sqlBlock list, set to current", pStmt->bInfo.tbFName);
2,328✔
905

906
    return TSDB_CODE_SUCCESS;
2,328✔
907
  }
908

909
  STMT_ERR_RET(stmtCleanBindInfo(pStmt));
×
910

911
  return TSDB_CODE_SUCCESS;
×
912
}
913

UNCOV
914
static int32_t stmtResetStmt(STscStmt2* pStmt) {
×
UNCOV
915
  STMT_ERR_RET(stmtCleanSQLInfo(pStmt));
×
916

UNCOV
917
  pStmt->sql.pTableCache = taosHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
×
UNCOV
918
  if (NULL == pStmt->sql.pTableCache) {
×
919
    STMT2_ELOG("fail to allocate memory for pTableCache in stmtResetStmt:%s", tstrerror(terrno));
×
920
    STMT_ERR_RET(terrno);
×
921
  }
922

UNCOV
923
  pStmt->sql.status = STMT_INIT;
×
924

UNCOV
925
  return TSDB_CODE_SUCCESS;
×
926
}
927

928
static void stmtAsyncOutput(STscStmt2* pStmt, void* param) {
1,511,257✔
929
  SStmtQNode* pParam = (SStmtQNode*)param;
1,511,257✔
930

931
  if (pParam->restoreTbCols) {
1,511,257✔
932
    for (int32_t i = 0; i < pStmt->sql.siInfo.pTableColsIdx; ++i) {
1,510,782✔
933
      SArray** p = (SArray**)TARRAY_GET_ELEM(pStmt->sql.siInfo.pTableCols, i);
1,016,646✔
934
      *p = taosArrayInit(20, POINTER_BYTES);
1,016,572✔
935
      if (*p == NULL) {
1,016,473✔
936
        pStmt->errCode = terrno;
13✔
937
      }
938
    }
939
    atomic_store_8((int8_t*)&pStmt->sql.siInfo.tableColsReady, true);
494,210✔
940
    STMT2_TLOG_E("restore pTableCols finished");
494,347✔
941
  } else {
942
    int code = qAppendStmt2TableOutput(pStmt->sql.pQuery, pStmt->sql.pVgHash, &pParam->tblData, pStmt->exec.pCurrBlock,
1,016,972✔
943
                                       &pStmt->sql.siInfo, pParam->pCreateTbReq);
944
    // taosMemoryFree(pParam->pTbData);
945
    if (code != TSDB_CODE_SUCCESS) {
1,016,869✔
946
      STMT2_ELOG("async append stmt output failed, tbname:%s, err:%s", pParam->tblData.tbName, tstrerror(code));
194✔
947
      pStmt->errCode = code;
194✔
948
    }
949
    (void)atomic_sub_fetch_64(&pStmt->sql.siInfo.tbRemainNum, 1);
1,016,869✔
950
  }
951
}
1,511,439✔
952

953
static void* stmtBindThreadFunc(void* param) {
124,981✔
954
  setThreadName("stmt2Bind");
124,981✔
955

956
  STscStmt2* pStmt = (STscStmt2*)param;
125,018✔
957
  STMT2_DLOG_E("stmt2 bind thread started");
125,018✔
958

959
  while (true) {
1,511,389✔
960
    SStmtQNode* asyncParam = NULL;
1,636,407✔
961

962
    if (!stmtDequeue(pStmt, &asyncParam)) {
1,636,444✔
963
      if (pStmt->queue.stopQueue && 0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) {
124,630✔
964
        STMT2_DLOG_E("queue is empty and stopQueue is set, thread will exit");
124,630✔
965
        break;
124,630✔
966
      }
967
      continue;
×
968
    }
969

970
    stmtAsyncOutput(pStmt, asyncParam);
1,511,198✔
971
  }
972

973
  STMT2_DLOG_E("stmt2 bind thread stopped");
124,556✔
974
  return NULL;
124,630✔
975
}
976

977
static int32_t stmtStartBindThread(STscStmt2* pStmt) {
124,987✔
978
  TdThreadAttr thAttr;
98,427✔
979
  if (taosThreadAttrInit(&thAttr) != 0) {
124,987✔
980
    return TSDB_CODE_TSC_INTERNAL_ERROR;
×
981
  }
982
  if (taosThreadAttrSetDetachState(&thAttr, PTHREAD_CREATE_JOINABLE) != 0) {
125,018✔
983
    return TSDB_CODE_TSC_INTERNAL_ERROR;
×
984
  }
985

986
  if (taosThreadCreate(&pStmt->bindThread, &thAttr, stmtBindThreadFunc, pStmt) != 0) {
124,950✔
987
    terrno = TAOS_SYSTEM_ERROR(ERRNO);
×
988
    STMT_ERR_RET(terrno);
×
989
  }
990

991
  pStmt->bindThreadInUse = true;
125,018✔
992

993
  (void)taosThreadAttrDestroy(&thAttr);
125,018✔
994
  return TSDB_CODE_SUCCESS;
125,018✔
995
}
996

997
static int32_t stmtInitQueue(STscStmt2* pStmt) {
125,018✔
998
  (void)taosThreadCondInit(&pStmt->queue.waitCond, NULL);
125,018✔
999
  (void)taosThreadMutexInit(&pStmt->queue.mutex, NULL);
125,018✔
1000
  STMT_ERR_RET(stmtAllocQNodeFromBuf(&pStmt->sql.siInfo.tbBuf, (void**)&pStmt->queue.head));
249,715✔
1001
  pStmt->queue.tail = pStmt->queue.head;
124,734✔
1002

1003
  return TSDB_CODE_SUCCESS;
124,950✔
1004
}
1005

1006
static int32_t stmtIniAsyncBind(STscStmt2* pStmt) {
140,565✔
1007
  (void)taosThreadCondInit(&pStmt->asyncBindParam.waitCond, NULL);
140,565✔
1008
  (void)taosThreadMutexInit(&pStmt->asyncBindParam.mutex, NULL);
140,565✔
1009
  pStmt->asyncBindParam.asyncBindNum = 0;
140,534✔
1010

1011
  return TSDB_CODE_SUCCESS;
140,565✔
1012
}
1013

1014
static int32_t stmtInitTableBuf(STableBufInfo* pTblBuf) {
124,995✔
1015
  pTblBuf->buffUnit = sizeof(SStmtQNode);
124,995✔
1016
  pTblBuf->buffSize = pTblBuf->buffUnit * 1000;
125,018✔
1017
  pTblBuf->pBufList = taosArrayInit(100, POINTER_BYTES);
124,995✔
1018
  if (NULL == pTblBuf->pBufList) {
125,018✔
1019
    return terrno;
×
1020
  }
1021
  void* buff = taosMemoryMalloc(pTblBuf->buffSize);
125,018✔
1022
  if (NULL == buff) {
125,018✔
1023
    return terrno;
×
1024
  }
1025

1026
  if (taosArrayPush(pTblBuf->pBufList, &buff) == NULL) {
250,036✔
1027
    return terrno;
×
1028
  }
1029

1030
  pTblBuf->pCurBuff = buff;
125,018✔
1031
  pTblBuf->buffIdx = 1;
125,018✔
1032
  pTblBuf->buffOffset = 0;
125,018✔
1033

1034
  return TSDB_CODE_SUCCESS;
125,018✔
1035
}
1036

1037
TAOS_STMT2* stmtInit2(STscObj* taos, TAOS_STMT2_OPTION* pOptions) {
134,983✔
1038
  STscObj*   pObj = (STscObj*)taos;
134,983✔
1039
  STscStmt2* pStmt = NULL;
134,983✔
1040
  int32_t    code = 0;
134,983✔
1041

1042
  pStmt = taosMemoryCalloc(1, sizeof(STscStmt2));
134,983✔
1043
  if (NULL == pStmt) {
135,361✔
1044
    STMT2_ELOG_E("fail to allocate memory for pStmt");
×
1045
    return NULL;
×
1046
  }
1047

1048
  pStmt->sql.pTableCache = taosHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
135,361✔
1049
  if (NULL == pStmt->sql.pTableCache) {
135,384✔
1050
    STMT2_ELOG("fail to allocate memory for pTableCache in stmtInit2:%s", tstrerror(terrno));
×
1051
    taosMemoryFree(pStmt);
×
1052
    return NULL;
×
1053
  }
1054

1055
  pStmt->taos = pObj;
135,384✔
1056
  if (taos->db[0] != '\0') {
135,384✔
1057
    pStmt->db = taosStrdup(taos->db);
124,714✔
1058
  }
1059
  pStmt->bInfo.needParse = true;
135,384✔
1060
  pStmt->sql.status = STMT_INIT;
135,384✔
1061
  pStmt->errCode = TSDB_CODE_SUCCESS;
135,361✔
1062

1063
  if (NULL != pOptions) {
135,361✔
1064
    (void)memcpy(&pStmt->options, pOptions, sizeof(pStmt->options));
135,365✔
1065
    if (pOptions->singleStbInsert && pOptions->singleTableBindOnce) {
135,365✔
1066
      pStmt->stbInterlaceMode = true;
119,837✔
1067
    }
1068

1069
    pStmt->reqid = pOptions->reqid;
135,388✔
1070
  }
1071

1072
  if (pStmt->stbInterlaceMode) {
135,361✔
1073
    pStmt->sql.siInfo.transport = taos->pAppInfo->pTransporter;
119,814✔
1074
    pStmt->sql.siInfo.acctId = taos->acctId;
119,814✔
1075
    pStmt->sql.siInfo.dbname = taos->db;
119,837✔
1076
    pStmt->sql.siInfo.mgmtEpSet = getEpSet_s(&pStmt->taos->pAppInfo->mgmtEp);
119,814✔
1077

1078
    pStmt->sql.siInfo.pTableHash = tSimpleHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY));
119,814✔
1079
    if (NULL == pStmt->sql.siInfo.pTableHash) {
119,598✔
1080
      STMT2_ELOG("fail to allocate memory for pTableHash:%s", tstrerror(terrno));
×
1081
      (void)stmtClose2(pStmt);
×
1082
      return NULL;
×
1083
    }
1084

1085
    pStmt->sql.siInfo.pTableRowDataHash = tSimpleHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY));
119,621✔
1086
    if (NULL == pStmt->sql.siInfo.pTableRowDataHash) {
119,598✔
1087
      STMT2_ELOG("fail to allocate memory for pTableRowDataHash:%s", tstrerror(terrno));
×
1088
      (void)stmtClose2(pStmt);
×
1089
      return NULL;
×
1090
    }
1091

1092
    pStmt->sql.siInfo.pTableCols = taosArrayInit(STMT_TABLE_COLS_NUM, POINTER_BYTES);
119,590✔
1093
    if (NULL == pStmt->sql.siInfo.pTableCols) {
119,814✔
1094
      STMT2_ELOG("fail to allocate memory for pTableCols:%s", tstrerror(terrno));
×
1095
      (void)stmtClose2(pStmt);
×
1096
      return NULL;
×
1097
    }
1098

1099
    code = stmtInitTableBuf(&pStmt->sql.siInfo.tbBuf);
119,837✔
1100
    if (TSDB_CODE_SUCCESS == code) {
119,837✔
1101
      code = stmtInitQueue(pStmt);
119,837✔
1102
    }
1103
    if (TSDB_CODE_SUCCESS == code) {
119,806✔
1104
      code = stmtStartBindThread(pStmt);
119,806✔
1105
    }
1106
    if (TSDB_CODE_SUCCESS != code) {
119,837✔
1107
      terrno = code;
×
1108
      STMT2_ELOG("fail to init stmt2 bind thread:%s", tstrerror(code));
×
1109
      (void)stmtClose2(pStmt);
×
1110
      return NULL;
×
1111
    }
1112
  }
1113

1114
  pStmt->sql.siInfo.tableColsReady = true;
135,407✔
1115
  if (pStmt->options.asyncExecFn) {
135,384✔
1116
    if (tsem_init(&pStmt->asyncExecSem, 0, 1) != 0) {
2,134✔
1117
      terrno = TAOS_SYSTEM_ERROR(ERRNO);
×
1118
      STMT2_ELOG("fail to init asyncExecSem:%s", tstrerror(terrno));
×
1119
      (void)stmtClose2(pStmt);
×
1120
      return NULL;
×
1121
    }
1122
  }
1123
  code = stmtIniAsyncBind(pStmt);
135,384✔
1124
  if (TSDB_CODE_SUCCESS != code) {
135,384✔
1125
    terrno = code;
×
1126
    STMT2_ELOG("fail to start init asyncExecSem:%s", tstrerror(code));
×
1127

1128
    (void)stmtClose2(pStmt);
×
1129
    return NULL;
×
1130
  }
1131

1132
  pStmt->execSemWaited = false;
135,384✔
1133

1134
  // STMT_LOG_SEQ(STMT_INIT);
1135

1136
  STMT2_DLOG("stmt2 initialize finished, seqId:%d, db:%s, interlaceMode:%d, asyncExec:%d", pStmt->seqId, pStmt->db,
135,353✔
1137
             pStmt->stbInterlaceMode, pStmt->options.asyncExecFn != NULL);
1138

1139
  return pStmt;
135,309✔
1140
}
1141

1142
static int stmtSetDbName2(TAOS_STMT2* stmt, const char* dbName) {
123,855✔
1143
  STscStmt2* pStmt = (STscStmt2*)stmt;
123,855✔
1144
  if (dbName == NULL || dbName[0] == '\0') {
123,855✔
1145
    STMT2_ELOG_E("dbname in sql is illegal");
31✔
1146
    return TSDB_CODE_TSC_SQL_SYNTAX_ERROR;
×
1147
  }
1148

1149
  STMT2_DLOG("dbname is specified in sql:%s", dbName);
124,071✔
1150
  if (pStmt->db == NULL || pStmt->db[0] == '\0') {
124,071✔
1151
    taosMemoryFreeClear(pStmt->db);
9,506✔
1152
    STMT2_DLOG("dbname:%s is by sql, not by taosconnect", dbName);
9,506✔
1153
    pStmt->db = taosStrdup(dbName);
9,506✔
1154
    (void)strdequote(pStmt->db);
9,506✔
1155
  }
1156
  STMT_ERR_RET(stmtCreateRequest(pStmt));
124,043✔
1157

1158
  // The SQL statement specifies a database name, overriding the previously specified database
1159
  taosMemoryFreeClear(pStmt->exec.pRequest->pDb);
123,855✔
1160
  pStmt->exec.pRequest->pDb = taosStrdup(dbName);
123,945✔
1161
  (void)strdequote(pStmt->exec.pRequest->pDb);
123,855✔
1162
  if (pStmt->exec.pRequest->pDb == NULL) {
124,012✔
1163
    return terrno;
×
1164
  }
1165
  if (pStmt->sql.stbInterlaceMode) {
123,796✔
1166
    pStmt->sql.siInfo.dbname = pStmt->exec.pRequest->pDb;
112,738✔
1167
  }
1168
  return TSDB_CODE_SUCCESS;
123,796✔
1169
}
1170
static int32_t stmtResetStbInterlaceCache(STscStmt2* pStmt) {
5,181✔
1171
  int32_t code = TSDB_CODE_SUCCESS;
5,181✔
1172

1173
  pStmt->sql.siInfo.pTableHash = tSimpleHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY));
5,181✔
1174
  if (NULL == pStmt->sql.siInfo.pTableHash) {
5,181✔
1175
    return terrno;
×
1176
  }
1177

1178
  pStmt->sql.siInfo.pTableRowDataHash = tSimpleHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY));
5,181✔
1179
  if (NULL == pStmt->sql.siInfo.pTableRowDataHash) {
5,181✔
1180
    return terrno;
×
1181
  }
1182

1183
  pStmt->sql.siInfo.pTableCols = taosArrayInit(STMT_TABLE_COLS_NUM, POINTER_BYTES);
5,181✔
1184
  if (NULL == pStmt->sql.siInfo.pTableCols) {
5,181✔
1185
    return terrno;
×
1186
  }
1187

1188
  code = stmtInitTableBuf(&pStmt->sql.siInfo.tbBuf);
5,181✔
1189

1190
  if (TSDB_CODE_SUCCESS == code) {
5,181✔
1191
    code = stmtInitQueue(pStmt);
5,181✔
1192
    pStmt->queue.stopQueue = false;
5,181✔
1193
  }
1194
  if (TSDB_CODE_SUCCESS == code) {
5,181✔
1195
    code = stmtStartBindThread(pStmt);
5,181✔
1196
  }
1197
  if (TSDB_CODE_SUCCESS != code) {
5,181✔
1198
    return code;
×
1199
  }
1200

1201
  return TSDB_CODE_SUCCESS;
5,181✔
1202
}
1203

1204
static int32_t stmtDeepReset(STscStmt2* pStmt) {
6,733✔
1205
  // Save state that needs to be preserved
1206
  char*             db = pStmt->db;
6,733✔
1207
  TAOS_STMT2_OPTION options = pStmt->options;
6,733✔
1208
  uint32_t          reqid = pStmt->reqid;
6,733✔
1209
  bool              stbInterlaceMode = pStmt->stbInterlaceMode;
6,733✔
1210

1211
  pStmt->errCode = 0;
6,733✔
1212

1213
  // Wait for async execution to complete
1214
  if (pStmt->options.asyncExecFn && !pStmt->execSemWaited) {
6,733✔
1215
    if (tsem_wait(&pStmt->asyncExecSem) != 0) {
388✔
1216
      STMT2_ELOG_E("bind param wait asyncExecSem failed");
×
1217
    }
1218
    pStmt->execSemWaited = true;
388✔
1219
  }
1220

1221
  // Stop bind thread if in use (similar to stmtClose2)
1222
  if (stbInterlaceMode && pStmt->bindThreadInUse) {
6,733✔
1223
    while (0 == atomic_load_8((int8_t*)&pStmt->sql.siInfo.tableColsReady)) {
5,181✔
1224
      taosUsleep(1);
×
1225
    }
1226
    (void)taosThreadMutexLock(&pStmt->queue.mutex);
5,181✔
1227
    pStmt->queue.stopQueue = true;
5,181✔
1228
    (void)taosThreadCondSignal(&(pStmt->queue.waitCond));
5,181✔
1229
    (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
5,181✔
1230

1231
    (void)taosThreadJoin(pStmt->bindThread, NULL);
5,181✔
1232
    pStmt->bindThreadInUse = false;
5,181✔
1233
    pStmt->queue.head = NULL;
5,181✔
1234
    pStmt->queue.tail = NULL;
5,181✔
1235
    pStmt->queue.qRemainNum = 0;
5,181✔
1236

1237
    (void)taosThreadCondDestroy(&pStmt->queue.waitCond);
5,181✔
1238
    (void)taosThreadMutexDestroy(&pStmt->queue.mutex);
5,181✔
1239
  }
1240

1241
  // Clean all SQL and execution info (stmtCleanSQLInfo already handles most cleanup)
1242
  pStmt->bInfo.boundColsCached = false;
6,733✔
1243
  if (stbInterlaceMode) {
6,733✔
1244
    pStmt->bInfo.tagsCached = false;
5,181✔
1245
  }
1246
  STMT_ERR_RET(stmtCleanSQLInfo(pStmt));
6,733✔
1247

1248
  // Reinitialize resources (similar to stmtInit2)
1249
  if (stbInterlaceMode) {
6,733✔
1250
    pStmt->sql.siInfo.transport = pStmt->taos->pAppInfo->pTransporter;
5,181✔
1251
    pStmt->sql.siInfo.acctId = pStmt->taos->acctId;
5,181✔
1252
    pStmt->sql.siInfo.dbname = pStmt->taos->db;
5,181✔
1253
    pStmt->sql.siInfo.mgmtEpSet = getEpSet_s(&pStmt->taos->pAppInfo->mgmtEp);
5,181✔
1254

1255
    if (NULL == pStmt->pCatalog) {
5,181✔
1256
      STMT_ERR_RET(catalogGetHandle(pStmt->taos->pAppInfo->clusterId, &pStmt->pCatalog));
582✔
1257
    }
1258
    pStmt->sql.siInfo.pCatalog = pStmt->pCatalog;
5,181✔
1259

1260
    STMT_ERR_RET(stmtResetStbInterlaceCache(pStmt));
5,181✔
1261

1262
    int32_t code = stmtIniAsyncBind(pStmt);
5,181✔
1263
    if (TSDB_CODE_SUCCESS != code) {
5,181✔
1264
      STMT2_ELOG("fail to reinit async bind in stmtDeepReset:%s", tstrerror(code));
×
1265
      return code;
×
1266
    }
1267
  }
1268

1269
  // Restore preserved state
1270
  pStmt->db = db;
6,733✔
1271
  pStmt->options = options;
6,733✔
1272
  pStmt->reqid = reqid;
6,733✔
1273
  pStmt->stbInterlaceMode = stbInterlaceMode;
6,733✔
1274

1275
  pStmt->sql.pTableCache = taosHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
6,733✔
1276
  if (NULL == pStmt->sql.pTableCache) {
6,733✔
1277
    STMT2_ELOG("fail to allocate memory for pTableCache in stmtDeepReset:%s", tstrerror(terrno));
×
1278
    return terrno;
×
1279
  }
1280

1281
  pStmt->bInfo.needParse = true;
6,733✔
1282
  pStmt->sql.status = STMT_INIT;
6,733✔
1283
  pStmt->sql.siInfo.tableColsReady = true;
6,733✔
1284

1285
  return TSDB_CODE_SUCCESS;
6,733✔
1286
}
1287

1288
int stmtPrepare2(TAOS_STMT2* stmt, const char* sql, unsigned long length) {
141,698✔
1289
  STscStmt2* pStmt = (STscStmt2*)stmt;
141,698✔
1290
  int32_t    code = 0;
141,698✔
1291

1292
  STMT2_DLOG("start to prepare with sql:%s", sql);
141,698✔
1293

1294
  if (stmt == NULL || sql == NULL) {
141,698✔
1295
    STMT2_ELOG_E("stmt or sql is NULL");
×
1296
    return TSDB_CODE_INVALID_PARA;
×
1297
  }
1298

1299
  if (pStmt->sql.status >= STMT_PREPARE) {
141,698✔
1300
    STMT2_DLOG("stmt status is %d, need to reset stmt2 cache before prepare", pStmt->sql.status);
6,733✔
1301
    STMT_ERR_RET(stmtDeepReset(pStmt));
6,733✔
1302
  }
1303

1304
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
141,729✔
1305
    STMT2_ELOG("errCode is not success before, ErrCode: 0x%x, errorsyt: %s\n. ", pStmt->errCode,
194✔
1306
               tstrerror(pStmt->errCode));
1307
    return pStmt->errCode;
194✔
1308
  }
1309

1310
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_PREPARE));
141,504✔
1311

1312
  if (length <= 0) {
141,535✔
1313
    length = strlen(sql);
138,660✔
1314
  }
1315
  pStmt->sql.sqlStr = taosStrndup(sql, length);
141,535✔
1316
  if (!pStmt->sql.sqlStr) {
141,473✔
1317
    STMT2_ELOG("fail to allocate memory for sqlStr:%s", tstrerror(terrno));
×
1318
    STMT_ERR_RET(terrno);
×
1319
  }
1320
  pStmt->sql.sqlLen = length;
141,504✔
1321
  STMT_ERR_RET(stmtCreateRequest(pStmt));
141,535✔
1322

1323
  if (stmt2IsInsert(pStmt)) {
141,476✔
1324
    pStmt->sql.stbInterlaceMode = pStmt->stbInterlaceMode;
135,125✔
1325
    char* dbName = NULL;
135,094✔
1326
    if (qParseDbName(sql, length, &dbName)) {
135,094✔
1327
      STMT_ERR_RET(stmtSetDbName2(stmt, dbName));
123,855✔
1328
      taosMemoryFreeClear(dbName);
123,759✔
1329
    } else if (pStmt->db != NULL && pStmt->db[0] != '\0') {
11,023✔
1330
      taosMemoryFreeClear(pStmt->exec.pRequest->pDb);
10,829✔
1331
      pStmt->exec.pRequest->pDb = taosStrdup(pStmt->db);
10,829✔
1332
      if (pStmt->exec.pRequest->pDb == NULL) {
10,829✔
1333
        STMT_ERR_RET(terrno);
×
1334
      }
1335
      (void)strdequote(pStmt->exec.pRequest->pDb);
10,829✔
1336

1337
      if (pStmt->sql.stbInterlaceMode) {
10,829✔
1338
        pStmt->sql.siInfo.dbname = pStmt->exec.pRequest->pDb;
5,960✔
1339
      }
1340
    }
1341

1342
  } else if (stmt2IsSelect(pStmt)) {
6,410✔
1343
    pStmt->sql.stbInterlaceMode = false;
5,828✔
1344
    STMT_ERR_RET(stmtParseSql(pStmt));
5,828✔
1345
  } else {
1346
    return stmtBuildErrorMsgWithCode(pStmt, "stmt only support 'SELECT' or 'INSERT'", TSDB_CODE_PAR_SYNTAX_ERROR);
582✔
1347
  }
1348
  return TSDB_CODE_SUCCESS;
140,475✔
1349
}
1350

1351
static int32_t stmtInitStbInterlaceTableInfo(STscStmt2* pStmt) {
11,780✔
1352
  STableDataCxt** pSrc = taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
11,780✔
1353
  if (!pSrc) {
11,780✔
1354
    return terrno;
×
1355
  }
1356
  STableDataCxt* pDst = NULL;
11,780✔
1357

1358
  STMT_ERR_RET(qCloneStmtDataBlock(&pDst, *pSrc, true));
11,780✔
1359
  pStmt->sql.siInfo.pDataCtx = pDst;
11,780✔
1360

1361
  SArray* pTblCols = NULL;
11,780✔
1362
  for (int32_t i = 0; i < STMT_TABLE_COLS_NUM; i++) {
11,784,900✔
1363
    pTblCols = taosArrayInit(20, POINTER_BYTES);
11,773,175✔
1364
    if (NULL == pTblCols) {
11,767,553✔
1365
      return terrno;
×
1366
    }
1367

1368
    if (taosArrayPush(pStmt->sql.siInfo.pTableCols, &pTblCols) == NULL) {
23,541,209✔
1369
      return terrno;
×
1370
    }
1371
  }
1372

1373
  pStmt->sql.siInfo.boundTags = pStmt->bInfo.boundTags;
11,725✔
1374

1375
  STMT2_TLOG("init stb interlace table info, tbName:%s, pDataCtx:%p, boundTags:%p", pStmt->bInfo.tbFName,
11,725✔
1376
             pStmt->sql.siInfo.pDataCtx, pStmt->sql.siInfo.boundTags);
1377

1378
  return TSDB_CODE_SUCCESS;
11,780✔
1379
}
1380

1381
bool stmt2IsInsert(TAOS_STMT2* stmt) {
1,199,721✔
1382
  STscStmt2* pStmt = (STscStmt2*)stmt;
1,199,721✔
1383
  if (pStmt->sql.type) {
1,199,721✔
1384
    return (STMT_TYPE_INSERT == pStmt->sql.type || STMT_TYPE_MULTI_INSERT == pStmt->sql.type);
1,030,675✔
1385
  }
1386

1387
  return qIsInsertValuesSql(pStmt->sql.sqlStr, pStmt->sql.sqlLen);
169,230✔
1388
}
1389

1390
bool stmt2IsSelect(TAOS_STMT2* stmt) {
128,144✔
1391
  STscStmt2* pStmt = (STscStmt2*)stmt;
128,144✔
1392

1393
  if (pStmt->sql.type) {
128,144✔
1394
    return STMT_TYPE_QUERY == pStmt->sql.type;
×
1395
  }
1396
  return qIsSelectFromSql(pStmt->sql.sqlStr, pStmt->sql.sqlLen);
128,113✔
1397
}
1398

1399
int stmtSetTbName2(TAOS_STMT2* stmt, const char* tbName) {
1,045,884✔
1400
  STscStmt2* pStmt = (STscStmt2*)stmt;
1,045,884✔
1401

1402
  int64_t startUs = taosGetTimestampUs();
1,046,047✔
1403

1404
  STMT2_TLOG("start to set tbName:%s", tbName);
1,046,047✔
1405

1406
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
1,046,272✔
1407
    return pStmt->errCode;
×
1408
  }
1409

1410
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_SETTBNAME));
1,046,416✔
1411

1412
  int32_t insert = 0;
1,046,026✔
1413
  if (!stmt2IsInsert(stmt)) {
1,046,026✔
1414
    STMT2_ELOG_E("set tb name not available for no-insert statement");
×
1415
    STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
×
1416
  }
1417
  // process tbname
1418
  STMT_ERR_RET(stmtCreateRequest(pStmt));
1,046,025✔
1419

1420
  STMT_ERR_RET(qCreateSName2(&pStmt->bInfo.sname, tbName, pStmt->taos->acctId, pStmt->exec.pRequest->pDb,
1,045,778✔
1421
                             pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen));
1422
  STMT_ERR_RET(tNameExtractFullName(&pStmt->bInfo.sname, pStmt->bInfo.tbFName));
1,044,562✔
1423
  tstrncpy(pStmt->bInfo.tbName, (char*)tNameGetTableName(&pStmt->bInfo.sname), TSDB_TABLE_NAME_LEN);
1,045,106✔
1424

1425
  if (!pStmt->sql.stbInterlaceMode || NULL == pStmt->sql.siInfo.pDataCtx) {
1,044,612✔
1426
    STMT_ERR_RET(stmtGetFromCache(pStmt));
40,451✔
1427

1428
    if (pStmt->bInfo.needParse) {
39,615✔
1429
      STMT_ERR_RET(stmtParseSql(pStmt));
18,977✔
1430
      if (!pStmt->sql.autoCreateTbl) {
18,977✔
1431
        uint64_t uid, suid;
2,359✔
1432
        int32_t  vgId;
2,359✔
1433
        int8_t   tableType;
2,359✔
1434

1435
        int32_t code = stmtGetTableMetaAndValidate(pStmt, &uid, &suid, &vgId, &tableType);
5,355✔
1436
        if (code != TSDB_CODE_SUCCESS) {
5,355✔
1437
          return code;
776✔
1438
        }
1439
      }
1440
    }
1441

1442
  } else {
1443
    pStmt->exec.pRequest->requestId++;
1,004,695✔
1444
    pStmt->bInfo.needParse = false;
1,005,076✔
1445
  }
1446

1447
  if (pStmt->sql.stbInterlaceMode && NULL == pStmt->sql.siInfo.pDataCtx) {
1,043,759✔
1448
    STMT_ERR_RET(stmtInitStbInterlaceTableInfo(pStmt));
11,780✔
1449
  }
1450

1451
  int64_t startUs2 = taosGetTimestampUs();
1,044,044✔
1452
  pStmt->stat.setTbNameUs += startUs2 - startUs;
1,044,044✔
1453

1454
  return TSDB_CODE_SUCCESS;
1,044,245✔
1455
}
1456

1457
int stmtSetTbTags2(TAOS_STMT2* stmt, TAOS_STMT2_BIND* tags, SVCreateTbReq** pCreateTbReq) {
525,344✔
1458
  STscStmt2* pStmt = (STscStmt2*)stmt;
525,344✔
1459

1460
  STMT2_TLOG_E("start to set tbTags");
525,344✔
1461
  if (qDebugFlag & DEBUG_TRACE) {
525,344✔
1462
    (void)stmtPrintBindv(stmt, tags, -1, true);
×
1463
  }
1464

1465
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
525,308✔
1466
    return pStmt->errCode;
×
1467
  }
1468

1469
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_SETTAGS));
525,308✔
1470

1471
  if (pStmt->bInfo.needParse && pStmt->sql.runTimes && pStmt->sql.type > 0 &&
525,368✔
1472
      STMT_TYPE_MULTI_INSERT != pStmt->sql.type) {
×
1473
    pStmt->bInfo.needParse = false;
×
1474
  }
1475
  STMT_ERR_RET(stmtCreateRequest(pStmt));
525,368✔
1476

1477
  if (pStmt->bInfo.needParse) {
525,248✔
1478
    STMT_ERR_RET(stmtParseSql(pStmt));
388✔
1479
  }
1480
  if (pStmt->sql.stbInterlaceMode && NULL == pStmt->sql.siInfo.pDataCtx) {
525,248✔
1481
    STMT_ERR_RET(stmtInitStbInterlaceTableInfo(pStmt));
×
1482
  }
1483

1484
  SBoundColInfo* tags_info = (SBoundColInfo*)pStmt->bInfo.boundTags;
525,308✔
1485

1486
  STableDataCxt** pDataBlock = NULL;
525,308✔
1487
  if (pStmt->exec.pCurrBlock) {
525,308✔
1488
    pDataBlock = &pStmt->exec.pCurrBlock;
514,576✔
1489
  } else {
1490
    pDataBlock =
1491
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
10,732✔
1492
    if (NULL == pDataBlock) {
10,732✔
1493
      STMT2_ELOG("table %s not found in exec blockHash:%p", pStmt->bInfo.tbFName, pStmt->exec.pBlockHash);
×
1494
      STMT_ERR_RET(TSDB_CODE_TSC_STMT_CACHE_ERROR);
×
1495
    }
1496
    if (pStmt->sql.stbInterlaceMode && (*pDataBlock)->pData->pCreateTbReq) {
10,732✔
1497
      tdDestroySVCreateTbReq((*pDataBlock)->pData->pCreateTbReq);
388✔
1498
      taosMemoryFreeClear((*pDataBlock)->pData->pCreateTbReq);
388✔
1499
      (*pDataBlock)->pData->pCreateTbReq = NULL;
388✔
1500
    }
1501
  }
1502
  if (pStmt->bInfo.inExecCache && !pStmt->sql.autoCreateTbl) {
525,308✔
1503
    return TSDB_CODE_SUCCESS;
×
1504
  }
1505

1506
  STMT2_TLOG_E("start to bind stmt tag values");
525,308✔
1507

1508
  void* boundTags = NULL;
525,200✔
1509
  if (pStmt->sql.stbInterlaceMode) {
525,200✔
1510
    boundTags = pStmt->sql.siInfo.boundTags;
501,980✔
1511
    *pCreateTbReq = taosMemoryCalloc(1, sizeof(SVCreateTbReq));
501,980✔
1512
    if (NULL == pCreateTbReq) {
502,172✔
1513
      return terrno;
×
1514
    }
1515
    int32_t vgId = -1;
502,172✔
1516
    STMT_ERR_RET(stmtTryAddTableVgroupInfo(pStmt, &vgId));
502,172✔
1517
    (*pCreateTbReq)->uid = vgId;
502,196✔
1518
  } else {
1519
    boundTags = pStmt->bInfo.boundTags;
23,220✔
1520
  }
1521

1522
  STMT_ERR_RET(qBindStmtTagsValue2(*pDataBlock, boundTags, pStmt->bInfo.tbSuid, pStmt->bInfo.stbFName,
525,416✔
1523
                                   pStmt->bInfo.sname.tname, tags, pStmt->exec.pRequest->msgBuf,
1524
                                   pStmt->exec.pRequest->msgBufLen, pStmt->taos->optionInfo.charsetCxt, *pCreateTbReq));
1525

1526
  return TSDB_CODE_SUCCESS;
525,186✔
1527
}
1528

1529
int stmtCheckTags2(TAOS_STMT2* stmt, SVCreateTbReq** pCreateTbReq) {
109,760✔
1530
  STscStmt2* pStmt = (STscStmt2*)stmt;
109,760✔
1531

1532
  STMT2_TLOG_E("start to clone createTbRequest for fixed tags");
109,760✔
1533

1534
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
109,812✔
1535
    return pStmt->errCode;
×
1536
  }
1537

1538
  if (!pStmt->sql.stbInterlaceMode) {
109,812✔
1539
    return TSDB_CODE_SUCCESS;
×
1540
  }
1541

1542
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_SETTAGS));
109,812✔
1543

1544
  if (pStmt->sql.fixValueTags) {
109,799✔
1545
    STMT2_TLOG_E("tags are fixed, use one createTbReq");
106,521✔
1546
    STMT_ERR_RET(cloneSVreateTbReq(pStmt->sql.fixValueTbReq, pCreateTbReq));
106,521✔
1547
    if ((*pCreateTbReq)->name) {
106,521✔
1548
      taosMemoryFree((*pCreateTbReq)->name);
106,521✔
1549
    }
1550
    (*pCreateTbReq)->name = taosStrdup(pStmt->bInfo.tbName);
106,456✔
1551
    int32_t vgId = -1;
106,534✔
1552
    STMT_ERR_RET(stmtTryAddTableVgroupInfo(pStmt, &vgId));
106,534✔
1553
    (*pCreateTbReq)->uid = vgId;
106,521✔
1554
    return TSDB_CODE_SUCCESS;
106,521✔
1555
  }
1556

1557
  STMT_ERR_RET(stmtCreateRequest(pStmt));
3,278✔
1558
  if (pStmt->bInfo.needParse) {
3,278✔
1559
    STMT_ERR_RET(stmtParseSql(pStmt));
×
1560
    if (!pStmt->sql.autoCreateTbl) {
×
1561
      STMT2_WLOG_E("don't need to create table, will not check tags");
×
1562
      return TSDB_CODE_SUCCESS;
×
1563
    }
1564
  }
1565

1566
  if (pStmt->sql.stbInterlaceMode && NULL == pStmt->sql.siInfo.pDataCtx) {
3,278✔
1567
    STMT_ERR_RET(stmtInitStbInterlaceTableInfo(pStmt));
×
1568
  }
1569

1570
  STableDataCxt** pDataBlock = NULL;
3,278✔
1571
  if (pStmt->exec.pCurrBlock) {
3,278✔
1572
    pDataBlock = &pStmt->exec.pCurrBlock;
×
1573
  } else {
1574
    pDataBlock =
1575
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
3,278✔
1576
    if (NULL == pDataBlock) {
3,278✔
1577
      STMT2_ELOG("table %s not found in exec blockHash:%p", pStmt->bInfo.tbFName, pStmt->exec.pBlockHash);
×
1578
      STMT_ERR_RET(TSDB_CODE_TSC_STMT_CACHE_ERROR);
×
1579
    }
1580
  }
1581

1582
  if (!((*pDataBlock)->pData->flags & SUBMIT_REQ_AUTO_CREATE_TABLE)) {
3,278✔
1583
    STMT2_DLOG_E("don't need to create, will not check tags");
×
1584
    return TSDB_CODE_SUCCESS;
×
1585
  }
1586

1587

1588
  if ((*pDataBlock)->pData->pCreateTbReq) {
3,278✔
1589
    STMT2_TLOG_E("tags are fixed, set createTbReq first time");
3,278✔
1590
    pStmt->sql.fixValueTags = true;
3,278✔
1591
    STMT_ERR_RET(cloneSVreateTbReq((*pDataBlock)->pData->pCreateTbReq, &pStmt->sql.fixValueTbReq));
3,278✔
1592
    STMT_ERR_RET(cloneSVreateTbReq(pStmt->sql.fixValueTbReq, pCreateTbReq));
3,278✔
1593
    (*pCreateTbReq)->uid = (*pDataBlock)->pMeta->vgId;
3,278✔
1594

1595
    // destroy the createTbReq in the data block
1596
    tdDestroySVCreateTbReq((*pDataBlock)->pData->pCreateTbReq);
3,278✔
1597
    taosMemoryFreeClear((*pDataBlock)->pData->pCreateTbReq);
3,278✔
1598
    (*pDataBlock)->pData->pCreateTbReq = NULL;
3,278✔
1599
  }
1600

1601
  return TSDB_CODE_SUCCESS;
3,278✔
1602
}
1603

1604
static int stmtFetchColFields2(STscStmt2* pStmt, int32_t* fieldNum, TAOS_FIELD_E** fields) {
×
1605
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
×
1606
    return pStmt->errCode;
×
1607
  }
1608

1609
  if (STMT_TYPE_QUERY == pStmt->sql.type) {
×
1610
    tscError("invalid operation to get query column fileds");
×
1611
    STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
×
1612
  }
1613

1614
  STableDataCxt** pDataBlock = NULL;
×
1615

1616
  if (pStmt->sql.stbInterlaceMode) {
×
1617
    pDataBlock = &pStmt->sql.siInfo.pDataCtx;
×
1618
  } else {
1619
    pDataBlock =
1620
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
×
1621
    if (NULL == pDataBlock) {
×
1622
      STMT2_ELOG("table %s not found in exec blockHash:%p", pStmt->bInfo.tbFName, pStmt->exec.pBlockHash);
×
1623
      STMT_ERR_RET(TSDB_CODE_APP_ERROR);
×
1624
    }
1625
  }
1626

1627
  STMT_ERR_RET(qBuildStmtColFields(*pDataBlock, fieldNum, fields));
×
1628

1629
  return TSDB_CODE_SUCCESS;
×
1630
}
1631

1632
static int stmtFetchStbColFields2(STscStmt2* pStmt, int32_t* fieldNum, TAOS_FIELD_ALL** fields) {
10,495✔
1633
  int32_t code = 0;
10,495✔
1634
  int32_t preCode = pStmt->errCode;
10,495✔
1635

1636
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
10,495✔
1637
    return pStmt->errCode;
×
1638
  }
1639

1640
  if (STMT_TYPE_QUERY == pStmt->sql.type) {
10,495✔
1641
    STMT2_ELOG_E("stmtFetchStbColFields2 only for insert statement");
×
1642
    STMT_ERRI_JRET(TSDB_CODE_TSC_STMT_API_ERROR);
×
1643
  }
1644

1645
  STableDataCxt** pDataBlock = NULL;
10,495✔
1646
  bool            cleanStb = false;
10,495✔
1647

1648
  if (pStmt->sql.stbInterlaceMode && pStmt->sql.siInfo.pDataCtx != NULL) {
10,495✔
1649
    pDataBlock = &pStmt->sql.siInfo.pDataCtx;
1,164✔
1650
  } else {
1651
    cleanStb = true;
9,331✔
1652
    pDataBlock =
1653
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
9,331✔
1654
  }
1655

1656
  if (NULL == pDataBlock || NULL == *pDataBlock) {
10,495✔
1657
    STMT2_ELOG("table %s not found in exec blockHash:%p", pStmt->bInfo.tbFName, pStmt->exec.pBlockHash);
×
1658
    STMT_ERRI_JRET(TSDB_CODE_APP_ERROR);
×
1659
  }
1660

1661
  pStmt->sql.placeholderOfTags = 0;
10,495✔
1662
  pStmt->sql.placeholderOfCols = 0;
10,495✔
1663
  int32_t totalNum = 0;
10,495✔
1664
  STMT_ERRI_JRET(qBuildStmtStbColFields(*pDataBlock, pStmt->bInfo.boundTags, pStmt->bInfo.fixedValueCols,
10,495✔
1665
                                        pStmt->bInfo.tbNameFlag, &totalNum, fields, &pStmt->sql.placeholderOfTags,
1666
                                        &pStmt->sql.placeholderOfCols));
1667

1668
  if (pStmt->bInfo.tbType == TSDB_SUPER_TABLE && cleanStb) {
10,495✔
1669
    taosMemoryFreeClear((*pDataBlock)->boundColsInfo.pColIndex);
7,391✔
1670
    qDestroyStmtDataBlock(*pDataBlock);
7,391✔
1671
    *pDataBlock = NULL;
7,391✔
1672
    if (taosHashRemove(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName)) != 0) {
7,391✔
1673
      STMT2_ELOG("fail to remove remove stb:%s exec blockHash", pStmt->bInfo.tbFName);
×
1674
      STMT_ERRI_JRET(TSDB_CODE_APP_ERROR);
×
1675
    }
1676
    pStmt->sql.autoCreateTbl = false;
7,391✔
1677
    pStmt->bInfo.tagsCached = false;
7,391✔
1678
    pStmt->bInfo.sname = (SName){0};
7,391✔
1679
    STMT_ERR_RET(stmtCleanBindInfo(pStmt));
7,391✔
1680
  }
1681

1682
  if (fieldNum != NULL) {
10,495✔
1683
    *fieldNum = totalNum;
10,495✔
1684
  }
1685

1686
  STMT2_DLOG("get insert fields totalNum:%d, tagsNum:%d, colsNum:%d", totalNum, pStmt->sql.placeholderOfTags,
10,495✔
1687
             pStmt->sql.placeholderOfCols);
1688

1689
_return:
10,495✔
1690

1691
  pStmt->errCode = preCode;
10,495✔
1692

1693
  return code;
10,495✔
1694
}
1695
/*
1696
SArray* stmtGetFreeCol(STscStmt2* pStmt, int32_t* idx) {
1697
  while (true) {
1698
    if (pStmt->exec.smInfo.pColIdx >= STMT_COL_BUF_SIZE) {
1699
      pStmt->exec.smInfo.pColIdx = 0;
1700
    }
1701

1702
    if ((pStmt->exec.smInfo.pColIdx + 1) == atomic_load_32(&pStmt->exec.smInfo.pColFreeIdx)) {
1703
      taosUsleep(1);
1704
      continue;
1705
    }
1706

1707
    *idx = pStmt->exec.smInfo.pColIdx;
1708
    return pStmt->exec.smInfo.pCols[pStmt->exec.smInfo.pColIdx++];
1709
  }
1710
}
1711
*/
1712
static int32_t stmtAppendTablePostHandle(STscStmt2* pStmt, SStmtQNode* param) {
1,016,709✔
1713
  if (NULL == pStmt->sql.siInfo.pVgroupHash) {
1,016,709✔
1714
    pStmt->sql.siInfo.pVgroupHash =
494,743✔
1715
        taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
494,632✔
1716
  }
1717
  if (NULL == pStmt->sql.siInfo.pVgroupList) {
1,017,091✔
1718
    pStmt->sql.siInfo.pVgroupList = taosArrayInit(64, POINTER_BYTES);
494,817✔
1719
  }
1720

1721
  if (NULL == pStmt->sql.siInfo.pRequest) {
1,017,017✔
1722
    STMT_ERR_RET(buildRequest(pStmt->taos->id, pStmt->sql.sqlStr, pStmt->sql.sqlLen, NULL, false,
11,392✔
1723
                              (SRequestObj**)&pStmt->sql.siInfo.pRequest, pStmt->reqid));
1724

1725
    if (pStmt->reqid != 0) {
11,392✔
1726
      pStmt->reqid++;
8✔
1727
    }
1728
    pStmt->exec.pRequest->syncQuery = true;
11,392✔
1729

1730
    pStmt->sql.siInfo.requestId = ((SRequestObj*)pStmt->sql.siInfo.pRequest)->requestId;
11,392✔
1731
    pStmt->sql.siInfo.requestSelf = ((SRequestObj*)pStmt->sql.siInfo.pRequest)->self;
11,392✔
1732
  }
1733

1734
  if (!pStmt->sql.siInfo.tbFromHash && pStmt->sql.siInfo.firstName[0] &&
1,017,013✔
1735
      0 == strcmp(pStmt->sql.siInfo.firstName, pStmt->bInfo.tbName)) {
27,562✔
1736
    pStmt->sql.siInfo.tbFromHash = true;
4,367✔
1737
  }
1738

1739
  if (0 == pStmt->sql.siInfo.firstName[0]) {
1,016,976✔
1740
    tstrncpy(pStmt->sql.siInfo.firstName, pStmt->bInfo.tbName, TSDB_TABLE_NAME_LEN);
11,392✔
1741
  }
1742

1743
  param->tblData.getFromHash = pStmt->sql.siInfo.tbFromHash;
1,016,972✔
1744
  param->next = NULL;
1,016,976✔
1745

1746
  (void)atomic_add_fetch_64(&pStmt->sql.siInfo.tbRemainNum, 1);
1,017,091✔
1747

1748
  if (pStmt->queue.stopQueue) {
1,017,265✔
1749
    STMT2_ELOG_E("bind thread already stopped, cannot enqueue");
×
1750
    return TSDB_CODE_TSC_STMT_API_ERROR;
×
1751
  }
1752
  stmtEnqueue(pStmt, param);
1,017,191✔
1753

1754
  return TSDB_CODE_SUCCESS;
1,016,982✔
1755
}
1756

1757
static FORCE_INLINE int32_t stmtGetTableColsFromCache(STscStmt2* pStmt, SArray** pTableCols) {
1758
  while (true) {
1759
    if (pStmt->sql.siInfo.pTableColsIdx < taosArrayGetSize(pStmt->sql.siInfo.pTableCols)) {
1,016,769✔
1760
      *pTableCols = (SArray*)taosArrayGetP(pStmt->sql.siInfo.pTableCols, pStmt->sql.siInfo.pTableColsIdx++);
1,017,186✔
1761
      break;
1,017,019✔
1762
    } else {
1763
      SArray* pTblCols = NULL;
×
1764
      for (int32_t i = 0; i < 100; i++) {
×
1765
        pTblCols = taosArrayInit(20, POINTER_BYTES);
×
1766
        if (NULL == pTblCols) {
×
1767
          return terrno;
×
1768
        }
1769

1770
        if (taosArrayPush(pStmt->sql.siInfo.pTableCols, &pTblCols) == NULL) {
×
1771
          return terrno;
×
1772
        }
1773
      }
1774
    }
1775
  }
1776

1777
  return TSDB_CODE_SUCCESS;
1,017,019✔
1778
}
1779

1780
static int32_t stmtCacheBlock(STscStmt2* pStmt) {
42,329,697✔
1781
  if (pStmt->sql.type != STMT_TYPE_MULTI_INSERT) {
42,329,697✔
1782
    return TSDB_CODE_SUCCESS;
42,513,852✔
1783
  }
1784

1785
  uint64_t uid = pStmt->bInfo.tbUid;
26,384✔
1786
  uint64_t cacheUid = (TSDB_CHILD_TABLE == pStmt->bInfo.tbType) ? pStmt->bInfo.tbSuid : uid;
26,384✔
1787

1788
  if (taosHashGet(pStmt->sql.pTableCache, &cacheUid, sizeof(cacheUid))) {
26,384✔
1789
    STMT2_TLOG("table %s already cached, no need to cache again", pStmt->bInfo.tbFName);
20,485✔
1790
    return TSDB_CODE_SUCCESS;
20,485✔
1791
  }
1792

1793
  STableDataCxt** pSrc = taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
6,033✔
1794
  if (!pSrc) {
6,033✔
1795
    STMT2_ELOG("table %s not found in exec blockHash:%p", pStmt->bInfo.tbFName, pStmt->exec.pBlockHash);
×
1796
    return terrno;
×
1797
  }
1798
  STableDataCxt* pDst = NULL;
6,033✔
1799

1800
  STMT_ERR_RET(qCloneStmtDataBlock(&pDst, *pSrc, true));
6,033✔
1801

1802
  SStmtTableCache cache = {
6,033✔
1803
      .pDataCtx = pDst,
1804
      .boundTags = pStmt->bInfo.boundTags,
6,033✔
1805
  };
1806

1807
  if (taosHashPut(pStmt->sql.pTableCache, &cacheUid, sizeof(cacheUid), &cache, sizeof(cache))) {
6,033✔
1808
    STMT2_ELOG("fail to put table cache:%s", tstrerror(terrno));
×
1809
    return terrno;
×
1810
  }
1811

1812
  if (pStmt->sql.autoCreateTbl) {
6,033✔
1813
    pStmt->bInfo.tagsCached = true;
5,063✔
1814
  } else {
1815
    pStmt->bInfo.boundTags = NULL;
970✔
1816
  }
1817

1818
  return TSDB_CODE_SUCCESS;
6,033✔
1819
}
1820

1821
static int stmtAddBatch2(TAOS_STMT2* stmt) {
42,890,207✔
1822
  STscStmt2* pStmt = (STscStmt2*)stmt;
42,890,207✔
1823

1824
  int64_t startUs = taosGetTimestampUs();
43,187,644✔
1825

1826
  // STMT2_TLOG_E("start to add batch");
1827

1828
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
43,187,644✔
1829
    return pStmt->errCode;
×
1830
  }
1831

1832
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_ADD_BATCH));
43,242,507✔
1833

1834
  if (pStmt->sql.stbInterlaceMode) {
43,172,022✔
1835
    int64_t startUs2 = taosGetTimestampUs();
494,298✔
1836
    pStmt->stat.addBatchUs += startUs2 - startUs;
494,298✔
1837

1838
    pStmt->sql.siInfo.tableColsReady = false;
494,261✔
1839

1840
    SStmtQNode* param = NULL;
494,224✔
1841
    STMT_ERR_RET(stmtAllocQNodeFromBuf(&pStmt->sql.siInfo.tbBuf, (void**)&param));
988,287✔
1842
    param->restoreTbCols = true;
494,063✔
1843
    param->next = NULL;
494,100✔
1844

1845
    if (pStmt->sql.autoCreateTbl) {
494,211✔
1846
      pStmt->bInfo.tagsCached = true;
211,743✔
1847
    }
1848
    pStmt->bInfo.boundColsCached = true;
494,063✔
1849

1850
    if (pStmt->queue.stopQueue) {
494,174✔
1851
      STMT2_ELOG_E("stmt bind thread is stopped,cannot enqueue bind request");
×
1852
      return TSDB_CODE_TSC_STMT_API_ERROR;
×
1853
    }
1854

1855
    stmtEnqueue(pStmt, param);
494,100✔
1856

1857
    return TSDB_CODE_SUCCESS;
494,297✔
1858
  }
1859

1860
  STMT_ERR_RET(stmtCacheBlock(pStmt));
42,712,384✔
1861

1862
  return TSDB_CODE_SUCCESS;
42,562,936✔
1863
}
1864
/*
1865
static int32_t stmtBackupQueryFields(STscStmt2* pStmt) {
1866
  SStmtQueryResInfo* pRes = &pStmt->sql.queryRes;
1867
  pRes->numOfCols = pStmt->exec.pRequest->body.resInfo.numOfCols;
1868
  pRes->precision = pStmt->exec.pRequest->body.resInfo.precision;
1869

1870
  int32_t size = pRes->numOfCols * sizeof(TAOS_FIELD);
1871
  pRes->fields = taosMemoryMalloc(size);
1872
  pRes->userFields = taosMemoryMalloc(size);
1873
  if (NULL == pRes->fields || NULL == pRes->userFields) {
1874
    STMT_ERR_RET(terrno);
1875
  }
1876
  (void)memcpy(pRes->fields, pStmt->exec.pRequest->body.resInfo.fields, size);
1877
  (void)memcpy(pRes->userFields, pStmt->exec.pRequest->body.resInfo.userFields, size);
1878

1879
  return TSDB_CODE_SUCCESS;
1880
}
1881

1882
static int32_t stmtRestoreQueryFields(STscStmt2* pStmt) {
1883
  SStmtQueryResInfo* pRes = &pStmt->sql.queryRes;
1884
  int32_t            size = pRes->numOfCols * sizeof(TAOS_FIELD);
1885

1886
  pStmt->exec.pRequest->body.resInfo.numOfCols = pRes->numOfCols;
1887
  pStmt->exec.pRequest->body.resInfo.precision = pRes->precision;
1888

1889
  if (NULL == pStmt->exec.pRequest->body.resInfo.fields) {
1890
    pStmt->exec.pRequest->body.resInfo.fields = taosMemoryMalloc(size);
1891
    if (NULL == pStmt->exec.pRequest->body.resInfo.fields) {
1892
      STMT_ERR_RET(terrno);
1893
    }
1894
    (void)memcpy(pStmt->exec.pRequest->body.resInfo.fields, pRes->fields, size);
1895
  }
1896

1897
  if (NULL == pStmt->exec.pRequest->body.resInfo.userFields) {
1898
    pStmt->exec.pRequest->body.resInfo.userFields = taosMemoryMalloc(size);
1899
    if (NULL == pStmt->exec.pRequest->body.resInfo.userFields) {
1900
      STMT_ERR_RET(terrno);
1901
    }
1902
    (void)memcpy(pStmt->exec.pRequest->body.resInfo.userFields, pRes->userFields, size);
1903
  }
1904

1905
  return TSDB_CODE_SUCCESS;
1906
}
1907
*/
1908

1909
/**
1910
 * Fetch metadata for query statement after parameter binding.
1911
 * This function collects metadata requirements from the query (after binding),
1912
 * fetches metadata synchronously from catalog, and returns it for parsing.
1913
 *
1914
 * Note: We fetch metadata on every bind because:
1915
 * 1. Parameter values in WHERE conditions (e.g., dataname IN (?,?)) may change
1916
 * 2. Different parameter values may require different vgroup lists for virtual tables
1917
 * 3. Metadata requirements can only be determined after parameters are bound
1918
 *
1919
 * @param pStmt Statement handle
1920
 * @param pCxt Parse context (must have catalog handle initialized)
1921
 * @param pMetaData Output: Fetched metadata (caller responsible for cleanup)
1922
 * @return TSDB_CODE_SUCCESS on success, error code otherwise
1923
 */
1924
// Callback parameter structure for synchronous catalog metadata fetch
1925
typedef struct {
1926
  SMetaData* pRsp;
1927
  int32_t    code;
1928
  tsem_t     sem;
1929
} SCatalogSyncCbParam;
1930

1931
// Callback function for catalogAsyncGetAllMeta to make it synchronous
1932
static void stmtCatalogSyncGetAllMetaCb(SMetaData* pResultMeta, void* param, int32_t code) {
7,574✔
1933
  SCatalogSyncCbParam* pCbParam = (SCatalogSyncCbParam*)param;
7,574✔
1934
  if (TSDB_CODE_SUCCESS == code && pResultMeta) {
7,574✔
1935
    *pCbParam->pRsp = *pResultMeta;
7,574✔
1936
    TAOS_MEMSET(pResultMeta, 0, sizeof(SMetaData));  // Clear to avoid double free
7,574✔
1937
  }
1938
  pCbParam->code = code;
7,574✔
1939
  if (tsem_post(&pCbParam->sem) != 0) {
7,574✔
1940
    tscError("failed to post semaphore");
×
1941
  }
1942
}
7,574✔
1943

1944
static int32_t stmtFetchMetadataForQuery(STscStmt2* pStmt, SParseContext* pCxt, SMetaData* pMetaData) {
7,574✔
1945
  int32_t          code = 0;
7,574✔
1946
  SParseMetaCache  metaCache = {0};
7,574✔
1947
  SCatalogReq      catalogReq = {0};
7,574✔
1948
  SRequestConnInfo conn = {.pTrans = pCxt->pTransporter,
7,574✔
1949
                           .requestId = pCxt->requestId,
7,574✔
1950
                           .requestObjRefId = pCxt->requestRid,
7,574✔
1951
                           .mgmtEps = pCxt->mgmtEpSet};
1952

1953
  TAOS_MEMSET(pMetaData, 0, sizeof(SMetaData));
7,574✔
1954

1955
  code = collectMetaKey(pCxt, pStmt->sql.pQuery, &metaCache);
7,574✔
1956
  if (TSDB_CODE_SUCCESS == code) {
7,574✔
1957
    code = buildCatalogReq(&metaCache, &catalogReq);
7,574✔
1958
  }
1959
  if (TSDB_CODE_SUCCESS == code) {
7,574✔
1960
    SCatalogSyncCbParam cbParam = {.pRsp = pMetaData, .code = TSDB_CODE_SUCCESS};
7,574✔
1961
    if (tsem_init(&cbParam.sem, 0, 0) != 0) {
7,574✔
1962
      code = TSDB_CODE_CTG_INTERNAL_ERROR;
×
1963
    } else {
1964
      code = catalogAsyncGetAllMeta(pCxt->pCatalog, &conn, &catalogReq, stmtCatalogSyncGetAllMetaCb, &cbParam, NULL);
7,574✔
1965
      if (TSDB_CODE_SUCCESS == code) {
7,574✔
1966
        code = tsem_wait(&cbParam.sem);
7,574✔
1967
        if (code != TSDB_CODE_SUCCESS) {
7,574✔
1968
          catalogFreeMetaData(pMetaData);
×
1969
          TAOS_MEMSET(pMetaData, 0, sizeof(SMetaData));
×
1970
        } else {
1971
          code = cbParam.code;
7,574✔
1972
        }
1973
      }
1974

1975
      if (tsem_destroy(&cbParam.sem) != 0) {
7,574✔
1976
        tscError("failed to destroy semaphore");
×
1977
        code = TSDB_CODE_CTG_INTERNAL_ERROR;
×
1978
        catalogFreeMetaData(pMetaData);
×
1979
        TAOS_MEMSET(pMetaData, 0, sizeof(SMetaData));
×
1980
      }
1981
    }
1982
  }
1983

1984
  destoryParseMetaCache(&metaCache, false);
7,574✔
1985
  destoryCatalogReq(&catalogReq);
7,574✔
1986

1987
  if (TSDB_CODE_SUCCESS != code) {
7,574✔
1988
    catalogFreeMetaData(pMetaData);
×
1989
  }
1990

1991
  return code;
7,574✔
1992
}
1993

1994
int stmtBindBatch2(TAOS_STMT2* stmt, TAOS_STMT2_BIND* bind, int32_t colIdx, SVCreateTbReq* pCreateTbReq) {
43,383,324✔
1995
  STscStmt2* pStmt = (STscStmt2*)stmt;
43,383,324✔
1996
  int32_t    code = 0;
43,383,324✔
1997

1998
  int64_t startUs = taosGetTimestampUs();
43,625,750✔
1999

2000
  if (qDebugFlag & DEBUG_TRACE) {
43,625,750✔
2001
    (void)stmtPrintBindv(stmt, bind, colIdx, false);
×
2002
  }
2003

2004
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
43,611,264✔
2005
    return pStmt->errCode;
194✔
2006
  }
2007

2008
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_BIND));
43,722,968✔
2009

2010
  if (pStmt->bInfo.needParse && pStmt->sql.runTimes && pStmt->sql.type > 0 &&
43,848,967✔
2011
      STMT_TYPE_MULTI_INSERT != pStmt->sql.type) {
×
2012
    pStmt->bInfo.needParse = false;
×
2013
  }
2014

2015
  if (pStmt->exec.pRequest && STMT_TYPE_QUERY == pStmt->sql.type && pStmt->sql.runTimes) {
43,542,232✔
2016
    resetRequest(pStmt);
1,746✔
2017
  }
2018

2019
  STMT_ERR_RET(stmtCreateRequest(pStmt));
43,484,770✔
2020
  if (pStmt->bInfo.needParse) {
43,447,137✔
2021
    code = stmtParseSql(pStmt);
105,267✔
2022
    if (code != TSDB_CODE_SUCCESS) {
104,954✔
2023
      goto cleanup_root;
×
2024
    }
2025
  }
2026

2027
  if (STMT_TYPE_QUERY == pStmt->sql.type) {
43,388,542✔
2028
    code = qStmtBindParams2(pStmt->sql.pQuery, bind, colIdx, pStmt->taos->optionInfo.charsetCxt);
7,574✔
2029
    if (code != TSDB_CODE_SUCCESS) {
7,574✔
2030
      goto cleanup_root;
×
2031
    }
2032
    SParseContext ctx = {.requestId = pStmt->exec.pRequest->requestId,
7,574✔
2033
                         .acctId = pStmt->taos->acctId,
7,574✔
2034
                         .db = pStmt->exec.pRequest->pDb,
7,574✔
2035
                         .topicQuery = false,
2036
                         .pSql = pStmt->sql.sqlStr,
7,574✔
2037
                         .sqlLen = pStmt->sql.sqlLen,
7,574✔
2038
                         .pMsg = pStmt->exec.pRequest->msgBuf,
7,574✔
2039
                         .msgLen = ERROR_MSG_BUF_DEFAULT_SIZE,
2040
                         .pTransporter = pStmt->taos->pAppInfo->pTransporter,
7,574✔
2041
                         .pStmtCb = NULL,
2042
                         .pUser = pStmt->taos->user,
7,574✔
2043
                         .stmtBindVersion = pStmt->exec.pRequest->stmtBindVersion};
7,574✔
2044
    ctx.mgmtEpSet = getEpSet_s(&pStmt->taos->pAppInfo->mgmtEp);
7,574✔
2045
    code = catalogGetHandle(pStmt->taos->pAppInfo->clusterId, &ctx.pCatalog);
7,574✔
2046
    if (code != TSDB_CODE_SUCCESS) {
7,574✔
2047
      goto cleanup_root;
×
2048
    }
2049

2050
    // Fetch metadata for query(vtable need)
2051
    SMetaData metaData = {0};
7,574✔
2052
    code = stmtFetchMetadataForQuery(pStmt, &ctx, &metaData);
7,574✔
2053
    if (TSDB_CODE_SUCCESS != code) {
7,574✔
2054
      goto cleanup_root;
×
2055
    }
2056

2057
    code = qStmtParseQuerySql(&ctx, pStmt->sql.pQuery, &metaData);
7,574✔
2058
    if (TSDB_CODE_SUCCESS == code) {
7,574✔
2059
      // Copy metaData to pRequest->parseMeta for potential future use
2060
      // Similar to doAsyncQueryFromAnalyse when parseOnly is true
2061
      (void)memcpy(&pStmt->exec.pRequest->parseMeta, &metaData, sizeof(SMetaData));
7,380✔
2062
      (void)memset(&metaData, 0, sizeof(SMetaData));  // Clear to avoid double free
7,380✔
2063
    } else {
2064
      // Clean up metaData on failure - free all arrays
2065
      if (metaData.pVStbRefDbs) {
194✔
2066
        taosArrayDestroy(metaData.pVStbRefDbs);
194✔
2067
        metaData.pVStbRefDbs = NULL;
194✔
2068
      }
2069
      // Note: Other fields in metaData are managed by catalog module if ctgFree is true
2070
      goto cleanup_root;
194✔
2071
    }
2072

2073
    if (pStmt->sql.pQuery->haveResultSet) {
7,380✔
2074
      STMT_ERR_RET(setResSchemaInfo(&pStmt->exec.pRequest->body.resInfo, pStmt->sql.pQuery->pResSchema,
14,954✔
2075
                                    pStmt->sql.pQuery->numOfResCols, pStmt->sql.pQuery->pResExtSchema, true));
2076
      taosMemoryFreeClear(pStmt->sql.pQuery->pResSchema);
7,380✔
2077
      taosMemoryFreeClear(pStmt->sql.pQuery->pResExtSchema);
7,380✔
2078
      setResPrecision(&pStmt->exec.pRequest->body.resInfo, pStmt->sql.pQuery->precision);
7,380✔
2079
    }
2080

2081
    TSWAP(pStmt->exec.pRequest->dbList, pStmt->sql.pQuery->pDbList);
7,380✔
2082
    TSWAP(pStmt->exec.pRequest->tableList, pStmt->sql.pQuery->pTableList);
7,380✔
2083
    TSWAP(pStmt->exec.pRequest->targetTableList, pStmt->sql.pQuery->pTargetTableList);
7,380✔
2084

2085
    // if (STMT_TYPE_QUERY == pStmt->sql.queryRes) {
2086
    //   STMT_ERR_RET(stmtRestoreQueryFields(pStmt));
2087
    // }
2088

2089
    // STMT_ERR_RET(stmtBackupQueryFields(pStmt));
2090

2091
    return TSDB_CODE_SUCCESS;
7,380✔
2092

2093
  cleanup_root:
194✔
2094
    STMT2_ELOG("parse query statment unexpected failed code:%d, need to clean node", code);
194✔
2095
    if (pStmt->sql.pQuery && pStmt->sql.pQuery->pRoot) {
194✔
2096
      nodesDestroyNode(pStmt->sql.pQuery->pRoot);
194✔
2097
      pStmt->sql.pQuery->pRoot = NULL;
194✔
2098
    }
2099
    STMT_ERR_RET(code);
194✔
2100
  }
2101

2102
  if (pStmt->sql.stbInterlaceMode && NULL == pStmt->sql.siInfo.pDataCtx) {
43,481,327✔
2103
    STMT_ERR_RET(stmtInitStbInterlaceTableInfo(pStmt));
×
2104
  }
2105

2106
  STableDataCxt** pDataBlock = NULL;
42,819,938✔
2107

2108
  if (pStmt->exec.pCurrBlock) {
42,819,938✔
2109
    pDataBlock = &pStmt->exec.pCurrBlock;
43,577,362✔
2110
  } else {
2111
    pDataBlock =
2112
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
122,763✔
2113
    if (NULL == pDataBlock) {
123,656✔
2114
      STMT2_ELOG("table %s not found in exec blockHash:%p", pStmt->bInfo.tbFName, pStmt->exec.pBlockHash);
×
2115
      STMT_ERR_RET(TSDB_CODE_TSC_STMT_CACHE_ERROR);
×
2116
    }
2117
    pStmt->exec.pCurrBlock = *pDataBlock;
123,656✔
2118
    if (pStmt->sql.stbInterlaceMode) {
123,411✔
2119
      taosArrayDestroy(pStmt->exec.pCurrBlock->pData->aCol);
11,780✔
2120
      (*pDataBlock)->pData->aCol = NULL;
11,780✔
2121
    }
2122
    if (colIdx < -1) {
123,440✔
2123
      pStmt->sql.bindRowFormat = true;
194✔
2124
      taosArrayDestroy((*pDataBlock)->pData->aCol);
194✔
2125
      (*pDataBlock)->pData->aCol = taosArrayInit(20, POINTER_BYTES);
194✔
2126
    }
2127
  }
2128

2129
  int64_t startUs2 = taosGetTimestampUs();
43,679,222✔
2130
  pStmt->stat.bindDataUs1 += startUs2 - startUs;
43,679,222✔
2131

2132
  SStmtQNode* param = NULL;
43,766,002✔
2133
  if (pStmt->sql.stbInterlaceMode) {
43,636,403✔
2134
    STMT_ERR_RET(stmtAllocQNodeFromBuf(&pStmt->sql.siInfo.tbBuf, (void**)&param));
2,033,570✔
2135
    STMT_ERR_RET(stmtGetTableColsFromCache(pStmt, &param->tblData.aCol));
2,033,673✔
2136
    taosArrayClear(param->tblData.aCol);
1,017,019✔
2137

2138
    // param->tblData.aCol = taosArrayInit(20, POINTER_BYTES);
2139

2140
    param->restoreTbCols = false;
1,016,342✔
2141
    param->tblData.isOrdered = true;
1,016,383✔
2142
    param->tblData.isDuplicateTs = false;
1,016,309✔
2143
    tstrncpy(param->tblData.tbName, pStmt->bInfo.tbName, TSDB_TABLE_NAME_LEN);
1,016,431✔
2144

2145
    param->pCreateTbReq = pCreateTbReq;
1,016,164✔
2146
  }
2147

2148
  int64_t startUs3 = taosGetTimestampUs();
43,584,667✔
2149
  pStmt->stat.bindDataUs2 += startUs3 - startUs2;
43,584,667✔
2150

2151
  SArray*   pCols = pStmt->sql.stbInterlaceMode ? param->tblData.aCol : (*pDataBlock)->pData->aCol;
43,522,275✔
2152
  SBlobSet* pBlob = NULL;
43,783,826✔
2153
  if (colIdx < 0) {
43,814,195✔
2154
    if (pStmt->sql.stbInterlaceMode) {
43,881,459✔
2155
      (*pDataBlock)->pData->flags &= ~SUBMIT_REQ_COLUMN_DATA_FORMAT;
1,016,767✔
2156
      code = qBindStmtStbColsValue2(*pDataBlock, pCols, pStmt->bInfo.fixedValueCols, bind, pStmt->exec.pRequest->msgBuf,
1,298,603✔
2157
                                    pStmt->exec.pRequest->msgBufLen, &pStmt->sql.siInfo.pTSchema, pStmt->sql.pBindInfo,
1,017,137✔
2158
                                    pStmt->taos->optionInfo.charsetCxt, &pBlob);
1,017,174✔
2159
      param->tblData.isOrdered = (*pDataBlock)->ordered;
1,017,261✔
2160
      param->tblData.isDuplicateTs = (*pDataBlock)->duplicateTs;
1,017,051✔
2161
    } else {
2162
      if (colIdx == -1) {
42,870,972✔
2163
        if (pStmt->sql.bindRowFormat) {
42,473,612✔
2164
          STMT2_ELOG_E("can't mix bind row format and bind column format");
194✔
2165
          STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
194✔
2166
        }
2167
        code = qBindStmtColsValue2(*pDataBlock, pCols, pStmt->bInfo.fixedValueCols, bind, pStmt->exec.pRequest->msgBuf,
79,666,147✔
2168
                                   pStmt->exec.pRequest->msgBufLen, pStmt->taos->optionInfo.charsetCxt);
79,424,200✔
2169
      } else {
2170
        code =
2171
            qBindStmt2RowValue(*pDataBlock, (*pDataBlock)->pData->aRowP, pStmt->bInfo.fixedValueCols, bind,
388✔
2172
                               pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen,
388✔
2173
                               &pStmt->sql.siInfo.pTSchema, pStmt->sql.pBindInfo, pStmt->taos->optionInfo.charsetCxt);
420,806✔
2174
      }
2175
    }
2176

2177
    if (code) {
43,326,405✔
2178
      STMT2_ELOG("bind cols or rows failed, error:%s", tstrerror(code));
388✔
2179
      STMT_ERR_RET(code);
388✔
2180
    }
2181
  } else {
2182
    if (pStmt->sql.stbInterlaceMode) {
1,164✔
2183
      STMT2_ELOG_E("bind single column not allowed in stb insert mode");
×
2184
      STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
×
2185
    }
2186

2187
    if (pStmt->sql.bindRowFormat) {
1,164✔
2188
      STMT2_ELOG_E("can't mix bind row format and bind column format");
×
2189
      STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
×
2190
    }
2191

2192
    if (colIdx != (pStmt->bInfo.sBindLastIdx + 1) && colIdx != 0) {
1,164✔
2193
      STMT2_ELOG_E("bind column index not in sequence");
×
2194
      STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
×
2195
    }
2196

2197
    pStmt->bInfo.sBindLastIdx = colIdx;
1,164✔
2198

2199
    if (0 == colIdx) {
1,164✔
2200
      pStmt->bInfo.sBindRowNum = bind->num;
582✔
2201
    }
2202

2203
    code = qBindStmtSingleColValue2(*pDataBlock, pCols, bind, pStmt->exec.pRequest->msgBuf,
1,164✔
2204
                                    pStmt->exec.pRequest->msgBufLen, colIdx, pStmt->bInfo.sBindRowNum,
1,164✔
2205
                                    pStmt->taos->optionInfo.charsetCxt);
1,164✔
2206
    if (code) {
1,164✔
2207
      STMT2_ELOG("bind single col failed, error:%s", tstrerror(code));
×
2208
      STMT_ERR_RET(code);
×
2209
    }
2210
  }
2211

2212
  int64_t startUs4 = taosGetTimestampUs();
43,493,516✔
2213
  pStmt->stat.bindDataUs3 += startUs4 - startUs3;
43,493,516✔
2214

2215
  if (pStmt->stbInterlaceMode) {
43,672,734✔
2216
    if (param) param->tblData.pBlobSet = pBlob;
43,771,450✔
2217
  }
2218

2219
  if (pStmt->sql.stbInterlaceMode) {
43,740,429✔
2220
    STMT_ERR_RET(stmtAppendTablePostHandle(pStmt, param));
1,016,947✔
2221
  } else {
2222
    STMT_ERR_RET(stmtAddBatch2(pStmt));
42,721,249✔
2223
  }
2224

2225
  pStmt->stat.bindDataUs4 += taosGetTimestampUs() - startUs4;
43,643,365✔
2226
  return TSDB_CODE_SUCCESS;
43,772,075✔
2227
}
2228

2229
/*
2230
int stmtUpdateTableUid(STscStmt2* pStmt, SSubmitRsp* pRsp) {
2231
  tscDebug("stmt start to update tbUid, blockNum:%d", pRsp->nBlocks);
2232

2233
  int32_t code = 0;
2234
  int32_t finalCode = 0;
2235
  size_t  keyLen = 0;
2236
  void*   pIter = taosHashIterate(pStmt->exec.pBlockHash, NULL);
2237
  while (pIter) {
2238
    STableDataCxt* pBlock = *(STableDataCxt**)pIter;
2239
    char*          key = taosHashGetKey(pIter, &keyLen);
2240

2241
    STableMeta* pMeta = qGetTableMetaInDataBlock(pBlock);
2242
    if (pMeta->uid) {
2243
      pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
2244
      continue;
2245
    }
2246

2247
    SSubmitBlkRsp* blkRsp = NULL;
2248
    int32_t        i = 0;
2249
    for (; i < pRsp->nBlocks; ++i) {
2250
      blkRsp = pRsp->pBlocks + i;
2251
      if (strlen(blkRsp->tblFName) != keyLen) {
2252
        continue;
2253
      }
2254

2255
      if (strncmp(blkRsp->tblFName, key, keyLen)) {
2256
        continue;
2257
      }
2258

2259
      break;
2260
    }
2261

2262
    if (i < pRsp->nBlocks) {
2263
      tscDebug("auto created table %s uid updated from %" PRIx64 " to %" PRIx64, blkRsp->tblFName, pMeta->uid,
2264
               blkRsp->uid);
2265

2266
      pMeta->uid = blkRsp->uid;
2267
      pStmt->bInfo.tbUid = blkRsp->uid;
2268
    } else {
2269
      tscDebug("table %s not found in submit rsp, will update from catalog", pStmt->bInfo.tbFName);
2270
      if (NULL == pStmt->pCatalog) {
2271
        code = catalogGetHandle(pStmt->taos->pAppInfo->clusterId, &pStmt->pCatalog);
2272
        if (code) {
2273
          pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
2274
          finalCode = code;
2275
          continue;
2276
        }
2277
      }
2278

2279
      code = stmtCreateRequest(pStmt);
2280
      if (code) {
2281
        pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
2282
        finalCode = code;
2283
        continue;
2284
      }
2285

2286
      STableMeta*      pTableMeta = NULL;
2287
      SRequestConnInfo conn = {.pTrans = pStmt->taos->pAppInfo->pTransporter,
2288
                               .requestId = pStmt->exec.pRequest->requestId,
2289
                               .requestObjRefId = pStmt->exec.pRequest->self,
2290
                               .mgmtEps = getEpSet_s(&pStmt->taos->pAppInfo->mgmtEp)};
2291
      code = catalogGetTableMeta(pStmt->pCatalog, &conn, &pStmt->bInfo.sname, &pTableMeta);
2292

2293
      pStmt->stat.ctgGetTbMetaNum++;
2294

2295
      taos_free_result(pStmt->exec.pRequest);
2296
      pStmt->exec.pRequest = NULL;
2297

2298
      if (code || NULL == pTableMeta) {
2299
        pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
2300
        finalCode = code;
2301
        taosMemoryFree(pTableMeta);
2302
        continue;
2303
      }
2304

2305
      pMeta->uid = pTableMeta->uid;
2306
      pStmt->bInfo.tbUid = pTableMeta->uid;
2307
      taosMemoryFree(pTableMeta);
2308
    }
2309

2310
    pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
2311
  }
2312

2313
  return finalCode;
2314
}
2315
*/
2316
/*
2317
int stmtStaticModeExec(TAOS_STMT* stmt) {
2318
  STscStmt2*   pStmt = (STscStmt2*)stmt;
2319
  int32_t     code = 0;
2320
  SSubmitRsp* pRsp = NULL;
2321
  if (pStmt->sql.staticMode) {
2322
    return TSDB_CODE_TSC_STMT_API_ERROR;
2323
  }
2324

2325
  STMT_DLOG_E("start to exec");
2326

2327
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_EXECUTE));
2328

2329
  STMT_ERR_RET(qBuildStmtOutputFromTbList(pStmt->sql.pQuery, pStmt->sql.pVgHash, pStmt->exec.pTbBlkList,
2330
pStmt->exec.pCurrBlock, pStmt->exec.tbBlkNum));
2331

2332
  launchQueryImpl(pStmt->exec.pRequest, pStmt->sql.pQuery, true, NULL);
2333

2334
  if (pStmt->exec.pRequest->code && NEED_CLIENT_HANDLE_ERROR(pStmt->exec.pRequest->code)) {
2335
    code = refreshMeta(pStmt->exec.pRequest->pTscObj, pStmt->exec.pRequest);
2336
    if (code) {
2337
      pStmt->exec.pRequest->code = code;
2338
    } else {
2339
      tFreeSSubmitRsp(pRsp);
2340
      STMT_ERR_RET(stmtResetStmt(pStmt));
2341
      STMT_ERR_RET(TSDB_CODE_NEED_RETRY);
2342
    }
2343
  }
2344

2345
  STMT_ERR_JRET(pStmt->exec.pRequest->code);
2346

2347
  pStmt->exec.affectedRows = taos_affected_rows(pStmt->exec.pRequest);
2348
  pStmt->affectedRows += pStmt->exec.affectedRows;
2349

2350
_return:
2351

2352
  stmtCleanExecInfo(pStmt, (code ? false : true), false);
2353

2354
  tFreeSSubmitRsp(pRsp);
2355

2356
  ++pStmt->sql.runTimes;
2357

2358
  STMT_RET(code);
2359
}
2360
*/
2361

2362
static int32_t createParseContext(const SRequestObj* pRequest, SParseContext** pCxt, SSqlCallbackWrapper* pWrapper) {
4,850✔
2363
  const STscObj* pTscObj = pRequest->pTscObj;
4,850✔
2364

2365
  *pCxt = taosMemoryCalloc(1, sizeof(SParseContext));
4,850✔
2366
  if (*pCxt == NULL) {
4,850✔
2367
    return terrno;
×
2368
  }
2369

2370
  **pCxt = (SParseContext){.requestId = pRequest->requestId,
×
2371
                           .requestRid = pRequest->self,
4,850✔
2372
                           .acctId = pTscObj->acctId,
4,850✔
2373
                           .db = pRequest->pDb,
4,850✔
2374
                           .topicQuery = false,
2375
                           .pSql = pRequest->sqlstr,
4,850✔
2376
                           .sqlLen = pRequest->sqlLen,
4,850✔
2377
                           .pMsg = pRequest->msgBuf,
4,850✔
2378
                           .msgLen = ERROR_MSG_BUF_DEFAULT_SIZE,
2379
                           .pTransporter = pTscObj->pAppInfo->pTransporter,
4,850✔
2380
                           .pStmtCb = NULL,
2381
                           .pUser = pTscObj->user,
4,850✔
2382
                           .userId = pTscObj->userId,
4,850✔
2383
                           .pEffectiveUser = pRequest->effectiveUser,
4,850✔
2384
                           .isSuperUser = (0 == strcmp(pTscObj->user, TSDB_DEFAULT_USER)),
4,850✔
2385
                           .enableSysInfo = pTscObj->sysInfo,
4,850✔
2386
                           .privInfo = pWrapper->pParseCtx ? pWrapper->pParseCtx->privInfo : 0,
4,850✔
2387
                           .async = true,
2388
                           .svrVer = pTscObj->sVer,
4,850✔
2389
                           .nodeOffline = (pTscObj->pAppInfo->onlineDnodes < pTscObj->pAppInfo->totalDnodes),
4,850✔
2390
                           .allocatorId = pRequest->allocatorRefId,
4,850✔
2391
                           .parseSqlFp = clientParseSql,
2392
                           .parseSqlParam = pWrapper};
2393
  int8_t biMode = atomic_load_8(&((STscObj*)pTscObj)->biMode);
4,850✔
2394
  (*pCxt)->biMode = biMode;
4,850✔
2395
  return TSDB_CODE_SUCCESS;
4,850✔
2396
}
2397

2398
static void asyncQueryCb(void* userdata, TAOS_RES* res, int code) {
4,850✔
2399
  STscStmt2*        pStmt = userdata;
4,850✔
2400
  __taos_async_fn_t fp = pStmt->options.asyncExecFn;
4,850✔
2401
  pStmt->asyncResultAvailable = true;
4,850✔
2402
  pStmt->exec.pRequest->inCallback = true;
4,850✔
2403

2404
  if (code == TSDB_CODE_SUCCESS) {
4,850✔
2405
    pStmt->exec.affectedRows = taos_affected_rows(res);
4,656✔
2406
    pStmt->affectedRows += pStmt->exec.affectedRows;
4,656✔
2407
  }
2408

2409
  if (fp) {
4,850✔
2410
    fp(pStmt->options.userdata, res, code);
4,850✔
2411
  }
2412

2413
  while (0 == atomic_load_8((int8_t*)&pStmt->sql.siInfo.tableColsReady)) {
4,948✔
2414
    taosUsleep(1);
98✔
2415
  }
2416
  (void)stmtCleanExecInfo(pStmt, (code ? false : true), false);
4,850✔
2417
  ++pStmt->sql.runTimes;
4,850✔
2418
  if (pStmt->exec.pRequest != NULL) {
4,850✔
2419
    pStmt->exec.pRequest->inCallback = false;
3,104✔
2420
  }
2421

2422
  if (tsem_post(&pStmt->asyncExecSem) != 0) {
4,850✔
2423
    STMT2_ELOG_E("fail to post asyncExecSem");
×
2424
  }
2425
}
4,850✔
2426

2427
int stmtExec2(TAOS_STMT2* stmt, int* affected_rows) {
618,465✔
2428
  STscStmt2* pStmt = (STscStmt2*)stmt;
618,465✔
2429
  int32_t    code = 0;
618,465✔
2430
  int64_t    startUs = taosGetTimestampUs();
618,934✔
2431

2432
  STMT2_DLOG_E("start to exec");
618,934✔
2433

2434
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
618,935✔
2435
    return pStmt->errCode;
194✔
2436
  }
2437

2438
  STMT_ERR_RET(taosThreadMutexLock(&pStmt->asyncBindParam.mutex));
618,667✔
2439
  while (atomic_load_8((int8_t*)&pStmt->asyncBindParam.asyncBindNum) > 0) {
618,766✔
2440
    (void)taosThreadCondWait(&pStmt->asyncBindParam.waitCond, &pStmt->asyncBindParam.mutex);
×
2441
  }
2442
  STMT_ERR_RET(taosThreadMutexUnlock(&pStmt->asyncBindParam.mutex));
618,437✔
2443

2444
  if (pStmt->sql.stbInterlaceMode) {
618,474✔
2445
    STMT_ERR_RET(stmtAddBatch2(pStmt));
494,246✔
2446
  }
2447

2448
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_EXECUTE));
618,439✔
2449

2450
  if (STMT_TYPE_QUERY != pStmt->sql.type) {
618,568✔
2451
    if (pStmt->sql.stbInterlaceMode) {
611,213✔
2452
      int64_t startTs = taosGetTimestampUs();
494,162✔
2453
      // wait for stmt bind thread to finish
2454
      while (atomic_load_64(&pStmt->sql.siInfo.tbRemainNum)) {
2,215,185✔
2455
        taosUsleep(1);
1,720,985✔
2456
      }
2457

2458
      if (pStmt->errCode != TSDB_CODE_SUCCESS) {
494,223✔
2459
        return pStmt->errCode;
194✔
2460
      }
2461

2462
      pStmt->stat.execWaitUs += taosGetTimestampUs() - startTs;
494,065✔
2463
      STMT_ERR_RET(qBuildStmtFinOutput(pStmt->sql.pQuery, pStmt->sql.pVgHash, pStmt->sql.siInfo.pVgroupList));
494,102✔
2464
      taosHashCleanup(pStmt->sql.siInfo.pVgroupHash);
494,004✔
2465
      pStmt->sql.siInfo.pVgroupHash = NULL;
494,091✔
2466
      pStmt->sql.siInfo.pVgroupList = NULL;
494,128✔
2467
    } else {
2468
      tDestroySubmitTbData(pStmt->exec.pCurrTbData, TSDB_MSG_FLG_ENCODE);
117,076✔
2469
      taosMemoryFreeClear(pStmt->exec.pCurrTbData);
116,872✔
2470

2471
      STMT_ERR_RET(qCloneCurrentTbData(pStmt->exec.pCurrBlock, &pStmt->exec.pCurrTbData));
116,872✔
2472

2473
      STMT_ERR_RET(qBuildStmtOutput(pStmt->sql.pQuery, pStmt->sql.pVgHash, pStmt->exec.pBlockHash));
116,647✔
2474
    }
2475
  }
2476

2477
  pStmt->asyncResultAvailable = false;
618,161✔
2478
  SRequestObj*      pRequest = pStmt->exec.pRequest;
618,167✔
2479
  __taos_async_fn_t fp = pStmt->options.asyncExecFn;
618,624✔
2480
  STMT2_DLOG("EXEC INFO :req:0x%" PRIx64 ", QID:0x%" PRIx64 ", exec sql:%s,  conn:%" PRId64, pRequest->self,
618,661✔
2481
             pRequest->requestId, pStmt->sql.sqlStr, pRequest->pTscObj->id);
2482

2483
  if (!fp) {
618,236✔
2484
    launchQueryImpl(pStmt->exec.pRequest, pStmt->sql.pQuery, true, NULL);
613,386✔
2485

2486
    if (pStmt->exec.pRequest->code && NEED_CLIENT_HANDLE_ERROR(pStmt->exec.pRequest->code)) {
613,234✔
UNCOV
2487
      STMT2_ELOG_E("exec failed errorcode:NEED_CLIENT_HANDLE_ERROR, need to refresh meta and retry");
×
UNCOV
2488
      code = refreshMeta(pStmt->exec.pRequest->pTscObj, pStmt->exec.pRequest);
×
UNCOV
2489
      if (code) {
×
2490
        pStmt->exec.pRequest->code = code;
×
2491

2492
      } else {
UNCOV
2493
        STMT_ERR_RET(stmtResetStmt(pStmt));
×
UNCOV
2494
        STMT_ERR_RET(TSDB_CODE_NEED_RETRY);
×
2495
      }
2496
    }
2497

2498
    STMT_ERR_JRET(pStmt->exec.pRequest->code);
613,505✔
2499

2500
    pStmt->exec.affectedRows = taos_affected_rows(pStmt->exec.pRequest);
612,573✔
2501
    if (affected_rows) {
613,386✔
2502
      *affected_rows = pStmt->exec.affectedRows;
605,064✔
2503
    }
2504
    pStmt->affectedRows += pStmt->exec.affectedRows;
612,490✔
2505

2506
    // wait for stmt bind thread to finish
2507
    while (0 == atomic_load_8((int8_t*)&pStmt->sql.siInfo.tableColsReady)) {
613,107✔
2508
      taosUsleep(1);
37✔
2509
    }
2510

2511
    STMT_ERR_RET(stmtCleanExecInfo(pStmt, (code ? false : true), false));
612,920✔
2512

2513
    ++pStmt->sql.runTimes;
612,950✔
2514
  } else {
2515
    SSqlCallbackWrapper* pWrapper = taosMemoryCalloc(1, sizeof(SSqlCallbackWrapper));
4,850✔
2516
    if (pWrapper == NULL) {
4,850✔
2517
      code = terrno;
×
2518
    } else {
2519
      pWrapper->pRequest = pRequest;
4,850✔
2520
      pRequest->pWrapper = pWrapper;
4,850✔
2521
    }
2522
    if (TSDB_CODE_SUCCESS == code) {
4,850✔
2523
      code = createParseContext(pRequest, &pWrapper->pParseCtx, pWrapper);
4,850✔
2524
    }
2525
    pRequest->syncQuery = false;
4,850✔
2526
    pRequest->body.queryFp = asyncQueryCb;
4,850✔
2527
    ((SSyncQueryParam*)(pRequest)->body.interParam)->userParam = pStmt;
4,850✔
2528

2529
    pStmt->execSemWaited = false;
4,850✔
2530
    launchAsyncQuery(pRequest, pStmt->sql.pQuery, NULL, pWrapper);
4,850✔
2531
  }
2532

2533
_return:
618,262✔
2534
  if (code) {
618,262✔
2535
    STMT2_ELOG("exec failed, error:%s", tstrerror(code));
388✔
2536
  }
2537
  pStmt->stat.execUseUs += taosGetTimestampUs() - startUs;
618,157✔
2538

2539
  STMT_RET(code);
618,444✔
2540
}
2541

2542
int stmtClose2(TAOS_STMT2* stmt) {
134,414✔
2543
  STscStmt2* pStmt = (STscStmt2*)stmt;
134,414✔
2544

2545
  STMT2_DLOG_E("start to close stmt");
134,414✔
2546
  taosMemoryFreeClear(pStmt->db);
134,414✔
2547

2548
  if (pStmt->bindThreadInUse) {
134,414✔
2549
    // wait for stmt bind thread to finish
2550
    while (0 == atomic_load_8((int8_t*)&pStmt->sql.siInfo.tableColsReady)) {
119,741✔
2551
      taosUsleep(1);
292✔
2552
    }
2553

2554
    (void)taosThreadMutexLock(&pStmt->queue.mutex);
119,449✔
2555
    pStmt->queue.stopQueue = true;
119,449✔
2556
    (void)taosThreadCondSignal(&(pStmt->queue.waitCond));
119,449✔
2557
    (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
119,449✔
2558

2559
    (void)taosThreadJoin(pStmt->bindThread, NULL);
119,449✔
2560
    pStmt->bindThreadInUse = false;
119,449✔
2561

2562
    (void)taosThreadCondDestroy(&pStmt->queue.waitCond);
119,449✔
2563
    (void)taosThreadMutexDestroy(&pStmt->queue.mutex);
119,449✔
2564
  }
2565

2566
  TSC_ERR_RET(taosThreadMutexLock(&pStmt->asyncBindParam.mutex));
134,414✔
2567
  while (atomic_load_8((int8_t*)&pStmt->asyncBindParam.asyncBindNum) > 0) {
134,414✔
2568
    (void)taosThreadCondWait(&pStmt->asyncBindParam.waitCond, &pStmt->asyncBindParam.mutex);
×
2569
  }
2570
  TSC_ERR_RET(taosThreadMutexUnlock(&pStmt->asyncBindParam.mutex));
134,414✔
2571

2572
  (void)taosThreadCondDestroy(&pStmt->asyncBindParam.waitCond);
134,414✔
2573
  (void)taosThreadMutexDestroy(&pStmt->asyncBindParam.mutex);
134,414✔
2574

2575
  if (pStmt->options.asyncExecFn && !pStmt->execSemWaited) {
134,414✔
2576
    if (tsem_wait(&pStmt->asyncExecSem) != 0) {
2,134✔
2577
      STMT2_ELOG_E("fail to wait asyncExecSem");
×
2578
    }
2579
  }
2580

2581
  /* On macOS dispatch_semaphore_dispose requires value >= orig (1). After tsem_wait above value is 0; post once before
2582
   * destroy. */
2583
  if (pStmt->options.asyncExecFn) {
134,414✔
2584
    if (tsem_post(&pStmt->asyncExecSem) != 0) {
2,134✔
2585
      STMT2_ELOG_E("fail to post asyncExecSem");
×
2586
    }
2587
  }
2588

2589
  STMT2_DLOG("stbInterlaceMode:%d, statInfo: ctgGetTbMetaNum=>%" PRId64 ", getCacheTbInfo=>%" PRId64
134,414✔
2590
             ", parseSqlNum=>%" PRId64 ", pStmt->stat.bindDataNum=>%" PRId64
2591
             ", settbnameAPI:%u, bindAPI:%u, addbatchAPI:%u, execAPI:%u"
2592
             ", setTbNameUs:%" PRId64 ", bindDataUs:%" PRId64 ",%" PRId64 ",%" PRId64 ",%" PRId64 " addBatchUs:%" PRId64
2593
             ", execWaitUs:%" PRId64 ", execUseUs:%" PRId64,
2594
             pStmt->sql.stbInterlaceMode, pStmt->stat.ctgGetTbMetaNum, pStmt->stat.getCacheTbInfo,
2595
             pStmt->stat.parseSqlNum, pStmt->stat.bindDataNum, pStmt->seqIds[STMT_SETTBNAME], pStmt->seqIds[STMT_BIND],
2596
             pStmt->seqIds[STMT_ADD_BATCH], pStmt->seqIds[STMT_EXECUTE], pStmt->stat.setTbNameUs,
2597
             pStmt->stat.bindDataUs1, pStmt->stat.bindDataUs2, pStmt->stat.bindDataUs3, pStmt->stat.bindDataUs4,
2598
             pStmt->stat.addBatchUs, pStmt->stat.execWaitUs, pStmt->stat.execUseUs);
2599
  if (pStmt->sql.stbInterlaceMode) {
134,414✔
2600
    pStmt->bInfo.tagsCached = false;
9,495✔
2601
  }
2602
  pStmt->bInfo.boundColsCached = false;
134,414✔
2603

2604
  STMT_ERR_RET(stmtCleanSQLInfo(pStmt));
134,414✔
2605

2606
  if (pStmt->options.asyncExecFn) {
134,414✔
2607
    if (tsem_destroy(&pStmt->asyncExecSem) != 0) {
2,134✔
2608
      STMT2_ELOG_E("fail to destroy asyncExecSem");
×
2609
    }
2610
  }
2611
  taosMemoryFree(stmt);
134,414✔
2612

2613
  return TSDB_CODE_SUCCESS;
134,391✔
2614
}
2615

2616
const char* stmt2Errstr(TAOS_STMT2* stmt) {
9,506✔
2617
  STscStmt2* pStmt = (STscStmt2*)stmt;
9,506✔
2618

2619
  if (stmt == NULL || NULL == pStmt->exec.pRequest) {
9,506✔
2620
    return (char*)tstrerror(terrno);
776✔
2621
  }
2622

2623
  // if stmt async exec ,error code is pStmt->exec.pRequest->code
2624
  if (!(pStmt->sql.status >= STMT_EXECUTE && pStmt->options.asyncExecFn != NULL && pStmt->asyncResultAvailable)) {
8,730✔
2625
    pStmt->exec.pRequest->code = terrno;
8,730✔
2626
  }
2627

2628
  SRequestObj* pRequest = pStmt->exec.pRequest;
8,730✔
2629
  if (NULL != pRequest->msgBuf && (strlen(pRequest->msgBuf) > 0 || pRequest->code == TSDB_CODE_RPC_FQDN_ERROR)) {
8,730✔
2630
    return pRequest->msgBuf;
6,014✔
2631
  }
2632
  return (const char*)tstrerror(pRequest->code);
2,716✔
2633
}
2634
/*
2635
int stmtAffectedRows(TAOS_STMT* stmt) { return ((STscStmt2*)stmt)->affectedRows; }
2636

2637
int stmtAffectedRowsOnce(TAOS_STMT* stmt) { return ((STscStmt2*)stmt)->exec.affectedRows; }
2638
*/
2639

2640
int stmtParseColFields2(TAOS_STMT2* stmt) {
13,793✔
2641
  int32_t    code = 0;
13,793✔
2642
  STscStmt2* pStmt = (STscStmt2*)stmt;
13,793✔
2643
  int32_t    preCode = pStmt->errCode;
13,793✔
2644

2645
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
13,793✔
2646
    return pStmt->errCode;
×
2647
  }
2648

2649
  if (STMT_TYPE_QUERY == pStmt->sql.type) {
13,793✔
2650
    STMT2_ELOG_E("stmtParseColFields2 only for insert");
×
2651
    STMT_ERRI_JRET(TSDB_CODE_TSC_STMT_API_ERROR);
×
2652
  }
2653

2654
  STMT_ERRI_JRET(stmtSwitchStatus(pStmt, STMT_FETCH_FIELDS));
13,793✔
2655

2656
  if (pStmt->bInfo.needParse && pStmt->sql.runTimes && pStmt->sql.type > 0 &&
13,793✔
2657
      STMT_TYPE_MULTI_INSERT != pStmt->sql.type) {
582✔
2658
    pStmt->bInfo.needParse = false;
×
2659
  }
2660
  if (pStmt->sql.stbInterlaceMode && pStmt->sql.siInfo.pDataCtx != NULL) {
13,793✔
2661
    pStmt->bInfo.needParse = false;
1,164✔
2662
  }
2663

2664
  STMT_ERRI_JRET(stmtCreateRequest(pStmt));
13,793✔
2665

2666
  if (pStmt->bInfo.needParse) {
13,793✔
2667
    STMT_ERRI_JRET(stmtParseSql(pStmt));
12,435✔
2668
  }
2669

2670
_return:
10,495✔
2671
  // compatible with previous versions
2672
  if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST && (pStmt->bInfo.tbNameFlag & NO_DATA_USING_CLAUSE) == 0x0) {
13,793✔
2673
    code = TSDB_CODE_TSC_STMT_TBNAME_ERROR;
776✔
2674
  }
2675

2676
  pStmt->errCode = preCode;
13,793✔
2677

2678
  return code;
13,793✔
2679
}
2680

2681
int stmtGetStbColFields2(TAOS_STMT2* stmt, int* nums, TAOS_FIELD_ALL** fields) {
13,793✔
2682
  int32_t code = stmtParseColFields2(stmt);
13,793✔
2683
  if (code != TSDB_CODE_SUCCESS) {
13,793✔
2684
    return code;
3,298✔
2685
  }
2686

2687
  return stmtFetchStbColFields2(stmt, nums, fields);
10,495✔
2688
}
2689

2690
int stmtGetParamNum2(TAOS_STMT2* stmt, int* nums) {
3,686✔
2691
  int32_t    code = 0;
3,686✔
2692
  STscStmt2* pStmt = (STscStmt2*)stmt;
3,686✔
2693
  int32_t    preCode = pStmt->errCode;
3,686✔
2694

2695
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
3,686✔
2696
    return pStmt->errCode;
×
2697
  }
2698

2699
  STMT_ERRI_JRET(stmtSwitchStatus(pStmt, STMT_FETCH_FIELDS));
3,686✔
2700

2701
  if (pStmt->bInfo.needParse && pStmt->sql.runTimes && pStmt->sql.type > 0 &&
3,686✔
2702
      STMT_TYPE_MULTI_INSERT != pStmt->sql.type) {
×
2703
    pStmt->bInfo.needParse = false;
×
2704
  }
2705

2706
  if (pStmt->exec.pRequest && STMT_TYPE_QUERY == pStmt->sql.type && pStmt->sql.runTimes) {
3,686✔
2707
    resetRequest(pStmt);
×
2708
  }
2709

2710
  STMT_ERRI_JRET(stmtCreateRequest(pStmt));
3,686✔
2711

2712
  if (pStmt->bInfo.needParse) {
3,686✔
2713
    STMT_ERRI_JRET(stmtParseSql(pStmt));
×
2714
  }
2715

2716
  if (STMT_TYPE_QUERY == pStmt->sql.type) {
3,686✔
2717
    *nums = taosArrayGetSize(pStmt->sql.pQuery->pPlaceholderValues);
3,686✔
2718
  } else {
2719
    STMT_ERRI_JRET(stmtFetchColFields2(stmt, nums, NULL));
×
2720
  }
2721

2722
  STMT2_DLOG("get param num success, nums:%d", *nums);
3,686✔
2723

2724
_return:
3,686✔
2725

2726
  pStmt->errCode = preCode;
3,686✔
2727

2728
  return code;
3,686✔
2729
}
2730

2731
TAOS_RES* stmtUseResult2(TAOS_STMT2* stmt) {
6,604✔
2732
  STscStmt2* pStmt = (STscStmt2*)stmt;
6,604✔
2733

2734
  STMT2_TLOG_E("start to use result");
6,604✔
2735

2736
  if (STMT_TYPE_QUERY != pStmt->sql.type) {
6,604✔
UNCOV
2737
    STMT2_ELOG_E("useResult only for query statement");
×
UNCOV
2738
    return NULL;
×
2739
  }
2740

2741
  if (pStmt->options.asyncExecFn != NULL && !pStmt->asyncResultAvailable) {
6,604✔
2742
    STMT2_ELOG_E("use result after callBackFn return");
194✔
2743
    return NULL;
194✔
2744
  }
2745

2746
  if (tsUseAdapter) {
6,410✔
2747
    TAOS_RES* res = (TAOS_RES*)pStmt->exec.pRequest;
4,462✔
2748
    pStmt->exec.pRequest = NULL;
4,462✔
2749
    return res;
4,462✔
2750
  }
2751

2752
  return pStmt->exec.pRequest;
1,948✔
2753
}
2754

2755
int32_t stmtAsyncBindThreadFunc(void* args) {
×
2756
  qInfo("async stmt bind thread started");
×
2757

2758
  ThreadArgs* targs = (ThreadArgs*)args;
×
2759
  STscStmt2*  pStmt = (STscStmt2*)targs->stmt;
×
2760

2761
  int code = taos_stmt2_bind_param(targs->stmt, targs->bindv, targs->col_idx);
×
2762
  targs->fp(targs->param, NULL, code);
×
2763
  (void)taosThreadMutexLock(&(pStmt->asyncBindParam.mutex));
×
2764
  (void)atomic_sub_fetch_8(&pStmt->asyncBindParam.asyncBindNum, 1);
×
2765
  (void)taosThreadCondSignal(&(pStmt->asyncBindParam.waitCond));
×
2766
  (void)taosThreadMutexUnlock(&(pStmt->asyncBindParam.mutex));
×
2767
  taosMemoryFree(args);
×
2768

2769
  qInfo("async stmt bind thread stopped");
×
2770

2771
  return code;
×
2772
}
2773

2774
void stmtBuildErrorMsg(STscStmt2* pStmt, const char* msg) {
582✔
2775
  if (pStmt == NULL || msg == NULL) {
582✔
2776
    return;
×
2777
  }
2778

2779
  if (pStmt->exec.pRequest == NULL) {
582✔
2780
    return;
×
2781
  }
2782

2783
  if (pStmt->exec.pRequest->msgBuf == NULL) {
582✔
2784
    return;
×
2785
  }
2786

2787
  size_t msgLen = strlen(msg);
582✔
2788
  size_t bufLen = pStmt->exec.pRequest->msgBufLen;
582✔
2789

2790
  if (msgLen >= bufLen) {
582✔
2791
    tstrncpy(pStmt->exec.pRequest->msgBuf, msg, bufLen - 1);
×
2792
    pStmt->exec.pRequest->msgBuf[bufLen - 1] = '\0';
×
2793
    pStmt->exec.pRequest->msgBufLen = bufLen - 1;
×
2794
  } else {
2795
    tstrncpy(pStmt->exec.pRequest->msgBuf, msg, bufLen);
582✔
2796
    pStmt->exec.pRequest->msgBufLen = msgLen;
582✔
2797
  }
2798

2799
  return;
582✔
2800
}
2801

2802
int32_t stmtBuildErrorMsgWithCode(STscStmt2* pStmt, const char* msg, int32_t errorCode) {
582✔
2803
  stmtBuildErrorMsg(pStmt, msg);
582✔
2804
  pStmt->errCode = errorCode;
582✔
2805

2806
  return errorCode;
582✔
2807
}
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