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

taosdata / TDengine / #4983

13 Mar 2026 03:38AM UTC coverage: 68.653% (+0.07%) from 68.587%
#4983

push

travis-ci

web-flow
feat/6641435300-save-audit-in-self (#34738)

434 of 584 new or added lines in 10 files covered. (74.32%)

434 existing lines in 121 files now uncovered.

212745 of 309883 relevant lines covered (68.65%)

134272959.11 hits per line

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

63.92
/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,656,184✔
15
    *pBuf = (char*)pTblBuf->pCurBuff + pTblBuf->buffOffset;
1,656,220✔
16
    pTblBuf->buffOffset += pTblBuf->buffUnit;
1,656,019✔
17
  } else if (pTblBuf->buffIdx < taosArrayGetSize(pTblBuf->pBufList)) {
52✔
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,655,816✔
41
}
42

43
static bool stmtDequeue(STscStmt2* pStmt, SStmtQNode** param) {
1,656,623✔
44
  int i = 0;
1,656,623✔
45
  while (0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) {
7,919,509✔
46
    if (pStmt->queue.stopQueue) {
6,379,815✔
47
      return false;
116,764✔
48
    }
49
    if (i < 10) {
6,263,089✔
50
      taosUsleep(1);
5,851,291✔
51
      i++;
5,850,562✔
52
    } else {
53
      (void)taosThreadMutexLock(&pStmt->queue.mutex);
411,798✔
54
      if (pStmt->queue.stopQueue) {
412,324✔
55
        (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
×
56
        return false;
×
57
      }
58
      if (0 == atomic_load_64((int64_t*)&pStmt->queue.qRemainNum)) {
412,324✔
59
        (void)taosThreadCondWait(&pStmt->queue.waitCond, &pStmt->queue.mutex);
412,065✔
60
      }
61
      (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
412,299✔
62
    }
63
  }
64

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

69
  (void)taosThreadMutexLock(&pStmt->queue.mutex);
1,539,175✔
70
  if (pStmt->queue.head == pStmt->queue.tail) {
1,540,018✔
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,539,942✔
78
  pStmt->queue.head->next = node->next;
1,539,942✔
79
  if (pStmt->queue.tail == node) {
1,540,022✔
80
    pStmt->queue.tail = pStmt->queue.head;
837,992✔
81
  }
82
  node->next = NULL;
1,540,174✔
83
  *param = node;
1,540,098✔
84

85
  (void)atomic_sub_fetch_64((int64_t*)&pStmt->queue.qRemainNum, 1);
1,539,934✔
86
  (void)taosThreadMutexUnlock(&pStmt->queue.mutex);
1,540,069✔
87

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

90
  return true;
1,540,021✔
91
}
92

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

99
  param->next = NULL;
1,539,624✔
100

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

103
  pStmt->queue.tail->next = param;
1,539,862✔
104
  pStmt->queue.tail = param;
1,539,862✔
105
  pStmt->stat.bindDataNum++;
1,539,744✔
106

107
  (void)atomic_add_fetch_64(&pStmt->queue.qRemainNum, 1);
1,539,862✔
108
  (void)taosThreadCondSignal(&(pStmt->queue.waitCond));
1,539,994✔
109

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

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

116
static int32_t stmtCreateRequest(STscStmt2* pStmt) {
65,088,210✔
117
  int32_t code = 0;
65,088,210✔
118

119
  if (pStmt->exec.pRequest == NULL) {
65,088,210✔
120
    code = buildRequest(pStmt->taos->id, pStmt->sql.sqlStr, pStmt->sql.sqlLen, NULL, false, &pStmt->exec.pRequest,
118,988✔
121
                        pStmt->reqid);
122
    if (pStmt->reqid != 0) {
118,988✔
123
      pStmt->reqid++;
9✔
124
    }
125
    pStmt->exec.pRequest->type = RES_TYPE__QUERY;
118,983✔
126
    if (pStmt->db != NULL) {
119,025✔
127
      taosMemoryFreeClear(pStmt->exec.pRequest->pDb);
119,025✔
128
      pStmt->exec.pRequest->pDb = taosStrdup(pStmt->db);
118,987✔
129
    }
130
    if (TSDB_CODE_SUCCESS == code) {
118,945✔
131
      pStmt->exec.pRequest->syncQuery = true;
118,945✔
132
      pStmt->exec.pRequest->stmtBindVersion = 2;
118,987✔
133
    }
134
    STMT2_DLOG("create request:0x%" PRIx64 ", QID:0x%" PRIx64, pStmt->exec.pRequest->self,
118,987✔
135
               pStmt->exec.pRequest->requestId);
136
  }
137

138
  return code;
65,615,918✔
139
}
140

141
static int32_t stmtSwitchStatus(STscStmt2* pStmt, STMT_STATUS newStatus) {
127,347,263✔
142
  int32_t code = 0;
127,347,263✔
143

144
  if (newStatus >= STMT_INIT && newStatus < STMT_MAX) {
127,347,263✔
145
    STMT2_LOG_SEQ(newStatus);
128,810,852✔
146
  }
147

148
  if (pStmt->errCode && newStatus != STMT_PREPARE) {
129,469,556✔
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) {
128,419,054✔
154
    case STMT_PREPARE:
116,767✔
155
      pStmt->errCode = 0;
116,767✔
156
      break;
13,946✔
157
    case STMT_SETTBNAME:
1,041,496✔
158
      if (STMT_STATUS_EQ(INIT)) {
1,041,496✔
159
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
160
      }
161
      if (!pStmt->sql.stbInterlaceMode && (STMT_STATUS_EQ(BIND) || STMT_STATUS_EQ(BIND_COL))) {
1,041,538✔
162
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
163
      }
164
      break;
1,041,072✔
165
    case STMT_SETTAGS:
629,331✔
166
      if (STMT_STATUS_EQ(INIT)) {
629,331✔
167
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
168
      }
169
      break;
629,331✔
170
    case STMT_FETCH_FIELDS:
27✔
171
      if (STMT_STATUS_EQ(INIT)) {
27✔
172
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
173
      }
174
      break;
27✔
175
    case STMT_BIND:
62,939,452✔
176
      if (STMT_STATUS_EQ(INIT) || STMT_STATUS_EQ(BIND_COL)) {
62,939,452✔
177
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
178
      }
179
      /*
180
            if ((pStmt->sql.type == STMT_TYPE_MULTI_INSERT) && ()) {
181
              code = TSDB_CODE_TSC_STMT_API_ERROR;
182
            }
183
      */
184
      break;
64,309,961✔
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:
63,079,941✔
191
      if (STMT_STATUS_NE(BIND) && STMT_STATUS_NE(BIND_COL) && STMT_STATUS_NE(FETCH_FIELDS)) {
63,079,941✔
192
        code = TSDB_CODE_TSC_STMT_API_ERROR;
×
193
      }
194
      break;
63,498,194✔
195
    case STMT_EXECUTE:
612,040✔
196
      if (STMT_TYPE_QUERY == pStmt->sql.type) {
612,040✔
197
        if (STMT_STATUS_NE(ADD_BATCH) && STMT_STATUS_NE(FETCH_FIELDS) && STMT_STATUS_NE(BIND) &&
11✔
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,991✔
203
          code = TSDB_CODE_TSC_STMT_API_ERROR;
×
204
        }
205
      }
206
      break;
612,002✔
207
    default:
×
208
      code = TSDB_CODE_APP_ERROR;
×
209
      break;
×
210
  }
211

212
  STMT_ERR_RET(code);
129,282,169✔
213

214
  pStmt->sql.status = newStatus;
129,282,169✔
215

216
  return TSDB_CODE_SUCCESS;
130,095,461✔
217
}
218

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

222
  pStmt->sql.type = STMT_TYPE_MULTI_INSERT;
3,183✔
223

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

229
  *tbName = pStmt->bInfo.tbName;
3,156✔
230

231
  return TSDB_CODE_SUCCESS;
3,156✔
232
}
233

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

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

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

251
  pStmt->bInfo.tbUid = autoCreateTbl ? 0 : pTableMeta->uid;
114,543✔
252
  pStmt->bInfo.tbSuid = pTableMeta->suid;
114,579✔
253
  pStmt->bInfo.tbVgId = pTableMeta->vgId;
114,806✔
254
  pStmt->bInfo.tbType = pTableMeta->tableType;
114,541✔
255

256
  if (!pStmt->bInfo.tagsCached) {
114,499✔
257
    qDestroyBoundColInfo(pStmt->bInfo.boundTags);
114,617✔
258
    taosMemoryFreeClear(pStmt->bInfo.boundTags);
114,579✔
259
  }
260

261
  // transfer ownership of cols to stmt
262
  if (cols) {
114,241✔
263
    pStmt->bInfo.fixedValueCols = *cols;
114,538✔
264
    *cols = NULL;
114,426✔
265
  }
266

267
  pStmt->bInfo.boundTags = tags;
113,902✔
268
  pStmt->bInfo.tagsCached = false;
114,468✔
269
  pStmt->bInfo.tbNameFlag = tbNameFlag;
114,731✔
270
  tstrncpy(pStmt->bInfo.stbFName, sTableName, sizeof(pStmt->bInfo.stbFName));
114,542✔
271

272
  if (pTableMeta->tableType != TSDB_CHILD_TABLE && pTableMeta->tableType != TSDB_SUPER_TABLE) {
114,954✔
273
    pStmt->sql.stbInterlaceMode = false;
1,660✔
274
  }
275

276
  return TSDB_CODE_SUCCESS;
114,580✔
277
}
278

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

282
  pStmt->sql.pVgHash = pVgHash;
114,580✔
283
  pStmt->exec.pBlockHash = pBlockHash;
114,998✔
284

285
  return TSDB_CODE_SUCCESS;
114,542✔
286
}
287

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

