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

taosdata / TDengine / #4822

27 Oct 2025 05:42AM UTC coverage: 59.732% (+1.0%) from 58.728%
#4822

push

travis-ci

web-flow
Merge pull request #33377 from taosdata/fix/main/rename-udf-path

fix: update UDF example links to correct file paths

121214 of 258518 branches covered (46.89%)

Branch coverage included in aggregate %.

193636 of 268583 relevant lines covered (72.1%)

4002399.5 hits per line

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

59.07
/source/libs/executor/src/scanoperator.c
1
/*
2
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3
 *
4
 * This program is free software: you can use, redistribute, and/or modify
5
 * it under the terms of the GNU Affero General Public License, version 3
6
 * or later ("AGPL"), as published by the Free Software Foundation.
7
 *
8
 * This program is distributed in the hope that it will be useful, but WITHOUT
9
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10
 * FITNESS FOR A PARTICULAR PURPOSE.
11
 *
12
 * You should have received a copy of the GNU Affero General Public License
13
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
14
 */
15

16
#include "executor.h"
17
#include "executorInt.h"
18
#include "filter.h"
19
#include "function.h"
20
#include "functionMgt.h"
21
#include "os.h"
22
#include "querynodes.h"
23
#include "streamexecutorInt.h"
24
#include "systable.h"
25
#include "tname.h"
26

27
#include "tdatablock.h"
28
#include "tmsg.h"
29
#include "ttime.h"
30

31
#include "operator.h"
32
#include "query.h"
33
#include "querytask.h"
34
#include "tcompare.h"
35
#include "thash.h"
36
#include "ttypes.h"
37

38
#include "function.h"
39
#include "storageapi.h"
40
#include "wal.h"
41

42
int32_t scanDebug = 0;
43

44
#define MULTI_READER_MAX_TABLE_NUM     5000
45
#define SET_REVERSE_SCAN_FLAG(_info)   ((_info)->scanFlag = REVERSE_SCAN)
46
#define SWITCH_ORDER(n)                (((n) = ((n) == TSDB_ORDER_ASC) ? TSDB_ORDER_DESC : TSDB_ORDER_ASC))
47
#define STREAM_SCAN_OP_NAME            "StreamScanOperator"
48
#define STREAM_SCAN_OP_STATE_NAME      "StreamScanFillHistoryState"
49
#define STREAM_SCAN_OP_CHECKPOINT_NAME "StreamScanOperator_Checkpoint"
50

51
typedef struct STableMergeScanExecInfo {
52
  SFileBlockLoadRecorder blockRecorder;
53
  SSortExecInfo          sortExecInfo;
54
} STableMergeScanExecInfo;
55

56
typedef struct STableMergeScanSortSourceParam {
57
  SOperatorInfo* pOperator;
58
  int32_t        readerIdx;
59
  uint64_t       uid;
60
  STsdbReader*   reader;
61
} STableMergeScanSortSourceParam;
62

63
typedef struct STableCountScanOperatorInfo {
64
  SReadHandle  readHandle;
65
  SSDataBlock* pRes;
66

67
  STableCountScanSupp supp;
68

69
  int32_t currGrpIdx;
70
  SArray* stbUidList;  // when group by db_name and/or stable_name
71
} STableCountScanOperatorInfo;
72

73
static bool    processBlockWithProbability(const SSampleExecInfo* pInfo);
74
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes);
75

76
bool processBlockWithProbability(const SSampleExecInfo* pInfo) {
987,068✔
77
#if 0
78
  if (pInfo->sampleRatio == 1) {
79
    return true;
80
  }
81

82
  uint32_t val = taosRandR((uint32_t*) &pInfo->seed);
83
  return (val % ((uint32_t)(1/pInfo->sampleRatio))) == 0;
84
#else
85
  return true;
987,068✔
86
#endif
87
}
88

89
static void switchCtxOrder(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
152✔
90
  for (int32_t i = 0; i < numOfOutput; ++i) {
152!
91
    SWITCH_ORDER(pCtx[i].order);
×
92
  }
93
}
152✔
94

95
static int32_t overlapWithTimeWindow(SInterval* pInterval, SDataBlockInfo* pBlockInfo, int32_t order, bool* overlap) {
1,024,829✔
96
  int32_t     code = TSDB_CODE_SUCCESS;
1,024,829✔
97
  STimeWindow w = {0};
1,024,829✔
98

99
  // 0 by default, which means it is not a interval operator of the upstream operator.
100
  if (pInterval->interval == 0) {
1,024,829✔
101
    *overlap = false;
948,317✔
102
    return code;
948,317✔
103
  }
104

105
  if (order == TSDB_ORDER_ASC) {
76,512✔
106
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
74,450✔
107
    if (w.ekey < pBlockInfo->window.skey) {
74,423!
108
      qError("w.ekey:%" PRId64 " < pBlockInfo->window.skey:%" PRId64, w.ekey, pBlockInfo->window.skey);
×
109
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
110
    }
111

112
    if (w.ekey < pBlockInfo->window.ekey) {
74,423✔
113
      *overlap = true;
65,770✔
114
      return code;
65,770✔
115
    }
116

117
    while (1) {
118
      getNextTimeWindow(pInterval, &w, order);
8,666✔
119
      if (w.skey > pBlockInfo->window.ekey) {
8,625✔
120
        break;
8,040✔
121
      }
122

123
      if (w.ekey <= pBlockInfo->window.ekey) {
585!
124
        qError("w.ekey:%" PRId64 " <= pBlockInfo->window.ekey:%" PRId64, w.ekey, pBlockInfo->window.ekey);
×
125
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
126
      }
127
      if (TMAX(w.skey, pBlockInfo->window.skey) <= pBlockInfo->window.ekey) {
585✔
128
        *overlap = true;
572✔
129
        return code;
572✔
130
      }
131
    }
132
  } else {
133
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.ekey);
2,062✔
134
    if (w.skey > pBlockInfo->window.ekey) {
2,122!
135
      qError("w.skey:%" PRId64 " > pBlockInfo->window.skey:%" PRId64, w.skey, pBlockInfo->window.ekey);
×
136
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
137
    }
138

139
    if (w.skey > pBlockInfo->window.skey) {
2,122✔
140
      *overlap = true;
1,839✔
141
      return code;
1,839✔
142
    }
143

144
    while (1) {
145
      getNextTimeWindow(pInterval, &w, order);
283✔
146
      if (w.ekey < pBlockInfo->window.skey) {
283✔
147
        break;
277✔
148
      }
149

150
      if (w.skey >= pBlockInfo->window.skey) {
6!
151
        qError("w.skey:%" PRId64 " >= pBlockInfo->window.skey:%" PRId64, w.skey, pBlockInfo->window.skey);
×
152
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
153
      }
154
      if (pBlockInfo->window.skey <= TMIN(w.ekey, pBlockInfo->window.ekey)) {
6!
155
        *overlap = true;
6✔
156
        return code;
6✔
157
      }
158
    }
159
  }
160

161
  *overlap = false;
8,317✔
162
  return code;
8,317✔
163
}
164

165
// this function is for table scanner to extract temporary results of upstream aggregate results.
166
static SResultRow* getTableGroupOutputBuf(SOperatorInfo* pOperator, uint64_t groupId, SFilePage** pPage) {
147,502✔
167
  if (pOperator->operatorType != QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
147,502!
168
    return NULL;
×
169
  }
170

171
  int64_t buf[2] = {0};
147,502✔
172
  SET_RES_WINDOW_KEY((char*)buf, &groupId, sizeof(groupId), groupId);
147,502✔
173

174
  STableScanInfo* pTableScanInfo = pOperator->info;
147,502✔
175

176
  SResultRowPosition* p1 = (SResultRowPosition*)tSimpleHashGet(pTableScanInfo->base.pdInfo.pAggSup->pResultRowHashTable,
147,502✔
177
                                                               buf, GET_RES_WINDOW_KEY_LEN(sizeof(groupId)));
178

179
  if (p1 == NULL) {
147,499✔
180
    return NULL;
81,190✔
181
  }
182

183
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
66,309✔
184
  if (NULL == *pPage) {
66,310!
185
    return NULL;
×
186
  }
187

188
  return (SResultRow*)((char*)(*pPage) + p1->offset);
66,310✔
189
}
190

191
static int32_t insertTableToScanIgnoreList(STableScanInfo* pTableScanInfo, uint64_t uid) {
682✔
192
  if (NULL == pTableScanInfo->pIgnoreTables) {
682✔
193
    int32_t tableNum = taosArrayGetSize(pTableScanInfo->base.pTableListInfo->pTableList);
360✔
194
    pTableScanInfo->pIgnoreTables =
360✔
195
        taosHashInit(tableNum, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, HASH_NO_LOCK);
360✔
196
    if (NULL == pTableScanInfo->pIgnoreTables) {
360!
197
      return terrno;
×
198
    }
199
  }
200

201
  int32_t tempRes = taosHashPut(pTableScanInfo->pIgnoreTables, &uid, sizeof(uid), &pTableScanInfo->scanTimes,
682✔
202
                                sizeof(pTableScanInfo->scanTimes));
203
  if (tempRes != TSDB_CODE_SUCCESS && tempRes != TSDB_CODE_DUP_KEY) {
683!
204
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(tempRes));
×
205
    return tempRes;
×
206
  }
207
  return TSDB_CODE_SUCCESS;
683✔
208
}
209

210
static int32_t doDynamicPruneDataBlock(SOperatorInfo* pOperator, SDataBlockInfo* pBlockInfo, uint32_t* status) {
1,161,791✔
211
  STableScanInfo* pTableScanInfo = pOperator->info;
1,161,791✔
212
  int32_t         code = TSDB_CODE_SUCCESS;
1,161,791✔
213

214
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
1,161,791✔
215
    return TSDB_CODE_SUCCESS;
1,014,667✔
216
  }
217

218
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
147,124✔
219

220
  SFilePage*  pPage = NULL;
147,124✔
221
  SResultRow* pRow = getTableGroupOutputBuf(pOperator, pBlockInfo->id.groupId, &pPage);
147,124✔
222

223
  if (pRow == NULL) {
147,494✔
224
    return TSDB_CODE_SUCCESS;
81,189✔
225
  }
226

227
  bool notLoadBlock = true;
66,305✔
228
  for (int32_t i = 0; i < pSup1->numOfExprs; ++i) {
69,641✔
229
    int32_t functionId = pSup1->pCtx[i].functionId;
68,960✔
230

231
    SResultRowEntryInfo* pEntry = getResultEntryInfo(pRow, i, pTableScanInfo->base.pdInfo.pExprSup->rowEntryInfoOffset);
68,960✔
232

233
    EFuncDataRequired reqStatus = fmFuncDynDataRequired(functionId, pEntry, pBlockInfo);
68,960✔
234
    if (reqStatus != FUNC_DATA_REQUIRED_NOT_LOAD) {
68,965✔
235
      notLoadBlock = false;
65,629✔
236
      break;
65,629✔
237
    }
238
  }
239

240
  // release buffer pages
241
  releaseBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, pPage);
66,310✔
242

243
  if (notLoadBlock) {
66,311✔
244
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
682✔
245
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
682✔
246
  }
247

248
  return code;
66,312✔
249
}
250

251
static int32_t doFilterByBlockSMA(SFilterInfo* pFilterInfo, SColumnDataAgg* pColsAgg, int32_t numOfCols,
29,616✔
252
                                  int32_t numOfRows, bool* keep) {
253
  if (pColsAgg == NULL || pFilterInfo == NULL) {
29,616!
254
    *keep = true;
×
255
    return TSDB_CODE_SUCCESS;
×
256
  }
257

258
  return filterRangeExecute(pFilterInfo, pColsAgg, numOfCols, numOfRows, keep);
29,616✔
259
}
260

261
static int32_t doLoadBlockSMA(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
273,250✔
262
                              bool* pLoad) {
263
  SStorageAPI* pAPI = &pTaskInfo->storageAPI;
273,250✔
264
  bool         allColumnsHaveAgg = true;
273,250✔
265
  bool         hasNullSMA = false;
273,250✔
266
  if (pLoad != NULL) {
273,250!
267
    *pLoad = false;
273,336✔
268
  }
269

270
  int32_t code = pAPI->tsdReader.tsdReaderRetrieveBlockSMAInfo(pTableScanInfo->dataReader, pBlock, &allColumnsHaveAgg,
273,250✔
271
                                                               &hasNullSMA);
272
  if (code != TSDB_CODE_SUCCESS) {
273,132!
273
    return code;
×
274
  }
275

276
  if (!allColumnsHaveAgg || hasNullSMA) {
273,132!
277
    *pLoad = false;
232,273✔
278
  } else {
279
    *pLoad = true;
40,859✔
280
  }
281

282
  return code;
273,132✔
283
}
284

285
static int32_t doSetTagColumnData(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
1,293,822✔
286
                                  int32_t rows) {
287
  int32_t    code = 0;
1,293,822✔
288
  SExprSupp* pSup = &pTableScanInfo->pseudoSup;
1,293,822✔
289
  if (pSup->numOfExprs > 0) {
1,293,822✔
290
    code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
611,138✔
291
                                  pTaskInfo, &pTableScanInfo->metaCache);
292
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
293
    if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
611,375!
294
      if (pTaskInfo->streamInfo.pState) blockDataCleanup(pBlock);
×
295
      code = 0;
×
296
    }
297
  }
298

299
  return code;
1,294,347✔
300
}
301

302
bool applyLimitOffset(SLimitInfo* pLimitInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
2,165,856✔
303
  SLimit*     pLimit = &pLimitInfo->limit;
2,165,856✔
304
  const char* id = GET_TASKID(pTaskInfo);
2,165,856✔
305

306
  if (pLimitInfo->remainOffset > 0) {
2,165,856✔
307
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
11,129✔
308
      pLimitInfo->remainOffset -= pBlock->info.rows;
5,622✔
309
      blockDataEmpty(pBlock);
5,622✔
310
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
5,622✔
311
      return false;
5,622✔
312
    } else {
313
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
5,507✔
314
      if (code != TSDB_CODE_SUCCESS) {
5,507!
315
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
316
        pTaskInfo->code = code;
×
317
        T_LONG_JMP(pTaskInfo->env, code);
×
318
      }
319
      pLimitInfo->remainOffset = 0;
5,507✔
320
    }
321
  }
322

323
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
2,160,234✔
324
    // limit the output rows
325
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
47,849✔
326
    blockDataKeepFirstNRows(pBlock, keep);
47,849✔
327

328
    pLimitInfo->numOfOutputRows += pBlock->info.rows;
47,852✔
329
    qDebug("output limit %" PRId64 " has reached, %s", pLimit->limit, id);
47,852✔
330
    return true;
47,854✔
331
  }
332

333
  pLimitInfo->numOfOutputRows += pBlock->info.rows;
2,112,385✔
334
  return false;
2,112,385✔
335
}
336

337
static bool isDynVtbScan(SOperatorInfo* pOperator) {
46,511✔
338
  return pOperator->dynamicTask && ((STableScanInfo*)(pOperator->info))->virtualStableScan;
46,511!
339
}
340

341
static int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanBase* pTableScanInfo, SSDataBlock* pBlock,
1,237,612✔
342
                             uint32_t* status) {
343
  int32_t        code = TSDB_CODE_SUCCESS;
1,237,612✔
344
  int32_t        lino = 0;
1,237,612✔
345
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
1,237,612✔
346
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
1,237,612✔
347
  bool           loadSMA = false;
1,237,612✔
348

349
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
1,237,612✔
350

351
  pCost->totalBlocks += 1;
1,237,612✔
352
  pCost->totalRows += pBlock->info.rows;
1,237,612✔
353
  *status = pTableScanInfo->dataBlockLoadFlag;
1,237,612✔
354

355
  if (pOperator->exprSupp.pFilterInfo != NULL) {
1,237,612✔
356
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
212,806✔
357
  } else {
358
    bool overlap = false;
1,024,806✔
359
    int  ret =
360
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
1,024,806✔
361
    if (ret != TSDB_CODE_SUCCESS) {
1,024,751!
362
      return ret;
×
363
    }
364
    if (overlap) {
1,024,751✔
365
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
68,173✔
366
    }
367
  }
368

369
  SDataBlockInfo* pBlockInfo = &pBlock->info;
1,237,557✔
370
  taosMemoryFreeClear(pBlock->pBlockAgg);
1,237,557!
371

372
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
1,237,557!
373
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
×
374
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
375
    pCost->filterOutBlocks += 1;
×
376
    pCost->totalRows += pBlock->info.rows;
×
377
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
378
    return TSDB_CODE_SUCCESS;
×
379
  } else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
1,237,557✔
380
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
60,733✔
381
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows,
382
           pBlockInfo->id.uid);
383
    code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
60,764✔
384
    pCost->skipBlocks += 1;
60,785✔
385
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
60,785✔
386
    return code;
60,786✔
387
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
1,176,824✔
388
    pCost->loadBlockStatis += 1;
60,558✔
389
    loadSMA = true;  // mark the operation of load sma;
60,558✔
390
    bool success = true;
60,558✔
391
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
60,558✔
392
    if (code) {
60,413!
393
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
394
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
395
      QUERY_CHECK_CODE(code, lino, _end);
×
396
    }
397

398
    if (success) {  // failed to load the block sma data, data block statistics does not exist, load data block instead
60,408✔
399
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
11,246✔
400
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
401
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
11,251✔
402
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
11,233✔
403
      return code;
11,722✔
404
    } else {
405
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
49,162✔
406
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
49,203✔
407
    }
408
  }
409

410
  if (*status != FUNC_DATA_REQUIRED_DATA_LOAD) {
1,165,469!
411
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
412
    qError("%s loadDataBlock invalid status:%d", GET_TASKID(pTaskInfo), *status);
×
413
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
414
  }
415

416
  // try to filter data block according to sma info
417
  if (pOperator->exprSupp.pFilterInfo != NULL && (!loadSMA)) {
1,165,469!
418
    bool success = true;
212,730✔
419
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
212,730✔
420
    if (code) {
212,726!
421
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
422
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
423
      QUERY_CHECK_CODE(code, lino, _end);
×
424
    }
425

426
    if (success) {
212,671✔
427
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
29,617✔
428
      bool   keep = false;
29,617✔
429
      code = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows, &keep);
29,617✔
430
      if (code) {
29,618!
431
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
432
        qError("%s failed to do filter by block sma, code:%s", GET_TASKID(pTaskInfo), tstrerror(code));
×
433
        QUERY_CHECK_CODE(code, lino, _end);
×
434
      }
435

436
      if (!keep) {
29,618✔
437
        qDebug("%s data block filter out by block SMA, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
2,888!
438
               GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
439
        pCost->filterOutBlocks += 1;
2,888✔
440
        (*status) = FUNC_DATA_REQUIRED_FILTEROUT;
2,888✔
441
        taosMemoryFreeClear(pBlock->pBlockAgg);
2,888!
442

443
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
2,888✔
444
        return TSDB_CODE_SUCCESS;
2,888✔
445
      }
446
    }
447
  }
448

449
  // free the sma info, since it should not be involved in *later computing process.
450
  taosMemoryFreeClear(pBlock->pBlockAgg);
1,162,523!
451

452
  // try to filter data block according to current results
453
  code = doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
1,162,524✔
454
  if (code) {
1,161,984✔
455
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
71✔
456
    QUERY_CHECK_CODE(code, lino, _end);
×
457
  }
458

459
  if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
1,161,913✔
460
    qDebug("%s data block skipped due to dynamic prune, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
683!
461
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
462
    pCost->skipBlocks += 1;
683✔
463
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
683✔
464

465
    STableScanInfo* p1 = pOperator->info;
683✔
466
    if (taosHashGetSize(p1->pIgnoreTables) == taosArrayGetSize(p1->base.pTableListInfo->pTableList)) {
683✔
467
      *status = FUNC_DATA_REQUIRED_ALL_FILTEROUT;
76✔
468
    } else {
469
      *status = FUNC_DATA_REQUIRED_FILTEROUT;
607✔
470
    }
471
    return TSDB_CODE_SUCCESS;
683✔
472
  }
473

474
  pCost->totalCheckedRows += pBlock->info.rows;
1,161,230✔
475
  pCost->loadBlocks += 1;
1,161,230✔
476

477
  SSDataBlock* p = NULL;
1,161,230✔
478
  code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pTableScanInfo->dataReader, &p, NULL);
1,161,230✔
479
  if (p == NULL || code != TSDB_CODE_SUCCESS || p != pBlock) {
1,161,149!
480
    return code;
×
481
  }
482

483
  if ((pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) &&
1,161,525✔
484
      ((STableScanInfo*)pOperator->info)->ignoreTag) {
911,669✔
485
    // do nothing
486
  } else {
487
    // dyn vtb scan do not read tag from origin tables.
488
    code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
1,129,172✔
489
    if (code) {
1,129,433!
490
      return code;
×
491
    }
492
  }
493

494
  // restore the previous value
495
  pCost->totalRows -= pBlock->info.rows;
1,161,786✔
496

497
  if (pOperator->exprSupp.pFilterInfo != NULL) {
1,161,786✔
498
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo, NULL);
209,839✔
499
    QUERY_CHECK_CODE(code, lino, _end);
209,794✔
500

501
    int64_t st = taosGetTimestampUs();
209,793✔
502
    double  el = (taosGetTimestampUs() - st) / 1000.0;
209,835✔
503
    pTableScanInfo->readRecorder.filterTime += el;
209,835✔
504

505
    if (pBlock->info.rows == 0) {
209,835✔
506
      pCost->filterOutBlocks += 1;
28,579✔
507
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
28,579✔
508
             GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows, el);
509
    } else {
510
      qDebug("%s data block filter applied, elapsed time:%.2f ms", GET_TASKID(pTaskInfo), el);
181,256✔
511
    }
512
  }
513

514
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
1,161,809✔
515
  if (limitReached) {  // set operator flag is done
1,161,521✔
516
    setOperatorCompleted(pOperator);
7,168✔
517
  }
518

519
  pCost->totalRows += pBlock->info.rows;
1,161,803✔
520

521
_end:
1,161,804✔
522
  if (code != TSDB_CODE_SUCCESS) {
1,161,804✔
523
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1!
524
  }
525
  return code;
1,161,886✔
526
}
527

528
static void prepareForDescendingScan(STableScanBase* pTableScanInfo, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
152✔
529
  SET_REVERSE_SCAN_FLAG(pTableScanInfo);
152✔
530

531
  switchCtxOrder(pCtx, numOfOutput);
152✔
532
  pTableScanInfo->cond.order = TSDB_ORDER_DESC;
152✔
533
  // STimeWindow* pTWindow = &pTableScanInfo->cond.twindows;
534
  // TSWAP(pTWindow->skey, pTWindow->ekey);
535
}
152✔
536

537
typedef struct STableCachedVal {
538
  const char* pName;
539
  STag*       pTags;
540
} STableCachedVal;
541

542
static void freeTableCachedVal(void* param) {
531,008✔
543
  if (param == NULL) {
531,008!
544
    return;
×
545
  }
546

547
  STableCachedVal* pVal = param;
531,008✔
548
  taosMemoryFree((void*)pVal->pName);
531,008!
549
  taosMemoryFree(pVal->pTags);
532,222!
550
  taosMemoryFree(pVal);
532,269!
551
}
552

553
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
532,184✔
554
  int32_t          code = TSDB_CODE_SUCCESS;
532,184✔
555
  int32_t          lino = 0;
532,184✔
556
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
532,184!
557
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
532,330!
558

559
  pVal->pTags = NULL;
532,330✔
560
  pVal->pName = taosStrdup(pMetaReader->me.name);
532,330!
561
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
532,205!
562

563
  // only child table has tag value
564
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
532,205✔
565
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
532,057✔
566
    pVal->pTags = taosMemoryMalloc(pTag->len);
532,057✔
567
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
531,981!
568
    memcpy(pVal->pTags, pTag, pTag->len);
531,981✔
569
  }
570

571
  (*ppResVal) = pVal;
532,129✔
572

573
_end:
532,129✔
574
  if (code != TSDB_CODE_SUCCESS) {
532,129!
575
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
576
    freeTableCachedVal(pVal);
×
577
  }
578
  return code;
532,067✔
579
}
580

581
// const void *key, size_t keyLen, void *value
582
static void freeCachedMetaItem(const void* key, size_t keyLen, void* value, void* ud) {
530,974✔
583
  (void)key;
584
  (void)keyLen;
585
  (void)ud;
586
  freeTableCachedVal(value);
530,974✔
587
}
532,314✔
588

589
static void doSetNullValue(SSDataBlock* pBlock, const SExprInfo* pExpr, int32_t numOfExpr) {
×
590
  for (int32_t j = 0; j < numOfExpr; ++j) {
×
591
    int32_t dstSlotId = pExpr[j].base.resSchema.slotId;
×
592

593
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
×
594
    colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
×
595
  }
596
}
×
597

598
static void freeTableCachedValObj(STableCachedVal* pVal) {
×
599
  taosMemoryFree((void*)pVal->pName);
×
600
  taosMemoryFree(pVal->pTags);
×
601
}
×
602

603
int32_t addTagPseudoColumnData(SReadHandle* pHandle, const SExprInfo* pExpr, int32_t numOfExpr, SSDataBlock* pBlock,
714,564✔
604
                               int32_t rows, SExecTaskInfo* pTask, STableMetaCacheInfo* pCache) {
605
  int32_t          code = TSDB_CODE_SUCCESS;
714,564✔
606
  int32_t          lino = 0;
714,564✔
607
  bool             freeReader = false;
714,564✔
608
  LRUHandle*       h = NULL;
714,564✔
609
  STableCachedVal  val = {0};
714,564✔
610
  SMetaReader      mr = {0};
714,564✔
611
  const char*      idStr = pTask->id.str;
714,564✔
612
  int32_t          insertRet = TAOS_LRU_STATUS_OK;
714,564✔
613
  STableCachedVal* pVal = NULL;
714,564✔
614

615
  // currently only the tbname pseudo column
616
  if (numOfExpr <= 0) {
714,564✔
617
    return TSDB_CODE_SUCCESS;
172✔
618
  }
619

620
  // todo: opt if only require the vgId and the vgVer;
621

622
  // backup the rows
623
  int32_t backupRows = pBlock->info.rows;
714,392✔
624
  pBlock->info.rows = rows;
714,392✔
625

626
  // todo refactor: extract method
627
  // the handling of the null data should be packed in the extracted method
628

629
  // 1. check if it is existed in meta cache
630
  if (pCache == NULL) {
714,392✔
631
    pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
208✔
632
    code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
208✔
633
    if (code != TSDB_CODE_SUCCESS) {
208!
634
      // when encounter the TSDB_CODE_PAR_TABLE_NOT_EXIST error, we proceed.
635
      if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
×
636
        qWarn("failed to get table meta, table may have been dropped, uid:0x%" PRIx64 ", code:%s, %s",
×
637
              pBlock->info.id.uid, tstrerror(code), idStr);
638

639
        // append null value before return to caller, since the caller will ignore this error code and proceed
640
        doSetNullValue(pBlock, pExpr, numOfExpr);
×
641
      } else {
642
        qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", pBlock->info.id.uid, tstrerror(code), idStr);
×
643
      }
644
      pHandle->api.metaReaderFn.clearReader(&mr);
×
645
      return code;
×
646
    }
647

648
    pHandle->api.metaReaderFn.readerReleaseLock(&mr);
208✔
649

650
    val.pName = mr.me.name;
208✔
651
    val.pTags = (STag*)mr.me.ctbEntry.pTags;
208✔
652

653
    freeReader = true;
208✔
654
  } else {
655
    pCache->metaFetch += 1;
714,184✔
656

657
    h = taosLRUCacheLookup(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(pBlock->info.id.uid));
714,184✔
658
    if (h == NULL) {
714,437✔
659
      pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
532,383✔
660
      freeReader = true;
532,410✔
661
      code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
532,410✔
662
      if (code != TSDB_CODE_SUCCESS) {
531,804!
663
        if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
×
664
          qWarn("failed to get table meta, table may have been dropped, uid:0x%" PRIx64 ", code:%s, %s",
×
665
                pBlock->info.id.uid, tstrerror(code), idStr);
666
          // append null value before return to caller, since the caller will ignore this error code and proceed
667
          doSetNullValue(pBlock, pExpr, numOfExpr);
×
668
        } else {
669
          qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", pBlock->info.id.uid, tstrerror(code),
×
670
                 idStr);
671
        }
672
        pHandle->api.metaReaderFn.clearReader(&mr);
×
673
        return code;
×
674
      }
