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

taosdata / TDengine / #4683

21 Aug 2025 12:03PM UTC coverage: 60.097% (+0.2%) from 59.918%
#4683

push

travis-ci

web-flow
jdbc release 3.7.3 (#32682)

* chore(jdbc): update jdbc version

* docs(jdbc): release 3.7.3

138078 of 292193 branches covered (47.26%)

Branch coverage included in aggregate %.

208479 of 284466 relevant lines covered (73.29%)

17837633.69 hits per line

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

55.35
/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) {
10,691,884✔
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;
10,691,884✔
85
#endif
86
}
87

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

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

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

104
  if (order == TSDB_ORDER_ASC) {
568,063✔
105
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
412,172✔
106
    if (w.ekey < pBlockInfo->window.skey) {
412,116!
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) {
412,116✔
112
      *overlap = true;
347,727✔
113
      return code;
347,727✔
114
    }
115

116
    while (1) {
117
      getNextTimeWindow(pInterval, &w, order);
64,388✔
118
      if (w.skey > pBlockInfo->window.ekey) {
64,297✔
119
        break;
50,610✔
120
      }
121

122
      if (w.ekey <= pBlockInfo->window.ekey) {
13,687!
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) {
13,687!
127
        *overlap = true;
13,688✔
128
        return code;
13,688✔
129
      }
130
    }
131
  } else {
132
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.ekey);
155,891✔
133
    if (w.skey > pBlockInfo->window.ekey) {
156,270!
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) {
156,270✔
139
      *overlap = true;
109,835✔
140
      return code;
109,835✔
141
    }
142

143
    while (1) {
144
      getNextTimeWindow(pInterval, &w, order);
46,435✔
145
      if (w.ekey < pBlockInfo->window.skey) {
46,450✔
146
        break;
38,237✔
147
      }
148

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

160
  *overlap = false;
88,847✔
161
  return code;
88,847✔
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) {
2,356,126✔
166
  if (pOperator->operatorType != QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
2,356,126!
167
    return NULL;
×
168
  }
169

170
  int64_t buf[2] = {0};
2,356,126✔
171
  SET_RES_WINDOW_KEY((char*)buf, &groupId, sizeof(groupId), groupId);
2,356,126✔
172

173
  STableScanInfo* pTableScanInfo = pOperator->info;
2,356,126✔
174

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

178
  if (p1 == NULL) {
2,356,113✔
179
    return NULL;
374,472✔
180
  }
181

182
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
1,981,641✔
183
  if (NULL == *pPage) {
1,981,590!
184
    return NULL;
×
185
  }
186

187
  return (SResultRow*)((char*)(*pPage) + p1->offset);
1,981,590✔
188
}
189

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

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

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

213
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
10,795,357✔
214
    return TSDB_CODE_SUCCESS;
8,440,748✔
215
  }
216

217
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
2,354,609✔
218

219
  SFilePage*  pPage = NULL;
2,354,609✔
220
  SResultRow* pRow = getTableGroupOutputBuf(pOperator, pBlockInfo->id.groupId, &pPage);
2,354,609✔
221

222
  if (pRow == NULL) {
2,356,052✔
223
    return TSDB_CODE_SUCCESS;
374,487✔
224
  }
225

226
  bool notLoadBlock = true;
1,981,565✔
227
  for (int32_t i = 0; i < pSup1->numOfExprs; ++i) {
2,107,131✔
228
    int32_t functionId = pSup1->pCtx[i].functionId;
2,091,901✔
229

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

232
    EFuncDataRequired reqStatus = fmFuncDynDataRequired(functionId, pEntry, pBlockInfo);
2,091,913✔
233
    if (reqStatus != FUNC_DATA_REQUIRED_NOT_LOAD) {
2,091,936✔
234
      notLoadBlock = false;
1,966,370✔
235
      break;
1,966,370✔
236
    }
237
  }
238

239
  // release buffer pages
240
  releaseBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, pPage);
1,981,600✔
241

242
  if (notLoadBlock) {
1,981,510✔
243
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
15,189✔
244
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
15,189✔
245
  }
246

247
  return code;
1,981,518✔
248
}
249

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

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

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

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

275
  if (!allColumnsHaveAgg || hasNullSMA) {
6,099,706✔
276
    *pLoad = false;
3,659,253✔
277
  } else {
278
    *pLoad = true;
2,440,453✔
279
  }
280

281
  return code;
6,099,706✔
282
}
283

284
static int32_t doSetTagColumnData(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
15,670,350✔
285
                                  int32_t rows) {
286
  int32_t    code = 0;
15,670,350✔
287
  SExprSupp* pSup = &pTableScanInfo->pseudoSup;
15,670,350✔
288
  if (pSup->numOfExprs > 0) {
15,670,350✔
289
    code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
4,008,297✔
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) {
4,009,059!
293
      if (pTaskInfo->streamInfo.pState) blockDataCleanup(pBlock);
×
294
      code = 0;
×
295
    }
296
  }
297

298
  return code;
15,672,540✔
299
}
300

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

305
  if (pLimitInfo->remainOffset > 0) {
37,434,388✔
306
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
60,508✔
307
      pLimitInfo->remainOffset -= pBlock->info.rows;
15,639✔
308
      blockDataEmpty(pBlock);
15,639✔
309
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
15,639✔
310
      return false;
15,639✔
311
    } else {
312
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
44,869✔
313
      if (code != TSDB_CODE_SUCCESS) {
44,869!
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;
44,869✔
319
    }
320
  }
321

322
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
37,418,749✔
323
    // limit the output rows
324
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
351,290✔
325
    blockDataKeepFirstNRows(pBlock, keep);
351,290✔
326

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

332
  pLimitInfo->numOfOutputRows += pBlock->info.rows;
37,067,459✔
333
  return false;
37,067,459✔
334
}
335

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

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

348
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
15,267,146✔
349

350
  pCost->totalBlocks += 1;
15,267,146✔
351
  pCost->totalRows += pBlock->info.rows;
15,267,146✔
352
  *status = pTableScanInfo->dataBlockLoadFlag;
15,267,146✔
353

354
  if (pOperator->exprSupp.pFilterInfo != NULL) {
15,267,146✔
355
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
3,430,190✔
356
  } else {
357
    bool overlap = false;
11,836,956✔
358
    int  ret =
359
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
11,836,956✔
360
    if (ret != TSDB_CODE_SUCCESS) {
11,837,127!
361
      return ret;
×
362
    }
363
    if (overlap) {
11,837,127✔
364
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
479,344✔
365
    }
366
  }
367

368
  SDataBlockInfo* pBlockInfo = &pBlock->info;
15,267,317✔
369
  taosMemoryFreeClear(pBlock->pBlockAgg);
15,267,317!
370

371
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
15,267,317!
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) {
15,267,317✔
379
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
2,054,489✔
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);
2,054,615✔
383
    pCost->skipBlocks += 1;
2,054,638✔
384
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
2,054,638✔
385
    return code;
2,054,640✔
386
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
13,212,828✔
387
    pCost->loadBlockStatis += 1;
2,669,833✔
388
    loadSMA = true;  // mark the operation of load sma;
2,669,833✔
389
    bool success = true;
2,669,833✔
390
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
2,669,833✔
391
    if (code) {
2,669,823!
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
2,669,786✔
398
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
2,411,361✔
399
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
400
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
2,411,364✔
401
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
2,411,364✔
402
      return code;
2,412,393✔
403
    } else {
404
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
258,425✔
405
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
258,800✔
406
    }
407
  }
408

409
  if (*status != FUNC_DATA_REQUIRED_DATA_LOAD) {
10,801,795!
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)) {
10,801,795!
417
    bool success = true;
3,429,971✔
418
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
3,429,971✔
419
    if (code) {
3,429,976!
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) {
3,429,645✔
426
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
29,087✔
427
      bool   keep = false;
29,089✔
428
      code = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows, &keep);
29,089✔
429
      if (code) {
29,095!
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) {
29,095✔
436
        qDebug("%s data block filter out by block SMA, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
3,265!
437
               GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
438
        pCost->filterOutBlocks += 1;
3,265✔
439
        (*status) = FUNC_DATA_REQUIRED_FILTEROUT;
3,265✔
440
        taosMemoryFreeClear(pBlock->pBlockAgg);
3,265!
441

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

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

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

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

464
    STableScanInfo* p1 = pOperator->info;
15,189✔
465
    if (taosHashGetSize(p1->pIgnoreTables) == taosArrayGetSize(p1->base.pTableListInfo->pTableList)) {
15,189✔
466
      *status = FUNC_DATA_REQUIRED_ALL_FILTEROUT;
3,495✔
467
    } else {
468
      *status = FUNC_DATA_REQUIRED_FILTEROUT;
11,694✔
469
    }
470
    return TSDB_CODE_SUCCESS;
15,189✔
471
  }
472

473
  pCost->totalCheckedRows += pBlock->info.rows;
10,781,060✔
474
  pCost->loadBlocks += 1;
10,781,060✔
475

476
  SSDataBlock* p = NULL;
10,781,060✔
477
  code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pTableScanInfo->dataReader, &p, NULL);
10,781,060✔
478
  if (p == NULL || code != TSDB_CODE_SUCCESS || p != pBlock) {
10,780,165!
479
    return code;
×
480
  }
481

482
  if ((pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) &&
10,781,235✔
483
      ((STableScanInfo*)pOperator->info)->ignoreTag) {
6,208,406!
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);
10,781,235✔
488
    if (code) {
10,782,592!
489
      return code;
×
490
    }
491
  }
492

493
  // restore the previous value
494
  pCost->totalRows -= pBlock->info.rows;
10,782,592✔
495

496
  if (pOperator->exprSupp.pFilterInfo != NULL) {
10,782,592✔
497
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo);
3,416,750✔
498
    QUERY_CHECK_CODE(code, lino, _end);
3,416,735✔
499

500
    int64_t st = taosGetTimestampUs();
3,416,532✔
501
    double  el = (taosGetTimestampUs() - st) / 1000.0;
3,416,754✔
502
    pTableScanInfo->readRecorder.filterTime += el;
3,416,754✔
503

504
    if (pBlock->info.rows == 0) {
3,416,754✔
505
      pCost->filterOutBlocks += 1;
111,489✔
506
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
111,489✔
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);
3,305,265✔
510
    }
511
  }
512

513
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
10,782,733✔
514
  if (limitReached) {  // set operator flag is done
10,781,034✔
515
    setOperatorCompleted(pOperator);
104,453✔
516
  }
517

518
  pCost->totalRows += pBlock->info.rows;
10,782,632✔
519

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

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

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

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

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

546
  STableCachedVal* pVal = param;
1,956,010✔
547
  taosMemoryFree((void*)pVal->pName);
1,956,010!
548
  taosMemoryFree(pVal->pTags);
1,957,430!
549
  taosMemoryFree(pVal);
1,957,401!
550
}
551

552
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
1,957,070✔
553
  int32_t          code = TSDB_CODE_SUCCESS;
1,957,070✔
554
  int32_t          lino = 0;
1,957,070✔
555
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
1,957,070!
556
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
1,957,372!
557

558
  pVal->pTags = NULL;
1,957,372✔
559
  pVal->pName = taosStrdup(pMetaReader->me.name);
1,957,372!
560
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
1,957,075!
561

562
  // only child table has tag value
563
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
1,957,075✔
564
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
1,956,408✔
565
    pVal->pTags = taosMemoryMalloc(pTag->len);
1,956,408✔
566
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
1,956,081!
567
    memcpy(pVal->pTags, pTag, pTag->len);
1,956,081✔
568
  }
569

570
  (*ppResVal) = pVal;
1,956,748✔
571

572
_end:
1,956,748✔
573
  if (code != TSDB_CODE_SUCCESS) {
1,956,748!
574
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
575
    freeTableCachedVal(pVal);
×
576
  }
577
  return code;
1,956,625✔
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,955,969✔
582
  (void)key;
583
  (void)keyLen;
584
  (void)ud;
585
  freeTableCachedVal(value);
1,955,969✔
586
}
1,957,404✔
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,
4,115,124✔
603
                               int32_t rows, SExecTaskInfo* pTask, STableMetaCacheInfo* pCache) {
604
  int32_t          code = TSDB_CODE_SUCCESS;
4,115,124✔
605
  int32_t          lino = 0;
4,115,124✔
606
  bool             freeReader = false;
4,115,124✔
607
  LRUHandle*       h = NULL;
4,115,124✔
608
  STableCachedVal  val = {0};
4,115,124✔
609
  SMetaReader      mr = {0};
4,115,124✔
610
  const char*      idStr = pTask->id.str;
4,115,124✔
611
  int32_t          insertRet = TAOS_LRU_STATUS_OK;
4,115,124✔
612
  STableCachedVal* pVal = NULL;
4,115,124✔
613

614
  // currently only the tbname pseudo column
615
  if (numOfExpr <= 0) {
4,115,124✔
616
    return TSDB_CODE_SUCCESS;
3,847✔
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;
4,111,277✔
623
  pBlock->info.rows = rows;
4,111,277✔
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) {
4,111,277✔
630
    pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
1,045✔
631
    code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
1,044✔
632
    if (code != TSDB_CODE_SUCCESS) {
1,043!
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);
1,043✔
648

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

652
    freeReader = true;
1,045✔
653
  } else {
654
    pCache->metaFetch += 1;
4,110,232✔
655

656
    h = taosLRUCacheLookup(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(pBlock->info.id.uid));
4,110,232✔
657
    if (h == NULL) {
4,111,023✔
658
      pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
1,957,544✔
659
      freeReader = true;
1,957,513✔
660
      code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
1,957,513✔
661
      if (code != TSDB_CODE_SUCCESS) {
1,956,410!
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,956,410✔
676

677
      code = createTableCacheVal(&mr, &pVal);
1,957,382✔
678
      QUERY_CHECK_CODE(code, lino, _end);
1,956,676!
679

680
      val = *pVal;
1,956,676✔
681
    } else {
682
      pCache->cacheHit += 1;
2,153,479✔
683
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
2,153,479✔
684
      val = *pValTmp;
2,153,413✔
685

686
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
2,153,413✔
687
      qTrace("release LRU cache, res %d", bRes);
2,153,678!
688
    }
689

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

694
  for (int32_t j = 0; j < numOfExpr; ++j) {
16,939,152✔
695
    const SExprInfo* pExpr1 = &pExpr[j];
12,826,840✔
696
    int32_t          dstSlotId = pExpr1->base.resSchema.slotId;
12,826,840✔
697

698
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
12,826,840✔
699
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
12,824,814✔
700

701
    int32_t functionId = pExpr1->pExpr->_function.functionId;
12,823,704✔
702

703
    // this is to handle the tbname
704
    if (fmIsScanPseudoColumnFunc(functionId)) {
12,823,704✔
705
      int32_t fType = pExpr1->pExpr->_function.functionType;
2,188,468✔
706
      if (fType == FUNCTION_TYPE_TBNAME) {
2,188,468!
707
        code = setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
2,188,468✔
708
        QUERY_CHECK_CODE(code, lino, _end);
2,188,440!
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};
10,636,552✔
718
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
10,636,552✔
719
      const char* p = pHandle->api.metaFn.extractTagVal(val.pTags, pColInfoData->info.type, &tagVal);
10,636,552✔
720

721
      char* data = NULL;
10,638,621✔
722
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
10,638,621✔
723
        data = tTagValToData((const STagVal*)p, false);
10,597,380✔
724
      } else {
725
        data = (char*)p;
41,241✔
726
      }
727

728
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
10,638,317✔
729
      if (isNullVal) {
10,638,317✔
730
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
38,400✔
731
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
10,599,917✔
732
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, false);
10,596,933✔
733
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
10,596,884!
734
          char* tmp = taosMemoryCalloc(1, varDataLen(data) + 1);
2,641,174!
735
          if (tmp != NULL) {
2,642,220!
736
            memcpy(tmp, varDataVal(data), varDataLen(data));
2,642,220✔
737
            qDebug("get tag value:%s, cid:%d, table name:%s, uid%" PRId64, tmp, tagVal.cid, val.pName,
2,642,220✔
738
                   pBlock->info.id.uid);
739
            taosMemoryFree(tmp);
2,642,222✔
740
          }
741
          taosMemoryFree(data);
2,642,209✔
742
        }
743
        QUERY_CHECK_CODE(code, lino, _end);
10,596,861!
744
      } else {  // todo opt for json tag
745
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
9,857✔
746
          code = colDataSetVal(pColInfoData, i, data, false);
6,873✔
747
          QUERY_CHECK_CODE(code, lino, _end);
6,873!
748
        }
749
      }
750
    }
