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

taosdata / TDengine / #4819

20 Oct 2025 02:42AM UTC coverage: 61.392% (+0.3%) from 61.125%
#4819

push

travis-ci

happyguoxy
add coverage result json

156672 of 324369 branches covered (48.3%)

Branch coverage included in aggregate %.

207936 of 269535 relevant lines covered (77.15%)

242336611.46 hits per line

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

56.21
/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 "executorInt.h"
17
#include "filter.h"
18
#include "function.h"
19
#include "functionMgt.h"
20
#include "os.h"
21
#include "querynodes.h"
22
#include "streamexecutorInt.h"
23
#include "systable.h"
24
#include "tname.h"
25

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

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

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

41
int32_t scanDebug = 0;
42

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

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

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

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

66
  STableCountScanSupp supp;
67

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

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

75
bool processBlockWithProbability(const SSampleExecInfo* pInfo) {
2,147,483,647✔
76
#if 0
77
  if (pInfo->sampleRatio == 1) {
78
    return true;
79
  }
80

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

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

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

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

104
  if (order == TSDB_ORDER_ASC) {
270,903,684✔
105
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
256,182,063✔
106
    if (w.ekey < pBlockInfo->window.skey) {
256,155,422!
107
      qError("w.ekey:%" PRId64 " < pBlockInfo->window.skey:%" PRId64, w.ekey, pBlockInfo->window.skey);
×
108
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
109
    }
110

111
    if (w.ekey < pBlockInfo->window.ekey) {
256,152,962✔
112
      *overlap = true;
224,307,211✔
113
      return code;
224,308,385✔
114
    }
115

116
    while (1) {
117
      getNextTimeWindow(pInterval, &w, order);
31,860,666✔
118
      if (w.skey > pBlockInfo->window.ekey) {
31,878,252✔
119
        break;
30,874,509✔
120
      }
121

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

138
    if (w.skey > pBlockInfo->window.skey) {
14,716,733✔
139
      *overlap = true;
13,741,663✔
140
      return code;
13,737,956✔
141
    }
142

143
    while (1) {
144
      getNextTimeWindow(pInterval, &w, order);
979,173✔
145
      if (w.ekey < pBlockInfo->window.skey) {
981,304✔
146
        break;
967,594✔
147
      }
148

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

160
  *overlap = false;
31,842,103✔
161
  return code;
31,831,431✔
162
}
163

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

170
  int64_t buf[2] = {0};
502,351,854✔
171
  SET_RES_WINDOW_KEY((char*)buf, &groupId, sizeof(groupId), groupId);
502,320,592✔
172

173
  STableScanInfo* pTableScanInfo = pOperator->info;
502,276,096✔
174

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

178
  if (p1 == NULL) {
502,371,621✔
179
    return NULL;
206,517,645✔
180
  }
181

182
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
295,853,976✔
183
  if (NULL == *pPage) {
295,829,206!
184
    return NULL;
×
185
  }
186

187
  return (SResultRow*)((char*)(*pPage) + p1->offset);
295,835,401✔
188
}
189

190
static int32_t insertTableToScanIgnoreList(STableScanInfo* pTableScanInfo, uint64_t uid) {
30,238,429✔
191
  if (NULL == pTableScanInfo->pIgnoreTables) {
30,238,429✔
192
    int32_t tableNum = taosArrayGetSize(pTableScanInfo->base.pTableListInfo->pTableList);
1,568,088✔
193
    pTableScanInfo->pIgnoreTables =
1,567,547✔
194
        taosHashInit(tableNum, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, HASH_NO_LOCK);
1,567,547✔
195
    if (NULL == pTableScanInfo->pIgnoreTables) {
1,567,547!
196
      return terrno;
×
197
    }
198
  }
199

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

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

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

217
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
502,316,182✔
218

219
  SFilePage*  pPage = NULL;
502,339,321✔
220
  SResultRow* pRow = getTableGroupOutputBuf(pOperator, pBlockInfo->id.groupId, &pPage);
502,334,838✔
221

222
  if (pRow == NULL) {
502,301,748✔
223
    return TSDB_CODE_SUCCESS;
206,507,460✔
224
  }
225

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

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

232
    EFuncDataRequired reqStatus = fmFuncDynDataRequired(functionId, pEntry, pBlockInfo);
2,147,483,647✔
233
    if (reqStatus != FUNC_DATA_REQUIRED_NOT_LOAD) {
2,147,483,647✔
234
      notLoadBlock = false;
265,541,970✔
235
      break;
265,541,970✔
236
    }
237
  }
238

239
  // release buffer pages
240
  releaseBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, pPage);
294,455,260✔
241

242
  if (notLoadBlock) {
295,794,595✔
243
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
30,246,045✔
244
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
30,246,045✔
245
  }
246

247
  return code;
295,775,765✔
248
}
249

250
static int32_t doFilterByBlockSMA(SFilterInfo* pFilterInfo, SColumnDataAgg* pColsAgg, int32_t numOfCols,
42,784,979✔
251
                                  int32_t numOfRows, bool* keep) {
252
  if (pColsAgg == NULL || pFilterInfo == NULL) {
42,784,979!
253
    *keep = true;
5,982✔
254
    return TSDB_CODE_SUCCESS;
×
255
  }
256

257
  return filterRangeExecute(pFilterInfo, pColsAgg, numOfCols, numOfRows, keep);
42,779,206✔
258
}
259

260
static int32_t doLoadBlockSMA(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
685,349,215✔
261
                              bool* pLoad) {
262
  SStorageAPI* pAPI = &pTaskInfo->storageAPI;
685,349,215✔
263
  bool         allColumnsHaveAgg = true;
685,450,100✔
264
  bool         hasNullSMA = false;
685,457,665✔
265
  if (pLoad != NULL) {
685,407,711✔
266
    *pLoad = false;
685,344,951✔
267
  }
268

269
  int32_t code = pAPI->tsdReader.tsdReaderRetrieveBlockSMAInfo(pTableScanInfo->dataReader, pBlock, &allColumnsHaveAgg,
685,396,024✔
270
                                                               &hasNullSMA);
271
  if (code != TSDB_CODE_SUCCESS) {
685,254,589!
272
    return code;
×
273
  }
274

275
  if (!allColumnsHaveAgg || hasNullSMA) {
685,254,589!
276
    *pLoad = false;
610,287,621✔
277
  } else {
278
    *pLoad = true;
74,966,968✔
279
  }
280

281
  return code;
685,267,713✔
282
}
283

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

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

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

305
  if (pLimitInfo->remainOffset > 0) {
2,147,483,647✔
306
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
29,973,601✔
307
      pLimitInfo->remainOffset -= pBlock->info.rows;
12,793,923✔
308
      blockDataEmpty(pBlock);
12,793,923✔
309
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
12,793,923✔
310
      return false;
12,793,923✔
311
    } else {
312
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
17,179,678✔
313
      if (code != TSDB_CODE_SUCCESS) {
17,179,678!
314
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
315
        pTaskInfo->code = code;
×
316
        T_LONG_JMP(pTaskInfo->env, code);
×
317
      }
318
      pLimitInfo->remainOffset = 0;
17,179,678✔
319
    }
320
  }
321

322
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
2,147,483,647✔
323
    // limit the output rows
324
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
138,394,299✔
325
    blockDataKeepFirstNRows(pBlock, keep);
138,390,974✔
326

327
    pLimitInfo->numOfOutputRows += pBlock->info.rows;
138,390,620✔
328
    qDebug("output limit %" PRId64 " has reached, %s", pLimit->limit, id);
138,388,268✔
329
    return true;
138,395,839✔
330
  }
331

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

336
static bool isDynVtbScan(SOperatorInfo* pOperator) {
67,349,055✔
337
  return pOperator->dynamicTask && ((STableScanInfo*)(pOperator->info))->virtualStableScan;
67,349,055!
338
}
339

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

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

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

354
  if (pOperator->exprSupp.pFilterInfo != NULL) {
2,147,483,647✔
355
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
436,923,058✔
356
  } else {
357
    bool overlap = false;
2,147,483,647✔
358
    int  ret =
359
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
2,147,483,647✔
360
    if (ret != TSDB_CODE_SUCCESS) {
2,147,483,647!
361
      return ret;
×
362
    }
363
    if (overlap) {
2,147,483,647!
364
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
239,043,532✔
365
    }
366
  }
367

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

371
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
2,147,483,647!
372
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
×
373
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
374
    pCost->filterOutBlocks += 1;
×
375
    pCost->totalRows += pBlock->info.rows;
×
376
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
377
    return TSDB_CODE_SUCCESS;
×
378
  } else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
2,147,483,647✔
379
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
166,062,534✔
380
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows,
381
           pBlockInfo->id.uid);
382
    code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
166,100,467✔
383
    pCost->skipBlocks += 1;
166,075,393✔
384
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
166,076,496✔
385
    return code;
166,075,460✔
386
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
2,147,483,647✔
387
    pCost->loadBlockStatis += 1;
248,440,439✔
388
    loadSMA = true;  // mark the operation of load sma;
248,419,916✔
389
    bool success = true;
248,419,916✔
390
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
248,442,480✔
391
    if (code) {
248,421,261!
392
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
393
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
394
      QUERY_CHECK_CODE(code, lino, _end);
×
395
    }
396

397
    if (success) {  // failed to load the block sma data, data block statistics does not exist, load data block instead
248,384,937!
398
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
32,192,956✔
399
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
400
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
32,195,010✔
401
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
32,193,016✔
402
      return code;
32,195,010✔
403
    } else {
404
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
216,191,981✔
405
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
216,291,054✔
406
    }
407
  }
408

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

415
  // try to filter data block according to sma info
416
  if (pOperator->exprSupp.pFilterInfo != NULL && (!loadSMA)) {
2,147,483,647!
417
    bool success = true;
436,896,333✔
418
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
436,931,709✔
419
    if (code) {
436,816,712!
420
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
421
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
422
      QUERY_CHECK_CODE(code, lino, _end);
×
423
    }
424

425
    if (success) {
436,751,204!
426
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
42,778,997✔
427
      bool   keep = false;
42,776,215✔
428
      code = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows, &keep);
42,776,215✔
429
      if (code) {
42,780,203!
430
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
431
        qError("%s failed to do filter by block sma, code:%s", GET_TASKID(pTaskInfo), tstrerror(code));
×
432
        QUERY_CHECK_CODE(code, lino, _end);
×
433
      }
434

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

442
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
4,309,131✔
443
        return TSDB_CODE_SUCCESS;
4,309,131✔
444
      }
445
    }
446
  }
447

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

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

458
  if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
2,147,483,647✔
459
    qDebug("%s data block skipped due to dynamic prune, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
30,246,919!
460
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
461
    pCost->skipBlocks += 1;
30,251,705✔
462
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
30,248,875✔
463

464
    STableScanInfo* p1 = pOperator->info;
30,251,705✔
465
    if (taosHashGetSize(p1->pIgnoreTables) == taosArrayGetSize(p1->base.pTableListInfo->pTableList)) {
30,251,705✔
466
      *status = FUNC_DATA_REQUIRED_ALL_FILTEROUT;
1,066,013✔
467
    } else {
468
      *status = FUNC_DATA_REQUIRED_FILTEROUT;
29,185,692✔
469
    }
470
    return TSDB_CODE_SUCCESS;
30,251,705✔
471
  }
472

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

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

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

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

496
  if (pOperator->exprSupp.pFilterInfo != NULL) {
2,147,483,647✔
497
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo, NULL);
432,570,387✔
498
    QUERY_CHECK_CODE(code, lino, _end);
432,539,440✔
499

500
    int64_t st = taosGetTimestampUs();
432,599,115✔
501
    double  el = (taosGetTimestampUs() - st) / 1000.0;
432,572,030✔
502
    pTableScanInfo->readRecorder.filterTime += el;
432,572,030✔
503

504
    if (pBlock->info.rows == 0) {
432,572,601✔
505
      pCost->filterOutBlocks += 1;
63,533,782✔
506
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
63,505,716✔
507
             GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows, el);
508
    } else {
509
      qDebug("%s data block filter applied, elapsed time:%.2f ms", GET_TASKID(pTaskInfo), el);
368,955,914✔
510
    }
511
  }
512

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

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

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

527
static void prepareForDescendingScan(STableScanBase* pTableScanInfo, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
265,976✔
528
  SET_REVERSE_SCAN_FLAG(pTableScanInfo);
265,976✔
529

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

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

541
static void freeTableCachedVal(void* param) {
1,084,819,810✔
542
  if (param == NULL) {
1,084,819,810!
543
    return;
×
544
  }
545

546
  STableCachedVal* pVal = param;
1,084,819,810✔
547
  taosMemoryFree((void*)pVal->pName);
1,084,819,810✔
548
  taosMemoryFree(pVal->pTags);
1,084,778,388✔
549
  taosMemoryFree(pVal);
1,084,762,947✔
550
}
551

552
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
1,083,949,368✔
553
  int32_t          code = TSDB_CODE_SUCCESS;
1,083,949,368✔
554
  int32_t          lino = 0;
1,083,949,368✔
555
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
1,083,949,368✔
556
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
1,084,494,722!
557

558
  pVal->pTags = NULL;
1,084,494,722✔
559
  pVal->pName = taosStrdup(pMetaReader->me.name);
1,084,371,231✔
560
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
1,084,048,080!
561

562
  // only child table has tag value
563
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
1,084,004,288!
564
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
1,083,273,293✔
565
    pVal->pTags = taosMemoryMalloc(pTag->len);
1,083,135,123✔
566
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
1,083,254,753!
567
    memcpy(pVal->pTags, pTag, pTag->len);
1,083,186,524!
568
  }
569

570
  (*ppResVal) = pVal;
1,084,435,226✔
571

572
_end:
1,084,476,568✔
573
  if (code != TSDB_CODE_SUCCESS) {
1,084,476,568!
574
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
575
    freeTableCachedVal(pVal);
×
576
  }
577
  return code;
1,083,871,750✔
578
}
579

580
// const void *key, size_t keyLen, void *value
581
static void freeCachedMetaItem(const void* key, size_t keyLen, void* value, void* ud) {
1,084,822,148✔
582
  (void)key;
583
  (void)keyLen;
584
  (void)ud;
585
  freeTableCachedVal(value);
1,084,822,148✔
586
}
1,084,787,618✔
587

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

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

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

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

614
  // currently only the tbname pseudo column
615
  if (numOfExpr <= 0) {
1,493,064,914✔
616
    return TSDB_CODE_SUCCESS;
1,974,545✔
617
  }
618

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

621
  // backup the rows
622
  int32_t backupRows = pBlock->info.rows;
1,491,090,369✔
623
  pBlock->info.rows = rows;
1,491,245,163✔
624

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

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

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

647
    pHandle->api.metaReaderFn.readerReleaseLock(&mr);
719,747✔
648

649
    val.pName = mr.me.name;
720,835✔
650
    val.pTags = (STag*)mr.me.ctbEntry.pTags;
720,835✔
651

652
    freeReader = true;
720,835✔
653
  } else {
654
    pCache->metaFetch += 1;
1,489,780,020✔
655

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

675
      pHandle->api.metaReaderFn.readerReleaseLock(&mr);
1,083,889,266✔
676

677
      code = createTableCacheVal(&mr, &pVal);
1,084,693,218✔
678
      QUERY_CHECK_CODE(code, lino, _end);
1,084,072,480!
679

680
      val = *pVal;
1,084,072,480✔
681
    } else {
682
      pCache->cacheHit += 1;
405,855,832✔
683
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
405,901,165✔
684
      val = *pValTmp;
405,909,942✔
685

686
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
405,929,436✔
687
      qTrace("release LRU cache, res %d", bRes);
405,930,975!
688
    }
689

690
    qDebug("retrieve table meta from cache:%" PRIu64 ", hit:%" PRIu64 " miss:%" PRIu64 ", %s", pCache->metaFetch,
1,490,011,933✔
691
           pCache->cacheHit, (pCache->metaFetch - pCache->cacheHit), idStr);
692
  }
693

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

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

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

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

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

728
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
2,147,483,647✔
729
      if (isNullVal) {
2,147,483,647✔
730
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
28,960,452!
731
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
2,147,483,647✔
732
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, 1, false);
2,147,483,647✔
733
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
2,147,483,647!
734
          char* tmp = taosMemoryCalloc(1, varDataLen(data) + 1);
1,047,942,901✔
735
          if (tmp != NULL) {
1,047,955,383!
736
            memcpy(tmp, varDataVal(data), varDataLen(data));
1,047,955,383!
737
            qDebug("get tag value:%s, cid:%d, table name:%s, uid%" PRId64, tmp, tagVal.cid, val.pName,
1,047,898,256✔
738
                   pBlock->info.id.uid);
739
            taosMemoryFree(tmp);
1,047,987,082✔
740
          }
741
          taosMemoryFree(data);
1,047,975,230✔
742
        }
743
        QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
744
      } else {  // todo opt for json tag
745
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
8,666,124✔
746
          code = colDataSetVal(pColInfoData, i, data, false);
6,692,059✔
747
          QUERY_CHECK_CODE(code, lino, _end);
6,692,059!
748
        }
749
      }
750
    }
751
  }
752

753
  // restore the rows
754
  pBlock->info.rows = backupRows;
1,491,704,306✔
755

756
_end:
1,491,514,839✔
757

758
  if (NULL != pVal) {
1,491,526,805✔
759
    insertRet =
1,084,827,128✔
760
        taosLRUCacheInsert(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(uint64_t), pVal,
1,084,853,468✔
761
                           sizeof(STableCachedVal), freeCachedMetaItem, NULL, NULL, TAOS_LRU_PRIORITY_LOW, NULL);
762
    if (insertRet != TAOS_LRU_STATUS_OK) {
1,084,827,128!
763
      qWarn("failed to put meta into lru cache, code:%d, %s", insertRet, idStr);
×
764
    }
765
  }
766

767
  if (freeReader) {
1,491,500,465✔
768
    pHandle->api.metaReaderFn.clearReader(&mr);
1,085,484,246✔
769
  }
770
  if (code != TSDB_CODE_SUCCESS) {
1,491,512,519!
771
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
772
  }
773
  return code;
1,491,407,729✔
774
}
775

776
int32_t setTbNameColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId,
582,490,513✔
777
                         const char* name) {
778
  int32_t                     code = TSDB_CODE_SUCCESS;
582,490,513✔
779
  int32_t                     lino = 0;
582,490,513✔
780
  struct SScalarFuncExecFuncs fpSet = {0};
582,490,513✔
781
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
582,496,319✔
782
  QUERY_CHECK_CODE(code, lino, _end);
582,495,905!
783

784
  size_t len = TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE;
582,495,905✔
785
  char   buf[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
582,495,905✔
786
  STR_TO_VARSTR(buf, name)
582,490,155!
787

788
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
582,497,734✔
789

790
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
582,495,026✔
791
  QUERY_CHECK_CODE(code, lino, _end);
582,482,552!
792

793
  code = colDataSetVal(&infoData, 0, buf, false);
582,482,552✔
794
  QUERY_CHECK_CODE(code, lino, _end);
582,490,940!
795

796
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
582,490,940✔
797
  SScalarParam param = {.columnData = pColInfoData};
582,487,236✔
798

799
  if (fpSet.process != NULL) {
582,478,621!
800
    code = fpSet.process(&srcParam, 1, &param);
582,478,621✔
801
    QUERY_CHECK_CODE(code, lino, _end);
582,486,268!
802
  } else {
803
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
×
804
  }
805

806
  colDataDestroy(&infoData);
582,486,268✔
807

808
_end:
582,480,207✔
809
  if (code != TSDB_CODE_SUCCESS) {
582,483,973!
810
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
811
  }
812
  return code;
582,483,973✔
813
}
814

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

887
  pInfo->tableStartIndex = TARRAY_ELEM_IDX(pInfo->base.pTableListInfo->pTableList, *pKeyInfo);
820,185,305!
888
  pInfo->tableEndIndex = (pInfo->tableStartIndex + (*size) - 1);
819,867,699✔
889
  pInfo->pResBlock->info.blankFill = false;
819,929,775✔
890
  taosMemoryFreeClear(pInfo->pResBlock->pBlockAgg);
819,957,726!
891

892
  if (!pInfo->needCountEmptyTable) {
820,219,143✔
893
    pInfo->countState = TABLE_COUNT_STATE_END;
727,766,788✔
894
  } else {
895
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
92,172,356✔
896
  }
897

898
_end:
819,972,660✔
899
  if (code != TSDB_CODE_SUCCESS) {
819,972,660!
900
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
901
  }
902
  return code;
819,690,962✔
903
}
904

905
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
2,147,483,647✔
906
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
2,147,483,647✔
907
    return;
2,147,483,647✔
908
  }
909
  if (pInfo->base.pTableListInfo->groupOffset) {
470,092,382✔
910
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
965,546✔
911
  } else {
912
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
469,106,574✔
913
    if (code != TSDB_CODE_SUCCESS) {
469,152,723✔
914
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
75,084,397!
915
    }
916
  }
917
}
918

919
static SSDataBlock* getOneRowResultBlock(SExecTaskInfo* pTaskInfo, STableScanBase* pBase, SSDataBlock* pBlock,
319,596,646✔
920
                                         const STableKeyInfo* tbInfo) {
921
  blockDataEmpty(pBlock);
319,596,646✔
922
  pBlock->info.rows = 1;
319,625,333✔
923
  pBlock->info.id.uid = tbInfo->uid;
319,625,333✔
924
  pBlock->info.id.groupId = tbInfo->groupId;
319,621,371✔
925
  pBlock->info.blankFill = true;
319,612,889✔
926

927
  // only one row: set all col data to null & hasNull
928
  int32_t col_num = blockDataGetNumOfCols(pBlock);
319,608,832✔
929
  for (int32_t i = 0; i < col_num; ++i) {
1,300,496,384✔
930
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
980,880,327✔
931
    colDataSetNULL(pColInfoData, 0);
932
  }
933

934
  // set tag/tbname
935
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
319,616,057✔
936

937
  return pBlock;
319,623,580✔
938
}
939

940
static SSDataBlock* getBlockForEmptyTable(SOperatorInfo* pOperator, const STableKeyInfo* tbInfo) {
307,088,457✔
941
  STableScanInfo* pTableScanInfo = pOperator->info;
307,088,457✔
942
  SSDataBlock*    pBlock =
943
      getOneRowResultBlock(pOperator->pTaskInfo, &pTableScanInfo->base, pTableScanInfo->pResBlock, tbInfo);
307,134,082✔
944

945
  pOperator->resultInfo.totalRows++;
307,148,352✔
946
  return pBlock;
307,159,824✔
947
}
948

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

959
  QRY_PARAM_CHECK(ppRes);
2,147,483,647!
960
  pBlock->info.dataLoad = false;
2,147,483,647✔
961

962
  while (true) {
82,996,118✔
963
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
2,147,483,647✔
964
    if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
965
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
3,080✔
966
      QUERY_CHECK_CODE(code, lino, _end);
×
967
    }
968

969
    if (!hasNext) {
2,147,483,647✔
970
      break;
683,750,969✔
971
    }
972

973
    if (isTaskKilled(pTaskInfo)) {
2,147,483,647✔
974
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
56,051✔
975
      code = pTaskInfo->code;
56,051✔
976
      goto _end;
56,051✔
977
    }
978

979
    if (pOperator->status == OP_EXEC_DONE) {
2,147,483,647✔
980
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
710,360✔
981
      break;
710,360✔
982
    }
983

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

990
    if (pBlock->info.id.uid) {
2,147,483,647✔
991
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
2,147,483,647✔
992
    }
993

994
    uint32_t status = 0;
2,147,483,647✔
995
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
2,147,483,647✔
996
    QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647✔
997

998
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
2,147,483,647✔
999
      break;
1,066,013✔
1000
    }
1001

1002
    // current block is filter out according to filter condition, continue load the next block
1003
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
2,147,483,647✔
1004
      continue;
82,996,118✔
1005
    }
1006

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

1010
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
2,147,483,647✔
1011
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
2,147,483,647✔
1012

1013
    (*ppRes) = pBlock;
2,147,483,647✔
1014
    return code;
2,147,483,647✔
1015
  }
1016

1017
_end:
685,589,706✔
1018
  if (code != TSDB_CODE_SUCCESS) {
685,629,422✔
1019
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
82,549!
1020
    pTaskInfo->code = code;
82,549✔
1021
    T_LONG_JMP(pTaskInfo->env, code);
60,660!
1022
  }
1023
  return code;
685,546,922✔
1024
}
1025

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

1033
  QRY_PARAM_CHECK(pBlock);
2,147,483,647!
1034

1035
  // The read handle is not initialized yet, since no qualified tables exists
