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

taosdata / TDengine / #4830

31 Oct 2025 03:37AM UTC coverage: 58.792% (+9.1%) from 49.734%
#4830

push

travis-ci

SallyHuo-TAOS
Merge remote-tracking branch 'origin/cover/3.0' into cover/3.0

# Conflicts:
#	test/ci/run.sh

149913 of 324176 branches covered (46.24%)

Branch coverage included in aggregate %.

199119 of 269498 relevant lines covered (73.89%)

239518383.75 hits per line

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

56.29
/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) {
2,147,483,647✔
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;
2,147,483,647✔
86
#endif
87
}
88

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

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

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

105
  if (order == TSDB_ORDER_ASC) {
303,700,002✔
106
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
290,275,678✔
107
    if (w.ekey < pBlockInfo->window.skey) {
290,274,477!
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) {
290,225,253✔
113
      *overlap = true;
252,912,906✔
114
      return code;
252,910,999✔
115
    }
116

117
    while (1) {
118
      getNextTimeWindow(pInterval, &w, order);
37,325,482✔
119
      if (w.skey > pBlockInfo->window.ekey) {
37,351,639✔
120
        break;
36,506,993✔
121
      }
122

123
      if (w.ekey <= pBlockInfo->window.ekey) {
829,401!
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) {
829,401!
128
        *overlap = true;
829,401✔
129
        return code;
829,401✔
130
      }
131
    }
132
  } else {
133
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.ekey);
13,424,324✔
134
    if (w.skey > pBlockInfo->window.ekey) {
13,426,237!
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) {
13,428,622✔
140
      *overlap = true;
12,606,593✔
141
      return code;
12,606,012✔
142
    }
143

144
    while (1) {
145
      getNextTimeWindow(pInterval, &w, order);
820,931✔
146
      if (w.ekey < pBlockInfo->window.skey) {
822,975✔
147
        break;
809,289✔
148
      }
149

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

161
  *overlap = false;
37,316,282✔
162
  return code;
37,308,924✔
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) {
507,988,037✔
167
  if (pOperator->operatorType != QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
507,988,037!
168
    return NULL;
×
169
  }
170

171
  int64_t buf[2] = {0};
507,999,495✔
172
  SET_RES_WINDOW_KEY((char*)buf, &groupId, sizeof(groupId), groupId);
507,967,481✔
173

174
  STableScanInfo* pTableScanInfo = pOperator->info;
507,918,119✔
175

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

179
  if (p1 == NULL) {
508,039,561✔
180
    return NULL;
193,055,210✔
181
  }
182

183
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
314,984,351✔
184
  if (NULL == *pPage) {
314,950,022!
185
    return NULL;
×
186
  }
187

188
  return (SResultRow*)((char*)(*pPage) + p1->offset);
314,951,808✔
189
}
190

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

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

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

214
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
2,147,483,647✔
215
    return TSDB_CODE_SUCCESS;
2,147,483,647✔
216
  }
217

218
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
507,990,431✔
219

220
  SFilePage*  pPage = NULL;
508,001,045✔
221
  SResultRow* pRow = getTableGroupOutputBuf(pOperator, pBlockInfo->id.groupId, &pPage);
507,994,817✔
222

223
  if (pRow == NULL) {
507,966,073✔
224
    return TSDB_CODE_SUCCESS;
193,020,042✔
225
  }
226

227
  bool notLoadBlock = true;
314,946,031✔
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) {
2,147,483,647✔
235
      notLoadBlock = false;
285,836,760✔
236
      break;
285,836,760✔
237
    }
238
  }
239

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

243
  if (notLoadBlock) {
314,943,637✔
244
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
29,097,046✔
245
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
29,097,046✔
246
  }
247

248
  return code;
314,913,920✔
249
}
250

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

258
  return filterRangeExecute(pFilterInfo, pColsAgg, numOfCols, numOfRows, keep);
43,187,270✔
259
}
260

261
static int32_t doLoadBlockSMA(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
662,046,564✔
262
                              bool* pLoad) {
263
  SStorageAPI* pAPI = &pTaskInfo->storageAPI;
662,046,564✔
264
  bool         allColumnsHaveAgg = true;
662,172,013✔
265
  bool         hasNullSMA = false;
662,206,517✔
266
  if (pLoad != NULL) {
662,170,535✔
267
    *pLoad = false;
662,142,228✔
268
  }
269

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

276
  if (!allColumnsHaveAgg || hasNullSMA) {
661,996,149!
277
    *pLoad = false;
588,250,744✔
278
  } else {
279
    *pLoad = true;
73,745,405✔
280
  }
281

282
  return code;
661,869,447✔
283
}
284

285
static int32_t doSetTagColumnData(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
2,147,483,647✔
286
                                  int32_t rows) {
287
  int32_t    code = 0;
2,147,483,647✔
288
  SExprSupp* pSup = &pTableScanInfo->pseudoSup;
2,147,483,647✔
289
  if (pSup->numOfExprs > 0) {
2,147,483,647✔
290
    code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
1,426,507,772✔
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) {
1,426,370,180!
294
      if (pTaskInfo->streamInfo.pState) blockDataCleanup(pBlock);
×
295
      code = 0;
×
296
    }
297
  }
298

299
  return code;
2,147,483,647✔
300
}
301

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

306
  if (pLimitInfo->remainOffset > 0) {
2,147,483,647✔
307
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
28,265,244✔
308
      pLimitInfo->remainOffset -= pBlock->info.rows;
12,022,453✔
309
      blockDataEmpty(pBlock);
12,022,453✔
310
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
12,022,453✔
311
      return false;
12,022,453✔
312
    } else {
313
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
16,242,791✔
314
      if (code != TSDB_CODE_SUCCESS) {
16,242,791!
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;
16,242,791✔
320
    }
321
  }
322

323
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
2,147,483,647✔
324
    // limit the output rows
325
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
139,830,491✔
326
    blockDataKeepFirstNRows(pBlock, keep);
139,824,465✔
327

328
    pLimitInfo->numOfOutputRows += pBlock->info.rows;
139,828,511✔
329
    qDebug("output limit %" PRId64 " has reached, %s", pLimit->limit, id);
139,828,341✔
330
    return true;
139,827,365✔
331
  }
332

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

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

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

349
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
2,147,483,647✔
350

351
  pCost->totalBlocks += 1;
2,147,483,647✔
352
  pCost->totalRows += pBlock->info.rows;
2,147,483,647✔
353
  *status = pTableScanInfo->dataBlockLoadFlag;
2,147,483,647✔
354

355
  if (pOperator->exprSupp.pFilterInfo != NULL) {
2,147,483,647✔
356
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
425,723,944✔
357
  } else {
358
    bool overlap = false;
2,147,483,647✔
359
    int  ret =
360
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
2,147,483,647✔
361
    if (ret != TSDB_CODE_SUCCESS) {
2,147,483,647!
362
      return ret;
×
363
    }
364
    if (overlap) {
2,147,483,647!
365
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
266,315,279✔
366
    }
367
  }
368

369
  SDataBlockInfo* pBlockInfo = &pBlock->info;
2,147,483,647✔
370
  taosMemoryFreeClear(pBlock->pBlockAgg);
2,147,483,647!
371

372
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
2,147,483,647!
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) {
2,147,483,647✔
380
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
128,019,838✔
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);
128,043,248✔
384
    pCost->skipBlocks += 1;
128,062,755✔
385
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
128,063,392✔
386
    return code;
128,061,931✔
387
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
2,147,483,647✔
388
    pCost->loadBlockStatis += 1;
236,394,643✔
389
    loadSMA = true;  // mark the operation of load sma;
236,408,517✔
390
    bool success = true;
236,408,517✔
391
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
236,407,149✔
392
    if (code) {
236,339,112!
393
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
394
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
395
      QUERY_CHECK_CODE(code, lino, _end);
251!
396
    }
397

398
    if (success) {  // failed to load the block sma data, data block statistics does not exist, load data block instead
236,310,783!
399
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
30,558,135✔
400
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
401
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
30,558,135✔
402
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
30,558,135✔
403
      return code;
30,558,135✔
404
    } else {
405
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
205,752,648✔
406
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
205,902,867✔
407
    }
408
  }
409

410
  if (*status != FUNC_DATA_REQUIRED_DATA_LOAD) {
2,147,483,647!
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)) {
2,147,483,647✔
418
    bool success = true;
425,682,366✔
419
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
425,679,439✔
420
    if (code) {
425,531,595!
421
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
422
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
423
      QUERY_CHECK_CODE(code, lino, _end);
53!
424
    }
425

426
    if (success) {
425,523,644!
427
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
43,187,270✔
428
      bool   keep = false;
43,187,270✔
429
      code = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows, &keep);
43,187,270✔
430
      if (code) {
43,187,270!
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) {
43,187,270!
437
        qDebug("%s data block filter out by block SMA, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
5,039,254!
438
               GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
439
        pCost->filterOutBlocks += 1;
5,039,254✔
440
        (*status) = FUNC_DATA_REQUIRED_FILTEROUT;
5,039,254✔
441
        taosMemoryFreeClear(pBlock->pBlockAgg);
5,039,254!
442

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

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

452
  // try to filter data block according to current results
453
  code = doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
2,147,483,647✔
454
  if (code) {
2,147,483,647!
455
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
456
    QUERY_CHECK_CODE(code, lino, _end);
×
457
  }
458

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

465
    STableScanInfo* p1 = pOperator->info;
29,097,046✔
466
    if (taosHashGetSize(p1->pIgnoreTables) == taosArrayGetSize(p1->base.pTableListInfo->pTableList)) {
29,097,046✔
467
      *status = FUNC_DATA_REQUIRED_ALL_FILTEROUT;
163,297✔
468
    } else {
469
      *status = FUNC_DATA_REQUIRED_FILTEROUT;
28,933,749✔
470
    }
471
    return TSDB_CODE_SUCCESS;
29,097,046✔
472
  }
473

474
  pCost->totalCheckedRows += pBlock->info.rows;
2,147,483,647✔
475
  pCost->loadBlocks += 1;
2,147,483,647✔
476

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

483
  if ((pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) &&
2,147,483,647✔
484
      ((STableScanInfo*)pOperator->info)->ignoreTag) {
2,127,200,759!
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);
2,147,483,647✔
489
    if (code) {
2,147,483,647!
490
      return code;
×
491
    }
492
  }
493

494
  // restore the previous value
495
  pCost->totalRows -= pBlock->info.rows;
2,147,483,647✔
496

497
  if (pOperator->exprSupp.pFilterInfo != NULL) {
2,147,483,647✔
498
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo, NULL);
420,556,214✔
499
    QUERY_CHECK_CODE(code, lino, _end);
420,506,176✔
500

501
    int64_t st = taosGetTimestampUs();
420,608,257✔
502
    double  el = (taosGetTimestampUs() - st) / 1000.0;
420,602,251✔
503
    pTableScanInfo->readRecorder.filterTime += el;
420,602,251✔
504

505
    if (pBlock->info.rows == 0) {
420,524,924✔
506
      pCost->filterOutBlocks += 1;
62,394,116✔
507
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
62,363,849✔
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);
358,020,230✔
511
    }
512
  }
513

514
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
2,147,483,647✔
515
  if (limitReached) {  // set operator flag is done
2,147,483,647✔
516
    setOperatorCompleted(pOperator);
21,464,794✔
517
  }
518

519
  pCost->totalRows += pBlock->info.rows;
2,147,483,647✔
520

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

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

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

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

542
static void freeTableCachedVal(void* param) {
1,050,331,347✔
543
  if (param == NULL) {
1,050,331,347!
544
    return;
×
545
  }
546

547
  STableCachedVal* pVal = param;
1,050,331,347✔
548
  taosMemoryFree((void*)pVal->pName);
1,050,331,347✔
549
  taosMemoryFree(pVal->pTags);
1,050,282,673✔
550
  taosMemoryFree(pVal);
1,050,270,477✔
551
}
552

553
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
1,049,571,882✔
554
  int32_t          code = TSDB_CODE_SUCCESS;
1,049,571,882✔
555
  int32_t          lino = 0;
1,049,571,882✔
556
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
1,049,571,882✔
557
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
1,049,823,052!
558

559
  pVal->pTags = NULL;
1,049,823,052✔
560
  pVal->pName = taosStrdup(pMetaReader->me.name);
1,049,863,881✔
561
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
1,049,853,106!
562

563
  // only child table has tag value
564
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
1,049,591,157!
565
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
1,048,608,585✔
566
    pVal->pTags = taosMemoryMalloc(pTag->len);
1,048,769,582✔
567
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
1,048,759,668!
568
    memcpy(pVal->pTags, pTag, pTag->len);
1,048,787,403!
569
  }
570

571
  (*ppResVal) = pVal;
1,049,779,559✔
572

573
_end:
1,049,985,945✔
574
  if (code != TSDB_CODE_SUCCESS) {
1,049,985,945!
575
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
576
    freeTableCachedVal(pVal);
×
577
  }
578
  return code;
1,049,727,115✔
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) {
1,050,338,367✔
583
  (void)key;
584
  (void)keyLen;
585
  (void)ud;
586
  freeTableCachedVal(value);
1,050,338,367✔
587
}
1,050,299,920✔
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,
1,544,576,710✔
604
                               int32_t rows, SExecTaskInfo* pTask, STableMetaCacheInfo* pCache) {
605
  int32_t          code = TSDB_CODE_SUCCESS;
1,544,576,710✔
606
  int32_t          lino = 0;
1,544,576,710✔
607
  bool             freeReader = false;
1,544,576,710✔
608
  LRUHandle*       h = NULL;
1,544,576,710✔
609
  STableCachedVal  val = {0};
1,544,576,710✔
610
  SMetaReader      mr = {0};
1,544,576,710✔
611
  const char*      idStr = pTask->id.str;
1,544,566,715✔
612
  int32_t          insertRet = TAOS_LRU_STATUS_OK;
1,544,693,036✔
613
  STableCachedVal* pVal = NULL;
1,544,693,036✔
614

615
  // currently only the tbname pseudo column
616
  if (numOfExpr <= 0) {
1,544,856,832✔
617
    return TSDB_CODE_SUCCESS;
2,026,266✔
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;
1,542,830,566✔
624
  pBlock->info.rows = rows;
1,542,848,566✔
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) {
1,542,381,994✔
631
    pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
694,443✔
632
    code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
694,443✔
633
    if (code != TSDB_CODE_SUCCESS) {
694,443!
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);
694,443✔
649

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

653
    freeReader = true;
693,073✔
654
  } else {
655
    pCache->metaFetch += 1;
1,541,687,551✔
656

657
    h = taosLRUCacheLookup(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(pBlock->info.id.uid));
1,542,039,408✔
658
    if (h == NULL) {
1,542,351,682✔
659
      pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
1,050,336,531✔
660
      freeReader = true;
1,050,301,019✔
661
      code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
1,050,301,019✔
662
      if (code != TSDB_CODE_SUCCESS) {
1,049,634,408!
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);
1,049,634,408✔
677

678
      code = createTableCacheVal(&mr, &pVal);
1,050,159,055✔
679
      QUERY_CHECK_CODE(code, lino, _end);
1,049,810,445!
680

681
      val = *pVal;
1,049,810,445✔
682
    } else {
683
      pCache->cacheHit += 1;
492,015,151✔
684
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
492,063,626✔
685
      val = *pValTmp;
492,113,712✔
686

687
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
492,142,781✔
688
      qTrace("release LRU cache, res %d", bRes);
492,135,209!
689
    }
690

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

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

699
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
2,147,483,647✔
700
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
2,147,483,647✔
701

702
    int32_t functionId = pExpr1->pExpr->_function.functionId;
2,147,483,647✔
703

704
    // this is to handle the tbname
705
    if (fmIsScanPseudoColumnFunc(functionId)) {
2,147,483,647✔
706
      int32_t fType = pExpr1->pExpr->_function.functionType;
607,312,588✔
707
      if (fType == FUNCTION_TYPE_TBNAME) {
607,308,783!
708
        code = setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
607,308,783✔
709
        QUERY_CHECK_CODE(code, lino, _end);
607,283,633!
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};
2,147,483,647✔
719
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
2,147,483,647✔
720
      const char* p = pHandle->api.metaFn.extractTagVal(val.pTags, pColInfoData->info.type, &tagVal);
2,147,483,647✔
721

722
      char* data = NULL;
2,147,483,647✔
723
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
2,147,483,647✔
724
        data = tTagValToData((const STagVal*)p, false);
2,147,483,647✔
725
      } else {
726
        data = (char*)p;
25,043,392✔
727
      }
728

729
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
2,147,483,647✔
730
      if (isNullVal) {
2,147,483,647✔
731
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
23,677,223!
732
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
733
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, 1, false);
2,147,483,647✔
734
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
2,147,483,647!
735
          char* tmp = taosMemoryCalloc(1, varDataLen(data) + 1);
1,178,714,328✔
736
          if (tmp != NULL) {
1,178,714,812!
737
            memcpy(tmp, varDataVal(data), varDataLen(data));
1,178,714,812!
738
            qDebug("get tag value:%s, cid:%d, table name:%s, uid%" PRId64, tmp, tagVal.cid, val.pName,
1,178,710,651✔
739
                   pBlock->info.id.uid);
740
            taosMemoryFree(tmp);
1,178,760,296✔
741
          }
742
          taosMemoryFree(data);
1,178,737,491✔
743
        }
744
        QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
745
      } else {  // todo opt for json tag
746
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
7,624,546✔
747
          code = colDataSetVal(pColInfoData, i, data, false);
6,258,377✔
748
          QUERY_CHECK_CODE(code, lino, _end);
6,258,377!
749
        }
750
      }
751
    }
752
  }
753

754
  // restore the rows
755
  pBlock->info.rows = backupRows;
1,543,176,194✔
756

757
_end:
1,543,124,804✔
758

759
  if (NULL != pVal) {
1,543,234,550✔
760
    insertRet =
1,050,307,409✔
761
        taosLRUCacheInsert(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(uint64_t), pVal,
1,050,340,038✔
762
                           sizeof(STableCachedVal), freeCachedMetaItem, NULL, NULL, TAOS_LRU_PRIORITY_LOW, NULL);
763
    if (insertRet != TAOS_LRU_STATUS_OK) {
1,050,307,409!
764
      qWarn("failed to put meta into lru cache, code:%d, %s", insertRet, idStr);
×
765
    }
766
  }
767

768
  if (freeReader) {
1,543,201,921✔
769
    pHandle->api.metaReaderFn.clearReader(&mr);
1,050,909,446✔
770
  }
771
  if (code != TSDB_CODE_SUCCESS) {
1,543,119,572!
772
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
773
  }
774
  return code;
1,543,166,958✔
775
}
776

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

785
  size_t len = TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE;
607,315,514✔
786
  char   buf[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
607,315,514✔
787
  STR_TO_VARSTR(buf, name)
607,317,689!
788

789
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
607,317,064✔
790

791
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
607,309,044✔
792
  QUERY_CHECK_CODE(code, lino, _end);
607,304,027!
793

794
  code = colDataSetVal(&infoData, 0, buf, false);
607,304,027✔
795
  QUERY_CHECK_CODE(code, lino, _end);
607,300,008!
796

797
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
607,300,008✔
798
  SScalarParam param = {.columnData = pColInfoData};
607,301,200✔
799

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

807
  colDataDestroy(&infoData);
607,295,381✔
808

809
_end:
607,283,979✔
810
  if (code != TSDB_CODE_SUCCESS) {
607,303,090!
811
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
812
  }
813
  return code;
607,303,090✔
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) {
694,815,051✔
883
  int32_t code = TSDB_CODE_SUCCESS;
694,815,051✔
884
  int32_t lino = 0;
694,815,051✔
885
  code = tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, pKeyInfo, size);
694,815,051✔
886
  QUERY_CHECK_CODE(code, lino, _end);
694,683,523!
887

888
  if (*size <= 0) {
694,683,523✔
889
    goto _end;
24,896,423✔
890
  }
891
  pInfo->tableStartIndex = TARRAY_ELEM_IDX(pInfo->base.pTableListInfo->pTableList, *pKeyInfo);
669,793,025!
892
  pInfo->tableEndIndex = (pInfo->tableStartIndex + (*size) - 1);
669,622,491✔
893
  pInfo->pResBlock->info.blankFill = false;
669,656,852✔
894
  taosMemoryFreeClear(pInfo->pResBlock->pBlockAgg);
669,747,525!
895

896
  if (!pInfo->needCountEmptyTable) {
669,789,210✔
897
    pInfo->countState = TABLE_COUNT_STATE_END;
587,574,443✔
898
  } else {
899
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
81,726,935✔
900
  }
901

902
_end:
694,295,741✔
903
  if (code != TSDB_CODE_SUCCESS) {
694,295,741!
904
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
905
  }
906
  return code;
694,418,739✔
907
}
908

909
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
2,147,483,647✔
910
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
2,147,483,647✔
911
    return;
2,046,753,731✔
912
  }
913
  if (pInfo->base.pTableListInfo->groupOffset) {
491,766,602✔
914
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
931,830✔
915
  } else {
916
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
490,816,477✔
917
    if (code != TSDB_CODE_SUCCESS) {
490,858,505✔
918
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
106,840,948!
919
    }
920
  }
921
}
922

923
static SSDataBlock* getOneRowResultBlock(SExecTaskInfo* pTaskInfo, STableScanBase* pBase, SSDataBlock* pBlock,
310,649,605✔
924
                                         const STableKeyInfo* tbInfo) {
925
  blockDataEmpty(pBlock);
310,649,605✔
926
  pBlock->info.rows = 1;
310,673,858✔
927
  pBlock->info.id.uid = tbInfo->uid;
310,675,651✔
928
  pBlock->info.id.groupId = tbInfo->groupId;
310,675,288✔
929
  pBlock->info.blankFill = true;
310,676,844✔
930

931
  // only one row: set all col data to null & hasNull
932
  int32_t col_num = blockDataGetNumOfCols(pBlock);
310,664,679✔
933
  for (int32_t i = 0; i < col_num; ++i) {
1,263,103,121✔
934
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
952,428,395✔
935
    colDataSetNULL(pColInfoData, 0);
936
  }
937

938
  // set tag/tbname
939
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
310,674,726✔
940

941
  return pBlock;
310,671,984✔
942
}
943

944
static SSDataBlock* getBlockForEmptyTable(SOperatorInfo* pOperator, const STableKeyInfo* tbInfo) {
298,516,841✔
945
  STableScanInfo* pTableScanInfo = pOperator->info;
298,516,841✔
946
  SSDataBlock*    pBlock =
947
      getOneRowResultBlock(pOperator->pTaskInfo, &pTableScanInfo->base, pTableScanInfo->pResBlock, tbInfo);
298,535,510✔
948

949
  pOperator->resultInfo.totalRows++;
298,534,954✔
950
  return pBlock;
298,555,259✔
951
}
952

953
static int32_t doTableScanImplNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
2,147,483,647✔
954
  int32_t         code = TSDB_CODE_SUCCESS;
2,147,483,647✔
955
  int32_t         lino = 0;
2,147,483,647✔
956
  STableScanInfo* pTableScanInfo = pOperator->info;
2,147,483,647✔
957
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
2,147,483,647✔
958
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
2,147,483,647✔
959
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
2,147,483,647✔
960
  bool            hasNext = false;
2,147,483,647✔
961
  int64_t         st = taosGetTimestampUs();
2,147,483,647✔
962

963
  QRY_PARAM_CHECK(ppRes);
2,147,483,647!
964
  pBlock->info.dataLoad = false;
2,147,483,647✔
965

966
  while (true) {
79,826,887✔
967
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
2,147,483,647✔
968
    if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
969
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
3,076✔
970
      QUERY_CHECK_CODE(code, lino, _end);
×
971
    }
972

973
    if (!hasNext) {
2,147,483,647✔
974
      break;
650,329,918✔
975
    }
976

977
    if (isTaskKilled(pTaskInfo)) {
2,147,483,647✔
978
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
41,873✔
979
      code = pTaskInfo->code;
41,873✔
980
      goto _end;
41,873✔
981
    }
982

983
    if (pOperator->status == OP_EXEC_DONE) {
2,147,483,647✔
984
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
1,555,624✔
985
      break;
1,555,624✔
986
    }
987

988
    // process this data block based on the probabilities
989
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
2,147,483,647✔
990
    if (!processThisBlock) {
2,147,483,647!
991
      continue;
×
992
    }
993

994
    if (pBlock->info.id.uid) {
2,147,483,647!
995
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
2,147,483,647✔
996
    }
997

998
    uint32_t status = 0;
2,147,483,647✔
999
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
2,147,483,647✔
1000
    QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647✔
1001

1002
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
2,147,483,647✔
1003
      break;
163,297✔
1004
    }
1005

1006
    // current block is filter out according to filter condition, continue load the next block
1007
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
2,147,483,647✔
1008
      continue;
79,826,887✔
1009
    }
1010

1011
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
2,147,483,647✔
1012
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
2,147,483,647✔
1013

1014
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
2,147,483,647✔
1015
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
2,147,483,647✔
1016

1017
    (*ppRes) = pBlock;
2,147,483,647✔
1018
    return code;
2,147,483,647✔
1019
  }
1020

1021
_end:
652,094,740✔
1022
  if (code != TSDB_CODE_SUCCESS) {
652,125,287✔
1023
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
70,330!
1024
    pTaskInfo->code = code;
70,330✔
1025
    T_LONG_JMP(pTaskInfo->env, code);
46,478!
1026
  }
1027
  return code;
652,054,957✔
1028
}
1029

1030
static int32_t doGroupedTableScan(SOperatorInfo* pOperator, SSDataBlock** pBlock) {
2,147,483,647✔
1031
  int32_t         code = TSDB_CODE_SUCCESS;
2,147,483,647✔
1032
  int32_t         lino = 0;
2,147,483,647✔
1033
  STableScanInfo* pTableScanInfo = pOperator->info;
2,147,483,647✔
1034
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
2,147,483,647✔
1035
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
2,147,483,647✔
1036

1037
  QRY_PARAM_CHECK(pBlock);
2,147,483,647!
1038

1039
  // The read handle is not initialized yet, since no qualified tables exists
1040
  if (pTableScanInfo->base.dataReader == NULL || pOperator->status == OP_EXEC_DONE) {
2,147,483,647!
1041
    return code;
11,600,466✔
1042
  }
1043

1044
  // do the ascending order traverse in the first place.
1045
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
2,147,483,647✔
1046
    SSDataBlock* p = NULL;
2,147,483,647✔
1047
    code = doTableScanImplNext(pOperator, &p);
2,147,483,647✔
1048
    QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
1049

1050
    if (p != NULL) {
2,147,483,647✔
1051
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
2,090,807,270✔
1052
      *pBlock = p;
2,090,778,961✔
1053
      return code;
2,090,986,321✔
1054
    }
1055

1056
    pTableScanInfo->scanTimes += 1;
619,005,189✔
1057
    taosHashClear(pTableScanInfo->pIgnoreTables);
619,097,031✔
1058

1059
    if (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
618,979,452✔
1060
      setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
308,513✔
1061
      pTableScanInfo->base.scanFlag = MAIN_SCAN;
308,513✔
1062
      pTableScanInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
308,513✔
1063
      qDebug("start to repeat ascending order scan data blocks due to query func required, %s", GET_TASKID(pTaskInfo));
308,513!
1064

1065
      // do prepare for the next round table scan operation
1066
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
308,513✔
1067
      QUERY_CHECK_CODE(code, lino, _end);
260,846!
1068
    }
1069
  }