675

676
      pHandle->api.metaReaderFn.readerReleaseLock(&mr);
531,804✔
677

678
      code = createTableCacheVal(&mr, &pVal);
532,289✔
679
      QUERY_CHECK_CODE(code, lino, _end);
532,094!
680

681
      val = *pVal;
532,094✔
682
    } else {
683
      pCache->cacheHit += 1;
182,054✔
684
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
182,054✔
685
      val = *pValTmp;
182,054✔
686

687
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
182,054✔
688
      qTrace("release LRU cache, res %d", bRes);
182,082!
689
    }
690

691
    qDebug("retrieve table meta from cache:%" PRIu64 ", hit:%" PRIu64 " miss:%" PRIu64 ", %s", pCache->metaFetch,
714,176✔
692
           pCache->cacheHit, (pCache->metaFetch - pCache->cacheHit), idStr);
693
  }
694

695
  for (int32_t j = 0; j < numOfExpr; ++j) {
2,429,270✔
696
    const SExprInfo* pExpr1 = &pExpr[j];
1,714,516✔
697
    int32_t          dstSlotId = pExpr1->base.resSchema.slotId;
1,714,516✔
698

699
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
1,714,516✔
700
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
1,714,485✔
701

702
    int32_t functionId = pExpr1->pExpr->_function.functionId;
1,714,413✔
703

704
    // this is to handle the tbname
705
    if (fmIsScanPseudoColumnFunc(functionId)) {
1,714,413✔
706
      int32_t fType = pExpr1->pExpr->_function.functionType;
168,957✔
707
      if (fType == FUNCTION_TYPE_TBNAME) {
168,957!
708
        code = setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
168,957✔
709
        QUERY_CHECK_CODE(code, lino, _end);
168,950!
710
      } else if (fType == FUNCTION_TYPE_VGID) {
×
711
        code = setVgIdColData(pBlock, pColInfoData, functionId, pTask->id.vgId);
×
712
        QUERY_CHECK_CODE(code, lino, _end);
×
713
      } else if (fType == FUNCTION_TYPE_VGVER) {
×
714
        code = setVgVerColData(pBlock, pColInfoData, functionId, pBlock->info.version);
×
715
        QUERY_CHECK_CODE(code, lino, _end);
×
716
      }
717
    } else {  // these are tags
718
      STagVal tagVal = {0};
1,545,424✔
719
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
1,545,424✔
720
      const char* p = pHandle->api.metaFn.extractTagVal(val.pTags, pColInfoData->info.type, &tagVal);
1,545,424✔
721

722
      char* data = NULL;
1,545,490✔
723
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
1,545,490✔
724
        data = tTagValToData((const STagVal*)p, false);
1,532,456✔
725
      } else {
726
        data = (char*)p;
13,034✔
727
      }
728

729
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
1,545,516✔
730
      if (isNullVal) {
1,545,516✔
731
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
12,057✔
732
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
1,533,459✔
733
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, 1, false);
1,532,461✔
734
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
1,532,462!
735
          char* tmp = taosMemoryCalloc(1, varDataLen(data) + 1);
599,226!
736
          if (tmp != NULL) {
599,339!
737
            memcpy(tmp, varDataVal(data), varDataLen(data));
599,339✔
738
            qDebug("get tag value:%s, cid:%d, table name:%s, uid%" PRId64, tmp, tagVal.cid, val.pName,
599,339✔
739
                   pBlock->info.id.uid);
740
            taosMemoryFree(tmp);
599,341✔
741
          }
742
          taosMemoryFree(data);
599,343!
743
        }
744
        QUERY_CHECK_CODE(code, lino, _end);
1,532,495!
745
      } else {  // todo opt for json tag
746
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
5,138✔
747
          code = colDataSetVal(pColInfoData, i, data, false);
4,140✔
748
          QUERY_CHECK_CODE(code, lino, _end);
4,140!
749
        }
750
      }
751
    }
752
  }
753

754
  // restore the rows
755
  pBlock->info.rows = backupRows;
714,754✔
756

757
_end:
714,754✔
758

759
  if (NULL != pVal) {
714,754✔
760
    insertRet =
532,453✔
761
        taosLRUCacheInsert(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(uint64_t), pVal,
532,458✔
762
                           sizeof(STableCachedVal), freeCachedMetaItem, NULL, NULL, TAOS_LRU_PRIORITY_LOW, NULL);
763
    if (insertRet != TAOS_LRU_STATUS_OK) {
532,453!
764
      qWarn("failed to put meta into lru cache, code:%d, %s", insertRet, idStr);
×
765
    }
766
  }
767

768
  if (freeReader) {
714,749✔
769
    pHandle->api.metaReaderFn.clearReader(&mr);
532,662✔
770
  }
771
  if (code != TSDB_CODE_SUCCESS) {
714,728!
772
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
773
  }
774
  return code;
714,728✔
775
}
776

777
int32_t setTbNameColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId,
168,956✔
778
                         const char* name) {
779
  int32_t                     code = TSDB_CODE_SUCCESS;
168,956✔
780
  int32_t                     lino = 0;
168,956✔
781
  struct SScalarFuncExecFuncs fpSet = {0};
168,956✔
782
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
168,956✔
783
  QUERY_CHECK_CODE(code, lino, _end);
168,956!
784

785
  size_t len = TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE;
168,956✔
786
  char   buf[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
168,956✔
787
  STR_TO_VARSTR(buf, name)
168,956✔
788

789
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
168,956✔
790

791
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
168,966✔
792
  QUERY_CHECK_CODE(code, lino, _end);
168,952!
793

794
  code = colDataSetVal(&infoData, 0, buf, false);
168,952✔
795
  QUERY_CHECK_CODE(code, lino, _end);
168,954!
796

797
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
168,954✔
798
  SScalarParam param = {.columnData = pColInfoData};
168,954✔
799

800
  if (fpSet.process != NULL) {
168,954!
801
    code = fpSet.process(&srcParam, 1, &param);
168,960✔
802
    QUERY_CHECK_CODE(code, lino, _end);
168,956!
803
  } else {
804
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
×
805
  }
806

807
  colDataDestroy(&infoData);
168,950✔
808

809
_end:
168,951✔
810
  if (code != TSDB_CODE_SUCCESS) {
168,951!
811
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
812
  }
813
  return code;
168,951✔
814
}
815

816
int32_t setVgIdColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId, int32_t vgId) {
×
817
  int32_t                     code = TSDB_CODE_SUCCESS;
×
818
  int32_t                     lino = 0;
×
819
  struct SScalarFuncExecFuncs fpSet = {0};
×
820
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
×
821
  QUERY_CHECK_CODE(code, lino, _end);
×
822

823
  SColumnInfoData infoData = createColumnInfoData(pColInfoData->info.type, pColInfoData->info.bytes, 1);
×
824

825
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
×
826
  QUERY_CHECK_CODE(code, lino, _end);
×
827

828
  code = colDataSetVal(&infoData, 0, (const char*)&vgId, false);
×
829
  QUERY_CHECK_CODE(code, lino, _end);
×
830

831
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
×
832
  SScalarParam param = {.columnData = pColInfoData};
×
833

834
  if (fpSet.process != NULL) {
×
835
    code = fpSet.process(&srcParam, 1, &param);
×
836
    QUERY_CHECK_CODE(code, lino, _end);
×
837
  } else {
838
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
×
839
  }
840

841
_end:
×
842
  colDataDestroy(&infoData);
×
843
  if (code != TSDB_CODE_SUCCESS) {
×
844
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
845
  }
846
  return code;
×
847
}
848

849
int32_t setVgVerColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId, int64_t vgVer) {
×
850
  int32_t                     code = TSDB_CODE_SUCCESS;
×
851
  int32_t                     lino = 0;
×
852
  struct SScalarFuncExecFuncs fpSet = {0};
×
853
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
×
854
  QUERY_CHECK_CODE(code, lino, _end);
×
855

856
  SColumnInfoData infoData = createColumnInfoData(pColInfoData->info.type, pColInfoData->info.bytes, 1);
×
857

858
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
×
859
  QUERY_CHECK_CODE(code, lino, _end);
×
860

861
  code = colDataSetVal(&infoData, 0, (const char*)&vgVer, false);
×
862
  QUERY_CHECK_CODE(code, lino, _end);
×
863

864
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
×
865
  SScalarParam param = {.columnData = pColInfoData};
×
866

867
  if (fpSet.process != NULL) {
×
868
    code = fpSet.process(&srcParam, 1, &param);
×
869
    QUERY_CHECK_CODE(code, lino, _end);
×
870
  } else {
871
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
×
872
  }
873

874
_end:
×
875
  colDataDestroy(&infoData);
×
876
  if (code != TSDB_CODE_SUCCESS) {
×
877
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
878
  }
879
  return code;
×
880
}
881

882
static int32_t initNextGroupScan(STableScanInfo* pInfo, STableKeyInfo** pKeyInfo, int32_t* size) {
314,055✔
883
  int32_t code = TSDB_CODE_SUCCESS;
314,055✔
884
  int32_t lino = 0;
314,055✔
885
  code = tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, pKeyInfo, size);
314,055✔
886
  QUERY_CHECK_CODE(code, lino, _end);
314,079!
887

888
  pInfo->tableStartIndex = TARRAY_ELEM_IDX(pInfo->base.pTableListInfo->pTableList, *pKeyInfo);
314,079✔
889
  pInfo->tableEndIndex = (pInfo->tableStartIndex + (*size) - 1);
314,079✔
890
  pInfo->pResBlock->info.blankFill = false;
314,079✔
891
  taosMemoryFreeClear(pInfo->pResBlock->pBlockAgg);
314,079!
892

893
  if (!pInfo->needCountEmptyTable) {
314,079✔
894
    pInfo->countState = TABLE_COUNT_STATE_END;
284,550✔
895
  } else {
896
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
29,529✔
897
  }
898

899
_end:
314,079✔
900
  if (code != TSDB_CODE_SUCCESS) {
314,079!
901
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
902
  }
903
  return code;
314,065✔
904
}
905

906
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
1,049,906✔
907
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
1,049,906✔
908
    return;
920,587✔
909
  }
910
  if (pInfo->base.pTableListInfo->groupOffset) {
129,319✔
911
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
731✔
912
  } else {
913
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
128,588✔
914
    if (code != TSDB_CODE_SUCCESS) {
128,760✔
915
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
10,933!
916
    }
917
  }
918
}
919

920
static SSDataBlock* getOneRowResultBlock(SExecTaskInfo* pTaskInfo, STableScanBase* pBase, SSDataBlock* pBlock,
92,896✔
921
                                         const STableKeyInfo* tbInfo) {
922
  blockDataEmpty(pBlock);
92,896✔
923
  pBlock->info.rows = 1;
92,884✔
924
  pBlock->info.id.uid = tbInfo->uid;
92,884✔
925
  pBlock->info.id.groupId = tbInfo->groupId;
92,884✔
926
  pBlock->info.blankFill = true;
92,884✔
927

928
  // only one row: set all col data to null & hasNull
929
  int32_t col_num = blockDataGetNumOfCols(pBlock);
92,884✔
930
  for (int32_t i = 0; i < col_num; ++i) {
381,386✔
931
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
288,558✔
932
    colDataSetNULL(pColInfoData, 0);
933
  }
934

935
  // set tag/tbname
936
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
92,828✔
937

938
  return pBlock;
92,908✔
939
}
940

941
static SSDataBlock* getBlockForEmptyTable(SOperatorInfo* pOperator, const STableKeyInfo* tbInfo) {
89,301✔
942
  STableScanInfo* pTableScanInfo = pOperator->info;
89,301✔
943
  SSDataBlock*    pBlock =
944
      getOneRowResultBlock(pOperator->pTaskInfo, &pTableScanInfo->base, pTableScanInfo->pResBlock, tbInfo);
89,301✔
945

946
  pOperator->resultInfo.totalRows++;
89,311✔
947
  return pBlock;
89,311✔
948
}
949

950
static int32_t doTableScanImplNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
1,266,130✔
951
  int32_t         code = TSDB_CODE_SUCCESS;
1,266,130✔
952
  int32_t         lino = 0;
1,266,130✔
953
  STableScanInfo* pTableScanInfo = pOperator->info;
1,266,130✔
954
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
1,266,130✔
955
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
1,266,130✔
956
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
1,266,130✔
957
  bool            hasNext = false;
1,266,130✔
958
  int64_t         st = taosGetTimestampUs();
1,266,512✔
959

960
  QRY_PARAM_CHECK(ppRes);
1,266,512!
961
  pBlock->info.dataLoad = false;
1,266,512✔
962

963
  while (true) {
26,417✔
964
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
1,292,929✔
965
    if (code != TSDB_CODE_SUCCESS) {
1,292,445!
966
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
×
967
      QUERY_CHECK_CODE(code, lino, _end);
187!
968
    }
969

970
    if (!hasNext) {
1,292,664✔
971
      break;
305,385✔
972
    }
973

974
    if (isTaskKilled(pTaskInfo)) {
987,579✔
975
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
95✔
976
      code = pTaskInfo->code;
186✔
977
      goto _end;
186✔
978
    }
979

980
    if (pOperator->status == OP_EXEC_DONE) {
987,648✔
981
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
224✔
982
      break;
224✔
983
    }
984

985
    // process this data block based on the probabilities
986
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
987,424✔
987
    if (!processThisBlock) {
987,122!
988
      continue;
26,417✔
989
    }
990

991
    if (pBlock->info.id.uid) {
987,122!
992
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
987,344✔
993
    }
994

995
    uint32_t status = 0;
987,472✔
996
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
987,472✔
997
    QUERY_CHECK_CODE(code, lino, _end);
987,561✔
998

999
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
987,560✔
1000
      break;
76✔
1001
    }
1002

1003
    // current block is filter out according to filter condition, continue load the next block
1004
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
987,484✔
1005
      continue;
26,417✔
1006
    }
1007

1008
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
961,067✔
1009
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
961,034✔
1010

1011
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
961,034✔
1012
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
961,034✔
1013

1014
    (*ppRes) = pBlock;
961,034✔
1015
    return code;
961,034✔
1016
  }
1017

1018
_end:
305,572✔
1019
  if (code != TSDB_CODE_SUCCESS) {
305,572✔
1020
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
187!
1021
    pTaskInfo->code = code;
187✔
1022
    T_LONG_JMP(pTaskInfo->env, code);
187!
1023
  }
1024
  return code;
305,385✔
1025
}
1026

1027
static int32_t doGroupedTableScan(SOperatorInfo* pOperator, SSDataBlock** pBlock) {
1,360,598✔
1028
  int32_t         code = TSDB_CODE_SUCCESS;
1,360,598✔
1029
  int32_t         lino = 0;
1,360,598✔
1030
  STableScanInfo* pTableScanInfo = pOperator->info;
1,360,598✔
1031
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
1,360,598✔
1032
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
1,360,598✔
1033

1034
  QRY_PARAM_CHECK(pBlock);
1,360,598!
1035

1036
  // The read handle is not initialized yet, since no qualified tables exists
1037
  if (pTableScanInfo->base.dataReader == NULL || pOperator->status == OP_EXEC_DONE) {
1,360,598!
1038
    return code;
5,270✔
1039
  }
1040

1041
  // do the ascending order traverse in the first place.
1042
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
1,640,171✔
1043
    SSDataBlock* p = NULL;
1,214,493✔
1044
    code = doTableScanImplNext(pOperator, &p);
1,214,493✔
1045
    QUERY_CHECK_CODE(code, lino, _end);
1,214,143!
1046

1047
    if (p != NULL) {
1,214,143✔
1048
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
929,453✔
1049
      *pBlock = p;
929,452✔
1050
      return code;
929,452✔
1051
    }
1052

1053
    pTableScanInfo->scanTimes += 1;
284,690✔
1054
    taosHashClear(pTableScanInfo->pIgnoreTables);
284,690✔
1055

1056
    if (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
284,843✔
1057
      setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
196✔
1058
      pTableScanInfo->base.scanFlag = MAIN_SCAN;
196✔
1059
      pTableScanInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
196✔
1060
      qDebug("start to repeat ascending order scan data blocks due to query func required, %s", GET_TASKID(pTaskInfo));
196!
1061

1062
      // do prepare for the next round table scan operation
1063
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
196✔
1064
      QUERY_CHECK_CODE(code, lino, _end);
196!
1065
    }
1066
  }
1067

1068
  int32_t total = pTableScanInfo->scanInfo.numOfAsc + pTableScanInfo->scanInfo.numOfDesc;
425,678✔
1069
  if (pTableScanInfo->scanTimes < total) {
425,678✔
1070
    if (pTableScanInfo->base.cond.order == TSDB_ORDER_ASC) {
51,917✔
1071
      prepareForDescendingScan(&pTableScanInfo->base, pOperator->exprSupp.pCtx, 0);
152✔
1072
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
152✔
1073
      QUERY_CHECK_CODE(code, lino, _end);
152!
1074
      qDebug("%s start to descending order scan data blocks due to query func required", GET_TASKID(pTaskInfo));
152!
1075
    }
1076

1077
    while (pTableScanInfo->scanTimes < total) {
72,439✔
1078
      SSDataBlock* p = NULL;
51,912✔
1079
      code = doTableScanImplNext(pOperator, &p);
51,912✔
1080
      QUERY_CHECK_CODE(code, lino, _end);
51,851!
1081

1082
      if (p != NULL) {
51,851✔
1083
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
31,326✔
1084
        *pBlock = p;
31,326✔
1085
        return code;
31,326✔
1086
      }
1087

1088
      pTableScanInfo->scanTimes += 1;
20,525✔
1089
      taosHashClear(pTableScanInfo->pIgnoreTables);
20,525✔
1090

1091
      if (pTableScanInfo->scanTimes < total) {
20,522✔
1092
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
2✔
1093
        pTableScanInfo->base.scanFlag = MAIN_SCAN;
×
1094

1095
        qDebug("%s start to repeat descending order scan data blocks", GET_TASKID(pTaskInfo));
×
1096
        code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
×
1097
        QUERY_CHECK_CODE(code, lino, _end);
×
1098
      }
1099
    }
1100
  }
1101

1102
  if (pTableScanInfo->countState < TABLE_COUNT_STATE_END) {
394,288✔
1103
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
117,910✔
1104
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
117,910✔
1105
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
240✔
1106
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
4✔
1107
        STableKeyInfo* pStart =
1108
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
4✔
1109

1110
        if (NULL == pStart) {
4!
1111
          return code;
×
1112
        }
1113

1114
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
4✔
1115
        return code;
4✔
1116
      }
1117
    } else {  // group by tag + no sort
1118
      int32_t numOfTables = 0;
117,670✔
1119
      code = tableListGetSize(pTableListInfo, &numOfTables);
117,670✔
1120
      QUERY_CHECK_CODE(code, lino, _end);
117,643!
1121

1122
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
117,643✔
1123
        // get empty group, mark processed & rm from hash
1124
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
116,810✔
1125
        if (pIte != NULL) {
116,875✔
1126
          size_t        keySize = 0;
89,279✔
1127
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
89,279✔
1128
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
89,282✔
1129
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
89,282✔
1130
          markGroupProcessed(pTableScanInfo, *pGroupId);
89,290✔
1131
          *pBlock = getBlockForEmptyTable(pOperator, &info);
89,300✔
1132

1133
          return code;
89,306✔
1134
        }
1135
      }
1136
    }
1137
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
28,665✔
1138
  }
1139

1140
_end:
276,378✔
1141
  if (code != TSDB_CODE_SUCCESS) {
305,043!
1142
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1143
    pTaskInfo->code = code;
×
1144
  }
1145

1146
  return code;
305,043✔
1147
}
1148

1149
static int32_t createTableListInfoFromParam(SOperatorInfo* pOperator) {
647✔
1150
  STableScanInfo*          pInfo = pOperator->info;
647✔
1151
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
647✔
1152
  int32_t                  code = 0;
647✔
1153
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
647✔
1154
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
647✔
1155
  int32_t                  num = taosArrayGetSize(pParam->pUidList);
647✔
1156
  if (num <= 0) {
647!
1157
    qError("empty table scan uid list");
×
1158
    return TSDB_CODE_INVALID_PARA;
×
1159
  }
1160

1161
  qDebug("vgId:%d add total %d dynamic tables to scan, tableSeq:%d, exist num:%" PRId64 ", operator status:%d",
647!
1162
         pTaskInfo->id.vgId, num, pParam->tableSeq, (int64_t)taosArrayGetSize(pListInfo->pTableList),
1163
         pOperator->status);
1164

1165
  if (pParam->tableSeq) {
647✔
1166
    pListInfo->oneTableForEachGroup = true;
12✔
1167
    if (taosArrayGetSize(pListInfo->pTableList) > 0) {
12!
1168
      taosHashClear(pListInfo->map);
×
1169
      taosArrayClear(pListInfo->pTableList);
×
1170
      pOperator->status = OP_EXEC_DONE;
×
1171
    }
1172
  } else {
1173
    pListInfo->oneTableForEachGroup = false;
635✔
1174
    pListInfo->numOfOuputGroups = 1;
635✔
1175
  }
1176

1177
  STableKeyInfo info = {.groupId = 0};
647✔
1178
  int32_t       tableIdx = 0;
647✔
1179
  for (int32_t i = 0; i < num; ++i) {
15,313✔
1180
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
14,668✔
1181
    if (!pUid) {
14,667!
1182
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1183
      return terrno;
×
1184
    }
1185

1186
    if (taosHashPut(pListInfo->map, pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
14,667✔
1187
      if (TSDB_CODE_DUP_KEY == terrno) {
13,232!
1188
        continue;
13,232✔
1189
      }
1190
      return terrno;
×
1191
    }
1192

1193
    info.uid = *pUid;
1,443✔
1194
    void* p = taosArrayPush(pListInfo->pTableList, &info);
1,443✔
1195
    if (p == NULL) {
1,443!
1196
      return terrno;
×
1197
    }
1198

1199
    tableIdx++;
1,443✔
1200
    qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
1,443!
1201
  }
1202

1203
  return code;
645✔
1204
}
1205

1206
static int32_t doInitReader(STableScanInfo* pInfo, SExecTaskInfo* pTaskInfo, SStorageAPI* pAPI, int32_t* pNum,
300,090✔
1207
                            STableKeyInfo** pList) {
1208
  const char* idStr = GET_TASKID(pTaskInfo);
300,090✔
1209
  int32_t     code = initNextGroupScan(pInfo, pList, pNum);
300,090✔
1210
  if (code) {
300,102!
1211
    qError("%s failed to init groupScan Info, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
×
1212
    return code;
×
1213
  }
1214

1215
  if (pInfo->base.dataReader != NULL) {
300,109!
1216
    qError("%s tsdb reader should be null", idStr);
×
1217
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1218
  }
1219

1220
  code = pAPI->tsdReader.tsdReaderOpen(pInfo->base.readHandle.vnode, &pInfo->base.cond, *pList, *pNum, pInfo->pResBlock,
300,109✔
1221
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
300,109✔
1222
  if (code) {
300,073✔
1223
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
10!
1224
  }
1225

1226
  return code;
300,049✔
1227
}
1228

1229
int compareColIdPair(const void* elem1, const void* elem2) {
204,418✔
1230
  SColIdPair* node1 = (SColIdPair*)elem1;
204,418✔
1231
  SColIdPair* node2 = (SColIdPair*)elem2;
204,418✔
1232

1233
  if (node1->orgColId < node2->orgColId) {
204,418✔
1234
    return -1;
204,246✔
1235
  }
1236

1237
  return node1->orgColId > node2->orgColId;
172✔
1238
}
1239

1240
static bool isNewScanParam(STableScanOperatorParam* pParam) {
45,867✔
1241
  return pParam->window.skey == INT64_MAX && pParam->window.ekey == INT64_MIN;
45,867!
1242
}
1243

1244
static int32_t createVTableScanInfoFromParam(SOperatorInfo* pOperator) {
45,864✔
1245
  int32_t                  code = 0;
45,864✔
1246
  int32_t                  lino = 0;
45,864✔
1247
  STableScanInfo*          pInfo = pOperator->info;
45,864✔
1248
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
45,864✔
1249
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
45,864✔
1250
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
45,864✔
1251
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
45,864✔
1252
  SMetaReader              orgTable = {0};
45,864✔
1253
  SMetaReader              superTable = {0};
45,864✔
1254
  SSchemaWrapper*          schema = NULL;
45,864✔
1255
  SArray*                  pColArray = NULL;
45,864✔
1256
  SArray*                  pBlockColArray = NULL;
45,864✔
1257
  int32_t                  num = 0;
45,864✔
1258
  STableKeyInfo*           pList = NULL;
45,864✔
1259

1260
  cleanupQueryTableDataCond(&pInfo->base.cond);
45,864✔
1261

1262
  pAPI->metaReaderFn.initReader(&orgTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
45,864✔
1263
  code = pAPI->metaReaderFn.getTableEntryByName(&orgTable, strstr(pParam->pOrgTbInfo->tbName, ".") + 1);
45,864✔
1264
  pAPI->metaReaderFn.readerReleaseLock(&orgTable);
45,864✔
1265
  qDebug("dynamic vtable scan for origin table:%s, %s", pParam->pOrgTbInfo->tbName, GET_TASKID(pTaskInfo));
45,864✔
1266
  QUERY_CHECK_CODE(code, lino, _return);
45,864!
1267
  switch (orgTable.me.type) {
45,864!
1268
    case TSDB_CHILD_TABLE:
16,783✔
1269
      pAPI->metaReaderFn.initReader(&superTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
16,783✔
1270
      code = pAPI->metaReaderFn.getTableEntryByUid(&superTable, orgTable.me.ctbEntry.suid);
16,783✔
1271
      pAPI->metaReaderFn.readerReleaseLock(&superTable);
16,783✔
1272
      QUERY_CHECK_CODE(code, lino, _return);
16,783!
1273
      schema = &superTable.me.stbEntry.schemaRow;
16,783✔
1274
      break;
16,783✔
1275
    case TSDB_NORMAL_TABLE:
29,081✔
1276
      schema = &orgTable.me.ntbEntry.schemaRow;
29,081✔
1277
      break;
29,081✔
1278
    default:
×
1279
      qError("invalid table type:%d", orgTable.me.type);
×
1280
      return TSDB_CODE_INVALID_PARA;
×
1281
      break;
1282
  }
1283

1284
  pListInfo->oneTableForEachGroup = true;
45,864✔
1285
  taosHashClear(pListInfo->map);
45,864✔
1286
  taosArrayClear(pListInfo->pTableList);
45,864✔
1287

1288
  uint64_t      pUid = orgTable.me.uid;
45,864✔
1289
  STableKeyInfo info = {.groupId = 0, .uid = pUid};
45,864✔
1290
  int32_t       tableIdx = 0;
45,864✔
1291
  code = taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t));
45,864✔
1292
  QUERY_CHECK_CODE(code, lino, _return);
45,864!
1293
  QUERY_CHECK_NULL(taosArrayPush(pListInfo->pTableList, &info), code, lino, _return, terrno);
91,728!
1294
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
45,864✔
1295

1296
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
45,864✔
1297
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
45,864!
1298
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(int32_t));
45,864✔
1299
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
45,864!
1300

1301
  // virtual table's origin table scan do not has ts column.
1302
  SColIdPair tsPair = {.vtbColId = PRIMARYKEY_TIMESTAMP_COL_ID, .orgColId = PRIMARYKEY_TIMESTAMP_COL_ID};
45,864✔
1303
  QUERY_CHECK_NULL(taosArrayPush(pColArray, &tsPair), code, lino, _return, terrno);
45,864!
1304

1305
  for (int32_t i = 0; i < taosArrayGetSize(pParam->pOrgTbInfo->colMap); ++i) {
131,261✔
1306
    SColIdNameKV* kv = taosArrayGet(pParam->pOrgTbInfo->colMap, i);
85,397✔
1307
    for (int32_t j = 0; j < schema->nCols; j++) {
701,678!
1308
      if (strcmp(kv->colName, schema->pSchema[j].name) == 0) {
701,678✔
1309
        SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(j + 1)};
85,397✔
1310
        QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
85,397!
1311
        break;
85,397✔
1312
      }
1313
    }
1314
  }
1315

1316
  for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
177,125✔
1317
    SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, i);
131,261✔
1318
    for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
723,600!
1319
      SColMatchItem* pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
723,600✔
1320
      if (pItem->colId == pPair->vtbColId) {
723,600✔
1321
        SColIdPair tmpPair = {.orgColId = pPair->orgColId, .vtbColId = pItem->dstSlotId};
131,261✔
1322
        QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &tmpPair), code, lino, _return, terrno);
