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

taosdata / TDengine / #4923

09 Jan 2026 08:13AM UTC coverage: 65.373% (+0.2%) from 65.161%
#4923

push

travis-ci

web-flow
merge: from main to 3.0 branch #34232

33 of 56 new or added lines in 8 files covered. (58.93%)

3042 existing lines in 131 files now uncovered.

198273 of 303297 relevant lines covered (65.37%)

118980690.73 hits per line

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

77.94
/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) {
559,944,061✔
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;
559,944,061✔
86
#endif
87
}
88

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

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

99
  // 0 by default, which means it is not a interval operator of the upstream operator.
100
  if (pInterval->interval == 0) {
599,290,231✔
101
    *overlap = false;
558,055,880✔
102
    return code;
558,063,705✔
103
  }
104

105
  if (order == TSDB_ORDER_ASC) {
41,196,559✔
106
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
39,248,420✔
107
    if (w.ekey < pBlockInfo->window.skey) {
39,249,010✔
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) {
39,243,591✔
113
      *overlap = true;
32,571,987✔
114
      return code;
32,573,653✔
115
    }
116

117
    while (1) {
118
      getNextTimeWindow(pInterval, &w, order);
6,671,326✔
119
      if (w.skey > pBlockInfo->window.ekey) {
6,678,453✔
120
        break;
6,415,470✔
121
      }
122

123
      if (w.ekey <= pBlockInfo->window.ekey) {
261,621✔
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) {
261,621✔
128
        *overlap = true;
261,621✔
129
        return code;
261,621✔
130
      }
131
    }
132
  } else {
133
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.ekey);
1,948,139✔
134
    if (w.skey > pBlockInfo->window.ekey) {
1,950,267✔
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) {
1,948,165✔
140
      *overlap = true;
1,513,519✔
141
      return code;
1,513,519✔
142
    }
143

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

150
      if (w.skey >= pBlockInfo->window.skey) {
2,508✔
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)) {
2,508✔
155
        *overlap = true;
2,508✔
156
        return code;
2,508✔
157
      }
158
    }
159
  }
160

161
  *overlap = false;
6,848,196✔
162
  return code;
6,847,742✔
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,834,655✔
167
  if (pOperator->operatorType != QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
147,834,655✔
168
    return NULL;
×
169
  }
170

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

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

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

179
  if (p1 == NULL) {
147,881,083✔
180
    return NULL;
48,020,918✔
181
  }
182

183
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
99,860,165✔
184
  if (NULL == *pPage) {
99,847,655✔
185
    return NULL;
×
186
  }
187

188
  return (SResultRow*)((char*)(*pPage) + p1->offset);
99,850,091✔
189
}
190

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

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

210
static int32_t doDynamicPruneDataBlock(SOperatorInfo* pOperator, SDataBlockInfo* pBlockInfo, uint32_t* status) {
666,056,222✔
211
  STableScanInfo* pTableScanInfo = pOperator->info;
666,056,222✔
212
  int32_t         code = TSDB_CODE_SUCCESS;
666,345,095✔
213

214
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
666,345,095✔
215
    return TSDB_CODE_SUCCESS;
518,463,728✔
216
  }
217

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

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

223
  if (pRow == NULL) {
147,862,118✔
224
    return TSDB_CODE_SUCCESS;
48,013,209✔
225
  }
226

227
  bool notLoadBlock = true;
99,848,909✔
228
  for (int32_t i = 0; i < pSup1->numOfExprs; ++i) {
2,147,483,647✔
229
    int32_t functionId = pSup1->pCtx[i].functionId;
2,147,483,647✔
230

231
    SResultRowEntryInfo* pEntry = getResultEntryInfo(pRow, i, pTableScanInfo->base.pdInfo.pExprSup->rowEntryInfoOffset);
2,147,483,647✔
232

233
    EFuncDataRequired reqStatus = fmFuncDynDataRequired(functionId, pEntry, pBlockInfo);
2,147,483,647✔
234
    if (reqStatus != FUNC_DATA_REQUIRED_NOT_LOAD && !pSup1->pCtx[i].skipDynDataCheck) {
2,147,483,647✔
235
      notLoadBlock = false;
90,848,401✔
236
      break;
90,848,401✔
237
    }
238
  }
239

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

243
  if (notLoadBlock) {
99,845,535✔
244
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
9,005,821✔
245
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
9,005,821✔
246
  }
247

248
  return code;
99,810,415✔
249
}
250

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

258
  return filterRangeExecute(pFilterInfo, pColsAgg, numOfCols, numOfRows, keep);
10,740,725✔
259
}
260

261
static int32_t doLoadBlockSMA(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
150,003,653✔
262
                              bool* pLoad) {
263
  SStorageAPI* pAPI = &pTaskInfo->storageAPI;
150,003,653✔
264
  bool         allColumnsHaveAgg = true;
150,045,991✔
265
  bool         hasNullSMA = false;
150,052,973✔
266
  if (pLoad != NULL) {
150,045,368✔
267
    *pLoad = false;
150,050,605✔
268
  }
269

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

276
  if (!allColumnsHaveAgg || hasNullSMA) {
149,997,105✔
277
    *pLoad = false;
134,107,067✔
278
  } else {
279
    *pLoad = true;
15,890,038✔
280
  }
281

282
  return code;
149,985,585✔
283
}
284

285
static int32_t doSetTagColumnData(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
726,599,144✔
286
                                  int32_t rows) {
287
  int32_t    code = 0;
726,599,144✔
288
  SExprSupp* pSup = &pTableScanInfo->pseudoSup;
726,599,144✔
289
  if (pSup->numOfExprs > 0) {
726,945,535✔
290
    code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
326,700,720✔
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) {
326,700,817✔
294
      if (pTaskInfo->streamInfo.pState) blockDataCleanup(pBlock);
×
295
      code = 0;
×
296
    }
297
  }
298

299
  return code;
726,889,566✔
300
}
301

302
bool applyLimitOffset(SLimitInfo* pLimitInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
1,188,823,276✔
303
  SLimit*     pLimit = &pLimitInfo->limit;
1,188,823,276✔
304
  const char* id = GET_TASKID(pTaskInfo);
1,188,945,160✔
305

306
  if (pLimitInfo->remainOffset > 0) {
1,189,020,461✔
307
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
5,038,868✔
308
      pLimitInfo->remainOffset -= pBlock->info.rows;
2,507,958✔
309
      blockDataEmpty(pBlock);
2,507,958✔
310
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
2,507,958✔
311
      return false;
2,507,958✔
312
    } else {
313
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
2,530,910✔
314
      if (code != TSDB_CODE_SUCCESS) {
2,530,910✔
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;
2,530,910✔
320
    }
321
  }
322

323
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
1,186,561,520✔
324
    // limit the output rows
325
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
22,857,738✔
326
    blockDataKeepFirstNRows(pBlock, keep);
22,856,401✔
327

328
    pLimitInfo->numOfOutputRows += pBlock->info.rows;
22,858,424✔
329
    qDebug("output limit %" PRId64 " has reached, %s", pLimit->limit, id);
22,859,320✔
330
    return true;
22,858,290✔
331
  }
332

333
  pLimitInfo->numOfOutputRows += pBlock->info.rows;
1,163,459,627✔
334
  return false;
1,163,625,770✔
335
}
336

337
static bool isDynVtbScan(SOperatorInfo* pOperator) {
×
338
  return pOperator->dynamicTask && ((STableScanInfo*)(pOperator->info))->virtualStableScan;
×
339
}
340

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

349
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
699,465,256✔
350

351
  pCost->totalBlocks += 1;
699,459,918✔
352
  pCost->totalRows += pBlock->info.rows;
699,469,527✔
353
  *status = pTableScanInfo->dataBlockLoadFlag;
699,400,681✔
354

355
  if (pOperator->exprSupp.pFilterInfo != NULL) {
699,415,564✔
356
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
100,056,655✔
357
  } else {
358
    bool overlap = false;
599,352,223✔
359
    int  ret =
360
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
599,362,217✔
361
    if (ret != TSDB_CODE_SUCCESS) {
598,961,156✔
362
      return ret;
×
363
    }
364
    if (overlap) {
598,961,156✔
365
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
34,345,355✔
366
    }
367
  }
368

369
  SDataBlockInfo* pBlockInfo = &pBlock->info;
699,281,051✔
370
  taosMemoryFreeClear(pBlock->pBlockAgg);
699,293,021✔
371

372
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
699,304,908✔
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) {
698,921,924✔
380
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
23,016,291✔
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);
23,026,991✔
384
    pCost->skipBlocks += 1;
23,030,763✔
385
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
23,031,727✔
386
    return code;
23,031,220✔
387
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
675,975,173✔
388
    pCost->loadBlockStatis += 1;
49,982,127✔
389
    loadSMA = true;  // mark the operation of load sma;
49,992,080✔
390
    bool success = true;
49,992,080✔
391
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
49,998,652✔
392
    if (code) {
49,982,687✔
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
49,974,907✔
399
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
5,150,347✔
400
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
401
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
5,150,347✔
402
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
5,150,877✔
403
      return code;
5,150,877✔
404
    } else {
405
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
44,824,560✔
406
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
44,849,094✔
407
    }
408
  }
409

410
  if (*status != FUNC_DATA_REQUIRED_DATA_LOAD) {
670,837,776✔
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)) {
670,677,936✔
418
    bool success = true;
100,031,104✔
419
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
100,044,826✔
420
    if (code) {
100,000,279✔
421
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
422
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
423
      QUERY_CHECK_CODE(code, lino, _end);
1,300✔
424
    }
425

426
    if (success) {
100,001,579✔
427
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
10,739,665✔
428
      bool   keep = false;
10,739,678✔
429
      code = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows, &keep);
10,739,678✔
430
      if (code) {
10,741,255✔
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) {
10,735,968✔
437
        qDebug("%s data block filter out by block SMA, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
4,768,738✔
438
               GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
439
        pCost->filterOutBlocks += 1;
4,768,208✔
440
        (*status) = FUNC_DATA_REQUIRED_FILTEROUT;
4,772,978✔
441
        taosMemoryFreeClear(pBlock->pBlockAgg);
4,772,978✔
442

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

449
  // free the sma info, since it should not be involved in *later computing process.
450
  taosMemoryFreeClear(pBlock->pBlockAgg);
666,100,146✔
451

452
  // try to filter data block according to current results
453
  code = doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
665,881,112✔
454
  if (code) {
666,232,247✔
455
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
456
    QUERY_CHECK_CODE(code, lino, _end);
×
457
  }
458

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

465
    STableScanInfo* p1 = pOperator->info;
9,006,248✔
466
    if (taosHashGetSize(p1->pIgnoreTables) == taosArrayGetSize(p1->base.pTableListInfo->pTableList)) {
9,006,248✔
467
      *status = FUNC_DATA_REQUIRED_ALL_FILTEROUT;
419,765✔
468
    } else {
469
      *status = FUNC_DATA_REQUIRED_FILTEROUT;
8,586,483✔
470
    }
471
    return TSDB_CODE_SUCCESS;
9,006,248✔
472
  }
473

474
  pCost->totalCheckedRows += pBlock->info.rows;
657,003,813✔
475
  pCost->loadBlocks += 1;
657,251,494✔
476

477
  SSDataBlock* p = NULL;
657,148,276✔
478
  code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pTableScanInfo->dataReader, &p);
657,367,164✔
479
  if (p == NULL || code != TSDB_CODE_SUCCESS || p != pBlock) {
657,211,536✔
480
    return code;
×
481
  }
482

483
  if ((pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) &&
657,211,536✔
484
      ((STableScanInfo*)pOperator->info)->ignoreTag) {
521,357,171✔
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);
653,203,621✔
489
    if (code) {
653,252,004✔
490
      return code;
×
491
    }
492
  }
493

494
  // restore the previous value
495
  pCost->totalRows -= pBlock->info.rows;
657,252,081✔
496

497
  if (pOperator->exprSupp.pFilterInfo != NULL) {
657,413,328✔
498
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo, NULL);
95,234,054✔
499
    QUERY_CHECK_CODE(code, lino, _end);
95,225,146✔
500

501
    int64_t st = taosGetTimestampUs();
94,526,915✔
502
    double  el = (taosGetTimestampUs() - st) / 1000.0;
94,545,519✔
503
    pTableScanInfo->readRecorder.filterTime += el;
94,545,519✔
504

505
    if (pBlock->info.rows == 0) {
94,538,057✔
506
      pCost->filterOutBlocks += 1;
20,352,423✔
507
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
20,353,857✔
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);
74,138,672✔
511
    }
512
  }
513

514
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
656,704,627✔
515
  if (limitReached) {  // set operator flag is done
656,564,348✔
516
    setOperatorCompleted(pOperator);
3,345,765✔
517
  }
518

519
  pCost->totalRows += pBlock->info.rows;
656,563,489✔
520

521
_end:
657,342,457✔
522
  if (code != TSDB_CODE_SUCCESS) {
657,393,456✔
523
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
708,670✔
524
  }
525
  return code;
657,217,553✔
526
}
527

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

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

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

542
static void freeTableCachedVal(void* param) {
236,695,494✔
543
  if (param == NULL) {
236,695,494✔
544
    return;
×
545
  }
546

547
  STableCachedVal* pVal = param;
236,695,494✔
548
  taosMemoryFree((void*)pVal->pName);
236,695,494✔
549
  taosMemoryFree(pVal->pTags);
236,678,775✔
550
  taosMemoryFree(pVal);
236,685,908✔
551
}
552

553
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
236,407,014✔
554
  int32_t          code = TSDB_CODE_SUCCESS;
236,407,014✔
555
  int32_t          lino = 0;
236,407,014✔
556
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
236,407,014✔
557
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
236,514,866✔
558

559
  pVal->pTags = NULL;
236,514,866✔
560
  pVal->pName = taosStrdup(pMetaReader->me.name);
236,496,906✔
561
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
236,415,779✔
562

563
  // only child table has tag value
564
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
236,401,660✔
565
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
236,171,778✔
566
    pVal->pTags = taosMemoryMalloc(pTag->len);
236,179,960✔
567
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
236,274,208✔
568
    memcpy(pVal->pTags, pTag, pTag->len);
236,288,063✔
569
  }
570

571
  (*ppResVal) = pVal;
236,444,925✔
572

573
_end:
236,526,070✔
574
  if (code != TSDB_CODE_SUCCESS) {
236,526,070✔
575
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
576
    freeTableCachedVal(pVal);
×
577
  }
578
  return code;
236,438,769✔
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) {
236,697,089✔
583
  (void)key;
584
  (void)keyLen;
585
  (void)ud;
586
  freeTableCachedVal(value);
236,697,089✔
587
}
236,690,547✔
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,
407,929,124✔
604
                               int32_t rows, SExecTaskInfo* pTask, STableMetaCacheInfo* pCache) {
605
  int32_t          code = TSDB_CODE_SUCCESS;
407,929,124✔
606
  int32_t          lino = 0;
407,929,124✔
607
  bool             freeReader = false;
407,929,124✔
608
  LRUHandle*       h = NULL;
407,929,124✔
609
  STableCachedVal  val = {0};
407,929,124✔
610
  SMetaReader      mr = {0};
407,929,124✔
611
  const char*      idStr = pTask->id.str;
407,996,954✔
612
  int32_t          insertRet = TAOS_LRU_STATUS_OK;
408,056,209✔
613
  STableCachedVal* pVal = NULL;
408,056,209✔
614

615
  // currently only the tbname pseudo column
616
  if (numOfExpr <= 0) {
408,129,061✔
617
    return TSDB_CODE_SUCCESS;
721,946✔
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;
407,407,115✔
624
  pBlock->info.rows = rows;
407,411,510✔
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 || pCache->pTableMetaEntryCache == NULL) {
407,405,316✔
631
    pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
1,948,624✔
632
    code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
1,770,631✔
633
    if (code != TSDB_CODE_SUCCESS) {
1,768,962✔
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);
1,768,962✔
649

650
    val.pName = mr.me.name;
1,770,393✔
651
    val.pTags = (STag*)mr.me.ctbEntry.pTags;
1,770,393✔
652

653
    freeReader = true;
1,770,393✔
654
  } else {
655
    pCache->metaFetch += 1;
405,455,548✔
656

657
    h = taosLRUCacheLookup(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(pBlock->info.id.uid));
405,596,501✔
658
    if (h == NULL) {
405,670,257✔
659
      pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
236,675,144✔
660
      freeReader = true;
236,688,053✔
661
      code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
236,688,053✔
662
      if (code != TSDB_CODE_SUCCESS) {
236,422,872✔
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);
236,422,872✔
677

678
      code = createTableCacheVal(&mr, &pVal);
236,657,589✔
679
      QUERY_CHECK_CODE(code, lino, _end);
236,450,625✔
680

681
      val = *pVal;
236,450,625✔
682
    } else {
683
      pCache->cacheHit += 1;
168,995,113✔
684
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
169,036,062✔
685
      val = *pValTmp;
169,020,730✔
686

687
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
169,028,734✔
688
      qTrace("release LRU cache, res %d", bRes);
169,034,624✔
689
    }
690

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

695
  for (int32_t j = 0; j < numOfExpr; ++j) {
1,706,958,599✔
696
    const SExprInfo* pExpr1 = &pExpr[j];
1,299,371,184✔
697
    int32_t          dstSlotId = pExpr1->base.resSchema.slotId;
1,299,421,425✔
698

699
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
1,299,463,963✔
700
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
1,299,370,905✔
701

702
    int32_t functionId = pExpr1->pExpr->_function.functionId;
1,299,403,405✔
703

704
    // this is to handle the tbname
705
    if (fmIsScanPseudoColumnFunc(functionId)) {
1,299,461,200✔
706
      int32_t fType = pExpr1->pExpr->_function.functionType;
132,226,201✔
707
      if (fType == FUNCTION_TYPE_TBNAME) {
132,226,786✔
708
        code = setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
132,226,786✔
709
        QUERY_CHECK_CODE(code, lino, _end);
132,222,305✔
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,167,141,581✔
719
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
1,167,067,283✔
720
      const char* p = pHandle->api.metaFn.extractTagVal(val.pTags, pColInfoData->info.type, &tagVal);
1,167,130,673✔
721

722
      char* data = NULL;
1,167,280,209✔
723
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
1,167,280,209✔
724
        data = tTagValToData((const STagVal*)p, false);
1,161,671,055✔
725
      } else {
726
        data = (char*)p;
5,610,520✔
727
      }
728

729
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
1,167,209,621✔
730
      if (isNullVal) {
1,167,219,465✔
731
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
5,143,102✔
732
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
1,162,076,363✔
733
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, 1, false);
1,161,627,536✔
734
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
1,161,622,659✔
735
          char* tmp = taosMemoryCalloc(1, varDataLen(data) + 1);
427,331,684✔
736
          if (tmp != NULL) {
427,348,238✔
737
            memcpy(tmp, varDataVal(data), varDataLen(data));
427,348,238✔
738
            qDebug("get tag value:%s, cid:%d, table name:%s, uid%" PRId64, tmp, tagVal.cid, val.pName,
427,350,201✔
739
                   pBlock->info.id.uid);
740
            taosMemoryFree(tmp);
427,352,908✔
741
          }
742
          taosMemoryFree(data);
427,331,855✔
743
        }
744
        QUERY_CHECK_CODE(code, lino, _end);
1,161,621,777✔
745
      } else {  // todo opt for json tag
746
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
2,941,236✔
747
          code = colDataSetVal(pColInfoData, i, data, false);
2,473,284✔
748
          QUERY_CHECK_CODE(code, lino, _end);
2,473,284✔
749
        }
750
      }
751
    }
752
  }
753

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

757
_end:
407,557,459✔
758

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

768
  if (freeReader) {
407,512,530✔
769
    pHandle->api.metaReaderFn.clearReader(&mr);
238,448,027✔
770
  }
771
  if (code != TSDB_CODE_SUCCESS) {
407,507,808✔
772
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
773
  }
774
  return code;
407,487,846✔
775
}
776

777
int32_t setTbNameColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId,
132,226,107✔
778
                         const char* name) {
779
  int32_t                     code = TSDB_CODE_SUCCESS;
132,226,107✔
780
  int32_t                     lino = 0;
132,226,107✔
781
  struct SScalarFuncExecFuncs fpSet = {0};
132,226,107✔
782
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
132,227,686✔
783
  QUERY_CHECK_CODE(code, lino, _end);
132,228,271✔
784

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

789
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
132,226,516✔
790

791
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
132,227,776✔
792
  QUERY_CHECK_CODE(code, lino, _end);
132,227,098✔
793

794
  code = colDataSetVal(&infoData, 0, buf, false);
132,227,098✔
795
  QUERY_CHECK_CODE(code, lino, _end);
132,226,146✔
796

797
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
132,226,146✔
798
  SScalarParam param = {.columnData = pColInfoData};
132,225,561✔
799

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

807
  colDataDestroy(&infoData);
132,225,301✔
808

809
_end:
132,216,327✔
810
  if (code != TSDB_CODE_SUCCESS) {
132,222,688✔
811
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
812
  }
813
  return code;
132,222,688✔
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) {
211,110,360✔
883
  int32_t code = TSDB_CODE_SUCCESS;
211,110,360✔
884
  int32_t lino = 0;
211,110,360✔
885
  code = tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, pKeyInfo, size);
211,110,360✔
886
  QUERY_CHECK_CODE(code, lino, _end);
211,010,141✔
887

888
  pInfo->tableStartIndex = TARRAY_ELEM_IDX(pInfo->base.pTableListInfo->pTableList, *pKeyInfo);
211,010,141✔
889
  pInfo->tableEndIndex = (pInfo->tableStartIndex + (*size) - 1);
211,098,104✔
890
  pInfo->pResBlock->info.blankFill = false;
211,031,457✔
891
  taosMemoryFreeClear(pInfo->pResBlock->pBlockAgg);
211,090,388✔
892

893
  if (!pInfo->needCountEmptyTable) {
211,100,016✔
894
    pInfo->countState = TABLE_COUNT_STATE_END;
191,639,038✔
895
  } else {
896
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
19,422,613✔
897
  }
898

899
_end:
211,100,381✔
900
  if (code != TSDB_CODE_SUCCESS) {
211,100,381✔
901
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
902
  }
903
  return code;
210,982,248✔
904
}
905

906
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
578,052,543✔
907
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
578,052,543✔
908
    return;
464,229,162✔
909
  }
910
  if (pInfo->base.pTableListInfo->groupOffset) {
113,897,300✔
911
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
216,670✔
912
  } else {
913
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
113,675,135✔
914
    if (code != TSDB_CODE_SUCCESS) {
113,687,284✔
915
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
48,933,257✔
916
    }
917
  }
918
}
919

920
static SSDataBlock* getOneRowResultBlock(SExecTaskInfo* pTaskInfo, STableScanBase* pBase, SSDataBlock* pBlock,
45,421,836✔
921
                                         const STableKeyInfo* tbInfo) {
922
  blockDataEmpty(pBlock);
45,421,836✔
923
  pBlock->info.rows = 1;
45,431,241✔
924
  pBlock->info.id.uid = tbInfo->uid;
45,434,706✔
925
  pBlock->info.id.groupId = tbInfo->groupId;
45,433,221✔
926
  pBlock->info.blankFill = true;
45,433,716✔
927

928
  // only one row: set all col data to null & hasNull
929
  int32_t col_num = blockDataGetNumOfCols(pBlock);
45,434,211✔
930
  for (int32_t i = 0; i < col_num; ++i) {
187,839,388✔
931
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
142,405,218✔
932
    colDataSetNULL(pColInfoData, 0);
933
  }
934

935
  // set tag/tbname
936
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
45,434,170✔
937

938
  return pBlock;
45,431,200✔
939
}
940

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

946
  pOperator->resultInfo.totalRows++;
43,656,844✔
947
  return pBlock;
43,666,249✔
948
}
949

950
static int32_t doTableScanImplNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
741,260,518✔
951
  int32_t         code = TSDB_CODE_SUCCESS;
741,260,518✔
952
  int32_t         lino = 0;
741,260,518✔
953
  STableScanInfo* pTableScanInfo = pOperator->info;
741,260,518✔
954
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
741,461,069✔
955
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
741,351,983✔
956
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
741,441,013✔
957
  bool            hasNext = false;
741,433,951✔
958
  int64_t         st = taosGetTimestampUs();
741,380,582✔
959

960
  QRY_PARAM_CHECK(ppRes);
741,380,582✔
961
  pBlock->info.dataLoad = false;
741,437,124✔
962

963
  while (true) {
24,510,005✔
964
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
765,889,539✔
965
    if (code != TSDB_CODE_SUCCESS) {
765,641,875✔
966
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
×
967
      QUERY_CHECK_CODE(code, lino, _end);
546✔
968
    }
969

970
    if (!hasNext) {
765,601,195✔
971
      break;
205,806,438✔
972
    }
973

974
    if (isTaskKilled(pTaskInfo)) {
559,794,757✔
975
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
3,507✔
976
      code = pTaskInfo->code;
3,507✔
977
      goto _end;
3,507✔
978
    }
979

980
    if (pOperator->status == OP_EXEC_DONE) {
560,238,199✔
981
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
58,128✔
982
      break;
58,128✔
983
    }
984

985
    // process this data block based on the probabilities
986
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
560,159,540✔
987
    if (!processThisBlock) {
560,063,854✔
988
      continue;
30✔
989
    }
990

991
    if (pBlock->info.id.uid) {
560,063,854✔
992
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
560,176,015✔
993
    }
994

995
    uint32_t status = 0;
560,260,382✔
996
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
560,292,465✔
997
    QUERY_CHECK_CODE(code, lino, _end);
559,933,842✔
998

999
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
559,241,396✔
1000
      break;
419,765✔
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) {
558,821,631✔
1005
      continue;
24,510,005✔
1006
    }
1007

1008
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
534,501,857✔
1009
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
534,543,818✔
1010

1011
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
534,592,482✔
1012
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
534,612,013✔
1013

1014
    (*ppRes) = pBlock;
534,526,241✔
1015
    return code;
534,459,777✔
1016
  }
1017

1018
_end:
206,976,767✔
1019
  if (code != TSDB_CODE_SUCCESS) {
207,009,602✔
1020
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
755,982✔
1021
    pTaskInfo->code = code;
755,982✔
1022
    T_LONG_JMP(pTaskInfo->env, code);
695,953✔
1023
  }
1024
  return code;
206,253,620✔
1025
}
1026

1027
static int32_t doGroupedTableScan(SOperatorInfo* pOperator, SSDataBlock** pBlock) {
790,204,380✔
1028
  int32_t         code = TSDB_CODE_SUCCESS;
790,204,380✔
1029
  int32_t         lino = 0;
790,204,380✔
1030
  STableScanInfo* pTableScanInfo = pOperator->info;
790,204,380✔
1031
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
790,436,377✔
1032
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
790,357,358✔
1033

1034
  QRY_PARAM_CHECK(pBlock);
790,428,094✔
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) {
790,412,630✔
1038
    return code;
5,431,053✔
1039
  }
1040

1041
  // do the ascending order traverse in the first place.
1042
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
979,670,803✔
1043
    SSDataBlock* p = NULL;
659,759,456✔
1044
    code = doTableScanImplNext(pOperator, &p);
659,723,354✔
1045
    QUERY_CHECK_CODE(code, lino, _end);
658,826,163✔
1046

1047
    if (p != NULL) {
658,826,163✔
1048
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
464,114,647✔
1049
      *pBlock = p;
464,075,907✔
1050
      return code;
464,123,101✔
1051
    }
1052

1053
    pTableScanInfo->scanTimes += 1;
194,711,516✔
1054
    taosHashClear(pTableScanInfo->pIgnoreTables);
194,727,426✔
1055

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

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

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

1077
    while (pTableScanInfo->scanTimes < total) {
93,330,078✔
1078
      SSDataBlock* p = NULL;
81,799,814✔
1079
      code = doTableScanImplNext(pOperator, &p);
81,803,363✔
1080
      QUERY_CHECK_CODE(code, lino, _end);
81,763,407✔
1081

1082
      if (p != NULL) {
81,763,407✔
1083
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
70,208,289✔
1084
        *pBlock = p;
70,220,494✔
1085
        return code;
70,221,774✔
1086
      }
1087

1088
      pTableScanInfo->scanTimes += 1;
11,555,118✔
1089
      taosHashClear(pTableScanInfo->pIgnoreTables);
11,557,383✔
1090

1091
      if (pTableScanInfo->scanTimes < total) {
11,557,363✔
1092
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
×
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) {
249,764,746✔
1103
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
62,787,235✔
1104
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
62,787,633✔
1105
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
115,616✔
1106
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
3,952✔
1107
        STableKeyInfo* pStart =
1108
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
3,952✔
1109

1110
        if (NULL == pStart) {
3,952✔
1111
          return code;
×
1112
        }
1113

1114
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
3,952✔
1115
        return code;
3,952✔
1116
      }
1117
    } else {  // group by tag + no sort
1118
      int32_t numOfTables = 0;
62,683,525✔
1119
      code = tableListGetSize(pTableListInfo, &numOfTables);
62,681,943✔
1120
      QUERY_CHECK_CODE(code, lino, _end);
62,683,904✔
1121

1122
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
62,683,904✔
1123
        // get empty group, mark processed & rm from hash
1124
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
61,357,638✔
1125
        if (pIte != NULL) {
61,356,559✔
1126
          size_t        keySize = 0;
43,658,832✔
1127
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
43,660,812✔
1128
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
43,659,822✔
1129
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
43,659,327✔
1130
          markGroupProcessed(pTableScanInfo, *pGroupId);
43,651,902✔
1131
          *pBlock = getBlockForEmptyTable(pOperator, &info);
43,648,437✔
1132

1133
          return code;
43,656,852✔
1134
        }
1135
      }
1136
    }
1137
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
19,135,657✔
1138
  }
1139

1140
_end:
187,017,215✔
1141
  if (code != TSDB_CODE_SUCCESS) {
206,061,136✔
1142
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1143
    pTaskInfo->code = code;
×
1144
  }
1145

1146
  return code;
206,061,136✔
1147
}
1148

