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

taosdata / TDengine / #4913

06 Jan 2026 01:30AM UTC coverage: 64.884% (-0.004%) from 64.888%
#4913

push

travis-ci

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

180 of 319 new or added lines in 14 files covered. (56.43%)

571 existing lines in 128 files now uncovered.

195016 of 300563 relevant lines covered (64.88%)

117540852.85 hits per line

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

78.04
/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) {
582,672,813✔
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;
582,672,813✔
86
#endif
87
}
88

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

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

99
  // 0 by default, which means it is not a interval operator of the upstream operator.
100
  if (pInterval->interval == 0) {
625,734,950✔
101
    *overlap = false;
583,058,858✔
102
    return code;
582,999,789✔
103
  }
104

105
  if (order == TSDB_ORDER_ASC) {
42,443,632✔
106
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
40,526,352✔
107
    if (w.ekey < pBlockInfo->window.skey) {
40,549,612✔
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) {
40,513,074✔
113
      *overlap = true;
32,886,499✔
114
      return code;
32,887,660✔
115
    }
116

117
    while (1) {
118
      getNextTimeWindow(pInterval, &w, order);
7,662,506✔
119
      if (w.skey > pBlockInfo->window.ekey) {
7,696,593✔
120
        break;
7,441,729✔
121
      }
122

123
      if (w.ekey <= pBlockInfo->window.ekey) {
252,670✔
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) {
252,670✔
128
        *overlap = true;
252,670✔
129
        return code;
252,670✔
130
      }
131
    }
132
  } else {
133
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.ekey);
1,917,280✔
134
    if (w.skey > pBlockInfo->window.ekey) {
1,917,308✔
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,917,887✔
140
      *overlap = true;
1,489,712✔
141
      return code;
1,489,712✔
142
    }
143

144
    while (1) {
145
      getNextTimeWindow(pInterval, &w, order);
427,005✔
146
      if (w.ekey < pBlockInfo->window.skey) {
428,203✔
147
        break;
425,737✔
148
      }
149

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

161
  *overlap = false;
7,867,466✔
162
  return code;
7,864,701✔
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) {
150,960,393✔
167
  if (pOperator->operatorType != QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
150,960,393✔
168
    return NULL;
×
169
  }
170

171
  int64_t buf[2] = {0};
150,987,893✔
172
  SET_RES_WINDOW_KEY((char*)buf, &groupId, sizeof(groupId), groupId);
150,939,427✔
173

174
  STableScanInfo* pTableScanInfo = pOperator->info;
150,932,403✔
175

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

179
  if (p1 == NULL) {
151,009,838✔
180
    return NULL;
50,121,425✔
181
  }
182

183
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
100,888,413✔
184
  if (NULL == *pPage) {
100,870,808✔
185
    return NULL;
×
186
  }
187

188
  return (SResultRow*)((char*)(*pPage) + p1->offset);
100,868,611✔
189
}
190

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

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

210
static int32_t doDynamicPruneDataBlock(SOperatorInfo* pOperator, SDataBlockInfo* pBlockInfo, uint32_t* status) {
689,602,792✔
211
  STableScanInfo* pTableScanInfo = pOperator->info;
689,602,792✔
212
  int32_t         code = TSDB_CODE_SUCCESS;
689,872,080✔
213

214
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
689,872,080✔
215
    return TSDB_CODE_SUCCESS;
538,590,333✔
216
  }
217

218
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
150,935,205✔
219

220
  SFilePage*  pPage = NULL;
150,983,353✔
221
  SResultRow* pRow = getTableGroupOutputBuf(pOperator, pBlockInfo->id.groupId, &pPage);
150,905,182✔
222

223
  if (pRow == NULL) {
150,962,050✔
224
    return TSDB_CODE_SUCCESS;
50,083,447✔
225
  }
226

227
  bool notLoadBlock = true;
100,878,603✔
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;
91,201,416✔
236
      break;
91,201,416✔
237
    }
238
  }
239

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

243
  if (notLoadBlock) {
100,871,391✔
244
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
9,674,925✔
245
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
9,674,551✔
246
  }
247

248
  return code;
100,852,822✔
249
}
250

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

258
  return filterRangeExecute(pFilterInfo, pColsAgg, numOfCols, numOfRows, keep);
8,296,158✔
259
}
260

261
static int32_t doLoadBlockSMA(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
150,270,549✔
262
                              bool* pLoad) {
263
  SStorageAPI* pAPI = &pTaskInfo->storageAPI;
150,270,549✔
264
  bool         allColumnsHaveAgg = true;
150,301,794✔
265
  bool         hasNullSMA = false;
150,322,703✔
266
  if (pLoad != NULL) {
150,300,246✔
267
    *pLoad = false;
150,279,028✔
268
  }
269

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

276
  if (!allColumnsHaveAgg || hasNullSMA) {
150,231,720✔
277
    *pLoad = false;
136,463,453✔
278
  } else {
279
    *pLoad = true;
13,768,267✔
280
  }
281

282
  return code;
150,162,343✔
283
}
284

285
static int32_t doSetTagColumnData(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
752,903,363✔
286
                                  int32_t rows) {
287
  int32_t    code = 0;
752,903,363✔
288
  SExprSupp* pSup = &pTableScanInfo->pseudoSup;
752,903,363✔
289
  if (pSup->numOfExprs > 0) {
753,301,477✔
290
    code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
320,298,784✔
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) {
320,103,912✔
294
      if (pTaskInfo->streamInfo.pState) blockDataCleanup(pBlock);
×
295
      code = 0;
×
296
    }
297
  }
298

299
  return code;
753,029,889✔
300
}
301

302
bool applyLimitOffset(SLimitInfo* pLimitInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
1,200,354,333✔
303
  SLimit*     pLimit = &pLimitInfo->limit;
1,200,354,333✔
304
  const char* id = GET_TASKID(pTaskInfo);
1,200,507,705✔
305

306
  if (pLimitInfo->remainOffset > 0) {
1,200,683,062✔
307
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
5,137,313✔
308
      pLimitInfo->remainOffset -= pBlock->info.rows;
2,571,221✔
309
      blockDataEmpty(pBlock);
2,571,221✔
310
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
2,571,221✔
311
      return false;
2,571,221✔
312
    } else {
313
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
2,566,092✔
314
      if (code != TSDB_CODE_SUCCESS) {
2,566,092✔
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,566,092✔
320
    }
321
  }
322

323
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
1,198,290,223✔
324
    // limit the output rows
325
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
27,034,969✔
326
    blockDataKeepFirstNRows(pBlock, keep);
27,031,159✔
327

328
    pLimitInfo->numOfOutputRows += pBlock->info.rows;
27,031,301✔
329
    qDebug("output limit %" PRId64 " has reached, %s", pLimit->limit, id);
27,031,804✔
330
    return true;
27,037,736✔
331
  }
332

333
  pLimitInfo->numOfOutputRows += pBlock->info.rows;
1,170,920,779✔
334
  return false;
1,171,337,760✔
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,
723,759,317✔
342
                             uint32_t* status) {
343
  int32_t        code = TSDB_CODE_SUCCESS;
723,759,317✔
344
  int32_t        lino = 0;
723,759,317✔
345
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
723,759,317✔
346
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
724,024,157✔
347
  bool           loadSMA = false;
723,850,349✔
348

349
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
723,850,349✔
350

351
  pCost->totalBlocks += 1;
723,704,355✔
352
  pCost->totalRows += pBlock->info.rows;
723,840,281✔
353
  *status = pTableScanInfo->dataBlockLoadFlag;
723,378,314✔
354

355
  if (pOperator->exprSupp.pFilterInfo != NULL) {
723,808,291✔
356
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
98,071,211✔
357
  } else {
358
    bool overlap = false;
625,507,669✔
359
    int  ret =
360
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
625,314,831✔
361
    if (ret != TSDB_CODE_SUCCESS) {
625,157,510✔
362
      return ret;
×
363
    }
364
    if (overlap) {
625,157,510✔
365
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
34,625,766✔
366
    }
367
  }
368

369
  SDataBlockInfo* pBlockInfo = &pBlock->info;
723,654,907✔
370
  taosMemoryFreeClear(pBlock->pBlockAgg);
723,801,816✔
371

372
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
723,478,856✔
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) {
723,187,267✔
380
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
25,135,430✔
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);
25,151,034✔
384
    pCost->skipBlocks += 1;
25,177,682✔
385
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
25,163,610✔
386
    return code;
25,151,084✔
387
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
697,977,760✔
388
    pCost->loadBlockStatis += 1;
52,175,725✔
389
    loadSMA = true;  // mark the operation of load sma;
52,200,257✔
390
    bool success = true;
52,200,257✔
391
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
52,196,794✔
392
    if (code) {
52,177,710✔
393
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
394
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
UNCOV
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
52,159,133✔
399
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
5,474,921✔
400
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
401
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
5,474,921✔
402
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
5,474,806✔
403
      return code;
5,475,151✔
404
    } else {
405
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
46,684,212✔
406
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
46,698,613✔
407
    }
408
  }
409

410
  if (*status != FUNC_DATA_REQUIRED_DATA_LOAD) {
692,466,196✔
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)) {
692,649,457✔
418
    bool success = true;
98,109,596✔
419
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
98,106,760✔
420
    if (code) {
97,991,384✔
421
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
422
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
423
      QUERY_CHECK_CODE(code, lino, _end);
71,284✔
424
    }
425

426
    if (success) {
98,062,668✔
427
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
8,296,963✔
428
      bool   keep = false;
8,295,876✔
429
      code = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows, &keep);
8,296,503✔
430
      if (code) {
8,295,646✔
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) {
8,291,276✔
437
        qDebug("%s data block filter out by block SMA, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
3,281,154✔
438
               GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
439
        pCost->filterOutBlocks += 1;
3,281,154✔
440
        (*status) = FUNC_DATA_REQUIRED_FILTEROUT;
3,281,154✔
441
        taosMemoryFreeClear(pBlock->pBlockAgg);
3,281,154✔
442

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

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

452
  // try to filter data block according to current results
453
  code = doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
689,598,012✔
454
  if (code) {
689,510,550✔
455
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
29✔
456
    QUERY_CHECK_CODE(code, lino, _end);
×
457
  }
458

459
  if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
689,510,521✔
460
    qDebug("%s data block skipped due to dynamic prune, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
9,674,226✔
461
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
462
    pCost->skipBlocks += 1;
9,680,423✔
463
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
9,675,348✔
464

465
    STableScanInfo* p1 = pOperator->info;
9,675,348✔
466
    if (taosHashGetSize(p1->pIgnoreTables) == taosArrayGetSize(p1->base.pTableListInfo->pTableList)) {
9,675,348✔
467
      *status = FUNC_DATA_REQUIRED_ALL_FILTEROUT;
1,084,380✔
468
    } else {
469
      *status = FUNC_DATA_REQUIRED_FILTEROUT;
8,590,968✔
470
    }
471
    return TSDB_CODE_SUCCESS;
9,675,348✔
472
  }
473

474
  pCost->totalCheckedRows += pBlock->info.rows;
679,404,019✔
475
  pCost->loadBlocks += 1;
679,492,408✔
476

477
  SSDataBlock* p = NULL;
679,649,456✔
478
  code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pTableScanInfo->dataReader, &p);
679,727,577✔
479
  if (p == NULL || code != TSDB_CODE_SUCCESS || p != pBlock) {
680,135,815✔
480
    return code;
×
481
  }
482

483
  if ((pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) &&
680,135,815✔
484
      ((STableScanInfo*)pOperator->info)->ignoreTag) {
540,900,210✔
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);
679,523,805✔
489
    if (code) {
679,487,541✔
490
      return code;
×
491
    }
492
  }
493

494
  // restore the previous value
495
  pCost->totalRows -= pBlock->info.rows;
679,661,987✔
496

497
  if (pOperator->exprSupp.pFilterInfo != NULL) {
679,676,260✔
498
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo, NULL);
94,755,435✔
499
    QUERY_CHECK_CODE(code, lino, _end);
94,803,534✔
500

501
    int64_t st = taosGetTimestampUs();
92,512,652✔
502
    double  el = (taosGetTimestampUs() - st) / 1000.0;
92,506,249✔
503
    pTableScanInfo->readRecorder.filterTime += el;
92,506,249✔
504

505
    if (pBlock->info.rows == 0) {
92,489,336✔
506
      pCost->filterOutBlocks += 1;
19,275,657✔
507
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
19,270,145✔
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);
73,184,880✔
511
    }
512
  }
513

514
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
677,785,304✔
515
  if (limitReached) {  // set operator flag is done
677,556,277✔
516
    setOperatorCompleted(pOperator);
3,496,482✔
517
  }
518

519
  pCost->totalRows += pBlock->info.rows;
677,556,696✔
520

521
_end:
680,011,106✔
522
  if (code != TSDB_CODE_SUCCESS) {
680,031,721✔
523
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
2,310,005✔
524
  }
525
  return code;
679,577,140✔
526
}
527

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

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

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

542
static void freeTableCachedVal(void* param) {
231,805,998✔
543
  if (param == NULL) {
231,805,998✔
544
    return;
×
545
  }
546

547
  STableCachedVal* pVal = param;
231,805,998✔
548
  taosMemoryFree((void*)pVal->pName);
231,805,998✔
549
  taosMemoryFree(pVal->pTags);
231,792,660✔
550
  taosMemoryFree(pVal);
231,791,467✔
551
}
552

553
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
231,656,736✔
554
  int32_t          code = TSDB_CODE_SUCCESS;
231,656,736✔
555
  int32_t          lino = 0;
231,656,736✔
556
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
231,656,736✔
557
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
231,700,036✔
558

559
  pVal->pTags = NULL;
231,700,036✔
560
  pVal->pName = taosStrdup(pMetaReader->me.name);
231,703,851✔
561
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
231,675,828✔
562

563
  // only child table has tag value
564
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
231,641,207✔
565
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
231,406,754✔
566
    pVal->pTags = taosMemoryMalloc(pTag->len);
231,421,596✔
567
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
231,439,980✔
568
    memcpy(pVal->pTags, pTag, pTag->len);
231,442,929✔
569
  }
570

571
  (*ppResVal) = pVal;
231,692,827✔
572

573
_end:
231,732,267✔
574
  if (code != TSDB_CODE_SUCCESS) {
231,732,267✔
575
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
576
    freeTableCachedVal(pVal);
×
577
  }
578
  return code;
231,687,524✔
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) {
231,805,400✔
583
  (void)key;
584
  (void)keyLen;
585
  (void)ud;
586
  freeTableCachedVal(value);
231,805,400✔
587
}
231,801,327✔
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,
382,469,446✔
604
                               int32_t rows, SExecTaskInfo* pTask, STableMetaCacheInfo* pCache) {
605
  int32_t          code = TSDB_CODE_SUCCESS;
382,469,446✔
606
  int32_t          lino = 0;
382,469,446✔
607
  bool             freeReader = false;
382,469,446✔
608
  LRUHandle*       h = NULL;
382,469,446✔
609
  STableCachedVal  val = {0};
382,469,446✔
610
  SMetaReader      mr = {0};
382,469,446✔
611
  const char*      idStr = pTask->id.str;
382,476,287✔
612
  int32_t          insertRet = TAOS_LRU_STATUS_OK;
382,503,934✔
613
  STableCachedVal* pVal = NULL;
382,503,934✔
614

615
  // currently only the tbname pseudo column
616
  if (numOfExpr <= 0) {
382,515,801✔
617
    return TSDB_CODE_SUCCESS;
171,171✔
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;
382,344,630✔
624
  pBlock->info.rows = rows;
382,356,512✔
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) {
382,361,518✔
631
    pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
1,773,385✔
632
    code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
1,674,442✔
633
    if (code != TSDB_CODE_SUCCESS) {
1,672,334✔
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,672,334✔
649

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

653
    freeReader = true;
1,674,430✔
654
  } else {
655
    pCache->metaFetch += 1;
380,590,248✔
656

657
    h = taosLRUCacheLookup(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(pBlock->info.id.uid));
380,683,858✔
658
    if (h == NULL) {
380,736,747✔
659
      pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
231,815,423✔
660
      freeReader = true;
231,798,578✔
661
      code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
231,798,578✔
662
      if (code != TSDB_CODE_SUCCESS) {
231,649,385✔
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);
231,649,385✔
677

678
      code = createTableCacheVal(&mr, &pVal);
231,771,972✔
679
      QUERY_CHECK_CODE(code, lino, _end);
231,686,466✔
680

681
      val = *pVal;
231,686,466✔
682
    } else {
683
      pCache->cacheHit += 1;
148,921,324✔
684
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
148,936,222✔
685
      val = *pValTmp;
148,931,425✔
686

687
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
148,939,017✔
688
      qTrace("release LRU cache, res %d", bRes);
148,950,697✔
689
    }
690

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

695
  for (int32_t j = 0; j < numOfExpr; ++j) {
1,539,623,948✔
696
    const SExprInfo* pExpr1 = &pExpr[j];
1,157,168,976✔
697
    int32_t          dstSlotId = pExpr1->base.resSchema.slotId;
1,157,166,746✔
698

699
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
1,157,181,578✔
700
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
1,157,156,343✔
701

702
    int32_t functionId = pExpr1->pExpr->_function.functionId;
1,157,135,410✔
703

704
    // this is to handle the tbname
705
    if (fmIsScanPseudoColumnFunc(functionId)) {
1,157,161,051✔
706
      int32_t fType = pExpr1->pExpr->_function.functionType;
128,069,997✔
707
      if (fType == FUNCTION_TYPE_TBNAME) {
128,070,855✔
708
        code = setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
128,070,855✔
709
        QUERY_CHECK_CODE(code, lino, _end);
128,066,238✔
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,029,079,560✔
719
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
1,029,076,791✔
720
      const char* p = pHandle->api.metaFn.extractTagVal(val.pTags, pColInfoData->info.type, &tagVal);
1,029,082,459✔
721

722
      char* data = NULL;
1,029,113,024✔
723
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
1,029,113,024✔
724
        data = tTagValToData((const STagVal*)p, false);
1,023,645,679✔
725
      } else {
726
        data = (char*)p;
5,469,432✔
727
      }
728

729
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
1,029,080,181✔
730
      if (isNullVal) {
1,029,089,118✔
731
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
5,005,817✔
732
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
1,024,083,301✔
733
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, 1, false);
1,023,624,894✔
734
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
1,023,620,788✔
735
          char* tmp = taosMemoryCalloc(1, varDataLen(data) + 1);
374,524,063✔
736
          if (tmp != NULL) {
374,532,567✔
737
            memcpy(tmp, varDataVal(data), varDataLen(data));
374,532,567✔
738
            qDebug("get tag value:%s, cid:%d, table name:%s, uid%" PRId64, tmp, tagVal.cid, val.pName,
374,534,932✔
739
                   pBlock->info.id.uid);
740
            taosMemoryFree(tmp);
374,546,882✔
741
          }
742
          taosMemoryFree(data);
374,542,640✔
743
        }
744
        QUERY_CHECK_CODE(code, lino, _end);
1,023,600,166✔
745
      } else {  // todo opt for json tag
746
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
2,907,281✔
747
          code = colDataSetVal(pColInfoData, i, data, false);
2,443,666✔
748
          QUERY_CHECK_CODE(code, lino, _end);
2,443,666✔
749
        }
750
      }
751
    }
752
  }