293
  STMT_ERR_RET(stmtUpdateBindInfo(stmt, pTableMeta, tags, cols, tbName, sTableName, autoCreateTbl, tbNameFlag));
114,580✔
294
  STMT_ERR_RET(stmtUpdateExecInfo(stmt, pVgHash, pBlockHash));
114,580✔
295

296
  pStmt->sql.autoCreateTbl = autoCreateTbl;
114,542✔
297

298
  return TSDB_CODE_SUCCESS;
114,992✔
299
}
300

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

304
  *pVgHash = pStmt->sql.pVgHash;
27✔
305
  pStmt->sql.pVgHash = NULL;
27✔
306

307
  *pBlockHash = pStmt->exec.pBlockHash;
27✔
308
  pStmt->exec.pBlockHash = NULL;
27✔
309

310
  return TSDB_CODE_SUCCESS;
27✔
311
}
312

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

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

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

326
  pStmt->stat.parseSqlNum++;
115,041✔
327

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

331
  pStmt->sql.siInfo.pQuery = pStmt->sql.pQuery;
114,855✔
332

333
  pStmt->bInfo.needParse = false;
115,082✔
334

335
  if (pStmt->sql.type == 0) {
115,008✔
336
    if (pStmt->sql.pQuery->pRoot && LEGAL_INSERT(nodeType(pStmt->sql.pQuery->pRoot))) {
111,822✔
337
      pStmt->sql.type = STMT_TYPE_INSERT;
111,847✔
338
      pStmt->sql.stbInterlaceMode = false;
111,847✔
339
    } else if (pStmt->sql.pQuery->pPrepareRoot && LEGAL_SELECT(nodeType(pStmt->sql.pQuery->pPrepareRoot))) {
11✔
340
      pStmt->sql.type = STMT_TYPE_QUERY;
11✔
341
      pStmt->sql.stbInterlaceMode = false;
11✔
342

343
      return TSDB_CODE_SUCCESS;
11✔
344
    } else {
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) {
3,224✔
352
    pStmt->sql.stbInterlaceMode = false;
×
353
    return TSDB_CODE_SUCCESS;
×
354
  } else if (pStmt->sql.type == STMT_TYPE_INSERT) {
3,224✔
355
    pStmt->sql.stbInterlaceMode = false;
×
356
  }
357

358
  STableDataCxt** pSrc =
359
      (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
115,071✔
360
  if (NULL == pSrc || NULL == *pSrc) {
114,996✔
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;
114,769✔
366
  if (pStmt->sql.stbInterlaceMode && pTableCtx->pData->pCreateTbReq && (pStmt->bInfo.tbNameFlag & USING_CLAUSE) == 0) {
114,769✔
367
    STMT2_TLOG("destroy pCreateTbReq for no-using insert, tbFName:%s", pStmt->bInfo.tbFName);
×
368
    tdDestroySVCreateTbReq(pTableCtx->pData->pCreateTbReq);
×
369
    taosMemoryFreeClear(pTableCtx->pData->pCreateTbReq);
×
370
    pTableCtx->pData->pCreateTbReq = NULL;
×
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) {
114,996✔
386
    pStmt->sql.pBindInfo = taosMemoryMalloc(pTableCtx->boundColsInfo.numOfBound * sizeof(*pStmt->sql.pBindInfo));
115,007✔
387
    if (NULL == pStmt->sql.pBindInfo) {
114,628✔
388
      STMT2_ELOG_E("fail to malloc pBindInfo");
×
389
      return terrno;
×
390
    }
391
  }
392

393
  return TSDB_CODE_SUCCESS;
114,617✔
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) {
230,902✔
520
  if (pStmt->exec.pRequest) {
230,902✔
521
    taos_free_result(pStmt->exec.pRequest);
119,025✔
522
    pStmt->exec.pRequest = NULL;
119,025✔
523
  }
524
  pStmt->asyncResultAvailable = false;
230,902✔
525
}
230,902✔
526

527
static int32_t stmtCleanBindInfo(STscStmt2* pStmt) {
731,543✔
528
  pStmt->bInfo.tbUid = 0;
731,543✔
529
  pStmt->bInfo.tbVgId = -1;
731,477✔
530
  pStmt->bInfo.tbType = 0;
731,439✔
531
  pStmt->bInfo.needParse = true;
731,439✔
532
  pStmt->bInfo.inExecCache = false;
731,401✔
533

534
  pStmt->bInfo.tbName[0] = 0;
731,477✔
535
  pStmt->bInfo.tbFName[0] = 0;
731,444✔
536
  if (!pStmt->bInfo.tagsCached) {
731,364✔
537
    qDestroyBoundColInfo(pStmt->bInfo.boundTags);
522,101✔
538
    taosMemoryFreeClear(pStmt->bInfo.boundTags);
522,209✔
539
    pStmt->bInfo.boundTags = NULL;
522,262✔
540
  }
541

542
  if (!pStmt->bInfo.boundColsCached) {
731,377✔
543
    tSimpleHashCleanup(pStmt->bInfo.fixedValueCols);
233,637✔
544
    pStmt->bInfo.fixedValueCols = NULL;
233,637✔
545
  }
546

547
  if (!pStmt->sql.autoCreateTbl) {
731,377✔
548
    pStmt->bInfo.stbFName[0] = 0;
521,740✔
549
    pStmt->bInfo.tbSuid = 0;
521,594✔
550
  }
551

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

555
  return TSDB_CODE_SUCCESS;
731,332✔
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) {
497,841✔
564
  pTblBuf->pCurBuff = taosArrayGetP(pTblBuf->pBufList, 0);
497,841✔
565
  if (NULL == pTblBuf->pCurBuff) {
497,826✔
566
    tscError("QInfo:%p, fail to get buffer from list", pTblBuf);
52✔
567
    return;
×
568
  }
569
  pTblBuf->buffIdx = 1;
497,774✔
570
  pTblBuf->buffOffset = sizeof(*pQueue->head);
497,774✔
571

572
  pQueue->head = pQueue->tail = pTblBuf->pCurBuff;
497,774✔
573
  pQueue->qRemainNum = 0;
497,749✔
574
  pQueue->head->next = NULL;
497,707✔
575
}
576

577
static int32_t stmtCleanExecInfo(STscStmt2* pStmt, bool keepTable, bool deepClean) {
728,560✔
578
  if (pStmt->sql.stbInterlaceMode) {
728,560✔
579
    if (deepClean) {
502,745✔
580
      taosHashCleanup(pStmt->exec.pBlockHash);
4,917✔
581
      pStmt->exec.pBlockHash = NULL;
4,917✔
582

583
      if (NULL != pStmt->exec.pCurrBlock) {
4,917✔
584
        taosMemoryFreeClear(pStmt->exec.pCurrBlock->boundColsInfo.pColIndex);
3,167✔
585
        taosMemoryFreeClear(pStmt->exec.pCurrBlock->pData);
3,167✔
586
        qDestroyStmtDataBlock(pStmt->exec.pCurrBlock);
3,167✔
587
        pStmt->exec.pCurrBlock = NULL;
3,167✔
588
      }
589
      if (STMT_TYPE_QUERY != pStmt->sql.type) {
4,917✔
590
        resetRequest(pStmt);
4,917✔
591
      }
592
    } else {
593
      pStmt->sql.siInfo.pTableColsIdx = 0;
497,828✔
594
      stmtResetQueueTableBuf(&pStmt->sql.siInfo.tbBuf, &pStmt->queue);
497,828✔
595
      tSimpleHashClear(pStmt->sql.siInfo.pTableRowDataHash);
497,790✔
596
    }
597
    if (NULL != pStmt->exec.pRequest) {
502,810✔
598
      pStmt->exec.pRequest->body.resInfo.numOfRows = 0;
497,893✔
599
    }
600
  } else {
601
    if (STMT_TYPE_QUERY != pStmt->sql.type || deepClean) {
225,957✔
602
      resetRequest(pStmt);
225,946✔
603
    }
604

605
    size_t keyLen = 0;
225,922✔
606
    void*  pIter = taosHashIterate(pStmt->exec.pBlockHash, NULL);
225,922✔
607
    while (pIter) {
452,163✔
608
      STableDataCxt* pBlocks = *(STableDataCxt**)pIter;
226,167✔
609
      char*          key = taosHashGetKey(pIter, &keyLen);
226,167✔
610
      STableMeta*    pMeta = qGetTableMetaInDataBlock(pBlocks);
226,167✔
611

612
      if (keepTable && pBlocks == pStmt->exec.pCurrBlock) {
226,167✔
613
        TSWAP(pBlocks->pData, pStmt->exec.pCurrTbData);
114,097✔
614
        STMT_ERR_RET(qResetStmtDataBlock(pBlocks, false));
123,237✔
615

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

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

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

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

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

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

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

643
  return TSDB_CODE_SUCCESS;
614,580✔
644
}
645

646
static void stmtFreeTbBuf(void* buf) {
116,727✔
647
  void* pBuf = *(void**)buf;
116,727✔
648
  taosMemoryFree(pBuf);
116,727✔
649
}
116,731✔
650

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

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

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

674
  void* pIter = taosHashIterate(pStmt->sql.pTableCache, NULL);
116,805✔
675
  while (pIter) {
116,835✔
676
    SStmtTableCache* pCache = (SStmtTableCache*)pIter;
30✔
677

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

682
    pIter = taosHashIterate(pStmt->sql.pTableCache, pIter);
30✔
683
  }
684
  taosHashCleanup(pStmt->sql.pTableCache);
116,805✔
685
  pStmt->sql.pTableCache = NULL;
116,805✔
686

687
  STMT_ERR_RET(stmtCleanExecInfo(pStmt, false, true));
116,805✔
688
  STMT_ERR_RET(stmtCleanBindInfo(pStmt));
116,805✔
689

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

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

703
  STMT2_TLOG_E("end to free SQL info");
116,699✔
704

705
  return TSDB_CODE_SUCCESS;
116,662✔
706
}
707

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

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

719
  int32_t code = catalogGetTableHashVgroup(pStmt->pCatalog, &conn, &pStmt->bInfo.sname, &vgInfo);
629,275✔
720
  if (TSDB_CODE_SUCCESS != code) {
629,314✔
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));
629,314✔
727
  if (TSDB_CODE_SUCCESS != code) {
629,327✔
728
    STMT2_ELOG("fail to put vgroup info, code:%d", code);
91✔
729
    return code;
×
730
  }
731

732
  *vgId = vgInfo.vgId;
629,236✔
733

734
  return TSDB_CODE_SUCCESS;
629,236✔
735
}
736

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

745
  pStmt->stat.ctgGetTbMetaNum++;
2,937✔
746

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

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

756
    STMT_ERR_RET(code);
×
757
  }
758

759
  STMT_ERR_RET(code);
2,937✔
760

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

767
  taosMemoryFree(pTableMeta);
2,937✔
768

769
  return TSDB_CODE_SUCCESS;
2,937✔
770
}
771

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

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