1149
static int32_t createTableListInfoFromParam(SOperatorInfo* pOperator) {
238,982✔
1150
  STableScanInfo*          pInfo = pOperator->info;
238,982✔
1151
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
238,982✔
1152
  int32_t                  code = 0;
238,982✔
1153
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
238,982✔
1154
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
238,982✔
1155
  int32_t                  num = taosArrayGetSize(pParam->pUidList);
238,982✔
1156
  if (num <= 0) {
238,982✔
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",
238,982✔
1162
         pTaskInfo->id.vgId, num, pParam->tableSeq, (int64_t)taosArrayGetSize(pListInfo->pTableList),
1163
         pOperator->status);
1164

1165
  if (pParam->tableSeq) {
238,982✔
1166
    pListInfo->oneTableForEachGroup = true;
4,200✔
1167
    if (taosArrayGetSize(pListInfo->pTableList) > 0) {
4,200✔
1168
      taosHashClear(pListInfo->map);
×
1169
      taosArrayClear(pListInfo->pTableList);
×
1170
      pOperator->status = OP_EXEC_DONE;
×
1171
    }
1172
  } else {
1173
    pListInfo->oneTableForEachGroup = false;
234,782✔
1174
    pListInfo->numOfOuputGroups = 1;
234,782✔
1175
  }
1176
  if (pParam->window.skey == INT64_MAX && pParam->window.ekey == INT64_MIN) {
238,982✔
1177
    // do nothing
1178
  } else {
1179
    pInfo->base.cond.twindows.skey = pParam->window.skey;
×
1180
    pInfo->base.cond.twindows.ekey = pParam->window.ekey;
×
1181
  }
1182

1183
  STableKeyInfo info = {.groupId = 0};
238,982✔
1184
  int32_t       tableIdx = 0;
238,982✔
1185
  for (int32_t i = 0; i < num; ++i) {
7,457,552✔
1186
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
7,218,570✔
1187
    if (!pUid) {
7,215,042✔
1188
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1189
      return terrno;
×
1190
    }
1191

1192
    if (taosHashPut(pListInfo->map, pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
7,215,042✔
1193
      if (TSDB_CODE_DUP_KEY == terrno) {
6,642,926✔
1194
        continue;
6,642,926✔
1195
      }
1196
      return terrno;
×
1197
    }
1198

1199
    info.uid = *pUid;
576,652✔
1200
    void* p = taosArrayPush(pListInfo->pTableList, &info);
576,652✔
1201
    if (p == NULL) {
576,652✔
1202
      return terrno;
×
1203
    }
1204

1205
    tableIdx++;
576,652✔
1206
    qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
576,652✔
1207
  }
1208

1209
  return code;
238,982✔
1210
}
1211

1212
static int32_t doInitReader(STableScanInfo* pInfo, SExecTaskInfo* pTaskInfo, SStorageAPI* pAPI, int32_t* pNum,
208,891,603✔
1213
                            STableKeyInfo** pList) {
1214
  const char* idStr = GET_TASKID(pTaskInfo);
208,891,603✔
1215
  int32_t     code = initNextGroupScan(pInfo, pList, pNum);
208,958,730✔
1216
  if (code) {
208,751,918✔
1217
    qError("%s failed to init groupScan Info, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
×
1218
    return code;
×
1219
  }
1220

1221
  if (pInfo->base.dataReader != NULL) {
208,751,918✔
1222
    qError("%s tsdb reader should be null", idStr);
×
1223
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1224
  }
1225

1226
  code = pAPI->tsdReader.tsdReaderOpen(pInfo->base.readHandle.vnode, &pInfo->base.cond, *pList, *pNum, pInfo->pResBlock,
417,619,502✔
1227
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
208,911,116✔
1228
  if (code) {
208,851,711✔
1229
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
2,105✔
1230
  }
1231

1232
  return code;
208,803,885✔
1233
}
1234

1235
int compareColIdPair(const void* elem1, const void* elem2) {
59,962,083✔
1236
  SColIdPair* node1 = (SColIdPair*)elem1;
59,962,083✔
1237
  SColIdPair* node2 = (SColIdPair*)elem2;
59,962,083✔
1238

1239
  if (node1->orgColId == node2->orgColId) {
59,962,083✔
1240
    return 0;
106,688✔
1241
  }
1242

1243
  return node1->orgColId < node2->orgColId ? -1 : 1;
59,855,395✔
1244
}
1245

1246
int compareColIdSlotIdPair(const void* elem1, const void* elem2) {
59,962,083✔
1247
  SColIdSlotIdPair* node1 = (SColIdSlotIdPair*)elem1;
59,962,083✔
1248
  SColIdSlotIdPair* node2 = (SColIdSlotIdPair*)elem2;
59,962,083✔
1249

1250
  if (node1->orgColId == node2->orgColId) {
59,962,083✔
1251
    return 0;
106,688✔
1252
  }
1253

1254
  return node1->orgColId < node2->orgColId ? -1 : 1;
59,855,395✔
1255
}
1256

1257
static bool isNewScanParam(STableScanOperatorParam* pParam) {
9,925,064✔
1258
  return pParam->isNewParam;
9,925,064✔
1259
}
1260

1261
static int32_t createVTableScanInfoFromBatchParam(SOperatorInfo* pOperator) {
×
1262
  int32_t                  code = 0;
×
1263
  int32_t                  lino = 0;
×
1264
  STableScanInfo*          pInfo = pOperator->info;
×
1265
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
×
1266
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
×
1267
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
×
1268
  SMetaReader              orgTable = {0};
×
1269
  SMetaReader              superTable = {0};
×
1270
  SSchemaWrapper*          schema = NULL;
×
1271
  SArray*                  pColArray = NULL;
×
1272
  SArray*                  pBlockColArray = NULL;
×
1273
  int32_t                  num = 0;
×
1274
  STableKeyInfo*           pList = NULL;
×
1275
  SOrgTbInfo*              pOrgTbInfo;
1276
  bool                     isNewTable;
1277

1278
  cleanupQueryTableDataCond(&pInfo->base.cond);
×
1279

1280
  if (pOperator->pOperatorGetParam) {
×
1281
    STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
×
1282
    pInfo->pBatchColMap = pParam->pBatchTbInfo;
×
1283
    pParam->pBatchTbInfo = NULL;
×
1284
    pInfo->cachedTimeWindow = pParam->window;
×
1285
    pInfo->currentBatchIdx = 0;
×
1286
    pInfo->lastBatchIdx = -1;
×
1287
    pInfo->cachedTagList = pParam->pTagList;
×
1288
    pParam->pTagList = NULL;
×
1289
    pInfo->cachedGroupId = pParam->groupid;
×
1290

1291
    qDebug("vgId:%d dynamic vtable scan with batch param, total %zd tables to scan, groupid:%" PRIu64 ", %s",
×
1292
           pTaskInfo->id.vgId, taosArrayGetSize(pInfo->pBatchColMap), pInfo->cachedGroupId, GET_TASKID(pTaskInfo));
1293
    taosHashClear(pInfo->pIgnoreTables);
×
1294

1295
    if (pInfo->base.pdInfo.pExprSup) {
×
1296
      SExprSupp* pSup = pInfo->base.pdInfo.pExprSup;
×
1297
      for (int32_t i = 0; i < pSup->numOfExprs; i++) {
×
1298
        SqlFunctionCtx ctx = pSup->pCtx[i];
×
1299
        ctx.skipDynDataCheck = false;
×
1300
      }
1301
    }
1302
  }
1303

1304
  pOrgTbInfo = taosArrayGet(pInfo->pBatchColMap, pInfo->currentBatchIdx);
×
1305
  QUERY_CHECK_NULL(pOrgTbInfo, code, lino, _return, terrno);
×
1306

1307
  if (pInfo->currentBatchIdx == pInfo->lastBatchIdx) {
×
1308
    isNewTable = false;
×
1309
  } else {
1310
    isNewTable = true;
×
1311
    pInfo->lastBatchIdx = pInfo->currentBatchIdx;
×
1312
  }
1313

1314
  if (isNewTable) {
×
1315
    pAPI->metaReaderFn.initReader(&orgTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
×
1316
    code = pAPI->metaReaderFn.getTableEntryByName(&orgTable, strstr(pOrgTbInfo->tbName, ".") + 1);
×
1317
    pAPI->metaReaderFn.readerReleaseLock(&orgTable);
×
1318
    qDebug("dynamic vtable scan for origin table:%s, %s", pOrgTbInfo->tbName, GET_TASKID(pTaskInfo));
×
1319
    QUERY_CHECK_CODE(code, lino, _return);
×
1320
    switch (orgTable.me.type) {
×
1321
      case TSDB_CHILD_TABLE:
×
1322
        pAPI->metaReaderFn.initReader(&superTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
×
1323
        code = pAPI->metaReaderFn.getTableEntryByUid(&superTable, orgTable.me.ctbEntry.suid);
×
1324
        pAPI->metaReaderFn.readerReleaseLock(&superTable);
×
1325
        QUERY_CHECK_CODE(code, lino, _return);
×
1326
        schema = &superTable.me.stbEntry.schemaRow;
×
1327
        break;
×
1328
      case TSDB_NORMAL_TABLE:
×
1329
        schema = &orgTable.me.ntbEntry.schemaRow;
×
1330
        break;
×
1331
      default:
×
1332
        qError("invalid table type:%d", orgTable.me.type);
×
1333
        return TSDB_CODE_INVALID_PARA;
×
1334
        break;
1335
    }
1336

1337
    pListInfo->oneTableForEachGroup = true;
×
1338
    taosHashClear(pListInfo->map);
×
1339
    taosArrayClear(pListInfo->pTableList);
×
1340

1341
    uint64_t      pUid = orgTable.me.uid;
×
1342
    STableKeyInfo info = {.groupId = pInfo->cachedGroupId, .uid = pUid};
×
1343
    int32_t       tableIdx = 0;
×
1344
    code = taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t));
×
1345
    QUERY_CHECK_CODE(code, lino, _return);
×
1346
    QUERY_CHECK_NULL(taosArrayPush(pListInfo->pTableList, &info), code, lino, _return, terrno);
×
1347
    qDebug("add dynamic table scan uid:%" PRIu64 ", %" PRIu64 ", %s", info.uid, info.groupId, GET_TASKID(pTaskInfo));
×
1348

1349
    pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
×
1350
    QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
×
1351
    pBlockColArray = taosArrayInit(schema->nCols, sizeof(SColIdSlotIdPair));
×
1352
    QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
×
1353

1354
    // virtual table's origin table scan do not has ts column.
1355
    SColIdPair tsPair = {.vtbColId = PRIMARYKEY_TIMESTAMP_COL_ID, .orgColId = PRIMARYKEY_TIMESTAMP_COL_ID};
×
1356
    QUERY_CHECK_NULL(taosArrayPush(pColArray, &tsPair), code, lino, _return, terrno);
×
1357

1358
    for (int32_t i = 0; i < taosArrayGetSize(pOrgTbInfo->colMap); ++i) {
×
1359
      SColIdNameKV* kv = taosArrayGet(pOrgTbInfo->colMap, i);
×
1360
      for (int32_t j = 0; j < schema->nCols; j++) {
×
1361
        if (strcmp(kv->colName, schema->pSchema[j].name) == 0) {
×
1362
          SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(schema->pSchema[j].colId)};
×
1363
          QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
×
1364
          qDebug("dynamic vtable scan map col:%s, orgColId:%d, vtbColId:%d, %s", kv->colName, pPair.orgColId,
×
1365
                 pPair.vtbColId, GET_TASKID(pTaskInfo));
1366
          break;
×
1367
        }
1368
      }
1369
    }
1370

1371
    for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
×
1372
      SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, i);
×
1373
      for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
×
1374
        SColMatchItem* pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
×
1375
        if (pItem->colId == pPair->vtbColId) {
×
1376
          SColIdSlotIdPair colIdSlotIdPair = {.orgColId = pPair->orgColId, .vtbSlotId = pItem->dstSlotId};
×
1377
          QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &colIdSlotIdPair), code, lino, _return, terrno);
×
1378
          qDebug("dynamic vtable scan block col map orgColId:%d, vtbSlotId:%d, %s", colIdSlotIdPair.orgColId,
×
1379
                 colIdSlotIdPair.vtbSlotId, GET_TASKID(pTaskInfo));
1380
          break;
×
1381
        }
1382
      }
1383
    }
1384

1385
    taosArraySort(pColArray, compareColIdPair);
×
1386
    taosArraySort(pBlockColArray, compareColIdSlotIdPair);
×
1387
    if (pInfo->pBlockColMap) {
×
1388
      taosArrayDestroy(pInfo->pBlockColMap);
×
1389
      pInfo->pBlockColMap = NULL;
×
1390
    }
1391
    pInfo->pBlockColMap = taosArrayDup(pBlockColArray, NULL);
×
1392
    QUERY_CHECK_NULL(pInfo->pBlockColMap, code, lino, _return, terrno)
×
1393

1394

1395
    taosArrayRemoveDuplicate(pColArray, compareColIdPair, NULL);
×
1396
    taosArrayRemoveDuplicate(pBlockColArray, compareColIdSlotIdPair, NULL);
×
1397

1398
    if (pInfo->base.pdInfo.pExprSup) {
×
1399
      SExprSupp* pSup = pInfo->base.pdInfo.pExprSup;
×
1400
      for (int32_t i = 0; i < pSup->numOfExprs; i++) {
×
1401
        SqlFunctionCtx* ctx = &pSup->pCtx[i];
×
1402
        bool           needScan = false;
×
1403
        for (int32_t j = 0; j < (fmIsImplicitTsFunc(ctx->functionId) ? ctx->numOfParams - 1 : ctx->numOfParams); j++) {
×
1404
          SFunctParam pParam = ctx->param[j];
×
1405
          for (int32_t k = 0; k < taosArrayGetSize(pColArray); k++) {
×
1406
            SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, k);
×
1407
            if (pParam.pCol && pParam.pCol->colId == pPair->vtbColId) {
×
1408
              needScan |= true;
×
1409
              break;
×
1410
            }
1411
          }
1412
        }
1413
        qDebug("dynamic vtable scan expr funcId:%d, needScan:%d, %s", ctx->functionId, needScan, GET_TASKID(pTaskInfo));
×
1414
        if (!needScan) {
×
1415
          ctx->skipDynDataCheck = true;
×
1416
        } else {
1417
          ctx->skipDynDataCheck = false;
×
1418
        }
1419
      }
1420
    }
1421

1422

1423
  } else {
1424
    pColArray = pInfo->lastColArray;
×
1425
    pBlockColArray = pInfo->lastBlockColArray;
×
1426
  }
1427

1428
  code = initQueryTableDataCondWithColArray(&pInfo->base.cond, &pInfo->base.orgCond, &pInfo->base.readHandle, pColArray);
×
1429
  QUERY_CHECK_CODE(code, lino, _return);
×
1430

1431
  if (pInfo->pResBlock) {
×
1432
    blockDataDestroy(pInfo->pResBlock);
×
1433
    pInfo->pResBlock = NULL;
×
1434
  }
1435

1436
  qDebug("dynamic vtable scan isNewTable:%d, order:%d, lastTimeWindow:[%" PRId64 ", %" PRId64 "], "
×
1437
         "cachedTimeWindow:[%" PRId64 ", %" PRId64 "] , "
1438
         "orgCondWindow: :[%" PRId64 ", %" PRId64 "], %s",
1439
         isNewTable, pInfo->base.orgCond.order,
1440
         pInfo->lastTimeWindow.skey, pInfo->lastTimeWindow.ekey,
1441
         pInfo->cachedTimeWindow.skey, pInfo->cachedTimeWindow.ekey,
1442
         pInfo->base.orgCond.twindows.skey, pInfo->base.orgCond.twindows.ekey, GET_TASKID(pTaskInfo));
1443

1444
  if (isNewTable) {
×
1445
    if (pInfo->cachedTimeWindow.skey == INT64_MAX && pInfo->cachedTimeWindow.ekey == INT64_MIN) {
×
1446
      pInfo->base.cond.twindows.skey = pInfo->base.orgCond.twindows.skey;
×
1447
      pInfo->base.cond.twindows.ekey = pInfo->base.orgCond.twindows.ekey;
×
1448
    } else {
1449
      pInfo->base.cond.twindows.skey = pInfo->cachedTimeWindow.skey;
×
1450
      pInfo->base.cond.twindows.ekey = pInfo->cachedTimeWindow.ekey;
×
1451
    }
1452
  } else {
1453
    if (pInfo->base.orgCond.order == TSDB_ORDER_DESC)
×
1454
      pInfo->base.cond.twindows.ekey = pInfo->lastTimeWindow.skey - 1;
×
1455
    else {
1456
      pInfo->base.cond.twindows.skey = pInfo->lastTimeWindow.ekey + 1;
×
1457
    }
1458
  }
1459

1460
  pInfo->base.cond.suid = orgTable.me.type == TSDB_CHILD_TABLE ? superTable.me.uid : 0;
×
1461
  pInfo->currentGroupId = 0;
×
1462
  pInfo->ignoreTag = true;
×
1463

1464
  code = createOneDataBlockWithColArray(pInfo->pOrgBlock, pBlockColArray, &pInfo->pResBlock);
×
1465
  QUERY_CHECK_CODE(code, lino, _return);
×
1466

1467
  if (isNewTable) {
×
1468
    if (pInfo->base.dataReader != NULL) {
×
1469
      pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
×
1470
    }
1471
    pInfo->base.dataReader = NULL;
×
1472

1473
    taosRLockLatch(&pTaskInfo->lock);
×
1474
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
×
1475
    taosRUnLockLatch(&pTaskInfo->lock);
×
1476
    QUERY_CHECK_CODE(code, lino, _return);
×
1477

1478
    pInfo->newReader = true;
×
1479
    if (pInfo->lastColArray) {
×
1480
      taosArrayDestroy(pInfo->lastColArray);
×
1481
      pInfo->lastColArray = NULL;
×
1482
    }
1483
    if (pInfo->lastBlockColArray) {
×
1484
      taosArrayDestroy(pInfo->lastBlockColArray);
×
1485
      pInfo->lastBlockColArray = NULL;
×
1486
    }
1487
    pInfo->lastColArray = pColArray;
×
1488
    pInfo->lastBlockColArray = pBlockColArray;
×
1489
  } else {
1490
    code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
×
1491
    QUERY_CHECK_CODE(code, lino, _return);
×
1492

1493
    pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, pInfo->pResBlock);
×
1494
    pInfo->newReader = false;
×
1495
    pInfo->scanTimes = 0;
×
1496
  }
1497

1498
  pOperator->status = OP_OPENED;
×
1499

1500
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
×
1501
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1502
  }
1503

1504
  pInfo->currentGroupId = -1;
×
1505
_return:
×
1506
  if (code) {
×
1507
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1508
  }
1509
  pAPI->metaReaderFn.clearReader(&superTable);
×
1510
  pAPI->metaReaderFn.clearReader(&orgTable);
×
1511
  return code;
×
1512
}
1513

1514
static int32_t createVTableScanInfoFromParam(SOperatorInfo* pOperator) {
5,918,998✔
1515
  int32_t                  code = 0;
5,918,998✔
1516
  int32_t                  lino = 0;
5,918,998✔
1517
  STableScanInfo*          pInfo = pOperator->info;
5,918,998✔
1518
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
5,918,998✔
1519
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
5,918,998✔
1520
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
5,918,998✔
1521
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
5,918,998✔
1522
  SMetaReader              orgTable = {0};
5,918,998✔
1523
  SMetaReader              superTable = {0};
5,918,998✔
1524
  SSchemaWrapper*          schema = NULL;
5,918,998✔
1525
  SArray*                  pColArray = NULL;
5,918,998✔
1526
  SArray*                  pBlockColArray = NULL;
5,918,998✔
1527
  int32_t                  num = 0;
5,918,998✔
1528
  STableKeyInfo*           pList = NULL;
5,918,998✔
1529
  SOrgTbInfo*              pOrgTbInfo;
1530

1531
  cleanupQueryTableDataCond(&pInfo->base.cond);
5,918,998✔
1532

1533
  pOrgTbInfo = pParam->pOrgTbInfo;
5,918,998✔
1534

1535
  QUERY_CHECK_NULL(pOrgTbInfo, code, lino, _return, terrno);
5,918,998✔
1536

1537
  pAPI->metaReaderFn.initReader(&orgTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
5,918,998✔
1538
  code = pAPI->metaReaderFn.getTableEntryByName(&orgTable, strstr(pOrgTbInfo->tbName, ".") + 1);
5,918,998✔
1539
  pAPI->metaReaderFn.readerReleaseLock(&orgTable);
5,918,998✔
1540
  qDebug("dynamic vtable scan for origin table:%s, %s", pOrgTbInfo->tbName, GET_TASKID(pTaskInfo));
5,918,998✔
1541
  QUERY_CHECK_CODE(code, lino, _return);
5,918,998✔
1542
  switch (orgTable.me.type) {
5,918,998✔
1543
    case TSDB_CHILD_TABLE:
2,038,998✔
1544
      pAPI->metaReaderFn.initReader(&superTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
2,038,998✔
1545
      code = pAPI->metaReaderFn.getTableEntryByUid(&superTable, orgTable.me.ctbEntry.suid);
2,038,998✔
1546
      pAPI->metaReaderFn.readerReleaseLock(&superTable);
2,038,998✔
1547
      QUERY_CHECK_CODE(code, lino, _return);
2,038,998✔
1548
      schema = &superTable.me.stbEntry.schemaRow;
2,038,998✔
1549
      break;
2,038,998✔
1550
    case TSDB_NORMAL_TABLE:
3,880,000✔
1551
      schema = &orgTable.me.ntbEntry.schemaRow;
3,880,000✔
1552
      break;
3,880,000✔
1553
    default:
×
1554
      qError("invalid table type:%d", orgTable.me.type);
×
1555
      return TSDB_CODE_INVALID_PARA;
×
1556
      break;
1557
  }
1558

1559
  pListInfo->oneTableForEachGroup = true;
5,918,998✔
1560
  taosHashClear(pListInfo->map);
5,918,998✔
1561
  taosArrayClear(pListInfo->pTableList);
5,918,998✔
1562

1563
  uint64_t      pUid = orgTable.me.uid;
5,918,998✔
1564
  STableKeyInfo info = {.groupId = 0, .uid = pUid};
5,918,998✔
1565
  int32_t       tableIdx = 0;
5,918,998✔
1566
  code = taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t));
5,918,998✔
1567
  QUERY_CHECK_CODE(code, lino, _return);
5,918,998✔
1568
  QUERY_CHECK_NULL(taosArrayPush(pListInfo->pTableList, &info), code, lino, _return, terrno);
11,837,996✔
1569
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
5,918,998✔
1570

1571
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
5,918,998✔
1572
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
5,918,998✔
1573
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(SColIdSlotIdPair));
5,918,998✔
1574
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
5,918,998✔
1575

1576
  // virtual table's origin table scan do not has ts column.
1577
  SColIdPair tsPair = {.vtbColId = PRIMARYKEY_TIMESTAMP_COL_ID, .orgColId = PRIMARYKEY_TIMESTAMP_COL_ID,
5,918,998✔
1578
                       .type.type = TSDB_DATA_TYPE_TIMESTAMP,
1579
                       .type.bytes = 8};
1580
  QUERY_CHECK_NULL(taosArrayPush(pColArray, &tsPair), code, lino, _return, terrno);
5,918,998✔
1581

1582
  for (int32_t i = 0; i < taosArrayGetSize(pOrgTbInfo->colMap); ++i) {
23,925,200✔
1583
    SColIdNameKV* kv = taosArrayGet(pOrgTbInfo->colMap, i);
18,006,202✔
1584
    for (int32_t j = 0; j < schema->nCols; j++) {
148,505,352✔
1585
      if (strcmp(kv->colName, schema->pSchema[j].name) == 0) {
148,505,352✔
1586
        SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(schema->pSchema[j].colId),
18,006,202✔
1587
                            .type.type = schema->pSchema[j].type,
18,006,202✔
1588
                            .type.bytes = schema->pSchema[j].bytes};
18,006,202✔
1589
        QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
18,006,202✔
1590
        break;
18,006,202✔
1591
      }
1592
    }
1593
  }
1594

1595
  for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
29,843,581✔
1596
    SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, i);
23,925,200✔
1597
    for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
150,452,208✔
1598
      SColMatchItem* pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
150,452,208✔
1599
      if (pItem->colId == pPair->vtbColId) {
150,452,208✔
1600
        if (pItem->dataType.type != pPair->type.type ||
23,925,200✔
1601
            pItem->dataType.bytes != pPair->type.bytes) {
23,924,583✔
1602
          qError("column type not match for vtable colId:%d, org colId:%d, org table name:%s", pPair->vtbColId, pPair->orgColId, orgTable.me.name);
617✔
1603
          code = TSDB_CODE_VTABLE_COLUMN_TYPE_MISMATCH;
617✔
1604
          goto _return;
617✔
1605
        }
1606
        SColIdSlotIdPair colIdSlotIdPair = {.orgColId = pPair->orgColId, .vtbSlotId = pItem->dstSlotId};
23,924,583✔
1607
        QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &colIdSlotIdPair), code, lino, _return, terrno);
23,924,583✔
1608
        break;
23,924,583✔
1609
      }
1610
    }
1611
  }
1612

1613
  taosArraySort(pColArray, compareColIdPair);
5,918,381✔
1614
  taosArraySort(pBlockColArray, compareColIdSlotIdPair);
5,918,381✔
1615
  if (pInfo->pBlockColMap) {
5,918,381✔
1616
    taosArrayDestroy(pInfo->pBlockColMap);
5,067,121✔
1617
    pInfo->pBlockColMap = NULL;
5,067,121✔
1618
  }
1619
  pInfo->pBlockColMap = taosArrayDup(pBlockColArray, NULL);
5,918,381✔
1620
  QUERY_CHECK_NULL(pInfo->pBlockColMap, code, lino, _return, terrno)
5,918,381✔
1621

1622

1623
  taosArrayRemoveDuplicate(pColArray, compareColIdPair, NULL);
5,918,381✔
1624
  taosArrayRemoveDuplicate(pBlockColArray, compareColIdSlotIdPair, NULL);
5,918,381✔
1625

1626
  code = initQueryTableDataCondWithColArray(&pInfo->base.cond, &pInfo->base.orgCond, &pInfo->base.readHandle, pColArray);
5,918,381✔
1627
  QUERY_CHECK_CODE(code, lino, _return);
5,918,381✔
1628

1629
  if (pInfo->pResBlock) {
5,918,381✔
1630
    blockDataDestroy(pInfo->pResBlock);
4,006,683✔
1631
    pInfo->pResBlock = NULL;
4,006,683✔
1632
  }
1633

1634
  if (isNewScanParam(pParam)) {
5,918,381✔
1635
    if (pParam->window.skey == INT64_MAX && pParam->window.ekey == INT64_MIN) {
1,918,304✔
1636
      pInfo->base.cond.twindows.skey = pInfo->base.orgCond.twindows.skey;
1,918,304✔
1637
      pInfo->base.cond.twindows.ekey = pInfo->base.orgCond.twindows.ekey;
1,918,304✔
1638
    } else {
1639
      pInfo->base.cond.twindows.skey = pParam->window.skey;
×
1640
      pInfo->base.cond.twindows.ekey = pParam->window.ekey;
×
1641
    }
1642
  } else {
1643
    pInfo->base.cond.twindows.skey = pParam->window.ekey + 1;
4,000,077✔
1644
  }
1645
  pInfo->base.cond.suid = orgTable.me.type == TSDB_CHILD_TABLE ? superTable.me.uid : 0;
5,918,381✔
1646
  pInfo->currentGroupId = 0;
5,918,381✔
1647
  pInfo->ignoreTag = true;
5,918,381✔
1648

1649
  code = createOneDataBlockWithColArray(pInfo->pOrgBlock, pBlockColArray, &pInfo->pResBlock);
5,918,381✔
1650
  QUERY_CHECK_CODE(code, lino, _return);
5,918,381✔
1651

1652
  void **reader = taosHashGet(pInfo->readerCache, &pUid, sizeof(uint64_t));
5,918,381✔
1653
  if (reader) {
5,918,381✔
1654
    if (isNewScanParam(pParam)) {
4,006,683✔
1655
      pAPI->tsdReader.tsdReaderClose(*reader);
6,606✔
1656
      pInfo->base.dataReader = NULL;
6,606✔
1657

1658
      code = taosHashRemove(pInfo->readerCache, &pUid, sizeof(uint64_t));
6,606✔
1659
      QUERY_CHECK_CODE(code, lino, _return);
6,606✔
1660

1661
      taosRLockLatch(&pTaskInfo->lock);
6,606✔
1662
      code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
6,606✔
1663
      taosRUnLockLatch(&pTaskInfo->lock);
6,606✔
1664
      QUERY_CHECK_CODE(code, lino, _return);
6,606✔
1665

1666
      code = taosHashPut(pInfo->readerCache, &pUid, sizeof(uint64_t), &pInfo->base.dataReader, POINTER_BYTES);
6,606✔
1667
      QUERY_CHECK_CODE(code, lino, _return);
6,606✔
1668
      pInfo->newReader = true;
6,606✔
1669
    } else {
1670
      pInfo->base.dataReader = *reader;
4,000,077✔
1671
      code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
4,000,077✔
1672
      QUERY_CHECK_CODE(code, lino, _return);
4,000,077✔
1673

1674
      pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, pInfo->pResBlock);
4,000,077✔
1675
      pInfo->newReader = false;
4,000,077✔
1676
      pInfo->scanTimes = 0;
4,000,077✔
1677
    }
1678
  } else {
1679
    pInfo->base.dataReader = NULL;
1,911,698✔
1680

1681
    taosRLockLatch(&pTaskInfo->lock);
1,911,698✔
1682
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
1,911,698✔
1683
    taosRUnLockLatch(&pTaskInfo->lock);
1,911,698✔
1684
    QUERY_CHECK_CODE(code, lino, _return);
1,911,698✔
1685

1686
    code = taosHashPut(pInfo->readerCache, &pUid, sizeof(uint64_t), &pInfo->base.dataReader, POINTER_BYTES);
1,911,698✔
1687
    QUERY_CHECK_CODE(code, lino, _return);
1,911,698✔
1688
    pInfo->newReader = true;
1,911,698✔
1689
  }
1690
  pOperator->status = OP_OPENED;
5,918,381✔
1691

1692
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
5,918,381✔
1693
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1694
  }
1695

1696
  pInfo->currentGroupId = -1;
5,918,381✔
1697
_return:
5,918,998✔
1698
  if (code) {
5,918,998✔
1699
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
617✔
1700
  }
1701
  taosArrayDestroy(pColArray);
5,918,998✔
1702
  taosArrayDestroy(pBlockColArray);
5,918,998✔
1703
  pAPI->metaReaderFn.clearReader(&superTable);
5,918,998✔
1704
  pAPI->metaReaderFn.clearReader(&orgTable);
5,918,998✔
1705
  return code;
5,918,998✔
1706
}
1707

1708
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
206,084,072✔
1709
  int32_t         code = TSDB_CODE_SUCCESS;
206,084,072✔
1710
  int32_t         lino = 0;
206,084,072✔
1711
  STableScanInfo* pInfo = pOperator->info;
206,084,072✔
1712
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
206,141,278✔
1713
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
206,132,675✔
1714
  int32_t         numOfTables = 0;
206,133,908✔
1715

1716
  QRY_PARAM_CHECK(pResult);
206,125,290✔
1717

1718
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
206,116,005✔
1719
  QUERY_CHECK_CODE(code, lino, _end);
206,145,195✔
1720

1721
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
206,145,195✔
1722
    setOperatorCompleted(pOperator);