753

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

757
_end:
382,430,770✔
758

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

768
  if (freeReader) {
382,417,672✔
769
    pHandle->api.metaReaderFn.clearReader(&mr);
233,461,853✔
770
  }
771
  if (code != TSDB_CODE_SUCCESS) {
382,426,384✔
772
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
773
  }
774
  return code;
382,426,838✔
775
}
776

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

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

789
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
128,070,929✔
790

791
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
128,069,958✔
792
  QUERY_CHECK_CODE(code, lino, _end);
128,070,430✔
793

794
  code = colDataSetVal(&infoData, 0, buf, false);
128,070,430✔
795
  QUERY_CHECK_CODE(code, lino, _end);
128,069,623✔
796

797
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
128,069,623✔
798
  SScalarParam param = {.columnData = pColInfoData};
128,069,810✔
799

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

807
  colDataDestroy(&infoData);
128,067,779✔
808

809
_end:
128,067,277✔
810
  if (code != TSDB_CODE_SUCCESS) {
128,068,024✔
811
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
812
  }
813
  return code;
128,068,024✔
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) {
228,006,083✔
883
  int32_t code = TSDB_CODE_SUCCESS;
228,006,083✔
884
  int32_t lino = 0;
228,006,083✔
885
  code = tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, pKeyInfo, size);
228,006,083✔
886
  QUERY_CHECK_CODE(code, lino, _end);
227,584,920✔
887

888
  pInfo->tableStartIndex = TARRAY_ELEM_IDX(pInfo->base.pTableListInfo->pTableList, *pKeyInfo);
227,584,920✔
889
  pInfo->tableEndIndex = (pInfo->tableStartIndex + (*size) - 1);
227,966,912✔
890
  pInfo->pResBlock->info.blankFill = false;
227,602,605✔
891
  taosMemoryFreeClear(pInfo->pResBlock->pBlockAgg);
227,501,785✔
892

893
  if (!pInfo->needCountEmptyTable) {
227,832,730✔
894
    pInfo->countState = TABLE_COUNT_STATE_END;
211,913,914✔
895
  } else {
896
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
16,091,890✔
897
  }
898

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

906
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
595,143,575✔
907
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
595,143,575✔
908
    return;
482,805,903✔
909
  }
910
  if (pInfo->base.pTableListInfo->groupOffset) {
112,497,640✔
911
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
213,710✔
912
  } else {
913
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
112,281,496✔
914
    if (code != TSDB_CODE_SUCCESS) {
112,286,300✔
915
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
50,419,392✔
916
    }
917
  }
918
}
919

920
static SSDataBlock* getOneRowResultBlock(SExecTaskInfo* pTaskInfo, STableScanBase* pBase, SSDataBlock* pBlock,
42,774,798✔
921
                                         const STableKeyInfo* tbInfo) {
922
  blockDataEmpty(pBlock);
42,774,798✔
923
  pBlock->info.rows = 1;
42,785,050✔
924
  pBlock->info.id.uid = tbInfo->uid;
42,785,050✔
925
  pBlock->info.id.groupId = tbInfo->groupId;
42,784,118✔
926
  pBlock->info.blankFill = true;
42,782,720✔
927

928
  // only one row: set all col data to null & hasNull
929
  int32_t col_num = blockDataGetNumOfCols(pBlock);
42,784,118✔
930
  for (int32_t i = 0; i < col_num; ++i) {
176,886,973✔
931
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
134,102,389✔
932
    colDataSetNULL(pColInfoData, 0);
933
  }
934

935
  // set tag/tbname
936
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
42,784,584✔
937

938
  return pBlock;
42,782,254✔
939
}
940

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

946
  pOperator->resultInfo.totalRows++;
41,106,223✔
947
  return pBlock;
41,110,417✔
948
}
949

950
static int32_t doTableScanImplNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
779,063,137✔
951
  int32_t         code = TSDB_CODE_SUCCESS;
779,063,137✔
952
  int32_t         lino = 0;
779,063,137✔
953
  STableScanInfo* pTableScanInfo = pOperator->info;
779,063,137✔
954
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
779,421,743✔
955
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
779,136,138✔
956
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
779,129,525✔
957
  bool            hasNext = false;
779,208,857✔
958
  int64_t         st = taosGetTimestampUs();
779,120,107✔
959

960
  QRY_PARAM_CHECK(ppRes);
779,120,107✔
961
  pBlock->info.dataLoad = false;
779,232,965✔
962

963
  while (true) {
24,952,846✔
964
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
803,935,726✔
965
    if (code != TSDB_CODE_SUCCESS) {
803,917,235✔
966
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
×
967
      QUERY_CHECK_CODE(code, lino, _end);
419✔
968
    }
969

970
    if (!hasNext) {
803,873,037✔
971
      break;
221,391,242✔
972
    }
973

974
    if (isTaskKilled(pTaskInfo)) {
582,481,795✔
975
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
10,568✔
976
      code = pTaskInfo->code;
10,568✔
977
      goto _end;
10,568✔
978
    }
979

980
    if (pOperator->status == OP_EXEC_DONE) {
582,966,975✔
981
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
86,016✔
982
      break;
86,016✔
983
    }
984

985
    // process this data block based on the probabilities
986
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
582,448,649✔
987
    if (!processThisBlock) {
582,801,199✔
988
      continue;
×
989
    }
990

991
    if (pBlock->info.id.uid) {
582,801,199✔
992
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
582,901,068✔
993
    }
994

995
    uint32_t status = 0;
583,083,978✔
996
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
583,070,535✔
997
    QUERY_CHECK_CODE(code, lino, _end);
582,302,084✔
998

999
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
580,076,182✔
1000
      break;
1,084,380✔
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) {
578,991,802✔
1005
      continue;
24,952,435✔
1006
    }
1007

1008
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
554,053,609✔
1009
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
554,456,846✔
1010

1011
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
554,555,368✔
1012
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
554,404,983✔
1013

1014
    (*ppRes) = pBlock;
554,357,572✔
1015
    return code;
554,101,961✔
1016
  }
1017

1018
_end:
224,833,046✔
1019
  if (code != TSDB_CODE_SUCCESS) {
224,863,794✔
1020
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
2,280,651✔
1021
    pTaskInfo->code = code;
2,280,651✔
1022
    T_LONG_JMP(pTaskInfo->env, code);
2,236,470✔
1023
  }
1024
  return code;
222,583,143✔
1025
}
1026

1027
static int32_t doGroupedTableScan(SOperatorInfo* pOperator, SSDataBlock** pBlock) {
825,618,266✔
1028
  int32_t         code = TSDB_CODE_SUCCESS;
825,618,266✔
1029
  int32_t         lino = 0;
825,618,266✔
1030
  STableScanInfo* pTableScanInfo = pOperator->info;
825,618,266✔
1031
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
825,919,123✔
1032
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
825,772,566✔
1033

1034
  QRY_PARAM_CHECK(pBlock);
825,661,731✔
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) {
825,861,364✔
1038
    return code;
5,489,750✔
1039
  }
1040

1041
  // do the ascending order traverse in the first place.
1042
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
1,033,446,700✔
1043
    SSDataBlock* p = NULL;
698,703,197✔
1044
    code = doTableScanImplNext(pOperator, &p);
698,589,796✔
1045
    QUERY_CHECK_CODE(code, lino, _end);
696,194,922✔
1046

1047
    if (p != NULL) {
696,194,922✔
1048
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
482,977,444✔
1049
      *pBlock = p;
482,746,037✔
1050
      return code;
483,058,922✔
1051
    }
1052

1053
    pTableScanInfo->scanTimes += 1;
213,217,478✔
1054
    taosHashClear(pTableScanInfo->pIgnoreTables);
213,220,816✔
1055

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

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

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

1077
    while (pTableScanInfo->scanTimes < total) {
89,964,607✔
1078
      SSDataBlock* p = NULL;
80,652,688✔
1079
      code = doTableScanImplNext(pOperator, &p);
80,650,807✔
1080
      QUERY_CHECK_CODE(code, lino, _end);
80,622,406✔
1081

1082
      if (p != NULL) {
80,622,406✔
1083
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
71,302,186✔
1084
        *pBlock = p;
71,302,395✔
1085
        return code;
71,298,079✔
1086
      }
1087

1088
      pTableScanInfo->scanTimes += 1;
9,320,220✔
1089
      taosHashClear(pTableScanInfo->pIgnoreTables);
9,325,655✔
1090

1091
      if (pTableScanInfo->scanTimes < total) {
9,323,247✔
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);
×
UNCOV
1097
        QUERY_CHECK_CODE(code, lino, _end);
×
1098
      }
1099
    }
1100
  }
1101

1102
  if (pTableScanInfo->countState < TABLE_COUNT_STATE_END) {
263,482,181✔
1103
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
56,679,732✔
1104
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
56,681,454✔
1105
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
113,839✔
1106
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
3,816✔
1107
        STableKeyInfo* pStart =
1108
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
3,816✔
1109

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

1114
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
3,816✔
1115
        return code;
3,816✔
1116
      }
1117
    } else {  // group by tag + no sort
1118
      int32_t numOfTables = 0;
56,574,821✔
1119
      code = tableListGetSize(pTableListInfo, &numOfTables);
56,574,710✔
1120
      QUERY_CHECK_CODE(code, lino, _end);
56,577,638✔
1121

1122
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
56,577,638✔
1123
        // get empty group, mark processed & rm from hash
1124
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
55,890,362✔
1125
        if (pIte != NULL) {
55,886,313✔
1126
          size_t        keySize = 0;
41,102,017✔
1127
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
41,103,415✔
1128
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
41,102,483✔
1129
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
41,102,483✔
1130
          markGroupProcessed(pTableScanInfo, *pGroupId);
41,100,543✔
1131
          *pBlock = getBlockForEmptyTable(pOperator, &info);
41,094,485✔
1132

1133
          return code;
41,104,271✔
1134
        }
1135
      }
1136
    }
1137
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
15,582,447✔
1138
  }
1139

1140
_end:
206,880,542✔
1141
  if (code != TSDB_CODE_SUCCESS) {
222,428,209✔
1142
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1143
    pTaskInfo->code = code;
×
1144
  }
1145

1146
  return code;
222,428,209✔
1147
}
1148

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

1165
  if (pParam->tableSeq) {
236,797✔
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;
232,597✔
1174
    pListInfo->numOfOuputGroups = 1;
232,597✔
1175
  }
1176
  if (pParam->window.skey == INT64_MAX && pParam->window.ekey == INT64_MIN) {
236,797✔
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};
236,797✔
1184
  int32_t       tableIdx = 0;
236,797✔
1185
  for (int32_t i = 0; i < num; ++i) {
7,379,659✔
1186
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
7,142,862✔
1187
    if (!pUid) {
7,141,866✔
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,141,866✔
1193
      if (TSDB_CODE_DUP_KEY == terrno) {
6,573,520✔
1194
        continue;
6,573,520✔
1195
      }
1196
      return terrno;
×
1197
    }
1198

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

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

1209
  return code;
236,797✔
1210
}
1211

1212
static int32_t doInitReader(STableScanInfo* pInfo, SExecTaskInfo* pTaskInfo, SStorageAPI* pAPI, int32_t* pNum,
227,606,260✔
1213
                            STableKeyInfo** pList) {
1214
  const char* idStr = GET_TASKID(pTaskInfo);
227,606,260✔
1215
  int32_t     code = initNextGroupScan(pInfo, pList, pNum);
227,806,382✔
1216
  if (code) {
227,080,462✔
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) {
227,080,462✔
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,
454,647,834✔
1227
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
227,543,894✔
1228
  if (code) {
227,665,948✔
1229
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
2,065✔
1230
  }
1231

1232
  return code;
227,557,477✔
1233
}
1234

1235
int compareColIdPair(const void* elem1, const void* elem2) {
3,105,629✔
1236
  SColIdPair* node1 = (SColIdPair*)elem1;
3,105,629✔
1237
  SColIdPair* node2 = (SColIdPair*)elem2;
3,105,629✔
1238

1239
  if (node1->orgColId == node2->orgColId) {
3,105,629✔
1240
    return 0;
105,486✔
1241
  }
1242

1243
  return node1->orgColId < node2->orgColId ? -1 : 1;
3,000,143✔
1244
}
1245

1246
int compareColIdSlotIdPair(const void* elem1, const void* elem2) {
3,105,629✔
1247
  SColIdSlotIdPair* node1 = (SColIdSlotIdPair*)elem1;
3,105,629✔
1248
  SColIdSlotIdPair* node2 = (SColIdSlotIdPair*)elem2;
3,105,629✔
1249

1250
  if (node1->orgColId == node2->orgColId) {
3,105,629✔
1251
    return 0;
105,486✔
1252
  }
1253

1254
  return node1->orgColId < node2->orgColId ? -1 : 1;
3,000,143✔
1255
}
1256

1257
static bool isNewScanParam(STableScanOperatorParam* pParam) {
510,887✔
1258
  return pParam->isNewParam;
510,887✔
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) {
332,093✔
1515
  int32_t                  code = 0;
332,093✔
1516
  int32_t                  lino = 0;
332,093✔
1517
  STableScanInfo*          pInfo = pOperator->info;
332,093✔
1518
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
332,093✔
1519
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
332,093✔
1520
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
332,093✔
1521
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
332,093✔
1522
  SMetaReader              orgTable = {0};
332,093✔
1523
  SMetaReader              superTable = {0};
332,093✔
1524
  SSchemaWrapper*          schema = NULL;
332,093✔
1525
  SArray*                  pColArray = NULL;
332,093✔
1526
  SArray*                  pBlockColArray = NULL;
332,093✔
1527
  int32_t                  num = 0;
332,093✔
1528
  STableKeyInfo*           pList = NULL;
332,093✔
1529
  SOrgTbInfo*              pOrgTbInfo;
1530

1531
  cleanupQueryTableDataCond(&pInfo->base.cond);
332,093✔
1532

1533
  pOrgTbInfo = pParam->pOrgTbInfo;
332,093✔
1534

1535
  QUERY_CHECK_NULL(pOrgTbInfo, code, lino, _return, terrno);
332,093✔
1536

1537
  pAPI->metaReaderFn.initReader(&orgTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
332,093✔
1538
  code = pAPI->metaReaderFn.getTableEntryByName(&orgTable, strstr(pOrgTbInfo->tbName, ".") + 1);
332,093✔
1539
  pAPI->metaReaderFn.readerReleaseLock(&orgTable);
332,093✔
1540
  qDebug("dynamic vtable scan for origin table:%s, %s", pOrgTbInfo->tbName, GET_TASKID(pTaskInfo));
332,093✔
1541
  QUERY_CHECK_CODE(code, lino, _return);
332,093✔
1542
  switch (orgTable.me.type) {
332,093✔
1543
    case TSDB_CHILD_TABLE:
197,293✔
1544
      pAPI->metaReaderFn.initReader(&superTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
197,293✔
1545
      code = pAPI->metaReaderFn.getTableEntryByUid(&superTable, orgTable.me.ctbEntry.suid);
197,293✔
1546
      pAPI->metaReaderFn.readerReleaseLock(&superTable);
197,293✔
1547
      QUERY_CHECK_CODE(code, lino, _return);
197,293✔
1548
      schema = &superTable.me.stbEntry.schemaRow;
197,293✔
1549
      break;
197,293✔
1550
    case TSDB_NORMAL_TABLE:
134,800✔
1551
      schema = &orgTable.me.ntbEntry.schemaRow;
134,800✔
1552
      break;
134,800✔
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;
332,093✔
1560
  taosHashClear(pListInfo->map);
332,093✔
1561
  taosArrayClear(pListInfo->pTableList);
332,093✔
1562

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

1571
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
332,093✔
1572
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
332,093✔
1573
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(SColIdSlotIdPair));
332,093✔
1574
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
332,093✔
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,
332,093✔
1578
                       .type.type = TSDB_DATA_TYPE_TIMESTAMP,
1579
                       .type.bytes = 8};
1580
  QUERY_CHECK_NULL(taosArrayPush(pColArray, &tsPair), code, lino, _return, terrno);
332,093✔
1581

1582
  for (int32_t i = 0; i < taosArrayGetSize(pOrgTbInfo->colMap); ++i) {
1,276,483✔
1583
    SColIdNameKV* kv = taosArrayGet(pOrgTbInfo->colMap, i);
944,390✔
1584
    for (int32_t j = 0; j < schema->nCols; j++) {
6,254,781✔
1585
      if (strcmp(kv->colName, schema->pSchema[j].name) == 0) {
6,254,781✔
1586
        SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(schema->pSchema[j].colId),
944,390✔
1587
                            .type.type = schema->pSchema[j].type,
944,390✔
1588
                            .type.bytes = schema->pSchema[j].bytes};
944,390✔
1589
        QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
944,390✔
1590
        break;
944,390✔
1591
      }
1592
    }
1593
  }
1594

1595
  for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
1,607,961✔
1596
    SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, i);
1,276,483✔
1597
    for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
6,503,352✔
1598
      SColMatchItem* pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
6,503,352✔
1599
      if (pItem->colId == pPair->vtbColId) {
6,503,352✔
1600
        if (pItem->dataType.type != pPair->type.type ||
1,276,483✔
1601
            pItem->dataType.bytes != pPair->type.bytes) {
1,275,868✔
1602
          qError("column type not match for vtable colId:%d, org colId:%d, org table name:%s", pPair->vtbColId, pPair->orgColId, orgTable.me.name);
615✔
1603
          code = TSDB_CODE_VTABLE_COLUMN_TYPE_MISMATCH;
615✔
1604
          goto _return;
615✔
1605
        }
1606
        SColIdSlotIdPair colIdSlotIdPair = {.orgColId = pPair->orgColId, .vtbSlotId = pItem->dstSlotId};
1,275,868✔
1607
        QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &colIdSlotIdPair), code, lino, _return, terrno);
1,275,868✔
1608
        break;
1,275,868✔
1609
      }
1610
    }
1611
  }
1612

1613
  taosArraySort(pColArray, compareColIdPair);
331,478✔
1614
  taosArraySort(pBlockColArray, compareColIdSlotIdPair);
331,478✔
1615
  if (pInfo->pBlockColMap) {
331,478✔
1616
    taosArrayDestroy(pInfo->pBlockColMap);
149,504✔
1617
    pInfo->pBlockColMap = NULL;
149,504✔
1618
  }
1619
  pInfo->pBlockColMap = taosArrayDup(pBlockColArray, NULL);
331,478✔
1620
  QUERY_CHECK_NULL(pInfo->pBlockColMap, code, lino, _return, terrno)
331,478✔
1621

1622

1623
  taosArrayRemoveDuplicate(pColArray, compareColIdPair, NULL);
331,478✔
1624
  taosArrayRemoveDuplicate(pBlockColArray, compareColIdSlotIdPair, NULL);
331,478✔
1625

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

1629
  if (pInfo->pResBlock) {
331,478✔
1630
    blockDataDestroy(pInfo->pResBlock);
179,409✔
1631
    pInfo->pResBlock = NULL;
179,409✔
1632
  }
1633