131,261!
1323
        break;
131,261✔
1324
      }
1325
    }
1326
  }
1327

1328
  taosArraySort(pColArray, compareColIdPair);
45,864✔
1329
  taosArraySort(pBlockColArray, compareColIdPair);
45,864✔
1330

1331
  code = initQueryTableDataCondWithColArray(&pInfo->base.cond, &pInfo->base.orgCond, &pInfo->base.readHandle, pColArray);
45,864✔
1332
  QUERY_CHECK_CODE(code, lino, _return);
45,864!
1333

1334
  if (pInfo->pResBlock) {
45,864✔
1335
    blockDataDestroy(pInfo->pResBlock);
32,330✔
1336
    pInfo->pResBlock = NULL;
32,330✔
1337
  }
1338

1339
  if (pParam->window.ekey > 0) {
45,864✔
1340
    pInfo->base.cond.twindows.skey = pParam->window.ekey + 1;
32,320✔
1341
  } else if (isNewScanParam(pParam)) {
13,544!
1342
    pInfo->base.cond.twindows.skey = pInfo->base.orgCond.twindows.skey;
13,544✔
1343
    pInfo->base.cond.twindows.ekey = pInfo->base.orgCond.twindows.ekey;
13,544✔
1344
  }
1345
  pInfo->base.cond.suid = orgTable.me.type == TSDB_CHILD_TABLE ? superTable.me.uid : 0;
45,864✔
1346
  pInfo->currentGroupId = 0;
45,864✔
1347
  pInfo->ignoreTag = true;
45,864✔
1348

1349
  code = createOneDataBlockWithColArray(pInfo->pOrgBlock, pBlockColArray, &pInfo->pResBlock);
45,864✔
1350
  QUERY_CHECK_CODE(code, lino, _return);
45,864!
1351

1352
  void **reader = taosHashGet(pInfo->readerCache, &pUid, sizeof(uint64_t));
45,864✔
1353
  if (reader) {
45,864✔
1354
    if (isNewScanParam(pParam)) {
32,323✔
1355
      pAPI->tsdReader.tsdReaderClose(*reader);
3✔
1356
      pInfo->base.dataReader = NULL;
3✔
1357

1358
      code = taosHashRemove(pInfo->readerCache, &pUid, sizeof(uint64_t));
3✔
1359
      QUERY_CHECK_CODE(code, lino, _return);
3!
1360

1361
      taosRLockLatch(&pTaskInfo->lock);
3✔
1362
      code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
3✔
1363
      taosRUnLockLatch(&pTaskInfo->lock);
3✔
1364
      QUERY_CHECK_CODE(code, lino, _return);
3!
1365

1366
      code = taosHashPut(pInfo->readerCache, &pUid, sizeof(uint64_t), &pInfo->base.dataReader, POINTER_BYTES);
3✔
1367
      QUERY_CHECK_CODE(code, lino, _return);
3!
1368
      pInfo->newReader = true;
3✔
1369
    } else {
1370
      pInfo->base.dataReader = *reader;
32,320✔
1371
      code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
32,320✔
1372
      QUERY_CHECK_CODE(code, lino, _return);
32,320!
1373

1374
      pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, pInfo->pResBlock);
32,320✔
1375
      pInfo->newReader = false;
32,320✔
1376
      pInfo->scanTimes = 0;
32,320✔
1377
    }
1378
  } else {
1379
    pInfo->base.dataReader = NULL;
13,541✔
1380

1381
    taosRLockLatch(&pTaskInfo->lock);
13,541✔
1382
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
13,541✔
1383
    taosRUnLockLatch(&pTaskInfo->lock);
13,541✔
1384
    QUERY_CHECK_CODE(code, lino, _return);
13,541!
1385

1386
    code = taosHashPut(pInfo->readerCache, &pUid, sizeof(uint64_t), &pInfo->base.dataReader, POINTER_BYTES);
13,541✔
1387
    QUERY_CHECK_CODE(code, lino, _return);
13,541!
1388
    pInfo->newReader = true;
13,541✔
1389
  }
1390
  pOperator->status = OP_OPENED;
45,864✔
1391
  
1392
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
45,864!
1393
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1394
  }
1395

1396
  pInfo->currentGroupId = -1;
45,864✔
1397
_return:
45,864✔
1398
  if (code) {
45,864!
1399
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1400
  }
1401
  taosArrayDestroy(pColArray);
45,864✔
1402
  taosArrayDestroy(pBlockColArray);
45,864✔
1403
  pAPI->metaReaderFn.clearReader(&superTable);
45,864✔
1404
  pAPI->metaReaderFn.clearReader(&orgTable);
45,864✔
1405
  return code;
45,864✔
1406
}
1407

1408
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
292,847✔
1409
  int32_t         code = TSDB_CODE_SUCCESS;
292,847✔
1410
  int32_t         lino = 0;
292,847✔
1411
  STableScanInfo* pInfo = pOperator->info;
292,847✔
1412
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
292,847✔
1413
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
292,847✔
1414
  int32_t         numOfTables = 0;
292,847✔
1415

1416
  QRY_PARAM_CHECK(pResult);
292,847!
1417

1418
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
292,847✔
1419
  QUERY_CHECK_CODE(code, lino, _end);
292,888!
1420

1421
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
292,888✔
1422
    setOperatorCompleted(pOperator);
278,910✔
1423
    if (pOperator->dynamicTask) {
278,914✔
1424
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
645✔
1425
      taosHashClear(pInfo->base.pTableListInfo->map);
644✔
1426
    }
1427
    return code;
278,915✔
1428
  }
1429

1430
  // reset value for the next group data output
1431
  pOperator->status = OP_OPENED;
13,957✔
1432
  resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
13,957✔
1433

1434
  int32_t        num = 0;
13,969✔
1435
  STableKeyInfo* pList = NULL;
13,969✔
1436
  code = initNextGroupScan(pInfo, &pList, &num);
13,969✔
1437
  QUERY_CHECK_CODE(code, lino, _end);
13,969!
1438

1439
  code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
13,969✔
1440
  QUERY_CHECK_CODE(code, lino, _end);
13,970!
1441

1442
  code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
13,970✔
1443
  QUERY_CHECK_CODE(code, lino, _end);
13,970!
1444
  pInfo->scanTimes = 0;
13,970✔
1445

1446
  code = doGroupedTableScan(pOperator, pResult);
13,970✔
1447
  QUERY_CHECK_CODE(code, lino, _end);
13,970!
1448

1449
  if (*pResult != NULL) {
13,970✔
1450
    if (pOperator->dynamicTask) {
13,873✔
1451
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
13,508✔
1452
    }
1453
    return code;
13,873✔
1454
  }
1455

1456
_end:
97✔
1457
  if (code != TSDB_CODE_SUCCESS) {
97!
1458
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1459
    pTaskInfo->code = code;
×
1460
  }
1461
  return code;
97✔
1462
}
1463

1464
static int32_t groupSeqTableScan(SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
1,277,170✔
1465
  int32_t         code = TSDB_CODE_SUCCESS;
1,277,170✔
1466
  int32_t         lino = 0;
1,277,170✔
1467
  STableScanInfo* pInfo = pOperator->info;
1,277,170✔
1468
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
1,277,170✔
1469
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
1,277,170✔
1470
  int32_t         num = 0;
1,277,170✔
1471
  STableKeyInfo*  pList = NULL;
1,277,170✔
1472
  SSDataBlock*    pResult = NULL;
1,277,170✔
1473
  const char*     idStr = GET_TASKID(pTaskInfo);
1,277,170✔
1474

1475
  QRY_PARAM_CHECK(pResBlock);
1,277,170!
1476

1477
  if (pInfo->currentGroupId == -1) {
1,277,170✔
1478
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
286,559!
1479
      setOperatorCompleted(pOperator);
×
1480
      return code;
×
1481
    }
1482

1483
    taosRLockLatch(&pTaskInfo->lock);
286,553✔
1484
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
286,595✔
1485
    taosRUnLockLatch(&pTaskInfo->lock);
286,491✔
1486
    QUERY_CHECK_CODE(code, lino, _end);
286,584✔
1487

1488
    if (pInfo->filesetDelimited) {
286,574✔
1489
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
639✔
1490
    }
1491

1492
    if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
286,574!
1493
      pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1494
    }
1495
  }
1496

1497
  code = doGroupedTableScan(pOperator, &pResult);
1,277,185✔
1498
  QUERY_CHECK_CODE(code, lino, _end);
1,277,209!
1499

1500
  if (pResult != NULL) {
1,277,209✔
1501
    if (pOperator->dynamicTask) {
998,151✔
1502
      pResult->info.id.groupId = pResult->info.id.uid;
1,346✔
1503
    }
1504

1505
    *pResBlock = pResult;
998,151✔
1506
    return code;
998,151✔
1507
  }
1508

1509
  while (true) {
1510
    code = startNextGroupScan(pOperator, &pResult);
279,113✔
1511
    QUERY_CHECK_CODE(code, lino, _end);
279,337!
1512

1513
    if (pResult || pOperator->status == OP_EXEC_DONE) {
279,337✔
1514
      *pResBlock = pResult;
279,282✔
1515
      return code;
279,282✔
1516
    }
1517
  }
1518

1519
_end:
10✔
1520
  if (code != TSDB_CODE_SUCCESS) {
10!
1521
    qError("%s %s failed at line %d since %s", idStr, __func__, lino, tstrerror(code));
10!
1522
    pTaskInfo->code = code;
10✔
1523
  }
1524

1525
  return code;
10✔
1526
}
1527

1528
static bool isEmptyQueryTimeWindow(STimeWindow* pWindow) {
45,864✔
1529
  return (pWindow == NULL) || (pWindow->skey > pWindow->ekey);
45,864!
1530
}
1531

1532
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
1,342,484✔
1533
  int32_t         code = TSDB_CODE_SUCCESS;
1,342,484✔
1534
  int32_t         lino = 0;
1,342,484✔
1535
  STableScanInfo* pInfo = pOperator->info;
1,342,484✔
1536
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
1,342,484✔
1537
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
1,342,484✔
1538
  QRY_PARAM_CHECK(ppRes);
1,342,484!
1539
  qTrace("%s call", __FUNCTION__);
1,342,484!
1540
  if (pOperator->pOperatorGetParam) {
1,342,382✔
1541
    pOperator->dynamicTask = true;
46,511✔
1542
    if (isDynVtbScan(pOperator)) {
46,511✔
1543
      code = createVTableScanInfoFromParam(pOperator);
45,864✔
1544

1545
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
45,864✔
1546
      pOperator->pOperatorGetParam = NULL;
45,864✔
1547
      QUERY_CHECK_CODE(code, lino, _end);
45,864!
1548

1549
      SSDataBlock* result = NULL;
45,864✔
1550
      if (isEmptyQueryTimeWindow(&pInfo->base.cond.twindows) && pInfo->base.cond.type == TIMEWINDOW_RANGE_CONTAINED) {
45,864!
1551
        (*ppRes) = result;
26✔
1552
        return code;
45,864✔
1553
      }
1554

1555
      if (pInfo->newReader) {
45,838✔
1556
        code = startNextGroupScan(pOperator, &result);
13,544✔
1557
        QUERY_CHECK_CODE(code, lino, _end);
13,544!
1558
      } else {
1559
        code = doGroupedTableScan(pOperator, &result);
32,294✔
1560
        QUERY_CHECK_CODE(code, lino, _end);
32,294!
1561
      }
1562

1563
      if (result) {
45,838✔
1564
        SSDataBlock* res = NULL;
32,353✔
1565
        pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, NULL);
32,353✔
1566
        code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, &res);
32,353✔
1567
        QUERY_CHECK_CODE(code, lino, _end);
32,353!
1568
        pInfo->pResBlock = res;
32,353✔
1569
        blockDataDestroy(result);
32,353✔
1570
        (*ppRes) = res;
32,353✔
1571
      } else {
1572
        STableKeyInfo *keyInfo = taosArrayGet(pInfo->base.pTableListInfo->pTableList, 0);
13,485✔
1573
        QUERY_CHECK_NULL(keyInfo, code, lino, _end, terrno)
13,485!
1574

1575
        blockDataDestroy(pInfo->pResBlock);
13,485✔
1576
        pInfo->pResBlock = NULL;
13,485✔
1577

1578
        void** reader = taosHashGet(pInfo->readerCache, &keyInfo->uid, sizeof(uint64_t));
13,485✔
1579
        if (reader) {
13,485!
1580
          if (*reader == pInfo->base.dataReader) {
13,485!
1581
            pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, NULL);
13,485✔
1582
            pInfo->base.dataReader = NULL;
13,485✔
1583
          }
1584
          if (pAPI->tsdReader.tsdReaderClose) {
13,485!
1585
            pAPI->tsdReader.tsdReaderClose(*reader);
13,485✔
1586
          }
1587
        }
1588

1589
        code = taosHashRemove(pInfo->readerCache, &keyInfo->uid, sizeof(uint64_t));
13,485✔
1590
        QUERY_CHECK_CODE(code, lino, _end);
13,485!
1591
      }
1592
      return code;
45,838✔
1593
    } else {
1594
      code = createTableListInfoFromParam(pOperator);
647✔
1595
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
647✔
1596
      pOperator->pOperatorGetParam = NULL;
647✔
1597
      QUERY_CHECK_CODE(code, lino, _end);
647!
1598

1599
      if (pOperator->status == OP_EXEC_DONE) {
647✔
1600
        pInfo->currentGroupId = -1;
6✔
1601
        pOperator->status = OP_OPENED;
6✔
1602
        SSDataBlock* result = NULL;
6✔
1603

1604
        while (true) {
1605
          code = startNextGroupScan(pOperator, &result);
6✔
1606
          QUERY_CHECK_CODE(code, lino, _end);
6!
1607

1608
          if (result || pOperator->status == OP_EXEC_DONE) {
6!
1609
            (*ppRes) = result;
6✔
1610
            return code;
6✔
1611
          }
1612
        }
1613
      }
1614
    }
1615
  }
1616

1617
  // scan table one by one sequentially
1618
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
1,296,512✔
1619
    int32_t       numOfTables = 0;
19,335✔
1620
    STableKeyInfo tInfo = {0};
19,335✔
1621
    pInfo->countState = TABLE_COUNT_STATE_END;
19,335✔
1622

1623
    while (1) {
17,826✔
1624
      SSDataBlock* result = NULL;
37,161✔
1625
      code = doGroupedTableScan(pOperator, &result);
37,161✔
1626
      QUERY_CHECK_CODE(code, lino, _end);
37,159!
1627

1628
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
37,159!
1629
        (*ppRes) = result;
19,179✔
1630
        return code;
19,330✔
1631
      }
1632

1633
      // if no data, switch to next table and continue scan
1634
      pInfo->currentTable++;
17,980✔
1635

1636
      taosRLockLatch(&pTaskInfo->lock);
17,980✔
1637
      numOfTables = 0;
17,983✔
1638
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
17,983✔
1639
      if (code != TSDB_CODE_SUCCESS) {
17,983!
1640
        taosRUnLockLatch(&pTaskInfo->lock);
×
1641
        TSDB_CHECK_CODE(code, lino, _end);
×
1642
      }
1643

1644
      if (pInfo->currentTable >= numOfTables) {
17,983✔
1645
        qDebug("all table checked in table list, total:%d, return NULL, %s", numOfTables, GET_TASKID(pTaskInfo));
151!
1646
        taosRUnLockLatch(&pTaskInfo->lock);
151✔
1647
        (*ppRes) = NULL;
151✔
1648
        return code;
151✔
1649
      }
1650

1651
      STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
17,832✔
1652
      if (!tmp) {
17,831!
1653
        taosRUnLockLatch(&pTaskInfo->lock);
×
1654
        (*ppRes) = NULL;
×
1655
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1656
      }
1657

1658
      tInfo = *tmp;
17,831✔
1659
      taosRUnLockLatch(&pTaskInfo->lock);
17,831✔
1660

1661
      code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, &tInfo, 1);
17,832✔
1662
      QUERY_CHECK_CODE(code, lino, _end);
17,829!
1663
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d/%d %s", tInfo.uid, numOfTables,
17,829!
1664
             pInfo->currentTable, numOfTables, GET_TASKID(pTaskInfo));
1665

1666
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
17,832✔
1667
      QUERY_CHECK_CODE(code, lino, _end);
17,826!
1668
      pInfo->scanTimes = 0;
17,826✔
1669
    }
1670
  } else {  // scan table group by group sequentially
1671
    code = groupSeqTableScan(pOperator, ppRes);
1,277,177✔
1672
    QUERY_CHECK_CODE(code, lino, _end);
1,277,203✔
1673
  }
1674

1675
_end:
1,277,193✔
1676
  if (code != TSDB_CODE_SUCCESS) {
1,277,203✔
1677
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
10!
1678
    pTaskInfo->code = code;
10✔
1679
    T_LONG_JMP(pTaskInfo->env, code);
10!
1680
  }
1681

1682
  return code;
1,277,193✔
1683
}
1684

1685
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
3,617✔
1686
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
3,617!
1687
  if (!pRecorder) {
3,617!
1688
    return terrno;
×
1689
  }
1690
  STableScanInfo* pTableScanInfo = pOptr->info;
3,617✔
1691
  *pRecorder = pTableScanInfo->base.readRecorder;
3,617✔
1692
  *pOptrExplain = pRecorder;
3,617✔
1693
  *len = sizeof(SFileBlockLoadRecorder);
3,617✔
1694
  return 0;
3,617✔
1695
}
1696

1697
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
355,914✔
1698
  cleanupQueryTableDataCond(&pBase->cond);
355,914✔
1699
  cleanupQueryTableDataCond(&pBase->orgCond);
355,917✔
1700

1701
  if (pAPI->tsdReaderClose) {
355,917!
1702
    pAPI->tsdReaderClose(pBase->dataReader);
355,917✔
1703
  }
1704
  pBase->dataReader = NULL;
355,910✔
1705

1706
  if (pBase->matchInfo.pList != NULL) {
355,910!
1707
    taosArrayDestroy(pBase->matchInfo.pList);
355,915✔
1708
  }
1709

1710
  tableListDestroy(pBase->pTableListInfo);
355,911✔
1711
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
355,915✔
1712
  cleanupExprSupp(&pBase->pseudoSup);
355,899✔
1713
}
355,883✔
1714

1715
static void cleanReaderForVTable(STableScanInfo* pInfo){
6,774✔
1716
  if (pInfo != NULL && pInfo->base.readerAPI.tsdReaderClose) {
6,774!
1717
    void *pIter = taosHashIterate(pInfo->readerCache, NULL);
6,774✔
1718
    while (pIter != NULL) {
6,830✔
1719
      void **reader = pIter;
56✔
1720
      if (*reader) {
56!
1721
        if (*reader == pInfo->base.dataReader) {
56✔
1722
          pInfo->base.dataReader = NULL;
49✔
1723
        }
1724
        pInfo->base.readerAPI.tsdReaderClose(*reader);
56✔
1725
      }
1726
      pIter = taosHashIterate(pInfo->readerCache, pIter);
56✔
1727
    }
1728
  }
1729
}
6,774✔
1730

1731
static void destroyTableScanOperatorInfo(void* param) {
303,169✔
1732
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
303,169✔
1733
  blockDataDestroy(pTableScanInfo->pResBlock);
303,169✔
1734
  blockDataDestroy(pTableScanInfo->pOrgBlock);
303,169✔
1735
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
303,169✔
1736
  if (pTableScanInfo->virtualStableScan && pTableScanInfo->readerCache) {
303,166!
1737
    cleanReaderForVTable(pTableScanInfo);
6,774✔
1738
    taosHashCleanup(pTableScanInfo->readerCache);
6,774✔
1739
  }
1740
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
303,167✔
1741
  taosMemoryFreeClear(param);
303,125!
1742
}
303,155✔
1743

1744
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
302,984✔
1745
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
302,984!
1746
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
18,356✔
1747
    for (int32_t i = 0; i < numOfCols; ++i) {
44,501✔
1748
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
26,144✔
1749
      if (pCol) {
26,145!
1750
        pCol->info.noData = true;
26,146✔
1751
      }
1752
    }
1753
  }
1754
}
302,985✔
1755

1756
static int32_t resetTableScanOperatorState(SOperatorInfo* pOper) {
1,525✔
1757
  int32_t         code = TSDB_CODE_SUCCESS;
1,525✔
1758
  STableScanInfo*   pInfo = pOper->info;
1,525✔
1759
  pOper->status = OP_NOT_OPENED;
1,525✔
1760

1761
  pInfo->scanTimes = 0;
1,525✔
1762
  pInfo->currentGroupId = -1;
1,525✔
1763
  pInfo->tableEndIndex = -1;
1,525✔
1764
  pInfo->tableStartIndex = 0;
1,525✔
1765
  pInfo->currentTable = 0;
1,525✔
1766
  pInfo->scanMode = 0;
1,525✔
1767
  pInfo->countState = 0;
1,525✔
1768
  if (!pInfo->virtualStableScan) {
1,525✔
1769
  // if (pInfo->virtualStableScan && pInfo->readerCache) {
1770
  //   cleanReaderForVTable(pInfo);
1771
  //   taosHashClear(pInfo->readerCache);
1772
  //   pInfo->readerCache = false;
1773
  // } else {
1774
    if (pInfo->base.readerAPI.tsdReaderClose) {
809!
1775
      pInfo->base.readerAPI.tsdReaderClose(pInfo->base.dataReader);
809✔
1776
    }
1777
    pInfo->base.dataReader = NULL;
809✔
1778
  }
1779

1780
  tableListDestroy(pInfo->base.pTableListInfo);
1,525✔
1781
  pInfo->base.pTableListInfo = tableListCreate();
1,525✔
1782
  if (!pInfo->base.pTableListInfo) {
1,525!
1783
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1784
    return terrno;
×
1785
  }
1786
  SExecTaskInfo*         pTaskInfo = pOper->pTaskInfo;
1,525✔
1787

1788
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
1,525✔
1789
  if (!pTableScanNode->scan.node.dynamicOp) {
1,525✔
1790
    code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
809✔
1791
                                    &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
1792
                                    pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
809✔
1793
    if (code) {
809!
1794
      qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
×
1795
      return code;
×
1796
    }
1797
  }
1798

1799
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->base.limitInfo);
1,525✔
1800

1801
  cleanupQueryTableDataCond(&pInfo->base.cond);
1,525✔
1802
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle);
1,525✔
1803
  if (code) {
1,525!
1804
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
1805
    return code;
×
1806
  }
1807
  if (pTableScanNode->scan.node.dynamicOp && pTableScanNode->scan.virtualStableScan) {
1,525!
1808
    cleanupQueryTableDataCond(&pInfo->base.orgCond);
716✔
1809
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
716✔
1810
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
716✔
1811
  }
1812

1813
  pOper->resultInfo.totalRows = 0;
1,525✔
1814
  blockDataEmpty(pInfo->pResBlock);
1,525✔
1815
  blockDataEmpty(pInfo->pOrgBlock);
1,525✔
1816
  taosHashClear(pInfo->pIgnoreTables);
1,525✔
1817
  return code;
1,525✔
1818
}
1819

1820
int32_t createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
303,082✔
1821
                                    STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
1822
                                    SOperatorInfo** pOptrInfo) {
1823
  QRY_PARAM_CHECK(pOptrInfo);
303,082!
1824

1825
  int32_t         code = TSDB_CODE_SUCCESS;
303,082✔
1826
  int32_t         lino = 0;
303,082✔
1827
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
303,082!
1828
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
303,135!
1829
  if (pInfo == NULL || pOperator == NULL) {
303,138!
1830
    code = terrno;
×
1831
    goto _error;
×
1832
  }
1833

1834
  pOperator->pPhyNode = pTableScanNode;
303,138✔
1835
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
303,138✔
1836
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
303,138✔
1837

1838
  int32_t numOfCols = 0;
303,138✔
1839
  code =
1840
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
303,138✔
1841
  QUERY_CHECK_CODE(code, lino, _error);
303,130!
1842

1843
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
303,130✔
1844
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
303,113✔
1845
  QUERY_CHECK_CODE(code, lino, _error);
303,100!
1846

1847
  if (pScanNode->pScanPseudoCols != NULL) {
303,100✔
1848
    SExprSupp* pSup = &pInfo->base.pseudoSup;
101,385✔
1849
    pSup->pExprInfo = NULL;
101,385✔
1850
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
101,385✔
1851
    QUERY_CHECK_CODE(code, lino, _error);
101,386!
1852

1853
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
101,386✔
1854
                                      &pTaskInfo->storageAPI.functionStore);
1855
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
101,374✔
1856
  }
1857

1858
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
303,066✔
1859
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
303,066✔
1860

1861
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
303,066✔
1862
  pInfo->base.readHandle = *readHandle;
303,080✔
1863
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
303,080✔
1864

1865
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
303,080✔
1866
  pInfo->sample.seed = taosGetTimestampSec();
303,080✔
1867

1868
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
303,088✔
1869
  initResultSizeInfo(&pOperator->resultInfo, 4096);
303,088✔
1870
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
303,044✔
1871
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
303,145✔
1872
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
303,095!
1873

1874
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
303,095✔
1875
  QUERY_CHECK_CODE(code, lino, _error);
303,068!
1876

1877
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
303,068✔
1878
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
303,068✔
1879
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
6,773✔
1880
    pInfo->pResBlock = NULL;
6,773✔
1881
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
6,773✔
1882
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
6,773✔
1883
  }
1884

1885
  if (pInfo->virtualStableScan) {
303,068✔
1886
    pInfo->readerCache = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
6,775✔
1887
    QUERY_CHECK_NULL(pInfo->readerCache, code, lino, _error, terrno);
6,774!
1888
  }
1889

1890
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
303,067✔
1891
                            pTaskInfo->pStreamRuntimeInfo);
303,067✔
1892
  QUERY_CHECK_CODE(code, lino, _error);
303,075✔
1893

1894
  pInfo->currentGroupId = -1;
303,039✔
1895

1896
  pInfo->tableEndIndex = -1;
303,039✔
1897
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
303,039✔
1898
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
303,039✔
1899

1900
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
303,039✔
1901
                  pTaskInfo);
1902
  pOperator->exprSupp.numOfExprs = numOfCols;
303,010✔
1903

1904
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
303,010✔
1905
  pInfo->ignoreTag = false;
303,010✔
1906

1907
  pInfo->base.pTableListInfo = pTableListInfo;
303,010✔
1908
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
303,010✔
1909
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
303,084!
1910
    code = terrno;
×
1911
    QUERY_CHECK_CODE(code, lino, _error);
×
1912
  }
1913

1914
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
303,084✔
1915

1916
  taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
303,084✔
1917
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
303,109✔
1918
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
1919

1920
  setOperatorResetStateFn(pOperator, resetTableScanOperatorState); 
303,058✔
1921
  // for non-blocking operator, the open cost is always 0
1922
  pOperator->cost.openCost = 0;
303,047✔
1923
  *pOptrInfo = pOperator;
303,047✔
1924
  return TSDB_CODE_SUCCESS;
303,047✔
1925