779
  return TSDB_CODE_SUCCESS;
193✔
780
}
781

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

789
  pStmt->bInfo.needParse = true;
3,390✔
790
  pStmt->bInfo.inExecCache = false;
3,348✔
791

792
  STableDataCxt** pCxtInExec = taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
3,348✔
793
  if (pCxtInExec) {
3,390✔
794
    pStmt->bInfo.needParse = false;
×
795
    pStmt->bInfo.inExecCache = true;
×
796

797
    pStmt->exec.pCurrBlock = *pCxtInExec;
×
798

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

805
  if (NULL == pStmt->pCatalog) {
3,390✔
806
    STMT_ERR_RET(catalogGetHandle(pStmt->taos->pAppInfo->clusterId, &pStmt->pCatalog));
3,054✔
807
    pStmt->sql.siInfo.pCatalog = pStmt->pCatalog;
2,971✔
808
  }
809

810
  if (NULL == pStmt->sql.pTableCache || taosHashGetSize(pStmt->sql.pTableCache) <= 0) {
3,266✔
811
    if (pStmt->bInfo.inExecCache) {
3,114✔
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);
3,197✔
818

819
    return TSDB_CODE_SUCCESS;
3,156✔
820
  }
821

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

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

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

836
      pStmt->exec.pCurrBlock = pNewBlock;
193✔
837

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

840
      return TSDB_CODE_SUCCESS;
193✔
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));
×
851

852
  uint64_t cacheUid = (TSDB_CHILD_TABLE == tableType) ? suid : uid;
×
853

854
  if (uid == pStmt->bInfo.tbUid) {
×
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) {
×
863
    SStmtTableCache* pCache = taosHashGet(pStmt->sql.pTableCache, &cacheUid, sizeof(cacheUid));
×
864
    if (NULL == pCache) {
×
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;
×
872

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

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

881
    return TSDB_CODE_SUCCESS;
×
882
  }
883

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

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

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

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

902
    pStmt->exec.pCurrBlock = pNewBlock;
×
903

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

906
    return TSDB_CODE_SUCCESS;
×
907
  }
908

909
  STMT_ERR_RET(stmtCleanBindInfo(pStmt));
×
910

911
  return TSDB_CODE_SUCCESS;
×
912
}
913

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

917
  pStmt->sql.pTableCache = taosHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
×
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

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

925
  return TSDB_CODE_SUCCESS;
×
926
}
927

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