1634
  if (isNewScanParam(pParam)) {
331,478✔
1635
    if (pParam->window.skey == INT64_MAX && pParam->window.ekey == INT64_MIN) {
157,032✔
1636
      pInfo->base.cond.twindows.skey = pInfo->base.orgCond.twindows.skey;
157,032✔
1637
      pInfo->base.cond.twindows.ekey = pInfo->base.orgCond.twindows.ekey;
157,032✔
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;
174,446✔
1644
  }
1645
  pInfo->base.cond.suid = orgTable.me.type == TSDB_CHILD_TABLE ? superTable.me.uid : 0;
331,478✔
1646
  pInfo->currentGroupId = 0;
331,478✔
1647
  pInfo->ignoreTag = true;
331,478✔
1648

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

1652
  void **reader = taosHashGet(pInfo->readerCache, &pUid, sizeof(uint64_t));
331,478✔
1653
  if (reader) {
331,478✔
1654
    if (isNewScanParam(pParam)) {
179,409✔
1655
      pAPI->tsdReader.tsdReaderClose(*reader);
4,963✔
1656
      pInfo->base.dataReader = NULL;
4,963✔
1657

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

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

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

1674
      pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, pInfo->pResBlock);
174,446✔
1675
      pInfo->newReader = false;
174,446✔
1676
      pInfo->scanTimes = 0;
174,446✔
1677
    }
1678
  } else {
1679
    pInfo->base.dataReader = NULL;
152,069✔
1680

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

1686
    code = taosHashPut(pInfo->readerCache, &pUid, sizeof(uint64_t), &pInfo->base.dataReader, POINTER_BYTES);
152,069✔
1687
    QUERY_CHECK_CODE(code, lino, _return);
152,069✔
1688
    pInfo->newReader = true;
152,069✔
1689
  }
1690
  pOperator->status = OP_OPENED;
331,478✔
1691

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

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

1708
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
222,556,179✔
1709
  int32_t         code = TSDB_CODE_SUCCESS;
222,556,179✔
1710
  int32_t         lino = 0;
222,556,179✔
1711
  STableScanInfo* pInfo = pOperator->info;
222,556,179✔
1712
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
222,576,271✔
1713
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
222,566,002✔
1714
  int32_t         numOfTables = 0;
222,616,357✔
1715

1716
  QRY_PARAM_CHECK(pResult);
222,542,562✔
1717

1718
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
222,536,757✔
1719
  QUERY_CHECK_CODE(code, lino, _end);
222,603,564✔
1720

1721
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
222,603,564✔
1722
    setOperatorCompleted(pOperator);
222,150,442✔
1723
    if (pOperator->dynamicTask) {
222,150,197✔
1724
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
235,762✔
1725
      taosHashClear(pInfo->base.pTableListInfo->map);
235,762✔
1726
    }
1727
    return code;
222,152,570✔
1728
  }
1729

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

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

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

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

1746
  code = doGroupedTableScan(pOperator, pResult);
393,966✔
1747
  QUERY_CHECK_CODE(code, lino, _end);
393,966✔
1748

1749
  if (*pResult != NULL) {
393,966✔
1750
    if (pOperator->dynamicTask) {
299,240✔
1751
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
104,526✔
1752
    }
1753
    return code;
299,240✔
1754
  }
1755

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

1764
static int32_t groupSeqTableScan(SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
814,094,188✔
1765
  int32_t         code = TSDB_CODE_SUCCESS;
814,094,188✔
1766
  int32_t         lino = 0;
814,094,188✔
1767
  STableScanInfo* pInfo = pOperator->info;
814,094,188✔
1768
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
814,415,411✔
1769
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
814,207,816✔
1770
  int32_t         num = 0;
814,017,055✔
1771
  STableKeyInfo*  pList = NULL;
814,238,079✔
1772
  SSDataBlock*    pResult = NULL;
814,054,197✔
1773
  const char*     idStr = GET_TASKID(pTaskInfo);
814,168,309✔
1774

1775
  QRY_PARAM_CHECK(pResBlock);
814,220,013✔
1776

1777
  if (pInfo->currentGroupId == -1) {
814,180,087✔
1778
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
227,248,990✔
1779
      setOperatorCompleted(pOperator);
×
1780
      return code;
×
1781
    }
1782

1783
    taosRLockLatch(&pTaskInfo->lock);
227,502,867✔
1784
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
227,654,895✔
1785
    taosRUnLockLatch(&pTaskInfo->lock);
227,517,129✔
1786
    QUERY_CHECK_CODE(code, lino, _end);
227,717,550✔
1787

1788
    if (pInfo->filesetDelimited) {
227,715,485✔
1789
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
234,015✔
1790
    }
1791

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

1797
  code = doGroupedTableScan(pOperator, &pResult);
814,464,536✔
1798
  QUERY_CHECK_CODE(code, lino, _end);
811,841,687✔
1799

1800
  if (pResult != NULL) {
811,841,687✔
1801
    if (pOperator->dynamicTask) {
589,815,450✔
1802
      pResult->info.id.groupId = pResult->info.id.uid;
532,812✔
1803
    }
1804

1805
    *pResBlock = pResult;
589,796,542✔
1806
    return code;
589,391,164✔
1807
  }
1808

1809
  while (true) {
1810
    code = startNextGroupScan(pOperator, &pResult);
222,066,357✔
1811
    QUERY_CHECK_CODE(code, lino, _end);
222,405,527✔
1812

1813
    if (pResult || pOperator->status == OP_EXEC_DONE) {
222,405,527✔
1814
      *pResBlock = pResult;
222,324,895✔
1815
      return code;
222,355,473✔
1816
    }
1817
  }
1818

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

1825
  return code;
2,065✔
1826
}
1827

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

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

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

1845
  if (pOperator->status == OP_EXEC_DONE) {
236,797✔
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:
234,697✔
1861
  if (code) {
234,697✔
1862
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1863
  }
1864
  return code;
234,697✔
1865
}
1866

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

1875
  QRY_PARAM_CHECK(ppRes);
332,093✔
1876
  qTrace("%s call", __FUNCTION__);
332,093✔
1877

1878
  code = createVTableScanInfoFromParam(pOperator);
332,093✔
1879
  freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
332,093✔
1880
  pOperator->pOperatorGetParam = NULL;
332,093✔
1881

1882
  QUERY_CHECK_CODE(code, lino, _end);
332,093✔
1883

1884
  SSDataBlock* result = NULL;
331,478✔
1885
  if (isEmptyQueryTimeWindow(&pInfo->base.cond.twindows) && pInfo->base.cond.type == TIMEWINDOW_RANGE_CONTAINED) {
331,478✔
1886
    (*ppRes) = result;
17,844✔
1887
    return code;
17,844✔
1888
  }
1889

1890
  if (pInfo->newReader) {
313,634✔
1891
    code = startNextGroupScan(pOperator, &result);
157,032✔
1892
    QUERY_CHECK_CODE(code, lino, _end);
157,032✔
1893
  } else {
1894
    code = doGroupedTableScan(pOperator, &result);
156,602✔
1895
    QUERY_CHECK_CODE(code, lino, _end);
156,602✔
1896
  }
1897

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

1906
    pInfo->pResBlock = res;
174,446✔
1907
    blockDataDestroy(result);
174,446✔
1908
    (*ppRes) = res;
174,446✔
1909
  } else {
1910
    STableKeyInfo *keyInfo = taosArrayGet(pInfo->base.pTableListInfo->pTableList, 0);
139,188✔
1911
    QUERY_CHECK_NULL(keyInfo, code, lino, _end, terrno)
139,188✔
1912

1913
    blockDataDestroy(pInfo->pResBlock);
139,188✔
1914
    pInfo->pResBlock = NULL;
139,188✔
1915

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

1927
    code = taosHashRemove(pInfo->readerCache, &keyInfo->uid, sizeof(uint64_t));
139,188✔
1928
    QUERY_CHECK_CODE(code, lino, _end);
139,188✔
1929
  }
1930
_end:
314,249✔
1931
  if (code) {
314,249✔
1932
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
615✔
1933
  }
1934
  return code;
314,249✔
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) {
568,890✔
2072
  int32_t                  code = TSDB_CODE_SUCCESS;
568,890✔
2073
  int32_t                  lino = 0;
568,890✔
2074
  STableScanInfo*          pInfo = pOperator->info;
568,890✔
2075
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
568,890✔
2076
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
568,890✔
2077

2078
  QRY_PARAM_CHECK(ppRes);
568,890✔
2079
  qTrace("%s call", __FUNCTION__);
568,890✔
2080

2081
  pOperator->dynamicTask = true;
568,890✔
2082

2083
  switch (type) {
568,890✔
2084
    case DYN_TYPE_STB_JOIN: {
236,797✔
2085
      code = doStbJoinDynamicTableScanNext(pOperator, ppRes);
236,797✔
2086
      QUERY_CHECK_CODE(code, lino, _end);
236,797✔
2087
      break;
236,797✔
2088
    }
2089
    case DYN_TYPE_VSTB_SINGLE_SCAN: {
332,093✔
2090
      code = doVstbSingleDynamicTableScanNext(pOperator, ppRes);
332,093✔
2091
      QUERY_CHECK_CODE(code, lino, _end);
332,093✔
2092
      break;
331,478✔
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:
568,890✔
2105
  return code;
568,890✔
2106
}
2107

2108
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
820,178,413✔
2109
  int32_t         code = TSDB_CODE_SUCCESS;
820,178,413✔
2110
  int32_t         lino = 0;
820,178,413✔
2111
  STableScanInfo* pInfo = pOperator->info;
820,178,413✔
2112
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
820,522,972✔
2113
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
820,443,179✔
2114
  QRY_PARAM_CHECK(ppRes);
820,255,168✔
2115
  qTrace("%s call", __FUNCTION__);
820,164,735✔
2116

2117
  if (pOperator->pOperatorGetParam || pInfo->pBatchColMap) {
820,164,735✔
2118
    ETableScanDynType type = pOperator->pOperatorGetParam ?
544,641✔
2119
                                                          ((STableScanOperatorParam*)pOperator->pOperatorGetParam->value)->type :
568,890✔
2120
                                                          DYN_TYPE_VSTB_BATCH_SCAN;
2121

2122
    code = doDynamicTableScanNext(pOperator, type, ppRes);
568,890✔
2123
    QUERY_CHECK_CODE(code, lino, _end);
568,890✔
2124

2125
    if (type != DYN_TYPE_STB_JOIN) {
568,275✔
2126
      return code;
331,478✔
2127
    }
2128
  }
2129

2130
  // scan table one by one sequentially
2131
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
819,841,130✔
2132
    int32_t       numOfTables = 0;
5,689,299✔
2133
    STableKeyInfo tInfo = {0};
5,689,299✔
2134
    pInfo->countState = TABLE_COUNT_STATE_END;
5,689,299✔
2135

2136
    while (1) {
5,303,316✔
2137
      SSDataBlock* result = NULL;
10,992,615✔
2138
      code = doGroupedTableScan(pOperator, &result);
10,990,544✔
2139
      QUERY_CHECK_CODE(code, lino, _end);
10,986,334✔
2140

2141
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
10,986,334✔
2142
        (*ppRes) = result;
5,641,059✔
2143
        return code;
5,646,743✔
2144
      }
2145

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

2149
      taosRLockLatch(&pTaskInfo->lock);
5,341,087✔
2150
      numOfTables = 0;
5,346,172✔
2151
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
5,346,172✔
2152
      if (code != TSDB_CODE_SUCCESS) {
5,343,481✔
2153
        taosRUnLockLatch(&pTaskInfo->lock);
×
2154
        TSDB_CHECK_CODE(code, lino, _end);
×
2155
      }
2156

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

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

2171
      tInfo = *tmp;
5,300,031✔
2172
      taosRUnLockLatch(&pTaskInfo->lock);
5,294,051✔
2173

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

2179
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
5,313,180✔
2180
      QUERY_CHECK_CODE(code, lino, _end);
5,301,244✔
2181
      pInfo->scanTimes = 0;
5,301,244✔
2182
    }
2183
  } else {  // scan table group by group sequentially
2184
    code = groupSeqTableScan(pOperator, ppRes);
814,110,104✔
2185
    QUERY_CHECK_CODE(code, lino, _end);
811,840,662✔
2186
  }
2187

2188
_end:
811,838,597✔
2189
  if (code != TSDB_CODE_SUCCESS) {
811,878,729✔
2190
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
2,680✔
2191
    pTaskInfo->code = code;
2,680✔
2192
    T_LONG_JMP(pTaskInfo->env, code);
2,680✔
2193
  }
2194

2195
  return code;
811,876,049✔
2196
}
2197

2198
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
2,395,633✔
2199
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
2,395,633✔
2200
  if (!pRecorder) {
2,394,108✔
2201
    return terrno;
×
2202
  }
2203
  STableScanInfo* pTableScanInfo = pOptr->info;
2,394,108✔
2204
  *pRecorder = pTableScanInfo->base.readRecorder;
2,394,107✔
2205
  *pOptrExplain = pRecorder;
2,396,141✔
2206
  *len = sizeof(SFileBlockLoadRecorder);
2,393,064✔
2207
  return 0;
2,396,141✔
2208
}
2209

2210
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
266,393,458✔
2211
  cleanupQueryTableDataCond(&pBase->cond);
266,393,458✔
2212
  cleanupQueryTableDataCond(&pBase->orgCond);
266,300,601✔
2213

2214
  if (pAPI->tsdReaderClose) {
266,354,084✔
2215
    pAPI->tsdReaderClose(pBase->dataReader);
266,408,459✔
2216
  }
2217
  pBase->dataReader = NULL;
266,233,014✔
2218

2219
  if (pBase->matchInfo.pList != NULL) {
266,272,285✔
2220
    taosArrayDestroy(pBase->matchInfo.pList);
266,314,380✔
2221
  }
2222

2223
  tableListDestroy(pBase->pTableListInfo);
266,149,534✔
2224
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
266,324,488✔
2225
  cleanupExprSupp(&pBase->pseudoSup);
266,550,457✔
2226
}
266,541,782✔
2227

2228
static void cleanReaderForVTable(STableScanInfo* pInfo){
52,481✔
2229
  if (pInfo != NULL && pInfo->base.readerAPI.tsdReaderClose) {
52,481✔
2230
    void *pIter = taosHashIterate(pInfo->readerCache, NULL);
52,481✔
2231
    while (pIter != NULL) {
65,362✔
2232
      void **reader = pIter;
12,881✔
2233
      if (*reader) {
12,881✔
2234
        if (*reader == pInfo->base.dataReader) {
12,881✔
2235
          pInfo->base.dataReader = NULL;
12,881✔
2236
        }
2237
        pInfo->base.readerAPI.tsdReaderClose(*reader);
12,881✔
2238
      }
2239
      pIter = taosHashIterate(pInfo->readerCache, pIter);
12,881✔
2240
    }
2241
  }
2242
}
52,481✔
2243

2244
static void destroyTableScanOperatorInfo(void* param) {
239,913,717✔
2245
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
239,913,717✔
2246
  blockDataDestroy(pTableScanInfo->pResBlock);
239,913,717✔
2247
  blockDataDestroy(pTableScanInfo->pOrgBlock);
239,890,453✔
2248
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
239,907,008✔
2249
  if (pTableScanInfo->virtualStableScan && pTableScanInfo->readerCache) {
239,751,630✔
2250
    cleanReaderForVTable(pTableScanInfo);
52,481✔
2251
    taosHashCleanup(pTableScanInfo->readerCache);
52,481✔
2252
  }
2253
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
239,757,092✔
2254
  taosArrayDestroy(pTableScanInfo->pBlockColMap);
240,029,817✔
2255
  taosMemoryFreeClear(param);
240,030,726✔
2256
}
240,022,502✔
2257

2258
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
239,680,957✔
2259
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
239,680,957✔
2260
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
11,656,811✔
2261
    for (int32_t i = 0; i < numOfCols; ++i) {
27,147,581✔
2262
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
15,517,496✔
2263
      if (pCol) {
15,520,363✔
2264
        pCol->info.noData = true;
15,516,889✔
2265
      }
2266
    }
2267
  }
2268
}
239,654,231✔
2269

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

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

2284
  if (!pInfo->virtualStableScan) {
216,977✔
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) {
83,897✔
2291
      pInfo->base.readerAPI.tsdReaderClose(pInfo->base.dataReader);
84,103✔
2292
    }
2293
    pInfo->base.dataReader = NULL;
83,691✔
2294
  }
2295
  tableListDestroy(pInfo->base.pTableListInfo);
216,977✔
2296
  pInfo->base.pTableListInfo = tableListCreate();
216,771✔
2297
  if (!pInfo->base.pTableListInfo) {
216,977✔
2298
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2299
    return terrno;
×
2300
  }
2301
  SExecTaskInfo*         pTaskInfo = pOper->pTaskInfo;
216,977✔
2302

2303
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
216,977✔
2304
  if (!pTableScanNode->scan.node.dynamicOp) {
216,977✔
2305
    code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
165,545✔
2306
                                    &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
2307
                                    pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
165,545✔
2308
    if (code) {
83,147✔
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);
216,977✔
2315
  cleanupQueryTableDataCond(&pInfo->base.cond);
216,977✔
2316
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle, true);
216,977✔
2317
  if (code) {
216,977✔
2318
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
2319
    return code;
×
2320
  }
2321
  if (pTableScanNode->scan.node.dynamicOp && pTableScanNode->scan.virtualStableScan) {
216,977✔
2322
    cleanupQueryTableDataCond(&pInfo->base.orgCond);
132,874✔
2323
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
132,874✔
2324
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
132,874✔
2325
  } 
2326

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

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

2341
  int32_t         code = TSDB_CODE_SUCCESS;
239,981,531✔
2342
  int32_t         lino = 0;
239,981,531✔
2343
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
239,981,531✔
2344
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
239,690,734✔
2345
  if (pInfo == NULL || pOperator == NULL) {
239,592,440✔
UNCOV
2346
    code = terrno;
×
2347
    goto _error;
×
2348
  }
2349

2350
  pOperator->pPhyNode = pTableScanNode;
239,596,506✔
2351
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
239,619,064✔
2352
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
239,643,616✔
2353

2354
  int32_t numOfCols = 0;
239,651,614✔
2355
  code =
2356
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
239,728,750✔
2357
  QUERY_CHECK_CODE(code, lino, _error);
239,731,688✔
2358

2359
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
239,731,688✔
2360
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, true);
239,553,201✔
2361
  QUERY_CHECK_CODE(code, lino, _error);
239,056,347✔
2362

2363
  if (pScanNode->pScanPseudoCols != NULL) {
239,056,347✔
2364
    SExprSupp* pSup = &pInfo->base.pseudoSup;
41,939,631✔
2365
    pSup->pExprInfo = NULL;
41,938,609✔
2366
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
41,939,553✔
2367
    QUERY_CHECK_CODE(code, lino, _error);
41,929,425✔
2368

2369
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
41,929,425✔
2370
                                      &pTaskInfo->storageAPI.functionStore);
2371
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
41,911,395✔
2372
  }
2373

2374
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
239,096,210✔
2375
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
239,255,292✔
2376

2377
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
239,152,507✔
2378
  pInfo->base.readHandle = *readHandle;
239,222,926✔
2379
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
239,301,902✔
2380

2381
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
239,809,690✔
2382
  pInfo->sample.seed = taosGetTimestampSec();