1070

1071
  int32_t total = pTableScanInfo->scanInfo.numOfAsc + pTableScanInfo->scanInfo.numOfDesc;
1,099,100,883✔
1072
  if (pTableScanInfo->scanTimes < total) {
1,099,020,339✔
1073
    if (pTableScanInfo->base.cond.order == TSDB_ORDER_ASC) {
181,970,113✔
1074
      prepareForDescendingScan(&pTableScanInfo->base, pOperator->exprSupp.pCtx, 0);
225,231✔
1075
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
225,231✔
1076
      QUERY_CHECK_CODE(code, lino, _end);
225,231!
1077
      qDebug("%s start to descending order scan data blocks due to query func required", GET_TASKID(pTaskInfo));
225,231!
1078
    }
1079

1080
    while (pTableScanInfo->scanTimes < total) {
215,047,924✔
1081
      SSDataBlock* p = NULL;
182,008,345✔
1082
      code = doTableScanImplNext(pOperator, &p);
181,958,329✔
1083
      QUERY_CHECK_CODE(code, lino, _end);
181,959,541!
1084

1085
      if (p != NULL) {
181,959,541✔
1086
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
148,917,318✔
1087
        *pBlock = p;
148,878,482✔
1088
        return code;
148,910,155✔
1089
      }
1090

1091
      pTableScanInfo->scanTimes += 1;
33,042,223✔
1092
      taosHashClear(pTableScanInfo->pIgnoreTables);
33,045,581✔
1093

1094
      if (pTableScanInfo->scanTimes < total) {
33,052,356!
1095
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
×
1096
        pTableScanInfo->base.scanFlag = MAIN_SCAN;
×
1097

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

1105
  if (pTableScanInfo->countState < TABLE_COUNT_STATE_END) {
949,981,079✔
1106
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
403,643,061✔
1107
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
403,602,713✔
1108
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
379,342✔
1109
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
26,404✔
1110
        STableKeyInfo* pStart =
1111
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
26,404✔
1112

1113
        if (NULL == pStart) {
26,404!
1114
          return code;
×
1115
        }
1116

1117
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
26,404✔
1118
        return code;
26,404✔
1119
      }
1120
    } else {  // group by tag + no sort
1121
      int32_t numOfTables = 0;
403,232,860✔
1122
      code = tableListGetSize(pTableListInfo, &numOfTables);
403,229,615✔
1123
      QUERY_CHECK_CODE(code, lino, _end);
403,307,202!
1124

1125
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
403,307,202✔
1126
        // get empty group, mark processed & rm from hash
1127
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
402,832,885✔
1128
        if (pIte != NULL) {
402,825,714✔
1129
          size_t        keySize = 0;
298,509,827✔
1130
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
298,509,833✔
1131
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
298,506,590✔
1132
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
298,510,921✔
1133
          markGroupProcessed(pTableScanInfo, *pGroupId);
298,513,064✔
1134
          *pBlock = getBlockForEmptyTable(pOperator, &info);
298,497,001✔
1135

1136
          return code;
298,522,212✔
1137
        }
1138
      }
1139
    }
1140
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
105,141,396✔
1141
  }
1142

1143
_end:
546,463,654✔
1144
  if (code != TSDB_CODE_SUCCESS) {
651,505,198!
1145
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1146
    pTaskInfo->code = code;
×
1147
  }
1148

1149
  return code;
651,505,198✔
1150
}
1151

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

1164
  qDebug("vgId:%d add total %d dynamic tables to scan, tableSeq:%d, exist num:%" PRId64 ", operator status:%d",
1,141,348!
1165
         pTaskInfo->id.vgId, num, pParam->tableSeq, (int64_t)taosArrayGetSize(pListInfo->pTableList),
1166
         pOperator->status);
1167

1168
  if (pParam->tableSeq) {
1,141,348!
1169
    pListInfo->oneTableForEachGroup = true;
51,264✔
1170
    if (taosArrayGetSize(pListInfo->pTableList) > 0) {
51,264!
1171
      taosHashClear(pListInfo->map);
×
1172
      taosArrayClear(pListInfo->pTableList);
×
1173
      pOperator->status = OP_EXEC_DONE;
×
1174
    }
1175
  } else {
1176
    pListInfo->oneTableForEachGroup = false;
1,090,084✔
1177
    pListInfo->numOfOuputGroups = 1;
1,090,084✔
1178
  }
1179

1180
  STableKeyInfo info = {.groupId = 0};
1,141,348✔
1181
  int32_t       tableIdx = 0;
1,141,348✔
1182
  for (int32_t i = 0; i < num; ++i) {
47,263,194✔
1183
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
46,121,846✔
1184
    if (!pUid) {
46,121,846!
1185
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1186
      return terrno;
×
1187
    }
1188

1189
    if (taosHashPut(pListInfo->map, pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
46,121,846✔
1190
      if (TSDB_CODE_DUP_KEY == terrno) {
43,091,676!
1191
        continue;
43,092,300✔
1192
      }
1193
      return terrno;
×
1194
    }
1195

1196
    info.uid = *pUid;
3,030,170✔
1197
    void* p = taosArrayPush(pListInfo->pTableList, &info);
3,030,170✔
1198
    if (p == NULL) {
3,030,170!
1199
      return terrno;
×
1200
    }
1201

1202
    tableIdx++;
3,030,170✔
1203
    qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
3,030,170✔
1204
  }
1205

1206
  return code;
1,141,348✔
1207
}
1208

1209
static int32_t doInitReader(STableScanInfo* pInfo, SExecTaskInfo* pTaskInfo, SStorageAPI* pAPI, int32_t* pNum,
668,690,667✔
1210
                            STableKeyInfo** pList) {
1211
  const char* idStr = GET_TASKID(pTaskInfo);
668,690,667✔
1212
  int32_t     code = initNextGroupScan(pInfo, pList, pNum);
668,840,423✔
1213
  if (code) {
668,072,986!
1214
    qError("%s failed to init groupScan Info, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
×
1215
    return code;
×
1216
  }
1217

1218
  if (pInfo->base.dataReader != NULL) {
668,072,986!
1219
    qError("%s tsdb reader should be null", idStr);
×
1220
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1221
  }
1222

1223
  code = pAPI->tsdReader.tsdReaderOpen(pInfo->base.readHandle.vnode, &pInfo->base.cond, *pList, *pNum, pInfo->pResBlock,
1,336,816,343✔
1224
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
668,624,330✔
1225
  if (code) {
668,486,067✔
1226
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
24,087!
1227
  }
1228

1229
  return code;
668,231,478✔
1230
}
1231

1232
int compareColIdPair(const void* elem1, const void* elem2) {
643,383,232✔
1233
  SColIdPair* node1 = (SColIdPair*)elem1;
643,383,232✔
1234
  SColIdPair* node2 = (SColIdPair*)elem2;
643,383,232✔
1235

1236
  if (node1->orgColId < node2->orgColId) {
643,383,232✔
1237
    return -1;
643,010,064✔
1238
  }
1239

1240
  return node1->orgColId > node2->orgColId;
373,168✔
1241
}
1242

1243
static bool isNewScanParam(STableScanOperatorParam* pParam) {
87,160,173✔
1244
  return pParam->window.skey == INT64_MAX && pParam->window.ekey == INT64_MIN;
87,160,173!
1245
}
1246

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

1263
  cleanupQueryTableDataCond(&pInfo->base.cond);
87,152,534✔
1264

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

1287
  pListInfo->oneTableForEachGroup = true;
87,152,534✔
1288
  taosHashClear(pListInfo->map);
87,152,534✔
1289
  taosArrayClear(pListInfo->pTableList);
87,152,534✔
1290

1291
  uint64_t      pUid = orgTable.me.uid;
87,152,534✔
1292
  STableKeyInfo info = {.groupId = 0, .uid = pUid};
87,152,534✔
1293
  int32_t       tableIdx = 0;
87,152,534✔
1294
  code = taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t));
87,152,534✔
1295
  QUERY_CHECK_CODE(code, lino, _return);
87,152,534!
1296
  QUERY_CHECK_NULL(taosArrayPush(pListInfo->pTableList, &info), code, lino, _return, terrno);
174,305,068!
1297
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
87,152,534✔
1298

1299
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
87,152,534✔
1300
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
87,152,534!
1301
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(int32_t));
87,152,534✔
1302
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
87,152,534!
1303

1304
  // virtual table's origin table scan do not has ts column.
1305
  SColIdPair tsPair = {.vtbColId = PRIMARYKEY_TIMESTAMP_COL_ID, .orgColId = PRIMARYKEY_TIMESTAMP_COL_ID};
87,152,534✔
1306
  QUERY_CHECK_NULL(taosArrayPush(pColArray, &tsPair), code, lino, _return, terrno);
87,152,534!
1307

1308
  for (int32_t i = 0; i < taosArrayGetSize(pParam->pOrgTbInfo->colMap); ++i) {
235,727,400✔
1309
    SColIdNameKV* kv = taosArrayGet(pParam->pOrgTbInfo->colMap, i);
148,574,866✔
1310
    for (int32_t j = 0; j < schema->nCols; j++) {
1,228,813,570!
1311
      if (strcmp(kv->colName, schema->pSchema[j].name) == 0) {
1,228,813,570!
1312
        SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(j + 1)};
148,574,866✔
1313
        QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
148,574,866!
1314
        break;
148,574,866✔
1315
      }
1316
    }
1317
  }
1318

1319
  for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
322,879,934✔
1320
    SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, i);
235,727,400✔
1321
    for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
1,274,942,264!
1322
      SColMatchItem* pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
1,274,942,264✔
1323
      if (pItem->colId == pPair->vtbColId) {
1,274,942,264✔
1324
        SColIdPair tmpPair = {.orgColId = pPair->orgColId, .vtbColId = pItem->dstSlotId};
235,727,400✔
1325
        QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &tmpPair), code, lino, _return, terrno);
235,727,400!
1326
        break;
235,727,400✔
1327
      }
1328
    }
1329
  }
1330

1331
  taosArraySort(pColArray, compareColIdPair);
87,152,534✔
1332
  taosArraySort(pBlockColArray, compareColIdPair);
87,152,534✔
1333

1334
  code = initQueryTableDataCondWithColArray(&pInfo->base.cond, &pInfo->base.orgCond, &pInfo->base.readHandle, pColArray);
87,152,534✔
1335
  QUERY_CHECK_CODE(code, lino, _return);
87,152,534!
1336

1337
  if (pInfo->pResBlock) {
87,152,534✔
1338
    blockDataDestroy(pInfo->pResBlock);
62,400,233✔
1339
    pInfo->pResBlock = NULL;
62,400,233✔
1340
  }
1341

1342
  if (pParam->window.ekey > 0) {
87,152,534✔
1343
    pInfo->base.cond.twindows.skey = pParam->window.ekey + 1;
62,361,337✔
1344
  } else if (isNewScanParam(pParam)) {
24,791,197!
1345
    pInfo->base.cond.twindows.skey = pInfo->base.orgCond.twindows.skey;
24,791,197✔
1346
    pInfo->base.cond.twindows.ekey = pInfo->base.orgCond.twindows.ekey;
24,791,197✔
1347
  }
1348
  pInfo->base.cond.suid = orgTable.me.type == TSDB_CHILD_TABLE ? superTable.me.uid : 0;
87,152,534✔
1349
  pInfo->currentGroupId = 0;
87,152,534✔
1350
  pInfo->ignoreTag = true;
87,152,534✔
1351

1352
  code = createOneDataBlockWithColArray(pInfo->pOrgBlock, pBlockColArray, &pInfo->pResBlock);
87,152,534✔
1353
  QUERY_CHECK_CODE(code, lino, _return);
87,152,534!
1354

1355
  void **reader = taosHashGet(pInfo->readerCache, &pUid, sizeof(uint64_t));
87,152,534✔
1356
  if (reader) {
87,152,534✔
1357
    if (isNewScanParam(pParam)) {
62,368,976✔
1358
      pAPI->tsdReader.tsdReaderClose(*reader);
7,639✔
1359
      pInfo->base.dataReader = NULL;
7,639✔
1360

1361
      code = taosHashRemove(pInfo->readerCache, &pUid, sizeof(uint64_t));
7,639✔
1362
      QUERY_CHECK_CODE(code, lino, _return);
7,639!
1363

1364
      taosRLockLatch(&pTaskInfo->lock);
7,639✔
1365
      code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
7,639✔
1366
      taosRUnLockLatch(&pTaskInfo->lock);
7,639✔
1367
      QUERY_CHECK_CODE(code, lino, _return);
7,639!
1368

1369
      code = taosHashPut(pInfo->readerCache, &pUid, sizeof(uint64_t), &pInfo->base.dataReader, POINTER_BYTES);
7,639✔
1370
      QUERY_CHECK_CODE(code, lino, _return);
7,639!
1371
      pInfo->newReader = true;
7,639✔
1372
    } else {
1373
      pInfo->base.dataReader = *reader;
62,361,337✔
1374
      code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
62,361,337✔
1375
      QUERY_CHECK_CODE(code, lino, _return);
62,361,337!
1376

1377
      pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, pInfo->pResBlock);
62,361,337✔
1378
      pInfo->newReader = false;
62,361,337✔
1379
      pInfo->scanTimes = 0;
62,361,337✔
1380
    }
1381
  } else {
1382
    pInfo->base.dataReader = NULL;
24,783,558✔
1383

1384
    taosRLockLatch(&pTaskInfo->lock);
24,783,558✔
1385
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
24,783,558✔
1386
    taosRUnLockLatch(&pTaskInfo->lock);
24,783,558✔
1387
    QUERY_CHECK_CODE(code, lino, _return);
24,783,558!
1388

1389
    code = taosHashPut(pInfo->readerCache, &pUid, sizeof(uint64_t), &pInfo->base.dataReader, POINTER_BYTES);
24,783,558✔
1390
    QUERY_CHECK_CODE(code, lino, _return);
24,783,558!
1391
    pInfo->newReader = true;
24,783,558✔
1392
  }
1393
  pOperator->status = OP_OPENED;
87,152,534✔
1394
  
1395
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
87,152,534!
1396
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1397
  }
1398

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

1411
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
647,715,166✔
1412
  int32_t         code = TSDB_CODE_SUCCESS;
647,715,166✔
1413
  int32_t         lino = 0;
647,715,166✔
1414
  STableScanInfo* pInfo = pOperator->info;
647,715,166✔
1415
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
647,729,722✔
1416
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
647,760,252✔
1417
  int32_t         numOfTables = 0;
647,701,929✔
1418

1419
  QRY_PARAM_CHECK(pResult);
647,661,205!
1420

1421
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
647,692,579✔
1422
  QUERY_CHECK_CODE(code, lino, _end);
647,799,367!
1423

1424
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
647,799,367✔
1425
    setOperatorCompleted(pOperator);
621,866,382✔
1426
    if (pOperator->dynamicTask) {
621,835,016!
1427
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
1,137,365✔
1428
      taosHashClear(pInfo->base.pTableListInfo->map);
1,136,050✔
1429
    }
1430
    return code;
621,814,166✔
1431
  }
1432

1433
  // reset value for the next group data output
1434
  pOperator->status = OP_OPENED;
25,872,763✔
1435
  resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
25,871,449✔
1436

1437
  int32_t        num = 0;
25,871,449✔
1438
  STableKeyInfo* pList = NULL;
25,871,449✔
1439
  code = initNextGroupScan(pInfo, &pList, &num);
25,871,449✔
1440
  QUERY_CHECK_CODE(code, lino, _end);
25,872,763!
1441

1442
  code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
25,872,763✔
1443
  QUERY_CHECK_CODE(code, lino, _end);
25,872,763!
1444

1445
  code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
25,872,763✔
1446
  QUERY_CHECK_CODE(code, lino, _end);
25,871,449!
1447
  pInfo->scanTimes = 0;
25,871,449✔
1448

1449
  code = doGroupedTableScan(pOperator, pResult);
25,871,449✔
1450
  QUERY_CHECK_CODE(code, lino, _end);
25,872,763!
1451

1452
  if (*pResult != NULL) {
25,872,763✔
1453
    if (pOperator->dynamicTask) {
25,670,425!
1454
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
24,759,409✔
1455
    }
1456
    return code;
25,670,425✔
1457
  }
1458

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

1467
static int32_t groupSeqTableScan(SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
2,147,483,647✔
1468
  int32_t         code = TSDB_CODE_SUCCESS;
2,147,483,647✔
1469
  int32_t         lino = 0;
2,147,483,647✔
1470
  STableScanInfo* pInfo = pOperator->info;
2,147,483,647✔
1471
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
2,147,483,647✔
1472
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
2,147,483,647✔
1473
  int32_t         num = 0;
2,147,483,647✔
1474
  STableKeyInfo*  pList = NULL;
2,147,483,647✔
1475
  SSDataBlock*    pResult = NULL;
2,147,483,647✔
1476
  const char*     idStr = GET_TASKID(pTaskInfo);
2,147,483,647✔
1477

1478
  QRY_PARAM_CHECK(pResBlock);
2,147,483,647!
1479

1480
  if (pInfo->currentGroupId == -1) {
2,147,483,647✔
1481
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
643,804,649!
1482
      setOperatorCompleted(pOperator);
×
1483
      return code;
×
1484
    }
1485

1486
    taosRLockLatch(&pTaskInfo->lock);
643,941,815✔
1487
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
643,998,577✔
1488
    taosRUnLockLatch(&pTaskInfo->lock);
643,651,749✔
1489
    QUERY_CHECK_CODE(code, lino, _end);
643,942,856✔
1490

1491
    if (pInfo->filesetDelimited) {
643,918,769!
1492
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
1,113,075✔
1493
    }
1494

1495
    if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
643,871,846!
1496
      pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1497
    }
1498
  }
1499

1500
  code = doGroupedTableScan(pOperator, &pResult);
2,147,483,647✔
1501
  QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
1502

1503
  if (pResult != NULL) {
2,147,483,647✔
1504
    if (pOperator->dynamicTask) {
2,147,483,647!
1505
      pResult->info.id.groupId = pResult->info.id.uid;
2,851,909✔
1506
    }
1507

1508
    *pResBlock = pResult;
2,147,483,647✔
1509
    return code;
2,147,483,647✔
1510
  }
1511

1512
  while (true) {
1513
    code = startNextGroupScan(pOperator, &pResult);
622,285,801✔
1514
    QUERY_CHECK_CODE(code, lino, _end);
622,878,730!
1515

1516
    if (pResult || pOperator->status == OP_EXEC_DONE) {
622,878,730✔
1517
      *pResBlock = pResult;
622,801,255✔
1518
      return code;
622,761,486✔
1519
    }
1520
  }
1521

1522
_end:
24,087✔
1523
  if (code != TSDB_CODE_SUCCESS) {
24,087!
1524
    qError("%s %s failed at line %d since %s", idStr, __func__, lino, tstrerror(code));
24,087!
1525
    pTaskInfo->code = code;
24,087✔
1526
  }
1527

1528
  return code;
24,087✔
1529
}
1530

1531
static bool isEmptyQueryTimeWindow(STimeWindow* pWindow) {
87,152,534✔
1532
  return (pWindow == NULL) || (pWindow->skey > pWindow->ekey);
87,152,534!
1533
}
1534

1535
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
2,147,483,647✔
1536
  int32_t         code = TSDB_CODE_SUCCESS;
2,147,483,647✔
1537
  int32_t         lino = 0;
2,147,483,647✔
1538
  STableScanInfo* pInfo = pOperator->info;
2,147,483,647✔
1539
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
2,147,483,647✔
1540
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
2,147,483,647✔
1541
  QRY_PARAM_CHECK(ppRes);
2,147,483,647!
1542
  qTrace("%s call", __FUNCTION__);
2,147,483,647!
1543
  if (pOperator->pOperatorGetParam) {
2,147,483,647✔
1544
    pOperator->dynamicTask = true;
88,293,882✔
1545
    if (isDynVtbScan(pOperator)) {
88,293,882✔
1546
      code = createVTableScanInfoFromParam(pOperator);
87,152,534✔
1547

1548
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
87,152,534✔
1549
      pOperator->pOperatorGetParam = NULL;
87,152,534✔
1550
      QUERY_CHECK_CODE(code, lino, _end);
87,152,534!
1551

1552
      SSDataBlock* result = NULL;
87,152,534✔
1553
      if (isEmptyQueryTimeWindow(&pInfo->base.cond.twindows) && pInfo->base.cond.type == TIMEWINDOW_RANGE_CONTAINED) {
87,152,534!
1554
        (*ppRes) = result;
33,098✔
1555
        return code;
33,098✔
1556
      }
1557

1558
      if (pInfo->newReader) {
87,119,436!
1559
        code = startNextGroupScan(pOperator, &result);
24,791,197✔
1560
        QUERY_CHECK_CODE(code, lino, _end);
24,791,197!
1561
      } else {
1562
        code = doGroupedTableScan(pOperator, &result);
62,328,239✔
1563
        QUERY_CHECK_CODE(code, lino, _end);
62,328,239!
1564
      }
1565

1566
      if (result) {
87,119,436✔
1567
        SSDataBlock* res = NULL;
62,493,816✔
1568
        pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, NULL);
62,493,816✔
1569
        code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, &res);
62,493,816✔
1570
        QUERY_CHECK_CODE(code, lino, _end);
62,493,816!
1571
        pInfo->pResBlock = res;
62,493,816✔
1572
        blockDataDestroy(result);
62,493,816✔
1573
        (*ppRes) = res;
62,493,816✔
1574
      } else {
1575
        STableKeyInfo *keyInfo = taosArrayGet(pInfo->base.pTableListInfo->pTableList, 0);
24,625,620✔
1576
        QUERY_CHECK_NULL(keyInfo, code, lino, _end, terrno)
24,625,620!
1577

1578
        blockDataDestroy(pInfo->pResBlock);
24,625,620✔
1579
        pInfo->pResBlock = NULL;
24,625,620✔
1580

1581
        void** reader = taosHashGet(pInfo->readerCache, &keyInfo->uid, sizeof(uint64_t));
24,625,620✔
1582
        if (reader) {
24,625,620!
1583
          if (*reader == pInfo->base.dataReader) {
24,625,620!
1584
            pAPI->tsdReader.tsdReaderSetDatablock(pInfo->base.dataReader, NULL);
24,625,620✔
1585
            pInfo->base.dataReader = NULL;
24,625,620✔
1586
          }
1587
          if (pAPI->tsdReader.tsdReaderClose) {
24,625,620!
1588
            pAPI->tsdReader.tsdReaderClose(*reader);
24,625,620✔
1589
          }
1590
        }
1591

1592
        code = taosHashRemove(pInfo->readerCache, &keyInfo->uid, sizeof(uint64_t));
24,625,620✔
1593
        QUERY_CHECK_CODE(code, lino, _end);
24,625,620!
1594
      }
1595
      return code;
87,119,436✔
1596
    } else {
1597
      code = createTableListInfoFromParam(pOperator);
1,141,348✔
1598
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
1,141,348✔
1599
      pOperator->pOperatorGetParam = NULL;
1,141,348✔
1600
      QUERY_CHECK_CODE(code, lino, _end);
1,141,348!
1601

1602
      if (pOperator->status == OP_EXEC_DONE) {
1,141,348✔
1603
        pInfo->currentGroupId = -1;
25,632✔
1604
        pOperator->status = OP_OPENED;
25,632✔
1605
        SSDataBlock* result = NULL;
25,632✔
1606

1607
        while (true) {
1608
          code = startNextGroupScan(pOperator, &result);
25,632✔
1609
          QUERY_CHECK_CODE(code, lino, _end);
25,632!
1610

1611
          if (result || pOperator->status == OP_EXEC_DONE) {
25,632!
1612
            (*ppRes) = result;
25,632✔
1613
            return code;
25,632✔
1614
          }
1615
        }
1616
      }
1617
    }
1618
  }
1619

1620
  // scan table one by one sequentially
1621
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
2,147,483,647✔
1622
    int32_t       numOfTables = 0;
16,651,525✔
1623
    STableKeyInfo tInfo = {0};
16,651,525✔
1624
    pInfo->countState = TABLE_COUNT_STATE_END;
16,651,525✔
1625

1626
    while (1) {
15,458,932✔
1627
      SSDataBlock* result = NULL;
32,110,457✔
1628
      code = doGroupedTableScan(pOperator, &result);
32,106,951✔
1629
      QUERY_CHECK_CODE(code, lino, _end);
32,069,704!
1630

1631
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
32,069,704!
1632
        (*ppRes) = result;
16,472,130✔
1633
        return code;
16,509,493✔
1634
      }
1635

1636
      // if no data, switch to next table and continue scan
1637
      pInfo->currentTable++;
15,587,163✔
1638

1639
      taosRLockLatch(&pTaskInfo->lock);
15,555,928✔
1640
      numOfTables = 0;
15,581,914✔
1641
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
15,581,914✔
1642
      if (code != TSDB_CODE_SUCCESS) {
15,578,443!
1643
        taosRUnLockLatch(&pTaskInfo->lock);
×
1644
        TSDB_CHECK_CODE(code, lino, _end);
×
1645
      }
1646

1647
      if (pInfo->currentTable >= numOfTables) {
15,578,443✔
1648
        qDebug("all table checked in table list, total:%d, return NULL, %s", numOfTables, GET_TASKID(pTaskInfo));
129,082!
1649
        taosRUnLockLatch(&pTaskInfo->lock);
129,082✔
1650
        (*ppRes) = NULL;
129,082✔
1651
        return code;
129,082✔
1652
      }
1653

1654
      STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
15,426,800✔
1655
      if (!tmp) {
15,455,440!
1656
        taosRUnLockLatch(&pTaskInfo->lock);
×
1657
        (*ppRes) = NULL;
×
1658
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1659
      }
1660

1661
      tInfo = *tmp;
15,455,440✔
1662
      taosRUnLockLatch(&pTaskInfo->lock);
15,429,398✔
1663

1664
      code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, &tInfo, 1);
15,467,585✔
1665
      QUERY_CHECK_CODE(code, lino, _end);