1036
  if (pTableScanInfo->base.dataReader == NULL || pOperator->status == OP_EXEC_DONE) {
2,147,483,647!
1037
    return code;
11,833,435✔
1038
  }
1039

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

1046
    if (p != NULL) {
2,147,483,647✔
1047
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
2,147,483,647✔
1048
      *pBlock = p;
2,147,483,647✔
1049
      return code;
2,147,483,647✔
1050
    }
1051

1052
    pTableScanInfo->scanTimes += 1;
644,704,558✔
1053
    taosHashClear(pTableScanInfo->pIgnoreTables);
644,769,385✔
1054

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

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

1067
  int32_t total = pTableScanInfo->scanInfo.numOfAsc + pTableScanInfo->scanInfo.numOfDesc;
1,118,864,671✔
1068
  if (pTableScanInfo->scanTimes < total) {
1,118,924,287✔
1069
    if (pTableScanInfo->base.cond.order == TSDB_ORDER_ASC) {
167,547,046✔
1070
      prepareForDescendingScan(&pTableScanInfo->base, pOperator->exprSupp.pCtx, 0);
265,976✔
1071
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
265,976✔
1072
      QUERY_CHECK_CODE(code, lino, _end);
265,976!
1073
      qDebug("%s start to descending order scan data blocks due to query func required", GET_TASKID(pTaskInfo));
265,976!
1074
    }
1075

1076
    while (pTableScanInfo->scanTimes < total) {
208,370,879✔
1077
      SSDataBlock* p = NULL;
167,556,557✔
1078
      code = doTableScanImplNext(pOperator, &p);
167,550,987✔
1079
      QUERY_CHECK_CODE(code, lino, _end);
167,511,024!
1080

1081
      if (p != NULL) {
167,511,024✔
1082
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
126,708,383✔
1083
        *pBlock = p;
126,697,028✔
1084
        return code;
126,725,146✔
1085
      }
1086

1087
      pTableScanInfo->scanTimes += 1;
40,802,641✔
1088
      taosHashClear(pTableScanInfo->pIgnoreTables);
40,804,685✔
1089

1090
      if (pTableScanInfo->scanTimes < total) {
40,804,477!
1091
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
×
1092
        pTableScanInfo->base.scanFlag = MAIN_SCAN;
×
1093

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

1101
  if (pTableScanInfo->countState < TABLE_COUNT_STATE_END) {
992,117,489✔
1102
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
394,305,741✔
1103
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
394,299,014✔
1104
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
385,746✔
1105
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
18,304✔
1106
        STableKeyInfo* pStart =
1107
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
18,304✔
1108

1109
        if (NULL == pStart) {
18,304!
1110
          return code;
×
1111
        }
1112

1113
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
18,304✔
1114
        return code;
18,304✔
1115
      }
1116
    } else {  // group by tag + no sort
1117
      int32_t numOfTables = 0;
393,938,964✔
1118
      code = tableListGetSize(pTableListInfo, &numOfTables);
393,936,575✔
1119
      QUERY_CHECK_CODE(code, lino, _end);
393,952,806!
1120

1121
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
393,952,806✔
1122
        // get empty group, mark processed & rm from hash
1123
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
390,459,387✔
1124
        if (pIte != NULL) {
390,441,880✔
1125
          size_t        keySize = 0;
307,112,313✔
1126
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
307,127,571✔
1127
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
307,102,510✔
1128
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
307,124,722✔
1129
          markGroupProcessed(pTableScanInfo, *pGroupId);
307,104,270✔
1130
          *pBlock = getBlockForEmptyTable(pOperator, &info);
307,091,631✔
1131

1132
          return code;
307,134,833✔
1133
        }
1134
      }
1135
    }
1136
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
87,191,718✔
1137
  }
1138

1139
_end:
597,825,236✔
1140
  if (code != TSDB_CODE_SUCCESS) {
684,970,823!
1141
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1142
    pTaskInfo->code = code;
×
1143
  }
1144

1145
  return code;
684,970,823✔
1146
}
1147

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

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

1164
  if (pParam->tableSeq) {
1,196,621!
1165
    pListInfo->oneTableForEachGroup = true;
50,924✔
1166
    if (taosArrayGetSize(pListInfo->pTableList) > 0) {
50,924!
1167
      taosHashClear(pListInfo->map);
×
1168
      taosArrayClear(pListInfo->pTableList);
×
1169
      pOperator->status = OP_EXEC_DONE;
×
1170
    }
1171
  } else {
1172
    pListInfo->oneTableForEachGroup = false;
1,145,697✔
1173
    pListInfo->numOfOuputGroups = 1;
1,145,697✔
1174
  }
1175

1176
  STableKeyInfo info = {.groupId = 0};
1,196,621✔
1177
  int32_t       tableIdx = 0;
1,196,621✔
1178
  for (int32_t i = 0; i < num; ++i) {
47,376,849✔
1179
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
46,180,228✔
1180
    if (!pUid) {
46,175,420!
1181
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1182
      return terrno;
×
1183
    }
1184

1185
    if (taosHashPut(pListInfo->map, pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
46,175,420✔
1186
      if (TSDB_CODE_DUP_KEY == terrno) {
43,078,801!
1187
        continue;
43,079,044✔
1188
      }
1189
      return terrno;
×
1190
    }
1191

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

1198
    tableIdx++;
3,107,391✔
1199
    qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
3,107,391✔
1200
  }
1201

1202
  return code;
1,196,621✔
1203
}
1204

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

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

1219
  code = pAPI->tsdReader.tsdReaderOpen(pInfo->base.readHandle.vnode, &pInfo->base.cond, *pList, *pNum, pInfo->pResBlock,
1,505,590,932✔
1220
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
752,890,750✔
1221
  if (code) {
752,920,713✔
1222
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
30,915!
1223
  }
1224

1225
  return code;
752,589,980✔
1226
}
1227

1228
int compareColIdPair(const void* elem1, const void* elem2) {
450,222,444✔
1229
  SColIdPair* node1 = (SColIdPair*)elem1;
450,222,444✔
1230
  SColIdPair* node2 = (SColIdPair*)elem2;
450,222,444✔
1231

1232
  if (node1->orgColId < node2->orgColId) {
450,222,444✔
1233
    return -1;
449,851,644✔
1234
  }
1235

1236
  return node1->orgColId > node2->orgColId;
370,800✔
1237
}
1238

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

1255
  cleanupQueryTableDataCond(&pInfo->base.cond);
66,152,434✔
1256
  if (pAPI->tsdReader.tsdReaderClose) {
66,152,434!
1257
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
66,152,434✔
1258
  }
1259
  pInfo->base.dataReader = NULL;
66,152,434✔
1260

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

1283
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
66,152,434✔
1284
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
66,152,434!
1285
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(int32_t));
66,152,434✔
1286
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
66,152,434!
1287

1288
  // virtual table's origin table scan do not has ts column.
1289
  SColIdPair tsPair = {.vtbColId = PRIMARYKEY_TIMESTAMP_COL_ID, .orgColId = PRIMARYKEY_TIMESTAMP_COL_ID};
66,152,434✔
1290
  QUERY_CHECK_NULL(taosArrayPush(pColArray, &tsPair), code, lino, _return, terrno);
66,152,434!
1291

1292
  for (int32_t i = 0; i < taosArrayGetSize(pParam->pOrgTbInfo->colMap); ++i) {
171,004,172✔
1293
    SColIdNameKV* kv = taosArrayGet(pParam->pOrgTbInfo->colMap, i);
104,851,738✔
1294
    for (int32_t j = 0; j < schema->nCols; j++) {
826,007,016!
1295
      if (strcmp(kv->colName, schema->pSchema[j].name) == 0) {
826,007,016!
1296
        SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(j + 1)};
104,851,738✔
1297
        QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
104,851,738!
1298
        break;
104,851,738✔
1299
      }
1300
    }
1301
  }
1302

1303
  for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
237,156,606✔
1304
    SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, i);
171,004,172✔
1305
    for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
859,593,192!
1306
      SColMatchItem* pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
859,593,192✔
1307
      if (pItem->colId == pPair->vtbColId) {
859,593,192✔
1308
        SColIdPair tmpPair = {.orgColId = pPair->orgColId, .vtbColId = pItem->dstSlotId};
171,004,172✔
1309
        QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &tmpPair), code, lino, _return, terrno);
171,004,172!
1310
        break;
171,004,172✔
1311
      }
1312
    }
1313
  }
1314

1315
  if (pInfo->pResBlock) {
66,152,434✔
1316
    blockDataDestroy(pInfo->pResBlock);
60,468,482✔
1317
    pInfo->pResBlock = NULL;
60,468,482✔
1318
  }
1319
  taosArraySort(pColArray, compareColIdPair);
66,152,434✔
1320
  taosArraySort(pBlockColArray, compareColIdPair);
66,152,434✔
1321
  code = createOneDataBlockWithColArray(pInfo->pOrgBlock, pBlockColArray, &pInfo->pResBlock);
66,152,434✔
1322
  QUERY_CHECK_CODE(code, lino, _return);
66,152,434!
1323
  code =
1324
      initQueryTableDataCondWithColArray(&pInfo->base.cond, &pInfo->base.orgCond, &pInfo->base.readHandle, pColArray);
66,152,434✔
1325
  QUERY_CHECK_CODE(code, lino, _return);
66,152,434!
1326
  if (pParam->window.ekey > 0) {
66,152,434✔
1327
    pInfo->base.cond.twindows.skey = pParam->window.ekey + 1;
49,485,090✔
1328
  }
1329
  pInfo->base.cond.suid = orgTable.me.type == TSDB_CHILD_TABLE ? superTable.me.uid : 0;
66,152,434✔
1330
  pInfo->currentGroupId = 0;
66,152,434✔
1331
  pInfo->ignoreTag = true;
66,152,434✔
1332

1333
  pListInfo->oneTableForEachGroup = true;
66,152,434✔
1334
  taosHashClear(pListInfo->map);
66,152,434✔
1335
  taosArrayClear(pListInfo->pTableList);
66,152,434✔
1336

1337
  uint64_t      pUid = orgTable.me.uid;
66,152,434✔
1338
  STableKeyInfo info = {.groupId = 0, .uid = pUid};
66,152,434✔
1339
  int32_t       tableIdx = 0;
66,152,434✔
1340
  code = taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t));
66,152,434✔
1341
  QUERY_CHECK_CODE(code, lino, _return);
66,152,434!
1342
  QUERY_CHECK_NULL(taosArrayPush(pListInfo->pTableList, &info), code, lino, _return, terrno);
132,304,868!
1343
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
66,152,434✔
1344

1345
  pOperator->status = OP_OPENED;
66,152,434✔
1346

1347
  taosRLockLatch(&pTaskInfo->lock);
66,152,434✔
1348
  code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
66,152,434✔
1349
  taosRUnLockLatch(&pTaskInfo->lock);
66,152,434✔
1350
  QUERY_CHECK_CODE(code, lino, _return);
66,152,434!
1351

1352
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
66,152,434!
1353
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1354
  }
1355

1356
  pInfo->currentGroupId = -1;
66,152,434✔
1357
_return:
66,152,434✔
1358
  if (code) {
66,152,434!
1359
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1360
  }
1361
  taosArrayDestroy(pColArray);
66,152,434✔
1362
  taosArrayDestroy(pBlockColArray);
66,152,434✔
1363
  pAPI->metaReaderFn.clearReader(&superTable);
66,152,434✔
1364
  pAPI->metaReaderFn.clearReader(&orgTable);
66,152,434✔
1365
  return code;
66,152,434✔
1366
}
1367

1368
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
746,986,377✔
1369
  int32_t         code = TSDB_CODE_SUCCESS;
746,986,377✔
1370
  int32_t         lino = 0;
746,986,377✔
1371
  STableScanInfo* pInfo = pOperator->info;
746,986,377✔
1372
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
747,036,939✔
1373
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
747,032,443✔
1374
  int32_t         numOfTables = 0;
747,092,821✔
1375

1376
  QRY_PARAM_CHECK(pResult);
747,023,062!
1377

1378
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
747,028,692✔
1379
  QUERY_CHECK_CODE(code, lino, _end);
747,099,994!
1380

1381
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
747,099,994✔
1382
    setOperatorCompleted(pOperator);
679,764,554✔
1383
    if (pOperator->dynamicTask) {
679,782,027!
1384
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
17,721,504✔
1385
      taosHashClear(pInfo->base.pTableListInfo->map);
17,721,504✔
1386
    }
1387
    return code;
679,753,559✔
1388
  }
1389

1390
  // reset value for the next group data output
1391
  pOperator->status = OP_OPENED;
67,261,358✔
1392
  resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
67,262,632✔
1393

1394
  int32_t        num = 0;
67,261,597✔
1395
  STableKeyInfo* pList = NULL;
67,261,597✔
1396
  code = initNextGroupScan(pInfo, &pList, &num);
67,261,597✔
1397
  QUERY_CHECK_CODE(code, lino, _end);
67,261,358!
1398

1399
  code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
67,261,358✔
1400
  QUERY_CHECK_CODE(code, lino, _end);
67,262,966!
1401

1402
  code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
67,262,966✔
1403
  QUERY_CHECK_CODE(code, lino, _end);
67,261,692!
1404
  pInfo->scanTimes = 0;
67,261,692✔
1405

1406
  code = doGroupedTableScan(pOperator, pResult);
67,261,692✔
1407
  QUERY_CHECK_CODE(code, lino, _end);
67,258,886!
1408

1409
  if (*pResult != NULL) {
67,258,886✔
1410
    if (pOperator->dynamicTask) {
50,563,301!
1411
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
49,632,095✔
1412
    }
1413
    return code;
50,563,301✔
1414
  }
1415

1416
_end:
16,695,585✔
1417
  if (code != TSDB_CODE_SUCCESS) {
16,696,921!
1418
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1419
    pTaskInfo->code = code;
×
1420
  }
1421
  return code;
16,696,921✔
1422
}
1423

1424
static int32_t groupSeqTableScan(SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
2,147,483,647✔
1425
  int32_t         code = TSDB_CODE_SUCCESS;
2,147,483,647✔
1426
  int32_t         lino = 0;
2,147,483,647✔
1427
  STableScanInfo* pInfo = pOperator->info;
2,147,483,647✔
1428
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
2,147,483,647✔
1429
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
2,147,483,647✔
1430
  int32_t         num = 0;
2,147,483,647✔
1431
  STableKeyInfo*  pList = NULL;
2,147,483,647✔
1432
  SSDataBlock*    pResult = NULL;
2,147,483,647✔
1433
  const char*     idStr = GET_TASKID(pTaskInfo);
2,147,483,647✔
1434

1435
  QRY_PARAM_CHECK(pResBlock);
2,147,483,647!
1436

1437
  if (pInfo->currentGroupId == -1) {
2,147,483,647✔
1438
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
686,693,403!
1439
      setOperatorCompleted(pOperator);
×
1440
      return code;
×
1441
    }
1442

1443
    taosRLockLatch(&pTaskInfo->lock);
686,931,911✔
1444
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
687,013,153✔
1445
    taosRUnLockLatch(&pTaskInfo->lock);
686,745,195✔
1446
    QUERY_CHECK_CODE(code, lino, _end);
686,947,546✔
1447

1448
    if (pInfo->filesetDelimited) {
686,916,631!
1449
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
1,165,654✔
1450
    }
1451

1452
    if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
686,618,176!
1453
      pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1454
    }
1455
  }
1456

1457
  code = doGroupedTableScan(pOperator, &pResult);
2,147,483,647✔
1458
  QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
1459

1460
  if (pResult != NULL) {
2,147,483,647✔
1461
    if (pOperator->dynamicTask) {
2,147,483,647!
1462
      pResult->info.id.groupId = pResult->info.id.uid;
2,923,520✔
1463
    }
1464

1465
    *pResBlock = pResult;
2,147,483,647✔
1466
    return code;
2,147,483,647✔
1467
  }
1468

1469
  while (true) {
1470
    code = startNextGroupScan(pOperator, &pResult);
663,677,756✔
1471
    QUERY_CHECK_CODE(code, lino, _end);
664,394,682!
1472

1473
    if (pResult || pOperator->status == OP_EXEC_DONE) {
664,394,682✔
1474
      *pResBlock = pResult;
664,198,889✔
1475
      return code;
664,145,153✔
1476
    }
1477
  }
1478

1479
_end:
30,915✔
1480
  if (code != TSDB_CODE_SUCCESS) {
30,915!
1481
    qError("%s %s failed at line %d since %s", idStr, __func__, lino, tstrerror(code));
30,915!
1482
    pTaskInfo->code = code;
30,915✔
1483
  }
1484

1485
  return code;
30,915✔
1486
}
1487

1488
static bool isEmptyQueryTimeWindow(STimeWindow* pWindow) {
66,152,434✔
1489
  return (pWindow == NULL) || (pWindow->skey > pWindow->ekey);
66,152,434!
1490
}
1491

1492
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
2,147,483,647✔
1493
  int32_t         code = TSDB_CODE_SUCCESS;
2,147,483,647✔
1494
  int32_t         lino = 0;
2,147,483,647✔
1495
  STableScanInfo* pInfo = pOperator->info;
2,147,483,647✔
1496
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
2,147,483,647✔
1497
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
2,147,483,647✔
1498
  QRY_PARAM_CHECK(ppRes);
2,147,483,647!
1499
  qTrace("%s call", __FUNCTION__);
2,147,483,647!
1500
  if (pOperator->pOperatorGetParam) {
2,147,483,647✔
1501
    pOperator->dynamicTask = true;
67,349,055✔
1502
    if (isDynVtbScan(pOperator)) {
67,349,055✔
1503
      code = createVTableScanInfoFromParam(pOperator);
66,152,434✔
1504

1505
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
66,152,434✔
1506
      pOperator->pOperatorGetParam = NULL;
66,152,434✔
1507
      QUERY_CHECK_CODE(code, lino, _end);
66,152,434!
1508

1509
      SSDataBlock* result = NULL;
66,152,434✔
1510
      if (isEmptyQueryTimeWindow(&pInfo->base.cond.twindows) && pInfo->base.cond.type == TIMEWINDOW_RANGE_CONTAINED) {
66,152,434!
1511
        (*ppRes) = result;
5,364✔
1512
        return code;
5,364✔
1513
      }
1514
      while (true) {
1515
        code = startNextGroupScan(pOperator, &result);
82,687,507✔
1516
        QUERY_CHECK_CODE(code, lino, _end);
82,686,171!
1517

1518
        if (result || pOperator->status == OP_EXEC_DONE) {
82,686,171✔
1519
          SSDataBlock* res = NULL;
66,145,734✔
1520
          if (result) {
66,147,070✔
1521
            code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, &res);
49,606,633✔
1522
            QUERY_CHECK_CODE(code, lino, _end);
49,606,633!
1523
            pInfo->pResBlock = res;
49,606,633✔
1524
            blockDataDestroy(result);
49,606,633✔
1525
          }
1526
          (*ppRes) = res;
66,147,070✔
1527
          return code;
66,147,070✔
1528
        }
1529
      }
1530
    } else {
1531
      code = createTableListInfoFromParam(pOperator);
1,196,621✔
1532
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
1,196,621✔
1533
      pOperator->pOperatorGetParam = NULL;
1,196,621✔
1534
      QUERY_CHECK_CODE(code, lino, _end);
1,196,621!
1535

1536
      if (pOperator->status == OP_EXEC_DONE) {
1,196,621✔
1537
        pInfo->currentGroupId = -1;
25,462✔
1538
        pOperator->status = OP_OPENED;
25,462✔
1539
        SSDataBlock* result = NULL;
25,462✔
1540

1541
        while (true) {
1542
          code = startNextGroupScan(pOperator, &result);
25,462✔
1543
          QUERY_CHECK_CODE(code, lino, _end);
25,462!
1544

1545
          if (result || pOperator->status == OP_EXEC_DONE) {
25,462!
1546
            (*ppRes) = result;
25,462✔
1547
            return code;
25,462✔
1548
          }
1549
        }
1550
      }
1551
    }
1552
  }
1553

1554
  // scan table one by one sequentially
1555
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
2,147,483,647✔
1556
    int32_t       numOfTables = 0;
16,999,319✔
1557
    STableKeyInfo tInfo = {0};
16,999,319✔
1558
    pInfo->countState = TABLE_COUNT_STATE_END;
16,999,319✔
1559

1560
    while (1) {
15,794,344✔
1561
      SSDataBlock* result = NULL;
32,793,663✔
1562
      code = doGroupedTableScan(pOperator, &result);
32,794,534✔
1563
      QUERY_CHECK_CODE(code, lino, _end);
32,763,292!
1564

1565
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
32,763,292!
1566
        (*ppRes) = result;
16,845,835✔
1567
        return code;
16,855,423✔
1568
      }
1569

1570
      // if no data, switch to next table and continue scan
1571
      pInfo->currentTable++;
15,930,485✔
1572

1573
      taosRLockLatch(&pTaskInfo->lock);
15,922,624✔
1574
      numOfTables = 0;
15,939,099✔
1575
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
15,939,099✔
1576
      if (code != TSDB_CODE_SUCCESS) {
15,915,662!
1577
        taosRUnLockLatch(&pTaskInfo->lock);
×
1578
        TSDB_CHECK_CODE(code, lino, _end);
×
1579
      }
1580

1581
      if (pInfo->currentTable >= numOfTables) {
15,915,662✔
1582
        qDebug("all table checked in table list, total:%d, return NULL, %s", numOfTables, GET_TASKID(pTaskInfo));
130,909!
1583
        taosRUnLockLatch(&pTaskInfo->lock);
130,909✔
1584
        (*ppRes) = NULL;
130,909✔
1585
        return code;
130,909✔
1586
      }
1587

1588
      STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
15,770,066✔
1589
      if (!tmp) {
15,784,800!
1590
        taosRUnLockLatch(&pTaskInfo->lock);
×
1591
        (*ppRes) = NULL;
×
1592
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1593
      }
1594

1595
      tInfo = *tmp;
15,784,800✔
1596
      taosRUnLockLatch(&pTaskInfo->lock);
15,763,970✔
1597

1598
      code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, &tInfo, 1);
15,807,336✔
1599
      QUERY_CHECK_CODE(code, lino, _end);
15,806,498!
1600
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d/%d %s", tInfo.uid, numOfTables,
15,806,498✔
1601
             pInfo->currentTable, numOfTables, GET_TASKID(pTaskInfo));
1602

1603
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
15,822,054✔
1604
      QUERY_CHECK_CODE(code, lino, _end);
15,776,987!
1605
      pInfo->scanTimes = 0;
15,776,987✔
1606
    }
1607
  } else {  // scan table group by group sequentially
1608
    code = groupSeqTableScan(pOperator, ppRes);
2,147,483,647✔
1609
    QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647✔
1610
  }
1611

1612
_end:
2,147,483,647✔
1613
  if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
1614
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
30,915!
1615
    pTaskInfo->code = code;
30,915✔
1616
    T_LONG_JMP(pTaskInfo->env, code);
30,915!
1617
  }
1618

1619
  return code;
2,147,483,647✔
1620
}
1621

1622
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
11,311,405✔
1623
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
11,311,405!
1624
  if (!pRecorder) {
11,310,808!
1625
    return terrno;
×
1626
  }
1627
  STableScanInfo* pTableScanInfo = pOptr->info;
11,310,808✔
1628
  *pRecorder = pTableScanInfo->base.readRecorder;
11,310,808✔
1629
  *pOptrExplain = pRecorder;
11,312,007✔
1630
  *len = sizeof(SFileBlockLoadRecorder);
11,312,007✔
1631
  return 0;
11,312,007✔
1632
}
1633

1634
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
870,182,085✔
1635
  cleanupQueryTableDataCond(&pBase->cond);
870,182,085✔
1636
  cleanupQueryTableDataCond(&pBase->orgCond);
870,171,383✔
1637

1638
  if (pAPI->tsdReaderClose) {
870,178,019✔
1639
    pAPI->tsdReaderClose(pBase->dataReader);
870,204,976✔
1640
  }
1641
  pBase->dataReader = NULL;
870,025,126✔
1642

1643
  if (pBase->matchInfo.pList != NULL) {
870,046,733!
1644
    taosArrayDestroy(pBase->matchInfo.pList);
870,079,809✔
1645
  }
1646

1647
  tableListDestroy(pBase->pTableListInfo);
869,979,737✔
1648
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
870,154,143✔
1649
  cleanupExprSupp(&pBase->pseudoSup);
870,252,826✔
1650
}
870,221,068✔
1651

1652
static void destroyTableScanOperatorInfo(void* param) {
726,299,814✔
1653
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
726,299,814✔
1654
  blockDataDestroy(pTableScanInfo->pResBlock);
726,299,814✔
1655
  blockDataDestroy(pTableScanInfo->pOrgBlock);
726,282,648✔
1656
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
726,292,775✔
1657
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
726,240,046✔
1658
  taosMemoryFreeClear(param);
726,279,344!
1659
}
726,246,891✔
1660