239,440,592✔
2383

2384
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
239,504,009✔
2385
  initResultSizeInfo(&pOperator->resultInfo, 4096);
239,414,705✔
2386
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
239,157,984✔
2387
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
239,922,156✔
2388
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
239,325,401✔
2389

2390
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
239,346,819✔
2391
  QUERY_CHECK_CODE(code, lino, _error);
239,229,614✔
2392

2393
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
239,229,614✔
2394
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
239,238,919✔
2395
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
52,481✔
2396
    pInfo->pResBlock = NULL;
52,481✔
2397
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
52,481✔
2398
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
52,481✔
2399
  }
2400

2401
  if (pInfo->virtualStableScan) {
239,101,691✔
2402
    pInfo->readerCache = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
52,481✔
2403
    QUERY_CHECK_NULL(pInfo->readerCache, code, lino, _error, terrno);
52,481✔
2404
  }
2405

2406
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
239,479,796✔
2407
                            pTaskInfo->pStreamRuntimeInfo);
239,480,987✔
2408
  QUERY_CHECK_CODE(code, lino, _error);
239,412,027✔
2409

2410
  pInfo->currentGroupId = -1;
239,390,115✔
2411

2412
  pInfo->tableEndIndex = -1;
239,461,168✔
2413
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
239,355,777✔
2414
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
239,403,557✔
2415

2416
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
239,285,850✔
2417
                  pTaskInfo);
2418
  pOperator->exprSupp.numOfExprs = numOfCols;
239,173,653✔
2419

2420
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
239,233,203✔
2421
  pInfo->ignoreTag = false;
239,304,810✔
2422

2423
  pInfo->lastBatchIdx = -1;
239,366,122✔
2424
  pInfo->currentBatchIdx = 0;
239,146,913✔
2425
  pInfo->base.pTableListInfo = pTableListInfo;
239,170,429✔
2426
  if (readHandle->streamRtInfo == NULL) {
238,837,841✔
2427
    pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
239,114,267✔
2428
    if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
239,752,941✔
2429
      code = terrno;
×
2430
      QUERY_CHECK_CODE(code, lino, _error);
×
2431
    }
2432
    taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
239,735,141✔
2433
  }
2434

2435
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
239,824,292✔
2436

2437
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
239,943,508✔
2438
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
2439

2440
  setOperatorResetStateFn(pOperator, resetTableScanOperatorState); 
239,717,659✔
2441
  // for non-blocking operator, the open cost is always 0
2442
  pOperator->cost.openCost = 0;
239,675,946✔
2443
  *pOptrInfo = pOperator;
239,604,429✔
2444
  return TSDB_CODE_SUCCESS;
239,477,366✔
2445

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

2452
  if (pOperator != NULL) {
21,912✔
2453
    pOperator->info = NULL;
21,912✔
2454
    destroyOperator(pOperator);
21,912✔
2455
  }
2456
  pTaskInfo->code = code;
21,912✔
2457
  return code;
21,912✔
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) {
12,745✔
2495
  int32_t code = TSDB_CODE_SUCCESS;
12,745✔
2496
  int32_t lino = 0;
12,745✔
2497
  if (pBlock->info.window.skey != offset->ts || offset->primaryKey.type == 0) {
12,745✔
2498
    return code;
1,557✔
2499
  }
2500
  bool* p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
11,188✔
2501
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
11,188✔
2502
  bool hasUnqualified = false;
11,188✔
2503

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

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

2512
  __compar_fn_t func = getComparFunc(pColPk->info.type, 0);
11,188✔
2513
  QUERY_CHECK_NULL(func, code, lino, _end, terrno);
11,188✔
2514
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
40,141,024✔
2515
    int64_t* ts = (int64_t*)colDataGetData(pColTs, i);
40,129,836✔
2516
    void*    data = colDataGetData(pColPk, i);
40,129,836✔
2517
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
40,131,988✔
2518
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
2,152✔
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,152✔
2522
      QUERY_CHECK_NULL(tmq, code, lino, _end, terrno);
2,152✔
2523
      memcpy(varDataVal(tmq), offset->primaryKey.pData, offset->primaryKey.nData);
2,152✔
2524
      varDataLen(tmq) = offset->primaryKey.nData;
2,152✔
2525
      p[i] = (*ts > offset->ts) || (func(data, tmq) > 0);
2,152✔
2526
      taosMemoryFree(tmq);
2,152✔
2527
    } else {
2528
      p[i] = (*ts > offset->ts) || (func(data, VALUE_GET_DATUM(&offset->primaryKey, pColPk->info.type)) > 0);
40,127,684✔
2529
    }
2530

2531
    if (!p[i]) {
40,129,836✔
2532
      hasUnqualified = true;
14,324,250✔
2533
    }
2534
  }
2535

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

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

2549
int32_t colIdComparFn(const void* param1, const void* param2) {
1,339,182,215✔
2550
  int32_t p1 = *(int32_t*)param1;
1,339,182,215✔
2551
  int32_t p2 = *(int32_t*)param2;
1,339,183,944✔
2552

2553
  if (p1 == p2) {
1,339,187,794✔
2554
    return 0;
×
2555
  } else {
2556
    return (p1 < p2) ? -1 : 1;
1,339,187,794✔
2557
  }
2558
}
2559

2560
static int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock) {
96,393,605✔
2561
  int32_t         code = TSDB_CODE_SUCCESS;
96,393,605✔
2562
  int32_t         lino = 0;
96,393,605✔
2563
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
96,393,605✔
2564
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
96,396,952✔
2565
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
96,397,616✔
2566
  const char*     id = GET_TASKID(pTaskInfo);
96,397,616✔
2567
  SArray* pColList = NULL;
96,397,948✔
2568

2569
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
96,397,948✔
2570
  QUERY_CHECK_CODE(code, lino, _end);
96,397,284✔
2571

2572
  pBlockInfo->rows = pBlock->info.rows;
96,397,284✔
2573
  pBlockInfo->id.uid = pBlock->info.id.uid;
96,383,005✔
2574
  pBlockInfo->type = STREAM_NORMAL;
96,396,952✔
2575
  pBlockInfo->version = pBlock->info.version;
96,397,616✔
2576

2577
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
96,394,960✔
2578
  pBlockInfo->id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
96,395,624✔
2579

2580
  pColList = taosArrayInit(4, sizeof(int32_t));
96,397,616✔
2581
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
96,395,044✔
2582

2583
  // todo extract method
2584
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
512,153,796✔
2585
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
415,761,293✔
2586
    if (!pColMatchInfo->needOutput) {
415,748,436✔
2587
      continue;
604✔
2588
    }
2589

2590
    bool colExists = false;
415,750,448✔
2591
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
1,294,965,628✔
2592
      SColumnInfoData* pResCol = NULL;
1,294,563,551✔
2593
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
1,294,514,382✔
2594
      QUERY_CHECK_CODE(code, lino, _end);
1,294,525,705✔
2595

2596
      if (pResCol->info.colId == pColMatchInfo->colId) {
1,294,525,705✔
2597
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
415,790,222✔
2598
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
415,775,610✔
2599
        QUERY_CHECK_CODE(code, lino, _end);
415,657,708✔
2600

2601
        colExists = true;
415,657,708✔
2602
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
415,657,708✔
2603
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
415,750,110✔
2604
        break;
415,750,110✔
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) {
415,758,148✔
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) {
96,395,012✔
2619
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
123,815,038✔
2620
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
61,907,932✔
2621
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
2622
    if (code) {
61,903,621✔
2623
      QUERY_CHECK_CODE(code, lino, _end);
×
2624
    }
2625

2626
    // reset the error code.
2627
    terrno = 0;
61,903,621✔
2628

2629
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
249,002,943✔
2630
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
187,086,365✔
2631
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
187,096,657✔
2632
    }
2633
  }
2634

2635
  taosArraySort(pColList, colIdComparFn);
96,383,074✔
2636

2637
  int32_t i = 0, j = 0;
96,393,673✔
2638
  while (i < taosArrayGetSize(pColList)) {
699,178,920✔
2639
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
602,762,453✔
2640
    if (slot1 > j) {
602,785,247✔
2641
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
×
2642
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
2643
      j += 1;
×
2644
    } else {
2645
      i += 1;
602,785,247✔
2646
      j += 1;
602,785,247✔
2647
    }
2648
  }
2649

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

2656
  code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL, NULL);
96,396,324✔
2657
  QUERY_CHECK_CODE(code, lino, _end);
96,393,576✔
2658

2659
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
96,393,576✔
2660
  QUERY_CHECK_CODE(code, lino, _end);
96,394,136✔
2661

2662
_end:
96,394,136✔
2663
  taosArrayDestroy(pColList);
95,903,055✔
2664
  if (code != TSDB_CODE_SUCCESS) {
96,390,893✔
2665
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2666
  }
2667
  return code;
96,387,378✔
2668
}
2669

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

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

2701
_end:
6,357,760✔
2702
  if (code != TSDB_CODE_SUCCESS) {
6,358,059✔
2703
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2704
  }
2705
  return code;
6,356,580✔
2706
}
2707

2708
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
119,572,656✔
2709
  int32_t        code = TSDB_CODE_SUCCESS;
119,572,656✔
2710
  int32_t        lino = 0;
119,572,656✔
2711
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
119,572,656✔
2712
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
119,574,619✔
2713

2714
  SStreamScanInfo* pInfo = pOperator->info;
119,573,329✔
2715
  const char*      id = GET_TASKID(pTaskInfo);
119,576,578✔
2716

2717
  qDebug("start to exec queue scan, %s", id);
119,572,969✔
2718

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

2724
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
119,576,578✔
2725
    while (1) {
3,497✔
2726
      SSDataBlock* pResult = NULL;
5,689,299✔
2727
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
5,689,299✔
2728
      QUERY_CHECK_CODE(code, lino, _end);
5,686,308✔
2729

2730
      if (pResult && pResult->info.rows > 0) {
5,686,308✔
2731
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
5,647,641✔
2732
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
5,647,341✔
2733
        QUERY_CHECK_CODE(code, lino, _end);
5,642,256✔
2734
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
5,642,256✔
2735
        if (pResult->info.rows > 0) {
5,649,733✔
2736
          (*ppRes) = pResult;
5,646,237✔
2737
          return code;
5,646,237✔
2738
        }
2739
      } else {
2740
        break;
2741
      }
2742
    }
2743

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

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

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

2758
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
113,919,896✔
2759
    while (1) {
713,910✔
2760
      bool hasResult =
2761
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
114,633,142✔
2762

2763
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
114,611,471✔
2764
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
114,627,016✔
2765

2766
      // curVersion move to next
2767
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pWalReader->curVersion);
114,628,726✔
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);
114,625,251✔
2771

2772
      if (hasResult) {
114,627,750✔
2773
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
96,391,820✔
2774
               pTaskInfo->streamInfo.currentOffset.version);
2775
        blockDataCleanup(pInfo->pRes);
96,400,600✔
2776
        code = setBlockIntoRes(pInfo, pRes);
96,396,620✔
2777
        QUERY_CHECK_CODE(code, lino, _end);
96,387,717✔
2778
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
96,387,717✔
2779
               pTaskInfo->streamInfo.currentOffset.version);
2780
        if (pInfo->pRes->info.rows > 0) {
96,392,530✔
2781
          (*ppRes) = pInfo->pRes;
95,683,062✔
2782
          return code;
95,682,707✔
2783
        }
2784
      } else {
2785
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
18,235,930✔
2786
        (*ppRes) = NULL;
18,235,930✔
2787
        return code;
18,236,190✔
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) {
304,474✔
2813
  int32_t code = TSDB_CODE_SUCCESS;
304,474✔
2814
  int32_t lino = 0;
304,474✔
2815
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));
304,474✔
2816
  QUERY_CHECK_NULL(tableIdList, code, lino, _end, terrno);
306,241✔
2817

2818
  // Transfer the Array of STableKeyInfo into uid list.
2819
  int32_t size = 0;
306,241✔
2820
  code = tableListGetSize(pTableListInfo, &size);
306,241✔
2821
  QUERY_CHECK_CODE(code, lino, _end);
306,539✔
2822
  for (int32_t i = 0; i < size; ++i) {
9,237,729✔
2823
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
8,930,500✔
2824
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
8,925,482✔
2825
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
8,925,482✔
2826
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
8,931,190✔
2827
  }
2828

2829
  (*ppArrayRes) = tableIdList;
307,229✔
2830

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

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

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

2848
  qDebug("tmqsnap doRawScan called");
1,432,801✔
2849
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
1,432,801✔
2850
    bool hasNext = false;
1,428,284✔
2851
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
1,428,284✔
2852
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
1,427,489✔
2853
      if (code != TSDB_CODE_SUCCESS) {
1,425,797✔
2854
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2855
        QUERY_CHECK_CODE(code, lino, _end);
×
2856
      }
2857
    }
2858

2859
    if (pInfo->dataReader && hasNext) {
1,426,592✔
2860
      if (isTaskKilled(pTaskInfo)) {
718,667✔
2861
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2862
        return code;
×
2863
      }
2864

2865
      SSDataBlock* pBlock = NULL;
718,667✔
2866
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock);
718,667✔
2867
      QUERY_CHECK_CODE(code, lino, _end);
717,539✔
2868

2869
      if (pBlock && pBlock->info.rows > 0) {
717,539✔
2870
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
718,385✔
2871
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
718,103✔
2872
        QUERY_CHECK_CODE(code, lino, _end);
717,821✔
2873
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
717,821✔
2874
        (*ppRes) = pBlock;
718,949✔
2875
        return code;
718,667✔
2876
      }
2877
    }
2878

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

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

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

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

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

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

2966
    (*ppRes) = NULL;
4,269✔
2967
    return code;
4,517✔
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) {
106,140✔
2982
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
106,140✔
2983
  pRawScan->pAPI->tsdReader.tsdReaderClose(pRawScan->dataReader);
106,140✔
2984
  pRawScan->pAPI->snapshotFn.destroySnapshot(pRawScan->sContext);
106,140✔
2985
  tableListDestroy(pRawScan->pTableListInfo);
105,892✔
2986
  taosMemoryFree(pRawScan);
106,140✔
2987
}
105,892✔
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) {
105,831✔
2993
  // create operator
2994
  // create tb reader
2995
  // create meta reader
2996
  // create tq reader
2997

2998
  QRY_PARAM_CHECK(pOptrInfo);
105,831✔
2999
  int32_t code = TSDB_CODE_SUCCESS;
106,140✔
3000
  int32_t lino = 0;
106,140✔
3001

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

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

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

3020
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScanNext, NULL, destroyTmqRawScanOperatorInfo, optrDefaultBufFn,
106,140✔
3021
                                         NULL, optrDefaultGetNextExtFn, NULL);
3022
  *pOptrInfo = pOperator;
106,140✔
3023
  return code;
106,140✔
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) {
306,539✔
3036
  if (param == NULL) {
306,539✔
3037
    return;
×
3038
  }
3039

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

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

3072
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
306,539✔
3073
  cleanupExprSupp(&pStreamScan->tagCalSup);
306,539✔
3074

3075
  blockDataDestroy(pStreamScan->pRes);
306,539✔
3076
  blockDataDestroy(pStreamScan->pUpdateRes);
306,240✔
3077
  blockDataDestroy(pStreamScan->pDeleteDataRes);
306,539✔
3078
  blockDataDestroy(pStreamScan->pUpdateDataRes);
306,539✔
3079
  blockDataDestroy(pStreamScan->pCreateTbRes);
306,539✔
3080
  taosArrayDestroy(pStreamScan->pBlockLists);
306,539✔
3081
  blockDataDestroy(pStreamScan->pCheckpointRes);
306,539✔
3082

3083
  taosMemoryFree(pStreamScan);
306,539✔
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,
306,216✔
3104
                                                  SNode* pTagCond, STableListInfo* pTableListInfo,
3105
                                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
3106
  QRY_PARAM_CHECK(pOptrInfo);
306,216✔
3107

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

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

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

3124
  pInfo->pTagCond = pTagCond;
306,539✔
3125
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
306,539✔
3126

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

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

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

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

3148
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
1,621,831✔
3149
      pInfo->primaryTsIndex = id->dstSlotId;
306,539✔
3150
    }
3151
    if (id->isPk) {
1,621,508✔
3152
      pInfo->primaryKeyIndex = id->dstSlotId;
1,398✔
3153
      pInfo->basic.primaryPkIndex = id->dstSlotId;
1,398✔
3154
      pkType = id->dataType;
1,398✔
3155
    }
3156
  }
3157

3158
  pInfo->pPartTbnameSup = NULL;
306,539✔
3159
  if (pTableScanNode->pSubtable != NULL) {
306,539✔
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) {
306,539✔
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));
306,539✔
3188
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
306,216✔
3189

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

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

3203
    STableKeyInfo* pList = NULL;
305,694✔
3204
    int32_t        num = 0;
305,692✔
3205
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
305,392✔
3206
    QUERY_CHECK_CODE(code, lino, _error);
305,692✔
3207

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

3213
    if (pHandle->initTqReader) {
304,479✔
3214
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
304,777✔
3215
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
306,539✔
3216
    } else {
3217
      pInfo->tqReader = pHandle->tqReader;
841✔
3218
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
3219
    }
3220

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

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

3234
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
306,241✔
3235
    QUERY_CHECK_CODE(code, lino, _error);
305,563✔
3236

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

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

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

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

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

3265
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
306,539✔
3266
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
306,539✔
3267
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
306,282✔
3268
  QUERY_CHECK_CODE(code, lino, _error);
305,975✔
3269

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

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

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

3295
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
305,975✔
3296
  QUERY_CHECK_CODE(code, lino, _error);
306,539✔
3297

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

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

3305
  *pOptrInfo = pOperator;
306,232✔
3306
  return code;
306,539✔
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) {
2,409,700✔
3330
  int32_t        code = TSDB_CODE_SUCCESS;
2,409,700✔
3331
  int32_t        lino = 0;
2,409,700✔
3332
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
2,409,700✔
3333
  STagScanInfo*  pInfo = pOperator->info;
2,409,708✔
3334
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
2,409,708✔
3335
  int32_t        count = pRes->info.rows;
2,409,708✔
3336

3337
  STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
2,409,704✔
3338
  if (!item) {
2,409,676✔
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);
2,409,676✔
3346
  tDecoderClear(&(*mr).coder);
2,409,700✔
3347
  if (code != TSDB_CODE_SUCCESS) {
2,409,670✔
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];
2,409,574✔
3354
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
2,761,425✔
3355
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
351,725✔
3356

3357
    // refactor later
3358
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
351,759✔
3359
      STR_TO_VARSTR(str, (*mr).me.name);
127,303✔
3360
      code = colDataSetVal(pDst, (count), str, false);
127,303✔
3361
      QUERY_CHECK_CODE(code, lino, _end);
127,303✔
3362
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
224,456✔
3363
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
14,293✔
3364
      QUERY_CHECK_CODE(code, lino, _end);
14,293✔
3365
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
210,163✔
3366
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
14,293✔
3367
      QUERY_CHECK_CODE(code, lino, _end);
14,293✔
3368
    } else {  // it is a tag value
3369
      STagVal val = {0};
195,870✔
3370
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
195,870✔
3371
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
195,870✔
3372

3373
      char* data = NULL;
195,870✔
3374
      if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
195,870✔
3375
        data = tTagValToData((const STagVal*)p, false);
194,547✔
3376
      } else {
3377
        data = (char*)p;
1,323✔
3378
      }