751
  }
752

753
  // restore the rows
754
  pBlock->info.rows = backupRows;
4,112,312✔
755

756
_end:
4,112,312✔
757

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

767
  if (freeReader) {
4,112,302✔
768
    pHandle->api.metaReaderFn.clearReader(&mr);
1,958,663✔
769
  }
770
  if (code != TSDB_CODE_SUCCESS) {
4,112,283!
771
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
772
  }
773
  return code;
4,112,348✔
774
}
775

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

784
  size_t len = TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE;
2,192,896✔
785
  char   buf[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
2,192,896✔
786
  STR_TO_VARSTR(buf, name)
2,192,896✔
787

788
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
2,192,896✔
789

790
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
2,192,984✔
791
  QUERY_CHECK_CODE(code, lino, _end);
2,192,908!
792

793
  code = colDataSetVal(&infoData, 0, buf, false);
2,192,908✔
794
  QUERY_CHECK_CODE(code, lino, _end);
2,192,868!
795

796
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
2,192,868✔
797
  SScalarParam param = {.columnData = pColInfoData};
2,192,868✔
798

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

806
  colDataDestroy(&infoData);
2,192,893✔
807

808
_end:
2,192,863✔
809
  if (code != TSDB_CODE_SUCCESS) {
2,192,863!
810
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
811
  }
812
  return code;
2,192,867✔
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) {
2,615,045✔
882
  int32_t code = TSDB_CODE_SUCCESS;
2,615,045✔
883
  int32_t lino = 0;
2,615,045✔
884
  code = tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, pKeyInfo, size);
2,615,045✔
885
  QUERY_CHECK_CODE(code, lino, _end);
2,612,847!
886

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

892
  if (!pInfo->needCountEmptyTable) {
2,612,847✔
893
    pInfo->countState = TABLE_COUNT_STATE_END;
2,399,212✔
894
  } else {
895
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
213,635✔
896
  }
897

898
_end:
2,612,847✔
899
  if (code != TSDB_CODE_SUCCESS) {
2,612,847!
900
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
901
  }
902
  return code;
2,612,712✔
903
}
904

905
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
10,987,060✔
906
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
10,987,060✔
907
    return;
9,387,603✔
908
  }
909
  if (pInfo->base.pTableListInfo->groupOffset) {
1,599,457✔
910
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
751✔
911
  } else {
912
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
1,598,706✔
913
    if (code != TSDB_CODE_SUCCESS) {
1,599,477✔
914
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
1,024,796✔
915
    }
916
  }
917
}
918

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

927
  // only one row: set all col data to null & hasNull
928
  int32_t col_num = blockDataGetNumOfCols(pBlock);
423,861✔
929
  for (int32_t i = 0; i < col_num; ++i) {
1,661,463✔
930
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
1,237,796✔
931
    colDataSetNULL(pColInfoData, 0);
932
  }
933

934
  // set tag/tbname
935
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
423,667✔
936

937
  return pBlock;
423,957✔
938
}
939

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

945
  pOperator->resultInfo.totalRows++;
409,557✔
946
  return pBlock;
409,557✔
947
}
948

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

959
  QRY_PARAM_CHECK(ppRes);
13,088,799!
960
  pBlock->info.dataLoad = false;
13,088,799✔
961

962
  while (true) {
110,132✔
963
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
13,198,931✔
964
    if (code != TSDB_CODE_SUCCESS) {
13,199,079!
965
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
×
966
      QUERY_CHECK_CODE(code, lino, _end);
810!
967
    }
968

969
    if (!hasNext) {
13,200,176✔
970
      break;
2,510,701✔
971
    }
972

973
    if (isTaskKilled(pTaskInfo)) {
10,694,370✔
974
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
202✔
975
      code = pTaskInfo->code;
805✔
976
      goto _end;
805✔
977
    }
978

979
    if (pOperator->status == OP_EXEC_DONE) {
10,694,689✔
980
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
1,400✔
981
      break;
1,400✔
982
    }
983

984
    // process this data block based on the probabilities
985
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
10,693,289✔
986
    if (!processThisBlock) {
10,692,075!
987
      continue;
110,132✔
988
    }
989

990
    if (pBlock->info.id.uid) {
10,692,075!
991
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
10,692,977✔
992
    }
993

994
    uint32_t status = 0;
10,692,897✔
995
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
10,692,897✔
996
    QUERY_CHECK_CODE(code, lino, _end);
10,693,061✔
997

998
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
10,693,060✔
999
      break;
3,495✔
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) {
10,689,565✔
1004
      continue;
110,132✔
1005
    }
1006

1007
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
10,579,433✔
1008
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
10,579,256✔
1009

1010
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
10,579,256✔
1011
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
10,579,256✔
1012

1013
    (*ppRes) = pBlock;
10,579,256✔
1014
    return code;
10,579,256✔
1015
  }
1016

1017
_end:
2,511,511✔
1018
  if (code != TSDB_CODE_SUCCESS) {
2,511,511✔
1019
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
810!
1020
    pTaskInfo->code = code;
811✔
1021
    T_LONG_JMP(pTaskInfo->env, code);
811!
1022
  }
1023
  return code;
2,510,701✔
1024
}
1025

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

1033
  QRY_PARAM_CHECK(pBlock);
13,589,513!
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) {
13,589,513!
1037
    return code;
91,351✔
1038
  }
1039

1040
  // do the ascending order traverse in the first place.
1041
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
15,298,924✔
1042
    SSDataBlock* p = NULL;
11,775,009✔
1043
    code = doTableScanImplNext(pOperator, &p);
11,775,009✔
1044
    QUERY_CHECK_CODE(code, lino, _end);
11,773,788!
1045

1046
    if (p != NULL) {
11,773,788✔
1047
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
9,973,782✔
1048
      *pBlock = p;
9,974,208✔
1049
      return code;
9,974,208✔
1050
    }
1051

1052
    pTableScanInfo->scanTimes += 1;
1,800,006✔
1053
    taosHashClear(pTableScanInfo->pIgnoreTables);
1,800,006✔
1054

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

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

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

1076
    while (pTableScanInfo->scanTimes < total) {
2,023,779✔
1077
      SSDataBlock* p = NULL;
1,313,890✔
1078
      code = doTableScanImplNext(pOperator, &p);
1,313,890✔
1079
      QUERY_CHECK_CODE(code, lino, _end);
1,314,195!
1080

1081
      if (p != NULL) {
1,314,195✔
1082
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
604,396✔
1083
        *pBlock = p;
604,423✔
1084
        return code;
604,423✔
1085
      }
1086

1087
      pTableScanInfo->scanTimes += 1;
709,799✔
1088
      taosHashClear(pTableScanInfo->pIgnoreTables);
709,799✔
1089

1090
      if (pTableScanInfo->scanTimes < total) {
709,808✔
1091
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
4✔
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);
×
1097
      }
1098
    }
1099
  }
1100

1101
  if (pTableScanInfo->countState < TABLE_COUNT_STATE_END) {
2,919,851✔
1102
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
620,366✔
1103
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
620,366✔
1104
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
363✔
1105
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
46✔
1106
        STableKeyInfo* pStart =
1107
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
46✔
1108

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

1113
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
46✔
1114
        return code;
46✔
1115
      }
1116
    } else {  // group by tag + no sort
1117
      int32_t numOfTables = 0;
620,003✔
1118
      code = tableListGetSize(pTableListInfo, &numOfTables);
620,003✔
1119
      QUERY_CHECK_CODE(code, lino, _end);
619,956!
1120

1121
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
619,956✔
1122
        // get empty group, mark processed & rm from hash
1123
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
618,855✔
1124
        if (pIte != NULL) {
618,979✔
1125
          size_t        keySize = 0;
409,407✔
1126
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
409,407✔
1127
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
409,447✔
1128
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
409,447✔
1129
          markGroupProcessed(pTableScanInfo, *pGroupId);
409,474✔
1130
          *pBlock = getBlockForEmptyTable(pOperator, &info);
409,451✔
1131

1132
          return code;
409,512✔
1133
        }
1134
      }
1135
    }
1136
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
210,990✔
1137
  }
1138

1139
_end:
2,299,485✔
1140
  if (code != TSDB_CODE_SUCCESS) {
2,510,475!
1141
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1142
    pTaskInfo->code = code;
×
1143
  }
1144

1145
  return code;
2,510,475✔
1146
}
1147

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

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

1176
  STableKeyInfo info = {.groupId = 0};
1,668✔
1177
  int32_t       tableIdx = 0;
1,668✔
1178
  for (int32_t i = 0; i < num; ++i) {
71,494✔
1179
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
69,836✔
1180
    if (!pUid) {
69,812!
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))) {
69,812✔
1186
      if (TSDB_CODE_DUP_KEY == terrno) {
65,303✔
1187
        continue;
65,281✔
1188
      }
1189
      return terrno;
1✔
1190
    }
1191

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

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

1202
  return code;
1,658✔
1203
}
1204

1205
static int32_t doInitReader(STableScanInfo* pInfo, SExecTaskInfo* pTaskInfo, SStorageAPI* pAPI, int32_t* pNum,
2,615,447✔
1206
                            STableKeyInfo** pList) {
1207
  const char* idStr = GET_TASKID(pTaskInfo);
2,615,447✔
1208
  int32_t     code = initNextGroupScan(pInfo, pList, pNum);
2,615,447✔
1209
  if (code) {
2,612,132!
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) {
2,612,353!
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,
2,612,353✔
1220
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
2,612,353✔
1221
  if (code) {
2,619,360!
1222
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
×
1223
  }
1224

1225
  return code;
2,619,127✔
1226
}
1227

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

1232
  if (node1->orgColId < node2->orgColId) {
×
1233
    return -1;
×
1234
  }
1235

1236
  return node1->orgColId > node2->orgColId;
×
1237
}
1238

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

1255
  cleanupQueryTableDataCond(&pInfo->base.cond);
×
1256
  if (pAPI->tsdReader.tsdReaderClose) {
×
1257
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
×
1258
  }
1259

1260
  pAPI->metaReaderFn.initReader(&orgTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
×
1261
  code = pAPI->metaReaderFn.getTableEntryByName(&orgTable, strstr(pParam->pOrgTbInfo->tbName, ".") + 1);
×
1262
  pAPI->metaReaderFn.readerReleaseLock(&orgTable);
×
1263
  QUERY_CHECK_CODE(code, lino, _return);
×
1264
  switch (orgTable.me.type) {
×
1265
    case TSDB_CHILD_TABLE:
×
1266
      pAPI->metaReaderFn.initReader(&superTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
×
1267
      code = pAPI->metaReaderFn.getTableEntryByUid(&superTable, orgTable.me.ctbEntry.suid);
×
1268
      pAPI->metaReaderFn.readerReleaseLock(&superTable);
×
1269
      QUERY_CHECK_CODE(code, lino, _return);
×
1270
      schema = &superTable.me.stbEntry.schemaRow;
×
1271
      break;
×
1272
    case TSDB_NORMAL_TABLE:
×
1273
      schema = &orgTable.me.ntbEntry.schemaRow;
×
1274
      break;
×
1275
    default:
×
1276
      qError("invalid table type:%d", orgTable.me.type);
×
1277
      return TSDB_CODE_INVALID_PARA;
×
1278
      break;
1279
  }
1280

1281
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
×
1282
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
×
1283
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(int32_t));
×
1284
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
×
1285

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

1290
  for (int32_t i = 0; i < taosArrayGetSize(pParam->pOrgTbInfo->colMap); ++i) {
×
1291
    SColIdNameKV* kv = taosArrayGet(pParam->pOrgTbInfo->colMap, i);
×
1292
    for (int32_t j = 0; j < schema->nCols; j++) {
×
1293
      if (strcmp(kv->colName, schema->pSchema[j].name) == 0) {
×
1294
        SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(j + 1)};
×
1295
        QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
×
1296
        break;
×
1297
      }
1298
    }
1299
  }
1300

1301
  for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
×
1302
    SColIdPair* pPair = (SColIdPair*)taosArrayGet(pColArray, i);
×
1303
    for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
×
1304
      SColMatchItem* pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
×
1305
      if (pItem->colId == pPair->vtbColId) {
×
1306
        SColIdPair tmpPair = {.orgColId = pPair->orgColId, .vtbColId = pItem->dstSlotId};
×
1307
        QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &tmpPair), code, lino, _return, terrno);
×
1308
        break;
×
1309
      }
1310
    }
1311
  }
1312

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

1332
  pListInfo->oneTableForEachGroup = true;
×
1333
  taosHashClear(pListInfo->map);
×
1334
  taosArrayClear(pListInfo->pTableList);
×
1335

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

1344
  pOperator->status = OP_OPENED;
×
1345

1346
  taosRLockLatch(&pTaskInfo->lock);
×
1347
  code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
×
1348
  taosRUnLockLatch(&pTaskInfo->lock);
×
1349
  QUERY_CHECK_CODE(code, lino, _return);
×
1350

1351
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
×
1352
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1353
  }
1354

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

1367
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
2,585,419✔
1368
  int32_t         code = TSDB_CODE_SUCCESS;
2,585,419✔
1369
  int32_t         lino = 0;
2,585,419✔
1370
  STableScanInfo* pInfo = pOperator->info;
2,585,419✔
1371
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
2,585,419✔
1372
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
2,585,419✔
1373
  int32_t         numOfTables = 0;
2,585,419✔
1374

1375
  QRY_PARAM_CHECK(pResult);
2,585,419!
1376

1377
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
2,585,419✔
1378
  QUERY_CHECK_CODE(code, lino, _end);
2,585,492!
1379

1380
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
2,585,492✔
1381
    setOperatorCompleted(pOperator);
2,584,704✔
1382
    if (pOperator->dynamicTask) {
2,584,841✔
1383
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
1,658✔
1384
      taosHashClear(pInfo->base.pTableListInfo->map);
1,657✔
1385
    }
1386
    return code;
2,584,876✔
1387
  }
1388

1389
  // reset value for the next group data output
1390
  pOperator->status = OP_OPENED;
680✔
1391
  resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
680✔
1392

1393
  int32_t        num = 0;
737✔
1394
  STableKeyInfo* pList = NULL;
737✔
1395
  code = initNextGroupScan(pInfo, &pList, &num);