1926
_error:
36✔
1927
  if (pInfo != NULL) {
36!
1928
    pInfo->base.pTableListInfo = NULL;  // this attribute will be destroy outside of this function
36✔
1929
    destroyTableScanOperatorInfo(pInfo);
36✔
1930
  }
1931

1932
  if (pOperator != NULL) {
36!
1933
    pOperator->info = NULL;
36✔
1934
    destroyOperator(pOperator);
36✔
1935
  }
1936
  pTaskInfo->code = code;
36✔
1937
  return code;
36✔
1938
}
1939

1940
int32_t createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
×
1941
  QRY_PARAM_CHECK(pOptrInfo);
×
1942

1943
  int32_t         code = 0;
×
1944
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
×
1945
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
1946
  if (pInfo == NULL || pOperator == NULL) {
×
1947
    code = terrno;
×
1948
    goto _end;
×
1949
  }
1950

1951
  pInfo->base.dataReader = pReadHandle;
×
1952
  //  pInfo->prevGroupId       = -1;
1953

1954
  setOperatorInfo(pOperator, "TableSeqScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN, false, OP_NOT_OPENED,
×
1955
                  pInfo, pTaskInfo);
1956
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanImplNext, NULL, NULL, optrDefaultBufFn, NULL,
×
1957
                                         optrDefaultGetNextExtFn, NULL);
1958
  *pOptrInfo = pOperator;
×
1959
  return code;
×
1960

1961
_end:
×
1962
  if (pInfo != NULL) {
×
1963
    taosMemoryFree(pInfo);
×
1964
  }
1965

1966
  if (pOperator != NULL) {
×
1967
    taosMemoryFree(pOperator);
×
1968
  }
1969

1970
  pTaskInfo->code = code;
×
1971
  return code;
×
1972
}
1973

1974
static int32_t doBlockDataPrimaryKeyFilter(SSDataBlock* pBlock, STqOffsetVal* offset) {
×
1975
  int32_t code = TSDB_CODE_SUCCESS;
×
1976
  int32_t lino = 0;
×
1977
  if (pBlock->info.window.skey != offset->ts || offset->primaryKey.type == 0) {
×
1978
    return code;
×
1979
  }
1980
  bool* p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
×
1981
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
×
1982
  bool hasUnqualified = false;
×
1983

1984
  SColumnInfoData* pColTs = taosArrayGet(pBlock->pDataBlock, 0);
×
1985
  SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
1986

1987
  qDebug("doBlockDataWindowFilter primary key, ts:%" PRId64 " %" PRId64, offset->ts,
×
1988
         VALUE_GET_TRIVIAL_DATUM(&offset->primaryKey));
1989
  QUERY_CHECK_CONDITION((pColPk->info.type == offset->primaryKey.type), code, lino, _end,
×
1990
                        TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
1991

1992
  __compar_fn_t func = getComparFunc(pColPk->info.type, 0);
×
1993
  QUERY_CHECK_NULL(func, code, lino, _end, terrno);
×
1994
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
×
1995
    int64_t* ts = (int64_t*)colDataGetData(pColTs, i);
×
1996
    void*    data = colDataGetData(pColPk, i);
×
1997
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
1998
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
×
1999
        QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY, lino, _end);
×
2000
      }
2001
      void* tmq = taosMemoryMalloc(offset->primaryKey.nData + VARSTR_HEADER_SIZE);
×
2002
      QUERY_CHECK_NULL(tmq, code, lino, _end, terrno);
×
2003
      memcpy(varDataVal(tmq), offset->primaryKey.pData, offset->primaryKey.nData);
×
2004
      varDataLen(tmq) = offset->primaryKey.nData;
×
2005
      p[i] = (*ts > offset->ts) || (func(data, tmq) > 0);
×
2006
      taosMemoryFree(tmq);
×
2007
    } else {
2008
      p[i] = (*ts > offset->ts) || (func(data, VALUE_GET_DATUM(&offset->primaryKey, pColPk->info.type)) > 0);
×
2009
    }
2010

2011
    if (!p[i]) {
×
2012
      hasUnqualified = true;
×
2013
    }
2014
  }
2015

2016
  if (hasUnqualified) {
×
2017
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
×
2018
    QUERY_CHECK_CODE(code, lino, _end);
×
2019
  }
2020

2021
_end:
×
2022
  taosMemoryFree(p);
×
2023
  if (code != TSDB_CODE_SUCCESS) {
×
2024
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2025
  }
2026
  return code;
×
2027
}
2028

2029
int32_t colIdComparFn(const void* param1, const void* param2) {
2,005,275✔
2030
  int32_t p1 = *(int32_t*)param1;
2,005,275✔
2031
  int32_t p2 = *(int32_t*)param2;
2,005,275✔
2032

2033
  if (p1 == p2) {
2,005,275!
2034
    return 0;
×
2035
  } else {
2036
    return (p1 < p2) ? -1 : 1;
2,005,275✔
2037
  }
2038
}
2039

2040
static int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock) {
209,244✔
2041
  int32_t         code = TSDB_CODE_SUCCESS;
209,244✔
2042
  int32_t         lino = 0;
209,244✔
2043
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
209,244✔
2044
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
209,244✔
2045
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
209,244✔
2046
  const char*     id = GET_TASKID(pTaskInfo);
209,244✔
2047
  bool            isVtableSourceScan = (pTaskInfo->pSubplan->pVTables != NULL);
209,244✔
2048

2049
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
209,244✔
2050
  QUERY_CHECK_CODE(code, lino, _end);
209,249!
2051

2052
  pBlockInfo->rows = pBlock->info.rows;
209,249✔
2053
  pBlockInfo->id.uid = pBlock->info.id.uid;
209,249✔
2054
  pBlockInfo->type = STREAM_NORMAL;
209,249✔
2055
  pBlockInfo->version = pBlock->info.version;
209,249✔
2056

2057
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
209,249✔
2058
  if (!isVtableSourceScan) {
209,249!
2059
    pBlockInfo->id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
209,250✔
2060
  } else {
2061
    // use original table uid as groupId for vtable
2062
    pBlockInfo->id.groupId = pBlock->info.id.groupId;
×
2063
  }
2064

2065
  SArray* pColList = taosArrayInit(4, sizeof(int32_t));
209,268✔
2066
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
209,261!
2067

2068
  // todo extract method
2069
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
1,160,236✔
2070
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
951,022✔
2071
    if (!pColMatchInfo->needOutput) {
950,989✔
2072
      continue;
2✔
2073
    }
2074

2075
    bool colExists = false;
950,987✔
2076
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
3,109,171✔
2077
      SColumnInfoData* pResCol = NULL;
3,109,167✔
2078
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
3,109,167✔
2079
      QUERY_CHECK_CODE(code, lino, _end);
3,108,953!
2080

2081
      if (pResCol->info.colId == pColMatchInfo->colId) {
3,108,953✔
2082
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
950,769✔
2083
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
950,719✔
2084
        QUERY_CHECK_CODE(code, lino, _end);
950,920!
2085

2086
        colExists = true;
950,920✔
2087
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
950,920✔
2088
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
950,972!
2089
        break;
950,972✔
2090
      }
2091
    }
2092

2093
    // the required column does not exists in submit block, let's set it to be all null value
2094
    if (!colExists) {
950,973✔
2095
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
4✔
2096
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
4!
2097
      void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
4✔
2098
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
4!
2099
    }
2100
  }
2101

2102
  // currently only the tbname pseudo column
2103
  if (pInfo->numOfPseudoExpr > 0 && !isVtableSourceScan) {
209,150!
2104
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
103,121✔
2105
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
103,121✔
2106
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
2107
    if (code) {
103,135!
2108
      QUERY_CHECK_CODE(code, lino, _end);
×
2109
    }
2110

2111
    // reset the error code.
2112
    terrno = 0;
103,135✔
2113

2114
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
524,471✔
2115
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
421,335✔
2116
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
421,337!
2117
    }
2118
  }
2119

2120
  taosArraySort(pColList, colIdComparFn);
209,165✔
2121

2122
  int32_t i = 0, j = 0;
209,259✔
2123
  while (i < taosArrayGetSize(pColList)) {
1,581,803✔
2124
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
1,372,528✔
2125
    if (slot1 > j) {
1,372,549✔
2126
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
5✔
2127
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
2128
      j += 1;
×
2129
    } else {
2130
      i += 1;
1,372,544✔
2131
      j += 1;
1,372,544✔
2132
    }
2133
  }
2134

2135
  while (j < taosArrayGetSize(pInfo->pRes->pDataBlock)) {
209,262✔
2136
    SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
2✔
2137
    colDataSetNNULL(pDst, 0, pBlockInfo->rows);
2!
2138
    j += 1;
2✔
2139
  }
2140

2141
  taosArrayDestroy(pColList);
209,254✔
2142

2143
  code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL, NULL);
209,267✔
2144
  QUERY_CHECK_CODE(code, lino, _end);
209,254!
2145

2146
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
209,254✔
2147
  QUERY_CHECK_CODE(code, lino, _end);
209,260!
2148
  if (pInfo->pRes->info.rows == 0) {
209,260✔
2149
    return 0;
2,330✔
2150
  }
2151

2152
_end:
206,930✔
2153
  if (code != TSDB_CODE_SUCCESS) {
206,930!
2154
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2155
  }
2156
  return code;
206,928✔
2157
}
2158

2159
static int32_t processPrimaryKey(SSDataBlock* pBlock, bool hasPrimaryKey, STqOffsetVal* offset) {
21,715✔
2160
  int32_t code = TSDB_CODE_SUCCESS;
21,715✔
2161
  int32_t lino = 0;
21,715✔
2162
  SValue  val = {0};
21,715✔
2163
  if (hasPrimaryKey) {
21,715!
2164
    code = doBlockDataPrimaryKeyFilter(pBlock, offset);
×
2165
    if (code != TSDB_CODE_SUCCESS) {
×
2166
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2167
      return code;
×
2168
    }
2169
    SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
2170

2171
    if (pBlock->info.rows < 1) {
×
2172
      return code;
×
2173
    }
2174
    void* tmp = colDataGetData(pColPk, pBlock->info.rows - 1);
×
2175
    val.type = pColPk->info.type;
×
2176
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
2177
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
×
2178
        return TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY;
×
2179
      }
2180
      val.pData = taosMemoryMalloc(varDataLen(tmp));
×
2181
      QUERY_CHECK_NULL(val.pData, code, lino, _end, terrno);
×
2182
      val.nData = varDataLen(tmp);
×
2183
      memcpy(val.pData, varDataVal(tmp), varDataLen(tmp));
×
2184
    } else {
2185
      valueSetDatum(&val, pColPk->info.type, tmp, pColPk->info.bytes);
×
2186
    }
2187
  }
2188
  tqOffsetResetToData(offset, pBlock->info.id.uid, pBlock->info.window.ekey, val);
43,431!
2189

2190
_end:
21,716✔
2191
  if (code != TSDB_CODE_SUCCESS) {
21,716!
2192
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2193
  }
2194
  return code;
21,714✔
2195
}
2196

2197
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
254,847✔
2198
  int32_t        code = TSDB_CODE_SUCCESS;
254,847✔
2199
  int32_t        lino = 0;
254,847✔
2200
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
254,847✔
2201
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
254,847✔
2202

2203
  SStreamScanInfo* pInfo = pOperator->info;
254,847✔
2204
  const char*      id = GET_TASKID(pTaskInfo);
254,847✔
2205

2206
  qDebug("start to exec queue scan, %s", id);
254,847!
2207

2208
  if (isTaskKilled(pTaskInfo)) {
254,852!
2209
    (*ppRes) = NULL;
×
2210
    return pTaskInfo->code;
×
2211
  }
2212

2213
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
254,853✔
2214
    while (1) {
×
2215
      SSDataBlock* pResult = NULL;
19,336✔
2216
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
19,336✔
2217
      QUERY_CHECK_CODE(code, lino, _end);
19,330!
2218

2219
      if (pResult && pResult->info.rows > 0) {
19,330!
2220
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
19,181✔
2221
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
19,184✔
2222
        QUERY_CHECK_CODE(code, lino, _end);
19,183!
2223
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
19,183!
2224
        if (pResult->info.rows > 0) {
19,185!
2225
          (*ppRes) = pResult;
19,185✔
2226
          return code;
19,185✔
2227
        }
2228
      } else {
2229
        break;
2230
      }
2231
    }
2232

2233
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
149✔
2234
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
149✔
2235

2236
    pTSInfo->base.dataReader = NULL;
151✔
2237
    int64_t validVer = pTaskInfo->streamInfo.snapshotVer + 1;
151✔
2238
    qDebug("queue scan tsdb over, switch to wal ver %" PRId64, validVer);
151!
2239
    if (pAPI->tqReaderFn.tqReaderSeek(pInfo->tqReader, validVer, pTaskInfo->id.str) < 0) {
151✔
2240
      (*ppRes) = NULL;
70✔
2241
      return code;
70✔
2242
    }
2243

2244
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, validVer);
81✔
2245
  }
2246

2247
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
235,598!
2248
    while (1) {
2,330✔
2249
      bool hasResult =
2250
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
237,928✔
2251

2252
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
237,888✔
2253
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
237,894✔
2254

2255
      // curVersion move to next
2256
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pWalReader->curVersion);
237,897✔
2257

2258
      // use ts to pass time when replay, because ts not used if type is log
2259
      pTaskInfo->streamInfo.currentOffset.ts = pAPI->tqReaderFn.tqGetResultBlockTime(pInfo->tqReader);
237,897✔
2260

2261
      if (hasResult) {
237,896✔
2262
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
209,239!
2263
               pTaskInfo->streamInfo.currentOffset.version);
2264
        blockDataCleanup(pInfo->pRes);
209,263✔
2265
        code = setBlockIntoRes(pInfo, pRes);
209,241✔
2266
        QUERY_CHECK_CODE(code, lino, _end);
209,251!
2267
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
209,251!
2268
               pTaskInfo->streamInfo.currentOffset.version);
2269
        if (pInfo->pRes->info.rows > 0) {
209,270✔
2270
          (*ppRes) = pInfo->pRes;
206,940✔
2271
          return code;
206,940✔
2272
        }
2273
      } else {
2274
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
28,657!
2275
        (*ppRes) = NULL;
28,657✔
2276
        return code;
28,657✔
2277
      }
2278
    }
2279
  } else {
2280
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
×
2281
    (*ppRes) = NULL;
×
2282
    return code;
×
2283
  }
2284

2285
_end:
×
2286
  if (code != TSDB_CODE_SUCCESS) {
×
2287
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2288
    pTaskInfo->code = code;
×
2289
    T_LONG_JMP(pTaskInfo->env, code);
×
2290
  }
2291
  (*ppRes) = NULL;
×
2292
  return code;
×
2293
}
2294

2295
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
×
2296
  SSDataBlock* pRes = NULL;
×
2297
  int32_t      code = doQueueScanNext(pOperator, &pRes);
×
2298
  return pRes;
×
2299
}
2300

2301
int32_t extractTableIdList(const STableListInfo* pTableListInfo, SArray** ppArrayRes) {
976✔
2302
  int32_t code = TSDB_CODE_SUCCESS;
976✔
2303
  int32_t lino = 0;
976✔
2304
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));
976✔
2305
  QUERY_CHECK_NULL(tableIdList, code, lino, _end, terrno);
978!
2306

2307
  // Transfer the Array of STableKeyInfo into uid list.
2308
  int32_t size = 0;
978✔
2309
  code = tableListGetSize(pTableListInfo, &size);
978✔
2310
  QUERY_CHECK_CODE(code, lino, _end);
977!
2311
  for (int32_t i = 0; i < size; ++i) {
27,084✔
2312
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
26,105✔
2313
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
26,109!
2314
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
26,109✔
2315
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
26,107!
2316
  }
2317

2318
  (*ppArrayRes) = tableIdList;
979✔
2319

2320
_end:
979✔
2321
  if (code != TSDB_CODE_SUCCESS) {
979!
2322
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2323
  }
2324
  return code;
978✔
2325
}
2326

2327
static int32_t doRawScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
5,063✔
2328
  int32_t        code = TSDB_CODE_SUCCESS;
5,063✔
2329
  int32_t        lino = 0;
5,063✔
2330
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5,063✔
2331
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
5,063✔
2332

2333
  SStreamRawScanInfo* pInfo = pOperator->info;
5,063✔
2334
  pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = NULL;  // use batchMetaReq != NULL to judge if data is meta
5,063✔
2335
  pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = NULL;
5,063✔
2336

2337
  qDebug("tmqsnap doRawScan called");
5,063!
2338
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
5,063✔
2339
    bool hasNext = false;
5,045✔
2340
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
5,045!
2341
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
5,041✔
2342
      if (code != TSDB_CODE_SUCCESS) {
5,041!
2343
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2344
        QUERY_CHECK_CODE(code, lino, _end);
×
2345
      }
2346
    }
2347

2348
    if (pInfo->dataReader && hasNext) {
5,045✔
2349
      if (isTaskKilled(pTaskInfo)) {
2,532!
2350
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2351
        return code;
2,532✔
2352
      }
2353

2354
      SSDataBlock* pBlock = NULL;
2,532✔
2355
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock, NULL);
2,532✔
2356
      QUERY_CHECK_CODE(code, lino, _end);
2,532!
2357

2358
      if (pBlock && pBlock->info.rows > 0) {
2,532!
2359
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
2,532✔
2360
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
2,531✔
2361
        QUERY_CHECK_CODE(code, lino, _end);
2,531!
2362
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
2,531!
2363
        (*ppRes) = pBlock;
2,532✔
2364
        return code;
2,532✔
2365
      }
2366
    }
2367

2368
    SMetaTableInfo mtInfo = {0};
2,513✔
2369
    code = pAPI->snapshotFn.getMetaTableInfoFromSnapshot(pInfo->sContext, &mtInfo);
2,513✔
2370
    QUERY_CHECK_CODE(code, lino, _end);
2,513!
2371
    if (code != 0) {
2,513!
2372
      destroyMetaTableInfo(&mtInfo);
2373
      QUERY_CHECK_CODE(code, lino, _end);
×
2374
    }
2375
    STqOffsetVal offset = {0};
2,513✔
2376
    if (mtInfo.uid == 0 || pInfo->sContext->withMeta == ONLY_META) {  // read snapshot done, change to get data from wal
2,513!
2377
      qDebug("tmqsnap read snapshot done, change to get data from wal");
31!
2378
      tqOffsetResetToLog(&offset, pInfo->sContext->snapVersion + 1);
31✔
2379
    } else {
2380
      SValue val = {0};
2,482✔
2381
      tqOffsetResetToData(&offset, mtInfo.uid, INT64_MIN, val);
2,482✔
2382
      qDebug("tmqsnap change get data uid:%" PRId64, mtInfo.uid);
2,482!
2383
    }
2384
    destroyMetaTableInfo(&mtInfo);
2385
    code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
2,512✔
2386
    QUERY_CHECK_CODE(code, lino, _end);
2,513!
2387
    (*ppRes) = NULL;
2,513✔
2388
    return code;
2,513✔
2389
  } else if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_META) {
18!
2390
    SSnapContext* sContext = pInfo->sContext;
18✔
2391
    for (int32_t i = 0; i < tmqRowSize; i++) {
154!
2392
      void*   data = NULL;
154✔
2393
      int32_t dataLen = 0;
154✔
2394
      int16_t type = 0;
154✔
2395
      int64_t uid = 0;
154✔
2396
      if (pAPI->snapshotFn.getTableInfoFromSnapshot(sContext, &data, &dataLen, &type, &uid) < 0) {
154!
2397
        qError("tmqsnap getTableInfoFromSnapshot error");
×
2398
        taosMemoryFreeClear(data);
×
2399
        break;
18✔
2400
      }
2401

2402
      if (!sContext->queryMeta) {  // change to get data next poll request
154✔
2403
        STqOffsetVal offset = {0};
18✔
2404
        SValue       val = {0};
18!
2405
        tqOffsetResetToData(&offset, 0, INT64_MIN, val);
18✔
2406
        code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
18✔
2407
        QUERY_CHECK_CODE(code, lino, _end);
18!
2408
        break;
18✔
2409
      } else {
2410
        tqOffsetResetToMeta(&pTaskInfo->streamInfo.currentOffset, uid);
136✔
2411
        SMqMetaRsp tmpMetaRsp = {0};
136✔
2412
        tmpMetaRsp.resMsgType = type;
136✔
2413
        tmpMetaRsp.metaRspLen = dataLen;
136✔
2414
        tmpMetaRsp.metaRsp = data;
136✔
2415
        if (!pTaskInfo->streamInfo.btMetaRsp.batchMetaReq) {
136✔
2416
          pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = taosArrayInit(4, POINTER_BYTES);
16✔
2417
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, code, lino, _end, terrno);
16!
2418

2419
          pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = taosArrayInit(4, sizeof(int32_t));
16✔
2420
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, code, lino, _end, terrno);
16!
2421
        }
2422
        int32_t  tempRes = TSDB_CODE_SUCCESS;
136✔
2423
        uint32_t len = 0;
136✔
2424
        tEncodeSize(tEncodeMqMetaRsp, &tmpMetaRsp, len, tempRes);
136!
2425
        if (TSDB_CODE_SUCCESS != tempRes) {
136!
2426
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2427
          taosMemoryFreeClear(data);
×
2428
          break;
×
2429
        }
2430

2431
        int32_t tLen = sizeof(SMqRspHead) + len;
136✔
2432
        void*   tBuf = taosMemoryCalloc(1, tLen);
136!
2433
        QUERY_CHECK_NULL(tBuf, code, lino, _end, terrno);
136!
2434

2435
        void*    metaBuff = POINTER_SHIFT(tBuf, sizeof(SMqRspHead));
136✔
2436
        SEncoder encoder = {0};
136✔
2437
        tEncoderInit(&encoder, metaBuff, len);
136✔
2438
        int32_t tempLen = tEncodeMqMetaRsp(&encoder, &tmpMetaRsp);
136✔
2439
        if (tempLen < 0) {
136!
2440
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2441
          tEncoderClear(&encoder);
×
2442
          taosMemoryFreeClear(tBuf);
×
2443
          taosMemoryFreeClear(data);
×
2444
          break;
×
2445
        }
2446
        taosMemoryFreeClear(data);
136!
2447
        void* tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, &tBuf);
136✔
2448
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
136!
2449

2450
        tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, &tLen);
136✔
2451
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
136!
2452
      }
2453
    }
2454

2455
    (*ppRes) = NULL;
18✔
2456
    return code;
18✔
2457
  }
2458

2459
_end:
×
2460
  if (code != TSDB_CODE_SUCCESS) {
×
2461
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2462
    pTaskInfo->code = code;
×
2463
    T_LONG_JMP(pTaskInfo->env, code);
×
2464
  }
2465

2466
  (*ppRes) = NULL;
×
2467
  return code;
×
2468
}
2469

2470
static void destroyTmqRawScanOperatorInfo(void* param) {
316✔
2471
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
316✔
2472
  pRawScan->pAPI->tsdReader.tsdReaderClose(pRawScan->dataReader);
316✔
2473
  pRawScan->pAPI->snapshotFn.destroySnapshot(pRawScan->sContext);
316✔
2474
  tableListDestroy(pRawScan->pTableListInfo);
317✔
2475
  taosMemoryFree(pRawScan);
317!
2476
}
317✔
2477

2478
// for subscribing db or stb (not including column),
2479
// if this scan is used, meta data can be return
2480
// and schemas are decided when scanning
2481
int32_t createTmqRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
316✔
2482
  // create operator
2483
  // create tb reader
2484
  // create meta reader
2485
  // create tq reader
2486

2487
  QRY_PARAM_CHECK(pOptrInfo);
316!
2488
  int32_t code = TSDB_CODE_SUCCESS;
316✔
2489
  int32_t lino = 0;
316✔
2490

2491
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
316!
2492
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
317!
2493
  if (pInfo == NULL || pOperator == NULL) {
317!
2494
    code = terrno;
×
2495
    lino = __LINE__;
×
2496
    goto _end;
×
2497
  }
2498

2499
  pInfo->pTableListInfo = tableListCreate();
317✔
2500
  QUERY_CHECK_NULL(pInfo->pTableListInfo, code, lino, _end, terrno);
317!
2501
  pInfo->vnode = pHandle->vnode;
317✔
2502
  pInfo->pAPI = &pTaskInfo->storageAPI;
317✔
2503

2504
  pInfo->sContext = pHandle->sContext;
317✔
2505
  setOperatorInfo(pOperator, "RawScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
317✔
2506
                  pTaskInfo);
2507

2508
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScanNext, NULL, destroyTmqRawScanOperatorInfo, optrDefaultBufFn,
317✔
2509
                                         NULL, optrDefaultGetNextExtFn, NULL);
2510
  *pOptrInfo = pOperator;
317✔
2511
  return code;
317✔
2512

2513
_end:
×
2514
  if (code != TSDB_CODE_SUCCESS) {
×
2515
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2516
  }
2517
  taosMemoryFree(pInfo);
×
2518
  taosMemoryFree(pOperator);
×
2519
  pTaskInfo->code = code;
×
2520
  return code;
×
2521
}
2522

2523
void destroyTmqScanOperatorInfo(void* param) {
977✔
2524
  if (param == NULL) {
977!
2525
    return;
×
2526
  }
2527

2528
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
977✔
2529
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
977!
2530
    destroyOperator(pStreamScan->pTableScanOp);
978✔
2531
  }
2532

2533
  if (pStreamScan->tqReader != NULL && pStreamScan->readerFn.tqReaderClose != NULL) {
977!
2534
    pStreamScan->readerFn.tqReaderClose(pStreamScan->tqReader);
978✔
2535
  }
2536
  if (pStreamScan->pVtableMergeHandles) {
977!
2537
    taosHashCleanup(pStreamScan->pVtableMergeHandles);
×
2538
    pStreamScan->pVtableMergeHandles = NULL;
×
2539
  }
2540
  if (pStreamScan->pVtableMergeBuf) {
977!
2541
    destroyDiskbasedBuf(pStreamScan->pVtableMergeBuf);
×
2542
    pStreamScan->pVtableMergeBuf = NULL;
×
2543
  }
2544
  if (pStreamScan->pVtableReadyHandles) {
977!
2545
    taosArrayDestroy(pStreamScan->pVtableReadyHandles);
×
2546
    pStreamScan->pVtableReadyHandles = NULL;
×
2547
  }
2548
  if (pStreamScan->pTableListInfo) {
977!
2549
    tableListDestroy(pStreamScan->pTableListInfo);
×
2550
    pStreamScan->pTableListInfo = NULL;
×
2551
  }
2552
  if (pStreamScan->matchInfo.pList) {
977!
2553
    taosArrayDestroy(pStreamScan->matchInfo.pList);
978✔
2554
  }
2555
  if (pStreamScan->pPseudoExpr) {
977✔
2556
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
554✔
2557
    taosMemoryFree(pStreamScan->pPseudoExpr);
553!
2558
  }
2559

2560
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
976✔
2561
  cleanupExprSupp(&pStreamScan->tagCalSup);
977✔
2562

2563
  blockDataDestroy(pStreamScan->pRes);
977✔
2564
  blockDataDestroy(pStreamScan->pUpdateRes);
978✔
2565
  blockDataDestroy(pStreamScan->pDeleteDataRes);
978✔
2566
  blockDataDestroy(pStreamScan->pUpdateDataRes);
978✔
2567
  blockDataDestroy(pStreamScan->pCreateTbRes);
978✔
2568
  taosArrayDestroy(pStreamScan->pBlockLists);
978✔
2569
  blockDataDestroy(pStreamScan->pCheckpointRes);
978✔
2570

2571
  taosMemoryFree(pStreamScan);
978!
2572
}
2573

2574
int32_t addPrimaryKeyCol(SSDataBlock* pBlock, uint8_t type, int32_t bytes) {
×
2575
  int32_t code = TSDB_CODE_SUCCESS;
×
2576
  int32_t lino = 0;
×
2577
  pBlock->info.rowSize += bytes;
×
2578
  SColumnInfoData infoData = {0};
×
2579
  infoData.info.type = type;
×
2580
  infoData.info.bytes = bytes;
×
2581
  void* tmp = taosArrayPush(pBlock->pDataBlock, &infoData);
×
2582
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
2583

2584
_end:
×
2585
  if (code != TSDB_CODE_SUCCESS) {
×
2586
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2587
  }
2588
  return code;
×
2589
}
2590