3379

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

3384
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
195,870✔
3385
          (data != NULL)) {
3386
        taosMemoryFree(data);
76,825✔
3387
      }
3388
    }
3389
  }
3390

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

3399
  return code;
2,409,696✔
3400
}
3401

3402
static void tagScanFreeUidTag(void* p) {
7,940,866✔
3403
  STUidTagInfo* pInfo = p;
7,940,866✔
3404
  if (pInfo->pTagVal != NULL) {
7,940,866✔
3405
    taosMemoryFree(pInfo->pTagVal);
7,944,905✔
3406
  }
3407
}
7,944,183✔
3408

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

3415
  pColumnData->info.type = pType->type;
1,533,526✔
3416
  pColumnData->info.bytes = pType->bytes;
1,531,815✔
3417
  pColumnData->info.scale = pType->scale;
1,532,408✔
3418
  pColumnData->info.precision = pType->precision;
1,532,622✔
3419

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

3427
  pParam->columnData = pColumnData;
1,532,751✔
3428
  pParam->colAlloced = true;
1,533,191✔
3429
  return TSDB_CODE_SUCCESS;
1,534,644✔
3430
}
3431

3432
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
9,352,137✔
3433
  int32_t                code = TSDB_CODE_SUCCESS;
9,352,137✔
3434
  int32_t                lino = 0;
9,352,137✔
3435
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
9,352,137✔
3436
  SColumnNode*           pSColumnNode = NULL;
9,352,137✔
3437
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
9,353,397✔
3438
    pSColumnNode = *(SColumnNode**)pNode;
2,620,641✔
3439
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
6,735,948✔
3440
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
478✔
3441
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
478✔
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;
478✔
3455
    }
3456
  } else {
3457
    return DEAL_RES_CONTINUE;
6,733,572✔
3458
  }
3459

3460
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
2,619,979✔
3461
  if (!data) {
2,621,097✔
3462
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
1,537,761✔
3463
    if (code == TSDB_CODE_DUP_KEY) {
1,538,758✔
3464
      code = TSDB_CODE_SUCCESS;
×
3465
    }
3466
    QUERY_CHECK_CODE(code, lino, _end);
1,538,758✔
3467
    pSColumnNode->slotId = pCtx->index++;
1,538,758✔
3468
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
1,536,849✔
3469
                         .type = pSColumnNode->node.resType.type,
1,536,850✔
3470
                         .bytes = pSColumnNode->node.resType.bytes};
1,535,914✔
3471
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
1,537,177✔
3472
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,538,758✔
3473
  } else {
3474
    SColumnNode* col = *(SColumnNode**)data;
1,083,336✔
3475
    pSColumnNode->slotId = col->slotId;
1,083,336✔
3476
  }
3477

3478
_end:
2,620,695✔
3479
  if (code != TSDB_CODE_SUCCESS) {
2,620,056✔
3480
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
335✔
3481
    return DEAL_RES_ERROR;
×
3482
  }
3483
  return DEAL_RES_CONTINUE;
2,619,721✔
3484
}
3485

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

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

3496
  pBlockList = taosArrayInit(1, POINTER_BYTES);
1,534,453✔
3497
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
1,533,844✔
3498

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

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

3504
  SScalarParam output = {0};
1,534,203✔
3505
  code = tagScanCreateResultData(&type, numOfTables, &output);
1,534,332✔
3506
  QUERY_CHECK_CODE(code, lino, _end);
1,532,447✔
3507

3508
  gTaskScalarExtra.pStreamInfo = (void*)pStreamRuntimeInfo;
1,532,447✔
3509
  gTaskScalarExtra.pStreamRange = NULL;
1,532,447✔
3510
  code = scalarCalculate(pTagCond, pBlockList, &output, &gTaskScalarExtra);
1,532,591✔
3511
  QUERY_CHECK_CODE(code, lino, _end);
1,531,614✔
3512

3513
  bool* result = (bool*)output.columnData->pData;
1,531,614✔
3514
  for (int32_t i = 0; i < numOfTables; ++i) {
3,834,772✔
3515
    if (result[i]) {
2,299,980✔
3516
      void* tmp = taosArrayPush(aFilterIdxs, &i);
2,185,646✔
3517
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,185,646✔
3518
    }
3519
  }
3520

3521
  colDataDestroy(output.columnData);
1,534,675✔
3522
  taosMemoryFreeClear(output.columnData);
1,533,799✔
3523

3524
_end:
1,533,982✔
3525
  blockDataDestroy(pResBlock);
1,531,005✔
3526
  taosArrayDestroy(pBlockList);
1,531,276✔
3527
  if (code != TSDB_CODE_SUCCESS) {
1,530,219✔
3528
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3529
  }
3530
  return code;
1,530,219✔
3531
}
3532

3533
static int32_t tagScanFillOneCellWithTag(SOperatorInfo* pOperator, const STUidTagInfo* pUidTagInfo,
28,152,814✔
3534
                                         SExprInfo* pExprInfo, SColumnInfoData* pColInfo, int rowIndex,
3535
                                         const SStorageAPI* pAPI, void* pVnode) {
3536
  int32_t code = TSDB_CODE_SUCCESS;
28,152,814✔
3537
  int32_t lino = 0;
28,152,814✔
3538
  if (QUERY_NODE_FUNCTION == pExprInfo->pExpr->nodeType) {
28,152,814✔
3539
    if (FUNCTION_TYPE_TBNAME == pExprInfo->pExpr->_function.functionType) {  // tbname
13,301,620✔
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,296,369✔
3546
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
6,646,217✔
3547
      QUERY_CHECK_CODE(code, lino, _end);
6,650,338✔
3548
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
6,651,137✔
3549
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
6,646,248✔
3550
      QUERY_CHECK_CODE(code, lino, _end);
6,650,475✔
3551
    }
3552
  } else {
3553
    STagVal tagVal = {0};
14,848,541✔
3554
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
14,848,085✔
3555
    if (pUidTagInfo->pTagVal == NULL) {
14,851,565✔
3556
      colDataSetNULL(pColInfo, rowIndex);
×
3557
    } else {
3558
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
14,848,717✔
3559

3560
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
14,854,505✔
3561
        colDataSetNULL(pColInfo, rowIndex);
7,740✔
3562
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
14,846,309✔
3563
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
3564
        QUERY_CHECK_CODE(code, lino, _end);
×
3565
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
18,365,683✔
3566
        if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
3,512,760✔
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,522,995✔
3570
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,519,928✔
3571

3572
        varDataSetLen(tmp, tagVal.nData);
3,519,928✔
3573
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
3,520,993✔
3574
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
3,521,977✔
3575
        taosMemoryFree(tmp);
3,523,301✔
3576
        QUERY_CHECK_CODE(code, lino, _end);
3,523,438✔
3577
      } else {
3578
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
11,312,286✔
3579
        QUERY_CHECK_CODE(code, lino, _end);
11,318,561✔
3580
      }
3581
    }
3582
  }
3583

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

3591
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
5,107,634✔
3592
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
3593
  int32_t       code = TSDB_CODE_SUCCESS;
5,107,634✔
3594
  int32_t       lino = 0;
5,107,634✔
3595
  STagScanInfo* pInfo = pOperator->info;
5,107,634✔
3596
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
5,108,994✔
3597
  if (!ignoreFilterIdx) {
5,108,507✔
3598
    size_t szTables = taosArrayGetSize(aFilterIdxs);
1,532,518✔
3599
    for (int i = 0; i < szTables; ++i) {
3,716,576✔
3600
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
2,183,185✔
3601
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
2,182,051✔
3602
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
2,183,059✔
3603
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
12,525,796✔
3604
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
10,340,465✔
3605
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
10,345,169✔
3606
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
10,345,169✔
3607
        QUERY_CHECK_CODE(code, lino, _end);
10,342,737✔
3608
      }
3609
    }
3610
  } else {
3611
    size_t szTables = taosArrayGetSize(aUidTags);
3,575,989✔
3612
    for (int i = 0; i < szTables; ++i) {
9,227,272✔
3613
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
5,649,595✔
3614
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
5,650,070✔
3615
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
23,452,383✔
3616
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
17,802,631✔
3617
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
17,803,612✔
3618
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
17,803,612✔
3619
        QUERY_CHECK_CODE(code, lino, _end);
17,802,313✔
3620
      }
3621
    }
3622
  }
3623

3624
_end:
3,577,677✔
3625
  if (code != TSDB_CODE_SUCCESS) {
5,111,068✔
3626
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3627
  }
3628
  return code;
5,110,691✔
3629
}
3630

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

3637
  STagScanInfo* pInfo = pOperator->info;
9,833,337✔
3638
  SSDataBlock*  pRes = pInfo->pRes;
9,830,157✔
3639

3640
  QRY_PARAM_CHECK(ppRes);
9,843,145✔
3641

3642
  if (pOperator->status == OP_EXEC_DONE) {
9,843,076✔
3643
    return TSDB_CODE_SUCCESS;
4,668,713✔
3644
  }
3645
  blockDataCleanup(pRes);
5,158,358✔
3646

3647
  if (pInfo->pCtbCursor == NULL) {
5,176,324✔
3648
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
5,174,864✔
3649
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
5,171,323✔
3650
  } else {
3651
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
3652
    QUERY_CHECK_CODE(code, lino, _end);
×
3653
  }
3654

3655
  SArray* aUidTags = pInfo->aUidTags;
5,169,207✔
3656
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
5,173,378✔
3657
  int32_t count = 0;
5,161,732✔
3658
  bool    ctbCursorFinished = false;
5,161,732✔
3659
  while (1) {
72,136✔
3660
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
5,233,868✔
3661
    taosArrayClear(aFilterIdxs);
5,245,720✔
3662

3663
    int32_t numTables = 0;
5,244,623✔
3664
    while (numTables < pOperator->resultInfo.capacity) {
13,186,368✔
3665
      SMCtbCursor* pCur = pInfo->pCtbCursor;
13,186,972✔
3666
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
13,187,383✔
3667
      if (uid == 0) {
13,187,274✔
3668
        ctbCursorFinished = true;
5,252,214✔
3669
        break;
5,252,214✔
3670
      }
3671
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
7,935,060✔
3672
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
7,933,386✔
3673
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
7,928,674✔
3674

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

3681
    if (numTables == 0) {
5,252,214✔
3682
      break;
139,976✔
3683
    }
3684
    bool ignoreFilterIdx = true;
5,112,238✔
3685
    if (pInfo->pTagCond != NULL) {
5,112,238✔
3686
      ignoreFilterIdx = false;
1,535,587✔
3687
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo, pTaskInfo->pStreamRuntimeInfo);
1,535,587✔
3688
      QUERY_CHECK_CODE(code, lino, _end);
1,531,638✔
3689
    } else {
3690
      ignoreFilterIdx = true;
3,577,126✔
3691
    }
3692

3693
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
5,108,764✔
3694
    QUERY_CHECK_CODE(code, lino, _end);
5,110,364✔
3695

3696
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
5,110,364✔
3697

3698
    if (count != 0) {
5,111,147✔
3699
      break;
5,039,011✔
3700
    }
3701
  }
3702

3703
  if (count > 0) {
5,178,987✔
3704
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
5,038,555✔
3705
  }
3706
  if (count == 0 || ctbCursorFinished) {
5,175,114✔
3707
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
5,177,496✔
3708
    pInfo->pCtbCursor = NULL;
5,169,778✔
3709
    setOperatorCompleted(pOperator);
5,170,234✔
3710
  }
3711

3712
  pRes->info.rows = count;
5,168,512✔
3713
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
5,169,745✔
3714
  if (bLimitReached) {
5,168,262✔
3715
    setOperatorCompleted(pOperator);
×
3716
  }
3717

3718
  pOperator->resultInfo.totalRows += pRes->info.rows;
5,168,262✔
3719
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
5,159,318✔
3720

3721
_end:
5,157,780✔
3722
  if (code != TSDB_CODE_SUCCESS) {
5,157,780✔
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;
5,157,780✔
3729
}
3730

3731
static int32_t createTagScanTableListInfoFromParam(SOperatorInfo* pOperator) {
24,847✔
3732
  STagScanInfo*          pInfo = pOperator->info;
24,847✔
3733
  SExecTaskInfo*         pTaskInfo = pOperator->pTaskInfo;
24,847✔
3734
  int32_t                code = 0;
24,847✔
3735
  STableListInfo*        pListInfo = pInfo->pTableListInfo;
24,847✔
3736
  STagScanOperatorParam* pParam = (STagScanOperatorParam*)pOperator->pOperatorGetParam->value;
24,847✔
3737
  tb_uid_t               pUid = pParam->vcUid;
24,847✔
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};
24,847✔
3744

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

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

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

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

3764
  return code;
24,847✔
3765
}
3766

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

3780
    if (pOperator->status == OP_EXEC_DONE) {
24,847✔
3781
      pOperator->status = OP_OPENED;
11,634✔
3782
    }
3783
  }
3784

3785
  if (pOperator->status == OP_EXEC_DONE) {
4,418,336✔
3786
    (*ppRes) = NULL;
2,067,435✔
3787
    return code;
2,067,435✔
3788
  }
3789

3790
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
2,350,901✔
3791
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
2,350,901✔
3792

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

3797
  int32_t size = 0;
2,350,901✔
3798
  code = tableListGetSize(pInfo->pTableListInfo, &size);
2,350,901✔
3799
  QUERY_CHECK_CODE(code, lino, _end);
2,350,901✔
3800

3801
  if (size == 0) {
2,350,901✔
3802
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
16,896✔
3803
    (*ppRes) = NULL;
16,896✔
3804
    return code;
16,896✔
3805
  }
3806

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

3811
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
4,743,701✔
3812
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
2,409,674✔
3813
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD &&
2,409,674✔
3814
        code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
3815
      // ignore other error
3816
      code = TSDB_CODE_SUCCESS;
2,409,063✔
3817
    }
3818
    QUERY_CHECK_CODE(code, lino, _end);
2,409,674✔
3819

3820
    if (++pInfo->curPos >= size) {
2,409,674✔
3821
      setOperatorCompleted(pOperator);
2,334,005✔
3822
    }
3823
  }
3824

3825
  pAPI->metaReaderFn.clearReader(&mr);
2,333,975✔
3826
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
2,334,005✔
3827
  if (bLimitReached) {
2,334,005✔
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) {
2,334,005✔
3833
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
2,334,001✔
3834
  }
3835

3836
  pOperator->resultInfo.totalRows += pRes->info.rows;
2,334,005✔
3837

3838
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
2,334,001✔
3839
_end:
2,334,001✔
3840
  if (code != TSDB_CODE_SUCCESS) {
2,334,001✔
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;
2,334,001✔
3846
}
3847

3848
static void destroyTagScanOperatorInfo(void* param) {
8,115,725✔
3849
  STagScanInfo* pInfo = (STagScanInfo*)param;
8,115,725✔
3850
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
8,115,725✔
3851
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
3852
  }
3853
  taosHashCleanup(pInfo->filterCtx.colHash);
8,119,170✔
3854
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
8,115,055✔
3855
  taosArrayDestroy(pInfo->aFilterIdxs);
8,113,829✔
3856
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
8,113,426✔
3857

3858
  blockDataDestroy(pInfo->pRes);
8,117,635✔
3859
  taosArrayDestroy(pInfo->matchInfo.pList);
8,119,560✔
3860
  tableListDestroy(pInfo->pTableListInfo);
8,119,709✔
3861

3862
  pInfo->pRes = NULL;
8,121,187✔
3863
  pInfo->pTableListInfo = NULL;
8,121,514✔
3864
  taosMemoryFreeClear(param);
8,121,567✔
3865
}
8,120,215✔
3866

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

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

3882
  return code;
4,590✔
3883
}
3884

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

3890
  int32_t         code = TSDB_CODE_SUCCESS;
8,112,215✔
3891
  int32_t         lino = 0;
8,112,215✔
3892
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
8,112,215✔
3893
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
8,112,215✔
3894
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
8,105,191✔
3895
  if (pInfo == NULL || pOperator == NULL) {
8,088,108✔
3896
    code = terrno;
×
3897
    goto _error;
×
3898
  }
3899
  pOperator->pPhyNode = pTagScanNode;
8,089,676✔
3900
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
8,091,974✔
3901

3902
  int32_t    numOfExprs = 0;
8,093,245✔
3903
  SExprInfo* pExprInfo = NULL;
8,097,944✔
3904

3905
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
8,099,157✔
3906
  QUERY_CHECK_CODE(code, lino, _error);
8,103,758✔
3907

3908
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
8,103,758✔
3909
  QUERY_CHECK_CODE(code, lino, _error);
8,103,460✔
3910

3911
  int32_t num = 0;
8,103,460✔
3912
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
8,100,740✔
3913
  QUERY_CHECK_CODE(code, lino, _error);
8,113,715✔
3914

3915
  pInfo->pTagCond = pTagCond;
8,113,715✔
3916
  pInfo->pTagIndexCond = pTagIndexCond;
8,113,715✔
3917
  pInfo->suid = pPhyNode->suid;
8,112,190✔
3918
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
8,108,251✔
3919

3920
  pInfo->pTableListInfo = pTableListInfo;
8,103,727✔
3921
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
8,098,322✔
3922
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
8,117,189✔
3923

3924
  pInfo->readHandle = *pReadHandle;
8,106,026✔
3925
  pInfo->curPos = 0;
8,101,901✔
3926

3927
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
8,105,593✔
3928
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
8,104,933✔
3929
                  pTaskInfo);
3930
  initResultSizeInfo(&pOperator->resultInfo, 4096);
8,107,985✔
3931
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
8,102,499✔
3932
  QUERY_CHECK_CODE(code, lino, _error);
8,120,279✔
3933

3934
  if (pTagScanNode->onlyMetaCtbIdx) {
8,120,279✔
3935
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
5,534,363✔
3936
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
5,540,840✔
3937

3938
    pInfo->aFilterIdxs = taosArrayInit(pOperator->resultInfo.capacity, sizeof(int32_t));
5,534,714✔
3939
    QUERY_CHECK_NULL(pInfo->aFilterIdxs, code, lino, _error, terrno);
5,538,914✔
3940

3941
    pInfo->filterCtx.colHash =
5,537,964✔
3942
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
5,536,121✔
3943
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
5,539,113✔
3944

3945
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
5,534,363✔
3946
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
5,540,550✔
3947

3948
    if (pInfo->pTagCond != NULL) {
5,531,937✔
3949
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
1,538,431✔
3950
    }
3951
  }
3952
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
8,119,162✔
3953
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
8,117,768✔
3954
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
3955
  setOperatorResetStateFn(pOperator, resetTagScanOperatorState);     
8,116,019✔
3956
  
3957
  *pOptrInfo = pOperator;
8,120,523✔
3958
  return code;