203,896,853✔
1723
    if (pOperator->dynamicTask) {
203,899,520✔
1724
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
238,282✔
1725
      taosHashClear(pInfo->base.pTableListInfo->map);
238,282✔
1726
    }
1727
    return code;
203,869,190✔
1728
  }
1729

1730
  // reset value for the next group data output
1731
  pOperator->status = OP_OPENED;
2,215,011✔
1732
  resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
2,215,011✔
1733

1734
  int32_t        num = 0;
2,215,011✔
1735
  STableKeyInfo* pList = NULL;
2,215,011✔
1736
  code = initNextGroupScan(pInfo, &pList, &num);
2,215,011✔
1737
  QUERY_CHECK_CODE(code, lino, _end);
2,215,011✔
1738

1739
  code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
2,215,011✔
1740
  QUERY_CHECK_CODE(code, lino, _end);
2,215,011✔
1741

1742
  code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
2,215,011✔
1743
  QUERY_CHECK_CODE(code, lino, _end);
2,215,011✔
1744
  pInfo->scanTimes = 0;
2,215,011✔
1745

1746
  code = doGroupedTableScan(pOperator, pResult);
2,215,011✔
1747
  QUERY_CHECK_CODE(code, lino, _end);
2,215,011✔
1748

1749
  if (*pResult != NULL) {
2,215,011✔
1750
    if (pOperator->dynamicTask) {
1,755,165✔
1751
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
1,510,839✔
1752
    }
1753
    return code;
1,755,165✔
1754
  }
1755

1756
_end:
459,846✔
1757
  if (code != TSDB_CODE_SUCCESS) {
459,846✔
1758
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1759
    pTaskInfo->code = code;
×
1760
  }
1761
  return code;
459,846✔
1762
}
1763

1764
static int32_t groupSeqTableScan(SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
772,920,918✔
1765
  int32_t         code = TSDB_CODE_SUCCESS;
772,920,918✔
1766
  int32_t         lino = 0;
772,920,918✔
1767
  STableScanInfo* pInfo = pOperator->info;
772,920,918✔
1768
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
773,200,733✔
1769
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
773,157,159✔
1770
  int32_t         num = 0;
773,127,776✔
1771
  STableKeyInfo*  pList = NULL;
773,193,811✔
1772
  SSDataBlock*    pResult = NULL;
773,084,593✔
1773
  const char*     idStr = GET_TASKID(pTaskInfo);
773,134,262✔
1774

1775
  QRY_PARAM_CHECK(pResBlock);
773,079,185✔
1776

1777
  if (pInfo->currentGroupId == -1) {
773,131,089✔
1778
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
206,920,537✔
1779
      setOperatorCompleted(pOperator);
×
1780
      return code;
×
1781
    }
1782

1783
    taosRLockLatch(&pTaskInfo->lock);
207,014,930✔
1784
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
207,047,650✔
1785
    taosRUnLockLatch(&pTaskInfo->lock);
206,962,399✔
1786
    QUERY_CHECK_CODE(code, lino, _end);
207,020,306✔
1787

1788
    if (pInfo->filesetDelimited) {
207,018,201✔
1789
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
235,829✔
1790
    }
1791

1792
    if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
207,020,028✔
1793
      pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1794
    }
1795
  }
1796

1797
  code = doGroupedTableScan(pOperator, &pResult);
773,132,375✔
1798
  QUERY_CHECK_CODE(code, lino, _end);
772,184,818✔
1799

1800
  if (pResult != NULL) {
772,184,818✔
1801
    if (pOperator->dynamicTask) {
568,202,049✔
1802
      pResult->info.id.groupId = pResult->info.id.uid;
535,350✔
1803
    }
1804

1805
    *pResBlock = pResult;
568,241,978✔
1806
    return code;
568,017,594✔
1807
  }
1808

1809
  while (true) {
1810
    code = startNextGroupScan(pOperator, &pResult);
204,027,860✔
1811
    QUERY_CHECK_CODE(code, lino, _end);
204,176,172✔
1812

1813
    if (pResult || pOperator->status == OP_EXEC_DONE) {
204,176,172✔
1814
      *pResBlock = pResult;
204,142,090✔
1815
      return code;
204,122,449✔
1816
    }
1817
  }
1818

1819
_end:
2,105✔
1820
  if (code != TSDB_CODE_SUCCESS) {
2,105✔
1821
    qError("%s %s failed at line %d since %s", idStr, __func__, lino, tstrerror(code));
2,105✔
1822
    pTaskInfo->code = code;
2,105✔
1823
  }
1824

1825
  return code;
2,105✔
1826
}
1827

1828
static bool isEmptyQueryTimeWindow(STimeWindow* pWindow) {
5,918,381✔
1829
  return (pWindow == NULL) || (pWindow->skey > pWindow->ekey);
5,918,381✔
1830
}
1831

1832
static int32_t doStbJoinDynamicTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
238,982✔
1833
  int32_t                  code = TSDB_CODE_SUCCESS;
238,982✔
1834
  int32_t                  lino = 0;
238,982✔
1835
  STableScanInfo*          pInfo = pOperator->info;
238,982✔
1836
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
238,982✔
1837
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
238,982✔
1838
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
238,982✔
1839

1840
  code = createTableListInfoFromParam(pOperator);
238,982✔
1841
  freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
238,982✔
1842
  pOperator->pOperatorGetParam = NULL;
238,982✔
1843
  QUERY_CHECK_CODE(code, lino, _end);
238,982✔
1844

1845
  if (pOperator->status == OP_EXEC_DONE) {
238,982✔
1846
    pInfo->currentGroupId = -1;
2,100✔
1847
    pOperator->status = OP_OPENED;
2,100✔
1848
    SSDataBlock* result = NULL;
2,100✔
1849

1850
    while (true) {
1851
      code = startNextGroupScan(pOperator, &result);
2,100✔
1852
      QUERY_CHECK_CODE(code, lino, _end);
2,100✔
1853

1854
      if (result || pOperator->status == OP_EXEC_DONE) {
2,100✔
1855
        (*ppRes) = result;
2,100✔
1856
        return code;
2,100✔
1857
      }
1858
    }
1859
  }
1860
_end:
236,882✔
1861
  if (code) {
236,882✔
1862
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1863
  }
1864
  return code;
236,882✔
1865
}
1866

1867
static int32_t doVstbSingleDynamicTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
5,918,998✔
1868
  int32_t                  code = TSDB_CODE_SUCCESS;
5,918,998✔
1869
  int32_t                  lino = 0;
5,918,998✔
1870
  STableScanInfo*          pInfo = pOperator->info;
5,918,998✔
1871
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
5,918,998✔
1872
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
5,918,998✔
1873
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
5,918,998✔
1874

1875
  QRY_PARAM_CHECK(ppRes);
5,918,998✔
1876
  qTrace("%s call", __FUNCTION__);
5,918,998✔
1877

1878
  code = createVTableScanInfoFromParam(pOperator);
5,918,998✔
1879
  freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
5,918,998✔
1880
  pOperator->pOperatorGetParam = NULL;
5,918,998✔
1881

1882
  QUERY_CHECK_CODE(code, lino, _end);
5,918,998✔
1883

1884
  SSDataBlock* result = NULL;
5,918,381✔
1885
  if (isEmptyQueryTimeWindow(&pInfo->base.cond.twindows) && pInfo->base.cond.type == TIMEWINDOW_RANGE_CONTAINED) {
5,918,381✔
1886
    (*ppRes) = result;
18,456✔
1887
    return code;
18,456✔
1888
  }
1889

1890
  if (pInfo->newReader) {
5,899,925✔
1891
    code = startNextGroupScan(pOperator, &result);
1,918,304✔
1892
    QUERY_CHECK_CODE(code, lino, _end);
1,918,304✔
1893
  } else {
1894
    code = doGroupedTableScan(pOperator, &result);
3,981,621✔
1895
    QUERY_CHECK_CODE(code, lino, _end);
3,981,621✔
1896
  }
1897

1898
  if (result) {
5,899,925✔
1899
    SSDataBlock* res = NULL;
4,000,077✔
1900
    pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, NULL);
4,000,077✔
1901
    pInfo->lastTimeWindow.skey = result->info.window.skey;
4,000,077✔
1902
    pInfo->lastTimeWindow.ekey = result->info.window.ekey;
4,000,077✔
1903
    code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, pInfo->pBlockColMap, &res);
4,000,077✔
1904
    QUERY_CHECK_CODE(code, lino, _end);
4,000,077✔
1905

1906
    pInfo->pResBlock = res;
4,000,077✔
1907
    blockDataDestroy(result);
4,000,077✔
1908
    (*ppRes) = res;
4,000,077✔
1909
  } else {
1910
    STableKeyInfo *keyInfo = taosArrayGet(pInfo->base.pTableListInfo->pTableList, 0);
1,899,848✔
1911
    QUERY_CHECK_NULL(keyInfo, code, lino, _end, terrno)
1,899,848✔
1912

1913
    blockDataDestroy(pInfo->pResBlock);
1,899,848✔
1914
    pInfo->pResBlock = NULL;
1,899,848✔
1915

1916
    void** reader = taosHashGet(pInfo->readerCache, &keyInfo->uid, sizeof(uint64_t));
1,899,848✔
1917
    if (reader) {
1,899,848✔
1918
      if (*reader == pInfo->base.dataReader) {
1,899,848✔
1919
        pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, NULL);
1,899,848✔
1920
        pInfo->base.dataReader = NULL;
1,899,848✔
1921
      }
1922
      if (pAPI->tsdReader.tsdReaderClose) {
1,899,848✔
1923
        pAPI->tsdReader.tsdReaderClose(*reader);
1,899,848✔
1924
      }
1925
    }
1926

1927
    code = taosHashRemove(pInfo->readerCache, &keyInfo->uid, sizeof(uint64_t));
1,899,848✔
1928
    QUERY_CHECK_CODE(code, lino, _end);
1,899,848✔
1929
  }
1930
_end:
5,900,542✔
1931
  if (code) {
5,900,542✔
1932
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
617✔
1933
  }
1934
  return code;
5,900,542✔
1935
}
1936

1937
static int32_t setTagValFromTagList(SOperatorInfo* pOperator, SSDataBlock* pRes) {
×
1938
  int32_t                  code = TSDB_CODE_SUCCESS;
×
1939
  int32_t                  lino = 0;
×
1940
  STableScanInfo*          pInfo = pOperator->info;
×
1941
  int32_t                  index = 0;
×
1942
  char*                    tagVal = NULL;
×
1943

1944
  for (int32_t i = taosArrayGetSize(pRes->pDataBlock) - taosArrayGetSize(pInfo->cachedTagList); i < taosArrayGetSize(pRes->pDataBlock); i++) {
×
1945
    SColumnInfoData* pTagCol = taosArrayGet(pRes->pDataBlock, i);
×
1946
    STagVal*         pTagVal = taosArrayGet(pInfo->cachedTagList, index);
×
1947

1948
    QUERY_CHECK_NULL(pTagVal, code, lino, _end, terrno);
×
1949
    QUERY_CHECK_NULL(pTagCol, code, lino, _end, terrno);
×
1950

1951
    for(int32_t j = 0; j < pRes->info.rows; j++) {
×
1952
      if (IS_VAR_DATA_TYPE(pTagVal->type)) {
×
1953
        tagVal = taosMemoryMalloc(pTagVal->nData + VARSTR_HEADER_SIZE + 1);
×
1954
        QUERY_CHECK_NULL(tagVal, code, lino, _end, terrno);
×
1955

1956
        varDataSetLen(tagVal, pTagVal->nData);
×
1957
        memcpy(tagVal + VARSTR_HEADER_SIZE, pTagVal->pData, pTagVal->nData);
×
1958
        code = colDataSetVal(pTagCol, j, tagVal, false);
×
1959
        QUERY_CHECK_CODE(code, lino, _end);
×
1960
        taosMemoryFreeClear(tagVal);
×
1961
      } else {
1962
        code = colDataSetVal(pTagCol, j, (const char*)&pTagVal->i64, false);
×
1963
        QUERY_CHECK_CODE(code, lino, _end);
×
1964
      }
1965
    }
1966
    index++;
×
1967
  }
1968
  return code;
×
1969
_end:
×
1970
  qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1971
  taosMemoryFreeClear(tagVal);
×
1972
  return code;
×
1973
}
1974

1975
static void clearVstbBatchDynamicTableScanInfo(STableScanInfo* pInfo) {
×
1976
  if (pInfo->pBatchColMap) {
×
1977
    for (int32_t i = 0; i < taosArrayGetSize(pInfo->pBatchColMap); i++) {
×
1978
      SOrgTbInfo* pOrgTbInfo = (SOrgTbInfo*)taosArrayGet(pInfo->pBatchColMap, i);
×
1979
      taosArrayDestroy(pOrgTbInfo->colMap);
×
1980
    }
1981
    taosArrayDestroy(pInfo->pBatchColMap);
×
1982
    pInfo->pBatchColMap = NULL;
×
1983
  }
1984
  if (pInfo->cachedTagList) {
×
1985
    for (int32_t i = 0; i < taosArrayGetSize(pInfo->cachedTagList); i++) {
×
1986
      STagVal* pTagVal = (STagVal*)taosArrayGet(pInfo->cachedTagList, i);
×
1987
      if (IS_VAR_DATA_TYPE(pTagVal->type)) {
×
1988
        taosMemoryFreeClear(pTagVal->pData);
×
1989
      }
1990
    }
1991
    taosArrayDestroy(pInfo->cachedTagList);
×
1992
    pInfo->cachedTagList = NULL;
×
1993
  }
1994
  if (pInfo->lastColArray) {
×
1995
    taosArrayDestroy(pInfo->lastColArray);
×
1996
    pInfo->lastColArray = NULL;
×
1997
  }
1998
  if (pInfo->lastBlockColArray) {
×
1999
    taosArrayDestroy(pInfo->lastBlockColArray);
×
2000
    pInfo->lastBlockColArray = NULL;
×
2001
  }
2002

2003
  blockDataDestroy(pInfo->pResBlock);
×
2004
  pInfo->pResBlock = NULL;
×
2005
}
×
2006

2007
static int32_t doVstbBatchDynamicTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
×
2008
  int32_t                  code = TSDB_CODE_SUCCESS;
×
2009
  int32_t                  lino = 0;
×
2010
  STableScanInfo*          pInfo = pOperator->info;
×
2011
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
×
2012
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
×
2013
  SSDataBlock*             result = NULL;
×
2014

2015
  QRY_PARAM_CHECK(ppRes);
×
2016
  qTrace("%s call", __FUNCTION__);
×
2017

2018
  while (!result) {
×
2019
    code = createVTableScanInfoFromBatchParam(pOperator);
×
2020
    freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
×
2021
    pOperator->pOperatorGetParam = NULL;
×
2022
    QUERY_CHECK_CODE(code, lino, _end);
×
2023

2024
    if (pInfo->newReader) {
×
2025
      code = startNextGroupScan(pOperator, &result);
×
2026
      QUERY_CHECK_CODE(code, lino, _end);
×
2027
    } else {
2028
      code = doGroupedTableScan(pOperator, &result);
×
2029
      QUERY_CHECK_CODE(code, lino, _end);
×
2030
    }
2031
    if (!result) {
×
2032
      qDebug("%s, finished batch scan for current table, switch to next table, idx:%d,  %s",
×
2033
             __func__ , pInfo->currentBatchIdx, GET_TASKID(pTaskInfo));
2034
      pInfo->currentBatchIdx++;
×
2035
      if (pInfo->currentBatchIdx >= taosArrayGetSize(pInfo->pBatchColMap)) {
×
2036
        // all batch scan finished
2037
        break;
×
2038
      }
2039
    }
2040
  }
2041

2042
  if (result) {
×
2043
    SSDataBlock* res = NULL;
×
2044
    pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, NULL);
×
2045
    pInfo->lastTimeWindow.skey = result->info.window.skey;
×
2046
    pInfo->lastTimeWindow.ekey = result->info.window.ekey;
×
2047

2048
    code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, pInfo->pBlockColMap, &res);
×
2049
    QUERY_CHECK_CODE(code, lino, _end);
×
2050

2051
    if (pInfo->cachedTagList) {
×
2052
      code = setTagValFromTagList(pOperator, res);
×
2053
      QUERY_CHECK_CODE(code, lino, _end);
×
2054
    }
2055

2056
    res->info.id.groupId = pInfo->cachedGroupId;
×
2057
    pInfo->pResBlock = res;
×
2058
    blockDataDestroy(result);
×
2059
    (*ppRes) = res;
×
2060

2061
    printDataBlock(res, __func__, pTaskInfo->id.str, pTaskInfo->id.queryId);
×
2062
  } else {
2063
    clearVstbBatchDynamicTableScanInfo(pInfo);
×
2064
    setOperatorCompleted(pOperator);
×
2065
  }
2066

2067
_end:
×
2068
  return code;
×
2069
}
2070

2071
static int32_t doDynamicTableScanNext(SOperatorInfo* pOperator, ETableScanDynType type, SSDataBlock** ppRes) {
6,157,980✔
2072
  int32_t                  code = TSDB_CODE_SUCCESS;
6,157,980✔
2073
  int32_t                  lino = 0;
6,157,980✔
2074
  STableScanInfo*          pInfo = pOperator->info;
6,157,980✔
2075
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
6,157,980✔
2076
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
6,157,980✔
2077

2078
  QRY_PARAM_CHECK(ppRes);
6,157,980✔
2079
  qTrace("%s call", __FUNCTION__);
6,157,980✔
2080

2081
  pOperator->dynamicTask = true;
6,157,980✔
2082

2083
  switch (type) {
6,157,980✔
2084
    case DYN_TYPE_STB_JOIN: {
238,982✔
2085
      code = doStbJoinDynamicTableScanNext(pOperator, ppRes);
238,982✔
2086
      QUERY_CHECK_CODE(code, lino, _end);
238,982✔
2087
      break;
238,982✔
2088
    }
2089
    case DYN_TYPE_VSTB_SINGLE_SCAN: {
5,918,998✔
2090
      code = doVstbSingleDynamicTableScanNext(pOperator, ppRes);
5,918,998✔
2091
      QUERY_CHECK_CODE(code, lino, _end);
5,918,998✔
2092
      break;
5,918,381✔
2093
    }
2094
    case DYN_TYPE_VSTB_BATCH_SCAN:
×
2095
      code = doVstbBatchDynamicTableScanNext(pOperator, ppRes);
×
2096
      QUERY_CHECK_CODE(code, lino, _end);
×
2097
      break;
×
2098
    default:
×
2099
      code = TSDB_CODE_INVALID_PARA;
×
2100
      QUERY_CHECK_CODE(code, lino, _end);
×
2101
      break;
×
2102
  }
2103

2104
_end:
6,157,980✔
2105
  return code;
6,157,980✔
2106
}
2107

2108
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
784,680,261✔
2109
  int32_t         code = TSDB_CODE_SUCCESS;
784,680,261✔
2110
  int32_t         lino = 0;
784,680,261✔
2111
  STableScanInfo* pInfo = pOperator->info;
784,680,261✔
2112
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
784,873,071✔
2113
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
784,866,427✔
2114
  QRY_PARAM_CHECK(ppRes);
784,878,867✔
2115
  qTrace("%s call", __FUNCTION__);
784,819,596✔
2116

2117
  if (pOperator->pOperatorGetParam || pInfo->pBatchColMap) {
784,819,626✔
2118
    ETableScanDynType type = pOperator->pOperatorGetParam ?
6,170,589✔
2119
                                                          ((STableScanOperatorParam*)pOperator->pOperatorGetParam->value)->type :
6,157,980✔
2120
                                                          DYN_TYPE_VSTB_BATCH_SCAN;
2121

2122
    code = doDynamicTableScanNext(pOperator, type, ppRes);
6,157,980✔
2123
    QUERY_CHECK_CODE(code, lino, _end);
6,157,980✔
2124

2125
    if (type != DYN_TYPE_STB_JOIN) {
6,157,363✔
2126
      return code;
5,918,381✔
2127
    }
2128
  }
2129

2130
  // scan table one by one sequentially
2131
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
778,879,865✔
2132
    int32_t       numOfTables = 0;
5,782,181✔
2133
    STableKeyInfo tInfo = {0};
5,782,181✔
2134
    pInfo->countState = TABLE_COUNT_STATE_END;
5,782,181✔
2135

2136
    while (1) {
5,405,458✔
2137
      SSDataBlock* result = NULL;
11,187,639✔
2138
      code = doGroupedTableScan(pOperator, &result);
11,186,433✔
2139
      QUERY_CHECK_CODE(code, lino, _end);
11,183,725✔
2140

2141
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
11,183,725✔
2142
        (*ppRes) = result;
5,741,607✔
2143
        return code;
5,744,336✔
2144
      }
2145

2146
      // if no data, switch to next table and continue scan
2147
      pInfo->currentTable++;
5,441,496✔
2148

2149
      taosRLockLatch(&pTaskInfo->lock);
5,441,514✔
2150
      numOfTables = 0;
5,443,303✔
2151
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
5,443,303✔
2152
      if (code != TSDB_CODE_SUCCESS) {
5,441,798✔
2153
        taosRUnLockLatch(&pTaskInfo->lock);
×
2154
        TSDB_CHECK_CODE(code, lino, _end);
×
2155
      }
2156

2157
      if (pInfo->currentTable >= numOfTables) {
5,441,798✔
2158
        qDebug("all table checked in table list, total:%d, return NULL, %s", numOfTables, GET_TASKID(pTaskInfo));
37,544✔
2159
        taosRUnLockLatch(&pTaskInfo->lock);
37,544✔
2160
        (*ppRes) = NULL;
37,544✔
2161
        return code;
37,544✔
2162
      }
2163

2164
      STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
5,402,163✔
2165
      if (!tmp) {
5,403,651✔
2166
        taosRUnLockLatch(&pTaskInfo->lock);
×
2167
        (*ppRes) = NULL;
×
2168
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
2169
      }
2170

2171
      tInfo = *tmp;
5,403,651✔
2172
      taosRUnLockLatch(&pTaskInfo->lock);
5,401,542✔
2173

2174
      code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, &tInfo, 1);
5,403,065✔
2175
      QUERY_CHECK_CODE(code, lino, _end);
5,402,204✔
2176
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d/%d %s", tInfo.uid, numOfTables,
5,402,204✔
2177
             pInfo->currentTable, numOfTables, GET_TASKID(pTaskInfo));
2178

2179
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
5,404,851✔
2180
      QUERY_CHECK_CODE(code, lino, _end);
5,401,559✔
2181
      pInfo->scanTimes = 0;
5,401,559✔
2182
    }
2183
  } else {  // scan table group by group sequentially
2184
    code = groupSeqTableScan(pOperator, ppRes);
773,017,863✔
2185
    QUERY_CHECK_CODE(code, lino, _end);
772,120,387✔
2186
  }
2187

2188
_end:
772,118,282✔
2189
  if (code != TSDB_CODE_SUCCESS) {
772,261,936✔
2190
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
2,722✔
2191
    pTaskInfo->code = code;
2,722✔
2192
    T_LONG_JMP(pTaskInfo->env, code);
2,722✔
2193
  }
2194

2195
  return code;
772,259,214✔
2196
}
2197

2198
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
2,207,276✔
2199
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
2,207,276✔
2200
  if (!pRecorder) {
2,206,766✔
2201
    return terrno;
×
2202
  }
2203
  STableScanInfo* pTableScanInfo = pOptr->info;
2,206,766✔
2204
  *pRecorder = pTableScanInfo->base.readRecorder;
2,206,766✔
2205
  *pOptrExplain = pRecorder;
2,207,276✔
2206
  *len = sizeof(SFileBlockLoadRecorder);
2,206,761✔
2207
  return 0;
2,207,276✔
2208
}
2209

2210
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
239,648,770✔
2211
  cleanupQueryTableDataCond(&pBase->cond);
239,648,770✔
2212
  cleanupQueryTableDataCond(&pBase->orgCond);
239,648,766✔
2213

2214
  if (pAPI->tsdReaderClose) {
239,674,668✔
2215
    pAPI->tsdReaderClose(pBase->dataReader);
239,682,279✔
2216
  }
2217
  pBase->dataReader = NULL;
239,660,185✔
2218

2219
  if (pBase->matchInfo.pList != NULL) {
239,666,875✔
2220
    taosArrayDestroy(pBase->matchInfo.pList);
239,661,407✔
2221
  }
2222

2223
  tableListDestroy(pBase->pTableListInfo);
239,602,185✔
2224
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
239,674,573✔
2225
  cleanupExprSupp(&pBase->pseudoSup);
239,725,676✔
2226
}
239,719,706✔
2227

2228
static void cleanReaderForVTable(STableScanInfo* pInfo){
866,331✔
2229
  if (pInfo != NULL && pInfo->base.readerAPI.tsdReaderClose) {
866,331✔
2230
    void *pIter = taosHashIterate(pInfo->readerCache, NULL);
866,331✔
2231
    while (pIter != NULL) {
878,181✔
2232
      void **reader = pIter;
11,850✔
2233
      if (*reader) {
11,850✔
2234
        if (*reader == pInfo->base.dataReader) {
11,850✔
2235
          pInfo->base.dataReader = NULL;
11,850✔
2236
        }
2237
        pInfo->base.readerAPI.tsdReaderClose(*reader);
11,850✔
2238
      }
2239
      pIter = taosHashIterate(pInfo->readerCache, pIter);
11,850✔
2240
    }
2241
  }
2242
}
866,331✔
2243

2244
static void destroyTableScanOperatorInfo(void* param) {
216,519,128✔
2245
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
216,519,128✔
2246
  blockDataDestroy(pTableScanInfo->pResBlock);
216,519,128✔
2247
  blockDataDestroy(pTableScanInfo->pOrgBlock);
216,536,521✔
2248
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
216,557,291✔
2249
  if (pTableScanInfo->virtualStableScan && pTableScanInfo->readerCache) {
216,504,864✔
2250
    cleanReaderForVTable(pTableScanInfo);
866,331✔
2251
    taosHashCleanup(pTableScanInfo->readerCache);
866,331✔
2252
  }
2253
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
216,505,399✔
2254
  taosArrayDestroy(pTableScanInfo->pBlockColMap);
216,577,166✔
2255
  taosMemoryFreeClear(param);
216,579,542✔
2256
}
216,576,601✔
2257

2258
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
216,455,637✔
2259
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
216,455,637✔
2260
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
10,065,820✔
2261
    for (int32_t i = 0; i < numOfCols; ++i) {
24,185,555✔
2262
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
14,125,058✔
2263
      if (pCol) {
14,123,123✔
2264
        pCol->info.noData = true;
14,122,567✔
2265
      }
2266
    }
2267
  }
2268
}
216,450,314✔
2269

2270
static int32_t resetTableScanOperatorState(SOperatorInfo* pOper) {
251,232✔
2271
  int32_t         code = TSDB_CODE_SUCCESS;
251,232✔
2272
  STableScanInfo*   pInfo = pOper->info;
251,232✔
2273
  pOper->status = OP_NOT_OPENED;
251,232✔
2274

2275
  pInfo->scanTimes = 0;
251,232✔
2276
  pInfo->currentGroupId = -1;
251,232✔
2277
  pInfo->tableEndIndex = -1;
251,232✔
2278
  pInfo->tableStartIndex = 0;
251,232✔
2279
  pInfo->currentTable = 0;
251,232✔
2280
  pInfo->scanMode = 0;
251,232✔
2281
  pInfo->countState = 0;
251,232✔
2282
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
251,232✔
2283

2284
  if (!pInfo->virtualStableScan) {
251,232✔
2285
  // if (pInfo->virtualStableScan && pInfo->readerCache) {
2286
  //   cleanReaderForVTable(pInfo);
2287
  //   taosHashClear(pInfo->readerCache);
2288
  //   pInfo->readerCache = false;
2289
  // } else {
2290
    if (pInfo->base.readerAPI.tsdReaderClose) {
66,055✔
2291
      pInfo->base.readerAPI.tsdReaderClose(pInfo->base.dataReader);
66,055✔
2292
    }
2293
    pInfo->base.dataReader = NULL;
66,055✔
2294
  }
2295
  tableListDestroy(pInfo->base.pTableListInfo);
251,232✔
2296
  pInfo->base.pTableListInfo = tableListCreate();
251,117✔
2297
  if (!pInfo->base.pTableListInfo) {
251,232✔
2298
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2299
    return terrno;
×
2300
  }
2301
  SExecTaskInfo*         pTaskInfo = pOper->pTaskInfo;
251,232✔
2302

2303
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
251,232✔
2304
  if (!pTableScanNode->scan.node.dynamicOp) {
251,232✔
2305
    code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
129,187✔
2306
                                    &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
2307
                                    pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
129,187✔
2308
    if (code) {
65,095✔
2309
      qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
×
2310
      return code;
×
2311
    }
2312
  }
2313

2314
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->base.limitInfo);
251,232✔
2315
  cleanupQueryTableDataCond(&pInfo->base.cond);
251,232✔
2316
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle, true);
251,232✔
2317
  if (code) {
251,232✔
2318
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
2319
    return code;
×
2320
  }
2321
  if (pTableScanNode->scan.node.dynamicOp && pTableScanNode->scan.virtualStableScan) {
251,232✔
2322
    cleanupQueryTableDataCond(&pInfo->base.orgCond);
185,177✔
2323
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
184,443✔
2324
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
184,443✔
2325
  } 
2326

2327
  pOper->resultInfo.totalRows = 0;
250,498✔
2328
  blockDataEmpty(pInfo->pResBlock);
251,232✔
2329
  blockDataEmpty(pInfo->pOrgBlock);
251,232✔
2330
  taosHashClear(pInfo->pIgnoreTables);
251,232✔
2331
  taosArrayDestroy(pInfo->pBlockColMap);
251,232✔
2332
  pInfo->pBlockColMap = NULL;
251,232✔
2333
  return code;
251,232✔
2334
}
2335

2336
int32_t createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
216,517,203✔
2337
                                    STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
2338
                                    SOperatorInfo** pOptrInfo) {
2339
  QRY_PARAM_CHECK(pOptrInfo);
216,517,203✔
2340

2341
  int32_t         code = TSDB_CODE_SUCCESS;
216,548,915✔
2342
  int32_t         lino = 0;
216,548,915✔
2343
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
216,548,915✔
2344
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
216,464,505✔
2345
  if (pInfo == NULL || pOperator == NULL) {
216,460,444✔
2346
    code = terrno;
×
2347
    goto _error;
×
2348
  }
2349

2350
  pOperator->pPhyNode = pTableScanNode;
216,464,208✔
2351
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
216,471,212✔
2352
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
216,479,703✔
2353

2354
  int32_t numOfCols = 0;
216,478,589✔
2355
  code =
2356
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
216,500,328✔
2357
  QUERY_CHECK_CODE(code, lino, _error);
216,528,879✔
2358

2359
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
216,528,879✔
2360
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, true);
216,467,714✔
2361
  QUERY_CHECK_CODE(code, lino, _error);