737✔
1396
  QUERY_CHECK_CODE(code, lino, _end);
737!
1397

1398
  code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
737✔
1399
  QUERY_CHECK_CODE(code, lino, _end);
740!
1400

1401
  code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
740✔
1402
  QUERY_CHECK_CODE(code, lino, _end);
739!
1403
  pInfo->scanTimes = 0;
739✔
1404

1405
  code = doGroupedTableScan(pOperator, pResult);
739✔
1406
  QUERY_CHECK_CODE(code, lino, _end);
740!
1407

1408
  if (*pResult != NULL) {
740✔
1409
    if (pOperator->dynamicTask) {
526✔
1410
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
24✔
1411
    }
1412
    return code;
526✔
1413
  }
1414

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

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

1434
  QRY_PARAM_CHECK(pResBlock);
13,545,953!
1435

1436
  if (pInfo->currentGroupId == -1) {
13,545,953✔
1437
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
2,616,910!
1438
      setOperatorCompleted(pOperator);
×
1439
      return code;
×
1440
    }
1441

1442
    taosRLockLatch(&pTaskInfo->lock);
2,616,453✔
1443
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
2,619,670✔
1444
    taosRUnLockLatch(&pTaskInfo->lock);
2,619,169✔
1445
    QUERY_CHECK_CODE(code, lino, _end);
2,620,202!
1446

1447
    if (pInfo->filesetDelimited) {
2,620,202✔
1448
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
1,640✔
1449
    }
1450

1451
    if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
2,620,202!
1452
      pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1453
    }
1454
  }
1455

1456
  code = doGroupedTableScan(pOperator, &pResult);
13,549,245✔
1457
  QUERY_CHECK_CODE(code, lino, _end);
13,552,534!
1458

1459
  if (pResult != NULL) {
13,552,534✔
1460
    if (pOperator->dynamicTask) {
10,968,476✔
1461
      pResult->info.id.groupId = pResult->info.id.uid;
4,208✔
1462
    }
1463

1464
    *pResBlock = pResult;
10,968,476✔
1465
    return code;
10,968,476✔
1466
  }
1467

1468
  while (true) {
1469
    code = startNextGroupScan(pOperator, &pResult);
2,584,272✔
1470
    QUERY_CHECK_CODE(code, lino, _end);
2,585,521!
1471

1472
    if (pResult || pOperator->status == OP_EXEC_DONE) {
2,585,521✔
1473
      *pResBlock = pResult;
2,585,307✔
1474
      return code;
2,585,307✔
1475
    }
1476
  }
1477

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

1484
  return code;
×
1485
}
1486

1487
static bool isEmptyQueryTimeWindow(STimeWindow* pWindow) {
×
1488
  return (pWindow == NULL) || (pWindow->skey > pWindow->ekey);
×
1489
}
1490

1491
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
13,565,537✔
1492
  int32_t         code = TSDB_CODE_SUCCESS;
13,565,537✔
1493
  int32_t         lino = 0;
13,565,537✔
1494
  STableScanInfo* pInfo = pOperator->info;
13,565,537✔
1495
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
13,565,537✔
1496
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
13,565,537✔
1497
  QRY_PARAM_CHECK(ppRes);
13,565,537!
1498
  qTrace("%s call", __FUNCTION__);
13,565,537!
1499
  if (pOperator->pOperatorGetParam) {
13,565,155✔
1500
    pOperator->dynamicTask = true;
1,668✔
1501
    if (isDynVtbScan(pOperator)) {
1,668!
1502
      code = createVTableScanInfoFromParam(pOperator);
×
1503

1504
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
×
1505
      pOperator->pOperatorGetParam = NULL;
×
1506
      QUERY_CHECK_CODE(code, lino, _end);
×
1507

1508
      SSDataBlock* result = NULL;
×
1509
      if (isEmptyQueryTimeWindow(&pInfo->base.cond.twindows) && pInfo->base.cond.type == TIMEWINDOW_RANGE_CONTAINED) {
×
1510
        (*ppRes) = result;
×
1511
        return code;
×
1512
      }
1513
      while (true) {
1514
        code = startNextGroupScan(pOperator, &result);
×
1515
        QUERY_CHECK_CODE(code, lino, _end);
×
1516

1517
        if (result || pOperator->status == OP_EXEC_DONE) {
×
1518
          SSDataBlock* res = NULL;
×
1519
          if (result) {
×
1520
            code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, &res);
×
1521
            QUERY_CHECK_CODE(code, lino, _end);
×
1522
            pInfo->pResBlock = res;
×
1523
            blockDataDestroy(result);
×
1524
          }
1525
          (*ppRes) = res;
×
1526
          return code;
×
1527
        }
1528
      }
1529
    } else {
1530
      code = createTableListInfoFromParam(pOperator);
1,668✔
1531
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
1,668✔
1532
      pOperator->pOperatorGetParam = NULL;
1,668✔
1533
      QUERY_CHECK_CODE(code, lino, _end);
1,668!
1534

1535
      if (pOperator->status == OP_EXEC_DONE) {
1,668✔
1536
        pInfo->currentGroupId = -1;
24✔
1537
        pOperator->status = OP_OPENED;
24✔
1538
        SSDataBlock* result = NULL;
24✔
1539

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

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

1553
  // scan table one by one sequentially
1554
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
13,565,131✔
1555
    int32_t       numOfTables = 0;
19,080✔
1556
    STableKeyInfo tInfo = {0};
19,080✔
1557
    pInfo->countState = TABLE_COUNT_STATE_END;
19,080✔
1558

1559
    while (1) {
17,555✔
1560
      SSDataBlock* result = NULL;
36,635✔
1561
      code = doGroupedTableScan(pOperator, &result);
36,635✔
1562
      QUERY_CHECK_CODE(code, lino, _end);
36,625!
1563

1564
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
36,625!
1565
        (*ppRes) = result;
18,898✔
1566
        return code;
19,068✔
1567
      }
1568

1569
      // if no data, switch to next table and continue scan
1570
      pInfo->currentTable++;
17,727✔
1571

1572
      taosRLockLatch(&pTaskInfo->lock);
17,727✔
1573
      numOfTables = 0;
17,731✔
1574
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
17,731✔
1575
      if (code != TSDB_CODE_SUCCESS) {
17,727!
1576
        taosRUnLockLatch(&pTaskInfo->lock);
×
1577
        TSDB_CHECK_CODE(code, lino, _end);
×
1578
      }
1579

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

1587
      STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
17,557✔
1588
      if (!tmp) {
17,561!
1589
        taosRUnLockLatch(&pTaskInfo->lock);
×
1590
        (*ppRes) = NULL;
×
1591
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1592
      }
1593

1594
      tInfo = *tmp;
17,561✔
1595
      taosRUnLockLatch(&pTaskInfo->lock);
17,561✔
1596

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

1602
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
17,560✔
1603
      QUERY_CHECK_CODE(code, lino, _end);
17,555!
1604
      pInfo->scanTimes = 0;
17,555✔
1605
    }
1606
  } else {  // scan table group by group sequentially
1607
    code = groupSeqTableScan(pOperator, ppRes);
13,546,051✔
1608
    QUERY_CHECK_CODE(code, lino, _end);
13,552,474!
1609
  }
1610

1611
_end:
13,552,474✔
1612
  if (code != TSDB_CODE_SUCCESS) {
13,552,474!
1613
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
×
1614
    pTaskInfo->code = code;
×
1615
    T_LONG_JMP(pTaskInfo->env, code);
×
1616
  }
1617

1618
  return code;
13,552,474✔
1619
}
1620

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

1633
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
4,345,737✔
1634
  cleanupQueryTableDataCond(&pBase->cond);
4,345,737✔
1635
  cleanupQueryTableDataCond(&pBase->orgCond);
4,346,031✔
1636

1637
  if (pAPI->tsdReaderClose) {
4,346,164!
1638
    pAPI->tsdReaderClose(pBase->dataReader);
4,346,200✔
1639
  }
1640
  pBase->dataReader = NULL;
4,346,204✔
1641

1642
  if (pBase->matchInfo.pList != NULL) {
4,346,204!
1643
    taosArrayDestroy(pBase->matchInfo.pList);
4,346,268✔
1644
  }
1645

1646
  tableListDestroy(pBase->pTableListInfo);
4,346,287✔
1647
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
4,346,347✔
1648
  cleanupExprSupp(&pBase->pseudoSup);
4,346,224✔
1649
}
4,345,969✔
1650

1651
static void destroyTableScanOperatorInfo(void* param) {
2,652,064✔
1652
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
2,652,064✔
1653
  blockDataDestroy(pTableScanInfo->pResBlock);
2,652,064✔
1654
  blockDataDestroy(pTableScanInfo->pOrgBlock);
2,652,273✔
1655
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
2,652,269✔
1656
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
2,652,110✔
1657
  taosMemoryFreeClear(param);
2,651,872!
1658
}
2,652,150✔
1659

1660
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
2,649,478✔
1661
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
2,649,478!
1662
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
153,523✔
1663
    for (int32_t i = 0; i < numOfCols; ++i) {
445,742✔
1664
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
292,217✔
1665
      if (pCol) {
292,224!
1666
        pCol->info.noData = true;
292,231✔
1667
      }
1668
    }
1669
  }
1670
}
2,649,480✔
1671

1672
int32_t createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
2,651,022✔
1673
                                    STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
1674
                                    SOperatorInfo** pOptrInfo) {
1675
  QRY_PARAM_CHECK(pOptrInfo);
2,651,022!
1676

1677
  int32_t         code = TSDB_CODE_SUCCESS;
2,651,022✔
1678
  int32_t         lino = 0;
2,651,022✔
1679
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
2,651,022!
1680
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
2,651,272!
1681
  if (pInfo == NULL || pOperator == NULL) {
2,651,407!
1682
    code = terrno;
×
1683
    goto _error;
×
1684
  }
1685

1686
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
2,651,460✔
1687
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
2,651,460✔
1688

1689
  int32_t numOfCols = 0;
2,651,460✔
1690
  code =
1691
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
2,651,460✔
1692
  QUERY_CHECK_CODE(code, lino, _error);
2,649,007!
1693

1694
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
2,649,007✔
1695
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
2,651,449✔
1696
  QUERY_CHECK_CODE(code, lino, _error);
2,651,523!
1697

1698
  if (pScanNode->pScanPseudoCols != NULL) {
2,651,523✔
1699
    SExprSupp* pSup = &pInfo->base.pseudoSup;
650,165✔
1700
    pSup->pExprInfo = NULL;
650,165✔
1701
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
650,165✔
1702
    QUERY_CHECK_CODE(code, lino, _error);
650,111!
1703

1704
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
650,111✔
1705
                                      &pTaskInfo->storageAPI.functionStore);
1706
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
649,855✔
1707
  }
1708

1709
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
2,650,031✔
1710
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
2,650,031✔
1711

1712
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
2,650,031✔
1713
  pInfo->base.readHandle = *readHandle;
2,650,121✔
1714
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
2,650,121✔
1715

1716
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
2,650,121✔
1717
  pInfo->sample.seed = taosGetTimestampSec();
2,650,121✔
1718

1719
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
2,650,759✔
1720
  initResultSizeInfo(&pOperator->resultInfo, 4096);
2,650,759✔
1721
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
2,649,631✔
1722
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
2,649,907✔
1723
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
2,650,227!
1724

1725
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
2,650,227✔
1726
  QUERY_CHECK_CODE(code, lino, _error);
2,647,581!
1727

1728
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
2,647,581✔
1729
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
2,647,581!
1730
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
×
1731
    pInfo->pResBlock = NULL;
×
1732
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
×
1733
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
×
1734
  }
1735

1736
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
2,647,581✔
1737
                            pTaskInfo->pStreamRuntimeInfo);
2,647,581✔
1738
  QUERY_CHECK_CODE(code, lino, _error);
2,648,515✔
1739

1740
  pInfo->currentGroupId = -1;
2,648,501✔
1741

1742
  pInfo->tableEndIndex = -1;
2,648,501✔
1743
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
2,648,501✔
1744
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
2,648,501✔
1745

1746
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
2,648,501✔
1747
                  pTaskInfo);
1748
  pOperator->exprSupp.numOfExprs = numOfCols;
2,648,200✔
1749

1750
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
2,648,200✔
1751
  pInfo->ignoreTag = false;
2,648,200✔
1752

1753
  pInfo->base.pTableListInfo = pTableListInfo;
2,648,200✔
1754
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
2,648,200✔
1755
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
2,651,138!
1756
    code = terrno;
×
1757
    QUERY_CHECK_CODE(code, lino, _error);
×
1758
  }
1759

1760
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
2,651,138✔
1761

1762
  taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
2,651,138✔
1763
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
2,651,979✔
1764
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
1765

1766
  // for non-blocking operator, the open cost is always 0
1767
  pOperator->cost.openCost = 0;
2,649,022✔
1768
  *pOptrInfo = pOperator;
2,649,022✔
1769
  return TSDB_CODE_SUCCESS;
2,649,022✔
1770

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1874
int32_t colIdComparFn(const void* param1, const void* param2) {
1,993,777✔
1875
  int32_t p1 = *(int32_t*)param1;
1,993,777✔
1876
  int32_t p2 = *(int32_t*)param2;
1,993,777✔
1877

1878
  if (p1 == p2) {
1,993,777!
1879
    return 0;
×
1880
  } else {
1881
    return (p1 < p2) ? -1 : 1;
1,993,777✔
1882
  }
1883
}
1884

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

1894
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
211,208✔
1895
  QUERY_CHECK_CODE(code, lino, _end);
211,212!
1896

1897
  pBlockInfo->rows = pBlock->info.rows;
211,212✔
1898
  pBlockInfo->id.uid = pBlock->info.id.uid;
211,212✔
1899
  pBlockInfo->type = STREAM_NORMAL;
211,212✔
1900
  pBlockInfo->version = pBlock->info.version;
211,212✔
1901

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

1910
  SArray* pColList = taosArrayInit(4, sizeof(int32_t));
211,225✔
1911
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
211,231!
1912

1913
  // todo extract method
1914
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
1,159,640✔
1915
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
948,477✔
1916
    if (!pColMatchInfo->needOutput) {
948,452✔
1917
      continue;
2✔
1918
    }
1919

1920
    bool colExists = false;
948,450✔
1921
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
3,096,819!
1922
      SColumnInfoData* pResCol = NULL;
3,096,760✔
1923
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
3,096,760✔
1924
      QUERY_CHECK_CODE(code, lino, _end);
3,096,588!
1925

1926
      if (pResCol->info.colId == pColMatchInfo->colId) {
3,096,588✔
1927
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
948,219✔
1928
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
948,242✔
1929
        QUERY_CHECK_CODE(code, lino, _end);
948,373!
1930

1931
        colExists = true;
948,373✔
1932
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
948,373✔
1933
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
948,407!
1934
        break;
948,407✔
1935
      }
1936
    }
1937

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

1947
  // currently only the tbname pseudo column
1948
  if (pInfo->numOfPseudoExpr > 0 && !isVtableSourceScan) {
211,096!
1949
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
102,184✔
1950
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
102,184✔
1951
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
1952
    if (code) {
102,210!
1953
      QUERY_CHECK_CODE(code, lino, _end);
×
1954
    }
1955

1956
    // reset the error code.
1957
    terrno = 0;
102,210✔
1958

1959
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
522,198✔
1960
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
419,995✔
1961
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
419,990!
1962
    }
1963
  }
1964