1661
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
725,906,399✔
1662
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
725,906,399!
1663
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
43,596,326✔
1664
    for (int32_t i = 0; i < numOfCols; ++i) {
108,054,649✔
1665
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
64,501,832✔
1666
      if (pCol) {
64,486,426!
1667
        pCol->info.noData = true;
64,488,445✔
1668
      }
1669
    }
1670
  }
1671
}
725,862,890✔
1672

1673
int32_t createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
726,235,049✔
1674
                                    STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
1675
                                    SOperatorInfo** pOptrInfo) {
1676
  QRY_PARAM_CHECK(pOptrInfo);
726,235,049!
1677

1678
  int32_t         code = TSDB_CODE_SUCCESS;
726,273,419✔
1679
  int32_t         lino = 0;
726,273,419✔
1680
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
726,273,419✔
1681
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
726,167,435✔
1682
  if (pInfo == NULL || pOperator == NULL) {
726,144,969!
1683
    code = terrno;
×
1684
    goto _error;
×
1685
  }
1686

1687
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
726,149,401✔
1688
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
726,151,692✔
1689

1690
  int32_t numOfCols = 0;
726,162,511✔
1691
  code =
1692
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
726,179,198✔
1693
  QUERY_CHECK_CODE(code, lino, _error);
726,117,274!
1694

1695
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
726,117,274✔
1696
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, true);
725,985,944✔
1697
  QUERY_CHECK_CODE(code, lino, _error);
725,676,649!
1698

1699
  if (pScanNode->pScanPseudoCols != NULL) {
725,676,649✔
1700
    SExprSupp* pSup = &pInfo->base.pseudoSup;
269,380,584✔
1701
    pSup->pExprInfo = NULL;
269,443,943✔
1702
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
269,421,076✔
1703
    QUERY_CHECK_CODE(code, lino, _error);
269,349,450!
1704

1705
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
269,349,450✔
1706
                                      &pTaskInfo->storageAPI.functionStore);
1707
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
269,202,799!
1708
  }
1709

1710
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
725,552,894✔
1711
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
725,715,630✔
1712

1713
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
725,337,844✔
1714
  pInfo->base.readHandle = *readHandle;
725,223,402✔
1715
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
725,456,933✔
1716

1717
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
725,128,584✔
1718
  pInfo->sample.seed = taosGetTimestampSec();
725,450,506✔
1719

1720
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
725,804,775✔
1721
  initResultSizeInfo(&pOperator->resultInfo, 4096);
725,651,218✔
1722
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
725,510,909✔
1723
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
726,040,634✔
1724
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
725,734,845!
1725

1726
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
725,347,704✔
1727
  QUERY_CHECK_CODE(code, lino, _error);
725,153,995!
1728

1729
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
725,153,995!
1730
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
725,371,686!
1731
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
8,414,135✔
1732
    pInfo->pResBlock = NULL;
8,414,298✔
1733
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
8,417,353!
1734
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
8,414,138!
1735
  }
1736

1737
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
725,605,695✔
1738
                            pTaskInfo->pStreamRuntimeInfo);
725,361,576✔
1739
  QUERY_CHECK_CODE(code, lino, _error);
725,603,593✔
1740

1741
  pInfo->currentGroupId = -1;
725,561,974✔
1742

1743
  pInfo->tableEndIndex = -1;
725,614,193✔
1744
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
725,449,128!
1745
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
725,525,685✔
1746

1747
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
725,359,508✔
1748
                  pTaskInfo);
1749
  pOperator->exprSupp.numOfExprs = numOfCols;
725,247,568✔
1750

1751
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
725,516,637!
1752
  pInfo->ignoreTag = false;
725,455,218✔
1753

1754
  pInfo->base.pTableListInfo = pTableListInfo;
725,419,519✔
1755
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
725,222,175✔
1756
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
725,835,909!
1757
    code = terrno;
×
1758
    QUERY_CHECK_CODE(code, lino, _error);
×
1759
  }
1760

1761
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
726,000,908!
1762

1763
  taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
725,971,673✔
1764
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
726,041,156✔
1765
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
1766

1767
  // for non-blocking operator, the open cost is always 0
1768
  pOperator->cost.openCost = 0;
725,845,266✔
1769
  *pOptrInfo = pOperator;
725,785,573✔
1770
  return TSDB_CODE_SUCCESS;
725,613,769✔
1771

1772
_error:
41,619✔
1773
  if (pInfo != NULL) {
41,619!
1774
    pInfo->base.pTableListInfo = NULL;  // this attribute will be destroy outside of this function
41,619✔
1775
    destroyTableScanOperatorInfo(pInfo);
41,619✔
1776
  }
1777

1778
  if (pOperator != NULL) {
41,619!
1779
    pOperator->info = NULL;
41,619✔
1780
    destroyOperator(pOperator);
41,619✔
1781
  }
1782
  pTaskInfo->code = code;
41,619✔
1783
  return code;
41,619✔
1784
}
1785

1786
int32_t createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
×
1787
  QRY_PARAM_CHECK(pOptrInfo);
×
1788

1789
  int32_t         code = 0;
×
1790
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
×
1791
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
1792
  if (pInfo == NULL || pOperator == NULL) {
×
1793
    code = terrno;
×
1794
    goto _end;
×
1795
  }
1796

1797
  pInfo->base.dataReader = pReadHandle;
×
1798
  //  pInfo->prevGroupId       = -1;
1799

1800
  setOperatorInfo(pOperator, "TableSeqScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN, false, OP_NOT_OPENED,
×
1801
                  pInfo, pTaskInfo);
1802
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanImplNext, NULL, NULL, optrDefaultBufFn, NULL,
×
1803
                                         optrDefaultGetNextExtFn, NULL);
1804
  *pOptrInfo = pOperator;
×
1805
  return code;
×
1806

1807
_end:
×
1808
  if (pInfo != NULL) {
×
1809
    taosMemoryFree(pInfo);
×
1810
  }
1811

1812
  if (pOperator != NULL) {
×
1813
    taosMemoryFree(pOperator);
×
1814
  }
1815

1816
  pTaskInfo->code = code;
×
1817
  return code;
×
1818
}
1819

1820
static int32_t doBlockDataPrimaryKeyFilter(SSDataBlock* pBlock, STqOffsetVal* offset) {
×
1821
  int32_t code = TSDB_CODE_SUCCESS;
×
1822
  int32_t lino = 0;
×
1823
  if (pBlock->info.window.skey != offset->ts || offset->primaryKey.type == 0) {
×
1824
    return code;
×
1825
  }
1826
  bool* p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
×
1827
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
×
1828
  bool hasUnqualified = false;
×
1829

1830
  SColumnInfoData* pColTs = taosArrayGet(pBlock->pDataBlock, 0);
×
1831
  SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
1832

1833
  qDebug("doBlockDataWindowFilter primary key, ts:%" PRId64 " %" PRId64, offset->ts,
×
1834
         VALUE_GET_TRIVIAL_DATUM(&offset->primaryKey));
1835
  QUERY_CHECK_CONDITION((pColPk->info.type == offset->primaryKey.type), code, lino, _end,
×
1836
                        TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
1837

1838
  __compar_fn_t func = getComparFunc(pColPk->info.type, 0);
×
1839
  QUERY_CHECK_NULL(func, code, lino, _end, terrno);
×
1840
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
×
1841
    int64_t* ts = (int64_t*)colDataGetData(pColTs, i);
×
1842
    void*    data = colDataGetData(pColPk, i);
×
1843
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
1844
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
×
1845
        QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY, lino, _end);
×
1846
      }
1847
      void* tmq = taosMemoryMalloc(offset->primaryKey.nData + VARSTR_HEADER_SIZE);
×
1848
      QUERY_CHECK_NULL(tmq, code, lino, _end, terrno);
×
1849
      memcpy(varDataVal(tmq), offset->primaryKey.pData, offset->primaryKey.nData);
×
1850
      varDataLen(tmq) = offset->primaryKey.nData;
×
1851
      p[i] = (*ts > offset->ts) || (func(data, tmq) > 0);
×
1852
      taosMemoryFree(tmq);
×
1853
    } else {
1854
      p[i] = (*ts > offset->ts) || (func(data, VALUE_GET_DATUM(&offset->primaryKey, pColPk->info.type)) > 0);
×
1855
    }
1856

1857
    if (!p[i]) {
×
1858
      hasUnqualified = true;
×
1859
    }
1860
  }
1861

1862
  if (hasUnqualified) {
×
1863
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
×
1864
    QUERY_CHECK_CODE(code, lino, _end);
×
1865
  }
1866

1867
_end:
×
1868
  taosMemoryFree(p);
×
1869
  if (code != TSDB_CODE_SUCCESS) {
×
1870
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1871
  }
1872
  return code;
×
1873
}
1874

1875
int32_t colIdComparFn(const void* param1, const void* param2) {
2,147,483,647✔
1876
  int32_t p1 = *(int32_t*)param1;
2,147,483,647✔
1877
  int32_t p2 = *(int32_t*)param2;
2,147,483,647✔
1878

1879
  if (p1 == p2) {
2,147,483,647!
1880
    return 0;
×
1881
  } else {
1882
    return (p1 < p2) ? -1 : 1;
2,147,483,647✔
1883
  }
1884
}
1885

1886
static int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock) {
202,045,349✔
1887
  int32_t         code = TSDB_CODE_SUCCESS;
202,045,349✔
1888
  int32_t         lino = 0;
202,045,349✔
1889
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
202,045,349✔
1890
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
202,045,269✔
1891
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
202,046,174✔
1892
  const char*     id = GET_TASKID(pTaskInfo);
202,045,269✔
1893
  bool            isVtableSourceScan = (pTaskInfo->pSubplan->pVTables != NULL);
202,044,364✔
1894

1895
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
202,045,280✔
1896
  QUERY_CHECK_CODE(code, lino, _end);
202,045,269!
1897

1898
  pBlockInfo->rows = pBlock->info.rows;
202,045,269✔
1899
  pBlockInfo->id.uid = pBlock->info.id.uid;
202,038,945✔
1900
  pBlockInfo->type = STREAM_NORMAL;
202,044,375✔
1901
  pBlockInfo->version = pBlock->info.version;
202,044,375✔
1902

1903
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
202,040,755✔
1904
  if (!isVtableSourceScan) {
202,045,280!
1905
    pBlockInfo->id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
202,045,280✔
1906
  } else {
1907
    // use original table uid as groupId for vtable
1908
    pBlockInfo->id.groupId = pBlock->info.id.groupId;
×
1909
  }
1910

1911
  SArray* pColList = taosArrayInit(4, sizeof(int32_t));
202,042,554✔
1912
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
202,044,375✔
1913

1914
  // todo extract method
1915
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
1,135,550,106✔
1916
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
933,529,001✔
1917
    if (!pColMatchInfo->needOutput) {
933,481,658!
1918
      continue;
1,680✔
1919
    }
1920

1921
    bool colExists = false;
933,480,465✔
1922
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
2,147,483,647!
1923
      SColumnInfoData* pResCol = NULL;
2,147,483,647✔
1924
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
2,147,483,647✔
1925
      QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
1926

1927
      if (pResCol->info.colId == pColMatchInfo->colId) {
2,147,483,647✔
1928
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
933,603,031✔
1929
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
933,585,712✔
1930
        QUERY_CHECK_CODE(code, lino, _end);
933,258,959!
1931

1932
        colExists = true;
933,258,959✔
1933
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
933,258,959✔
1934
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
933,490,230!
1935
        break;
933,490,230✔
1936
      }
1937
    }
1938

1939
    // the required column does not exists in submit block, let's set it to be all null value
1940
    if (!colExists) {
933,504,956✔
1941
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
3,356✔
1942
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
3,356!
1943
      void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
3,356✔
1944
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,356!
1945
    }
1946
  }
1947

1948
  // currently only the tbname pseudo column
1949
  if (pInfo->numOfPseudoExpr > 0 && !isVtableSourceScan) {
202,038,036!
1950
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
208,253,459✔
1951
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
104,126,343✔
1952
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
1953
    if (code) {
104,117,293!
1954
      QUERY_CHECK_CODE(code, lino, _end);
×
1955
    }
1956

1957
    // reset the error code.
1958
    terrno = 0;
104,117,293✔
1959

1960
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
528,730,546✔
1961
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
424,587,072✔
1962
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
424,605,108!
1963
    }
1964
  }
1965

1966
  taosArraySort(pColList, colIdComparFn);
201,996,509✔
1967

1968
  int32_t i = 0, j = 0;
202,026,467✔
1969
  while (i < taosArrayGetSize(pColList)) {
1,559,884,564✔
1970
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
1,357,792,060✔
1971
    if (slot1 > j) {
1,357,858,097!
1972
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
×
1973
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
1974
      j += 1;
×
1975
    } else {
1976
      i += 1;
1,357,858,097✔
1977
      j += 1;
1,357,858,097✔
1978
    }
1979
  }
1980

1981
  while (j < taosArrayGetSize(pInfo->pRes->pDataBlock)) {
202,039,869✔
1982
    SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
1,680✔
1983
    colDataSetNNULL(pDst, 0, pBlockInfo->rows);
1,680!
1984
    j += 1;
1,680✔
1985
  }
1986

1987
  taosArrayDestroy(pColList);
202,032,851✔
1988

1989
  code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL, NULL);
202,035,542✔
1990
  QUERY_CHECK_CODE(code, lino, _end);
202,030,554!
1991

1992
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
202,030,554✔
1993
  QUERY_CHECK_CODE(code, lino, _end);
202,032,803!
1994
  if (pInfo->pRes->info.rows == 0) {
202,032,803✔
1995
    return 0;
2,045,260✔
1996
  }
1997

1998
_end:
199,967,838✔
1999
  if (code != TSDB_CODE_SUCCESS) {
197,593,101!
2000
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2001
  }
2002
  return code;
199,978,716✔
2003
}
2004

2005
static int32_t processPrimaryKey(SSDataBlock* pBlock, bool hasPrimaryKey, STqOffsetVal* offset) {
19,015,810✔
2006
  int32_t code = TSDB_CODE_SUCCESS;
19,015,810✔
2007
  int32_t lino = 0;
19,015,810✔
2008
  SValue  val = {0};
19,015,810✔
2009
  if (hasPrimaryKey) {
19,018,379!
2010
    code = doBlockDataPrimaryKeyFilter(pBlock, offset);
×
2011
    if (code != TSDB_CODE_SUCCESS) {
×
2012
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
2013
      return code;
×
2014
    }
2015
    SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
2016

2017
    if (pBlock->info.rows < 1) {
×
2018
      return code;
×
2019
    }
2020
    void* tmp = colDataGetData(pColPk, pBlock->info.rows - 1);
×
2021
    val.type = pColPk->info.type;
×
2022
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
2023
      if (IS_STR_DATA_BLOB(pColPk->info.type)) {
×
2024
        return TSDB_CODE_BLOB_NOT_SUPPORT_PRIMARY_KEY;
×
2025
      }
2026
      val.pData = taosMemoryMalloc(varDataLen(tmp));
×
2027
      QUERY_CHECK_NULL(val.pData, code, lino, _end, terrno);
×
2028
      val.nData = varDataLen(tmp);
×
2029
      memcpy(val.pData, varDataVal(tmp), varDataLen(tmp));
×
2030
    } else {
2031
      valueSetDatum(&val, pColPk->info.type, tmp, pColPk->info.bytes);
×
2032
    }
2033
  }
2034
  tqOffsetResetToData(offset, pBlock->info.id.uid, pBlock->info.window.ekey, val);
19,018,379!
2035

2036
_end:
19,013,234✔
2037
  if (code != TSDB_CODE_SUCCESS) {
19,014,085!
2038
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2039
  }
2040
  return code;
19,014,980✔
2041
}
2042

2043
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
254,031,547✔
2044
  int32_t        code = TSDB_CODE_SUCCESS;
254,031,547✔
2045
  int32_t        lino = 0;
254,031,547✔
2046
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
254,031,547✔
2047
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
254,037,706✔
2048

2049
  SStreamScanInfo* pInfo = pOperator->info;
254,039,626✔
2050
  const char*      id = GET_TASKID(pTaskInfo);
254,039,648✔
2051

2052
  qDebug("start to exec queue scan, %s", id);
254,031,134!
2053

2054
  if (isTaskKilled(pTaskInfo)) {
254,042,677!
2055
    (*ppRes) = NULL;
×
2056
    return pTaskInfo->code;
×
2057
  }
2058

2059
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
254,050,237✔
2060
    while (1) {
×
2061
      SSDataBlock* pResult = NULL;
16,999,319✔
2062
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
16,999,319✔
2063
      QUERY_CHECK_CODE(code, lino, _end);
16,988,069!
2064

2065
      if (pResult && pResult->info.rows > 0) {
16,988,069!
2066
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
16,862,363✔
2067
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
16,862,335✔
2068
        QUERY_CHECK_CODE(code, lino, _end);
16,849,431!
2069
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
16,849,431!
2070
        if (pResult->info.rows > 0) {
16,864,992!
2071
          (*ppRes) = pResult;
16,868,410✔
2072
          return code;
16,868,410✔
2073
        }
2074
      } else {
2075
        break;
2076
      }
2077
    }
2078

2079
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
130,909✔
2080
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
130,909✔
2081

2082
    pTSInfo->base.dataReader = NULL;
130,909✔
2083
    int64_t validVer = pTaskInfo->streamInfo.snapshotVer + 1;
130,909✔
2084
    qDebug("queue scan tsdb over, switch to wal ver %" PRId64, validVer);
130,909!
2085
    if (pAPI->tqReaderFn.tqReaderSeek(pInfo->tqReader, validVer, pTaskInfo->id.str) < 0) {
130,909✔
2086
      (*ppRes) = NULL;
41,210✔
2087
      return code;
41,210✔
2088
    }
2089

2090
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, validVer);
89,699✔
2091
  }
2092

2093
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
237,139,750!
2094
    while (1) {
2,045,260✔
2095
      bool hasResult =
2096
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
239,184,067✔
2097

2098
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
239,148,121✔
2099
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
239,169,840✔
2100

2101
      // curVersion move to next
2102
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pWalReader->curVersion);
239,168,913✔
2103

2104
      // use ts to pass time when replay, because ts not used if type is log
2105
      pTaskInfo->streamInfo.currentOffset.ts = pAPI->tqReaderFn.tqGetResultBlockTime(pInfo->tqReader);
239,159,951✔
2106

2107
      if (hasResult) {
239,166,203✔
2108
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
202,027,316✔
2109
               pTaskInfo->streamInfo.currentOffset.version);
2110
        blockDataCleanup(pInfo->pRes);
202,043,978✔
2111
        code = setBlockIntoRes(pInfo, pRes);
202,043,459✔
2112
        QUERY_CHECK_CODE(code, lino, _end);
202,003,205!
2113
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
202,003,205!
2114
               pTaskInfo->streamInfo.currentOffset.version);
2115
        if (pInfo->pRes->info.rows > 0) {
202,026,321✔
2116
          (*ppRes) = pInfo->pRes;
200,000,009✔
2117
          return code;
199,999,104✔
2118
        }
2119
      } else {
2120
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
37,138,887!
2121
        (*ppRes) = NULL;
37,139,703✔
2122
        return code;
37,139,703✔
2123
      }
2124
    }
2125
  } else {
2126
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
×
2127
    (*ppRes) = NULL;
×
2128
    return code;
×
2129
  }
2130

2131
_end:
×
2132
  if (code != TSDB_CODE_SUCCESS) {
×
2133
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2134
    pTaskInfo->code = code;
×
2135
    T_LONG_JMP(pTaskInfo->env, code);
×
2136
  }
2137
  (*ppRes) = NULL;
×
2138
  return code;
×
2139
}
2140

2141
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
×
2142
  SSDataBlock* pRes = NULL;
×
2143
  int32_t      code = doQueueScanNext(pOperator, &pRes);
×
2144
  return pRes;
×
2145
}
2146

2147
int32_t extractTableIdList(const STableListInfo* pTableListInfo, SArray** ppArrayRes) {
908,561✔
2148
  int32_t code = TSDB_CODE_SUCCESS;
908,561✔
2149
  int32_t lino = 0;
908,561✔
2150
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));
908,561✔
2151
  QUERY_CHECK_NULL(tableIdList, code, lino, _end, terrno);
914,678!
2152

2153
  // Transfer the Array of STableKeyInfo into uid list.
2154
  int32_t size = 0;
914,678✔
2155
  code = tableListGetSize(pTableListInfo, &size);
913,789✔
2156
  QUERY_CHECK_CODE(code, lino, _end);
915,548!
2157
  for (int32_t i = 0; i < size; ++i) {
27,914,567✔
2158
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
26,997,321✔
2159
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
26,956,338!
2160
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
26,956,338✔
2161
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
26,999,019!
2162
  }
2163

2164
  (*ppArrayRes) = tableIdList;
917,246✔
2165

2166
_end:
913,822✔
2167
  if (code != TSDB_CODE_SUCCESS) {
915,548!
2168
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2169
  }
2170
  return code;
915,548✔
2171
}
2172

2173
static int32_t doRawScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
4,316,707✔
2174
  int32_t        code = TSDB_CODE_SUCCESS;
4,316,707✔
2175
  int32_t        lino = 0;
4,316,707✔
2176
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4,316,707✔
2177
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
4,316,707✔
2178

2179
  SStreamRawScanInfo* pInfo = pOperator->info;
4,317,558✔
2180
  pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = NULL;  // use batchMetaReq != NULL to judge if data is meta
4,317,558✔
2181
  pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = NULL;
4,317,558✔
2182

2183
  qDebug("tmqsnap doRawScan called");
4,317,558✔
2184
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
4,318,443✔
2185
    bool hasNext = false;
4,302,284✔
2186
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
4,302,284!
2187
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
4,298,927✔
2188
      if (code != TSDB_CODE_SUCCESS) {
4,295,511!
2189
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2190
        QUERY_CHECK_CODE(code, lino, _end);
×
2191
      }
2192
    }
2193

2194
    if (pInfo->dataReader && hasNext) {
4,298,868!
2195
      if (isTaskKilled(pTaskInfo)) {
2,156,941!
2196
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2197
        return code;
×
2198
      }
2199

2200
      SSDataBlock* pBlock = NULL;
2,159,494✔
2201
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock, NULL);
2,159,494✔
2202
      QUERY_CHECK_CODE(code, lino, _end);
2,156,941!
2203

2204
      if (pBlock && pBlock->info.rows > 0) {
2,156,941!
2205
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
2,157,792✔
2206
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
2,159,494✔
2207
        QUERY_CHECK_CODE(code, lino, _end);
2,157,792!
2208
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
2,157,792✔
2209
        (*ppRes) = pBlock;
2,161,196✔
2210
        return code;
2,160,345✔
2211
      }
2212
    }
2213

2214
    SMetaTableInfo mtInfo = {0};
2,141,927✔
2215
    code = pAPI->snapshotFn.getMetaTableInfoFromSnapshot(pInfo->sContext, &mtInfo);
2,141,939✔
2216
    QUERY_CHECK_CODE(code, lino, _end);
2,135,946!
2217
    if (code != 0) {
2,135,946!
2218
      destroyMetaTableInfo(&mtInfo);
2219
      QUERY_CHECK_CODE(code, lino, _end);
×
2220
    }
2221
    STqOffsetVal offset = {0};
2,135,946✔
2222
    if (mtInfo.uid == 0 || pInfo->sContext->withMeta == ONLY_META) {  // read snapshot done, change to get data from wal
2,136,821✔
2223
      qDebug("tmqsnap read snapshot done, change to get data from wal");
29,881✔
2224
      tqOffsetResetToLog(&offset, pInfo->sContext->snapVersion + 1);
29,881✔
2225
    } else {
2226
      SValue val = {0};
2,112,070✔
2227
      tqOffsetResetToData(&offset, mtInfo.uid, INT64_MIN, val);
2,112,070!
2228
      qDebug("tmqsnap change get data uid:%" PRId64, mtInfo.uid);
2,108,654!
2229
    }
2230
    destroyMetaTableInfo(&mtInfo);
2231
    code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
2,141,939✔
2232
    QUERY_CHECK_CODE(code, lino, _end);
2,141,939!
2233
    (*ppRes) = NULL;
2,141,939✔
2234
    return code;