216,356,086✔
2362

2363
  if (pScanNode->pScanPseudoCols != NULL) {
216,356,086✔
2364
    SExprSupp* pSup = &pInfo->base.pseudoSup;
47,423,399✔
2365
    pSup->pExprInfo = NULL;
47,431,197✔
2366
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
47,427,937✔
2367
    QUERY_CHECK_CODE(code, lino, _error);
47,411,994✔
2368

2369
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
47,411,994✔
2370
                                      &pTaskInfo->storageAPI.functionStore);
2371
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
47,414,737✔
2372
  }
2373

2374
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
216,340,759✔
2375
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
216,406,771✔
2376

2377
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
216,360,808✔
2378
  pInfo->base.readHandle = *readHandle;
216,262,776✔
2379
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
216,335,415✔
2380

2381
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
216,372,019✔
2382
  pInfo->sample.seed = taosGetTimestampSec();
216,386,080✔
2383

2384
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
216,420,903✔
2385
  initResultSizeInfo(&pOperator->resultInfo, 4096);
216,475,725✔
2386
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
216,351,309✔
2387
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
216,551,387✔
2388
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
216,402,040✔
2389

2390
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
216,384,027✔
2391
  QUERY_CHECK_CODE(code, lino, _error);
216,380,825✔
2392

2393
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
216,380,825✔
2394
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
216,327,399✔
2395
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
865,768✔
2396
    pInfo->pResBlock = NULL;
866,331✔
2397
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
865,768✔
2398
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
865,205✔
2399
  }
2400

2401
  if (pInfo->virtualStableScan) {
216,293,684✔
2402
    pInfo->readerCache = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
864,642✔
2403
    QUERY_CHECK_NULL(pInfo->readerCache, code, lino, _error, terrno);
866,331✔
2404
  }
2405

2406
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
216,412,527✔
2407
                            pTaskInfo->pStreamRuntimeInfo);
216,305,884✔
2408
  QUERY_CHECK_CODE(code, lino, _error);
216,332,831✔
2409

2410
  pInfo->currentGroupId = -1;
216,310,715✔
2411

2412
  pInfo->tableEndIndex = -1;
216,342,714✔
2413
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
216,385,041✔
2414
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
216,338,424✔
2415

2416
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
216,326,942✔
2417
                  pTaskInfo);
2418
  pOperator->exprSupp.numOfExprs = numOfCols;
216,307,325✔
2419

2420
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
216,370,177✔
2421
  pInfo->ignoreTag = false;
216,390,519✔
2422

2423
  pInfo->lastBatchIdx = -1;
216,327,119✔
2424
  pInfo->currentBatchIdx = 0;
216,398,426✔
2425
  pInfo->base.pTableListInfo = pTableListInfo;
216,259,922✔
2426
  if (readHandle->streamRtInfo == NULL) {
216,334,508✔
2427
    pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
216,283,175✔
2428
    if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
216,415,659✔
2429
      code = terrno;
×
2430
      QUERY_CHECK_CODE(code, lino, _error);
×
2431
    }
2432
    taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
216,426,477✔
2433
  }
2434

2435
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
216,440,262✔
2436

2437
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
216,523,899✔
2438
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
2439

2440
  setOperatorResetStateFn(pOperator, resetTableScanOperatorState); 
216,506,482✔
2441
  // for non-blocking operator, the open cost is always 0
2442
  pOperator->cost.openCost = 0;
216,475,472✔
2443
  *pOptrInfo = pOperator;
216,436,434✔
2444
  return TSDB_CODE_SUCCESS;
216,429,672✔
2445

2446
_error:
22,116✔
2447
  if (pInfo != NULL) {
22,116✔
2448
    pInfo->base.pTableListInfo = NULL;  // this attribute will be destroy outside of this function
22,116✔
2449
    destroyTableScanOperatorInfo(pInfo);
22,116✔
2450
  }
2451

2452
  if (pOperator != NULL) {
22,116✔
2453
    pOperator->info = NULL;
22,116✔
2454
    destroyOperator(pOperator);
22,116✔
2455
  }
2456
  pTaskInfo->code = code;
22,116✔
2457
  return code;
22,116✔
2458
}
2459

2460
int32_t createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
×
2461
  QRY_PARAM_CHECK(pOptrInfo);
×
2462

2463
  int32_t         code = 0;
×
2464
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
×
2465
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
2466
  if (pInfo == NULL || pOperator == NULL) {
×
2467
    code = terrno;
×
2468
    goto _end;
×
2469
  }
2470

2471
  pInfo->base.dataReader = pReadHandle;
×
2472
  //  pInfo->prevGroupId       = -1;
2473

2474
  setOperatorInfo(pOperator, "TableSeqScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN, false, OP_NOT_OPENED,
×
2475
                  pInfo, pTaskInfo);
2476
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanImplNext, NULL, NULL, optrDefaultBufFn, NULL,
×
2477
                                         optrDefaultGetNextExtFn, NULL);
2478
  *pOptrInfo = pOperator;
×
2479
  return code;
×
2480

2481
_end:
×
2482
  if (pInfo != NULL) {
×
2483
    taosMemoryFree(pInfo);
×
2484
  }
2485

2486
  if (pOperator != NULL) {
×
2487
    taosMemoryFree(pOperator);
×
2488
  }
2489

2490
  pTaskInfo->code = code;
×
2491
  return code;
×
2492
}
2493

2494
static int32_t doBlockDataPrimaryKeyFilter(SSDataBlock* pBlock, STqOffsetVal* offset) {
13,177✔
2495
  int32_t code = TSDB_CODE_SUCCESS;
13,177✔
2496
  int32_t lino = 0;
13,177✔
2497
  if (pBlock->info.window.skey != offset->ts || offset->primaryKey.type == 0) {
13,177✔
2498
    return code;
1,577✔
2499
  }
2500
  bool* p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
11,600✔
2501
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
11,600✔
2502
  bool hasUnqualified = false;
11,600✔
2503

2504
  SColumnInfoData* pColTs = taosArrayGet(pBlock->pDataBlock, 0);
11,600✔
2505
  SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
11,600✔
2506

2507
  qDebug("doBlockDataWindowFilter primary key, ts:%" PRId64 " %" PRId64, offset->ts,
11,600✔
2508
         VALUE_GET_TRIVIAL_DATUM(&offset->primaryKey));
2509
  QUERY_CHECK_CONDITION((pColPk->info.type == offset->primaryKey.type), code, lino, _end,
11,600✔
2510
                        TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
2511

2512
  __compar_fn_t func = getComparFunc(pColPk->info.type, 0);
11,600✔
2513
  QUERY_CHECK_NULL(func, code, lino, _end, terrno);
11,600✔
2514
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
41,731,392✔
2515
    int64_t* ts = (int64_t*)colDataGetData(pColTs, i);
41,719,792✔
2516
    void*    data = colDataGetData(pColPk, i);
41,719,792✔
2517
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
41,722,040✔
2518
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
2,248✔
2519
        QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY, lino, _end);
×
2520
      }
2521
      void* tmq = taosMemoryMalloc(offset->primaryKey.nData + VARSTR_HEADER_SIZE);
2,248✔
2522
      QUERY_CHECK_NULL(tmq, code, lino, _end, terrno);
2,248✔
2523
      memcpy(varDataVal(tmq), offset->primaryKey.pData, offset->primaryKey.nData);
2,248✔
2524
      varDataLen(tmq) = offset->primaryKey.nData;
2,248✔
2525
      p[i] = (*ts > offset->ts) || (func(data, tmq) > 0);
2,248✔
2526
      taosMemoryFree(tmq);
2,248✔
2527
    } else {
2528
      p[i] = (*ts > offset->ts) || (func(data, VALUE_GET_DATUM(&offset->primaryKey, pColPk->info.type)) > 0);
41,717,544✔
2529
    }
2530

2531
    if (!p[i]) {
41,719,792✔
2532
      hasUnqualified = true;
14,856,754✔
2533
    }
2534
  }
2535

2536
  if (hasUnqualified) {
11,600✔
2537
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
3,908✔
2538
    QUERY_CHECK_CODE(code, lino, _end);
3,908✔
2539
  }
2540

2541
_end:
11,600✔
2542
  taosMemoryFree(p);
11,600✔
2543
  if (code != TSDB_CODE_SUCCESS) {
11,600✔
2544
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2545
  }
2546
  return code;
11,600✔
2547
}
2548

2549
int32_t colIdComparFn(const void* param1, const void* param2) {
2,147,483,647✔
2550
  int32_t p1 = *(int32_t*)param1;
2,147,483,647✔
2551
  int32_t p2 = *(int32_t*)param2;
2,147,483,647✔
2552

2553
  if (p1 == p2) {
2,147,483,647✔
2554
    return 0;
×
2555
  } else {
2556
    return (p1 < p2) ? -1 : 1;
2,147,483,647✔
2557
  }
2558
}
2559

2560
static int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock) {
115,017,769✔
2561
  int32_t         code = TSDB_CODE_SUCCESS;
115,017,769✔
2562
  int32_t         lino = 0;
115,017,769✔
2563
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
115,017,769✔
2564
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
115,024,941✔
2565
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
115,024,941✔
2566
  const char*     id = GET_TASKID(pTaskInfo);
115,025,919✔
2567
  SArray* pColList = NULL;
115,024,941✔
2568

2569
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
115,024,941✔
2570
  QUERY_CHECK_CODE(code, lino, _end);
115,022,659✔
2571

2572
  pBlockInfo->rows = pBlock->info.rows;
115,022,659✔
2573
  pBlockInfo->id.uid = pBlock->info.id.uid;
115,021,029✔
2574
  pBlockInfo->type = STREAM_NORMAL;
115,022,659✔
2575
  pBlockInfo->version = pBlock->info.version;
115,023,311✔
2576

2577
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
115,020,377✔
2578
  pBlockInfo->id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
115,022,007✔
2579

2580
  pColList = taosArrayInit(4, sizeof(int32_t));
115,024,941✔
2581
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
115,017,113✔
2582

2583
  // todo extract method
2584
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
727,233,502✔
2585
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
612,282,078✔
2586
    if (!pColMatchInfo->needOutput) {
612,218,892✔
2587
      continue;
626✔
2588
    }
2589

2590
    bool colExists = false;
612,271,921✔
2591
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
2,147,483,647✔
2592
      SColumnInfoData* pResCol = NULL;
2,147,483,647✔
2593
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
2,147,483,647✔
2594
      QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647✔
2595

2596
      if (pResCol->info.colId == pColMatchInfo->colId) {
2,147,483,647✔
2597
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
612,337,342✔
2598
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
612,311,306✔
2599
        QUERY_CHECK_CODE(code, lino, _end);
612,161,046✔
2600

2601
        colExists = true;
612,161,046✔
2602
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
612,161,046✔
2603
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
612,236,931✔
2604
        break;
612,236,931✔
2605
      }
2606
    }
2607

2608
    // the required column does not exists in submit block, let's set it to be all null value
2609
    if (!colExists) {
612,215,763✔
2610
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
×
2611
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
2612
      void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
×
2613
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
2614
    }
2615
  }
2616

2617
  // currently only the tbname pseudo column
2618
  if (pInfo->numOfPseudoExpr > 0) {
115,016,856✔
2619
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
161,171,763✔
2620
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
80,585,961✔
2621
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
2622
    if (code) {
80,584,983✔
2623
      QUERY_CHECK_CODE(code, lino, _end);
×
2624
    }
2625

2626
    // reset the error code.
2627
    terrno = 0;
80,584,983✔
2628

2629
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
427,114,077✔
2630
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
346,523,878✔
2631
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
346,535,940✔
2632
    }
2633
  }
2634

2635
  taosArraySort(pColList, colIdComparFn);
114,957,935✔
2636

2637
  int32_t i = 0, j = 0;
115,018,155✔
2638
  while (i < taosArrayGetSize(pColList)) {
1,073,495,179✔
2639
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
958,465,055✔
2640
    if (slot1 > j) {
958,477,024✔
2641
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
×
2642
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
2643
      j += 1;
×
2644
    } else {
2645
      i += 1;
958,477,024✔
2646
      j += 1;
958,477,024✔
2647
    }
2648
  }
2649

2650
  while (j < taosArrayGetSize(pInfo->pRes->pDataBlock)) {
115,013,182✔
2651
    SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
626✔
2652
    colDataSetNNULL(pDst, 0, pBlockInfo->rows);
626✔
2653
    j += 1;
626✔
2654
  }
2655

2656
  code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL, NULL);
115,018,441✔
2657
  QUERY_CHECK_CODE(code, lino, _end);
115,017,330✔
2658

2659
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
115,017,330✔
2660
  QUERY_CHECK_CODE(code, lino, _end);
115,013,680✔
2661

2662
_end:
115,013,680✔
2663
  taosArrayDestroy(pColList);
113,530,513✔
2664
  if (code != TSDB_CODE_SUCCESS) {
115,014,726✔
2665
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2666
  }
2667
  return code;
115,003,327✔
2668
}
2669

2670
static int32_t processPrimaryKey(SSDataBlock* pBlock, bool hasPrimaryKey, STqOffsetVal* offset) {
6,447,534✔
2671
  int32_t code = TSDB_CODE_SUCCESS;
6,447,534✔
2672
  int32_t lino = 0;
6,447,534✔
2673
  SValue  val = {0};
6,447,534✔
2674
  if (hasPrimaryKey) {
6,448,407✔
2675
    code = doBlockDataPrimaryKeyFilter(pBlock, offset);
13,177✔
2676
    if (code != TSDB_CODE_SUCCESS) {
13,177✔
2677
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2678
      return code;
×
2679
    }
2680
    SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
13,177✔
2681

2682
    if (pBlock->info.rows < 1) {
13,177✔
2683
      return code;
3,627✔
2684
    }
2685
    void* tmp = colDataGetData(pColPk, pBlock->info.rows - 1);
9,550✔
2686
    val.type = pColPk->info.type;
9,550✔
2687
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
9,550✔
2688
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
605✔
2689
        return TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY;
×
2690
      }
2691
      val.pData = taosMemoryMalloc(varDataLen(tmp));
605✔
2692
      QUERY_CHECK_NULL(val.pData, code, lino, _end, terrno);
605✔
2693
      val.nData = varDataLen(tmp);
605✔
2694
      memcpy(val.pData, varDataVal(tmp), varDataLen(tmp));
605✔
2695
    } else {
2696
      valueSetDatum(&val, pColPk->info.type, tmp, pColPk->info.bytes);
8,945✔
2697
    }
2698
  }
2699
  tqOffsetResetToData(offset, pBlock->info.id.uid, pBlock->info.window.ekey, val);
6,444,780✔
2700

2701
_end:
6,443,022✔
2702
  if (code != TSDB_CODE_SUCCESS) {
6,443,323✔
2703
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2704
  }
2705
  return code;
6,442,441✔
2706
}
2707

2708
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
143,867,897✔
2709
  int32_t        code = TSDB_CODE_SUCCESS;
143,867,897✔
2710
  int32_t        lino = 0;
143,867,897✔
2711
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
143,867,897✔
2712
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
143,873,063✔
2713

2714
  SStreamScanInfo* pInfo = pOperator->info;
143,877,928✔
2715
  const char*      id = GET_TASKID(pTaskInfo);
143,878,906✔
2716

2717
  qDebug("start to exec queue scan, %s", id);
143,877,627✔
2718

2719
  if (isTaskKilled(pTaskInfo)) {
143,866,167✔
2720
    (*ppRes) = NULL;
×
2721
    return pTaskInfo->code;
×
2722
  }
2723

2724
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
143,878,580✔
2725
    while (1) {
3,627✔
2726
      SSDataBlock* pResult = NULL;
5,782,181✔
2727
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
5,782,181✔
2728
      QUERY_CHECK_CODE(code, lino, _end);
5,781,880✔
2729

2730
      if (pResult && pResult->info.rows > 0) {
5,781,880✔
2731
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
5,744,637✔
2732
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
5,744,637✔
2733
        QUERY_CHECK_CODE(code, lino, _end);
5,742,566✔
2734
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
5,742,566✔
2735
        if (pResult->info.rows > 0) {
5,744,071✔
2736
          (*ppRes) = pResult;
5,741,010✔
2737
          return code;
5,741,010✔
2738
        }
2739
      } else {
2740
        break;
2741
      }
2742
    }
2743

2744
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
37,544✔
2745
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
37,544✔
2746

2747
    pTSInfo->base.dataReader = NULL;
37,544✔
2748
    int64_t validVer = pTaskInfo->streamInfo.snapshotVer + 1;
37,544✔
2749
    qDebug("queue scan tsdb over, switch to wal ver %" PRId64, validVer);
37,544✔
2750
    if (pAPI->tqReaderFn.tqReaderSeek(pInfo->tqReader, validVer, pTaskInfo->id.str) < 0) {
37,544✔
2751
      (*ppRes) = NULL;
7,030✔
2752
      return code;
7,030✔
2753
    }
2754

2755
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, validVer);
30,514✔
2756
  }
2757

2758
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
138,131,192✔
2759
    while (1) {
727,420✔
2760
      bool hasResult =
2761
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
138,858,286✔
2762

2763
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
138,825,527✔
2764
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
138,848,830✔
2765

2766
      // curVersion move to next
2767
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pWalReader->curVersion);
138,847,865✔
2768

2769
      // use ts to pass time when replay, because ts not used if type is log
2770
      pTaskInfo->streamInfo.currentOffset.ts = pAPI->tqReaderFn.tqGetResultBlockTime(pInfo->tqReader);
138,836,095✔
2771

2772
      if (hasResult) {
138,829,166✔
2773
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
114,994,191✔
2774
               pTaskInfo->streamInfo.currentOffset.version);
2775
        blockDataCleanup(pInfo->pRes);
115,012,724✔
2776
        code = setBlockIntoRes(pInfo, pRes);
115,021,355✔
2777
        QUERY_CHECK_CODE(code, lino, _end);
114,993,580✔
2778
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
114,993,580✔
2779
               pTaskInfo->streamInfo.currentOffset.version);
2780
        if (pInfo->pRes->info.rows > 0) {
115,000,023✔
2781
          (*ppRes) = pInfo->pRes;
114,295,891✔
2782
          return code;
114,293,935✔
2783
        }
2784
      } else {
2785
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
23,834,975✔
2786
        (*ppRes) = NULL;
23,834,975✔
2787
        return code;
23,834,975✔
2788
      }
2789
    }
2790
  } else {
2791
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
×
2792
    (*ppRes) = NULL;
×
2793
    return code;
×
2794
  }
2795

2796
_end:
×
2797
  if (code != TSDB_CODE_SUCCESS) {
×
2798
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2799
    pTaskInfo->code = code;
×
2800
    T_LONG_JMP(pTaskInfo->env, code);
×
2801
  }
2802
  (*ppRes) = NULL;
×
2803
  return code;
×
2804
}
2805

2806
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
×
2807
  SSDataBlock* pRes = NULL;
×
2808
  int32_t      code = doQueueScanNext(pOperator, &pRes);
×
2809
  return pRes;
×
2810
}
2811

2812
int32_t extractTableIdList(const STableListInfo* pTableListInfo, SArray** ppArrayRes) {
308,562✔
2813
  int32_t code = TSDB_CODE_SUCCESS;
308,562✔
2814
  int32_t lino = 0;
308,562✔
2815
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));
308,562✔
2816
  QUERY_CHECK_NULL(tableIdList, code, lino, _end, terrno);
310,000✔
2817

2818
  // Transfer the Array of STableKeyInfo into uid list.
2819
  int32_t size = 0;
310,000✔
2820
  code = tableListGetSize(pTableListInfo, &size);
309,696✔
2821
  QUERY_CHECK_CODE(code, lino, _end);
310,000✔
2822
  for (int32_t i = 0; i < size; ++i) {
9,362,622✔
2823
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
9,052,622✔
2824
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
9,052,014✔
2825
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
9,052,014✔
2826
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
9,052,622✔
2827
  }
2828

2829
  (*ppArrayRes) = tableIdList;
310,000✔
2830

2831
_end:
309,719✔
2832
  if (code != TSDB_CODE_SUCCESS) {
309,719✔
2833
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2834
  }
2835
  return code;
309,719✔
2836
}
2837

2838
static int32_t doRawScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
1,422,252✔
2839
  int32_t        code = TSDB_CODE_SUCCESS;
1,422,252✔
2840
  int32_t        lino = 0;
1,422,252✔
2841
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
1,422,252✔
2842
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
1,422,252✔
2843

2844
  SStreamRawScanInfo* pInfo = pOperator->info;
1,422,252✔
2845
  pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = NULL;  // use batchMetaReq != NULL to judge if data is meta
1,422,252✔
2846
  pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = NULL;
1,421,969✔
2847

2848
  qDebug("tmqsnap doRawScan called");
1,421,969✔
2849
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
1,422,252✔
2850
    bool hasNext = false;
1,417,485✔
2851
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
1,417,485✔
2852
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
1,416,644✔
2853
      if (code != TSDB_CODE_SUCCESS) {
1,415,512✔
2854
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2855
        QUERY_CHECK_CODE(code, lino, _end);
×
2856
      }
2857
    }
2858

2859
    if (pInfo->dataReader && hasNext) {
1,416,353✔
2860
      if (isTaskKilled(pTaskInfo)) {
704,336✔
2861
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2862
        return code;
×
2863
      }
2864

2865
      SSDataBlock* pBlock = NULL;
704,619✔
2866
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock);
704,053✔
2867
      QUERY_CHECK_CODE(code, lino, _end);
704,053✔
2868

2869
      if (pBlock && pBlock->info.rows > 0) {
704,053✔
2870
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
704,336✔
2871
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
704,330✔
2872
        QUERY_CHECK_CODE(code, lino, _end);
702,915✔
2873
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
702,915✔
2874
        (*ppRes) = pBlock;
704,613✔
2875
        return code;
704,619✔
2876
      }
2877
    }
2878

2879
    SMetaTableInfo mtInfo = {0};
712,300✔
2880
    code = pAPI->snapshotFn.getMetaTableInfoFromSnapshot(pInfo->sContext, &mtInfo);
712,866✔
2881
    QUERY_CHECK_CODE(code, lino, _end);
712,583✔
2882
    if (code != 0) {
712,583✔
2883
      destroyMetaTableInfo(&mtInfo);
2884
      QUERY_CHECK_CODE(code, lino, _end);
×
2885
    }
2886
    STqOffsetVal offset = {0};
712,583✔
2887
    if (mtInfo.uid == 0 || pInfo->sContext->withMeta == ONLY_META) {  // read snapshot done, change to get data from wal
712,300✔
2888
      qDebug("tmqsnap read snapshot done, change to get data from wal");
8,556✔
2889
      tqOffsetResetToLog(&offset, pInfo->sContext->snapVersion + 1);
8,556✔
2890
    } else {
2891
      SValue val = {0};
703,744✔
2892
      tqOffsetResetToData(&offset, mtInfo.uid, INT64_MIN, val);
703,744✔
2893
      qDebug("tmqsnap change get data uid:%" PRId64, mtInfo.uid);
703,178✔
2894
    }
2895
    destroyMetaTableInfo(&mtInfo);
2896
    code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
712,866✔
2897
    QUERY_CHECK_CODE(code, lino, _end);
712,866✔
2898
    (*ppRes) = NULL;
712,866✔
2899
    return code;
712,866✔
2900
  } else if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_META) {
4,767✔
2901
    SSnapContext* sContext = pInfo->sContext;
4,767✔
2902
    for (int32_t i = 0; i < tmqRowSize; i++) {
40,030✔
2903
      void*   data = NULL;
40,030✔
2904
      int32_t dataLen = 0;
40,030✔
2905
      int16_t type = 0;
40,030✔
2906
      int64_t uid = 0;
40,030✔
2907
      if (pAPI->snapshotFn.getTableInfoFromSnapshot(sContext, &data, &dataLen, &type, &uid) < 0) {
40,030✔
2908
        qError("tmqsnap getTableInfoFromSnapshot error");
×
2909
        taosMemoryFreeClear(data);
×
2910
        break;
×
2911
      }
2912

2913
      if (!sContext->queryMeta) {  // change to get data next poll request
40,030✔
2914
        STqOffsetVal offset = {0};
4,767✔
2915
        SValue       val = {0};
4,767✔
2916
        tqOffsetResetToData(&offset, 0, INT64_MIN, val);
×
2917
        code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
4,767✔
2918
        QUERY_CHECK_CODE(code, lino, _end);
4,767✔
2919
        break;
4,767✔
2920
      } else {
2921
        tqOffsetResetToMeta(&pTaskInfo->streamInfo.currentOffset, uid);
35,263✔
2922
        SMqMetaRsp tmpMetaRsp = {0};
35,263✔
2923
        tmpMetaRsp.resMsgType = type;
35,263✔
2924
        tmpMetaRsp.metaRspLen = dataLen;
35,263✔
2925
        tmpMetaRsp.metaRsp = data;
35,263✔
2926
        if (!pTaskInfo->streamInfo.btMetaRsp.batchMetaReq) {
35,263✔
2927
          pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = taosArrayInit(4, POINTER_BYTES);
4,235✔
2928
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, code, lino, _end, terrno);
4,235✔
2929

2930
          pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = taosArrayInit(4, sizeof(int32_t));
4,235✔
2931
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, code, lino, _end, terrno);
4,235✔
2932
        }
2933
        int32_t  tempRes = TSDB_CODE_SUCCESS;
35,263✔
2934
        uint32_t len = 0;
35,263✔
2935
        tEncodeSize(tEncodeMqMetaRsp, &tmpMetaRsp, len, tempRes);
35,263✔
2936
        if (TSDB_CODE_SUCCESS != tempRes) {
35,263✔
2937
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2938
          taosMemoryFreeClear(data);
×
2939
          break;
×
2940
        }
2941

2942
        int32_t tLen = sizeof(SMqRspHead) + len;
35,263✔
2943
        void*   tBuf = taosMemoryCalloc(1, tLen);
35,263✔
2944
        QUERY_CHECK_NULL(tBuf, code, lino, _end, terrno);
35,263✔
2945

2946
        void*    metaBuff = POINTER_SHIFT(tBuf, sizeof(SMqRspHead));
35,263✔
2947
        SEncoder encoder = {0};
35,263✔
2948
        tEncoderInit(&encoder, metaBuff, len);
35,263✔
2949
        int32_t tempLen = tEncodeMqMetaRsp(&encoder, &tmpMetaRsp);
35,263✔
2950
        if (tempLen < 0) {
35,263✔
2951
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2952
          tEncoderClear(&encoder);
×
2953
          taosMemoryFreeClear(tBuf);
×
2954
          taosMemoryFreeClear(data);
×
2955
          break;
×
2956
        }
2957
        taosMemoryFreeClear(data);
35,263✔
2958
        void* tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, &tBuf);
35,529✔
2959
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
35,263✔
2960

2961
        tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, &tLen);
35,263✔
2962
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
35,263✔
2963
      }
2964
    }
2965

2966
    (*ppRes) = NULL;
4,767✔
2967
    return code;
4,767✔
2968
  }
2969

2970
_end:
×
2971
  if (code != TSDB_CODE_SUCCESS) {
×
2972
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2973
    pTaskInfo->code = code;
×
2974
    T_LONG_JMP(pTaskInfo->env, code);
×
2975
  }
2976

2977
  (*ppRes) = NULL;
×
2978
  return code;
×
2979
}
2980

2981
static void destroyTmqRawScanOperatorInfo(void* param) {
112,191✔
2982
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
112,191✔
2983
  pRawScan->pAPI->tsdReader.tsdReaderClose(pRawScan->dataReader);
112,191✔
2984
  pRawScan->pAPI->snapshotFn.destroySnapshot(pRawScan->sContext);
112,191✔
2985
  tableListDestroy(pRawScan->pTableListInfo);
112,191✔
2986
  taosMemoryFree(pRawScan);
112,191✔
2987
}
112,191✔
2988

2989
// for subscribing db or stb (not including column),
2990
// if this scan is used, meta data can be return
2991
// and schemas are decided when scanning
2992
int32_t createTmqRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
112,191✔
2993
  // create operator
2994
  // create tb reader
2995
  // create meta reader
2996
  // create tq reader
2997

2998
  QRY_PARAM_CHECK(pOptrInfo);
112,191✔
2999
  int32_t code = TSDB_CODE_SUCCESS;
112,191✔
3000
  int32_t lino = 0;
112,191✔
3001

3002
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
112,191✔
3003
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
112,191✔
3004
  if (pInfo == NULL || pOperator == NULL) {
112,191✔
3005
    code = terrno;
×
3006
    lino = __LINE__;
×
3007
    goto _end;
×
3008
  }
3009

3010
  pInfo->pTableListInfo = tableListCreate();
112,191✔
3011
  QUERY_CHECK_NULL(pInfo->pTableListInfo, code, lino, _end, terrno);
112,191✔
3012
  pInfo->vnode = pHandle->vnode;
112,191✔
3013
  pInfo->pAPI = &pTaskInfo->storageAPI;
112,191✔
3014

3015
  pInfo->sContext = pHandle->sContext;
112,191✔
3016
  pHandle->sContext = NULL;  // transfer the ownership
112,191✔
3017
  setOperatorInfo(pOperator, "RawScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
112,191✔
3018
                  pTaskInfo);
3019

3020
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScanNext, NULL, destroyTmqRawScanOperatorInfo, optrDefaultBufFn,
111,925✔
3021
                                         NULL, optrDefaultGetNextExtFn, NULL);
3022
  *pOptrInfo = pOperator;
111,925✔
3023
  return code;
112,191✔
3024

3025
_end:
×
3026
  if (code != TSDB_CODE_SUCCESS) {
×
3027
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3028
  }
3029
  taosMemoryFree(pInfo);
×
3030
  taosMemoryFree(pOperator);
×
3031
  pTaskInfo->code = code;
×
3032
  return code;
×
3033
}
3034

3035
void destroyTmqScanOperatorInfo(void* param) {
310,000✔
3036
  if (param == NULL) {
310,000✔
3037
    return;
×
3038
  }
3039

3040
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
310,000✔
3041
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
310,000✔
3042
    destroyOperator(pStreamScan->pTableScanOp);
310,000✔
3043
  }
3044

3045
  if (pStreamScan->tqReader != NULL && pStreamScan->readerFn.tqReaderClose != NULL) {
310,000✔
3046
    pStreamScan->readerFn.tqReaderClose(pStreamScan->tqReader);
309,717✔
3047
  }
3048
  if (pStreamScan->pVtableMergeHandles) {
310,566✔
3049
    taosHashCleanup(pStreamScan->pVtableMergeHandles);
×
3050
    pStreamScan->pVtableMergeHandles = NULL;
×
3051
  }
3052
  if (pStreamScan->pVtableMergeBuf) {
310,000✔
3053
    destroyDiskbasedBuf(pStreamScan->pVtableMergeBuf);
×
3054
    pStreamScan->pVtableMergeBuf = NULL;
×
3055
  }