931
  if (pParam->restoreTbCols) {
1,539,876✔
932
    for (int32_t i = 0; i < pStmt->sql.siInfo.pTableColsIdx; ++i) {
1,539,800✔
933
      SArray** p = (SArray**)TARRAY_GET_ELEM(pStmt->sql.siInfo.pTableCols, i);
1,042,023✔
934
      *p = taosArrayInit(20, POINTER_BYTES);
1,042,023✔
935
      if (*p == NULL) {
1,041,934✔
936
        pStmt->errCode = terrno;
91✔
937
      }
938
    }
939
    atomic_store_8((int8_t*)&pStmt->sql.siInfo.tableColsReady, true);
497,815✔
940
    STMT2_TLOG_E("restore pTableCols finished");
497,932✔
941
  } else {
942
    int code = qAppendStmt2TableOutput(pStmt->sql.pQuery, pStmt->sql.pVgHash, &pParam->tblData, pStmt->exec.pCurrBlock,
1,041,995✔
943
                                       &pStmt->sql.siInfo, pParam->pCreateTbReq);
944
    // taosMemoryFree(pParam->pTbData);
945
    if (code != TSDB_CODE_SUCCESS) {
1,041,929✔
946
      STMT2_ELOG("async append stmt output failed, tbname:%s, err:%s", pParam->tblData.tbName, tstrerror(code));
×
947
      pStmt->errCode = code;
×
948
    }
949
    (void)atomic_sub_fetch_64(&pStmt->sql.siInfo.tbRemainNum, 1);
1,041,929✔
950
  }
951
}
1,540,047✔
952

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

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

959
  while (true) {
1,540,021✔
960
    SStmtQNode* asyncParam = NULL;
1,656,785✔
961

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

970
    stmtAsyncOutput(pStmt, asyncParam);
1,539,966✔
971
  }
972

973
  STMT2_DLOG_E("stmt2 bind thread stopped");
116,726✔
974
  return NULL;
116,726✔
975
}
976

977
static int32_t stmtStartBindThread(STscStmt2* pStmt) {
116,764✔
978
  TdThreadAttr thAttr;
107,112✔
979
  if (taosThreadAttrInit(&thAttr) != 0) {
116,764✔
980
    return TSDB_CODE_TSC_INTERNAL_ERROR;
×
981
  }
982
  if (taosThreadAttrSetDetachState(&thAttr, PTHREAD_CREATE_JOINABLE) != 0) {
116,727✔
983
    return TSDB_CODE_TSC_INTERNAL_ERROR;
×
984
  }
985

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

991
  pStmt->bindThreadInUse = true;
116,727✔
992

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

997
static int32_t stmtInitQueue(STscStmt2* pStmt) {
116,764✔
998
  (void)taosThreadCondInit(&pStmt->queue.waitCond, NULL);
116,764✔
999
  (void)taosThreadMutexInit(&pStmt->queue.mutex, NULL);
116,764✔
1000
  STMT_ERR_RET(stmtAllocQNodeFromBuf(&pStmt->sql.siInfo.tbBuf, (void**)&pStmt->queue.head));
233,528✔
1001
  pStmt->queue.tail = pStmt->queue.head;
116,764✔
1002

1003
  return TSDB_CODE_SUCCESS;
116,764✔
1004
}
1005

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

1011
  return TSDB_CODE_SUCCESS;
116,767✔
1012
}
1013

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

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

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

1034
  return TSDB_CODE_SUCCESS;
116,764✔
1035
}
1036

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

1042
  pStmt = taosMemoryCalloc(1, sizeof(STscStmt2));
116,582✔
1043
  if (NULL == pStmt) {
116,627✔
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);
116,627✔
1049
  if (NULL == pStmt->sql.pTableCache) {
116,662✔
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;
116,662✔
1056
  if (taos->db[0] != '\0') {
116,662✔
1057
    pStmt->db = taosStrdup(taos->db);
116,662✔
1058
  }
1059
  pStmt->bInfo.needParse = true;
116,662✔
1060
  pStmt->sql.status = STMT_INIT;
116,662✔
1061
  pStmt->errCode = TSDB_CODE_SUCCESS;
116,662✔
1062

1063
  if (NULL != pOptions) {
116,662✔
1064
    (void)memcpy(&pStmt->options, pOptions, sizeof(pStmt->options));
116,633✔
1065
    if (pOptions->singleStbInsert && pOptions->singleTableBindOnce) {
116,633✔
1066
      pStmt->stbInterlaceMode = true;
116,621✔
1067
    }
1068

1069
    pStmt->reqid = pOptions->reqid;
116,633✔
1070
  }
1071

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

1078
    pStmt->sql.siInfo.pTableHash = tSimpleHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY));
116,621✔
1079
    if (NULL == pStmt->sql.siInfo.pTableHash) {
116,394✔
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));
116,394✔
1086
    if (NULL == pStmt->sql.siInfo.pTableRowDataHash) {
116,621✔
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);
116,621✔
1093
    if (NULL == pStmt->sql.siInfo.pTableCols) {
116,621✔
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);
116,621✔
1100
    if (TSDB_CODE_SUCCESS == code) {
116,621✔
1101
      code = stmtInitQueue(pStmt);
116,621✔
1102
    }
1103
    if (TSDB_CODE_SUCCESS == code) {
116,584✔
1104
      code = stmtStartBindThread(pStmt);
116,584✔
1105
    }
1106
    if (TSDB_CODE_SUCCESS != code) {
116,584✔
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;
116,625✔
1115
  if (pStmt->options.asyncExecFn) {
116,625✔
1116
    if (tsem_init(&pStmt->asyncExecSem, 0, 1) != 0) {
×
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);
116,625✔
1124
  if (TSDB_CODE_SUCCESS != code) {
116,624✔
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;
116,624✔
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,
116,624✔
1137
             pStmt->stbInterlaceMode, pStmt->options.asyncExecFn != NULL);
1138

1139
  return pStmt;
116,624✔
1140
}
1141

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

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

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

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

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

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

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

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

1201
  return TSDB_CODE_SUCCESS;
143✔
1202
}
1203

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

1211
  pStmt->errCode = 0;
143✔
1212

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

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

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

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

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

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

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

1260
    STMT_ERR_RET(stmtResetStbInterlaceCache(pStmt));
143✔
1261

1262
    int32_t code = stmtIniAsyncBind(pStmt);
143✔
1263
    if (TSDB_CODE_SUCCESS != code) {
143✔
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;
143✔
1271
  pStmt->options = options;
143✔
1272
  pStmt->reqid = reqid;
143✔
1273
  pStmt->stbInterlaceMode = stbInterlaceMode;
143✔
1274

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

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

1285
  return TSDB_CODE_SUCCESS;
143✔
1286
}
1287

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

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

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

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

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

1310
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_PREPARE));
116,730✔
1311

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

1323
  if (stmt2IsInsert(pStmt)) {
116,688✔
1324
    pStmt->sql.stbInterlaceMode = pStmt->stbInterlaceMode;
116,794✔
1325
    char* dbName = NULL;
116,794✔
1326
    if (qParseDbName(sql, length, &dbName)) {
116,794✔
1327
      STMT_ERR_RET(stmtSetDbName2(stmt, dbName));
113,597✔
1328
      taosMemoryFreeClear(dbName);
113,634✔
1329
    } else if (pStmt->db != NULL && pStmt->db[0] != '\0') {
3,142✔
1330
      taosMemoryFreeClear(pStmt->exec.pRequest->pDb);
3,197✔
1331
      pStmt->exec.pRequest->pDb = taosStrdup(pStmt->db);
3,155✔
1332
      if (pStmt->exec.pRequest->pDb == NULL) {
3,197✔
1333
        STMT_ERR_RET(terrno);
×
1334
      }
1335
      (void)strdequote(pStmt->exec.pRequest->pDb);
3,155✔
1336

1337
      if (pStmt->sql.stbInterlaceMode) {
3,155✔
1338
        pStmt->sql.siInfo.dbname = pStmt->exec.pRequest->pDb;
3,125✔
1339
      }
1340
    }
1341

1342
  } else if (stmt2IsSelect(pStmt)) {
11✔
1343
    pStmt->sql.stbInterlaceMode = false;
11✔
1344
    STMT_ERR_RET(stmtParseSql(pStmt));
11✔
1345
  } else {
1346
    return stmtBuildErrorMsgWithCode(pStmt, "stmt only support 'SELECT' or 'INSERT'", TSDB_CODE_PAR_SYNTAX_ERROR);
×
1347
  }