1965
  taosArraySort(pColList, colIdComparFn);
211,115✔
1966

1967
  int32_t i = 0, j = 0;
211,224✔
1968
  while (i < taosArrayGetSize(pColList)) {
1,579,908✔
1969
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
1,368,655✔
1970
    if (slot1 > j) {
1,368,687✔
1971
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
3✔
1972
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
1973
      j += 1;
×
1974
    } else {
1975
      i += 1;
1,368,684✔
1976
      j += 1;
1,368,684✔
1977
    }
1978
  }
1979

1980
  while (j < taosArrayGetSize(pInfo->pRes->pDataBlock)) {
211,226✔
1981
    SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
2✔
1982
    colDataSetNNULL(pDst, 0, pBlockInfo->rows);
2!
1983
    j += 1;
2✔
1984
  }
1985

1986
  taosArrayDestroy(pColList);
211,223✔
1987

1988
  code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
211,229✔
1989
  QUERY_CHECK_CODE(code, lino, _end);
211,223!
1990

1991
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
211,223✔
1992
  QUERY_CHECK_CODE(code, lino, _end);
211,226!
1993
  if (pInfo->pRes->info.rows == 0) {
211,226✔
1994
    return 0;
2,330✔
1995
  }
1996

1997
_end:
208,896✔
1998
  if (code != TSDB_CODE_SUCCESS) {
208,896!
1999
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2000
  }
2001
  return code;
208,896✔
2002
}
2003

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

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

2035
_end:
21,435✔
2036
  if (code != TSDB_CODE_SUCCESS) {
21,435!
2037
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2038
  }
2039
  return code;
21,430✔
2040
}
2041

2042
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
256,658✔
2043
  int32_t        code = TSDB_CODE_SUCCESS;
256,658✔
2044
  int32_t        lino = 0;
256,658✔
2045
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
256,658✔
2046
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
256,658✔
2047

2048
  SStreamScanInfo* pInfo = pOperator->info;
256,658✔
2049
  const char*      id = GET_TASKID(pTaskInfo);
256,658✔
2050

2051
  qDebug("start to exec queue scan, %s", id);
256,658!
2052

2053
  if (isTaskKilled(pTaskInfo)) {
256,659!
2054
    (*ppRes) = NULL;
×
2055
    return pTaskInfo->code;
×
2056
  }
2057

2058
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
256,660✔
2059
    while (1) {
×
2060
      SSDataBlock* pResult = NULL;
19,080✔
2061
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
19,080✔
2062
      QUERY_CHECK_CODE(code, lino, _end);
19,066!
2063

2064
      if (pResult && pResult->info.rows > 0) {
19,066!
2065
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
18,907✔
2066
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
18,905✔
2067
        QUERY_CHECK_CODE(code, lino, _end);
18,902!
2068
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
18,902!
2069
        if (pResult->info.rows > 0) {
18,910!
2070
          (*ppRes) = pResult;
18,910✔
2071
          return code;
18,910✔
2072
        }
2073
      } else {
2074
        break;
2075
      }
2076
    }
2077

2078
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
159✔
2079
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
159✔
2080

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

2089
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, validVer);
84✔
2090
  }
2091

2092
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
237,664!
2093
    while (1) {
2,330✔
2094
      bool hasResult =
2095
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
239,994✔
2096

2097
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
239,973✔
2098
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
239,973✔
2099

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

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

2106
      if (hasResult) {
239,963✔
2107
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
211,199!
2108
               pTaskInfo->streamInfo.currentOffset.version);
2109
        blockDataCleanup(pInfo->pRes);
211,229✔
2110
        code = setBlockIntoRes(pInfo, pRes);
211,203✔
2111
        QUERY_CHECK_CODE(code, lino, _end);
211,220!
2112
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
211,220!
2113
               pTaskInfo->streamInfo.currentOffset.version);
2114
        if (pInfo->pRes->info.rows > 0) {
211,229✔
2115
          (*ppRes) = pInfo->pRes;
208,899✔
2116
          return code;
208,899✔
2117
        }
2118
      } else {
2119
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
28,764✔
2120
        (*ppRes) = NULL;
28,760✔
2121
        return code;
28,760✔
2122
      }
2123
    }
2124
  } else {
2125
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
×
2126
    (*ppRes) = NULL;
×
2127
    return code;
×
2128
  }
2129

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

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

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

2152
  // Transfer the Array of STableKeyInfo into uid list.
2153
  int32_t size = 0;
1,185✔
2154
  code = tableListGetSize(pTableListInfo, &size);
1,185✔
2155
  QUERY_CHECK_CODE(code, lino, _end);
1,185!
2156
  for (int32_t i = 0; i < size; ++i) {
26,744✔
2157
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
25,561✔
2158
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
25,526!
2159
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
25,526✔
2160
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
25,559!
2161
  }
2162

2163
  (*ppArrayRes) = tableIdList;
1,183✔
2164

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

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

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

2182
  qDebug("tmqsnap doRawScan called");
5,058!
2183
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
5,058✔
2184
    bool hasNext = false;
5,042✔
2185
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
5,042!
2186
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
5,039✔
2187
      if (code != TSDB_CODE_SUCCESS) {
5,039!
2188
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2189
        QUERY_CHECK_CODE(code, lino, _end);
×
2190
      }
2191
    }
2192

2193
    if (pInfo->dataReader && hasNext) {
5,042✔
2194
      if (isTaskKilled(pTaskInfo)) {
2,531!
2195
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
2196
        return code;
2,531✔
2197
      }
2198

2199
      SSDataBlock* pBlock = NULL;
2,531✔
2200
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock, NULL);
2,531✔
2201
      QUERY_CHECK_CODE(code, lino, _end);
2,530!
2202

2203
      if (pBlock && pBlock->info.rows > 0) {
2,530!
2204
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
2,531✔
2205
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
2,531✔
2206
        QUERY_CHECK_CODE(code, lino, _end);
2,529!
2207
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
2,529!
2208
        (*ppRes) = pBlock;
2,531✔
2209
        return code;
2,531✔
2210
      }
2211
    }
2212

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

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

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

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

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

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

2300
    (*ppRes) = NULL;
16✔
2301
    return code;
16✔
2302
  }
2303

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

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

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

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

2332
  QRY_PARAM_CHECK(pOptrInfo);
295!
2333
  int32_t code = TSDB_CODE_SUCCESS;
295✔
2334
  int32_t lino = 0;
295✔
2335

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

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

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

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

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

2368
void destroyTmqScanOperatorInfo(void* param) {
1,185✔
2369
  if (param == NULL) {
1,185!
2370
    return;
×
2371
  }
2372

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

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

2405
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
1,184✔
2406
  cleanupExprSupp(&pStreamScan->tagCalSup);
1,185✔
2407

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

2416
  taosMemoryFree(pStreamScan);
1,185!
2417
}
2418

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

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

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

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

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

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

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

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

2482
  if (pInfo == NULL || pOperator == NULL) {
1,185!
2483
    code = terrno;
×
2484
    goto _error;
×
2485
  }
2486

2487
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
1,185✔
2488
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
1,185✔
2489

2490
  pInfo->pTagCond = pTagCond;
1,185✔
2491
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
1,185✔
2492

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

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

2506
  for (int32_t i = 0; i < numOfOutput; ++i) {
7,414✔
2507
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
6,230✔
2508
    QUERY_CHECK_NULL(id, code, lino, _error, terrno);
6,230!
2509

2510
    int16_t colId = id->colId;
6,230✔
2511
    void*   tmp = taosArrayPush(pColIds, &colId);
6,229✔
2512
    QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
6,229!
2513

2514
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
6,229✔
2515
      pInfo->primaryTsIndex = id->dstSlotId;
1,185✔
2516
    }
2517
    if (id->isPk) {
6,229!
2518
      pInfo->primaryKeyIndex = id->dstSlotId;
×
2519
      pInfo->basic.primaryPkIndex = id->dstSlotId;
×
2520
      pkType = id->dataType;
×
2521
    }
2522
  }
2523

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

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

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

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

2553
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
1,184✔
2554
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
1,185!
2555

2556
  if (pHandle->vnode) {
1,185!
2557
    SOperatorInfo* pTableScanOp = NULL;
1,185✔
2558
    code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pTableScanOp);
1,185✔
2559
    if (pTableScanOp == NULL || code != 0) {
1,183!
2560
      qError("createTableScanOperatorInfo error, code:%d", pTaskInfo->code);
×
2561
      goto _error;
×
2562
    }
2563

2564
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
1,183✔
2565
    if (pHandle->version > 0) {
1,183!
2566
      pTSInfo->base.cond.endVersion = pHandle->version;
1,183✔
2567
    }
2568

2569
    STableKeyInfo* pList = NULL;
1,183✔
2570
    int32_t        num = 0;
1,183✔
2571
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
1,183✔
2572
    QUERY_CHECK_CODE(code, lino, _error);
1,185!
2573

2574
    if (pHandle->initTableReader) {
1,185!
2575
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
1,185✔
2576
      pTSInfo->base.dataReader = NULL;
1,185✔
2577
    }
2578

2579
    if (pHandle->initTqReader) {
1,185✔
2580
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
1,184✔
2581
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
1,185!
2582
    } else {
2583
      pInfo->tqReader = pHandle->tqReader;
1✔
2584
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
1!
2585
    }
2586

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

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

2603
    pInfo->readHandle = *pHandle;
1,186✔
2604
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
1,186✔
2605
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
1,186✔
2606
    QUERY_CHECK_NULL(pInfo->pCreateTbRes, code, lino, _error, terrno);
1,185!
2607
    pInfo->hasPart = false;
1,185✔
2608

2609
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
1,185✔
2610
    QUERY_CHECK_CODE(code, lino, _error);
1,185!
2611

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

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

2628
  // clear the local variable to avoid repeatly free
2629
  pColIds = NULL;
1,185✔
2630

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

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

2640
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
1,185✔
2641
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
1,185!
2642
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
1,185✔
2643
  QUERY_CHECK_CODE(code, lino, _error);
1,185!
2644

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

2654
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
1,185✔
2655
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateDataRes);
1,185✔
2656
  QUERY_CHECK_CODE(code, lino, _error);
1,185!
2657

2658
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
1,185✔
2659
  pInfo->partitionSup.needCalc = false;
1,185✔
2660
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
1,185✔
2661
  pInfo->igExpired = pTableScanNode->igExpired;
1,185✔
2662
  pInfo->twAggSup.maxTs = INT64_MIN;
1,185✔
2663
  pInfo->pState = pTaskInfo->streamInfo.pState;
1,185✔
2664
  pInfo->stateStore = pTaskInfo->storageAPI.stateStore;
1,185✔
2665
  pInfo->readerFn = pTaskInfo->storageAPI.tqReaderFn;
1,185✔
2666
  pInfo->pFillSup = NULL;
1,185✔
2667
  pInfo->useGetResultRange = false;
1,185✔
2668
  pInfo->pRangeScanRes = NULL;
1,185✔
2669

2670
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
1,185✔
2671
  QUERY_CHECK_CODE(code, lino, _error);
1,185!
2672

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

2677
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doQueueScanNext, NULL, destroyTmqScanOperatorInfo, optrDefaultBufFn,
1,185✔
2678
                                         NULL, optrDefaultGetNextExtFn, NULL);
2679

2680
  *pOptrInfo = pOperator;
1,185✔
2681
  return code;
1,185✔
2682

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

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

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

2704
static int32_t doTagScanOneTable(SOperatorInfo* pOperator, SSDataBlock* pRes, SMetaReader* mr, SStorageAPI* pAPI) {
20,818✔
2705
  int32_t        code = TSDB_CODE_SUCCESS;
20,818✔
2706
  int32_t        lino = 0;
20,818✔
2707
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
20,818✔
2708
  STagScanInfo*  pInfo = pOperator->info;
20,818✔
2709
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
20,818✔
2710
  int32_t        count = pRes->info.rows;
20,818✔
2711

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

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

2728
  char str[512];
2729
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
57,422✔
2730
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
36,589✔
2731

2732
    // refactor later
2733
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
36,525✔
2734
      STR_TO_VARSTR(str, (*mr).me.name);
20,624✔
2735
      code = colDataSetVal(pDst, (count), str, false);
20,624✔
2736
      QUERY_CHECK_CODE(code, lino, _end);
20,674!
2737
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
15,901✔
2738
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
160✔
2739
      QUERY_CHECK_CODE(code, lino, _end);
160!
2740
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
15,741✔
2741
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
160✔
2742
      QUERY_CHECK_CODE(code, lino, _end);
160!
2743
    } else {  // it is a tag value
2744
      STagVal val = {0};
15,581✔
2745
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
15,581✔
2746
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
15,581✔
2747

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

2755
      code = colDataSetVal(pDst, (count), data,
15,659✔
2756
                           (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
15,659!
2757
      QUERY_CHECK_CODE(code, lino, _end);
15,574!
2758

2759
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
15,574!
2760
          (data != NULL)) {
2761
        taosMemoryFree(data);
3,297!
2762
      }
2763
    }
2764
  }
2765

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

2774
  return code;
20,833✔
2775
}
2776

2777
static void tagScanFreeUidTag(void* p) {
50,422✔
2778
  STUidTagInfo* pInfo = p;
50,422✔
2779
  if (pInfo->pTagVal != NULL) {
50,422!
2780
    taosMemoryFree(pInfo->pTagVal);
50,425!
2781
  }
2782
}
50,430✔
2783

2784
static int32_t tagScanCreateResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
11,889✔
2785
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
11,889!
2786
  if (pColumnData == NULL) {
11,896!
2787
    return terrno;
×
2788
  }
2789

2790
  pColumnData->info.type = pType->type;
11,896✔
2791
  pColumnData->info.bytes = pType->bytes;
11,896✔
2792
  pColumnData->info.scale = pType->scale;
11,896✔
2793
  pColumnData->info.precision = pType->precision;
11,896✔
2794

2795
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
11,896✔
2796
  if (code != TSDB_CODE_SUCCESS) {
11,896✔
2797
    terrno = code;
5✔
2798
    taosMemoryFree(pColumnData);
×
2799
    return terrno;
×
2800
  }
2801

2802
  pParam->columnData = pColumnData;
11,891✔
2803
  pParam->colAlloced = true;
11,891✔
2804
  return TSDB_CODE_SUCCESS;
11,891✔
2805
}
2806

2807
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
73,700✔
2808
  int32_t                code = TSDB_CODE_SUCCESS;
73,700✔
2809
  int32_t                lino = 0;
73,700✔
2810
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
73,700✔
2811
  SColumnNode*           pSColumnNode = NULL;
73,700✔
2812
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
73,700✔
2813
    pSColumnNode = *(SColumnNode**)pNode;
20,730✔
2814
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
52,970!
2815
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
×
2816
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
×
2817
      pSColumnNode = NULL;
×
2818
      pCtx->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
×
2819
      if (NULL == pSColumnNode) {
×
2820
        return DEAL_RES_ERROR;
×
2821
      }
2822
      pSColumnNode->colId = -1;
×
2823
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
×
2824
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
×
2825
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
×
2826
      nodesDestroyNode(*pNode);
×
2827
      *pNode = (SNode*)pSColumnNode;
×
2828
    } else {
2829
      return DEAL_RES_CONTINUE;
×
2830
    }
2831
  } else {
2832
    return DEAL_RES_CONTINUE;
52,970✔
2833
  }
2834