3056
  if (pStreamScan->pVtableReadyHandles) {
310,000✔
3057
    taosArrayDestroy(pStreamScan->pVtableReadyHandles);
×
3058
    pStreamScan->pVtableReadyHandles = NULL;
×
3059
  }
3060
  if (pStreamScan->pTableListInfo) {
310,000✔
3061
    tableListDestroy(pStreamScan->pTableListInfo);
×
3062
    pStreamScan->pTableListInfo = NULL;
×
3063
  }
3064
  if (pStreamScan->matchInfo.pList) {
310,000✔
3065
    taosArrayDestroy(pStreamScan->matchInfo.pList);
310,000✔
3066
  }
3067
  if (pStreamScan->pPseudoExpr) {
310,000✔
3068
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
177,158✔
3069
    taosMemoryFree(pStreamScan->pPseudoExpr);
177,158✔
3070
  }
3071

3072
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
310,000✔
3073
  cleanupExprSupp(&pStreamScan->tagCalSup);
310,000✔
3074

3075
  blockDataDestroy(pStreamScan->pRes);
310,000✔
3076
  blockDataDestroy(pStreamScan->pUpdateRes);
310,000✔
3077
  blockDataDestroy(pStreamScan->pDeleteDataRes);
310,000✔
3078
  blockDataDestroy(pStreamScan->pUpdateDataRes);
309,717✔
3079
  blockDataDestroy(pStreamScan->pCreateTbRes);
309,717✔
3080
  taosArrayDestroy(pStreamScan->pBlockLists);
310,000✔
3081
  blockDataDestroy(pStreamScan->pCheckpointRes);
309,667✔
3082

3083
  taosMemoryFree(pStreamScan);
310,000✔
3084
}
3085

3086
int32_t addPrimaryKeyCol(SSDataBlock* pBlock, uint8_t type, int32_t bytes) {
×
3087
  int32_t code = TSDB_CODE_SUCCESS;
×
3088
  int32_t lino = 0;
×
3089
  pBlock->info.rowSize += bytes;
×
3090
  SColumnInfoData infoData = {0};
×
3091
  infoData.info.type = type;
×
3092
  infoData.info.bytes = bytes;
×
3093
  void* tmp = taosArrayPush(pBlock->pDataBlock, &infoData);
×
3094
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
3095

3096
_end:
×
3097
  if (code != TSDB_CODE_SUCCESS) {
×
3098
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3099
  }
3100
  return code;
×
3101
}
3102

3103
 int32_t createTmqScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode,
309,101✔
3104
                                                  SNode* pTagCond, STableListInfo* pTableListInfo,
3105
                                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
3106
  QRY_PARAM_CHECK(pOptrInfo);
309,101✔
3107

3108
  int32_t          code = TSDB_CODE_SUCCESS;
310,000✔
3109
  int32_t          lino = 0;
310,000✔
3110
  SArray*          pColIds = NULL;
310,000✔
3111
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
310,000✔
3112
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
310,000✔
3113
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
310,000✔
3114
  const char*      idstr = pTaskInfo->id.str;
310,000✔
3115

3116
  if (pInfo == NULL || pOperator == NULL) {
310,000✔
3117
    code = terrno;
×
3118
    goto _error;
×
3119
  }
3120

3121
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
310,000✔
3122
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
310,000✔
3123

3124
  pInfo->pTagCond = pTagCond;
310,000✔
3125
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
310,000✔
3126

3127
  int32_t numOfCols = 0;
310,000✔
3128
  code = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
310,000✔
3129
  if (code != TSDB_CODE_SUCCESS) {
310,000✔
3130
    goto _error;
×
3131
  }
3132

3133
  SDataType pkType = {0};
310,000✔
3134
  pInfo->primaryKeyIndex = -1;
310,000✔
3135
  pInfo->basic.primaryPkIndex = -1;
310,000✔
3136
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
310,000✔
3137
  pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
310,000✔
3138
  QUERY_CHECK_NULL(pColIds, code, lino, _error, terrno);
310,000✔
3139

3140
  for (int32_t i = 0; i < numOfOutput; ++i) {
1,954,324✔
3141
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
1,644,324✔
3142
    QUERY_CHECK_NULL(id, code, lino, _error, terrno);
1,644,324✔
3143

3144
    int16_t colId = id->colId;
1,644,324✔
3145
    void*   tmp = taosArrayPush(pColIds, &colId);
1,644,324✔
3146
    QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
1,644,324✔
3147

3148
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
1,644,324✔
3149
      pInfo->primaryTsIndex = id->dstSlotId;
310,000✔
3150
    }
3151
    if (id->isPk) {
1,644,324✔
3152
      pInfo->primaryKeyIndex = id->dstSlotId;
1,444✔
3153
      pInfo->basic.primaryPkIndex = id->dstSlotId;
1,444✔
3154
      pkType = id->dataType;
1,444✔
3155
    }
3156
  }
3157

3158
  pInfo->pPartTbnameSup = NULL;
310,000✔
3159
  if (pTableScanNode->pSubtable != NULL) {
310,000✔
3160
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
×
3161
    if (pSubTableExpr == NULL) {
×
3162
      code = terrno;
×
3163
      goto _error;
×
3164
    }
3165

3166
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
×
3167
    code = createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
×
3168
    QUERY_CHECK_CODE(code, lino, _error);
×
3169

3170
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1, &pTaskInfo->storageAPI.functionStore) != 0) {
×
3171
      goto _error;
×
3172
    }
3173
  }
3174

3175
  if (pTableScanNode->pTags != NULL) {
310,000✔
3176
    int32_t    numOfTags;
×
3177
    SExprInfo* pTagExpr = createExpr(pTableScanNode->pTags, &numOfTags);
×
3178
    if (pTagExpr == NULL) {
×
3179
      goto _error;
×
3180
    }
3181
    code = initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags, &pTaskInfo->storageAPI.functionStore);
×
3182
    if (code != 0) {
×
3183
      goto _error;
×
3184
    }
3185
  }
3186

3187
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
310,000✔
3188
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
310,000✔
3189

3190
  if (pHandle->vnode) {
310,000✔
3191
    SOperatorInfo* pTableScanOp = NULL;
310,000✔
3192
    code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pTableScanOp);
310,000✔
3193
    if (pTableScanOp == NULL || code != 0) {
309,438✔
UNCOV
3194
      qError("createTableScanOperatorInfo error, code:%d", pTaskInfo->code);
×
3195
      goto _error;
×
3196
    }
3197

3198
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
309,719✔
3199
    if (pHandle->version > 0) {
309,719✔
3200
      pTSInfo->base.cond.endVersion = pHandle->version;
309,719✔
3201
    }
3202

3203
    STableKeyInfo* pList = NULL;
310,000✔
3204
    int32_t        num = 0;
310,000✔
3205
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
310,000✔
3206
    QUERY_CHECK_CODE(code, lino, _error);
309,719✔
3207

3208
    if (pHandle->initTableReader) {
309,719✔
3209
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
309,719✔
3210
      pTSInfo->base.dataReader = NULL;
309,719✔
3211
    }
3212

3213
    if (pHandle->initTqReader) {
309,438✔
3214
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
309,438✔
3215
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
310,000✔
3216
    } else {
3217
      pInfo->tqReader = pHandle->tqReader;
×
3218
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3219
    }
3220

3221
    pInfo->pUpdateInfo = NULL;
309,667✔
3222
    pInfo->pTableScanOp = pTableScanOp;
310,000✔
3223
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
309,667✔
3224
      pAPI->stateStore.streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1,
×
3225
                                            pInfo->primaryTsIndex);
3226
    }
3227

3228
    pInfo->readHandle = *pHandle;
310,000✔
3229
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
310,000✔
3230
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
310,000✔
3231
    QUERY_CHECK_NULL(pInfo->pCreateTbRes, code, lino, _error, terrno);
309,696✔
3232
    pInfo->hasPart = false;
309,105✔
3233

3234
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
309,719✔
3235
    QUERY_CHECK_CODE(code, lino, _error);
309,409✔
3236

3237
    // set the extract column id to streamHandle
3238
    code = pAPI->tqReaderFn.tqReaderSetColIdList(pInfo->tqReader, pColIds, idstr);
309,409✔
3239
    QUERY_CHECK_CODE(code, lino, _error);
310,000✔
3240

3241
    SArray* tableIdList = NULL;
310,000✔
3242
    code = extractTableIdList(((STableScanInfo*)(pInfo->pTableScanOp->info))->base.pTableListInfo, &tableIdList);
310,000✔
3243
    QUERY_CHECK_CODE(code, lino, _error);
309,719✔
3244
    code = pAPI->tqReaderFn.tqReaderSetQueryTableList(pInfo->tqReader, tableIdList, idstr);
309,719✔
3245
    QUERY_CHECK_CODE(code, lino, _error);
310,000✔
3246
    taosArrayDestroy(tableIdList);
310,000✔
3247
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->base.cond, sizeof(SQueryTableDataCond));
310,000✔
3248
  } else {
3249
    taosArrayDestroy(pColIds);
×
3250
    tableListDestroy(pTableListInfo);
×
3251
  }
3252

3253
  // clear the local variable to avoid repeatly free
3254
  pColIds = NULL;
310,000✔
3255

3256
  // create the pseduo columns info
3257
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
310,000✔
3258
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->pPseudoExpr, &pInfo->numOfPseudoExpr);
177,158✔
3259
    QUERY_CHECK_CODE(code, lino, _error);
177,158✔
3260
  }
3261

3262
  code = filterInitFromNode((SNode*)pScanPhyNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0, NULL);
310,000✔
3263
  QUERY_CHECK_CODE(code, lino, _error);
309,719✔
3264

3265
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
309,719✔
3266
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
310,000✔
3267
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
310,000✔
3268
  QUERY_CHECK_CODE(code, lino, _error);
309,719✔
3269

3270
  pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
309,719✔
3271
  pInfo->windowSup = (SWindowSupporter){.pStreamAggSup = NULL, .gap = -1, .parentType = QUERY_NODE_PHYSICAL_PLAN};
309,719✔
3272
  pInfo->groupId = 0;
309,409✔
3273
  pInfo->igCheckGroupId = false;
309,409✔
3274
  pInfo->pStreamScanOp = pOperator;
309,719✔
3275
  pInfo->deleteDataIndex = 0;
309,409✔
3276
  code = createSpecialDataBlock(STREAM_DELETE_DATA, &pInfo->pDeleteDataRes);
309,690✔
3277
  QUERY_CHECK_CODE(code, lino, _error);
310,000✔
3278

3279
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
310,000✔
3280
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateDataRes);
310,000✔
3281
  QUERY_CHECK_CODE(code, lino, _error);
310,000✔
3282

3283
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
310,000✔
3284
  pInfo->partitionSup.needCalc = false;
310,000✔
3285
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
310,000✔
3286
  pInfo->igExpired = pTableScanNode->igExpired;
310,000✔
3287
  pInfo->twAggSup.maxTs = INT64_MIN;
310,000✔
3288
  pInfo->pState = pTaskInfo->streamInfo.pState;
310,000✔
3289
  pInfo->stateStore = pTaskInfo->storageAPI.stateStore;
310,000✔
3290
  pInfo->readerFn = pTaskInfo->storageAPI.tqReaderFn;
310,000✔
3291
  pInfo->pFillSup = NULL;
309,719✔
3292
  pInfo->useGetResultRange = false;
309,719✔
3293
  pInfo->pRangeScanRes = NULL;
309,719✔
3294

3295
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
309,719✔
3296
  QUERY_CHECK_CODE(code, lino, _error);
310,000✔
3297

3298
  setOperatorInfo(pOperator, STREAM_SCAN_OP_NAME, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, false, OP_NOT_OPENED, pInfo,
310,000✔
3299
                  pTaskInfo);
3300
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
310,000✔
3301

3302
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doQueueScanNext, NULL, destroyTmqScanOperatorInfo, optrDefaultBufFn,
309,719✔
3303
                                         NULL, optrDefaultGetNextExtFn, NULL);
3304

3305
  *pOptrInfo = pOperator;
309,719✔
3306
  return code;
309,719✔
3307

3308
_error:
×
3309
  if (pColIds != NULL) {
×
3310
    taosArrayDestroy(pColIds);
×
3311
  }
3312

3313
  if (pInfo != NULL && pInfo->pTableScanOp != NULL) {
×
3314
    STableScanInfo* p = (STableScanInfo*)pInfo->pTableScanOp->info;
×
3315
    if (p != NULL) {
×
3316
      p->base.pTableListInfo = NULL;
×
3317
    }
3318
    destroyTmqScanOperatorInfo(pInfo);
×
3319
  }
3320

3321
  if (pOperator != NULL) {
×
3322
    pOperator->info = NULL;
×
3323
    destroyOperator(pOperator);
×
3324
  }
3325
  pTaskInfo->code = code;
×
3326
  return code;
×
3327
}
3328

3329
static int32_t doTagScanOneTable(SOperatorInfo* pOperator, SSDataBlock* pRes, SMetaReader* mr, SStorageAPI* pAPI) {
3,244,463✔
3330
  int32_t        code = TSDB_CODE_SUCCESS;
3,244,463✔
3331
  int32_t        lino = 0;
3,244,463✔
3332
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3,244,463✔
3333
  STagScanInfo*  pInfo = pOperator->info;
3,244,463✔
3334
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
3,244,463✔
3335
  int32_t        count = pRes->info.rows;
3,244,463✔
3336

3337
  STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
3,244,463✔
3338
  if (!item) {
3,244,463✔
3339
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
3340
           GET_TASKID(pTaskInfo));
3341
    tDecoderClear(&(*mr).coder);
×
3342
    goto _end;
×
3343
  }
3344

3345
  code = pAPI->metaReaderFn.getTableEntryByUid(mr, item->uid);
3,244,463✔
3346
  tDecoderClear(&(*mr).coder);
3,244,463✔
3347
  if (code != TSDB_CODE_SUCCESS) {
3,244,463✔
3348
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
3349
           GET_TASKID(pTaskInfo));
3350
    goto _end;
×
3351
  }
3352

3353
  char str[512];
3,244,367✔
3354
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
6,428,971✔
3355
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
3,184,508✔
3356

3357
    // refactor later
3358
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
3,184,508✔
3359
      STR_TO_VARSTR(str, (*mr).me.name);
212,357✔
3360
      code = colDataSetVal(pDst, (count), str, false);
212,357✔
3361
      QUERY_CHECK_CODE(code, lino, _end);
212,357✔
3362
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
2,972,151✔
3363
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
14,370✔
3364
      QUERY_CHECK_CODE(code, lino, _end);
14,370✔
3365
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
2,957,781✔
3366
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
14,370✔
3367
      QUERY_CHECK_CODE(code, lino, _end);
14,370✔
3368
    } else {  // it is a tag value
3369
      STagVal val = {0};
2,943,411✔
3370
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
2,943,411✔
3371
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
2,943,411✔
3372

3373
      char* data = NULL;
2,943,411✔
3374
      if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
2,943,411✔
3375
        data = tTagValToData((const STagVal*)p, false);
2,942,085✔
3376
      } else {
3377
        data = (char*)p;
1,326✔
3378
      }
3379

3380
      code = colDataSetVal(pDst, (count), data,
2,943,411✔
3381
                           (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
2,943,411✔
3382
      QUERY_CHECK_CODE(code, lino, _end);
2,943,411✔
3383

3384
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
2,943,411✔
3385
          (data != NULL)) {
3386
        taosMemoryFree(data);
820,268✔
3387
      }
3388
    }
3389
  }
3390

3391
_end:
3,244,463✔
3392
  if (code != TSDB_CODE_SUCCESS) {
3,244,463✔
3393
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3394
    pTaskInfo->code = code;
×
3395
  } else {
3396
    pRes->info.rows++;
3,244,463✔
3397
  }
3398

3399
  return code;
3,244,463✔
3400
}
3401

3402
static void tagScanFreeUidTag(void* p) {
7,105,798✔
3403
  STUidTagInfo* pInfo = p;
7,105,798✔
3404
  if (pInfo->pTagVal != NULL) {
7,105,798✔
3405
    taosMemoryFree(pInfo->pTagVal);
7,109,467✔
3406
  }
3407
}
7,106,228✔
3408

3409
static int32_t tagScanCreateResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
1,586,857✔
3410
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
1,586,857✔
3411
  if (pColumnData == NULL) {
1,586,873✔
3412
    return terrno;
×
3413
  }
3414

3415
  pColumnData->info.type = pType->type;
1,586,873✔
3416
  pColumnData->info.bytes = pType->bytes;
1,586,873✔
3417
  pColumnData->info.scale = pType->scale;
1,586,873✔
3418
  pColumnData->info.precision = pType->precision;
1,587,351✔
3419

3420
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
1,587,351✔
3421
  if (code != TSDB_CODE_SUCCESS) {
1,587,351✔
3422
    terrno = code;
×
3423
    taosMemoryFree(pColumnData);
×
3424
    return terrno;
×
3425
  }
3426

3427
  pParam->columnData = pColumnData;
1,587,351✔
3428
  pParam->colAlloced = true;
1,586,857✔
3429
  return TSDB_CODE_SUCCESS;
1,586,857✔
3430
}
3431

3432
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
9,804,880✔
3433
  int32_t                code = TSDB_CODE_SUCCESS;
9,804,880✔
3434
  int32_t                lino = 0;
9,804,880✔
3435
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
9,804,880✔
3436
  SColumnNode*           pSColumnNode = NULL;
9,804,880✔
3437
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
9,805,712✔
3438
    pSColumnNode = *(SColumnNode**)pNode;
2,752,686✔
3439
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
7,052,870✔
3440
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
480✔
3441
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
480✔
3442
      pSColumnNode = NULL;
×
3443
      pCtx->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
×
3444
      if (NULL == pSColumnNode) {
×
3445
        return DEAL_RES_ERROR;
×
3446
      }
3447
      pSColumnNode->colId = -1;
×
3448
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
×
3449
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
×
3450
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
×
3451
      nodesDestroyNode(*pNode);
×
3452
      *pNode = (SNode*)pSColumnNode;
×
3453
    } else {
3454
      return DEAL_RES_CONTINUE;
480✔
3455
    }
3456
  } else {
3457
    return DEAL_RES_CONTINUE;
7,052,390✔
3458
  }
3459

3460
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
2,752,686✔
3461
  if (!data) {
2,752,498✔
3462
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
1,590,410✔
3463
    if (code == TSDB_CODE_DUP_KEY) {
1,590,748✔
3464
      code = TSDB_CODE_SUCCESS;
×
3465
    }
3466
    QUERY_CHECK_CODE(code, lino, _end);
1,590,748✔
3467
    pSColumnNode->slotId = pCtx->index++;
1,590,748✔
3468
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
1,590,254✔
3469
                         .type = pSColumnNode->node.resType.type,
1,590,060✔
3470
                         .bytes = pSColumnNode->node.resType.bytes};
1,590,748✔
3471
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
1,589,566✔
3472
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,590,598✔
3473
  } else {
3474
    SColumnNode* col = *(SColumnNode**)data;
1,162,088✔
3475
    pSColumnNode->slotId = col->slotId;
1,162,088✔
3476
  }
3477

3478
_end:
2,753,180✔
3479
  if (code != TSDB_CODE_SUCCESS) {
2,752,492✔
3480
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
338✔
3481
    return DEAL_RES_ERROR;
×
3482
  }
3483
  return DEAL_RES_CONTINUE;
2,752,160✔
3484
}
3485

3486
static int32_t tagScanFilterByTagCond(SArray* aUidTags, SNode* pTagCond, SArray* aFilterIdxs, void* pVnode,
1,587,842✔
3487
                                      SStorageAPI* pAPI, STagScanInfo* pInfo, void* pStreamRuntimeInfo) {
3488
  int32_t code = TSDB_CODE_SUCCESS;
1,587,842✔
3489
  int32_t lino = 0;
1,587,842✔
3490
  int32_t numOfTables = taosArrayGetSize(aUidTags);
1,587,842✔
3491
  SArray* pBlockList = NULL;
1,587,504✔
3492

3493
  SSDataBlock* pResBlock = createTagValBlockForFilter(pInfo->filterCtx.cInfoList, numOfTables, aUidTags, pVnode, pAPI);
1,587,504✔
3494
  QUERY_CHECK_NULL(pResBlock, code, lino, _end, terrno);
1,586,035✔
3495

3496
  pBlockList = taosArrayInit(1, POINTER_BYTES);
1,586,035✔
3497
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
1,587,842✔
3498

3499
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
1,586,857✔
3500
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,586,857✔
3501

3502
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
1,586,857✔
3503

3504
  SScalarParam output = {0};
1,586,857✔
3505
  code = tagScanCreateResultData(&type, numOfTables, &output);
1,586,857✔
3506
  QUERY_CHECK_CODE(code, lino, _end);
1,587,348✔
3507

3508
  gTaskScalarExtra.pStreamInfo = (void*)pStreamRuntimeInfo;
1,587,348✔
3509
  gTaskScalarExtra.pStreamRange = NULL;
1,587,348✔
3510
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
1,587,351✔
3511
  QUERY_CHECK_CODE(code, lino, _end);
1,587,010✔
3512

3513
  bool* result = (bool*)output.columnData->pData;
1,587,010✔
3514
  for (int32_t i = 0; i < numOfTables; ++i) {
3,967,745✔
3515
    if (result[i]) {
2,380,743✔
3516
      void* tmp = taosArrayPush(aFilterIdxs, &i);
2,236,492✔
3517
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,236,492✔
3518
    }
3519
  }
3520

3521
  colDataDestroy(output.columnData);
1,585,676✔
3522
  taosMemoryFreeClear(output.columnData);
1,586,857✔
3523

3524
_end:
1,586,032✔
3525
  blockDataDestroy(pResBlock);
1,584,709✔
3526
  taosArrayDestroy(pBlockList);
1,586,032✔
3527
  if (code != TSDB_CODE_SUCCESS) {
1,585,063✔
3528
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3529
  }
3530
  return code;
1,585,063✔
3531
}
3532

3533
static int32_t tagScanFillOneCellWithTag(SOperatorInfo* pOperator, const STUidTagInfo* pUidTagInfo,
28,960,279✔
3534
                                         SExprInfo* pExprInfo, SColumnInfoData* pColInfo, int rowIndex,
3535
                                         const SStorageAPI* pAPI, void* pVnode) {
3536
  int32_t code = TSDB_CODE_SUCCESS;
28,960,279✔
3537
  int32_t lino = 0;
28,960,279✔
3538
  if (QUERY_NODE_FUNCTION == pExprInfo->pExpr->nodeType) {
28,960,279✔
3539
    if (FUNCTION_TYPE_TBNAME == pExprInfo->pExpr->_function.functionType) {  // tbname
13,764,368✔
3540
      char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
3541
      STR_TO_VARSTR(str, "ctbidx");
×
3542

3543
      code = colDataSetVal(pColInfo, rowIndex, str, false);
×
3544
      QUERY_CHECK_CODE(code, lino, _end);
×
3545
    } else if (FUNCTION_TYPE_TBUID == pExprInfo->pExpr->_function.functionType) {
13,759,187✔
3546
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
6,878,902✔
3547
      QUERY_CHECK_CODE(code, lino, _end);
6,880,718✔
3548
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
6,882,184✔
3549
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
6,879,702✔
3550
      QUERY_CHECK_CODE(code, lino, _end);
6,881,706✔
3551
    }
3552
  } else {
3553
    STagVal tagVal = {0};
15,196,902✔
3554
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
15,197,058✔
3555
    if (pUidTagInfo->pTagVal == NULL) {
15,199,052✔
3556
      colDataSetNULL(pColInfo, rowIndex);
×
3557
    } else {
3558
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
15,198,063✔
3559

3560
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
15,199,717✔
3561
        colDataSetNULL(pColInfo, rowIndex);
7,510✔
3562
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
15,192,700✔
3563
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
3564
        QUERY_CHECK_CODE(code, lino, _end);
×
3565
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
19,168,596✔
3566
        if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
3,979,398✔
3567
          QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
3568
        }
3569
        char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
3,978,215✔
3570
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,977,721✔
3571

3572
        varDataSetLen(tmp, tagVal.nData);
3,977,721✔
3573
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
3,977,571✔
3574
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
3,978,064✔
3575
        taosMemoryFree(tmp);
3,977,570✔
3576
        QUERY_CHECK_CODE(code, lino, _end);
3,978,556✔
3577
      } else {
3578
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
11,209,963✔
3579
        QUERY_CHECK_CODE(code, lino, _end);
11,212,468✔
3580
      }
3581
    }
3582
  }
3583

3584
_end:
28,959,656✔
3585
  if (code != TSDB_CODE_SUCCESS) {
28,959,656✔
3586
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3587
  }
3588
  return code;
28,955,343✔
3589
}
3590

3591
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
4,544,822✔
3592
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
3593
  int32_t       code = TSDB_CODE_SUCCESS;
4,544,822✔
3594
  int32_t       lino = 0;
4,544,822✔
3595
  STagScanInfo* pInfo = pOperator->info;
4,544,822✔
3596
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
4,545,807✔
3597
  if (!ignoreFilterIdx) {
4,546,476✔
3598
    size_t szTables = taosArrayGetSize(aFilterIdxs);
1,586,364✔
3599
    for (int i = 0; i < szTables; ++i) {
3,824,045✔
3600
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
2,234,526✔
3601
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
2,234,526✔
3602
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
2,235,358✔
3603
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
12,720,938✔
3604
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
10,481,617✔
3605
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
10,484,247✔
3606
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
10,484,247✔
3607
        QUERY_CHECK_CODE(code, lino, _end);
10,485,580✔
3608
      }
3609
    }
3610
  } else {
3611
    size_t szTables = taosArrayGetSize(aUidTags);
2,960,112✔
3612
    for (int i = 0; i < szTables; ++i) {
7,687,996✔
3613
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
4,727,884✔
3614
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
4,727,884✔
3615
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
23,198,124✔
3616
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
18,469,583✔
3617
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
18,470,717✔
3618
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
18,470,717✔
3619
        QUERY_CHECK_CODE(code, lino, _end);
18,470,240✔
3620
      }
3621
    }
3622
  }
3623

3624
_end:
2,960,112✔
3625
  if (code != TSDB_CODE_SUCCESS) {
4,549,631✔
3626
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3627
  }
3628
  return code;
4,545,788✔
3629
}
3630

3631
static int32_t doTagScanFromCtbIdxNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
9,051,899✔
3632
  int32_t        code = TSDB_CODE_SUCCESS;
9,051,899✔
3633
  int32_t        lino = 0;
9,051,899✔
3634
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
9,051,899✔
3635
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
9,054,059✔
3636

3637
  STagScanInfo* pInfo = pOperator->info;
9,052,742✔
3638
  SSDataBlock*  pRes = pInfo->pRes;
9,052,904✔
3639

3640
  QRY_PARAM_CHECK(ppRes);
9,051,142✔
3641

3642
  if (pOperator->status == OP_EXEC_DONE) {
9,050,307✔
3643
    return TSDB_CODE_SUCCESS;
4,434,064✔
3644
  }
3645
  blockDataCleanup(pRes);
4,612,420✔
3646

3647
  if (pInfo->pCtbCursor == NULL) {
4,615,767✔
3648
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
4,615,429✔
3649
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
4,615,413✔
3650
  } else {
3651
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
3652
    QUERY_CHECK_CODE(code, lino, _end);
×
3653
  }
3654

3655
  SArray* aUidTags = pInfo->aUidTags;
4,615,413✔
3656
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
4,615,907✔
3657
  int32_t count = 0;
4,615,907✔
3658
  bool    ctbCursorFinished = false;
4,615,907✔
3659
  while (1) {
92,693✔
3660
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
4,708,600✔
3661
    taosArrayClear(aFilterIdxs);
4,709,282✔
3662

3663
    int32_t numTables = 0;
4,708,788✔
3664
    while (numTables < pOperator->resultInfo.capacity) {
11,820,833✔
3665
      SMCtbCursor* pCur = pInfo->pCtbCursor;
11,817,261✔
3666
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
11,817,605✔
3667
      if (uid == 0) {
11,816,761✔
3668
        ctbCursorFinished = true;
4,709,282✔
3669
        break;
4,709,282✔
3670
      }
3671
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
7,107,479✔
3672
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
7,106,982✔
3673
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
7,107,941✔
3674

3675
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
7,107,941✔
3676
      void* tmp = taosArrayPush(aUidTags, &info);
7,109,155✔
3677
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
7,109,155✔
3678
      ++numTables;
7,109,155✔
3679
    }
3680

3681
    if (numTables == 0) {
4,709,282✔
3682
      break;
161,328✔
3683
    }
3684
    bool ignoreFilterIdx = true;
4,547,954✔
3685
    if (pInfo->pTagCond != NULL) {
4,547,954✔
3686
      ignoreFilterIdx = false;
1,587,842✔
3687
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo, pTaskInfo->pStreamRuntimeInfo);
1,587,842✔
3688
      QUERY_CHECK_CODE(code, lino, _end);
1,584,709✔
3689
    } else {
3690
      ignoreFilterIdx = true;
2,960,112✔
3691
    }
3692

3693
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
4,544,821✔
3694
    QUERY_CHECK_CODE(code, lino, _end);
4,545,788✔
3695

3696
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
4,545,788✔
3697

3698
    if (count != 0) {
4,546,279✔
3699
      break;
4,453,586✔
3700
    }
3701
  }
3702

3703
  if (count > 0) {
4,614,914✔
3704
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
4,453,586✔
3705
  }
3706
  if (count == 0 || ctbCursorFinished) {
4,615,752✔
3707
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
4,615,752✔
3708
    pInfo->pCtbCursor = NULL;
4,615,413✔
3709
    setOperatorCompleted(pOperator);
4,615,413✔
3710
  }
3711

3712
  pRes->info.rows = count;
4,615,064✔
3713
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
4,614,080✔
3714
  if (bLimitReached) {
4,614,438✔
3715
    setOperatorCompleted(pOperator);
×
3716
  }
3717

3718
  pOperator->resultInfo.totalRows += pRes->info.rows;
4,614,438✔
3719
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
4,613,946✔
3720

3721
_end:
4,611,931✔
3722
  if (code != TSDB_CODE_SUCCESS) {
4,611,931✔
3723
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3724
    pTaskInfo->code = code;
×
3725
    T_LONG_JMP(pTaskInfo->env, code);
×
3726
  }
3727

3728
  return code;
4,611,931✔
3729
}
3730