2591
static SSDataBlock* createStreamVtableBlock(SColMatchInfo* pMatchInfo, const char* idstr) {
×
2592
  int32_t      code = TSDB_CODE_SUCCESS;
×
2593
  int32_t      lino = 0;
×
2594
  SSDataBlock* pRes = NULL;
×
2595

2596
  QUERY_CHECK_NULL(pMatchInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
2597

2598
  code = createDataBlock(&pRes);
×
2599
  QUERY_CHECK_CODE(code, lino, _end);
×
2600
  int32_t numOfOutput = taosArrayGetSize(pMatchInfo->pList);
×
2601
  for (int32_t i = 0; i < numOfOutput; ++i) {
×
2602
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
×
2603
    if (!pItem->needOutput) {
×
2604
      continue;
×
2605
    }
2606
    SColumnInfoData colInfo = createColumnInfoData(pItem->dataType.type, pItem->dataType.bytes, pItem->colId);
×
2607
    code = blockDataAppendColInfo(pRes, &colInfo);
×
2608
    QUERY_CHECK_CODE(code, lino, _end);
×
2609
  }
2610

2611
_end:
×
2612
  if (code != TSDB_CODE_SUCCESS) {
×
2613
    qError("%s failed at line %d since %s, id: %s", __func__, lino, tstrerror(code), idstr);
×
2614
    if (pRes != NULL) {
×
2615
      blockDataDestroy(pRes);
×
2616
    }
2617
    pRes = NULL;
×
2618
    terrno = code;
×
2619
  }
2620
  return pRes;
×
2621
}
2622

2623
 int32_t createTmqScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode,
974✔
2624
                                                  SNode* pTagCond, STableListInfo* pTableListInfo,
2625
                                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
2626
  QRY_PARAM_CHECK(pOptrInfo);
974!
2627

2628
  int32_t          code = TSDB_CODE_SUCCESS;
974✔
2629
  int32_t          lino = 0;
974✔
2630
  SArray*          pColIds = NULL;
974✔
2631
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
974!
2632
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
978!
2633
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
978✔
2634
  const char*      idstr = pTaskInfo->id.str;
978✔
2635
  SSHashObj*       pVtableInfos = pTaskInfo->pSubplan->pVTables;
978✔
2636

2637
  if (pInfo == NULL || pOperator == NULL) {
978!
2638
    code = terrno;
×
2639
    goto _error;
×
2640
  }
2641

2642
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
978✔
2643
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
978✔
2644

2645
  pInfo->pTagCond = pTagCond;
978✔
2646
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
978✔
2647

2648
  int32_t numOfCols = 0;
978✔
2649
  code = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
978✔
2650
  if (code != TSDB_CODE_SUCCESS) {
978!
2651
    goto _error;
×
2652
  }
2653

2654
  SDataType pkType = {0};
978✔
2655
  pInfo->primaryKeyIndex = -1;
978✔
2656
  pInfo->basic.primaryPkIndex = -1;
978✔
2657
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
978✔
2658
  pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
978✔
2659
  QUERY_CHECK_NULL(pColIds, code, lino, _error, terrno);
978!
2660

2661
  for (int32_t i = 0; i < numOfOutput; ++i) {
6,670✔
2662
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
5,692✔
2663
    QUERY_CHECK_NULL(id, code, lino, _error, terrno);
5,691!
2664

2665
    int16_t colId = id->colId;
5,691✔
2666
    void*   tmp = taosArrayPush(pColIds, &colId);
5,692✔
2667
    QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
5,692!
2668

2669
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
5,692✔
2670
      pInfo->primaryTsIndex = id->dstSlotId;
977✔
2671
    }
2672
    if (id->isPk) {
5,692!
2673
      pInfo->primaryKeyIndex = id->dstSlotId;
×
2674
      pInfo->basic.primaryPkIndex = id->dstSlotId;
×
2675
      pkType = id->dataType;
×
2676
    }
2677
  }
2678

2679
  pInfo->pPartTbnameSup = NULL;
978✔
2680
  if (pTableScanNode->pSubtable != NULL) {
978!
2681
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
×
2682
    if (pSubTableExpr == NULL) {
×
2683
      code = terrno;
×
2684
      goto _error;
×
2685
    }
2686

2687
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
×
2688
    code = createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
×
2689
    QUERY_CHECK_CODE(code, lino, _error);
×
2690

2691
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1, &pTaskInfo->storageAPI.functionStore) != 0) {
×
2692
      goto _error;
×
2693
    }
2694
  }
2695

2696
  if (pTableScanNode->pTags != NULL) {
978!
2697
    int32_t    numOfTags;
2698
    SExprInfo* pTagExpr = createExpr(pTableScanNode->pTags, &numOfTags);
×
2699
    if (pTagExpr == NULL) {
×
2700
      goto _error;
×
2701
    }
2702
    code = initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags, &pTaskInfo->storageAPI.functionStore);
×
2703
    if (code != 0) {
×
2704
      goto _error;
×
2705
    }
2706
  }
2707

2708
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
978✔
2709
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
978✔
2710

2711
  if (pHandle->vnode) {
977!
2712
    SOperatorInfo* pTableScanOp = NULL;
977✔
2713
    code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pTableScanOp);
977✔
2714
    if (pTableScanOp == NULL || code != 0) {
978!
2715
      qError("createTableScanOperatorInfo error, code:%d", pTaskInfo->code);
×
2716
      goto _error;
×
2717
    }
2718

2719
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
978✔
2720
    if (pHandle->version > 0) {
978!
2721
      pTSInfo->base.cond.endVersion = pHandle->version;
978✔
2722
    }
2723

2724
    STableKeyInfo* pList = NULL;
978✔
2725
    int32_t        num = 0;
978✔
2726
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
978✔
2727
    QUERY_CHECK_CODE(code, lino, _error);
976!
2728

2729
    if (pHandle->initTableReader) {
976!
2730
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
976✔
2731
      pTSInfo->base.dataReader = NULL;
976✔
2732
    }
2733

2734
    if (pHandle->initTqReader) {
976!
2735
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
976✔
2736
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
978!
2737
    } else {
2738
      pInfo->tqReader = pHandle->tqReader;
×
2739
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
2740
    }
2741

2742
    if (pVtableInfos != NULL) {
978!
2743
      // save vtable info into tqReader for vtable source scan
2744
      SSDataBlock* pResBlock = createStreamVtableBlock(&pInfo->matchInfo, idstr);
×
2745
      QUERY_CHECK_CODE(code, lino, _error);
×
2746
      code = pAPI->tqReaderFn.tqReaderSetVtableInfo(pInfo->tqReader, pHandle->vnode, pAPI, pVtableInfos, &pResBlock,
×
2747
                                                    idstr);
2748
      QUERY_CHECK_CODE(code, lino, _error);
×
2749
    }
2750

2751
    pInfo->pUpdateInfo = NULL;
978✔
2752
    pInfo->pTableScanOp = pTableScanOp;
978✔
2753
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
978!
2754
      pAPI->stateStore.streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1,
×
2755
                                            pInfo->primaryTsIndex);
2756
    }
2757

2758
    pInfo->readHandle = *pHandle;
978✔
2759
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
978✔
2760
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
978✔
2761
    QUERY_CHECK_NULL(pInfo->pCreateTbRes, code, lino, _error, terrno);
978!
2762
    pInfo->hasPart = false;
978✔
2763

2764
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
978✔
2765
    QUERY_CHECK_CODE(code, lino, _error);
977!
2766

2767
    // set the extract column id to streamHandle
2768
    code = pAPI->tqReaderFn.tqReaderSetColIdList(pInfo->tqReader, pColIds, idstr);
977✔
2769
    QUERY_CHECK_CODE(code, lino, _error);
976!
2770

2771
    SArray* tableIdList = NULL;
976✔
2772
    code = extractTableIdList(((STableScanInfo*)(pInfo->pTableScanOp->info))->base.pTableListInfo, &tableIdList);
976✔
2773
    QUERY_CHECK_CODE(code, lino, _error);
978!
2774
    code = pAPI->tqReaderFn.tqReaderSetQueryTableList(pInfo->tqReader, tableIdList, idstr);
978✔
2775
    QUERY_CHECK_CODE(code, lino, _error);
978!
2776
    taosArrayDestroy(tableIdList);
978✔
2777
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->base.cond, sizeof(SQueryTableDataCond));
978✔
2778
  } else {
2779
    taosArrayDestroy(pColIds);
×
2780
    tableListDestroy(pTableListInfo);
×
2781
  }
2782

2783
  // clear the local variable to avoid repeatly free
2784
  pColIds = NULL;
978✔
2785

2786
  // create the pseduo columns info
2787
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
978✔
2788
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->pPseudoExpr, &pInfo->numOfPseudoExpr);
554✔
2789
    QUERY_CHECK_CODE(code, lino, _error);
554!
2790
  }
2791

2792
  code = filterInitFromNode((SNode*)pScanPhyNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0, NULL);
978✔
2793
  QUERY_CHECK_CODE(code, lino, _error);
978!
2794

2795
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
978✔
2796
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
978!
2797
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
978✔
2798
  QUERY_CHECK_CODE(code, lino, _error);
978!
2799

2800
  pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
978✔
2801
  pInfo->windowSup = (SWindowSupporter){.pStreamAggSup = NULL, .gap = -1, .parentType = QUERY_NODE_PHYSICAL_PLAN};
978✔
2802
  pInfo->groupId = 0;
978✔
2803
  pInfo->igCheckGroupId = false;
978✔
2804
  pInfo->pStreamScanOp = pOperator;
978✔
2805
  pInfo->deleteDataIndex = 0;
978✔
2806
  code = createSpecialDataBlock(STREAM_DELETE_DATA, &pInfo->pDeleteDataRes);
978✔
2807
  QUERY_CHECK_CODE(code, lino, _error);
978!
2808

2809
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
978✔
2810
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateDataRes);
978✔
2811
  QUERY_CHECK_CODE(code, lino, _error);
978!
2812

2813
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
978✔
2814
  pInfo->partitionSup.needCalc = false;
978✔
2815
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
978✔
2816
  pInfo->igExpired = pTableScanNode->igExpired;
978✔
2817
  pInfo->twAggSup.maxTs = INT64_MIN;
978✔
2818
  pInfo->pState = pTaskInfo->streamInfo.pState;
978✔
2819
  pInfo->stateStore = pTaskInfo->storageAPI.stateStore;
978✔
2820
  pInfo->readerFn = pTaskInfo->storageAPI.tqReaderFn;
978✔
2821
  pInfo->pFillSup = NULL;
978✔
2822
  pInfo->useGetResultRange = false;
978✔
2823
  pInfo->pRangeScanRes = NULL;
978✔
2824

2825
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
978✔
2826
  QUERY_CHECK_CODE(code, lino, _error);
978!
2827

2828
  setOperatorInfo(pOperator, STREAM_SCAN_OP_NAME, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, false, OP_NOT_OPENED, pInfo,
978✔
2829
                  pTaskInfo);
2830
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
978✔
2831

2832
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doQueueScanNext, NULL, destroyTmqScanOperatorInfo, optrDefaultBufFn,
978✔
2833
                                         NULL, optrDefaultGetNextExtFn, NULL);
2834

2835
  *pOptrInfo = pOperator;
978✔
2836
  return code;
978✔
2837

2838
_error:
×
2839
  if (pColIds != NULL) {
×
2840
    taosArrayDestroy(pColIds);
×
2841
  }
2842

2843
  if (pInfo != NULL && pInfo->pTableScanOp != NULL) {
×
2844
    STableScanInfo* p = (STableScanInfo*)pInfo->pTableScanOp->info;
×
2845
    if (p != NULL) {
×
2846
      p->base.pTableListInfo = NULL;
×
2847
    }
2848
    destroyTmqScanOperatorInfo(pInfo);
×
2849
  }
2850

2851
  if (pOperator != NULL) {
×
2852
    pOperator->info = NULL;
×
2853
    destroyOperator(pOperator);
×
2854
  }
2855
  pTaskInfo->code = code;
×
2856
  return code;
×
2857
}
2858

2859
static int32_t doTagScanOneTable(SOperatorInfo* pOperator, SSDataBlock* pRes, SMetaReader* mr, SStorageAPI* pAPI) {
7,839✔
2860
  int32_t        code = TSDB_CODE_SUCCESS;
7,839✔
2861
  int32_t        lino = 0;
7,839✔
2862
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
7,839✔
2863
  STagScanInfo*  pInfo = pOperator->info;
7,839✔
2864
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
7,839✔
2865
  int32_t        count = pRes->info.rows;
7,839✔
2866

2867
  STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
7,839✔
2868
  if (!item) {
7,838!
2869
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
2870
           GET_TASKID(pTaskInfo));
2871
    tDecoderClear(&(*mr).coder);
×
2872
    goto _end;
×
2873
  }
2874

2875
  code = pAPI->metaReaderFn.getTableEntryByUid(mr, item->uid);
7,838✔
2876
  tDecoderClear(&(*mr).coder);
7,809✔
2877
  if (code != TSDB_CODE_SUCCESS) {
7,858!
2878
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
2879
           GET_TASKID(pTaskInfo));
2880
    goto _end;
×
2881
  }
2882

2883
  char str[512];
2884
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
24,546✔
2885
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
16,699✔
2886

2887
    // refactor later
2888
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
16,692✔
2889
      STR_TO_VARSTR(str, (*mr).me.name);
3,886✔
2890
      code = colDataSetVal(pDst, (count), str, false);
3,886✔
2891
      QUERY_CHECK_CODE(code, lino, _end);
3,880!
2892
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
12,806✔
2893
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
41✔
2894
      QUERY_CHECK_CODE(code, lino, _end);
41!
2895
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
12,765✔
2896
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
41✔
2897
      QUERY_CHECK_CODE(code, lino, _end);
41!
2898
    } else {  // it is a tag value
2899
      STagVal val = {0};
12,724✔
2900
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
12,724✔
2901
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
12,724✔
2902

2903
      char* data = NULL;
12,723✔
2904
      if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
12,723!
2905
        data = tTagValToData((const STagVal*)p, false);
12,725✔
2906
      } else {
2907
        data = (char*)p;
×
2908
      }
2909

2910
      code = colDataSetVal(pDst, (count), data,
12,724✔
2911
                           (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
12,724!
2912
      QUERY_CHECK_CODE(code, lino, _end);
12,724!
2913

2914
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
12,724!
2915
          (data != NULL)) {
2916
        taosMemoryFree(data);
3,377!
2917
      }
2918
    }
2919
  }
2920

2921
_end:
7,847✔
2922
  if (code != TSDB_CODE_SUCCESS) {
7,847!
2923
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2924
    pTaskInfo->code = code;
×
2925
  } else {
2926
    pRes->info.rows++;
7,847✔
2927
  }
2928

2929
  return code;
7,847✔
2930
}
2931

2932
static void tagScanFreeUidTag(void* p) {
15,587✔
2933
  STUidTagInfo* pInfo = p;
15,587✔
2934
  if (pInfo->pTagVal != NULL) {
15,587!
2935
    taosMemoryFree(pInfo->pTagVal);
15,587!
2936
  }
2937
}
15,588✔
2938

2939
static int32_t tagScanCreateResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
3,544✔
2940
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
3,544!
2941
  if (pColumnData == NULL) {
3,544!
2942
    return terrno;
×
2943
  }
2944

2945
  pColumnData->info.type = pType->type;
3,544✔
2946
  pColumnData->info.bytes = pType->bytes;
3,544✔
2947
  pColumnData->info.scale = pType->scale;
3,544✔
2948
  pColumnData->info.precision = pType->precision;
3,544✔
2949

2950
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
3,544✔
2951
  if (code != TSDB_CODE_SUCCESS) {
3,543!
2952
    terrno = code;
×
2953
    taosMemoryFree(pColumnData);
×
2954
    return terrno;
×
2955
  }
2956

2957
  pParam->columnData = pColumnData;
3,543✔
2958
  pParam->colAlloced = true;
3,543✔
2959
  return TSDB_CODE_SUCCESS;
3,543✔
2960
}
2961

2962
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
21,504✔
2963
  int32_t                code = TSDB_CODE_SUCCESS;
21,504✔
2964
  int32_t                lino = 0;
21,504✔
2965
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
21,504✔
2966
  SColumnNode*           pSColumnNode = NULL;
21,504✔
2967
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
21,504✔
2968
    pSColumnNode = *(SColumnNode**)pNode;
6,018✔
2969
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
15,486!
2970
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
×
2971
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
×
2972
      pSColumnNode = NULL;
×
2973
      pCtx->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
×
2974
      if (NULL == pSColumnNode) {
×
2975
        return DEAL_RES_ERROR;
×
2976
      }
2977
      pSColumnNode->colId = -1;
×
2978
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
×
2979
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
×
2980
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
×
2981
      nodesDestroyNode(*pNode);
×
2982
      *pNode = (SNode*)pSColumnNode;
×
2983
    } else {
2984
      return DEAL_RES_CONTINUE;
×
2985
    }
2986
  } else {
2987
    return DEAL_RES_CONTINUE;
15,486✔
2988
  }
2989

2990
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
6,018✔
2991
  if (!data) {
6,016✔
2992
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
3,553✔
2993
    if (code == TSDB_CODE_DUP_KEY) {
3,554!
2994
      code = TSDB_CODE_SUCCESS;
×
2995
    }
2996
    QUERY_CHECK_CODE(code, lino, _end);
3,554!
2997
    pSColumnNode->slotId = pCtx->index++;
3,554✔
2998
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
3,554✔
2999
                         .type = pSColumnNode->node.resType.type,
3,554✔
3000
                         .bytes = pSColumnNode->node.resType.bytes};
3,554✔
3001
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
3,554✔
3002
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,554!
3003
  } else {
3004
    SColumnNode* col = *(SColumnNode**)data;
2,463✔
3005
    pSColumnNode->slotId = col->slotId;
2,463✔
3006
  }
3007

3008
_end:
6,017✔
3009
  if (code != TSDB_CODE_SUCCESS) {
6,017!
3010
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3011
    return DEAL_RES_ERROR;
×
3012
  }
3013
  return DEAL_RES_CONTINUE;
6,017✔
3014
}
3015

3016
static int32_t tagScanFilterByTagCond(SArray* aUidTags, SNode* pTagCond, SArray* aFilterIdxs, void* pVnode,
3,544✔
3017
                                      SStorageAPI* pAPI, STagScanInfo* pInfo) {
3018
  int32_t code = TSDB_CODE_SUCCESS;
3,544✔
3019
  int32_t lino = 0;
3,544✔
3020
  int32_t numOfTables = taosArrayGetSize(aUidTags);
3,544✔
3021
  SArray* pBlockList = NULL;
3,544✔
3022

3023
  SSDataBlock* pResBlock = createTagValBlockForFilter(pInfo->filterCtx.cInfoList, numOfTables, aUidTags, pVnode, pAPI);
3,544✔
3024
  QUERY_CHECK_NULL(pResBlock, code, lino, _end, terrno);
3,543!
3025

3026
  pBlockList = taosArrayInit(1, POINTER_BYTES);
3,543✔
3027
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
3,544!
3028

3029
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
3,544✔
3030
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,544!
3031

3032
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
3,544✔
3033

3034
  SScalarParam output = {0};
3,544✔
3035
  code = tagScanCreateResultData(&type, numOfTables, &output);
3,544✔
3036
  QUERY_CHECK_CODE(code, lino, _end);
3,543!
3037

3038
  code = scalarCalculate(pTagCond, pBlockList, &output, NULL, NULL);
3,543✔
3039
  QUERY_CHECK_CODE(code, lino, _end);
3,543!
3040

3041
  bool* result = (bool*)output.columnData->pData;
3,543✔
3042
  for (int32_t i = 0; i < numOfTables; ++i) {
8,860✔
3043
    if (result[i]) {
5,317✔
3044
      void* tmp = taosArrayPush(aFilterIdxs, &i);
4,977✔
3045
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
4,977!
3046
    }
3047
  }
3048

3049
  colDataDestroy(output.columnData);
3,543✔
3050
  taosMemoryFreeClear(output.columnData);
3,544!
3051

3052
_end:
×
3053
  blockDataDestroy(pResBlock);
3,543✔
3054
  taosArrayDestroy(pBlockList);
3,544✔
3055
  if (code != TSDB_CODE_SUCCESS) {
3,544!
3056
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3057
  }
3058
  return code;
3,543✔
3059
}
3060

3061
static int32_t tagScanFillOneCellWithTag(SOperatorInfo* pOperator, const STUidTagInfo* pUidTagInfo,
64,094✔
3062
                                         SExprInfo* pExprInfo, SColumnInfoData* pColInfo, int rowIndex,
3063
                                         const SStorageAPI* pAPI, void* pVnode) {
3064
  int32_t code = TSDB_CODE_SUCCESS;
64,094✔
3065
  int32_t lino = 0;
64,094✔
3066
  if (QUERY_NODE_FUNCTION == pExprInfo->pExpr->nodeType) {
64,094✔
3067
    if (FUNCTION_TYPE_TBNAME == pExprInfo->pExpr->_function.functionType) {  // tbname
30,458!
3068
      char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
3069
      STR_TO_VARSTR(str, "ctbidx");
×
3070

3071
      code = colDataSetVal(pColInfo, rowIndex, str, false);
×
3072
      QUERY_CHECK_CODE(code, lino, _end);
×
3073
    } else if (FUNCTION_TYPE_TBUID == pExprInfo->pExpr->_function.functionType) {
30,458✔
3074
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
15,230✔
3075
      QUERY_CHECK_CODE(code, lino, _end);
15,232!
3076
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
15,228!
3077
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
15,231✔
3078
      QUERY_CHECK_CODE(code, lino, _end);
15,234!
3079
    }
3080
  } else {
3081
    STagVal tagVal = {0};
33,636✔
3082
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
33,636✔
3083
    if (pUidTagInfo->pTagVal == NULL) {
33,636!
3084
      colDataSetNULL(pColInfo, rowIndex);
×
3085
    } else {
3086
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
33,636✔
3087

3088
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
33,652!
3089
        colDataSetNULL(pColInfo, rowIndex);
15!
3090
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
33,637!
3091
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
3092
        QUERY_CHECK_CODE(code, lino, _end);
×
3093
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
41,886!
3094
        if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
8,242!
3095
          QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
3096
        }
3097
        char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
8,248!
3098
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
8,248!
3099

3100
        varDataSetLen(tmp, tagVal.nData);
8,248✔
3101
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
8,248✔
3102
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
8,248✔
3103
        taosMemoryFree(tmp);
8,251!
3104
        QUERY_CHECK_CODE(code, lino, _end);
8,249!
3105
      } else {
3106
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
25,395✔
3107
        QUERY_CHECK_CODE(code, lino, _end);
25,394!
3108
      }
3109
    }
3110
  }
3111

3112
_end:
64,115✔
3113
  if (code != TSDB_CODE_SUCCESS) {
64,115!
3114
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3115
  }
3116
  return code;
64,107✔
3117
}
3118

3119
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
9,997✔
3120
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
3121
  int32_t       code = TSDB_CODE_SUCCESS;
9,997✔
3122
  int32_t       lino = 0;
9,997✔
3123
  STagScanInfo* pInfo = pOperator->info;
9,997✔
3124
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
9,997✔
3125
  if (!ignoreFilterIdx) {
9,997✔
3126
    size_t szTables = taosArrayGetSize(aFilterIdxs);
3,544✔
3127
    for (int i = 0; i < szTables; ++i) {
8,524✔
3128
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
4,975✔
3129
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
4,975✔
3130
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
4,976!
3131
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
28,335✔
3132
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
23,355✔
3133
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
23,349!
3134
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
23,349✔
3135
        QUERY_CHECK_CODE(code, lino, _end);
23,359!
3136
      }
3137
    }
3138
  } else {
3139
    size_t szTables = taosArrayGetSize(aUidTags);
6,453✔
3140
    for (int i = 0; i < szTables; ++i) {
16,729✔
3141
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
10,271✔
3142
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
10,271!
3143
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
51,020✔
3144
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
40,745✔
3145
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
40,744!
3146
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
40,744✔
3147
        QUERY_CHECK_CODE(code, lino, _end);
40,749!
3148
      }
3149
    }
3150
  }
3151

3152
_end:
6,458✔
3153
  if (code != TSDB_CODE_SUCCESS) {
10,007!
3154
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3155
  }
3156
  return code;
9,998✔
3157
}
3158

3159
static int32_t doTagScanFromCtbIdxNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
19,964✔
3160
  int32_t        code = TSDB_CODE_SUCCESS;
19,964✔
3161
  int32_t        lino = 0;
19,964✔
3162
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
19,964✔
3163
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
19,964✔
3164

3165
  STagScanInfo* pInfo = pOperator->info;
19,964✔
3166
  SSDataBlock*  pRes = pInfo->pRes;
19,964✔
3167

3168
  QRY_PARAM_CHECK(ppRes);
19,964!
3169

3170
  if (pOperator->status == OP_EXEC_DONE) {
19,964✔
3171
    return TSDB_CODE_SUCCESS;
9,783✔
3172
  }
3173
  blockDataCleanup(pRes);
10,181✔
3174

3175
  if (pInfo->pCtbCursor == NULL) {
10,183!
3176
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
10,183✔
3177
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
10,183!
3178
  } else {
3179
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
3180
    QUERY_CHECK_CODE(code, lino, _end);
×
3181
  }
3182

3183
  SArray* aUidTags = pInfo->aUidTags;
10,183✔
3184
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
10,183✔
3185
  int32_t count = 0;
10,183✔
3186
  bool    ctbCursorFinished = false;
10,183✔
3187
  while (1) {
215✔
3188
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
10,398✔
3189
    taosArrayClear(aFilterIdxs);
10,397✔
3190

3191
    int32_t numTables = 0;
10,396✔
3192
    while (numTables < pOperator->resultInfo.capacity) {
25,985!
3193
      SMCtbCursor* pCur = pInfo->pCtbCursor;
25,985✔
3194
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
25,985✔
3195
      if (uid == 0) {
25,979✔
3196
        ctbCursorFinished = true;
10,396✔
3197
        break;
10,396✔
3198
      }
3199
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
15,583✔
3200
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
15,583!
3201
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
15,590!
3202

3203
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
15,590✔
3204
      void* tmp = taosArrayPush(aUidTags, &info);
15,589✔
3205
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
15,589!
3206
      ++numTables;
15,589✔
3207
    }
3208

3209
    if (numTables == 0) {
10,396✔
3210
      break;
398✔
3211
    }
3212
    bool ignoreFilterIdx = true;
9,998✔
3213
    if (pInfo->pTagCond != NULL) {
9,998✔
3214
      ignoreFilterIdx = false;
3,544✔
3215
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo);
3,544✔
3216
      QUERY_CHECK_CODE(code, lino, _end);
3,543!
3217
    } else {
3218
      ignoreFilterIdx = true;
6,454✔
3219
    }
3220

3221
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
9,997✔
3222
    QUERY_CHECK_CODE(code, lino, _end);
9,998!
3223

3224
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
9,998✔
3225

3226
    if (count != 0) {
9,998✔
3227
      break;
9,783✔
3228
    }
3229
  }
3230

3231
  if (count > 0) {
10,181✔
3232
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
9,783✔
3233
  }
3234
  if (count == 0 || ctbCursorFinished) {
10,181!
3235
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
10,181✔
3236
    pInfo->pCtbCursor = NULL;
10,183✔
3237
    setOperatorCompleted(pOperator);
10,183✔
3238
  }
3239

3240
  pRes->info.rows = count;
10,183✔
3241
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
10,183✔
3242
  if (bLimitReached) {
10,183!
3243
    setOperatorCompleted(pOperator);
×
3244
  }
3245

3246
  pOperator->resultInfo.totalRows += pRes->info.rows;
10,183✔
3247
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
10,183✔
3248