1348
  return TSDB_CODE_SUCCESS;
116,768✔
1349
}
1350

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

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

1361
  SArray* pTblCols = NULL;
3,129✔
1362
  for (int32_t i = 0; i < STMT_TABLE_COLS_NUM; i++) {
3,156,899✔
1363
    pTblCols = taosArrayInit(20, POINTER_BYTES);
3,153,818✔
1364
    if (NULL == pTblCols) {
3,128,410✔
1365
      return terrno;
×
1366
    }
1367

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

1373
  pStmt->sql.siInfo.boundTags = pStmt->bInfo.boundTags;
3,081✔
1374

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

1378
  return TSDB_CODE_SUCCESS;
3,167✔
1379
}
1380

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

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

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

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

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

1402
  int64_t startUs = taosGetTimestampUs();
1,041,253✔
1403

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

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

1410
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_SETTBNAME));
1,041,739✔
1411

1412
  int32_t insert = 0;
1,041,395✔
1413
  if (!stmt2IsInsert(stmt)) {
1,041,395✔
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,041,378✔
1419

1420
  STMT_ERR_RET(qCreateSName2(&pStmt->bInfo.sname, tbName, pStmt->taos->acctId, pStmt->exec.pRequest->pDb,
1,041,379✔
1421
                             pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen));
1422
  STMT_ERR_RET(tNameExtractFullName(&pStmt->bInfo.sname, pStmt->bInfo.tbFName));
1,041,626✔
1423
  tstrncpy(pStmt->bInfo.tbName, (char*)tNameGetTableName(&pStmt->bInfo.sname), TSDB_TABLE_NAME_LEN);
1,041,637✔
1424

1425
  if (!pStmt->sql.stbInterlaceMode || NULL == pStmt->sql.siInfo.pDataCtx) {
1,041,459✔
1426
    STMT_ERR_RET(stmtGetFromCache(pStmt));
3,668✔
1427

1428
    if (pStmt->bInfo.needParse) {
3,349✔
1429
      STMT_ERR_RET(stmtParseSql(pStmt));
3,156✔
1430
      if (!pStmt->sql.autoCreateTbl) {
3,197✔
1431
        uint64_t uid, suid;
2,614✔
1432
        int32_t  vgId;
2,572✔
1433
        int8_t   tableType;
2,572✔
1434

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

1442
  } else {
1443
    pStmt->exec.pRequest->requestId++;
1,037,669✔
1444
    pStmt->bInfo.needParse = false;
1,037,996✔
1445
  }
1446

1447
  if (pStmt->sql.stbInterlaceMode && NULL == pStmt->sql.siInfo.pDataCtx) {
1,041,085✔
1448
    STMT_ERR_RET(stmtInitStbInterlaceTableInfo(pStmt));
3,167✔
1449
  }
1450

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

1454
  return TSDB_CODE_SUCCESS;
1,042,009✔
1455
}
1456

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

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

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

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

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

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

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

1486
  STableDataCxt** pDataBlock = NULL;
525,318✔
1487
  if (pStmt->exec.pCurrBlock) {
525,318✔
1488
    pDataBlock = &pStmt->exec.pCurrBlock;
525,240✔
1489
  } else {
1490
    pDataBlock =
1491
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
78✔
1492
    if (NULL == pDataBlock) {
78✔
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) {
78✔
1497
      tdDestroySVCreateTbReq((*pDataBlock)->pData->pCreateTbReq);
×
1498
      taosMemoryFreeClear((*pDataBlock)->pData->pCreateTbReq);
×
1499
      (*pDataBlock)->pData->pCreateTbReq = NULL;
×
1500
    }
1501
  }
1502
  if (pStmt->bInfo.inExecCache && !pStmt->sql.autoCreateTbl) {
525,318✔
1503
    return TSDB_CODE_SUCCESS;
×
1504
  }
1505

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

1508
  void* boundTags = NULL;
525,292✔
1509
  if (pStmt->sql.stbInterlaceMode) {
525,292✔
1510
    boundTags = pStmt->sql.siInfo.boundTags;
525,069✔
1511
    *pCreateTbReq = taosMemoryCalloc(1, sizeof(SVCreateTbReq));
525,069✔
1512
    if (NULL == pCreateTbReq) {
525,147✔
1513
      return terrno;
×
1514
    }
1515
    int32_t vgId = -1;
525,147✔
1516
    STMT_ERR_RET(stmtTryAddTableVgroupInfo(pStmt, &vgId));
525,147✔
1517
    (*pCreateTbReq)->uid = vgId;
525,290✔
1518
  } else {
1519
    boundTags = pStmt->bInfo.boundTags;
223✔
1520
  }
1521

1522
  STMT_ERR_RET(qBindStmtTagsValue2(*pDataBlock, boundTags, pStmt->bInfo.tbSuid, pStmt->bInfo.stbFName,
525,513✔
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,396✔
1527
}
1528

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

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

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

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

1542
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_SETTAGS));
103,974✔
1543

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

1557
  STMT_ERR_RET(stmtCreateRequest(pStmt));
182✔
1558
  if (pStmt->bInfo.needParse) {
182✔
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) {
182✔
1567
    STMT_ERR_RET(stmtInitStbInterlaceTableInfo(pStmt));
×
1568
  }
1569

1570
  STableDataCxt** pDataBlock = NULL;
182✔
1571
  if (pStmt->exec.pCurrBlock) {
182✔
1572
    pDataBlock = &pStmt->exec.pCurrBlock;
×
1573
  } else {
1574
    pDataBlock =
1575
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
182✔
1576
    if (NULL == pDataBlock) {
182✔
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)) {
182✔
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) {
182✔
1589
    STMT2_TLOG_E("tags are fixed, set createTbReq first time");
182✔
1590
    pStmt->sql.fixValueTags = true;
182✔
1591
    STMT_ERR_RET(cloneSVreateTbReq((*pDataBlock)->pData->pCreateTbReq, &pStmt->sql.fixValueTbReq));
182✔
1592
    STMT_ERR_RET(cloneSVreateTbReq(pStmt->sql.fixValueTbReq, pCreateTbReq));
182✔
1593
    (*pCreateTbReq)->uid = (*pDataBlock)->pMeta->vgId;
182✔
1594

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

1601
  return TSDB_CODE_SUCCESS;
182✔
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) {
27✔
1633
  int32_t code = 0;
27✔
1634
  int32_t preCode = pStmt->errCode;
27✔
1635

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

1640
  if (STMT_TYPE_QUERY == pStmt->sql.type) {
27✔
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;
27✔
1646
  bool            cleanStb = false;
27✔
1647

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

1656
  if (NULL == pDataBlock || NULL == *pDataBlock) {
27✔
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;
27✔
1662
  pStmt->sql.placeholderOfCols = 0;
27✔
1663
  int32_t totalNum = 0;
27✔
1664
  STMT_ERRI_JRET(qBuildStmtStbColFields(*pDataBlock, pStmt->bInfo.boundTags, pStmt->bInfo.fixedValueCols,
27✔
1665
                                        pStmt->bInfo.tbNameFlag, &totalNum, fields, &pStmt->sql.placeholderOfTags,
1666
                                        &pStmt->sql.placeholderOfCols));
1667

1668
  if (pStmt->bInfo.tbType == TSDB_SUPER_TABLE && cleanStb) {
27✔
1669
    taosMemoryFreeClear((*pDataBlock)->boundColsInfo.pColIndex);
27✔
1670
    qDestroyStmtDataBlock(*pDataBlock);
27✔
1671
    *pDataBlock = NULL;
27✔
1672
    if (taosHashRemove(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName)) != 0) {
27✔
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;
27✔
1677
    pStmt->bInfo.tagsCached = false;
27✔
1678
    pStmt->bInfo.sname = (SName){0};
27✔
1679
    STMT_ERR_RET(stmtCleanBindInfo(pStmt));
27✔
1680
  }
1681

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

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

1689
_return:
27✔
1690

1691
  pStmt->errCode = preCode;
27✔
1692

1693
  return code;
27✔
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,041,343✔
1713
  if (NULL == pStmt->sql.siInfo.pVgroupHash) {
1,041,343✔
1714
    pStmt->sql.siInfo.pVgroupHash =
497,840✔
1715
        taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, HASH_ENTRY_LOCK);
497,753✔
1716
  }
1717
  if (NULL == pStmt->sql.siInfo.pVgroupList) {
1,041,968✔
1718
    pStmt->sql.siInfo.pVgroupList = taosArrayInit(64, POINTER_BYTES);
497,865✔
1719
  }
1720

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

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

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

1734
  if (!pStmt->sql.siInfo.tbFromHash && pStmt->sql.siInfo.firstName[0] &&
1,041,896✔
1735
      0 == strcmp(pStmt->sql.siInfo.firstName, pStmt->bInfo.tbName)) {
18,996✔
1736
    pStmt->sql.siInfo.tbFromHash = true;
1,901✔
1737
  }
1738

1739
  if (0 == pStmt->sql.siInfo.firstName[0]) {
1,041,853✔
1740
    tstrncpy(pStmt->sql.siInfo.firstName, pStmt->bInfo.tbName, TSDB_TABLE_NAME_LEN);
3,167✔
1741
  }
1742

1743
  param->tblData.getFromHash = pStmt->sql.siInfo.tbFromHash;
1,041,896✔
1744
  param->next = NULL;
1,041,892✔
1745

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

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

1754
  return TSDB_CODE_SUCCESS;
1,041,939✔
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,041,439✔
1760
      *pTableCols = (SArray*)taosArrayGetP(pStmt->sql.siInfo.pTableCols, pStmt->sql.siInfo.pTableColsIdx++);
1,041,649✔
1761
      break;
1,041,127✔
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,041,127✔
1778
}
1779

1780
static int32_t stmtCacheBlock(STscStmt2* pStmt) {
61,895,092✔
1781
  if (pStmt->sql.type != STMT_TYPE_MULTI_INSERT) {
61,895,092✔
1782
    return TSDB_CODE_SUCCESS;
62,936,064✔
1783
  }
1784

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

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

1793
  STableDataCxt** pSrc = taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
30✔
1794
  if (!pSrc) {
30✔
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;
30✔
1799

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

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

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

1812
  if (pStmt->sql.autoCreateTbl) {
30✔
1813
    pStmt->bInfo.tagsCached = true;
30✔
1814
  } else {
1815
    pStmt->bInfo.boundTags = NULL;
×
1816
  }
1817

1818
  return TSDB_CODE_SUCCESS;
30✔
1819
}
1820

1821
static int stmtAddBatch2(TAOS_STMT2* stmt) {
62,337,219✔
1822
  STscStmt2* pStmt = (STscStmt2*)stmt;
62,337,219✔
1823

1824
  int64_t startUs = taosGetTimestampUs();
63,695,656✔
1825

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

1828
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
63,695,656✔
1829
    return pStmt->errCode;
×
1830
  }
1831

1832
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_ADD_BATCH));
63,977,139✔
1833

1834
  if (pStmt->sql.stbInterlaceMode) {
63,671,958✔
1835
    int64_t startUs2 = taosGetTimestampUs();
497,855✔
1836
    pStmt->stat.addBatchUs += startUs2 - startUs;
497,855✔
1837

1838
    pStmt->sql.siInfo.tableColsReady = false;
497,817✔
1839

1840
    SStmtQNode* param = NULL;
497,893✔
1841
    STMT_ERR_RET(stmtAllocQNodeFromBuf(&pStmt->sql.siInfo.tbBuf, (void**)&param));
995,659✔
1842
    param->restoreTbCols = true;
497,804✔
1843
    param->next = NULL;
497,880✔
1844

1845
    if (pStmt->sql.autoCreateTbl) {
497,880✔
1846
      pStmt->bInfo.tagsCached = true;
209,091✔
1847
    }
1848
    pStmt->bInfo.boundColsCached = true;
497,804✔
1849

1850
    if (pStmt->queue.stopQueue) {
497,804✔
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);
497,766✔
1856

1857
    return TSDB_CODE_SUCCESS;
497,881✔
1858
  }
1859

1860
  STMT_ERR_RET(stmtCacheBlock(pStmt));
63,327,672✔
1861

1862
  return TSDB_CODE_SUCCESS;
63,128,056✔
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) {
11✔
1933
  SCatalogSyncCbParam* pCbParam = (SCatalogSyncCbParam*)param;
11✔
1934
  if (TSDB_CODE_SUCCESS == code && pResultMeta) {
11✔
1935
    *pCbParam->pRsp = *pResultMeta;
11✔
1936
    TAOS_MEMSET(pResultMeta, 0, sizeof(SMetaData));  // Clear to avoid double free
11✔
1937
  }
1938
  pCbParam->code = code;
11✔
1939
  if (tsem_post(&pCbParam->sem) != 0) {
11✔
1940
    tscError("failed to post semaphore");
×
1941
  }
1942
}
11✔
1943

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

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

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