2,141,939✔
2235
  } else if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_META) {
15,274!
2236
    SSnapContext* sContext = pInfo->sContext;
15,274✔
2237
    for (int32_t i = 0; i < tmqRowSize; i++) {
128,129!
2238
      void*   data = NULL;
128,129✔
2239
      int32_t dataLen = 0;
128,129✔
2240
      int16_t type = 0;
128,129✔
2241
      int64_t uid = 0;
128,129✔
2242
      if (pAPI->snapshotFn.getTableInfoFromSnapshot(sContext, &data, &dataLen, &type, &uid) < 0) {
128,129!
2243
        qError("tmqsnap getTableInfoFromSnapshot error");
×
2244
        taosMemoryFreeClear(data);
×
2245
        break;
×
2246
      }
2247

2248
      if (!sContext->queryMeta) {  // change to get data next poll request
128,129✔
2249
        STqOffsetVal offset = {0};
15,274✔
2250
        SValue       val = {0};
15,274!
2251
        tqOffsetResetToData(&offset, 0, INT64_MIN, val);
×
2252
        code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
15,274✔
2253
        QUERY_CHECK_CODE(code, lino, _end);
15,274!
2254
        break;
15,274✔
2255
      } else {
2256
        tqOffsetResetToMeta(&pTaskInfo->streamInfo.currentOffset, uid);
112,855✔
2257
        SMqMetaRsp tmpMetaRsp = {0};
112,855✔
2258
        tmpMetaRsp.resMsgType = type;
112,855✔
2259
        tmpMetaRsp.metaRspLen = dataLen;
112,855✔
2260
        tmpMetaRsp.metaRsp = data;
112,855✔
2261
        if (!pTaskInfo->streamInfo.btMetaRsp.batchMetaReq) {
112,855✔
2262
          pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = taosArrayInit(4, POINTER_BYTES);
12,783✔
2263
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, code, lino, _end, terrno);
13,642!
2264

2265
          pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = taosArrayInit(4, sizeof(int32_t));
13,642✔
2266
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, code, lino, _end, terrno);
13,642!
2267
        }
2268
        int32_t  tempRes = TSDB_CODE_SUCCESS;
113,714✔
2269
        uint32_t len = 0;
113,714✔
2270
        tEncodeSize(tEncodeMqMetaRsp, &tmpMetaRsp, len, tempRes);
113,714!
2271
        if (TSDB_CODE_SUCCESS != tempRes) {
112,855!
2272
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2273
          taosMemoryFreeClear(data);
×
2274
          break;
×
2275
        }
2276

2277
        int32_t tLen = sizeof(SMqRspHead) + len;
112,855✔
2278
        void*   tBuf = taosMemoryCalloc(1, tLen);
112,855!
2279
        QUERY_CHECK_NULL(tBuf, code, lino, _end, terrno);
112,855!
2280

2281
        void*    metaBuff = POINTER_SHIFT(tBuf, sizeof(SMqRspHead));
112,855✔
2282
        SEncoder encoder = {0};
112,855✔
2283
        tEncoderInit(&encoder, metaBuff, len);
112,855✔
2284
        int32_t tempLen = tEncodeMqMetaRsp(&encoder, &tmpMetaRsp);
112,855✔
2285
        if (tempLen < 0) {
111,996!
2286
          qError("tmqsnap tEncodeMqMetaRsp error");
×
2287
          tEncoderClear(&encoder);
×
2288
          taosMemoryFreeClear(tBuf);
×
2289
          taosMemoryFreeClear(data);
×
2290
          break;
×
2291
        }
2292
        taosMemoryFreeClear(data);
111,996!
2293
        void* tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, &tBuf);
112,855✔
2294
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
112,855!
2295

2296
        tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, &tLen);
112,855✔
2297
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
112,855!
2298
      }
2299
    }
2300

2301
    (*ppRes) = NULL;
15,274✔
2302
    return code;
15,274✔
2303
  }
2304

2305
_end:
×
2306
  if (code != TSDB_CODE_SUCCESS) {
×
2307
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2308
    pTaskInfo->code = code;
×
2309
    T_LONG_JMP(pTaskInfo->env, code);
×
2310
  }
2311

2312
  (*ppRes) = NULL;
×
2313
  return code;
×
2314
}
2315

2316
static void destroyTmqRawScanOperatorInfo(void* param) {
291,306✔
2317
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
291,306✔
2318
  pRawScan->pAPI->tsdReader.tsdReaderClose(pRawScan->dataReader);
291,306✔
2319
  pRawScan->pAPI->snapshotFn.destroySnapshot(pRawScan->sContext);
291,306✔
2320
  tableListDestroy(pRawScan->pTableListInfo);
291,306✔
2321
  taosMemoryFree(pRawScan);
291,306!
2322
}
291,306✔
2323

2324
// for subscribing db or stb (not including column),
2325
// if this scan is used, meta data can be return
2326
// and schemas are decided when scanning
2327
int32_t createTmqRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
290,421✔
2328
  // create operator
2329
  // create tb reader
2330
  // create meta reader
2331
  // create tq reader
2332

2333
  QRY_PARAM_CHECK(pOptrInfo);
290,421!
2334
  int32_t code = TSDB_CODE_SUCCESS;
291,306✔
2335
  int32_t lino = 0;
291,306✔
2336

2337
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
291,306!
2338
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
291,306!
2339
  if (pInfo == NULL || pOperator == NULL) {
291,306!
2340
    code = terrno;
×
2341
    lino = __LINE__;
×
2342
    goto _end;
×
2343
  }
2344

2345
  pInfo->pTableListInfo = tableListCreate();
291,306✔
2346
  QUERY_CHECK_NULL(pInfo->pTableListInfo, code, lino, _end, terrno);
291,306!
2347
  pInfo->vnode = pHandle->vnode;
291,306✔
2348
  pInfo->pAPI = &pTaskInfo->storageAPI;
291,306✔
2349

2350
  pInfo->sContext = pHandle->sContext;
291,306✔
2351
  setOperatorInfo(pOperator, "RawScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
291,306✔
2352
                  pTaskInfo);
2353

2354
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScanNext, NULL, destroyTmqRawScanOperatorInfo, optrDefaultBufFn,
291,306✔
2355
                                         NULL, optrDefaultGetNextExtFn, NULL);
2356
  *pOptrInfo = pOperator;
291,306✔
2357
  return code;
291,306✔
2358

2359
_end:
×
2360
  if (code != TSDB_CODE_SUCCESS) {
×
2361
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2362
  }
2363
  taosMemoryFree(pInfo);
×
2364
  taosMemoryFree(pOperator);
×
2365
  pTaskInfo->code = code;
×
2366
  return code;
×
2367
}
2368

2369
void destroyTmqScanOperatorInfo(void* param) {
915,548✔
2370
  if (param == NULL) {
915,548!
2371
    return;
×
2372
  }
2373

2374
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
915,548✔
2375
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
915,548!
2376
    destroyOperator(pStreamScan->pTableScanOp);
915,548✔
2377
  }
2378

2379
  if (pStreamScan->tqReader != NULL && pStreamScan->readerFn.tqReaderClose != NULL) {
915,548!
2380
    pStreamScan->readerFn.tqReaderClose(pStreamScan->tqReader);
915,548✔
2381
  }
2382
  if (pStreamScan->pVtableMergeHandles) {
915,548!
2383
    taosHashCleanup(pStreamScan->pVtableMergeHandles);
×
2384
    pStreamScan->pVtableMergeHandles = NULL;
×
2385
  }
2386
  if (pStreamScan->pVtableMergeBuf) {
915,548!
2387
    destroyDiskbasedBuf(pStreamScan->pVtableMergeBuf);
×
2388
    pStreamScan->pVtableMergeBuf = NULL;
×
2389
  }
2390
  if (pStreamScan->pVtableReadyHandles) {
915,548!
2391
    taosArrayDestroy(pStreamScan->pVtableReadyHandles);
×
2392
    pStreamScan->pVtableReadyHandles = NULL;
×
2393
  }
2394
  if (pStreamScan->pTableListInfo) {
915,548!
2395
    tableListDestroy(pStreamScan->pTableListInfo);
×
2396
    pStreamScan->pTableListInfo = NULL;
×
2397
  }
2398
  if (pStreamScan->matchInfo.pList) {
915,548!
2399
    taosArrayDestroy(pStreamScan->matchInfo.pList);
915,548✔
2400
  }
2401
  if (pStreamScan->pPseudoExpr) {
915,548✔
2402
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
532,340✔
2403
    taosMemoryFree(pStreamScan->pPseudoExpr);
531,468!
2404
  }
2405

2406
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
915,548✔
2407
  cleanupExprSupp(&pStreamScan->tagCalSup);
915,548✔
2408

2409
  blockDataDestroy(pStreamScan->pRes);
915,548✔
2410
  blockDataDestroy(pStreamScan->pUpdateRes);
915,548✔
2411
  blockDataDestroy(pStreamScan->pDeleteDataRes);
915,548✔
2412
  blockDataDestroy(pStreamScan->pUpdateDataRes);
915,548✔
2413
  blockDataDestroy(pStreamScan->pCreateTbRes);
915,548✔
2414
  taosArrayDestroy(pStreamScan->pBlockLists);
915,548✔
2415
  blockDataDestroy(pStreamScan->pCheckpointRes);
915,548✔
2416

2417
  taosMemoryFree(pStreamScan);
915,548!
2418
}
2419

2420
int32_t addPrimaryKeyCol(SSDataBlock* pBlock, uint8_t type, int32_t bytes) {
×
2421
  int32_t code = TSDB_CODE_SUCCESS;
×
2422
  int32_t lino = 0;
×
2423
  pBlock->info.rowSize += bytes;
×
2424
  SColumnInfoData infoData = {0};
×
2425
  infoData.info.type = type;
×
2426
  infoData.info.bytes = bytes;
×
2427
  void* tmp = taosArrayPush(pBlock->pDataBlock, &infoData);
×
2428
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
2429

2430
_end:
×
2431
  if (code != TSDB_CODE_SUCCESS) {
×
2432
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2433
  }
2434
  return code;
×
2435
}
2436

2437
static SSDataBlock* createStreamVtableBlock(SColMatchInfo* pMatchInfo, const char* idstr) {
×
2438
  int32_t      code = TSDB_CODE_SUCCESS;
×
2439
  int32_t      lino = 0;
×
2440
  SSDataBlock* pRes = NULL;
×
2441

2442
  QUERY_CHECK_NULL(pMatchInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
2443

2444
  code = createDataBlock(&pRes);
×
2445
  QUERY_CHECK_CODE(code, lino, _end);
×
2446
  int32_t numOfOutput = taosArrayGetSize(pMatchInfo->pList);
×
2447
  for (int32_t i = 0; i < numOfOutput; ++i) {
×
2448
    SColMatchItem* pItem = taosArrayGet(pMatchInfo->pList, i);
×
2449
    if (!pItem->needOutput) {
×
2450
      continue;
×
2451
    }
2452
    SColumnInfoData colInfo = createColumnInfoData(pItem->dataType.type, pItem->dataType.bytes, pItem->colId);
×
2453
    code = blockDataAppendColInfo(pRes, &colInfo);
×
2454
    QUERY_CHECK_CODE(code, lino, _end);
×
2455
  }
2456

2457
_end:
×
2458
  if (code != TSDB_CODE_SUCCESS) {
×
2459
    qError("%s failed at line %d since %s, id: %s", __func__, lino, tstrerror(code), idstr);
×
2460
    if (pRes != NULL) {
×
2461
      blockDataDestroy(pRes);
×
2462
    }
2463
    pRes = NULL;
×
2464
    terrno = code;
×
2465
  }
2466
  return pRes;
×
2467
}
2468

2469
 int32_t createTmqScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode,
911,157✔
2470
                                                  SNode* pTagCond, STableListInfo* pTableListInfo,
2471
                                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
2472
  QRY_PARAM_CHECK(pOptrInfo);
911,157!
2473

2474
  int32_t          code = TSDB_CODE_SUCCESS;
915,548✔
2475
  int32_t          lino = 0;
915,548✔
2476
  SArray*          pColIds = NULL;
915,548✔
2477
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
915,548!
2478
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
915,548!
2479
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
915,548✔
2480
  const char*      idstr = pTaskInfo->id.str;
915,548✔
2481
  SSHashObj*       pVtableInfos = pTaskInfo->pSubplan->pVTables;
915,548✔
2482

2483
  if (pInfo == NULL || pOperator == NULL) {
915,548!
2484
    code = terrno;
×
2485
    goto _error;
×
2486
  }
2487

2488
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
915,548✔
2489
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
915,548✔
2490

2491
  pInfo->pTagCond = pTagCond;
915,548✔
2492
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
915,548✔
2493

2494
  int32_t numOfCols = 0;
915,548✔
2495
  code = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
915,548✔
2496
  if (code != TSDB_CODE_SUCCESS) {
912,904!
2497
    goto _error;
×
2498
  }
2499

2500
  SDataType pkType = {0};
912,904✔
2501
  pInfo->primaryKeyIndex = -1;
912,904✔
2502
  pInfo->basic.primaryPkIndex = -1;
912,904✔
2503
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
912,006✔
2504
  pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
914,650✔
2505
  QUERY_CHECK_NULL(pColIds, code, lino, _error, terrno);
914,679!
2506

2507
  for (int32_t i = 0; i < numOfOutput; ++i) {
6,768,353✔
2508
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
5,852,798✔
2509
    QUERY_CHECK_NULL(id, code, lino, _error, terrno);
5,852,798!
2510

2511
    int16_t colId = id->colId;
5,852,798✔
2512
    void*   tmp = taosArrayPush(pColIds, &colId);
5,853,674✔
2513
    QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
5,853,674!
2514

2515
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
5,853,674✔
2516
      pInfo->primaryTsIndex = id->dstSlotId;
913,805✔
2517
    }
2518
    if (id->isPk) {
5,851,900!
2519
      pInfo->primaryKeyIndex = id->dstSlotId;
×
2520
      pInfo->basic.primaryPkIndex = id->dstSlotId;
×
2521
      pkType = id->dataType;
×
2522
    }
2523
  }
2524

2525
  pInfo->pPartTbnameSup = NULL;
915,555✔
2526
  if (pTableScanNode->pSubtable != NULL) {
915,548!
2527
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
×
2528
    if (pSubTableExpr == NULL) {
×
2529
      code = terrno;
×
2530
      goto _error;
×
2531
    }
2532

2533
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
×
2534
    code = createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
×
2535
    QUERY_CHECK_CODE(code, lino, _error);
×
2536

2537
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1, &pTaskInfo->storageAPI.functionStore) != 0) {
×
2538
      goto _error;
×
2539
    }
2540
  }
2541

2542
  if (pTableScanNode->pTags != NULL) {
913,805!
2543
    int32_t    numOfTags;
×
2544
    SExprInfo* pTagExpr = createExpr(pTableScanNode->pTags, &numOfTags);
×
2545
    if (pTagExpr == NULL) {
×
2546
      goto _error;
×
2547
    }
2548
    code = initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags, &pTaskInfo->storageAPI.functionStore);
×
2549
    if (code != 0) {
×
2550
      goto _error;
×
2551
    }
2552
  }
2553

2554
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
915,548✔
2555
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
913,774!
2556

2557
  if (pHandle->vnode) {
912,937!
2558
    SOperatorInfo* pTableScanOp = NULL;
914,643✔
2559
    code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pTableScanOp);
912,900✔
2560
    if (pTableScanOp == NULL || code != 0) {
911,264!
2561
      qError("createTableScanOperatorInfo error, code:%d", pTaskInfo->code);
2,672!
2562
      goto _error;
×
2563
    }
2564

2565
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
908,592✔
2566
    if (pHandle->version > 0) {
910,399✔
2567
      pTSInfo->base.cond.endVersion = pHandle->version;
912,901✔
2568
    }
2569

2570
    STableKeyInfo* pList = NULL;
910,401✔
2571
    int32_t        num = 0;
912,074✔
2572
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
913,811✔
2573
    QUERY_CHECK_CODE(code, lino, _error);
912,137!
2574

2575
    if (pHandle->initTableReader) {
912,137✔
2576
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
911,296✔
2577
      pTSInfo->base.dataReader = NULL;
912,938✔
2578
    }
2579

2580
    if (pHandle->initTqReader) {
907,759!
2581
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
908,627✔
2582
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
914,650!
2583
    } else {
2584
      pInfo->tqReader = pHandle->tqReader;
×
2585
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
2586
    }
2587

2588
    if (pVtableInfos != NULL) {
914,684!
2589
      // save vtable info into tqReader for vtable source scan
2590
      SSDataBlock* pResBlock = createStreamVtableBlock(&pInfo->matchInfo, idstr);
×
2591
      QUERY_CHECK_CODE(code, lino, _error);
×
2592
      code = pAPI->tqReaderFn.tqReaderSetVtableInfo(pInfo->tqReader, pHandle->vnode, pAPI, pVtableInfos, &pResBlock,
×
2593
                                                    idstr);
2594
      QUERY_CHECK_CODE(code, lino, _error);
×
2595
    }
2596

2597
    pInfo->pUpdateInfo = NULL;
914,684✔
2598
    pInfo->pTableScanOp = pTableScanOp;
914,650✔
2599
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
915,548!
2600
      pAPI->stateStore.streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1,
×
2601
                                            pInfo->primaryTsIndex);
2602
    }
2603

2604
    pInfo->readHandle = *pHandle;
915,548✔
2605
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
915,548✔
2606
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
914,650✔
2607
    QUERY_CHECK_NULL(pInfo->pCreateTbRes, code, lino, _error, terrno);
913,843!
2608
    pInfo->hasPart = false;
909,469✔
2609

2610
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
912,071✔
2611
    QUERY_CHECK_CODE(code, lino, _error);
910,333!
2612

2613
    // set the extract column id to streamHandle
2614
    code = pAPI->tqReaderFn.tqReaderSetColIdList(pInfo->tqReader, pColIds, idstr);
910,333✔
2615
    QUERY_CHECK_CODE(code, lino, _error);
910,361!
2616

2617
    SArray* tableIdList = NULL;
910,361✔
2618
    code = extractTableIdList(((STableScanInfo*)(pInfo->pTableScanOp->info))->base.pTableListInfo, &tableIdList);
912,118✔
2619
    QUERY_CHECK_CODE(code, lino, _error);
913,822!
2620
    code = pAPI->tqReaderFn.tqReaderSetQueryTableList(pInfo->tqReader, tableIdList, idstr);
913,822✔
2621
    QUERY_CHECK_CODE(code, lino, _error);
915,548!
2622
    taosArrayDestroy(tableIdList);
915,548✔
2623
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->base.cond, sizeof(SQueryTableDataCond));
915,548!
2624
  } else {
2625
    taosArrayDestroy(pColIds);
×
2626
    tableListDestroy(pTableListInfo);
×
2627
  }
2628

2629
  // clear the local variable to avoid repeatly free
2630
  pColIds = NULL;
915,548✔
2631

2632
  // create the pseduo columns info
2633
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
915,548✔
2634
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->pPseudoExpr, &pInfo->numOfPseudoExpr);
532,340✔
2635
    QUERY_CHECK_CODE(code, lino, _error);
532,340!
2636
  }
2637

2638
  code = filterInitFromNode((SNode*)pScanPhyNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0, NULL);
915,548✔
2639
  QUERY_CHECK_CODE(code, lino, _error);
915,548!
2640

2641
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
915,548✔
2642
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
915,548!
2643
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
915,548✔
2644
  QUERY_CHECK_CODE(code, lino, _error);
914,680!
2645

2646
  pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
914,680✔
2647
  pInfo->windowSup = (SWindowSupporter){.pStreamAggSup = NULL, .gap = -1, .parentType = QUERY_NODE_PHYSICAL_PLAN};
914,680✔
2648
  pInfo->groupId = 0;
913,855✔
2649
  pInfo->igCheckGroupId = false;
914,680✔
2650
  pInfo->pStreamScanOp = pOperator;
914,680✔
2651
  pInfo->deleteDataIndex = 0;
913,855✔
2652
  code = createSpecialDataBlock(STREAM_DELETE_DATA, &pInfo->pDeleteDataRes);
914,680✔
2653
  QUERY_CHECK_CODE(code, lino, _error);
915,548!
2654

2655
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
915,548✔
2656
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateDataRes);
913,810✔
2657
  QUERY_CHECK_CODE(code, lino, _error);
914,723!
2658

2659
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
914,723!
2660
  pInfo->partitionSup.needCalc = false;
913,853✔
2661
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
914,678✔
2662
  pInfo->igExpired = pTableScanNode->igExpired;
914,678✔
2663
  pInfo->twAggSup.maxTs = INT64_MIN;
914,678✔
2664
  pInfo->pState = pTaskInfo->streamInfo.pState;
913,853✔
2665
  pInfo->stateStore = pTaskInfo->storageAPI.stateStore;
913,853✔
2666
  pInfo->readerFn = pTaskInfo->storageAPI.tqReaderFn;
914,678✔
2667
  pInfo->pFillSup = NULL;
914,674✔
2668
  pInfo->useGetResultRange = false;
912,426✔
2669
  pInfo->pRangeScanRes = NULL;
913,804✔
2670

2671
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
913,300✔
2672
  QUERY_CHECK_CODE(code, lino, _error);
915,548!
2673

2674
  setOperatorInfo(pOperator, STREAM_SCAN_OP_NAME, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, false, OP_NOT_OPENED, pInfo,
915,548✔
2675
                  pTaskInfo);
2676
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
915,548✔
2677

2678
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doQueueScanNext, NULL, destroyTmqScanOperatorInfo, optrDefaultBufFn,
914,170✔
2679
                                         NULL, optrDefaultGetNextExtFn, NULL);
2680

2681
  *pOptrInfo = pOperator;
913,841✔
2682
  return code;
912,463✔
2683

2684
_error:
×
2685
  if (pColIds != NULL) {
×
2686
    taosArrayDestroy(pColIds);
×
2687
  }
2688

2689
  if (pInfo != NULL && pInfo->pTableScanOp != NULL) {
×
2690
    STableScanInfo* p = (STableScanInfo*)pInfo->pTableScanOp->info;
×
2691
    if (p != NULL) {
×
2692
      p->base.pTableListInfo = NULL;
×
2693
    }
2694
    destroyTmqScanOperatorInfo(pInfo);
×
2695
  }
2696

2697
  if (pOperator != NULL) {
×
2698
    pOperator->info = NULL;
×
2699
    destroyOperator(pOperator);
×
2700
  }
2701
  pTaskInfo->code = code;
×
2702
  return code;
×
2703
}
2704

2705
static int32_t doTagScanOneTable(SOperatorInfo* pOperator, SSDataBlock* pRes, SMetaReader* mr, SStorageAPI* pAPI) {
18,585,235✔
2706
  int32_t        code = TSDB_CODE_SUCCESS;
18,585,235✔
2707
  int32_t        lino = 0;
18,585,235✔
2708
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
18,585,235✔
2709
  STagScanInfo*  pInfo = pOperator->info;
18,585,918✔
2710
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
18,584,536✔
2711
  int32_t        count = pRes->info.rows;
18,586,590✔
2712

2713
  STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
18,586,590✔
2714
  if (!item) {
18,587,617!
2715
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
2716
           GET_TASKID(pTaskInfo));
2717
    tDecoderClear(&(*mr).coder);
×
2718
    goto _end;
×
2719
  }
2720

2721
  code = pAPI->metaReaderFn.getTableEntryByUid(mr, item->uid);
18,587,617✔
2722
  tDecoderClear(&(*mr).coder);
18,587,617✔
2723
  if (code != TSDB_CODE_SUCCESS) {
18,587,617!
2724
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
2725
           GET_TASKID(pTaskInfo));
2726
    goto _end;
×
2727
  }
2728

2729
  char str[512];
18,587,617✔
2730
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
45,792,338✔
2731
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
27,207,132✔
2732

2733
    // refactor later
2734
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
27,208,159✔
2735
      STR_TO_VARSTR(str, (*mr).me.name);
13,520,241!
2736
      code = colDataSetVal(pDst, (count), str, false);
13,520,924✔
2737
      QUERY_CHECK_CODE(code, lino, _end);
13,519,214!
2738
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
13,687,235✔
2739
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
105,320✔
2740
      QUERY_CHECK_CODE(code, lino, _end);
105,320!
2741
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
13,581,915✔
2742
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
105,320✔
2743
      QUERY_CHECK_CODE(code, lino, _end);
105,320!
2744
    } else {  // it is a tag value
2745
      STagVal val = {0};
13,476,595✔
2746
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
13,476,595✔
2747
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
13,476,595✔
2748

2749
      char* data = NULL;
13,476,595✔
2750
      if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
13,476,595!
2751
        data = tTagValToData((const STagVal*)p, false);
13,476,595✔
2752
      } else {
2753
        data = (char*)p;
×
2754
      }
2755