15,449,368!
1666
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d/%d %s", tInfo.uid, numOfTables,
15,449,368!
1667
             pInfo->currentTable, numOfTables, GET_TASKID(pTaskInfo));
1668

1669
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
15,476,270✔
1670
      QUERY_CHECK_CODE(code, lino, _end);
15,438,079!
1671
      pInfo->scanTimes = 0;
15,438,079✔
1672
    }
1673
  } else {  // scan table group by group sequentially
1674
    code = groupSeqTableScan(pOperator, ppRes);
2,147,483,647✔
1675
    QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647✔
1676
  }
1677

1678
_end:
2,147,483,647✔
1679
  if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
1680
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
24,087!
1681
    pTaskInfo->code = code;
24,087✔
1682
    T_LONG_JMP(pTaskInfo->env, code);
24,087!
1683
  }
1684

1685
  return code;
2,147,483,647✔
1686
}
1687

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

1700
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
822,184,695✔
1701
  cleanupQueryTableDataCond(&pBase->cond);
822,184,695✔
1702
  cleanupQueryTableDataCond(&pBase->orgCond);
822,165,697✔
1703

1704
  if (pAPI->tsdReaderClose) {
822,215,587!
1705
    pAPI->tsdReaderClose(pBase->dataReader);
822,232,169✔
1706
  }
1707
  pBase->dataReader = NULL;
822,063,956✔
1708

1709
  if (pBase->matchInfo.pList != NULL) {
822,081,308!
1710
    taosArrayDestroy(pBase->matchInfo.pList);
822,124,172✔
1711
  }
1712

1713
  tableListDestroy(pBase->pTableListInfo);
821,983,199✔
1714
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
822,191,020✔
1715
  cleanupExprSupp(&pBase->pseudoSup);
822,264,483✔
1716
}
822,242,202✔
1717

1718
static void cleanReaderForVTable(STableScanInfo* pInfo){
13,146,409✔
1719
  if (pInfo != NULL && pInfo->base.readerAPI.tsdReaderClose) {
13,146,409!
1720
    void *pIter = taosHashIterate(pInfo->readerCache, NULL);
13,146,409✔
1721
    while (pIter != NULL) {
13,304,347✔
1722
      void **reader = pIter;
157,938✔
1723
      if (*reader) {
157,938!
1724
        if (*reader == pInfo->base.dataReader) {
157,938✔
1725
          pInfo->base.dataReader = NULL;
126,681✔
1726
        }
1727
        pInfo->base.readerAPI.tsdReaderClose(*reader);
157,938✔
1728
      }
1729
      pIter = taosHashIterate(pInfo->readerCache, pIter);
157,938✔
1730
    }
1731
  }
1732
}
13,146,409✔
1733

1734
static void destroyTableScanOperatorInfo(void* param) {
687,763,665✔
1735
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
687,763,665✔
1736
  blockDataDestroy(pTableScanInfo->pResBlock);
687,763,665✔
1737
  blockDataDestroy(pTableScanInfo->pOrgBlock);
687,750,355✔
1738
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
687,759,316✔
1739
  if (pTableScanInfo->virtualStableScan && pTableScanInfo->readerCache) {
687,712,487!
1740
    cleanReaderForVTable(pTableScanInfo);
13,146,409✔
1741
    taosHashCleanup(pTableScanInfo->readerCache);
13,146,409✔
1742
  }
1743
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
687,711,454✔
1744
  taosMemoryFreeClear(param);
687,761,266✔
1745
}
687,729,169✔
1746

1747
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
687,361,375✔
1748
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
687,361,375!
1749
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
41,849,127✔
1750
    for (int32_t i = 0; i < numOfCols; ++i) {
104,145,793✔
1751
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
62,334,768✔
1752
      if (pCol) {
62,319,339✔
1753
        pCol->info.noData = true;
62,318,737✔
1754
      }
1755
    }
1756
  }
1757
}
687,323,273✔
1758

1759
static int32_t resetTableScanOperatorState(SOperatorInfo* pOper) {
1,180,575✔
1760
  int32_t         code = TSDB_CODE_SUCCESS;
1,180,575✔
1761
  STableScanInfo*   pInfo = pOper->info;
1,180,575✔
1762
  pOper->status = OP_NOT_OPENED;
1,180,575✔
1763

1764
  pInfo->scanTimes = 0;
1,180,575✔
1765
  pInfo->currentGroupId = -1;
1,180,575✔
1766
  pInfo->tableEndIndex = -1;
1,180,575✔
1767
  pInfo->tableStartIndex = 0;
1,180,575✔
1768
  pInfo->currentTable = 0;
1,180,575✔
1769
  pInfo->scanMode = 0;
1,180,575✔
1770
  pInfo->countState = 0;
1,180,575✔
1771
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
1,180,575!
1772

1773
  tableListDestroy(pInfo->base.pTableListInfo);
1,180,575✔
1774
  pInfo->base.pTableListInfo = tableListCreate();
1,180,575✔
1775
  if (!pInfo->base.pTableListInfo) {
1,180,575!
1776
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1777
    return terrno;
×
1778
  }
1779
  SExecTaskInfo*         pTaskInfo = pOper->pTaskInfo;
1,180,575✔
1780

1781
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
1,180,575✔
1782
  if (!pTableScanNode->scan.node.dynamicOp) {
1,180,575✔
1783
    code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
534,186!
1784
                                    &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
1785
                                    pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
534,186✔
1786
    if (code) {
267,786!
1787
      qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
×
1788
      return code;
×
1789
    }
1790
  }
1791

1792
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->base.limitInfo);
1,180,575✔
1793
  cleanupQueryTableDataCond(&pInfo->base.cond);
1,180,575✔
1794
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle, true);
1,180,575✔
1795
  if (code) {
1,180,575!
1796
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
1797
    return code;
×
1798
  }
1799
  pInfo->base.cond.startVersion = 0;
1,180,575✔
1800
  pInfo->base.cond.endVersion = pInfo->base.readHandle.version;
1,180,575✔
1801
  if (pTableScanNode->scan.node.dynamicOp && pTableScanNode->scan.virtualStableScan) {
1,180,575!
1802
    cleanupQueryTableDataCond(&pInfo->base.orgCond);
912,789✔
1803
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
912,789!
1804
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
912,789!
1805
  } else {
1806
    // reuse tsdb reader
1807
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
267,786!
1808
      setOperatorCompleted(pOper);
×
1809
      return code;
×
1810
    }
1811

1812
    taosRLockLatch(&pTaskInfo->lock);
267,786✔
1813
    do {
1814
      int32_t        num = 0;
267,786✔
1815
      STableKeyInfo* pList = NULL;
267,786✔
1816
      code = initNextGroupScan(pInfo, &pList, &num);
267,786✔
1817
      if (code) {
267,786!
1818
        qError("%s failed to initNextGroupScan, code:%s", __func__, tstrerror(code));
×
1819
        break;
×
1820
      }
1821
      code = pInfo->base.readerAPI.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
267,786✔
1822
      if (code) {
267,786!
1823
        qError("%s failed to tsdSetQueryTableList, code:%s", __func__, tstrerror(code));
×
1824
        break;
×
1825
      }
1826
      code = pInfo->base.readerAPI.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
267,786✔
1827
      if (code) {
267,786!
1828
        qError("%s failed to tsdReaderResetStatus, code:%s", __func__, tstrerror(code));
×
1829
        break;
×
1830
      }
1831
      pInfo->base.readerAPI.tsdReaderResetVer(pInfo->base.dataReader, &pInfo->base.cond);
267,786✔
1832
      code = pInfo->base.readerAPI.tsdReaderResetExTimeWindow(pInfo->base.dataReader, &pInfo->base.cond);
267,786✔
1833
      if (code) {
267,786!
1834
        qError("%s failed to tsdReaderResetStatus, code:%s", __func__, tstrerror(code));
×
1835
        break;
×
1836
      }
1837
        
1838
    } while (0);
1839
    taosRUnLockLatch(&pTaskInfo->lock);
267,786✔
1840
    if (code) return code;
267,786!
1841

1842
    if (pInfo->filesetDelimited) {
267,786!
1843
      pInfo->base.readerAPI.tsdSetFilesetDelimited(pInfo->base.dataReader);
×
1844
    }
1845

1846
    if (pInfo->pResBlock->info.capacity > pOper->resultInfo.capacity) {
267,786!
1847
      pOper->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1848
    }
1849
  }
1850

1851
  pOper->resultInfo.totalRows = 0;
1,180,575✔
1852
  blockDataEmpty(pInfo->pResBlock);
1,180,575✔
1853
  blockDataEmpty(pInfo->pOrgBlock);
1,180,575✔
1854
  taosHashClear(pInfo->pIgnoreTables);
1,180,575✔
1855
  return code;
1,180,575✔
1856
}
1857

1858
int32_t createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
687,728,318✔
1859
                                    STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
1860
                                    SOperatorInfo** pOptrInfo) {
1861
  QRY_PARAM_CHECK(pOptrInfo);
687,728,318!
1862

1863
  int32_t         code = TSDB_CODE_SUCCESS;
687,757,250✔
1864
  int32_t         lino = 0;
687,757,250✔
1865
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
687,757,250✔
1866
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
687,638,828✔
1867
  if (pInfo == NULL || pOperator == NULL) {
687,628,496!
1868
    code = terrno;
747✔
1869
    goto _error;
×
1870
  }
1871

1872
  pOperator->pPhyNode = pTableScanNode;
687,628,025✔
1873
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
687,630,585✔
1874
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
687,636,513✔
1875

1876
  int32_t numOfCols = 0;
687,638,505✔
1877
  code =
1878
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
687,682,226✔
1879
  QUERY_CHECK_CODE(code, lino, _error);
687,613,508!
1880

1881
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
687,613,508✔
1882
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, true);
687,424,331✔
1883
  QUERY_CHECK_CODE(code, lino, _error);
687,127,161!
1884

1885
  if (pScanNode->pScanPseudoCols != NULL) {
687,127,161✔
1886
    SExprSupp* pSup = &pInfo->base.pseudoSup;
250,807,302✔
1887
    pSup->pExprInfo = NULL;
250,758,459✔
1888
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
250,722,958✔
1889
    QUERY_CHECK_CODE(code, lino, _error);
250,769,860!
1890

1891
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
250,769,860✔
1892
                                      &pTaskInfo->storageAPI.functionStore);
1893
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
250,614,746!
1894
  }
1895

1896
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
686,895,637✔
1897
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
686,919,220✔
1898

1899
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
686,914,985✔
1900
  pInfo->base.readHandle = *readHandle;
686,759,342✔
1901
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
686,585,238✔
1902

1903
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
686,920,092✔
1904
  pInfo->sample.seed = taosGetTimestampSec();
687,037,827✔
1905

1906
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
687,243,667✔
1907
  initResultSizeInfo(&pOperator->resultInfo, 4096);
687,255,419✔
1908
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
686,748,025✔
1909
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
687,672,942✔
1910
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
686,804,637!
1911

1912
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
686,814,552✔
1913
  QUERY_CHECK_CODE(code, lino, _error);
687,395,197!
1914

1915
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
687,395,197!
1916
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
687,231,941!
1917
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
13,139,087✔
1918
    pInfo->pResBlock = NULL;
13,141,996✔
1919
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
13,134,722!
1920
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
13,139,198!
1921
  }
1922

1923
  if (pInfo->virtualStableScan) {
687,121,113!
1924
    pInfo->readerCache = taosHashInit(1024, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
13,139,150✔
1925
    QUERY_CHECK_NULL(pInfo->readerCache, code, lino, _error, terrno);
13,146,409!
1926
  }
1927

1928
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
687,036,381✔
1929
                            pTaskInfo->pStreamRuntimeInfo);
686,680,737✔
1930
  QUERY_CHECK_CODE(code, lino, _error);
687,054,677✔
1931

1932
  pInfo->currentGroupId = -1;
687,001,109✔
1933

1934
  pInfo->tableEndIndex = -1;
687,024,512✔
1935
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
686,793,360!
1936
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
687,004,670✔
1937

1938
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
686,749,754✔
1939
                  pTaskInfo);
1940
  pOperator->exprSupp.numOfExprs = numOfCols;
686,712,227✔
1941

1942
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
686,791,508!
1943
  pInfo->ignoreTag = false;
686,793,874✔
1944

1945
  pInfo->base.pTableListInfo = pTableListInfo;
687,031,390✔
1946
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
686,785,809✔
1947
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
687,251,566!
1948
    code = terrno;
×
1949
    QUERY_CHECK_CODE(code, lino, _error);
×
1950
  }
1951

1952
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
687,419,373!
1953

1954
  taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
687,243,018✔
1955
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
687,581,567✔
1956
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
1957

1958
  setOperatorResetStateFn(pOperator, resetTableScanOperatorState); 
687,315,176✔
1959
  // for non-blocking operator, the open cost is always 0
1960
  pOperator->cost.openCost = 0;
687,025,167✔
1961
  *pOptrInfo = pOperator;
687,047,978✔
1962
  return TSDB_CODE_SUCCESS;
686,850,282✔
1963

1964
_error:
53,568✔
1965
  if (pInfo != NULL) {
53,568!
1966
    pInfo->base.pTableListInfo = NULL;  // this attribute will be destroy outside of this function
53,568✔
1967
    destroyTableScanOperatorInfo(pInfo);
53,568✔
1968
  }
1969

1970
  if (pOperator != NULL) {
53,568!
1971
    pOperator->info = NULL;
53,568✔
1972
    destroyOperator(pOperator);
53,568✔
1973
  }
1974
  pTaskInfo->code = code;
53,568✔
1975
  return code;
53,568✔
1976
}
1977

1978
int32_t createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
×
1979
  QRY_PARAM_CHECK(pOptrInfo);
×
1980

1981
  int32_t         code = 0;
×
1982
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
×
1983
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
1984
  if (pInfo == NULL || pOperator == NULL) {
×
1985
    code = terrno;
×
1986
    goto _end;
×
1987
  }
1988

1989
  pInfo->base.dataReader = pReadHandle;
×
1990
  //  pInfo->prevGroupId       = -1;
1991

1992
  setOperatorInfo(pOperator, "TableSeqScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN, false, OP_NOT_OPENED,
×
1993
                  pInfo, pTaskInfo);
1994
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanImplNext, NULL, NULL, optrDefaultBufFn, NULL,
×
1995
                                         optrDefaultGetNextExtFn, NULL);
1996
  *pOptrInfo = pOperator;
×
1997
  return code;
×
1998

1999
_end:
×
2000
  if (pInfo != NULL) {
×
2001
    taosMemoryFree(pInfo);
×
2002
  }
2003

2004
  if (pOperator != NULL) {
×
2005
    taosMemoryFree(pOperator);
×
2006
  }
2007

2008
  pTaskInfo->code = code;
×
2009
  return code;
×
2010
}
2011

2012
static int32_t doBlockDataPrimaryKeyFilter(SSDataBlock* pBlock, STqOffsetVal* offset) {
×
2013
  int32_t code = TSDB_CODE_SUCCESS;
×
2014
  int32_t lino = 0;
×
2015
  if (pBlock->info.window.skey != offset->ts || offset->primaryKey.type == 0) {
×
2016
    return code;
×
2017
  }
2018
  bool* p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
×
2019
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
×
2020
  bool hasUnqualified = false;
×
2021

2022
  SColumnInfoData* pColTs = taosArrayGet(pBlock->pDataBlock, 0);
×
2023
  SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
2024

2025
  qDebug("doBlockDataWindowFilter primary key, ts:%" PRId64 " %" PRId64, offset->ts,
×
2026
         VALUE_GET_TRIVIAL_DATUM(&offset->primaryKey));
2027
  QUERY_CHECK_CONDITION((pColPk->info.type == offset->primaryKey.type), code, lino, _end,
×
2028
                        TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
2029

2030
  __compar_fn_t func = getComparFunc(pColPk->info.type, 0);
×
2031
  QUERY_CHECK_NULL(func, code, lino, _end, terrno);
×
2032
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
×
2033
    int64_t* ts = (int64_t*)colDataGetData(pColTs, i);
×
2034
    void*    data = colDataGetData(pColPk, i);
×
2035
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
2036
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
×
2037
        QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY, lino, _end);
×
2038
      }
2039
      void* tmq = taosMemoryMalloc(offset->primaryKey.nData + VARSTR_HEADER_SIZE);
×
2040
      QUERY_CHECK_NULL(tmq, code, lino, _end, terrno);
×
2041
      memcpy(varDataVal(tmq), offset->primaryKey.pData, offset->primaryKey.nData);
×
2042
      varDataLen(tmq) = offset->primaryKey.nData;
×
2043
      p[i] = (*ts > offset->ts) || (func(data, tmq) > 0);
×
2044
      taosMemoryFree(tmq);
×
2045
    } else {
2046
      p[i] = (*ts > offset->ts) || (func(data, VALUE_GET_DATUM(&offset->primaryKey, pColPk->info.type)) > 0);
×
2047
    }
2048

2049
    if (!p[i]) {
×
2050
      hasUnqualified = true;
×
2051
    }
2052
  }
2053

2054
  if (hasUnqualified) {
×
2055
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
×
2056
    QUERY_CHECK_CODE(code, lino, _end);
×
2057
  }
2058

2059
_end:
×
2060
  taosMemoryFree(p);
×
2061
  if (code != TSDB_CODE_SUCCESS) {
×
2062
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2063
  }
2064
  return code;
×
2065
}
2066

2067
int32_t colIdComparFn(const void* param1, const void* param2) {
2,147,483,647✔
2068
  int32_t p1 = *(int32_t*)param1;
2,147,483,647✔
2069
  int32_t p2 = *(int32_t*)param2;
2,147,483,647✔
2070

2071
  if (p1 == p2) {
2,147,483,647!
2072
    return 0;
×
2073
  } else {
2074
    return (p1 < p2) ? -1 : 1;
2,147,483,647✔
2075
  }
2076
}
2077

2078
static int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock) {
211,582,119✔
2079
  int32_t         code = TSDB_CODE_SUCCESS;
211,582,119✔
2080
  int32_t         lino = 0;
211,582,119✔
2081
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
211,582,119✔
2082
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
211,583,933✔
2083
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
211,580,305✔
2084
  const char*     id = GET_TASKID(pTaskInfo);
211,579,398✔
2085
  bool            isVtableSourceScan = (pTaskInfo->pSubplan->pVTables != NULL);
211,583,933✔
2086

2087
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
211,581,212✔
2088
  QUERY_CHECK_CODE(code, lino, _end);
211,581,212!
2089

2090
  pBlockInfo->rows = pBlock->info.rows;
211,581,212✔
2091
  pBlockInfo->id.uid = pBlock->info.id.uid;
211,573,049✔
2092
  pBlockInfo->type = STREAM_NORMAL;
211,581,212✔
2093
  pBlockInfo->version = pBlock->info.version;
211,582,119✔
2094

2095
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
211,564,159✔
2096
  if (!isVtableSourceScan) {
211,580,305!
2097
    pBlockInfo->id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
211,580,305✔
2098
  } else {
2099
    // use original table uid as groupId for vtable
2100
    pBlockInfo->id.groupId = pBlock->info.id.groupId;
×
2101
  }
2102

2103
  SArray* pColList = taosArrayInit(4, sizeof(int32_t));
211,579,398✔
2104
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
211,577,608!
2105

2106
  // todo extract method
2107
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
1,219,954,055✔
2108
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
1,008,334,942✔
2109
    if (!pColMatchInfo->needOutput) {
1,008,281,765!
2110
      continue;
1,682✔
2111
    }
2112

2113
    bool colExists = false;
1,008,283,214✔
2114
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
2,147,483,647!
2115
      SColumnInfoData* pResCol = NULL;
2,147,483,647✔
2116
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
2,147,483,647✔
2117
      QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
2118

2119
      if (pResCol->info.colId == pColMatchInfo->colId) {
2,147,483,647✔
2120
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
1,008,026,527✔
2121
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
1,008,380,203✔
2122
        QUERY_CHECK_CODE(code, lino, _end);
1,008,111,873!
2123

2124
        colExists = true;
1,008,111,873✔
2125
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
1,008,111,873✔
2126
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,008,376,809!
2127
        break;
1,008,376,809✔
2128
      }
2129
    }
2130

2131
    // the required column does not exists in submit block, let's set it to be all null value
2132
    if (!colExists) {
1,008,373,869✔
2133
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
3,356✔
2134
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
3,356!
2135
      void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
3,356✔
2136
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,356!
2137
    }
2138
  }
2139

2140
  // currently only the tbname pseudo column
2141
  if (pInfo->numOfPseudoExpr > 0 && !isVtableSourceScan) {
211,580,316!
2142
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
231,451,737✔
2143
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
115,725,921✔
2144
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
2145
    if (code) {
115,721,386!
2146
      QUERY_CHECK_CODE(code, lino, _end);
×
2147
    }
2148

2149
    // reset the error code.
2150
    terrno = 0;
115,721,386✔
2151

2152
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
598,473,356✔
2153
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
482,762,854✔
2154
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
482,770,110!
2155
    }
2156
  }
2157

2158
  taosArraySort(pColList, colIdComparFn);
211,534,966✔
2159

2160
  int32_t i = 0, j = 0;
211,569,677✔
2161
  while (i < taosArrayGetSize(pColList)) {
1,702,270,886✔
2162
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
1,490,636,634✔
2163
    if (slot1 > j) {
1,490,701,209!
2164
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
×
2165
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
2166
      j += 1;
×
2167
    } else {
2168
      i += 1;
1,490,701,209✔
2169
      j += 1;
1,490,701,209✔
2170
    }
2171
  }
2172

2173
  while (j < taosArrayGetSize(pInfo->pRes->pDataBlock)) {
211,566,656✔
2174
    SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
1,682✔
2175
    colDataSetNNULL(pDst, 0, pBlockInfo->rows);
1,682!
2176
    j += 1;
1,682✔
2177
  }
2178

2179
  taosArrayDestroy(pColList);
211,569,616✔
2180

2181
  code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL, NULL);
211,576,855✔
2182
  QUERY_CHECK_CODE(code, lino, _end);
211,559,276!
2183

2184
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
211,559,276✔
2185
  QUERY_CHECK_CODE(code, lino, _end);
211,568,819!
2186
  if (pInfo->pRes->info.rows == 0) {
211,568,819✔
2187
    return 0;
2,046,660✔
2188
  }
2189

2190
_end:
209,521,707✔
2191
  if (code != TSDB_CODE_SUCCESS) {
205,589,696!
2192
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2193
  }
2194
  return code;
209,511,830✔
2195
}
2196

2197
static int32_t processPrimaryKey(SSDataBlock* pBlock, bool hasPrimaryKey, STqOffsetVal* offset) {
18,646,231✔
2198
  int32_t code = TSDB_CODE_SUCCESS;
18,646,231✔
2199
  int32_t lino = 0;
18,646,231✔
2200
  SValue  val = {0};
18,646,231✔
2201
  if (hasPrimaryKey) {
18,654,797!
2202
    code = doBlockDataPrimaryKeyFilter(pBlock, offset);
×
2203
    if (code != TSDB_CODE_SUCCESS) {
×
2204
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2205
      return code;
×
2206
    }
2207
    SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
2208

2209
    if (pBlock->info.rows < 1) {
×
2210
      return code;
×
2211
    }
2212
    void* tmp = colDataGetData(pColPk, pBlock->info.rows - 1);
×
2213
    val.type = pColPk->info.type;
×
2214
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
2215
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
×
2216
        return TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY;
×
2217
      }
2218
      val.pData = taosMemoryMalloc(varDataLen(tmp));
×
2219
      QUERY_CHECK_NULL(val.pData, code, lino, _end, terrno);
×
2220
      val.nData = varDataLen(tmp);
×
2221
      memcpy(val.pData, varDataVal(tmp), varDataLen(tmp));
×
2222
    } else {
2223
      valueSetDatum(&val, pColPk->info.type, tmp, pColPk->info.bytes);
×
2224
    }
2225
  }
2226
  tqOffsetResetToData(offset, pBlock->info.id.uid, pBlock->info.window.ekey, val);
18,654,797!
2227

2228
_end:
18,636,752✔
2229
  if (code != TSDB_CODE_SUCCESS) {
18,641,054!
2230
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2231
  }
2232
  return code;
18,641,815✔
2233
}
2234

2235
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
261,751,160✔
2236
  int32_t        code = TSDB_CODE_SUCCESS;
261,751,160✔
2237
  int32_t        lino = 0;
261,751,160✔
2238
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
261,751,160✔
2239
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
261,770,113✔
2240

2241
  SStreamScanInfo* pInfo = pOperator->info;
261,784,458✔
2242
  const char*      id = GET_TASKID(pTaskInfo);
261,781,676✔
2243

2244
  qDebug("start to exec queue scan, %s", id);
261,780,568✔
2245

2246
  if (isTaskKilled(pTaskInfo)) {
261,785,221!
2247
    (*ppRes) = NULL;
×
2248
    return pTaskInfo->code;
×
2249
  }
2250

2251
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
261,787,992✔
2252
    while (1) {
×
2253
      SSDataBlock* pResult = NULL;
16,651,525✔
2254
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
16,651,525✔
2255
      QUERY_CHECK_CODE(code, lino, _end);
16,638,505!
2256

2257
      if (pResult && pResult->info.rows > 0) {
16,638,505!
2258
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
16,513,807✔
2259
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
16,516,411✔
2260
        QUERY_CHECK_CODE(code, lino, _end);
16,496,473!
2261
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
16,496,473!
2262
        if (pResult->info.rows > 0) {
16,517,261!
2263
          (*ppRes) = pResult;
16,522,443✔
2264
          return code;
16,521,575✔
2265
        }
2266
      } else {
2267
        break;
2268
      }
2269
    }
2270

2271
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
129,082✔
2272
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
129,082✔
2273

2274
    pTSInfo->base.dataReader = NULL;
129,082✔
2275
    int64_t validVer = pTaskInfo->streamInfo.snapshotVer + 1;
129,082✔
2276
    qDebug("queue scan tsdb over, switch to wal ver %" PRId64, validVer);
129,082!
2277
    if (pAPI->tqReaderFn.tqReaderSeek(pInfo->tqReader, validVer, pTaskInfo->id.str) < 0) {
129,082✔
2278
      (*ppRes) = NULL;
40,245✔
2279
      return code;
40,245✔
2280
    }
2281

2282
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, validVer);
88,837✔
2283
  }
2284

2285
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
245,224,397!
2286
    while (1) {
2,046,660✔
2287
      bool hasResult =
2288
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
247,269,243✔
2289

2290
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
247,232,277✔
2291
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
247,266,581✔
2292

2293
      // curVersion move to next
2294
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pWalReader->curVersion);
247,251,322✔
2295

2296
      // use ts to pass time when replay, because ts not used if type is log
2297
      pTaskInfo->streamInfo.currentOffset.ts = pAPI->tqReaderFn.tqGetResultBlockTime(pInfo->tqReader);