3731
static int32_t createTagScanTableListInfoFromParam(SOperatorInfo* pOperator) {
829,936✔
3732
  STagScanInfo*          pInfo = pOperator->info;
829,936✔
3733
  SExecTaskInfo*         pTaskInfo = pOperator->pTaskInfo;
829,936✔
3734
  int32_t                code = 0;
829,936✔
3735
  STableListInfo*        pListInfo = pInfo->pTableListInfo;
829,936✔
3736
  STagScanOperatorParam* pParam = (STagScanOperatorParam*)pOperator->pOperatorGetParam->value;
829,936✔
3737
  tb_uid_t               pUid = pParam->vcUid;
829,936✔
3738

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

3743
  STableKeyInfo info = {.groupId = 0};
829,936✔
3744

3745
  int32_t tableIdx = 0;
829,936✔
3746
  taosHashClear(pListInfo->map);
829,936✔
3747
  taosArrayClear(pListInfo->pTableList);
829,936✔
3748

3749
  if (taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
829,936✔
3750
    if (TSDB_CODE_DUP_KEY == terrno) {
×
3751
    } else {
3752
      return terrno;
×
3753
    }
3754
  }
3755

3756
  info.uid = pUid;
829,936✔
3757
  void* p = taosArrayPush(pListInfo->pTableList, &info);
829,936✔
3758
  if (p == NULL) {
829,936✔
3759
    return terrno;
×
3760
  }
3761

3762
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
829,936✔
3763

3764
  return code;
829,936✔
3765
}
3766

3767
static int32_t doTagScanFromMetaEntryNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
5,236,277✔
3768
  int32_t       code = TSDB_CODE_SUCCESS;
5,236,277✔
3769
  int32_t       lino = 0;
5,236,277✔
3770
  STagScanInfo* pInfo = pOperator->info;
5,236,277✔
3771
  if (pOperator->pOperatorGetParam) {
5,236,277✔
3772
    pOperator->resultInfo.totalRows = 0;
829,936✔
3773
    pOperator->dynamicTask = true;
829,936✔
3774
    pInfo->curPos = 0;
829,936✔
3775
    code = createTagScanTableListInfoFromParam(pOperator);
829,936✔
3776
    freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
829,936✔
3777
    pOperator->pOperatorGetParam = NULL;
829,936✔
3778
    QUERY_CHECK_CODE(code, lino, _end);
829,936✔
3779

3780
    if (pOperator->status == OP_EXEC_DONE) {
829,936✔
3781
      pOperator->status = OP_OPENED;
327,018✔
3782
    }
3783
  }
3784

3785
  if (pOperator->status == OP_EXEC_DONE) {
5,236,277✔
3786
    (*ppRes) = NULL;
2,070,100✔
3787
    return code;
2,070,100✔
3788
  }
3789

3790
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3,166,177✔
3791
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
3,166,177✔
3792

3793
  SExprInfo*   pExprInfo = &pOperator->exprSupp.pExprInfo[0];
3,166,177✔
3794
  SSDataBlock* pRes = pInfo->pRes;
3,166,177✔
3795
  blockDataCleanup(pRes);
3,166,177✔
3796

3797
  int32_t size = 0;
3,166,177✔
3798
  code = tableListGetSize(pInfo->pTableListInfo, &size);
3,166,177✔
3799
  QUERY_CHECK_CODE(code, lino, _end);
3,166,177✔
3800

3801
  if (size == 0) {
3,166,177✔
3802
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
17,117✔
3803
    (*ppRes) = NULL;
17,117✔
3804
    return code;
17,117✔
3805
  }
3806

3807
  SMetaReader mr = {0};
3,149,060✔
3808
  pAPI->metaReaderFn.initReader(&mr, pInfo->readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
3,149,060✔
3809
  pRes->info.rows = 0;
3,149,060✔
3810

3811
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
6,393,523✔
3812
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
3,244,463✔
3813
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD &&
3,244,463✔
3814
        code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
3815
      // ignore other error
3816
      code = TSDB_CODE_SUCCESS;
3,244,463✔
3817
    }
3818
    QUERY_CHECK_CODE(code, lino, _end);
3,244,463✔
3819

3820
    if (++pInfo->curPos >= size) {
3,244,463✔
3821
      setOperatorCompleted(pOperator);
3,149,060✔
3822
    }
3823
  }
3824

3825
  pAPI->metaReaderFn.clearReader(&mr);
3,149,060✔
3826
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
3,149,060✔
3827
  if (bLimitReached) {
3,149,060✔
3828
    setOperatorCompleted(pOperator);
×
3829
  }
3830

3831
  // qDebug("QInfo:0x%" PRIx64 ", create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
3832
  if (pOperator->status == OP_EXEC_DONE) {
3,149,060✔
3833
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
3,149,060✔
3834
  }
3835

3836
  pOperator->resultInfo.totalRows += pRes->info.rows;
3,149,060✔
3837

3838
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
3,149,060✔
3839
_end:
3,149,060✔
3840
  if (code != TSDB_CODE_SUCCESS) {
3,149,060✔
3841
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3842
    pTaskInfo->code = code;
×
3843
    T_LONG_JMP(pTaskInfo->env, code);
×
3844
  }
3845
  return code;
3,149,060✔
3846
}
3847

3848
static void destroyTagScanOperatorInfo(void* param) {
7,926,834✔
3849
  STagScanInfo* pInfo = (STagScanInfo*)param;
7,926,834✔
3850
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
7,926,834✔
3851
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
3852
  }
3853
  taosHashCleanup(pInfo->filterCtx.colHash);
7,927,828✔
3854
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
7,928,026✔
3855
  taosArrayDestroy(pInfo->aFilterIdxs);
7,928,708✔
3856
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
7,927,108✔
3857

3858
  blockDataDestroy(pInfo->pRes);
7,927,996✔
3859
  taosArrayDestroy(pInfo->matchInfo.pList);
7,929,814✔
3860
  tableListDestroy(pInfo->pTableListInfo);
7,929,658✔
3861

3862
  pInfo->pRes = NULL;
7,928,870✔
3863
  pInfo->pTableListInfo = NULL;
7,928,526✔
3864
  taosMemoryFreeClear(param);
7,929,342✔
3865
}
7,927,832✔
3866

3867
static int32_t resetTagScanOperatorState(SOperatorInfo* pOper) {
4,440✔
3868
  int32_t         code = TSDB_CODE_SUCCESS;
4,440✔
3869
  STagScanInfo*   pInfo = pOper->info;
4,440✔
3870

3871
  pOper->status = OP_NOT_OPENED;
4,440✔
3872
  STagScanPhysiNode* pTagScanNode = (STagScanPhysiNode*)pOper->pPhyNode;
4,440✔
3873
  if (pTagScanNode->onlyMetaCtbIdx) {
4,440✔
3874
      SExecTaskInfo* pTaskInfo = pOper->pTaskInfo;
1,920✔
3875
      SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
1,920✔
3876
      pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
1,920✔
3877
      pInfo->pCtbCursor = NULL;
1,920✔
3878
  } else {
3879
      pInfo->curPos = 0;
2,520✔
3880
  }
3881

3882
  return code;
4,440✔
3883
}
3884

3885
int32_t createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pTagScanNode,
7,928,832✔
3886
                                  STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3887
                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
3888
  QRY_PARAM_CHECK(pOptrInfo);
7,928,832✔
3889

3890
  int32_t         code = TSDB_CODE_SUCCESS;
7,929,326✔
3891
  int32_t         lino = 0;
7,929,326✔
3892
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
7,929,326✔
3893
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
7,929,326✔
3894
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
7,928,495✔
3895
  if (pInfo == NULL || pOperator == NULL) {
7,924,517✔
3896
    code = terrno;
×
3897
    goto _error;
×
3898
  }
3899
  pOperator->pPhyNode = pTagScanNode;
7,924,517✔
3900
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
7,924,517✔
3901

3902
  int32_t    numOfExprs = 0;
7,924,517✔
3903
  SExprInfo* pExprInfo = NULL;
7,924,517✔
3904

3905
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
7,924,517✔
3906
  QUERY_CHECK_CODE(code, lino, _error);
7,929,929✔
3907

3908
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
7,929,929✔
3909
  QUERY_CHECK_CODE(code, lino, _error);
7,924,375✔
3910

3911
  int32_t num = 0;
7,924,375✔
3912
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
7,922,571✔
3913
  QUERY_CHECK_CODE(code, lino, _error);
7,926,703✔
3914

3915
  pInfo->pTagCond = pTagCond;
7,926,703✔
3916
  pInfo->pTagIndexCond = pTagIndexCond;
7,926,359✔
3917
  pInfo->suid = pPhyNode->suid;
7,927,277✔
3918
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
7,928,414✔
3919

3920
  pInfo->pTableListInfo = pTableListInfo;
7,927,207✔
3921
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
7,926,588✔
3922
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
7,928,956✔
3923

3924
  pInfo->readHandle = *pReadHandle;
7,925,209✔
3925
  pInfo->curPos = 0;
7,924,465✔
3926

3927
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
7,924,422✔
3928
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
7,924,871✔
3929
                  pTaskInfo);
3930
  initResultSizeInfo(&pOperator->resultInfo, 4096);
7,921,234✔
3931
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
7,923,155✔
3932
  QUERY_CHECK_CODE(code, lino, _error);
7,929,778✔
3933

3934
  if (pTagScanNode->onlyMetaCtbIdx) {
7,929,778✔
3935
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
4,630,103✔
3936
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
4,630,597✔
3937

3938
    pInfo->aFilterIdxs = taosArrayInit(pOperator->resultInfo.capacity, sizeof(int32_t));
4,630,103✔
3939
    QUERY_CHECK_NULL(pInfo->aFilterIdxs, code, lino, _error, terrno);
4,629,428✔
3940

3941
    pInfo->filterCtx.colHash =
4,630,104✔
3942
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
4,629,428✔
3943
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
4,630,104✔
3944

3945
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
4,629,611✔
3946
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
4,630,103✔
3947

3948
    if (pInfo->pTagCond != NULL) {
4,628,622✔
3949
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
1,590,598✔
3950
    }
3951
  }
3952
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
7,928,947✔
3953
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
7,927,385✔
3954
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
3955
  setOperatorResetStateFn(pOperator, resetTagScanOperatorState);     
7,928,373✔
3956
  
3957
  *pOptrInfo = pOperator;
7,929,514✔
3958
  return code;
7,929,514✔
3959

3960
_error:
×
3961
  if (pInfo) {
×
3962
    pInfo->pTableListInfo = NULL;
×
3963
  }
3964

3965
  if (pInfo != NULL) destroyTagScanOperatorInfo(pInfo);
×
3966
  if (pOperator != NULL) {
×
3967
    pOperator->info = NULL;
×
3968
    destroyOperator(pOperator);
×
3969
  }
3970
  return code;
×
3971
}
3972

3973
// table merge scan operator
3974

3975
static int32_t subTblRowCompareTsFn(const void* pLeft, const void* pRight, void* param) {
228,563,032✔
3976
  int32_t                 left = *(int32_t*)pLeft;
228,563,032✔
3977
  int32_t                 right = *(int32_t*)pRight;
228,564,408✔
3978
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
228,566,128✔
3979

3980
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
228,566,128✔
3981
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
228,566,816✔
3982

3983
  if (leftIdx == -1) {
228,571,632✔
3984
    return 1;
195,799,152✔
3985
  } else if (rightIdx == -1) {
32,772,480✔
3986
    return -1;
324,480✔
3987
  }
3988

3989
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
32,448,000✔
3990
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
32,448,000✔
3991
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
32,448,000✔
3992
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
32,448,000✔
3993
    ret = -1 * ret;
×
3994
  }
3995
  return ret;
32,448,000✔
3996
}
3997

3998
static int32_t subTblRowCompareTsPkFn(const void* pLeft, const void* pRight, void* param) {
×
3999
  int32_t                 left = *(int32_t*)pLeft;
×
4000
  int32_t                 right = *(int32_t*)pRight;
×
4001
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
×
4002

4003
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
×
4004
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
×
4005

4006
  if (leftIdx == -1) {
×
4007
    return 1;
×
4008
  } else if (rightIdx == -1) {
×
4009
    return -1;
×
4010
  }
4011

4012
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
×
4013
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
×
4014
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
×
4015
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
×
4016
    ret = -1 * ret;
×
4017
  }
4018
  if (ret == 0 && pInfo->pPkOrderInfo) {
×
4019
    ret = tsortComparBlockCell(pInfo->aInputs[left].pInputBlock, pInfo->aInputs[right].pInputBlock, leftIdx, rightIdx,
×
4020
                               pInfo->pPkOrderInfo);
×
4021
  }
4022
  return ret;
×
4023
}
4024

4025
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
2,907,860✔
4026
  int32_t code = TSDB_CODE_SUCCESS;
2,907,860✔
4027
  int32_t lino = 0;
2,907,860✔
4028
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
2,907,860✔
4029
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
2,907,860✔
4030
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
2,900,726✔
4031
  for (int i = 0; i < src->numOfCols; i++) {
9,652,926✔
4032
    dst->colList[i] = src->colList[i];
6,752,586✔
4033
  }
4034

4035
_end:
2,907,522✔
4036
  if (code != TSDB_CODE_SUCCESS) {
2,907,522✔
4037
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4038
  }
4039
  return code;
2,900,388✔
4040
}
4041

4042
static int32_t fetchNextSubTableBlockFromReader(SOperatorInfo* pOperator, STmsSubTableInput* pInput,
4,359,144✔
4043
                                                bool* pSubTableHasBlock) {
4044
  int32_t code = 0;
4,359,144✔
4045

4046
  STableMergeScanInfo*    pInfo = pOperator->info;
4,359,144✔
4047
  SReadHandle*            pHandle = &pInfo->base.readHandle;
4,359,494✔
4048
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
4,360,870✔
4049
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
4,361,884✔
4050
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
4,361,214✔
4051

4052
  blockDataCleanup(pInput->pReaderBlock);
4,360,526✔
4053
  if (!pInput->bInMemReader) {
4,361,214✔
4054
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
4,381,464✔
4055
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
2,190,234✔
4056
    if (code != 0) {
2,190,222✔
4057
      return code;
×
4058
    }
4059
  }
4060

4061
  pInfo->base.dataReader = pInput->pReader;
4,361,220✔
4062

4063
  while (true) {
7,224✔
4064
    bool hasNext = false;
4,370,128✔
4065
    code = pAPI->tsdReader.tsdNextDataBlock(pInfo->base.dataReader, &hasNext);
4,370,128✔
4066
    if (code != 0) {
4,370,478✔
4067
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
4068
      pInfo->base.dataReader = NULL;
×
4069
      return code;
×
4070
    }
4071

4072
    if (!hasNext || isTaskKilled(pTaskInfo)) {
4,370,478✔
4073
      if (isTaskKilled(pTaskInfo)) {
2,912,616✔
4074
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
4075
        pInfo->base.dataReader = NULL;
×
4076
        return code;
×
4077
      }
4078

4079
      *pSubTableHasBlock = false;
2,912,278✔
4080
      break;
2,912,278✔
4081
    }
4082

4083
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
1,458,200✔
4084
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
1,429,992✔
4085
    } else {
4086
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
28,208✔
4087
    }
4088

4089
    uint32_t status = 0;
1,458,200✔
4090
    code = loadDataBlock(pOperator, &pInfo->base, pInput->pReaderBlock, &status);
1,458,200✔
4091
    if (code != 0) {
1,458,200✔
4092
      pInfo->base.dataReader = NULL;
×
4093
      return code;
×
4094
    }
4095

4096
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
1,458,200✔
4097
      *pSubTableHasBlock = false;
×
4098
      break;
×
4099
    }
4100
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pInput->pReaderBlock->info.rows == 0) {
1,458,200✔
4101
      continue;
7,224✔
4102
    }
4103

4104
    *pSubTableHasBlock = true;
1,450,976✔
4105
    break;
1,450,976✔
4106
  }
4107

4108
  if (*pSubTableHasBlock) {
4,363,592✔
4109
    pInput->pReaderBlock->info.id.groupId =
2,901,270✔
4110
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
2,901,614✔
4111
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
1,450,976✔
4112
  }
4113
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
4,363,254✔
4114
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
3,280,854✔
4115
    pInput->pReader = NULL;
3,280,160✔
4116
  }
4117

4118
  pInfo->base.dataReader = NULL;
4,362,216✔
4119
  return TSDB_CODE_SUCCESS;
4,363,248✔
4120
}
4121

4122
static int32_t setGroupStartEndIndex(STableMergeScanInfo* pInfo) {
1,818,592✔
4123
  pInfo->bGroupProcessed = false;
1,818,592✔
4124

4125
  int32_t numOfTables = 0;
1,819,606✔
4126
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
1,813,184✔
4127
  if (code != TSDB_CODE_SUCCESS) {
1,807,384✔
4128
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4129
    return code;
×
4130
  }
4131

4132
  int32_t i = pInfo->tableStartIndex + 1;
1,807,384✔
4133
  for (; i < numOfTables; ++i) {
2,894,612✔
4134
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
1,145,320✔
4135
    if (!tableKeyInfo) {
1,144,632✔
4136
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4137
      return terrno;
×
4138
    }
4139
    if (tableKeyInfo->groupId != pInfo->groupId) {
1,144,632✔
4140
      break;
57,760✔
4141
    }
4142
  }
4143
  pInfo->tableEndIndex = i - 1;
1,807,052✔
4144
  return TSDB_CODE_SUCCESS;
1,820,614✔
4145
}
4146

4147
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
1,823,010✔
4148
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
4,734,950✔
4149
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
2,912,278✔
4150
    if (pInput->rowIdx == -1) {
2,911,602✔
4151
      continue;
1,461,302✔
4152
    }
4153

4154
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
1,450,632✔
4155
      pInput->rowIdx = 0;
1,450,976✔
4156
      pInput->pageIdx = -1;
1,450,632✔
4157
    }
4158

4159
    pInput->pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,449,600✔
4160
    SColumnInfoData* col = taosArrayGet(pInput->pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
1,450,632✔
4161
    if (!col) {
1,450,976✔
4162
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4163
      return terrno;
×
4164
    }
4165
    pInput->aTs = (int64_t*)col->pData;
1,450,976✔
4166
  }
4167

4168
  __merge_compare_fn_t mergeCompareFn = (!pSubTblsInfo->pPkOrderInfo) ? subTblRowCompareTsFn : subTblRowCompareTsPkFn;
1,824,030✔
4169
  return tMergeTreeCreate(&pSubTblsInfo->pTree, pSubTblsInfo->numSubTables, pSubTblsInfo, mergeCompareFn);
1,823,692✔
4170
}
4171

4172
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
1,819,274✔
4173
  int32_t code = setGroupStartEndIndex(pInfo);
1,819,274✔
4174
  if (code != TSDB_CODE_SUCCESS) {
1,819,588✔
4175
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4176
    return code;
×
4177
  }
4178
  STmsSubTablesMergeInfo* pSubTblsInfo = taosMemoryCalloc(1, sizeof(STmsSubTablesMergeInfo));
1,819,588✔
4179
  if (pSubTblsInfo == NULL) {
1,816,220✔
4180
    return terrno;
×
4181
  }
4182
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
1,816,220✔
4183
  if (!pSubTblsInfo->pTsOrderInfo) {
1,820,638✔
4184
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4185
    return terrno;
×
4186
  }
4187
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
1,817,916✔
4188
    pSubTblsInfo->pPkOrderInfo = taosArrayGet(pInfo->pSortInfo, 1);
×
4189
    if (!pSubTblsInfo->pPkOrderInfo) {
×
4190
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4191
      return terrno;
×
4192
    }
4193
  } else {
4194
    pSubTblsInfo->pPkOrderInfo = NULL;
1,816,890✔
4195
  }
4196
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
1,819,962✔
4197
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
1,816,220✔
4198
  if (pSubTblsInfo->aInputs == NULL) {
1,810,438✔
4199
    taosMemoryFree(pSubTblsInfo);
×
4200
    return terrno;
×
4201
  }
4202
  int32_t bufPageSize = pInfo->bufPageSize;
1,812,478✔
4203
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
1,812,822✔
4204
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
1,814,850✔
4205
  if (code != TSDB_CODE_SUCCESS) {
1,814,180✔
4206
    taosMemoryFree(pSubTblsInfo->aInputs);
×
4207
    taosMemoryFree(pSubTblsInfo);
×
4208
    return code;
×
4209
  }
4210
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
1,814,180✔
4211
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
1,820,970✔
4212

4213
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
1,817,578✔
4214
  return TSDB_CODE_SUCCESS;
1,813,130✔
4215
}
4216

4217
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
1,816,564✔
4218
  int32_t        code = TSDB_CODE_SUCCESS;
1,816,564✔
4219
  int32_t        lino = 0;
1,816,564✔
4220
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
1,816,564✔
4221
  SReadHandle*   pHandle = &pInfo->base.readHandle;
1,819,274✔
4222
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
1,818,272✔
4223

4224
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
1,818,924✔
4225

4226
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
4,724,044✔
4227
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
2,906,840✔
4228
    pInput->type = SUB_TABLE_MEM_BLOCK;
2,905,826✔
4229

4230
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
2,906,164✔
4231
    QUERY_CHECK_CODE(code, lino, _end);
2,900,062✔
4232

4233
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
2,900,062✔
4234
    QUERY_CHECK_CODE(code, lino, _end);
2,909,562✔
4235

4236
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
2,909,562✔
4237
    QUERY_CHECK_CODE(code, lino, _end);
2,912,278✔
4238

4239
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
2,912,278✔
4240
    pInput->pKeyInfo = keyInfo;
2,911,934✔
4241

4242
    if (isTaskKilled(pTaskInfo)) {
2,911,934✔
4243
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
×
4244
    }
4245

4246
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
2,909,218✔
4247
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
2,176,152✔
4248
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
1,088,248✔
4249
      QUERY_CHECK_CODE(code, lino, _end);
1,087,216✔
4250
      pInput->bInMemReader = true;
1,087,216✔
4251
    } else {
4252
      pInput->pReader = NULL;
1,822,672✔
4253
      pInput->bInMemReader = false;
1,821,996✔
4254
    }
4255
    bool hasNext = true;
2,909,188✔
4256
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
2,909,188✔
4257
    QUERY_CHECK_CODE(code, lino, _end);
2,911,252✔
4258
    if (!hasNext) {
2,911,252✔
4259
      pInput->rowIdx = -1;
1,460,964✔
4260
      ++pSubTblsInfo->numSubTablesCompleted;
1,461,302✔
4261
      continue;
1,460,620✔
4262
    } else {
4263
      pInput->rowIdx = 0;
1,450,288✔
4264
      pInput->pageIdx = -1;
1,450,976✔
4265
    }
4266
  }
4267

4268
_end:
1,823,686✔
4269
  if (code != TSDB_CODE_SUCCESS) {
1,823,686✔
4270
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4271
  }
4272
  return code;
1,822,666✔
4273
}
4274

4275
static void adjustSubTableFromMemBlock(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
1,450,976✔
4276
  int32_t              code = TSDB_CODE_SUCCESS;
1,450,976✔
4277
  int32_t              lino = 0;
1,450,976✔
4278
  SExecTaskInfo*       pTaskInfo = pOperatorInfo->pTaskInfo;
1,450,976✔
4279
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
1,450,976✔
4280
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,450,976✔
4281
  bool                 hasNext = true;
1,450,976✔
4282
  code = fetchNextSubTableBlockFromReader(pOperatorInfo, pInput, &hasNext);
1,450,976✔
4283
  QUERY_CHECK_CODE(code, lino, _end);
1,450,976✔
4284

4285
  if (!hasNext) {
1,450,976✔
4286
    pInput->rowIdx = -1;
1,450,976✔
4287
    ++pSubTblsInfo->numSubTablesCompleted;
1,450,976✔
4288
  } else {
4289
    pInput->rowIdx = 0;
×
4290
  }
4291

4292
_end:
1,450,976✔
4293
  if (code != TSDB_CODE_SUCCESS) {
1,450,976✔
4294
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4295
    pTaskInfo->code = code;
×
4296
    T_LONG_JMP(pTaskInfo->env, code);
×
4297
  }
4298
}
1,450,976✔
4299

4300
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
264,080,836✔
4301
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
264,080,836✔
4302
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
264,069,060✔
4303

4304
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
264,080,552✔
4305
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
264,110,252✔
4306
    ++pInput->rowIdx;
262,651,110✔
4307
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
1,450,976✔
4308
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
1,450,976✔
4309
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
1,450,976✔
4310
    }
4311
    if (pInput->rowIdx != -1) {
1,450,976✔
4312
      SColumnInfoData* col = taosArrayGet(pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
×
4313
      if (!col) {
×
4314
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4315
        return terrno;
×
4316
      }
4317
      pInput->pInputBlock = pInputBlock;
×
4318
      pInput->aTs = (int64_t*)col->pData;
×
4319
    }
4320
  }
4321

4322
  return tMergeTreeAdjust(pSubTblsInfo->pTree, tMergeTreeGetAdjustIndex(pSubTblsInfo->pTree));
264,107,542✔
4323
}
4324

4325
static int32_t appendChosenRowToDataBlock(STmsSubTablesMergeInfo* pSubTblsInfo, SSDataBlock* pBlock) {
263,967,252✔
4326
  int32_t            code = TSDB_CODE_SUCCESS;
263,967,252✔
4327
  int32_t            lino = 0;
263,967,252✔
4328
  STmsSubTableInput* pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
263,967,252✔
4329
  SSDataBlock*       pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
263,986,234✔
4330

4331
  for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
1,079,471,484✔
4332
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
814,867,554✔
4333
    QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
814,529,384✔
4334

4335
    SColumnInfoData* pSrcColInfo = taosArrayGet(pInputBlock->pDataBlock, i);
814,529,384✔
4336
    QUERY_CHECK_NULL(pSrcColInfo, code, lino, _end, terrno);
814,020,550✔
4337
    bool isNull = colDataIsNull(pSrcColInfo, pInputBlock->info.rows, pInput->rowIdx, NULL);
814,020,550✔
4338

4339
    if (isNull) {
814,884,976✔
4340
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
91,648,104✔
4341
      QUERY_CHECK_CODE(code, lino, _end);
91,637,440✔
4342
    } else {
4343
      if (pSrcColInfo->pData != NULL) {
723,236,872✔
4344
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
723,222,532✔
4345
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
724,325,680✔
4346
        QUERY_CHECK_CODE(code, lino, _end);
723,849,070✔
4347
      }
4348
    }
4349
  }
4350
  pBlock->info.dataLoad = 1;
264,116,952✔
4351
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
264,117,658✔
4352
  pBlock->info.rows += 1;
264,109,432✔
4353

4354
_end:
264,109,872✔
4355
  if (code != TSDB_CODE_SUCCESS) {
264,109,872✔
4356
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4357
  }
4358
  return code;
264,101,990✔
4359
}
4360

4361
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
2,949,844✔
4362
                                       SSDataBlock** pResBlock) {
4363
  int32_t                 code = TSDB_CODE_SUCCESS;
2,949,844✔
4364
  int32_t                 lino = 0;
2,949,844✔
4365
  STableMergeScanInfo*    pInfo = pOperator->info;
2,949,844✔
4366
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
2,950,864✔
4367
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
2,950,520✔
4368
  bool                    finished = false;
2,948,824✔
4369

4370
  QRY_PARAM_CHECK(pResBlock);
2,948,824✔
4371

4372
  blockDataCleanup(pBlock);
2,949,162✔
4373

4374
  while (true) {
×
4375
    while (true) {
4376
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
266,948,764✔
4377
        finished = true;
2,950,526✔
4378
        break;
2,950,526✔
4379
      }
4380

4381
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
263,984,598✔
4382
      QUERY_CHECK_CODE(code, lino, _end);
264,091,814✔
4383

4384
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
264,091,814✔
4385
      QUERY_CHECK_CODE(code, lino, _end);
263,989,710✔
4386

4387
      if (pBlock->info.rows >= capacity) {
263,989,710✔
4388
        break;
×
4389
      }
4390
    }
4391

4392
    if (isTaskKilled(pTaskInfo)) {
2,950,526✔
4393
      return pTaskInfo->code;
×
4394
    }
4395

4396
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo);
2,950,526✔
4397
    if (finished || limitReached || pBlock->info.rows > 0) {
2,950,526✔
4398
      break;
4399
    }
4400
  }
4401

4402
  if (pBlock->info.rows > 0) {
2,950,526✔
4403
    *pResBlock = pBlock;
1,126,496✔
4404
  }
4405

4406
_end:
1,823,354✔
4407
  if (code != TSDB_CODE_SUCCESS) {
2,949,850✔
4408
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4409
    pTaskInfo->code = code;
×
4410
  }
4411
  return code;
2,950,526✔
4412
}
4413

4414
static int32_t startSubTablesTableMergeScan(SOperatorInfo* pOperator) {
1,819,612✔
4415
  int32_t              code = TSDB_CODE_SUCCESS;
1,819,612✔
4416
  int32_t              lino = 0;
1,819,612✔
4417
  STableMergeScanInfo* pInfo = pOperator->info;
1,819,612✔
4418

4419
  code = initSubTablesMergeInfo(pInfo);
1,823,010✔
4420
  QUERY_CHECK_CODE(code, lino, _end);
1,814,512✔
4421

4422
  code = initSubTableInputs(pOperator, pInfo);
1,814,512✔
4423
  QUERY_CHECK_CODE(code, lino, _end);
1,823,010✔
4424

4425
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
1,823,010✔
4426
  QUERY_CHECK_CODE(code, lino, _end);
1,821,990✔
4427

4428
_end:
1,821,990✔
4429
  if (code != TSDB_CODE_SUCCESS) {
1,821,990✔
4430
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4431
  }
4432
  return code;
1,822,334✔
4433
}
4434

4435
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
27,134,938✔
4436
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
27,134,938✔
4437
  if (pSubTblsInfo != NULL) {
27,136,762✔
4438
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
1,824,030✔
4439

4440
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
4,736,646✔
4441
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
2,912,616✔
4442
      taosMemoryFree(pInput->tblCond.colList);
2,912,278✔
4443
      blockDataDestroy(pInput->pReaderBlock);
2,912,278✔
4444
      blockDataDestroy(pInput->pPageBlock);
2,911,940✔
4445
      taosArrayDestroy(pInput->aBlockPages);
2,912,616✔
4446
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
2,912,616✔
4447
      pInput->pReader = NULL;
2,912,616✔
4448
    }
4449

4450
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
1,824,368✔
4451
    taosMemoryFree(pSubTblsInfo->aInputs);
1,824,030✔
4452

4453
    taosMemoryFree(pSubTblsInfo);
1,824,030✔
4454
    pInfo->pSubTablesMergeInfo = NULL;
1,823,692✔
4455

4456
    // taosMemoryTrim(0);
4457
  }
4458
}
27,136,762✔
4459

4460
int32_t doTableMergeScanParaSubTablesNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
2,935,138✔
4461
  QRY_PARAM_CHECK(ppRes);
2,935,138✔
4462

4463
  int32_t lino = 0;
2,937,498✔
4464
  int32_t tableListSize = 0;
2,937,498✔
4465
  int64_t st = taosGetTimestampUs();
2,932,760✔
4466

4467
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
2,932,760✔
4468
  STableMergeScanInfo* pInfo = pOperator->info;
2,932,416✔
4469