2756
      code = colDataSetVal(pDst, (count), data,
13,476,595✔
2757
                           (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
13,476,595!
2758
      QUERY_CHECK_CODE(code, lino, _end);
13,476,595!
2759

2760
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
13,476,595!
2761
          (data != NULL)) {
2762
        taosMemoryFree(data);
3,016,945!
2763
      }
2764
    }
2765
  }
2766

2767
_end:
18,586,608✔
2768
  if (code != TSDB_CODE_SUCCESS) {
18,585,889!
2769
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2770
    pTaskInfo->code = code;
×
2771
  } else {
2772
    pRes->info.rows++;
18,585,889✔
2773
  }
2774

2775
  return code;
18,586,916✔
2776
}
2777

2778
static void tagScanFreeUidTag(void* p) {
47,536,667✔
2779
  STUidTagInfo* pInfo = p;
47,536,667✔
2780
  if (pInfo->pTagVal != NULL) {
47,536,667✔
2781
    taosMemoryFree(pInfo->pTagVal);
47,535,442!
2782
  }
2783
}
47,542,858✔
2784

2785
static int32_t tagScanCreateResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
10,572,683✔
2786
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
10,572,683!
2787
  if (pColumnData == NULL) {
10,578,387!
2788
    return terrno;
×
2789
  }
2790

2791
  pColumnData->info.type = pType->type;
10,578,387✔
2792
  pColumnData->info.bytes = pType->bytes;
10,574,863✔
2793
  pColumnData->info.scale = pType->scale;
10,578,956✔
2794
  pColumnData->info.precision = pType->precision;
10,569,897✔
2795

2796
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
10,568,598✔
2797
  if (code != TSDB_CODE_SUCCESS) {
10,573,843!
2798
    terrno = code;
×
2799
    taosMemoryFree(pColumnData);
×
2800
    return terrno;
×
2801
  }
2802

2803
  pParam->columnData = pColumnData;
10,573,843✔
2804
  pParam->colAlloced = true;
10,580,273✔
2805
  return TSDB_CODE_SUCCESS;
10,580,178✔
2806
}
2807

2808
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
66,728,409✔
2809
  int32_t                code = TSDB_CODE_SUCCESS;
66,728,409✔
2810
  int32_t                lino = 0;
66,728,409✔
2811
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
66,728,409✔
2812
  SColumnNode*           pSColumnNode = NULL;
66,728,409✔
2813
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
66,735,016✔
2814
    pSColumnNode = *(SColumnNode**)pNode;
18,753,487✔
2815
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
48,010,447!
2816
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
×
2817
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
×
2818
      pSColumnNode = NULL;
×
2819
      pCtx->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
×
2820
      if (NULL == pSColumnNode) {
×
2821
        return DEAL_RES_ERROR;
×
2822
      }
2823
      pSColumnNode->colId = -1;
×
2824
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
×
2825
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
×
2826
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
×
2827
      nodesDestroyNode(*pNode);
×
2828
      *pNode = (SNode*)pSColumnNode;
×
2829
    } else {
2830
      return DEAL_RES_CONTINUE;
×
2831
    }
2832
  } else {
2833
    return DEAL_RES_CONTINUE;
47,998,376✔
2834
  }
2835

2836
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
18,744,214✔
2837
  if (!data) {
18,747,625✔
2838
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
10,591,286✔
2839
    if (code == TSDB_CODE_DUP_KEY) {
10,601,207!
2840
      code = TSDB_CODE_SUCCESS;
×
2841
    }
2842
    QUERY_CHECK_CODE(code, lino, _end);
10,601,207!
2843
    pSColumnNode->slotId = pCtx->index++;
10,601,207✔
2844
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
10,593,002✔
2845
                         .type = pSColumnNode->node.resType.type,
10,582,090✔
2846
                         .bytes = pSColumnNode->node.resType.bytes};
10,583,590✔
2847
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
10,590,085✔
2848
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
10,594,685!
2849
  } else {
2850
    SColumnNode* col = *(SColumnNode**)data;
8,156,339✔
2851
    pSColumnNode->slotId = col->slotId;
8,157,400✔
2852
  }
2853

2854
_end:
18,743,427✔
2855
  if (code != TSDB_CODE_SUCCESS) {
18,737,687✔
2856
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,890!
2857
    return DEAL_RES_ERROR;
×
2858
  }
2859
  return DEAL_RES_CONTINUE;
18,735,885✔
2860
}
2861

2862
static int32_t tagScanFilterByTagCond(SArray* aUidTags, SNode* pTagCond, SArray* aFilterIdxs, void* pVnode,
10,590,480✔
2863
                                      SStorageAPI* pAPI, STagScanInfo* pInfo) {
2864
  int32_t code = TSDB_CODE_SUCCESS;
10,590,480✔
2865
  int32_t lino = 0;
10,590,480✔
2866
  int32_t numOfTables = taosArrayGetSize(aUidTags);
10,590,480✔
2867
  SArray* pBlockList = NULL;
10,590,480✔
2868

2869
  SSDataBlock* pResBlock = createTagValBlockForFilter(pInfo->filterCtx.cInfoList, numOfTables, aUidTags, pVnode, pAPI);
10,590,480✔
2870
  QUERY_CHECK_NULL(pResBlock, code, lino, _end, terrno);
10,577,119!
2871

2872
  pBlockList = taosArrayInit(1, POINTER_BYTES);
10,577,119✔
2873
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
10,582,362!
2874

2875
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
10,589,352✔
2876
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
10,589,352!
2877

2878
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
10,589,352✔
2879

2880
  SScalarParam output = {0};
10,584,716✔
2881
  code = tagScanCreateResultData(&type, numOfTables, &output);
10,581,604✔
2882
  QUERY_CHECK_CODE(code, lino, _end);
10,577,358!
2883

2884
  code = scalarCalculate(pTagCond, pBlockList, &output, NULL, NULL);
10,577,358✔
2885
  QUERY_CHECK_CODE(code, lino, _end);
10,560,433!
2886

2887
  bool* result = (bool*)output.columnData->pData;
10,560,433✔
2888
  for (int32_t i = 0; i < numOfTables; ++i) {
26,479,571✔
2889
    if (result[i]) {
15,897,818!
2890
      void* tmp = taosArrayPush(aFilterIdxs, &i);
15,140,822✔
2891
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
15,140,822!
2892
    }
2893
  }
2894

2895
  colDataDestroy(output.columnData);
10,584,688✔
2896
  taosMemoryFreeClear(output.columnData);
10,571,043!
2897

2898
_end:
10,565,916✔
2899
  blockDataDestroy(pResBlock);
10,569,283✔
2900
  taosArrayDestroy(pBlockList);
10,556,347✔
2901
  if (code != TSDB_CODE_SUCCESS) {
10,562,166!
2902
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2903
  }
2904
  return code;
10,562,166✔
2905
}
2906

2907
static int32_t tagScanFillOneCellWithTag(SOperatorInfo* pOperator, const STUidTagInfo* pUidTagInfo,
196,118,551✔
2908
                                         SExprInfo* pExprInfo, SColumnInfoData* pColInfo, int rowIndex,
2909
                                         const SStorageAPI* pAPI, void* pVnode) {
2910
  int32_t code = TSDB_CODE_SUCCESS;
196,118,551✔
2911
  int32_t lino = 0;
196,118,551✔
2912
  if (QUERY_NODE_FUNCTION == pExprInfo->pExpr->nodeType) {
196,118,551✔
2913
    if (FUNCTION_TYPE_TBNAME == pExprInfo->pExpr->_function.functionType) {  // tbname
93,526,937!
2914
      char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
2915
      STR_TO_VARSTR(str, "ctbidx");
×
2916

2917
      code = colDataSetVal(pColInfo, rowIndex, str, false);
×
2918
      QUERY_CHECK_CODE(code, lino, _end);
×
2919
    } else if (FUNCTION_TYPE_TBUID == pExprInfo->pExpr->_function.functionType) {
93,469,699✔
2920
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
46,700,372✔
2921
      QUERY_CHECK_CODE(code, lino, _end);
46,758,040!
2922
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
46,764,409✔
2923
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
46,740,546✔
2924
      QUERY_CHECK_CODE(code, lino, _end);
46,761,776!
2925
    }
2926
  } else {
2927
    STagVal tagVal = {0};
102,578,971✔
2928
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
102,576,140✔
2929
    if (pUidTagInfo->pTagVal == NULL) {
102,609,814!
2930
      colDataSetNULL(pColInfo, rowIndex);
×
2931
    } else {
2932
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
102,568,751✔
2933

2934
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
102,600,314!
2935
        colDataSetNULL(pColInfo, rowIndex);
31,709!
2936
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
102,571,213!
2937
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
2938
        QUERY_CHECK_CODE(code, lino, _end);
×
2939
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
126,994,559!
2940
        if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
24,437,069!
2941
          QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
×
2942
        }
2943
        char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
24,460,720!
2944
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
24,457,006!
2945

2946
        varDataSetLen(tmp, tagVal.nData);
24,457,006✔
2947
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
24,453,848!
2948
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
24,450,351✔
2949
        taosMemoryFree(tmp);
24,469,639!
2950
        QUERY_CHECK_CODE(code, lino, _end);
24,464,686!
2951
      } else {
2952
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
78,045,983✔
2953
        QUERY_CHECK_CODE(code, lino, _end);
78,103,928!
2954
      }
2955
    }
2956
  }
2957

2958
_end:
196,125,875✔
2959
  if (code != TSDB_CODE_SUCCESS) {
196,125,875!
2960
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2961
  }
2962
  return code;
196,058,283✔
2963
}
2964

2965
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
30,553,359✔
2966
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
2967
  int32_t       code = TSDB_CODE_SUCCESS;
30,553,359✔
2968
  int32_t       lino = 0;
30,553,359✔
2969
  STagScanInfo* pInfo = pOperator->info;
30,553,359✔
2970
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
30,561,165✔
2971
  if (!ignoreFilterIdx) {
30,543,797✔
2972
    size_t szTables = taosArrayGetSize(aFilterIdxs);
10,560,747✔
2973
    for (int i = 0; i < szTables; ++i) {
25,694,990✔
2974
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
15,107,797✔
2975
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
15,098,958✔
2976
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
15,115,517!
2977
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
86,383,153✔
2978
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
71,214,545✔
2979
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
71,266,482!
2980
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
71,266,482✔
2981
        QUERY_CHECK_CODE(code, lino, _end);
71,267,636!
2982
      }
2983
    }
2984
  } else {
2985
    size_t szTables = taosArrayGetSize(aUidTags);
19,983,050✔
2986
    for (int i = 0; i < szTables; ++i) {
51,622,455✔
2987
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
31,638,844✔
2988
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
31,639,178!
2989
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
156,418,715✔
2990
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
124,785,076✔
2991
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
124,807,351!
2992
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
124,807,351✔
2993
        QUERY_CHECK_CODE(code, lino, _end);
124,778,476!
2994
      }
2995
    }
2996
  }
2997

2998
_end:
19,983,611✔
2999
  if (code != TSDB_CODE_SUCCESS) {
30,570,804!
3000
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3001
  }
3002
  return code;
30,558,958✔
3003
}
3004

3005
static int32_t doTagScanFromCtbIdxNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
60,872,207✔
3006
  int32_t        code = TSDB_CODE_SUCCESS;
60,872,207✔
3007
  int32_t        lino = 0;
60,872,207✔
3008
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
60,872,207✔
3009
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
60,876,326✔
3010

3011
  STagScanInfo* pInfo = pOperator->info;
60,872,808✔
3012
  SSDataBlock*  pRes = pInfo->pRes;
60,862,376✔
3013

3014
  QRY_PARAM_CHECK(ppRes);
60,859,592!
3015

3016
  if (pOperator->status == OP_EXEC_DONE) {
60,864,963✔
3017
    return TSDB_CODE_SUCCESS;
30,073,593✔
3018
  }
3019
  blockDataCleanup(pRes);
30,782,709✔
3020

3021
  if (pInfo->pCtbCursor == NULL) {
30,813,064!
3022
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
30,810,759✔
3023
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
30,830,973!
3024
  } else {
3025
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
3026
    QUERY_CHECK_CODE(code, lino, _end);
×
3027
  }
3028

3029
  SArray* aUidTags = pInfo->aUidTags;
30,831,230✔
3030
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
30,829,061✔
3031
  int32_t count = 0;
30,830,871✔
3032
  bool    ctbCursorFinished = false;
30,830,871✔
3033
  while (1) {
494,574✔
3034
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
31,325,445✔
3035
    taosArrayClear(aFilterIdxs);
31,313,049✔
3036

3037
    int32_t numTables = 0;
31,313,810✔
3038
    while (numTables < pOperator->resultInfo.capacity) {
78,884,005✔
3039
      SMCtbCursor* pCur = pInfo->pCtbCursor;
78,869,867✔
3040
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
78,867,959✔
3041
      if (uid == 0) {
78,839,507✔
3042
        ctbCursorFinished = true;
31,324,809✔
3043
        break;
31,324,809✔
3044
      }
3045
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
47,514,698✔
3046
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
47,519,316!
3047
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
47,482,734!
3048

3049
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
47,482,734!
3050
      void* tmp = taosArrayPush(aUidTags, &info);
47,553,218✔
3051
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
47,553,218!
3052
      ++numTables;
47,553,218✔
3053
    }
3054

3055
    if (numTables == 0) {
31,323,435✔
3056
      break;
748,822✔
3057
    }
3058
    bool ignoreFilterIdx = true;
30,574,613✔
3059
    if (pInfo->pTagCond != NULL) {
30,574,613✔
3060
      ignoreFilterIdx = false;
10,590,480✔
3061
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo);
10,590,480✔
3062
      QUERY_CHECK_CODE(code, lino, _end);
10,566,248!
3063
    } else {
3064
      ignoreFilterIdx = true;
19,984,133✔
3065
    }
3066

3067
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
30,550,381✔
3068
    QUERY_CHECK_CODE(code, lino, _end);
30,559,577!
3069

3070
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
30,559,577✔
3071

3072
    if (count != 0) {
30,560,731✔
3073
      break;
30,066,157✔
3074
    }
3075
  }
3076

3077
  if (count > 0) {
30,814,979✔
3078
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
30,066,886✔
3079
  }
3080
  if (count == 0 || ctbCursorFinished) {
30,828,044✔
3081
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
30,827,620✔
3082
    pInfo->pCtbCursor = NULL;
30,803,623✔
3083
    setOperatorCompleted(pOperator);
30,797,014✔
3084
  }
3085

3086
  pRes->info.rows = count;
30,800,777✔
3087
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
30,799,248✔
3088
  if (bLimitReached) {
30,804,717!
3089
    setOperatorCompleted(pOperator);
×
3090
  }
3091

3092
  pOperator->resultInfo.totalRows += pRes->info.rows;
30,804,717✔
3093
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
30,782,386✔
3094

3095
_end:
30,775,660✔
3096
  if (code != TSDB_CODE_SUCCESS) {
30,775,660!
3097
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3098
    pTaskInfo->code = code;
×
3099
    T_LONG_JMP(pTaskInfo->env, code);
×
3100
  }
3101

3102
  return code;
30,775,660✔
3103
}
3104

3105
static int32_t createTagScanTableListInfoFromParam(SOperatorInfo* pOperator) {
5,029,906✔
3106
  STagScanInfo*          pInfo = pOperator->info;
5,029,906✔
3107
  SExecTaskInfo*         pTaskInfo = pOperator->pTaskInfo;
5,029,906✔
3108
  int32_t                code = 0;
5,029,906✔
3109
  STableListInfo*        pListInfo = pInfo->pTableListInfo;
5,029,906✔
3110
  STagScanOperatorParam* pParam = (STagScanOperatorParam*)pOperator->pOperatorGetParam->value;
5,029,906✔
3111
  tb_uid_t               pUid = pParam->vcUid;
5,029,906✔
3112

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

3117
  STableKeyInfo info = {.groupId = 0};
5,029,906✔
3118

3119
  int32_t tableIdx = 0;
5,029,906✔
3120
  taosHashClear(pListInfo->map);
5,029,906✔
3121
  taosArrayClear(pListInfo->pTableList);
5,029,906✔
3122

3123
  if (taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
5,029,906!
3124
    if (TSDB_CODE_DUP_KEY == terrno) {
×
3125
    } else {
3126
      return terrno;
×
3127
    }
3128
  }
3129

3130
  info.uid = pUid;
5,029,906✔
3131
  void* p = taosArrayPush(pListInfo->pTableList, &info);
5,029,906✔
3132
  if (p == NULL) {
5,029,906!
3133
    return terrno;
×
3134
  }
3135

3136
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
5,029,906✔
3137

3138
  return code;
5,029,906✔
3139
}
3140

3141
static int32_t doTagScanFromMetaEntryNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
7,365,544✔
3142
  int32_t       code = TSDB_CODE_SUCCESS;
7,365,544✔
3143
  int32_t       lino = 0;
7,365,544✔
3144
  STagScanInfo* pInfo = pOperator->info;
7,365,544✔
3145
  if (pOperator->pOperatorGetParam) {
7,365,544✔
3146
    pOperator->resultInfo.totalRows = 0;
5,029,906✔
3147
    pOperator->dynamicTask = true;
5,029,906✔
3148
    pInfo->curPos = 0;
5,029,906✔
3149
    code = createTagScanTableListInfoFromParam(pOperator);
5,029,906✔
3150
    freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
5,029,906✔
3151
    pOperator->pOperatorGetParam = NULL;
5,029,906✔
3152
    QUERY_CHECK_CODE(code, lino, _end);
5,029,906!
3153

3154
    if (pOperator->status == OP_EXEC_DONE) {
5,029,906✔
3155
      pOperator->status = OP_OPENED;
2,504,856✔
3156
    }
3157
  }
3158

3159
  if (pOperator->status == OP_EXEC_DONE) {
7,365,544✔
3160
    (*ppRes) = NULL;
1,114,243✔
3161
    return code;
1,114,243✔
3162
  }
3163

3164
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
6,251,301✔
3165
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
6,251,301✔
3166

3167
  SExprInfo*   pExprInfo = &pOperator->exprSupp.pExprInfo[0];
6,251,301✔
3168
  SSDataBlock* pRes = pInfo->pRes;
6,251,301✔
3169
  blockDataCleanup(pRes);
6,251,301✔
3170

3171
  int32_t size = 0;
6,251,301✔
3172
  code = tableListGetSize(pInfo->pTableListInfo, &size);
6,251,301✔
3173
  QUERY_CHECK_CODE(code, lino, _end);
6,251,301!
3174

3175
  if (size == 0) {
6,251,301✔
3176
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
106,451✔
3177
    (*ppRes) = NULL;
106,451✔
3178
    return code;
106,451✔
3179
  }
3180

3181
  SMetaReader mr = {0};
6,144,850✔
3182
  pAPI->metaReaderFn.initReader(&mr, pInfo->readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
6,144,850✔
3183
  pRes->info.rows = 0;
6,144,850✔
3184

3185
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
24,731,784✔
3186
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
18,584,880✔
3187
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD &&
18,586,590!
3188
        code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
3189
      // ignore other error
3190
      code = TSDB_CODE_SUCCESS;
18,586,590✔
3191
    }
3192
    QUERY_CHECK_CODE(code, lino, _end);
18,586,590!
3193

3194
    if (++pInfo->curPos >= size) {
18,586,590✔
3195
      setOperatorCompleted(pOperator);
6,144,850✔
3196
    }
3197
  }
3198

3199
  pAPI->metaReaderFn.clearReader(&mr);
6,147,587✔
3200
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
6,144,850✔
3201
  if (bLimitReached) {
6,144,850!
3202
    setOperatorCompleted(pOperator);
×
3203
  }
3204

3205
  // qDebug("QInfo:0x%" PRIx64 ", create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
3206
  if (pOperator->status == OP_EXEC_DONE) {
6,144,850!
3207
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
6,144,850✔
3208
  }
3209

3210
  pOperator->resultInfo.totalRows += pRes->info.rows;
6,144,149✔
3211

3212
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
6,144,149!
3213
_end:
6,144,149✔
3214
  if (code != TSDB_CODE_SUCCESS) {
6,144,149!
3215
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3216
    pTaskInfo->code = code;
×
3217
    T_LONG_JMP(pTaskInfo->env, code);
×
3218
  }
3219
  return code;
6,144,149✔
3220
}
3221

3222
static SSDataBlock* doTagScanFromMetaEntry(SOperatorInfo* pOperator) {
×
3223
  SSDataBlock* pRes = NULL;
×
3224
  int32_t      code = doTagScanFromMetaEntryNext(pOperator, &pRes);
×
3225
  return pRes;
×
3226
}
3227

3228
static void destroyTagScanOperatorInfo(void* param) {
34,553,333✔
3229
  STagScanInfo* pInfo = (STagScanInfo*)param;
34,553,333✔
3230
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
34,553,333!
3231
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
3232
  }
3233
  taosHashCleanup(pInfo->filterCtx.colHash);
34,564,794✔
3234
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
34,546,370✔
3235
  taosArrayDestroy(pInfo->aFilterIdxs);
34,547,137✔
3236
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
34,553,659✔
3237

3238
  blockDataDestroy(pInfo->pRes);
34,580,124✔
3239
  taosArrayDestroy(pInfo->matchInfo.pList);
34,580,760✔
3240
  tableListDestroy(pInfo->pTableListInfo);
34,556,729✔
3241

3242
  pInfo->pRes = NULL;
34,584,075✔
3243
  pInfo->pTableListInfo = NULL;
34,584,833✔
3244
  taosMemoryFreeClear(param);
34,581,750!
3245
}
34,575,473✔
3246

3247
int32_t createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pTagScanNode,
34,574,521✔
3248
                                  STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
3249
                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
3250
  QRY_PARAM_CHECK(pOptrInfo);
34,574,521!
3251

3252
  int32_t         code = TSDB_CODE_SUCCESS;
34,580,496✔
3253
  int32_t         lino = 0;
34,580,496✔
3254
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
34,580,496✔
3255
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
34,580,496!
3256
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
34,569,149!
3257
  if (pInfo == NULL || pOperator == NULL) {
34,526,764!
3258
    code = terrno;
×
3259
    goto _error;
×
3260
  }
3261

3262
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
34,527,829✔
3263

3264
  int32_t    numOfExprs = 0;
34,529,439✔
3265
  SExprInfo* pExprInfo = NULL;
34,532,414✔
3266

3267
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
34,553,875✔
3268
  QUERY_CHECK_CODE(code, lino, _error);
34,602,712!
3269

3270
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
34,602,712✔
3271
  QUERY_CHECK_CODE(code, lino, _error);
34,506,837!
3272

3273
  int32_t num = 0;
34,506,837✔
3274
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
34,521,271✔
3275
  QUERY_CHECK_CODE(code, lino, _error);
34,564,654!
3276

3277
  pInfo->pTagCond = pTagCond;
34,564,654✔
3278
  pInfo->pTagIndexCond = pTagIndexCond;
34,577,406✔
3279
  pInfo->suid = pPhyNode->suid;
34,564,150✔
3280
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
34,535,605✔
3281

3282
  pInfo->pTableListInfo = pTableListInfo;
34,526,129✔
3283
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
34,479,625✔
3284
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
34,591,831!
3285

3286
  pInfo->readHandle = *pReadHandle;
34,508,875✔
3287
  pInfo->curPos = 0;
34,533,850✔
3288

3289
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
34,485,937✔
3290
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
34,490,588✔
3291
                  pTaskInfo);
3292
  initResultSizeInfo(&pOperator->resultInfo, 4096);
34,483,862✔
3293
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
34,518,764✔
3294
  QUERY_CHECK_CODE(code, lino, _error);
34,596,356!
3295

3296
  if (pTagScanNode->onlyMetaCtbIdx) {
34,596,356!
3297
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
30,824,632✔
3298
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
30,832,761!
3299

3300
    pInfo->aFilterIdxs = taosArrayInit(pOperator->resultInfo.capacity, sizeof(int32_t));
30,829,909✔
3301
    QUERY_CHECK_NULL(pInfo->aFilterIdxs, code, lino, _error, terrno);
30,830,496!
3302

3303
    pInfo->filterCtx.colHash =
30,824,094✔
3304
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
30,822,437✔
3305
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
30,821,269!
3306

3307
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
30,820,828✔
3308
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
30,823,504!
3309

3310
    if (pInfo->pTagCond != NULL) {
30,819,574✔
3311
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
10,595,531✔
3312
    }
3313
  }
3314
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
34,598,243!
3315
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
34,595,979✔
3316
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
3317
  *pOptrInfo = pOperator;
34,582,943✔
3318
  return code;
34,581,733✔
3319

3320
_error:
×
3321
  if (pInfo) {
×
3322
    pInfo->pTableListInfo = NULL;
×
3323
  }
3324

3325
  if (pInfo != NULL) destroyTagScanOperatorInfo(pInfo);