8,121,536✔
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) {
221,344,035✔
3976
  int32_t                 left = *(int32_t*)pLeft;
221,344,035✔
3977
  int32_t                 right = *(int32_t*)pRight;
221,344,035✔
3978
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
221,343,365✔
3979

3980
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
221,343,365✔
3981
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
221,351,740✔
3982

3983
  if (leftIdx == -1) {
221,354,420✔
3984
    return 1;
189,648,500✔
3985
  } else if (rightIdx == -1) {
31,705,920✔
3986
    return -1;
313,920✔
3987
  }
3988

3989
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
31,392,000✔
3990
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
31,392,000✔
3991
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
31,392,000✔
3992
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
31,392,000✔
3993
    ret = -1 * ret;
×
3994
  }
3995
  return ret;
31,392,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,815,518✔
4026
  int32_t code = TSDB_CODE_SUCCESS;
2,815,518✔
4027
  int32_t lino = 0;
2,815,518✔
4028
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
2,815,518✔
4029
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
2,815,518✔
4030
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
2,805,309✔
4031
  for (int i = 0; i < src->numOfCols; i++) {
9,472,903✔
4032
    dst->colList[i] = src->colList[i];
6,690,939✔
4033
  }
4034

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

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

4046
  STableMergeScanInfo*    pInfo = pOperator->info;
4,226,012✔
4047
  SReadHandle*            pHandle = &pInfo->base.readHandle;
4,228,381✔
4048
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
4,224,768✔
4049
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
4,227,065✔
4050
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
4,225,749✔
4051

4052
  blockDataCleanup(pInput->pReaderBlock);
4,221,745✔
4053
  if (!pInput->bInMemReader) {
4,226,746✔
4054
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
4,246,657✔
4055
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
2,123,173✔
4056
    if (code != 0) {
2,123,173✔
4057
      return code;
×
4058
    }
4059
  }
4060

4061
  pInfo->base.dataReader = pInput->pReader;
4,225,430✔
4062

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

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

4079
      *pSubTableHasBlock = false;
2,825,033✔
4080
      break;
2,823,725✔
4081
    }
4082

4083
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
1,414,020✔
4084
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
1,387,890✔
4085
    } else {
4086
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
26,130✔
4087
    }
4088

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

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

4104
    *pSubTableHasBlock = true;
1,406,315✔
4105
    break;
1,406,650✔
4106
  }
4107

4108
  if (*pSubTableHasBlock) {
4,231,348✔
4109
    pInput->pReaderBlock->info.id.groupId =
2,812,630✔
4110
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
2,811,960✔
4111
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
1,406,650✔
4112
  }
4113
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
4,230,686✔
4114
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
3,180,762✔
4115
    pInput->pReader = NULL;
3,180,108✔
4116
  }
4117

4118
  pInfo->base.dataReader = NULL;
4,229,043✔
4119
  return TSDB_CODE_SUCCESS;
4,230,351✔
4120
}
4121

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

4125
  int32_t numOfTables = 0;
1,762,515✔
4126
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
1,742,129✔
4127
  if (code != TSDB_CODE_SUCCESS) {
1,729,360✔
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,729,360✔
4133
  for (; i < numOfTables; ++i) {
2,784,644✔
4134
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
1,109,950✔
4135
    if (!tableKeyInfo) {
1,108,610✔
4136
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4137
      return terrno;
×
4138
    }
4139
    if (tableKeyInfo->groupId != pInfo->groupId) {
1,108,610✔
4140
      break;
55,990✔
4141
    }
4142
  }
4143
  pInfo->tableEndIndex = i - 1;
1,730,684✔
4144
  return TSDB_CODE_SUCCESS;
1,765,123✔
4145
}
4146

4147
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
1,768,417✔
4148
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
4,590,826✔
4149
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
2,823,717✔
4150
    if (pInput->rowIdx == -1) {
2,823,709✔
4151
      continue;
1,417,729✔
4152
    }
4153

4154
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
1,405,980✔
4155
      pInput->rowIdx = 0;
1,405,980✔
4156
      pInput->pageIdx = -1;
1,406,315✔
4157
    }
4158

4159
    pInput->pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,405,645✔
4160
    SColumnInfoData* col = taosArrayGet(pInput->pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
1,405,645✔
4161
    if (!col) {
1,405,980✔
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,405,980✔
4166
  }
4167

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

4172
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
1,760,864✔
4173
  int32_t code = setGroupStartEndIndex(pInfo);
1,760,864✔
4174
  if (code != TSDB_CODE_SUCCESS) {
1,761,159✔
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,761,159✔
4179
  if (pSubTblsInfo == NULL) {
1,748,661✔
4180
    return terrno;
×
4181
  }
4182
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
1,748,661✔
4183
  if (!pSubTblsInfo->pTsOrderInfo) {
1,755,903✔
4184
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4185
    return terrno;
×
4186
  }
4187
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
1,746,683✔
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,740,143✔
4195
  }
4196
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
1,754,284✔
4197
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
1,739,186✔
4198
  if (pSubTblsInfo->aInputs == NULL) {
1,735,246✔
4199
    taosMemoryFree(pSubTblsInfo);
×
4200
    return terrno;
×
4201
  }
4202
  int32_t bufPageSize = pInfo->bufPageSize;
1,741,842✔
4203
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
1,744,785✔
4204
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
1,750,041✔
4205
  if (code != TSDB_CODE_SUCCESS) {
1,747,425✔
4206
    taosMemoryFree(pSubTblsInfo->aInputs);
×
4207
    taosMemoryFree(pSubTblsInfo);
×
4208
    return code;
×
4209
  }
4210
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
1,747,425✔
4211
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
1,752,665✔
4212

4213
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
1,750,057✔
4214
  return TSDB_CODE_SUCCESS;
1,732,638✔
4215
}
4216

4217
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
1,752,346✔
4218
  int32_t        code = TSDB_CODE_SUCCESS;
1,752,346✔
4219
  int32_t        lino = 0;
1,752,346✔
4220
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
1,752,346✔
4221
  SReadHandle*   pHandle = &pInfo->base.readHandle;
1,751,381✔
4222
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
1,741,531✔
4223

4224
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
1,752,346✔
4225

4226
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
4,559,609✔
4227
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
2,813,867✔
4228
    pInput->type = SUB_TABLE_MEM_BLOCK;
2,807,614✔
4229

4230
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
2,796,145✔
4231
    QUERY_CHECK_CODE(code, lino, _end);
2,799,104✔
4232

4233
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
2,799,104✔
4234
    QUERY_CHECK_CODE(code, lino, _end);
2,821,763✔
4235

4236
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
2,821,763✔
4237
    QUERY_CHECK_CODE(code, lino, _end);
2,824,698✔
4238

4239
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
2,824,698✔
4240
    pInput->pKeyInfo = keyInfo;
2,820,774✔
4241

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

4246
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
2,812,543✔
4247
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
2,108,590✔
4248
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
1,054,630✔
4249
      QUERY_CHECK_CODE(code, lino, _end);
1,054,295✔
4250
      pInput->bInMemReader = true;
1,054,295✔
4251
    } else {
4252
      pInput->pReader = NULL;
1,767,109✔
4253
      pInput->bInMemReader = false;
1,766,120✔
4254
    }
4255
    bool hasNext = true;
2,822,688✔
4256
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
2,819,354✔
4257
    QUERY_CHECK_CODE(code, lino, _end);
2,823,366✔
4258
    if (!hasNext) {
2,823,366✔
4259
      pInput->rowIdx = -1;
1,417,386✔
4260
      ++pSubTblsInfo->numSubTablesCompleted;
1,418,040✔
4261
      continue;
1,418,383✔
4262
    } else {
4263
      pInput->rowIdx = 0;
1,405,980✔
4264
      pInput->pageIdx = -1;
1,406,315✔
4265
    }
4266
  }
4267

4268
_end:
1,768,417✔
4269
  if (code != TSDB_CODE_SUCCESS) {
1,769,398✔
4270
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4271
  }
4272
  return code;
1,766,774✔
4273
}
4274

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

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

4292
_end:
1,406,315✔
4293
  if (code != TSDB_CODE_SUCCESS) {
1,406,315✔
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,406,315✔
4299

4300
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
256,038,673✔
4301
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
256,038,673✔
4302
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
256,040,659✔
4303

4304
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
256,060,232✔
4305
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
256,049,728✔
4306
    ++pInput->rowIdx;
254,629,176✔
4307
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
1,406,650✔
4308
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
1,406,650✔
4309
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
1,406,650✔
4310
    }
4311
    if (pInput->rowIdx != -1) {
1,406,315✔
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));
256,057,010✔
4323
}
4324

4325
static int32_t appendChosenRowToDataBlock(STmsSubTablesMergeInfo* pSubTblsInfo, SSDataBlock* pBlock) {
255,969,237✔
4326
  int32_t            code = TSDB_CODE_SUCCESS;
255,969,237✔
4327
  int32_t            lino = 0;
255,969,237✔
4328
  STmsSubTableInput* pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
255,969,237✔
4329
  SSDataBlock*       pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
256,026,575✔
4330

4331
  for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
1,056,785,257✔
4332
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
800,065,735✔
4333
    QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
800,021,482✔
4334

4335
    SColumnInfoData* pSrcColInfo = taosArrayGet(pInputBlock->pDataBlock, i);
800,021,482✔
4336
    QUERY_CHECK_NULL(pSrcColInfo, code, lino, _end, terrno);
799,642,090✔
4337
    bool isNull = colDataIsNull(pSrcColInfo, pInputBlock->info.rows, pInput->rowIdx, NULL);
799,642,090✔
4338

4339
    if (isNull) {
800,167,847✔
4340
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
79,363,720✔
4341
      QUERY_CHECK_CODE(code, lino, _end);
79,349,315✔
4342
    } else {
4343
      if (pSrcColInfo->pData != NULL) {
720,804,127✔
4344
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
720,799,868✔
4345
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
721,617,299✔
4346
        QUERY_CHECK_CODE(code, lino, _end);
721,471,433✔
4347
      }
4348
    }
4349
  }
4350
  pBlock->info.dataLoad = 1;
256,073,025✔
4351
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
256,072,706✔
4352
  pBlock->info.rows += 1;
256,066,700✔
4353

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

4361
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
2,860,820✔
4362
                                       SSDataBlock** pResBlock) {
4363
  int32_t                 code = TSDB_CODE_SUCCESS;
2,860,820✔
4364
  int32_t                 lino = 0;
2,860,820✔
4365
  STableMergeScanInfo*    pInfo = pOperator->info;
2,860,820✔
4366
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
2,862,463✔
4367
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
2,861,482✔
4368
  bool                    finished = false;
2,859,831✔
4369

4370
  QRY_PARAM_CHECK(pResBlock);
2,859,831✔
4371

4372
  blockDataCleanup(pBlock);
2,858,515✔
4373

4374
  while (true) {
×
4375
    while (true) {
4376
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
258,838,025✔
4377
        finished = true;
2,862,790✔
4378
        break;
2,862,790✔
4379
      }
4380

4381
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
255,975,897✔
4382
      QUERY_CHECK_CODE(code, lino, _end);
256,061,109✔
4383

4384
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
256,061,109✔
4385
      QUERY_CHECK_CODE(code, lino, _end);
255,971,295✔
4386

4387
      if (pBlock->info.rows >= capacity) {
255,971,295✔
4388
        break;
×
4389
      }
4390
    }
4391

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

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

4402
  if (pBlock->info.rows > 0) {
2,861,155✔
4403
    *pResBlock = pBlock;
1,092,730✔
4404
  }
4405

4406
_end:
1,768,425✔
4407
  if (code != TSDB_CODE_SUCCESS) {
2,861,155✔
4408
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4409
    pTaskInfo->code = code;
×
4410
  }
4411
  return code;
2,861,809✔
4412
}
4413

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

4419
  code = initSubTablesMergeInfo(pInfo);
1,762,188✔
4420
  QUERY_CHECK_CODE(code, lino, _end);
1,737,894✔
4421

4422
  code = initSubTableInputs(pOperator, pInfo);
1,737,894✔
4423
  QUERY_CHECK_CODE(code, lino, _end);
1,768,744✔
4424

4425
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
1,768,744✔
4426
  QUERY_CHECK_CODE(code, lino, _end);
1,766,128✔
4427

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

4435
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
30,296,721✔
4436
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
30,296,721✔
4437
  if (pSubTblsInfo != NULL) {
30,297,954✔
4438
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
1,766,790✔
4439

4440
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
4,592,796✔
4441
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
2,823,725✔
4442
      taosMemoryFree(pInput->tblCond.colList);
2,824,706✔
4443
      blockDataDestroy(pInput->pReaderBlock);
2,824,379✔
4444
      blockDataDestroy(pInput->pPageBlock);
2,824,379✔
4445
      taosArrayDestroy(pInput->aBlockPages);
2,824,371✔
4446
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
2,824,371✔
4447
      pInput->pReader = NULL;
2,824,379✔
4448
    }
4449

4450
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
1,769,079✔
4451
    taosMemoryFree(pSubTblsInfo->aInputs);
1,767,771✔
4452

4453
    taosMemoryFree(pSubTblsInfo);
1,768,736✔
4454
    pInfo->pSubTablesMergeInfo = NULL;
1,767,755✔
4455

4456
    // taosMemoryTrim(0);
4457
  }
4458
}
30,298,919✔
4459

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

4463
  int32_t lino = 0;
2,841,545✔
4464
  int32_t tableListSize = 0;
2,841,545✔
4465
  int64_t st = taosGetTimestampUs();
2,844,456✔
4466

4467
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
2,844,456✔
4468
  STableMergeScanInfo* pInfo = pOperator->info;
2,834,614✔
4469

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

4474
  int32_t code = pOperator->fpSet._openFn(pOperator);
2,820,816✔
4475
  QUERY_CHECK_CODE(code, lino, _end);
2,820,513✔
4476

4477
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
2,820,513✔
4478
  QUERY_CHECK_CODE(code, lino, _end);
2,834,957✔
4479

4480
  if (!pInfo->hasGroupId) {
2,834,957✔
4481
    pInfo->hasGroupId = true;
1,741,892✔
4482

4483
    if (tableListSize == 0) {
1,727,440✔
4484
      setOperatorCompleted(pOperator);
45,225✔
4485
      (*ppRes) = NULL;
45,225✔
4486
      return code;
45,225✔
4487
    }
4488

4489
    pInfo->tableStartIndex = 0;
1,682,215✔
4490
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
1,683,180✔
4491
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
1,677,637✔
4492
    pInfo->groupId = pTmpGpId->groupId;
1,677,637✔
4493
    code = startSubTablesTableMergeScan(pOperator);
1,696,356✔
4494
    QUERY_CHECK_CODE(code, lino, _end);
1,711,446✔
4495
  }
4496

4497
  SSDataBlock* pBlock = NULL;
2,805,181✔
4498
  while (pInfo->tableStartIndex < tableListSize) {
2,859,839✔
4499
    if (isTaskKilled(pTaskInfo)) {
2,861,474✔
4500
      break;
×
4501
    }
4502

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

4506
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
2,861,155✔
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,859,512✔
4514
      pBlock->info.id.groupId = pInfo->groupId;
1,092,395✔
4515
      pOperator->resultInfo.totalRows += pBlock->info.rows;
1,092,730✔
4516
      pInfo->bGroupProcessed = true;
1,092,730✔
4517
      break;
1,092,395✔
4518
    } else {
4519
      // Data of this group are all dumped, let's try the next group
4520
      stopSubTablesTableMergeScan(pInfo);
1,767,117✔
4521
      if (pInfo->tableEndIndex >= tableListSize - 1) {
1,767,755✔
4522
        setOperatorCompleted(pOperator);
1,713,089✔
4523
        break;
1,712,084✔
4524
      }
4525

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

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

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

4539
_end:
2,805,484✔
4540
  if (code != TSDB_CODE_SUCCESS) {
2,805,492✔
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,805,492✔
4546
  }
4547

4548
  return code;
2,805,157✔
4549
}
4550

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

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

4572
_end:
3,302,273✔
4573
  if (code != TSDB_CODE_SUCCESS) {
3,302,273✔
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,298,689✔
4579

4580
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
169,108,915✔
4581
  STableMergeScanInfo* pInfo = pOperator->info;
169,108,915✔
4582
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
169,119,510✔
4583
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
169,121,825✔
4584
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
169,104,977✔
4585
  int32_t              code = 0;
169,100,854✔
4586
  bool                 hasNext = false;
169,100,854✔
4587
  STsdbReader*         reader = pInfo->base.dataReader;
169,111,764✔
4588

4589
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
169,111,545✔
4590
  if (code != 0) {
169,071,903✔
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)) {
169,071,903✔
4598
    if (isTaskKilled(pTaskInfo)) {
29,546,329✔
4599
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
822✔
4600
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
822✔
4601
    }
4602
    *pFinished = true;
29,526,642✔
4603
    return code;
29,524,815✔
4604
  }
4605

4606
  uint32_t status = 0;
139,553,376✔
4607
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
139,549,332✔
4608

4609
  if (code != TSDB_CODE_SUCCESS) {
139,528,101✔
4610
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
84,103✔
4611
    pTaskInfo->code = code;
84,103✔
4612
    return code;
84,103✔
4613
  }
4614

4615
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
139,443,998✔
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) {
139,443,998✔
4622
    *pSkipped = true;
7,010,126✔
4623
    return code;
7,008,158✔
4624
  }
4625

4626
  return code;
132,423,136✔
4627
}
4628

4629
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
168,703,822✔
4630
  STableMergeScanSortSourceParam* source = param;
168,703,822✔
4631

4632
  SOperatorInfo*       pOperator = source->pOperator;
168,703,822✔
4633
  STableMergeScanInfo* pInfo = pOperator->info;
168,711,689✔
4634
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
168,716,704✔
4635
  SSDataBlock*         pBlock = NULL;
168,712,141✔
4636
  int64_t              st = taosGetTimestampUs();
168,718,144✔
4637
  int32_t              code = TSDB_CODE_SUCCESS;
168,718,144✔
4638

4639
  while (true) {
4640
    if (pInfo->rtnNextDurationBlocks) {
181,421,206✔
4641
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
12,302,551✔
4642
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
4643

4644
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
12,309,137✔
4645
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
5,695,928✔
4646
        ++pInfo->nextDurationBlocksIdx;
5,695,928✔
4647
      } else {
4648
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
12,307,130✔
4649
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
5,694,904✔
4650
          pInfo->nextDurationBlocks[i] = NULL;
5,694,904✔
4651
        }
4652

4653
        pInfo->rtnNextDurationBlocks = false;
6,612,226✔
4654
        pInfo->nextDurationBlocksIdx = 0;
6,612,226✔
4655
        pInfo->numNextDurationBlocks = 0;
6,612,226✔
4656
        continue;
6,612,226✔
4657
      }