247,240,432✔
2298

2299
      if (hasResult) {
247,243,142✔
2300
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
211,553,297✔
2301
               pTaskInfo->streamInfo.currentOffset.version);
2302
        blockDataCleanup(pInfo->pRes);
211,581,201✔
2303
        code = setBlockIntoRes(pInfo, pRes);
211,571,235✔
2304
        QUERY_CHECK_CODE(code, lino, _end);
211,552,153!
2305
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
211,552,153!
2306
               pTaskInfo->streamInfo.currentOffset.version);
2307
        if (pInfo->pRes->info.rows > 0) {
211,572,181✔
2308
          (*ppRes) = pInfo->pRes;
209,535,459✔
2309
          return code;
209,530,924✔
2310
        }
2311
      } else {
2312
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
35,689,845!
2313
        (*ppRes) = NULL;
35,689,021✔
2314
        return code;
35,689,021✔
2315
      }
2316
    }
2317
  } else {
2318
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
×
2319
    (*ppRes) = NULL;
×
2320
    return code;
×
2321
  }
2322

2323
_end:
×
2324
  if (code != TSDB_CODE_SUCCESS) {
×
2325
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2326
    pTaskInfo->code = code;
×
2327
    T_LONG_JMP(pTaskInfo->env, code);
×
2328
  }
2329
  (*ppRes) = NULL;
×
2330
  return code;
×
2331
}
2332

2333
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
×
2334
  SSDataBlock* pRes = NULL;
×
2335
  int32_t      code = doQueueScanNext(pOperator, &pRes);
×
2336
  return pRes;
×
2337
}
2338

2339
int32_t extractTableIdList(const STableListInfo* pTableListInfo, SArray** ppArrayRes) {
901,023✔
2340
  int32_t code = TSDB_CODE_SUCCESS;
901,023✔
2341
  int32_t lino = 0;
901,023✔
2342
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));
901,023✔
2343
  QUERY_CHECK_NULL(tableIdList, code, lino, _end, terrno);
907,171!
2344

2345
  // Transfer the Array of STableKeyInfo into uid list.
2346
  int32_t size = 0;
907,171✔
2347
  code = tableListGetSize(pTableListInfo, &size);
907,171✔
2348
  QUERY_CHECK_CODE(code, lino, _end);
908,031!
2349
  for (int32_t i = 0; i < size; ++i) {
27,787,612✔
2350
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
26,878,722✔
2351
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
26,878,722!
2352
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
26,878,722✔
2353
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
26,879,581!
2354
  }
2355

2356
  (*ppArrayRes) = tableIdList;
908,890✔
2357

2358
_end:
907,171✔
2359
  if (code != TSDB_CODE_SUCCESS) {
908,031!
2360
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2361
  }
2362
  return code;
908,031✔
2363
}
2364

2365
static int32_t doRawScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
4,308,414✔
2366
  int32_t        code = TSDB_CODE_SUCCESS;
4,308,414✔
2367
  int32_t        lino = 0;
4,308,414✔
2368
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4,308,414✔
2369
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
4,309,263✔
2370

2371
  SStreamRawScanInfo* pInfo = pOperator->info;
4,308,414✔
2372
  pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = NULL;  // use batchMetaReq != NULL to judge if data is meta
4,308,414✔
2373
  pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = NULL;
4,308,414✔
2374

2375
  qDebug("tmqsnap doRawScan called");
4,307,565!
2376
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
4,309,263✔
2377
    bool hasNext = false;
4,293,983✔
2378
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
4,293,983!
2379
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
4,290,647✔
2380
      if (code != TSDB_CODE_SUCCESS) {
4,265,998!
2381
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2382
        QUERY_CHECK_CODE(code, lino, _end);
×
2383
      }
2384
    }
2385

2386
    if (pInfo->dataReader && hasNext) {
4,269,334!
2387
      if (isTaskKilled(pTaskInfo)) {
2,153,775!
2388
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2389
        return code;
×
2390
      }
2391

2392
      SSDataBlock* pBlock = NULL;
2,152,926✔
2393
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock, NULL);
2,153,775✔
2394
      QUERY_CHECK_CODE(code, lino, _end);
2,141,889!
2395

2396
      if (pBlock && pBlock->info.rows > 0) {
2,141,889!
2397
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
2,152,077✔
2398
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
2,147,832✔
2399
        QUERY_CHECK_CODE(code, lino, _end);
2,144,436!
2400
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
2,144,436✔
2401
        (*ppRes) = pBlock;
2,162,265✔
2402
        return code;
2,156,322✔
2403
      }
2404
    }
2405

2406
    SMetaTableInfo mtInfo = {0};
2,120,653✔
2407
    code = pAPI->snapshotFn.getMetaTableInfoFromSnapshot(pInfo->sContext, &mtInfo);
2,135,963✔
2408
    QUERY_CHECK_CODE(code, lino, _end);
2,122,351!
2409
    if (code != 0) {
2,122,351!
2410
      destroyMetaTableInfo(&mtInfo);
2411
      QUERY_CHECK_CODE(code, lino, _end);
×
2412
    }
2413
    STqOffsetVal offset = {0};
2,122,351✔
2414
    if (mtInfo.uid == 0 || pInfo->sContext->withMeta == ONLY_META) {  // read snapshot done, change to get data from wal
2,119,818✔
2415
      qDebug("tmqsnap read snapshot done, change to get data from wal");
39,296✔
2416
      tqOffsetResetToLog(&offset, pInfo->sContext->snapVersion + 1);
39,296✔
2417
    } else {
2418
      SValue val = {0};
2,097,572✔
2419
      tqOffsetResetToData(&offset, mtInfo.uid, INT64_MIN, val);
2,097,572!
2420
      qDebug("tmqsnap change get data uid:%" PRId64, mtInfo.uid);
2,100,982!
2421
    }
2422
    destroyMetaTableInfo(&mtInfo);
2423
    code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
2,137,661✔
2424
    QUERY_CHECK_CODE(code, lino, _end);
2,137,661!
2425
    (*ppRes) = NULL;
2,137,661✔
2426
    return code;
2,137,661✔
2427
  } else if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_META) {
15,280!
2428
    SSnapContext* sContext = pInfo->sContext;
15,280✔
2429
    for (int32_t i = 0; i < tmqRowSize; i++) {
127,676!
2430
      void*   data = NULL;
127,676✔
2431
      int32_t dataLen = 0;
127,676✔
2432
      int16_t type = 0;
127,676✔
2433
      int64_t uid = 0;
127,676✔
2434
      if (pAPI->snapshotFn.getTableInfoFromSnapshot(sContext, &data, &dataLen, &type, &uid) < 0) {
127,676!
2435
        qError("tmqsnap getTableInfoFromSnapshot error");
×
2436
        taosMemoryFreeClear(data);
×
2437
        break;
×
2438
      }
2439

2440
      if (!sContext->queryMeta) {  // change to get data next poll request
127,676✔
2441
        STqOffsetVal offset = {0};
15,280✔
2442
        SValue       val = {0};
15,280!
2443
        tqOffsetResetToData(&offset, 0, INT64_MIN, val);
×
2444
        code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
15,280✔
2445
        QUERY_CHECK_CODE(code, lino, _end);
15,280!
2446
        break;
15,280✔
2447
      } else {
2448
        tqOffsetResetToMeta(&pTaskInfo->streamInfo.currentOffset, uid);
112,396✔
2449
        SMqMetaRsp tmpMetaRsp = {0};
112,396✔
2450
        tmpMetaRsp.resMsgType = type;
112,396✔
2451
        tmpMetaRsp.metaRspLen = dataLen;
112,396✔
2452
        tmpMetaRsp.metaRsp = data;
112,396✔
2453
        if (!pTaskInfo->streamInfo.btMetaRsp.batchMetaReq) {
112,396✔
2454
          pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = taosArrayInit(4, POINTER_BYTES);
13,662✔
2455
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, code, lino, _end, terrno);
13,662!
2456

2457
          pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = taosArrayInit(4, sizeof(int32_t));
13,662✔
2458
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, code, lino, _end, terrno);
13,662!
2459
        }
2460
        int32_t  tempRes = TSDB_CODE_SUCCESS;
112,396✔
2461
        uint32_t len = 0;
112,396✔
2462
        tEncodeSize(tEncodeMqMetaRsp, &tmpMetaRsp, len, tempRes);
112,396!
2463
        if (TSDB_CODE_SUCCESS != tempRes) {
112,396!
2464
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2465
          taosMemoryFreeClear(data);
×
2466
          break;
×
2467
        }
2468

2469
        int32_t tLen = sizeof(SMqRspHead) + len;
112,396✔
2470
        void*   tBuf = taosMemoryCalloc(1, tLen);
112,396!
2471
        QUERY_CHECK_NULL(tBuf, code, lino, _end, terrno);
111,587!
2472

2473
        void*    metaBuff = POINTER_SHIFT(tBuf, sizeof(SMqRspHead));
111,587✔
2474
        SEncoder encoder = {0};
111,587✔
2475
        tEncoderInit(&encoder, metaBuff, len);
111,587✔
2476
        int32_t tempLen = tEncodeMqMetaRsp(&encoder, &tmpMetaRsp);
111,587✔
2477
        if (tempLen < 0) {
112,396!
2478
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2479
          tEncoderClear(&encoder);
×
2480
          taosMemoryFreeClear(tBuf);
×
2481
          taosMemoryFreeClear(data);
×
2482
          break;
×
2483
        }
2484
        taosMemoryFreeClear(data);
112,396!
2485
        void* tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, &tBuf);
112,396✔
2486
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
112,396!
2487

2488
        tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, &tLen);
112,396✔
2489
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
112,396!
2490
      }
2491
    }
2492

2493
    (*ppRes) = NULL;
15,280✔
2494
    return code;
15,280✔
2495
  }
2496

2497
_end:
×
2498
  if (code != TSDB_CODE_SUCCESS) {
×
2499
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2500
    pTaskInfo->code = code;
×
2501
    T_LONG_JMP(pTaskInfo->env, code);
×
2502
  }
2503

2504
  (*ppRes) = NULL;
×
2505
  return code;
×
2506
}
2507

2508
static void destroyTmqRawScanOperatorInfo(void* param) {
290,731✔
2509
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
290,731✔
2510
  pRawScan->pAPI->tsdReader.tsdReaderClose(pRawScan->dataReader);
290,731✔
2511
  pRawScan->pAPI->snapshotFn.destroySnapshot(pRawScan->sContext);
292,478✔
2512
  tableListDestroy(pRawScan->pTableListInfo);
291,615✔
2513
  taosMemoryFree(pRawScan);
292,478!
2514
}
292,478✔
2515

2516
// for subscribing db or stb (not including column),
2517
// if this scan is used, meta data can be return
2518
// and schemas are decided when scanning
2519
int32_t createTmqRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
289,840✔
2520
  // create operator
2521
  // create tb reader
2522
  // create meta reader
2523
  // create tq reader
2524

2525
  QRY_PARAM_CHECK(pOptrInfo);
289,840!
2526
  int32_t code = TSDB_CODE_SUCCESS;
292,478✔
2527
  int32_t lino = 0;
292,478✔
2528

2529
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
292,478!
2530
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
292,478!
2531
  if (pInfo == NULL || pOperator == NULL) {
292,478!
2532
    code = terrno;
×
2533
    lino = __LINE__;
×
2534
    goto _end;
×
2535
  }
2536

2537
  pInfo->pTableListInfo = tableListCreate();
292,478✔
2538
  QUERY_CHECK_NULL(pInfo->pTableListInfo, code, lino, _end, terrno);
292,478!
2539
  pInfo->vnode = pHandle->vnode;
292,478✔
2540
  pInfo->pAPI = &pTaskInfo->storageAPI;
292,478✔
2541

2542
  pInfo->sContext = pHandle->sContext;
292,478✔
2543
  setOperatorInfo(pOperator, "RawScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
292,478✔
2544
                  pTaskInfo);
2545

2546
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScanNext, NULL, destroyTmqRawScanOperatorInfo, optrDefaultBufFn,
292,478✔
2547
                                         NULL, optrDefaultGetNextExtFn, NULL);
2548
  *pOptrInfo = pOperator;
292,478✔
2549
  return code;
292,478✔
2550

2551
_end:
×
2552
  if (code != TSDB_CODE_SUCCESS) {
×
2553
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2554
  }
2555
  taosMemoryFree(pInfo);
×
2556
  taosMemoryFree(pOperator);
×
2557
  pTaskInfo->code = code;
×
2558
  return code;
×
2559
}
2560

2561
void destroyTmqScanOperatorInfo(void* param) {
908,031✔
2562
  if (param == NULL) {
908,031!
2563
    return;
×
2564
  }
2565

2566
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
908,031✔
2567
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
908,031!
2568
    destroyOperator(pStreamScan->pTableScanOp);
908,031✔
2569
  }
2570

2571
  if (pStreamScan->tqReader != NULL && pStreamScan->readerFn.tqReaderClose != NULL) {
908,031!
2572
    pStreamScan->readerFn.tqReaderClose(pStreamScan->tqReader);
908,031✔
2573
  }
2574
  if (pStreamScan->pVtableMergeHandles) {
908,031!
2575
    taosHashCleanup(pStreamScan->pVtableMergeHandles);
×
2576
    pStreamScan->pVtableMergeHandles = NULL;
×
2577
  }
2578
  if (pStreamScan->pVtableMergeBuf) {
908,031!
2579
    destroyDiskbasedBuf(pStreamScan->pVtableMergeBuf);
×
2580
    pStreamScan->pVtableMergeBuf = NULL;
×
2581
  }
2582
  if (pStreamScan->pVtableReadyHandles) {
908,031!
2583
    taosArrayDestroy(pStreamScan->pVtableReadyHandles);
×
2584
    pStreamScan->pVtableReadyHandles = NULL;
×
2585
  }
2586
  if (pStreamScan->pTableListInfo) {
908,031!
2587
    tableListDestroy(pStreamScan->pTableListInfo);
×
2588
    pStreamScan->pTableListInfo = NULL;
×
2589
  }
2590
  if (pStreamScan->matchInfo.pList) {
908,031!
2591
    taosArrayDestroy(pStreamScan->matchInfo.pList);
908,031✔
2592
  }
2593
  if (pStreamScan->pPseudoExpr) {
908,031✔
2594
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
524,902✔
2595
    taosMemoryFree(pStreamScan->pPseudoExpr);
524,902!
2596
  }
2597

2598
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
908,031✔
2599
  cleanupExprSupp(&pStreamScan->tagCalSup);
908,031✔
2600

2601
  blockDataDestroy(pStreamScan->pRes);
908,031✔
2602
  blockDataDestroy(pStreamScan->pUpdateRes);
908,031✔
2603
  blockDataDestroy(pStreamScan->pDeleteDataRes);
908,031✔
2604
  blockDataDestroy(pStreamScan->pUpdateDataRes);
908,031✔
2605
  blockDataDestroy(pStreamScan->pCreateTbRes);
908,031✔
2606
  taosArrayDestroy(pStreamScan->pBlockLists);
907,222✔
2607
  blockDataDestroy(pStreamScan->pCheckpointRes);
908,031✔
2608

2609
  taosMemoryFree(pStreamScan);
908,031!
2610
}
2611

2612
int32_t addPrimaryKeyCol(SSDataBlock* pBlock, uint8_t type, int32_t bytes) {
×
2613
  int32_t code = TSDB_CODE_SUCCESS;
×
2614
  int32_t lino = 0;
×
2615
  pBlock->info.rowSize += bytes;
×
2616
  SColumnInfoData infoData = {0};
×
2617
  infoData.info.type = type;
×
2618
  infoData.info.bytes = bytes;
×
2619
  void* tmp = taosArrayPush(pBlock->pDataBlock, &infoData);
×
2620
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
2621

2622
_end:
×
2623
  if (code != TSDB_CODE_SUCCESS) {
×
2624
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2625
  }
2626
  return code;
×
2627
}
2628

2629
static SSDataBlock* createStreamVtableBlock(SColMatchInfo* pMatchInfo, const char* idstr) {
×
2630
  int32_t      code = TSDB_CODE_SUCCESS;
×
2631
  int32_t      lino = 0;
×
2632
  SSDataBlock* pRes = NULL;
×
2633

2634
  QUERY_CHECK_NULL(pMatchInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
2635

2636
  code = createDataBlock(&pRes);
×
2637
  QUERY_CHECK_CODE(code, lino, _end);
×
2638
  int32_t numOfOutput = taosArrayGetSize(pMatchInfo->pList);
×
2639
  for (int32_t i = 0; i < numOfOutput; ++i) {
×
2640
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
×
2641
    if (!pItem->needOutput) {
×
2642
      continue;
×
2643
    }
2644
    SColumnInfoData colInfo = createColumnInfoData(pItem->dataType.type, pItem->dataType.bytes, pItem->colId);
×
2645
    code = blockDataAppendColInfo(pRes, &colInfo);
×
2646
    QUERY_CHECK_CODE(code, lino, _end);
×
2647
  }
2648

2649
_end:
×
2650
  if (code != TSDB_CODE_SUCCESS) {
×
2651
    qError("%s failed at line %d since %s, id: %s", __func__, lino, tstrerror(code), idstr);
×
2652
    if (pRes != NULL) {
×
2653
      blockDataDestroy(pRes);
×
2654
    }
2655
    pRes = NULL;
×
2656
    terrno = code;
×
2657
  }
2658
  return pRes;
×
2659
}
2660

2661
 int32_t createTmqScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode,
905,489✔
2662
                                                  SNode* pTagCond, STableListInfo* pTableListInfo,
2663
                                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
2664
  QRY_PARAM_CHECK(pOptrInfo);
905,489!
2665

2666
  int32_t          code = TSDB_CODE_SUCCESS;
908,031✔
2667
  int32_t          lino = 0;
908,031✔
2668
  SArray*          pColIds = NULL;
908,031✔
2669
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
908,031!
2670
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
908,031!
2671
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
908,031✔
2672
  const char*      idstr = pTaskInfo->id.str;
908,031✔
2673
  SSHashObj*       pVtableInfos = pTaskInfo->pSubplan->pVTables;
908,031✔
2674

2675
  if (pInfo == NULL || pOperator == NULL) {
908,031!
2676
    code = terrno;
×
2677
    goto _error;
×
2678
  }
2679

2680
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
908,031✔
2681
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
908,031✔
2682

2683
  pInfo->pTagCond = pTagCond;
908,031✔
2684
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
908,031✔
2685

2686
  int32_t numOfCols = 0;
908,031✔
2687
  code = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
908,031✔
2688
  if (code != TSDB_CODE_SUCCESS) {
908,031!
2689
    goto _error;
×
2690
  }
2691

2692
  SDataType pkType = {0};
908,031✔
2693
  pInfo->primaryKeyIndex = -1;
908,031✔
2694
  pInfo->basic.primaryPkIndex = -1;
908,031✔
2695
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
908,031✔
2696
  pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
908,031✔
2697
  QUERY_CHECK_NULL(pColIds, code, lino, _error, terrno);
908,031!
2698

2699
  for (int32_t i = 0; i < numOfOutput; ++i) {
6,652,067✔
2700
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
5,744,036✔
2701
    QUERY_CHECK_NULL(id, code, lino, _error, terrno);
5,744,036!
2702

2703
    int16_t colId = id->colId;
5,744,036✔
2704
    void*   tmp = taosArrayPush(pColIds, &colId);
5,744,036✔
2705
    QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
5,744,036!
2706

2707
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
5,744,036✔
2708
      pInfo->primaryTsIndex = id->dstSlotId;
908,031✔
2709
    }
2710
    if (id->isPk) {
5,744,036!
2711
      pInfo->primaryKeyIndex = id->dstSlotId;
×
2712
      pInfo->basic.primaryPkIndex = id->dstSlotId;
×
2713
      pkType = id->dataType;
×
2714
    }
2715
  }
2716

2717
  pInfo->pPartTbnameSup = NULL;
908,031✔
2718
  if (pTableScanNode->pSubtable != NULL) {
908,031!
2719
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
×
2720
    if (pSubTableExpr == NULL) {
×
2721
      code = terrno;
×
2722
      goto _error;
×
2723
    }
2724

2725
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
×
2726
    code = createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
×
2727
    QUERY_CHECK_CODE(code, lino, _error);
×
2728

2729
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1, &pTaskInfo->storageAPI.functionStore) != 0) {
×
2730
      goto _error;
×
2731
    }
2732
  }
2733

2734
  if (pTableScanNode->pTags != NULL) {
908,031!
2735
    int32_t    numOfTags;
×
2736
    SExprInfo* pTagExpr = createExpr(pTableScanNode->pTags, &numOfTags);
×
2737
    if (pTagExpr == NULL) {
×
2738
      goto _error;
×
2739
    }
2740
    code = initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags, &pTaskInfo->storageAPI.functionStore);
×
2741
    if (code != 0) {
×
2742
      goto _error;
×
2743
    }
2744
  }
2745

2746
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
908,031✔
2747
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
908,031!
2748

2749
  if (pHandle->vnode) {
908,031!
2750
    SOperatorInfo* pTableScanOp = NULL;
908,031✔
2751
    code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pTableScanOp);
908,031✔
2752
    if (pTableScanOp == NULL || code != 0) {
907,171!
2753
      qError("createTableScanOperatorInfo error, code:%d", pTaskInfo->code);
2,149!
2754
      goto _error;
×
2755
    }
2756

2757
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
905,022✔
2758
    if (pHandle->version > 0) {
908,031!
2759
      pTSInfo->base.cond.endVersion = pHandle->version;
908,031✔
2760
    }
2761

2762
    STableKeyInfo* pList = NULL;
904,154✔
2763
    int32_t        num = 0;
906,718✔
2764
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
906,327✔
2765
    QUERY_CHECK_CODE(code, lino, _error);
907,171!
2766

2767
    if (pHandle->initTableReader) {
907,171✔
2768
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
906,303✔
2769
      pTSInfo->base.dataReader = NULL;
907,171✔
2770
    }
2771

2772
    if (pHandle->initTqReader) {
905,890!
2773
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
906,327✔
2774
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
908,031!
2775
    } else {
2776
      pInfo->tqReader = pHandle->tqReader;
×
2777
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
2778
    }
2779

2780
    if (pVtableInfos != NULL) {
908,031!
2781
      // save vtable info into tqReader for vtable source scan
2782
      SSDataBlock* pResBlock = createStreamVtableBlock(&pInfo->matchInfo, idstr);
×
2783
      QUERY_CHECK_CODE(code, lino, _error);
×
2784
      code = pAPI->tqReaderFn.tqReaderSetVtableInfo(pInfo->tqReader, pHandle->vnode, pAPI, pVtableInfos, &pResBlock,
×
2785
                                                    idstr);
2786
      QUERY_CHECK_CODE(code, lino, _error);
×
2787
    }
2788

2789
    pInfo->pUpdateInfo = NULL;
908,031✔
2790
    pInfo->pTableScanOp = pTableScanOp;
908,031✔
2791
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
908,031!
2792
      pAPI->stateStore.streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1,
×
2793
                                            pInfo->primaryTsIndex);
2794
    }
2795

2796
    pInfo->readHandle = *pHandle;
908,031✔
2797
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
908,031✔
2798
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
908,031✔
2799
    QUERY_CHECK_NULL(pInfo->pCreateTbRes, code, lino, _error, terrno);
906,335!
2800
    pInfo->hasPart = false;
906,335✔
2801

2802
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
908,031✔
2803
    QUERY_CHECK_CODE(code, lino, _error);
906,272!
2804

2805
    // set the extract column id to streamHandle
2806
    code = pAPI->tqReaderFn.tqReaderSetColIdList(pInfo->tqReader, pColIds, idstr);
906,272✔
2807
    QUERY_CHECK_CODE(code, lino, _error);
904,576!
2808

2809
    SArray* tableIdList = NULL;
904,576✔
2810
    code = extractTableIdList(((STableScanInfo*)(pInfo->pTableScanOp->info))->base.pTableListInfo, &tableIdList);
905,436✔
2811
    QUERY_CHECK_CODE(code, lino, _error);
907,195!
2812
    code = pAPI->tqReaderFn.tqReaderSetQueryTableList(pInfo->tqReader, tableIdList, idstr);
907,195✔
2813
    QUERY_CHECK_CODE(code, lino, _error);
908,031!
2814
    taosArrayDestroy(tableIdList);
908,031✔
2815
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->base.cond, sizeof(SQueryTableDataCond));
908,031!
2816
  } else {
2817
    taosArrayDestroy(pColIds);
×
2818
    tableListDestroy(pTableListInfo);
×
2819
  }
2820

2821
  // clear the local variable to avoid repeatly free
2822
  pColIds = NULL;
908,031✔
2823

2824
  // create the pseduo columns info
2825
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
908,031✔
2826
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->pPseudoExpr, &pInfo->numOfPseudoExpr);
524,902✔
2827
    QUERY_CHECK_CODE(code, lino, _error);
524,902!
2828
  }
2829

2830
  code = filterInitFromNode((SNode*)pScanPhyNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0, NULL);
908,031✔
2831
  QUERY_CHECK_CODE(code, lino, _error);
908,031!
2832

2833
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
908,031✔
2834
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
908,031!
2835
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
908,031✔
2836
  QUERY_CHECK_CODE(code, lino, _error);
908,031!
2837

2838
  pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
908,031✔
2839
  pInfo->windowSup = (SWindowSupporter){.pStreamAggSup = NULL, .gap = -1, .parentType = QUERY_NODE_PHYSICAL_PLAN};
908,031✔
2840
  pInfo->groupId = 0;
907,195✔
2841
  pInfo->igCheckGroupId = false;
907,195✔
2842
  pInfo->pStreamScanOp = pOperator;
908,031✔
2843
  pInfo->deleteDataIndex = 0;
908,031✔
2844
  code = createSpecialDataBlock(STREAM_DELETE_DATA, &pInfo->pDeleteDataRes);
908,031✔
2845
  QUERY_CHECK_CODE(code, lino, _error);
908,031!
2846

2847
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
908,031✔
2848
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateDataRes);
908,031✔
2849
  QUERY_CHECK_CODE(code, lino, _error);
908,031!
2850

2851
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
908,031!
2852
  pInfo->partitionSup.needCalc = false;
907,195✔
2853
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
908,031✔
2854
  pInfo->igExpired = pTableScanNode->igExpired;
908,031✔
2855
  pInfo->twAggSup.maxTs = INT64_MIN;
907,207✔
2856
  pInfo->pState = pTaskInfo->streamInfo.pState;
908,031✔
2857
  pInfo->stateStore = pTaskInfo->storageAPI.stateStore;
908,031✔
2858
  pInfo->readerFn = pTaskInfo->storageAPI.tqReaderFn;