×
3326
  if (pOperator != NULL) {
×
3327
    pOperator->info = NULL;
×
3328
    destroyOperator(pOperator);
×
3329
  }
3330
  return code;
×
3331
}
3332

3333
// table merge scan operator
3334

3335
static int32_t subTblRowCompareTsFn(const void* pLeft, const void* pRight, void* param) {
1,230,149,517✔
3336
  int32_t                 left = *(int32_t*)pLeft;
1,230,149,517✔
3337
  int32_t                 right = *(int32_t*)pRight;
1,230,173,798✔
3338
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
1,230,176,401✔
3339

3340
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
1,230,176,401✔
3341
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
1,230,181,023✔
3342

3343
  if (leftIdx == -1) {
1,230,176,580✔
3344
    return 1;
1,053,321,540✔
3345
  } else if (rightIdx == -1) {
176,855,040✔
3346
    return -1;
1,751,040✔
3347
  }
3348

3349
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
175,104,000✔
3350
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
175,104,000✔
3351
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
175,104,000!
3352
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
175,104,000!
3353
    ret = -1 * ret;
×
3354
  }
3355
  return ret;
175,104,000✔
3356
}
3357

3358
static int32_t subTblRowCompareTsPkFn(const void* pLeft, const void* pRight, void* param) {
×
3359
  int32_t                 left = *(int32_t*)pLeft;
×
3360
  int32_t                 right = *(int32_t*)pRight;
×
3361
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
×
3362

3363
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
×
3364
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
×
3365

3366
  if (leftIdx == -1) {
×
3367
    return 1;
×
3368
  } else if (rightIdx == -1) {
×
3369
    return -1;
×
3370
  }
3371

3372
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
×
3373
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
×
3374
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
×
3375
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
×
3376
    ret = -1 * ret;
×
3377
  }
3378
  if (ret == 0 && pInfo->pPkOrderInfo) {
×
3379
    ret = tsortComparBlockCell(pInfo->aInputs[left].pInputBlock, pInfo->aInputs[right].pInputBlock, leftIdx, rightIdx,
×
3380
                               pInfo->pPkOrderInfo);
×
3381
  }
3382
  return ret;
×
3383
}
3384

3385
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
15,626,816✔
3386
  int32_t code = TSDB_CODE_SUCCESS;
15,626,816✔
3387
  int32_t lino = 0;
15,626,816✔
3388
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
15,626,816!
3389
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
15,626,816!
3390
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
15,585,071!
3391
  for (int i = 0; i < src->numOfCols; i++) {
53,189,151✔
3392
    dst->colList[i] = src->colList[i];
37,620,480✔
3393
  }
3394

3395
_end:
15,600,796✔
3396
  if (code != TSDB_CODE_SUCCESS) {
15,600,796!
3397
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3398
  }
3399
  return code;
15,575,573✔
3400
}
3401

3402
static int32_t fetchNextSubTableBlockFromReader(SOperatorInfo* pOperator, STmsSubTableInput* pInput,
23,437,214✔
3403
                                                bool* pSubTableHasBlock) {
3404
  int32_t code = 0;
23,437,214✔
3405

3406
  STableMergeScanInfo*    pInfo = pOperator->info;
23,437,214✔
3407
  SReadHandle*            pHandle = &pInfo->base.readHandle;
23,432,522✔
3408
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
23,435,823✔
3409
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
23,433,058✔
3410
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
23,423,369✔
3411

3412
  blockDataCleanup(pInput->pReaderBlock);
23,434,307✔
3413
  if (!pInput->bInMemReader) {
23,432,809!
3414
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
23,504,152✔
3415
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
11,767,926✔
3416
    if (code != 0) {
11,762,131!
3417
      return code;
×
3418
    }
3419
  }
3420

3421
  pInfo->base.dataReader = pInput->pReader;
23,436,332✔
3422

3423
  while (true) {
42,097✔
3424
    bool hasNext = false;
23,475,957✔
3425
    code = pAPI->tsdReader.tsdNextDataBlock(pInfo->base.dataReader, &hasNext);
23,478,864✔
3426
    if (code != 0) {
23,495,363!
3427
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3428
      pInfo->base.dataReader = NULL;
×
3429
      return code;
×
3430
    }
3431

3432
    if (!hasNext || isTaskKilled(pTaskInfo)) {
23,495,363!
3433
      if (isTaskKilled(pTaskInfo)) {
15,658,993!
3434
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3435
        pInfo->base.dataReader = NULL;
×
3436
        return code;
×
3437
      }
3438

3439
      *pSubTableHasBlock = false;
15,661,282✔
3440
      break;
15,658,690✔
3441
    }
3442

3443
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
7,836,370✔
3444
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
7,689,484✔
3445
    } else {
3446
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
146,886✔
3447
    }
3448

3449
    uint32_t status = 0;
7,836,370✔
3450
    code = loadDataBlock(pOperator, &pInfo->base, pInput->pReaderBlock, &status);
7,836,370✔
3451
    if (code != 0) {
7,836,370!
3452
      pInfo->base.dataReader = NULL;
×
3453
      return code;
×
3454
    }
3455

3456
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
7,836,370!
3457
      *pSubTableHasBlock = false;
×
3458
      break;
×
3459
    }
3460
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pInput->pReaderBlock->info.rows == 0) {
7,836,370✔
3461
      continue;
42,097✔
3462
    }
3463

3464
    *pSubTableHasBlock = true;
7,793,207✔
3465
    break;
7,794,273✔
3466
  }
3467

3468
  if (*pSubTableHasBlock) {
23,456,005!
3469
    pInput->pReaderBlock->info.id.groupId =
15,585,964✔
3470
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
15,585,656✔
3471
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
7,794,273✔
3472
  }
3473
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
23,456,005!
3474
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
17,628,850✔
3475
    pInput->pReader = NULL;
17,630,690✔
3476
  }
3477

3478
  pInfo->base.dataReader = NULL;
23,452,649✔
3479
  return TSDB_CODE_SUCCESS;
23,453,104✔
3480
}
3481

3482
static int32_t setGroupStartEndIndex(STableMergeScanInfo* pInfo) {
9,759,416✔
3483
  pInfo->bGroupProcessed = false;
9,759,416✔
3484

3485
  int32_t numOfTables = 0;
9,754,254✔
3486
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
9,717,457✔
3487
  if (code != TSDB_CODE_SUCCESS) {
9,659,368!
3488
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3489
    return code;
×
3490
  }
3491

3492
  int32_t i = pInfo->tableStartIndex + 1;
9,659,368✔
3493
  for (; i < numOfTables; ++i) {
15,510,063✔
3494
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
6,157,748✔
3495
    if (!tableKeyInfo) {
6,145,810!
3496
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3497
      return terrno;
×
3498
    }
3499
    if (tableKeyInfo->groupId != pInfo->groupId) {
6,145,810✔
3500
      break;
309,803✔
3501
    }
3502
  }
3503
  pInfo->tableEndIndex = i - 1;
9,662,118✔
3504
  return TSDB_CODE_SUCCESS;
9,785,536✔
3505
}
3506

3507
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
9,803,526✔
3508
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
25,456,078✔
3509
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
15,656,216✔
3510
    if (pInput->rowIdx == -1) {
15,654,388✔
3511
      continue;
7,865,305✔
3512
    }
3513

3514
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
7,792,747!
3515
      pInput->rowIdx = 0;
7,792,747✔
3516
      pInput->pageIdx = -1;
7,791,670✔
3517
    }
3518

3519
    pInput->pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
7,791,670!
3520
    SColumnInfoData* col = taosArrayGet(pInput->pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
7,792,433✔
3521
    if (!col) {
7,790,593!
3522
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3523
      return terrno;
×
3524
    }
3525
    pInput->aTs = (int64_t*)col->pData;
7,790,593✔
3526
  }
3527

3528
  __merge_compare_fn_t mergeCompareFn = (!pSubTblsInfo->pPkOrderInfo) ? subTblRowCompareTsFn : subTblRowCompareTsPkFn;
9,801,690!
3529
  return tMergeTreeCreate(&pSubTblsInfo->pTree, pSubTblsInfo->numSubTables, pSubTblsInfo, mergeCompareFn);
9,803,206✔
3530
}
3531

3532
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
9,775,197✔
3533
  int32_t code = setGroupStartEndIndex(pInfo);
9,775,197✔
3534
  if (code != TSDB_CODE_SUCCESS) {
9,769,446!
3535
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3536
    return code;
×
3537
  }
3538
  STmsSubTablesMergeInfo* pSubTblsInfo = taosMemoryCalloc(1, sizeof(STmsSubTablesMergeInfo));
9,769,446!
3539
  if (pSubTblsInfo == NULL) {
9,726,955!
3540
    return terrno;
×
3541
  }
3542
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
9,726,955✔
3543
  if (!pSubTblsInfo->pTsOrderInfo) {
9,758,044!
3544
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3545
    return terrno;
×
3546
  }
3547
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
9,721,997!
3548
    pSubTblsInfo->pPkOrderInfo = taosArrayGet(pInfo->pSortInfo, 1);
×
3549
    if (!pSubTblsInfo->pPkOrderInfo) {
×
3550
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3551
      return terrno;
×
3552
    }
3553
  } else {
3554
    pSubTblsInfo->pPkOrderInfo = NULL;
9,709,171✔
3555
  }
3556
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
9,723,722✔
3557
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
9,719,113!
3558
  if (pSubTblsInfo->aInputs == NULL) {
9,700,696!
3559
    taosMemoryFree(pSubTblsInfo);
×
3560
    return terrno;
×
3561
  }
3562
  int32_t bufPageSize = pInfo->bufPageSize;
9,737,690✔
3563
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
9,730,242✔
3564
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
9,719,280✔
3565
  if (code != TSDB_CODE_SUCCESS) {
9,732,562!
3566
    taosMemoryFree(pSubTblsInfo->aInputs);
×
3567
    taosMemoryFree(pSubTblsInfo);
×
3568
    return code;
×
3569
  }
3570
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
9,732,562✔
3571
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
9,746,245✔
3572

3573
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
9,742,858✔
3574
  return TSDB_CODE_SUCCESS;
9,690,172✔
3575
}
3576

3577
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
9,741,737✔
3578
  int32_t        code = TSDB_CODE_SUCCESS;
9,741,737✔
3579
  int32_t        lino = 0;
9,741,737✔
3580
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
9,741,737✔
3581
  SReadHandle*   pHandle = &pInfo->base.readHandle;
9,743,064✔
3582
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
9,721,397✔
3583

3584
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
9,731,033✔
3585

3586
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
25,368,465✔
3587
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
15,606,449✔
3588
    pInput->type = SUB_TABLE_MEM_BLOCK;
15,524,883✔
3589

3590
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
15,540,906✔
3591
    QUERY_CHECK_CODE(code, lino, _end);
15,560,761!
3592

3593
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
15,560,761✔
3594
    QUERY_CHECK_CODE(code, lino, _end);
15,658,068!
3595

3596
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
15,658,068✔
3597
    QUERY_CHECK_CODE(code, lino, _end);
15,662,793!
3598

3599
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
15,662,793✔
3600
    pInput->pKeyInfo = keyInfo;
15,657,618✔
3601

3602
    if (isTaskKilled(pTaskInfo)) {
15,644,055!
3603
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
×
3604
    }
3605

3606
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
15,630,367✔
3607
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
11,691,312✔
3608
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
5,848,887✔
3609
      QUERY_CHECK_CODE(code, lino, _end);
5,848,708!
3610
      pInput->bInMemReader = true;
5,848,708✔
3611
    } else {
3612
      pInput->pReader = NULL;
9,800,494✔
3613
      pInput->bInMemReader = false;
9,801,252✔
3614
    }
3615
    bool hasNext = true;
15,643,499✔
3616
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
15,634,483✔
3617
    QUERY_CHECK_CODE(code, lino, _end);
15,656,980!
3618
    if (!hasNext) {
15,656,980!
3619
      pInput->rowIdx = -1;
7,863,784✔
3620
      ++pSubTblsInfo->numSubTablesCompleted;
7,869,283✔
3621
      continue;
7,867,762✔
3622
    } else {
3623
      pInput->rowIdx = 0;
7,793,196✔
3624
      pInput->pageIdx = -1;
7,794,273✔
3625
    }
3626
  }
3627

3628
_end:
9,808,564✔
3629
  if (code != TSDB_CODE_SUCCESS) {
9,806,421!
3630
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3631
  }
3632
  return code;
9,802,308✔
3633
}
3634

3635
static void adjustSubTableFromMemBlock(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
7,794,273✔
3636
  int32_t              code = TSDB_CODE_SUCCESS;
7,794,273✔
3637
  int32_t              lino = 0;
7,794,273✔
3638
  SExecTaskInfo*       pTaskInfo = pOperatorInfo->pTaskInfo;
7,794,273✔
3639
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
7,794,273✔
3640
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
7,794,273✔
3641
  bool                 hasNext = true;
7,794,273✔
3642
  code = fetchNextSubTableBlockFromReader(pOperatorInfo, pInput, &hasNext);
7,794,273✔
3643
  QUERY_CHECK_CODE(code, lino, _end);
7,794,273!
3644

3645
  if (!hasNext) {
7,794,273!
3646
    pInput->rowIdx = -1;
7,794,273✔
3647
    ++pSubTblsInfo->numSubTablesCompleted;
7,794,273✔
3648
  } else {
3649
    pInput->rowIdx = 0;
×
3650
  }
3651

3652
_end:
7,794,273✔
3653
  if (code != TSDB_CODE_SUCCESS) {
7,794,273!
3654
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3655
    pTaskInfo->code = code;
×
3656
    T_LONG_JMP(pTaskInfo->env, code);
×
3657
  }
3658
}
7,794,273✔
3659

3660
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
1,418,225,420✔
3661
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
1,418,225,420✔
3662
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,418,235,795✔
3663

3664
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,418,267,200!
3665
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
1,418,425,297✔
3666
    ++pInput->rowIdx;
1,410,602,935✔
3667
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
7,794,273!
3668
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
7,794,273!
3669
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
7,794,273✔
3670
    }
3671
    if (pInput->rowIdx != -1) {
7,794,273!
3672
      SColumnInfoData* col = taosArrayGet(pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
×
3673
      if (!col) {
×
3674
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3675
        return terrno;
×
3676
      }
3677
      pInput->pInputBlock = pInputBlock;
×
3678
      pInput->aTs = (int64_t*)col->pData;
×
3679
    }
3680
  }
3681

3682
  return tMergeTreeAdjust(pSubTblsInfo->pTree, tMergeTreeGetAdjustIndex(pSubTblsInfo->pTree));
1,418,405,612✔
3683
}
3684

3685
static int32_t appendChosenRowToDataBlock(STmsSubTablesMergeInfo* pSubTblsInfo, SSDataBlock* pBlock) {
1,417,700,327✔
3686
  int32_t            code = TSDB_CODE_SUCCESS;
1,417,700,327✔
3687
  int32_t            lino = 0;
1,417,700,327✔
3688
  STmsSubTableInput* pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,417,700,327✔
3689
  SSDataBlock*       pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,417,948,681!
3690

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

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

3699
    if (isNull) {
2,147,483,647✔
3700
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
520,112,448✔
3701
      QUERY_CHECK_CODE(code, lino, _end);
520,108,812!
3702
    } else {
3703
      if (pSrcColInfo->pData != NULL) {
2,147,483,647!
3704
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
2,147,483,647!
3705
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
2,147,483,647✔
3706
        QUERY_CHECK_CODE(code, lino, _end);
2,147,483,647!
3707
      }
3708
    }
3709
  }
3710
  pBlock->info.dataLoad = 1;
1,418,312,370✔
3711
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
1,418,314,973✔
3712
  pBlock->info.rows += 1;
1,418,311,601✔
3713

3714
_end:
1,418,292,913✔
3715
  if (code != TSDB_CODE_SUCCESS) {
1,418,292,913!
3716
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3717
  }
3718
  return code;
1,418,267,871✔
3719
}
3720

3721
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
15,851,792✔
3722
                                       SSDataBlock** pResBlock) {
3723
  int32_t                 code = TSDB_CODE_SUCCESS;
15,851,792✔
3724
  int32_t                 lino = 0;
15,851,792✔
3725
  STableMergeScanInfo*    pInfo = pOperator->info;
15,851,792✔
3726
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
15,853,318✔
3727
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
15,847,527✔
3728
  bool                    finished = false;
15,843,727✔
3729

3730
  QRY_PARAM_CHECK(pResBlock);
15,843,727!
3731

3732
  blockDataCleanup(pBlock);
15,845,238✔
3733

3734
  while (true) {
×
3735
    while (true) {
3736
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
1,433,673,514✔
3737
        finished = true;
15,854,076✔
3738
        break;
15,854,076✔
3739
      }
3740

3741
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
1,417,757,225✔
3742
      QUERY_CHECK_CODE(code, lino, _end);
1,418,217,551!
3743

3744
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
1,418,217,551✔
3745
      QUERY_CHECK_CODE(code, lino, _end);
1,417,875,449!
3746

3747
      if (pBlock->info.rows >= capacity) {
1,417,875,449!
3748
        break;
×
3749
      }
3750
    }
3751

3752
    if (isTaskKilled(pTaskInfo)) {
15,854,076!
3753
      return pTaskInfo->code;
×
3754
    }
3755

3756
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo);
15,854,076✔
3757
    if (finished || limitReached || pBlock->info.rows > 0) {
15,852,999!
3758
      break;
3759
    }
3760
  }
3761

3762
  if (pBlock->info.rows > 0) {
15,852,999✔
3763
    *pResBlock = pBlock;
6,043,233✔
3764
  }
3765

3766
_end:
9,809,008✔
3767
  if (code != TSDB_CODE_SUCCESS) {
15,852,241!
3768
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3769
    pTaskInfo->code = code;
×
3770
  }
3771
  return code;
15,850,417✔
3772
}
3773

3774
static int32_t startSubTablesTableMergeScan(SOperatorInfo* pOperator) {
9,771,225✔
3775
  int32_t              code = TSDB_CODE_SUCCESS;
9,771,225✔
3776
  int32_t              lino = 0;
9,771,225✔
3777
  STableMergeScanInfo* pInfo = pOperator->info;
9,771,225✔
3778

3779
  code = initSubTablesMergeInfo(pInfo);
9,784,026✔
3780
  QUERY_CHECK_CODE(code, lino, _end);
9,684,795!
3781

3782
  code = initSubTableInputs(pOperator, pInfo);
9,684,795✔
3783
  QUERY_CHECK_CODE(code, lino, _end);
9,801,550!
3784

3785
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
9,801,550✔
3786
  QUERY_CHECK_CODE(code, lino, _end);
9,804,278!
3787

3788
_end:
9,804,278✔
3789
  if (code != TSDB_CODE_SUCCESS) {
9,804,278!
3790
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3791
  }
3792
  return code;
9,799,098✔
3793
}
3794

3795
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
153,745,636✔
3796
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
153,745,636✔
3797
  if (pSubTblsInfo != NULL) {
153,754,819✔
3798
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
9,809,019✔
3799

3800
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
25,472,867✔
3801
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
15,663,540✔
3802
      taosMemoryFree(pInput->tblCond.colList);
15,662,793!
3803
      blockDataDestroy(pInput->pReaderBlock);
15,660,211✔
3804
      blockDataDestroy(pInput->pPageBlock);
15,662,348✔
3805
      taosArrayDestroy(pInput->aBlockPages);
15,663,226✔
3806
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
15,664,303✔
3807
      pInput->pReader = NULL;
15,662,024✔
3808
    }
3809

3810
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
9,809,777✔
3811
    taosMemoryFree(pSubTblsInfo->aInputs);
9,805,810!
3812

3813
    taosMemoryFree(pSubTblsInfo);
9,805,360!
3814
    pInfo->pSubTablesMergeInfo = NULL;
9,809,327✔
3815

3816
    // taosMemoryTrim(0);
3817
  }
3818
}
153,755,127✔
3819

3820
int32_t doTableMergeScanParaSubTablesNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
15,758,525✔
3821
  QRY_PARAM_CHECK(ppRes);
15,758,525!
3822

3823
  int32_t lino = 0;
15,729,913✔
3824
  int32_t tableListSize = 0;
15,729,913✔
3825
  int64_t st = taosGetTimestampUs();
15,747,266✔
3826

3827
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
15,747,266✔
3828
  STableMergeScanInfo* pInfo = pOperator->info;
15,708,097✔
3829

3830
  if (pOperator->status == OP_EXEC_DONE) {
15,713,545!
3831
    return TSDB_CODE_SUCCESS;
×
3832
  }
3833

3834
  int32_t code = pOperator->fpSet._openFn(pOperator);
15,705,321✔
3835
  QUERY_CHECK_CODE(code, lino, _end);
15,712,473!
3836

3837
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
15,712,473✔
3838
  QUERY_CHECK_CODE(code, lino, _end);
15,725,148!
3839

3840
  if (!pInfo->hasGroupId) {
15,725,148!
3841
    pInfo->hasGroupId = true;
9,679,045✔
3842

3843
    if (tableListSize == 0) {
9,657,238✔
3844
      setOperatorCompleted(pOperator);
252,350✔
3845
      (*ppRes) = NULL;
253,562✔
3846
      return code;
253,562✔
3847
    }
3848

3849
    pInfo->tableStartIndex = 0;
9,404,888✔
3850
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
9,399,506✔
3851
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
9,392,568!
3852
    pInfo->groupId = pTmpGpId->groupId;
9,392,568✔
3853
    code = startSubTablesTableMergeScan(pOperator);
9,446,673✔
3854
    QUERY_CHECK_CODE(code, lino, _end);
9,493,393!
3855
  }
3856

3857
  SSDataBlock* pBlock = NULL;
15,549,253✔
3858
  while (pInfo->tableStartIndex < tableListSize) {
15,847,203✔
3859
    if (isTaskKilled(pTaskInfo)) {
15,850,109!
3860
      break;
×
3861
    }
3862

3863
    code = getSubTablesSortedBlock(pOperator, pInfo->pResBlock, pOperator->resultInfo.capacity, &pBlock);
15,849,346✔
3864
    QUERY_CHECK_CODE(code, lino, _end);
15,853,318!
3865

3866
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
15,853,318!
3867
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
×
3868
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
×
3869

3870
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
×
3871
    }
3872

3873
    if (pBlock != NULL) {
15,850,736✔
3874
      pBlock->info.id.groupId = pInfo->groupId;
6,043,233✔
3875
      pOperator->resultInfo.totalRows += pBlock->info.rows;
6,043,233✔
3876
      pInfo->bGroupProcessed = true;
6,043,233✔
3877
      break;
6,043,233✔
3878
    } else {
3879
      // Data of this group are all dumped, let's try the next group
3880
      stopSubTablesTableMergeScan(pInfo);
9,807,503✔
3881
      if (pInfo->tableEndIndex >= tableListSize - 1) {
9,808,569✔
3882
        setOperatorCompleted(pOperator);
9,496,612✔
3883
        break;
9,494,914✔
3884
      }
3885

3886
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
310,880✔
3887
      STableKeyInfo* pTmpGpId = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
310,880✔
3888
      QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
310,880!
3889

3890
      pInfo->groupId = pTmpGpId->groupId;
310,880✔
3891
      code = startSubTablesTableMergeScan(pOperator);
310,880✔
3892
      QUERY_CHECK_CODE(code, lino, _end);
309,803!
3893
      resetLimitInfoForNextGroup(&pInfo->limitInfo);
309,803✔
3894
    }
3895
  }
3896

3897
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
15,541,059✔
3898

3899
_end:
15,539,548✔
3900
  if (code != TSDB_CODE_SUCCESS) {
15,534,823!
3901
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3902
    pTaskInfo->code = code;
×
3903
    T_LONG_JMP(pTaskInfo->env, code);
×
3904
  } else {
3905
    (*ppRes) = pBlock;
15,534,823✔
3906
  }
3907

3908
  return code;
15,536,955✔
3909
}
3910