4658
    } else {
4659
      bool bFinished = false;
169,111,769✔
4660
      bool bSkipped = false;
169,107,151✔
4661

4662
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
169,124,223✔
4663
      if (code != 0) {
169,051,313✔
4664
        return code;
84,103✔
4665
      }
4666

4667
      pBlock = pInfo->pReaderBlock;
168,967,210✔
4668
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
169,006,659✔
4669
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
4670
      if (bFinished) {
168,930,422✔
4671
        pInfo->bNewFilesetEvent = false;
29,527,848✔
4672
        break;
29,528,320✔
4673
      }
4674

4675
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
139,402,574✔
4676
        if (!bSkipped) {
6,617,265✔
4677
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
5,695,593✔
4678
          if (code) {
5,695,539✔
4679
            *ppBlock = NULL;
×
4680
            return code;
×
4681
          }
4682

4683
          ++pInfo->numNextDurationBlocks;
5,695,539✔
4684
          if (pInfo->numNextDurationBlocks > 2) {
5,693,992✔
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) {
6,615,664✔
4693
          pInfo->rtnNextDurationBlocks = true;
6,612,092✔
4694
          *ppBlock = NULL;
6,612,092✔
4695
          return code;
6,612,092✔
4696
        }
4697

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

4706
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
132,468,565✔
4707

4708
    pOperator->resultInfo.totalRows += pBlock->info.rows;
132,508,245✔
4709
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
132,473,930✔
4710
    *ppBlock = pBlock;
132,462,937✔
4711

4712
    return code;
132,452,374✔
4713
  }
4714

4715
  *ppBlock = NULL;
29,527,848✔
4716
  return code;
29,526,615✔
4717
}
4718

4719
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
26,472,728✔
4720
  int32_t code = TSDB_CODE_SUCCESS;
26,472,728✔
4721
  int32_t lino = 0;
26,472,728✔
4722
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
26,472,728✔
4723
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
26,427,894✔
4724
  SBlockOrderInfo biTs = {0};
26,427,894✔
4725
  SBlockOrderInfo biPk = {0};
26,428,305✔
4726

4727
  int32_t tsTargetSlotId = 0;
26,436,136✔
4728
  int32_t pkTargetSlotId = -1;
26,436,136✔
4729
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
96,756,607✔
4730
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
70,383,419✔
4731
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
70,386,870✔
4732
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
70,386,870✔
4733
      tsTargetSlotId = colInfo->dstSlotId;
26,483,406✔
4734
      biTs.order = order;
26,459,678✔
4735
      biTs.slotId = tsTargetSlotId;
26,459,678✔
4736
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
26,459,678✔
4737
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
26,459,678✔
4738
    }
4739
    // TODO: order by just ts
4740
    if (colInfo->isPk) {
70,277,168✔
4741
      pkTargetSlotId = colInfo->dstSlotId;
1,360,958✔
4742
      biPk.order = order;
1,360,958✔
4743
      biPk.slotId = pkTargetSlotId;
1,360,958✔
4744
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
1,360,958✔
4745
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
1,360,958✔
4746
    }
4747
  }
4748

4749
  void* tmp = taosArrayPush(pSortInfo, &biTs);
26,485,226✔
4750
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
26,485,226✔
4751
  if (pkTargetSlotId != -1) {
26,485,226✔
4752
    tmp = taosArrayPush(pSortInfo, &biPk);
1,362,751✔
4753
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,362,751✔
4754
  }
4755

4756
  (*ppSortArray) = pSortInfo;
26,485,805✔
4757

4758
_end:
26,434,563✔
4759
  if (code != TSDB_CODE_SUCCESS) {
26,435,903✔
4760
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4761
  }
4762
  return code;
26,435,903✔
4763
}
4764

4765
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
19,979,290✔
4766
  STableMergeScanInfo* pTmsInfo = param;
19,979,290✔
4767
  if (type == TSD_READER_NOTIFY_DURATION_START) {
19,979,290✔
4768
    pTmsInfo->bNewFilesetEvent = true;
19,981,874✔
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);
19,977,849✔
4773

4774
  return;
19,995,027✔
4775
}
4776

4777
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
36,382,617✔
4778
  STableMergeScanInfo* pInfo = pOperator->info;
36,382,617✔
4779
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
36,393,846✔
4780

4781
  int32_t code = TSDB_CODE_SUCCESS;
36,418,411✔
4782
  int32_t lino = 0;
36,418,411✔
4783
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
36,418,411✔
4784

4785
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
36,392,698✔
4786
  pInfo->bNewFilesetEvent = false;
36,412,003✔
4787
  pInfo->bNextDurationBlockEvent = false;
36,391,373✔
4788

4789
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
36,365,655✔
4790
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
36,388,318✔
4791

4792
  pInfo->pSortHandle = NULL;
36,332,174✔
4793
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
72,754,530✔
4794
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
36,332,434✔
4795
  if (code) {
36,347,806✔
4796
    return code;
×
4797
  }
4798

4799
  if (pInfo->bSortRowId && numOfTable != 1) {
36,347,806✔
4800
    int32_t memSize = 512 * 1024 * 1024;
3,070,856✔
4801
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
3,070,856✔
4802
    QUERY_CHECK_CODE(code, lino, _end);
3,072,593✔
4803
  }
4804

4805
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
36,335,472✔
4806
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
36,337,761✔
4807
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
36,373,236✔
4808

4809
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
36,284,563✔
4810
  QUERY_CHECK_CODE(code, lino, _end);
36,305,134✔
4811

4812
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
36,305,134✔
4813
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
36,296,672✔
4814
  param->pOperator = pOperator;
36,296,672✔
4815

4816
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
36,299,488✔
4817
  if (ps == NULL) {
36,242,009✔
4818
    taosMemoryFree(param);
×
4819
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
4820
  }
4821

4822
  ps->param = param;
36,242,009✔
4823
  ps->onlyRef = false;
36,271,199✔
4824
  code = tsortAddSource(pInfo->pSortHandle, ps);
36,334,635✔
4825
  QUERY_CHECK_CODE(code, lino, _end);
36,415,652✔
4826

4827
  if (numOfTable == 1) {
36,415,652✔
4828
    tsortSetSingleTableMerge(pInfo->pSortHandle);
24,650,063✔
4829
  } else {
4830
    code = tsortOpen(pInfo->pSortHandle);
11,765,589✔
4831
    QUERY_CHECK_CODE(code, lino, _end);
11,776,629✔
4832
  }
4833

4834
_end:
11,734,783✔
4835
  if (code != TSDB_CODE_SUCCESS) {
36,413,326✔
4836
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
41,846✔
4837
  }
4838
  return code;
36,412,615✔
4839
}
4840

4841
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
36,355,414✔
4842
  STableMergeScanInfo* pInfo = pOperator->info;
36,355,414✔
4843
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
36,353,611✔
4844
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
36,355,764✔
4845

4846
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
36,355,764✔
4847
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
36,353,611✔
4848
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
36,354,531✔
4849
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
36,351,384✔
4850
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
36,345,276✔
4851
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
36,350,482✔
4852

4853
  tsortDestroySortHandle(pInfo->pSortHandle);
36,340,283✔
4854
  pInfo->pSortHandle = NULL;
36,350,482✔
4855
}
36,350,893✔
4856

4857
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
29,767,297✔
4858
  int32_t              code = TSDB_CODE_SUCCESS;
29,767,297✔
4859
  int32_t              lino = 0;
29,767,297✔
4860
  STableMergeScanInfo* pInfo = pOperator->info;
29,767,297✔
4861
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
29,742,628✔
4862
  SReadHandle*         pHandle = &pInfo->base.readHandle;
29,728,257✔
4863
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
29,750,686✔
4864
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
29,728,752✔
4865

4866
  {
4867
    int32_t numOfTables = 0;
29,730,171✔
4868
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
29,692,575✔
4869
    QUERY_CHECK_CODE(code, lino, _end);
29,729,307✔
4870

4871
    int32_t i = pInfo->tableStartIndex + 1;
29,729,307✔
4872
    for (; i < numOfTables; ++i) {
134,308,114✔
4873
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
110,324,996✔
4874
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
110,325,961✔
4875
      if (tableKeyInfo->groupId != pInfo->groupId) {
110,325,961✔
4876
        break;
5,729,020✔
4877
      }
4878
    }
4879
    pInfo->tableEndIndex = i - 1;
29,712,138✔
4880
  }
4881
  pInfo->bGroupProcessed = false;
29,810,592✔
4882
  int32_t tableStartIdx = pInfo->tableStartIndex;
29,817,181✔
4883
  int32_t tableEndIdx = pInfo->tableEndIndex;
29,812,851✔
4884

4885
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
29,809,524✔
4886
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
29,809,524✔
4887
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
59,563,125✔
4888
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
29,736,647✔
4889
  QUERY_CHECK_CODE(code, lino, _end);
29,799,161✔
4890
  if (pInfo->filesetDelimited) {
29,799,161✔
4891
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
29,810,533✔
4892
  }
4893
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
29,793,657✔
4894

4895
  code = startDurationForGroupTableMergeScan(pOperator);
29,748,118✔
4896
  QUERY_CHECK_CODE(code, lino, _end);
29,799,751✔
4897

4898
_end:
29,757,905✔
4899
  if (code != TSDB_CODE_SUCCESS) {
29,808,562✔
4900
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
41,846✔
4901
    pTaskInfo->code = code;
41,846✔
4902
    T_LONG_JMP(pTaskInfo->env, code);
41,846✔
4903
  }
4904
}
29,766,716✔
4905

4906
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
29,740,459✔
4907
  STableMergeScanInfo* pInfo = pOperator->info;
29,740,459✔
4908
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
29,744,378✔
4909
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
29,740,870✔
4910

4911
  stopDurationForGroupTableMergeScan(pOperator);
29,743,397✔
4912

4913
  if (pInfo->base.dataReader != NULL) {
29,737,643✔
4914
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
29,738,404✔
4915
    pInfo->base.dataReader = NULL;
29,727,741✔
4916
  }
4917
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
29,729,160✔
4918
    if (pInfo->nextDurationBlocks[i]) {
1,024✔
4919
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
1,024✔
4920
      pInfo->nextDurationBlocks[i] = NULL;
1,024✔
4921
    }
4922
    pInfo->numNextDurationBlocks = 0;
1,024✔
4923
    pInfo->nextDurationBlocksIdx = 0;
1,024✔
4924
  }
4925
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
29,731,645✔
4926
  taosHashCleanup(pInfo->mSkipTables);
29,723,005✔
4927
  pInfo->mSkipTables = NULL;
29,727,085✔
4928
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
29,727,387✔
4929
}
29,738,698✔
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,
72,553,035✔
4934
                                              SOperatorInfo* pOperator) {
4935
  STableMergeScanInfo* pInfo = pOperator->info;
72,553,035✔
4936
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
72,564,921✔
4937
  STupleHandle*        pTupleHandle = NULL;
72,560,362✔
4938

4939
  blockDataCleanup(pResBlock);
72,562,219✔
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✔
UNCOV
4946
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
UNCOV
4947
        T_LONG_JMP(pOperator->pTaskInfo->env, code);
×
4948
      }
4949
      if (pTupleHandle == NULL) {
2,147,483,647✔
4950
        break;
60,085,183✔
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,443,799✔
4961
      }
4962
    }
4963

4964
    if (tsortIsClosed(pHandle)) {
72,528,982✔
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);
72,525,171✔
4970
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
72,536,396✔
4971
           pInfo->limitInfo.numOfOutputRows);
4972
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
72,532,509✔
4973
      break;
4974
    }
4975
  }
4976
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
72,532,509✔
4977
}
4978

4979
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
62,913,328✔
4980
  if (pOperator->status == OP_EXEC_DONE) {
62,913,328✔
4981
    (*ppRes) = NULL;
24,737✔
4982
    return TSDB_CODE_SUCCESS;
24,737✔
4983
  }
4984

4985
  int32_t              code = TSDB_CODE_SUCCESS;
62,846,380✔
4986
  int32_t              lino = 0;
62,846,380✔
4987
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
62,846,380✔
4988
  STableMergeScanInfo* pInfo = pOperator->info;
62,852,715✔
4989

4990
  code = pOperator->fpSet._openFn(pOperator);
62,836,316✔
4991
  QUERY_CHECK_CODE(code, lino, _end);
62,832,221✔
4992

4993
  int64_t st = taosGetTimestampUs();
62,832,447✔
4994

4995
  int32_t tableListSize = 0;
62,832,447✔
4996
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
62,829,337✔
4997
  QUERY_CHECK_CODE(code, lino, _end);
62,908,778✔
4998

4999
  if (!pInfo->hasGroupId) {
62,908,778✔
5000
    pInfo->hasGroupId = true;
26,720,515✔
5001

5002
    if (tableListSize == 0) {
26,682,319✔
5003
      setOperatorCompleted(pOperator);
2,667,888✔
5004
      (*ppRes) = NULL;
2,667,888✔
5005
      return code;
2,667,888✔
5006
    }
5007
    pInfo->tableStartIndex = 0;
24,014,431✔
5008
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
24,015,621✔
5009
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
23,992,233✔
5010
    pInfo->groupId = tmp->groupId;
23,992,233✔
5011
    startGroupTableMergeScan(pOperator);
24,047,297✔
5012
  }
5013

5014
  SSDataBlock* pBlock = NULL;
60,221,236✔
5015
  while (pInfo->tableStartIndex < tableListSize) {
72,560,264✔
5016
    if (isTaskKilled(pTaskInfo)) {
72,564,614✔
5017
      goto _end;
822✔
5018
    }
5019

5020
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
72,562,614✔
5021
                                              pOperator);
5022
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
72,533,111✔
5023
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
1,675,099✔
5024
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
1,675,099✔
5025
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
1,675,099✔
5026
    }
5027
    if (pBlock != NULL) {
72,531,835✔
5028
      pBlock->info.id.groupId = pInfo->groupId;
36,172,624✔
5029
      pOperator->resultInfo.totalRows += pBlock->info.rows;
36,172,339✔
5030
      pInfo->bGroupProcessed = true;
36,170,067✔
5031
      break;
36,169,342✔
5032
    } else {
5033
      if (pInfo->bNewFilesetEvent) {
36,359,211✔
5034
        stopDurationForGroupTableMergeScan(pOperator);
6,613,250✔
5035
        code = startDurationForGroupTableMergeScan(pOperator);
6,612,915✔
5036
        QUERY_CHECK_CODE(code, lino, _end);
6,610,008✔
5037
      } else {
5038
        // Data of this group are all dumped, let's try the next group
5039
        stopGroupTableMergeScan(pOperator);
29,746,844✔
5040
        if (pInfo->tableEndIndex >= tableListSize - 1) {
29,735,465✔
5041
          setOperatorCompleted(pOperator);
24,001,541✔
5042
          break;
24,007,929✔
5043
        }
5044

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

5055
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
60,178,369✔
5056

5057
_end:
60,179,708✔
5058
  if (code != TSDB_CODE_SUCCESS) {
60,181,763✔
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;
60,181,763✔
5064
  }
5065

5066
  return code;
60,176,527✔
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) {
26,503,648✔
5076
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
26,503,648✔
5077

5078
  // start one reader variable
5079
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
26,503,648✔
5080
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
26,505,911✔
5081
    pTableScanInfo->base.dataReader = NULL;
26,500,849✔
5082
  }
5083

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

5091
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
26,504,531✔
5092
  pTableScanInfo->pSortHandle = NULL;
26,492,675✔
5093
  taosHashCleanup(pTableScanInfo->mSkipTables);
26,492,264✔
5094
  pTableScanInfo->mSkipTables = NULL;
26,490,900✔
5095
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
26,493,509✔
5096
  pTableScanInfo->pSortInputBlock = NULL;
26,507,949✔
5097
  // end one reader variable
5098

5099
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
26,507,538✔
5100

5101
  blockDataDestroy(pTableScanInfo->pResBlock);
26,512,278✔
5102
  pTableScanInfo->pResBlock = NULL;
26,513,578✔
5103

5104
  // remove it from the task->result list
5105
  blockDataDestroy(pTableScanInfo->pReaderBlock);
26,514,789✔
5106
  pTableScanInfo->pReaderBlock = NULL;
26,512,497✔
5107
  taosArrayDestroy(pTableScanInfo->pSortInfo);
26,512,086✔
5108

5109
  stopSubTablesTableMergeScan(pTableScanInfo);
26,510,219✔
5110

5111
  taosMemoryFreeClear(param);
26,510,214✔
5112
}
26,504,970✔
5113

5114
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
1,094,982✔
5115
  if (pOptr == NULL) {
1,094,982✔
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,094,982✔
5121
  if (!execInfo) {
1,095,490✔
5122
    return terrno;
×
5123
  }
5124
  STableMergeScanInfo* pInfo = pOptr->info;
1,095,490✔
5125
  execInfo->blockRecorder = pInfo->base.readRecorder;
1,095,490✔
5126
  execInfo->sortExecInfo = pInfo->sortExecInfo;
1,095,490✔
5127

5128
  *pOptrExplain = execInfo;
1,095,490✔
5129
  *len = sizeof(STableMergeScanExecInfo);
1,094,982✔
5130

5131
  return TSDB_CODE_SUCCESS;
1,095,490✔
5132
}
5133

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

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

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

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

5160
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
2,015,478✔
5161
  code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
4,030,956✔
5162
                                  &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
5163
                                  pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
4,030,956✔
5164
  if (code) {
2,015,478✔
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,015,478✔
5170
  cleanupQueryTableDataCond(&pInfo->base.cond);
2,015,478✔
5171
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle, false);
2,015,478✔
5172
  if (code) {
2,015,257✔
5173
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
5174
    return code;
×
5175
  }
5176

5177
  tsortDestroySortHandle(pInfo->pSortHandle);
2,015,257✔
5178
  pInfo->pSortHandle = NULL;
2,015,478✔
5179

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

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

5199
  stopSubTablesTableMergeScan(pInfo);
2,015,478✔
5200
  return code;
2,015,257✔
5201
}
5202

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

5208
  int32_t              code = TSDB_CODE_SUCCESS;
26,435,080✔
5209
  int32_t              lino = 0;
26,435,080✔
5210
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
26,435,080✔
5211
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
26,345,806✔
5212
  if (pInfo == NULL || pOperator == NULL) {
26,335,981✔
5213
    code = terrno;
61✔
5214
    goto _error;
×
5215
  }
5216

5217
  pOperator->pPhyNode = pTableScanNode;
26,336,415✔
5218
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
26,338,279✔
5219

5220
  int32_t numOfCols = 0;
26,365,004✔
5221
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
26,389,415✔
5222
                             &pInfo->base.matchInfo);
5223
  QUERY_CHECK_CODE(code, lino, _error);
26,433,101✔
5224

5225
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, false);
26,433,101✔
5226
  QUERY_CHECK_CODE(code, lino, _error);
26,334,294✔
5227

5228
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
26,334,294✔
5229
    SExprSupp* pSup = &pInfo->base.pseudoSup;
9,885,903✔
5230
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
9,877,825✔
5231
    QUERY_CHECK_CODE(code, lino, _error);
9,890,554✔
5232

5233
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
9,890,554✔
5234
                                      &pTaskInfo->storageAPI.functionStore);
5235
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
9,873,674✔
5236
  }
5237

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

5240
  if (readHandle->streamRtInfo == NULL) {
26,337,011✔
5241
    pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
26,352,441✔
5242
    QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
26,472,533✔
5243
  }
5244
  
5245
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
26,430,995✔
5246
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
26,476,963✔
5247
  pInfo->base.scanFlag = MAIN_SCAN;
26,406,169✔
5248
  pInfo->base.readHandle = *readHandle;
26,420,984✔
5249

5250
  pInfo->readIdx = -1;
26,380,209✔
5251

5252
  pInfo->base.limitInfo.limit.limit = -1;
26,413,351✔
5253
  pInfo->base.limitInfo.slimit.limit = -1;