907,207✔
2859
  pInfo->pFillSup = NULL;
906,371✔
2860
  pInfo->useGetResultRange = false;
906,371✔
2861
  pInfo->pRangeScanRes = NULL;
907,195✔
2862

2863
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
906,371✔
2864
  QUERY_CHECK_CODE(code, lino, _error);
908,031!
2865

2866
  setOperatorInfo(pOperator, STREAM_SCAN_OP_NAME, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, false, OP_NOT_OPENED, pInfo,
908,031✔
2867
                  pTaskInfo);
2868
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
908,031✔
2869

2870
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doQueueScanNext, NULL, destroyTmqScanOperatorInfo, optrDefaultBufFn,
907,171✔
2871
                                         NULL, optrDefaultGetNextExtFn, NULL);
2872

2873
  *pOptrInfo = pOperator;
906,335✔
2874
  return code;
907,207✔
2875

2876
_error:
×
2877
  if (pColIds != NULL) {
×
2878
    taosArrayDestroy(pColIds);
×
2879
  }
2880

2881
  if (pInfo != NULL && pInfo->pTableScanOp != NULL) {
×
2882
    STableScanInfo* p = (STableScanInfo*)pInfo->pTableScanOp->info;
×
2883
    if (p != NULL) {
×
2884
      p->base.pTableListInfo = NULL;
×
2885
    }
2886
    destroyTmqScanOperatorInfo(pInfo);
×
2887
  }
2888

2889
  if (pOperator != NULL) {
×
2890
    pOperator->info = NULL;
×
2891
    destroyOperator(pOperator);
×
2892
  }
2893
  pTaskInfo->code = code;
×
2894
  return code;
×
2895
}
2896

2897
static int32_t doTagScanOneTable(SOperatorInfo* pOperator, SSDataBlock* pRes, SMetaReader* mr, SStorageAPI* pAPI) {
8,644,400✔
2898
  int32_t        code = TSDB_CODE_SUCCESS;
8,644,400✔
2899
  int32_t        lino = 0;
8,644,400✔
2900
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
8,644,400✔
2901
  STagScanInfo*  pInfo = pOperator->info;
8,645,605✔
2902
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
8,645,605✔
2903
  int32_t        count = pRes->info.rows;
8,645,605✔
2904

2905
  STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
8,645,605✔
2906
  if (!item) {
8,645,605!
2907
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
2908
           GET_TASKID(pTaskInfo));
2909
    tDecoderClear(&(*mr).coder);
×
2910
    goto _end;
×
2911
  }
2912

2913
  code = pAPI->metaReaderFn.getTableEntryByUid(mr, item->uid);
8,645,605✔
2914
  tDecoderClear(&(*mr).coder);
8,644,437✔
2915
  if (code != TSDB_CODE_SUCCESS) {
8,645,605!
2916
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
2917
           GET_TASKID(pTaskInfo));
2918
    goto _end;
×
2919
  }
2920

2921
  char str[512];
8,645,605✔
2922
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
30,090,289✔
2923
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
21,444,647✔
2924

2925
    // refactor later
2926
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
21,445,852✔
2927
      STR_TO_VARSTR(str, (*mr).me.name);
1,890,317!
2928
      code = colDataSetVal(pDst, (count), str, false);
1,890,317✔
2929
      QUERY_CHECK_CODE(code, lino, _end);
1,889,149!
2930
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
19,555,535✔
2931
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
54,427✔
2932
      QUERY_CHECK_CODE(code, lino, _end);
54,427!
2933
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
19,501,108✔
2934
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
54,427✔
2935
      QUERY_CHECK_CODE(code, lino, _end);
54,427!
2936
    } else {  // it is a tag value
2937
      STagVal val = {0};
19,445,476✔
2938
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
19,445,476✔
2939
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
19,445,476✔
2940

2941
      char* data = NULL;
19,445,476✔
2942
      if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
19,445,476!
2943
        data = tTagValToData((const STagVal*)p, false);
19,446,681✔
2944
      } else {
2945
        data = (char*)p;
×
2946
      }
2947

2948
      code = colDataSetVal(pDst, (count), data,
19,446,681✔
2949
                           (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
19,446,681!
2950
      QUERY_CHECK_CODE(code, lino, _end);
19,446,681!
2951

2952
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
19,446,681!
2953
          (data != NULL)) {
2954
        taosMemoryFree(data);
4,975,350!
2955
      }
2956
    }
2957
  }
2958

2959
_end:
8,644,437✔
2960
  if (code != TSDB_CODE_SUCCESS) {
8,645,605!
2961
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2962
    pTaskInfo->code = code;
×
2963
  } else {
2964
    pRes->info.rows++;
8,645,605✔
2965
  }
2966

2967
  return code;
8,645,605✔
2968
}
2969

2970
static void tagScanFreeUidTag(void* p) {
47,174,109✔
2971
  STUidTagInfo* pInfo = p;
47,174,109✔
2972
  if (pInfo->pTagVal != NULL) {
47,174,109✔
2973
    taosMemoryFree(pInfo->pTagVal);
47,185,656!
2974
  }
2975
}
47,178,478✔
2976

2977
static int32_t tagScanCreateResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
10,475,058✔
2978
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
10,475,058!
2979
  if (pColumnData == NULL) {
10,481,778!
2980
    return terrno;
×
2981
  }
2982

2983
  pColumnData->info.type = pType->type;
10,481,778✔
2984
  pColumnData->info.bytes = pType->bytes;
10,474,276✔
2985
  pColumnData->info.scale = pType->scale;
10,473,667✔
2986
  pColumnData->info.precision = pType->precision;
10,467,313✔
2987

2988
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
10,471,262✔
2989
  if (code != TSDB_CODE_SUCCESS) {
10,470,411!
2990
    terrno = code;
×
2991
    taosMemoryFree(pColumnData);
×
2992
    return terrno;
×
2993
  }
2994

2995
  pParam->columnData = pColumnData;
10,470,411✔
2996
  pParam->colAlloced = true;
10,470,151✔
2997
  return TSDB_CODE_SUCCESS;
10,477,094✔
2998
}
2999

3000
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
65,937,205✔
3001
  int32_t                code = TSDB_CODE_SUCCESS;
65,937,205✔
3002
  int32_t                lino = 0;
65,937,205✔
3003
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
65,937,205✔
3004
  SColumnNode*           pSColumnNode = NULL;
65,937,205✔
3005
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
65,943,313✔
3006
    pSColumnNode = *(SColumnNode**)pNode;
18,427,785✔
3007
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
47,549,412!
3008
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
×
3009
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
×
3010
      pSColumnNode = NULL;
×
3011
      pCtx->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
×
3012
      if (NULL == pSColumnNode) {
×
3013
        return DEAL_RES_ERROR;
×
3014
      }
3015
      pSColumnNode->colId = -1;
×
3016
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
×
3017
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
×
3018
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
×
3019
      nodesDestroyNode(*pNode);
×
3020
      *pNode = (SNode*)pSColumnNode;
×
3021
    } else {
3022
      return DEAL_RES_CONTINUE;
×
3023
    }
3024
  } else {
3025
    return DEAL_RES_CONTINUE;
47,523,692✔
3026
  }
3027

3028
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
18,422,883✔
3029
  if (!data) {
18,418,561✔
3030
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
10,489,451✔
3031
    if (code == TSDB_CODE_DUP_KEY) {
10,495,424!
3032
      code = TSDB_CODE_SUCCESS;
×
3033
    }
3034
    QUERY_CHECK_CODE(code, lino, _end);
10,495,424!
3035
    pSColumnNode->slotId = pCtx->index++;
10,495,424✔
3036
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
10,490,017✔
3037
                         .type = pSColumnNode->node.resType.type,
10,491,088✔
3038
                         .bytes = pSColumnNode->node.resType.bytes};
10,485,091✔
3039
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
10,490,979✔
3040
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
10,497,383!
3041
  } else {
3042
    SColumnNode* col = *(SColumnNode**)data;
7,929,110✔
3043
    pSColumnNode->slotId = col->slotId;
7,929,110✔
3044
  }
3045

3046
_end:
18,419,269✔
3047
  if (code != TSDB_CODE_SUCCESS) {
18,415,903✔
3048
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
5,514!
3049
    return DEAL_RES_ERROR;
×
3050
  }
3051
  return DEAL_RES_CONTINUE;
18,410,389✔
3052
}
3053

3054
static int32_t tagScanFilterByTagCond(SArray* aUidTags, SNode* pTagCond, SArray* aFilterIdxs, void* pVnode,
10,485,700✔
3055
                                      SStorageAPI* pAPI, STagScanInfo* pInfo) {
3056
  int32_t code = TSDB_CODE_SUCCESS;
10,485,700✔
3057
  int32_t lino = 0;
10,485,700✔
3058
  int32_t numOfTables = taosArrayGetSize(aUidTags);
10,485,700✔
3059
  SArray* pBlockList = NULL;
10,485,030✔
3060

3061
  SSDataBlock* pResBlock = createTagValBlockForFilter(pInfo->filterCtx.cInfoList, numOfTables, aUidTags, pVnode, pAPI);
10,485,030✔
3062
  QUERY_CHECK_NULL(pResBlock, code, lino, _end, terrno);
10,479,675!
3063

3064
  pBlockList = taosArrayInit(1, POINTER_BYTES);
10,479,675✔
3065
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
10,478,690!
3066

3067
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
10,483,458✔
3068
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
10,483,458!
3069

3070
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
10,483,458✔
3071

3072
  SScalarParam output = {0};
10,478,728✔
3073
  code = tagScanCreateResultData(&type, numOfTables, &output);
10,475,282✔
3074
  QUERY_CHECK_CODE(code, lino, _end);
10,474,835!
3075

3076
  code = scalarCalculate(pTagCond, pBlockList, &output, NULL, NULL);
10,474,835✔
3077
  QUERY_CHECK_CODE(code, lino, _end);
10,468,286!
3078

3079
  bool* result = (bool*)output.columnData->pData;
10,468,286✔
3080
  for (int32_t i = 0; i < numOfTables; ++i) {
26,221,668✔
3081
    if (result[i]) {
15,745,541!
3082
      void* tmp = taosArrayPush(aFilterIdxs, &i);
14,968,266✔
3083
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
14,968,266!
3084
    }
3085
  }
3086

3087
  colDataDestroy(output.columnData);
10,474,941✔
3088
  taosMemoryFreeClear(output.columnData);
10,470,617!
3089

3090
_end:
10,459,816✔
3091
  blockDataDestroy(pResBlock);
10,464,657✔
3092
  taosArrayDestroy(pBlockList);
10,467,036✔
3093
  if (code != TSDB_CODE_SUCCESS) {
10,446,633!
3094
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3095
  }
3096
  return code;
10,446,633✔
3097
}
3098

3099
static int32_t tagScanFillOneCellWithTag(SOperatorInfo* pOperator, const STUidTagInfo* pUidTagInfo,
194,509,191✔
3100
                                         SExprInfo* pExprInfo, SColumnInfoData* pColInfo, int rowIndex,
3101
                                         const SStorageAPI* pAPI, void* pVnode) {
3102
  int32_t code = TSDB_CODE_SUCCESS;
194,509,191✔
3103
  int32_t lino = 0;
194,509,191✔
3104
  if (QUERY_NODE_FUNCTION == pExprInfo->pExpr->nodeType) {
194,509,191✔
3105
    if (FUNCTION_TYPE_TBNAME == pExprInfo->pExpr->_function.functionType) {  // tbname
92,816,437!
3106
      char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
3107
      STR_TO_VARSTR(str, "ctbidx");
×
3108

3109
      code = colDataSetVal(pColInfo, rowIndex, str, false);
×
3110
      QUERY_CHECK_CODE(code, lino, _end);
×
3111
    } else if (FUNCTION_TYPE_TBUID == pExprInfo->pExpr->_function.functionType) {
92,771,105✔
3112
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
46,347,508✔
3113
      QUERY_CHECK_CODE(code, lino, _end);
46,402,447!
3114
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
46,410,067✔
3115
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
46,386,561✔
3116
      QUERY_CHECK_CODE(code, lino, _end);
46,402,147!
3117
    }
3118
  } else {
3119
    STagVal tagVal = {0};
101,675,850✔
3120
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
101,676,077✔
3121
    if (pUidTagInfo->pTagVal == NULL) {
101,705,215!
3122
      colDataSetNULL(pColInfo, rowIndex);
×
3123
    } else {
3124
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
101,679,947✔
3125

3126
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
101,697,051!
3127
        colDataSetNULL(pColInfo, rowIndex);
28,026!
3128
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
101,670,584!
3129
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
3130
        QUERY_CHECK_CODE(code, lino, _end);
×
3131
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
125,596,239!
3132
        if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
23,967,527✔
3133
          QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
3134
        }
3135
        char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
23,952,218!
3136
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
23,945,490!
3137

3138
        varDataSetLen(tmp, tagVal.nData);
23,945,490✔
3139
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
23,943,633!
3140
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
23,934,305✔
3141
        taosMemoryFree(tmp);
23,967,868!
3142
        QUERY_CHECK_CODE(code, lino, _end);
23,965,392!
3143
      } else {
3144
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
77,638,127✔
3145
        QUERY_CHECK_CODE(code, lino, _end);
77,701,011!
3146
      }
3147
    }
3148
  }
3149

3150
_end:
194,501,173✔
3151
  if (code != TSDB_CODE_SUCCESS) {
194,501,173!
3152
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3153
  }
3154
  return code;
194,466,388✔
3155
}
3156

3157
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
30,311,938✔
3158
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
3159
  int32_t       code = TSDB_CODE_SUCCESS;
30,311,938✔
3160
  int32_t       lino = 0;
30,311,938✔
3161
  STagScanInfo* pInfo = pOperator->info;
30,311,938✔
3162
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
30,319,339✔
3163
  if (!ignoreFilterIdx) {
30,305,334✔
3164
    size_t szTables = taosArrayGetSize(aFilterIdxs);
10,445,532✔
3165
    for (int i = 0; i < szTables; ++i) {
25,432,016✔
3166
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
14,954,501✔
3167
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
14,955,639✔
3168
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
14,954,514!
3169
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
85,366,498✔
3170
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
70,375,656✔
3171
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
70,400,465!
3172
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
70,400,465✔
3173
        QUERY_CHECK_CODE(code, lino, _end);
70,411,984!
3174
      }
3175
    }
3176
  } else {
3177
    size_t szTables = taosArrayGetSize(aUidTags);
19,859,802✔
3178
    for (int i = 0; i < szTables; ++i) {
51,303,889✔
3179
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
31,445,485✔
3180
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
31,446,891!
3181
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
155,490,774✔
3182
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
124,053,240✔
3183
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
124,063,494!
3184
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
124,063,494✔
3185
        QUERY_CHECK_CODE(code, lino, _end);
124,043,883!
3186
      }
3187
    }
3188
  }
3189

3190
_end:
19,858,404✔
3191
  if (code != TSDB_CODE_SUCCESS) {
30,335,919!
3192
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3193
  }
3194
  return code;
30,339,440✔
3195
}
3196

3197
static int32_t doTagScanFromCtbIdxNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
60,407,197✔
3198
  int32_t        code = TSDB_CODE_SUCCESS;
60,407,197✔
3199
  int32_t        lino = 0;
60,407,197✔
3200
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
60,407,197✔
3201
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
60,413,228✔
3202

3203
  STagScanInfo* pInfo = pOperator->info;
60,390,353✔
3204
  SSDataBlock*  pRes = pInfo->pRes;
60,405,464✔
3205

3206
  QRY_PARAM_CHECK(ppRes);
60,401,141!
3207

3208
  if (pOperator->status == OP_EXEC_DONE) {
60,406,752✔
3209
    return TSDB_CODE_SUCCESS;
29,840,840✔
3210
  }
3211
  blockDataCleanup(pRes);
30,556,552✔
3212

3213
  if (pInfo->pCtbCursor == NULL) {
30,579,764!
3214
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
30,575,687✔
3215
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
30,579,755!
3216
  } else {
3217
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
3218
    QUERY_CHECK_CODE(code, lino, _end);
×
3219
  }
3220

3221
  SArray* aUidTags = pInfo->aUidTags;
30,579,752✔
3222
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
30,576,853✔
3223
  int32_t count = 0;
30,591,078✔
3224
  bool    ctbCursorFinished = false;
30,591,078✔
3225
  while (1) {
496,426✔
3226
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
31,087,504✔
3227
    taosArrayClear(aFilterIdxs);
31,080,940✔
3228

3229
    int32_t numTables = 0;
31,082,665✔
3230
    while (numTables < pOperator->resultInfo.capacity) {
78,312,101✔
3231
      SMCtbCursor* pCur = pInfo->pCtbCursor;
78,275,653✔
3232
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
78,277,422✔
3233
      if (uid == 0) {
78,265,298✔
3234
        ctbCursorFinished = true;
31,090,717✔
3235
        break;
31,090,717✔
3236
      }
3237
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
47,174,581✔
3238
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
47,181,926!
3239
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
47,152,713!
3240

3241
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
47,152,713!
3242
      void* tmp = taosArrayPush(aUidTags, &info);
47,204,113✔
3243
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
47,204,113!
3244
      ++numTables;
47,204,113✔
3245
    }
3246

3247
    if (numTables == 0) {
31,090,717✔
3248
      break;
743,074✔
3249
    }
3250
    bool ignoreFilterIdx = true;
30,347,643✔
3251
    if (pInfo->pTagCond != NULL) {
30,347,643✔
3252
      ignoreFilterIdx = false;
10,485,700✔
3253
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo);
10,485,700✔
3254
      QUERY_CHECK_CODE(code, lino, _end);
10,454,610!
3255
    } else {
3256
      ignoreFilterIdx = true;
19,859,802✔
3257
    }
3258

3259
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
30,314,412✔
3260
    QUERY_CHECK_CODE(code, lino, _end);
30,340,427!
3261

3262
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
30,340,427✔
3263

3264
    if (count != 0) {
30,340,828✔
3265
      break;
29,844,402✔
3266
    }
3267
  }
3268

3269
  if (count > 0) {
30,587,476✔
3270
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
29,843,655✔
3271
  }
3272
  if (count == 0 || ctbCursorFinished) {
30,583,959!
3273
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
30,587,284✔
3274
    pInfo->pCtbCursor = NULL;
30,569,226✔
3275
    setOperatorCompleted(pOperator);
30,564,684✔
3276
  }
3277

3278
  pRes->info.rows = count;
30,560,020✔
3279
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
30,568,519✔
3280
  if (bLimitReached) {
30,564,730!
3281
    setOperatorCompleted(pOperator);
×
3282
  }
3283

3284
  pOperator->resultInfo.totalRows += pRes->info.rows;
30,564,730✔
3285
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
30,552,323✔
3286

3287
_end:
30,542,914✔
3288
  if (code != TSDB_CODE_SUCCESS) {
30,542,914!
3289
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3290
    pTaskInfo->code = code;
×
3291
    T_LONG_JMP(pTaskInfo->env, code);
×
3292
  }
3293

3294
  return code;
30,542,914✔
3295
}
3296

3297
static int32_t createTagScanTableListInfoFromParam(SOperatorInfo* pOperator) {
6,418,010✔
3298
  STagScanInfo*          pInfo = pOperator->info;
6,418,010✔
3299
  SExecTaskInfo*         pTaskInfo = pOperator->pTaskInfo;
6,418,010✔
3300
  int32_t                code = 0;
6,418,010✔
3301
  STableListInfo*        pListInfo = pInfo->pTableListInfo;
6,418,010✔
3302
  STagScanOperatorParam* pParam = (STagScanOperatorParam*)pOperator->pOperatorGetParam->value;
6,418,010✔
3303
  tb_uid_t               pUid = pParam->vcUid;
6,418,010✔
3304

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

3309
  STableKeyInfo info = {.groupId = 0};
6,418,010✔
3310

3311
  int32_t tableIdx = 0;
6,418,010✔
3312
  taosHashClear(pListInfo->map);
6,418,010✔
3313
  taosArrayClear(pListInfo->pTableList);
6,418,010✔
3314

3315
  if (taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
6,418,010!
3316
    if (TSDB_CODE_DUP_KEY == terrno) {
×
3317
    } else {
3318
      return terrno;
×
3319
    }
3320
  }
3321

3322
  info.uid = pUid;
6,418,010✔
3323
  void* p = taosArrayPush(pListInfo->pTableList, &info);
6,418,010✔
3324
  if (p == NULL) {
6,418,010!
3325
    return terrno;
×
3326
  }
3327

3328
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
6,418,010✔
3329

3330
  return code;
6,418,010✔
3331
}
3332

3333
static int32_t doTagScanFromMetaEntryNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
8,737,529✔
3334
  int32_t       code = TSDB_CODE_SUCCESS;
8,737,529✔
3335
  int32_t       lino = 0;
8,737,529✔
3336
  STagScanInfo* pInfo = pOperator->info;
8,737,529✔
3337
  if (pOperator->pOperatorGetParam) {
8,737,529✔
3338
    pOperator->resultInfo.totalRows = 0;
6,418,010✔
3339
    pOperator->dynamicTask = true;
6,418,010✔
3340
    pInfo->curPos = 0;
6,418,010✔
3341
    code = createTagScanTableListInfoFromParam(pOperator);
6,418,010✔
3342
    freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
6,418,010✔
3343
    pOperator->pOperatorGetParam = NULL;
6,418,010✔
3344
    QUERY_CHECK_CODE(code, lino, _end);
6,418,010!
3345

3346
    if (pOperator->status == OP_EXEC_DONE) {
6,418,010✔
3347
      pOperator->status = OP_OPENED;
2,896,926✔
3348
    }
3349
  }
3350

3351
  if (pOperator->status == OP_EXEC_DONE) {
8,736,356✔
3352
    (*ppRes) = NULL;
1,117,314✔
3353
    return code;
1,117,314✔
3354
  }
3355

3356
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
7,619,042✔
3357
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
7,620,215✔
3358

3359
  SExprInfo*   pExprInfo = &pOperator->exprSupp.pExprInfo[0];
7,620,215✔
3360
  SSDataBlock* pRes = pInfo->pRes;
7,619,042✔
3361
  blockDataCleanup(pRes);
7,620,215✔
3362

3363
  int32_t size = 0;
7,618,774✔
3364
  code = tableListGetSize(pInfo->pTableListInfo, &size);
7,618,774✔
3365
  QUERY_CHECK_CODE(code, lino, _end);
7,620,215!
3366

3367
  if (size == 0) {
7,620,215✔
3368
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
84,891✔
3369
    (*ppRes) = NULL;
84,891✔
3370
    return code;
84,891✔
3371
  }
3372

3373
  SMetaReader mr = {0};
7,535,324✔
3374
  pAPI->metaReaderFn.initReader(&mr, pInfo->readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
7,535,324✔
3375
  pRes->info.rows = 0;
7,535,324✔
3376

3377
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
16,179,761!
3378
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
8,645,605✔
3379
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD &&
8,645,605!
3380
        code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
3381
      // ignore other error
3382
      code = TSDB_CODE_SUCCESS;
8,645,605✔
3383
    }
3384
    QUERY_CHECK_CODE(code, lino, _end);
8,645,605!
3385

3386
    if (++pInfo->curPos >= size) {
8,645,605✔
3387
      setOperatorCompleted(pOperator);
7,535,324✔
3388
    }
3389
  }
3390

3391
  pAPI->metaReaderFn.clearReader(&mr);
7,535,324✔
3392
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
7,535,324✔
3393
  if (bLimitReached) {
7,535,324!
3394
    setOperatorCompleted(pOperator);
×
3395
  }
3396

3397
  // qDebug("QInfo:0x%" PRIx64 ", create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
3398
  if (pOperator->status == OP_EXEC_DONE) {
7,535,324!
3399
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
7,535,324✔
3400
  }
3401

3402
  pOperator->resultInfo.totalRows += pRes->info.rows;
7,535,324✔
3403

3404
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
7,535,324!
3405
_end:
7,535,324✔
3406
  if (code != TSDB_CODE_SUCCESS) {
7,535,324!
3407
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3408
    pTaskInfo->code = code;
×
3409
    T_LONG_JMP(pTaskInfo->env, code);
×
3410
  }
3411
  return code;
7,535,324✔
3412
}
3413

3414
static SSDataBlock* doTagScanFromMetaEntry(SOperatorInfo* pOperator) {
×
3415
  SSDataBlock* pRes = NULL;
×
3416
  int32_t      code = doTagScanFromMetaEntryNext(pOperator, &pRes);
×
3417
  return pRes;
×
3418
}
3419

3420
static void destroyTagScanOperatorInfo(void* param) {
36,071,470✔
3421
  STagScanInfo* pInfo = (STagScanInfo*)param;
36,071,470✔
3422
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
36,071,470!
3423
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
3424
  }
3425
  taosHashCleanup(pInfo->filterCtx.colHash);
36,082,020✔
3426
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
36,071,132✔
3427
  taosArrayDestroy(pInfo->aFilterIdxs);
36,073,677✔
3428
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
36,073,269✔
3429

3430
  blockDataDestroy(pInfo->pRes);
36,102,112✔
3431
  taosArrayDestroy(pInfo->matchInfo.pList);
36,096,344✔
3432
  tableListDestroy(pInfo->pTableListInfo);
36,084,068✔
3433

3434
  pInfo->pRes = NULL;
36,099,637✔
3435
  pInfo->pTableListInfo = NULL;
36,100,401✔
3436
  taosMemoryFreeClear(param);
36,089,475!
3437
}
36,090,017✔
3438

3439
int32_t createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pTagScanNode,
36,084,291✔
3440
                                  STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3441
                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
3442
  QRY_PARAM_CHECK(pOptrInfo);
36,084,291!
3443

3444
  int32_t         code = TSDB_CODE_SUCCESS;
36,091,551✔
3445
  int32_t         lino = 0;
36,091,551✔
3446
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
36,091,551✔
3447
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
36,091,551!
3448
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
36,073,886!
3449
  if (pInfo == NULL || pOperator == NULL) {
36,039,509!
3450
    code = terrno;
1,562✔
3451
    goto _error;
×
3452
  }
3453

3454
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
36,037,947✔
3455

3456
  int32_t    numOfExprs = 0;
36,037,441✔
3457
  SExprInfo* pExprInfo = NULL;
36,048,147✔
3458

3459
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
36,058,870✔
3460
  QUERY_CHECK_CODE(code, lino, _error);
36,111,181!
3461

3462
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
36,111,181✔
3463
  QUERY_CHECK_CODE(code, lino, _error);
36,030,176!
3464

3465
  int32_t num = 0;
36,030,176✔
3466
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
36,044,188✔
3467
  QUERY_CHECK_CODE(code, lino, _error);
36,078,824!
3468

3469
  pInfo->pTagCond = pTagCond;
36,078,824✔
3470
  pInfo->pTagIndexCond = pTagIndexCond;