3911
static void tableMergeScanDoSkipTable(uint64_t uid, void* pTableMergeOpInfo) {
29,697,061✔
3912
  int32_t              code = TSDB_CODE_SUCCESS;
29,697,061✔
3913
  int32_t              lino = 0;
29,697,061✔
3914
  SOperatorInfo*       pOperator = (SOperatorInfo*)pTableMergeOpInfo;
29,697,061✔
3915
  STableMergeScanInfo* pInfo = pOperator->info;
29,697,061✔
3916
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
29,697,200✔
3917

3918
  if (pInfo->mSkipTables == NULL) {
29,697,061✔
3919
    pInfo->mSkipTables = taosHashInit(pInfo->tableEndIndex - pInfo->tableStartIndex + 1,
2,021,428✔
3920
                                      taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_NO_LOCK);
3921
    QUERY_CHECK_NULL(pInfo->mSkipTables, code, lino, _end, terrno);
2,021,567!
3922
  }
3923
  int bSkip = 1;
29,697,339✔
3924
  if (pInfo->mSkipTables != NULL) {
29,697,200!
3925
    code = taosHashPut(pInfo->mSkipTables, &uid, sizeof(uid), &bSkip, sizeof(bSkip));
29,697,200✔
3926
    if (code == TSDB_CODE_DUP_KEY) {
29,697,200!
3927
      code = TSDB_CODE_SUCCESS;
×
3928
    }
3929
    QUERY_CHECK_CODE(code, lino, _end);
29,697,200!
3930
  }
3931

3932
_end:
29,697,200✔
3933
  if (code != TSDB_CODE_SUCCESS) {
29,697,200!
3934
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3935
    pTaskInfo->code = code;
×
3936
    T_LONG_JMP(pTaskInfo->env, code);
×
3937
  }
3938
}
29,693,006✔
3939

3940
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
728,892,972✔
3941
  STableMergeScanInfo* pInfo = pOperator->info;
728,892,972✔
3942
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
728,938,077✔
3943
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
728,843,159✔
3944
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
728,881,676✔
3945
  int32_t              code = 0;
728,873,917✔
3946
  bool                 hasNext = false;
728,873,917✔
3947
  STsdbReader*         reader = pInfo->base.dataReader;
728,882,264✔
3948

3949
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
728,868,337✔
3950
  if (code != 0) {
728,826,897!
3951
    pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
3952
    qError("table merge scan fetch next data block error code: %d, %s", code, GET_TASKID(pTaskInfo));
×
3953
    pTaskInfo->code = code;
×
3954
    return code;
×
3955
  }
3956

3957
  if (!hasNext || isTaskKilled(pTaskInfo)) {
728,826,897!
3958
    if (isTaskKilled(pTaskInfo)) {
161,725,730✔
3959
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
792!
3960
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
792✔
3961
    }
3962
    *pFinished = true;
161,573,863✔
3963
    return code;
161,554,093✔
3964
  }
3965

3966
  uint32_t status = 0;
567,246,306✔
3967
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
567,286,097✔
3968

3969
  if (code != TSDB_CODE_SUCCESS) {
567,197,650!
3970
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
×
3971
    pTaskInfo->code = code;
×
3972
    return code;
×
3973
  }
3974

3975
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
567,197,650!
3976
    *pFinished = true;
×
3977
    return code;
×
3978
  }
3979

3980
  // current block is filter out according to filter condition, continue load the next block
3981
  if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
567,197,650✔
3982
    *pSkipped = true;
17,217,042✔
3983
    return code;
17,289,631✔
3984
  }
3985

3986
  return code;
549,975,713✔
3987
}
3988

3989
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
733,135,441✔
3990
  STableMergeScanSortSourceParam* source = param;
733,135,441✔
3991

3992
  SOperatorInfo*       pOperator = source->pOperator;
733,135,441✔
3993
  STableMergeScanInfo* pInfo = pOperator->info;
733,280,251✔
3994
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
733,313,679✔
3995
  SSDataBlock*         pBlock = NULL;
733,320,360✔
3996
  int64_t              st = taosGetTimestampUs();
733,316,846✔
3997
  int32_t              code = TSDB_CODE_SUCCESS;
733,316,846✔
3998

3999
  while (true) {
4000
    if (pInfo->rtnNextDurationBlocks) {
769,197,187!
4001
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
40,288,430!
4002
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
4003

4004
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
40,292,138✔
4005
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
18,593,080✔
4006
        ++pInfo->nextDurationBlocksIdx;
18,593,080✔
4007
      } else {
4008
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
40,293,737✔
4009
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
18,590,710✔
4010
          pInfo->nextDurationBlocks[i] = NULL;
18,590,710✔
4011
        }
4012

4013
        pInfo->rtnNextDurationBlocks = false;
21,703,027✔
4014
        pInfo->nextDurationBlocksIdx = 0;
21,703,027✔
4015
        pInfo->numNextDurationBlocks = 0;
21,703,027✔
4016
        continue;
21,703,027✔
4017
      }
4018
    } else {
4019
      bool bFinished = false;
728,897,029✔
4020
      bool bSkipped = false;
728,818,176✔
4021

4022
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
728,951,158✔
4023
      if (code != 0) {
728,804,191!
4024
        return code;
×
4025
      }
4026

4027
      pBlock = pInfo->pReaderBlock;
728,804,191✔
4028
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
728,866,597!
4029
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
4030
      if (bFinished) {
729,006,501!
4031
        pInfo->bNewFilesetEvent = false;
161,579,767✔
4032
        break;
161,582,055✔
4033
      }
4034

4035
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
567,426,734!
4036
        if (!bSkipped) {
21,730,370!
4037
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
18,593,080✔
4038
          if (code) {
18,593,080!
4039
            *ppBlock = NULL;
×
4040
            return code;
×
4041
          }
4042

4043
          ++pInfo->numNextDurationBlocks;
18,593,080✔
4044
          if (pInfo->numNextDurationBlocks > 2) {
18,590,119!
4045
            qError("%s table merge scan prefetch %d next duration blocks. end early.", GET_TASKID(pTaskInfo),
×
4046
                   pInfo->numNextDurationBlocks);
4047
            pInfo->bNewFilesetEvent = false;
×
4048
            break;
×
4049
          }
4050
        }
4051

4052
        if (pInfo->bNewFilesetEvent) {
21,727,093!
4053
          pInfo->rtnNextDurationBlocks = true;
21,704,338✔
4054
          *ppBlock = NULL;
21,704,339✔
4055
          return code;
21,700,532✔
4056
        }
4057

4058
        if (pInfo->bNextDurationBlockEvent) {
×
4059
          pInfo->bNextDurationBlockEvent = false;
×
4060
          continue;
×
4061
        }
4062
      }
4063
      if (bSkipped) continue;
545,699,497!
4064
    }
4065

4066
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
550,146,820✔
4067

4068
    pOperator->resultInfo.totalRows += pBlock->info.rows;
550,143,525✔
4069
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
550,143,128✔
4070
    *ppBlock = pBlock;
550,142,386✔
4071

4072
    return code;
550,139,934✔
4073
  }
4074

4075
  *ppBlock = NULL;
161,577,794✔
4076
  return code;
161,581,051✔
4077
}
4078

4079
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
143,849,859✔
4080
  int32_t code = TSDB_CODE_SUCCESS;
143,849,859✔
4081
  int32_t lino = 0;
143,849,859✔
4082
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
143,849,859✔
4083
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
143,752,180!
4084
  SBlockOrderInfo biTs = {0};
143,752,180✔
4085
  SBlockOrderInfo biPk = {0};
143,772,188✔
4086

4087
  int32_t tsTargetSlotId = 0;
143,794,320✔
4088
  int32_t pkTargetSlotId = -1;
143,794,320✔
4089
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
559,069,408✔
4090
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
415,475,195✔
4091
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
415,493,922!
4092
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
415,493,922✔
4093
      tsTargetSlotId = colInfo->dstSlotId;
143,860,715✔
4094
      biTs.order = order;
143,876,662✔
4095
      biTs.slotId = tsTargetSlotId;
143,876,662✔
4096
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
143,876,662✔
4097
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
143,876,662✔
4098
    }
4099
    // TODO: order by just ts
4100
    if (colInfo->isPk) {
415,460,644!
4101
      pkTargetSlotId = colInfo->dstSlotId;
6,314,783✔
4102
      biPk.order = order;
6,314,783✔
4103
      biPk.slotId = pkTargetSlotId;
6,314,783✔
4104
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
6,314,783✔
4105
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
6,314,783✔
4106
    }
4107
  }
4108

4109
  void* tmp = taosArrayPush(pSortInfo, &biTs);
143,928,409✔
4110
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
143,928,409!
4111
  if (pkTargetSlotId != -1) {
143,928,409✔
4112
    tmp = taosArrayPush(pSortInfo, &biPk);
6,314,232✔
4113
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
6,314,232!
4114
  }
4115

4116
  (*ppSortArray) = pSortInfo;
143,928,409✔
4117

4118
_end:
143,659,176✔
4119
  if (code != TSDB_CODE_SUCCESS) {
143,765,445!
4120
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4121
  }
4122
  return code;
143,765,445✔
4123
}
4124

4125
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
57,596,047✔
4126
  STableMergeScanInfo* pTmsInfo = param;
57,596,047✔
4127
  if (type == TSD_READER_NOTIFY_DURATION_START) {
57,596,047!
4128
    pTmsInfo->bNewFilesetEvent = true;
57,611,985✔
4129
  } else if (type == TSD_READER_NOTIFY_NEXT_DURATION_BLOCK) {
×
4130
    pTmsInfo->bNextDurationBlockEvent = true;
×
4131
  }
4132
  qDebug("table merge scan receive notification. type %d, fileset %d", type, info->duration.filesetId);
57,555,268!
4133

4134
  return;
57,635,272✔
4135
}
4136

4137
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
184,333,252✔
4138
  STableMergeScanInfo* pInfo = pOperator->info;
184,333,252✔
4139
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
184,411,865✔
4140

4141
  int32_t code = TSDB_CODE_SUCCESS;
184,432,462✔
4142
  int32_t lino = 0;
184,432,462✔
4143
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
184,432,462✔
4144

4145
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
184,352,664✔
4146
  pInfo->bNewFilesetEvent = false;
184,514,401✔
4147
  pInfo->bNextDurationBlockEvent = false;
184,461,613✔
4148

4149
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
184,411,139✔
4150
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
184,471,221!
4151

4152
  pInfo->pSortHandle = NULL;
184,447,705✔
4153
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
368,887,650✔
4154
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
184,428,368✔
4155
  if (code) {
184,381,749!
4156
    return code;
×
4157
  }
4158

4159
  if (pInfo->bSortRowId && numOfTable != 1) {
184,381,749!
4160
    int32_t memSize = 512 * 1024 * 1024;
15,691,517✔
4161
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
15,691,517✔
4162
    QUERY_CHECK_CODE(code, lino, _end);
15,698,077!
4163
  }
4164

4165
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
184,404,497✔
4166
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
184,343,679✔
4167
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
184,323,142✔
4168

4169
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
184,337,342✔
4170
  QUERY_CHECK_CODE(code, lino, _end);
184,316,997!
4171

4172
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
184,316,997!
4173
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
184,356,729!
4174
  param->pOperator = pOperator;
184,356,729✔
4175

4176
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
184,336,122!
4177
  if (ps == NULL) {
184,294,442!
4178
    taosMemoryFree(param);
×
4179
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
4180
  }
4181

4182
  ps->param = param;
184,294,442✔
4183
  ps->onlyRef = false;
184,362,947✔
4184
  code = tsortAddSource(pInfo->pSortHandle, ps);
184,446,548✔
4185
  QUERY_CHECK_CODE(code, lino, _end);
184,467,695!
4186

4187
  if (numOfTable == 1) {
184,467,695✔
4188
    tsortSetSingleTableMerge(pInfo->pSortHandle);
132,597,907✔
4189
  } else {
4190
    code = tsortOpen(pInfo->pSortHandle);
51,869,788✔
4191
    QUERY_CHECK_CODE(code, lino, _end);
51,896,915!
4192
  }
4193

4194
_end:
51,896,915✔
4195
  if (code != TSDB_CODE_SUCCESS) {
184,425,287!
4196
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4197
  }
4198
  return code;
184,456,111✔
4199
}
4200

4201
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
184,547,071✔
4202
  STableMergeScanInfo* pInfo = pOperator->info;
184,547,071✔
4203
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
184,546,364✔
4204
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
184,548,163✔
4205

4206
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
184,548,163✔
4207
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
184,547,440✔
4208
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
184,542,305✔
4209
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
184,546,401✔
4210
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
184,544,702✔
4211
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
184,535,014✔
4212

4213
  tsortDestroySortHandle(pInfo->pSortHandle);
184,539,341✔
4214
  pInfo->pSortHandle = NULL;
184,545,267✔
4215
}
184,538,781✔
4216

4217
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
162,759,971✔
4218
  int32_t              code = TSDB_CODE_SUCCESS;
162,759,971✔
4219
  int32_t              lino = 0;
162,759,971✔
4220
  STableMergeScanInfo* pInfo = pOperator->info;
162,759,971✔
4221
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
162,729,499✔
4222
  SReadHandle*         pHandle = &pInfo->base.readHandle;
162,674,432✔
4223
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
162,667,595✔
4224
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
162,605,181✔
4225

4226
  {
4227
    int32_t numOfTables = 0;
162,614,787✔
4228
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
162,509,278✔
4229
    QUERY_CHECK_CODE(code, lino, _end);
162,643,313!
4230

4231
    int32_t i = pInfo->tableStartIndex + 1;
162,643,313✔
4232
    for (; i < numOfTables; ++i) {
627,229,204✔
4233
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
504,084,012✔
4234
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
504,027,447!
4235
      if (tableKeyInfo->groupId != pInfo->groupId) {
504,027,447✔
4236
        break;
39,489,503✔
4237
      }
4238
    }
4239
    pInfo->tableEndIndex = i - 1;
162,634,695✔
4240
  }
4241
  pInfo->bGroupProcessed = false;
162,782,237✔
4242
  int32_t tableStartIdx = pInfo->tableStartIndex;
162,785,845✔
4243
  int32_t tableEndIdx = pInfo->tableEndIndex;
162,766,882✔
4244

4245
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
162,768,018✔
4246
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
162,768,018✔
4247
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
325,467,117✔
4248
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
162,683,452✔
4249
  QUERY_CHECK_CODE(code, lino, _end);
162,735,013!
4250
  if (pInfo->filesetDelimited) {
162,735,013!
4251
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
162,684,194✔
4252
  }
4253
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
162,832,567✔
4254

4255
  code = startDurationForGroupTableMergeScan(pOperator);
162,572,252✔
4256
  QUERY_CHECK_CODE(code, lino, _end);
162,744,748!
4257

4258
_end:
162,744,748✔
4259
  if (code != TSDB_CODE_SUCCESS) {
162,744,297!
4260
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4261
    pTaskInfo->code = code;
×
4262
    T_LONG_JMP(pTaskInfo->env, code);
×
4263
  }
4264
}
162,744,297✔
4265

4266
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
162,839,046✔
4267
  STableMergeScanInfo* pInfo = pOperator->info;
162,839,046✔
4268
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
162,842,896✔
4269
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
162,842,757✔
4270

4271
  stopDurationForGroupTableMergeScan(pOperator);
162,843,587✔
4272

4273
  if (pInfo->base.dataReader != NULL) {
162,838,954!
4274
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
162,842,580✔
4275
    pInfo->base.dataReader = NULL;
162,744,800✔
4276
  }
4277
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
162,769,550✔
4278
    if (pInfo->nextDurationBlocks[i]) {
2,370!
4279
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
2,370✔
4280
      pInfo->nextDurationBlocks[i] = NULL;
2,370✔
4281
    }
4282
    pInfo->numNextDurationBlocks = 0;
2,370✔
4283
    pInfo->nextDurationBlocksIdx = 0;
2,370✔
4284
  }
4285
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
162,779,299✔
4286
  taosHashCleanup(pInfo->mSkipTables);
162,714,687✔
4287
  pInfo->mSkipTables = NULL;
162,733,511✔
4288
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
162,745,477✔
4289
}
162,852,181✔
4290

4291
// all data produced by this function only belongs to one group
4292
// slimit/soffset does not need to be concerned here, since this function only deal with data within one group.
4293
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
360,017,406✔
4294
                                              SOperatorInfo* pOperator) {
4295
  STableMergeScanInfo* pInfo = pOperator->info;
360,017,406✔
4296
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
360,039,949✔
4297
  STupleHandle*        pTupleHandle = NULL;
360,004,454✔
4298

4299
  blockDataCleanup(pResBlock);
359,977,920✔
4300

4301
  while (1) {
×
4302
    while (1) {
2,147,483,647✔
4303
      pTupleHandle = NULL;
2,147,483,647✔
4304
      int32_t code = tsortNextTuple(pHandle, &pTupleHandle);
2,147,483,647✔
4305
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
4306
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
1,234,442!
4307
        T_LONG_JMP(pOperator->pTaskInfo->env, code);
1,234,442!
4308
      }
4309
      if (pTupleHandle == NULL) {
2,147,483,647✔
4310
        break;
289,110,146✔
4311
      }
4312

4313
      code = tsortAppendTupleToBlock(pInfo->pSortHandle, pResBlock, pTupleHandle);
2,147,483,647✔
4314
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647!
4315
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
4316
        T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4317
      }
4318

4319
      if (pResBlock->info.rows >= capacity) {
2,147,483,647✔
4320
        break;
70,974,711✔
4321
      }
4322
    }
4323

4324
    if (tsortIsClosed(pHandle)) {
360,084,857!
4325
      terrno = TSDB_CODE_TSC_QUERY_CANCELLED;
×
4326
      T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4327
    }
4328

4329
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
360,108,681✔
4330
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
360,104,525✔
4331
           pInfo->limitInfo.numOfOutputRows);
4332
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
360,103,913!
4333
      break;
4334
    }
4335
  }
4336
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
360,103,913✔
4337
}
4338

4339
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
309,767,513✔
4340
  if (pOperator->status == OP_EXEC_DONE) {
309,767,513✔
4341
    (*ppRes) = NULL;
105,226✔
4342
    return TSDB_CODE_SUCCESS;
105,226✔
4343
  }
4344

4345
  int32_t              code = TSDB_CODE_SUCCESS;
309,553,180✔
4346
  int32_t              lino = 0;
309,553,180✔
4347
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
309,553,180✔
4348
  STableMergeScanInfo* pInfo = pOperator->info;
309,569,959✔
4349

4350
  code = pOperator->fpSet._openFn(pOperator);
309,546,655✔
4351
  QUERY_CHECK_CODE(code, lino, _end);
309,524,156!
4352

4353
  int64_t st = taosGetTimestampUs();
309,573,514✔
4354

4355
  int32_t tableListSize = 0;
309,573,514✔
4356
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
309,536,056✔
4357
  QUERY_CHECK_CODE(code, lino, _end);
309,568,782!
4358

4359
  if (!pInfo->hasGroupId) {
309,568,782!
4360
    pInfo->hasGroupId = true;
133,981,742✔
4361

4362
    if (tableListSize == 0) {
133,968,906✔
4363
      setOperatorCompleted(pOperator);
10,830,156✔
4364
      (*ppRes) = NULL;
10,830,958✔
4365
      return code;
10,829,634✔
4366
    }
4367
    pInfo->tableStartIndex = 0;
123,138,750✔
4368
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
123,157,303✔
4369
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
123,044,294!
4370
    pInfo->groupId = tmp->groupId;
123,044,294✔
4371
    startGroupTableMergeScan(pOperator);
123,219,494✔
4372
  }
4373

4374
  SSDataBlock* pBlock = NULL;
298,821,730✔
4375
  while (pInfo->tableStartIndex < tableListSize) {
360,012,583✔
4376
    if (isTaskKilled(pTaskInfo)) {
360,039,590✔
4377
      goto _end;
2,436✔
4378
    }
4379

4380
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
360,029,441✔
4381
                                              pOperator);
4382
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
360,105,445!
4383
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
12,471,917✔
4384
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
12,471,917!
4385
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
12,471,917✔
4386
    }
4387
    if (pBlock != NULL) {
360,102,424✔
4388
      pBlock->info.id.groupId = pInfo->groupId;
175,557,379✔
4389
      pOperator->resultInfo.totalRows += pBlock->info.rows;
175,559,624✔
4390
      pInfo->bGroupProcessed = true;
175,560,614✔
4391
      break;
175,559,240✔
4392
    } else {
4393
      if (pInfo->bNewFilesetEvent) {
184,545,045!
4394
        stopDurationForGroupTableMergeScan(pOperator);
21,705,397✔
4395
        code = startDurationForGroupTableMergeScan(pOperator);
21,704,371✔
4396
        QUERY_CHECK_CODE(code, lino, _end);
21,700,354!
4397
      } else {
4398
        // Data of this group are all dumped, let's try the next group
4399
        stopGroupTableMergeScan(pOperator);
162,842,525✔
4400
        if (pInfo->tableEndIndex >= tableListSize - 1) {
162,829,147✔
4401
          setOperatorCompleted(pOperator);
123,334,930✔
4402
          break;
123,335,321✔
4403
        }
4404

4405
        pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
39,489,509✔
4406
        STableKeyInfo* tmp = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
39,489,509✔
4407
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
39,489,509!
4408
        pInfo->groupId = tmp->groupId;
39,489,509✔
4409
        startGroupTableMergeScan(pOperator);
39,490,499✔
4410
        resetLimitInfoForNextGroup(&pInfo->limitInfo);
39,488,875✔
4411
      }
4412
    }
4413
  }
4414

4415
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
298,880,169✔
4416

4417
_end:
298,908,380✔
4418
  if (code != TSDB_CODE_SUCCESS) {
298,906,864!
4419
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4420
    pTaskInfo->code = code;
×
4421
    T_LONG_JMP(pTaskInfo->env, code);
×
4422
  } else {
4423
    (*ppRes) = pBlock;
298,906,864✔
4424
  }
4425

4426
  return code;
298,897,531✔
4427
}
4428

4429
static SSDataBlock* doTableMergeScan(SOperatorInfo* pOperator) {
×
4430
  SSDataBlock* pRes = NULL;
×
4431
  int32_t      code = doTableMergeScanNext(pOperator, &pRes);
×
4432
  return pRes;
×
4433
}
4434

4435
void destroyTableMergeScanOperatorInfo(void* param) {
143,931,791✔
4436
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
143,931,791✔
4437

4438
  // start one reader variable
4439
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
143,931,791✔
4440
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
143,922,790✔
4441
    pTableScanInfo->base.dataReader = NULL;
143,923,538✔
4442
  }
4443

4444
  for (int32_t i = 0; i < pTableScanInfo->numNextDurationBlocks; ++i) {
143,925,069!
4445
    if (pTableScanInfo->nextDurationBlocks[i] != NULL) {
×
4446
      blockDataDestroy(pTableScanInfo->nextDurationBlocks[i]);
×
4447
      pTableScanInfo->nextDurationBlocks[i] = NULL;
×
4448
    }
4449
  }
4450

4451
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
143,927,307✔
4452
  pTableScanInfo->pSortHandle = NULL;
143,924,004✔
4453
  taosHashCleanup(pTableScanInfo->mSkipTables);
143,927,280✔
4454
  pTableScanInfo->mSkipTables = NULL;
143,914,104✔
4455
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
143,917,303✔
4456
  pTableScanInfo->pSortInputBlock = NULL;
143,937,384✔
4457
  // end one reader variable
4458

4459
  cleanupQueryTableDataCond(&pTableScanInfo->base.cond);
143,937,718✔
4460
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
143,907,334✔
4461

4462
  blockDataDestroy(pTableScanInfo->pResBlock);
143,947,190✔
4463
  pTableScanInfo->pResBlock = NULL;
143,943,468✔
4464

4465
  // remove it from the task->result list
4466
  blockDataDestroy(pTableScanInfo->pReaderBlock);
143,942,094✔
4467
  pTableScanInfo->pReaderBlock = NULL;
143,945,067✔
4468
  taosArrayDestroy(pTableScanInfo->pSortInfo);
143,946,985✔
4469

4470
  stopSubTablesTableMergeScan(pTableScanInfo);
143,932,448✔
4471

4472
  taosMemoryFreeClear(param);
143,935,426!
4473
}
143,923,709✔
4474

4475
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
4,471,591✔
4476
  if (pOptr == NULL) {
4,471,591!
4477
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
4478
    return TSDB_CODE_INVALID_PARA;
×
4479
  }
4480
  // TODO: merge these two info into one struct
4481
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
4,471,591!
4482
  if (!execInfo) {
4,471,591!
4483
    return terrno;
×
4484
  }
4485
  STableMergeScanInfo* pInfo = pOptr->info;
4,471,591✔
4486
  execInfo->blockRecorder = pInfo->base.readRecorder;
4,471,591✔
4487
  execInfo->sortExecInfo = pInfo->sortExecInfo;
4,471,591✔
4488

4489
  *pOptrExplain = execInfo;
4,471,591✔
4490
  *len = sizeof(STableMergeScanExecInfo);
4,471,591✔
4491

4492
  return TSDB_CODE_SUCCESS;
4,471,591✔
4493
}
4494