1975
      if (tsem_destroy(&cbParam.sem) != 0) {
11✔
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);
11✔
1985
  destoryCatalogReq(&catalogReq);
11✔
1986

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

1991
  return code;
11✔
1992
}
1993

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

1998
  int64_t startUs = taosGetTimestampUs();
63,059,691✔
1999

2000
  if (qDebugFlag & DEBUG_TRACE) {
63,059,691✔
2001
    (void)stmtPrintBindv(stmt, bind, colIdx, false);
×
2002
  }
2003

2004
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
63,087,316✔
2005
    return pStmt->errCode;
×
2006
  }
2007

2008
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_BIND));
63,310,150✔
2009

2010
  if (pStmt->bInfo.needParse && pStmt->sql.runTimes && pStmt->sql.type > 0 &&
64,325,823✔
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) {
63,958,883✔
2016
    resetRequest(pStmt);
×
2017
  }
2018

2019
  STMT_ERR_RET(stmtCreateRequest(pStmt));
64,084,409✔
2020
  if (pStmt->bInfo.needParse) {
63,681,697✔
2021
    code = stmtParseSql(pStmt);
111,847✔
2022
    if (code != TSDB_CODE_SUCCESS) {
111,355✔
2023
      goto cleanup_root;
×
2024
    }
2025
  }
2026

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

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

2057
    code = qStmtParseQuerySql(&ctx, pStmt->sql.pQuery, &metaData);
11✔
2058
    if (TSDB_CODE_SUCCESS == code) {
11✔
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));
11✔
2062
      (void)memset(&metaData, 0, sizeof(SMetaData));  // Clear to avoid double free