2835
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
20,730✔
2836
  if (!data) {
20,727✔
2837
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
11,906✔
2838
    if (code == TSDB_CODE_DUP_KEY) {
11,917!
2839
      code = TSDB_CODE_SUCCESS;
×
2840
    }
2841
    QUERY_CHECK_CODE(code, lino, _end);
11,917!
2842
    pSColumnNode->slotId = pCtx->index++;
11,917✔
2843
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
11,917✔
2844
                         .type = pSColumnNode->node.resType.type,
11,917✔
2845
                         .bytes = pSColumnNode->node.resType.bytes};
11,917✔
2846
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
11,917✔
2847
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
11,916!
2848
  } else {
2849
    SColumnNode* col = *(SColumnNode**)data;
8,821✔
2850
    pSColumnNode->slotId = col->slotId;
8,821✔
2851
  }
2852

2853
_end:
20,737✔
2854
  if (code != TSDB_CODE_SUCCESS) {
20,737!
2855
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2856
    return DEAL_RES_ERROR;
×
2857
  }
2858
  return DEAL_RES_CONTINUE;
20,737✔
2859
}
2860

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

2868
  SSDataBlock* pResBlock = createTagValBlockForFilter(pInfo->filterCtx.cInfoList, numOfTables, aUidTags, pVnode, pAPI);
11,897✔
2869
  QUERY_CHECK_NULL(pResBlock, code, lino, _end, terrno);
11,888!
2870

2871
  pBlockList = taosArrayInit(1, POINTER_BYTES);
11,888✔
2872
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
11,895!
2873

2874
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
11,892✔
2875
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
11,892!
2876

2877
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
11,892✔
2878

2879
  SScalarParam output = {0};
11,892✔
2880
  code = tagScanCreateResultData(&type, numOfTables, &output);
11,892✔
2881
  QUERY_CHECK_CODE(code, lino, _end);
11,891!
2882

2883
  code = scalarCalculate(pTagCond, pBlockList, &output, NULL, NULL);
11,891✔
2884
  QUERY_CHECK_CODE(code, lino, _end);
11,887!
2885

2886
  bool* result = (bool*)output.columnData->pData;
11,887✔
2887
  for (int32_t i = 0; i < numOfTables; ++i) {
29,750✔
2888
    if (result[i]) {
17,865✔
2889
      void* tmp = taosArrayPush(aFilterIdxs, &i);
16,760✔
2890
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
16,760!
2891
    }
2892
  }
2893

2894
  colDataDestroy(output.columnData);
11,885✔
2895
  taosMemoryFreeClear(output.columnData);
11,895!
2896

2897
_end:
×
2898
  blockDataDestroy(pResBlock);
11,893✔
2899
  taosArrayDestroy(pBlockList);
11,894✔
2900
  if (code != TSDB_CODE_SUCCESS) {
11,896!
2901
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2902
  }
2903
  return code;
11,894✔
2904
}
2905

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

2916
      code = colDataSetVal(pColInfo, rowIndex, str, false);
×
2917
      QUERY_CHECK_CODE(code, lino, _end);
×
2918
    } else if (FUNCTION_TYPE_TBUID == pExprInfo->pExpr->_function.functionType) {
98,593✔
2919
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
49,305✔
2920
      QUERY_CHECK_CODE(code, lino, _end);
49,307!
2921
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
49,288!
2922
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
49,306✔
2923
      QUERY_CHECK_CODE(code, lino, _end);
49,309!
2924
    }
2925
  } else {
2926
    STagVal tagVal = {0};
106,963✔
2927
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
106,963✔
2928
    if (pUidTagInfo->pTagVal == NULL) {
106,963!
2929
      colDataSetNULL(pColInfo, rowIndex);
×
2930
    } else {
2931
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
106,963✔
2932

2933
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
107,081!
2934
        colDataSetNULL(pColInfo, rowIndex);
35!
2935
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
107,046!
2936
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
2937
        QUERY_CHECK_CODE(code, lino, _end);
2!
2938
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
132,874!
2939
        if (IS_STR_DATA_BLOB(pColInfo->info.type)) {
25,829✔
2940
          QUERY_CHECK_CODE(code = TSDB_CODE_BLOB_NOT_SUPPORT_TAG, lino, _end);
2✔
2941
        }
2942
        char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
25,827!
2943
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
25,831!
2944

2945
        varDataSetLen(tmp, tagVal.nData);
25,831✔
2946
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
25,831✔
2947
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
25,831✔
2948
        taosMemoryFree(tmp);
25,833!
2949
        QUERY_CHECK_CODE(code, lino, _end);
25,828!
2950
      } else {
2951
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
81,217✔
2952
        QUERY_CHECK_CODE(code, lino, _end);
81,208!
2953
      }
2954
    }
2955
  }
2956

2957
_end:
205,671✔
2958
  if (code != TSDB_CODE_SUCCESS) {
205,671!
2959
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2960
  }
2961
  return code;
205,587✔
2962
}
2963

2964
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
31,804✔
2965
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
2966
  int32_t       code = TSDB_CODE_SUCCESS;
31,804✔
2967
  int32_t       lino = 0;
31,804✔
2968
  STagScanInfo* pInfo = pOperator->info;
31,804✔
2969
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
31,804✔
2970
  if (!ignoreFilterIdx) {
31,804✔
2971
    size_t szTables = taosArrayGetSize(aFilterIdxs);
11,893✔
2972
    for (int i = 0; i < szTables; ++i) {
28,656✔
2973
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
16,765✔
2974
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
16,764✔
2975
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
16,767!
2976
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
95,435✔
2977
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
78,670✔
2978
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
78,656!
2979
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
78,656✔
2980
        QUERY_CHECK_CODE(code, lino, _end);
78,668!
2981
      }
2982
    }
2983
  } else {
2984
    size_t szTables = taosArrayGetSize(aUidTags);
19,911✔
2985
    for (int i = 0; i < szTables; ++i) {
52,478✔
2986
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
32,559✔
2987
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
32,560✔
2988
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
159,483✔
2989
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
126,923✔
2990
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
126,917!
2991
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
126,917✔
2992
        QUERY_CHECK_CODE(code, lino, _end);
126,924!
2993
      }
2994
    }
2995
  }
2996

2997
_end:
19,919✔
2998
  if (code != TSDB_CODE_SUCCESS) {
31,810!
2999
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3000
  }
3001
  return code;
31,808✔
3002
}
3003

3004
static int32_t doTagScanFromCtbIdxNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
63,262✔
3005
  int32_t        code = TSDB_CODE_SUCCESS;
63,262✔
3006
  int32_t        lino = 0;
63,262✔
3007
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
63,262✔
3008
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
63,262✔
3009

3010
  STagScanInfo* pInfo = pOperator->info;
63,262✔
3011
  SSDataBlock*  pRes = pInfo->pRes;
63,262✔
3012

3013
  QRY_PARAM_CHECK(ppRes);
63,262!
3014

3015
  if (pOperator->status == OP_EXEC_DONE) {
63,262✔
3016
    return TSDB_CODE_SUCCESS;
31,113✔
3017
  }
3018
  blockDataCleanup(pRes);
32,149✔
3019

3020
  if (pInfo->pCtbCursor == NULL) {
32,176!
3021
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
32,177✔
3022
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
32,173!
3023
  } else {
3024
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
3025
    QUERY_CHECK_CODE(code, lino, _end);
×
3026
  }
3027

3028
  SArray* aUidTags = pInfo->aUidTags;
32,174✔
3029
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
32,174✔
3030
  int32_t count = 0;
32,174✔
3031
  bool    ctbCursorFinished = false;
32,174✔
3032
  while (1) {
700✔
3033
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
32,874✔
3034
    taosArrayClear(aFilterIdxs);
32,876✔
3035

3036
    int32_t numTables = 0;
32,861✔
3037
    while (numTables < pOperator->resultInfo.capacity) {
83,283!
3038
      SMCtbCursor* pCur = pInfo->pCtbCursor;
83,283✔
3039
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
83,283✔
3040
      if (uid == 0) {
83,288✔
3041
        ctbCursorFinished = true;
32,876✔
3042
        break;
32,876✔
3043
      }
3044
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
50,412✔
3045
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
50,412!
3046
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
50,431!
3047

3048
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
50,431✔
3049
      void* tmp = taosArrayPush(aUidTags, &info);
50,422✔
3050
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
50,422!
3051
      ++numTables;
50,422✔
3052
    }
3053

3054
    if (numTables == 0) {
32,876✔
3055
      break;
1,062✔
3056
    }
3057
    bool ignoreFilterIdx = true;
31,814✔
3058
    if (pInfo->pTagCond != NULL) {
31,814✔
3059
      ignoreFilterIdx = false;
11,897✔
3060
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo);
11,897✔
3061
      QUERY_CHECK_CODE(code, lino, _end);
11,891!
3062
    } else {
3063
      ignoreFilterIdx = true;
19,917✔
3064
    }
3065

3066
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
31,808✔
3067
    QUERY_CHECK_CODE(code, lino, _end);
31,809!
3068

3069
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
31,809✔
3070

3071
    if (count != 0) {
31,805✔
3072
      break;
31,105✔
3073
    }
3074
  }
3075

3076
  if (count > 0) {
32,167✔
3077
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
31,106✔
3078
  }
3079
  if (count == 0 || ctbCursorFinished) {
32,177!
3080
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
32,177✔
3081
    pInfo->pCtbCursor = NULL;
32,178✔
3082
    setOperatorCompleted(pOperator);
32,178✔
3083
  }
3084

3085
  pRes->info.rows = count;
32,175✔
3086
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
32,175✔
3087
  if (bLimitReached) {
32,164!
3088
    setOperatorCompleted(pOperator);
×
3089
  }
3090

3091
  pOperator->resultInfo.totalRows += pRes->info.rows;
32,172✔
3092
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
32,172✔
3093

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

3101
  return code;
32,172✔
3102
}
3103

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

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

3116
  STableKeyInfo info = {.groupId = 0};
×
3117

3118
  int32_t tableIdx = 0;
×
3119
  taosHashClear(pListInfo->map);
×
3120
  taosArrayClear(pListInfo->pTableList);
×
3121

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

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

3135
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
×
3136

3137
  return code;
×
3138
}
3139

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

3153
    if (pOperator->status == OP_EXEC_DONE) {
2!
3154
      pOperator->status = OP_OPENED;
×
3155
    }
3156
  }
3157

3158
  if (pOperator->status == OP_EXEC_DONE) {
7,977✔
3159
    (*ppRes) = NULL;
3,934✔
3160
    return code;
3,934✔
3161
  }
3162

3163
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4,043✔
3164
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
4,043✔
3165

3166
  SExprInfo*   pExprInfo = &pOperator->exprSupp.pExprInfo[0];
4,043✔
3167
  SSDataBlock* pRes = pInfo->pRes;
4,043✔
3168
  blockDataCleanup(pRes);
4,043✔
3169

3170
  int32_t size = 0;
4,046✔
3171
  code = tableListGetSize(pInfo->pTableListInfo, &size);
4,046✔
3172
  QUERY_CHECK_CODE(code, lino, _end);
4,038!
3173

3174
  if (size == 0) {
4,038✔
3175
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
114✔
3176
    (*ppRes) = NULL;
114✔
3177
    return code;
114✔
3178
  }
3179

3180
  SMetaReader mr = {0};
3,924✔
3181
  pAPI->metaReaderFn.initReader(&mr, pInfo->readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
3,924✔
3182
  pRes->info.rows = 0;
3,938✔
3183

3184
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
24,754!
3185
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
20,820✔
3186
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD &&
20,813!
3187
        code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
3188
      // ignore other error
3189
      code = TSDB_CODE_SUCCESS;
20,814✔
3190
    }
3191
    QUERY_CHECK_CODE(code, lino, _end);
20,813!
3192

3193
    if (++pInfo->curPos >= size) {
20,813✔
3194
      setOperatorCompleted(pOperator);
3,931✔
3195
    }
3196
  }
3197

3198
  pAPI->metaReaderFn.clearReader(&mr);
3,934✔
3199
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
3,938✔
3200
  if (bLimitReached) {
3,933!
3201
    setOperatorCompleted(pOperator);
×
3202
  }
3203

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

3209
  pOperator->resultInfo.totalRows += pRes->info.rows;
3,935✔
3210

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

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

3227
static void destroyTagScanOperatorInfo(void* param) {
36,201✔
3228
  STagScanInfo* pInfo = (STagScanInfo*)param;
36,201✔
3229
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
36,201!
3230
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
3231
  }
3232
  taosHashCleanup(pInfo->filterCtx.colHash);
36,201✔
3233
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
36,221✔
3234
  taosArrayDestroy(pInfo->aFilterIdxs);
36,214✔
3235
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
36,220✔
3236

3237
  blockDataDestroy(pInfo->pRes);
36,221✔
3238
  taosArrayDestroy(pInfo->matchInfo.pList);
36,228✔
3239
  tableListDestroy(pInfo->pTableListInfo);
36,229✔
3240

3241
  pInfo->pRes = NULL;
36,231✔
3242
  pInfo->pTableListInfo = NULL;
36,231✔
3243
  taosMemoryFreeClear(param);
36,231!
3244
}
36,228✔
3245

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

3251
  int32_t         code = TSDB_CODE_SUCCESS;
36,171✔
3252
  int32_t         lino = 0;
36,171✔
3253
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
36,171✔
3254
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
36,171!
3255
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
36,196!
3256
  if (pInfo == NULL || pOperator == NULL) {
36,218!
3257
    code = terrno;
×
3258
    goto _error;
×
3259
  }
3260

3261
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
36,219✔
3262

3263
  int32_t    numOfExprs = 0;
36,219✔
3264
  SExprInfo* pExprInfo = NULL;
36,219✔
3265

3266
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
36,219✔
3267
  QUERY_CHECK_CODE(code, lino, _error);
36,223!
3268

3269
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
36,223✔
3270
  QUERY_CHECK_CODE(code, lino, _error);
36,198!
3271

3272
  int32_t num = 0;
36,198✔
3273
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
36,198✔
3274
  QUERY_CHECK_CODE(code, lino, _error);
36,190!
3275

3276
  pInfo->pTagCond = pTagCond;
36,190✔
3277
  pInfo->pTagIndexCond = pTagIndexCond;
36,190✔
3278
  pInfo->suid = pPhyNode->suid;
36,190✔
3279
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
36,190✔
3280

3281
  pInfo->pTableListInfo = pTableListInfo;
36,190✔
3282
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
36,190✔
3283
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
36,216!
3284

3285
  pInfo->readHandle = *pReadHandle;
36,216✔
3286
  pInfo->curPos = 0;
36,216✔
3287

3288
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
36,216✔
3289
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
36,227✔
3290
                  pTaskInfo);
3291
  initResultSizeInfo(&pOperator->resultInfo, 4096);
36,191✔
3292
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
36,193✔
3293
  QUERY_CHECK_CODE(code, lino, _error);
36,221!
3294

3295
  if (pTagScanNode->onlyMetaCtbIdx) {
36,221✔
3296
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
32,175✔
3297
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
32,175!
3298

3299
    pInfo->aFilterIdxs = taosArrayInit(pOperator->resultInfo.capacity, sizeof(int32_t));
32,175✔
3300
    QUERY_CHECK_NULL(pInfo->aFilterIdxs, code, lino, _error, terrno);
32,169!
3301

3302
    pInfo->filterCtx.colHash =
32,171✔
3303
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
32,169✔
3304
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
32,171!
3305

3306
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
32,171✔
3307
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
32,179!
3308

3309
    if (pInfo->pTagCond != NULL) {
32,179✔
3310
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
11,918✔
3311
    }
3312
  }