4470
  if (pOperator->status == OP_EXEC_DONE) {
2,933,436✔
4471
    return TSDB_CODE_SUCCESS;
×
4472
  }
4473

4474
  int32_t code = pOperator->fpSet._openFn(pOperator);
2,929,712✔
4475
  QUERY_CHECK_CODE(code, lino, _end);
2,930,364✔
4476

4477
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
2,930,364✔
4478
  QUERY_CHECK_CODE(code, lino, _end);
2,936,152✔
4479

4480
  if (!pInfo->hasGroupId) {
2,936,152✔
4481
    pInfo->hasGroupId = true;
1,810,332✔
4482

4483
    if (tableListSize == 0) {
1,805,576✔
4484
      setOperatorCompleted(pOperator);
47,128✔
4485
      (*ppRes) = NULL;
47,472✔
4486
      return code;
47,472✔
4487
    }
4488

4489
    pInfo->tableStartIndex = 0;
1,758,448✔
4490
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
1,758,466✔
4491
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
1,756,076✔
4492
    pInfo->groupId = pTmpGpId->groupId;
1,756,076✔
4493
    code = startSubTablesTableMergeScan(pOperator);
1,762,196✔
4494
    QUERY_CHECK_CODE(code, lino, _end);
1,764,574✔
4495
  }
4496

4497
  SSDataBlock* pBlock = NULL;
2,890,732✔
4498
  while (pInfo->tableStartIndex < tableListSize) {
2,948,830✔
4499
    if (isTaskKilled(pTaskInfo)) {
2,950,182✔
4500
      break;
×
4501
    }
4502

4503
    code = getSubTablesSortedBlock(pOperator, pInfo->pResBlock, pOperator->resultInfo.capacity, &pBlock);
2,950,526✔
4504
    QUERY_CHECK_CODE(code, lino, _end);
2,950,526✔
4505

4506
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
2,950,526✔
4507
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
×
4508
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
×
4509

4510
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
×
4511
    }
4512

4513
    if (pBlock != NULL) {
2,949,174✔
4514
      pBlock->info.id.groupId = pInfo->groupId;
1,126,496✔
4515
      pOperator->resultInfo.totalRows += pBlock->info.rows;
1,126,496✔
4516
      pInfo->bGroupProcessed = true;
1,126,496✔
4517
      break;
1,126,496✔
4518
    } else {
4519
      // Data of this group are all dumped, let's try the next group
4520
      stopSubTablesTableMergeScan(pInfo);
1,822,678✔
4521
      if (pInfo->tableEndIndex >= tableListSize - 1) {
1,824,368✔
4522
        setOperatorCompleted(pOperator);
1,766,270✔
4523
        break;
1,765,932✔
4524
      }
4525

4526
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
57,760✔
4527
      STableKeyInfo* pTmpGpId = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
57,760✔
4528
      QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
57,760✔
4529

4530
      pInfo->groupId = pTmpGpId->groupId;
57,760✔
4531
      code = startSubTablesTableMergeScan(pOperator);
57,760✔
4532
      QUERY_CHECK_CODE(code, lino, _end);
57,760✔
4533
      resetLimitInfoForNextGroup(&pInfo->limitInfo);
57,760✔
4534
    }
4535
  }
4536

4537
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
2,892,428✔
4538

4539
_end:
2,892,428✔
4540
  if (code != TSDB_CODE_SUCCESS) {
2,892,766✔
4541
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4542
    pTaskInfo->code = code;
×
4543
    T_LONG_JMP(pTaskInfo->env, code);
×
4544
  } else {
4545
    (*ppRes) = pBlock;
2,892,766✔
4546
  }
4547

4548
  return code;
2,892,428✔
4549
}
4550

4551
static void tableMergeScanDoSkipTable(uint64_t uid, void* pTableMergeOpInfo) {
3,444,258✔
4552
  int32_t              code = TSDB_CODE_SUCCESS;
3,444,258✔
4553
  int32_t              lino = 0;
3,444,258✔
4554
  SOperatorInfo*       pOperator = (SOperatorInfo*)pTableMergeOpInfo;
3,444,258✔
4555
  STableMergeScanInfo* pInfo = pOperator->info;
3,444,258✔
4556
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
3,444,258✔
4557

4558
  if (pInfo->mSkipTables == NULL) {
3,444,258✔
4559
    pInfo->mSkipTables = taosHashInit(pInfo->tableEndIndex - pInfo->tableStartIndex + 1,
289,684✔
4560
                                      taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_NO_LOCK);
4561
    QUERY_CHECK_NULL(pInfo->mSkipTables, code, lino, _end, terrno);
289,684✔
4562
  }
4563
  int bSkip = 1;
3,444,258✔
4564
  if (pInfo->mSkipTables != NULL) {
3,444,258✔
4565
    code = taosHashPut(pInfo->mSkipTables, &uid, sizeof(uid), &bSkip, sizeof(bSkip));
3,444,258✔
4566
    if (code == TSDB_CODE_DUP_KEY) {
3,444,258✔
4567
      code = TSDB_CODE_SUCCESS;
×
4568
    }
4569
    QUERY_CHECK_CODE(code, lino, _end);
3,444,258✔
4570
  }
4571

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

4580
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
164,598,739✔
4581
  STableMergeScanInfo* pInfo = pOperator->info;
164,598,739✔
4582
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
164,612,673✔
4583
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
164,613,129✔
4584
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
164,612,142✔
4585
  int32_t              code = 0;
164,607,121✔
4586
  bool                 hasNext = false;
164,607,121✔
4587
  STsdbReader*         reader = pInfo->base.dataReader;
164,612,744✔
4588

4589
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
164,612,804✔
4590
  if (code != 0) {
164,576,241✔
4591
    pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
4592
    qError("table merge scan fetch next data block error code: %d, %s", code, GET_TASKID(pTaskInfo));
×
4593
    pTaskInfo->code = code;
×
4594
    return code;
×
4595
  }
4596

4597
  if (!hasNext || isTaskKilled(pTaskInfo)) {
164,576,241✔
4598
    if (isTaskKilled(pTaskInfo)) {
26,827,019✔
4599
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
507✔
4600
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
507✔
4601
    }
4602
    *pFinished = true;
26,825,074✔
4603
    return code;
26,821,782✔
4604
  }
4605

4606
  uint32_t status = 0;
137,778,397✔
4607
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
137,769,541✔
4608

4609
  if (code != TSDB_CODE_SUCCESS) {
137,760,589✔
4610
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
16,224✔
4611
    pTaskInfo->code = code;
16,224✔
4612
    return code;
16,224✔
4613
  }
4614

4615
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
137,744,365✔
4616
    *pFinished = true;
×
4617
    return code;
×
4618
  }
4619

4620
  // current block is filter out according to filter condition, continue load the next block
4621
  if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
137,744,365✔
4622
    *pSkipped = true;
9,984,818✔
4623
    return code;
9,992,066✔
4624
  }
4625

4626
  return code;
127,760,209✔
4627
}
4628

4629
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
165,146,475✔
4630
  STableMergeScanSortSourceParam* source = param;
165,146,475✔
4631

4632
  SOperatorInfo*       pOperator = source->pOperator;
165,146,475✔
4633
  STableMergeScanInfo* pInfo = pOperator->info;
165,147,924✔
4634
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
165,162,029✔
4635
  SSDataBlock*         pBlock = NULL;
165,160,232✔
4636
  int64_t              st = taosGetTimestampUs();
165,152,549✔
4637
  int32_t              code = TSDB_CODE_SUCCESS;
165,152,549✔
4638

4639
  while (true) {
4640
    if (pInfo->rtnNextDurationBlocks) {
184,312,680✔
4641
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
19,702,636✔
4642
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
4643

4644
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
19,712,697✔
4645
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
9,169,684✔
4646
        ++pInfo->nextDurationBlocksIdx;
9,169,684✔
4647
      } else {
4648
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
19,708,153✔
4649
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
9,168,650✔
4650
          pInfo->nextDurationBlocks[i] = NULL;
9,168,650✔
4651
        }
4652

4653
        pInfo->rtnNextDurationBlocks = false;
10,540,088✔
4654
        pInfo->nextDurationBlocksIdx = 0;
10,539,503✔
4655
        pInfo->numNextDurationBlocks = 0;
10,540,088✔
4656
        continue;
10,539,503✔
4657
      }
4658
    } else {
4659
      bool bFinished = false;
164,605,515✔
4660
      bool bSkipped = false;
164,607,127✔
4661

4662
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
164,615,884✔
4663
      if (code != 0) {
164,581,358✔
4664
        return code;
16,224✔
4665
      }
4666

4667
      pBlock = pInfo->pReaderBlock;
164,565,134✔
4668
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
164,590,400✔
4669
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
4670
      if (bFinished) {
164,602,544✔
4671
        pInfo->bNewFilesetEvent = false;
26,825,575✔
4672
        break;
26,825,575✔
4673
      }
4674

4675
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
137,776,969✔
4676
        if (!bSkipped) {
10,539,305✔
4677
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
9,167,929✔
4678
          if (code) {
9,169,684✔
4679
            *ppBlock = NULL;
×
4680
            return code;
×
4681
          }
4682

4683
          ++pInfo->numNextDurationBlocks;
9,169,684✔
4684
          if (pInfo->numNextDurationBlocks > 2) {
9,168,953✔
4685
            qError("%s table merge scan prefetch %d next duration blocks. end early.", GET_TASKID(pTaskInfo),
×
4686
                   pInfo->numNextDurationBlocks);
4687
            pInfo->bNewFilesetEvent = false;
×
4688
            break;
×
4689
          }
4690
        }
4691

4692
        if (pInfo->bNewFilesetEvent) {
10,540,084✔
4693
          pInfo->rtnNextDurationBlocks = true;
10,540,779✔
4694
          *ppBlock = NULL;
10,539,606✔
4695
          return code;
10,539,463✔
4696
        }
4697

4698
        if (pInfo->bNextDurationBlockEvent) {
×
4699
          pInfo->bNextDurationBlockEvent = false;
×
4700
          continue;
×
4701
        }
4702
      }
4703
      if (bSkipped) continue;
127,239,423✔
4704
    }
4705

4706
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
127,787,482✔
4707

4708
    pOperator->resultInfo.totalRows += pBlock->info.rows;
127,789,493✔
4709
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
127,787,957✔
4710
    *ppBlock = pBlock;
127,788,983✔
4711

4712
    return code;
127,786,465✔
4713
  }
4714

4715
  *ppBlock = NULL;
26,825,575✔
4716
  return code;
26,825,575✔
4717
}
4718

4719
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
23,126,596✔
4720
  int32_t code = TSDB_CODE_SUCCESS;
23,126,596✔
4721
  int32_t lino = 0;
23,126,596✔
4722
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
23,126,596✔
4723
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
23,126,612✔
4724
  SBlockOrderInfo biTs = {0};
23,126,612✔
4725
  SBlockOrderInfo biPk = {0};
23,125,259✔
4726

4727
  int32_t tsTargetSlotId = 0;
23,124,861✔
4728
  int32_t pkTargetSlotId = -1;
23,124,861✔
4729
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
95,675,940✔
4730
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
72,564,751✔
4731
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
72,567,743✔
4732
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
72,567,743✔
4733
      tsTargetSlotId = colInfo->dstSlotId;
23,136,480✔
4734
      biTs.order = order;
23,133,044✔
4735
      biTs.slotId = tsTargetSlotId;
23,133,044✔
4736
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
23,133,044✔
4737
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
23,133,044✔
4738
    }
4739
    // TODO: order by just ts
4740
    if (colInfo->isPk) {
72,555,612✔
4741
      pkTargetSlotId = colInfo->dstSlotId;
2,190,649✔
4742
      biPk.order = order;
2,190,649✔
4743
      biPk.slotId = pkTargetSlotId;
2,190,649✔
4744
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
2,190,649✔
4745
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
2,190,649✔
4746
    }
4747
  }
4748

4749
  void* tmp = taosArrayPush(pSortInfo, &biTs);
23,138,550✔
4750
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
23,138,550✔
4751
  if (pkTargetSlotId != -1) {
23,138,550✔
4752
    tmp = taosArrayPush(pSortInfo, &biPk);
2,190,649✔
4753
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,190,649✔
4754
  }
4755

4756
  (*ppSortArray) = pSortInfo;
23,139,138✔
4757

4758
_end:
23,127,198✔
4759
  if (code != TSDB_CODE_SUCCESS) {
23,124,529✔
4760
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4761
  }
4762
  return code;
23,124,529✔
4763
}
4764

4765
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
37,250,629✔
4766
  STableMergeScanInfo* pTmsInfo = param;
37,250,629✔
4767
  if (type == TSD_READER_NOTIFY_DURATION_START) {
37,250,629✔
4768
    pTmsInfo->bNewFilesetEvent = true;
37,262,640✔
4769
  } else if (type == TSD_READER_NOTIFY_NEXT_DURATION_BLOCK) {
×
4770
    pTmsInfo->bNextDurationBlockEvent = true;
×
4771
  }
4772
  qDebug("table merge scan receive notification. type %d, fileset %d", type, info->duration.filesetId);
37,265,676✔
4773

4774
  return;
37,286,936✔
4775
}
4776

4777
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
37,357,201✔
4778
  STableMergeScanInfo* pInfo = pOperator->info;
37,357,201✔
4779
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
37,369,636✔
4780

4781
  int32_t code = TSDB_CODE_SUCCESS;
37,370,227✔
4782
  int32_t lino = 0;
37,370,227✔
4783
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
37,370,227✔
4784

4785
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
37,373,021✔
4786
  pInfo->bNewFilesetEvent = false;
37,386,079✔
4787
  pInfo->bNextDurationBlockEvent = false;
37,375,025✔
4788

4789
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
37,377,957✔
4790
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
37,378,506✔
4791

4792
  pInfo->pSortHandle = NULL;
37,376,969✔
4793
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
74,755,811✔
4794
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
37,373,065✔
4795
  if (code) {
37,361,832✔
4796
    return code;
×
4797
  }
4798

4799
  if (pInfo->bSortRowId && numOfTable != 1) {
37,361,832✔
4800
    int32_t memSize = 512 * 1024 * 1024;
3,964,673✔
4801
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
3,964,673✔
4802
    QUERY_CHECK_CODE(code, lino, _end);
3,967,013✔
4803
  }
4804

4805
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
37,369,227✔
4806
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
37,368,699✔
4807
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
37,368,395✔
4808

4809
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
37,362,939✔
4810
  QUERY_CHECK_CODE(code, lino, _end);
37,367,348✔
4811

4812
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
37,367,348✔
4813
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
37,359,516✔
4814
  param->pOperator = pOperator;
37,359,516✔
4815

4816
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
37,360,316✔
4817
  if (ps == NULL) {
37,361,670✔
4818
    taosMemoryFree(param);
×
4819
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
4820
  }
4821

4822
  ps->param = param;
37,361,670✔
4823
  ps->onlyRef = false;
37,369,480✔
4824
  code = tsortAddSource(pInfo->pSortHandle, ps);
37,371,575✔
4825
  QUERY_CHECK_CODE(code, lino, _end);
37,374,043✔
4826

4827
  if (numOfTable == 1) {
37,374,043✔
4828
    tsortSetSingleTableMerge(pInfo->pSortHandle);
25,111,731✔
4829
  } else {
4830
    code = tsortOpen(pInfo->pSortHandle);
12,262,312✔
4831
    QUERY_CHECK_CODE(code, lino, _end);
12,266,673✔
4832
  }
4833

4834
_end:
12,258,561✔
4835
  if (code != TSDB_CODE_SUCCESS) {
37,375,508✔
4836
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
8,112✔
4837
  }
4838
  return code;
37,378,420✔
4839
}
4840

4841
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
37,368,209✔
4842
  STableMergeScanInfo* pInfo = pOperator->info;
37,368,209✔
4843
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
37,368,209✔
4844
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
37,367,854✔
4845

4846
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
37,367,269✔
4847
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
37,367,195✔
4848
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
37,367,195✔
4849
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
37,367,195✔
4850
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
37,366,710✔
4851
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
37,367,207✔
4852

4853
  tsortDestroySortHandle(pInfo->pSortHandle);
37,365,683✔
4854
  pInfo->pSortHandle = NULL;
37,367,372✔
4855
}
37,367,715✔
4856

4857
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
26,830,207✔
4858
  int32_t              code = TSDB_CODE_SUCCESS;
26,830,207✔
4859
  int32_t              lino = 0;
26,830,207✔
4860
  STableMergeScanInfo* pInfo = pOperator->info;
26,830,207✔
4861
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
26,836,828✔
4862
  SReadHandle*         pHandle = &pInfo->base.readHandle;
26,830,954✔
4863
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
26,826,575✔
4864
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
26,823,611✔
4865

4866
  {
4867
    int32_t numOfTables = 0;
26,824,737✔
4868
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
26,818,625✔
4869
    QUERY_CHECK_CODE(code, lino, _end);
26,829,110✔
4870

4871
    int32_t i = pInfo->tableStartIndex + 1;
26,829,110✔
4872
    for (; i < numOfTables; ++i) {
130,413,765✔
4873
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
109,628,589✔
4874
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
109,629,067✔
4875
      if (tableKeyInfo->groupId != pInfo->groupId) {
109,629,067✔
4876
        break;
6,038,548✔
4877
      }
4878
    }
4879
    pInfo->tableEndIndex = i - 1;
26,823,724✔
4880
  }
4881
  pInfo->bGroupProcessed = false;
26,835,632✔
4882
  int32_t tableStartIdx = pInfo->tableStartIndex;
26,837,066✔
4883
  int32_t tableEndIdx = pInfo->tableEndIndex;
26,837,589✔
4884

4885
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
26,836,662✔
4886
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
26,836,662✔
4887
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
53,665,521✔
4888
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
26,832,997✔
4889
  QUERY_CHECK_CODE(code, lino, _end);
26,825,591✔
4890
  if (pInfo->filesetDelimited) {
26,825,591✔
4891
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
26,834,373✔
4892
  }
4893
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
26,825,305✔
4894

4895
  code = startDurationForGroupTableMergeScan(pOperator);
26,824,385✔
4896
  QUERY_CHECK_CODE(code, lino, _end);
26,831,139✔
4897

4898
_end:
26,823,027✔
4899
  if (code != TSDB_CODE_SUCCESS) {
26,834,071✔
4900
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
8,112✔
4901
    pTaskInfo->code = code;
8,112✔
4902
    T_LONG_JMP(pTaskInfo->env, code);
8,112✔
4903
  }
4904
}
26,825,959✔
4905

4906
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
26,827,087✔
4907
  STableMergeScanInfo* pInfo = pOperator->info;
26,827,087✔
4908
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
26,827,087✔
4909
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
26,827,087✔
4910

4911
  stopDurationForGroupTableMergeScan(pOperator);
26,827,087✔
4912

4913
  if (pInfo->base.dataReader != NULL) {
26,824,253✔
4914
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
26,825,246✔
4915
    pInfo->base.dataReader = NULL;
26,820,246✔
4916
  }
4917
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
26,824,258✔
4918
    if (pInfo->nextDurationBlocks[i]) {
1,034✔
4919
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
1,034✔
4920
      pInfo->nextDurationBlocks[i] = NULL;
1,034✔
4921
    }
4922
    pInfo->numNextDurationBlocks = 0;
1,034✔
4923
    pInfo->nextDurationBlocksIdx = 0;
1,034✔
4924
  }
4925
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
26,823,584✔
4926
  taosHashCleanup(pInfo->mSkipTables);
26,817,570✔
4927
  pInfo->mSkipTables = NULL;
26,814,452✔
4928
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
26,815,644✔
4929
}
26,827,763✔
4930

4931
// all data produced by this function only belongs to one group
4932
// slimit/soffset does not need to be concerned here, since this function only deal with data within one group.
4933
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
73,493,684✔
4934
                                              SOperatorInfo* pOperator) {
4935
  STableMergeScanInfo* pInfo = pOperator->info;
73,493,684✔
4936
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
73,498,743✔
4937
  STupleHandle*        pTupleHandle = NULL;
73,500,836✔
4938

4939
  blockDataCleanup(pResBlock);
73,495,567✔
4940

4941
  while (1) {
×
4942
    while (1) {
2,147,483,647✔
4943
      pTupleHandle = NULL;
2,147,483,647✔
4944
      int32_t code = tsortNextTuple(pHandle, &pTupleHandle);
2,147,483,647✔
4945
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
4946
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4947
        T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
4948
      }
4949
      if (pTupleHandle == NULL) {
2,147,483,647✔
4950
        break;
61,098,194✔
4951
      }
4952

4953
      code = tsortAppendTupleToBlock(pInfo->pSortHandle, pResBlock, pTupleHandle);
2,147,483,647✔
4954
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
4955
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4956
        T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4957
      }
4958

4959
      if (pResBlock->info.rows >= capacity) {
2,147,483,647✔
4960
        break;
12,390,059✔
4961
      }
4962
    }
4963

4964
    if (tsortIsClosed(pHandle)) {
73,488,253✔
4965
      terrno = TSDB_CODE_TSC_QUERY_CANCELLED;
×
4966
      T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4967
    }
4968

4969
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
73,493,547✔
4970
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
73,493,138✔
4971
           pInfo->limitInfo.numOfOutputRows);
4972
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
73,494,223✔
4973
      break;
4974
    }
4975
  }
4976
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
73,494,223✔
4977
}
4978

4979
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
59,637,039✔
4980
  if (pOperator->status == OP_EXEC_DONE) {
59,637,039✔
4981
    (*ppRes) = NULL;
25,284✔
4982
    return TSDB_CODE_SUCCESS;
25,284✔
4983
  }
4984

4985
  int32_t              code = TSDB_CODE_SUCCESS;
59,608,937✔
4986
  int32_t              lino = 0;
59,608,937✔
4987
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
59,608,937✔
4988
  STableMergeScanInfo* pInfo = pOperator->info;
59,610,367✔
4989

4990
  code = pOperator->fpSet._openFn(pOperator);
59,610,381✔
4991
  QUERY_CHECK_CODE(code, lino, _end);
59,612,980✔
4992

4993
  int64_t st = taosGetTimestampUs();
59,608,816✔
4994

4995
  int32_t tableListSize = 0;
59,608,816✔
4996
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
59,606,484✔
4997
  QUERY_CHECK_CODE(code, lino, _end);
59,623,600✔
4998

4999
  if (!pInfo->hasGroupId) {
59,623,600✔
5000
    pInfo->hasGroupId = true;
23,496,363✔
5001

5002
    if (tableListSize == 0) {
23,483,045✔
5003
      setOperatorCompleted(pOperator);
2,695,575✔
5004
      (*ppRes) = NULL;
2,695,309✔
5005
      return code;
2,695,309✔
5006
    }
5007
    pInfo->tableStartIndex = 0;
20,787,470✔
5008
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
20,789,950✔
5009
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
20,790,676✔
5010
    pInfo->groupId = tmp->groupId;
20,790,676✔
5011
    startGroupTableMergeScan(pOperator);
20,793,879✔
5012
  }
5013

5014
  SSDataBlock* pBlock = NULL;
56,914,752✔
5015
  while (pInfo->tableStartIndex < tableListSize) {
73,491,915✔
5016
    if (isTaskKilled(pTaskInfo)) {
73,498,563✔
5017
      goto _end;
×
5018
    }
5019

5020
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
73,497,208✔
5021
                                              pOperator);
5022
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
73,495,643✔
5023
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
1,769,901✔
5024
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
1,769,901✔
5025
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
1,769,901✔
5026
    }
5027
    if (pBlock != NULL) {
73,495,752✔
5028
      pBlock->info.id.groupId = pInfo->groupId;
36,127,941✔
5029
      pOperator->resultInfo.totalRows += pBlock->info.rows;
36,128,178✔
5030
      pInfo->bGroupProcessed = true;
36,126,657✔
5031
      break;
36,126,669✔
5032
    } else {
5033
      if (pInfo->bNewFilesetEvent) {
37,367,811✔
5034
        stopDurationForGroupTableMergeScan(pOperator);
10,541,122✔
5035
        code = startDurationForGroupTableMergeScan(pOperator);
10,541,122✔
5036
        QUERY_CHECK_CODE(code, lino, _end);
10,539,609✔
5037
      } else {
5038
        // Data of this group are all dumped, let's try the next group
5039
        stopGroupTableMergeScan(pOperator);
26,827,087✔
5040
        if (pInfo->tableEndIndex >= tableListSize - 1) {
26,824,668✔
5041
          setOperatorCompleted(pOperator);
20,787,060✔
5042
          break;
20,787,821✔
5043
        }
5044

5045
        pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
6,038,548✔
5046
        STableKeyInfo* tmp = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
6,038,548✔
5047
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
6,038,548✔
5048
        pInfo->groupId = tmp->groupId;
6,038,548✔
5049
        startGroupTableMergeScan(pOperator);
6,038,548✔
5050
        resetLimitInfoForNextGroup(&pInfo->limitInfo);
6,038,051✔
5051
      }
5052
    }
5053
  }
5054

5055
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
56,913,563✔
5056

5057
_end:
56,913,309✔
5058
  if (code != TSDB_CODE_SUCCESS) {
56,913,787✔
5059
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5060
    pTaskInfo->code = code;
×
5061
    T_LONG_JMP(pTaskInfo->env, code);
×
5062
  } else {
5063
    (*ppRes) = pBlock;
56,913,787✔
5064
  }
5065

5066
  return code;
56,914,788✔
5067
}
5068

5069
static SSDataBlock* doTableMergeScan(SOperatorInfo* pOperator) {
×
5070
  SSDataBlock* pRes = NULL;
×
5071
  int32_t      code = doTableMergeScanNext(pOperator, &pRes);
×
5072
  return pRes;
×
5073
}
5074

5075
void destroyTableMergeScanOperatorInfo(void* param) {
23,140,646✔
5076
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
23,140,646✔
5077

5078
  // start one reader variable
5079
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
23,140,646✔
5080
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
23,141,491✔
5081
    pTableScanInfo->base.dataReader = NULL;
23,140,490✔
5082
  }
5083

5084
  for (int32_t i = 0; i < pTableScanInfo->numNextDurationBlocks; ++i) {
23,140,646✔
5085
    if (pTableScanInfo->nextDurationBlocks[i] != NULL) {
×
5086
      blockDataDestroy(pTableScanInfo->nextDurationBlocks[i]);
×
5087
      pTableScanInfo->nextDurationBlocks[i] = NULL;
×
5088
    }
5089
  }
5090

5091
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
23,140,302✔
5092
  pTableScanInfo->pSortHandle = NULL;
23,139,505✔
5093
  taosHashCleanup(pTableScanInfo->mSkipTables);
23,139,505✔
5094
  pTableScanInfo->mSkipTables = NULL;
23,139,311✔
5095
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
23,140,005✔
5096
  pTableScanInfo->pSortInputBlock = NULL;
23,141,496✔
5097
  // end one reader variable
5098

5099
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
23,141,840✔
5100

5101
  blockDataDestroy(pTableScanInfo->pResBlock);
23,140,534✔
5102
  pTableScanInfo->pResBlock = NULL;
23,140,852✔
5103

5104
  // remove it from the task->result list
5105
  blockDataDestroy(pTableScanInfo->pReaderBlock);
23,140,987✔
5106
  pTableScanInfo->pReaderBlock = NULL;
23,140,603✔
5107
  taosArrayDestroy(pTableScanInfo->pSortInfo);
23,140,946✔
5108

5109
  stopSubTablesTableMergeScan(pTableScanInfo);
23,140,701✔
5110

5111
  taosMemoryFreeClear(param);
23,139,508✔
5112
}
23,135,795✔
5113

5114
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
1,192,218✔
5115
  if (pOptr == NULL) {
1,192,218✔
5116
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
5117
    return TSDB_CODE_INVALID_PARA;
×
5118
  }
5119
  // TODO: merge these two info into one struct
5120
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
1,192,218✔
5121
  if (!execInfo) {
1,192,218✔
5122
    return terrno;
×
5123
  }
5124
  STableMergeScanInfo* pInfo = pOptr->info;
1,192,218✔
5125
  execInfo->blockRecorder = pInfo->base.readRecorder;
1,192,218✔
5126
  execInfo->sortExecInfo = pInfo->sortExecInfo;
1,192,218✔
5127

5128
  *pOptrExplain = execInfo;
1,192,218✔
5129
  *len = sizeof(STableMergeScanExecInfo);
1,192,218✔
5130

5131
  return TSDB_CODE_SUCCESS;
1,192,218✔
5132
}
5133

5134
static int32_t resetTableMergeScanOperatorState(SOperatorInfo* pOper) {
2,172,240✔
5135
  int32_t         code = TSDB_CODE_SUCCESS;
2,172,240✔
5136
  STableMergeScanInfo*   pInfo = pOper->info;
2,172,240✔
5137
  pOper->status = OP_NOT_OPENED;
2,172,240✔
5138

5139
  pInfo->tableEndIndex = 0;
2,172,240✔
5140
  pInfo->tableStartIndex = 0;
2,172,240✔
5141
  pInfo->hasGroupId = false;
2,172,240✔
5142
  if (pInfo->base.readerAPI.tsdReaderClose) {
2,172,240✔
5143
    pInfo->base.readerAPI.tsdReaderClose(pInfo->base.dataReader);
2,172,240✔
5144
  }
5145
  pInfo->base.dataReader = NULL;
2,172,240✔
5146
  pInfo->base.scanFlag = MAIN_SCAN;
2,172,240✔
5147

5148
  pInfo->base.limitInfo = (SLimitInfo){0};
2,172,240✔
5149
  pInfo->base.limitInfo.limit.limit = -1;
2,172,240✔
5150
  pInfo->base.limitInfo.slimit.limit = -1;
2,172,240✔
5151

5152
  tableListDestroy(pInfo->base.pTableListInfo);
2,172,240✔
5153
  pInfo->base.pTableListInfo = tableListCreate();
2,172,240✔
5154
  if (!pInfo->base.pTableListInfo) {
2,172,240✔
5155
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5156
    return terrno;
×
5157
  }
5158
  SExecTaskInfo*         pTaskInfo = pOper->pTaskInfo;
2,172,240✔
5159

5160
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
2,172,240✔
5161
  code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
4,344,480✔
5162
                                  &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
5163
                                  pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
4,344,480✔
5164
  if (code) {
2,172,240✔
5165
    qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
×
5166
    return code;
×
5167
  }
5168

5169
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
2,172,240✔
5170
  cleanupQueryTableDataCond(&pInfo->base.cond);
2,172,240✔
5171
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle, false);
2,172,240✔
5172
  if (code) {
2,171,999✔
5173
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
5174
    return code;
×
5175
  }
5176

5177
  tsortDestroySortHandle(pInfo->pSortHandle);
2,171,999✔
5178
  pInfo->pSortHandle = NULL;
2,171,762✔
5179

5180
  taosHashCleanup(pInfo->mSkipTables);
2,172,003✔
5181
  pInfo->mSkipTables = NULL;