36,094,830✔
3471
  pInfo->suid = pPhyNode->suid;
36,079,660✔
3472
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
36,075,590✔
3473

3474
  pInfo->pTableListInfo = pTableListInfo;
36,046,230✔
3475
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
36,046,492✔
3476
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
36,109,058!
3477

3478
  pInfo->readHandle = *pReadHandle;
36,054,872✔
3479
  pInfo->curPos = 0;
36,068,082✔
3480

3481
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
36,021,575✔
3482
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
36,017,656✔
3483
                  pTaskInfo);
3484
  initResultSizeInfo(&pOperator->resultInfo, 4096);
36,018,132✔
3485
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
36,022,494✔
3486
  QUERY_CHECK_CODE(code, lino, _error);
36,110,519!
3487

3488
  if (pTagScanNode->onlyMetaCtbIdx) {
36,110,519!
3489
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
30,581,904✔
3490
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
30,592,509!
3491

3492
    pInfo->aFilterIdxs = taosArrayInit(pOperator->resultInfo.capacity, sizeof(int32_t));
30,588,686✔
3493
    QUERY_CHECK_NULL(pInfo->aFilterIdxs, code, lino, _error, terrno);
30,592,645!
3494

3495
    pInfo->filterCtx.colHash =
30,586,118✔
3496
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
30,589,067✔
3497
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
30,585,680!
3498

3499
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
30,585,951✔
3500
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
30,588,122!
3501

3502
    if (pInfo->pTagCond != NULL) {
30,575,686✔
3503
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
10,494,209✔
3504
    }
3505
  }
3506
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
36,107,271!
3507
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
36,105,821✔
3508
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
3509
  *pOptrInfo = pOperator;
36,101,108✔
3510
  return code;
36,096,055✔
3511

3512
_error:
×
3513
  if (pInfo) {
×
3514
    pInfo->pTableListInfo = NULL;
×
3515
  }
3516

3517
  if (pInfo != NULL) destroyTagScanOperatorInfo(pInfo);
×
3518
  if (pOperator != NULL) {
×
3519
    pOperator->info = NULL;
×
3520
    destroyOperator(pOperator);
×
3521
  }
3522
  return code;
×
3523
}
3524

3525
// table merge scan operator
3526

3527
static int32_t subTblRowCompareTsFn(const void* pLeft, const void* pRight, void* param) {
1,227,445,068✔
3528
  int32_t                 left = *(int32_t*)pLeft;
1,227,445,068✔
3529
  int32_t                 right = *(int32_t*)pRight;
1,227,559,052✔
3530
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
1,227,564,088✔
3531

3532
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
1,227,564,088✔
3533
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
1,227,573,856✔
3534

3535
  if (leftIdx == -1) {
1,227,570,192✔
3536
    return 1;
1,051,587,792✔
3537
  } else if (rightIdx == -1) {
175,982,400✔
3538
    return -1;
1,742,400✔
3539
  }
3540

3541
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
174,240,000✔
3542
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
174,240,000✔
3543
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
174,240,000!
3544
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
174,240,000!
3545
    ret = -1 * ret;
×
3546
  }
3547
  return ret;
174,240,000✔
3548
}
3549

3550
static int32_t subTblRowCompareTsPkFn(const void* pLeft, const void* pRight, void* param) {
×
3551
  int32_t                 left = *(int32_t*)pLeft;
×
3552
  int32_t                 right = *(int32_t*)pRight;
×
3553
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
×
3554

3555
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
×
3556
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
×
3557

3558
  if (leftIdx == -1) {
×
3559
    return 1;
×
3560
  } else if (rightIdx == -1) {
×
3561
    return -1;
×
3562
  }
3563

3564
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
×
3565
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
×
3566
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
×
3567
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
×
3568
    ret = -1 * ret;
×
3569
  }
3570
  if (ret == 0 && pInfo->pPkOrderInfo) {
×
3571
    ret = tsortComparBlockCell(pInfo->aInputs[left].pInputBlock, pInfo->aInputs[right].pInputBlock, leftIdx, rightIdx,
×
3572
                               pInfo->pPkOrderInfo);
×
3573
  }
3574
  return ret;
×
3575
}
3576

3577
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
15,600,202✔
3578
  int32_t code = TSDB_CODE_SUCCESS;
15,600,202✔
3579
  int32_t lino = 0;
15,600,202✔
3580
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
15,600,202!
3581
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
15,600,202!
3582
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
15,576,962!
3583
  for (int i = 0; i < src->numOfCols; i++) {
52,476,565✔
3584
    dst->colList[i] = src->colList[i];
36,970,977✔
3585
  }
3586

3587
_end:
15,585,920✔
3588
  if (code != TSDB_CODE_SUCCESS) {
15,585,920!
3589
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3590
  }
3591
  return code;
15,550,112✔
3592
}
3593

3594
static int32_t fetchNextSubTableBlockFromReader(SOperatorInfo* pOperator, STmsSubTableInput* pInput,
23,417,340✔
3595
                                                bool* pSubTableHasBlock) {
3596
  int32_t code = 0;
23,417,340✔
3597

3598
  STableMergeScanInfo*    pInfo = pOperator->info;
23,417,340✔
3599
  SReadHandle*            pHandle = &pInfo->base.readHandle;
23,411,274✔
3600
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
23,418,480✔
3601
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
23,409,359✔
3602
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
23,404,182✔
3603

3604
  blockDataCleanup(pInput->pReaderBlock);
23,415,700✔
3605
  if (!pInput->bInMemReader) {
23,423,013!
3606
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
23,526,419✔
3607
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
11,768,083✔
3608
    if (code != 0) {
11,767,674!
3609
      return code;
×
3610
    }
3611
  }
3612

3613
  pInfo->base.dataReader = pInput->pReader;
23,422,918✔
3614

3615
  while (true) {
35,416✔
3616
    bool hasNext = false;
23,447,741✔
3617
    code = pAPI->tsdReader.tsdNextDataBlock(pInfo->base.dataReader, &hasNext);
23,450,646✔
3618
    if (code != 0) {
23,474,998!
3619
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3620
      pInfo->base.dataReader = NULL;
×
3621
      return code;
×
3622
    }
3623

3624
    if (!hasNext || isTaskKilled(pTaskInfo)) {
23,474,998!
3625
      if (isTaskKilled(pTaskInfo)) {
15,641,966!
3626
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3627
        pInfo->base.dataReader = NULL;
×
3628
        return code;
×
3629
      }
3630

3631
      *pSubTableHasBlock = false;
15,642,428✔
3632
      break;
15,643,484✔
3633
    }
3634

3635
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
7,833,032✔
3636
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
7,679,448✔
3637
    } else {
3638
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
153,584✔
3639
    }
3640

3641
    uint32_t status = 0;
7,832,268✔
3642
    code = loadDataBlock(pOperator, &pInfo->base, pInput->pReaderBlock, &status);
7,833,032✔
3643
    if (code != 0) {
7,832,268!
3644
      pInfo->base.dataReader = NULL;
×
3645
      return code;
×
3646
    }
3647

3648
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
7,832,268!
3649
      *pSubTableHasBlock = false;
×
3650
      break;
×
3651
    }
3652
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pInput->pReaderBlock->info.rows == 0) {
7,832,268!
3653
      continue;
35,416✔
3654
    }
3655

3656
    *pSubTableHasBlock = true;
7,796,852✔
3657
    break;
7,797,616✔
3658
  }
3659

3660
  if (*pSubTableHasBlock) {
23,439,280!
3661
    pInput->pReaderBlock->info.id.groupId =
15,593,704✔
3662
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
15,592,940✔
3663
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
7,797,616✔
3664
  }
3665
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
23,440,341!
3666
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
17,626,460✔
3667
    pInput->pReader = NULL;
17,621,302✔
3668
  }
3669

3670
  pInfo->base.dataReader = NULL;
23,435,959✔
3671
  return TSDB_CODE_SUCCESS;
23,436,706✔
3672
}
3673

3674
static int32_t setGroupStartEndIndex(STableMergeScanInfo* pInfo) {
9,760,924✔
3675
  pInfo->bGroupProcessed = false;
9,760,924✔
3676

3677
  int32_t numOfTables = 0;
9,746,968✔
3678
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
9,724,812✔
3679
  if (code != TSDB_CODE_SUCCESS) {
9,669,147!
3680
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3681
    return code;
×
3682
  }
3683

3684
  int32_t i = pInfo->tableStartIndex + 1;
9,669,147✔
3685
  for (; i < numOfTables; ++i) {
15,505,574✔
3686
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
6,144,204✔
3687
    if (!tableKeyInfo) {
6,139,168!
3688
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3689
      return terrno;
×
3690
    }
3691
    if (tableKeyInfo->groupId != pInfo->groupId) {
6,139,168✔
3692
      break;
309,400✔
3693
    }
3694
  }
3695
  pInfo->tableEndIndex = i - 1;
9,670,770✔
3696
  return TSDB_CODE_SUCCESS;
9,771,517✔
3697
}
3698

3699
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
9,798,469✔
3700
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
25,431,922✔
3701
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
15,642,270✔
3702
    if (pInput->rowIdx == -1) {
15,633,161✔
3703
      continue;
7,841,474✔
3704
    }
3705

3706
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
7,796,852!
3707
      pInput->rowIdx = 0;
7,796,088✔
3708
      pInput->pageIdx = -1;
7,795,020✔
3709
    }
3710

3711
    pInput->pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
7,793,952!
3712
    SColumnInfoData* col = taosArrayGet(pInput->pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
7,796,852✔
3713
    if (!col) {
7,794,728!
3714
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3715
      return terrno;
×
3716
    }
3717
    pInput->aTs = (int64_t*)col->pData;
7,794,728✔
3718
  }
3719

3720
  __merge_compare_fn_t mergeCompareFn = (!pSubTblsInfo->pPkOrderInfo) ? subTblRowCompareTsFn : subTblRowCompareTsPkFn;
9,795,440!
3721
  return tMergeTreeCreate(&pSubTblsInfo->pTree, pSubTblsInfo->numSubTables, pSubTblsInfo, mergeCompareFn);
9,798,311✔
3722
}
3723

3724
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
9,761,379✔
3725
  int32_t code = setGroupStartEndIndex(pInfo);
9,761,379✔
3726
  if (code != TSDB_CODE_SUCCESS) {
9,766,335!
3727
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3728
    return code;
×
3729
  }
3730
  STmsSubTablesMergeInfo* pSubTblsInfo = taosMemoryCalloc(1, sizeof(STmsSubTablesMergeInfo));
9,766,335!
3731
  if (pSubTblsInfo == NULL) {
9,706,625!
3732
    return terrno;
×
3733
  }
3734
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
9,706,625✔
3735
  if (!pSubTblsInfo->pTsOrderInfo) {
9,752,316!
3736
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3737
    return terrno;
×
3738
  }
3739
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
9,708,382!
3740
    pSubTblsInfo->pPkOrderInfo = taosArrayGet(pInfo->pSortInfo, 1);
×
3741
    if (!pSubTblsInfo->pPkOrderInfo) {
×
3742
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3743
      return terrno;
×
3744
    }
3745
  } else {
3746
    pSubTblsInfo->pPkOrderInfo = NULL;
9,698,913✔
3747
  }
3748
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
9,740,552✔
3749
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
9,710,647!
3750
  if (pSubTblsInfo->aInputs == NULL) {
9,682,797!
3751
    taosMemoryFree(pSubTblsInfo);
×
3752
    return terrno;
×
3753
  }
3754
  int32_t bufPageSize = pInfo->bufPageSize;
9,704,494✔
3755
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
9,708,115✔
3756
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
9,729,065✔
3757
  if (code != TSDB_CODE_SUCCESS) {
9,713,912!
3758
    taosMemoryFree(pSubTblsInfo->aInputs);
×
3759
    taosMemoryFree(pSubTblsInfo);
×
3760
    return code;
×
3761
  }
3762
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
9,713,912✔
3763
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
9,736,054✔
3764

3765
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
9,734,363✔
3766
  return TSDB_CODE_SUCCESS;
9,665,639✔
3767
}
3768

3769
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
9,750,015✔
3770
  int32_t        code = TSDB_CODE_SUCCESS;
9,750,015✔
3771
  int32_t        lino = 0;
9,750,015✔
3772
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
9,750,015✔
3773
  SReadHandle*   pHandle = &pInfo->base.readHandle;
9,744,299✔
3774
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
9,733,454✔
3775

3776
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
9,718,641✔
3777

3778
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
25,335,574✔
3779
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
15,609,267✔
3780
    pInput->type = SUB_TABLE_MEM_BLOCK;
15,586,492✔
3781

3782
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
15,569,832✔
3783
    QUERY_CHECK_CODE(code, lino, _end);
15,531,636!
3784

3785
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
15,531,636✔
3786
    QUERY_CHECK_CODE(code, lino, _end);
15,638,302!
3787

3788
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
15,638,302✔
3789
    QUERY_CHECK_CODE(code, lino, _end);
15,643,496!
3790

3791
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
15,643,496✔
3792
    pInput->pKeyInfo = keyInfo;
15,635,740✔
3793

3794
    if (isTaskKilled(pTaskInfo)) {
15,628,930!
3795
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
×
3796
    }
3797

3798
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
15,607,036✔
3799
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
11,660,908✔
3800
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
5,836,940✔
3801
      QUERY_CHECK_CODE(code, lino, _end);
5,830,836!
3802
      pInput->bInMemReader = true;
5,830,836✔
3803
    } else {
3804
      pInput->pReader = NULL;
9,786,029✔
3805
      pInput->bInMemReader = false;
9,784,966✔
3806
    }
3807
    bool hasNext = true;
15,613,045✔
3808
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
15,605,087✔
3809
    QUERY_CHECK_CODE(code, lino, _end);
15,641,197!
3810
    if (!hasNext) {
15,641,197!
3811
      pInput->rowIdx = -1;
7,843,581✔
3812
      ++pSubTblsInfo->numSubTablesCompleted;
7,845,413✔
3813
      continue;
7,845,121✔
3814
    } else {
3815
      pInput->rowIdx = 0;
7,797,616✔
3816
      pInput->pageIdx = -1;
7,797,616✔
3817
    }
3818
  }
3819

3820
_end:
9,800,284✔
3821
  if (code != TSDB_CODE_SUCCESS) {
9,800,284!
3822
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3823
  }
3824
  return code;
9,791,017✔
3825
}
3826

3827
static void adjustSubTableFromMemBlock(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
7,797,616✔
3828
  int32_t              code = TSDB_CODE_SUCCESS;
7,797,616✔
3829
  int32_t              lino = 0;
7,797,616✔
3830
  SExecTaskInfo*       pTaskInfo = pOperatorInfo->pTaskInfo;
7,797,616✔
3831
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
7,797,616✔
3832
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
7,797,616✔
3833
  bool                 hasNext = true;
7,797,616✔
3834
  code = fetchNextSubTableBlockFromReader(pOperatorInfo, pInput, &hasNext);
7,797,616✔
3835
  QUERY_CHECK_CODE(code, lino, _end);
7,797,616!
3836

3837
  if (!hasNext) {
7,797,616!
3838
    pInput->rowIdx = -1;
7,797,616✔
3839
    ++pSubTblsInfo->numSubTablesCompleted;
7,797,616✔
3840
  } else {
3841
    pInput->rowIdx = 0;
×
3842
  }
3843

3844
_end:
7,796,857✔
3845
  if (code != TSDB_CODE_SUCCESS) {
7,796,857!
3846
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3847
    pTaskInfo->code = code;
×
3848
    T_LONG_JMP(pTaskInfo->env, code);
×
3849
  }
3850
}
7,796,857✔
3851

3852
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
1,418,262,588✔
3853
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
1,418,262,588✔
3854
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,418,282,959✔
3855

3856
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,418,265,456!
3857
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
1,418,469,932✔
3858
    ++pInput->rowIdx;
1,410,682,844✔
3859
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
7,797,616!
3860
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
7,797,616!
3861
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
7,797,616✔
3862
    }
3863
    if (pInput->rowIdx != -1) {
7,795,789!
3864
      SColumnInfoData* col = taosArrayGet(pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
×
3865
      if (!col) {
×
3866
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3867
        return terrno;
×
3868
      }
3869
      pInput->pInputBlock = pInputBlock;
×
3870
      pInput->aTs = (int64_t*)col->pData;
×
3871
    }
3872
  }
3873

3874
  return tMergeTreeAdjust(pSubTblsInfo->pTree, tMergeTreeGetAdjustIndex(pSubTblsInfo->pTree));
1,418,484,868✔
3875
}
3876

3877
static int32_t appendChosenRowToDataBlock(STmsSubTablesMergeInfo* pSubTblsInfo, SSDataBlock* pBlock) {
1,417,808,379✔
3878
  int32_t            code = TSDB_CODE_SUCCESS;
1,417,808,379✔
3879
  int32_t            lino = 0;
1,417,808,379✔
3880
  STmsSubTableInput* pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,417,808,379✔
3881
  SSDataBlock*       pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,418,180,430!
3882

3883
  for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
2,147,483,647✔
3884
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
2,147,483,647✔
3885
    QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
2,147,483,647!
3886

3887
    SColumnInfoData* pSrcColInfo = taosArrayGet(pInputBlock->pDataBlock, i);
2,147,483,647✔
3888
    QUERY_CHECK_NULL(pSrcColInfo, code, lino, _end, terrno);
2,147,483,647!
3889
    bool isNull = colDataIsNull(pSrcColInfo, pInputBlock->info.rows, pInput->rowIdx, NULL);
2,147,483,647!
3890

3891
    if (isNull) {
2,147,483,647✔
3892
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
612,656,351✔
3893
      QUERY_CHECK_CODE(code, lino, _end);
612,608,606!
3894
    } else {
3895
      if (pSrcColInfo->pData != NULL) {
2,147,483,647!
3896
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
2,147,483,647!
3897
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
2,147,483,647✔
3898
        QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
3899
      }
3900
    }
3901
  }
3902
  pBlock->info.dataLoad = 1;
1,418,472,105✔
3903
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
1,418,475,448✔
3904
  pBlock->info.rows += 1;
1,418,472,256✔
3905

3906
_end:
1,418,442,201✔
3907
  if (code != TSDB_CODE_SUCCESS) {
1,418,442,201!
3908
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3909
  }
3910
  return code;
1,418,269,503✔
3911
}
3912

3913
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
15,853,977✔
3914
                                       SSDataBlock** pResBlock) {
3915
  int32_t                 code = TSDB_CODE_SUCCESS;
15,853,977✔
3916
  int32_t                 lino = 0;
15,853,977✔
3917
  STableMergeScanInfo*    pInfo = pOperator->info;
15,853,977✔
3918
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
15,855,050✔
3919
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
15,848,795✔
3920
  bool                    finished = false;
15,846,992✔
3921

3922
  QRY_PARAM_CHECK(pResBlock);
15,846,992!
3923

3924
  blockDataCleanup(pBlock);
15,850,629✔
3925

3926
  while (true) {
×
3927
    while (true) {
3928
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
1,434,015,174✔
3929
        finished = true;
15,856,865✔
3930
        break;
15,856,865✔
3931
      }
3932

3933
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
1,418,113,016✔
3934
      QUERY_CHECK_CODE(code, lino, _end);
1,418,298,298!
3935

3936
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
1,418,298,298✔
3937
      QUERY_CHECK_CODE(code, lino, _end);
1,418,130,139!
3938

3939
      if (pBlock->info.rows >= capacity) {
1,418,130,139!
3940
        break;
×
3941
      }
3942
    }
3943

3944
    if (isTaskKilled(pTaskInfo)) {
15,856,865!
3945
      return pTaskInfo->code;
×
3946
    }
3947

3948
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo);
15,854,291✔
3949
    if (finished || limitReached || pBlock->info.rows > 0) {
15,854,588!
3950
      break;
3951
    }
3952
  }
3953

3954
  if (pBlock->info.rows > 0) {
15,854,588✔
3955
    *pResBlock = pBlock;
6,054,457✔
3956
  }
3957

3958
_end:
9,796,501✔
3959
  if (code != TSDB_CODE_SUCCESS) {
15,850,958!
3960
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3961
    pTaskInfo->code = code;
×
3962
  }
3963
  return code;
15,852,476✔
3964
}
3965

3966
static int32_t startSubTablesTableMergeScan(SOperatorInfo* pOperator) {
9,741,492✔
3967
  int32_t              code = TSDB_CODE_SUCCESS;
9,741,492✔
3968
  int32_t              lino = 0;
9,741,492✔
3969
  STableMergeScanInfo* pInfo = pOperator->info;
9,741,492✔
3970

3971
  code = initSubTablesMergeInfo(pInfo);
9,750,362✔
3972
  QUERY_CHECK_CODE(code, lino, _end);
9,687,935!
3973

3974
  code = initSubTableInputs(pOperator, pInfo);
9,687,935✔
3975
  QUERY_CHECK_CODE(code, lino, _end);
9,793,007!
3976

3977
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
9,793,007✔
3978
  QUERY_CHECK_CODE(code, lino, _end);
9,791,788!
3979

3980
_end:
9,791,788✔
3981
  if (code != TSDB_CODE_SUCCESS) {
9,791,788!
3982
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3983
  }
3984
  return code;
9,793,294✔
3985
}
3986

3987
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
160,380,166✔
3988
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
160,380,166✔
3989
  if (pSubTblsInfo != NULL) {
160,384,579✔
3990
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
9,798,778✔
3991

3992
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
25,444,683✔
3993
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
15,638,942✔
3994
      taosMemoryFree(pInput->tblCond.colList);
15,638,942!
3995
      blockDataDestroy(pInput->pReaderBlock);
15,641,372✔
3996
      blockDataDestroy(pInput->pPageBlock);
15,645,608✔
3997
      taosArrayDestroy(pInput->aBlockPages);
15,645,608✔
3998
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
15,645,608✔
3999
      pInput->pReader = NULL;
15,645,608✔
4000
    }
4001

4002
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
9,802,408✔
4003
    taosMemoryFree(pSubTblsInfo->aInputs);
9,799,372!
4004

4005
    taosMemoryFree(pSubTblsInfo);
9,795,742!
4006
    pInfo->pSubTablesMergeInfo = NULL;
9,796,963✔
4007

4008
    // taosMemoryTrim(0);
4009
  }
4010
}
160,384,579✔
4011

4012
int32_t doTableMergeScanParaSubTablesNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
15,714,302✔
4013
  QRY_PARAM_CHECK(ppRes);
15,714,302!
4014

4015
  int32_t lino = 0;
15,726,306✔
4016
  int32_t tableListSize = 0;
15,726,306✔
4017
  int64_t st = taosGetTimestampUs();
15,693,413✔
4018

4019
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
15,693,413✔
4020
  STableMergeScanInfo* pInfo = pOperator->info;
15,681,840✔
4021

4022
  if (pOperator->status == OP_EXEC_DONE) {
15,687,581!
4023
    return TSDB_CODE_SUCCESS;
×
4024
  }
4025

4026
  int32_t code = pOperator->fpSet._openFn(pOperator);
15,680,340✔
4027
  QUERY_CHECK_CODE(code, lino, _end);
15,675,474!
4028

4029
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
15,675,474✔
4030
  QUERY_CHECK_CODE(code, lino, _end);
15,725,513!
4031

4032
  if (!pInfo->hasGroupId) {
15,725,513!
4033
    pInfo->hasGroupId = true;
9,665,609✔
4034

4035
    if (tableListSize == 0) {
9,607,940✔
4036
      setOperatorCompleted(pOperator);
216,472✔
4037
      (*ppRes) = NULL;
215,708✔
4038
      return code;
215,708✔
4039
    }
4040

4041
    pInfo->tableStartIndex = 0;
9,391,468✔
4042
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
9,406,207✔
4043
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
9,393,276!
4044
    pInfo->groupId = pTmpGpId->groupId;
9,393,276✔
4045
    code = startSubTablesTableMergeScan(pOperator);
9,452,190✔
4046
    QUERY_CHECK_CODE(code, lino, _end);
9,488,597!
4047
  }
4048

4049
  SSDataBlock* pBlock = NULL;
15,548,501✔
4050
  while (pInfo->tableStartIndex < tableListSize) {
15,852,449✔
4051
    if (isTaskKilled(pTaskInfo)) {
15,849,267!
4052
      break;
×
4053
    }
4054

4055
    code = getSubTablesSortedBlock(pOperator, pInfo->pResBlock, pOperator->resultInfo.capacity, &pBlock);
15,850,033✔
4056
    QUERY_CHECK_CODE(code, lino, _end);
15,849,440!
4057

4058
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
15,849,440!
4059
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
×
4060
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
×
4061

4062
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
×
4063
    }
4064

4065
    if (pBlock != NULL) {
15,849,902✔
4066
      pBlock->info.id.groupId = pInfo->groupId;
6,054,457✔
4067
      pOperator->resultInfo.totalRows += pBlock->info.rows;
6,054,457✔
4068
      pInfo->bGroupProcessed = true;
6,054,457✔
4069
      break;
6,053,389✔
4070
    } else {
4071
      // Data of this group are all dumped, let's try the next group
4072
      stopSubTablesTableMergeScan(pInfo);
9,795,445✔
4073
      if (pInfo->tableEndIndex >= tableListSize - 1) {
9,797,722✔
4074
        setOperatorCompleted(pOperator);
9,490,137✔
4075
        break;
9,489,675✔
4076
      }
4077

4078
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
309,400✔
4079
      STableKeyInfo* pTmpGpId = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
309,400✔
4080
      QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
309,400!
4081

4082
      pInfo->groupId = pTmpGpId->groupId;
309,400✔
4083
      code = startSubTablesTableMergeScan(pOperator);
309,400✔
4084
      QUERY_CHECK_CODE(code, lino, _end);
309,400!
4085
      resetLimitInfoForNextGroup(&pInfo->limitInfo);
309,400✔
4086
    }
4087
  }
4088

4089
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
15,544,891✔
4090

4091
_end:
15,542,767✔
4092
  if (code != TSDB_CODE_SUCCESS) {
15,540,183!
4093
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4094
    pTaskInfo->code = code;
×
4095
    T_LONG_JMP(pTaskInfo->env, code);
×
4096
  } else {
4097
    (*ppRes) = pBlock;
15,540,183✔
4098
  }
4099

4100
  return code;
15,541,711✔
4101
}
4102