3313
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
36,216✔
3314
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
36,216✔
3315
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
3316
  *pOptrInfo = pOperator;
36,210✔
3317
  return code;
36,210✔
3318

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

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

3332
// table merge scan operator
3333

3334
static int32_t subTblRowCompareTsFn(const void* pLeft, const void* pRight, void* param) {
668,955✔
3335
  int32_t                 left = *(int32_t*)pLeft;
668,955✔
3336
  int32_t                 right = *(int32_t*)pRight;
668,955✔
3337
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
668,955✔
3338

3339
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
668,955✔
3340
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
668,955✔
3341

3342
  if (leftIdx == -1) {
668,955✔
3343
    return 1;
572,146✔
3344
  } else if (rightIdx == -1) {
96,809✔
3345
    return -1;
960✔
3346
  }
3347

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

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

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

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

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

3384
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
8,507✔
3385
  int32_t code = TSDB_CODE_SUCCESS;
8,507✔
3386
  int32_t lino = 0;
8,507✔
3387
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
8,507✔
3388
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
8,507!
3389
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
8,516!
3390
  for (int i = 0; i < src->numOfCols; i++) {
28,500✔
3391
    dst->colList[i] = src->colList[i];
19,984✔
3392
  }
3393

3394
_end:
8,516✔
3395
  if (code != TSDB_CODE_SUCCESS) {
8,516!
3396
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3397
  }
3398
  return code;
8,521✔
3399
}
3400

3401
static int32_t fetchNextSubTableBlockFromReader(SOperatorInfo* pOperator, STmsSubTableInput* pInput,
12,767✔
3402
                                                bool* pSubTableHasBlock) {
3403
  int32_t code = 0;
12,767✔
3404

3405
  STableMergeScanInfo*    pInfo = pOperator->info;
12,767✔
3406
  SReadHandle*            pHandle = &pInfo->base.readHandle;
12,767✔
3407
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
12,767✔
3408
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
12,767✔
3409
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
12,767✔
3410

3411
  blockDataCleanup(pInput->pReaderBlock);
12,767✔
3412
  if (!pInput->bInMemReader) {
12,766✔
3413
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
6,417✔
3414
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
6,417✔
3415
    if (code != 0) {
6,419!
3416
      return code;
×
3417
    }
3418
  }
3419

3420
  pInfo->base.dataReader = pInput->pReader;
12,768✔
3421

3422
  while (true) {
22✔
3423
    bool hasNext = false;
12,790✔
3424
    code = pAPI->tsdReader.tsdNextDataBlock(pInfo->base.dataReader, &hasNext);
12,790✔
3425
    if (code != 0) {
12,792!
3426
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3427
      pInfo->base.dataReader = NULL;
×
3428
      return code;
×
3429
    }
3430

3431
    if (!hasNext || isTaskKilled(pTaskInfo)) {
12,792!
3432
      if (isTaskKilled(pTaskInfo)) {
8,527!
3433
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
3434
        pInfo->base.dataReader = NULL;
×
3435
        return code;
×
3436
      }
3437

3438
      *pSubTableHasBlock = false;
8,529✔
3439
      break;
8,529✔
3440
    }
3441

3442
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
4,265✔
3443
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
4,181✔
3444
    } else {
3445
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
84✔
3446
    }
3447

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

3455
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
4,264!
3456
      *pSubTableHasBlock = false;
×
3457
      break;
×
3458
    }
3459
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pInput->pReaderBlock->info.rows == 0) {
4,264!
3460
      continue;
22✔
3461
    }
3462

3463
    *pSubTableHasBlock = true;
4,242✔
3464
    break;
4,242✔
3465
  }
3466

3467
  if (*pSubTableHasBlock) {
12,771✔
3468
    pInput->pReaderBlock->info.id.groupId =
8,485✔
3469
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
4,242✔
3470
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
4,243✔
3471
  }
3472
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
12,772✔
3473
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
9,604✔
3474
    pInput->pReader = NULL;
9,603✔
3475
  }
3476

3477
  pInfo->base.dataReader = NULL;
12,771✔
3478
  return TSDB_CODE_SUCCESS;
12,771✔
3479
}
3480

3481
static int32_t setGroupStartEndIndex(STableMergeScanInfo* pInfo) {
5,318✔
3482
  pInfo->bGroupProcessed = false;
5,318✔
3483

3484
  int32_t numOfTables = 0;
5,318✔
3485
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
5,318✔
3486
  if (code != TSDB_CODE_SUCCESS) {
5,338✔
3487
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
14!
3488
    return code;
×
3489
  }
3490

3491
  int32_t i = pInfo->tableStartIndex + 1;
5,324✔
3492
  for (; i < numOfTables; ++i) {
8,506✔
3493
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
3,351✔
3494
    if (!tableKeyInfo) {
3,353✔
3495
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
2!
3496
      return terrno;
2✔
3497
    }
3498
    if (tableKeyInfo->groupId != pInfo->groupId) {
3,351✔
3499
      break;
169✔
3500
    }
3501
  }
3502
  pInfo->tableEndIndex = i - 1;
5,324✔
3503
  return TSDB_CODE_SUCCESS;
5,324✔
3504
}
3505

3506
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
5,346✔
3507
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
13,876✔
3508
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
8,530✔
3509
    if (pInput->rowIdx == -1) {
8,530✔
3510
      continue;
4,287✔
3511
    }
3512

3513
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
4,243!
3514
      pInput->rowIdx = 0;
4,243✔
3515
      pInput->pageIdx = -1;
4,243✔
3516
    }
3517

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

3527
  __merge_compare_fn_t mergeCompareFn = (!pSubTblsInfo->pPkOrderInfo) ? subTblRowCompareTsFn : subTblRowCompareTsPkFn;
5,346!
3528
  return tMergeTreeCreate(&pSubTblsInfo->pTree, pSubTblsInfo->numSubTables, pSubTblsInfo, mergeCompareFn);
5,346✔
3529
}
3530

3531
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
5,323✔
3532
  int32_t code = setGroupStartEndIndex(pInfo);
5,323✔
3533
  if (code != TSDB_CODE_SUCCESS) {
5,332!
3534
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3535
    return code;
×
3536
  }
3537
  STmsSubTablesMergeInfo* pSubTblsInfo = taosMemoryCalloc(1, sizeof(STmsSubTablesMergeInfo));
5,332!
3538
  if (pSubTblsInfo == NULL) {
5,343!
3539
    return terrno;
×
3540
  }
3541
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
5,343✔
3542
  if (!pSubTblsInfo->pTsOrderInfo) {
5,341!
3543
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3544
    return terrno;
×
3545
  }
3546
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
5,341✔
3547
    pSubTblsInfo->pPkOrderInfo = taosArrayGet(pInfo->pSortInfo, 1);
4✔
3548
    if (!pSubTblsInfo->pPkOrderInfo) {
×
3549
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3550
      return terrno;
×
3551
    }
3552
  } else {
3553
    pSubTblsInfo->pPkOrderInfo = NULL;
5,332✔
3554
  }
3555
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
5,332✔
3556
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
5,332!
3557
  if (pSubTblsInfo->aInputs == NULL) {
5,337!
3558
    taosMemoryFree(pSubTblsInfo);
×
3559
    return terrno;
×
3560
  }
3561
  int32_t bufPageSize = pInfo->bufPageSize;
5,337✔
3562
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
5,337✔
3563
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
5,337✔
3564
  if (code != TSDB_CODE_SUCCESS) {
5,339✔
3565
    taosMemoryFree(pSubTblsInfo->aInputs);
4!
3566
    taosMemoryFree(pSubTblsInfo);
×
3567
    return code;
×
3568
  }
3569
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
5,335✔
3570
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
5,335✔
3571

3572
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
5,335✔
3573
  return TSDB_CODE_SUCCESS;
5,335✔
3574
}
3575

3576
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
5,331✔
3577
  int32_t        code = TSDB_CODE_SUCCESS;
5,331✔
3578
  int32_t        lino = 0;
5,331✔
3579
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5,331✔
3580
  SReadHandle*   pHandle = &pInfo->base.readHandle;
5,331✔
3581
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
5,331✔
3582

3583
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
5,331✔
3584

3585
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
13,861✔
3586
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
8,511✔
3587
    pInput->type = SUB_TABLE_MEM_BLOCK;
8,511✔
3588

3589
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
8,511✔
3590
    QUERY_CHECK_CODE(code, lino, _end);
8,521!
3591

3592
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
8,521✔
3593
    QUERY_CHECK_CODE(code, lino, _end);
8,529!
3594

3595
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
8,529✔
3596
    QUERY_CHECK_CODE(code, lino, _end);
8,525!
3597

3598
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
8,525✔
3599
    pInput->pKeyInfo = keyInfo;
8,524✔
3600

3601
    if (isTaskKilled(pTaskInfo)) {
8,524!
3602
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
×
3603
    }
3604

3605
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
8,527✔
3606
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
3,183✔
3607
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
3,183✔
3608
      QUERY_CHECK_CODE(code, lino, _end);
3,183!
3609
      pInput->bInMemReader = true;
3,183✔
3610
    } else {
3611
      pInput->pReader = NULL;
5,344✔
3612
      pInput->bInMemReader = false;
5,344✔
3613
    }
3614
    bool hasNext = true;
8,527✔
3615
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
8,527✔
3616
    QUERY_CHECK_CODE(code, lino, _end);
8,530!
3617
    if (!hasNext) {
8,530✔
3618
      pInput->rowIdx = -1;
4,287✔
3619
      ++pSubTblsInfo->numSubTablesCompleted;
4,287✔
3620
      continue;
4,287✔
3621
    } else {
3622
      pInput->rowIdx = 0;
4,243✔
3623
      pInput->pageIdx = -1;
4,243✔
3624
    }
3625
  }
3626

3627
_end:
5,350✔
3628
  if (code != TSDB_CODE_SUCCESS) {
5,350!
3629
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3630
  }
3631
  return code;
5,346✔
3632
}
3633

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

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

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

3659
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
763,049✔
3660
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
763,049✔
3661
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
763,049✔
3662

3663
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
763,049!
3664
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
763,049✔
3665
    ++pInput->rowIdx;
758,988✔
3666
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
4,061!
3667
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
4,243!
3668
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
4,243✔
3669
    }
3670
    if (pInput->rowIdx != -1) {
4,243!
3671
      SColumnInfoData* col = taosArrayGet(pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
×
3672
      if (!col) {
×
3673
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
3674
        return terrno;
×
3675
      }
3676
      pInput->pInputBlock = pInputBlock;
×
3677
      pInput->aTs = (int64_t*)col->pData;
×
3678
    }
3679
  }
3680

3681
  return tMergeTreeAdjust(pSubTblsInfo->pTree, tMergeTreeGetAdjustIndex(pSubTblsInfo->pTree));
763,049✔
3682
}
3683

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

3690
  for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
3,102,950✔
3691
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
2,294,553✔
3692
    QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
2,323,924!
3693

3694
    SColumnInfoData* pSrcColInfo = taosArrayGet(pInputBlock->pDataBlock, i);
2,323,924✔
3695
    QUERY_CHECK_NULL(pSrcColInfo, code, lino, _end, terrno);
2,313,085!
3696
    bool isNull = colDataIsNull(pSrcColInfo, pInputBlock->info.rows, pInput->rowIdx, NULL);
2,321,219✔
3697

3698
    if (isNull) {
2,321,219✔
3699
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
277,613✔
3700
      QUERY_CHECK_CODE(code, lino, _end);
278,133!
3701
    } else {
3702
      if (pSrcColInfo->pData != NULL) {
2,043,606✔
3703
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
2,042,257!
3704
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
2,042,257✔
3705
        QUERY_CHECK_CODE(code, lino, _end);
2,059,068!
3706
      }
3707
    }
3708
  }
3709
  pBlock->info.dataLoad = 1;
763,701✔
3710
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
763,701✔
3711
  pBlock->info.rows += 1;
763,701✔
3712

3713
_end:
763,701✔
3714
  if (code != TSDB_CODE_SUCCESS) {
763,701!
3715
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3716
  }
3717
  return code;
763,601✔
3718
}
3719

3720
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
8,628✔
3721
                                       SSDataBlock** pResBlock) {
3722
  int32_t                 code = TSDB_CODE_SUCCESS;
8,628✔
3723
  int32_t                 lino = 0;
8,628✔
3724
  STableMergeScanInfo*    pInfo = pOperator->info;
8,628✔
3725
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
8,628✔
3726
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
8,628✔
3727
  bool                    finished = false;
8,628✔
3728

3729
  QRY_PARAM_CHECK(pResBlock);
8,628!
3730

3731
  blockDataCleanup(pBlock);
8,628✔
3732

3733
  while (true) {
×
3734
    while (true) {
3735
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
773,117✔
3736
        finished = true;
8,628✔
3737
        break;
8,628✔
3738
      }
3739

3740
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
764,489✔
3741
      QUERY_CHECK_CODE(code, lino, _end);
763,534!
3742

3743
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
763,534✔
3744
      QUERY_CHECK_CODE(code, lino, _end);
764,664!
3745

3746
      if (pBlock->info.rows >= capacity) {
764,664!
3747
        break;
×
3748
      }
3749
    }
3750

3751
    if (isTaskKilled(pTaskInfo)) {
8,628!
3752
      return pTaskInfo->code;
×
3753
    }
3754

3755
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo);
8,628✔
3756
    if (finished || limitReached || pBlock->info.rows > 0) {
8,628!
3757
      break;
3758
    }
3759
  }
3760

3761
  if (pBlock->info.rows > 0) {
8,628✔
3762
    *pResBlock = pBlock;
3,283✔
3763
  }
3764

3765
_end:
5,345✔
3766
  if (code != TSDB_CODE_SUCCESS) {
8,628!
3767
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3768
    pTaskInfo->code = code;
×
3769
  }
3770
  return code;
8,628✔
3771
}
3772

3773
static int32_t startSubTablesTableMergeScan(SOperatorInfo* pOperator) {
5,327✔
3774
  int32_t              code = TSDB_CODE_SUCCESS;
5,327✔
3775
  int32_t              lino = 0;
5,327✔
3776
  STableMergeScanInfo* pInfo = pOperator->info;
5,327✔
3777

3778
  code = initSubTablesMergeInfo(pInfo);
5,327✔
3779
  QUERY_CHECK_CODE(code, lino, _end);
5,334!
3780

3781
  code = initSubTableInputs(pOperator, pInfo);
5,334✔
3782
  QUERY_CHECK_CODE(code, lino, _end);
5,346!
3783

3784
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
5,346✔
3785
  QUERY_CHECK_CODE(code, lino, _end);
5,345!
3786

3787
_end:
5,345✔
3788
  if (code != TSDB_CODE_SUCCESS) {
5,345!
3789
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3790
  }
3791
  return code;
5,345✔
3792
}
3793

3794
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
1,699,348✔
3795
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
1,699,348✔
3796
  if (pSubTblsInfo != NULL) {
1,699,348✔
3797
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
5,346✔
3798

3799
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
13,876✔
3800
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
8,530✔
3801
      taosMemoryFree(pInput->tblCond.colList);
8,530!
3802
      blockDataDestroy(pInput->pReaderBlock);
8,530✔
3803
      blockDataDestroy(pInput->pPageBlock);
8,530✔
3804
      taosArrayDestroy(pInput->aBlockPages);
8,530✔
3805
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
8,530✔
3806
      pInput->pReader = NULL;
8,530✔
3807
    }