3249
_end:
10,183✔
3250
  if (code != TSDB_CODE_SUCCESS) {
10,183!
3251
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3252
    pTaskInfo->code = code;
×
3253
    T_LONG_JMP(pTaskInfo->env, code);
×
3254
  }
3255

3256
  return code;
10,183✔
3257
}
3258

3259
static int32_t createTagScanTableListInfoFromParam(SOperatorInfo* pOperator) {
3,937✔
3260
  STagScanInfo*          pInfo = pOperator->info;
3,937✔
3261
  SExecTaskInfo*         pTaskInfo = pOperator->pTaskInfo;
3,937✔
3262
  int32_t                code = 0;
3,937✔
3263
  STableListInfo*        pListInfo = pInfo->pTableListInfo;
3,937✔
3264
  STagScanOperatorParam* pParam = (STagScanOperatorParam*)pOperator->pOperatorGetParam->value;
3,937✔
3265
  tb_uid_t               pUid = pParam->vcUid;
3,937✔
3266

3267
  // qDebug("vgId:%d add total %d dynamic tables to scan, tableSeq:%d, exist num:%" PRId64 ", operator status:%d",
3268
  //        pTaskInfo->id.vgId, num, pParam->tableSeq, (int64_t)taosArrayGetSize(pListInfo->pTableList),
3269
  //        pOperator->status);
3270

3271
  STableKeyInfo info = {.groupId = 0};
3,937✔
3272

3273
  int32_t tableIdx = 0;
3,937✔
3274
  taosHashClear(pListInfo->map);
3,937✔
3275
  taosArrayClear(pListInfo->pTableList);
3,937✔
3276

3277
  if (taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
3,937!
3278
    if (TSDB_CODE_DUP_KEY == terrno) {
×
3279
    } else {
3280
      return terrno;
×
3281
    }
3282
  }
3283

3284
  info.uid = pUid;
3,937✔
3285
  void* p = taosArrayPush(pListInfo->pTableList, &info);
3,937✔
3286
  if (p == NULL) {
3,937!
3287
    return terrno;
×
3288
  }
3289

3290
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
3,937✔
3291

3292
  return code;
3,937✔
3293
}
3294

3295
static int32_t doTagScanFromMetaEntryNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
5,197✔
3296
  int32_t       code = TSDB_CODE_SUCCESS;
5,197✔
3297
  int32_t       lino = 0;
5,197✔
3298
  STagScanInfo* pInfo = pOperator->info;
5,197✔
3299
  if (pOperator->pOperatorGetParam) {
5,197✔
3300
    pOperator->resultInfo.totalRows = 0;
3,937✔
3301
    pOperator->dynamicTask = true;
3,937✔
3302
    pInfo->curPos = 0;
3,937✔
3303
    code = createTagScanTableListInfoFromParam(pOperator);
3,937✔
3304
    freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
3,937✔
3305
    pOperator->pOperatorGetParam = NULL;
3,935✔
3306
    QUERY_CHECK_CODE(code, lino, _end);
3,935!
3307

3308
    if (pOperator->status == OP_EXEC_DONE) {
3,935✔
3309
      pOperator->status = OP_OPENED;
1,802✔
3310
    }
3311
  }
3312

3313
  if (pOperator->status == OP_EXEC_DONE) {
5,195✔
3314
    (*ppRes) = NULL;
594✔
3315
    return code;
594✔
3316
  }
3317

3318
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4,601✔
3319
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
4,601✔
3320

3321
  SExprInfo*   pExprInfo = &pOperator->exprSupp.pExprInfo[0];
4,601✔
3322
  SSDataBlock* pRes = pInfo->pRes;
4,601✔
3323
  blockDataCleanup(pRes);
4,601✔
3324

3325
  int32_t size = 0;
4,601✔
3326
  code = tableListGetSize(pInfo->pTableListInfo, &size);
4,601✔
3327
  QUERY_CHECK_CODE(code, lino, _end);
4,602!
3328

3329
  if (size == 0) {
4,602✔
3330
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
69✔
3331
    (*ppRes) = NULL;
69✔
3332
    return code;
69✔
3333
  }
3334

3335
  SMetaReader mr = {0};
4,533✔
3336
  pAPI->metaReaderFn.initReader(&mr, pInfo->readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
4,533✔
3337
  pRes->info.rows = 0;
4,533✔
3338

3339
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
12,379!
3340
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
7,846✔
3341
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD &&
7,846!
3342
        code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
3343
      // ignore other error
3344
      code = TSDB_CODE_SUCCESS;
7,847✔
3345
    }
3346
    QUERY_CHECK_CODE(code, lino, _end);
7,846!
3347

3348
    if (++pInfo->curPos >= size) {
7,846✔
3349
      setOperatorCompleted(pOperator);
4,533✔
3350
    }
3351
  }
3352

3353
  pAPI->metaReaderFn.clearReader(&mr);
4,533✔
3354
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
4,533✔
3355
  if (bLimitReached) {
4,532!
3356
    setOperatorCompleted(pOperator);
×
3357
  }
3358

3359
  // qDebug("QInfo:0x%" PRIx64 ", create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
3360
  if (pOperator->status == OP_EXEC_DONE) {
4,532!
3361
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
4,532✔
3362
  }
3363

3364
  pOperator->resultInfo.totalRows += pRes->info.rows;
4,532✔
3365

3366
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
4,532!
3367
_end:
4,532✔
3368
  if (code != TSDB_CODE_SUCCESS) {
4,532!
3369
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3370
    pTaskInfo->code = code;
×
3371
    T_LONG_JMP(pTaskInfo->env, code);
×
3372
  }
3373
  return code;
4,532✔
3374
}
3375

3376
static SSDataBlock* doTagScanFromMetaEntry(SOperatorInfo* pOperator) {
×
3377
  SSDataBlock* pRes = NULL;
×
3378
  int32_t      code = doTagScanFromMetaEntryNext(pOperator, &pRes);
×
3379
  return pRes;
×
3380
}
3381

3382
static void destroyTagScanOperatorInfo(void* param) {
13,433✔
3383
  STagScanInfo* pInfo = (STagScanInfo*)param;
13,433✔
3384
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
13,433!
3385
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
3386
  }
3387
  taosHashCleanup(pInfo->filterCtx.colHash);
13,433✔
3388
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
13,434✔
3389
  taosArrayDestroy(pInfo->aFilterIdxs);
13,435✔
3390
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
13,436✔
3391

3392
  blockDataDestroy(pInfo->pRes);
13,435✔
3393
  taosArrayDestroy(pInfo->matchInfo.pList);
13,435✔
3394
  tableListDestroy(pInfo->pTableListInfo);
13,435✔
3395

3396
  pInfo->pRes = NULL;
13,435✔
3397
  pInfo->pTableListInfo = NULL;
13,435✔
3398
  taosMemoryFreeClear(param);
13,435!
3399
}
13,436✔
3400

3401
int32_t createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pTagScanNode,
13,424✔
3402
                                  STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3403
                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
3404
  QRY_PARAM_CHECK(pOptrInfo);
13,424!
3405

3406
  int32_t         code = TSDB_CODE_SUCCESS;
13,424✔
3407
  int32_t         lino = 0;
13,424✔
3408
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
13,424✔
3409
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
13,424!
3410
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
13,430!
3411
  if (pInfo == NULL || pOperator == NULL) {
13,432!
3412
    code = terrno;
×
3413
    goto _error;
×
3414
  }
3415

3416
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
13,432✔
3417

3418
  int32_t    numOfExprs = 0;
13,432✔
3419
  SExprInfo* pExprInfo = NULL;
13,432✔
3420

3421
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
13,432✔
3422
  QUERY_CHECK_CODE(code, lino, _error);
13,433!
3423

3424
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
13,433✔
3425
  QUERY_CHECK_CODE(code, lino, _error);
13,432!
3426

3427
  int32_t num = 0;
13,432✔
3428
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
13,432✔
3429
  QUERY_CHECK_CODE(code, lino, _error);
13,432!
3430

3431
  pInfo->pTagCond = pTagCond;
13,432✔
3432
  pInfo->pTagIndexCond = pTagIndexCond;
13,432✔
3433
  pInfo->suid = pPhyNode->suid;
13,432✔
3434
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
13,432✔
3435

3436
  pInfo->pTableListInfo = pTableListInfo;
13,432✔
3437
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
13,432✔
3438
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
13,433!
3439

3440
  pInfo->readHandle = *pReadHandle;
13,433✔
3441
  pInfo->curPos = 0;
13,433✔
3442

3443
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
13,433✔
3444
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
13,431✔
3445
                  pTaskInfo);
3446
  initResultSizeInfo(&pOperator->resultInfo, 4096);
13,430✔
3447
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
13,430✔
3448
  QUERY_CHECK_CODE(code, lino, _error);
13,434!
3449

3450
  if (pTagScanNode->onlyMetaCtbIdx) {
13,434✔
3451
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
10,181✔
3452
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
10,180!
3453

3454
    pInfo->aFilterIdxs = taosArrayInit(pOperator->resultInfo.capacity, sizeof(int32_t));
10,180✔
3455
    QUERY_CHECK_NULL(pInfo->aFilterIdxs, code, lino, _error, terrno);
10,181!
3456

3457
    pInfo->filterCtx.colHash =
10,182✔
3458
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
10,181✔
3459
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
10,182!
3460

3461
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
10,182✔
3462
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
10,183!
3463

3464
    if (pInfo->pTagCond != NULL) {
10,183✔
3465
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
3,554✔
3466
    }
3467
  }
3468
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
13,435✔
3469
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
13,435✔
3470
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
3471
  *pOptrInfo = pOperator;
13,434✔
3472
  return code;
13,434✔
3473

3474
_error:
×
3475
  if (pInfo) {
×
3476
    pInfo->pTableListInfo = NULL;
×
3477
  }
3478

3479
  if (pInfo != NULL) destroyTagScanOperatorInfo(pInfo);
×
3480
  if (pOperator != NULL) {
×
3481
    pOperator->info = NULL;
×
3482
    destroyOperator(pOperator);
×
3483
  }
3484
  return code;
×
3485
}
3486

3487
// table merge scan operator
3488

3489
static int32_t subTblRowCompareTsFn(const void* pLeft, const void* pRight, void* param) {
672,260✔
3490
  int32_t                 left = *(int32_t*)pLeft;
672,260✔
3491
  int32_t                 right = *(int32_t*)pRight;
672,260✔
3492
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
672,260✔
3493

3494
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
672,260✔
3495
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
672,260✔
3496

3497
  if (leftIdx == -1) {
672,260✔
3498
    return 1;
575,381✔
3499
  } else if (rightIdx == -1) {
96,879✔
3500
    return -1;
960✔
3501
  }
3502

3503
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
95,919✔
3504
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
95,919✔
3505
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
95,919!
3506
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
95,919!
3507
    ret = -1 * ret;
×
3508
  }
3509
  return ret;
95,919✔
3510
}
3511

3512
static int32_t subTblRowCompareTsPkFn(const void* pLeft, const void* pRight, void* param) {
×
3513
  int32_t                 left = *(int32_t*)pLeft;
×
3514
  int32_t                 right = *(int32_t*)pRight;
×
3515
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
×
3516

3517
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
×
3518
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
×
3519

3520
  if (leftIdx == -1) {
×
3521
    return 1;
×
3522
  } else if (rightIdx == -1) {
×
3523
    return -1;
×
3524
  }
3525

3526
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
×
3527
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
×
3528
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
×
3529
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
×
3530
    ret = -1 * ret;
×
3531
  }
3532
  if (ret == 0 && pInfo->pPkOrderInfo) {
×
3533
    ret = tsortComparBlockCell(pInfo->aInputs[left].pInputBlock, pInfo->aInputs[right].pInputBlock, leftIdx, rightIdx,
×
3534
                               pInfo->pPkOrderInfo);
×
3535
  }
3536
  return ret;
×
3537
}
3538

3539
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
8,575✔
3540
  int32_t code = TSDB_CODE_SUCCESS;
8,575✔
3541
  int32_t lino = 0;
8,575✔
3542
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
8,575✔
3543
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
8,575!
3544
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
8,592!
3545
  for (int i = 0; i < src->numOfCols; i++) {
30,391✔
3546
    dst->colList[i] = src->colList[i];
21,799✔
3547
  }
3548

3549
_end:
8,592✔
3550
  if (code != TSDB_CODE_SUCCESS) {
8,592!
3551
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3552
  }
3553
  return code;
8,596✔
3554
}
3555

3556
static int32_t fetchNextSubTableBlockFromReader(SOperatorInfo* pOperator, STmsSubTableInput* pInput,
12,875✔
3557
                                                bool* pSubTableHasBlock) {
3558
  int32_t code = 0;
12,875✔
3559

3560
  STableMergeScanInfo*    pInfo = pOperator->info;
12,875✔
3561
  SReadHandle*            pHandle = &pInfo->base.readHandle;
12,875✔
3562
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
12,875✔
3563
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
12,875✔
3564
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
12,875✔
3565

3566
  blockDataCleanup(pInput->pReaderBlock);
12,875✔
3567
  if (!pInput->bInMemReader) {
12,875✔
3568
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
6,479✔
3569
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
6,479✔
3570
    if (code != 0) {
6,472!
3571
      return code;
×
3572
    }
3573
  }
3574

3575
  pInfo->base.dataReader = pInput->pReader;
12,868✔
3576

3577
  while (true) {
26✔
3578
    bool hasNext = false;
12,894✔
3579
    code = pAPI->tsdReader.tsdNextDataBlock(pInfo->base.dataReader, &hasNext);
12,894✔
3580
    if (code != 0) {
12,900!
3581
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3582
      pInfo->base.dataReader = NULL;
×
3583
      return code;
×
3584
    }
3585

3586
    if (!hasNext || isTaskKilled(pTaskInfo)) {
12,900!
3587
      if (isTaskKilled(pTaskInfo)) {
8,598!
3588
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3589
        pInfo->base.dataReader = NULL;
×
3590
        return code;
×
3591
      }
3592

3593
      *pSubTableHasBlock = false;
8,600✔
3594
      break;
8,600✔
3595
    }
3596

3597
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
4,303✔
3598
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
4,219✔
3599
    } else {
3600
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
84✔
3601
    }
3602

3603
    uint32_t status = 0;
4,303✔
3604
    code = loadDataBlock(pOperator, &pInfo->base, pInput->pReaderBlock, &status);
4,303✔
3605
    if (code != 0) {
4,302!
3606
      pInfo->base.dataReader = NULL;
×
3607
      return code;
×
3608
    }
3609

3610
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
4,302!
3611
      *pSubTableHasBlock = false;
×
3612
      break;
×
3613
    }
3614
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pInput->pReaderBlock->info.rows == 0) {
4,302!
3615
      continue;
26✔
3616
    }
3617

3618
    *pSubTableHasBlock = true;
4,276✔
3619
    break;
4,276✔
3620
  }
3621

3622
  if (*pSubTableHasBlock) {
12,876✔
3623
    pInput->pReaderBlock->info.id.groupId =
8,553✔
3624
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
4,276✔
3625
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
4,277✔
3626
  }
3627
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
12,877✔
3628
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
9,685✔
3629
    pInput->pReader = NULL;
9,687✔
3630
  }
3631

3632
  pInfo->base.dataReader = NULL;
12,879✔
3633
  return TSDB_CODE_SUCCESS;
12,879✔
3634
}
3635

3636
static int32_t setGroupStartEndIndex(STableMergeScanInfo* pInfo) {
5,359✔
3637
  pInfo->bGroupProcessed = false;
5,359✔
3638

3639
  int32_t numOfTables = 0;
5,359✔
3640
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
5,359✔
3641
  if (code != TSDB_CODE_SUCCESS) {
5,381!
3642
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3643
    return code;
×
3644
  }
3645

3646
  int32_t i = pInfo->tableStartIndex + 1;
5,385✔
3647
  for (; i < numOfTables; ++i) {
8,596✔
3648
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
3,379✔
3649
    if (!tableKeyInfo) {
3,382✔
3650
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
1!
3651
      return terrno;
1✔
3652
    }
3653
    if (tableKeyInfo->groupId != pInfo->groupId) {
3,381✔
3654
      break;
170✔
3655
    }
3656
  }
3657
  pInfo->tableEndIndex = i - 1;
5,387✔
3658
  return TSDB_CODE_SUCCESS;
5,387✔
3659
}
3660

3661
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
5,390✔
3662
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
13,990✔
3663
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
8,601✔
3664
    if (pInput->rowIdx == -1) {
8,601✔
3665
      continue;
4,324✔
3666
    }
3667

3668
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
4,277!
3669
      pInput->rowIdx = 0;
4,277✔
3670
      pInput->pageIdx = -1;
4,277✔
3671
    }
3672

3673
    pInput->pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
4,277!
3674
    SColumnInfoData* col = taosArrayGet(pInput->pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
4,277✔
3675
    if (!col) {
4,277✔
3676
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
1!
3677
      return terrno;
1✔
3678
    }
3679
    pInput->aTs = (int64_t*)col->pData;
4,276✔
3680
  }
3681

3682
  __merge_compare_fn_t mergeCompareFn = (!pSubTblsInfo->pPkOrderInfo) ? subTblRowCompareTsFn : subTblRowCompareTsPkFn;
5,389!
3683
  return tMergeTreeCreate(&pSubTblsInfo->pTree, pSubTblsInfo->numSubTables, pSubTblsInfo, mergeCompareFn);
5,389✔
3684
}
3685

3686
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
5,357✔
3687
  int32_t code = setGroupStartEndIndex(pInfo);
5,357✔
3688
  if (code != TSDB_CODE_SUCCESS) {
5,380!
3689
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3690
    return code;
×
3691
  }
3692
  STmsSubTablesMergeInfo* pSubTblsInfo = taosMemoryCalloc(1, sizeof(STmsSubTablesMergeInfo));
5,380!
3693
  if (pSubTblsInfo == NULL) {
5,382!
3694
    return terrno;
×
3695
  }
3696
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
5,382✔
3697
  if (!pSubTblsInfo->pTsOrderInfo) {
5,380!
3698
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3699
    return terrno;
×
3700
  }
3701
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
5,380✔
3702
    pSubTblsInfo->pPkOrderInfo = taosArrayGet(pInfo->pSortInfo, 1);
6✔
3703
    if (!pSubTblsInfo->pPkOrderInfo) {
×
3704
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3705
      return terrno;
×
3706
    }
3707
  } else {
3708
    pSubTblsInfo->pPkOrderInfo = NULL;
5,372✔
3709
  }
3710
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
5,372✔
3711
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
5,372!
3712
  if (pSubTblsInfo->aInputs == NULL) {
5,375!
3713
    taosMemoryFree(pSubTblsInfo);
×
3714
    return terrno;
×
3715
  }
3716
  int32_t bufPageSize = pInfo->bufPageSize;
5,375✔
3717
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
5,375✔
3718
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
5,375✔
3719
  if (code != TSDB_CODE_SUCCESS) {
5,378✔
3720
    taosMemoryFree(pSubTblsInfo->aInputs);
4!
3721
    taosMemoryFree(pSubTblsInfo);
×
3722
    return code;
×
3723
  }
3724
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
5,374✔
3725
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
5,374✔
3726

3727
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
5,374✔
3728
  return TSDB_CODE_SUCCESS;
5,374✔
3729
}
3730

3731
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
5,370✔
3732
  int32_t        code = TSDB_CODE_SUCCESS;
5,370✔
3733
  int32_t        lino = 0;
5,370✔
3734
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5,370✔
3735
  SReadHandle*   pHandle = &pInfo->base.readHandle;
5,370✔
3736
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
5,370✔
3737

3738
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
5,370✔
3739

3740
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
13,972✔
3741
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
8,581✔
3742
    pInput->type = SUB_TABLE_MEM_BLOCK;
8,581✔
3743

3744
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
8,581✔
3745
    QUERY_CHECK_CODE(code, lino, _end);
8,597!
3746

3747
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
8,597✔
3748
    QUERY_CHECK_CODE(code, lino, _end);
8,599!
3749

3750
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
8,599✔
3751
    QUERY_CHECK_CODE(code, lino, _end);
8,600!
3752

3753
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
8,600✔
3754
    pInput->pKeyInfo = keyInfo;
8,600✔
3755

3756
    if (isTaskKilled(pTaskInfo)) {
8,600!
3757
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
×
3758
    }
3759

3760
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
8,600✔
3761
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
3,210✔
3762
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
3,210✔
3763
      QUERY_CHECK_CODE(code, lino, _end);
3,212!
3764
      pInput->bInMemReader = true;
3,212✔
3765
    } else {
3766
      pInput->pReader = NULL;
5,390✔
3767
      pInput->bInMemReader = false;
5,390✔
3768
    }
3769
    bool hasNext = true;
8,602✔
3770
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
8,602✔
3771
    QUERY_CHECK_CODE(code, lino, _end);
8,602!
3772
    if (!hasNext) {
8,602✔
3773
      pInput->rowIdx = -1;
4,325✔
3774
      ++pSubTblsInfo->numSubTablesCompleted;
4,325✔
3775
      continue;
4,325✔
3776
    } else {
3777
      pInput->rowIdx = 0;
4,277✔
3778
      pInput->pageIdx = -1;
4,277✔
3779
    }
3780
  }
3781

3782
_end:
5,391✔
3783
  if (code != TSDB_CODE_SUCCESS) {
5,391!
3784
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3785
  }
3786
  return code;
5,390✔
3787
}
3788

3789
static void adjustSubTableFromMemBlock(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
4,277✔
3790
  int32_t              code = TSDB_CODE_SUCCESS;
4,277✔
3791
  int32_t              lino = 0;
4,277✔
3792
  SExecTaskInfo*       pTaskInfo = pOperatorInfo->pTaskInfo;
4,277✔
3793
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
4,277✔
3794
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
4,277✔
3795
  bool                 hasNext = true;
4,277✔
3796
  code = fetchNextSubTableBlockFromReader(pOperatorInfo, pInput, &hasNext);
4,277✔
3797
  QUERY_CHECK_CODE(code, lino, _end);
4,277!
3798

3799
  if (!hasNext) {
4,277!
3800
    pInput->rowIdx = -1;
4,277✔
3801
    ++pSubTblsInfo->numSubTablesCompleted;
4,277✔
3802
  } else {
3803
    pInput->rowIdx = 0;
×
3804
  }
3805

3806
_end:
4,277✔
3807
  if (code != TSDB_CODE_SUCCESS) {
4,277!
3808
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3809
    pTaskInfo->code = code;
×
3810
    T_LONG_JMP(pTaskInfo->env, code);
×
3811
  }
3812
}
4,277✔
3813

3814
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
769,423✔
3815
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
769,423✔
3816
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
769,423✔
3817

3818
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
769,423✔
3819
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
769,423✔
3820
    ++pInput->rowIdx;
764,682✔
3821
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
4,741✔
3822
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
4,277!
3823
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
4,277✔
3824
    }
3825
    if (pInput->rowIdx != -1) {
4,277!
3826
      SColumnInfoData* col = taosArrayGet(pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
×
3827
      if (!col) {
×
3828
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3829
        return terrno;
×
3830
      }
3831
      pInput->pInputBlock = pInputBlock;
×
3832
      pInput->aTs = (int64_t*)col->pData;
×
3833
    }
3834
  }
3835

3836
  return tMergeTreeAdjust(pSubTblsInfo->pTree, tMergeTreeGetAdjustIndex(pSubTblsInfo->pTree));
769,423✔
3837
}
3838

3839
static int32_t appendChosenRowToDataBlock(STmsSubTablesMergeInfo* pSubTblsInfo, SSDataBlock* pBlock) {
769,856✔
3840
  int32_t            code = TSDB_CODE_SUCCESS;
769,856✔
3841
  int32_t            lino = 0;
769,856✔
3842
  STmsSubTableInput* pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
769,856✔
3843
  SSDataBlock*       pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
769,856!
3844

3845
  for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
3,245,333✔
3846
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
2,429,153✔
3847
    QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
2,459,379!
3848

3849
    SColumnInfoData* pSrcColInfo = taosArrayGet(pInputBlock->pDataBlock, i);
2,459,379✔
3850
    QUERY_CHECK_NULL(pSrcColInfo, code, lino, _end, terrno);
2,443,248!
3851
    bool isNull = colDataIsNull(pSrcColInfo, pInputBlock->info.rows, pInput->rowIdx, NULL);
2,459,801✔
3852

3853
    if (isNull) {
2,459,801✔
3854
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
287,331✔
3855
      QUERY_CHECK_CODE(code, lino, _end);
287,820!
3856
    } else {
3857
      if (pSrcColInfo->pData != NULL) {
2,172,470✔
3858
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
2,170,334!
3859
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
2,170,334✔
3860
        QUERY_CHECK_CODE(code, lino, _end);
2,185,521!
3861
      }
3862
    }
3863
  }
3864
  pBlock->info.dataLoad = 1;
770,188✔
3865
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
770,188✔
3866
  pBlock->info.rows += 1;
770,188✔
3867

3868
_end:
770,188✔
3869
  if (code != TSDB_CODE_SUCCESS) {
770,188!
3870
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3871
  }
3872
  return code;
770,194✔
3873
}
3874

3875
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
8,707✔
3876
                                       SSDataBlock** pResBlock) {
3877
  int32_t                 code = TSDB_CODE_SUCCESS;
8,707✔
3878
  int32_t                 lino = 0;
8,707✔
3879
  STableMergeScanInfo*    pInfo = pOperator->info;
8,707✔
3880
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
8,707✔
3881
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
8,707✔
3882
  bool                    finished = false;
8,707✔
3883

3884
  QRY_PARAM_CHECK(pResBlock);
8,707!
3885

3886
  blockDataCleanup(pBlock);
8,707✔
3887

3888
  while (true) {
×
3889
    while (true) {
3890
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
778,749✔
3891
        finished = true;
8,705✔
3892
        break;
8,705✔
3893
      }
3894

3895
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
770,044✔
3896
      QUERY_CHECK_CODE(code, lino, _end);
770,110!
3897

3898
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
770,110✔
3899
      QUERY_CHECK_CODE(code, lino, _end);
769,993!
3900

3901
      if (pBlock->info.rows >= capacity) {
769,993!
3902
        break;
×
3903
      }
3904
    }
3905

3906
    if (isTaskKilled(pTaskInfo)) {
8,705!
3907
      return pTaskInfo->code;
×
3908
    }
3909

3910
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo);
8,705✔
3911
    if (finished || limitReached || pBlock->info.rows > 0) {
8,703!
3912
      break;
3913
    }
3914
  }
3915

3916
  if (pBlock->info.rows > 0) {
8,703✔
3917
    *pResBlock = pBlock;
3,317✔
3918
  }
3919

3920
_end:
5,386✔
3921
  if (code != TSDB_CODE_SUCCESS) {
8,703!
3922
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3923
    pTaskInfo->code = code;
×
3924
  }
3925
  return code;
8,703✔
3926
}
3927

3928
static int32_t startSubTablesTableMergeScan(SOperatorInfo* pOperator) {
5,354✔
3929
  int32_t              code = TSDB_CODE_SUCCESS;
5,354✔
3930
  int32_t              lino = 0;
5,354✔
3931
  STableMergeScanInfo* pInfo = pOperator->info;
5,354✔
3932

3933
  code = initSubTablesMergeInfo(pInfo);
5,354✔
3934
  QUERY_CHECK_CODE(code, lino, _end);
5,375!
3935

3936
  code = initSubTableInputs(pOperator, pInfo);
5,375✔
3937
  QUERY_CHECK_CODE(code, lino, _end);
5,390!
3938

3939
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
5,390✔
3940
  QUERY_CHECK_CODE(code, lino, _end);
5,390!
3941

3942
_end:
5,390✔
3943
  if (code != TSDB_CODE_SUCCESS) {
5,390!
3944
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3945
  }
3946
  return code;
5,390✔
3947
}
3948