4103
static void tableMergeScanDoSkipTable(uint64_t uid, void* pTableMergeOpInfo) {
29,751,468✔
4104
  int32_t              code = TSDB_CODE_SUCCESS;
29,751,468✔
4105
  int32_t              lino = 0;
29,751,468✔
4106
  SOperatorInfo*       pOperator = (SOperatorInfo*)pTableMergeOpInfo;
29,751,468✔
4107
  STableMergeScanInfo* pInfo = pOperator->info;
29,751,468✔
4108
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
29,752,868✔
4109

4110
  if (pInfo->mSkipTables == NULL) {
29,752,171✔
4111
    pInfo->mSkipTables = taosHashInit(pInfo->tableEndIndex - pInfo->tableStartIndex + 1,
2,857,677✔
4112
                                      taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_NO_LOCK);
4113
    QUERY_CHECK_NULL(pInfo->mSkipTables, code, lino, _end, terrno);
2,857,677!
4114
  }
4115
  int bSkip = 1;
29,752,505✔
4116
  if (pInfo->mSkipTables != NULL) {
29,751,473!
4117
    code = taosHashPut(pInfo->mSkipTables, &uid, sizeof(uid), &bSkip, sizeof(bSkip));
29,752,868✔
4118
    if (code == TSDB_CODE_DUP_KEY) {
29,752,169!
4119
      code = TSDB_CODE_SUCCESS;
×
4120
    }
4121
    QUERY_CHECK_CODE(code, lino, _end);
29,752,169!
4122
  }
4123

4124
_end:
29,752,169✔
4125
  if (code != TSDB_CODE_SUCCESS) {
29,752,169!
4126
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4127
    pTaskInfo->code = code;
×
4128
    T_LONG_JMP(pTaskInfo->env, code);
×
4129
  }
4130
}
29,728,960✔
4131

4132
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
779,028,747✔
4133
  STableMergeScanInfo* pInfo = pOperator->info;
779,028,747✔
4134
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
779,090,564✔
4135
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
778,893,566✔
4136
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
779,046,959✔
4137
  int32_t              code = 0;
779,009,056✔
4138
  bool                 hasNext = false;
779,009,056✔
4139
  STsdbReader*         reader = pInfo->base.dataReader;
779,021,914✔
4140

4141
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
779,028,254✔
4142
  if (code != 0) {
778,908,615!
4143
    pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
4144
    qError("table merge scan fetch next data block error code: %d, %s", code, GET_TASKID(pTaskInfo));
×
4145
    pTaskInfo->code = code;
×
4146
    return code;
×
4147
  }
4148

4149
  if (!hasNext || isTaskKilled(pTaskInfo)) {
778,908,615!
4150
    if (isTaskKilled(pTaskInfo)) {
168,907,628!
4151
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
×
4152
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
4153
    }
4154
    *pFinished = true;
168,902,068✔
4155
    return code;
168,892,271✔
4156
  }
4157

4158
  uint32_t status = 0;
610,112,487✔
4159
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
610,086,050✔
4160

4161
  if (code != TSDB_CODE_SUCCESS) {
610,032,463!
4162
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
×
4163
    pTaskInfo->code = code;
×
4164
    return code;
×
4165
  }
4166

4167
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
610,032,463!
4168
    *pFinished = true;
×
4169
    return code;
×
4170
  }
4171

4172
  // current block is filter out according to filter condition, continue load the next block
4173
  if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
610,032,463✔
4174
    *pSkipped = true;
20,353,241✔
4175
    return code;
20,569,773✔
4176
  }
4177

4178
  return code;
589,621,315✔
4179
}
4180

4181
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
791,685,194✔
4182
  STableMergeScanSortSourceParam* source = param;
791,685,194✔
4183

4184
  SOperatorInfo*       pOperator = source->pOperator;
791,685,194✔
4185
  STableMergeScanInfo* pInfo = pOperator->info;
791,726,542✔
4186
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
791,749,923✔
4187
  SSDataBlock*         pBlock = NULL;
791,766,386✔
4188
  int64_t              st = taosGetTimestampUs();
791,758,791✔
4189
  int32_t              code = TSDB_CODE_SUCCESS;
791,758,791✔
4190

4191
  while (true) {
4192
    if (pInfo->rtnNextDurationBlocks) {
840,009,967!
4193
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
60,949,228✔
4194
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
4195

4196
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
60,959,571✔
4197
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
27,683,951✔
4198
        ++pInfo->nextDurationBlocksIdx;
27,683,951✔
4199
      } else {
4200
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
60,955,369✔
4201
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
27,681,069✔
4202
          pInfo->nextDurationBlocks[i] = NULL;
27,681,069✔
4203
        }
4204

4205
        pInfo->rtnNextDurationBlocks = false;
33,274,300✔
4206
        pInfo->nextDurationBlocksIdx = 0;
33,274,300✔
4207
        pInfo->numNextDurationBlocks = 0;
33,274,300✔
4208
        continue;
33,274,300✔
4209
      }
4210
    } else {
4211
      bool bFinished = false;
779,044,202✔
4212
      bool bSkipped = false;
779,011,508✔
4213

4214
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
779,081,958✔
4215
      if (code != 0) {
778,960,273!
4216
        return code;
×
4217
      }
4218

4219
      pBlock = pInfo->pReaderBlock;
778,960,273✔
4220
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
779,029,737!
4221
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
4222
      if (bFinished) {
779,019,010!
4223
        pInfo->bNewFilesetEvent = false;
168,910,846✔
4224
        break;
168,909,894✔
4225
      }
4226

4227
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
610,108,164!
4228
        if (!bSkipped) {
33,296,917!
4229
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
27,683,951✔
4230
          if (code) {
27,681,625!
4231
            *ppBlock = NULL;
×
4232
            return code;
×
4233
          }
4234

4235
          ++pInfo->numNextDurationBlocks;
27,681,625✔
4236
          if (pInfo->numNextDurationBlocks > 2) {
27,680,273!
4237
            qError("%s table merge scan prefetch %d next duration blocks. end early.", GET_TASKID(pTaskInfo),
×
4238
                   pInfo->numNextDurationBlocks);
4239
            pInfo->bNewFilesetEvent = false;
×
4240
            break;
×
4241
          }
4242
        }
4243

4244
        if (pInfo->bNewFilesetEvent) {
33,295,077!
4245
          pInfo->rtnNextDurationBlocks = true;
33,272,190✔
4246
          *ppBlock = NULL;
33,275,342✔
4247
          return code;
33,275,868✔
4248
        }
4249

4250
        if (pInfo->bNextDurationBlockEvent) {
1,314!
4251
          pInfo->bNextDurationBlockEvent = false;
×
4252
          continue;
×
4253
        }
4254
      }
4255
      if (bSkipped) continue;
576,948,281!
4256
    }
4257

4258
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
589,674,658✔
4259

4260
    pOperator->resultInfo.totalRows += pBlock->info.rows;
589,669,235✔
4261
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
589,671,034✔
4262
    *ppBlock = pBlock;
589,674,232✔
4263

4264
    return code;
589,669,261✔
4265
  }
4266

4267
  *ppBlock = NULL;
168,913,032✔
4268
  return code;
168,913,996✔
4269
}
4270

4271
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
134,381,887✔
4272
  int32_t code = TSDB_CODE_SUCCESS;
134,381,887✔
4273
  int32_t lino = 0;
134,381,887✔
4274
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
134,381,887✔
4275
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
134,259,508!
4276
  SBlockOrderInfo biTs = {0};
134,259,508✔
4277
  SBlockOrderInfo biPk = {0};
134,308,617✔
4278

4279
  int32_t tsTargetSlotId = 0;
134,324,832✔
4280
  int32_t pkTargetSlotId = -1;
134,324,832✔
4281
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
533,620,701✔
4282
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
399,524,509✔
4283
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
399,568,852!
4284
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
399,568,852✔
4285
      tsTargetSlotId = colInfo->dstSlotId;
134,376,966✔
4286
      biTs.order = order;
134,380,578✔
4287
      biTs.slotId = tsTargetSlotId;
134,380,578✔
4288
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
134,380,578✔
4289
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
134,380,578✔
4290
    }
4291
    // TODO: order by just ts
4292
    if (colInfo->isPk) {
399,333,783!
4293
      pkTargetSlotId = colInfo->dstSlotId;
6,196,034✔
4294
      biPk.order = order;
6,199,078✔
4295
      biPk.slotId = pkTargetSlotId;
6,199,078✔
4296
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
6,199,078✔
4297
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
6,199,078✔
4298
    }
4299
  }
4300

4301
  void* tmp = taosArrayPush(pSortInfo, &biTs);
134,442,702✔
4302
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
134,442,702!
4303
  if (pkTargetSlotId != -1) {
134,442,702✔
4304
    tmp = taosArrayPush(pSortInfo, &biPk);
6,205,166✔
4305
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
6,205,166!
4306
  }
4307

4308
  (*ppSortArray) = pSortInfo;
134,445,291✔
4309

4310
_end:
134,259,845✔
4311
  if (code != TSDB_CODE_SUCCESS) {
134,353,093!
4312
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4313
  }
4314
  return code;
134,353,093✔
4315
}
4316

4317
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
97,751,173✔
4318
  STableMergeScanInfo* pTmsInfo = param;
97,751,173✔
4319
  if (type == TSD_READER_NOTIFY_DURATION_START) {
97,751,173!
4320
    pTmsInfo->bNewFilesetEvent = true;
97,767,653✔
4321
  } else if (type == TSD_READER_NOTIFY_NEXT_DURATION_BLOCK) {
×
4322
    pTmsInfo->bNextDurationBlockEvent = true;
×
4323
  }
4324
  qDebug("table merge scan receive notification. type %d, fileset %d", type, info->duration.filesetId);
97,741,614✔
4325

4326
  return;
97,815,917✔
4327
}
4328

4329
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
202,692,761✔
4330
  STableMergeScanInfo* pInfo = pOperator->info;
202,692,761✔
4331
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
202,751,026✔
4332

4333
  int32_t code = TSDB_CODE_SUCCESS;
202,771,360✔
4334
  int32_t lino = 0;
202,771,360✔
4335
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
202,771,360✔
4336

4337
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
202,718,661✔
4338
  pInfo->bNewFilesetEvent = false;
202,851,727✔
4339
  pInfo->bNextDurationBlockEvent = false;
202,799,140✔
4340

4341
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
202,741,466✔
4342
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
202,782,781!
4343

4344
  pInfo->pSortHandle = NULL;
202,768,497✔
4345
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
405,527,442✔
4346
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
202,777,236✔
4347
  if (code) {
202,684,660!
4348
    return code;
×
4349
  }
4350

4351
  if (pInfo->bSortRowId && numOfTable != 1) {
202,684,660!
4352
    int32_t memSize = 512 * 1024 * 1024;
19,823,802✔
4353
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
19,823,802✔
4354
    QUERY_CHECK_CODE(code, lino, _end);
19,838,607!
4355
  }
4356

4357
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
202,719,853✔
4358
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
202,684,376✔
4359
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
202,684,816✔
4360

4361
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
202,627,179✔
4362
  QUERY_CHECK_CODE(code, lino, _end);
202,598,034!
4363

4364
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
202,598,034!
4365
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
202,648,323!
4366
  param->pOperator = pOperator;
202,648,323✔
4367

4368
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
202,641,913!
4369
  if (ps == NULL) {
202,593,626!
4370
    taosMemoryFree(param);
×
4371
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
4372
  }
4373

4374
  ps->param = param;
202,593,626✔
4375
  ps->onlyRef = false;
202,606,339✔
4376
  code = tsortAddSource(pInfo->pSortHandle, ps);
202,720,480✔
4377
  QUERY_CHECK_CODE(code, lino, _end);
202,756,273!
4378

4379
  if (numOfTable == 1) {
202,756,273✔
4380
    tsortSetSingleTableMerge(pInfo->pSortHandle);
134,281,751✔
4381
  } else {
4382
    code = tsortOpen(pInfo->pSortHandle);
68,474,522✔
4383
    QUERY_CHECK_CODE(code, lino, _end);
68,500,095!
4384
  }
4385

4386
_end:
68,500,095✔
4387
  if (code != TSDB_CODE_SUCCESS) {
202,751,528!
4388
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4389
  }
4390
  return code;
202,763,615✔
4391
}
4392

4393
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
202,825,346✔
4394
  STableMergeScanInfo* pInfo = pOperator->info;
202,825,346✔
4395
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
202,827,502✔
4396
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
202,827,256✔
4397

4398
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
202,827,256✔
4399
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
202,823,006✔
4400
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
202,817,330✔
4401
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
202,826,768✔
4402
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
202,826,125✔
4403
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
202,816,651✔
4404

4405
  tsortDestroySortHandle(pInfo->pSortHandle);
202,824,661✔
4406
  pInfo->pSortHandle = NULL;
202,826,291✔
4407
}
202,825,957✔
4408

4409
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
169,505,210✔
4410
  int32_t              code = TSDB_CODE_SUCCESS;
169,505,210✔
4411
  int32_t              lino = 0;
169,505,210✔
4412
  STableMergeScanInfo* pInfo = pOperator->info;
169,505,210✔
4413
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
169,506,243✔
4414
  SReadHandle*         pHandle = &pInfo->base.readHandle;
169,433,947✔
4415
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
169,396,714✔
4416
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
169,361,283✔
4417

4418
  {
4419
    int32_t numOfTables = 0;
169,373,904✔
4420
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
169,328,470✔
4421
    QUERY_CHECK_CODE(code, lino, _end);
169,458,819!
4422

4423
    int32_t i = pInfo->tableStartIndex + 1;
169,458,819✔
4424
    for (; i < numOfTables; ++i) {
655,625,684✔
4425
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
524,913,389✔
4426
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
524,917,376!
4427
      if (tableKeyInfo->groupId != pInfo->groupId) {
524,917,376✔
4428
        break;
38,752,503✔
4429
      }
4430
    }
4431
    pInfo->tableEndIndex = i - 1;
169,464,798✔
4432
  }
4433
  pInfo->bGroupProcessed = false;
169,544,859✔
4434
  int32_t tableStartIdx = pInfo->tableStartIndex;
169,533,035✔
4435
  int32_t tableEndIdx = pInfo->tableEndIndex;
169,522,424✔
4436

4437
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
169,531,681✔
4438
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
169,531,681✔
4439
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
338,973,601✔
4440
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
169,455,486✔
4441
  QUERY_CHECK_CODE(code, lino, _end);
169,471,209!
4442
  if (pInfo->filesetDelimited) {
169,471,209!
4443
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
169,415,512✔
4444
  }
4445
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
169,560,483✔
4446

4447
  code = startDurationForGroupTableMergeScan(pOperator);
169,347,067✔
4448
  QUERY_CHECK_CODE(code, lino, _end);
169,472,116!
4449

4450
_end:
169,472,116✔
4451
  if (code != TSDB_CODE_SUCCESS) {
169,467,311!
4452
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4453
    pTaskInfo->code = code;
×
4454
    T_LONG_JMP(pTaskInfo->env, code);
×
4455
  }
4456
}
169,467,311✔
4457

4458
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
169,545,798✔
4459
  STableMergeScanInfo* pInfo = pOperator->info;
169,545,798✔
4460
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
169,550,297✔
4461
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
169,556,256✔
4462

4463
  stopDurationForGroupTableMergeScan(pOperator);
169,548,003✔
4464

4465
  if (pInfo->base.dataReader != NULL) {
169,548,425✔
4466
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
169,548,339✔
4467
    pInfo->base.dataReader = NULL;
169,486,095✔
4468
  }
4469
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
169,500,700✔
4470
    if (pInfo->nextDurationBlocks[i]) {
2,882!
4471
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
2,882✔
4472
      pInfo->nextDurationBlocks[i] = NULL;
2,882✔
4473
    }
4474
    pInfo->numNextDurationBlocks = 0;
2,882✔
4475
    pInfo->nextDurationBlocksIdx = 0;
2,882✔
4476
  }
4477
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
169,496,727✔
4478
  taosHashCleanup(pInfo->mSkipTables);
169,449,536✔
4479
  pInfo->mSkipTables = NULL;
169,444,941✔
4480
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
169,452,451✔
4481
}
169,536,024✔
4482

4483
// all data produced by this function only belongs to one group
4484
// slimit/soffset does not need to be concerned here, since this function only deal with data within one group.
4485
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
384,007,606✔
4486
                                              SOperatorInfo* pOperator) {
4487
  STableMergeScanInfo* pInfo = pOperator->info;
384,007,606✔
4488
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
384,045,612✔
4489
  STupleHandle*        pTupleHandle = NULL;
384,028,206✔
4490

4491
  blockDataCleanup(pResBlock);
383,992,519✔
4492

4493
  while (1) {
×
4494
    while (1) {
2,147,483,647✔
4495
      pTupleHandle = NULL;
2,147,483,647✔
4496
      int32_t code = tsortNextTuple(pHandle, &pTupleHandle);
2,147,483,647✔
4497
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
4498
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
1,221,581!
4499
        T_LONG_JMP(pOperator->pTaskInfo->env, code);
1,221,581!
4500
      }
4501
      if (pTupleHandle == NULL) {
2,147,483,647✔
4502
        break;
320,718,792✔
4503
      }
4504

4505
      code = tsortAppendTupleToBlock(pInfo->pSortHandle, pResBlock, pTupleHandle);
2,147,483,647✔
4506
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647!
4507
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4508
        T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4509
      }
4510

4511
      if (pResBlock->info.rows >= capacity) {
2,147,483,647✔
4512
        break;
63,349,566✔
4513
      }
4514
    }
4515

4516
    if (tsortIsClosed(pHandle)) {
384,068,358!
4517
      terrno = TSDB_CODE_TSC_QUERY_CANCELLED;
×
4518
      T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4519
    }
4520

4521
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
384,087,591✔
4522
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
384,086,998✔
4523
           pInfo->limitInfo.numOfOutputRows);
4524
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
384,086,919!
4525
      break;
4526
    }
4527
  }
4528
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
384,086,919✔
4529
}
4530

4531
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
322,132,962✔
4532
  if (pOperator->status == OP_EXEC_DONE) {
322,132,962✔
4533
    (*ppRes) = NULL;
92,053✔
4534
    return TSDB_CODE_SUCCESS;
92,053✔
4535
  }
4536

4537
  int32_t              code = TSDB_CODE_SUCCESS;
321,958,651✔
4538
  int32_t              lino = 0;
321,958,651✔
4539
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
321,958,651✔
4540
  STableMergeScanInfo* pInfo = pOperator->info;
322,006,214✔
4541

4542
  code = pOperator->fpSet._openFn(pOperator);
322,023,253✔
4543
  QUERY_CHECK_CODE(code, lino, _end);
321,937,090!
4544

4545
  int64_t st = taosGetTimestampUs();
321,990,948✔
4546

4547
  int32_t tableListSize = 0;
321,990,948✔
4548
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
321,947,402✔
4549
  QUERY_CHECK_CODE(code, lino, _end);
322,041,442!
4550

4551
  if (!pInfo->hasGroupId) {
322,041,442!
4552
    pInfo->hasGroupId = true;
140,785,204✔
4553

4554
    if (tableListSize == 0) {
140,595,410✔
4555
      setOperatorCompleted(pOperator);
10,045,013✔
4556
      (*ppRes) = NULL;
10,043,935✔
4557
      return code;
10,043,935✔
4558
    }
4559
    pInfo->tableStartIndex = 0;
130,550,397✔
4560
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
130,636,876✔
4561
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
130,657,414!
4562
    pInfo->groupId = tmp->groupId;
130,657,414✔
4563
    startGroupTableMergeScan(pOperator);
130,722,073✔
4564
  }
4565

4566
  SSDataBlock* pBlock = NULL;
312,017,422✔
4567
  while (pInfo->tableStartIndex < tableListSize) {
384,044,129!
4568
    if (isTaskKilled(pTaskInfo)) {
384,045,441✔
4569
      goto _end;
16,920✔
4570
    }
4571

4572
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
384,030,817✔
4573
                                              pOperator);
4574
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
384,093,616!
4575
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
12,127,745✔
4576
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
12,127,745!
4577
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
12,127,745✔
4578
    }
4579
    if (pBlock != NULL) {
384,090,745✔
4580
      pBlock->info.id.groupId = pInfo->groupId;
181,261,759✔
4581
      pOperator->resultInfo.totalRows += pBlock->info.rows;
181,263,068✔
4582
      pInfo->bGroupProcessed = true;
181,261,754✔
4583
      break;
181,262,703✔
4584
    } else {
4585
      if (pInfo->bNewFilesetEvent) {
202,828,986!
4586
        stopDurationForGroupTableMergeScan(pOperator);
33,277,182✔
4587
        code = startDurationForGroupTableMergeScan(pOperator);
33,277,182✔
4588
        QUERY_CHECK_CODE(code, lino, _end);
33,274,434!
4589
      } else {
4590
        // Data of this group are all dumped, let's try the next group
4591
        stopGroupTableMergeScan(pOperator);
169,548,132✔
4592
        if (pInfo->tableEndIndex >= tableListSize - 1) {
169,547,005✔
4593
          setOperatorCompleted(pOperator);
130,781,224✔
4594
          break;
130,788,246✔
4595
        }
4596

4597
        pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
38,753,238✔
4598
        STableKeyInfo* tmp = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
38,753,238✔
4599
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
38,753,238!
4600
        pInfo->groupId = tmp->groupId;
38,753,238✔
4601
        startGroupTableMergeScan(pOperator);
38,753,238✔
4602
        resetLimitInfoForNextGroup(&pInfo->limitInfo);
38,752,030✔
4603
      }
4604
    }
4605
  }
4606

4607
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
312,043,814✔
4608

4609
_end:
312,071,973✔
4610
  if (code != TSDB_CODE_SUCCESS) {
312,073,381!
4611
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4612
    pTaskInfo->code = code;
×
4613
    T_LONG_JMP(pTaskInfo->env, code);
×
4614
  } else {
4615
    (*ppRes) = pBlock;
312,073,381✔
4616
  }
4617

4618
  return code;
312,067,516✔
4619
}
4620

4621
static SSDataBlock* doTableMergeScan(SOperatorInfo* pOperator) {
×
4622
  SSDataBlock* pRes = NULL;
×
4623
  int32_t      code = doTableMergeScanNext(pOperator, &pRes);
×
4624
  return pRes;
×
4625
}
4626

4627
void destroyTableMergeScanOperatorInfo(void* param) {
134,478,553✔
4628
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
134,478,553✔
4629

4630
  // start one reader variable
4631
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
134,478,553✔
4632
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
134,479,419✔
4633
    pTableScanInfo->base.dataReader = NULL;
134,470,612✔
4634
  }
4635

4636
  for (int32_t i = 0; i < pTableScanInfo->numNextDurationBlocks; ++i) {
134,469,191!
4637
    if (pTableScanInfo->nextDurationBlocks[i] != NULL) {
×
4638
      blockDataDestroy(pTableScanInfo->nextDurationBlocks[i]);
×
4639
      pTableScanInfo->nextDurationBlocks[i] = NULL;
×
4640
    }
4641
  }
4642

4643
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
134,480,798✔
4644
  pTableScanInfo->pSortHandle = NULL;
134,465,768✔
4645
  taosHashCleanup(pTableScanInfo->mSkipTables);
134,467,807✔
4646
  pTableScanInfo->mSkipTables = NULL;
134,452,376✔
4647
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
134,456,584✔
4648
  pTableScanInfo->pSortInputBlock = NULL;
134,481,391✔
4649
  // end one reader variable
4650

4651
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
134,480,413✔
4652

4653
  blockDataDestroy(pTableScanInfo->pResBlock);
134,485,939✔
4654
  pTableScanInfo->pResBlock = NULL;
134,493,522✔
4655

4656
  // remove it from the task->result list
4657
  blockDataDestroy(pTableScanInfo->pReaderBlock);
134,492,182✔
4658
  pTableScanInfo->pReaderBlock = NULL;
134,488,981✔
4659
  taosArrayDestroy(pTableScanInfo->pSortInfo);
134,489,484✔
4660

4661
  stopSubTablesTableMergeScan(pTableScanInfo);
134,489,039✔
4662

4663
  taosMemoryFreeClear(param);
134,482,043!
4664
}
134,468,099✔
4665

4666
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
4,282,598✔
4667
  if (pOptr == NULL) {
4,282,598!
4668
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
4669
    return TSDB_CODE_INVALID_PARA;
×
4670
  }
4671
  // TODO: merge these two info into one struct
4672
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
4,282,598!
4673
  if (!execInfo) {
4,281,284!
4674
    return terrno;
×
4675
  }
4676
  STableMergeScanInfo* pInfo = pOptr->info;
4,281,284✔
4677
  execInfo->blockRecorder = pInfo->base.readRecorder;
4,281,284✔
4678
  execInfo->sortExecInfo = pInfo->sortExecInfo;
4,281,284✔
4679

4680
  *pOptrExplain = execInfo;
4,283,925✔
4681
  *len = sizeof(STableMergeScanExecInfo);
4,282,598✔
4682

4683
  return TSDB_CODE_SUCCESS;
4,282,598✔
4684
}
4685

4686
static int32_t resetTableMergeScanOperatorState(SOperatorInfo* pOper) {
16,094,540✔
4687
  int32_t         code = TSDB_CODE_SUCCESS;
16,094,540✔
4688
  STableMergeScanInfo*   pInfo = pOper->info;
16,094,540✔
4689
  pOper->status = OP_NOT_OPENED;
16,094,540✔
4690

4691
  pInfo->tableEndIndex = 0;
16,094,540✔
4692
  pInfo->tableStartIndex = 0;
16,094,540✔
4693
  pInfo->hasGroupId = false;
16,094,540✔
4694
  if (pInfo->base.readerAPI.tsdReaderClose) {
16,094,540!
4695
    pInfo->base.readerAPI.tsdReaderClose(pInfo->base.dataReader);
16,094,540✔
4696
  }
4697
  pInfo->base.dataReader = NULL;
16,093,243✔
4698
  pInfo->base.scanFlag = MAIN_SCAN;
16,094,540✔
4699

4700
  pInfo->base.limitInfo = (SLimitInfo){0};
16,094,540✔
4701
  pInfo->base.limitInfo.limit.limit = -1;
16,094,540✔
4702
  pInfo->base.limitInfo.slimit.limit = -1;
16,094,540✔
4703

4704
  tableListDestroy(pInfo->base.pTableListInfo);
16,094,540✔
4705
  pInfo->base.pTableListInfo = tableListCreate();
16,094,540✔
4706
  if (!pInfo->base.pTableListInfo) {
16,094,540!
4707
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
4708
    return terrno;
×
4709
  }
4710
  SExecTaskInfo*         pTaskInfo = pOper->pTaskInfo;
16,094,540✔
4711

4712
  STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pTaskInfo->pSubplan->pNode;
16,094,540✔
4713
  code = createScanTableListInfo(&pTableScanNode->scan, pTableScanNode->pGroupTags, pTableScanNode->groupSort,
32,189,080!
4714
                                  &pInfo->base.readHandle, pInfo->base.pTableListInfo, 
4715
                                  pTaskInfo->pSubplan->pTagCond, pTaskInfo->pSubplan->pTagIndexCond, pTaskInfo, NULL);
32,189,080✔
4716
  if (code) {
16,094,540!
4717
    qError("%s failed to createScanTableListInfo, code:%s", __func__, tstrerror(code));
×
4718
    return code;
×
4719
  }
4720

4721
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
16,094,540✔
4722
  cleanupQueryTableDataCond(&pInfo->base.cond);
16,094,540✔
4723
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, &pInfo->base.readHandle, false);
16,094,540✔
4724
  if (code) {
16,094,540!
4725
    qError("%s failed to initQueryTableDataCond, code:%s", __func__, tstrerror(code));
×
4726
    return code;
×
4727
  }