11✔
2063
    } else {
2064
      // Clean up metaData on failure - free all arrays
2065
      if (metaData.pVStbRefDbs) {
×
2066
        taosArrayDestroy(metaData.pVStbRefDbs);
×
2067
        metaData.pVStbRefDbs = NULL;
×
2068
      }
2069
      // Note: Other fields in metaData are managed by catalog module if ctgFree is true
2070
      goto cleanup_root;
×
2071
    }
2072

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

2081
    TSWAP(pStmt->exec.pRequest->dbList, pStmt->sql.pQuery->pDbList);
11✔
2082
    TSWAP(pStmt->exec.pRequest->tableList, pStmt->sql.pQuery->pTableList);
11✔
2083
    TSWAP(pStmt->exec.pRequest->targetTableList, pStmt->sql.pQuery->pTargetTableList);
11✔
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;
11✔
2092

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

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

2106
  STableDataCxt** pDataBlock = NULL;
63,163,052✔
2107

2108
  if (pStmt->exec.pCurrBlock) {
63,163,052✔
2109
    pDataBlock = &pStmt->exec.pCurrBlock;
63,954,030✔
2110
  } else {
2111
    pDataBlock =
2112
        (STableDataCxt**)taosHashGet(pStmt->exec.pBlockHash, pStmt->bInfo.tbFName, strlen(pStmt->bInfo.tbFName));
114,516✔
2113
    if (NULL == pDataBlock) {
115,044✔
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;
115,044✔
2118
    if (pStmt->sql.stbInterlaceMode) {
115,044✔
2119
      taosArrayDestroy(pStmt->exec.pCurrBlock->pData->aCol);
3,167✔
2120
      (*pDataBlock)->pData->aCol = NULL;
3,167✔
2121
    }
2122
    if (colIdx < -1) {
114,817✔
2123
      pStmt->sql.bindRowFormat = true;
×
2124
      taosArrayDestroy((*pDataBlock)->pData->aCol);
×
2125
      (*pDataBlock)->pData->aCol = taosArrayInit(20, POINTER_BYTES);
×
2126
    }
2127
  }
2128

2129
  int64_t startUs2 = taosGetTimestampUs();
64,088,598✔
2130
  pStmt->stat.bindDataUs1 += startUs2 - startUs;
64,088,598✔
2131

2132
  SStmtQNode* param = NULL;
64,432,282✔
2133
  if (pStmt->sql.stbInterlaceMode) {
63,066,684✔
2134
    STMT_ERR_RET(stmtAllocQNodeFromBuf(&pStmt->sql.siInfo.tbBuf, (void**)&param));
2,082,350✔
2135
    STMT_ERR_RET(stmtGetTableColsFromCache(pStmt, &param->tblData.aCol));
2,082,051✔
2136
    taosArrayClear(param->tblData.aCol);
1,041,127✔
2137

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

2140
    param->restoreTbCols = false;
1,040,083✔
2141
    param->tblData.isOrdered = true;
1,040,125✔
2142
    param->tblData.isDuplicateTs = false;
1,040,125✔
2143
    tstrncpy(param->tblData.tbName, pStmt->bInfo.tbName, TSDB_TABLE_NAME_LEN);
1,040,455✔
2144

2145
    param->pCreateTbReq = pCreateTbReq;
1,039,939✔
2146
  }
2147

2148
  int64_t startUs3 = taosGetTimestampUs();
63,540,757✔
2149
  pStmt->stat.bindDataUs2 += startUs3 - startUs2;
63,540,757✔
2150

2151
  SArray*   pCols = pStmt->sql.stbInterlaceMode ? param->tblData.aCol : (*pDataBlock)->pData->aCol;
62,571,990✔
2152
  SBlobSet* pBlob = NULL;
64,691,487✔
2153
  if (colIdx < 0) {
64,737,651✔
2154
    if (pStmt->sql.stbInterlaceMode) {
64,828,505✔
2155
      (*pDataBlock)->pData->flags &= ~SUBMIT_REQ_COLUMN_DATA_FORMAT;
1,041,489✔
2156
      code = qBindStmtStbColsValue2(*pDataBlock, pCols, pStmt->bInfo.fixedValueCols, bind, pStmt->exec.pRequest->msgBuf,
1,332,038✔
2157
                                    pStmt->exec.pRequest->msgBufLen, &pStmt->sql.siInfo.pTSchema, pStmt->sql.pBindInfo,
1,041,687✔
2158
                                    pStmt->taos->optionInfo.charsetCxt, &pBlob);
1,041,749✔
2159
      param->tblData.isOrdered = (*pDataBlock)->ordered;
1,041,739✔
2160
      param->tblData.isDuplicateTs = (*pDataBlock)->duplicateTs;
1,041,792✔
2161
    } else {
2162
      if (colIdx == -1) {
63,774,839✔
2163
        if (pStmt->sql.bindRowFormat) {
62,347,171✔
2164
          STMT2_ELOG_E("can't mix bind row format and bind column format");
×
2165
          STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
×
2166
        }
2167
        code = qBindStmtColsValue2(*pDataBlock, pCols, pStmt->bInfo.fixedValueCols, bind, pStmt->exec.pRequest->msgBuf,
121,454,609✔
2168
                                   pStmt->exec.pRequest->msgBufLen, pStmt->taos->optionInfo.charsetCxt);
120,914,943✔
2169
      } else {
2170
        code =
2171
            qBindStmt2RowValue(*pDataBlock, (*pDataBlock)->pData->aRowP, pStmt->bInfo.fixedValueCols, bind,
23,598✔
2172
                               pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen,
23,598✔
2173
                               &pStmt->sql.siInfo.pTSchema, pStmt->sql.pBindInfo, pStmt->taos->optionInfo.charsetCxt);
1,427,668✔
2174
      }
2175
    }
2176

2177
    if (code) {
62,788,557✔
2178
      STMT2_ELOG("bind cols or rows failed, error:%s", tstrerror(code));
×
2179
      STMT_ERR_RET(code);
×
2180
    }
2181
  } else {
2182
    if (pStmt->sql.stbInterlaceMode) {
×
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) {
×
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) {
×
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;
×
2198

2199
    if (0 == colIdx) {
×
2200
      pStmt->bInfo.sBindRowNum = bind->num;
×
2201
    }
2202

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

2212
  int64_t startUs4 = taosGetTimestampUs();
63,097,858✔
2213
  pStmt->stat.bindDataUs3 += startUs4 - startUs3;
63,097,858✔
2214

2215
  if (pStmt->stbInterlaceMode) {
63,692,219✔
2216
    if (param) param->tblData.pBlobSet = pBlob;
64,256,635✔
2217
  }
2218

2219
  if (pStmt->sql.stbInterlaceMode) {
63,889,577✔
2220
    STMT_ERR_RET(stmtAppendTablePostHandle(pStmt, param));
1,041,728✔
2221
  } else {
2222
    STMT_ERR_RET(stmtAddBatch2(pStmt));
63,460,380✔
2223
  }
2224

2225
  pStmt->stat.bindDataUs4 += taosGetTimestampUs() - startUs4;
63,448,376✔
2226
  return TSDB_CODE_SUCCESS;
64,420,448✔
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) {
×
2363
  const STscObj* pTscObj = pRequest->pTscObj;
×
2364

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

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

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

2404
  if (code == TSDB_CODE_SUCCESS) {
×
2405
    pStmt->exec.affectedRows = taos_affected_rows(res);
×
2406
    pStmt->affectedRows += pStmt->exec.affectedRows;
×
2407
  }
2408

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

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

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

2427
int stmtExec2(TAOS_STMT2* stmt, int* affected_rows) {
610,996✔
2428
  STscStmt2* pStmt = (STscStmt2*)stmt;
610,996✔
2429
  int32_t    code = 0;
610,996✔
2430
  int64_t    startUs = taosGetTimestampUs();
611,964✔
2431

2432
  STMT2_DLOG_E("start to exec");
611,964✔
2433

2434
  if (pStmt->errCode != TSDB_CODE_SUCCESS) {
611,964✔
2435
    return pStmt->errCode;
×
2436
  }
2437

2438
  STMT_ERR_RET(taosThreadMutexLock(&pStmt->asyncBindParam.mutex));
611,964✔
2439
  while (atomic_load_8((int8_t*)&pStmt->asyncBindParam.asyncBindNum) > 0) {
612,027✔
2440
    (void)taosThreadCondWait(&pStmt->asyncBindParam.waitCond, &pStmt->asyncBindParam.mutex);
×
2441
  }
2442
  STMT_ERR_RET(taosThreadMutexUnlock(&pStmt->asyncBindParam.mutex));
611,977✔
2443

2444
  if (pStmt->sql.stbInterlaceMode) {
611,235✔
2445
    STMT_ERR_RET(stmtAddBatch2(pStmt));
497,932✔
2446
  }
2447

2448
  STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_EXECUTE));
611,926✔
2449

2450
  if (STMT_TYPE_QUERY != pStmt->sql.type) {
611,913✔
2451
    if (pStmt->sql.stbInterlaceMode) {
611,952✔
2452
      int64_t startTs = taosGetTimestampUs();
497,881✔
2453
      // wait for stmt bind thread to finish
2454
      while (atomic_load_64(&pStmt->sql.siInfo.tbRemainNum)) {
1,894,336✔
2455
        taosUsleep(1);
1,396,467✔
2456
      }
2457

2458
      if (pStmt->errCode != TSDB_CODE_SUCCESS) {
497,855✔
2459
        return pStmt->errCode;
×
2460
      }
2461

2462
      pStmt->stat.execWaitUs += taosGetTimestampUs() - startTs;
497,868✔
2463
      STMT_ERR_RET(qBuildStmtFinOutput(pStmt->sql.pQuery, pStmt->sql.pVgHash, pStmt->sql.siInfo.pVgroupList));
497,906✔
2464
      taosHashCleanup(pStmt->sql.siInfo.pVgroupHash);
497,855✔
2465
      pStmt->sql.siInfo.pVgroupHash = NULL;
497,881✔
2466
      pStmt->sql.siInfo.pVgroupList = NULL;
497,881✔
2467
    } else {
2468
      tDestroySubmitTbData(pStmt->exec.pCurrTbData, TSDB_MSG_FLG_ENCODE);
114,109✔
2469
      taosMemoryFreeClear(pStmt->exec.pCurrTbData);
114,060✔
2470

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

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

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

2483
  if (!fp) {
611,951✔
2484
    launchQueryImpl(pStmt->exec.pRequest, pStmt->sql.pQuery, true, NULL);
611,951✔
2485

2486
    if (pStmt->exec.pRequest->code && NEED_CLIENT_HANDLE_ERROR(pStmt->exec.pRequest->code)) {
611,897✔
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);
×
2489
      if (code) {
×
2490
        pStmt->exec.pRequest->code = code;
×
2491

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

2498
    STMT_ERR_JRET(pStmt->exec.pRequest->code);
611,964✔
2499

2500
    pStmt->exec.affectedRows = taos_affected_rows(pStmt->exec.pRequest);
611,933✔
2501
    if (affected_rows) {
611,940✔
2502
      *affected_rows = pStmt->exec.affectedRows;
609,564✔
2503
    }
2504
    pStmt->affectedRows += pStmt->exec.affectedRows;
611,760✔
2505

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

2511
    STMT_ERR_RET(stmtCleanExecInfo(pStmt, (code ? false : true), false));
611,858✔
2512

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

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

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

2539
  STMT_RET(code);
611,988✔
2540
}
2541

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

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

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

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

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

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

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

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

2575
  if (pStmt->options.asyncExecFn && !pStmt->execSemWaited) {
116,662✔
2576
    if (tsem_wait(&pStmt->asyncExecSem) != 0) {
×
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) {
116,662✔
2584
    if (tsem_post(&pStmt->asyncExecSem) != 0) {
×
2585
      STMT2_ELOG_E("fail to post asyncExecSem");
×
2586
    }
2587
  }
2588

2589
  STMT2_DLOG("stbInterlaceMode:%d, statInfo: ctgGetTbMetaNum=>%" PRId64 ", getCacheTbInfo=>%" PRId64
116,662✔
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) {
116,662✔
2600
    pStmt->bInfo.tagsCached = false;
4,774✔
2601
  }
2602
  pStmt->bInfo.boundColsCached = false;
116,662✔
2603

2604
  STMT_ERR_RET(stmtCleanSQLInfo(pStmt));
116,662✔
2605

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

2613
  return TSDB_CODE_SUCCESS;
116,627✔
2614
}
2615

2616
const char* stmt2Errstr(TAOS_STMT2* stmt) {
×
2617
  STscStmt2* pStmt = (STscStmt2*)stmt;
×
2618

2619
  if (stmt == NULL || NULL == pStmt->exec.pRequest) {
×
2620
    return (char*)tstrerror(terrno);
×
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)) {
×
2625
    pStmt->exec.pRequest->code = terrno;
×
2626
  }
2627

2628
  SRequestObj* pRequest = pStmt->exec.pRequest;
×
2629
  if (NULL != pRequest->msgBuf && (strlen(pRequest->msgBuf) > 0 || pRequest->code == TSDB_CODE_RPC_FQDN_ERROR)) {
×
2630
    return pRequest->msgBuf;
×
2631
  }
2632
  return (const char*)tstrerror(pRequest->code);
×
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) {
27✔
2641
  int32_t    code = 0;
27✔
2642
  STscStmt2* pStmt = (STscStmt2*)stmt;
27✔
2643
  int32_t    preCode = pStmt->errCode;
27✔
2644

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

2649
  if (STMT_TYPE_QUERY == pStmt->sql.type) {
27✔
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));
27✔
2655

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

2664
  STMT_ERRI_JRET(stmtCreateRequest(pStmt));
27✔
2665

2666
  if (pStmt->bInfo.needParse) {
27✔
2667
    STMT_ERRI_JRET(stmtParseSql(pStmt));
27✔
2668
  }
2669

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

2676
  pStmt->errCode = preCode;
27✔
2677

2678
  return code;
27✔
2679
}
2680

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

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

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

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

2699
  STMT_ERRI_JRET(stmtSwitchStatus(pStmt, STMT_FETCH_FIELDS));
×
2700

2701
  if (pStmt->bInfo.needParse && pStmt->sql.runTimes && pStmt->sql.type > 0 &&
×
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) {
×
2707
    resetRequest(pStmt);
×
2708
  }
2709

2710
  STMT_ERRI_JRET(stmtCreateRequest(pStmt));
×
2711

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

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

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

2724
_return:
×
2725

2726
  pStmt->errCode = preCode;
×
2727

2728
  return code;
×
2729
}
2730

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

2734
  STMT2_TLOG_E("start to use result");
11✔
2735

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

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

2746
  if (tsUseAdapter) {
11✔
2747
    TAOS_RES* res = (TAOS_RES*)pStmt->exec.pRequest;
×
2748
    pStmt->exec.pRequest = NULL;
×
2749
    return res;
×
2750
  }
2751

2752
  return pStmt->exec.pRequest;
11✔
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) {
×
2775
  if (pStmt == NULL || msg == NULL) {
×
2776
    return;
×
2777
  }
2778

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

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

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

2790
  if (msgLen >= bufLen) {
×
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);
×
2796
    pStmt->exec.pRequest->msgBufLen = msgLen;
×
2797
  }
2798

2799
  return;
×
2800
}
2801

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

2806
  return errorCode;
×
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