4495
int32_t createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
143,775,035✔
4496
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
4497
                                         SOperatorInfo** pOptrInfo) {
4498
  QRY_PARAM_CHECK(pOptrInfo);
143,775,035!
4499

4500
  int32_t              code = TSDB_CODE_SUCCESS;
143,769,864✔
4501
  int32_t              lino = 0;
143,769,864✔
4502
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
143,769,864!
4503
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
143,532,142!
4504
  if (pInfo == NULL || pOperator == NULL) {
143,490,068!
4505
    code = terrno;
×
4506
    goto _error;
×
4507
  }
4508

4509
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
143,491,330✔
4510

4511
  int32_t numOfCols = 0;
143,534,429✔
4512
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
143,574,573✔
4513
                             &pInfo->base.matchInfo);
4514
  QUERY_CHECK_CODE(code, lino, _error);
143,706,628!
4515

4516
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle, false);
143,706,628✔
4517
  QUERY_CHECK_CODE(code, lino, _error);
143,558,749!
4518

4519
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
143,558,749✔
4520
    SExprSupp* pSup = &pInfo->base.pseudoSup;
70,770,183✔
4521
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
70,832,223✔
4522
    QUERY_CHECK_CODE(code, lino, _error);
70,845,537!
4523

4524
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
70,845,537✔
4525
                                      &pTaskInfo->storageAPI.functionStore);
4526
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
70,723,380!
4527
  }
4528

4529
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
143,303,123✔
4530

4531
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
143,272,711✔
4532
  QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
143,829,186!
4533

4534
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
143,704,132✔
4535
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
143,808,104✔
4536
  pInfo->base.scanFlag = MAIN_SCAN;
143,578,605✔
4537
  pInfo->base.readHandle = *readHandle;
143,580,209✔
4538

4539
  pInfo->readIdx = -1;
143,669,643✔
4540

4541
  pInfo->base.limitInfo.limit.limit = -1;
143,544,581✔
4542
  pInfo->base.limitInfo.slimit.limit = -1;
143,575,491✔
4543
  pInfo->base.pTableListInfo = pTableListInfo;
143,683,389✔
4544

4545
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
143,551,767✔
4546
  pInfo->sample.seed = taosGetTimestampSec();
143,783,337✔
4547

4548
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
143,823,731✔
4549
                            pTaskInfo->pStreamRuntimeInfo);
143,808,343✔
4550
  QUERY_CHECK_CODE(code, lino, _error);
143,672,519!
4551

4552
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
143,672,519✔
4553

4554
  pInfo->mergeLimit = -1;
143,487,218✔
4555
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
143,505,403✔
4556
  if (hasLimit) {
143,439,501✔
4557
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1
25,746,994✔
4558
                            ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset
5,687,229✔
4559
                            : pInfo->limitInfo.limit.limit;
18,551,783✔
4560
    pInfo->mSkipTables = NULL;
12,868,233✔
4561
  }
4562

4563
  initResultSizeInfo(&pOperator->resultInfo, 1024);
143,432,544✔
4564
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
143,683,061✔
4565
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
143,907,519!
4566
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
143,622,676✔
4567
  QUERY_CHECK_CODE(code, lino, _error);
143,908,061!
4568
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
143,908,061!
4569
    pInfo->bSortRowId = true;
31,327,966✔
4570
  } else {
4571
    pInfo->bSortRowId = false;
112,586,587✔
4572
  }
4573

4574
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
143,833,806✔
4575
  QUERY_CHECK_CODE(code, lino, _error);
143,815,545!
4576

4577
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
143,815,545✔
4578
  QUERY_CHECK_CODE(code, lino, _error);
143,736,676!
4579

4580
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
143,736,676✔
4581
  QUERY_CHECK_CODE(code, lino, _error);
143,804,070!
4582

4583
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
143,804,070!
4584

4585
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
143,861,404✔
4586
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
143,761,285✔
4587

4588
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
143,603,803✔
4589

4590
  // start one reader variable
4591
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
143,877,108✔
4592
  QUERY_CHECK_CODE(code, lino, _error);
143,832,167!
4593

4594
  if (!tsExperimental) {
143,832,167!
4595
    pInfo->filesetDelimited = false;
×
4596
  } else {
4597
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
143,832,167!
4598
  }
4599
  // end one reader variable
4600
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
143,765,452✔
4601
                  pInfo, pTaskInfo);
4602
  pOperator->exprSupp.numOfExprs = numOfCols;
143,640,344✔
4603

4604
  pOperator->fpSet = createOperatorFpSet(
143,622,870✔
4605
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
143,754,477!
4606
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
4607
      NULL);
4608
  pOperator->cost.openCost = 0;
143,660,643✔
4609

4610
  *pOptrInfo = pOperator;
143,683,508✔
4611
  return code;
143,578,161✔
4612

4613
_error:
×
4614
  if (code != TSDB_CODE_SUCCESS) {
×
4615
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4616
  }
4617
  pTaskInfo->code = code;
×
4618
  if (pInfo != NULL) {
×
4619
    pInfo->base.pTableListInfo = NULL;
×
4620
    destroyTableMergeScanOperatorInfo(pInfo);
×
4621
  }
4622
  if (pOperator != NULL) {
×
4623
    pOperator->info = NULL;
×
4624
    destroyOperator(pOperator);
×
4625
  }
4626
  return code;
×
4627
}
4628

4629
// ====================================================================================================================
4630
// TableCountScanOperator
4631
static void        destoryTableCountScanOperator(void* param);
4632
static int32_t     buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
4633
                                                  SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI);
4634
static int32_t     buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
4635
                                                  SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI);
4636
static int32_t     buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4637
                                             STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName);
4638
static int32_t     buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4639
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName);
4640
static int32_t     buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4641
                                          STableCountScanSupp* pSupp, SSDataBlock* pRes);
4642
static void        buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
4643
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
4644
                                               size_t perfdbTableNum);
4645
static void        buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
4646
                                              size_t infodbTableNum, size_t perfdbTableNum);
4647
static const char* GROUP_TAG_DB_NAME = "db_name";
4648
static const char* GROUP_TAG_STABLE_NAME = "stable_name";
4649

4650
int32_t tblCountScanGetGroupTagsSlotId(const SNodeList* scanCols, STableCountScanSupp* supp) {
×
4651
  if (scanCols != NULL) {
×
4652
    SNode* pNode = NULL;
×
4653
    FOREACH(pNode, scanCols) {
×
4654
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
4655
        return TSDB_CODE_QRY_SYS_ERROR;
×
4656
      }
4657
      STargetNode* targetNode = (STargetNode*)pNode;
×
4658
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
×
4659
        return TSDB_CODE_QRY_SYS_ERROR;
×
4660
      }
4661
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
×
4662
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
4663
        supp->dbNameSlotId = targetNode->slotId;
×
4664
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
4665
        supp->stbNameSlotId = targetNode->slotId;
×
4666
      }
4667
    }
4668
  }
4669
  return TSDB_CODE_SUCCESS;
×
4670
}
4671

4672
int32_t tblCountScanGetCountSlotId(const SNodeList* pseudoCols, STableCountScanSupp* supp) {
×
4673
  if (pseudoCols != NULL) {
×
4674
    SNode* pNode = NULL;
×
4675
    FOREACH(pNode, pseudoCols) {
×
4676
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
4677
        return TSDB_CODE_QRY_SYS_ERROR;
×
4678
      }
4679
      STargetNode* targetNode = (STargetNode*)pNode;
×
4680
      if (nodeType(targetNode->pExpr) != QUERY_NODE_FUNCTION) {
×
4681
        return TSDB_CODE_QRY_SYS_ERROR;
×
4682
      }
4683
      SFunctionNode* funcNode = (SFunctionNode*)(targetNode->pExpr);
×
4684
      if (funcNode->funcType == FUNCTION_TYPE_TABLE_COUNT) {
×
4685
        supp->tbCountSlotId = targetNode->slotId;
×
4686
      }
4687
    }
4688
  }
4689
  return TSDB_CODE_SUCCESS;
×
4690
}
4691

4692
int32_t tblCountScanGetInputs(SNodeList* groupTags, SName* tableName, STableCountScanSupp* supp) {
×
4693
  if (groupTags != NULL) {
×
4694
    SNode* pNode = NULL;
×
4695
    FOREACH(pNode, groupTags) {
×
4696
      if (nodeType(pNode) != QUERY_NODE_COLUMN) {
×
4697
        return TSDB_CODE_QRY_SYS_ERROR;
×
4698
      }
4699
      SColumnNode* colNode = (SColumnNode*)pNode;
×
4700
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
4701
        supp->groupByDbName = true;
×
4702
      }
4703
      if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
4704
        supp->groupByStbName = true;
×
4705
      }
4706
    }
4707
  } else {
4708
    tstrncpy(supp->dbNameFilter, tNameGetDbNameP(tableName), TSDB_DB_NAME_LEN);
×
4709
    tstrncpy(supp->stbNameFilter, tNameGetTableName(tableName), TSDB_TABLE_NAME_LEN);
×
4710
  }
4711
  return TSDB_CODE_SUCCESS;
×
4712
}
4713

4714
int32_t getTableCountScanSupp(SNodeList* groupTags, SName* tableName, SNodeList* scanCols, SNodeList* pseudoCols,
×
4715
                              STableCountScanSupp* supp, SExecTaskInfo* taskInfo) {
4716
  int32_t code = 0;
×
4717
  code = tblCountScanGetInputs(groupTags, tableName, supp);
×
4718
  if (code != TSDB_CODE_SUCCESS) {
×
4719
    qError("%s get table count scan supp. get inputs error", GET_TASKID(taskInfo));
×
4720
    return code;
×
4721
  }
4722

4723
  supp->dbNameSlotId = -1;
×
4724
  supp->stbNameSlotId = -1;
×
4725
  supp->tbCountSlotId = -1;
×
4726

4727
  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
×
4728
  if (code != TSDB_CODE_SUCCESS) {
×
4729
    qError("%s get table count scan supp. get group tags slot id error", GET_TASKID(taskInfo));
×
4730
    return code;
×
4731
  }
4732

4733
  code = tblCountScanGetCountSlotId(pseudoCols, supp);
×
4734
  if (code != TSDB_CODE_SUCCESS) {
×
4735
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
×
4736
    return code;
×
4737
  }
4738
  return code;
×
4739
}
4740

4741
int32_t createTableCountScanOperatorInfo(SReadHandle* readHandle, STableCountScanPhysiNode* pTblCountScanNode,
×
4742
                                         SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
4743
  QRY_PARAM_CHECK(pOptrInfo);
×
4744

4745
  int32_t                      code = TSDB_CODE_SUCCESS;
×
4746
  int32_t                      lino = 0;
×
4747
  SScanPhysiNode*              pScanNode = &pTblCountScanNode->scan;
×
4748
  STableCountScanOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableCountScanOperatorInfo));
×
4749
  SOperatorInfo*               pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
4750
  if (!pInfo || !pOperator) {
×
4751
    code = terrno;
×
4752
    goto _error;
×
4753
  }
4754

4755
  pInfo->readHandle = *readHandle;
×
4756

4757
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
×
4758
  initResultSizeInfo(&pOperator->resultInfo, 1);
×
4759
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
×
4760
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
×
4761

4762
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
×
4763
  QUERY_CHECK_CODE(code, lino, _error);
×
4764

4765
  code = getTableCountScanSupp(pTblCountScanNode->pGroupTags, &pTblCountScanNode->scan.tableName,
×
4766
                               pTblCountScanNode->scan.pScanCols, pTblCountScanNode->scan.pScanPseudoCols, &pInfo->supp,
4767
                               pTaskInfo);
4768
  QUERY_CHECK_CODE(code, lino, _error);
×
4769

4770
  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
×
4771
                  pInfo, pTaskInfo);
4772
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScanNext, NULL, destoryTableCountScanOperator,
×
4773
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
4774
  *pOptrInfo = pOperator;
×
4775
  return code;
×
4776

4777
_error:
×
4778
  if (pInfo != NULL) {
×
4779
    destoryTableCountScanOperator(pInfo);
×
4780
  }
4781
  if (pOperator != NULL) {
×
4782
    pOperator->info = NULL;
×
4783
    destroyOperator(pOperator);
×
4784
  }
4785
  pTaskInfo->code = code;
×
4786
  return code;
×
4787
}
4788

4789
int32_t fillTableCountScanDataBlock(STableCountScanSupp* pSupp, char* dbName, char* stbName, int64_t count,
×
4790
                                    SSDataBlock* pRes) {
4791
  int32_t code = TSDB_CODE_SUCCESS;
×
4792
  int32_t lino = 0;
×
4793
  if (pSupp->dbNameSlotId != -1) {
×
4794
    QUERY_CHECK_CONDITION((strlen(dbName) > 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4795
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
×
4796
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
4797

4798
    char varDbName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
4799
    tstrncpy(varDataVal(varDbName), dbName, TSDB_DB_NAME_LEN);
×
4800

4801
    varDataSetLen(varDbName, strlen(dbName));
×
4802
    code = colDataSetVal(colInfoData, 0, varDbName, false);
×
4803
    QUERY_CHECK_CODE(code, lino, _end);
×
4804
  }
4805

4806
  if (pSupp->stbNameSlotId != -1) {
×
4807
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
×
4808
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
4809
    if (strlen(stbName) != 0) {
×
4810
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
4811
      tstrncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
×
4812
      varDataSetLen(varStbName, strlen(stbName));
×
4813
      code = colDataSetVal(colInfoData, 0, varStbName, false);
×
4814
      QUERY_CHECK_CODE(code, lino, _end);
×
4815
    } else {
4816
      colDataSetNULL(colInfoData, 0);
4817
    }
4818
  }
4819

4820
  if (pSupp->tbCountSlotId != -1) {
×
4821
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
×
4822
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
4823
    code = colDataSetVal(colInfoData, 0, (char*)&count, false);
×
4824
    QUERY_CHECK_CODE(code, lino, _end);
×
4825
  }
4826
  pRes->info.rows = 1;
×
4827

4828
_end:
×
4829
  if (code != TSDB_CODE_SUCCESS) {
×
4830
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4831
  }
4832
  return code;
×
4833
}
4834

4835
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
×
4836
  STableCountScanSupp* pSupp = &pInfo->supp;
×
4837
  SSDataBlock*         pRes = pInfo->pRes;
×
4838

4839
  size_t infodbTableNum;
×
4840
  getInfosDbMeta(NULL, &infodbTableNum);
×
4841
  infodbTableNum -= 1;
×
4842
  size_t perfdbTableNum;
×
4843
  getPerfDbMeta(NULL, &perfdbTableNum);
×
4844

4845
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
4846
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
4847
    return (pRes->info.rows > 0) ? pRes : NULL;
×
4848
  } else {
4849
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
4850
    return (pRes->info.rows > 0) ? pRes : NULL;
×
4851
  }
4852
}
4853

4854
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
×
4855
                                       size_t infodbTableNum, size_t perfdbTableNum) {
4856
  int32_t        code = TSDB_CODE_SUCCESS;
×
4857
  int32_t        lino = 0;
×
4858
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
4859
  if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
×
4860
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
4861
    QUERY_CHECK_CODE(code, lino, _end);
×
4862
  } else if (strcmp(pSupp->dbNameFilter, TSDB_PERFORMANCE_SCHEMA_DB) == 0) {
×
4863
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
4864
    QUERY_CHECK_CODE(code, lino, _end);
×
4865
  } else if (strlen(pSupp->dbNameFilter) == 0) {
×
4866
    code = fillTableCountScanDataBlock(pSupp, "", "", infodbTableNum + perfdbTableNum, pRes);
×
4867
    QUERY_CHECK_CODE(code, lino, _end);
×
4868
  }
4869

4870
_end:
×
4871
  if (code != TSDB_CODE_SUCCESS) {
×
4872
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4873
    pTaskInfo->code = code;
×
4874
    T_LONG_JMP(pTaskInfo->env, code);
×
4875
  }
4876
  setOperatorCompleted(pOperator);
×
4877
}
×
4878

4879
static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
4880
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
4881
                                        size_t perfdbTableNum) {
4882
  int32_t        code = TSDB_CODE_SUCCESS;
×
4883
  int32_t        lino = 0;
×
4884
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
4885
  if (pInfo->currGrpIdx == 0) {
×
4886
    uint64_t groupId = 0;
×
4887
    if (pSupp->groupByDbName) {
×
4888
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
×
4889
    } else {
4890
      groupId = calcGroupId("", 0);
×
4891
    }
4892

4893
    pRes->info.id.groupId = groupId;
×
4894
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
4895
    QUERY_CHECK_CODE(code, lino, _end);
×
4896
  } else if (pInfo->currGrpIdx == 1) {
×
4897
    uint64_t groupId = 0;
×
4898
    if (pSupp->groupByDbName) {
×
4899
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
×
4900
    } else {
4901
      groupId = calcGroupId("", 0);
×
4902
    }
4903

4904
    pRes->info.id.groupId = groupId;
×
4905
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
4906
    QUERY_CHECK_CODE(code, lino, _end);
×
4907
  } else {
4908
    setOperatorCompleted(pOperator);
×
4909
  }
4910
  pInfo->currGrpIdx++;
×
4911

4912
_end:
×
4913
  if (code != TSDB_CODE_SUCCESS) {
×
4914
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4915
    pTaskInfo->code = code;
×
4916
    T_LONG_JMP(pTaskInfo->env, code);
×
4917
  }
4918
}
×
4919

4920
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
×
4921
  int32_t                      code = TSDB_CODE_SUCCESS;
×
4922
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
×
4923
  STableCountScanOperatorInfo* pInfo = pOperator->info;
×
4924
  STableCountScanSupp*         pSupp = &pInfo->supp;
×
4925
  SSDataBlock*                 pRes = pInfo->pRes;
×
4926

4927
  blockDataCleanup(pRes);
×
4928
  QRY_PARAM_CHECK(ppRes);
×
4929

4930
  if (pOperator->status == OP_EXEC_DONE) {
×
4931
    return code;
×
4932
  }
4933

4934
  if (pInfo->readHandle.mnd != NULL) {
×
4935
    (*ppRes) = buildSysDbTableCount(pOperator, pInfo);
×
4936
    return code;
×
4937
  }
4938

4939
  code = buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
×
4940
  if (code != TSDB_CODE_SUCCESS) {
×
4941
    qError("%s failed since %s", __func__, tstrerror(code));
×
4942
    pTaskInfo->code = code;
×
4943
    T_LONG_JMP(pTaskInfo->env, code);
×
4944
  }
4945
  if (pRes->info.rows > 0) {
×
4946
    *ppRes = pRes;
×
4947
  }
4948

4949
  return code;
×
4950
}
4951

4952
static int32_t buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
4953
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes) {
4954
  int32_t        code = TSDB_CODE_SUCCESS;
×
4955
  int32_t        lino = 0;
×
4956
  const char*    db = NULL;
×
4957
  int32_t        vgId = 0;
×
4958
  char           dbName[TSDB_DB_NAME_LEN] = {0};
×
4959
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
4960
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
4961

4962
  // get dbname
4963
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, &db, &vgId, NULL, NULL);
×
4964
  SName sn = {0};
×
4965

4966
  code = tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
×
4967
  QUERY_CHECK_CODE(code, lino, _end);
×
4968

4969
  code = tNameGetDbName(&sn, dbName);
×
4970
  QUERY_CHECK_CODE(code, lino, _end);
×
4971

4972
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
4973
    code = buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
×
4974
  } else {
4975
    code = buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
×
4976
  }
4977

4978
_end:
×
4979
  if (code != TSDB_CODE_SUCCESS) {
×
4980
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4981
    pTaskInfo->code = code;
×
4982
  }
4983

4984
  return code;
×
4985
}
4986

4987
static int32_t buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
4988
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
4989
  int32_t        code = TSDB_CODE_SUCCESS;
×
4990
  int32_t        lino = 0;
×
4991
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
4992
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
4993

4994
  if (pSupp->groupByStbName) {
×
4995
    if (pInfo->stbUidList == NULL) {
×
4996
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
×
4997
      QUERY_CHECK_NULL(pInfo->stbUidList, code, lino, _end, terrno);
×
4998
      code = pAPI->metaFn.storeGetTableList(pInfo->readHandle.vnode, TSDB_SUPER_TABLE, pInfo->stbUidList);
×
4999
      QUERY_CHECK_CODE(code, lino, _end);
×
5000
    }
5001

5002
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
×
5003
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
×
5004
      code = buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid, pAPI);
×
5005
      QUERY_CHECK_CODE(code, lino, _end);
×
5006

5007
      pInfo->currGrpIdx++;
×
5008
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
×
5009
      code = buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName, pAPI);
×
5010
      QUERY_CHECK_CODE(code, lino, _end);
×
5011

5012
      pInfo->currGrpIdx++;
×
5013
    } else {
5014
      setOperatorCompleted(pOperator);
×
5015
    }
5016
  } else {
5017
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
×
5018
    pRes->info.id.groupId = groupId;
×
5019

5020
    int64_t dbTableCount = 0;
×
5021
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &dbTableCount, NULL);
×
5022
    code = fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
×
5023
    QUERY_CHECK_CODE(code, lino, _end);
×
5024
    setOperatorCompleted(pOperator);
×
5025
  }
5026

5027
_end:
×
5028
  if (code != TSDB_CODE_SUCCESS) {
×
5029
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5030
    pTaskInfo->code = code;
×
5031
  }
5032
  return code;
×
5033
}
5034

5035
static int32_t buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
5036
                                         STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
5037
  int32_t        code = TSDB_CODE_SUCCESS;
×
5038
  int32_t        lino = 0;
×
5039
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
5040
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
5041

5042
  if (strlen(pSupp->dbNameFilter) != 0) {
×
5043
    if (strlen(pSupp->stbNameFilter) != 0) {
×
5044
      uint64_t uid = 0;
×
5045
      code = pAPI->metaFn.getTableUidByName(pInfo->readHandle.vnode, pSupp->stbNameFilter, &uid);
×
5046
      QUERY_CHECK_CODE(code, lino, _end);
×
5047

5048
      int64_t numOfChildTables = 0;
×
5049
      code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, uid, &numOfChildTables, NULL, NULL);
×
5050
      QUERY_CHECK_CODE(code, lino, _end);
×
5051

5052
      code = fillTableCountScanDataBlock(pSupp, dbName, pSupp->stbNameFilter, numOfChildTables, pRes);
×
5053
      QUERY_CHECK_CODE(code, lino, _end);
×
5054
    } else {
5055
      int64_t tbNumVnode = 0;
×
5056
      pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
5057
      code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
5058
      QUERY_CHECK_CODE(code, lino, _end);
×
5059
    }
5060
  } else {
5061
    int64_t tbNumVnode = 0;
×
5062
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
5063
    code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
5064
    QUERY_CHECK_CODE(code, lino, _end);
×
5065
  }
5066

5067
_end:
×
5068
  if (code != TSDB_CODE_SUCCESS) {
×
5069
    pTaskInfo->code = code;
×
5070
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5071
  }
5072

5073
  setOperatorCompleted(pOperator);
×
5074
  return code;
×
5075
}
5076

5077
static int32_t buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
5078
                                              SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI) {
5079
  int32_t code = TSDB_CODE_SUCCESS;
×
5080
  int32_t lino = 0;
×
5081
  char    fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
5082
  if (pSupp->groupByDbName) {
×
5083
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
×
5084
  }
5085

5086
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
5087
  pRes->info.id.groupId = groupId;
×
5088

5089
  int64_t numOfTables = 0;
×
5090
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, NULL, &numOfTables);
×
5091

5092
  if (numOfTables != 0) {
×
5093
    code = fillTableCountScanDataBlock(pSupp, dbName, "", numOfTables, pRes);
×
5094
  }
5095

5096
_end:
×
5097
  if (code != TSDB_CODE_SUCCESS) {
×
5098
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5099
  }
5100
  return code;
×
5101
}
5102

5103
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
5104
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
5105
  int32_t code = TSDB_CODE_SUCCESS;
×
5106
  int32_t lino = 0;
×
5107
  char    stbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5108
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
×
5109
  QUERY_CHECK_CODE(code, lino, _end);
×
5110

5111
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
5112
  if (pSupp->groupByDbName) {
×
5113
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
×
5114
  } else {
5115
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
×
5116
  }
5117

5118
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
5119
  pRes->info.id.groupId = groupId;
×
5120

5121
  int64_t ctbNum = 0;
×
5122
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL, NULL);
×
5123
  QUERY_CHECK_CODE(code, lino, _end);
×
5124
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
×
5125
  QUERY_CHECK_CODE(code, lino, _end);
×
5126

5127
_end:
×
5128
  if (code != TSDB_CODE_SUCCESS) {
×
5129
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5130
  }
5131
  return code;
×
5132
}
5133

5134
static void destoryTableCountScanOperator(void* param) {
×
5135
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
×
5136
  blockDataDestroy(pTableCountScanInfo->pRes);
×
5137

5138
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
×
5139
  taosMemoryFreeClear(param);
×
5140
}
×
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