2,172,003✔
5182
  pOper->resultInfo.totalRows = 0;
2,172,003✔
5183
  
5184
  pInfo->bGroupProcessed = false;
2,172,003✔
5185
  pInfo->bNewFilesetEvent = false;
2,171,999✔
5186
  pInfo->bNextDurationBlockEvent = false;
2,171,999✔
5187
  pInfo->rtnNextDurationBlocks = false;
2,171,999✔
5188
  pInfo->nextDurationBlocksIdx = 0;
2,171,999✔
5189
  pInfo->bSortRowId = false;
2,171,999✔
5190

5191
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
2,171,999✔
5192
    if (pInfo->nextDurationBlocks[i] != NULL) {
×
5193
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
×
5194
      pInfo->nextDurationBlocks[i] = NULL;
×
5195
    }
5196
  }
5197
  pInfo->numNextDurationBlocks = 0;
2,172,240✔
5198

5199
  stopSubTablesTableMergeScan(pInfo);
2,172,240✔
5200
  return code;
2,171,766✔
5201
}
5202

5203
int32_t createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
23,117,451✔
5204
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
5205
                                         SOperatorInfo** pOptrInfo) {
5206
  QRY_PARAM_CHECK(pOptrInfo);
23,117,451✔
5207

5208
  int32_t              code = TSDB_CODE_SUCCESS;
23,122,172✔
5209
  int32_t              lino = 0;
23,122,172✔
5210
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
23,122,172✔
5211
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
23,098,125✔
5212
  if (pInfo == NULL || pOperator == NULL) {
23,098,586✔
5213
    code = terrno;
×
5214
    goto _error;
×
5215
  }
5216

5217
  pOperator->pPhyNode = pTableScanNode;
23,098,924✔
5218
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
23,098,927✔
5219

5220
  int32_t numOfCols = 0;
23,099,205✔
5221
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
23,114,728✔
5222
                             &pInfo->base.matchInfo);
5223
  QUERY_CHECK_CODE(code, lino, _error);
23,126,251✔
5224

5225
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, false);
23,126,251✔
5226
  QUERY_CHECK_CODE(code, lino, _error);
23,092,997✔
5227

5228
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
23,092,997✔
5229
    SExprSupp* pSup = &pInfo->base.pseudoSup;
10,292,747✔
5230
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
10,288,877✔
5231
    QUERY_CHECK_CODE(code, lino, _error);
10,295,578✔
5232

5233
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
10,295,578✔
5234
                                      &pTaskInfo->storageAPI.functionStore);
5235
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
10,289,834✔
5236
  }
5237

5238
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
23,093,042✔
5239

5240
  if (readHandle->streamRtInfo == NULL) {
23,079,007✔
5241
    pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
23,093,310✔
5242
    QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
23,126,928✔
5243
  }
5244
  
5245
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
23,109,777✔
5246
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
23,134,541✔
5247
  pInfo->base.scanFlag = MAIN_SCAN;
23,120,219✔
5248
  pInfo->base.readHandle = *readHandle;
23,115,285✔
5249

5250
  pInfo->readIdx = -1;
23,115,471✔
5251

5252
  pInfo->base.limitInfo.limit.limit = -1;
23,119,495✔
5253
  pInfo->base.limitInfo.slimit.limit = -1;
23,117,977✔
5254
  pInfo->base.pTableListInfo = pTableListInfo;
23,127,226✔
5255

5256
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
23,126,167✔
5257
  pInfo->sample.seed = taosGetTimestampSec();
23,131,400✔
5258

5259
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
23,136,354✔
5260
                            pTaskInfo->pStreamRuntimeInfo);
23,124,332✔
5261
  QUERY_CHECK_CODE(code, lino, _error);
23,110,567✔
5262

5263
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
23,110,567✔
5264

5265
  pInfo->mergeLimit = -1;
23,104,465✔
5266
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
23,107,504✔
5267
  if (hasLimit) {
23,116,202✔
5268
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1
3,105,736✔
5269
                            ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset
861,638✔
5270
                            : pInfo->limitInfo.limit.limit;
2,414,742✔
5271
    pInfo->mSkipTables = NULL;
1,551,938✔
5272
  }
5273

5274
  initResultSizeInfo(&pOperator->resultInfo, 1024);
23,114,698✔
5275
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
23,120,970✔
5276
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
23,139,439✔
5277
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
23,119,467✔
5278
  QUERY_CHECK_CODE(code, lino, _error);
23,134,458✔
5279
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
23,134,458✔
5280
    pInfo->bSortRowId = true;
5,291,723✔
5281
  } else {
5282
    pInfo->bSortRowId = false;
17,843,429✔
5283
  }
5284

5285
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
23,131,713✔
5286
  QUERY_CHECK_CODE(code, lino, _error);
23,130,183✔
5287

5288
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
23,130,183✔
5289
  QUERY_CHECK_CODE(code, lino, _error);
23,112,261✔
5290

5291
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
23,112,261✔
5292
  QUERY_CHECK_CODE(code, lino, _error);
23,127,846✔
5293

5294
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
23,127,846✔
5295

5296
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
23,126,359✔
5297
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
23,125,733✔
5298

5299
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
23,116,413✔
5300

5301
  // start one reader variable
5302
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
23,135,608✔
5303
  QUERY_CHECK_CODE(code, lino, _error);
23,126,831✔
5304

5305
  if (!tsExperimental) {
23,126,831✔
5306
    pInfo->filesetDelimited = false;
×
5307
  } else {
5308
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
23,126,831✔
5309
  }
5310
  // end one reader variable
5311
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
23,132,800✔
5312
                  pInfo, pTaskInfo);
5313
  pOperator->exprSupp.numOfExprs = numOfCols;
23,123,534✔
5314

5315
  pOperator->fpSet = createOperatorFpSet(
23,120,378✔
5316
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
23,133,369✔
5317
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
5318
      NULL);
5319
  setOperatorResetStateFn(pOperator, resetTableMergeScanOperatorState);     
23,115,760✔
5320
  pOperator->cost.openCost = 0;
23,122,620✔
5321

5322
  *pOptrInfo = pOperator;
23,123,605✔
5323
  return code;
23,118,422✔
5324

5325
_error:
×
5326
  if (code != TSDB_CODE_SUCCESS) {
×
5327
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5328
  }
5329
  pTaskInfo->code = code;
×
5330
  if (pInfo != NULL) {
×
5331
    pInfo->base.pTableListInfo = NULL;
×
5332
    destroyTableMergeScanOperatorInfo(pInfo);
×
5333
  }
5334
  if (pOperator != NULL) {
×
5335
    pOperator->info = NULL;
×
5336
    destroyOperator(pOperator);
×
5337
  }
5338
  return code;
×
5339
}
5340

5341
// ====================================================================================================================
5342
// TableCountScanOperator
5343
static void        destoryTableCountScanOperator(void* param);
5344
static int32_t     buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
5345
                                                  SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI);
5346
static int32_t     buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
5347
                                                  SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI);
5348
static int32_t     buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
5349
                                             STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName);
5350
static int32_t     buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
5351
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName);
5352
static int32_t     buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
5353
                                          STableCountScanSupp* pSupp, SSDataBlock* pRes);
5354
static void        buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
5355
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
5356
                                               size_t perfdbTableNum);
5357
static void        buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
5358
                                              size_t infodbTableNum, size_t perfdbTableNum);
5359
static const char* GROUP_TAG_DB_NAME = "db_name";
5360
static const char* GROUP_TAG_STABLE_NAME = "stable_name";
5361

5362
int32_t tblCountScanGetGroupTagsSlotId(const SNodeList* scanCols, STableCountScanSupp* supp) {
24,171✔
5363
  if (scanCols != NULL) {
24,171✔
5364
    SNode* pNode = NULL;
13,503✔
5365
    FOREACH(pNode, scanCols) {
31,499✔
5366
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
17,996✔
5367
        return TSDB_CODE_QRY_SYS_ERROR;
×
5368
      }
5369
      STargetNode* targetNode = (STargetNode*)pNode;
17,996✔
5370
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
17,996✔
5371
        return TSDB_CODE_QRY_SYS_ERROR;
×
5372
      }
5373
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
17,996✔
5374
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
17,996✔
5375
        supp->dbNameSlotId = targetNode->slotId;
8,827✔
5376
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
9,169✔
5377
        supp->stbNameSlotId = targetNode->slotId;
9,169✔
5378
      }
5379
    }
5380
  }
5381
  return TSDB_CODE_SUCCESS;
24,171✔
5382
}
5383

5384
int32_t tblCountScanGetCountSlotId(const SNodeList* pseudoCols, STableCountScanSupp* supp) {
24,171✔
5385
  if (pseudoCols != NULL) {
24,171✔
5386
    SNode* pNode = NULL;
24,171✔
5387
    FOREACH(pNode, pseudoCols) {
48,342✔
5388
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
24,171✔
5389
        return TSDB_CODE_QRY_SYS_ERROR;
×
5390
      }
5391
      STargetNode* targetNode = (STargetNode*)pNode;
24,171✔
5392
      if (nodeType(targetNode->pExpr) != QUERY_NODE_FUNCTION) {
24,171✔
5393
        return TSDB_CODE_QRY_SYS_ERROR;
×
5394
      }
5395
      SFunctionNode* funcNode = (SFunctionNode*)(targetNode->pExpr);
24,171✔
5396
      if (funcNode->funcType == FUNCTION_TYPE_TABLE_COUNT) {
24,171✔
5397
        supp->tbCountSlotId = targetNode->slotId;
24,171✔
5398
      }
5399
    }
5400
  }
5401
  return TSDB_CODE_SUCCESS;
24,171✔
5402
}
5403

5404
int32_t tblCountScanGetInputs(SNodeList* groupTags, SName* tableName, STableCountScanSupp* supp) {
24,171✔
5405
  if (groupTags != NULL) {
24,171✔
5406
    SNode* pNode = NULL;
13,503✔
5407
    FOREACH(pNode, groupTags) {
31,499✔
5408
      if (nodeType(pNode) != QUERY_NODE_COLUMN) {
17,996✔
5409
        return TSDB_CODE_QRY_SYS_ERROR;
×
5410
      }
5411
      SColumnNode* colNode = (SColumnNode*)pNode;
17,996✔
5412
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
17,996✔
5413
        supp->groupByDbName = true;
8,827✔
5414
      }
5415
      if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
17,996✔
5416
        supp->groupByStbName = true;
9,169✔
5417
      }
5418
    }
5419
  } else {
5420
    tstrncpy(supp->dbNameFilter, tNameGetDbNameP(tableName), TSDB_DB_NAME_LEN);
10,668✔
5421
    tstrncpy(supp->stbNameFilter, tNameGetTableName(tableName), TSDB_TABLE_NAME_LEN);
10,668✔
5422
  }
5423
  return TSDB_CODE_SUCCESS;
24,171✔
5424
}
5425

5426
int32_t getTableCountScanSupp(SNodeList* groupTags, SName* tableName, SNodeList* scanCols, SNodeList* pseudoCols,
24,171✔
5427
                              STableCountScanSupp* supp, SExecTaskInfo* taskInfo) {
5428
  int32_t code = 0;
24,171✔
5429
  code = tblCountScanGetInputs(groupTags, tableName, supp);
24,171✔
5430
  if (code != TSDB_CODE_SUCCESS) {
24,171✔
5431
    qError("%s get table count scan supp. get inputs error", GET_TASKID(taskInfo));
×
5432
    return code;
×
5433
  }
5434

5435
  supp->dbNameSlotId = -1;
24,171✔
5436
  supp->stbNameSlotId = -1;
24,171✔
5437
  supp->tbCountSlotId = -1;
24,171✔
5438

5439
  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
24,171✔
5440
  if (code != TSDB_CODE_SUCCESS) {
24,171✔
5441
    qError("%s get table count scan supp. get group tags slot id error", GET_TASKID(taskInfo));
×
5442
    return code;
×
5443
  }
5444

5445
  code = tblCountScanGetCountSlotId(pseudoCols, supp);
24,171✔
5446
  if (code != TSDB_CODE_SUCCESS) {
24,171✔
5447
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
×
5448
    return code;
×
5449
  }
5450
  return code;
24,171✔
5451
}
5452

5453
int32_t createTableCountScanOperatorInfo(SReadHandle* readHandle, STableCountScanPhysiNode* pTblCountScanNode,
24,171✔
5454
                                         SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
5455
  QRY_PARAM_CHECK(pOptrInfo);
24,171✔
5456

5457
  int32_t                      code = TSDB_CODE_SUCCESS;
24,171✔
5458
  int32_t                      lino = 0;
24,171✔
5459
  SScanPhysiNode*              pScanNode = &pTblCountScanNode->scan;
24,171✔
5460
  STableCountScanOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableCountScanOperatorInfo));
24,171✔
5461
  SOperatorInfo*               pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
24,171✔
5462
  if (!pInfo || !pOperator) {
24,171✔
5463
    code = terrno;
×
5464
    goto _error;
×
5465
  }
5466

5467
  pInfo->readHandle = *readHandle;
24,171✔
5468

5469
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
24,171✔
5470
  initResultSizeInfo(&pOperator->resultInfo, 1);
24,171✔
5471
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
24,171✔
5472
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
24,171✔
5473

5474
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
24,171✔
5475
  QUERY_CHECK_CODE(code, lino, _error);
24,171✔
5476

5477
  code = getTableCountScanSupp(pTblCountScanNode->pGroupTags, &pTblCountScanNode->scan.tableName,
24,171✔
5478
                               pTblCountScanNode->scan.pScanCols, pTblCountScanNode->scan.pScanPseudoCols, &pInfo->supp,
5479
                               pTaskInfo);
5480
  QUERY_CHECK_CODE(code, lino, _error);
24,171✔
5481

5482
  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
24,171✔
5483
                  pInfo, pTaskInfo);
5484
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScanNext, NULL, destoryTableCountScanOperator,
24,171✔
5485
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
5486
  *pOptrInfo = pOperator;
24,171✔
5487
  return code;
24,171✔
5488

5489
_error:
×
5490
  if (pInfo != NULL) {
×
5491
    destoryTableCountScanOperator(pInfo);
×
5492
  }
5493
  if (pOperator != NULL) {
×
5494
    pOperator->info = NULL;
×
5495
    destroyOperator(pOperator);
×
5496
  }
5497
  pTaskInfo->code = code;
×
5498
  return code;
×
5499
}
5500

5501
int32_t fillTableCountScanDataBlock(STableCountScanSupp* pSupp, char* dbName, char* stbName, int64_t count,
33,434✔
5502
                                    SSDataBlock* pRes) {
5503
  int32_t code = TSDB_CODE_SUCCESS;
33,434✔
5504
  int32_t lino = 0;
33,434✔
5505
  if (pSupp->dbNameSlotId != -1) {
33,434✔
5506
    QUERY_CHECK_CONDITION((strlen(dbName) > 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
13,298✔
5507
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
13,298✔
5508
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
13,298✔
5509

5510
    char varDbName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
13,298✔
5511
    tstrncpy(varDataVal(varDbName), dbName, TSDB_DB_NAME_LEN);
13,298✔
5512

5513
    varDataSetLen(varDbName, strlen(dbName));
13,298✔
5514
    code = colDataSetVal(colInfoData, 0, varDbName, false);
13,298✔
5515
    QUERY_CHECK_CODE(code, lino, _end);
12,867✔
5516
  }
5517

5518
  if (pSupp->stbNameSlotId != -1) {
33,164✔
5519
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
17,274✔
5520
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
17,248✔
5521
    if (strlen(stbName) != 0) {
17,248✔
5522
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
12,560✔
5523
      tstrncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
12,560✔
5524
      varDataSetLen(varStbName, strlen(stbName));
12,560✔
5525
      code = colDataSetVal(colInfoData, 0, varStbName, false);
12,560✔
5526
      QUERY_CHECK_CODE(code, lino, _end);
12,560✔
5527
    } else {
5528
      colDataSetNULL(colInfoData, 0);
5529
    }
5530
  }
5531

5532
  if (pSupp->tbCountSlotId != -1) {
33,273✔
5533
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
33,434✔
5534
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
33,434✔
5535
    code = colDataSetVal(colInfoData, 0, (char*)&count, false);
33,434✔
5536
    QUERY_CHECK_CODE(code, lino, _end);
33,434✔
5537
  }
5538
  pRes->info.rows = 1;
33,434✔
5539

5540
_end:
33,434✔
5541
  if (code != TSDB_CODE_SUCCESS) {
33,434✔
5542
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5543
  }
5544
  return code;
33,434✔
5545
}
5546

5547
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
10,017✔
5548
  STableCountScanSupp* pSupp = &pInfo->supp;
10,017✔
5549
  SSDataBlock*         pRes = pInfo->pRes;
10,017✔
5550

5551
  size_t infodbTableNum;
10,017✔
5552
  getInfosDbMeta(NULL, &infodbTableNum);
10,017✔
5553
  infodbTableNum -= 1;
10,017✔
5554
  size_t perfdbTableNum;
10,017✔
5555
  getPerfDbMeta(NULL, &perfdbTableNum);
10,017✔
5556

5557
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
10,017✔
5558
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
8,589✔
5559
    return (pRes->info.rows > 0) ? pRes : NULL;
8,589✔
5560
  } else {
5561
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
1,428✔
5562
    return (pRes->info.rows > 0) ? pRes : NULL;
1,428✔
5563
  }
5564
}
5565

5566
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator,
1,428✔
5567
                                       STableCountScanSupp* pSupp,
5568
                                       SSDataBlock* pRes,
5569
                                       size_t infodbTableNum,
5570
                                       size_t perfdbTableNum) {
5571
  int32_t        code = TSDB_CODE_SUCCESS;
1,428✔
5572
  int32_t        lino = 0;
1,428✔
5573
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
1,428✔
5574
  if (strlen(pSupp->stbNameFilter) != 0) {
1,428✔
5575
    // sys dbs do not contain any stable, so the count is always 0
5576
    code = fillTableCountScanDataBlock(pSupp, "", "", 0, pRes);
486✔
5577
    QUERY_CHECK_CODE(code, lino, _end);
486✔
5578
  } else if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
942✔
5579
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB,
162✔
5580
                                       "", infodbTableNum, pRes);
5581
    QUERY_CHECK_CODE(code, lino, _end);
162✔
5582
  } else if (strcmp(pSupp->dbNameFilter, TSDB_PERFORMANCE_SCHEMA_DB) == 0) {
780✔
5583
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB,
×
5584
                                       "", perfdbTableNum, pRes);
5585
    QUERY_CHECK_CODE(code, lino, _end);
×
5586
  } else if (strlen(pSupp->dbNameFilter) == 0) {
780✔
5587
    code = fillTableCountScanDataBlock(pSupp, "", "",
618✔
5588
                                       infodbTableNum + perfdbTableNum, pRes);
618✔
5589
    QUERY_CHECK_CODE(code, lino, _end);
618✔
5590
  } else {
5591
    // other dbs do not exist in sys dbs, so the count is always 0
5592
    code = fillTableCountScanDataBlock(pSupp, pSupp->dbNameFilter,
162✔
5593
                                       pSupp->stbNameFilter, 0, pRes);
162✔
5594
    QUERY_CHECK_CODE(code, lino, _end);
162✔
5595
  }
5596

5597
_end:
162✔
5598
  if (code != TSDB_CODE_SUCCESS) {
1,428✔
5599
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5600
    pTaskInfo->code = code;
×
5601
    T_LONG_JMP(pTaskInfo->env, code);
×
5602
  }
5603
  setOperatorCompleted(pOperator);
1,428✔
5604
}
1,428✔
5605

5606
static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
8,589✔
5607
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
5608
                                        size_t perfdbTableNum) {
5609
  int32_t        code = TSDB_CODE_SUCCESS;
8,589✔
5610
  int32_t        lino = 0;
8,589✔
5611
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
8,589✔
5612
  if (pInfo->currGrpIdx == 0) {
8,589✔
5613
    uint64_t groupId = 0;
2,863✔
5614
    if (pSupp->groupByDbName) {
2,863✔
5615
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
1,937✔
5616
    } else {
5617
      groupId = calcGroupId("", 0);
926✔
5618
    }
5619

5620
    pRes->info.id.groupId = groupId;
2,863✔
5621
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
2,863✔
5622
    QUERY_CHECK_CODE(code, lino, _end);
2,863✔
5623
  } else if (pInfo->currGrpIdx == 1) {
5,726✔
5624
    uint64_t groupId = 0;
2,863✔
5625
    if (pSupp->groupByDbName) {
2,863✔
5626
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
1,937✔
5627
    } else {
5628
      groupId = calcGroupId("", 0);
926✔
5629
    }
5630

5631
    pRes->info.id.groupId = groupId;
2,863✔
5632
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
2,863✔
5633
    QUERY_CHECK_CODE(code, lino, _end);
2,863✔
5634
  } else {
5635
    setOperatorCompleted(pOperator);
2,863✔
5636
  }
5637
  pInfo->currGrpIdx++;
8,589✔
5638

5639
_end:
8,589✔
5640
  if (code != TSDB_CODE_SUCCESS) {
8,589✔
5641
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5642
    pTaskInfo->code = code;
×
5643
    T_LONG_JMP(pTaskInfo->env, code);
×
5644
  }
5645
}
8,589✔
5646

5647
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
57,605✔
5648
  int32_t                      code = TSDB_CODE_SUCCESS;
57,605✔
5649
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
57,605✔
5650
  STableCountScanOperatorInfo* pInfo = pOperator->info;
57,605✔
5651
  STableCountScanSupp*         pSupp = &pInfo->supp;
57,605✔
5652
  SSDataBlock*                 pRes = pInfo->pRes;
57,605✔
5653

5654
  blockDataCleanup(pRes);
57,605✔
5655
  QRY_PARAM_CHECK(ppRes);
57,605✔
5656

5657
  if (pOperator->status == OP_EXEC_DONE) {
57,605✔
5658
    return code;
14,114✔
5659
  }
5660

5661
  if (pInfo->readHandle.mnd != NULL) {
43,491✔
5662
    (*ppRes) = buildSysDbTableCount(pOperator, pInfo);
10,017✔
5663
    return code;
10,017✔
5664
  }
5665

5666
  code = buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
33,474✔
5667
  if (code != TSDB_CODE_SUCCESS) {
33,339✔
5668
    qError("%s failed since %s", __func__, tstrerror(code));
×
5669
    pTaskInfo->code = code;
×
5670
    T_LONG_JMP(pTaskInfo->env, code);
×
5671
  }
5672
  if (pRes->info.rows > 0) {
33,339✔
5673
    *ppRes = pRes;
26,280✔
5674
  }
5675

5676
  return code;
33,474✔
5677
}
5678

5679
static int32_t buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
33,339✔
5680
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes) {
5681
  int32_t        code = TSDB_CODE_SUCCESS;
33,339✔
5682
  int32_t        lino = 0;
33,339✔
5683
  const char*    db = NULL;
33,339✔
5684
  int32_t        vgId = 0;
33,474✔
5685
  char           dbName[TSDB_DB_NAME_LEN] = {0};
33,474✔
5686
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
33,474✔
5687
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
33,474✔
5688

5689
  // get dbname
5690
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, &db, &vgId, NULL, NULL);
33,474✔
5691
  SName sn = {0};
33,474✔
5692

5693
  code = tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
33,474✔
5694
  QUERY_CHECK_CODE(code, lino, _end);
33,339✔
5695

5696
  code = tNameGetDbName(&sn, dbName);
33,339✔
5697
  QUERY_CHECK_CODE(code, lino, _end);
33,339✔
5698

5699
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
33,339✔
5700
    code = buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
24,234✔
5701
  } else {
5702
    code = buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
9,240✔
5703
  }
5704

5705
_end:
33,474✔
5706
  if (code != TSDB_CODE_SUCCESS) {
33,474✔
5707
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5708
    pTaskInfo->code = code;
×
5709
  }
5710

5711
  return code;
33,474✔
5712
}
5713

5714
static int32_t buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
24,099✔
5715
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
5716
  int32_t        code = TSDB_CODE_SUCCESS;
24,099✔
5717
  int32_t        lino = 0;
24,099✔
5718
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
24,099✔
5719
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
24,099✔
5720

5721
  if (pSupp->groupByStbName) {
24,234✔
5722
    if (pInfo->stbUidList == NULL) {
20,653✔
5723
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
7,194✔
5724
      QUERY_CHECK_NULL(pInfo->stbUidList, code, lino, _end, terrno);
7,194✔
5725
      code = pAPI->metaFn.storeGetTableList(pInfo->readHandle.vnode, TSDB_SUPER_TABLE, pInfo->stbUidList);
7,194✔
5726
      QUERY_CHECK_CODE(code, lino, _end);
7,194✔
5727
    }
5728

5729
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
20,788✔
5730
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
12,560✔
5731
      code = buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid, pAPI);
12,560✔
5732
      QUERY_CHECK_CODE(code, lino, _end);
12,560✔
5733

5734
      pInfo->currGrpIdx++;
12,560✔
5735
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
8,228✔
5736
      code = buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName, pAPI);
7,194✔
5737
      QUERY_CHECK_CODE(code, lino, _end);
7,194✔
5738

5739
      pInfo->currGrpIdx++;
7,194✔
5740
    } else {
5741
      setOperatorCompleted(pOperator);
1,034✔
5742
    }
5743
  } else {
5744
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
3,446✔
5745
    pRes->info.id.groupId = groupId;
3,446✔
5746

5747
    int64_t dbTableCount = 0;
3,446✔
5748
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &dbTableCount, NULL);
3,446✔
5749
    code = fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
3,446✔
5750
    QUERY_CHECK_CODE(code, lino, _end);
3,446✔
5751
    setOperatorCompleted(pOperator);
3,446✔
5752
  }
5753

5754
_end:
24,234✔
5755
  if (code != TSDB_CODE_SUCCESS) {
24,234✔
5756
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5757
    pTaskInfo->code = code;
×
5758
  }
5759
  return code;
24,234✔
5760
}
5761

5762
static int32_t buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
9,240✔
5763
                                         STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
5764
  int32_t        code = TSDB_CODE_SUCCESS;
9,240✔
5765
  int32_t        lino = 0;
9,240✔
5766
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
9,240✔
5767
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
9,240✔
5768

5769
  if (strlen(pSupp->dbNameFilter) != 0 &&
9,240✔
5770
      strcmp(dbName, pSupp->dbNameFilter) != 0) {
6,384✔
5771
    // db name not match, return count 0
5772
    code = fillTableCountScanDataBlock(pSupp, dbName,
×
5773
                                       pSupp->stbNameFilter, 0, pRes);
×
5774
    QUERY_CHECK_CODE(code, lino, _end);
×
5775
  } else if (strlen(pSupp->stbNameFilter) != 0) {
9,240✔
5776
    uint64_t uid = 0;
2,856✔
5777
    code = pAPI->metaFn.getTableUidByName(pInfo->readHandle.vnode,
5,712✔
5778
                                          pSupp->stbNameFilter, &uid);
2,856✔
5779
    if (code == TSDB_CODE_TDB_TABLE_NOT_EXIST) {
2,856✔
5780
      // table not exist, return count 0
5781
      code = fillTableCountScanDataBlock(pSupp, dbName,
1,296✔
5782
                                         pSupp->stbNameFilter, 0, pRes);
1,296✔
5783
      QUERY_CHECK_CODE(code, lino, _end);
1,296✔
5784
    } else {
5785
      QUERY_CHECK_CODE(code, lino, _end);
1,560✔
5786

5787
      int64_t numOfChildTables = 0;
1,560✔
5788
      code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, uid,
1,560✔
5789
                                              &numOfChildTables, NULL, NULL);
5790
      QUERY_CHECK_CODE(code, lino, _end);
1,560✔
5791

5792
      code = fillTableCountScanDataBlock(pSupp, dbName, pSupp->stbNameFilter,
1,560✔
5793
                                         numOfChildTables, pRes);
5794
      QUERY_CHECK_CODE(code, lino, _end);
1,560✔
5795
    }
5796
  } else {
5797
    int64_t tbNumVnode = 0;
6,384✔
5798
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode,
6,384✔
5799
                              NULL, NULL, &tbNumVnode, NULL);
5800
    code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
6,384✔
5801
    QUERY_CHECK_CODE(code, lino, _end);
6,384✔
5802
  }
5803

5804
_end:
9,240✔
5805
  if (code != TSDB_CODE_SUCCESS) {
9,240✔
5806
    pTaskInfo->code = code;
×
5807
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5808
  }
5809

5810
  setOperatorCompleted(pOperator);
9,240✔
5811
  return code;
9,240✔
5812
}
5813

5814
static int32_t buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
7,194✔
5815
                                              SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI) {
5816
  int32_t code = TSDB_CODE_SUCCESS;
7,194✔
5817
  int32_t lino = 0;
7,194✔
5818
  char    fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
7,194✔
5819
  if (pSupp->groupByDbName) {
7,194✔
5820
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
3,444✔
5821
  }
5822

5823
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
7,194✔
5824
  pRes->info.id.groupId = groupId;
7,194✔
5825

5826
  int64_t numOfTables = 0;
7,194✔
5827
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, NULL, &numOfTables);
7,194✔
5828

5829
  if (numOfTables != 0) {
7,194✔
5830
    code = fillTableCountScanDataBlock(pSupp, dbName, "", numOfTables, pRes);
1,034✔
5831
  }
5832

5833
_end:
6,160✔
5834
  if (code != TSDB_CODE_SUCCESS) {
7,194✔
5835
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5836
  }
5837
  return code;
7,194✔
5838
}
5839

5840
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
12,560✔
5841
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
5842
  int32_t code = TSDB_CODE_SUCCESS;
12,560✔
5843
  int32_t lino = 0;
12,560✔
5844
  char    stbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
12,560✔
5845
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
12,560✔
5846
  QUERY_CHECK_CODE(code, lino, _end);
12,560✔
5847

5848
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
12,560✔
5849
  if (pSupp->groupByDbName) {
12,560✔
5850
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
5,708✔
5851
  } else {
5852
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
6,852✔
5853
  }
5854

5855
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
12,560✔
5856
  pRes->info.id.groupId = groupId;
12,560✔
5857

5858
  int64_t ctbNum = 0;
12,560✔
5859
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL, NULL);
12,560✔
5860
  QUERY_CHECK_CODE(code, lino, _end);
12,425✔
5861
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
12,425✔
5862
  QUERY_CHECK_CODE(code, lino, _end);
12,560✔
5863

5864
_end:
12,560✔
5865
  if (code != TSDB_CODE_SUCCESS) {
12,560✔
5866
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5867
  }
5868
  return code;
12,560✔
5869
}
5870

5871
static void destoryTableCountScanOperator(void* param) {
24,171✔
5872
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
24,171✔
5873
  blockDataDestroy(pTableCountScanInfo->pRes);
24,171✔
5874

5875
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
24,171✔
5876
  taosMemoryFreeClear(param);
24,171✔
5877
}
24,171✔
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