26,395,711✔
5254
  pInfo->base.pTableListInfo = pTableListInfo;
26,452,666✔
5255

5256
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
26,426,008✔
5257
  pInfo->sample.seed = taosGetTimestampSec();
26,447,845✔
5258

5259
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
26,456,447✔
5260
                            pTaskInfo->pStreamRuntimeInfo);
26,457,635✔
5261
  QUERY_CHECK_CODE(code, lino, _error);
26,432,557✔
5262

5263
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
26,432,557✔
5264

5265
  pInfo->mergeLimit = -1;
26,403,986✔
5266
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
26,412,281✔
5267
  if (hasLimit) {
26,424,375✔
5268
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1
3,937,423✔
5269
                            ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset
891,537✔
5270
                            : pInfo->limitInfo.limit.limit;
2,859,153✔
5271
    pInfo->mSkipTables = NULL;
1,967,143✔
5272
  }
5273

5274
  initResultSizeInfo(&pOperator->resultInfo, 1024);
26,422,103✔
5275
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
26,401,107✔
5276
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
26,479,363✔
5277
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
26,409,043✔
5278
  QUERY_CHECK_CODE(code, lino, _error);
26,481,521✔
5279
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
26,481,521✔
5280
    pInfo->bSortRowId = true;
4,922,502✔
5281
  } else {
5282
    pInfo->bSortRowId = false;
21,565,081✔
5283
  }
5284

5285
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
26,467,475✔
5286
  QUERY_CHECK_CODE(code, lino, _error);
26,456,968✔
5287

5288
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
26,456,968✔
5289
  QUERY_CHECK_CODE(code, lino, _error);
26,429,885✔
5290

5291
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
26,429,885✔
5292
  QUERY_CHECK_CODE(code, lino, _error);
26,412,565✔
5293

5294
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
26,412,565✔
5295

5296
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
26,444,327✔
5297
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
26,430,940✔
5298

5299
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
26,393,790✔
5300

5301
  // start one reader variable
5302
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
26,482,086✔
5303
  QUERY_CHECK_CODE(code, lino, _error);
26,446,680✔
5304

5305
  if (!tsExperimental) {
26,446,680✔
5306
    pInfo->filesetDelimited = false;
×
5307
  } else {
5308
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
26,446,680✔
5309
  }
5310
  // end one reader variable
5311
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
26,461,950✔
5312
                  pInfo, pTaskInfo);
5313
  pOperator->exprSupp.numOfExprs = numOfCols;
26,421,980✔
5314

5315
  pOperator->fpSet = createOperatorFpSet(
26,393,989✔
5316
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
26,448,591✔
5317
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
5318
      NULL);
5319
  setOperatorResetStateFn(pOperator, resetTableMergeScanOperatorState);     
26,406,437✔
5320
  pOperator->cost.openCost = 0;
26,422,244✔
5321

5322
  *pOptrInfo = pOperator;
26,421,228✔
5323
  return code;
26,372,386✔
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) {
23,342✔
5363
  if (scanCols != NULL) {
23,342✔
5364
    SNode* pNode = NULL;
13,460✔
5365
    FOREACH(pNode, scanCols) {
31,404✔
5366
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
17,944✔
5367
        return TSDB_CODE_QRY_SYS_ERROR;
×
5368
      }
5369
      STargetNode* targetNode = (STargetNode*)pNode;
17,944✔
5370
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
17,944✔
5371
        return TSDB_CODE_QRY_SYS_ERROR;
×
5372
      }
5373
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
17,944✔
5374
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
17,944✔
5375
        supp->dbNameSlotId = targetNode->slotId;
8,811✔
5376
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
9,133✔
5377
        supp->stbNameSlotId = targetNode->slotId;
9,133✔
5378
      }
5379
    }
5380
  }
5381
  return TSDB_CODE_SUCCESS;
23,342✔
5382
}
5383

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

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

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

5435
  supp->dbNameSlotId = -1;
23,342✔
5436
  supp->stbNameSlotId = -1;
23,342✔
5437
  supp->tbCountSlotId = -1;
23,342✔
5438

5439
  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
23,342✔
5440
  if (code != TSDB_CODE_SUCCESS) {
23,342✔
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);
23,342✔
5446
  if (code != TSDB_CODE_SUCCESS) {
23,342✔
5447
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
×
5448
    return code;
×
5449
  }
5450
  return code;
23,342✔
5451
}
5452

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

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

5467
  pInfo->readHandle = *readHandle;
23,342✔
5468

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

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

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

5482
  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
23,342✔
5483
                  pInfo, pTaskInfo);
5484
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScanNext, NULL, destoryTableCountScanOperator,
23,342✔
5485
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
5486
  *pOptrInfo = pOperator;
23,342✔
5487
  return code;
23,342✔
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,
32,564✔
5502
                                    SSDataBlock* pRes) {
5503
  int32_t code = TSDB_CODE_SUCCESS;
32,564✔
5504
  int32_t lino = 0;
32,564✔
5505
  if (pSupp->dbNameSlotId != -1) {
32,564✔
5506
    QUERY_CHECK_CONDITION((strlen(dbName) > 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
13,280✔
5507
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
13,280✔
5508
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
13,280✔
5509

5510
    char varDbName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
13,280✔
5511
    tstrncpy(varDataVal(varDbName), dbName, TSDB_DB_NAME_LEN);
13,280✔
5512

5513
    varDataSetLen(varDbName, strlen(dbName));
13,280✔
5514
    code = colDataSetVal(colInfoData, 0, varDbName, false);
13,280✔
5515
    QUERY_CHECK_CODE(code, lino, _end);
13,280✔
5516
  }
5517

5518
  if (pSupp->stbNameSlotId != -1) {
32,564✔
5519
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
17,468✔
5520
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
17,468✔
5521
    if (strlen(stbName) != 0) {
17,468✔
5522
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
12,508✔
5523
      tstrncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
12,508✔
5524
      varDataSetLen(varStbName, strlen(stbName));
12,508✔
5525
      code = colDataSetVal(colInfoData, 0, varStbName, false);
12,347✔
5526
      QUERY_CHECK_CODE(code, lino, _end);
12,508✔
5527
    } else {
5528
      colDataSetNULL(colInfoData, 0);
5529
    }
5530
  }
5531

5532
  if (pSupp->tbCountSlotId != -1) {
32,564✔
5533
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
32,564✔
5534
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
32,564✔
5535
    code = colDataSetVal(colInfoData, 0, (char*)&count, false);
32,564✔
5536
    QUERY_CHECK_CODE(code, lino, _end);
32,564✔
5537
  }
5538
  pRes->info.rows = 1;
32,564✔
5539

5540
_end:
32,564✔
5541
  if (code != TSDB_CODE_SUCCESS) {
32,564✔
5542
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5543
  }
5544
  return code;
32,564✔
5545
}
5546

5547
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
9,994✔
5548
  STableCountScanSupp* pSupp = &pInfo->supp;
9,994✔
5549
  SSDataBlock*         pRes = pInfo->pRes;
9,994✔
5550

5551
  size_t infodbTableNum;
9,994✔
5552
  getInfosDbMeta(NULL, &infodbTableNum);
9,994✔
5553
  infodbTableNum -= 1;
9,994✔
5554
  size_t perfdbTableNum;
9,994✔
5555
  getPerfDbMeta(NULL, &perfdbTableNum);
9,994✔
5556

5557
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
9,994✔
5558
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
8,562✔
5559
    return (pRes->info.rows > 0) ? pRes : NULL;
8,562✔
5560
  } else {
5561
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
1,432✔
5562
    return (pRes->info.rows > 0) ? pRes : NULL;
1,432✔
5563
  }
5564
}
5565

5566
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator,
1,432✔
5567
                                       STableCountScanSupp* pSupp,
5568
                                       SSDataBlock* pRes,
5569
                                       size_t infodbTableNum,
5570
                                       size_t perfdbTableNum) {
5571
  int32_t        code = TSDB_CODE_SUCCESS;
1,432✔
5572
  int32_t        lino = 0;
1,432✔
5573
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
1,432✔
5574
  if (strlen(pSupp->stbNameFilter) != 0) {
1,432✔
5575
    // sys dbs do not contain any stable, so the count is always 0
5576
    code = fillTableCountScanDataBlock(pSupp, "", "", 0, pRes);
489✔
5577
    QUERY_CHECK_CODE(code, lino, _end);
489✔
5578
  } else if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
943✔
5579
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB,
163✔
5580
                                       "", infodbTableNum, pRes);
5581
    QUERY_CHECK_CODE(code, lino, _end);
163✔
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, "", "",
617✔
5588
                                       infodbTableNum + perfdbTableNum, pRes);
617✔
5589
    QUERY_CHECK_CODE(code, lino, _end);
617✔
5590
  } else {
5591
    // other dbs do not exist in sys dbs, so the count is always 0
5592
    code = fillTableCountScanDataBlock(pSupp, pSupp->dbNameFilter,
163✔
5593
                                       pSupp->stbNameFilter, 0, pRes);
163✔
5594
    QUERY_CHECK_CODE(code, lino, _end);
163✔
5595
  }
5596

5597
_end:
163✔
5598
  if (code != TSDB_CODE_SUCCESS) {
1,432✔
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,432✔
5604
}
1,432✔
5605

5606
static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
8,562✔
5607
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
5608
                                        size_t perfdbTableNum) {
5609
  int32_t        code = TSDB_CODE_SUCCESS;
8,562✔
5610
  int32_t        lino = 0;
8,562✔
5611
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
8,562✔
5612
  if (pInfo->currGrpIdx == 0) {
8,562✔
5613
    uint64_t groupId = 0;
2,854✔
5614
    if (pSupp->groupByDbName) {
2,854✔
5615
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
1,935✔
5616
    } else {
5617
      groupId = calcGroupId("", 0);
919✔
5618
    }
5619

5620
    pRes->info.id.groupId = groupId;
2,854✔
5621
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
2,854✔
5622
    QUERY_CHECK_CODE(code, lino, _end);
2,854✔
5623
  } else if (pInfo->currGrpIdx == 1) {
5,708✔
5624
    uint64_t groupId = 0;
2,854✔
5625
    if (pSupp->groupByDbName) {
2,854✔
5626
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
1,935✔
5627
    } else {
5628
      groupId = calcGroupId("", 0);
919✔
5629
    }
5630

5631
    pRes->info.id.groupId = groupId;
2,854✔
5632
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
2,854✔
5633
    QUERY_CHECK_CODE(code, lino, _end);
2,854✔
5634
  } else {
5635
    setOperatorCompleted(pOperator);
2,854✔
5636
  }
5637
  pInfo->currGrpIdx++;
8,562✔
5638

5639
_end:
8,562✔
5640
  if (code != TSDB_CODE_SUCCESS) {
8,562✔
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,562✔
5646

5647
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
55,906✔
5648
  int32_t                      code = TSDB_CODE_SUCCESS;
55,906✔
5649
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
55,906✔
5650
  STableCountScanOperatorInfo* pInfo = pOperator->info;
55,906✔
5651
  STableCountScanSupp*         pSupp = &pInfo->supp;
55,906✔
5652
  SSDataBlock*                 pRes = pInfo->pRes;
55,906✔
5653

5654
  blockDataCleanup(pRes);
55,906✔
5655
  QRY_PARAM_CHECK(ppRes);
55,906✔
5656

5657
  if (pOperator->status == OP_EXEC_DONE) {
55,906✔
5658
    return code;
13,322✔
5659
  }
5660

5661
  if (pInfo->readHandle.mnd != NULL) {
42,584✔
5662
    (*ppRes) = buildSysDbTableCount(pOperator, pInfo);
9,994✔
5663
    return code;
9,994✔
5664
  }
5665

5666
  code = buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
32,590✔
5667
  if (code != TSDB_CODE_SUCCESS) {
32,590✔
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) {
32,590✔
5673
    *ppRes = pRes;
25,424✔
5674
  }
5675

5676
  return code;
32,590✔
5677
}
5678

5679
static int32_t buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
32,590✔
5680
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes) {
5681
  int32_t        code = TSDB_CODE_SUCCESS;
32,590✔
5682
  int32_t        lino = 0;
32,590✔
5683
  const char*    db = NULL;
32,590✔
5684
  int32_t        vgId = 0;
32,590✔
5685
  char           dbName[TSDB_DB_NAME_LEN] = {0};
32,590✔
5686
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
32,590✔
5687
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
32,590✔
5688

5689
  // get dbname
5690
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, &db, &vgId, NULL, NULL);
32,590✔
5691
  SName sn = {0};
32,590✔
5692

5693
  code = tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
32,590✔
5694
  QUERY_CHECK_CODE(code, lino, _end);
32,590✔
5695

5696
  code = tNameGetDbName(&sn, dbName);
32,590✔
5697
  QUERY_CHECK_CODE(code, lino, _end);
32,590✔
5698

5699
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
32,590✔
5700
    code = buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
24,140✔
5701
  } else {
5702
    code = buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
8,450✔
5703
  }
5704

5705
_end:
32,590✔
5706
  if (code != TSDB_CODE_SUCCESS) {
32,590✔
5707
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5708
    pTaskInfo->code = code;
×
5709
  }
5710

5711
  return code;
32,590✔
5712
}
5713

5714
static int32_t buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
24,140✔
5715
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
5716
  int32_t        code = TSDB_CODE_SUCCESS;
24,140✔
5717
  int32_t        lino = 0;
24,140✔
5718
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
24,140✔
5719
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
24,140✔
5720

5721
  if (pSupp->groupByStbName) {
24,140✔
5722
    if (pInfo->stbUidList == NULL) {
20,700✔
5723
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
7,166✔
5724
      QUERY_CHECK_NULL(pInfo->stbUidList, code, lino, _end, terrno);
7,166✔
5725
      code = pAPI->metaFn.storeGetTableList(pInfo->readHandle.vnode, TSDB_SUPER_TABLE, pInfo->stbUidList);
7,166✔
5726
      QUERY_CHECK_CODE(code, lino, _end);
7,166✔
5727
    }
5728

5729
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
20,700✔
5730
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
12,508✔
5731
      code = buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid, pAPI);
12,508✔
5732
      QUERY_CHECK_CODE(code, lino, _end);
12,508✔
5733

5734
      pInfo->currGrpIdx++;
12,508✔
5735
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
8,192✔
5736
      code = buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName, pAPI);
7,166✔
5737
      QUERY_CHECK_CODE(code, lino, _end);
7,166✔
5738

5739
      pInfo->currGrpIdx++;
7,166✔
5740
    } else {
5741
      setOperatorCompleted(pOperator);
1,026✔
5742
    }
5743
  } else {
5744
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
3,440✔
5745
    pRes->info.id.groupId = groupId;
3,440✔
5746

5747
    int64_t dbTableCount = 0;
3,440✔
5748
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &dbTableCount, NULL);
3,440✔
5749
    code = fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
3,440✔
5750
    QUERY_CHECK_CODE(code, lino, _end);
3,440✔
5751
    setOperatorCompleted(pOperator);
3,440✔
5752
  }
5753

5754
_end:
24,140✔
5755
  if (code != TSDB_CODE_SUCCESS) {
24,140✔
5756
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5757
    pTaskInfo->code = code;
×
5758
  }
5759
  return code;
24,140✔
5760
}
5761

5762
static int32_t buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
8,450✔
5763
                                         STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
5764
  int32_t        code = TSDB_CODE_SUCCESS;
8,450✔
5765
  int32_t        lino = 0;
8,450✔
5766
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
8,450✔
5767
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
8,450✔
5768

5769
  if (strlen(pSupp->dbNameFilter) != 0 &&
8,450✔
5770
      strcmp(dbName, pSupp->dbNameFilter) != 0) {
5,586✔
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) {
8,450✔
5776
    uint64_t uid = 0;
2,864✔
5777
    code = pAPI->metaFn.getTableUidByName(pInfo->readHandle.vnode,
5,728✔
5778
                                          pSupp->stbNameFilter, &uid);
2,864✔
5779
    if (code == TSDB_CODE_TDB_TABLE_NOT_EXIST) {
2,864✔
5780
      // table not exist, return count 0
5781
      code = fillTableCountScanDataBlock(pSupp, dbName,
1,304✔
5782
                                         pSupp->stbNameFilter, 0, pRes);
1,304✔
5783
      QUERY_CHECK_CODE(code, lino, _end);
1,304✔
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;
5,586✔
5798
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode,
5,586✔
5799
                              NULL, NULL, &tbNumVnode, NULL);
5800
    code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
5,586✔
5801
    QUERY_CHECK_CODE(code, lino, _end);
5,586✔
5802
  }
5803

5804
_end:
8,450✔
5805
  if (code != TSDB_CODE_SUCCESS) {
8,450✔
5806
    pTaskInfo->code = code;
×
5807
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5808
  }
5809

5810
  setOperatorCompleted(pOperator);
8,450✔
5811
  return code;
8,450✔
5812
}
5813

5814
static int32_t buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
7,166✔
5815
                                              SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI) {
5816
  int32_t code = TSDB_CODE_SUCCESS;
7,166✔
5817
  int32_t lino = 0;
7,166✔
5818
  char    fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
7,166✔
5819
  if (pSupp->groupByDbName) {
7,166✔
5820
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
3,436✔
5821
  }
5822

5823
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
7,166✔
5824
  pRes->info.id.groupId = groupId;
7,166✔
5825

5826
  int64_t numOfTables = 0;
7,166✔
5827
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, NULL, &numOfTables);
7,166✔
5828

5829
  if (numOfTables != 0) {
7,166✔
5830
    code = fillTableCountScanDataBlock(pSupp, dbName, "", numOfTables, pRes);
1,026✔
5831
  }
5832

5833
_end:
6,140✔
5834
  if (code != TSDB_CODE_SUCCESS) {
7,166✔
5835
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5836
  }
5837
  return code;
7,166✔
5838
}
5839

5840
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
12,508✔
5841
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
5842
  int32_t code = TSDB_CODE_SUCCESS;
12,508✔
5843
  int32_t lino = 0;
12,508✔
5844
  char    stbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
12,508✔
5845
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
12,508✔
5846
  QUERY_CHECK_CODE(code, lino, _end);
12,508✔
5847

5848
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
12,508✔
5849
  if (pSupp->groupByDbName) {
12,508✔
5850
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
5,700✔
5851
  } else {
5852
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
6,808✔
5853
  }
5854

5855
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
12,508✔
5856
  pRes->info.id.groupId = groupId;
12,508✔
5857

5858
  int64_t ctbNum = 0;
12,508✔
5859
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL, NULL);
12,508✔
5860
  QUERY_CHECK_CODE(code, lino, _end);
12,508✔
5861
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
12,508✔
5862
  QUERY_CHECK_CODE(code, lino, _end);
12,508✔
5863

5864
_end:
12,508✔
5865
  if (code != TSDB_CODE_SUCCESS) {
12,508✔
5866
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5867
  }
5868
  return code;
12,508✔
5869
}
5870

5871
static void destoryTableCountScanOperator(void* param) {
23,342✔
5872
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
23,342✔
5873
  blockDataDestroy(pTableCountScanInfo->pRes);
23,342✔
5874

5875
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
23,315✔
5876
  taosMemoryFreeClear(param);
23,342✔
5877
}
23,315✔
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