4728

4729
  tsortDestroySortHandle(pInfo->pSortHandle);
16,094,540✔
4730
  pInfo->pSortHandle = NULL;
16,093,300✔
4731

4732
  taosHashCleanup(pInfo->mSkipTables);
16,093,300✔
4733
  pInfo->mSkipTables = NULL;
16,093,300✔
4734
  pOper->resultInfo.totalRows = 0;
16,094,540✔
4735
  
4736
  pInfo->bGroupProcessed = false;
16,093,248✔
4737
  pInfo->bNewFilesetEvent = false;
16,094,540✔
4738
  pInfo->bNextDurationBlockEvent = false;
16,094,540✔
4739
  pInfo->rtnNextDurationBlocks = false;
16,094,540✔
4740
  pInfo->nextDurationBlocksIdx = 0;
16,092,008✔
4741
  pInfo->bSortRowId = false;
16,094,540✔
4742

4743
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
16,094,540!
4744
    if (pInfo->nextDurationBlocks[i] != NULL) {
×
4745
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
×
4746
      pInfo->nextDurationBlocks[i] = NULL;
×
4747
    }
4748
  }
4749
  pInfo->numNextDurationBlocks = 0;
16,094,540✔
4750

4751
  stopSubTablesTableMergeScan(pInfo);
16,093,248✔
4752
  return code;
16,093,248✔
4753
}
4754

4755
int32_t createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
134,341,746✔
4756
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
4757
                                         SOperatorInfo** pOptrInfo) {
4758
  QRY_PARAM_CHECK(pOptrInfo);
134,341,746!
4759

4760
  int32_t              code = TSDB_CODE_SUCCESS;
134,361,525✔
4761
  int32_t              lino = 0;
134,361,525✔
4762
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
134,361,525!
4763
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
134,061,851!
4764
  if (pInfo == NULL || pOperator == NULL) {
134,068,911!
4765
    code = terrno;
×
4766
    goto _error;
×
4767
  }
4768

4769
  pOperator->pPhyNode = pTableScanNode;
134,073,164✔
4770
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
134,089,577✔
4771

4772
  int32_t numOfCols = 0;
134,172,932✔
4773
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
134,242,325✔
4774
                             &pInfo->base.matchInfo);
4775
  QUERY_CHECK_CODE(code, lino, _error);
134,296,790!
4776

4777
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, false);
134,296,790✔
4778
  QUERY_CHECK_CODE(code, lino, _error);
134,184,104!
4779

4780
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
134,184,104✔
4781
    SExprSupp* pSup = &pInfo->base.pseudoSup;
60,043,831✔
4782
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
60,024,665✔
4783
    QUERY_CHECK_CODE(code, lino, _error);
60,067,188!
4784

4785
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
60,067,188✔
4786
                                      &pTaskInfo->storageAPI.functionStore);
4787
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
59,975,313!
4788
  }
4789

4790
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
134,126,784✔
4791

4792
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
133,935,585✔
4793
  QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
134,351,498!
4794

4795
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
134,194,920✔
4796
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
134,267,829✔
4797
  pInfo->base.scanFlag = MAIN_SCAN;
134,165,029✔
4798
  pInfo->base.readHandle = *readHandle;
134,186,921✔
4799

4800
  pInfo->readIdx = -1;
134,235,549✔
4801

4802
  pInfo->base.limitInfo.limit.limit = -1;
134,083,255✔
4803
  pInfo->base.limitInfo.slimit.limit = -1;
134,170,120✔
4804
  pInfo->base.pTableListInfo = pTableListInfo;
134,202,450✔
4805

4806
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
134,195,587✔
4807
  pInfo->sample.seed = taosGetTimestampSec();
134,272,285✔
4808

4809
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
134,350,103✔
4810
                            pTaskInfo->pStreamRuntimeInfo);
134,269,029✔
4811
  QUERY_CHECK_CODE(code, lino, _error);
134,197,812!
4812

4813
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
134,197,812✔
4814

4815
  pInfo->mergeLimit = -1;
134,159,071✔
4816
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
134,147,734!
4817
  if (hasLimit) {
134,074,061✔
4818
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1
26,632,261✔
4819
                            ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset
5,682,360✔
4820
                            : pInfo->limitInfo.limit.limit;
18,992,090✔
4821
    pInfo->mSkipTables = NULL;
13,327,868✔
4822
  }
4823

4824
  initResultSizeInfo(&pOperator->resultInfo, 1024);
134,076,428✔
4825
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
134,123,769✔
4826
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
134,450,188!
4827
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
134,251,161✔
4828
  QUERY_CHECK_CODE(code, lino, _error);
134,434,811!
4829
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
134,434,811!
4830
    pInfo->bSortRowId = true;
30,990,950✔
4831
  } else {
4832
    pInfo->bSortRowId = false;
103,430,113✔
4833
  }
4834

4835
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
134,386,653✔
4836
  QUERY_CHECK_CODE(code, lino, _error);
134,429,153!
4837

4838
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
134,429,153✔
4839
  QUERY_CHECK_CODE(code, lino, _error);
134,182,663!
4840

4841
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
134,182,663✔
4842
  QUERY_CHECK_CODE(code, lino, _error);
134,341,914!
4843

4844
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
134,341,914!
4845

4846
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
134,410,967✔
4847
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
134,180,561✔
4848

4849
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
134,220,638✔
4850

4851
  // start one reader variable
4852
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
134,403,694✔
4853
  QUERY_CHECK_CODE(code, lino, _error);
134,350,348!
4854

4855
  if (!tsExperimental) {
134,350,348!
4856
    pInfo->filesetDelimited = false;
×
4857
  } else {
4858
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
134,350,348!
4859
  }
4860
  // end one reader variable
4861
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
134,312,607✔
4862
                  pInfo, pTaskInfo);
4863
  pOperator->exprSupp.numOfExprs = numOfCols;
134,239,675✔
4864

4865
  pOperator->fpSet = createOperatorFpSet(
134,175,023✔
4866
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
134,277,274!
4867
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
4868
      NULL);
4869
  setOperatorResetStateFn(pOperator, resetTableMergeScanOperatorState);     
134,152,936✔
4870
  pOperator->cost.openCost = 0;
134,287,924✔
4871

4872
  *pOptrInfo = pOperator;
134,338,680✔
4873
  return code;
134,163,201✔
4874

4875
_error:
×
4876
  if (code != TSDB_CODE_SUCCESS) {
×
4877
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4878
  }
4879
  pTaskInfo->code = code;
×
4880
  if (pInfo != NULL) {
×
4881
    pInfo->base.pTableListInfo = NULL;
×
4882
    destroyTableMergeScanOperatorInfo(pInfo);
×
4883
  }
4884
  if (pOperator != NULL) {
×
4885
    pOperator->info = NULL;
×
4886
    destroyOperator(pOperator);
×
4887
  }
4888
  return code;
×
4889
}
4890

4891
// ====================================================================================================================
4892
// TableCountScanOperator
4893
static void        destoryTableCountScanOperator(void* param);
4894
static int32_t     buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
4895
                                                  SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI);
4896
static int32_t     buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
4897
                                                  SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI);
4898
static int32_t     buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4899
                                             STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName);
4900
static int32_t     buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4901
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName);
4902
static int32_t     buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4903
                                          STableCountScanSupp* pSupp, SSDataBlock* pRes);
4904
static void        buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4905
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
4906
                                               size_t perfdbTableNum);
4907
static void        buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
4908
                                              size_t infodbTableNum, size_t perfdbTableNum);
4909
static const char* GROUP_TAG_DB_NAME = "db_name";
4910
static const char* GROUP_TAG_STABLE_NAME = "stable_name";
4911

4912
int32_t tblCountScanGetGroupTagsSlotId(const SNodeList* scanCols, STableCountScanSupp* supp) {
×
4913
  if (scanCols != NULL) {
×
4914
    SNode* pNode = NULL;
×
4915
    FOREACH(pNode, scanCols) {
×
4916
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
4917
        return TSDB_CODE_QRY_SYS_ERROR;
×
4918
      }
4919
      STargetNode* targetNode = (STargetNode*)pNode;
×
4920
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
×
4921
        return TSDB_CODE_QRY_SYS_ERROR;
×
4922
      }
4923
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
×
4924
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
4925
        supp->dbNameSlotId = targetNode->slotId;
×
4926
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
4927
        supp->stbNameSlotId = targetNode->slotId;
×
4928
      }
4929
    }
4930
  }
4931
  return TSDB_CODE_SUCCESS;
×
4932
}
4933

4934
int32_t tblCountScanGetCountSlotId(const SNodeList* pseudoCols, STableCountScanSupp* supp) {
×
4935
  if (pseudoCols != NULL) {
×
4936
    SNode* pNode = NULL;
×
4937
    FOREACH(pNode, pseudoCols) {
×
4938
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
4939
        return TSDB_CODE_QRY_SYS_ERROR;
×
4940
      }
4941
      STargetNode* targetNode = (STargetNode*)pNode;
×
4942
      if (nodeType(targetNode->pExpr) != QUERY_NODE_FUNCTION) {
×
4943
        return TSDB_CODE_QRY_SYS_ERROR;
×
4944
      }
4945
      SFunctionNode* funcNode = (SFunctionNode*)(targetNode->pExpr);
×
4946
      if (funcNode->funcType == FUNCTION_TYPE_TABLE_COUNT) {
×
4947
        supp->tbCountSlotId = targetNode->slotId;
×
4948
      }
4949
    }
4950
  }
4951
  return TSDB_CODE_SUCCESS;
×
4952
}
4953

4954
int32_t tblCountScanGetInputs(SNodeList* groupTags, SName* tableName, STableCountScanSupp* supp) {
×
4955
  if (groupTags != NULL) {
×
4956
    SNode* pNode = NULL;
×
4957
    FOREACH(pNode, groupTags) {
×
4958
      if (nodeType(pNode) != QUERY_NODE_COLUMN) {
×
4959
        return TSDB_CODE_QRY_SYS_ERROR;
×
4960
      }
4961
      SColumnNode* colNode = (SColumnNode*)pNode;
×
4962
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
4963
        supp->groupByDbName = true;
×
4964
      }
4965
      if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
4966
        supp->groupByStbName = true;
×
4967
      }
4968
    }
4969
  } else {
4970
    tstrncpy(supp->dbNameFilter, tNameGetDbNameP(tableName), TSDB_DB_NAME_LEN);
×
4971
    tstrncpy(supp->stbNameFilter, tNameGetTableName(tableName), TSDB_TABLE_NAME_LEN);
×
4972
  }
4973
  return TSDB_CODE_SUCCESS;
×
4974
}
4975

4976
int32_t getTableCountScanSupp(SNodeList* groupTags, SName* tableName, SNodeList* scanCols, SNodeList* pseudoCols,
×
4977
                              STableCountScanSupp* supp, SExecTaskInfo* taskInfo) {
4978
  int32_t code = 0;
×
4979
  code = tblCountScanGetInputs(groupTags, tableName, supp);
×
4980
  if (code != TSDB_CODE_SUCCESS) {
×
4981
    qError("%s get table count scan supp. get inputs error", GET_TASKID(taskInfo));
×
4982
    return code;
×
4983
  }
4984

4985
  supp->dbNameSlotId = -1;
×
4986
  supp->stbNameSlotId = -1;
×
4987
  supp->tbCountSlotId = -1;
×
4988

4989
  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
×
4990
  if (code != TSDB_CODE_SUCCESS) {
×
4991
    qError("%s get table count scan supp. get group tags slot id error", GET_TASKID(taskInfo));
×
4992
    return code;
×
4993
  }
4994

4995
  code = tblCountScanGetCountSlotId(pseudoCols, supp);
×
4996
  if (code != TSDB_CODE_SUCCESS) {
×
4997
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
×
4998
    return code;
×
4999
  }
5000
  return code;
×
5001
}
5002

5003
int32_t createTableCountScanOperatorInfo(SReadHandle* readHandle, STableCountScanPhysiNode* pTblCountScanNode,
×
5004
                                         SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
5005
  QRY_PARAM_CHECK(pOptrInfo);
×
5006

5007
  int32_t                      code = TSDB_CODE_SUCCESS;
×
5008
  int32_t                      lino = 0;
×
5009
  SScanPhysiNode*              pScanNode = &pTblCountScanNode->scan;
×
5010
  STableCountScanOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableCountScanOperatorInfo));
×
5011
  SOperatorInfo*               pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
5012
  if (!pInfo || !pOperator) {
×
5013
    code = terrno;
×
5014
    goto _error;
×
5015
  }
5016

5017
  pInfo->readHandle = *readHandle;
×
5018

5019
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
×
5020
  initResultSizeInfo(&pOperator->resultInfo, 1);
×
5021
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
×
5022
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
×
5023

5024
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
×
5025
  QUERY_CHECK_CODE(code, lino, _error);
×
5026

5027
  code = getTableCountScanSupp(pTblCountScanNode->pGroupTags, &pTblCountScanNode->scan.tableName,
×
5028
                               pTblCountScanNode->scan.pScanCols, pTblCountScanNode->scan.pScanPseudoCols, &pInfo->supp,
5029
                               pTaskInfo);
5030
  QUERY_CHECK_CODE(code, lino, _error);
×
5031

5032
  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
×
5033
                  pInfo, pTaskInfo);
5034
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScanNext, NULL, destoryTableCountScanOperator,
×
5035
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
5036
  *pOptrInfo = pOperator;
×
5037
  return code;
×
5038

5039
_error:
×
5040
  if (pInfo != NULL) {
×
5041
    destoryTableCountScanOperator(pInfo);
×
5042
  }
5043
  if (pOperator != NULL) {
×
5044
    pOperator->info = NULL;
×
5045
    destroyOperator(pOperator);
×
5046
  }
5047
  pTaskInfo->code = code;
×
5048
  return code;
×
5049
}
5050

5051
int32_t fillTableCountScanDataBlock(STableCountScanSupp* pSupp, char* dbName, char* stbName, int64_t count,
×
5052
                                    SSDataBlock* pRes) {
5053
  int32_t code = TSDB_CODE_SUCCESS;
×
5054
  int32_t lino = 0;
×
5055
  if (pSupp->dbNameSlotId != -1) {
×
5056
    QUERY_CHECK_CONDITION((strlen(dbName) > 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
5057
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
×
5058
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
5059

5060
    char varDbName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5061
    tstrncpy(varDataVal(varDbName), dbName, TSDB_DB_NAME_LEN);
×
5062

5063
    varDataSetLen(varDbName, strlen(dbName));
×
5064
    code = colDataSetVal(colInfoData, 0, varDbName, false);
×
5065
    QUERY_CHECK_CODE(code, lino, _end);
×
5066
  }
5067

5068
  if (pSupp->stbNameSlotId != -1) {
×
5069
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
×
5070
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
5071
    if (strlen(stbName) != 0) {
×
5072
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5073
      tstrncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
×
5074
      varDataSetLen(varStbName, strlen(stbName));
×
5075
      code = colDataSetVal(colInfoData, 0, varStbName, false);
×
5076
      QUERY_CHECK_CODE(code, lino, _end);
×
5077
    } else {
5078
      colDataSetNULL(colInfoData, 0);
5079
    }
5080
  }
5081

5082
  if (pSupp->tbCountSlotId != -1) {
×
5083
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
×
5084
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
5085
    code = colDataSetVal(colInfoData, 0, (char*)&count, false);
×
5086
    QUERY_CHECK_CODE(code, lino, _end);
×
5087
  }
5088
  pRes->info.rows = 1;
×
5089

5090
_end:
×
5091
  if (code != TSDB_CODE_SUCCESS) {
×
5092
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5093
  }
5094
  return code;
×
5095
}
5096

5097
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
×
5098
  STableCountScanSupp* pSupp = &pInfo->supp;
×
5099
  SSDataBlock*         pRes = pInfo->pRes;
×
5100

5101
  size_t infodbTableNum;
×
5102
  getInfosDbMeta(NULL, &infodbTableNum);
×
5103
  infodbTableNum -= 1;
×
5104
  size_t perfdbTableNum;
×
5105
  getPerfDbMeta(NULL, &perfdbTableNum);
×
5106

5107
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
5108
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
5109
    return (pRes->info.rows > 0) ? pRes : NULL;
×
5110
  } else {
5111
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
5112
    return (pRes->info.rows > 0) ? pRes : NULL;
×
5113
  }
5114
}
5115

5116
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
×
5117
                                       size_t infodbTableNum, size_t perfdbTableNum) {
5118
  int32_t        code = TSDB_CODE_SUCCESS;
×
5119
  int32_t        lino = 0;
×
5120
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5121
  if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
×
5122
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
5123
    QUERY_CHECK_CODE(code, lino, _end);
×
5124
  } else if (strcmp(pSupp->dbNameFilter, TSDB_PERFORMANCE_SCHEMA_DB) == 0) {
×
5125
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
5126
    QUERY_CHECK_CODE(code, lino, _end);
×
5127
  } else if (strlen(pSupp->dbNameFilter) == 0) {
×
5128
    code = fillTableCountScanDataBlock(pSupp, "", "", infodbTableNum + perfdbTableNum, pRes);
×
5129
    QUERY_CHECK_CODE(code, lino, _end);
×
5130
  }
5131

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

5141
static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5142
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
5143
                                        size_t perfdbTableNum) {
5144
  int32_t        code = TSDB_CODE_SUCCESS;
×
5145
  int32_t        lino = 0;
×
5146
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5147
  if (pInfo->currGrpIdx == 0) {
×
5148
    uint64_t groupId = 0;
×
5149
    if (pSupp->groupByDbName) {
×
5150
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
×
5151
    } else {
5152
      groupId = calcGroupId("", 0);
×
5153
    }
5154

5155
    pRes->info.id.groupId = groupId;
×
5156
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
5157
    QUERY_CHECK_CODE(code, lino, _end);
×
5158
  } else if (pInfo->currGrpIdx == 1) {
×
5159
    uint64_t groupId = 0;
×
5160
    if (pSupp->groupByDbName) {
×
5161
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
×
5162
    } else {
5163
      groupId = calcGroupId("", 0);
×
5164
    }
5165

5166
    pRes->info.id.groupId = groupId;
×
5167
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
5168
    QUERY_CHECK_CODE(code, lino, _end);
×
5169
  } else {
5170
    setOperatorCompleted(pOperator);
×
5171
  }
5172
  pInfo->currGrpIdx++;
×
5173

5174
_end:
×
5175
  if (code != TSDB_CODE_SUCCESS) {
×
5176
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5177
    pTaskInfo->code = code;
×
5178
    T_LONG_JMP(pTaskInfo->env, code);
×
5179
  }
5180
}
×
5181

5182
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
×
5183
  int32_t                      code = TSDB_CODE_SUCCESS;
×
5184
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
×
5185
  STableCountScanOperatorInfo* pInfo = pOperator->info;
×
5186
  STableCountScanSupp*         pSupp = &pInfo->supp;
×
5187
  SSDataBlock*                 pRes = pInfo->pRes;
×
5188

5189
  blockDataCleanup(pRes);
×
5190
  QRY_PARAM_CHECK(ppRes);
×
5191

5192
  if (pOperator->status == OP_EXEC_DONE) {
×
5193
    return code;
×
5194
  }
5195

5196
  if (pInfo->readHandle.mnd != NULL) {
×
5197
    (*ppRes) = buildSysDbTableCount(pOperator, pInfo);
×
5198
    return code;
×
5199
  }
5200

5201
  code = buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
×
5202
  if (code != TSDB_CODE_SUCCESS) {
×
5203
    qError("%s failed since %s", __func__, tstrerror(code));
×
5204
    pTaskInfo->code = code;
×
5205
    T_LONG_JMP(pTaskInfo->env, code);
×
5206
  }
5207
  if (pRes->info.rows > 0) {
×
5208
    *ppRes = pRes;
×
5209
  }
5210

5211
  return code;
×
5212
}
5213

5214
static int32_t buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5215
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes) {
5216
  int32_t        code = TSDB_CODE_SUCCESS;
×
5217
  int32_t        lino = 0;
×
5218
  const char*    db = NULL;
×
5219
  int32_t        vgId = 0;
×
5220
  char           dbName[TSDB_DB_NAME_LEN] = {0};
×
5221
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5222
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
5223

5224
  // get dbname
5225
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, &db, &vgId, NULL, NULL);
×
5226
  SName sn = {0};
×
5227

5228
  code = tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
×
5229
  QUERY_CHECK_CODE(code, lino, _end);
×
5230

5231
  code = tNameGetDbName(&sn, dbName);
×
5232
  QUERY_CHECK_CODE(code, lino, _end);
×
5233

5234
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
5235
    code = buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
×
5236
  } else {
5237
    code = buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
×
5238
  }
5239

5240
_end:
×
5241
  if (code != TSDB_CODE_SUCCESS) {
×
5242
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5243
    pTaskInfo->code = code;
×
5244
  }
5245

5246
  return code;
×
5247
}
5248

5249
static int32_t buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5250
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
5251
  int32_t        code = TSDB_CODE_SUCCESS;
×
5252
  int32_t        lino = 0;
×
5253
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5254
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
5255

5256
  if (pSupp->groupByStbName) {
×
5257
    if (pInfo->stbUidList == NULL) {
×
5258
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
×
5259
      QUERY_CHECK_NULL(pInfo->stbUidList, code, lino, _end, terrno);
×
5260
      code = pAPI->metaFn.storeGetTableList(pInfo->readHandle.vnode, TSDB_SUPER_TABLE, pInfo->stbUidList);
×
5261
      QUERY_CHECK_CODE(code, lino, _end);
×
5262
    }
5263

5264
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
×
5265
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
×
5266
      code = buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid, pAPI);
×
5267
      QUERY_CHECK_CODE(code, lino, _end);
×
5268

5269
      pInfo->currGrpIdx++;
×
5270
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
×
5271
      code = buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName, pAPI);
×
5272
      QUERY_CHECK_CODE(code, lino, _end);
×
5273

5274
      pInfo->currGrpIdx++;
×
5275
    } else {
5276
      setOperatorCompleted(pOperator);
×
5277
    }
5278
  } else {
5279
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
×
5280
    pRes->info.id.groupId = groupId;
×
5281

5282
    int64_t dbTableCount = 0;
×
5283
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &dbTableCount, NULL);
×
5284
    code = fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
×
5285
    QUERY_CHECK_CODE(code, lino, _end);
×
5286
    setOperatorCompleted(pOperator);
×
5287
  }
5288

5289
_end:
×
5290
  if (code != TSDB_CODE_SUCCESS) {
×
5291
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5292
    pTaskInfo->code = code;
×
5293
  }
5294
  return code;
×
5295
}
5296

5297
static int32_t buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5298
                                         STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
5299
  int32_t        code = TSDB_CODE_SUCCESS;
×
5300
  int32_t        lino = 0;
×
5301
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5302
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
5303

5304
  if (strlen(pSupp->dbNameFilter) != 0) {
×
5305
    if (strlen(pSupp->stbNameFilter) != 0) {
×
5306
      uint64_t uid = 0;
×
5307
      code = pAPI->metaFn.getTableUidByName(pInfo->readHandle.vnode, pSupp->stbNameFilter, &uid);
×
5308
      QUERY_CHECK_CODE(code, lino, _end);
×
5309

5310
      int64_t numOfChildTables = 0;
×
5311
      code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, uid, &numOfChildTables, NULL, NULL);
×
5312
      QUERY_CHECK_CODE(code, lino, _end);
×
5313

5314
      code = fillTableCountScanDataBlock(pSupp, dbName, pSupp->stbNameFilter, numOfChildTables, pRes);
×
5315
      QUERY_CHECK_CODE(code, lino, _end);
×
5316
    } else {
5317
      int64_t tbNumVnode = 0;
×
5318
      pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
5319
      code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
5320
      QUERY_CHECK_CODE(code, lino, _end);
×
5321
    }
5322
  } else {
5323
    int64_t tbNumVnode = 0;
×
5324
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
5325
    code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
5326
    QUERY_CHECK_CODE(code, lino, _end);
×
5327
  }
5328

5329
_end:
×
5330
  if (code != TSDB_CODE_SUCCESS) {
×
5331
    pTaskInfo->code = code;
×
5332
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5333
  }
5334

5335
  setOperatorCompleted(pOperator);
×
5336
  return code;
×
5337
}
5338

5339
static int32_t buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
5340
                                              SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI) {
5341
  int32_t code = TSDB_CODE_SUCCESS;
×
5342
  int32_t lino = 0;
×
5343
  char    fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
5344
  if (pSupp->groupByDbName) {
×
5345
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
×
5346
  }
5347

5348
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
5349
  pRes->info.id.groupId = groupId;
×
5350

5351
  int64_t numOfTables = 0;
×
5352
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, NULL, &numOfTables);
×
5353

5354
  if (numOfTables != 0) {
×
5355
    code = fillTableCountScanDataBlock(pSupp, dbName, "", numOfTables, pRes);
×
5356
  }
5357

5358
_end:
×
5359
  if (code != TSDB_CODE_SUCCESS) {
×
5360
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5361
  }
5362
  return code;
×
5363
}
5364

5365
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
5366
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
5367
  int32_t code = TSDB_CODE_SUCCESS;
×
5368
  int32_t lino = 0;
×
5369
  char    stbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5370
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
×
5371
  QUERY_CHECK_CODE(code, lino, _end);
×
5372

5373
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
5374
  if (pSupp->groupByDbName) {
×
5375
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
×
5376
  } else {
5377
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
×
5378
  }
5379

5380
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
5381
  pRes->info.id.groupId = groupId;
×
5382

5383
  int64_t ctbNum = 0;
×
5384
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL, NULL);
×
5385
  QUERY_CHECK_CODE(code, lino, _end);
×
5386
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
×
5387
  QUERY_CHECK_CODE(code, lino, _end);
×
5388

5389
_end:
×
5390
  if (code != TSDB_CODE_SUCCESS) {
×
5391
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5392
  }
5393
  return code;
×
5394
}
5395

5396
static void destoryTableCountScanOperator(void* param) {
×
5397
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
×
5398
  blockDataDestroy(pTableCountScanInfo->pRes);
×
5399

5400
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
×
5401
  taosMemoryFreeClear(param);
×
5402
}
×
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