3949
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
70,276✔
3950
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
70,276✔
3951
  if (pSubTblsInfo != NULL) {
70,276✔
3952
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
5,389✔
3953

3954
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
13,991✔
3955
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
8,601✔
3956
      taosMemoryFree(pInput->tblCond.colList);
8,601!
3957
      blockDataDestroy(pInput->pReaderBlock);
8,602✔
3958
      blockDataDestroy(pInput->pPageBlock);
8,602✔
3959
      taosArrayDestroy(pInput->aBlockPages);
8,601✔
3960
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
8,601✔
3961
      pInput->pReader = NULL;
8,601✔
3962
    }
3963

3964
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
5,390✔
3965
    taosMemoryFree(pSubTblsInfo->aInputs);
5,390!
3966

3967
    taosMemoryFree(pSubTblsInfo);
5,390!
3968
    pInfo->pSubTablesMergeInfo = NULL;
5,390✔
3969

3970
    // taosMemoryTrim(0);
3971
  }
3972
}
70,277✔
3973

3974
int32_t doTableMergeScanParaSubTablesNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
8,611✔
3975
  QRY_PARAM_CHECK(ppRes);
8,611!
3976

3977
  int32_t lino = 0;
8,611✔
3978
  int32_t tableListSize = 0;
8,611✔
3979
  int64_t st = taosGetTimestampUs();
8,633✔
3980

3981
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
8,633✔
3982
  STableMergeScanInfo* pInfo = pOperator->info;
8,633✔
3983

3984
  if (pOperator->status == OP_EXEC_DONE) {
8,633!
3985
    return TSDB_CODE_SUCCESS;
×
3986
  }
3987

3988
  int32_t code = pOperator->fpSet._openFn(pOperator);
8,633✔
3989
  QUERY_CHECK_CODE(code, lino, _end);
8,615!
3990

3991
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
8,615✔
3992
  QUERY_CHECK_CODE(code, lino, _end);
8,636!
3993

3994
  if (!pInfo->hasGroupId) {
8,636✔
3995
    pInfo->hasGroupId = true;
5,320✔
3996

3997
    if (tableListSize == 0) {
5,320✔
3998
      setOperatorCompleted(pOperator);
110✔
3999
      (*ppRes) = NULL;
110✔
4000
      return code;
110✔
4001
    }
4002

4003
    pInfo->tableStartIndex = 0;
5,210✔
4004
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
5,210✔
4005
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
5,202!
4006
    pInfo->groupId = pTmpGpId->groupId;
5,202✔
4007
    code = startSubTablesTableMergeScan(pOperator);
5,202✔
4008
    QUERY_CHECK_CODE(code, lino, _end);
5,220!
4009
  }
4010

4011
  SSDataBlock* pBlock = NULL;
8,536✔
4012
  while (pInfo->tableStartIndex < tableListSize) {
8,706!
4013
    if (isTaskKilled(pTaskInfo)) {
8,707!
4014
      break;
×
4015
    }
4016

4017
    code = getSubTablesSortedBlock(pOperator, pInfo->pResBlock, pOperator->resultInfo.capacity, &pBlock);
8,707✔
4018
    QUERY_CHECK_CODE(code, lino, _end);
8,705!
4019

4020
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
8,705!
4021
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
×
4022
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
×
4023

4024
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
×
4025
    }
4026

4027
    if (pBlock != NULL) {
8,705✔
4028
      pBlock->info.id.groupId = pInfo->groupId;
3,317✔
4029
      pOperator->resultInfo.totalRows += pBlock->info.rows;
3,317✔
4030
      pInfo->bGroupProcessed = true;
3,317✔
4031
      break;
3,317✔
4032
    } else {
4033
      // Data of this group are all dumped, let's try the next group
4034
      stopSubTablesTableMergeScan(pInfo);
5,388✔
4035
      if (pInfo->tableEndIndex >= tableListSize - 1) {
5,390✔
4036
        setOperatorCompleted(pOperator);
5,220✔
4037
        break;
5,220✔
4038
      }
4039

4040
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
170✔
4041
      STableKeyInfo* pTmpGpId = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
170✔
4042
      QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
170!
4043

4044
      pInfo->groupId = pTmpGpId->groupId;
170✔
4045
      code = startSubTablesTableMergeScan(pOperator);
170✔
4046
      QUERY_CHECK_CODE(code, lino, _end);
170!
4047
      resetLimitInfoForNextGroup(&pInfo->limitInfo);
170✔
4048
    }
4049
  }
4050

4051
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
8,535✔
4052

4053
_end:
8,535✔
4054
  if (code != TSDB_CODE_SUCCESS) {
8,535!
4055
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4056
    pTaskInfo->code = code;
×
4057
    T_LONG_JMP(pTaskInfo->env, code);
×
4058
  } else {
4059
    (*ppRes) = pBlock;
8,535✔
4060
  }
4061

4062
  return code;
8,535✔
4063
}
4064

4065
static void tableMergeScanDoSkipTable(uint64_t uid, void* pTableMergeOpInfo) {
6,952✔
4066
  int32_t              code = TSDB_CODE_SUCCESS;
6,952✔
4067
  int32_t              lino = 0;
6,952✔
4068
  SOperatorInfo*       pOperator = (SOperatorInfo*)pTableMergeOpInfo;
6,952✔
4069
  STableMergeScanInfo* pInfo = pOperator->info;
6,952✔
4070
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
6,952✔
4071

4072
  if (pInfo->mSkipTables == NULL) {
6,952✔
4073
    pInfo->mSkipTables = taosHashInit(pInfo->tableEndIndex - pInfo->tableStartIndex + 1,
621✔
4074
                                      taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_NO_LOCK);
4075
    QUERY_CHECK_NULL(pInfo->mSkipTables, code, lino, _end, terrno);
621!
4076
  }
4077
  int bSkip = 1;
6,952✔
4078
  if (pInfo->mSkipTables != NULL) {
6,952!
4079
    code = taosHashPut(pInfo->mSkipTables, &uid, sizeof(uid), &bSkip, sizeof(bSkip));
6,952✔
4080
    if (code == TSDB_CODE_DUP_KEY) {
6,949!
4081
      code = TSDB_CODE_SUCCESS;
×
4082
    }
4083
    QUERY_CHECK_CODE(code, lino, _end);
6,949!
4084
  }
4085

4086
_end:
6,949✔
4087
  if (code != TSDB_CODE_SUCCESS) {
6,949!
4088
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4089
    pTaskInfo->code = code;
×
4090
    T_LONG_JMP(pTaskInfo->env, code);
×
4091
  }
4092
}
6,949✔
4093

4094
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
311,066✔
4095
  STableMergeScanInfo* pInfo = pOperator->info;
311,066✔
4096
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
311,066✔
4097
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
311,066✔
4098
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
311,066✔
4099
  int32_t              code = 0;
311,066✔
4100
  bool                 hasNext = false;
311,066✔
4101
  STsdbReader*         reader = pInfo->base.dataReader;
311,066✔
4102

4103
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
311,066✔
4104
  if (code != 0) {
311,018!
4105
    pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
4106
    qError("table merge scan fetch next data block error code: %d, %s", code, GET_TASKID(pTaskInfo));
×
4107
    pTaskInfo->code = code;
×
4108
    return code;
×
4109
  }
4110

4111
  if (!hasNext || isTaskKilled(pTaskInfo)) {
311,018!
4112
    if (isTaskKilled(pTaskInfo)) {
65,248!
4113
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
×
4114
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
4115
    }
4116
    *pFinished = true;
65,283✔
4117
    return code;
65,283✔
4118
  }
4119

4120
  uint32_t status = 0;
245,776✔
4121
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
245,776✔
4122

4123
  if (code != TSDB_CODE_SUCCESS) {
245,720!
4124
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
×
4125
    pTaskInfo->code = code;
×
4126
    return code;
×
4127
  }
4128

4129
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
245,721!
4130
    *pFinished = true;
×
4131
    return code;
×
4132
  }
4133

4134
  // current block is filter out according to filter condition, continue load the next block
4135
  if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
245,721!
4136
    *pSkipped = true;
7,601✔
4137
    return code;
7,601✔
4138
  }
4139

4140
  return code;
238,120✔
4141
}
4142

4143
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
310,700✔
4144
  STableMergeScanSortSourceParam* source = param;
310,700✔
4145

4146
  SOperatorInfo*       pOperator = source->pOperator;
310,700✔
4147
  STableMergeScanInfo* pInfo = pOperator->info;
310,700✔
4148
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
310,700✔
4149
  SSDataBlock*         pBlock = NULL;
310,700✔
4150
  int64_t              st = taosGetTimestampUs();
310,709✔
4151
  int32_t              code = TSDB_CODE_SUCCESS;
310,709✔
4152

4153
  while (true) {
4154
    if (pInfo->rtnNextDurationBlocks) {
325,336✔
4155
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
14,259!
4156
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
4157

4158
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
14,261✔
4159
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
7,040✔
4160
        ++pInfo->nextDurationBlocksIdx;
7,040✔
4161
      } else {
4162
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
14,259✔
4163
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
7,038✔
4164
          pInfo->nextDurationBlocks[i] = NULL;
7,038✔
4165
        }
4166

4167
        pInfo->rtnNextDurationBlocks = false;
7,221✔
4168
        pInfo->nextDurationBlocksIdx = 0;
7,221✔
4169
        pInfo->numNextDurationBlocks = 0;
7,221✔
4170
        continue;
7,221✔
4171
      }
4172
    } else {
4173
      bool bFinished = false;
311,077✔
4174
      bool bSkipped = false;
311,077✔
4175

4176
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
311,077✔
4177
      if (code != 0) {
310,968!
4178
        return code;
7,222✔
4179
      }
4180

4181
      pBlock = pInfo->pReaderBlock;
310,968✔
4182
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
310,968✔
4183
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
4184
      if (bFinished) {
311,120✔
4185
        pInfo->bNewFilesetEvent = false;
65,285✔
4186
        break;
65,285✔
4187
      }
4188

4189
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
245,835!
4190
        if (!bSkipped) {
7,222✔
4191
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
7,040✔
4192
          if (code) {
7,040!
4193
            *ppBlock = NULL;
×
4194
            return code;
×
4195
          }
4196

4197
          ++pInfo->numNextDurationBlocks;
7,040✔
4198
          if (pInfo->numNextDurationBlocks > 2) {
7,040!
4199
            qError("%s table merge scan prefetch %d next duration blocks. end early.", GET_TASKID(pTaskInfo),
×
4200
                   pInfo->numNextDurationBlocks);
4201
            pInfo->bNewFilesetEvent = false;
×
4202
            break;
×
4203
          }
4204
        }
4205

4206
        if (pInfo->bNewFilesetEvent) {
7,222!
4207
          pInfo->rtnNextDurationBlocks = true;
7,222✔
4208
          *ppBlock = NULL;
7,222✔
4209
          return code;
7,222✔
4210
        }
4211

4212
        if (pInfo->bNextDurationBlockEvent) {
×
4213
          pInfo->bNextDurationBlockEvent = false;
×
4214
          continue;
7,406✔
4215
        }
4216
      }
4217
      if (bSkipped) continue;
238,613✔
4218
    }
4219

4220
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
238,247✔
4221

4222
    pOperator->resultInfo.totalRows += pBlock->info.rows;
238,225✔
4223
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
238,227✔
4224
    *ppBlock = pBlock;
238,227✔
4225

4226
    return code;
238,227✔
4227
  }
4228

4229
  *ppBlock = NULL;
65,285✔
4230
  return code;
65,285✔
4231
}
4232

4233
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
52,688✔
4234
  int32_t code = TSDB_CODE_SUCCESS;
52,688✔
4235
  int32_t lino = 0;
52,688✔
4236
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
52,688✔
4237
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
52,721!
4238
  SBlockOrderInfo biTs = {0};
52,727✔
4239
  SBlockOrderInfo biPk = {0};
52,727✔
4240

4241
  int32_t tsTargetSlotId = 0;
52,727✔
4242
  int32_t pkTargetSlotId = -1;
52,727✔
4243
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
186,860✔
4244
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
134,156✔
4245
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
134,156!
4246
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
134,160✔
4247
      tsTargetSlotId = colInfo->dstSlotId;
52,714✔
4248
      biTs.order = order;
52,714✔
4249
      biTs.slotId = tsTargetSlotId;
52,714✔
4250
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
52,714✔
4251
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
52,714✔
4252
    }
4253
    // TODO: order by just ts
4254
    if (colInfo->isPk) {
134,132✔
4255
      pkTargetSlotId = colInfo->dstSlotId;
1,705✔
4256
      biPk.order = order;
1,705✔
4257
      biPk.slotId = pkTargetSlotId;
1,705✔
4258
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
1,705✔
4259
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
1,705✔
4260
    }
4261
  }
4262

4263
  void* tmp = taosArrayPush(pSortInfo, &biTs);
52,709✔
4264
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
52,709!
4265
  if (pkTargetSlotId != -1) {
52,709✔
4266
    tmp = taosArrayPush(pSortInfo, &biPk);
1,707✔
4267
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,707!
4268
  }
4269

4270
  (*ppSortArray) = pSortInfo;
52,709✔
4271

4272
_end:
52,709✔
4273
  if (code != TSDB_CODE_SUCCESS) {
52,709!
4274
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4275
  }
4276
  return code;
52,661✔
4277
}
4278

4279
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
9,685✔
4280
  STableMergeScanInfo* pTmsInfo = param;
9,685✔
4281
  if (type == TSD_READER_NOTIFY_DURATION_START) {
9,685!
4282
    pTmsInfo->bNewFilesetEvent = true;
9,687✔
4283
  } else if (type == TSD_READER_NOTIFY_NEXT_DURATION_BLOCK) {
×
4284
    pTmsInfo->bNextDurationBlockEvent = true;
×
4285
  }
4286
  qDebug("table merge scan receive notification. type %d, fileset %d", type, info->duration.filesetId);
9,685!
4287

4288
  return;
9,690✔
4289
}
4290

4291
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
72,937✔
4292
  STableMergeScanInfo* pInfo = pOperator->info;
72,937✔
4293
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
72,937✔
4294

4295
  int32_t code = TSDB_CODE_SUCCESS;
72,937✔
4296
  int32_t lino = 0;
72,937✔
4297
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
72,937✔
4298

4299
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
72,937✔
4300
  pInfo->bNewFilesetEvent = false;
72,954✔
4301
  pInfo->bNextDurationBlockEvent = false;
72,954✔
4302

4303
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
72,954✔
4304
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
72,954✔
4305

4306
  pInfo->pSortHandle = NULL;
72,954✔
4307
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
72,954✔
4308
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
72,954✔
4309
  if (code) {
72,959!
4310
    return code;
×
4311
  }
4312

4313
  if (pInfo->bSortRowId && numOfTable != 1) {
72,959✔
4314
    int32_t memSize = 512 * 1024 * 1024;
3,218✔
4315
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
3,218✔
4316
    QUERY_CHECK_CODE(code, lino, _end);
3,220!
4317
  }
4318

4319
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
72,961✔
4320
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
72,956✔
4321
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
72,946✔
4322

4323
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
72,959✔
4324
  QUERY_CHECK_CODE(code, lino, _end);
72,961!
4325

4326
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
72,961!
4327
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
72,977!
4328
  param->pOperator = pOperator;
72,977✔
4329

4330
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
72,977✔
4331
  if (ps == NULL) {
72,975!
4332
    taosMemoryFree(param);
×
4333
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
4334
  }
4335

4336
  ps->param = param;
72,975✔
4337
  ps->onlyRef = false;
72,975✔
4338
  code = tsortAddSource(pInfo->pSortHandle, ps);
72,975✔
4339
  QUERY_CHECK_CODE(code, lino, _end);
72,965!
4340

4341
  if (numOfTable == 1) {
72,965✔
4342
    tsortSetSingleTableMerge(pInfo->pSortHandle);
54,540✔
4343
  } else {
4344
    code = tsortOpen(pInfo->pSortHandle);
18,425✔
4345
    QUERY_CHECK_CODE(code, lino, _end);
18,430!
4346
  }
4347

4348
_end:
18,430✔
4349
  if (code != TSDB_CODE_SUCCESS) {
72,967!
4350
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4351
  }
4352
  return code;
72,967✔
4353
}
4354

4355
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
72,976✔
4356
  STableMergeScanInfo* pInfo = pOperator->info;
72,976✔
4357
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
72,976✔
4358
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
72,976✔
4359

4360
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
72,978✔
4361
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
72,978✔
4362
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
72,978✔
4363
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
72,978✔
4364
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
72,978✔
4365
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
72,978✔
4366

4367
  tsortDestroySortHandle(pInfo->pSortHandle);
72,978✔
4368
  pInfo->pSortHandle = NULL;
72,980✔
4369
}
72,980✔
4370

4371
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
65,727✔
4372
  int32_t              code = TSDB_CODE_SUCCESS;
65,727✔
4373
  int32_t              lino = 0;
65,727✔
4374
  STableMergeScanInfo* pInfo = pOperator->info;
65,727✔
4375
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
65,727✔
4376
  SReadHandle*         pHandle = &pInfo->base.readHandle;
65,727✔
4377
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
65,727✔
4378
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
65,727✔
4379

4380
  {
4381
    int32_t numOfTables = 0;
65,728✔
4382
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
65,728✔
4383
    QUERY_CHECK_CODE(code, lino, _end);
65,746!
4384

4385
    int32_t i = pInfo->tableStartIndex + 1;
65,746✔
4386
    for (; i < numOfTables; ++i) {
270,146✔
4387
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
217,171✔
4388
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
217,171!
4389
      if (tableKeyInfo->groupId != pInfo->groupId) {
217,171✔
4390
        break;
12,771✔
4391
      }
4392
    }
4393
    pInfo->tableEndIndex = i - 1;
65,746✔
4394
  }
4395
  pInfo->bGroupProcessed = false;
65,746✔
4396
  int32_t tableStartIdx = pInfo->tableStartIndex;
65,746✔
4397
  int32_t tableEndIdx = pInfo->tableEndIndex;
65,746✔
4398

4399
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
65,746✔
4400
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
65,746✔
4401
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
65,750✔
4402
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
65,750✔
4403
  QUERY_CHECK_CODE(code, lino, _end);
65,724!
4404
  if (pInfo->filesetDelimited) {
65,724!
4405
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
65,740✔
4406
  }
4407
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
65,692✔
4408

4409
  code = startDurationForGroupTableMergeScan(pOperator);
65,738✔
4410
  QUERY_CHECK_CODE(code, lino, _end);
65,741!
4411

4412
_end:
65,741✔
4413
  if (code != TSDB_CODE_SUCCESS) {
65,741!
4414
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4415
    pTaskInfo->code = code;
×
4416
    T_LONG_JMP(pTaskInfo->env, code);
×
4417
  }
4418
}
65,741✔
4419

4420
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
65,753✔
4421
  STableMergeScanInfo* pInfo = pOperator->info;
65,753✔
4422
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
65,753✔
4423
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
65,753✔
4424

4425
  stopDurationForGroupTableMergeScan(pOperator);
65,753✔
4426

4427
  if (pInfo->base.dataReader != NULL) {
65,757✔
4428
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
65,756✔
4429
    pInfo->base.dataReader = NULL;
65,755✔
4430
  }
4431
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
65,758✔
4432
    if (pInfo->nextDurationBlocks[i]) {
2!
4433
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
2✔
4434
      pInfo->nextDurationBlocks[i] = NULL;
2✔
4435
    }
4436
    pInfo->numNextDurationBlocks = 0;
2✔
4437
    pInfo->nextDurationBlocksIdx = 0;
2✔
4438
  }
4439
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
65,756✔
4440
  taosHashCleanup(pInfo->mSkipTables);
65,753✔
4441
  pInfo->mSkipTables = NULL;
65,750✔
4442
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
65,750✔
4443
}
65,754✔
4444

4445
// all data produced by this function only belongs to one group
4446
// slimit/soffset does not need to be concerned here, since this function only deal with data within one group.
4447
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
136,186✔
4448
                                              SOperatorInfo* pOperator) {
4449
  STableMergeScanInfo* pInfo = pOperator->info;
136,186✔
4450
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
136,186✔
4451
  STupleHandle*        pTupleHandle = NULL;
136,186✔
4452

4453
  blockDataCleanup(pResBlock);
136,186✔
4454

4455
  while (1) {
×
4456
    while (1) {
24,315,354✔
4457
      pTupleHandle = NULL;
24,451,542✔
4458
      int32_t code = tsortNextTuple(pHandle, &pTupleHandle);
24,451,542✔
4459
      if (code != TSDB_CODE_SUCCESS) {
24,158,930✔
4460
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
11,415!
4461
        T_LONG_JMP(pOperator->pTaskInfo->env, code);
11,415!
4462
      }
4463
      if (pTupleHandle == NULL) {
24,147,515✔
4464
        break;
116,782✔
4465
      }
4466

4467
      code = tsortAppendTupleToBlock(pInfo->pSortHandle, pResBlock, pTupleHandle);
24,030,733✔
4468
      if (code != TSDB_CODE_SUCCESS) {
24,322,713!
4469
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4470
        T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4471
      }
4472

4473
      if (pResBlock->info.rows >= capacity) {
24,334,773✔
4474
        break;
19,419✔
4475
      }
4476
    }
4477

4478
    if (tsortIsClosed(pHandle)) {
136,201!
4479
      terrno = TSDB_CODE_TSC_QUERY_CANCELLED;
×
4480
      T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4481
    }
4482

4483
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
136,215✔
4484
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
136,207✔
4485
           pInfo->limitInfo.numOfOutputRows);
4486
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
136,213!
4487
      break;
4488
    }
4489
  }
4490
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
136,213✔
4491
}
4492

4493
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
122,826✔
4494
  if (pOperator->status == OP_EXEC_DONE) {
122,826✔
4495
    (*ppRes) = NULL;
58✔
4496
    return TSDB_CODE_SUCCESS;
58✔
4497
  }
4498

4499
  int32_t              code = TSDB_CODE_SUCCESS;
122,768✔
4500
  int32_t              lino = 0;
122,768✔
4501
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
122,768✔
4502
  STableMergeScanInfo* pInfo = pOperator->info;
122,768✔
4503

4504
  code = pOperator->fpSet._openFn(pOperator);
122,768✔
4505
  QUERY_CHECK_CODE(code, lino, _end);
122,774!
4506

4507
  int64_t st = taosGetTimestampUs();
122,795✔
4508

4509
  int32_t tableListSize = 0;
122,795✔
4510
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
122,795✔
4511
  QUERY_CHECK_CODE(code, lino, _end);
122,784!
4512

4513
  if (!pInfo->hasGroupId) {
122,784✔
4514
    pInfo->hasGroupId = true;
59,561✔
4515

4516
    if (tableListSize == 0) {
59,561✔
4517
      setOperatorCompleted(pOperator);
6,584✔
4518
      (*ppRes) = NULL;
6,584✔
4519
      return code;
6,584✔
4520
    }
4521
    pInfo->tableStartIndex = 0;
52,977✔
4522
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
52,977✔
4523
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
52,975!
4524
    pInfo->groupId = tmp->groupId;
52,975✔
4525
    startGroupTableMergeScan(pOperator);
52,975✔
4526
  }
4527

4528
  SSDataBlock* pBlock = NULL;
116,199✔
4529
  while (pInfo->tableStartIndex < tableListSize) {
136,190!
4530
    if (isTaskKilled(pTaskInfo)) {
136,190!
4531
      goto _end;
×
4532
    }
4533

4534
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
136,193✔
4535
                                              pOperator);
4536
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
136,213✔
4537
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
3,597✔
4538
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
3,597!
4539
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
3,597✔
4540
    }
4541
    if (pBlock != NULL) {
136,214✔
4542
      pBlock->info.id.groupId = pInfo->groupId;
63,237✔
4543
      pOperator->resultInfo.totalRows += pBlock->info.rows;
63,237✔
4544
      pInfo->bGroupProcessed = true;
63,237✔
4545
      break;
63,237✔
4546
    } else {
4547
      if (pInfo->bNewFilesetEvent) {
72,977✔
4548
        stopDurationForGroupTableMergeScan(pOperator);
7,223✔
4549
        code = startDurationForGroupTableMergeScan(pOperator);
7,222✔
4550
        QUERY_CHECK_CODE(code, lino, _end);
7,223!
4551
      } else {
4552
        // Data of this group are all dumped, let's try the next group
4553
        stopGroupTableMergeScan(pOperator);
65,754✔
4554
        if (pInfo->tableEndIndex >= tableListSize - 1) {
65,754✔
4555
          setOperatorCompleted(pOperator);
52,983✔
4556
          break;
52,986✔
4557
        }
4558

4559
        pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
12,771✔
4560
        STableKeyInfo* tmp = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
12,771✔
4561
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
12,771!
4562
        pInfo->groupId = tmp->groupId;
12,771✔
4563
        startGroupTableMergeScan(pOperator);
12,771✔
4564
        resetLimitInfoForNextGroup(&pInfo->limitInfo);
12,771✔
4565
      }
4566
    }
4567
  }
4568

4569
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
116,220✔
4570

4571
_end:
116,220✔
4572
  if (code != TSDB_CODE_SUCCESS) {
116,220!
4573
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4574
    pTaskInfo->code = code;
×
4575
    T_LONG_JMP(pTaskInfo->env, code);
×
4576
  } else {
4577
    (*ppRes) = pBlock;
116,220✔
4578
  }
4579

4580
  return code;
116,220✔
4581
}
4582

4583
static SSDataBlock* doTableMergeScan(SOperatorInfo* pOperator) {
×
4584
  SSDataBlock* pRes = NULL;
×
4585
  int32_t      code = doTableMergeScanNext(pOperator, &pRes);
×
4586
  return pRes;
×
4587
}
4588

4589
void destroyTableMergeScanOperatorInfo(void* param) {
52,748✔
4590
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
52,748✔
4591

4592
  // start one reader variable
4593
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
52,748!
4594
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
52,749✔
4595
    pTableScanInfo->base.dataReader = NULL;
52,748✔
4596
  }
4597

4598
  for (int32_t i = 0; i < pTableScanInfo->numNextDurationBlocks; ++i) {
52,747!
4599
    if (pTableScanInfo->nextDurationBlocks[i] != NULL) {
×
4600
      blockDataDestroy(pTableScanInfo->nextDurationBlocks[i]);
×
4601
      pTableScanInfo->nextDurationBlocks[i] = NULL;
×
4602
    }
4603
  }
4604

4605
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
52,747✔
4606
  pTableScanInfo->pSortHandle = NULL;
52,748✔
4607
  taosHashCleanup(pTableScanInfo->mSkipTables);
52,748✔
4608
  pTableScanInfo->mSkipTables = NULL;
52,747✔
4609
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
52,747✔
4610
  pTableScanInfo->pSortInputBlock = NULL;
52,750✔
4611
  // end one reader variable
4612

4613
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
52,750✔
4614

4615
  blockDataDestroy(pTableScanInfo->pResBlock);
52,733✔
4616
  pTableScanInfo->pResBlock = NULL;
52,748✔
4617

4618
  // remove it from the task->result list
4619
  blockDataDestroy(pTableScanInfo->pReaderBlock);
52,748✔
4620
  pTableScanInfo->pReaderBlock = NULL;
52,743✔
4621
  taosArrayDestroy(pTableScanInfo->pSortInfo);
52,743✔
4622

4623
  stopSubTablesTableMergeScan(pTableScanInfo);
52,745✔
4624

4625
  taosMemoryFreeClear(param);
52,737!
4626
}
52,748✔
4627

4628
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
3,210✔
4629
  if (pOptr == NULL) {
3,210!
4630
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
4631
    return TSDB_CODE_INVALID_PARA;
×
4632
  }
4633
  // TODO: merge these two info into one struct
4634
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
3,210!
4635
  if (!execInfo) {
3,210!
4636
    return terrno;
×
4637
  }
4638
  STableMergeScanInfo* pInfo = pOptr->info;
3,210✔
4639
  execInfo->blockRecorder = pInfo->base.readRecorder;
3,210✔
4640
  execInfo->sortExecInfo = pInfo->sortExecInfo;
3,210✔
4641

4642
  *pOptrExplain = execInfo;
3,210✔
4643
  *len = sizeof(STableMergeScanExecInfo);
3,210✔
4644