3808

3809
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
5,346✔
3810
    taosMemoryFree(pSubTblsInfo->aInputs);
5,346!
3811

3812
    taosMemoryFree(pSubTblsInfo);
5,346!
3813
    pInfo->pSubTablesMergeInfo = NULL;
5,346✔
3814

3815
    // taosMemoryTrim(0);
3816
  }
3817
}
1,699,348✔
3818

3819
int32_t doTableMergeScanParaSubTablesNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
8,584✔
3820
  QRY_PARAM_CHECK(ppRes);
8,584!
3821

3822
  int32_t lino = 0;
8,584✔
3823
  int32_t tableListSize = 0;
8,584✔
3824
  int64_t st = taosGetTimestampUs();
8,601✔
3825

3826
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
8,601✔
3827
  STableMergeScanInfo* pInfo = pOperator->info;
8,601✔
3828

3829
  if (pOperator->status == OP_EXEC_DONE) {
8,601!
3830
    return TSDB_CODE_SUCCESS;
×
3831
  }
3832

3833
  int32_t code = pOperator->fpSet._openFn(pOperator);
8,601✔
3834
  QUERY_CHECK_CODE(code, lino, _end);
8,597!
3835

3836
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
8,597✔
3837
  QUERY_CHECK_CODE(code, lino, _end);
8,602!
3838

3839
  if (!pInfo->hasGroupId) {
8,602✔
3840
    pInfo->hasGroupId = true;
5,311✔
3841

3842
    if (tableListSize == 0) {
5,311✔
3843
      setOperatorCompleted(pOperator);
153✔
3844
      (*ppRes) = NULL;
154✔
3845
      return code;
154✔
3846
    }
3847

3848
    pInfo->tableStartIndex = 0;
5,158✔
3849
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
5,158✔
3850
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
5,166!
3851
    pInfo->groupId = pTmpGpId->groupId;
5,166✔
3852
    code = startSubTablesTableMergeScan(pOperator);
5,166✔
3853
    QUERY_CHECK_CODE(code, lino, _end);
5,175!
3854
  }
3855

3856
  SSDataBlock* pBlock = NULL;
8,466✔
3857
  while (pInfo->tableStartIndex < tableListSize) {
8,636✔
3858
    if (isTaskKilled(pTaskInfo)) {
8,628!
3859
      break;
×
3860
    }
3861

3862
    code = getSubTablesSortedBlock(pOperator, pInfo->pResBlock, pOperator->resultInfo.capacity, &pBlock);
8,628✔
3863
    QUERY_CHECK_CODE(code, lino, _end);
8,628!
3864

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

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

3872
    if (pBlock != NULL) {
8,628✔
3873
      pBlock->info.id.groupId = pInfo->groupId;
3,283✔
3874
      pOperator->resultInfo.totalRows += pBlock->info.rows;
3,283✔
3875
      pInfo->bGroupProcessed = true;
3,283✔
3876
      break;
3,283✔
3877
    } else {
3878
      // Data of this group are all dumped, let's try the next group
3879
      stopSubTablesTableMergeScan(pInfo);
5,345✔
3880
      if (pInfo->tableEndIndex >= tableListSize - 1) {
5,346✔
3881
        setOperatorCompleted(pOperator);
5,176✔
3882
        break;
5,176✔
3883
      }
3884

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

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

3896
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
8,459✔
3897

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

3907
  return code;
8,459✔
3908
}
3909

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

3917
  if (pInfo->mSkipTables == NULL) {
43,740✔
3918
    pInfo->mSkipTables = taosHashInit(pInfo->tableEndIndex - pInfo->tableStartIndex + 1,
14,926✔
3919
                                      taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_NO_LOCK);
3920
    QUERY_CHECK_NULL(pInfo->mSkipTables, code, lino, _end, terrno);
14,928!
3921
  }
3922
  int bSkip = 1;
43,742✔
3923
  if (pInfo->mSkipTables != NULL) {
43,742!
3924
    code = taosHashPut(pInfo->mSkipTables, &uid, sizeof(uid), &bSkip, sizeof(bSkip));
43,742✔
3925
    if (code == TSDB_CODE_DUP_KEY) {
43,743!
3926
      code = TSDB_CODE_SUCCESS;
×
3927
    }
3928
    QUERY_CHECK_CODE(code, lino, _end);
43,743!
3929
  }
3930

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

3939
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
6,212,928✔
3940
  STableMergeScanInfo* pInfo = pOperator->info;
6,212,928✔
3941
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
6,212,928✔
3942
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
6,212,928✔
3943
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
6,212,928✔
3944
  int32_t              code = 0;
6,212,928✔
3945
  bool                 hasNext = false;
6,212,928✔
3946
  STsdbReader*         reader = pInfo->base.dataReader;
6,212,928✔
3947

3948
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
6,212,928✔
3949
  if (code != 0) {
6,215,120!
3950
    pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
3951
    qError("table merge scan fetch next data block error code: %d, %s", code, GET_TASKID(pTaskInfo));
×
3952
    pTaskInfo->code = code;
×
3953
    return code;
×
3954
  }
3955

3956
  if (!hasNext || isTaskKilled(pTaskInfo)) {
6,215,120!
3957
    if (isTaskKilled(pTaskInfo)) {
1,645,188!
3958
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
×
3959
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
3960
    }
3961
    *pFinished = true;
1,645,417✔
3962
    return code;
1,645,417✔
3963
  }
3964

3965
  uint32_t status = 0;
4,569,937✔
3966
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
4,569,937✔
3967

3968
  if (code != TSDB_CODE_SUCCESS) {
4,569,761✔
3969
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
71!
3970
    pTaskInfo->code = code;
×
3971
    return code;
×
3972
  }
3973

3974
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
4,569,690!
3975
    *pFinished = true;
×
3976
    return code;
×
3977
  }
3978

3979
  // current block is filter out according to filter condition, continue load the next block
3980
  if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
4,569,690!
3981
    *pSkipped = true;
24,077✔
3982
    return code;
24,077✔
3983
  }
3984

3985
  return code;
4,545,613✔
3986
}
3987

3988
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
7,784,914✔
3989
  STableMergeScanSortSourceParam* source = param;
7,784,914✔
3990

3991
  SOperatorInfo*       pOperator = source->pOperator;
7,784,914✔
3992
  STableMergeScanInfo* pInfo = pOperator->info;
7,784,914✔
3993
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
7,784,914✔
3994
  SSDataBlock*         pBlock = NULL;
7,784,914✔
3995
  int64_t              st = taosGetTimestampUs();
7,786,557✔
3996
  int32_t              code = TSDB_CODE_SUCCESS;
7,786,557✔
3997

3998
  while (true) {
3999
    if (pInfo->rtnNextDurationBlocks) {
9,400,616✔
4000
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
3,187,339✔
4001
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
4002

4003
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
3,187,427✔
4004
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
1,589,781✔
4005
        ++pInfo->nextDurationBlocksIdx;
1,589,781✔
4006
      } else {
4007
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
3,187,456✔
4008
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
1,589,796✔
4009
          pInfo->nextDurationBlocks[i] = NULL;
1,589,810✔
4010
        }
4011

4012
        pInfo->rtnNextDurationBlocks = false;
1,597,660✔
4013
        pInfo->nextDurationBlocksIdx = 0;
1,597,660✔
4014
        pInfo->numNextDurationBlocks = 0;
1,597,660✔
4015
        continue;
1,597,660✔
4016
      }
4017
    } else {
4018
      bool bFinished = false;
6,213,277✔
4019
      bool bSkipped = false;
6,213,277✔
4020

4021
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
6,213,277✔
4022
      if (code != 0) {
6,214,937!
4023
        return code;
1,599,084✔
4024
      }
4025

4026
      pBlock = pInfo->pReaderBlock;
6,214,937✔
4027
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
6,214,937✔
4028
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
4029
      if (bFinished) {
6,215,722✔
4030
        pInfo->bNewFilesetEvent = false;
1,645,427✔
4031
        break;
1,645,427✔
4032
      }
4033

4034
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
4,570,295!
4035
        if (!bSkipped) {
1,599,102✔
4036
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
1,591,250✔
4037
          if (code) {
1,591,232!
4038
            *ppBlock = NULL;
×
4039
            return code;
×
4040
          }
4041

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

4051
        if (pInfo->bNewFilesetEvent) {
1,599,084!
4052
          pInfo->rtnNextDurationBlocks = true;
1,599,084✔
4053
          *ppBlock = NULL;
1,599,084✔
4054
          return code;
1,599,084✔
4055
        }
4056

4057
        if (pInfo->bNextDurationBlockEvent) {
×
4058
          pInfo->bNextDurationBlockEvent = false;
×
4059
          continue;
16,399✔
4060
        }
4061
      }
4062
      if (bSkipped) continue;
2,971,193✔
4063
    }
4064

4065
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
4,544,575✔
4066

4067
    pOperator->resultInfo.totalRows += pBlock->info.rows;
4,544,504✔
4068
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
4,544,506✔
4069
    *ppBlock = pBlock;
4,544,506✔
4070

4071
    return code;
4,544,506✔
4072
  }
4073

4074
  *ppBlock = NULL;
1,645,427✔
4075
  return code;
1,645,427✔
4076
}
4077

4078
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
1,688,113✔
4079
  int32_t code = TSDB_CODE_SUCCESS;
1,688,113✔
4080
  int32_t lino = 0;
1,688,113✔
4081
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
1,688,113✔
4082
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
1,692,601!
4083
  SBlockOrderInfo biTs = {0};
1,693,191✔
4084
  SBlockOrderInfo biPk = {0};
1,693,191✔
4085

4086
  int32_t tsTargetSlotId = 0;
1,693,191✔
4087
  int32_t pkTargetSlotId = -1;
1,693,191✔
4088
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
10,270,772✔
4089
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
8,583,715✔
4090
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
8,576,544!
4091
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
8,577,629✔
4092
      tsTargetSlotId = colInfo->dstSlotId;
1,689,806✔
4093
      biTs.order = order;
1,689,806✔
4094
      biTs.slotId = tsTargetSlotId;
1,689,806✔
4095
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
1,689,806✔
4096
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
1,689,806✔
4097
    }
4098
    // TODO: order by just ts
4099
    if (colInfo->isPk) {
8,577,582✔
4100
      pkTargetSlotId = colInfo->dstSlotId;
7,693✔
4101
      biPk.order = order;
7,693✔
4102
      biPk.slotId = pkTargetSlotId;
7,693✔
4103
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
7,693✔
4104
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
7,693✔
4105
    }
4106
  }
4107

4108
  void* tmp = taosArrayPush(pSortInfo, &biTs);
1,690,813✔
4109
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,690,813!
4110
  if (pkTargetSlotId != -1) {
1,690,813✔
4111
    tmp = taosArrayPush(pSortInfo, &biPk);
7,695✔
4112
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
7,695!
4113
  }
4114

4115
  (*ppSortArray) = pSortInfo;
1,690,814✔
4116

4117
_end:
1,690,814✔
4118
  if (code != TSDB_CODE_SUCCESS) {
1,690,814!
4119
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4120
  }
4121
  return code;
1,689,502✔
4122
}
4123

4124
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
2,157,350✔
4125
  STableMergeScanInfo* pTmsInfo = param;
2,157,350✔
4126
  if (type == TSD_READER_NOTIFY_DURATION_START) {
2,157,350!
4127
    pTmsInfo->bNewFilesetEvent = true;
2,157,773✔
4128
  } else if (type == TSD_READER_NOTIFY_NEXT_DURATION_BLOCK) {
×
4129
    pTmsInfo->bNextDurationBlockEvent = true;
×
4130
  }
4131
  qDebug("table merge scan receive notification. type %d, fileset %d", type, info->duration.filesetId);
2,157,350✔
4132

4133
  return;
2,158,040✔
4134
}
4135

4136
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
3,243,837✔
4137
  STableMergeScanInfo* pInfo = pOperator->info;
3,243,837✔
4138
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
3,243,837✔
4139

4140
  int32_t code = TSDB_CODE_SUCCESS;
3,243,837✔
4141
  int32_t lino = 0;
3,243,837✔
4142
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
3,243,837✔
4143

4144
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
3,243,837✔
4145
  pInfo->bNewFilesetEvent = false;
3,243,867✔
4146
  pInfo->bNextDurationBlockEvent = false;
3,243,867✔
4147

4148
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
3,243,867✔
4149
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
3,243,867✔
4150

4151
  pInfo->pSortHandle = NULL;
3,243,867✔
4152
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
3,243,867✔
4153
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
3,243,867✔
4154
  if (code) {
3,243,016!
4155
    return code;
×
4156
  }
4157

4158
  if (pInfo->bSortRowId && numOfTable != 1) {
3,243,016✔
4159
    int32_t memSize = 512 * 1024 * 1024;
649,023✔
4160
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
649,023✔
4161
    QUERY_CHECK_CODE(code, lino, _end);
649,268!
4162
  }
4163

4164
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
3,243,261✔
4165
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
3,242,766✔
4166
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
3,242,606✔
4167

4168
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
3,242,761✔
4169
  QUERY_CHECK_CODE(code, lino, _end);
3,243,262!
4170

4171
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
3,243,262!
4172
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
3,243,796!
4173
  param->pOperator = pOperator;
3,243,796✔
4174

4175
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
3,243,796!
4176
  if (ps == NULL) {
3,244,585!
4177
    taosMemoryFree(param);
×
4178
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
4179
  }
4180

4181
  ps->param = param;
3,244,585✔
4182
  ps->onlyRef = false;
3,244,585✔
4183
  code = tsortAddSource(pInfo->pSortHandle, ps);
3,244,585✔
4184
  QUERY_CHECK_CODE(code, lino, _end);
3,243,731!
4185

4186
  if (numOfTable == 1) {
3,243,731✔
4187
    tsortSetSingleTableMerge(pInfo->pSortHandle);
1,588,535✔
4188
  } else {
4189
    code = tsortOpen(pInfo->pSortHandle);
1,655,196✔
4190
    QUERY_CHECK_CODE(code, lino, _end);
1,656,112!
4191
  }
4192

4193
_end:
1,656,112✔
4194
  if (code != TSDB_CODE_SUCCESS) {
3,244,673!
4195
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4196
  }
4197
  return code;
3,244,589✔
4198
}
4199

4200
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
3,242,075✔
4201
  STableMergeScanInfo* pInfo = pOperator->info;
3,242,075✔
4202
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
3,242,075✔
4203
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
3,242,075✔
4204

4205
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
3,242,076✔
4206
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
3,242,217✔
4207
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
3,242,217✔
4208
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
3,242,217✔
4209
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
3,242,217✔
4210
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
3,242,217✔
4211

4212
  tsortDestroySortHandle(pInfo->pSortHandle);
3,242,217✔
4213
  pInfo->pSortHandle = NULL;
3,242,375✔
4214
}
3,242,375✔
4215

4216
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
1,639,898✔
4217
  int32_t              code = TSDB_CODE_SUCCESS;
1,639,898✔
4218
  int32_t              lino = 0;
1,639,898✔
4219
  STableMergeScanInfo* pInfo = pOperator->info;
1,639,898✔
4220
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
1,639,898✔
4221
  SReadHandle*         pHandle = &pInfo->base.readHandle;
1,639,898✔
4222
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
1,639,898✔
4223
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
1,639,898✔
4224

4225
  {
4226
    int32_t numOfTables = 0;
1,639,900✔
4227
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
1,639,900✔
4228
    QUERY_CHECK_CODE(code, lino, _end);
1,642,472!
4229

4230
    int32_t i = pInfo->tableStartIndex + 1;
1,642,472✔
4231
    for (; i < numOfTables; ++i) {
4,237,978✔
4232
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
2,657,488✔
4233
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
2,655,744!
4234
      if (tableKeyInfo->groupId != pInfo->groupId) {
2,655,973✔
4235
        break;
60,467✔
4236
      }
4237
    }
4238
    pInfo->tableEndIndex = i - 1;
1,640,957✔
4239
  }
4240
  pInfo->bGroupProcessed = false;
1,640,957✔
4241
  int32_t tableStartIdx = pInfo->tableStartIndex;
1,640,957✔
4242
  int32_t tableEndIdx = pInfo->tableEndIndex;
1,640,957✔
4243

4244
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
1,640,957✔
4245
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
1,640,957✔
4246
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
1,643,578✔
4247
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
1,643,578✔
4248
  QUERY_CHECK_CODE(code, lino, _end);
1,646,251!
4249
  if (pInfo->filesetDelimited) {
1,646,251!
4250
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
1,646,557✔
4251
  }
4252
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
1,645,554✔
4253

4254
  code = startDurationForGroupTableMergeScan(pOperator);
1,646,565✔
4255
  QUERY_CHECK_CODE(code, lino, _end);
1,647,013!
4256

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

4265
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
1,644,493✔
4266
  STableMergeScanInfo* pInfo = pOperator->info;
1,644,493✔
4267
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
1,644,493✔
4268
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
1,644,493✔
4269

4270
  stopDurationForGroupTableMergeScan(pOperator);
1,644,493✔
4271

4272
  if (pInfo->base.dataReader != NULL) {
1,644,725✔
4273
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
1,644,718✔
4274
    pInfo->base.dataReader = NULL;
1,644,705✔
4275
  }
4276
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
1,644,714✔
4277
    if (pInfo->nextDurationBlocks[i]) {
2!
4278
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
2✔
4279
      pInfo->nextDurationBlocks[i] = NULL;
2✔
4280
    }
4281
    pInfo->numNextDurationBlocks = 0;
2✔
4282
    pInfo->nextDurationBlocksIdx = 0;
2✔
4283
  }
4284
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
1,644,712✔
4285
  taosHashCleanup(pInfo->mSkipTables);
1,644,614✔
4286
  pInfo->mSkipTables = NULL;
1,644,476✔
4287
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
1,644,476✔
4288
}
1,644,488✔
4289

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

4298
  blockDataCleanup(pResBlock);
10,647,280✔
4299

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

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

4318
      if (pResBlock->info.rows >= capacity) {
2,147,483,647✔
4319
        break;
5,328,210✔
4320
      }
4321
    }
4322

4323
    if (tsortIsClosed(pHandle)) {
10,647,952!
4324
      terrno = TSDB_CODE_TSC_QUERY_CANCELLED;
×
4325
      T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
4326
    }
4327

4328
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
10,648,238✔
4329
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
10,647,990✔
4330
           pInfo->limitInfo.numOfOutputRows);
4331
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
10,648,169!
4332
      break;
4333
    }
4334
  }
4335
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
10,648,169✔
4336
}
4337

4338
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
9,111,816✔
4339
  if (pOperator->status == OP_EXEC_DONE) {
9,111,816✔
4340
    (*ppRes) = NULL;
26,384✔
4341
    return TSDB_CODE_SUCCESS;
26,384✔
4342
  }
4343

4344
  int32_t              code = TSDB_CODE_SUCCESS;
9,085,432✔
4345
  int32_t              lino = 0;
9,085,432✔
4346
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
9,085,432✔
4347
  STableMergeScanInfo* pInfo = pOperator->info;
9,085,432✔
4348

4349
  code = pOperator->fpSet._openFn(pOperator);
9,085,432✔
4350
  QUERY_CHECK_CODE(code, lino, _end);
9,086,804!
4351

4352
  int64_t st = taosGetTimestampUs();
9,089,730✔
4353

4354
  int32_t tableListSize = 0;
9,089,730✔
4355
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
9,089,730✔
4356
  QUERY_CHECK_CODE(code, lino, _end);
9,085,974!
4357

4358
  if (!pInfo->hasGroupId) {
9,085,974✔
4359
    pInfo->hasGroupId = true;
1,683,334✔
4360

4361
    if (tableListSize == 0) {
1,683,334✔
4362
      setOperatorCompleted(pOperator);
101,932✔
4363
      (*ppRes) = NULL;
101,998✔
4364
      return code;
101,998✔
4365
    }
4366
    pInfo->tableStartIndex = 0;
1,581,402✔
4367
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
1,581,402✔
4368
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,579,989!
4369
    pInfo->groupId = tmp->groupId;
1,579,989✔
4370
    startGroupTableMergeScan(pOperator);
1,579,989✔
4371
  }
4372

4373
  SSDataBlock* pBlock = NULL;
8,989,479✔
4374
  while (pInfo->tableStartIndex < tableListSize) {
10,647,378!
4375
    if (isTaskKilled(pTaskInfo)) {
10,647,378!
4376
      goto _end;
×
4377
    }
4378

4379
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
10,647,397✔
4380
                                              pOperator);
4381
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
10,648,029✔
4382
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
14,400✔
4383
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
14,400!
4384
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
14,400✔
4385
    }
4386
    if (pBlock != NULL) {
10,648,145✔
4387
      pBlock->info.id.groupId = pInfo->groupId;
7,405,967✔
4388
      pOperator->resultInfo.totalRows += pBlock->info.rows;
7,405,967✔
4389
      pInfo->bGroupProcessed = true;
7,405,967✔
4390
      break;
7,405,967✔
4391
    } else {
4392
      if (pInfo->bNewFilesetEvent) {
3,242,178✔
4393
        stopDurationForGroupTableMergeScan(pOperator);
1,597,671✔
4394
        code = startDurationForGroupTableMergeScan(pOperator);
1,597,666✔
4395
        QUERY_CHECK_CODE(code, lino, _end);
1,597,638!
4396
      } else {
4397
        // Data of this group are all dumped, let's try the next group
4398
        stopGroupTableMergeScan(pOperator);
1,644,507✔
4399
        if (pInfo->tableEndIndex >= tableListSize - 1) {
1,644,461✔
4400
          setOperatorCompleted(pOperator);
1,584,022✔
4401
          break;
1,584,070✔
4402
        }
4403

4404
        pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
60,439✔
4405
        STableKeyInfo* tmp = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
60,439✔
4406
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
60,504!
4407
        pInfo->groupId = tmp->groupId;
60,504✔
4408
        startGroupTableMergeScan(pOperator);
60,504✔
4409
        resetLimitInfoForNextGroup(&pInfo->limitInfo);
60,504✔
4410
      }
4411
    }
4412
  }
4413

4414
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
8,990,042✔
4415

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

4425
  return code;
8,990,042✔
4426
}
4427

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

4434
void destroyTableMergeScanOperatorInfo(void* param) {
1,694,024✔
4435
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
1,694,024✔
4436

4437
  // start one reader variable
4438
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
1,694,024!
4439
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
1,694,077✔
4440
    pTableScanInfo->base.dataReader = NULL;
1,694,044✔
4441
  }
4442

4443
  for (int32_t i = 0; i < pTableScanInfo->numNextDurationBlocks; ++i) {
1,695,424✔
4444
    if (pTableScanInfo->nextDurationBlocks[i] != NULL) {
1,433!
4445
      blockDataDestroy(pTableScanInfo->nextDurationBlocks[i]);
1,433✔
4446
      pTableScanInfo->nextDurationBlocks[i] = NULL;
1,433✔
4447
    }
4448
  }
4449

4450
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
1,693,991✔
4451
  pTableScanInfo->pSortHandle = NULL;
1,694,031✔
4452
  taosHashCleanup(pTableScanInfo->mSkipTables);
1,694,031✔
4453
  pTableScanInfo->mSkipTables = NULL;
1,694,052✔
4454
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
1,694,052✔
4455
  pTableScanInfo->pSortInputBlock = NULL;
1,694,183✔
4456
  // end one reader variable
4457

4458
  cleanupQueryTableDataCond(&pTableScanInfo->base.cond);
1,694,183✔
4459
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
1,694,178✔
4460

4461
  blockDataDestroy(pTableScanInfo->pResBlock);
1,693,969✔
4462
  pTableScanInfo->pResBlock = NULL;
1,694,184✔
4463

4464
  // remove it from the task->result list
4465
  blockDataDestroy(pTableScanInfo->pReaderBlock);
1,694,184✔
4466
  pTableScanInfo->pReaderBlock = NULL;
1,694,138✔
4467
  taosArrayDestroy(pTableScanInfo->pSortInfo);
1,694,138✔
4468

4469
  stopSubTablesTableMergeScan(pTableScanInfo);
1,694,193✔
4470

4471
  taosMemoryFreeClear(param);
1,694,031!
4472
}
1,694,149✔
4473

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

4488
  *pOptrExplain = execInfo;
205,778✔
4489
  *len = sizeof(STableMergeScanExecInfo);
205,778✔
4490

4491
  return TSDB_CODE_SUCCESS;
205,778✔
4492
}
4493

4494
int32_t createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
1,689,403✔
4495
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
4496
                                         SOperatorInfo** pOptrInfo) {
4497
  QRY_PARAM_CHECK(pOptrInfo);
1,689,403!
4498

4499
  int32_t              code = TSDB_CODE_SUCCESS;
1,689,403✔
4500
  int32_t              lino = 0;
1,689,403✔
4501
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
1,689,403!
4502
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
1,691,191!
4503
  if (pInfo == NULL || pOperator == NULL) {
1,691,938!
4504
    code = terrno;
×
4505
    goto _error;
×
4506
  }
4507

4508
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
1,692,084✔
4509

4510
  int32_t numOfCols = 0;
1,692,084✔
4511
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
1,692,084✔
4512
                             &pInfo->base.matchInfo);
4513
  QUERY_CHECK_CODE(code, lino, _error);
1,691,449!
4514

4515
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
1,691,449✔
4516
  QUERY_CHECK_CODE(code, lino, _error);
1,693,171!
4517

4518
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
1,693,171✔
4519
    SExprSupp* pSup = &pInfo->base.pseudoSup;
1,021,395✔
4520
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
1,021,395✔
4521
    QUERY_CHECK_CODE(code, lino, _error);
1,022,276!
4522

4523
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
1,022,276✔
4524
                                      &pTaskInfo->storageAPI.functionStore);
4525
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
1,021,188!
4526
  }
4527

4528
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
1,692,964✔
4529

4530
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
1,692,964✔
4531
  QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
1,693,127!
4532

4533
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
1,693,127✔
4534
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
1,693,127✔
4535
  pInfo->base.scanFlag = MAIN_SCAN;
1,693,127✔
4536
  pInfo->base.readHandle = *readHandle;
1,693,127✔
4537

4538
  pInfo->readIdx = -1;
1,693,127✔
4539

4540
  pInfo->base.limitInfo.limit.limit = -1;
1,693,127✔
4541
  pInfo->base.limitInfo.slimit.limit = -1;
1,693,127✔
4542
  pInfo->base.pTableListInfo = pTableListInfo;
1,693,127✔
4543

4544
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
1,693,127✔
4545
  pInfo->sample.seed = taosGetTimestampSec();
1,693,127✔
4546

4547
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0,
1,692,893✔
4548
                            pTaskInfo->pStreamRuntimeInfo);
1,692,893✔
4549
  QUERY_CHECK_CODE(code, lino, _error);
1,691,555!
4550

4551
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
1,691,555✔
4552

4553
  pInfo->mergeLimit = -1;
1,692,051✔
4554
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
1,692,051!
4555
  if (hasLimit) {
1,692,051✔
4556
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1
218,712✔
4557
                            ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset
29,896✔
4558
                            : pInfo->limitInfo.limit.limit;
109,356✔
4559
    pInfo->mSkipTables = NULL;
109,356✔
4560
  }
4561

4562
  initResultSizeInfo(&pOperator->resultInfo, 1024);
1,692,051✔
4563
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
1,689,645✔
4564
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
1,692,652!
4565
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
1,692,652✔
4566
  QUERY_CHECK_CODE(code, lino, _error);
1,693,291!
4567
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
1,693,291!
4568
    pInfo->bSortRowId = true;
841,747✔
4569
  } else {
4570
    pInfo->bSortRowId = false;
850,740✔
4571
  }
4572

4573
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
1,692,487✔
4574
  QUERY_CHECK_CODE(code, lino, _error);
1,688,330!
4575

4576
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
1,688,330✔
4577
  QUERY_CHECK_CODE(code, lino, _error);
1,689,840!
4578

4579
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
1,689,840✔
4580
  QUERY_CHECK_CODE(code, lino, _error);
1,690,159!
4581

4582
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
1,690,159✔
4583

4584
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
1,690,159✔
4585
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
1,690,159✔
4586

4587
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
1,691,098✔
4588

4589
  // start one reader variable
4590
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
1,691,312✔
4591
  QUERY_CHECK_CODE(code, lino, _error);
1,691,928!
4592

4593
  if (!tsExperimental) {
1,691,928!
4594
    pInfo->filesetDelimited = false;
×
4595
  } else {
4596
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
1,691,928✔
4597
  }
4598
  // end one reader variable
4599
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
1,691,928✔
4600
                  pInfo, pTaskInfo);
4601
  pOperator->exprSupp.numOfExprs = numOfCols;
1,691,296✔
4602

4603
  pOperator->fpSet = createOperatorFpSet(
1,691,296✔
4604
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
1,691,296✔
4605
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
4606
      NULL);
4607
  pOperator->cost.openCost = 0;
1,690,470✔
4608

4609
  *pOptrInfo = pOperator;
1,690,470✔
4610
  return code;
1,690,470✔
4611

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

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

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

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

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

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

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

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

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

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

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

4754
  pInfo->readHandle = *readHandle;
×
4755

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

4948
  return code;
×
4949
}
4950

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

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

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

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

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

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

4983
  return code;
×
4984
}
4985

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

5102
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
5103
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
5104
  int32_t code = TSDB_CODE_SUCCESS;
×
5105
  int32_t lino = 0;
×
5106
  char    stbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5107
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
×
5108
  QUERY_CHECK_CODE(code, lino, _end);
×
5109

5110
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
5111
  if (pSupp->groupByDbName) {
×
5112
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
×
5113
  } else {
5114
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
×
5115
  }
5116

5117
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
5118
  pRes->info.id.groupId = groupId;
×
5119

5120
  int64_t ctbNum = 0;
×
5121
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL, NULL);
×
5122
  QUERY_CHECK_CODE(code, lino, _end);
×
5123
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
×
5124
  QUERY_CHECK_CODE(code, lino, _end);
×
5125

5126
_end:
×
5127
  if (code != TSDB_CODE_SUCCESS) {
×
5128
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5129
  }
5130
  return code;
×
5131
}
5132

5133
static void destoryTableCountScanOperator(void* param) {
×
5134
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
×
5135
  blockDataDestroy(pTableCountScanInfo->pRes);
×
5136

5137
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
×
5138
  taosMemoryFreeClear(param);
×
5139
}
×
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