4645
  return TSDB_CODE_SUCCESS;
3,210✔
4646
}
4647

4648
static int32_t resetTableMergeScanOperatorState(SOperatorInfo* pOper) {
12,150✔
4649
  int32_t         code = TSDB_CODE_SUCCESS;
12,150✔
4650
  STableMergeScanInfo*   pInfo = pOper->info;
12,150✔
4651
  pOper->status = OP_NOT_OPENED;
12,150✔
4652

4653
  pInfo->tableEndIndex = 0;
12,150✔
4654
  pInfo->tableStartIndex = 0;
12,150✔
4655
  pInfo->hasGroupId = false;
12,150✔
4656
  if (pInfo->base.readerAPI.tsdReaderClose) {
12,150!
4657
    pInfo->base.readerAPI.tsdReaderClose(pInfo->base.dataReader);
12,150✔
4658
  }
4659
  pInfo->base.dataReader = NULL;
12,150✔
4660

4661
  pInfo->base.limitInfo = (SLimitInfo){0};
12,150✔
4662
  pInfo->base.limitInfo.limit.limit = -1;
12,150✔
4663
  pInfo->base.limitInfo.slimit.limit = -1;
12,150✔
4664

4665
  tableListDestroy(pInfo->base.pTableListInfo);
12,150✔
4666
  pInfo->base.pTableListInfo = tableListCreate();
12,150✔
4667
  if (!pInfo->base.pTableListInfo) {
12,149!
4668
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4669
    return terrno;
×
4670
  }
4671
  SExecTaskInfo*         pTaskInfo = pOper->pTaskInfo;
12,149✔
4672

4673
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
12,149✔
4674
  code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
12,149✔
4675
                                  &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
4676
                                  pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
12,149✔
4677
  if (code) {
12,150!
4678
    qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
×
4679
    return code;
×
4680
  }
4681

4682
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
12,150✔
4683
  cleanupQueryTableDataCond(&pInfo->base.cond);
12,150✔
4684
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle);
12,150✔
4685
  if (code) {
12,150!
4686
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
4687
    return code;
×
4688
  }
4689

4690
  tsortDestroySortHandle(pInfo->pSortHandle);
12,150✔
4691
  pInfo->pSortHandle = NULL;
12,150✔
4692

4693
  taosHashCleanup(pInfo->mSkipTables);
12,150✔
4694
  pInfo->mSkipTables = NULL;
12,150✔
4695
  pOper->resultInfo.totalRows = 0;
12,150✔
4696
  
4697
  pInfo->bGroupProcessed = false;
12,150✔
4698
  pInfo->bNewFilesetEvent = false;
12,150✔
4699
  pInfo->bNextDurationBlockEvent = false;
12,150✔
4700
  pInfo->rtnNextDurationBlocks = false;
12,150✔
4701
  pInfo->nextDurationBlocksIdx = 0;
12,150✔
4702
  pInfo->bSortRowId = false;
12,150✔
4703

4704
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
12,150!
4705
    if (pInfo->nextDurationBlocks[i] != NULL) {
×
4706
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
×
4707
      pInfo->nextDurationBlocks[i] = NULL;
×
4708
    }
4709
  }
4710
  pInfo->numNextDurationBlocks = 0;
12,150✔
4711

4712
  stopSubTablesTableMergeScan(pInfo);
12,150✔
4713
  return code;
12,150✔
4714
}
4715

4716
int32_t createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
52,656✔
4717
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
4718
                                         SOperatorInfo** pOptrInfo) {
4719
  QRY_PARAM_CHECK(pOptrInfo);
52,656!
4720

4721
  int32_t              code = TSDB_CODE_SUCCESS;
52,656✔
4722
  int32_t              lino = 0;
52,656✔
4723
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
52,656!
4724
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
52,697!
4725
  if (pInfo == NULL || pOperator == NULL) {
52,701!
4726
    code = terrno;
×
4727
    goto _error;
×
4728
  }
4729

4730
  pOperator->pPhyNode = pTableScanNode;
52,705✔
4731
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
52,705✔
4732

4733
  int32_t numOfCols = 0;
52,705✔
4734
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
52,705✔
4735
                             &pInfo->base.matchInfo);
4736
  QUERY_CHECK_CODE(code, lino, _error);
52,700!
4737

4738
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
52,700✔
4739
  QUERY_CHECK_CODE(code, lino, _error);
52,711!
4740

4741
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
52,711✔
4742
    SExprSupp* pSup = &pInfo->base.pseudoSup;
22,417✔
4743
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
22,417✔
4744
    QUERY_CHECK_CODE(code, lino, _error);
22,402!
4745

4746
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
22,402✔
4747
                                      &pTaskInfo->storageAPI.functionStore);
4748
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
22,377!
4749
  }
4750

4751
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
52,671✔
4752

4753
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
52,671✔
4754
  QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
52,719!
4755

4756
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
52,719✔
4757
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
52,719✔
4758
  pInfo->base.scanFlag = MAIN_SCAN;
52,719✔
4759
  pInfo->base.readHandle = *readHandle;
52,719✔
4760

4761
  pInfo->readIdx = -1;
52,719✔
4762

4763
  pInfo->base.limitInfo.limit.limit = -1;
52,719✔
4764
  pInfo->base.limitInfo.slimit.limit = -1;
52,719✔
4765
  pInfo->base.pTableListInfo = pTableListInfo;
52,719✔
4766

4767
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
52,719✔
4768
  pInfo->sample.seed = taosGetTimestampSec();
52,719✔
4769

4770
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
52,731✔
4771
                            pTaskInfo->pStreamRuntimeInfo);
52,731✔
4772
  QUERY_CHECK_CODE(code, lino, _error);
52,710!
4773

4774
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
52,710✔
4775

4776
  pInfo->mergeLimit = -1;
52,706✔
4777
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
52,706!
4778
  if (hasLimit) {
52,706✔
4779
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1
8,482✔
4780
                            ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset
1,750✔
4781
                            : pInfo->limitInfo.limit.limit;
4,241✔
4782
    pInfo->mSkipTables = NULL;
4,241✔
4783
  }
4784

4785
  initResultSizeInfo(&pOperator->resultInfo, 1024);
52,706✔
4786
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
52,685✔
4787
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
52,698!
4788
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
52,698✔
4789
  QUERY_CHECK_CODE(code, lino, _error);
52,727!
4790
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
52,727!
4791
    pInfo->bSortRowId = true;
10,108✔
4792
  } else {
4793
    pInfo->bSortRowId = false;
42,604✔
4794
  }
4795

4796
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
52,712✔
4797
  QUERY_CHECK_CODE(code, lino, _error);
52,713!
4798

4799
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
52,713✔
4800
  QUERY_CHECK_CODE(code, lino, _error);
52,670!
4801

4802
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
52,670✔
4803
  QUERY_CHECK_CODE(code, lino, _error);
52,704!
4804

4805
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
52,704✔
4806

4807
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
52,704✔
4808
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
52,704✔
4809

4810
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
52,700✔
4811

4812
  // start one reader variable
4813
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
52,713✔
4814
  QUERY_CHECK_CODE(code, lino, _error);
52,721!
4815

4816
  if (!tsExperimental) {
52,721!
4817
    pInfo->filesetDelimited = false;
×
4818
  } else {
4819
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
52,721✔
4820
  }
4821
  // end one reader variable
4822
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
52,721✔
4823
                  pInfo, pTaskInfo);
4824
  pOperator->exprSupp.numOfExprs = numOfCols;
52,692✔
4825

4826
  pOperator->fpSet = createOperatorFpSet(
52,692✔
4827
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
52,692✔
4828
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
4829
      NULL);
4830
  setOperatorResetStateFn(pOperator, resetTableMergeScanOperatorState);     
52,684✔
4831
  pOperator->cost.openCost = 0;
52,675✔
4832

4833
  *pOptrInfo = pOperator;
52,675✔
4834
  return code;
52,675✔
4835

4836
_error:
×
4837
  if (code != TSDB_CODE_SUCCESS) {
×
4838
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4839
  }
4840
  pTaskInfo->code = code;
×
4841
  if (pInfo != NULL) {
×
4842
    pInfo->base.pTableListInfo = NULL;
×
4843
    destroyTableMergeScanOperatorInfo(pInfo);
×
4844
  }
4845
  if (pOperator != NULL) {
×
4846
    pOperator->info = NULL;
×
4847
    destroyOperator(pOperator);
×
4848
  }
4849
  return code;
×
4850
}
4851

4852
// ====================================================================================================================
4853
// TableCountScanOperator
4854
static void        destoryTableCountScanOperator(void* param);
4855
static int32_t     buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
4856
                                                  SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI);
4857
static int32_t     buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
4858
                                                  SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI);
4859
static int32_t     buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4860
                                             STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName);
4861
static int32_t     buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4862
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName);
4863
static int32_t     buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4864
                                          STableCountScanSupp* pSupp, SSDataBlock* pRes);
4865
static void        buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4866
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
4867
                                               size_t perfdbTableNum);
4868
static void        buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
4869
                                              size_t infodbTableNum, size_t perfdbTableNum);
4870
static const char* GROUP_TAG_DB_NAME = "db_name";
4871
static const char* GROUP_TAG_STABLE_NAME = "stable_name";
4872

4873
int32_t tblCountScanGetGroupTagsSlotId(const SNodeList* scanCols, STableCountScanSupp* supp) {
×
4874
  if (scanCols != NULL) {
×
4875
    SNode* pNode = NULL;
×
4876
    FOREACH(pNode, scanCols) {
×
4877
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
4878
        return TSDB_CODE_QRY_SYS_ERROR;
×
4879
      }
4880
      STargetNode* targetNode = (STargetNode*)pNode;
×
4881
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
×
4882
        return TSDB_CODE_QRY_SYS_ERROR;
×
4883
      }
4884
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
×
4885
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
4886
        supp->dbNameSlotId = targetNode->slotId;
×
4887
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
4888
        supp->stbNameSlotId = targetNode->slotId;
×
4889
      }
4890
    }
4891
  }
4892
  return TSDB_CODE_SUCCESS;
×
4893
}
4894

4895
int32_t tblCountScanGetCountSlotId(const SNodeList* pseudoCols, STableCountScanSupp* supp) {
×
4896
  if (pseudoCols != NULL) {
×
4897
    SNode* pNode = NULL;
×
4898
    FOREACH(pNode, pseudoCols) {
×
4899
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
4900
        return TSDB_CODE_QRY_SYS_ERROR;
×
4901
      }
4902
      STargetNode* targetNode = (STargetNode*)pNode;
×
4903
      if (nodeType(targetNode->pExpr) != QUERY_NODE_FUNCTION) {
×
4904
        return TSDB_CODE_QRY_SYS_ERROR;
×
4905
      }
4906
      SFunctionNode* funcNode = (SFunctionNode*)(targetNode->pExpr);
×
4907
      if (funcNode->funcType == FUNCTION_TYPE_TABLE_COUNT) {
×
4908
        supp->tbCountSlotId = targetNode->slotId;
×
4909
      }
4910
    }
4911
  }
4912
  return TSDB_CODE_SUCCESS;
×
4913
}
4914

4915
int32_t tblCountScanGetInputs(SNodeList* groupTags, SName* tableName, STableCountScanSupp* supp) {
×
4916
  if (groupTags != NULL) {
×
4917
    SNode* pNode = NULL;
×
4918
    FOREACH(pNode, groupTags) {
×
4919
      if (nodeType(pNode) != QUERY_NODE_COLUMN) {
×
4920
        return TSDB_CODE_QRY_SYS_ERROR;
×
4921
      }
4922
      SColumnNode* colNode = (SColumnNode*)pNode;
×
4923
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
4924
        supp->groupByDbName = true;
×
4925
      }
4926
      if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
4927
        supp->groupByStbName = true;
×
4928
      }
4929
    }
4930
  } else {
4931
    tstrncpy(supp->dbNameFilter, tNameGetDbNameP(tableName), TSDB_DB_NAME_LEN);
×
4932
    tstrncpy(supp->stbNameFilter, tNameGetTableName(tableName), TSDB_TABLE_NAME_LEN);
×
4933
  }
4934
  return TSDB_CODE_SUCCESS;
×
4935
}
4936

4937
int32_t getTableCountScanSupp(SNodeList* groupTags, SName* tableName, SNodeList* scanCols, SNodeList* pseudoCols,
×
4938
                              STableCountScanSupp* supp, SExecTaskInfo* taskInfo) {
4939
  int32_t code = 0;
×
4940
  code = tblCountScanGetInputs(groupTags, tableName, supp);
×
4941
  if (code != TSDB_CODE_SUCCESS) {
×
4942
    qError("%s get table count scan supp. get inputs error", GET_TASKID(taskInfo));
×
4943
    return code;
×
4944
  }
4945

4946
  supp->dbNameSlotId = -1;
×
4947
  supp->stbNameSlotId = -1;
×
4948
  supp->tbCountSlotId = -1;
×
4949

4950
  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
×
4951
  if (code != TSDB_CODE_SUCCESS) {
×
4952
    qError("%s get table count scan supp. get group tags slot id error", GET_TASKID(taskInfo));
×
4953
    return code;
×
4954
  }
4955

4956
  code = tblCountScanGetCountSlotId(pseudoCols, supp);
×
4957
  if (code != TSDB_CODE_SUCCESS) {
×
4958
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
×
4959
    return code;
×
4960
  }
4961
  return code;
×
4962
}
4963

4964
int32_t createTableCountScanOperatorInfo(SReadHandle* readHandle, STableCountScanPhysiNode* pTblCountScanNode,
×
4965
                                         SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
4966
  QRY_PARAM_CHECK(pOptrInfo);
×
4967

4968
  int32_t                      code = TSDB_CODE_SUCCESS;
×
4969
  int32_t                      lino = 0;
×
4970
  SScanPhysiNode*              pScanNode = &pTblCountScanNode->scan;
×
4971
  STableCountScanOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableCountScanOperatorInfo));
×
4972
  SOperatorInfo*               pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
4973
  if (!pInfo || !pOperator) {
×
4974
    code = terrno;
×
4975
    goto _error;
×
4976
  }
4977

4978
  pInfo->readHandle = *readHandle;
×
4979

4980
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
×
4981
  initResultSizeInfo(&pOperator->resultInfo, 1);
×
4982
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
×
4983
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
×
4984

4985
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
×
4986
  QUERY_CHECK_CODE(code, lino, _error);
×
4987

4988
  code = getTableCountScanSupp(pTblCountScanNode->pGroupTags, &pTblCountScanNode->scan.tableName,
×
4989
                               pTblCountScanNode->scan.pScanCols, pTblCountScanNode->scan.pScanPseudoCols, &pInfo->supp,
4990
                               pTaskInfo);
4991
  QUERY_CHECK_CODE(code, lino, _error);
×
4992

4993
  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
×
4994
                  pInfo, pTaskInfo);
4995
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScanNext, NULL, destoryTableCountScanOperator,
×
4996
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
4997
  *pOptrInfo = pOperator;
×
4998
  return code;
×
4999

5000
_error:
×
5001
  if (pInfo != NULL) {
×
5002
    destoryTableCountScanOperator(pInfo);
×
5003
  }
5004
  if (pOperator != NULL) {
×
5005
    pOperator->info = NULL;
×
5006
    destroyOperator(pOperator);
×
5007
  }
5008
  pTaskInfo->code = code;
×
5009
  return code;
×
5010
}
5011

5012
int32_t fillTableCountScanDataBlock(STableCountScanSupp* pSupp, char* dbName, char* stbName, int64_t count,
×
5013
                                    SSDataBlock* pRes) {
5014
  int32_t code = TSDB_CODE_SUCCESS;
×
5015
  int32_t lino = 0;
×
5016
  if (pSupp->dbNameSlotId != -1) {
×
5017
    QUERY_CHECK_CONDITION((strlen(dbName) > 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
5018
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
×
5019
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
5020

5021
    char varDbName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5022
    tstrncpy(varDataVal(varDbName), dbName, TSDB_DB_NAME_LEN);
×
5023

5024
    varDataSetLen(varDbName, strlen(dbName));
×
5025
    code = colDataSetVal(colInfoData, 0, varDbName, false);
×
5026
    QUERY_CHECK_CODE(code, lino, _end);
×
5027
  }
5028

5029
  if (pSupp->stbNameSlotId != -1) {
×
5030
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
×
5031
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
5032
    if (strlen(stbName) != 0) {
×
5033
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5034
      tstrncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
×
5035
      varDataSetLen(varStbName, strlen(stbName));
×
5036
      code = colDataSetVal(colInfoData, 0, varStbName, false);
×
5037
      QUERY_CHECK_CODE(code, lino, _end);
×
5038
    } else {
5039
      colDataSetNULL(colInfoData, 0);
5040
    }
5041
  }
5042

5043
  if (pSupp->tbCountSlotId != -1) {
×
5044
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
×
5045
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
5046
    code = colDataSetVal(colInfoData, 0, (char*)&count, false);
×
5047
    QUERY_CHECK_CODE(code, lino, _end);
×
5048
  }
5049
  pRes->info.rows = 1;
×
5050

5051
_end:
×
5052
  if (code != TSDB_CODE_SUCCESS) {
×
5053
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5054
  }
5055
  return code;
×
5056
}
5057

5058
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
×
5059
  STableCountScanSupp* pSupp = &pInfo->supp;
×
5060
  SSDataBlock*         pRes = pInfo->pRes;
×
5061

5062
  size_t infodbTableNum;
5063
  getInfosDbMeta(NULL, &infodbTableNum);
×
5064
  infodbTableNum -= 1;
×
5065
  size_t perfdbTableNum;
5066
  getPerfDbMeta(NULL, &perfdbTableNum);
×
5067

5068
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
5069
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
5070
    return (pRes->info.rows > 0) ? pRes : NULL;
×
5071
  } else {
5072
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
5073
    return (pRes->info.rows > 0) ? pRes : NULL;
×
5074
  }
5075
}
5076

5077
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
×
5078
                                       size_t infodbTableNum, size_t perfdbTableNum) {
5079
  int32_t        code = TSDB_CODE_SUCCESS;
×
5080
  int32_t        lino = 0;
×
5081
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5082
  if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
×
5083
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
5084
    QUERY_CHECK_CODE(code, lino, _end);
×
5085
  } else if (strcmp(pSupp->dbNameFilter, TSDB_PERFORMANCE_SCHEMA_DB) == 0) {
×
5086
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
5087
    QUERY_CHECK_CODE(code, lino, _end);
×
5088
  } else if (strlen(pSupp->dbNameFilter) == 0) {
×
5089
    code = fillTableCountScanDataBlock(pSupp, "", "", infodbTableNum + perfdbTableNum, pRes);
×
5090
    QUERY_CHECK_CODE(code, lino, _end);
×
5091
  }
5092

5093
_end:
×
5094
  if (code != TSDB_CODE_SUCCESS) {
×
5095
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5096
    pTaskInfo->code = code;
×
5097
    T_LONG_JMP(pTaskInfo->env, code);
×
5098
  }
5099
  setOperatorCompleted(pOperator);
×
5100
}
×
5101

5102
static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5103
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
5104
                                        size_t perfdbTableNum) {
5105
  int32_t        code = TSDB_CODE_SUCCESS;
×
5106
  int32_t        lino = 0;
×
5107
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5108
  if (pInfo->currGrpIdx == 0) {
×
5109
    uint64_t groupId = 0;
×
5110
    if (pSupp->groupByDbName) {
×
5111
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
×
5112
    } else {
5113
      groupId = calcGroupId("", 0);
×
5114
    }
5115

5116
    pRes->info.id.groupId = groupId;
×
5117
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
5118
    QUERY_CHECK_CODE(code, lino, _end);
×
5119
  } else if (pInfo->currGrpIdx == 1) {
×
5120
    uint64_t groupId = 0;
×
5121
    if (pSupp->groupByDbName) {
×
5122
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
×
5123
    } else {
5124
      groupId = calcGroupId("", 0);
×
5125
    }
5126

5127
    pRes->info.id.groupId = groupId;
×
5128
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
5129
    QUERY_CHECK_CODE(code, lino, _end);
×
5130
  } else {
5131
    setOperatorCompleted(pOperator);
×
5132
  }
5133
  pInfo->currGrpIdx++;
×
5134

5135
_end:
×
5136
  if (code != TSDB_CODE_SUCCESS) {
×
5137
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5138
    pTaskInfo->code = code;
×
5139
    T_LONG_JMP(pTaskInfo->env, code);
×
5140
  }
5141
}
×
5142

5143
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
×
5144
  int32_t                      code = TSDB_CODE_SUCCESS;
×
5145
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
×
5146
  STableCountScanOperatorInfo* pInfo = pOperator->info;
×
5147
  STableCountScanSupp*         pSupp = &pInfo->supp;
×
5148
  SSDataBlock*                 pRes = pInfo->pRes;
×
5149

5150
  blockDataCleanup(pRes);
×
5151
  QRY_PARAM_CHECK(ppRes);
×
5152

5153
  if (pOperator->status == OP_EXEC_DONE) {
×
5154
    return code;
×
5155
  }
5156

5157
  if (pInfo->readHandle.mnd != NULL) {
×
5158
    (*ppRes) = buildSysDbTableCount(pOperator, pInfo);
×
5159
    return code;
×
5160
  }
5161

5162
  code = buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
×
5163
  if (code != TSDB_CODE_SUCCESS) {
×
5164
    qError("%s failed since %s", __func__, tstrerror(code));
×
5165
    pTaskInfo->code = code;
×
5166
    T_LONG_JMP(pTaskInfo->env, code);
×
5167
  }
5168
  if (pRes->info.rows > 0) {
×
5169
    *ppRes = pRes;
×
5170
  }
5171

5172
  return code;
×
5173
}
5174

5175
static int32_t buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5176
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes) {
5177
  int32_t        code = TSDB_CODE_SUCCESS;
×
5178
  int32_t        lino = 0;
×
5179
  const char*    db = NULL;
×
5180
  int32_t        vgId = 0;
×
5181
  char           dbName[TSDB_DB_NAME_LEN] = {0};
×
5182
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5183
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
5184

5185
  // get dbname
5186
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, &db, &vgId, NULL, NULL);
×
5187
  SName sn = {0};
×
5188

5189
  code = tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
×
5190
  QUERY_CHECK_CODE(code, lino, _end);
×
5191

5192
  code = tNameGetDbName(&sn, dbName);
×
5193
  QUERY_CHECK_CODE(code, lino, _end);
×
5194

5195
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
5196
    code = buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
×
5197
  } else {
5198
    code = buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
×
5199
  }
5200

5201
_end:
×
5202
  if (code != TSDB_CODE_SUCCESS) {
×
5203
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5204
    pTaskInfo->code = code;
×
5205
  }
5206

5207
  return code;
×
5208
}
5209

5210
static int32_t buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5211
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
5212
  int32_t        code = TSDB_CODE_SUCCESS;
×
5213
  int32_t        lino = 0;
×
5214
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5215
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
5216

5217
  if (pSupp->groupByStbName) {
×
5218
    if (pInfo->stbUidList == NULL) {
×
5219
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
×
5220
      QUERY_CHECK_NULL(pInfo->stbUidList, code, lino, _end, terrno);
×
5221
      code = pAPI->metaFn.storeGetTableList(pInfo->readHandle.vnode, TSDB_SUPER_TABLE, pInfo->stbUidList);
×
5222
      QUERY_CHECK_CODE(code, lino, _end);
×
5223
    }
5224

5225
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
×
5226
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
×
5227
      code = buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid, pAPI);
×
5228
      QUERY_CHECK_CODE(code, lino, _end);
×
5229

5230
      pInfo->currGrpIdx++;
×
5231
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
×
5232
      code = buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName, pAPI);
×
5233
      QUERY_CHECK_CODE(code, lino, _end);
×
5234

5235
      pInfo->currGrpIdx++;
×
5236
    } else {
5237
      setOperatorCompleted(pOperator);
×
5238
    }
5239
  } else {
5240
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
×
5241
    pRes->info.id.groupId = groupId;
×
5242

5243
    int64_t dbTableCount = 0;
×
5244
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &dbTableCount, NULL);
×
5245
    code = fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
×
5246
    QUERY_CHECK_CODE(code, lino, _end);
×
5247
    setOperatorCompleted(pOperator);
×
5248
  }
5249

5250
_end:
×
5251
  if (code != TSDB_CODE_SUCCESS) {
×
5252
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5253
    pTaskInfo->code = code;
×
5254
  }
5255
  return code;
×
5256
}
5257

5258
static int32_t buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5259
                                         STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
5260
  int32_t        code = TSDB_CODE_SUCCESS;
×
5261
  int32_t        lino = 0;
×
5262
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5263
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
5264

5265
  if (strlen(pSupp->dbNameFilter) != 0) {
×
5266
    if (strlen(pSupp->stbNameFilter) != 0) {
×
5267
      uint64_t uid = 0;
×
5268
      code = pAPI->metaFn.getTableUidByName(pInfo->readHandle.vnode, pSupp->stbNameFilter, &uid);
×
5269
      QUERY_CHECK_CODE(code, lino, _end);
×
5270

5271
      int64_t numOfChildTables = 0;
×
5272
      code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, uid, &numOfChildTables, NULL, NULL);
×
5273
      QUERY_CHECK_CODE(code, lino, _end);
×
5274

5275
      code = fillTableCountScanDataBlock(pSupp, dbName, pSupp->stbNameFilter, numOfChildTables, pRes);
×
5276
      QUERY_CHECK_CODE(code, lino, _end);
×
5277
    } else {
5278
      int64_t tbNumVnode = 0;
×
5279
      pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
5280
      code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
5281
      QUERY_CHECK_CODE(code, lino, _end);
×
5282
    }
5283
  } else {
5284
    int64_t tbNumVnode = 0;
×
5285
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
5286
    code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
5287
    QUERY_CHECK_CODE(code, lino, _end);
×
5288
  }
5289

5290
_end:
×
5291
  if (code != TSDB_CODE_SUCCESS) {
×
5292
    pTaskInfo->code = code;
×
5293
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5294
  }
5295

5296
  setOperatorCompleted(pOperator);
×
5297
  return code;
×
5298
}
5299

5300
static int32_t buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
5301
                                              SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI) {
5302
  int32_t code = TSDB_CODE_SUCCESS;
×
5303
  int32_t lino = 0;
×
5304
  char    fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
5305
  if (pSupp->groupByDbName) {
×
5306
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
×
5307
  }
5308

5309
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
5310
  pRes->info.id.groupId = groupId;
×
5311

5312
  int64_t numOfTables = 0;
×
5313
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, NULL, &numOfTables);
×
5314

5315
  if (numOfTables != 0) {
×
5316
    code = fillTableCountScanDataBlock(pSupp, dbName, "", numOfTables, pRes);
×
5317
  }
5318

5319
_end:
×
5320
  if (code != TSDB_CODE_SUCCESS) {
×
5321
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5322
  }
5323
  return code;
×
5324
}
5325

5326
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
5327
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
5328
  int32_t code = TSDB_CODE_SUCCESS;
×
5329
  int32_t lino = 0;
×
5330
  char    stbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5331
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
×
5332
  QUERY_CHECK_CODE(code, lino, _end);
×
5333

5334
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
5335
  if (pSupp->groupByDbName) {
×
5336
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
×
5337
  } else {
5338
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
×
5339
  }
5340

5341
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
5342
  pRes->info.id.groupId = groupId;
×
5343

5344
  int64_t ctbNum = 0;
×
5345
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL, NULL);
×
5346
  QUERY_CHECK_CODE(code, lino, _end);
×
5347
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
×
5348
  QUERY_CHECK_CODE(code, lino, _end);
×
5349

5350
_end:
×
5351
  if (code != TSDB_CODE_SUCCESS) {
×
5352
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5353
  }
5354
  return code;
×
5355
}
5356

5357
static void destoryTableCountScanOperator(void* param) {
×
5358
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
×
5359
  blockDataDestroy(pTableCountScanInfo->pRes);
×
5360

5361
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
×
5362
  taosMemoryFreeClear(param);
×
5363
}
×
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