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

taosdata / TDengine / #3778

28 Mar 2025 06:51AM UTC coverage: 63.374% (+0.4%) from 62.934%
#3778

push

travis-ci

web-flow
fix(tdb): disable page recycling (#30529)

155771 of 313582 branches covered (49.67%)

Branch coverage included in aggregate %.

241569 of 313390 relevant lines covered (77.08%)

20705705.27 hits per line

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

63.53
/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 "storageapi.h"
38
#include "wal.h"
39

40
int32_t scanDebug = 0;
41

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

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

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

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

65
  STableCountScanSupp supp;
66

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

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

74
bool processBlockWithProbability(const SSampleExecInfo* pInfo) {
10,953,732✔
75
#if 0
76
  if (pInfo->sampleRatio == 1) {
77
    return true;
78
  }
79

80
  uint32_t val = taosRandR((uint32_t*) &pInfo->seed);
81
  return (val % ((uint32_t)(1/pInfo->sampleRatio))) == 0;
82
#else
83
  return true;
10,953,732✔
84
#endif
85
}
86

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

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

97
  // 0 by default, which means it is not a interval operator of the upstream operator.
98
  if (pInterval->interval == 0) {
11,774,927✔
99
    *overlap = false;
10,920,840✔
100
    return code;
10,920,840✔
101
  }
102

103
  if (order == TSDB_ORDER_ASC) {
854,087✔
104
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
563,237✔
105
    if (w.ekey < pBlockInfo->window.skey) {
563,157!
106
      qError("w.ekey:%" PRId64 " < pBlockInfo->window.skey:%" PRId64, w.ekey, pBlockInfo->window.skey);
×
107
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
108
    }
109

110
    if (w.ekey < pBlockInfo->window.ekey) {
563,157✔
111
      *overlap = true;
453,860✔
112
      return code;
453,860✔
113
    }
114

115
    while (1) {
116
      getNextTimeWindow(pInterval, &w, order);
109,310✔
117
      if (w.skey > pBlockInfo->window.ekey) {
109,263✔
118
        break;
83,893✔
119
      }
120

121
      if (w.ekey <= pBlockInfo->window.ekey) {
25,370!
122
        qError("w.ekey:%" PRId64 " <= pBlockInfo->window.ekey:%" PRId64, w.ekey, pBlockInfo->window.ekey);
×
123
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
124
      }
125
      if (TMAX(w.skey, pBlockInfo->window.skey) <= pBlockInfo->window.ekey) {
25,370✔
126
        *overlap = true;
25,357✔
127
        return code;
25,357✔
128
      }
129
    }
130
  } else {
131
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.ekey);
290,850✔
132
    if (w.skey > pBlockInfo->window.ekey) {
291,415!
133
      qError("w.skey:%" PRId64 " > pBlockInfo->window.skey:%" PRId64, w.skey, pBlockInfo->window.ekey);
×
134
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
135
    }
136

137
    if (w.skey > pBlockInfo->window.skey) {
291,415✔
138
      *overlap = true;
211,784✔
139
      return code;
211,784✔
140
    }
141

142
    while (1) {
143
      getNextTimeWindow(pInterval, &w, order);
79,631✔
144
      if (w.ekey < pBlockInfo->window.skey) {
79,626✔
145
        break;
65,509✔
146
      }
147

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

159
  *overlap = false;
149,402✔
160
  return code;
149,402✔
161
}
162

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

169
  int64_t buf[2] = {0};
2,241,142✔
170
  SET_RES_WINDOW_KEY((char*)buf, &groupId, sizeof(groupId), groupId);
2,241,142✔
171

172
  STableScanInfo* pTableScanInfo = pOperator->info;
2,241,142✔
173

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

177
  if (p1 == NULL) {
2,241,595✔
178
    return NULL;
485,750✔
179
  }
180

181
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
1,755,845✔
182
  if (NULL == *pPage) {
1,755,745!
183
    return NULL;
×
184
  }
185

186
  return (SResultRow*)((char*)(*pPage) + p1->offset);
1,755,745✔
187
}
188

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

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

208
static int32_t doDynamicPruneDataBlock(SOperatorInfo* pOperator, SDataBlockInfo* pBlockInfo, uint32_t* status) {
12,762,656✔
209
  STableScanInfo* pTableScanInfo = pOperator->info;
12,762,656✔
210
  int32_t         code = TSDB_CODE_SUCCESS;
12,762,656✔
211

212
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
12,762,656✔
213
    return TSDB_CODE_SUCCESS;
10,523,294✔
214
  }
215

216
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
2,239,362✔
217

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

221
  if (pRow == NULL) {
2,241,482✔
222
    return TSDB_CODE_SUCCESS;
485,796✔
223
  }
224

225
  bool notLoadBlock = true;
1,755,686✔
226
  for (int32_t i = 0; i < pSup1->numOfExprs; ++i) {
1,879,793✔
227
    int32_t functionId = pSup1->pCtx[i].functionId;
1,857,311✔
228

229
    SResultRowEntryInfo* pEntry = getResultEntryInfo(pRow, i, pTableScanInfo->base.pdInfo.pExprSup->rowEntryInfoOffset);
1,857,311✔
230

231
    EFuncDataRequired reqStatus = fmFuncDynDataRequired(functionId, pEntry, pBlockInfo);
1,857,241✔
232
    if (reqStatus != FUNC_DATA_REQUIRED_NOT_LOAD) {
1,857,258✔
233
      notLoadBlock = false;
1,733,151✔
234
      break;
1,733,151✔
235
    }
236
  }
237

238
  // release buffer pages
239
  releaseBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, pPage);
1,755,633✔
240

241
  if (notLoadBlock) {
1,755,574✔
242
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
22,489✔
243
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
22,489✔
244
  }
245

246
  return code;
1,755,603✔
247
}
248

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

256
  return filterRangeExecute(pFilterInfo, pColsAgg, numOfCols, numOfRows, keep);
86,991✔
257
}
258

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

268
  int32_t code = pAPI->tsdReader.tsdReaderRetrieveBlockSMAInfo(pTableScanInfo->dataReader, pBlock, &allColumnsHaveAgg,
7,402,079✔
269
                                                               &hasNullSMA);
270
  if (code != TSDB_CODE_SUCCESS) {
7,402,016!
271
    return code;
×
272
  }
273

274
  if (!allColumnsHaveAgg || hasNullSMA) {
7,402,016!
275
    *pLoad = false;
5,214,426✔
276
  } else {
277
    *pLoad = true;
2,187,590✔
278
  }
279

280
  return code;
7,402,016✔
281
}
282

283
static int32_t doSetTagColumnData(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
17,277,636✔
284
                                  int32_t rows) {
285
  int32_t    code = 0;
17,277,636✔
286
  SExprSupp* pSup = &pTableScanInfo->pseudoSup;
17,277,636✔
287
  if (pSup->numOfExprs > 0) {
17,277,636✔
288
    code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
4,977,777✔
289
                                  pTaskInfo, &pTableScanInfo->metaCache);
290
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
291
    if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
4,977,749!
292
      if (pTaskInfo->streamInfo.pState) blockDataCleanup(pBlock);
×
293
      code = 0;
×
294
    }
295
  }
296

297
  return code;
17,278,265✔
298
}
299

300
bool applyLimitOffset(SLimitInfo* pLimitInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
41,625,643✔
301
  SLimit*     pLimit = &pLimitInfo->limit;
41,625,643✔
302
  const char* id = GET_TASKID(pTaskInfo);
41,625,643✔
303

304
  if (pLimitInfo->remainOffset > 0) {
41,625,643✔
305
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
74,173✔
306
      pLimitInfo->remainOffset -= pBlock->info.rows;
21,223✔
307
      blockDataEmpty(pBlock);
21,223✔
308
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
21,223✔
309
      return false;
21,223✔
310
    } else {
311
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
52,950✔
312
      if (code != TSDB_CODE_SUCCESS) {
52,950!
313
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
314
        pTaskInfo->code = code;
×
315
        T_LONG_JMP(pTaskInfo->env, code);
×
316
      }
317
      pLimitInfo->remainOffset = 0;
52,950✔
318
    }
319
  }
320

321
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
41,604,420✔
322
    // limit the output rows
323
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
430,657✔
324
    blockDataKeepFirstNRows(pBlock, keep);
430,657✔
325

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

331
  pLimitInfo->numOfOutputRows += pBlock->info.rows;
41,173,763✔
332
  return false;
41,173,763✔
333
}
334

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

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

347
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
16,800,658✔
348

349
  pCost->totalBlocks += 1;
16,800,658✔
350
  pCost->totalRows += pBlock->info.rows;
16,800,658✔
351
  *status = pTableScanInfo->dataBlockLoadFlag;
16,800,658✔
352

353
  if (pOperator->exprSupp.pFilterInfo != NULL) {
16,800,658✔
354
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
5,025,750✔
355
  } else {
356
    bool overlap = false;
11,774,908✔
357
    int  ret =
358
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
11,774,908✔
359
    if (ret != TSDB_CODE_SUCCESS) {
11,774,662!
360
      return ret;
×
361
    }
362
    if (overlap) {
11,774,662✔
363
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
705,130✔
364
    }
365
  }
366

367
  SDataBlockInfo* pBlockInfo = &pBlock->info;
16,800,412✔
368
  taosMemoryFreeClear(pBlock->pBlockAgg);
16,800,412!
369

370
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
16,800,412!
371
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
×
372
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
373
    pCost->filterOutBlocks += 1;
×
374
    pCost->totalRows += pBlock->info.rows;
×
375
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
376
    return TSDB_CODE_SUCCESS;
×
377
  } else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
16,800,412✔
378
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
1,933,737✔
379
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows,
380
           pBlockInfo->id.uid);
381
    code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
1,933,737✔
382
    pCost->skipBlocks += 1;
1,933,743✔
383
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
1,933,743✔
384
    return code;
1,933,883✔
385
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
14,866,675✔
386
    pCost->loadBlockStatis += 1;
2,376,730✔
387
    loadSMA = true;  // mark the operation of load sma;
2,376,730✔
388
    bool success = true;
2,376,730✔
389
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
2,376,730✔
390
    if (code) {
2,376,650!
391
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
392
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
393
      QUERY_CHECK_CODE(code, lino, _end);
×
394
    }
395

396
    if (success) {  // failed to load the block sma data, data block statistics does not exist, load data block instead
2,376,487✔
397
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
2,100,605✔
398
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
399
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
2,100,605✔
400
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
2,100,601✔
401
      return code;
2,103,304✔
402
    } else {
403
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
275,882✔
404
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
275,880✔
405
    }
406
  }
407

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

414
  // try to filter data block according to sma info
415
  if (pOperator->exprSupp.pFilterInfo != NULL && (!loadSMA)) {
12,765,825!
416
    bool success = true;
5,025,480✔
417
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
5,025,480✔
418
    if (code) {
5,025,502!
419
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
420
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
421
      QUERY_CHECK_CODE(code, lino, _end);
×
422
    }
423

424
    if (success) {
5,025,223✔
425
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
86,991✔
426
      bool   keep = false;
86,991✔
427
      code = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows, &keep);
86,991✔
428
      if (code) {
86,991!
429
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
430
        qError("%s failed to do filter by block sma, code:%s", GET_TASKID(pTaskInfo), tstrerror(code));
×
431
        QUERY_CHECK_CODE(code, lino, _end);
×
432
      }
433

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

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

447
  // free the sma info, since it should not be involved in *later computing process.
448
  taosMemoryFreeClear(pBlock->pBlockAgg);
12,765,565!
449

450
  // try to filter data block according to current results
451
  code = doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
12,765,566✔
452
  if (code) {
12,764,471✔
453
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
1,479✔
454
    QUERY_CHECK_CODE(code, lino, _end);
×
455
  }
456

457
  if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
12,762,992✔
458
    qDebug("%s data block skipped due to dynamic prune, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
22,491✔
459
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
460
    pCost->skipBlocks += 1;
22,491✔
461
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
22,491✔
462

463
    STableScanInfo* p1 = pOperator->info;
22,490✔
464
    if (taosHashGetSize(p1->pIgnoreTables) == taosArrayGetSize(p1->base.pTableListInfo->pTableList)) {
22,490✔
465
      *status = FUNC_DATA_REQUIRED_ALL_FILTEROUT;
4,830✔
466
    } else {
467
      *status = FUNC_DATA_REQUIRED_FILTEROUT;
17,663✔
468
    }
469
    return TSDB_CODE_SUCCESS;
22,493✔
470
  }
471

472
  pCost->totalCheckedRows += pBlock->info.rows;
12,740,501✔
473
  pCost->loadBlocks += 1;
12,740,501✔
474

475
  SSDataBlock* p = NULL;
12,740,501✔
476
  code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pTableScanInfo->dataReader, &p, NULL);
12,740,501✔
477
  if (p == NULL || code != TSDB_CODE_SUCCESS || p != pBlock) {
12,740,443!
478
    return code;
×
479
  }
480

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

491
  // restore the previous value
492
  pCost->totalRows -= pBlock->info.rows;
12,741,322✔
493

494
  if (pOperator->exprSupp.pFilterInfo != NULL) {
12,741,322✔
495
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo);
5,013,940✔
496
    QUERY_CHECK_CODE(code, lino, _end);
5,014,107✔
497

498
    int64_t st = taosGetTimestampUs();
5,013,902✔
499
    double  el = (taosGetTimestampUs() - st) / 1000.0;
5,014,036✔
500
    pTableScanInfo->readRecorder.filterTime += el;
5,014,036✔
501

502
    if (pBlock->info.rows == 0) {
5,014,036✔
503
      pCost->filterOutBlocks += 1;
93,878✔
504
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
93,878✔
505
             GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows, el);
506
    } else {
507
      qDebug("%s data block filter applied, elapsed time:%.2f ms", GET_TASKID(pTaskInfo), el);
4,920,158✔
508
    }
509
  }
510

511
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
12,741,423✔
512
  if (limitReached) {  // set operator flag is done
12,740,025✔
513
    setOperatorCompleted(pOperator);
107,435✔
514
  }
515

516
  pCost->totalRows += pBlock->info.rows;
12,741,615✔
517

518
_end:
12,741,616✔
519
  if (code != TSDB_CODE_SUCCESS) {
12,741,616✔
520
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1!
521
  }
522
  return code;
12,741,230✔
523
}
524

525
static void prepareForDescendingScan(STableScanBase* pTableScanInfo, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
191✔
526
  SET_REVERSE_SCAN_FLAG(pTableScanInfo);
191✔
527

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

534
typedef struct STableCachedVal {
535
  const char* pName;
536
  STag*       pTags;
537
} STableCachedVal;
538

539
static void freeTableCachedVal(void* param) {
2,018,322✔
540
  if (param == NULL) {
2,018,322!
541
    return;
×
542
  }
543

544
  STableCachedVal* pVal = param;
2,018,322✔
545
  taosMemoryFree((void*)pVal->pName);
2,018,322!
546
  taosMemoryFree(pVal->pTags);
2,019,711!
547
  taosMemoryFree(pVal);
2,019,730!
548
}
549

550
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
2,018,786✔
551
  int32_t          code = TSDB_CODE_SUCCESS;
2,018,786✔
552
  int32_t          lino = 0;
2,018,786✔
553
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
2,018,786!
554
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
2,019,284!
555

556
  pVal->pTags = NULL;
2,019,284✔
557
  pVal->pName = taosStrdup(pMetaReader->me.name);
2,019,284!
558
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
2,018,882!
559

560
  // only child table has tag value
561
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
2,018,882✔
562
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
2,017,672✔
563
    pVal->pTags = taosMemoryMalloc(pTag->len);
2,017,672✔
564
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
2,016,963!
565
    memcpy(pVal->pTags, pTag, pTag->len);
2,016,963✔
566
  }
567

568
  (*ppResVal) = pVal;
2,018,173✔
569

570
_end:
2,018,173✔
571
  if (code != TSDB_CODE_SUCCESS) {
2,018,173!
572
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
573
    freeTableCachedVal(pVal);
×
574
  }
575
  return code;
2,018,224✔
576
}
577

578
// const void *key, size_t keyLen, void *value
579
static void freeCachedMetaItem(const void* key, size_t keyLen, void* value, void* ud) {
2,018,282✔
580
  (void)key;
581
  (void)keyLen;
582
  (void)ud;
583
  freeTableCachedVal(value);
2,018,282✔
584
}
2,019,707✔
585

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

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

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

600
int32_t addTagPseudoColumnData(SReadHandle* pHandle, const SExprInfo* pExpr, int32_t numOfExpr, SSDataBlock* pBlock,
5,352,519✔
601
                               int32_t rows, SExecTaskInfo* pTask, STableMetaCacheInfo* pCache) {
602
  int32_t          code = TSDB_CODE_SUCCESS;
5,352,519✔
603
  int32_t          lino = 0;
5,352,519✔
604
  bool             freeReader = false;
5,352,519✔
605
  LRUHandle*       h = NULL;
5,352,519✔
606
  STableCachedVal  val = {0};
5,352,519✔
607
  SMetaReader      mr = {0};
5,352,519✔
608
  const char*      idStr = pTask->id.str;
5,352,519✔
609
  int32_t          insertRet = TAOS_LRU_STATUS_OK;
5,352,519✔
610
  STableCachedVal* pVal = NULL;
5,352,519✔
611

612
  // currently only the tbname pseudo column
613
  if (numOfExpr <= 0) {
5,352,519✔
614
    return TSDB_CODE_SUCCESS;
5,700✔
615
  }
616

617
  // todo: opt if only require the vgId and the vgVer;
618

619
  // backup the rows
620
  int32_t backupRows = pBlock->info.rows;
5,346,819✔
621
  pBlock->info.rows = rows;
5,346,819✔
622

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

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

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

645
    pHandle->api.metaReaderFn.readerReleaseLock(&mr);
1,078✔
646

647
    val.pName = mr.me.name;
1,078✔
648
    val.pTags = (STag*)mr.me.ctbEntry.pTags;
1,078✔
649

650
    freeReader = true;
1,078✔
651
  } else {
652
    pCache->metaFetch += 1;
5,345,741✔
653

654
    h = taosLRUCacheLookup(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(pBlock->info.id.uid));
5,345,741✔
655
    if (h == NULL) {
5,347,102✔
656
      pHandle->api.metaReaderFn.initReader(&mr, pHandle->vnode, META_READER_LOCK, &pHandle->api.metaFn);
2,019,640✔
657
      freeReader = true;
2,019,538✔
658
      code = pHandle->api.metaReaderFn.getEntryGetUidCache(&mr, pBlock->info.id.uid);
2,019,538✔
659
      if (code != TSDB_CODE_SUCCESS) {
2,017,876!
660
        if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
×
661
          qWarn("failed to get table meta, table may have been dropped, uid:0x%" PRIx64 ", code:%s, %s",
×
662
                pBlock->info.id.uid, tstrerror(code), idStr);
663
          // append null value before return to caller, since the caller will ignore this error code and proceed
664
          doSetNullValue(pBlock, pExpr, numOfExpr);
×
665
        } else {
666
          qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", pBlock->info.id.uid, tstrerror(code),
×
667
                 idStr);
668
        }
669
        pHandle->api.metaReaderFn.clearReader(&mr);
×
670
        return code;
×
671
      }
672

673
      pHandle->api.metaReaderFn.readerReleaseLock(&mr);
2,017,876✔
674

675
      code = createTableCacheVal(&mr, &pVal);
2,019,396✔
676
      QUERY_CHECK_CODE(code, lino, _end);
2,018,034!
677

678
      val = *pVal;
2,018,034✔
679
    } else {
680
      pCache->cacheHit += 1;
3,327,462✔
681
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
3,327,462✔
682
      val = *pValTmp;
3,327,376✔
683

684
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
3,327,376✔
685
      qTrace("release LRU cache, res %d", bRes);
3,327,652✔
686
    }
687

688
    qDebug("retrieve table meta from cache:%" PRIu64 ", hit:%" PRIu64 " miss:%" PRIu64 ", %s", pCache->metaFetch,
5,345,686✔
689
           pCache->cacheHit, (pCache->metaFetch - pCache->cacheHit), idStr);
690
  }
691

692
  for (int32_t j = 0; j < numOfExpr; ++j) {
22,252,908✔
693
    const SExprInfo* pExpr1 = &pExpr[j];
16,904,312✔
694
    int32_t          dstSlotId = pExpr1->base.resSchema.slotId;
16,904,312✔
695

696
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
16,904,312✔
697
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
16,901,786✔
698

699
    int32_t functionId = pExpr1->pExpr->_function.functionId;
16,900,701✔
700

701
    // this is to handle the tbname
702
    if (fmIsScanPseudoColumnFunc(functionId)) {
16,900,701✔
703
      int32_t fType = pExpr1->pExpr->_function.functionType;
2,520,325✔
704
      if (fType == FUNCTION_TYPE_TBNAME) {
2,520,325✔
705
        code = setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
2,445,176✔
706
        QUERY_CHECK_CODE(code, lino, _end);
2,445,622!
707
      } else if (fType == FUNCTION_TYPE_VGID) {
75,149✔
708
        code = setVgIdColData(pBlock, pColInfoData, functionId, pTask->id.vgId);
37,575✔
709
        QUERY_CHECK_CODE(code, lino, _end);
37,575!
710
      } else if (fType == FUNCTION_TYPE_VGVER) {
37,574!
711
        code = setVgVerColData(pBlock, pColInfoData, functionId, pBlock->info.version);
37,575✔
712
        QUERY_CHECK_CODE(code, lino, _end);
37,575!
713
      }
714
    } else {  // these are tags
715
      STagVal tagVal = {0};
14,381,311✔
716
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
14,381,311✔
717
      const char* p = pHandle->api.metaFn.extractTagVal(val.pTags, pColInfoData->info.type, &tagVal);
14,381,311✔
718

719
      char* data = NULL;
14,383,746✔
720
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
14,383,746✔
721
        data = tTagValToData((const STagVal*)p, false);
14,248,210✔
722
      } else {
723
        data = (char*)p;
135,536✔
724
      }
725

726
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
14,384,632✔
727
      if (isNullVal) {
14,384,633✔
728
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
132,737✔
729
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
14,251,896✔
730
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, false);
14,248,025✔
731
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
14,249,830!
732
          taosMemoryFree(data);
3,728,106!
733
        }
734
        QUERY_CHECK_CODE(code, lino, _end);
14,248,669!
735
      } else {  // todo opt for json tag
736
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
7,428✔
737
          code = colDataSetVal(pColInfoData, i, data, false);
3,556✔
738
          QUERY_CHECK_CODE(code, lino, _end);
3,557!
739
        }
740
      }
741
    }
742
  }
743

744
  // restore the rows
745
  pBlock->info.rows = backupRows;
5,348,596✔
746

747
_end:
5,348,596✔
748

749
  if (NULL != pVal) {
5,348,596✔
750
    insertRet = taosLRUCacheInsert(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(uint64_t), pVal,
2,019,183✔
751
                                   sizeof(STableCachedVal), freeCachedMetaItem, NULL, NULL, TAOS_LRU_PRIORITY_LOW, NULL);
752
    if (insertRet != TAOS_LRU_STATUS_OK) {
2,019,379!
753
      qWarn("failed to put meta into lru cache, code:%d, %s", insertRet, idStr);
×
754
    }
755
  }
756

757
  if (freeReader) {
5,348,792✔
758
    pHandle->api.metaReaderFn.clearReader(&mr);
2,020,209✔
759
  }
760
  if (code != TSDB_CODE_SUCCESS) {
5,349,006!
761
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
762
  }
763
  return code;
5,347,324✔
764
}
765

766
int32_t setTbNameColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId,
2,445,049✔
767
                         const char* name) {
768
  int32_t                     code = TSDB_CODE_SUCCESS;
2,445,049✔
769
  int32_t                     lino = 0;
2,445,049✔
770
  struct SScalarFuncExecFuncs fpSet = {0};
2,445,049✔
771
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
2,445,049✔
772
  QUERY_CHECK_CODE(code, lino, _end);
2,445,012!
773

774
  size_t len = TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE;
2,445,012✔
775
  char   buf[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
2,445,012✔
776
  STR_TO_VARSTR(buf, name)
2,445,012✔
777

778
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
2,445,012✔
779

780
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
2,446,017✔
781
  QUERY_CHECK_CODE(code, lino, _end);
2,445,851!
782

783
  code = colDataSetVal(&infoData, 0, buf, false);
2,445,851✔
784
  QUERY_CHECK_CODE(code, lino, _end);
2,445,834!
785

786
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
2,445,834✔
787
  SScalarParam param = {.columnData = pColInfoData};
2,445,834✔
788

789
  if (fpSet.process != NULL) {
2,445,834✔
790
    code = fpSet.process(&srcParam, 1, &param);
2,445,805✔
791
    QUERY_CHECK_CODE(code, lino, _end);
2,445,930!
792
  } else {
793
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
29!
794
  }
795

796
  colDataDestroy(&infoData);
2,445,959✔
797

798
_end:
2,445,648✔
799
  if (code != TSDB_CODE_SUCCESS) {
2,445,648!
800
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
801
  }
802
  return code;
2,445,637✔
803
}
804

805
int32_t setVgIdColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId, int32_t vgId) {
37,575✔
806
  int32_t                     code = TSDB_CODE_SUCCESS;
37,575✔
807
  int32_t                     lino = 0;
37,575✔
808
  struct SScalarFuncExecFuncs fpSet = {0};
37,575✔
809
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
37,575✔
810
  QUERY_CHECK_CODE(code, lino, _end);
37,575!
811

812
  SColumnInfoData infoData = createColumnInfoData(pColInfoData->info.type, pColInfoData->info.bytes, 1);
37,575✔
813

814
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
37,575✔
815
  QUERY_CHECK_CODE(code, lino, _end);
37,575!
816

817
  code = colDataSetVal(&infoData, 0, (const char*)&vgId, false);
37,575✔
818
  QUERY_CHECK_CODE(code, lino, _end);
37,575!
819

820
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
37,575✔
821
  SScalarParam param = {.columnData = pColInfoData};
37,575✔
822

823
  if (fpSet.process != NULL) {
37,575!
824
    code = fpSet.process(&srcParam, 1, &param);
37,575✔
825
    QUERY_CHECK_CODE(code, lino, _end);
37,575!
826
  } else {
827
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
×
828
  }
829

830
_end:
×
831
  colDataDestroy(&infoData);
37,575✔
832
  if (code != TSDB_CODE_SUCCESS) {
37,574!
833
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
834
  }
835
  return code;
37,575✔
836
}
837

838
int32_t setVgVerColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId, int64_t vgVer) {
37,574✔
839
  int32_t                     code = TSDB_CODE_SUCCESS;
37,574✔
840
  int32_t                     lino = 0;
37,574✔
841
  struct SScalarFuncExecFuncs fpSet = {0};
37,574✔
842
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
37,574✔
843
  QUERY_CHECK_CODE(code, lino, _end);
37,575!
844

845
  SColumnInfoData infoData = createColumnInfoData(pColInfoData->info.type, pColInfoData->info.bytes, 1);
37,575✔
846

847
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
37,575✔
848
  QUERY_CHECK_CODE(code, lino, _end);
37,575!
849

850
  code = colDataSetVal(&infoData, 0, (const char*)&vgVer, false);
37,575✔
851
  QUERY_CHECK_CODE(code, lino, _end);
37,575!
852

853
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
37,575✔
854
  SScalarParam param = {.columnData = pColInfoData};
37,575✔
855

856
  if (fpSet.process != NULL) {
37,575✔
857
    code = fpSet.process(&srcParam, 1, &param);
37,574✔
858
    QUERY_CHECK_CODE(code, lino, _end);
37,575!
859
  } else {
860
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
1!
861
  }
862

863
_end:
1✔
864
  colDataDestroy(&infoData);
37,576✔
865
  if (code != TSDB_CODE_SUCCESS) {
37,575!
866
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
867
  }
868
  return code;
37,575✔
869
}
870

871
static int32_t initNextGroupScan(STableScanInfo* pInfo, STableKeyInfo** pKeyInfo, int32_t* size) {
3,202,327✔
872
  int32_t code = TSDB_CODE_SUCCESS;
3,202,327✔
873
  int32_t lino = 0;
3,202,327✔
874
  code = tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, pKeyInfo, size);
3,202,327✔
875
  QUERY_CHECK_CODE(code, lino, _end);
3,203,098!
876

877
  pInfo->tableStartIndex = TARRAY_ELEM_IDX(pInfo->base.pTableListInfo->pTableList, *pKeyInfo);
3,203,098✔
878
  pInfo->tableEndIndex = (pInfo->tableStartIndex + (*size) - 1);
3,203,098✔
879
  pInfo->pResBlock->info.blankFill = false;
3,203,098✔
880
  taosMemoryFreeClear(pInfo->pResBlock->pBlockAgg);
3,203,098!
881

882
  if (!pInfo->needCountEmptyTable) {
3,203,098✔
883
    pInfo->countState = TABLE_COUNT_STATE_END;
2,791,399✔
884
  } else {
885
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
411,699✔
886
  }
887

888
_end:
3,203,098✔
889
  if (code != TSDB_CODE_SUCCESS) {
3,203,098!
890
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
891
  }
892
  return code;
3,203,278✔
893
}
894

895
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
11,335,614✔
896
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
11,335,614✔
897
    return;
9,648,749✔
898
  }
899
  if (pInfo->base.pTableListInfo->groupOffset) {
1,686,865✔
900
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
406✔
901
  } else {
902
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
1,686,459✔
903
    if (code != TSDB_CODE_SUCCESS) {
1,687,793✔
904
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
905,365✔
905
    }
906
  }
907
}
908

909
static SSDataBlock* getOneRowResultBlock(SExecTaskInfo* pTaskInfo, STableScanBase* pBase, SSDataBlock* pBlock,
501,955✔
910
                                         const STableKeyInfo* tbInfo) {
911
  blockDataEmpty(pBlock);
501,955✔
912
  pBlock->info.rows = 1;
502,004✔
913
  pBlock->info.id.uid = tbInfo->uid;
502,004✔
914
  pBlock->info.id.groupId = tbInfo->groupId;
502,004✔
915
  pBlock->info.blankFill = true;
502,004✔
916

917
  // only one row: set all col data to null & hasNull
918
  int32_t col_num = blockDataGetNumOfCols(pBlock);
502,004✔
919
  for (int32_t i = 0; i < col_num; ++i) {
1,896,221✔
920
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
1,394,632✔
921
    colDataSetNULL(pColInfoData, 0);
922
  }
923

924
  // set tag/tbname
925
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
501,589✔
926

927
  return pBlock;
501,812✔
928
}
929

930
static SSDataBlock* getBlockForEmptyTable(SOperatorInfo* pOperator, const STableKeyInfo* tbInfo) {
487,736✔
931
  STableScanInfo* pTableScanInfo = pOperator->info;
487,736✔
932
  SSDataBlock*    pBlock =
933
      getOneRowResultBlock(pOperator->pTaskInfo, &pTableScanInfo->base, pTableScanInfo->pResBlock, tbInfo);
487,736✔
934

935
  pOperator->resultInfo.totalRows++;
487,606✔
936
  return pBlock;
487,606✔
937
}
938

939
static int32_t doTableScanImplNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
13,945,028✔
940
  int32_t         code = TSDB_CODE_SUCCESS;
13,945,028✔
941
  int32_t         lino = 0;
13,945,028✔
942
  STableScanInfo* pTableScanInfo = pOperator->info;
13,945,028✔
943
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
13,945,028✔
944
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
13,945,028✔
945
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
13,945,028✔
946
  bool            hasNext = false;
13,945,028✔
947
  int64_t         st = taosGetTimestampUs();
13,949,051✔
948

949
  QRY_PARAM_CHECK(ppRes);
13,949,051!
950
  pBlock->info.dataLoad = false;
13,949,051✔
951

952
  while (true) {
98,068✔
953
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
14,047,119✔
954
    if (code != TSDB_CODE_SUCCESS) {
14,049,873!
955
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
×
956
      QUERY_CHECK_CODE(code, lino, _end);
1,085!
957
    }
958

959
    if (!hasNext) {
14,051,132✔
960
      break;
3,100,773✔
961
    }
962

963
    if (isTaskKilled(pTaskInfo)) {
10,956,588✔
964
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
609✔
965
      code = pTaskInfo->code;
1,080✔
966
      goto _end;
1,080✔
967
    }
968

969
    if (pOperator->status == OP_EXEC_DONE) {
10,956,647✔
970
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
1,399✔
971
      break;
1,399✔
972
    }
973

974
    // process this data block based on the probabilities
975
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
10,955,248✔
976
    if (!processThisBlock) {
10,953,909!
977
      continue;
98,068✔
978
    }
979

980
    if (pBlock->info.id.uid) {
10,953,909!
981
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
10,954,728✔
982
    }
983

984
    uint32_t status = 0;
10,955,062✔
985
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
10,955,062✔
986
    QUERY_CHECK_CODE(code, lino, _end);
10,953,349✔
987

988
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
10,953,348✔
989
      break;
4,830✔
990
    }
991

992
    // current block is filter out according to filter condition, continue load the next block
993
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
10,948,518✔
994
      continue;
98,068✔
995
    }
996

997
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
10,850,450✔
998
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
10,849,874✔
999

1000
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
10,849,874✔
1001
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
10,849,874✔
1002

1003
    (*ppRes) = pBlock;
10,849,874✔
1004
    return code;
10,849,874✔
1005
  }
1006

1007
_end:
3,101,858✔
1008
  if (code != TSDB_CODE_SUCCESS) {
3,101,858✔
1009
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,085!
1010
    pTaskInfo->code = code;
1,085✔
1011
    T_LONG_JMP(pTaskInfo->env, code);
1,085!
1012
  }
1013
  return code;
3,100,773✔
1014
}
1015

1016
static int32_t doGroupedTableScan(SOperatorInfo* pOperator, SSDataBlock** pBlock) {
14,536,443✔
1017
  int32_t         code = TSDB_CODE_SUCCESS;
14,536,443✔
1018
  int32_t         lino = 0;
14,536,443✔
1019
  STableScanInfo* pTableScanInfo = pOperator->info;
14,536,443✔
1020
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
14,536,443✔
1021
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
14,536,443✔
1022

1023
  QRY_PARAM_CHECK(pBlock);
14,536,443!
1024

1025
  // The read handle is not initialized yet, since no qualified tables exists
1026
  if (pTableScanInfo->base.dataReader == NULL || pOperator->status == OP_EXEC_DONE) {
14,536,443!
1027
    return code;
99,266✔
1028
  }
1029

1030
  // do the ascending order traverse in the first place.
1031
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
16,704,771✔
1032
    SSDataBlock* p = NULL;
12,059,178✔
1033
    code = doTableScanImplNext(pOperator, &p);
12,059,178✔
1034
    QUERY_CHECK_CODE(code, lino, _end);
12,057,923!
1035

1036
    if (p != NULL) {
12,057,923✔
1037
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
9,791,648✔
1038
      *pBlock = p;
9,791,751✔
1039
      return code;
9,791,751✔
1040
    }
1041

1042
    pTableScanInfo->scanTimes += 1;
2,266,275✔
1043
    taosHashClear(pTableScanInfo->pIgnoreTables);
2,266,275✔
1044

1045
    if (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
2,267,597✔
1046
      setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
916✔
1047
      pTableScanInfo->base.scanFlag = MAIN_SCAN;
913✔
1048
      pTableScanInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
913✔
1049
      qDebug("start to repeat ascending order scan data blocks due to query func required, %s", GET_TASKID(pTaskInfo));
913✔
1050

1051
      // do prepare for the next round table scan operation
1052
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
913✔
1053
      QUERY_CHECK_CODE(code, lino, _end);
913!
1054
    }
1055
  }
1056

1057
  int32_t total = pTableScanInfo->scanInfo.numOfAsc + pTableScanInfo->scanInfo.numOfDesc;
4,645,593✔
1058
  if (pTableScanInfo->scanTimes < total) {
4,645,593✔
1059
    if (pTableScanInfo->base.cond.order == TSDB_ORDER_ASC) {
1,888,556✔
1060
      prepareForDescendingScan(&pTableScanInfo->base, pOperator->exprSupp.pCtx, 0);
191✔
1061
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
191✔
1062
      QUERY_CHECK_CODE(code, lino, _end);
191!
1063
      qDebug("%s start to descending order scan data blocks due to query func required", GET_TASKID(pTaskInfo));
191✔
1064
    }
1065

1066
    while (pTableScanInfo->scanTimes < total) {
2,721,483✔
1067
      SSDataBlock* p = NULL;
1,888,184✔
1068
      code = doTableScanImplNext(pOperator, &p);
1,888,184✔
1069
      QUERY_CHECK_CODE(code, lino, _end);
1,890,027!
1070

1071
      if (p != NULL) {
1,890,027✔
1072
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
1,057,331✔
1073
        *pBlock = p;
1,057,503✔
1074
        return code;
1,057,503✔
1075
      }
1076

1077
      pTableScanInfo->scanTimes += 1;
832,696✔
1078
      taosHashClear(pTableScanInfo->pIgnoreTables);
832,696✔
1079

1080
      if (pTableScanInfo->scanTimes < total) {
832,705✔
1081
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
8✔
1082
        pTableScanInfo->base.scanFlag = MAIN_SCAN;
×
1083

1084
        qDebug("%s start to repeat descending order scan data blocks", GET_TASKID(pTaskInfo));
×
1085
        code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
×
1086
        QUERY_CHECK_CODE(code, lino, _end);
×
1087
      }
1088
    }
1089
  }
1090

1091
  if (pTableScanInfo->countState < TABLE_COUNT_STATE_END) {
3,590,336✔
1092
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
895,066✔
1093
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
895,066✔
1094
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
198!
1095
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
×
1096
        STableKeyInfo* pStart =
1097
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
×
1098

1099
        if (NULL == pStart) {
×
1100
          return code;
×
1101
        }
1102

1103
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
×
1104
        return code;
×
1105
      }
1106
    } else {  // group by tag + no sort
1107
      int32_t numOfTables = 0;
894,868✔
1108
      code = tableListGetSize(pTableListInfo, &numOfTables);
894,868✔
1109
      QUERY_CHECK_CODE(code, lino, _end);
894,861!
1110

1111
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
894,861✔
1112
        // get empty group, mark processed & rm from hash
1113
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
889,752✔
1114
        if (pIte != NULL) {
890,236✔
1115
          size_t        keySize = 0;
487,855✔
1116
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
487,855✔
1117
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
487,835✔
1118
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
487,835✔
1119
          markGroupProcessed(pTableScanInfo, *pGroupId);
487,823✔
1120
          *pBlock = getBlockForEmptyTable(pOperator, &info);
487,782✔
1121

1122
          return code;
487,590✔
1123
        }
1124
      }
1125
    }
1126
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
407,688✔
1127
  }
1128

1129
_end:
2,695,270✔
1130
  if (code != TSDB_CODE_SUCCESS) {
3,102,958!
1131
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1132
    pTaskInfo->code = code;
×
1133
  }
1134

1135
  return code;
3,102,958✔
1136
}
1137

1138
static int32_t createTableListInfoFromParam(SOperatorInfo* pOperator) {
1,135✔
1139
  STableScanInfo*          pInfo = pOperator->info;
1,135✔
1140
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
1,135✔
1141
  int32_t                  code = 0;
1,135✔
1142
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
1,135✔
1143
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
1,135✔
1144
  int32_t                  num = taosArrayGetSize(pParam->pUidList);
1,135✔
1145
  if (num <= 0) {
1,135!
1146
    qError("empty table scan uid list");
×
1147
    return TSDB_CODE_INVALID_PARA;
×
1148
  }
1149

1150
  qDebug("vgId:%d add total %d dynamic tables to scan, tableSeq:%d, exist num:%" PRId64 ", operator status:%d",
1,135✔
1151
         pTaskInfo->id.vgId, num, pParam->tableSeq, (int64_t)taosArrayGetSize(pListInfo->pTableList),
1152
         pOperator->status);
1153

1154
  if (pParam->tableSeq) {
1,135✔
1155
    pListInfo->oneTableForEachGroup = true;
48✔
1156
    if (taosArrayGetSize(pListInfo->pTableList) > 0) {
48!
1157
      taosHashClear(pListInfo->map);
×
1158
      taosArrayClear(pListInfo->pTableList);
×
1159
      pOperator->status = OP_EXEC_DONE;
×
1160
    }
1161
  } else {
1162
    pListInfo->oneTableForEachGroup = false;
1,087✔
1163
    pListInfo->numOfOuputGroups = 1;
1,087✔
1164
  }
1165

1166
  STableKeyInfo info = {.groupId = 0};
1,135✔
1167
  int32_t       tableIdx = 0;
1,135✔
1168
  for (int32_t i = 0; i < num; ++i) {
69,934✔
1169
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
68,798✔
1170
    if (!pUid) {
68,747!
1171
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1172
      return terrno;
×
1173
    }
1174

1175
    if (taosHashPut(pListInfo->map, pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
68,747✔
1176
      if (TSDB_CODE_DUP_KEY == terrno) {
65,060✔
1177
        continue;
64,996✔
1178
      }
1179
      return terrno;
1✔
1180
    }
1181

1182
    info.uid = *pUid;
3,813✔
1183
    void* p = taosArrayPush(pListInfo->pTableList, &info);
3,813✔
1184
    if (p == NULL) {
3,852!
1185
      return terrno;
×
1186
    }
1187

1188
    tableIdx++;
3,852✔
1189
    qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
3,852✔
1190
  }
1191

1192
  return code;
1,136✔
1193
}
1194

1195
static int32_t doInitReader(STableScanInfo* pInfo, SExecTaskInfo* pTaskInfo, SStorageAPI* pAPI, int32_t* pNum,
3,203,167✔
1196
                            STableKeyInfo** pList) {
1197
  const char* idStr = GET_TASKID(pTaskInfo);
3,203,167✔
1198
  int32_t     code = initNextGroupScan(pInfo, pList, pNum);
3,203,167✔
1199
  if (code) {
3,202,358!
1200
    qError("%s failed to init groupScan Info, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
×
1201
    return code;
×
1202
  }
1203

1204
  if (pInfo->base.dataReader != NULL) {
3,204,039!
1205
    qError("%s tsdb reader should be null", idStr);
×
1206
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1207
  }
1208

1209
  code = pAPI->tsdReader.tsdReaderOpen(pInfo->base.readHandle.vnode, &pInfo->base.cond, *pList, *pNum, pInfo->pResBlock,
3,204,039✔
1210
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
3,204,039✔
1211
  if (code) {
3,209,565✔
1212
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
16!
1213
  }
1214

1215
  return code;
3,209,069✔
1216
}
1217

1218
static int32_t createVTableScanInfoFromParam(SOperatorInfo* pOperator) {
×
1219
  int32_t                  code = 0;
×
1220
  int32_t                  lino = 0;
×
1221
  STableScanInfo*          pInfo = pOperator->info;
×
1222
  SExecTaskInfo*           pTaskInfo = pOperator->pTaskInfo;
×
1223
  SStorageAPI*             pAPI = &pTaskInfo->storageAPI;
×
1224
  STableListInfo*          pListInfo = pInfo->base.pTableListInfo;
×
1225
  STableScanOperatorParam* pParam = (STableScanOperatorParam*)pOperator->pOperatorGetParam->value;
×
1226
  SMetaReader              orgTable = {0};
×
1227
  SMetaReader              superTable = {0};
×
1228
  SSchemaWrapper*          schema = NULL;
×
1229
  SArray*                  pColArray = NULL;
×
1230
  SArray*                  pBlockColArray = NULL;
×
1231
  int32_t                  num = 0;
×
1232
  STableKeyInfo*           pList = NULL;
×
1233

1234
  cleanupQueryTableDataCond(&pInfo->base.cond);
×
1235
  if (pAPI->tsdReader.tsdReaderClose) {
×
1236
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
×
1237
  }
1238

1239
  pAPI->metaReaderFn.initReader(&orgTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
×
1240
  code = pAPI->metaReaderFn.getTableEntryByName(&orgTable, strstr(pParam->pOrgTbInfo->tbName, ".") + 1);
×
1241
  QUERY_CHECK_CODE(code, lino, _return);
×
1242
  switch (orgTable.me.type) {
×
1243
    case TSDB_CHILD_TABLE:
×
1244
      pAPI->metaReaderFn.initReader(&superTable, pInfo->base.readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
×
1245
      code = pAPI->metaReaderFn.getTableEntryByUid(&superTable, orgTable.me.ctbEntry.suid);
×
1246
      QUERY_CHECK_CODE(code, lino, _return);
×
1247
      schema = &superTable.me.stbEntry.schemaRow;
×
1248
      break;
×
1249
    case TSDB_NORMAL_TABLE:
×
1250
      schema = &orgTable.me.ntbEntry.schemaRow;
×
1251
      break;
×
1252
    default:
×
1253
      qError("invalid table type:%d", orgTable.me.type);
×
1254
      return TSDB_CODE_INVALID_PARA;
×
1255
      break;
1256
  }
1257

1258
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
×
1259
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
×
1260
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(int32_t));
×
1261
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
×
1262

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

1267
  for (int32_t i = 0; i < taosArrayGetSize(pParam->pOrgTbInfo->colMap); ++i) {
×
1268
    SColIdNameKV *kv = taosArrayGet(pParam->pOrgTbInfo->colMap, i);
×
1269
    for (int32_t j = 0; j < schema->nCols; j++) {
×
1270
      if (strncmp(kv->colName, schema->pSchema[j].name, strlen(schema->pSchema[j].name)) == 0) {
×
1271
        SColIdPair pPair = {.vtbColId = kv->colId, .orgColId = (col_id_t)(j + 1)};
×
1272
        QUERY_CHECK_NULL(taosArrayPush(pColArray, &pPair), code, lino, _return, terrno);
×
1273
        break;
×
1274
      }
1275
    }
1276
  }
1277

1278
  for (int32_t i = 0; i < taosArrayGetSize(pColArray); i++) {
×
1279
    SColIdPair *pPair = (SColIdPair*)taosArrayGet(pColArray, i);
×
1280
    for (int32_t j = 0; j < taosArrayGetSize(pInfo->base.matchInfo.pList); j++) {
×
1281
      SColMatchItem *pItem = taosArrayGet(pInfo->base.matchInfo.pList, j);
×
1282
      if (pItem->colId == pPair->vtbColId) {
×
1283
        SColIdPair tmpPair = {.orgColId = pPair->orgColId, .vtbColId = pItem->dstSlotId};
×
1284
        QUERY_CHECK_NULL(taosArrayPush(pBlockColArray, &tmpPair), code, lino, _return, terrno);
×
1285
        break;
×
1286
      }
1287
    }
1288
  }
1289

1290
  if (pInfo->pResBlock) {
×
1291
    blockDataDestroy(pInfo->pResBlock);
×
1292
    pInfo->pResBlock = NULL;
×
1293
  }
1294
  code = createOneDataBlockWithColArray(pInfo->pOrgBlock, pBlockColArray, &pInfo->pResBlock);
×
1295
  QUERY_CHECK_CODE(code, lino, _return);
×
1296
  code = initQueryTableDataCondWithColArray(&pInfo->base.cond, &pInfo->base.orgCond, &pInfo->base.readHandle, pColArray);
×
1297
  QUERY_CHECK_CODE(code, lino, _return);
×
1298
  pInfo->base.cond.twindows.skey = pParam->window.ekey + 1;
×
1299
  pInfo->base.cond.suid = orgTable.me.type == TSDB_CHILD_TABLE ? superTable.me.uid : 0;
×
1300
  pInfo->currentGroupId = 0;
×
1301
  pInfo->base.dataReader = NULL;
×
1302
  pInfo->ignoreTag = true;
×
1303

1304
  pListInfo->oneTableForEachGroup = true;
×
1305
  taosHashClear(pListInfo->map);
×
1306
  taosArrayClear(pListInfo->pTableList);
×
1307

1308
  uint64_t      pUid = orgTable.me.uid;
×
1309
  STableKeyInfo info = {.groupId = 0, .uid = pUid};
×
1310
  int32_t       tableIdx = 0;
×
1311
  code = taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t));
×
1312
  QUERY_CHECK_CODE(code, lino, _return);
×
1313
  QUERY_CHECK_NULL(taosArrayPush(pListInfo->pTableList, &info), code, lino, _return, terrno);
×
1314
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
×
1315

1316
  pOperator->status = OP_OPENED;
×
1317

1318
  taosRLockLatch(&pTaskInfo->lock);
×
1319
  code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
×
1320
  taosRUnLockLatch(&pTaskInfo->lock);
×
1321
  QUERY_CHECK_CODE(code, lino, _return);
×
1322

1323
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
×
1324
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1325
  }
1326

1327
  pInfo->currentGroupId = -1;
×
1328
_return:
×
1329
  if (code) {
×
1330
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1331
  }
1332
  taosArrayDestroy(pColArray);
×
1333
  taosArrayDestroy(pBlockColArray);
×
1334
  pAPI->metaReaderFn.clearReader(&superTable);
×
1335
  pAPI->metaReaderFn.clearReader(&orgTable);
×
1336
  return code;
×
1337
}
1338

1339
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
3,183,980✔
1340
  int32_t         code = TSDB_CODE_SUCCESS;
3,183,980✔
1341
  int32_t         lino = 0;
3,183,980✔
1342
  STableScanInfo* pInfo = pOperator->info;
3,183,980✔
1343
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
3,183,980✔
1344
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
3,183,980✔
1345
  int32_t         numOfTables = 0;
3,183,980✔
1346

1347
  QRY_PARAM_CHECK(pResult);
3,183,980!
1348

1349
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
3,183,980✔
1350
  QUERY_CHECK_CODE(code, lino, _end);
3,184,594!
1351

1352
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
3,184,594✔
1353
    setOperatorCompleted(pOperator);
3,183,471✔
1354
    if (pOperator->dynamicTask) {
3,183,730✔
1355
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
1,125✔
1356
      taosHashClear(pInfo->base.pTableListInfo->map);
1,125✔
1357
    }
1358
    return code;
3,183,816✔
1359
  }
1360

1361
  // reset value for the next group data output
1362
  pOperator->status = OP_OPENED;
753✔
1363
  resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
753✔
1364

1365
  int32_t        num = 0;
919✔
1366
  STableKeyInfo* pList = NULL;
919✔
1367
  code = initNextGroupScan(pInfo, &pList, &num);
919✔
1368
  QUERY_CHECK_CODE(code, lino, _end);
919!
1369

1370
  code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
919✔
1371
  QUERY_CHECK_CODE(code, lino, _end);
919!
1372

1373
  code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
919✔
1374
  QUERY_CHECK_CODE(code, lino, _end);
919!
1375
  pInfo->scanTimes = 0;
919✔
1376

1377
  code = doGroupedTableScan(pOperator, pResult);
919✔
1378
  QUERY_CHECK_CODE(code, lino, _end);
918!
1379

1380
  if (*pResult != NULL) {
918✔
1381
    if (pOperator->dynamicTask) {
807✔
1382
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
24✔
1383
    }
1384
    return code;
807✔
1385
  }
1386

1387
_end:
111✔
1388
  if (code != TSDB_CODE_SUCCESS) {
111!
1389
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1390
    pTaskInfo->code = code;
×
1391
  }
1392
  return code;
111✔
1393
}
1394

1395
static int32_t groupSeqTableScan(SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
14,485,770✔
1396
  int32_t         code = TSDB_CODE_SUCCESS;
14,485,770✔
1397
  int32_t         lino = 0;
14,485,770✔
1398
  STableScanInfo* pInfo = pOperator->info;
14,485,770✔
1399
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
14,485,770✔
1400
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
14,485,770✔
1401
  int32_t         num = 0;
14,485,770✔
1402
  STableKeyInfo*  pList = NULL;
14,485,770✔
1403
  SSDataBlock*    pResult = NULL;
14,485,770✔
1404
  const char*     idStr = GET_TASKID(pTaskInfo);
14,485,770✔
1405

1406
  QRY_PARAM_CHECK(pResBlock);
14,485,770!
1407

1408
  if (pInfo->currentGroupId == -1) {
14,485,770✔
1409
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
3,204,288!
1410
      setOperatorCompleted(pOperator);
×
1411
      return code;
×
1412
    }
1413

1414
    taosRLockLatch(&pTaskInfo->lock);
3,203,418✔
1415
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
3,212,332✔
1416
    taosRUnLockLatch(&pTaskInfo->lock);
3,209,000✔
1417
    QUERY_CHECK_CODE(code, lino, _end);
3,212,371✔
1418

1419
    if (pInfo->filesetDelimited) {
3,212,355✔
1420
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
1,099✔
1421
    }
1422

1423
    if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
3,212,354!
1424
      pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1425
    }
1426
  }
1427

1428
  code = doGroupedTableScan(pOperator, &pResult);
14,493,836✔
1429
  QUERY_CHECK_CODE(code, lino, _end);
14,497,521!
1430

1431
  if (pResult != NULL) {
14,497,521✔
1432
    if (pOperator->dynamicTask) {
11,315,388✔
1433
      pResult->info.id.groupId = pResult->info.id.uid;
3,437✔
1434
    }
1435

1436
    *pResBlock = pResult;
11,315,388✔
1437
    return code;
11,315,388✔
1438
  }
1439

1440
  while (true) {
1441
    code = startNextGroupScan(pOperator, &pResult);
3,182,244✔
1442
    QUERY_CHECK_CODE(code, lino, _end);
3,184,708!
1443

1444
    if (pResult || pOperator->status == OP_EXEC_DONE) {
3,184,708✔
1445
      *pResBlock = pResult;
3,184,597✔
1446
      return code;
3,184,597✔
1447
    }
1448
  }
1449

1450
_end:
16✔
1451
  if (code != TSDB_CODE_SUCCESS) {
16!
1452
    qError("%s %s failed at line %d since %s", idStr, __func__, lino, tstrerror(code));
16!
1453
    pTaskInfo->code = code;
16✔
1454
  }
1455

1456
  return code;
16✔
1457
}
1458

1459
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
14,506,812✔
1460
  int32_t         code = TSDB_CODE_SUCCESS;
14,506,812✔
1461
  int32_t         lino = 0;
14,506,812✔
1462
  STableScanInfo* pInfo = pOperator->info;
14,506,812✔
1463
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
14,506,812✔
1464
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
14,506,812✔
1465
  QRY_PARAM_CHECK(ppRes);
14,506,812!
1466

1467
  if (pOperator->pOperatorGetParam) {
14,506,812✔
1468
    pOperator->dynamicTask = true;
1,135✔
1469
    if (isDynVtbScan(pOperator)) {
1,135!
1470
      code = createVTableScanInfoFromParam(pOperator);
×
1471

1472
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
×
1473
      pOperator->pOperatorGetParam = NULL;
×
1474
      QUERY_CHECK_CODE(code, lino, _end);
×
1475

1476
      SSDataBlock* result = NULL;
×
1477
      while (true) {
1478
        code = startNextGroupScan(pOperator, &result);
×
1479
        QUERY_CHECK_CODE(code, lino, _end);
×
1480

1481
        if (result || pOperator->status == OP_EXEC_DONE) {
×
1482
          SSDataBlock* res = NULL;
×
1483
          if (result) {
×
1484
            code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, &res);
×
1485
            QUERY_CHECK_CODE(code, lino, _end);
×
1486
            pInfo->pResBlock = res;
×
1487
            blockDataDestroy(result);
×
1488
          }
1489
          (*ppRes) = res;
×
1490
          return code;
×
1491
        }
1492
      }
1493
    } else {
1494
      code = createTableListInfoFromParam(pOperator);
1,135✔
1495
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
1,135✔
1496
      pOperator->pOperatorGetParam = NULL;
678✔
1497
      QUERY_CHECK_CODE(code, lino, _end);
678!
1498

1499
      if (pOperator->status == OP_EXEC_DONE) {
678✔
1500
        pInfo->currentGroupId = -1;
24✔
1501
        pOperator->status = OP_OPENED;
24✔
1502
        SSDataBlock* result = NULL;
24✔
1503

1504
        while (true) {
1505
          code = startNextGroupScan(pOperator, &result);
24✔
1506
          QUERY_CHECK_CODE(code, lino, _end);
24!
1507

1508
          if (result || pOperator->status == OP_EXEC_DONE) {
24!
1509
            (*ppRes) = result;
24✔
1510
            return code;
24✔
1511
          }
1512
        }
1513
      }
1514
    }
1515
  }
1516

1517
  // scan table one by one sequentially
1518
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
14,506,331✔
1519
    int32_t       numOfTables = 0;
20,332✔
1520
    STableKeyInfo tInfo = {0};
20,332✔
1521
    pInfo->countState = TABLE_COUNT_STATE_END;
20,332✔
1522

1523
    while (1) {
19,008✔
1524
      SSDataBlock* result = NULL;
39,340✔
1525
      code = doGroupedTableScan(pOperator, &result);
39,340✔
1526
      QUERY_CHECK_CODE(code, lino, _end);
39,330!
1527

1528
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
39,330!
1529
        (*ppRes) = result;
20,098✔
1530
        return code;
20,299✔
1531
      }
1532

1533
      // if no data, switch to next table and continue scan
1534
      pInfo->currentTable++;
19,231✔
1535

1536
      taosRLockLatch(&pTaskInfo->lock);
19,231✔
1537
      numOfTables = 0;
19,254✔
1538
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
19,254✔
1539
      if (code != TSDB_CODE_SUCCESS) {
19,252!
1540
        taosRUnLockLatch(&pTaskInfo->lock);
×
1541
        TSDB_CHECK_CODE(code, lino, _end);
×
1542
      }
1543

1544
      if (pInfo->currentTable >= numOfTables) {
19,252✔
1545
        qDebug("all table checked in table list, total:%d, return NULL, %s", numOfTables, GET_TASKID(pTaskInfo));
201✔
1546
        taosRUnLockLatch(&pTaskInfo->lock);
201✔
1547
        (*ppRes) = NULL;
201✔
1548
        return code;
201✔
1549
      }
1550

1551
      STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
19,051✔
1552
      if (!tmp) {
19,048!
1553
        taosRUnLockLatch(&pTaskInfo->lock);
×
1554
        (*ppRes) = NULL;
×
1555
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1556
      }
1557

1558
      tInfo = *tmp;
19,048✔
1559
      taosRUnLockLatch(&pTaskInfo->lock);
19,048✔
1560

1561
      code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, &tInfo, 1);
19,053✔
1562
      QUERY_CHECK_CODE(code, lino, _end);
19,052!
1563
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d/%d %s", tInfo.uid, numOfTables,
19,052✔
1564
             pInfo->currentTable, numOfTables, GET_TASKID(pTaskInfo));
1565

1566
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
19,052✔
1567
      QUERY_CHECK_CODE(code, lino, _end);
19,008!
1568
      pInfo->scanTimes = 0;
19,008✔
1569
    }
1570
  } else {  // scan table group by group sequentially
1571
    code = groupSeqTableScan(pOperator, ppRes);
14,485,999✔
1572
    QUERY_CHECK_CODE(code, lino, _end);
14,497,377✔
1573
  }
1574

1575
_end:
14,497,361✔
1576
  if (code != TSDB_CODE_SUCCESS) {
14,497,377✔
1577
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
16!
1578
    pTaskInfo->code = code;
16✔
1579
    T_LONG_JMP(pTaskInfo->env, code);
16!
1580
  }
1581

1582
  return code;
14,497,361✔
1583
}
1584

1585
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
155,074✔
1586
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
155,074!
1587
  if (!pRecorder) {
155,102!
1588
    return terrno;
×
1589
  }
1590
  STableScanInfo* pTableScanInfo = pOptr->info;
155,102✔
1591
  *pRecorder = pTableScanInfo->base.readRecorder;
155,102✔
1592
  *pOptrExplain = pRecorder;
155,102✔
1593
  *len = sizeof(SFileBlockLoadRecorder);
155,102✔
1594
  return 0;
155,102✔
1595
}
1596

1597
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
5,280,736✔
1598
  cleanupQueryTableDataCond(&pBase->cond);
5,280,736✔
1599
  cleanupQueryTableDataCond(&pBase->orgCond);
5,281,647✔
1600

1601
  if (pAPI->tsdReaderClose) {
5,281,764!
1602
    pAPI->tsdReaderClose(pBase->dataReader);
5,281,764✔
1603
  }
1604
  pBase->dataReader = NULL;
5,281,649✔
1605

1606
  if (pBase->matchInfo.pList != NULL) {
5,281,649!
1607
    taosArrayDestroy(pBase->matchInfo.pList);
5,281,826✔
1608
  }
1609

1610
  tableListDestroy(pBase->pTableListInfo);
5,281,831✔
1611
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
5,281,915✔
1612
  cleanupExprSupp(&pBase->pseudoSup);
5,281,958✔
1613
}
5,281,946✔
1614

1615
static void destroyTableScanOperatorInfo(void* param) {
3,253,834✔
1616
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
3,253,834✔
1617
  blockDataDestroy(pTableScanInfo->pResBlock);
3,253,834✔
1618
  blockDataDestroy(pTableScanInfo->pOrgBlock);
3,254,646✔
1619
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
3,254,634✔
1620
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
3,254,152✔
1621
  taosMemoryFreeClear(param);
3,254,291!
1622
}
3,254,576✔
1623

1624
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
3,247,422✔
1625
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
3,247,422!
1626
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
355,686✔
1627
    for (int32_t i = 0; i < numOfCols; ++i) {
1,008,150✔
1628
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
652,486✔
1629
      if (pCol) {
652,472!
1630
        pCol->info.noData = true;
652,487✔
1631
      }
1632
    }
1633
  }
1634
}
3,247,400✔
1635

1636
int32_t createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
3,250,300✔
1637
                                    STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
1638
                                    SOperatorInfo** pOptrInfo) {
1639
  QRY_PARAM_CHECK(pOptrInfo);
3,250,300!
1640

1641
  int32_t         code = TSDB_CODE_SUCCESS;
3,250,300✔
1642
  int32_t         lino = 0;
3,250,300✔
1643
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
3,250,300!
1644
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
3,251,582!
1645
  if (pInfo == NULL || pOperator == NULL) {
3,251,803!
1646
    code = terrno;
×
1647
    goto _error;
×
1648
  }
1649

1650
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
3,252,029✔
1651
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
3,252,029✔
1652

1653
  int32_t numOfCols = 0;
3,252,029✔
1654
  code =
1655
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
3,252,029✔
1656
  QUERY_CHECK_CODE(code, lino, _error);
3,250,407!
1657

1658
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
3,250,407✔
1659
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
3,251,273✔
1660
  QUERY_CHECK_CODE(code, lino, _error);
3,253,348!
1661

1662
  if (pScanNode->pScanPseudoCols != NULL) {
3,253,348✔
1663
    SExprSupp* pSup = &pInfo->base.pseudoSup;
726,817✔
1664
    pSup->pExprInfo = NULL;
726,817✔
1665
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
726,817✔
1666
    QUERY_CHECK_CODE(code, lino, _error);
726,488!
1667

1668
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
726,488✔
1669
                                      &pTaskInfo->storageAPI.functionStore);
1670
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
725,877✔
1671
  }
1672

1673
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
3,248,606✔
1674
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
3,248,606✔
1675

1676
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
3,248,606✔
1677
  pInfo->base.readHandle = *readHandle;
3,248,895✔
1678
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
3,248,895✔
1679

1680
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
3,248,895✔
1681
  pInfo->sample.seed = taosGetTimestampSec();
3,248,895✔
1682

1683
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
3,249,612✔
1684
  initResultSizeInfo(&pOperator->resultInfo, 4096);
3,249,612✔
1685
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
3,246,904✔
1686
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
3,248,782✔
1687
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
3,250,013!
1688

1689
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
3,250,013✔
1690
  QUERY_CHECK_CODE(code, lino, _error);
3,248,907!
1691

1692
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
3,248,907✔
1693
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
3,248,907!
1694
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
×
1695
    pInfo->pResBlock = NULL;
×
1696
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
×
1697
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
×
1698
  }
1699

1700
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
3,248,907✔
1701
  QUERY_CHECK_CODE(code, lino, _error);
3,245,607✔
1702

1703
  pInfo->currentGroupId = -1;
3,245,385✔
1704

1705
  pInfo->tableEndIndex = -1;
3,245,385✔
1706
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
3,245,385✔
1707
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
3,245,385✔
1708

1709
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
3,245,385✔
1710
                  pTaskInfo);
1711
  pOperator->exprSupp.numOfExprs = numOfCols;
3,244,440✔
1712

1713
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
3,244,440✔
1714
  pInfo->ignoreTag = false;
3,244,440✔
1715

1716
  pInfo->base.pTableListInfo = pTableListInfo;
3,244,440✔
1717
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
3,244,440✔
1718
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
3,252,751!
1719
    code = terrno;
×
1720
    QUERY_CHECK_CODE(code, lino, _error);
×
1721
  }
1722

1723
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
3,252,751✔
1724

1725
  taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
3,252,751✔
1726
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
3,253,343✔
1727
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
1728

1729
  // for non-blocking operator, the open cost is always 0
1730
  pOperator->cost.openCost = 0;
3,249,410✔
1731
  *pOptrInfo = pOperator;
3,249,410✔
1732
  return TSDB_CODE_SUCCESS;
3,249,410✔
1733

1734
_error:
222✔
1735
  if (pInfo != NULL) {
222!
1736
    pInfo->base.pTableListInfo = NULL;  // this attribute will be destroy outside of this function
222✔
1737
    destroyTableScanOperatorInfo(pInfo);
222✔
1738
  }
1739

1740
  if (pOperator != NULL) {
222!
1741
    pOperator->info = NULL;
222✔
1742
    destroyOperator(pOperator);
222✔
1743
  }
1744
  pTaskInfo->code = code;
222✔
1745
  return code;
222✔
1746
}
1747

1748
int32_t createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
×
1749
  QRY_PARAM_CHECK(pOptrInfo);
×
1750

1751
  int32_t         code = 0;
×
1752
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
×
1753
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
1754
  if (pInfo == NULL || pOperator == NULL) {
×
1755
    code = terrno;
×
1756
    goto _end;
×
1757
  }
1758

1759
  pInfo->base.dataReader = pReadHandle;
×
1760
  //  pInfo->prevGroupId       = -1;
1761

1762
  setOperatorInfo(pOperator, "TableSeqScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN, false, OP_NOT_OPENED,
×
1763
                  pInfo, pTaskInfo);
1764
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanImplNext, NULL, NULL, optrDefaultBufFn, NULL,
×
1765
                                         optrDefaultGetNextExtFn, NULL);
1766
  *pOptrInfo = pOperator;
×
1767
  return code;
×
1768

1769
_end:
×
1770
  if (pInfo != NULL) {
×
1771
    taosMemoryFree(pInfo);
×
1772
  }
1773

1774
  if (pOperator != NULL) {
×
1775
    taosMemoryFree(pOperator);
×
1776
  }
1777

1778
  pTaskInfo->code = code;
×
1779
  return code;
×
1780
}
1781

1782
FORCE_INLINE void doClearBufferedBlocks(SStreamScanInfo* pInfo) {
310✔
1783
  qDebug("clear buff blocks:%d", (int32_t)taosArrayGetSize(pInfo->pBlockLists));
45,457!
1784
  taosArrayClear(pInfo->pBlockLists);
45,457✔
1785
  pInfo->validBlockIndex = 0;
45,456✔
1786
}
45,146✔
1787

1788
static bool isSessionWindow(SStreamScanInfo* pInfo) {
1,885✔
1789
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION ||
3,253✔
1790
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION ||
3,166✔
1791
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION;
1,281!
1792
}
1793

1794
static bool isStateWindow(SStreamScanInfo* pInfo) {
1,339✔
1795
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE ||
1,718✔
1796
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_EVENT;
379✔
1797
}
1798

1799
static bool isIntervalWindow(SStreamScanInfo* pInfo) {
11,933✔
1800
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL ||
20,489✔
1801
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL ||
8,556✔
1802
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_MID_INTERVAL ||
7,293!
1803
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL ||
7,293!
1804
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_INTERVAL ||
7,293✔
1805
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_INTERVAL ||
25,867✔
1806
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_INTERVAL;
5,378!
1807
}
1808

1809
static bool isSignleIntervalWindow(SStreamScanInfo* pInfo) {
30,227,235✔
1810
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL;
30,227,235✔
1811
}
1812

1813
static bool isSlidingWindow(SStreamScanInfo* pInfo) {
7,844✔
1814
  return isIntervalWindow(pInfo) && pInfo->interval.interval != pInfo->interval.sliding;
7,844✔
1815
}
1816

1817
static bool isCountSlidingWindow(SStreamScanInfo* pInfo) {
7,528✔
1818
  return pInfo->windowSup.pStreamAggSup &&
12,600✔
1819
         (pInfo->windowSup.pStreamAggSup->windowCount != pInfo->windowSup.pStreamAggSup->windowSliding);
5,072✔
1820
}
1821

1822
static bool isCountWindow(SStreamScanInfo* pInfo) {
290✔
1823
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_COUNT;
290✔
1824
}
1825

1826
static bool isTimeSlice(SStreamScanInfo* pInfo) {
165✔
1827
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERP_FUNC;
165✔
1828
}
1829

1830
static void setGroupId(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t groupColIndex, int32_t rowIndex) {
7,844✔
1831
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, groupColIndex);
7,844✔
1832
  uint64_t*        groupCol = (uint64_t*)pColInfo->pData;
7,844✔
1833
  if (colDataIsNull_s(pColInfo, rowIndex)) {
15,688✔
1834
    pInfo->igCheckGroupId = true;
4,176✔
1835
  } else {
1836
    pInfo->groupId = groupCol[rowIndex];
3,668✔
1837
  }
1838
}
7,844✔
1839

1840
void resetTableScanInfo(STableScanInfo* pTableScanInfo, STimeWindow* pWin, int64_t ver) {
7,855✔
1841
  pTableScanInfo->base.cond.twindows = *pWin;
7,855✔
1842
  pTableScanInfo->base.cond.startVersion = 0;
7,855✔
1843
  pTableScanInfo->base.cond.endVersion = ver;
7,855✔
1844
  pTableScanInfo->scanTimes = 0;
7,855✔
1845
  pTableScanInfo->currentGroupId = -1;
7,855✔
1846
  pTableScanInfo->tableEndIndex = -1;
7,855✔
1847
  pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
7,855✔
1848
  pTableScanInfo->base.dataReader = NULL;
7,854✔
1849
  pTableScanInfo->scanMode = TABLE_SCAN__BLOCK_ORDER;
7,854✔
1850
}
7,854✔
1851

1852
SSDataBlock* readPreVersionData(SOperatorInfo* pTableScanOp, uint64_t tbUid, TSKEY startTs, TSKEY endTs,
715✔
1853
                                int64_t maxVersion) {
1854
  int32_t       code = TSDB_CODE_SUCCESS;
715✔
1855
  int32_t       lino = 0;
715✔
1856
  STableKeyInfo tblInfo = {.uid = tbUid, .groupId = 0};
715✔
1857

1858
  STableScanInfo*     pTableScanInfo = pTableScanOp->info;
715✔
1859
  SQueryTableDataCond cond = pTableScanInfo->base.cond;
715✔
1860

1861
  cond.startVersion = -1;
715✔
1862
  cond.endVersion = maxVersion;
715✔
1863
  cond.twindows = (STimeWindow){.skey = startTs, .ekey = endTs};
715✔
1864

1865
  SExecTaskInfo* pTaskInfo = pTableScanOp->pTaskInfo;
715✔
1866
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
715✔
1867

1868
  SSDataBlock* pBlock = pTableScanInfo->pResBlock;
715✔
1869
  STsdbReader* pReader = NULL;
715✔
1870
  code = pAPI->tsdReader.tsdReaderOpen(pTableScanInfo->base.readHandle.vnode, &cond, &tblInfo, 1, pBlock,
715✔
1871
                                       (void**)&pReader, GET_TASKID(pTaskInfo), NULL);
715✔
1872
  QUERY_CHECK_CODE(code, lino, _end);
715!
1873

1874
  bool hasNext = false;
715✔
1875
  code = pAPI->tsdReader.tsdNextDataBlock(pReader, &hasNext);
715✔
1876
  QUERY_CHECK_CODE(code, lino, _end);
715!
1877

1878
  if (hasNext) {
715✔
1879
    SSDataBlock* p = NULL;
681✔
1880
    code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pReader, &p, NULL);
681✔
1881
    QUERY_CHECK_CODE(code, lino, _end);
681!
1882

1883
    code = doSetTagColumnData(&pTableScanInfo->base, pBlock, pTaskInfo, pBlock->info.rows);
681✔
1884
    QUERY_CHECK_CODE(code, lino, _end);
681!
1885

1886
    pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
681✔
1887
  }
1888

1889
_end:
34✔
1890
  pAPI->tsdReader.tsdReaderClose(pReader);
715✔
1891
  qDebug("retrieve prev rows:%" PRId64 ", skey:%" PRId64 ", ekey:%" PRId64 " uid:%" PRIu64 ", max ver:%" PRId64
715✔
1892
         ", suid:%" PRIu64,
1893
         pBlock->info.rows, startTs, endTs, tbUid, maxVersion, cond.suid);
1894
  if (code != TSDB_CODE_SUCCESS) {
715!
1895
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1896
    terrno = code;
×
1897
    return NULL;
×
1898
  }
1899

1900
  return pBlock->info.rows > 0 ? pBlock : NULL;
715✔
1901
}
1902

1903
bool comparePrimaryKey(SColumnInfoData* pCol, int32_t rowId, void* pVal) {
2✔
1904
  // coverity scan
1905
  if (!pVal || !pCol) {
2!
1906
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
1907
    return false;
×
1908
  }
1909
  void* pData = colDataGetData(pCol, rowId);
2!
1910
  if (IS_VAR_DATA_TYPE(pCol->info.type)) {
2!
1911
    int32_t colLen = varDataLen(pData);
×
1912
    int32_t keyLen = varDataLen(pVal);
×
1913
    if (pCol->info.type == TSDB_DATA_TYPE_JSON) {
×
1914
      colLen = getJsonValueLen(pData);
×
1915
      keyLen = getJsonValueLen(pVal);
×
1916
    }
1917

1918
    if (colLen == keyLen && memcmp(pData, pVal, colLen) == 0) {
×
1919
      return true;
×
1920
    }
1921
  } else {
1922
    if (memcmp(pData, pVal, pCol->info.bytes) == 0) {
2!
1923
      return true;
2✔
1924
    }
1925
  }
1926
  return false;
×
1927
}
1928

1929
static bool hasPrimaryKeyCol(SStreamScanInfo* pInfo) { return pInfo->primaryKeyIndex != -1; }
51,081,068✔
1930

1931
static uint64_t getGroupIdByCol(SStreamScanInfo* pInfo, uint64_t uid, TSKEY ts, int64_t maxVersion, void* pVal) {
646✔
1932
  SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, uid, ts, ts, maxVersion);
646✔
1933
  if (!pPreRes || pPreRes->info.rows == 0) {
646!
1934
    if (terrno != TSDB_CODE_SUCCESS) {
34!
1935
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1936
    }
1937
    return 0;
34✔
1938
  }
1939

1940
  int32_t rowId = 0;
612✔
1941
  if (hasPrimaryKeyCol(pInfo)) {
612✔
1942
    SColumnInfoData* pPkCol = taosArrayGet(pPreRes->pDataBlock, pInfo->primaryKeyIndex);
2✔
1943
    for (; rowId < pPreRes->info.rows; rowId++) {
2!
1944
      if (comparePrimaryKey(pPkCol, rowId, pVal)) {
2!
1945
        break;
2✔
1946
      }
1947
    }
1948
  }
1949
  if (rowId >= pPreRes->info.rows) {
612!
1950
    qInfo("===stream===read preversion data of primary key failed. ts:%" PRId64 ",version:%" PRId64, ts, maxVersion);
×
1951
    return 0;
×
1952
  }
1953
  return calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, rowId);
612✔
1954
}
1955

1956
static uint64_t getGroupIdByUid(SStreamScanInfo* pInfo, uint64_t uid) {
2,574✔
1957
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
2,574✔
1958
  return tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, uid);
2,574✔
1959
}
1960

1961
static uint64_t getGroupIdByData(SStreamScanInfo* pInfo, uint64_t uid, TSKEY ts, int64_t maxVersion, void* pVal) {
2,294✔
1962
  if (pInfo->partitionSup.needCalc) {
2,294✔
1963
    return getGroupIdByCol(pInfo, uid, ts, maxVersion, pVal);
646✔
1964
  }
1965

1966
  return getGroupIdByUid(pInfo, uid);
1,648✔
1967
}
1968

1969
void prepareRangeScan(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t* pRowIndex, bool* pRes) {
15,359✔
1970
  int32_t code = TSDB_CODE_SUCCESS;
15,359✔
1971
  int32_t lino = 0;
15,359✔
1972
  if (pBlock->info.rows == 0) {
15,359✔
1973
    if (pRes) {
1!
1974
      (*pRes) = false;
×
1975
    }
1976
    goto _end;
1✔
1977
  }
1978
  if ((*pRowIndex) == pBlock->info.rows) {
15,358✔
1979
    if (pRes) {
7,514!
1980
      (*pRes) = false;
7,514✔
1981
    }
1982
    goto _end;
7,514✔
1983
  }
1984

1985
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
7,844✔
1986
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
7,845✔
1987
  SColumnInfoData* pEndTsCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
7,845✔
1988
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
7,845✔
1989
  STimeWindow      win = {.skey = startData[*pRowIndex], .ekey = endData[*pRowIndex]};
7,845✔
1990
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
7,845✔
1991
  uint64_t*        gpData = (uint64_t*)pGpCol->pData;
7,845✔
1992
  uint64_t         groupId = gpData[*pRowIndex];
7,845✔
1993

1994
  SColumnInfoData* pCalStartTsCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
7,845✔
1995
  TSKEY*           calStartData = (TSKEY*)pCalStartTsCol->pData;
7,845✔
1996
  SColumnInfoData* pCalEndTsCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
7,845✔
1997
  TSKEY*           calEndData = (TSKEY*)pCalEndTsCol->pData;
7,845✔
1998

1999
  setGroupId(pInfo, pBlock, GROUPID_COLUMN_INDEX, *pRowIndex);
7,845✔
2000
  if (isSlidingWindow(pInfo) || isCountSlidingWindow(pInfo)) {
7,845✔
2001
    pInfo->updateWin.skey = calStartData[*pRowIndex];
332✔
2002
    pInfo->updateWin.ekey = calEndData[*pRowIndex];
332✔
2003
  }
2004
  (*pRowIndex)++;
7,845✔
2005

2006
  for (; *pRowIndex < pBlock->info.rows; (*pRowIndex)++) {
7,855✔
2007
    if (win.skey == startData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
339✔
2008
      win.ekey = TMAX(win.ekey, endData[*pRowIndex]);
10✔
2009
      continue;
10✔
2010
    }
2011

2012
    if (win.skey == endData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
329!
2013
      win.skey = TMIN(win.skey, startData[*pRowIndex]);
×
2014
      continue;
×
2015
    }
2016
    break;
329✔
2017
  }
2018

2019
  STableScanInfo* pTScanInfo = pInfo->pTableScanOp->info;
7,845✔
2020

2021
  int64_t maxVer = -1;
7,845✔
2022
  if (!isRecalculateOperator(&pInfo->basic)) {
7,845✔
2023
    // coverity scan
2024
    QUERY_CHECK_NULL(pInfo->pUpdateInfo, code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
7,825!
2025

2026
    qDebug("prepare range scan start:%" PRId64 ",end:%" PRId64 ",maxVer:%" PRId64, win.skey, win.ekey,
7,825✔
2027
          pInfo->pUpdateInfo->maxDataVersion);
2028
    maxVer = pInfo->pUpdateInfo->maxDataVersion;
7,825✔
2029
  }
2030
  resetTableScanInfo(pInfo->pTableScanOp->info, &win, maxVer);
7,844✔
2031
  pInfo->pTableScanOp->status = OP_OPENED;
7,844✔
2032
  if (pRes) {
7,844✔
2033
    (*pRes) = true;
329✔
2034
  }
2035

2036
_end:
7,515✔
2037
  if (code != TSDB_CODE_SUCCESS) {
15,359!
2038
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2039
  }
2040
}
15,359✔
2041

2042
STimeWindow getSlidingWindow(TSKEY* startTsCol, TSKEY* endTsCol, uint64_t* gpIdCol, SInterval* pInterval,
1,709✔
2043
                             SDataBlockInfo* pDataBlockInfo, int32_t* pRowIndex, bool hasGroup) {
2044
  SResultRowInfo dumyInfo = {0};
1,709✔
2045
  dumyInfo.cur.pageId = -1;
1,709✔
2046
  STimeWindow win = getActiveTimeWindow(NULL, &dumyInfo, startTsCol[*pRowIndex], pInterval, TSDB_ORDER_ASC);
1,709✔
2047
  STimeWindow endWin = win;
1,709✔
2048
  STimeWindow preWin = win;
1,709✔
2049
  uint64_t    groupId = gpIdCol[*pRowIndex];
1,709✔
2050

2051
  while (1) {
2052
    if (hasGroup) {
1,709✔
2053
      (*pRowIndex) += 1;
147✔
2054
    } else {
2055
      while ((groupId == gpIdCol[(*pRowIndex)] && startTsCol[*pRowIndex] <= endWin.ekey)) {
1,603✔
2056
        (*pRowIndex) += 1;
1,569✔
2057
        if ((*pRowIndex) == pDataBlockInfo->rows) {
1,569✔
2058
          break;
1,528✔
2059
        }
2060
      }
2061
    }
2062

2063
    do {
2064
      preWin = endWin;
2,549✔
2065
      getNextTimeWindow(pInterval, &endWin, TSDB_ORDER_ASC);
2,549✔
2066
    } while (endTsCol[(*pRowIndex) - 1] >= endWin.skey);
2,549✔
2067
    endWin = preWin;
1,709✔
2068
    if (win.ekey == endWin.ekey || (*pRowIndex) == pDataBlockInfo->rows || groupId != gpIdCol[*pRowIndex]) {
1,709!
2069
      win.ekey = endWin.ekey;
1,709✔
2070
      return win;
1,709✔
2071
    }
2072
    win.ekey = endWin.ekey;
×
2073
  }
2074
}
2075

2076
int32_t doRangeScan(SStreamScanInfo* pInfo, SSDataBlock* pSDB, int32_t tsColIndex, int32_t* pRowIndex,
10,615✔
2077
                    SSDataBlock** ppRes) {
2078
  qDebug("do stream range scan. windows index:%d", *pRowIndex);
10,615✔
2079

2080
  int32_t code = TSDB_CODE_SUCCESS;
10,615✔
2081
  int32_t lino = 0;
10,615✔
2082
  bool    prepareRes = true;
10,615✔
2083

2084
  while (1) {
699✔
2085
    SSDataBlock* pResult = NULL;
11,314✔
2086
    code = doTableScanNext(pInfo->pTableScanOp, &pResult);
11,314✔
2087
    QUERY_CHECK_CODE(code, lino, _end);
21,929!
2088

2089
    if (!pResult) {
11,314✔
2090
      prepareRangeScan(pInfo, pSDB, pRowIndex, &prepareRes);
7,843✔
2091
      // scan next window data
2092
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
7,843✔
2093
      QUERY_CHECK_CODE(code, lino, _end);
7,843!
2094
    }
2095

2096
    if (!pResult) {
11,314✔
2097
      if (prepareRes) {
7,635✔
2098
        continue;
121✔
2099
      }
2100
      blockDataCleanup(pSDB);
7,514✔
2101
      *pRowIndex = 0;
7,514✔
2102
      pInfo->updateWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
7,514✔
2103
      STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
7,514✔
2104
      pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
7,514✔
2105
      pTableScanInfo->base.dataReader = NULL;
7,514✔
2106
      (*ppRes) = NULL;
7,514✔
2107
      goto _end;
7,514✔
2108
    }
2109

2110
    code = doFilter(pResult, pInfo->pTableScanOp->exprSupp.pFilterInfo, NULL);
3,679✔
2111
    QUERY_CHECK_CODE(code, lino, _end);
3,679!
2112
    if (pResult->info.rows == 0) {
3,679!
2113
      continue;
×
2114
    }
2115

2116
    if (pInfo->igCheckGroupId == true) {
3,679✔
2117
      pResult->info.calWin = pInfo->updateWin;
579✔
2118
      (*ppRes) = pResult;
579✔
2119
      goto _end;
579✔
2120
    }
2121

2122
    if (pInfo->partitionSup.needCalc) {
3,100✔
2123
      SSDataBlock* tmpBlock = NULL;
655✔
2124
      code = createOneDataBlock(pResult, true, &tmpBlock);
655✔
2125
      QUERY_CHECK_CODE(code, lino, _end);
1,112!
2126

2127
      blockDataCleanup(pResult);
655✔
2128
      for (int32_t i = 0; i < tmpBlock->info.rows; i++) {
2,281✔
2129
        if (calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, tmpBlock, i) == pInfo->groupId) {
1,626✔
2130
          for (int32_t j = 0; j < pInfo->pTableScanOp->exprSupp.numOfExprs; j++) {
10,618✔
2131
            SColumnInfoData* pSrcCol = taosArrayGet(tmpBlock->pDataBlock, j);
9,836✔
2132
            SColumnInfoData* pDestCol = taosArrayGet(pResult->pDataBlock, j);
9,835✔
2133
            bool             isNull = colDataIsNull(pSrcCol, tmpBlock->info.rows, i, NULL);
9,837✔
2134
            char*            pSrcData = NULL;
9,837✔
2135
            if (!isNull) pSrcData = colDataGetData(pSrcCol, i);
9,837!
2136
            code = colDataSetVal(pDestCol, pResult->info.rows, pSrcData, isNull);
9,837✔
2137
            QUERY_CHECK_CODE(code, lino, _end);
9,836!
2138
          }
2139
          pResult->info.rows++;
782✔
2140
        }
2141
      }
2142

2143
      blockDataDestroy(tmpBlock);
655✔
2144

2145
      if (pResult->info.rows > 0) {
655✔
2146
        pResult->info.calWin = pInfo->updateWin;
457✔
2147
        (*ppRes) = pResult;
457✔
2148
        goto _end;
457✔
2149
      }
2150
    } else if (pResult->info.id.groupId == pInfo->groupId) {
2,445✔
2151
      pResult->info.calWin = pInfo->updateWin;
2,065✔
2152
      (*ppRes) = pResult;
2,065✔
2153
      goto _end;
2,065✔
2154
    }
2155
  }
2156

2157
_end:
10,615✔
2158
  if (code != TSDB_CODE_SUCCESS) {
10,615!
2159
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2160
  }
2161
  return code;
10,615✔
2162
}
2163

2164
static void getPreSessionWindow(SStreamAggSupporter* pAggSup, TSKEY startTs, TSKEY endTs, uint64_t groupId,
2✔
2165
                                SSessionKey* pKey) {
2166
  pKey->win.skey = startTs;
2✔
2167
  pKey->win.ekey = endTs;
2✔
2168
  pKey->groupId = groupId;
2✔
2169

2170
  void*   pCur = pAggSup->stateStore.streamStateSessionSeekKeyCurrentPrev(pAggSup->pState, pKey);
2✔
2171
  int32_t code = pAggSup->stateStore.streamStateSessionGetKVByCur(pCur, pKey, NULL, 0);
2✔
2172
  if (code != TSDB_CODE_SUCCESS) {
2!
2173
    SET_SESSION_WIN_KEY_INVALID(pKey);
×
2174
  }
2175

2176
  taosMemoryFree(pCur);
2!
2177
}
2✔
2178

2179
int32_t appendOneRowToSpecialBlockImpl(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, TSKEY* pCalStartTs,
7,844✔
2180
                                       TSKEY* pCalEndTs, uint64_t* pUid, uint64_t* pGp, void* pTbName, void* pPkData) {
2181
  int32_t          code = TSDB_CODE_SUCCESS;
7,844✔
2182
  int32_t          lino = 0;
7,844✔
2183
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
7,844✔
2184
  SColumnInfoData* pEndTsCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
7,843✔
2185
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
7,843✔
2186
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
7,843✔
2187
  SColumnInfoData* pCalStartCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
7,842✔
2188
  SColumnInfoData* pCalEndCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
7,842✔
2189
  SColumnInfoData* pTableCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
7,842✔
2190
  code = colDataSetVal(pStartTsCol, pBlock->info.rows, (const char*)pStartTs, false);
7,841✔
2191
  QUERY_CHECK_CODE(code, lino, _end);
7,841!
2192

2193
  code = colDataSetVal(pEndTsCol, pBlock->info.rows, (const char*)pEndTs, false);
7,841✔
2194
  QUERY_CHECK_CODE(code, lino, _end);
7,841!
2195

2196
  code = colDataSetVal(pUidCol, pBlock->info.rows, (const char*)pUid, pUid == NULL);
7,841✔
2197
  QUERY_CHECK_CODE(code, lino, _end);
7,841!
2198

2199
  code = colDataSetVal(pGpCol, pBlock->info.rows, (const char*)pGp, pGp == NULL);
7,841✔
2200
  QUERY_CHECK_CODE(code, lino, _end);
7,841!
2201

2202
  code = colDataSetVal(pCalStartCol, pBlock->info.rows, (const char*)pCalStartTs, false);
7,841✔
2203
  QUERY_CHECK_CODE(code, lino, _end);
7,841!
2204

2205
  code = colDataSetVal(pCalEndCol, pBlock->info.rows, (const char*)pCalEndTs, false);
7,841✔
2206
  QUERY_CHECK_CODE(code, lino, _end);
7,841!
2207

2208
  code = colDataSetVal(pTableCol, pBlock->info.rows, (const char*)pTbName, pTbName == NULL);
7,841✔
2209
  QUERY_CHECK_CODE(code, lino, _end);
7,841!
2210

2211
  if (taosArrayGetSize(pBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
7,841✔
2212
    SColumnInfoData* pPkCol = taosArrayGet(pBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
3✔
2213
    code = colDataSetVal(pPkCol, pBlock->info.rows, (const char*)pPkData, pPkData == NULL);
3✔
2214
    QUERY_CHECK_CODE(code, lino, _end);
3!
2215
  }
2216
  pBlock->info.rows++;
7,841✔
2217

2218
_end:
7,841✔
2219
  if (code != TSDB_CODE_SUCCESS) {
7,841!
2220
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2221
  }
2222
  return code;
7,841✔
2223
}
2224

2225
int32_t appendPkToSpecialBlock(SSDataBlock* pBlock, TSKEY* pTsArray, SColumnInfoData* pPkCol, int32_t rowId,
1,809✔
2226
                               uint64_t* pUid, uint64_t* pGp, void* pTbName) {
2227
  void* pVal = NULL;
1,809✔
2228
  if (pPkCol) {
1,809✔
2229
    pVal = colDataGetData(pPkCol, rowId);
9!
2230
  }
2231
  return appendOneRowToSpecialBlockImpl(pBlock, pTsArray + rowId, pTsArray + rowId, pTsArray + rowId, pTsArray + rowId,
1,809✔
2232
                                        pUid, pGp, pTbName, pVal);
2233
}
2234

2235
static int32_t getPreVersionDataBlock(uint64_t uid, TSKEY startTs, TSKEY endTs, int64_t version, char* taskIdStr,
7✔
2236
                                      SStreamScanInfo* pInfo, SSDataBlock* pBlock) {
2237
  int32_t      code = TSDB_CODE_SUCCESS;
7✔
2238
  int32_t      lino = 0;
7✔
2239
  SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, uid, startTs, endTs, version);
7✔
2240
  printDataBlock(pPreRes, "pre res", taskIdStr);
7✔
2241
  blockDataCleanup(pBlock);
7✔
2242
  if (!pPreRes) {
7!
2243
    if (terrno != TSDB_CODE_SUCCESS) {
×
2244
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2245
    }
2246
    goto _end;
×
2247
  }
2248
  code = blockDataEnsureCapacity(pBlock, pPreRes->info.rows);
7✔
2249
  QUERY_CHECK_CODE(code, lino, _end);
7!
2250

2251
  SColumnInfoData* pTsCol = (SColumnInfoData*)taosArrayGet(pPreRes->pDataBlock, pInfo->primaryTsIndex);
7✔
2252
  SColumnInfoData* pPkCol = NULL;
7✔
2253
  if (hasPrimaryKeyCol(pInfo)) {
7✔
2254
    pPkCol = (SColumnInfoData*)taosArrayGet(pPreRes->pDataBlock, pInfo->primaryKeyIndex);
2✔
2255
  }
2256
  for (int32_t i = 0; i < pPreRes->info.rows; i++) {
28✔
2257
    uint64_t groupId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, i);
21✔
2258
    code = appendPkToSpecialBlock(pBlock, (TSKEY*)pTsCol->pData, pPkCol, i, &uid, &groupId, NULL);
21✔
2259
    QUERY_CHECK_CODE(code, lino, _end);
21!
2260
  }
2261
  printDataBlock(pBlock, "new delete", taskIdStr);
7✔
2262

2263
_end:
7✔
2264
  if (code != TSDB_CODE_SUCCESS) {
7!
2265
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2266
  }
2267
  return code;
7✔
2268
}
2269

2270
static int32_t generateSessionScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
1,021✔
2271
                                        EStreamType mode) {
2272
  int32_t code = TSDB_CODE_SUCCESS;
1,021✔
2273
  int32_t lino = 0;
1,021✔
2274
  if (pSrcBlock->info.rows == 0) {
1,021!
2275
    return TSDB_CODE_SUCCESS;
×
2276
  }
2277
  SSHashObj*       pScanRange = tSimpleHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
1,021✔
2278
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
1,021✔
2279
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
1,021✔
2280
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
1,021✔
2281
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
1,021✔
2282
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
1,021✔
2283
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
1,021✔
2284
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
1,021✔
2285
  SColumnInfoData* pGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
1,021✔
2286
  SColumnInfoData* pSrcPkCol = NULL;
1,021✔
2287
  uint64_t*        pSrcGp = (uint64_t*)pGpCol->pData;
1,021✔
2288
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
1,021✔
2289
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
1✔
2290
  }
2291
  int64_t ver = pSrcBlock->info.version - 1;
1,021✔
2292

2293
  if (pInfo->partitionSup.needCalc &&
1,021✔
2294
      (startData[0] != endData[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
484✔
2295
    code = getPreVersionDataBlock(uidCol[0], startData[0], endData[0], ver, GET_TASKID(pTaskInfo), pInfo, pSrcBlock);
4✔
2296
    QUERY_CHECK_CODE(code, lino, _end);
4!
2297
    startData = (TSKEY*)pStartTsCol->pData;
4✔
2298
    endData = (TSKEY*)pEndTsCol->pData;
4✔
2299
    uidCol = (uint64_t*)pUidCol->pData;
4✔
2300
    pSrcGp = (uint64_t*)pGpCol->pData;
4✔
2301
  }
2302
  blockDataCleanup(pDestBlock);
1,021✔
2303
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
1,021✔
2304
  QUERY_CHECK_CODE(code, lino, _end);
1,021!
2305

2306
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
1,021✔
2307
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
1,021✔
2308
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
1,021✔
2309
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
1,021✔
2310
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
1,021✔
2311
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
1,021✔
2312
  SColumnInfoData* pDestTableNameInxCol = taosArrayGet(pDestBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
1,021✔
2313
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
2,088✔
2314
    uint64_t groupId = pSrcGp[i];
1,067✔
2315
    if (groupId == 0) {
1,067✔
2316
      void* pVal = NULL;
891✔
2317
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
891!
2318
        pVal = colDataGetData(pSrcPkCol, i);
1!
2319
      }
2320
      groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], ver, pVal);
891✔
2321
    }
2322
    // gap must be 0.
2323
    SSessionKey startWin = {0};
1,067✔
2324
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, startData[i], startData[i], groupId, &startWin);
1,067✔
2325
    if (IS_INVALID_SESSION_WIN_KEY(startWin)) {
1,067✔
2326
      // window has been closed.
2327
      continue;
71✔
2328
    }
2329
    SSessionKey endWin = {0};
1,001✔
2330
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, endData[i], endData[i], groupId, &endWin);
1,001✔
2331
    if (IS_INVALID_SESSION_WIN_KEY(endWin)) {
1,001✔
2332
      getPreSessionWindow(pInfo->windowSup.pStreamAggSup, endData[i], endData[i], groupId, &endWin);
2✔
2333
    }
2334
    if (IS_INVALID_SESSION_WIN_KEY(startWin)) {
1,001!
2335
      // window has been closed.
2336
      qError("generate session scan range failed. rang start:%" PRIx64 ", end:%" PRIx64, startData[i], endData[i]);
×
2337
      continue;
×
2338
    }
2339

2340
    SSessionKey checkKey = {.groupId = groupId, .win.skey = startWin.win.skey, .win.ekey = endWin.win.ekey};
1,001✔
2341
    if (tSimpleHashGet(pScanRange, &checkKey, sizeof(SSessionKey)) != NULL) {
1,001✔
2342
      continue;
5✔
2343
    }
2344
    code = tSimpleHashPut(pScanRange, &checkKey, sizeof(SSessionKey), NULL, 0);
996✔
2345
    QUERY_CHECK_CODE(code, lino, _end);
996!
2346

2347
    code = colDataSetVal(pDestStartCol, i, (const char*)&startWin.win.skey, false);
996✔
2348
    QUERY_CHECK_CODE(code, lino, _end);
996!
2349

2350
    code = colDataSetVal(pDestEndCol, i, (const char*)&endWin.win.ekey, false);
996✔
2351
    QUERY_CHECK_CODE(code, lino, _end);
996!
2352

2353
    colDataSetNULL(pDestUidCol, i);
996!
2354
    code = colDataSetVal(pDestGpCol, i, (const char*)&groupId, false);
996✔
2355
    QUERY_CHECK_CODE(code, lino, _end);
996!
2356

2357
    colDataSetNULL(pDestCalStartTsCol, i);
996!
2358
    colDataSetNULL(pDestTableNameInxCol, i);
996!
2359
    pDestBlock->info.rows++;
996✔
2360
  }
2361

2362
_end:
1,021✔
2363
  tSimpleHashCleanup(pScanRange);
1,021✔
2364
  if (code != TSDB_CODE_SUCCESS) {
1,021!
2365
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2366
  }
2367
  return code;
1,021✔
2368
}
2369

2370
static int32_t generateCountScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
67✔
2371
                                      EStreamType mode) {
2372
  int32_t code = TSDB_CODE_SUCCESS;
67✔
2373
  int32_t lino = 0;
67✔
2374
  blockDataCleanup(pDestBlock);
67✔
2375
  if (pSrcBlock->info.rows == 0) {
67!
2376
    return TSDB_CODE_SUCCESS;
×
2377
  }
2378
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
67✔
2379
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
67✔
2380
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
67✔
2381
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
67✔
2382
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
67✔
2383
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
67✔
2384
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
67✔
2385
  SColumnInfoData* pGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
67✔
2386
  uint64_t*        pSrcGp = (uint64_t*)pGpCol->pData;
67✔
2387
  SColumnInfoData* pSrcPkCol = NULL;
67✔
2388
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
67!
2389
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
×
2390
  }
2391
  int64_t ver = pSrcBlock->info.version - 1;
67✔
2392

2393
  if (pInfo->partitionSup.needCalc &&
67✔
2394
      (startData[0] != endData[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
16!
2395
    code = getPreVersionDataBlock(uidCol[0], startData[0], endData[0], ver, GET_TASKID(pTaskInfo), pInfo, pSrcBlock);
1✔
2396
    QUERY_CHECK_CODE(code, lino, _end);
1!
2397
    startData = (TSKEY*)pStartTsCol->pData;
1✔
2398
    endData = (TSKEY*)pEndTsCol->pData;
1✔
2399
    uidCol = (uint64_t*)pUidCol->pData;
1✔
2400
    pSrcGp = (uint64_t*)pGpCol->pData;
1✔
2401
  }
2402

2403
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
67✔
2404
  QUERY_CHECK_CODE(code, lino, _end);
67!
2405

2406
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
67✔
2407
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
67✔
2408
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
67✔
2409
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
67✔
2410
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
67✔
2411
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
67✔
2412
  SColumnInfoData* pDestTableNameInxCol = taosArrayGet(pDestBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
67✔
2413
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
136✔
2414
    uint64_t groupId = pSrcGp[i];
69✔
2415
    if (groupId == 0) {
69✔
2416
      void* pVal = NULL;
55✔
2417
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
55!
2418
        pVal = colDataGetData(pSrcPkCol, i);
×
2419
      }
2420
      groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], ver, pVal);
55✔
2421
    }
2422
    SSessionKey startWin = {.win.skey = startData[i], .win.ekey = endData[i], .groupId = groupId};
69✔
2423
    SSessionKey range = {0};
69✔
2424
    getCountWinRange(pInfo->windowSup.pStreamAggSup, &startWin, mode, &range);
69✔
2425
    code = colDataSetVal(pDestStartCol, i, (const char*)&range.win.skey, false);
69✔
2426
    QUERY_CHECK_CODE(code, lino, _end);
69!
2427

2428
    code = colDataSetVal(pDestEndCol, i, (const char*)&range.win.ekey, false);
69✔
2429
    QUERY_CHECK_CODE(code, lino, _end);
69!
2430

2431
    colDataSetNULL(pDestUidCol, i);
69!
2432
    code = colDataSetVal(pDestGpCol, i, (const char*)&groupId, false);
69✔
2433
    QUERY_CHECK_CODE(code, lino, _end);
69!
2434

2435
    code = colDataSetVal(pDestCalStartTsCol, i, (const char*)&range.win.skey, false);
69✔
2436
    QUERY_CHECK_CODE(code, lino, _end);
69!
2437

2438
    code = colDataSetVal(pDestCalEndTsCol, i, (const char*)&range.win.ekey, false);
69✔
2439
    QUERY_CHECK_CODE(code, lino, _end);
69!
2440

2441
    colDataSetNULL(pDestTableNameInxCol, i);
69!
2442

2443
    pDestBlock->info.rows++;
69✔
2444
  }
2445

2446
_end:
67✔
2447
  if (code != TSDB_CODE_SUCCESS) {
67!
2448
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2449
  }
2450
  return code;
67✔
2451
}
2452

2453
static int32_t setDelRangeEndKey(SStreamAggSupporter* pAggSup, SStreamFillSupporter* pFillSup, SWinKey* pEndKey, STimeWindow* pScanRange, bool* pRes) {
24✔
2454
  int32_t     code = TSDB_CODE_SUCCESS;
24✔
2455
  int32_t     lino = 0;
24✔
2456
  SSlicePoint nextPoint = {.key.groupId = pEndKey->groupId};
24✔
2457
  int32_t     vLen = 0;
24✔
2458
  int32_t     winCode = TSDB_CODE_SUCCESS;
24✔
2459
  code = pAggSup->stateStore.streamStateFillGetNext(pAggSup->pState, pEndKey, &nextPoint.key, (void**)&nextPoint.pResPos, &vLen, &winCode);
24✔
2460
  QUERY_CHECK_CODE(code, lino, _end);
24!
2461
  if (winCode == TSDB_CODE_SUCCESS) {
24✔
2462
    setPointBuff(&nextPoint, pFillSup);
16✔
2463
    if (HAS_ROW_DATA(nextPoint.pLeftRow) && pEndKey->ts < nextPoint.pLeftRow->key) {
16!
2464
      pScanRange->ekey = nextPoint.pLeftRow->key;
8✔
2465
      *pRes = true;
8✔
2466
    } else if (pEndKey->ts < nextPoint.pRightRow->key) {
8✔
2467
      pScanRange->ekey = nextPoint.pRightRow->key;
2✔
2468
      *pRes = true;
2✔
2469
    } else {
2470
      *pEndKey = nextPoint.key;
6✔
2471
      pScanRange->ekey = TMAX(nextPoint.pRightRow->key, nextPoint.key.ts);
6✔
2472
      *pRes = false;
6✔
2473
    }
2474
  }
2475

2476
_end:
8✔
2477
  if (code != TSDB_CODE_SUCCESS) {
24!
2478
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2479
  }
2480
  return code;
24✔
2481
}
2482

2483
int32_t getTimeSliceWinRange(SStreamAggSupporter* pAggSup, SStreamFillSupporter* pFillSup, SInterval* pInterval, TSKEY start, TSKEY end,
15✔
2484
                             int64_t groupId, STimeWindow* pScanRange) {
2485
  int32_t        code = TSDB_CODE_SUCCESS;
15✔
2486
  int32_t        lino = 0;
15✔
2487
  int32_t        winCode = TSDB_CODE_SUCCESS;
15✔
2488
  SResultRowInfo dumyInfo = {0};
15✔
2489
  dumyInfo.cur.pageId = -1;
15✔
2490
  STimeWindow sWin = getActiveTimeWindow(NULL, &dumyInfo, start, pInterval, TSDB_ORDER_ASC);
15✔
2491
  SWinKey     startKey = {.groupId = groupId, .ts = sWin.skey};
15✔
2492

2493
  sWin = getActiveTimeWindow(NULL, &dumyInfo, end, pInterval, TSDB_ORDER_ASC);
15✔
2494
  SWinKey endKey = {.groupId = groupId, .ts = sWin.ekey};
15✔
2495

2496
  SSlicePoint prevPoint = {.key.groupId = groupId};
15✔
2497
  SSlicePoint nextPoint = {.key.groupId = groupId};
15✔
2498
  int32_t     vLen = 0;
15✔
2499
  code = pAggSup->stateStore.streamStateFillGetPrev(pAggSup->pState, &startKey, &prevPoint.key, (void**)&prevPoint.pResPos, &vLen, &winCode);
15✔
2500
  QUERY_CHECK_CODE(code, lino, _end);
15!
2501
  if (winCode == TSDB_CODE_SUCCESS) {
15✔
2502
    setPointBuff(&prevPoint, pFillSup);
10✔
2503
    if (HAS_ROW_DATA(prevPoint.pRightRow)) {
10!
2504
      pScanRange->skey = prevPoint.pRightRow->key;
9✔
2505
    } else {
2506
      pScanRange->skey = prevPoint.pLeftRow->key;
1✔
2507
    }
2508
  } else {
2509
    pScanRange->skey = startKey.ts;
5✔
2510
  }
2511

2512
  bool res = false;
15✔
2513
  SWinKey curKey = endKey;
15✔
2514
  code = setDelRangeEndKey(pAggSup, pFillSup, &curKey, pScanRange, &res);
15✔
2515
  QUERY_CHECK_CODE(code, lino, _end);
15!
2516
  if (res == false) {
15✔
2517
    code = setDelRangeEndKey(pAggSup, pFillSup, &curKey, pScanRange, &res);
9✔
2518
    QUERY_CHECK_CODE(code, lino, _end);
9!
2519
  }
2520
  if (res == false) {
15✔
2521
    pScanRange->ekey = TMAX(endKey.ts, pScanRange->ekey);
5✔
2522
  }
2523

2524
_end:
10✔
2525
  if (code != TSDB_CODE_SUCCESS) {
15!
2526
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2527
  }
2528
  return code;
15✔
2529
}
2530

2531
static int32_t generateTimeSliceScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
15✔
2532
                                          EStreamType mode) {
2533
  int32_t code = TSDB_CODE_SUCCESS;
15✔
2534
  int32_t lino = 0;
15✔
2535
  blockDataCleanup(pDestBlock);
15✔
2536
  if (pSrcBlock->info.rows == 0) {
15!
2537
    return TSDB_CODE_SUCCESS;
×
2538
  }
2539
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
15✔
2540
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
15✔
2541
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
15✔
2542
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
15✔
2543
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
15✔
2544
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
15✔
2545
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
15✔
2546
  SColumnInfoData* pGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
15✔
2547
  uint64_t*        pSrcGp = (uint64_t*)pGpCol->pData;
15✔
2548
  SColumnInfoData* pSrcPkCol = NULL;
15✔
2549
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
15!
2550
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
×
2551
  }
2552
  int64_t ver = pSrcBlock->info.version - 1;
15✔
2553

2554
  if (pInfo->partitionSup.needCalc &&
15!
2555
      (startData[0] != endData[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
×
2556
    code = getPreVersionDataBlock(uidCol[0], startData[0], endData[0], ver, GET_TASKID(pTaskInfo), pInfo, pSrcBlock);
×
2557
    QUERY_CHECK_CODE(code, lino, _end);
×
2558
    startData = (TSKEY*)pStartTsCol->pData;
×
2559
    endData = (TSKEY*)pEndTsCol->pData;
×
2560
    uidCol = (uint64_t*)pUidCol->pData;
×
2561
    pSrcGp = (uint64_t*)pGpCol->pData;
×
2562
  }
2563

2564
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
15✔
2565
  QUERY_CHECK_CODE(code, lino, _end);
15!
2566

2567
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
15✔
2568
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
15✔
2569
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
15✔
2570
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
15✔
2571
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
15✔
2572
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
15✔
2573
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
30✔
2574
    uint64_t groupId = pSrcGp[i];
15✔
2575
    if (groupId == 0) {
15!
2576
      void* pVal = NULL;
15✔
2577
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
15!
2578
        pVal = colDataGetData(pSrcPkCol, i);
×
2579
      }
2580
      groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], ver, pVal);
15✔
2581
    }
2582

2583
    STimeWindow scanRange = {0};
15✔
2584
    code = getTimeSliceWinRange(pInfo->windowSup.pStreamAggSup, pInfo->pFillSup, &pInfo->interval, startData[i], endData[i], groupId,
15✔
2585
                                &scanRange);
2586
    QUERY_CHECK_CODE(code, lino, _end);
15!
2587

2588
    code = colDataSetVal(pDestStartCol, i, (const char*)&scanRange.skey, false);
15✔
2589
    QUERY_CHECK_CODE(code, lino, _end);
15!
2590

2591
    code = colDataSetVal(pDestEndCol, i, (const char*)&scanRange.ekey, false);
15✔
2592
    QUERY_CHECK_CODE(code, lino, _end);
15!
2593

2594
    colDataSetNULL(pDestUidCol, i);
15!
2595
    code = colDataSetVal(pDestGpCol, i, (const char*)&groupId, false);
15✔
2596
    QUERY_CHECK_CODE(code, lino, _end);
15!
2597

2598
    code = colDataSetVal(pDestCalStartTsCol, i, (const char*)&scanRange.skey, false);
15✔
2599
    QUERY_CHECK_CODE(code, lino, _end);
15!
2600

2601
    code = colDataSetVal(pDestCalEndTsCol, i, (const char*)&scanRange.ekey, false);
15✔
2602
    QUERY_CHECK_CODE(code, lino, _end);
15!
2603

2604
    pDestBlock->info.rows++;
15✔
2605
  }
2606

2607
_end:
15✔
2608
  if (code != TSDB_CODE_SUCCESS) {
15!
2609
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2610
  }
2611
  return code;
15✔
2612
}
2613

2614
static int32_t generateIntervalScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
1,646✔
2615
                                         EStreamType mode) {
2616
  int32_t code = TSDB_CODE_SUCCESS;
1,646✔
2617
  int32_t lino = 0;
1,646✔
2618
  blockDataCleanup(pDestBlock);
1,646✔
2619
  if (pSrcBlock->info.rows == 0) {
1,646!
2620
    return TSDB_CODE_SUCCESS;
×
2621
  }
2622
  SSHashObj*       pScanRange = tSimpleHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
1,646✔
2623
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
1,646✔
2624
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
1,646✔
2625
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
1,646✔
2626
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
1,646✔
2627
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
1,646✔
2628
  SColumnInfoData* pSrcPkCol = NULL;
1,646✔
2629
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
1,646✔
2630
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
1✔
2631
  }
2632

2633
  uint64_t* srcUidData = (uint64_t*)pSrcUidCol->pData;
1,646✔
2634
  TSKEY*    srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
1,646✔
2635
  TSKEY*    srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
1,646✔
2636
  int64_t   ver = pSrcBlock->info.version - 1;
1,646✔
2637

2638
  if (pInfo->partitionSup.needCalc &&
1,646✔
2639
      (srcStartTsCol[0] != srcEndTsCol[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
129✔
2640
    code = getPreVersionDataBlock(srcUidData[0], srcStartTsCol[0], srcEndTsCol[0], ver, GET_TASKID(pTaskInfo), pInfo,
2✔
2641
                                  pSrcBlock);
2642
    QUERY_CHECK_CODE(code, lino, _end);
2!
2643

2644
    srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
2✔
2645
    srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
2✔
2646
    srcUidData = (uint64_t*)pSrcUidCol->pData;
2✔
2647
  }
2648

2649
  uint64_t* srcGp = (uint64_t*)pSrcGpCol->pData;
1,646✔
2650
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
1,646✔
2651
  QUERY_CHECK_CODE(code, lino, _end);
1,647!
2652

2653
  SColumnInfoData* pStartTsCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
1,647✔
2654
  SColumnInfoData* pEndTsCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
1,647✔
2655
  SColumnInfoData* pDeUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
1,647✔
2656
  SColumnInfoData* pGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
1,647✔
2657
  SColumnInfoData* pCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
1,647✔
2658
  SColumnInfoData* pCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
1,647✔
2659
  SColumnInfoData* pDestTableNameInxCol = taosArrayGet(pDestBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
1,647✔
2660
  for (int32_t i = 0; i < pSrcBlock->info.rows;) {
3,346✔
2661
    uint64_t srcUid = srcUidData[i];
1,699✔
2662
    uint64_t groupId = srcGp[i];
1,699✔
2663
    if (groupId == 0) {
1,699✔
2664
      void* pVal = NULL;
1,284✔
2665
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
1,284!
2666
        pVal = colDataGetData(pSrcPkCol, i);
1!
2667
      }
2668
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], ver, pVal);
1,284✔
2669
    }
2670
    TSKEY calStartTs = srcStartTsCol[i];
1,699✔
2671
    code = colDataSetVal(pCalStartTsCol, pDestBlock->info.rows, (const char*)(&calStartTs), false);
1,699✔
2672
    QUERY_CHECK_CODE(code, lino, _end);
1,699!
2673

2674
    STimeWindow win = getSlidingWindow(srcStartTsCol, srcEndTsCol, srcGp, &pInfo->interval, &pSrcBlock->info, &i,
1,699✔
2675
                                       pInfo->partitionSup.needCalc);
1,699✔
2676
    TSKEY       calEndTs = srcStartTsCol[i - 1];
1,699✔
2677
    code = colDataSetVal(pCalEndTsCol, pDestBlock->info.rows, (const char*)(&calEndTs), false);
1,699✔
2678
    QUERY_CHECK_CODE(code, lino, _end);
1,699!
2679

2680
    code = colDataSetVal(pDeUidCol, pDestBlock->info.rows, (const char*)(&srcUid), false);
1,699✔
2681
    QUERY_CHECK_CODE(code, lino, _end);
1,699!
2682

2683
    SSessionKey checkKey = {.groupId = groupId, .win = win};
1,699✔
2684
    if (tSimpleHashGet(pScanRange, &checkKey, sizeof(SSessionKey)) != NULL) {
1,699!
2685
      continue;
×
2686
    }
2687
    code = tSimpleHashPut(pScanRange, &checkKey, sizeof(SSessionKey), NULL, 0);
1,699✔
2688
    QUERY_CHECK_CODE(code, lino, _end);
1,699!
2689

2690
    code = colDataSetVal(pStartTsCol, pDestBlock->info.rows, (const char*)(&win.skey), false);
1,699✔
2691
    QUERY_CHECK_CODE(code, lino, _end);
1,699!
2692

2693
    code = colDataSetVal(pEndTsCol, pDestBlock->info.rows, (const char*)(&win.ekey), false);
1,699✔
2694
    QUERY_CHECK_CODE(code, lino, _end);
1,699!
2695

2696
    code = colDataSetVal(pGpCol, pDestBlock->info.rows, (const char*)(&groupId), false);
1,699✔
2697
    QUERY_CHECK_CODE(code, lino, _end);
1,699!
2698

2699
    colDataSetNULL(pDestTableNameInxCol,  pDestBlock->info.rows);
1,699!
2700

2701
    pDestBlock->info.rows++;
1,699✔
2702
  }
2703

2704
_end:
1,647✔
2705
  tSimpleHashCleanup(pScanRange);
1,647✔
2706
  if (code != TSDB_CODE_SUCCESS) {
1,647!
2707
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2708
  }
2709
  return code;
1,647✔
2710
}
2711

2712
int32_t calBlockTbName(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t rowId) {
903,393✔
2713
  int32_t code = TSDB_CODE_SUCCESS;
903,393✔
2714
  int32_t lino = 0;
903,393✔
2715
  blockDataCleanup(pInfo->pCreateTbRes);
903,393✔
2716
  if (pInfo->tbnameCalSup.numOfExprs == 0 && pInfo->tagCalSup.numOfExprs == 0) {
903,379✔
2717
    pBlock->info.parTbName[0] = 0;
882,873✔
2718
    if (pInfo->hasPart == false) {
882,873✔
2719
      pInfo->stateStore.streamStateSetParNameInvalid(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState);
877,495✔
2720
    }
2721
  } else {
2722
    code = appendCreateTableRow(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, &pInfo->tbnameCalSup,
20,506✔
2723
                                &pInfo->tagCalSup, pBlock->info.id.groupId, pBlock, rowId, pInfo->pCreateTbRes,
2724
                                &pInfo->stateStore);
2725
    QUERY_CHECK_CODE(code, lino, _end);
20,520!
2726
  }
2727

2728
_end:
20,520✔
2729
  if (code != TSDB_CODE_SUCCESS) {
903,375!
2730
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2731
  }
2732
  return code;
903,384✔
2733
}
2734

2735
static int32_t generatePartitionDelResBlock(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
62✔
2736
  int32_t          code = TSDB_CODE_SUCCESS;
62✔
2737
  int32_t          lino = 0;
62✔
2738
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
62✔
2739
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
62✔
2740
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
62✔
2741
  uint64_t*        srcUidData = (uint64_t*)pSrcUidCol->pData;
62✔
2742
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
62✔
2743
  uint64_t*        srcGp = (uint64_t*)pSrcGpCol->pData;
62✔
2744

2745
  TSKEY*  srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
62✔
2746
  TSKEY*  srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
62✔
2747
  int64_t ver = pSrcBlock->info.version - 1;
62✔
2748
  for (int32_t delI = 0; delI < pSrcBlock->info.rows; delI++) {
124✔
2749
    uint64_t     groupId = 0;
62✔
2750
    uint64_t     srcUid = srcUidData[delI];
62✔
2751
    char         tbname[VARSTR_HEADER_SIZE + TSDB_TABLE_NAME_LEN] = {0};
62✔
2752
    SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, srcUid, srcStartTsCol[delI], srcEndTsCol[delI], ver);
62✔
2753
    if (!pPreRes) {
62!
2754
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2755
      continue;
×
2756
    }
2757
    code = blockDataEnsureCapacity(pDestBlock, pDestBlock->info.rows + pPreRes->info.rows);
62✔
2758
    QUERY_CHECK_CODE(code, lino, _end);
62!
2759
    for (int32_t preJ = 0; preJ < pPreRes->info.rows; preJ++) {
129✔
2760
      groupId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, preJ);
67✔
2761
      if (pInfo->pPartTbnameSup) {
67!
2762
        void*   parTbname = NULL;
67✔
2763
        int32_t winCode = TSDB_CODE_SUCCESS;
67✔
2764
        code = pInfo->stateStore.streamStateGetParName(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, groupId,
67✔
2765
                                                       &parTbname, false, &winCode);
2766
        QUERY_CHECK_CODE(code, lino, _end);
67!
2767

2768
        if (winCode != TSDB_CODE_SUCCESS) {
67✔
2769
          code = calBlockTbName(pInfo, pPreRes, preJ);
64✔
2770
          QUERY_CHECK_CODE(code, lino, _end);
64!
2771
          memcpy(varDataVal(tbname), pPreRes->info.parTbName, strlen(pPreRes->info.parTbName));
64✔
2772
        } else {
2773
          memcpy(varDataVal(tbname), parTbname, TSDB_TABLE_NAME_LEN);
3✔
2774
        }
2775
        varDataSetLen(tbname, strlen(varDataVal(tbname)));
67✔
2776
        pInfo->stateStore.streamStateFreeVal(parTbname);
67✔
2777
      }
2778
      code = appendDataToSpecialBlock(pDestBlock, srcStartTsCol + delI, srcEndTsCol + delI, srcUidData + delI, &groupId,
67✔
2779
                                      tbname[0] == 0 ? NULL : tbname);
67✔
2780
      QUERY_CHECK_CODE(code, lino, _end);
67!
2781
    }
2782
  }
2783

2784
_end:
62✔
2785
  if (code != TSDB_CODE_SUCCESS) {
62!
2786
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2787
  }
2788
  return code;
62✔
2789
}
2790

2791
static int32_t generateDeleteResultBlockImpl(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
73✔
2792
  int32_t          code = TSDB_CODE_SUCCESS;
73✔
2793
  int32_t          lino = 0;
73✔
2794
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
73✔
2795
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
73✔
2796
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
73✔
2797
  uint64_t*        srcUidData = (uint64_t*)pSrcUidCol->pData;
73✔
2798
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
73✔
2799
  uint64_t*        srcGp = (uint64_t*)pSrcGpCol->pData;
73✔
2800
  SColumnInfoData* pSrcPkCol = NULL;
73✔
2801
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
73!
2802
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
×
2803
  }
2804

2805
  TSKEY*  srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
73✔
2806
  TSKEY*  srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
73✔
2807
  int64_t ver = pSrcBlock->info.version - 1;
73✔
2808
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
146✔
2809
    uint64_t srcUid = srcUidData[i];
73✔
2810
    uint64_t groupId = srcGp[i];
73✔
2811
    char     tbname[VARSTR_HEADER_SIZE + TSDB_TABLE_NAME_LEN] = {0};
73✔
2812
    if (groupId == 0) {
73✔
2813
      void* pVal = NULL;
49✔
2814
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
49!
2815
        pVal = colDataGetData(pSrcPkCol, i);
×
2816
      }
2817
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], ver, pVal);
49✔
2818
    }
2819
    if (pInfo->tbnameCalSup.pExprInfo) {
73✔
2820
      void*   parTbname = NULL;
25✔
2821
      int32_t winCode = TSDB_CODE_SUCCESS;
25✔
2822
      code = pInfo->stateStore.streamStateGetParName(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, groupId,
25✔
2823
                                                     &parTbname, false, &winCode);
2824
      QUERY_CHECK_CODE(code, lino, _end);
25!
2825

2826
      if (winCode != TSDB_CODE_SUCCESS) {
25!
2827
        SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, srcUid, srcStartTsCol[i], srcStartTsCol[i], ver);
×
2828
        if (!pPreRes) {
×
2829
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2830
          continue;
×
2831
        }
2832
        printDataBlock(pPreRes, "pre res", GET_TASKID(pInfo->pStreamScanOp->pTaskInfo));
×
2833
        code = calBlockTbName(pInfo, pPreRes, 0);
×
2834
        QUERY_CHECK_CODE(code, lino, _end);
×
2835
        memcpy(varDataVal(tbname), pPreRes->info.parTbName, strlen(pPreRes->info.parTbName));
×
2836
      } else {
2837
        memcpy(varDataVal(tbname), parTbname, TSDB_TABLE_NAME_LEN);
25✔
2838
      }
2839
      varDataSetLen(tbname, strlen(varDataVal(tbname)));
25✔
2840
      pInfo->stateStore.streamStateFreeVal(parTbname);
25✔
2841
    }
2842
    code = appendDataToSpecialBlock(pDestBlock, srcStartTsCol + i, srcEndTsCol + i, srcUidData + i, &groupId,
73✔
2843
                                    tbname[0] == 0 ? NULL : tbname);
73✔
2844
    QUERY_CHECK_CODE(code, lino, _end);
73!
2845
  }
2846

2847
_end:
73✔
2848
  if (code != TSDB_CODE_SUCCESS) {
73!
2849
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2850
  }
2851
  return code;
73✔
2852
}
2853

2854
int32_t generateDeleteResultBlock(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
135✔
2855
  blockDataCleanup(pDestBlock);
135✔
2856
  int32_t rows = pSrcBlock->info.rows;
135✔
2857
  if (rows == 0) {
135!
2858
    return TSDB_CODE_SUCCESS;
×
2859
  }
2860
  int32_t code = blockDataEnsureCapacity(pDestBlock, rows);
135✔
2861
  if (code != TSDB_CODE_SUCCESS) {
135!
2862
    return code;
×
2863
  }
2864
  if (pInfo->partitionSup.needCalc) {
135✔
2865
    return generatePartitionDelResBlock(pInfo, pSrcBlock, pDestBlock);
62✔
2866
  }
2867
  return generateDeleteResultBlockImpl(pInfo, pSrcBlock, pDestBlock);
73✔
2868
}
2869

2870
int32_t generateScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock, EStreamType type) {
2,810✔
2871
  int32_t code = TSDB_CODE_SUCCESS;
2,810✔
2872
  int32_t lino = 0;
2,810✔
2873
  if (isIntervalWindow(pInfo)) {
2,810✔
2874
    code = generateIntervalScanRange(pInfo, pSrcBlock, pDestBlock, type);
1,646✔
2875
    QUERY_CHECK_CODE(code, lino, _end);
1,647!
2876
  } else if (isSessionWindow(pInfo) || isStateWindow(pInfo)) {
1,163✔
2877
    code = generateSessionScanRange(pInfo, pSrcBlock, pDestBlock, type);
1,021✔
2878
    QUERY_CHECK_CODE(code, lino, _end);
1,021!
2879
  } else if (isCountWindow(pInfo)) {
142✔
2880
    code = generateCountScanRange(pInfo, pSrcBlock, pDestBlock, type);
67✔
2881
    QUERY_CHECK_CODE(code, lino, _end);
67!
2882
  } else if (isTimeSlice(pInfo)) {
75✔
2883
    code = generateTimeSliceScanRange(pInfo, pSrcBlock, pDestBlock, type);
15✔
2884
    QUERY_CHECK_CODE(code, lino, _end);
15!
2885
  } else {
2886
    code = generateDeleteResultBlock(pInfo, pSrcBlock, pDestBlock);
60✔
2887
    QUERY_CHECK_CODE(code, lino, _end);
60!
2888
  }
2889
  pDestBlock->info.type = STREAM_CLEAR;
2,810✔
2890
  pDestBlock->info.version = pSrcBlock->info.version;
2,810✔
2891
  pDestBlock->info.dataLoad = 1;
2,810✔
2892
  code = blockDataUpdateTsWindow(pDestBlock, 0);
2,810✔
2893
  QUERY_CHECK_CODE(code, lino, _end);
2,810!
2894

2895
_end:
2,810✔
2896
  if (code != TSDB_CODE_SUCCESS) {
2,810!
2897
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2898
  }
2899
  return code;
2,810✔
2900
}
2901

2902
int32_t appendDataToSpecialBlock(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, uint64_t* pUid, uint64_t* pGp,
6,014✔
2903
                                 void* pTbName) {
2904
  return appendOneRowToSpecialBlockImpl(pBlock, pStartTs, pEndTs, pStartTs, pEndTs, pUid, pGp, pTbName, NULL);
6,014✔
2905
}
2906

2907
bool checkExpiredData(SStateStore* pAPI, SUpdateInfo* pUpdateInfo, STimeWindowAggSupp* pTwSup, uint64_t tableId,
6,625✔
2908
                      TSKEY ts, void* pPkVal, int32_t len) {
2909
  bool isExpired = false;
6,625✔
2910
  bool isInc = pAPI->isIncrementalTimeStamp(pUpdateInfo, tableId, ts, pPkVal, len);
6,625✔
2911
  if (!isInc) {
6,625✔
2912
    isExpired = isOverdue(ts, pTwSup);
363✔
2913
  }
2914
  return isExpired;
6,625✔
2915
}
2916

2917
int32_t checkUpdateData(SStreamScanInfo* pInfo, bool invertible, SSDataBlock* pBlock, bool out) {
335,153✔
2918
  int32_t code = TSDB_CODE_SUCCESS;
335,153✔
2919
  int32_t lino = 0;
335,153✔
2920
  if (out) {
335,153✔
2921
    blockDataCleanup(pInfo->pUpdateDataRes);
332,663✔
2922
    code = blockDataEnsureCapacity(pInfo->pUpdateDataRes, pBlock->info.rows * 2);
332,652✔
2923
    QUERY_CHECK_CODE(code, lino, _end);
332,654!
2924
  }
2925
  SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, pInfo->primaryTsIndex);
335,144✔
2926
  TSKEY*           tsCol = (TSKEY*)pColDataInfo->pData;
335,143✔
2927
  SColumnInfoData* pPkColDataInfo = NULL;
335,143✔
2928
  if (hasPrimaryKeyCol(pInfo)) {
335,143✔
2929
    pPkColDataInfo = taosArrayGet(pBlock->pDataBlock, pInfo->primaryKeyIndex);
308,514✔
2930
  }
2931

2932
  bool tableInserted = pInfo->stateStore.updateInfoIsTableInserted(pInfo->pUpdateInfo, pBlock->info.id.uid);
335,135✔
2933
  for (int32_t rowId = 0; rowId < pBlock->info.rows; rowId++) {
51,128,442✔
2934
    SResultRowInfo dumyInfo;
2935
    dumyInfo.cur.pageId = -1;
50,799,003✔
2936
    bool        isClosed = false;
50,799,003✔
2937
    STimeWindow win = {.skey = INT64_MIN, .ekey = INT64_MAX};
50,799,003✔
2938
    bool        overDue = isOverdue(tsCol[rowId], &pInfo->twAggSup);
50,799,003✔
2939
    if (pInfo->igExpired && overDue) {
50,792,603✔
2940
      continue;
304✔
2941
    }
2942

2943
    if (tableInserted && overDue) {
50,792,299✔
2944
      win = getActiveTimeWindow(NULL, &dumyInfo, tsCol[rowId], &pInfo->interval, TSDB_ORDER_ASC);
30,715,343✔
2945
      isClosed = isCloseWindow(&win, &pInfo->twAggSup);
30,677,152✔
2946
    }
2947
    // must check update info first.
2948
    void*   pPkVal = NULL;
50,742,157✔
2949
    int32_t pkLen = 0;
50,742,157✔
2950
    if (hasPrimaryKeyCol(pInfo)) {
50,742,157✔
2951
      pPkVal = colDataGetData(pPkColDataInfo, rowId);
30,761,927!
2952
      pkLen = colDataGetRowLength(pPkColDataInfo, rowId);
30,761,927✔
2953
    }
2954
    bool update =
2955
        pInfo->stateStore.updateInfoIsUpdated(pInfo->pUpdateInfo, pBlock->info.id.uid, tsCol[rowId], pPkVal, pkLen);
50,738,327✔
2956
    bool isDeleted =
50,792,981✔
2957
        isClosed && isSignleIntervalWindow(pInfo) &&
81,027,712!
2958
        isDeletedStreamWindow(&win, pBlock->info.id.groupId, pInfo->pState, &pInfo->twAggSup, &pInfo->stateStore);
30,231,862✔
2959
    if ((update || isDeleted) && out) {
50,792,981!
2960
      qDebug("stream update check not pass, update %d, deleted Win %d", update, isDeleted);
1,694✔
2961
      uint64_t gpId = 0;
1,694✔
2962
      code = appendPkToSpecialBlock(pInfo->pUpdateDataRes, tsCol, pPkColDataInfo, rowId, &pBlock->info.id.uid, &gpId,
1,694✔
2963
                                    NULL);
2964
      QUERY_CHECK_CODE(code, lino, _end);
1,690!
2965
      if (isDeleted && pInfo->partitionSup.needCalc) {
1,690!
2966
        gpId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pBlock, rowId);
×
2967
        code = appendPkToSpecialBlock(pInfo->pUpdateDataRes, tsCol, pPkColDataInfo, rowId, &pBlock->info.id.uid, &gpId,
×
2968
                                      NULL);
2969
        QUERY_CHECK_CODE(code, lino, _end);
×
2970
      }
2971
    }
2972
  }
2973
  if (out && pInfo->pUpdateDataRes->info.rows > 0) {
329,439!
2974
    pInfo->pUpdateDataRes->info.version = pBlock->info.version;
1,604✔
2975
    pInfo->pUpdateDataRes->info.dataLoad = 1;
1,604✔
2976
    code = blockDataUpdateTsWindow(pInfo->pUpdateDataRes, 0);
1,604✔
2977
    QUERY_CHECK_CODE(code, lino, _end);
1,605!
2978
    pInfo->pUpdateDataRes->info.type = pInfo->partitionSup.needCalc ? STREAM_DELETE_DATA : STREAM_CLEAR;
1,605✔
2979
  }
2980

2981
_end:
327,835✔
2982
  if (code != TSDB_CODE_SUCCESS) {
329,440!
2983
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2984
  }
2985
  return code;
335,177✔
2986
}
2987

2988
int32_t doBlockDataWindowFilter(SSDataBlock* pBlock, int32_t tsIndex, STimeWindow* pWindow, const char* id) {
860,516✔
2989
  int32_t code = TSDB_CODE_SUCCESS;
860,516✔
2990
  int32_t lino = 0;
860,516✔
2991
  bool*   p = NULL;
860,516✔
2992
  if (pWindow->skey != INT64_MIN || pWindow->ekey != INT64_MAX) {
860,516✔
2993
    p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
43,829!
2994
    QUERY_CHECK_NULL(p, code, lino, _end, terrno);
43,878!
2995
    bool hasUnqualified = false;
43,878✔
2996

2997
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, tsIndex);
43,878✔
2998

2999
    if (pWindow->skey != INT64_MIN) {
43,878✔
3000
      qDebug("%s filter for additional history window, skey:%" PRId64, id, pWindow->skey);
24,151✔
3001

3002
      QUERY_CHECK_NULL(pCol->pData, code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
24,151!
3003
      for (int32_t i = 0; i < pBlock->info.rows; ++i) {
2,305,007✔
3004
        int64_t* ts = (int64_t*)colDataGetData(pCol, i);
2,280,856!
3005
        p[i] = (*ts >= pWindow->skey);
2,280,856✔
3006

3007
        if (!p[i]) {
2,280,856✔
3008
          hasUnqualified = true;
1,304✔
3009
        }
3010
      }
3011
    } else if (pWindow->ekey != INT64_MAX) {
19,727!
3012
      qDebug("%s filter for additional history window, ekey:%" PRId64, id, pWindow->ekey);
19,727✔
3013
      for (int32_t i = 0; i < pBlock->info.rows; ++i) {
1,939,165✔
3014
        int64_t* ts = (int64_t*)colDataGetData(pCol, i);
1,919,438!
3015
        p[i] = (*ts <= pWindow->ekey);
1,919,438✔
3016

3017
        if (!p[i]) {
1,919,438✔
3018
          hasUnqualified = true;
1,918,952✔
3019
        }
3020
      }
3021
    }
3022

3023
    if (hasUnqualified) {
43,878✔
3024
      code = trimDataBlock(pBlock, pBlock->info.rows, p);
20,545✔
3025
      QUERY_CHECK_CODE(code, lino, _end);
20,545!
3026
    }
3027
  }
3028

3029
_end:
860,565✔
3030
  taosMemoryFree(p);
860,565!
3031
  if (code != TSDB_CODE_SUCCESS) {
860,549!
3032
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3033
  }
3034
  return code;
860,545✔
3035
}
3036

3037
static int32_t doBlockDataPrimaryKeyFilter(SSDataBlock* pBlock, STqOffsetVal* offset) {
×
3038
  int32_t code = TSDB_CODE_SUCCESS;
×
3039
  int32_t lino = 0;
×
3040
  if (pBlock->info.window.skey != offset->ts || offset->primaryKey.type == 0) {
×
3041
    return code;
×
3042
  }
3043
  bool* p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
×
3044
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
×
3045
  bool hasUnqualified = false;
×
3046

3047
  SColumnInfoData* pColTs = taosArrayGet(pBlock->pDataBlock, 0);
×
3048
  SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
3049

3050
  qDebug("doBlockDataWindowFilter primary key, ts:%" PRId64 " %" PRId64, offset->ts,
×
3051
         VALUE_GET_TRIVIAL_DATUM(&offset->primaryKey));
3052
  QUERY_CHECK_CONDITION((pColPk->info.type == offset->primaryKey.type), code, lino, _end,
×
3053
                        TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
3054

3055
  __compar_fn_t func = getComparFunc(pColPk->info.type, 0);
×
3056
  QUERY_CHECK_NULL(func, code, lino, _end, terrno);
×
3057
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
×
3058
    int64_t* ts = (int64_t*)colDataGetData(pColTs, i);
×
3059
    void*    data = colDataGetData(pColPk, i);
×
3060
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
3061
      void* tmq = taosMemoryMalloc(offset->primaryKey.nData + VARSTR_HEADER_SIZE);
×
3062
      QUERY_CHECK_NULL(tmq, code, lino, _end, terrno);
×
3063
      memcpy(varDataVal(tmq), offset->primaryKey.pData, offset->primaryKey.nData);
×
3064
      varDataLen(tmq) = offset->primaryKey.nData;
×
3065
      p[i] = (*ts > offset->ts) || (func(data, tmq) > 0);
×
3066
      taosMemoryFree(tmq);
×
3067
    } else {
3068
      p[i] = (*ts > offset->ts) || (func(data, VALUE_GET_DATUM(&offset->primaryKey, pColPk->info.type)) > 0);
×
3069
    }
3070

3071
    if (!p[i]) {
×
3072
      hasUnqualified = true;
×
3073
    }
3074
  }
3075

3076
  if (hasUnqualified) {
×
3077
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
×
3078
    QUERY_CHECK_CODE(code, lino, _end);
×
3079
  }
3080

3081
_end:
×
3082
  taosMemoryFree(p);
×
3083
  if (code != TSDB_CODE_SUCCESS) {
×
3084
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3085
  }
3086
  return code;
×
3087
}
3088

3089
// re-build the delete block, ONLY according to the split timestamp
3090
int32_t rebuildDeleteBlockData(SSDataBlock* pBlock, STimeWindow* pWindow, const char* id) {
2,481✔
3091
  int32_t code = TSDB_CODE_SUCCESS;
2,481✔
3092
  int32_t lino = 0;
2,481✔
3093
  int32_t numOfRows = pBlock->info.rows;
2,481✔
3094
  bool*   p = taosMemoryCalloc(numOfRows, sizeof(bool));
2,481!
3095
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
2,481!
3096

3097
  bool    hasUnqualified = false;
2,481✔
3098
  int64_t skey = pWindow->skey;
2,481✔
3099
  int64_t ekey = pWindow->ekey;
2,481✔
3100

3101
  SColumnInfoData* pSrcStartCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
2,481✔
3102
  uint64_t*        tsStartCol = (uint64_t*)pSrcStartCol->pData;
2,481✔
3103
  SColumnInfoData* pSrcEndCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
2,481✔
3104
  uint64_t*        tsEndCol = (uint64_t*)pSrcEndCol->pData;
2,481✔
3105

3106
  if (pWindow->skey != INT64_MIN) {
2,481!
3107
    for (int32_t i = 0; i < numOfRows; i++) {
×
3108
      if (tsStartCol[i] < skey) {
×
3109
        tsStartCol[i] = skey;
×
3110
      }
3111

3112
      if (tsEndCol[i] >= skey) {
×
3113
        p[i] = true;
×
3114
      } else {  // this row should be removed, since it is not in this query time window, which is [skey, INT64_MAX]
3115
        hasUnqualified = true;
×
3116
      }
3117
    }
3118
  } else if (pWindow->ekey != INT64_MAX) {
2,481!
3119
    for (int32_t i = 0; i < numOfRows; ++i) {
×
3120
      if (tsEndCol[i] > ekey) {
×
3121
        tsEndCol[i] = ekey;
×
3122
      }
3123

3124
      if (tsStartCol[i] <= ekey) {
×
3125
        p[i] = true;
×
3126
      } else {
3127
        hasUnqualified = true;
×
3128
      }
3129
    }
3130
  }
3131

3132
  if (hasUnqualified) {
2,481!
3133
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
×
3134
    qDebug("%s re-build delete datablock, start key revised to:%" PRId64 ", rows:%" PRId64, id, skey,
×
3135
           pBlock->info.rows);
3136
    QUERY_CHECK_CODE(code, lino, _end);
×
3137
  } else {
3138
    qDebug("%s not update the delete block", id);
2,481✔
3139
  }
3140

3141
_end:
1,173✔
3142
  taosMemoryFree(p);
2,481!
3143
  if (code != TSDB_CODE_SUCCESS) {
2,481!
3144
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3145
  }
3146
  return code;
2,481✔
3147
}
3148

3149
int32_t colIdComparFn(const void* param1, const void* param2) {
5,326,832✔
3150
  int32_t p1 = *(int32_t*)param1;
5,326,832✔
3151
  int32_t p2 = *(int32_t*)param2;
5,326,832✔
3152

3153
  if (p1 == p2) {
5,326,832!
3154
    return 0;
×
3155
  } else {
3156
    return (p1 < p2) ? -1 : 1;
5,326,832✔
3157
  }
3158
}
3159

3160
int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock, STimeWindow* pTimeWindow, bool filter) {
860,533✔
3161
  int32_t         code = TSDB_CODE_SUCCESS;
860,533✔
3162
  int32_t         lino = 0;
860,533✔
3163
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
860,533✔
3164
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
860,533✔
3165
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
860,533✔
3166
  const char*     id = GET_TASKID(pTaskInfo);
860,533✔
3167
  bool            isVtableSourceScan = (pTaskInfo->pSubplan->pVTables != NULL);
860,533✔
3168

3169
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
860,533✔
3170
  QUERY_CHECK_CODE(code, lino, _end);
860,563!
3171

3172
  pBlockInfo->rows = pBlock->info.rows;
860,563✔
3173
  pBlockInfo->id.uid = pBlock->info.id.uid;
860,563✔
3174
  pBlockInfo->type = STREAM_NORMAL;
860,563✔
3175
  pBlockInfo->version = pBlock->info.version;
860,563✔
3176

3177
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
860,563✔
3178
  if (!isVtableSourceScan) {
860,563✔
3179
    pBlockInfo->id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
860,556✔
3180
  } else {
3181
    // use original table uid as groupId for vtable
3182
    pBlockInfo->id.groupId = pBlock->info.id.groupId;
7✔
3183
  }
3184

3185
  SArray* pColList = taosArrayInit(4, sizeof(int32_t));
860,615✔
3186
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
860,595!
3187

3188
  // todo extract method
3189
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
4,114,266✔
3190
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
3,253,768✔
3191
    if (!pColMatchInfo->needOutput) {
3,253,689✔
3192
      continue;
2✔
3193
    }
3194

3195
    bool colExists = false;
3,253,687✔
3196
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
8,805,341!
3197
      SColumnInfoData* pResCol = NULL;
8,805,344✔
3198
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
8,805,344✔
3199
      QUERY_CHECK_CODE(code, lino, _end);
8,804,892!
3200

3201
      if (pResCol->info.colId == pColMatchInfo->colId) {
8,804,892✔
3202
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
3,253,238✔
3203
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
3,253,119✔
3204
        QUERY_CHECK_CODE(code, lino, _end);
3,253,439!
3205

3206
        colExists = true;
3,253,439✔
3207
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
3,253,439✔
3208
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
3,253,668!
3209
        break;
3,253,668✔
3210
      }
3211
    }
3212

3213
    // the required column does not exists in submit block, let's set it to be all null value
3214
    if (!colExists) {
3,253,666✔
3215
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
4✔
3216
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
4!
3217
      void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
4✔
3218
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
4!
3219
    }
3220
  }
3221

3222
  // currently only the tbname pseudo column
3223
  if (pInfo->numOfPseudoExpr > 0 && !isVtableSourceScan) {
860,317!
3224
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
368,253✔
3225
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
368,253✔
3226
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
3227
    if (code) {
368,275!
3228
      QUERY_CHECK_CODE(code, lino, _end);
×
3229
    }
3230

3231
    // reset the error code.
3232
    terrno = 0;
368,275✔
3233

3234
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
1,399,925✔
3235
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
1,031,675✔
3236
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,031,653!
3237
    }
3238
  }
3239

3240
  taosArraySort(pColList, colIdComparFn);
860,314✔
3241

3242
  int32_t i = 0, j = 0;
860,552✔
3243
  while (i < taosArrayGetSize(pColList)) {
5,145,993✔
3244
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
4,285,319✔
3245
    if (slot1 > j) {
4,285,451✔
3246
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
10✔
3247
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
3248
      j += 1;
×
3249
    } else {
3250
      i += 1;
4,285,441✔
3251
      j += 1;
4,285,441✔
3252
    }
3253
  }
3254

3255
  while (j < taosArrayGetSize(pInfo->pRes->pDataBlock)) {
862,253✔
3256
    SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
1,742✔
3257
    colDataSetNNULL(pDst, 0, pBlockInfo->rows);
1,742✔
3258
    j += 1;
1,742✔
3259
  }
3260

3261
  taosArrayDestroy(pColList);
860,501✔
3262

3263
  if (filter) {
860,572✔
3264
    code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
490,283✔
3265
    QUERY_CHECK_CODE(code, lino, _end);
490,234!
3266
  }
3267

3268
  // filter the block extracted from WAL files, according to the time window apply additional time window filter
3269
  code = doBlockDataWindowFilter(pInfo->pRes, pInfo->primaryTsIndex, pTimeWindow, id);
860,523✔
3270
  QUERY_CHECK_CODE(code, lino, _end);
860,545!
3271
  pInfo->pRes->info.dataLoad = 1;
860,545✔
3272

3273
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
860,545✔
3274
  QUERY_CHECK_CODE(code, lino, _end);
860,555!
3275
  if (pInfo->pRes->info.rows == 0) {
860,555✔
3276
    return 0;
22,874✔
3277
  }
3278

3279
  code = calBlockTbName(pInfo, pInfo->pRes, 0);
837,681✔
3280
  QUERY_CHECK_CODE(code, lino, _end);
837,598!
3281

3282
_end:
837,598✔
3283
  if (code != TSDB_CODE_SUCCESS) {
837,598!
3284
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3285
  }
3286
  return code;
837,600✔
3287
}
3288

3289
static int32_t processPrimaryKey(SSDataBlock* pBlock, bool hasPrimaryKey, STqOffsetVal* offset) {
22,622✔
3290
  int32_t code = TSDB_CODE_SUCCESS;
22,622✔
3291
  int32_t lino = 0;
22,622✔
3292
  SValue  val = {0};
22,622✔
3293
  if (hasPrimaryKey) {
22,622!
3294
    code = doBlockDataPrimaryKeyFilter(pBlock, offset);
×
3295
    if (code != TSDB_CODE_SUCCESS) {
×
3296
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3297
      return code;
×
3298
    }
3299
    SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
3300

3301
    if (pBlock->info.rows < 1) {
×
3302
      return code;
×
3303
    }
3304
    void* tmp = colDataGetData(pColPk, pBlock->info.rows - 1);
×
3305
    val.type = pColPk->info.type;
×
3306
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
3307
      val.pData = taosMemoryMalloc(varDataLen(tmp));
×
3308
      QUERY_CHECK_NULL(val.pData, code, lino, _end, terrno);
×
3309
      val.nData = varDataLen(tmp);
×
3310
      memcpy(val.pData, varDataVal(tmp), varDataLen(tmp));
×
3311
    } else {
3312
      valueSetDatum(&val, pColPk->info.type, tmp, pColPk->info.bytes);
×
3313
    }
3314
  }
3315
  tqOffsetResetToData(offset, pBlock->info.id.uid, pBlock->info.window.ekey, val);
45,263!
3316

3317
_end:
22,641✔
3318
  if (code != TSDB_CODE_SUCCESS) {
22,641!
3319
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3320
  }
3321
  return code;
22,639✔
3322
}
3323

3324
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
586,027✔
3325
  int32_t        code = TSDB_CODE_SUCCESS;
586,027✔
3326
  int32_t        lino = 0;
586,027✔
3327
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
586,027✔
3328
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
586,027✔
3329

3330
  SStreamScanInfo* pInfo = pOperator->info;
586,027✔
3331
  const char*      id = GET_TASKID(pTaskInfo);
586,027✔
3332

3333
  qDebug("start to exec queue scan, %s", id);
586,027✔
3334

3335
  if (isTaskKilled(pTaskInfo)) {
586,031!
3336
    (*ppRes) = NULL;
×
3337
    return pTaskInfo->code;
×
3338
  }
3339

3340
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
586,037✔
3341
    while (1) {
×
3342
      SSDataBlock* pResult = NULL;
20,332✔
3343
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
20,332✔
3344
      QUERY_CHECK_CODE(code, lino, _end);
20,275!
3345

3346
      if (pResult && pResult->info.rows > 0) {
20,275!
3347
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
20,082✔
3348
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
20,123✔
3349
        QUERY_CHECK_CODE(code, lino, _end);
20,106!
3350
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
20,106✔
3351
        if (pResult->info.rows > 0) {
20,113!
3352
          (*ppRes) = pResult;
20,113✔
3353
          return code;
20,113✔
3354
        }
3355
      } else {
3356
        break;
3357
      }
3358
    }
3359

3360
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
193✔
3361
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
193✔
3362

3363
    pTSInfo->base.dataReader = NULL;
201✔
3364
    int64_t validVer = pTaskInfo->streamInfo.snapshotVer + 1;
201✔
3365
    qDebug("queue scan tsdb over, switch to wal ver %" PRId64, validVer);
201✔
3366
    if (pAPI->tqReaderFn.tqReaderSeek(pInfo->tqReader, validVer, pTaskInfo->id.str) < 0) {
201✔
3367
      (*ppRes) = NULL;
108✔
3368
      return code;
108✔
3369
    }
3370

3371
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, validVer);
93✔
3372
  }
3373

3374
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
565,798!
3375
    while (1) {
2,329✔
3376
      bool hasResult =
3377
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
568,127✔
3378

3379
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
568,110✔
3380
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
568,102✔
3381

3382
      // curVersion move to next
3383
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pWalReader->curVersion);
568,105✔
3384

3385
      // use ts to pass time when replay, because ts not used if type is log
3386
      pTaskInfo->streamInfo.currentOffset.ts = pAPI->tqReaderFn.tqGetResultBlockTime(pInfo->tqReader);
568,105✔
3387

3388
      if (hasResult) {
568,102✔
3389
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
490,282✔
3390
               pTaskInfo->streamInfo.currentOffset.version);
3391
        blockDataCleanup(pInfo->pRes);
490,290✔
3392
        STimeWindow defaultWindow = {.skey = INT64_MIN, .ekey = INT64_MAX};
490,240✔
3393
        code = setBlockIntoRes(pInfo, pRes, &defaultWindow, true);
490,240✔
3394
        QUERY_CHECK_CODE(code, lino, _end);
490,172!
3395
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
490,172✔
3396
               pTaskInfo->streamInfo.currentOffset.version);
3397
        if (pInfo->pRes->info.rows > 0) {
490,203✔
3398
          (*ppRes) = pInfo->pRes;
487,874✔
3399
          return code;
487,874✔
3400
        }
3401
      } else {
3402
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
77,820✔
3403
        (*ppRes) = NULL;
77,818✔
3404
        return code;
77,818✔
3405
      }
3406
    }
3407
  } else {
3408
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
×
3409
    (*ppRes) = NULL;
×
3410
    return code;
×
3411
  }
3412

3413
_end:
×
3414
  if (code != TSDB_CODE_SUCCESS) {
×
3415
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3416
    pTaskInfo->code = code;
×
3417
    T_LONG_JMP(pTaskInfo->env, code);
×
3418
  }
3419
  (*ppRes) = NULL;
×
3420
  return code;
×
3421
}
3422

3423
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
×
3424
  SSDataBlock* pRes = NULL;
×
3425
  int32_t      code = doQueueScanNext(pOperator, &pRes);
×
3426
  return pRes;
×
3427
}
3428

3429
int32_t filterDelBlockByUid(SSDataBlock* pDst, const SSDataBlock* pSrc, STqReader* pReader, SStoreTqReader* pReaderFn) {
2,481✔
3430
  int32_t    code = TSDB_CODE_SUCCESS;
2,481✔
3431
  int32_t    lino = 0;
2,481✔
3432
  int32_t    rows = pSrc->info.rows;
2,481✔
3433
  code = blockDataEnsureCapacity(pDst, rows);
2,481✔
3434
  QUERY_CHECK_CODE(code, lino, _end);
2,481!
3435

3436
  SColumnInfoData* pSrcStartCol = taosArrayGet(pSrc->pDataBlock, START_TS_COLUMN_INDEX);
2,481✔
3437
  uint64_t*        startCol = (uint64_t*)pSrcStartCol->pData;
2,481✔
3438
  SColumnInfoData* pSrcEndCol = taosArrayGet(pSrc->pDataBlock, END_TS_COLUMN_INDEX);
2,481✔
3439
  uint64_t*        endCol = (uint64_t*)pSrcEndCol->pData;
2,481✔
3440
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrc->pDataBlock, UID_COLUMN_INDEX);
2,481✔
3441
  uint64_t*        uidCol = (uint64_t*)pSrcUidCol->pData;
2,481✔
3442

3443
  SColumnInfoData* pDstStartCol = taosArrayGet(pDst->pDataBlock, START_TS_COLUMN_INDEX);
2,481✔
3444
  SColumnInfoData* pDstEndCol = taosArrayGet(pDst->pDataBlock, END_TS_COLUMN_INDEX);
2,481✔
3445
  SColumnInfoData* pDstUidCol = taosArrayGet(pDst->pDataBlock, UID_COLUMN_INDEX);
2,481✔
3446

3447
  int32_t j = 0;
2,481✔
3448
  for (int32_t i = 0; i < rows; i++) {
4,969✔
3449
    if (pReaderFn->tqReaderIsQueriedTable(pReader, uidCol[i])) {
2,488✔
3450
      code = colDataSetVal(pDstStartCol, j, (const char*)&startCol[i], false);
1,366✔
3451
      QUERY_CHECK_CODE(code, lino, _end);
1,366!
3452

3453
      code = colDataSetVal(pDstEndCol, j, (const char*)&endCol[i], false);
1,366✔
3454
      QUERY_CHECK_CODE(code, lino, _end);
1,366!
3455

3456
      code = colDataSetVal(pDstUidCol, j, (const char*)&uidCol[i], false);
1,366✔
3457
      QUERY_CHECK_CODE(code, lino, _end);
1,366!
3458

3459
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, GROUPID_COLUMN_INDEX), j);
1,366✔
3460
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX), j);
1,366✔
3461
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX), j);
1,366✔
3462
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, TABLE_NAME_COLUMN_INDEX), j);
1,366✔
3463
      j++;
1,366✔
3464
    }
3465
  }
3466

3467
  uint32_t cap = pDst->info.capacity;
2,481✔
3468
  pDst->info = pSrc->info;
2,481✔
3469
  pDst->info.rows = j;
2,481✔
3470
  pDst->info.capacity = cap;
2,481✔
3471

3472
_end:
2,481✔
3473
  if (code != TSDB_CODE_SUCCESS) {
2,481!
3474
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3475
  }
3476
  return code;
2,481✔
3477
}
3478

3479
// for partition by tag
3480
int32_t setBlockGroupIdByUid(SStreamScanInfo* pInfo, SSDataBlock* pBlock) {
2,481✔
3481
  int32_t          code = TSDB_CODE_SUCCESS;
2,481✔
3482
  int32_t          lino = 0;
2,481✔
3483
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
2,481✔
3484
  TSKEY*           startTsCol = (TSKEY*)pStartTsCol->pData;
2,481✔
3485
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
2,481✔
3486
  uint64_t*        gpCol = (uint64_t*)pGpCol->pData;
2,481✔
3487
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
2,481✔
3488
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
2,481✔
3489
  int32_t          rows = pBlock->info.rows;
2,481✔
3490
  if (!pInfo->partitionSup.needCalc) {
2,481✔
3491
    for (int32_t i = 0; i < rows; i++) {
2,584✔
3492
      uint64_t groupId = getGroupIdByUid(pInfo, uidCol[i]);
926✔
3493
      code = colDataSetVal(pGpCol, i, (const char*)&groupId, false);
926✔
3494
      QUERY_CHECK_CODE(code, lino, _end);
926!
3495
    }
3496
  }
3497

3498
_end:
2,481✔
3499
  if (code != TSDB_CODE_SUCCESS) {
2,481!
3500
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3501
  }
3502
  return code;
2,481✔
3503
}
3504

3505
int32_t doCheckUpdate(SStreamScanInfo* pInfo, TSKEY endKey, SSDataBlock* pBlock) {
349,364✔
3506
  int32_t code = TSDB_CODE_SUCCESS;
349,364✔
3507
  int32_t lino = 0;
349,364✔
3508
  if (pInfo->pUpdateInfo) {
349,364✔
3509
    pInfo->pUpdateInfo->maxDataVersion = TMAX(pInfo->pUpdateInfo->maxDataVersion, pBlock->info.version);
348,168✔
3510
  }
3511
  if (!pInfo->igCheckUpdate && pInfo->pUpdateInfo) {
349,364✔
3512
    code = checkUpdateData(pInfo, true, pBlock, true);
332,667✔
3513
    QUERY_CHECK_CODE(code, lino, _end);
332,686!
3514
    pInfo->twAggSup.maxTs = TMAX(pInfo->twAggSup.maxTs, endKey);
332,686✔
3515
    if (pInfo->pUpdateDataRes->info.rows > 0) {
332,686✔
3516
      pInfo->updateResIndex = 0;
1,607✔
3517
      if (pInfo->pUpdateDataRes->info.type == STREAM_CLEAR) {
1,607✔
3518
        pInfo->scanMode = STREAM_SCAN_FROM_UPDATERES;
1,350✔
3519
      } else if (pInfo->pUpdateDataRes->info.type == STREAM_INVERT) {
257!
3520
        pInfo->scanMode = STREAM_SCAN_FROM_RES;
×
3521
        // return pInfo->pUpdateDataRes;
3522
      } else if (pInfo->pUpdateDataRes->info.type == STREAM_DELETE_DATA) {
257✔
3523
        pInfo->scanMode = STREAM_SCAN_FROM_DELETE_DATA;
256✔
3524
      }
3525
    }
3526
  }
3527

3528
_end:
347,777✔
3529
  if (code != TSDB_CODE_SUCCESS) {
349,383!
3530
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3531
  }
3532
  return code;
349,396✔
3533
}
3534

3535
int32_t streamScanOperatorEncode(SStreamScanInfo* pInfo, void** pBuff, int32_t* pLen) {
55✔
3536
  int32_t   code = TSDB_CODE_SUCCESS;
55✔
3537
  int32_t   lino = 0;
55✔
3538
  int32_t   len = 0;
55✔
3539
  SEncoder* pEnCoder = NULL;
55✔
3540
  SEncoder* pScanEnCoder = NULL;
55✔
3541

3542
  len += encodeSTimeWindowAggSupp(NULL, &pInfo->twAggSup);
55✔
3543
  SEncoder encoder = {0};
55✔
3544
  pEnCoder = &encoder;
55✔
3545
  tEncoderInit(pEnCoder, NULL, 0);
55✔
3546
  if (tStartEncode(pEnCoder) != 0) {
55!
3547
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3548
    QUERY_CHECK_CODE(code, lino, _end);
×
3549
  }
3550
  code = pInfo->stateStore.updateInfoSerialize(pEnCoder, pInfo->pUpdateInfo);
55✔
3551
  QUERY_CHECK_CODE(code, lino, _end);
55!
3552

3553
  if (tEncodeI64(pEnCoder, pInfo->lastScanRange.skey) < 0) {
110!
3554
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3555
    QUERY_CHECK_CODE(code, lino, _end);
×
3556
  }
3557
  if (tEncodeI64(pEnCoder, pInfo->lastScanRange.ekey) < 0) {
110!
3558
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3559
    QUERY_CHECK_CODE(code, lino, _end);
×
3560
  }
3561

3562
  tEndEncode(pEnCoder);
55✔
3563
  len += encoder.pos;
55✔
3564
  tEncoderClear(pEnCoder);
55✔
3565
  pEnCoder = NULL;
55✔
3566

3567
  *pBuff = taosMemoryCalloc(1, len);
55!
3568
  if (!(*pBuff)) {
55!
3569
    code = terrno;
×
3570
    QUERY_CHECK_CODE(code, lino, _end);
×
3571
  }
3572
  void* buf = *pBuff;
55✔
3573
  int32_t stwLen = encodeSTimeWindowAggSupp(&buf, &pInfo->twAggSup);
55✔
3574

3575
  SEncoder scanEncoder = {0};
55✔
3576
  pScanEnCoder = &scanEncoder;
55✔
3577
  tEncoderInit(pScanEnCoder, buf, len - stwLen);
55✔
3578
  if (tStartEncode(pScanEnCoder) != 0) {
55!
3579
    code = TSDB_CODE_FAILED;
×
3580
    QUERY_CHECK_CODE(code, lino, _end);
×
3581
  }
3582
  code = pInfo->stateStore.updateInfoSerialize(pScanEnCoder, pInfo->pUpdateInfo);
55✔
3583
  QUERY_CHECK_CODE(code, lino, _end);
55!
3584

3585
  if (tEncodeI64(pScanEnCoder, pInfo->lastScanRange.skey) < 0) {
110!
3586
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3587
    QUERY_CHECK_CODE(code, lino, _end);
×
3588
  }
3589
  if (tEncodeI64(pScanEnCoder, pInfo->lastScanRange.ekey) < 0) {
110!
3590
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3591
    QUERY_CHECK_CODE(code, lino, _end);
×
3592
  }
3593

3594
  qDebug("%s last scan range %d. %" PRId64 ",%" PRId64, __func__, __LINE__, pInfo->lastScanRange.skey, pInfo->lastScanRange.ekey);
55✔
3595

3596
  *pLen = len;
55✔
3597

3598
_end:
55✔
3599
  if (code != TSDB_CODE_SUCCESS) {
55!
3600
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3601
  }
3602
  if (pEnCoder != NULL) {
55!
3603
    tEndEncode(pEnCoder);
×
3604
    tEncoderClear(pEnCoder);
×
3605
  }
3606
  if (pScanEnCoder != NULL) {
55!
3607
    tEndEncode(pScanEnCoder);
55✔
3608
    tEncoderClear(pScanEnCoder);
55✔
3609
  }
3610
  return code;
55✔
3611
}
3612

3613
void streamScanOperatorSaveCheckpoint(SStreamScanInfo* pInfo) {
3,852✔
3614
  int32_t code = TSDB_CODE_SUCCESS;
3,852✔
3615
  int32_t lino = 0;
3,852✔
3616
  if (!pInfo->pState) {
3,852!
3617
    return;
×
3618
  }
3619
  if (needSaveStreamOperatorInfo(&pInfo->basic)) {
3,852✔
3620
    void*   pBuf = NULL;
55✔
3621
    int32_t len = 0;
55✔
3622
    code = streamScanOperatorEncode(pInfo, &pBuf, &len);
55✔
3623
    QUERY_CHECK_CODE(code, lino, _end);
55!
3624
    pInfo->stateStore.streamStateSaveInfo(pInfo->pState, STREAM_SCAN_OP_CHECKPOINT_NAME,
55✔
3625
                                          strlen(STREAM_SCAN_OP_CHECKPOINT_NAME), pBuf, len);
3626
    taosMemoryFree(pBuf);
55!
3627
    saveStreamOperatorStateComplete(&pInfo->basic);
55✔
3628
  }
3629

3630
_end:
3,797✔
3631
  if (code != TSDB_CODE_SUCCESS) {
3,852!
3632
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3633
  }
3634
}
3635

3636
// other properties are recovered from the execution plan
3637
void streamScanOperatorDecode(void* pBuff, int32_t len, SStreamScanInfo* pInfo) {
26✔
3638
  int32_t   code = TSDB_CODE_SUCCESS;
26✔
3639
  int32_t   lino = 0;
26✔
3640
  SDecoder* pDeCoder = NULL;
26✔
3641
  if (!pBuff || len == 0) {
26!
3642
    lino = __LINE__;
×
3643
    goto _end;
×
3644
  }
3645
  void* buf = pBuff;
26✔
3646
  buf = decodeSTimeWindowAggSupp(buf, &pInfo->twAggSup);
26✔
3647
  int32_t tlen = len - encodeSTimeWindowAggSupp(NULL, &pInfo->twAggSup);
26✔
3648
  if (tlen == 0) {
26!
3649
    lino = __LINE__;
×
3650
    goto _end;
×
3651
  }
3652

3653
  SDecoder decoder = {0};
26✔
3654
  pDeCoder = &decoder;
26✔
3655
  tDecoderInit(pDeCoder, buf, tlen);
26✔
3656
  if (tStartDecode(pDeCoder) < 0) {
26!
3657
    lino = __LINE__;
×
3658
    goto _end;
×
3659
  }
3660

3661
  void* pUpInfo = taosMemoryCalloc(1, sizeof(SUpdateInfo));
26!
3662
  if (!pUpInfo) {
26!
3663
    lino = __LINE__;
×
3664
    goto _end;
×
3665
  }
3666
  code = pInfo->stateStore.updateInfoDeserialize(pDeCoder, pUpInfo);
26✔
3667
  if (code == TSDB_CODE_SUCCESS) {
26✔
3668
    pInfo->stateStore.updateInfoDestroy(pInfo->pUpdateInfo);
23✔
3669
    pInfo->pUpdateInfo = pUpInfo;
23✔
3670
    qDebug("%s line:%d. stream scan updateinfo deserialize success", __func__, __LINE__);
23✔
3671
  } else {
3672
    taosMemoryFree(pUpInfo);
3!
3673
    code = TSDB_CODE_SUCCESS;
3✔
3674
    qDebug("%s line:%d. stream scan did not have updateinfo", __func__, __LINE__);
3!
3675
  }
3676

3677
  if (tDecodeIsEnd(pDeCoder)) {
26!
3678
    lino = __LINE__;
×
3679
    goto _end;
×
3680
  }
3681

3682
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.skey);
26✔
3683
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.ekey);
26✔
3684

3685
  if (tDecodeI64(pDeCoder, &pInfo->lastScanRange.skey) < 0) {
52!
3686
    lino = __LINE__;
×
3687
    goto _end;
×
3688
  }
3689

3690
  if (tDecodeI64(pDeCoder, &pInfo->lastScanRange.ekey) < 0) {
52!
3691
    lino = __LINE__;
×
3692
    goto _end;
×
3693
  }
3694
  qDebug("%s last scan range %d. %" PRId64 ",%" PRId64, __func__, __LINE__, pInfo->lastScanRange.skey, pInfo->lastScanRange.ekey);
26✔
3695

3696
_end:
20✔
3697
  if (pDeCoder != NULL) {
26!
3698
    tEndDecode(pDeCoder);
26✔
3699
    tDecoderClear(pDeCoder);
26✔
3700
  }
3701
  qInfo("%s end at line %d", __func__, lino);
26!
3702
}
26✔
3703

3704
bool hasScanRange(SStreamScanInfo* pInfo) {
386,176✔
3705
  SStreamAggSupporter* pSup = pInfo->windowSup.pStreamAggSup;
386,176✔
3706
  return pSup && pSup->pScanBlock->info.rows > 0 && (isStateWindow(pInfo) || isCountWindow(pInfo));
386,176!
3707
}
3708

3709
bool isStreamWindow(SStreamScanInfo* pInfo) {
1,279✔
3710
  return isIntervalWindow(pInfo) || isSessionWindow(pInfo) || isStateWindow(pInfo) || isCountWindow(pInfo) ||
1,369✔
3711
         isTimeSlice(pInfo);
90✔
3712
}
3713

3714
int32_t copyGetResultBlock(SSDataBlock* dest, TSKEY start, TSKEY end) {
4,176✔
3715
  int32_t code = blockDataEnsureCapacity(dest, 1);
4,176✔
3716
  if (code != TSDB_CODE_SUCCESS) {
4,176!
3717
    return code;
×
3718
  }
3719
  return appendDataToSpecialBlock(dest, &start, &end, NULL, NULL, NULL);
4,176✔
3720
}
3721

3722
int32_t deletePartName(SStateStore* pStore, SStreamState* pState, SSDataBlock* pBlock, int32_t *deleteNum) {
185✔
3723
  int32_t code = TSDB_CODE_SUCCESS;
185✔
3724
  int32_t lino = 0;
185✔
3725
  for (int32_t i = 0; i < pBlock->info.rows; i++) {
375✔
3726
    // uid is the same as gid
3727
    SColumnInfoData* pGpIdCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
190✔
3728
    SColumnInfoData* pTbnameCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
190✔
3729
    int64_t*         gpIdCol = (int64_t*)pGpIdCol->pData;
190✔
3730
    void*            pParName = NULL;
190✔
3731
    int32_t          winCode = 0;
190✔
3732
    code = pStore->streamStateGetParName(pState, gpIdCol[i],
190✔
3733
                                                   &pParName, false, &winCode);
3734
    if (TSDB_CODE_SUCCESS == code && winCode != 0) {
190!
3735
      qDebug("delete stream part Name for:%"PRId64 " not found", gpIdCol[i]);
135!
3736
      colDataSetNULL(pTbnameCol, i);
135!
3737
      continue;
135✔
3738
    }
3739
    (*deleteNum)++;
55✔
3740
    QUERY_CHECK_CODE(code, lino, _end);
55!
3741
    char varTbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE + 1] = {0};
55✔
3742
    varDataSetLen(varTbName, strlen(pParName));
55✔
3743
    int64_t len = tsnprintf(varTbName + VARSTR_HEADER_SIZE, TSDB_TABLE_NAME_LEN + 1, "%s", pParName);
55✔
3744
    code = colDataSetVal(pTbnameCol, i, varTbName, false);
55✔
3745
    qDebug("delete stream part for:%"PRId64 " res tb: %s", gpIdCol[i], (char*)pParName);
55!
3746
    pStore->streamStateFreeVal(pParName);
55✔
3747
    QUERY_CHECK_CODE(code, lino, _end);
55!
3748
    code = pStore->streamStateDeleteParName(pState, gpIdCol[i]);
55✔
3749
    QUERY_CHECK_CODE(code, lino, _end);
55!
3750
    pBlock->info.id.groupId = gpIdCol[i];
55✔
3751
    // currently, only one valid row in pBlock
3752
    memcpy(pBlock->info.parTbName, varTbName + VARSTR_HEADER_SIZE, TSDB_TABLE_NAME_LEN + 1);
55✔
3753
  }
3754

3755
_end:
185✔
3756
  if (code != TSDB_CODE_SUCCESS) {
185!
3757
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3758
  }
3759
  return code;
185✔
3760
}
3761

3762
static int32_t doStreamScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
484,771✔
3763
  // NOTE: this operator does never check if current status is done or not
3764
  int32_t        code = TSDB_CODE_SUCCESS;
484,771✔
3765
  int32_t        lino = 0;
484,771✔
3766
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
484,771✔
3767
  const char*    id = GET_TASKID(pTaskInfo);
484,771✔
3768

3769
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
484,771✔
3770
  SStreamScanInfo* pInfo = pOperator->info;
484,771✔
3771
  SStreamTaskInfo* pStreamInfo = &pTaskInfo->streamInfo;
484,771✔
3772
  SSHashObj*       pVtableInfos = pTaskInfo->pSubplan->pVTables;
484,771✔
3773

3774
  qDebug("stream scan started, %s", id);
484,771✔
3775

3776
  if (pVtableInfos != NULL && pStreamInfo->recoverStep != STREAM_RECOVER_STEP__NONE) {
484,743!
3777
    qError("stream vtable source scan should not have recovery step: %d", pStreamInfo->recoverStep);
×
3778
    pStreamInfo->recoverStep = STREAM_RECOVER_STEP__NONE;
×
3779
  }
3780

3781
  if (pVtableInfos != NULL && !pInfo->igCheckUpdate) {
484,743!
3782
    qError("stream vtable source scan should have igCheckUpdate");
×
3783
    pInfo->igCheckUpdate = false;
×
3784
  }
3785

3786
  if (pStreamInfo->recoverStep == STREAM_RECOVER_STEP__PREPARE1 ||
484,743✔
3787
      pStreamInfo->recoverStep == STREAM_RECOVER_STEP__PREPARE2) {
482,589✔
3788
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
4,983✔
3789
    memcpy(&pTSInfo->base.cond, &pStreamInfo->tableCond, sizeof(SQueryTableDataCond));
4,983✔
3790

3791
    if (pStreamInfo->recoverStep == STREAM_RECOVER_STEP__PREPARE1) {
4,983✔
3792
      pTSInfo->base.cond.startVersion = pStreamInfo->fillHistoryVer.minVer;
2,355✔
3793
      pTSInfo->base.cond.endVersion = pStreamInfo->fillHistoryVer.maxVer;
2,355✔
3794

3795
      pTSInfo->base.cond.twindows = pStreamInfo->fillHistoryWindow;
2,355✔
3796
      qDebug("stream scan step1, verRange:%" PRId64 "-%" PRId64 " window:%" PRId64 "-%" PRId64 ", %s",
2,355✔
3797
             pTSInfo->base.cond.startVersion, pTSInfo->base.cond.endVersion, pTSInfo->base.cond.twindows.skey,
3798
             pTSInfo->base.cond.twindows.ekey, id);
3799
      pStreamInfo->recoverStep = STREAM_RECOVER_STEP__SCAN1;
2,355✔
3800
      pStreamInfo->recoverScanFinished = false;
2,355✔
3801
    } else {
3802
      pTSInfo->base.cond.startVersion = pStreamInfo->fillHistoryVer.minVer;
2,628✔
3803
      pTSInfo->base.cond.endVersion = pStreamInfo->fillHistoryVer.maxVer;
2,628✔
3804
      pTSInfo->base.cond.twindows = pStreamInfo->fillHistoryWindow;
2,628✔
3805
      qDebug("stream scan step2 (scan wal), verRange:%" PRId64 " - %" PRId64 ", window:%" PRId64 "-%" PRId64 ", %s",
2,628✔
3806
             pTSInfo->base.cond.startVersion, pTSInfo->base.cond.endVersion, pTSInfo->base.cond.twindows.skey,
3807
             pTSInfo->base.cond.twindows.ekey, id);
3808
      pStreamInfo->recoverStep = STREAM_RECOVER_STEP__NONE;
2,628✔
3809
    }
3810

3811
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
4,983✔
3812

3813
    pTSInfo->base.dataReader = NULL;
4,983✔
3814
    pInfo->pTableScanOp->status = OP_OPENED;
4,983✔
3815

3816
    pTSInfo->scanTimes = 0;
4,983✔
3817
    pTSInfo->currentGroupId = -1;
4,983✔
3818
  }
3819

3820
  if (pStreamInfo->recoverStep == STREAM_RECOVER_STEP__SCAN1) {
484,743✔
3821
    if (isTaskKilled(pTaskInfo)) {
67,219✔
3822
      qInfo("===stream===stream scan is killed. task id:%s, code %s", id, tstrerror(pTaskInfo->code));
2!
3823
      (*ppRes) = NULL;
2✔
3824
      return code;
2✔
3825
    }
3826

3827
    switch (pInfo->scanMode) {
67,217✔
3828
      case STREAM_SCAN_FROM_RES: {
2,257✔
3829
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
2,257✔
3830
        printSpecDataBlock(pInfo->pRecoverRes, getStreamOpName(pOperator->operatorType), "recover",
2,257✔
3831
                           GET_TASKID(pTaskInfo));
2,257✔
3832
        (*ppRes) = pInfo->pRecoverRes;
2,257✔
3833
        return code;
2,257✔
3834
      } break;
3835
      default:
64,960✔
3836
        break;
64,960✔
3837
    }
3838

3839
    code = doTableScanNext(pInfo->pTableScanOp, &pInfo->pRecoverRes);
64,960✔
3840
    QUERY_CHECK_CODE(code, lino, _end);
64,961!
3841

3842
    if (pInfo->pRecoverRes != NULL) {
64,961✔
3843
      code = calBlockTbName(pInfo, pInfo->pRecoverRes, 0);
62,607✔
3844
      QUERY_CHECK_CODE(code, lino, _end);
62,607!
3845
      if (!pInfo->igCheckUpdate && pInfo->pUpdateInfo) {
62,607✔
3846
        TSKEY maxTs = INT64_MIN;
61,782✔
3847
        code = pAPI->stateStore.updateInfoFillBlockData(pInfo->pUpdateInfo, pInfo->pRecoverRes, pInfo->primaryTsIndex,
61,782✔
3848
                                                        pInfo->primaryKeyIndex, &maxTs);
3849
        QUERY_CHECK_CODE(code, lino, _end);
61,782!
3850
        pInfo->twAggSup.maxTs = TMAX(pInfo->twAggSup.maxTs, maxTs);
61,782✔
3851
      }
3852
      if (pInfo->pCreateTbRes->info.rows > 0) {
62,607✔
3853
        pInfo->scanMode = STREAM_SCAN_FROM_RES;
2,257✔
3854
        printSpecDataBlock(pInfo->pCreateTbRes, getStreamOpName(pOperator->operatorType), "recover",
2,257✔
3855
                           GET_TASKID(pTaskInfo));
2,257✔
3856
        (*ppRes) = pInfo->pCreateTbRes;
2,257✔
3857
        return code;
2,257✔
3858
      }
3859

3860
      qDebug("stream recover scan get block, rows %" PRId64, pInfo->pRecoverRes->info.rows);
60,350✔
3861
      printSpecDataBlock(pInfo->pRecoverRes, getStreamOpName(pOperator->operatorType), "recover",
60,350✔
3862
                         GET_TASKID(pTaskInfo));
60,350✔
3863
      (*ppRes) = pInfo->pRecoverRes;
60,350✔
3864
      return code;
60,350✔
3865
    }
3866
    pStreamInfo->recoverStep = STREAM_RECOVER_STEP__NONE;
2,354✔
3867
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
2,354✔
3868
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
2,354✔
3869

3870
    pTSInfo->base.dataReader = NULL;
2,354✔
3871

3872
    pTSInfo->base.cond.startVersion = -1;
2,354✔
3873
    pTSInfo->base.cond.endVersion = -1;
2,354✔
3874

3875
    pStreamInfo->recoverScanFinished = true;
2,354✔
3876
    (*ppRes) = NULL;
2,354✔
3877
    return code;
2,354✔
3878
  }
3879

3880
  size_t total = taosArrayGetSize(pInfo->pBlockLists);
417,524✔
3881
// TODO: refactor
3882
FETCH_NEXT_BLOCK:
418,862✔
3883
  if (pInfo->blockType == STREAM_INPUT__DATA_BLOCK) {
418,862✔
3884
    if (pVtableInfos != NULL) {
18,624!
3885
      qInfo("stream vtable source scan would ignore all data blocks");
×
3886
      pInfo->validBlockIndex = total;
×
3887
    }
3888
    if (pInfo->validBlockIndex >= total) {
18,624✔
3889
      doClearBufferedBlocks(pInfo);
3890
      (*ppRes) = NULL;
4,294✔
3891
      return code;
4,294✔
3892
    }
3893

3894
    int32_t current = pInfo->validBlockIndex++;
14,330✔
3895
    qDebug("process %d/%d input data blocks, %s", current, (int32_t)total, id);
14,330✔
3896

3897
    SPackedData* pPacked = taosArrayGet(pInfo->pBlockLists, current);
14,330✔
3898
    QUERY_CHECK_NULL(pPacked, code, lino, _end, terrno);
14,330!
3899

3900
    SSDataBlock* pBlock = pPacked->pDataBlock;
14,330✔
3901
    if (pBlock->info.parTbName[0]) {
14,330✔
3902
      code =
3903
          pAPI->stateStore.streamStatePutParName(pStreamInfo->pState, pBlock->info.id.groupId, pBlock->info.parTbName);
1,443✔
3904
      QUERY_CHECK_CODE(code, lino, _end);
1,443!
3905
    }
3906

3907
    // TODO move into scan
3908
    pBlock->info.calWin.skey = INT64_MIN;
14,330✔
3909
    pBlock->info.calWin.ekey = INT64_MAX;
14,330✔
3910
    pBlock->info.dataLoad = 1;
14,330✔
3911
    if (pInfo->pUpdateInfo) {
14,330✔
3912
      pInfo->pUpdateInfo->maxDataVersion = TMAX(pInfo->pUpdateInfo->maxDataVersion, pBlock->info.version);
8,857✔
3913
    }
3914

3915
    code = blockDataUpdateTsWindow(pBlock, 0);
14,330✔
3916
    QUERY_CHECK_CODE(code, lino, _end);
14,331!
3917
    switch (pBlock->info.type) {
14,331!
3918
      case STREAM_NORMAL:
3,524✔
3919
      case STREAM_GET_ALL:
3920
        printDataBlock(pBlock, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
3,524✔
3921
        setStreamOperatorState(&pInfo->basic, pBlock->info.type);
3,524✔
3922
        (*ppRes) = pBlock;
3,524✔
3923
        return code;
3,524✔
3924
      case STREAM_RETRIEVE: {
519✔
3925
        pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
519✔
3926
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RETRIEVE;
519✔
3927
        code = copyDataBlock(pInfo->pUpdateRes, pBlock);
519✔
3928
        QUERY_CHECK_CODE(code, lino, _end);
519!
3929
        pInfo->updateResIndex = 0;
519✔
3930
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
519✔
3931
        pAPI->stateStore.updateInfoAddCloseWindowSBF(pInfo->pUpdateInfo);
519✔
3932
      } break;
519✔
3933
      case STREAM_DELETE_DATA: {
2,401✔
3934
        printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "delete recv", GET_TASKID(pTaskInfo));
2,401✔
3935
        SSDataBlock* pDelBlock = NULL;
2,401✔
3936
        if (pInfo->tqReader) {
2,401!
3937
          code = createSpecialDataBlock(STREAM_DELETE_DATA, &pDelBlock);
2,401✔
3938
          QUERY_CHECK_CODE(code, lino, _end);
2,401!
3939

3940
          code = filterDelBlockByUid(pDelBlock, pBlock, pInfo->tqReader, &pInfo->readerFn);
2,401✔
3941
          QUERY_CHECK_CODE(code, lino, _end);
2,401!
3942
        } else {
3943
          pDelBlock = pBlock;
×
3944
        }
3945

3946
        code = setBlockGroupIdByUid(pInfo, pDelBlock);
2,401✔
3947
        QUERY_CHECK_CODE(code, lino, _end);
2,401!
3948
        code = rebuildDeleteBlockData(pDelBlock, &pStreamInfo->fillHistoryWindow, id);
2,401✔
3949
        QUERY_CHECK_CODE(code, lino, _end);
2,401!
3950
        printSpecDataBlock(pDelBlock, getStreamOpName(pOperator->operatorType), "delete recv filtered",
2,401✔
3951
                           GET_TASKID(pTaskInfo));
2,401✔
3952
        if (pDelBlock->info.rows == 0) {
2,401✔
3953
          if (pInfo->tqReader) {
1,122!
3954
            blockDataDestroy(pDelBlock);
1,122✔
3955
          }
3956
          goto FETCH_NEXT_BLOCK;
1,122✔
3957
        }
3958

3959
        if (!isStreamWindow(pInfo)) {
1,279✔
3960
          code = generateDeleteResultBlock(pInfo, pDelBlock, pInfo->pDeleteDataRes);
75✔
3961
          QUERY_CHECK_CODE(code, lino, _end);
75!
3962
          if (pInfo->partitionSup.needCalc) {
75✔
3963
            pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
2✔
3964
          } else {
3965
            pInfo->pDeleteDataRes->info.type = STREAM_DELETE_RESULT;
73✔
3966
          }
3967
          blockDataDestroy(pDelBlock);
75✔
3968

3969
          if (pInfo->pDeleteDataRes->info.rows > 0) {
75!
3970
            printSpecDataBlock(pInfo->pDeleteDataRes, getStreamOpName(pOperator->operatorType), "delete result",
75✔
3971
                               GET_TASKID(pTaskInfo));
75✔
3972
            setStreamOperatorState(&pInfo->basic, pInfo->pDeleteDataRes->info.type);
75✔
3973
            (*ppRes) = pInfo->pDeleteDataRes;
75✔
3974
            return code;
1,278✔
3975
          } else {
3976
            goto FETCH_NEXT_BLOCK;
×
3977
          }
3978
        } else {
3979
          pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
1,204✔
3980
          pInfo->updateResIndex = 0;
1,204✔
3981
          code = generateScanRange(pInfo, pDelBlock, pInfo->pUpdateRes, STREAM_DELETE_DATA);
1,204✔
3982
          QUERY_CHECK_CODE(code, lino, _end);
1,204!
3983
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
1,204✔
3984
          code = copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
1,204✔
3985
          QUERY_CHECK_CODE(code, lino, _end);
1,204!
3986
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
1,204✔
3987
          if (pInfo->tqReader) {
1,204!
3988
            blockDataDestroy(pDelBlock);
1,204✔
3989
          }
3990
          if (pInfo->pDeleteDataRes->info.rows > 0) {
1,204✔
3991
            pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
1,203✔
3992
            printSpecDataBlock(pInfo->pDeleteDataRes, getStreamOpName(pOperator->operatorType), "delete result",
1,203✔
3993
                               GET_TASKID(pTaskInfo));
1,203✔
3994
            setStreamOperatorState(&pInfo->basic, pInfo->pDeleteDataRes->info.type);
1,203✔
3995
            (*ppRes) = pInfo->pDeleteDataRes;
1,203✔
3996
            return code;
1,203✔
3997
          } else {
3998
            goto FETCH_NEXT_BLOCK;
1✔
3999
          }
4000
        }
4001
      } break;
4002
      case STREAM_GET_RESULT: {
4,176✔
4003
        pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
4,176✔
4004
        pInfo->updateResIndex = 0;
4,176✔
4005
        pInfo->lastScanRange = pBlock->info.window;
4,176✔
4006
        TSKEY endKey = getNextTimeWindowStart(&pInfo->interval, pBlock->info.window.skey, TSDB_ORDER_ASC) - 1;
4,176✔
4007
        if (pInfo->useGetResultRange == true) {
4,176✔
4008
          endKey = pBlock->info.window.ekey;
876✔
4009
        }
4010
        code = copyGetResultBlock(pInfo->pUpdateRes, pBlock->info.window.skey, endKey);
4,176✔
4011
        QUERY_CHECK_CODE(code, lino, _end);
4,176!
4012
        pInfo->pUpdateInfo->maxDataVersion = -1;
4,176✔
4013
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
4,176✔
4014
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
4,176✔
4015
      } break;
4,176✔
4016
      case STREAM_DROP_CHILD_TABLE: {
185✔
4017
        int32_t deleteNum = 0;
185✔
4018
        code = deletePartName(&pInfo->stateStore, pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, pBlock, &deleteNum);
185✔
4019
        QUERY_CHECK_CODE(code, lino, _end);
185!
4020
        if (deleteNum == 0) {
185✔
4021
          printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "block recv", GET_TASKID(pTaskInfo));
130✔
4022
          qDebug("===stream=== ignore block type 18, delete num is 0");
130!
4023
          goto FETCH_NEXT_BLOCK;
130✔
4024
        }
4025
      } break;
55✔
4026
      case STREAM_CHECKPOINT: {
×
4027
        qError("stream check point error. msg type: STREAM_INPUT__DATA_BLOCK");
×
4028
      } break;
×
4029
      default:
3,526✔
4030
        break;
3,526✔
4031
    }
4032
    printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "block recv", GET_TASKID(pTaskInfo));
8,276✔
4033
    setStreamOperatorState(&pInfo->basic, pBlock->info.type);
8,276✔
4034
    (*ppRes) = pBlock;
8,276✔
4035
    return code;
8,276✔
4036
  } else if (pInfo->blockType == STREAM_INPUT__DATA_SUBMIT) {
400,238✔
4037
    qDebug("stream scan mode:%d, %s", pInfo->scanMode, id);
392,464✔
4038
    if (pVtableInfos != NULL && pInfo->scanMode != STREAM_SCAN_FROM_READERHANDLE) {
392,390!
4039
      qError("stream vtable source scan should not have scan mode: %d", pInfo->scanMode);
×
4040
      pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
×
4041
    }
4042
    switch (pInfo->scanMode) {
392,390✔
4043
      case STREAM_SCAN_FROM_RES: {
1,522✔
4044
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
1,522✔
4045
        code = doCheckUpdate(pInfo, pInfo->pRes->info.window.ekey, pInfo->pRes);
1,522✔
4046
        QUERY_CHECK_CODE(code, lino, _end);
1,522!
4047
        setStreamOperatorState(&pInfo->basic, pInfo->pRes->info.type);
1,522✔
4048
        code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
1,522✔
4049
        QUERY_CHECK_CODE(code, lino, _end);
1,522!
4050
        pInfo->pRes->info.dataLoad = 1;
1,522✔
4051
        code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
1,522✔
4052
        QUERY_CHECK_CODE(code, lino, _end);
1,522!
4053
        if (pInfo->pRes->info.rows > 0) {
1,522!
4054
          printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
1,522✔
4055
          (*ppRes) = pInfo->pRes;
1,522✔
4056
          return code;
1,522✔
4057
        }
4058
      } break;
×
4059
      case STREAM_SCAN_FROM_DELETE_DATA: {
254✔
4060
        code = generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes, STREAM_PARTITION_DELETE_DATA);
254✔
4061
        QUERY_CHECK_CODE(code, lino, _end);
254!
4062
        if (pInfo->pUpdateRes->info.rows > 0) {
254✔
4063
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
220✔
4064
          pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
220✔
4065
          code = copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
220✔
4066
          QUERY_CHECK_CODE(code, lino, _end);
220!
4067
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
220✔
4068
          (*ppRes) = pInfo->pDeleteDataRes;
220✔
4069
          return code;
220✔
4070
        }
4071
        qError("%s===stream=== %s failed at line %d since pInfo->pUpdateRes is empty", GET_TASKID(pTaskInfo), __func__,
34!
4072
               __LINE__);
4073
        blockDataCleanup(pInfo->pUpdateDataRes);
34✔
4074
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
34✔
4075
      } break;
34✔
4076
      case STREAM_SCAN_FROM_UPDATERES: {
1,352✔
4077
        code = generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes, STREAM_CLEAR);
1,352✔
4078
        QUERY_CHECK_CODE(code, lino, _end);
1,352!
4079
        if (pInfo->pUpdateRes->info.rows > 0) {
1,352✔
4080
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
1,335✔
4081
          pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
1,336✔
4082
          (*ppRes) = pInfo->pUpdateRes;
1,336✔
4083
          return code;
1,336✔
4084
        }
4085
        qError("%s===stream=== %s failed at line %d since pInfo->pUpdateRes is empty", GET_TASKID(pTaskInfo), __func__,
17!
4086
               __LINE__);
4087
        blockDataCleanup(pInfo->pUpdateDataRes);
17✔
4088
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
16✔
4089
      } break;
16✔
4090
      case STREAM_SCAN_FROM_DATAREADER_RANGE:
10,626✔
4091
      case STREAM_SCAN_FROM_DATAREADER_RETRIEVE: {
4092
        if (pInfo->pRangeScanRes != NULL) {
10,626✔
4093
          (*ppRes) = pInfo->pRangeScanRes;
26✔
4094
          pInfo->pRangeScanRes = NULL;
26✔
4095
          return code;
3,117✔
4096
        } 
4097
        SSDataBlock* pSDB = NULL;
10,600✔
4098
        code = doRangeScan(pInfo, pInfo->pUpdateRes, pInfo->primaryTsIndex, &pInfo->updateResIndex, &pSDB);
10,600✔
4099
        QUERY_CHECK_CODE(code, lino, _end);
10,601!
4100
        if (pSDB) {
10,601✔
4101
          STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
3,091✔
4102
          pSDB->info.type = pInfo->scanMode == STREAM_SCAN_FROM_DATAREADER_RANGE ? STREAM_NORMAL : STREAM_PULL_DATA;
3,091✔
4103
          if (!pInfo->igCheckUpdate && pInfo->pUpdateInfo) {
3,091!
4104
            code = checkUpdateData(pInfo, true, pSDB, false);
2,491✔
4105
            QUERY_CHECK_CODE(code, lino, _end);
2,491!
4106
          }
4107
          printSpecDataBlock(pSDB, getStreamOpName(pOperator->operatorType), "update", GET_TASKID(pTaskInfo));
3,091✔
4108
          code = calBlockTbName(pInfo, pSDB, 0);
3,091✔
4109
          QUERY_CHECK_CODE(code, lino, _end);
3,091!
4110

4111
          if (pInfo->pCreateTbRes->info.rows > 0) {
3,091✔
4112
            printSpecDataBlock(pInfo->pCreateTbRes, getStreamOpName(pOperator->operatorType), "update",
26✔
4113
                               GET_TASKID(pTaskInfo));
26✔
4114
            (*ppRes) = pInfo->pCreateTbRes;
26✔
4115
            pInfo->pRangeScanRes = pSDB;
26✔
4116
            return code;
26✔
4117
          }
4118

4119
          (*ppRes) = pSDB;
3,065✔
4120
          return code;
3,065✔
4121
        }
4122
        blockDataCleanup(pInfo->pUpdateDataRes);
7,510✔
4123
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
7,510✔
4124
      } break;
7,510✔
4125
      default:
378,636✔
4126
        break;
378,636✔
4127
    }
4128

4129
    if (hasScanRange(pInfo)) {
386,196✔
4130
      pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
58✔
4131
      pInfo->updateResIndex = 0;
58✔
4132
      SStreamAggSupporter* pSup = pInfo->windowSup.pStreamAggSup;
58✔
4133
      code = copyDataBlock(pInfo->pUpdateRes, pSup->pScanBlock);
58✔
4134
      QUERY_CHECK_CODE(code, lino, _end);
58!
4135
      blockDataCleanup(pSup->pScanBlock);
58✔
4136
      prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
58✔
4137
      pInfo->pUpdateRes->info.type = STREAM_DELETE_DATA;
58✔
4138
      printSpecDataBlock(pInfo->pUpdateRes, getStreamOpName(pOperator->operatorType), "rebuild", GET_TASKID(pTaskInfo));
58✔
4139
      (*ppRes) = pInfo->pUpdateRes;
58✔
4140
      return code;
58✔
4141
    }
4142

4143
    SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
386,224✔
4144
    int32_t         totalBlocks = taosArrayGetSize(pInfo->pBlockLists);
386,224✔
4145

4146
  NEXT_SUBMIT_BLK:
386,176✔
4147
    while (1) {
4148
      if (pInfo->readerFn.tqReaderCurrentBlockConsumed(pInfo->tqReader)) {
829,554✔
4149
        if (pInfo->validBlockIndex >= totalBlocks) {
480,398✔
4150
          pAPI->stateStore.updateInfoDestoryColseWinSBF(pInfo->pUpdateInfo);
37,008✔
4151
          doClearBufferedBlocks(pInfo);
4152

4153
          qDebug("stream scan return empty, all %d submit blocks consumed, %s", totalBlocks, id);
37,000✔
4154
          (*ppRes) = NULL;
37,000✔
4155
          return code;
37,000✔
4156
        }
4157

4158
        int32_t      current = pInfo->validBlockIndex++;
443,390✔
4159
        SPackedData* pSubmit = taosArrayGet(pInfo->pBlockLists, current);
443,390✔
4160
        QUERY_CHECK_NULL(pSubmit, code, lino, _end, terrno);
443,374!
4161

4162
        qDebug("set %d/%d as the input submit block, %s", current + 1, totalBlocks, id);
443,374✔
4163
        if (pAPI->tqReaderFn.tqReaderSetSubmitMsg(pInfo->tqReader, pSubmit->msgStr, pSubmit->msgLen, pSubmit->ver, NULL) <
443,375!
4164
            0) {
4165
          qError("submit msg messed up when initializing stream submit block %p, current %d/%d, %s", pSubmit, current,
×
4166
                 totalBlocks, id);
4167
          continue;
×
4168
        }
4169
      }
4170

4171
      blockDataCleanup(pInfo->pRes);
792,709✔
4172

4173
      while (pAPI->tqReaderFn.tqNextBlockImpl(pInfo->tqReader, id)) {
813,210✔
4174
        SSDataBlock* pRes = NULL;
369,975✔
4175

4176
        code = pAPI->tqReaderFn.tqRetrieveBlock(pInfo->tqReader, &pRes, id);
369,975✔
4177
        qDebug("retrieve data from submit completed code:%s rows:%" PRId64 " %s", tstrerror(code), pRes->info.rows, id);
369,983✔
4178

4179
        if (code != TSDB_CODE_SUCCESS || pRes->info.rows == 0) {
369,984✔
4180
          qDebug("retrieve data failed, try next block in submit block, %s", id);
27✔
4181
          continue;
20,569✔
4182
        }
4183

4184
        code = setBlockIntoRes(pInfo, pRes, &pStreamInfo->fillHistoryWindow, false);
369,957✔
4185
        if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
369,930!
4186
          pInfo->pRes->info.rows = 0;
×
4187
          code = TSDB_CODE_SUCCESS;
×
4188
        } else {
4189
          QUERY_CHECK_CODE(code, lino, _end);
369,930!
4190
        }
4191

4192
        if (pInfo->pRes->info.rows == 0) {
369,930✔
4193
          continue;
20,545✔
4194
        }
4195

4196
        if (pVtableInfos != NULL && pInfo->pCreateTbRes->info.rows > 0) {
349,385!
4197
          qError("stream vtable source scan should not have create table res");
×
4198
          blockDataCleanup(pInfo->pCreateTbRes);
×
4199
        }
4200

4201
        if (pInfo->pCreateTbRes->info.rows > 0) {
349,372✔
4202
          pInfo->scanMode = STREAM_SCAN_FROM_RES;
1,523✔
4203
          qDebug("create table res exists, rows:%" PRId64 " return from stream scan, %s",
1,523✔
4204
                 pInfo->pCreateTbRes->info.rows, id);
4205
          (*ppRes) = pInfo->pCreateTbRes;
1,523✔
4206
          return code;
1,523✔
4207
        }
4208

4209
        code = doCheckUpdate(pInfo, pBlockInfo->window.ekey, pInfo->pRes);
347,849✔
4210
        QUERY_CHECK_CODE(code, lino, _end);
347,876!
4211
        setStreamOperatorState(&pInfo->basic, pInfo->pRes->info.type);
347,876✔
4212
        if (pVtableInfos == NULL) {
347,884!
4213
          // filter should be applied in merge task for vtables
4214
          code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
347,885✔
4215
          QUERY_CHECK_CODE(code, lino, _end);
347,875!
4216
        }
4217

4218
        code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
347,874✔
4219
        QUERY_CHECK_CODE(code, lino, _end);
347,882!
4220

4221
        int64_t numOfUpdateRes = pInfo->pUpdateDataRes->info.rows;
347,882✔
4222
        qDebug("%s %" PRId64 " rows in datablock, update res:%" PRId64, id, pBlockInfo->rows, numOfUpdateRes);
347,882✔
4223
        if (pBlockInfo->rows > 0 || numOfUpdateRes > 0) {
347,859✔
4224
          break;
4225
        }
4226
      }
4227

4228
      if (pBlockInfo->rows > 0 || pInfo->pUpdateDataRes->info.rows > 0) {
791,177!
4229
        break;
4230
      } else {
4231
        continue;
443,378✔
4232
      }
4233
    }
4234

4235
    // record the scan action.
4236
    pInfo->numOfExec++;
347,799✔
4237
    pOperator->resultInfo.totalRows += pBlockInfo->rows;
347,799✔
4238

4239
    qDebug("stream scan completed, and return source rows:%" PRId64 ", %s", pBlockInfo->rows, id);
347,799✔
4240
    if (pBlockInfo->rows > 0) {
347,804✔
4241
      printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
347,800✔
4242
      (*ppRes) = pInfo->pRes;
347,884✔
4243
      return code;
347,884✔
4244
    }
4245

4246
    if (pInfo->pUpdateDataRes->info.rows > 0) {
4✔
4247
      goto FETCH_NEXT_BLOCK;
3✔
4248
    }
4249

4250
    goto NEXT_SUBMIT_BLK;
1✔
4251
  } else if (pInfo->blockType == STREAM_INPUT__CHECKPOINT) {
7,774✔
4252
    if (pInfo->validBlockIndex >= total || pVtableInfos != NULL) {
7,704!
4253
      doClearBufferedBlocks(pInfo);
4254
      (*ppRes) = NULL;
3,852✔
4255
      return code;
3,852✔
4256
    }
4257

4258
    int32_t current = pInfo->validBlockIndex++;
3,852✔
4259
    qDebug("process %d/%d input data blocks, %s", current, (int32_t)total, id);
3,852✔
4260

4261
    SPackedData* pData = taosArrayGet(pInfo->pBlockLists, current);
3,852✔
4262
    QUERY_CHECK_NULL(pData, code, lino, _end, terrno);
3,852!
4263
    SSDataBlock* pBlock = taosArrayGet(pData->pDataBlock, 0);
3,852✔
4264
    QUERY_CHECK_NULL(pBlock, code, lino, _end, terrno);
3,852!
4265

4266
    if (pBlock->info.type == STREAM_CHECKPOINT) {
3,852!
4267
      streamScanOperatorSaveCheckpoint(pInfo);
3,852✔
4268
    }
4269
    // printDataBlock(pInfo->pCheckpointRes, "stream scan ck", GET_TASKID(pTaskInfo));
4270
    (*ppRes) = pInfo->pCheckpointRes;
3,852✔
4271
    return code;
3,852✔
4272
  } else {
4273
    qError("stream scan error, invalid block type %d, %s", pInfo->blockType, id);
70!
4274
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4275
  }
4276

4277
_end:
×
4278
  if (code != TSDB_CODE_SUCCESS) {
×
4279
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4280
    pTaskInfo->code = code;
×
4281
    T_LONG_JMP(pTaskInfo->env, code);
×
4282
  }
4283
  (*ppRes) = NULL;
×
4284
  return code;
×
4285
}
4286

4287
static SSDataBlock* doStreamScan(SOperatorInfo* pOperator) {
×
4288
  SSDataBlock* pRes = NULL;
×
4289
  int32_t      code = doStreamScanNext(pOperator, &pRes);
×
4290
  return pRes;
×
4291
}
4292
int32_t extractTableIdList(const STableListInfo* pTableListInfo, SArray** ppArrayRes) {
8,539✔
4293
  int32_t code = TSDB_CODE_SUCCESS;
8,539✔
4294
  int32_t lino = 0;
8,539✔
4295
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));
8,539✔
4296
  QUERY_CHECK_NULL(tableIdList, code, lino, _end, terrno);
8,555!
4297

4298
  // Transfer the Array of STableKeyInfo into uid list.
4299
  int32_t size = 0;
8,555✔
4300
  code = tableListGetSize(pTableListInfo, &size);
8,555✔
4301
  QUERY_CHECK_CODE(code, lino, _end);
8,555!
4302
  for (int32_t i = 0; i < size; ++i) {
160,717✔
4303
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
152,196✔
4304
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
152,091!
4305
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
152,091✔
4306
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
152,162!
4307
  }
4308

4309
  (*ppArrayRes) = tableIdList;
8,521✔
4310

4311
_end:
8,521✔
4312
  if (code != TSDB_CODE_SUCCESS) {
8,521!
4313
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4314
  }
4315
  return code;
8,553✔
4316
}
4317

4318
static int32_t doRawScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
5,048✔
4319
  int32_t        code = TSDB_CODE_SUCCESS;
5,048✔
4320
  int32_t        lino = 0;
5,048✔
4321
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5,048✔
4322
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
5,048✔
4323

4324
  SStreamRawScanInfo* pInfo = pOperator->info;
5,048✔
4325
  pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = NULL;  // use batchMetaReq != NULL to judge if data is meta
5,048✔
4326
  pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = NULL;
5,048✔
4327

4328
  qDebug("tmqsnap doRawScan called");
5,048✔
4329
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
5,048✔
4330
    bool hasNext = false;
5,032✔
4331
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
5,032!
4332
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
5,027✔
4333
      if (code != TSDB_CODE_SUCCESS) {
5,024!
4334
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
4335
        QUERY_CHECK_CODE(code, lino, _end);
×
4336
      }
4337
    }
4338

4339
    if (pInfo->dataReader && hasNext) {
5,029✔
4340
      if (isTaskKilled(pTaskInfo)) {
2,525!
4341
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
4342
        return code;
2,524✔
4343
      }
4344

4345
      SSDataBlock* pBlock = NULL;
2,525✔
4346
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock, NULL);
2,525✔
4347
      QUERY_CHECK_CODE(code, lino, _end);
2,524!
4348

4349
      if (pBlock && pBlock->info.rows > 0) {
2,524!
4350
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
2,524✔
4351
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
2,524✔
4352
        QUERY_CHECK_CODE(code, lino, _end);
2,524!
4353
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
2,524✔
4354
        (*ppRes) = pBlock;
2,524✔
4355
        return code;
2,524✔
4356
      }
4357
    }
4358

4359
    SMetaTableInfo mtInfo = {0};
2,504✔
4360
    code = pAPI->snapshotFn.getMetaTableInfoFromSnapshot(pInfo->sContext, &mtInfo);
2,504✔
4361
    QUERY_CHECK_CODE(code, lino, _end);
2,505!
4362
    if (code != 0) {
2,505!
4363
      destroyMetaTableInfo(&mtInfo);
4364
      QUERY_CHECK_CODE(code, lino, _end);
×
4365
    }
4366
    STqOffsetVal offset = {0};
2,505✔
4367
    if (mtInfo.uid == 0 || pInfo->sContext->withMeta == ONLY_META) {  // read snapshot done, change to get data from wal
2,505!
4368
      qDebug("tmqsnap read snapshot done, change to get data from wal");
28✔
4369
      tqOffsetResetToLog(&offset, pInfo->sContext->snapVersion + 1);
29✔
4370
    } else {
4371
      SValue val = {0};
2,477✔
4372
      tqOffsetResetToData(&offset, mtInfo.uid, INT64_MIN, val);
2,477!
4373
      qDebug("tmqsnap change get data uid:%" PRId64, mtInfo.uid);
2,477✔
4374
    }
4375
    destroyMetaTableInfo(&mtInfo);
4376
    code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
2,507✔
4377
    QUERY_CHECK_CODE(code, lino, _end);
2,503!
4378
    (*ppRes) = NULL;
2,503✔
4379
    return code;
2,503✔
4380
  } else if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_META) {
16!
4381
    SSnapContext* sContext = pInfo->sContext;
16✔
4382
    for (int32_t i = 0; i < tmqRowSize; i++) {
143!
4383
      void*   data = NULL;
143✔
4384
      int32_t dataLen = 0;
143✔
4385
      int16_t type = 0;
143✔
4386
      int64_t uid = 0;
143✔
4387
      if (pAPI->snapshotFn.getTableInfoFromSnapshot(sContext, &data, &dataLen, &type, &uid) < 0) {
143!
4388
        qError("tmqsnap getTableInfoFromSnapshot error");
×
4389
        taosMemoryFreeClear(data);
×
4390
        break;
16✔
4391
      }
4392

4393
      if (!sContext->queryMeta) {  // change to get data next poll request
143✔
4394
        STqOffsetVal offset = {0};
16✔
4395
        SValue       val = {0};
16!
4396
        tqOffsetResetToData(&offset, 0, INT64_MIN, val);
16✔
4397
        code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
16✔
4398
        QUERY_CHECK_CODE(code, lino, _end);
16!
4399
        break;
16✔
4400
      } else {
4401
        tqOffsetResetToMeta(&pTaskInfo->streamInfo.currentOffset, uid);
127✔
4402
        SMqMetaRsp tmpMetaRsp = {0};
127✔
4403
        tmpMetaRsp.resMsgType = type;
127✔
4404
        tmpMetaRsp.metaRspLen = dataLen;
127✔
4405
        tmpMetaRsp.metaRsp = data;
127✔
4406
        if (!pTaskInfo->streamInfo.btMetaRsp.batchMetaReq) {
127✔
4407
          pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = taosArrayInit(4, POINTER_BYTES);
14✔
4408
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, code, lino, _end, terrno);
14!
4409

4410
          pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = taosArrayInit(4, sizeof(int32_t));
14✔
4411
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, code, lino, _end, terrno);
14!
4412
        }
4413
        int32_t  tempRes = TSDB_CODE_SUCCESS;
127✔
4414
        uint32_t len = 0;
127✔
4415
        tEncodeSize(tEncodeMqMetaRsp, &tmpMetaRsp, len, tempRes);
127!
4416
        if (TSDB_CODE_SUCCESS != tempRes) {
127!
4417
          qError("tmqsnap tEncodeMqMetaRsp error");
×
4418
          taosMemoryFreeClear(data);
×
4419
          break;
×
4420
        }
4421

4422
        int32_t tLen = sizeof(SMqRspHead) + len;
127✔
4423
        void*   tBuf = taosMemoryCalloc(1, tLen);
127!
4424
        QUERY_CHECK_NULL(tBuf, code, lino, _end, terrno);
127!
4425

4426
        void*    metaBuff = POINTER_SHIFT(tBuf, sizeof(SMqRspHead));
127✔
4427
        SEncoder encoder = {0};
127✔
4428
        tEncoderInit(&encoder, metaBuff, len);
127✔
4429
        int32_t tempLen = tEncodeMqMetaRsp(&encoder, &tmpMetaRsp);
127✔
4430
        if (tempLen < 0) {
127!
4431
          qError("tmqsnap tEncodeMqMetaRsp error");
×
4432
          tEncoderClear(&encoder);
×
4433
          taosMemoryFreeClear(tBuf);
×
4434
          taosMemoryFreeClear(data);
×
4435
          break;
×
4436
        }
4437
        taosMemoryFreeClear(data);
127!
4438
        void* tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, &tBuf);
127✔
4439
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
127!
4440

4441
        tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, &tLen);
127✔
4442
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
127!
4443
      }
4444
    }
4445

4446
    (*ppRes) = NULL;
16✔
4447
    return code;
16✔
4448
  }
4449

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

4457
  (*ppRes) = NULL;
×
4458
  return code;
×
4459
}
4460

4461
static void destroyRawScanOperatorInfo(void* param) {
348✔
4462
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
348✔
4463
  pRawScan->pAPI->tsdReader.tsdReaderClose(pRawScan->dataReader);
348✔
4464
  pRawScan->pAPI->snapshotFn.destroySnapshot(pRawScan->sContext);
348✔
4465
  tableListDestroy(pRawScan->pTableListInfo);
348✔
4466
  taosMemoryFree(pRawScan);
348!
4467
}
348✔
4468

4469
// for subscribing db or stb (not including column),
4470
// if this scan is used, meta data can be return
4471
// and schemas are decided when scanning
4472
int32_t createRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
342✔
4473
  // create operator
4474
  // create tb reader
4475
  // create meta reader
4476
  // create tq reader
4477

4478
  QRY_PARAM_CHECK(pOptrInfo);
342!
4479
  int32_t code = TSDB_CODE_SUCCESS;
342✔
4480
  int32_t lino = 0;
342✔
4481

4482
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
342!
4483
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
348!
4484
  if (pInfo == NULL || pOperator == NULL) {
347!
4485
    code = terrno;
×
4486
    lino = __LINE__;
×
4487
    goto _end;
×
4488
  }
4489

4490
  pInfo->pTableListInfo = tableListCreate();
348✔
4491
  QUERY_CHECK_NULL(pInfo->pTableListInfo, code, lino, _end, terrno);
348!
4492
  pInfo->vnode = pHandle->vnode;
348✔
4493
  pInfo->pAPI = &pTaskInfo->storageAPI;
348✔
4494

4495
  pInfo->sContext = pHandle->sContext;
348✔
4496
  setOperatorInfo(pOperator, "RawScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
348✔
4497
                  pTaskInfo);
4498

4499
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScanNext, NULL, destroyRawScanOperatorInfo, optrDefaultBufFn, NULL,
348✔
4500
                                         optrDefaultGetNextExtFn, NULL);
4501
  *pOptrInfo = pOperator;
348✔
4502
  return code;
348✔
4503

4504
_end:
×
4505
  if (code != TSDB_CODE_SUCCESS) {
×
4506
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4507
  }
4508
  taosMemoryFree(pInfo);
×
4509
  taosMemoryFree(pOperator);
×
4510
  pTaskInfo->code = code;
×
4511
  return code;
×
4512
}
4513

4514
void destroyStreamScanOperatorInfo(void* param) {
8,771✔
4515
  if (param == NULL) {
8,771!
4516
    return;
×
4517
  }
4518

4519
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
8,771✔
4520
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
8,771!
4521
    destroyOperator(pStreamScan->pTableScanOp);
8,490✔
4522
  }
4523

4524
  if (pStreamScan->tqReader != NULL && pStreamScan->readerFn.tqReaderClose != NULL) {
8,772!
4525
    pStreamScan->readerFn.tqReaderClose(pStreamScan->tqReader);
8,491✔
4526
  }
4527
  if (pStreamScan->pVtableMergeHandles) {
8,772!
4528
    taosHashCleanup(pStreamScan->pVtableMergeHandles);
×
4529
    pStreamScan->pVtableMergeHandles = NULL;
×
4530
  }
4531
  if (pStreamScan->pVtableMergeBuf) {
8,772!
4532
    destroyDiskbasedBuf(pStreamScan->pVtableMergeBuf);
×
4533
    pStreamScan->pVtableMergeBuf = NULL;
×
4534
  }
4535
  if (pStreamScan->pVtableReadyHandles) {
8,772!
4536
    taosArrayDestroy(pStreamScan->pVtableReadyHandles);
×
4537
    pStreamScan->pVtableReadyHandles = NULL;
×
4538
  }
4539
  if (pStreamScan->pTableListInfo) {
8,772!
4540
    tableListDestroy(pStreamScan->pTableListInfo);
×
4541
    pStreamScan->pTableListInfo = NULL;
×
4542
  }
4543
  if (pStreamScan->matchInfo.pList) {
8,772!
4544
    taosArrayDestroy(pStreamScan->matchInfo.pList);
8,773✔
4545
  }
4546
  if (pStreamScan->pPseudoExpr) {
8,772✔
4547
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
5,223✔
4548
    taosMemoryFree(pStreamScan->pPseudoExpr);
5,223!
4549
  }
4550

4551
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
8,772✔
4552
  cleanupExprSupp(&pStreamScan->tagCalSup);
8,772✔
4553

4554
  if (pStreamScan->stateStore.updateInfoDestroy) {
8,773!
4555
    pStreamScan->stateStore.updateInfoDestroy(pStreamScan->pUpdateInfo);
8,773✔
4556
  }
4557

4558
  blockDataDestroy(pStreamScan->pRes);
8,773✔
4559
  blockDataDestroy(pStreamScan->pUpdateRes);
8,773✔
4560
  blockDataDestroy(pStreamScan->pDeleteDataRes);
8,773✔
4561
  blockDataDestroy(pStreamScan->pUpdateDataRes);
8,773✔
4562
  blockDataDestroy(pStreamScan->pCreateTbRes);
8,773✔
4563
  taosArrayDestroy(pStreamScan->pBlockLists);
8,772✔
4564
  blockDataDestroy(pStreamScan->pCheckpointRes);
8,773✔
4565

4566
  taosMemoryFree(pStreamScan);
8,773!
4567
}
4568

4569
void streamScanReleaseState(SOperatorInfo* pOperator) {
2,166✔
4570
  int32_t          code = TSDB_CODE_SUCCESS;
2,166✔
4571
  int32_t          lino = 0;
2,166✔
4572
  SStreamScanInfo* pInfo = pOperator->info;
2,166✔
4573
  void*            pBuff = NULL;
2,166✔
4574
  SEncoder*        pEnCoder = NULL;
2,166✔
4575
  SEncoder*        pScanEnCoder = NULL;
2,166✔
4576
  if (!pInfo->pState) {
2,166!
4577
    return;
395✔
4578
  }
4579
  if (!pInfo->pUpdateInfo) {
2,166✔
4580
    qDebug("stask:%s streamScanReleaseState cancel", GET_TASKID(pOperator->pTaskInfo));
395✔
4581
    return;
395✔
4582
  }
4583
  int32_t  len = 0;
1,771✔
4584
  SEncoder encoder = {0};
1,771✔
4585
  pEnCoder = &encoder;
1,771✔
4586
  tEncoderInit(pEnCoder, NULL, 0);
1,771✔
4587
  if (tStartEncode(pEnCoder) != 0) {
1,771!
4588
    code = TSDB_CODE_FAILED;
×
4589
    QUERY_CHECK_CODE(code, lino, _end);
×
4590
  }
4591
  code = pInfo->stateStore.updateInfoSerialize(pEnCoder, pInfo->pUpdateInfo);
1,771✔
4592
  QUERY_CHECK_CODE(code, lino, _end);
1,771!
4593

4594
  tEndEncode(pEnCoder);
1,771✔
4595
  len += encoder.pos;
1,771✔
4596
  tEncoderClear(pEnCoder);
1,771✔
4597
  pEnCoder = NULL;
1,771✔
4598

4599
  pBuff = taosMemoryCalloc(1, len);
1,771!
4600
  if (!pBuff) {
1,771!
4601
    code = terrno;
×
4602
    QUERY_CHECK_CODE(code, lino, _end);
×
4603
  }
4604

4605
  SEncoder scanEncoder = {0};
1,771✔
4606
  pScanEnCoder = &scanEncoder;
1,771✔
4607
  tEncoderInit(pScanEnCoder, pBuff, len);
1,771✔
4608
  if (tStartEncode(pScanEnCoder) != 0) {
1,771!
4609
    code = TSDB_CODE_FAILED;
×
4610
    QUERY_CHECK_CODE(code, lino, _end);
×
4611
  }
4612
  code = pInfo->stateStore.updateInfoSerialize(pScanEnCoder, pInfo->pUpdateInfo);
1,771✔
4613
  QUERY_CHECK_CODE(code, lino, _end);
1,771!
4614

4615
  tEndEncode(pScanEnCoder);
1,771✔
4616
  tEncoderClear(pScanEnCoder);
1,771✔
4617
  pScanEnCoder = NULL;
1,771✔
4618

4619
  pInfo->stateStore.streamStateSaveInfo(pInfo->pState, STREAM_SCAN_OP_STATE_NAME, strlen(STREAM_SCAN_OP_STATE_NAME),
1,771✔
4620
                                        pBuff, len);
4621
_end:
1,771✔
4622
  if (code != TSDB_CODE_SUCCESS) {
1,771!
4623
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4624
  }
4625
  if (pEnCoder != NULL) {
1,771!
4626
    tEndEncode(pEnCoder);
×
4627
    tEncoderClear(pEnCoder);
×
4628
  }
4629
  if (pScanEnCoder != NULL) {
1,771!
4630
    tEndEncode(pScanEnCoder);
×
4631
    tEncoderClear(pScanEnCoder);
×
4632
  }
4633
  taosMemoryFree(pBuff);
1,771!
4634
}
4635

4636
void streamScanReloadState(SOperatorInfo* pOperator) {
2,166✔
4637
  int32_t          code = TSDB_CODE_SUCCESS;
2,166✔
4638
  int32_t          lino = 0;
2,166✔
4639
  SDecoder*        pDeCoder = NULL;
2,166✔
4640
  SStreamScanInfo* pInfo = pOperator->info;
2,166✔
4641
  if (!pInfo->pState) {
2,166!
4642
    return;
395✔
4643
  }
4644
  if (!pInfo->pUpdateInfo) {
2,166✔
4645
    qDebug("stask:%s streamScanReloadState cancel", GET_TASKID(pOperator->pTaskInfo));
395✔
4646
    return;
395✔
4647
  }
4648
  void*   pBuff = NULL;
1,771✔
4649
  int32_t len = 0;
1,771✔
4650
  code = pInfo->stateStore.streamStateGetInfo(pInfo->pState, STREAM_SCAN_OP_STATE_NAME,
1,771✔
4651
                                              strlen(STREAM_SCAN_OP_STATE_NAME), &pBuff, &len);
4652
  QUERY_CHECK_CODE(code, lino, _end);
1,771!
4653

4654
  SUpdateInfo* pUpInfo = taosMemoryCalloc(1, sizeof(SUpdateInfo));
1,771!
4655
  if (!pUpInfo) {
1,771!
4656
    code = terrno;
×
4657
    QUERY_CHECK_CODE(code, lino, _end);
×
4658
  }
4659

4660
  SDecoder decoder = {0};
1,771✔
4661
  pDeCoder = &decoder;
1,771✔
4662
  tDecoderInit(pDeCoder, pBuff, len);
1,771✔
4663
  if (tStartDecode(pDeCoder) < 0) {
1,771!
4664
    lino = __LINE__;
×
4665
    goto _end;
×
4666
  }
4667
  int32_t winCode = pInfo->stateStore.updateInfoDeserialize(pDeCoder, pUpInfo);
1,771✔
4668
  taosMemoryFree(pBuff);
1,771!
4669
  if (winCode == TSDB_CODE_SUCCESS && pInfo->pUpdateInfo) {
1,771!
4670
    if (pInfo->pUpdateInfo->minTS < 0) {
1,771✔
4671
      pInfo->stateStore.updateInfoDestroy(pInfo->pUpdateInfo);
1,767✔
4672
      pInfo->pUpdateInfo = pUpInfo;
1,767✔
4673
    } else {
4674
      pInfo->stateStore.windowSBfDelete(pInfo->pUpdateInfo, 1);
4✔
4675
      code = pInfo->stateStore.windowSBfAdd(pInfo->pUpdateInfo, 1);
4✔
4676
      QUERY_CHECK_CODE(code, lino, _end);
4!
4677

4678
      QUERY_CHECK_CONDITION((pInfo->pUpdateInfo->minTS > pUpInfo->minTS), code, lino, _end,
4!
4679
                            TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
4680
      pInfo->pUpdateInfo->maxDataVersion = TMAX(pInfo->pUpdateInfo->maxDataVersion, pUpInfo->maxDataVersion);
4✔
4681
      SHashObj* curMap = pInfo->pUpdateInfo->pMap;
4✔
4682
      void*     pIte = taosHashIterate(curMap, NULL);
4✔
4683
      while (pIte != NULL) {
355✔
4684
        size_t   keySize = 0;
351✔
4685
        int64_t* pUid = taosHashGetKey(pIte, &keySize);
351✔
4686
        code = taosHashPut(pUpInfo->pMap, pUid, sizeof(int64_t), pIte, sizeof(TSKEY));
351✔
4687
        if (code == TSDB_CODE_DUP_KEY) {
351!
4688
          code = TSDB_CODE_SUCCESS;
×
4689
        }
4690
        QUERY_CHECK_CODE(code, lino, _end);
351!
4691

4692
        pIte = taosHashIterate(curMap, pIte);
351✔
4693
      }
4694
      taosHashCleanup(curMap);
4✔
4695
      pInfo->pUpdateInfo->pMap = pUpInfo->pMap;
4✔
4696
      pUpInfo->pMap = NULL;
4✔
4697
      pInfo->stateStore.updateInfoDestroy(pUpInfo);
4✔
4698
    }
4699
  } else {
4700
    pInfo->stateStore.updateInfoDestroy(pUpInfo);
×
4701
  }
4702

4703
_end:
1,771✔
4704
  if (pDeCoder != NULL) {
1,771!
4705
    tEndDecode(pDeCoder);
1,771✔
4706
    tDecoderClear(pDeCoder);
1,771✔
4707
  }
4708
  if (code != TSDB_CODE_SUCCESS) {
1,771!
4709
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4710
  }
4711
}
4712

4713
int32_t addPrimaryKeyCol(SSDataBlock* pBlock, uint8_t type, int32_t bytes) {
73✔
4714
  int32_t code = TSDB_CODE_SUCCESS;
73✔
4715
  int32_t lino = 0;
73✔
4716
  pBlock->info.rowSize += bytes;
73✔
4717
  SColumnInfoData infoData = {0};
73✔
4718
  infoData.info.type = type;
73✔
4719
  infoData.info.bytes = bytes;
73✔
4720
  void* tmp = taosArrayPush(pBlock->pDataBlock, &infoData);
73✔
4721
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
73!
4722

4723
_end:
73✔
4724
  if (code != TSDB_CODE_SUCCESS) {
73!
4725
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4726
  }
4727
  return code;
73✔
4728
}
4729

4730
static SSDataBlock* createStreamVtableBlock(SColMatchInfo *pMatchInfo, const char *idstr) {
×
4731
  int32_t code = TSDB_CODE_SUCCESS;
×
4732
  int32_t lino = 0;
×
4733
  SSDataBlock *pRes = NULL;
×
4734

4735
  QUERY_CHECK_NULL(pMatchInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
4736

4737
  code = createDataBlock(&pRes);
×
4738
  QUERY_CHECK_CODE(code, lino, _end);
×
4739
  int32_t numOfOutput = taosArrayGetSize(pMatchInfo->pList);
×
4740
  for (int32_t i = 0; i < numOfOutput; ++i) {
×
4741
    SColMatchItem*  pItem = taosArrayGet(pMatchInfo->pList, i);
×
4742
    if (!pItem->needOutput) {
×
4743
      continue;
×
4744
    }
4745
    SColumnInfoData colInfo = createColumnInfoData(pItem->dataType.type, pItem->dataType.bytes, pItem->colId);
×
4746
    code = blockDataAppendColInfo(pRes, &colInfo);
×
4747
    QUERY_CHECK_CODE(code, lino, _end);
×
4748
  }
4749

4750
_end:
×
4751
  if (code != TSDB_CODE_SUCCESS) {
×
4752
    qError("%s failed at line %d since %s, id: %s", __func__, lino, tstrerror(code), idstr);
×
4753
    if (pRes != NULL) {
×
4754
      blockDataDestroy(pRes);
×
4755
    }
4756
    pRes = NULL;
×
4757
    terrno = code;
×
4758
  }
4759
  return pRes;
×
4760
}
4761

4762
static int32_t createStreamNormalScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode,
8,771✔
4763
                                                  SNode* pTagCond, STableListInfo* pTableListInfo,
4764
                                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
4765
  QRY_PARAM_CHECK(pOptrInfo);
8,771!
4766

4767
  int32_t          code = TSDB_CODE_SUCCESS;
8,771✔
4768
  int32_t          lino = 0;
8,771✔
4769
  SArray*          pColIds = NULL;
8,771✔
4770
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
8,771!
4771
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
8,771!
4772
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
8,771✔
4773
  const char*      idstr = pTaskInfo->id.str;
8,771✔
4774
  SSHashObj*       pVtableInfos = pTaskInfo->pSubplan->pVTables;
8,771✔
4775

4776
  if (pInfo == NULL || pOperator == NULL) {
8,771!
4777
    code = terrno;
×
4778
    goto _error;
×
4779
  }
4780

4781
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
8,772✔
4782
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
8,772✔
4783

4784
  pInfo->pTagCond = pTagCond;
8,772✔
4785
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
8,772✔
4786

4787
  int32_t numOfCols = 0;
8,772✔
4788
  code = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
8,772✔
4789
  if (code != TSDB_CODE_SUCCESS) {
8,772!
4790
    goto _error;
×
4791
  }
4792

4793
  SDataType pkType = {0};
8,772✔
4794
  pInfo->primaryKeyIndex = -1;
8,772✔
4795
  pInfo->basic.primaryPkIndex = -1;
8,772✔
4796
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
8,772✔
4797
  pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
8,772✔
4798
  QUERY_CHECK_NULL(pColIds, code, lino, _error, terrno);
8,773!
4799

4800
  for (int32_t i = 0; i < numOfOutput; ++i) {
221,096✔
4801
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
212,322✔
4802
    QUERY_CHECK_NULL(id, code, lino, _error, terrno);
212,321!
4803

4804
    int16_t colId = id->colId;
212,321✔
4805
    void*   tmp = taosArrayPush(pColIds, &colId);
212,323✔
4806
    QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
212,323!
4807

4808
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
212,323✔
4809
      pInfo->primaryTsIndex = id->dstSlotId;
8,489✔
4810
    }
4811
    if (id->isPk) {
212,323✔
4812
      pInfo->primaryKeyIndex = id->dstSlotId;
73✔
4813
      pInfo->basic.primaryPkIndex = id->dstSlotId;
73✔
4814
      pkType = id->dataType;
73✔
4815
    }
4816
  }
4817

4818
  pInfo->pPartTbnameSup = NULL;
8,774✔
4819
  if (pTableScanNode->pSubtable != NULL) {
8,774✔
4820
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
3,204!
4821
    if (pSubTableExpr == NULL) {
3,204!
4822
      code = terrno;
×
4823
      goto _error;
×
4824
    }
4825

4826
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
3,204✔
4827
    code = createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
3,204✔
4828
    QUERY_CHECK_CODE(code, lino, _error);
3,204!
4829

4830
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1, &pTaskInfo->storageAPI.functionStore) != 0) {
3,204!
4831
      goto _error;
×
4832
    }
4833
  }
4834

4835
  if (pTableScanNode->pTags != NULL) {
8,774✔
4836
    int32_t    numOfTags;
4837
    SExprInfo* pTagExpr = createExpr(pTableScanNode->pTags, &numOfTags);
2,311✔
4838
    if (pTagExpr == NULL) {
2,311!
4839
      goto _error;
×
4840
    }
4841
    code = initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags, &pTaskInfo->storageAPI.functionStore);
2,311✔
4842
    if (code != 0) {
2,311!
4843
      goto _error;
×
4844
    }
4845
  }
4846

4847
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
8,774✔
4848
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
8,772✔
4849

4850
  if (pHandle->vnode) {
8,771✔
4851
    SOperatorInfo* pTableScanOp = NULL;
8,489✔
4852
    code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pTableScanOp);
8,489✔
4853
    if (pTableScanOp == NULL || code != 0) {
8,489!
4854
      qError("createTableScanOperatorInfo error, code:%d", pTaskInfo->code);
2!
4855
      goto _error;
×
4856
    }
4857

4858
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
8,487✔
4859
    if (pHandle->version > 0) {
8,487✔
4860
      pTSInfo->base.cond.endVersion = pHandle->version;
1,268✔
4861
    }
4862

4863
    STableKeyInfo* pList = NULL;
8,487✔
4864
    int32_t        num = 0;
8,487✔
4865
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
8,487✔
4866
    QUERY_CHECK_CODE(code, lino, _error);
8,489!
4867

4868
    if (pHandle->initTableReader) {
8,489✔
4869
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
1,266✔
4870
      pTSInfo->base.dataReader = NULL;
1,266✔
4871
    }
4872

4873
    if (pHandle->initTqReader) {
8,489!
4874
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
8,490✔
4875
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
8,490!
4876
    } else {
4877
      pInfo->tqReader = pHandle->tqReader;
×
4878
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4879
    }
4880

4881
    if (pVtableInfos != NULL) {
8,489!
4882
      // save vtable info into tqReader for vtable source scan
4883
      SSDataBlock* pResBlock = createStreamVtableBlock(&pInfo->matchInfo, idstr);
×
4884
      QUERY_CHECK_CODE(code, lino, _error);
×
4885
      code = pAPI->tqReaderFn.tqReaderSetVtableInfo(pInfo->tqReader, pHandle->vnode, pAPI, pVtableInfos, &pResBlock,
×
4886
                                                    idstr);
4887
      QUERY_CHECK_CODE(code, lino, _error);
×
4888
    }
4889

4890
    pInfo->pUpdateInfo = NULL;
8,489✔
4891
    pInfo->pTableScanOp = pTableScanOp;
8,489✔
4892
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
8,489✔
4893
      pAPI->stateStore.streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1,
7,222✔
4894
                                            pInfo->primaryTsIndex);
4895
    }
4896

4897
    pInfo->readHandle = *pHandle;
8,489✔
4898
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
8,489✔
4899
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
8,489✔
4900
    QUERY_CHECK_NULL(pInfo->pCreateTbRes, code, lino, _error, terrno);
8,490!
4901
    pInfo->hasPart = false;
8,490✔
4902

4903
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
8,490✔
4904
    QUERY_CHECK_CODE(code, lino, _error);
8,489!
4905

4906
    // set the extract column id to streamHandle
4907
    code = pAPI->tqReaderFn.tqReaderSetColIdList(pInfo->tqReader, pColIds, idstr);
8,489✔
4908
    QUERY_CHECK_CODE(code, lino, _error);
8,489!
4909

4910
    SArray* tableIdList = NULL;
8,489✔
4911
    code = extractTableIdList(((STableScanInfo*)(pInfo->pTableScanOp->info))->base.pTableListInfo, &tableIdList);
8,489✔
4912
    QUERY_CHECK_CODE(code, lino, _error);
8,489!
4913
    code = pAPI->tqReaderFn.tqReaderSetQueryTableList(pInfo->tqReader, tableIdList, idstr);
8,489✔
4914
    QUERY_CHECK_CODE(code, lino, _error);
8,491!
4915
    taosArrayDestroy(tableIdList);
8,491✔
4916
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->base.cond, sizeof(SQueryTableDataCond));
8,491✔
4917
  } else {
4918
    taosArrayDestroy(pColIds);
282✔
4919
    tableListDestroy(pTableListInfo);
282✔
4920
  }
4921

4922
  // clear the local variable to avoid repeatly free
4923
  pColIds = NULL;
8,773✔
4924

4925
  // create the pseduo columns info
4926
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
8,773✔
4927
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->pPseudoExpr, &pInfo->numOfPseudoExpr);
5,223✔
4928
    QUERY_CHECK_CODE(code, lino, _error);
5,223!
4929
  }
4930

4931
  code = filterInitFromNode((SNode*)pScanPhyNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
8,773✔
4932
  QUERY_CHECK_CODE(code, lino, _error);
8,773!
4933

4934
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
8,773✔
4935
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
8,772!
4936
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
8,772✔
4937
  QUERY_CHECK_CODE(code, lino, _error);
8,772!
4938

4939
  pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
8,772✔
4940
  pInfo->windowSup = (SWindowSupporter){.pStreamAggSup = NULL, .gap = -1, .parentType = QUERY_NODE_PHYSICAL_PLAN};
8,772✔
4941
  pInfo->groupId = 0;
8,772✔
4942
  pInfo->igCheckGroupId = false;
8,772✔
4943
  pInfo->pStreamScanOp = pOperator;
8,772✔
4944
  pInfo->deleteDataIndex = 0;
8,772✔
4945
  code = createSpecialDataBlock(STREAM_DELETE_DATA, &pInfo->pDeleteDataRes);
8,772✔
4946
  QUERY_CHECK_CODE(code, lino, _error);
8,773!
4947

4948
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
8,773✔
4949
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateDataRes);
8,773✔
4950
  QUERY_CHECK_CODE(code, lino, _error);
8,773!
4951

4952
  if (hasPrimaryKeyCol(pInfo)) {
8,773✔
4953
    code = addPrimaryKeyCol(pInfo->pUpdateDataRes, pkType.type, pkType.bytes);
73✔
4954
    QUERY_CHECK_CODE(code, lino, _error);
73!
4955

4956
    pInfo->pkColType = pkType.type;
73✔
4957
    pInfo->pkColLen = pkType.bytes;
73✔
4958
  }
4959

4960
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
8,773✔
4961
  pInfo->partitionSup.needCalc = false;
8,773✔
4962
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
8,773✔
4963
  pInfo->igExpired = pTableScanNode->igExpired;
8,773✔
4964
  pInfo->twAggSup.maxTs = INT64_MIN;
8,773✔
4965
  pInfo->pState = pTaskInfo->streamInfo.pState;
8,773✔
4966
  pInfo->stateStore = pTaskInfo->storageAPI.stateStore;
8,773✔
4967
  pInfo->readerFn = pTaskInfo->storageAPI.tqReaderFn;
8,773✔
4968
  pInfo->pFillSup = NULL;
8,773✔
4969
  pInfo->useGetResultRange = false;
8,773✔
4970
  pInfo->pRangeScanRes = NULL;
8,773✔
4971

4972
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
8,773✔
4973
  QUERY_CHECK_CODE(code, lino, _error);
8,773!
4974

4975
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.skey);
8,773✔
4976
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.ekey);
8,773✔
4977
  // for stream
4978
  if (pTaskInfo->streamInfo.pState) {
8,773✔
4979
    void*   buff = NULL;
7,504✔
4980
    int32_t len = 0;
7,504✔
4981
    int32_t res = pAPI->stateStore.streamStateGetInfo(pTaskInfo->streamInfo.pState, STREAM_SCAN_OP_CHECKPOINT_NAME,
7,504✔
4982
                                                      strlen(STREAM_SCAN_OP_CHECKPOINT_NAME), &buff, &len);
4983
    if (res == TSDB_CODE_SUCCESS) {
7,503✔
4984
      streamScanOperatorDecode(buff, len, pInfo);
26✔
4985
      taosMemoryFree(buff);
26!
4986
    }
4987
  }
4988

4989
  setOperatorInfo(pOperator, STREAM_SCAN_OP_NAME, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, false, OP_NOT_OPENED, pInfo,
8,768✔
4990
                  pTaskInfo);
4991
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
8,769✔
4992

4993
  __optr_fn_t nextFn = (pTaskInfo->execModel == OPTR_EXEC_MODEL_STREAM) ? doStreamScanNext : doQueueScanNext;
8,771✔
4994
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, nextFn, NULL, destroyStreamScanOperatorInfo, optrDefaultBufFn,
8,771✔
4995
                                         NULL, optrDefaultGetNextExtFn, NULL);
4996
  setOperatorStreamStateFn(pOperator, streamScanReleaseState, streamScanReloadState);
8,769✔
4997

4998
  *pOptrInfo = pOperator;
8,767✔
4999
  return code;
8,767✔
5000

5001
_error:
×
5002
  if (pColIds != NULL) {
×
5003
    taosArrayDestroy(pColIds);
×
5004
  }
5005

5006
  if (pInfo != NULL && pInfo->pTableScanOp != NULL) {
×
5007
    STableScanInfo* p = (STableScanInfo*)pInfo->pTableScanOp->info;
×
5008
    if (p != NULL) {
×
5009
      p->base.pTableListInfo = NULL;
×
5010
    }
5011
    destroyStreamScanOperatorInfo(pInfo);
×
5012
  }
5013

5014
  if (pOperator != NULL) {
×
5015
    pOperator->info = NULL;
×
5016
    destroyOperator(pOperator);
×
5017
  }
5018
  pTaskInfo->code = code;
×
5019
  return code;
×
5020
}
5021

5022
int32_t createStreamScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode, SNode* pTagCond,
8,844✔
5023
                                     STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
5024
                                     SOperatorInfo** pOptrInfo) {
5025
  if (pTableScanNode->triggerType == STREAM_TRIGGER_CONTINUOUS_WINDOW_CLOSE) {
8,844✔
5026
    return createStreamDataScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, pOptrInfo);
82✔
5027
  } else {
5028
    return createStreamNormalScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, pOptrInfo);
8,762✔
5029
  }
5030
}
5031

5032
static int32_t doTagScanOneTable(SOperatorInfo* pOperator, SSDataBlock* pRes, SMetaReader* mr, SStorageAPI* pAPI) {
21,478✔
5033
  int32_t        code = TSDB_CODE_SUCCESS;
21,478✔
5034
  int32_t        lino = 0;
21,478✔
5035
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
21,478✔
5036
  STagScanInfo*  pInfo = pOperator->info;
21,478✔
5037
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
21,478✔
5038
  int32_t        count = pRes->info.rows;
21,478✔
5039

5040
  STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
21,478✔
5041
  if (!item) {
21,468!
5042
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
5043
           GET_TASKID(pTaskInfo));
5044
    tDecoderClear(&(*mr).coder);
×
5045
    goto _end;
×
5046
  }
5047

5048
  code = pAPI->metaReaderFn.getTableEntryByUid(mr, item->uid);
21,468✔
5049
  tDecoderClear(&(*mr).coder);
21,359✔
5050
  if (code != TSDB_CODE_SUCCESS) {
21,566!
5051
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
5052
           GET_TASKID(pTaskInfo));
5053
    goto _end;
×
5054
  }
5055

5056
  char str[512];
5057
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
59,491✔
5058
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
37,970✔
5059

5060
    // refactor later
5061
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
37,915✔
5062
      STR_TO_VARSTR(str, (*mr).me.name);
21,490✔
5063
      code = colDataSetVal(pDst, (count), str, false);
21,490✔
5064
      QUERY_CHECK_CODE(code, lino, _end);
21,482!
5065
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
16,425✔
5066
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
150✔
5067
      QUERY_CHECK_CODE(code, lino, _end);
150!
5068
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
16,275✔
5069
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
150✔
5070
      QUERY_CHECK_CODE(code, lino, _end);
150!
5071
    } else {  // it is a tag value
5072
      STagVal val = {0};
16,125✔
5073
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
16,125✔
5074
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
16,125✔
5075

5076
      char* data = NULL;
16,167✔
5077
      if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
16,167!
5078
        data = tTagValToData((const STagVal*)p, false);
16,167✔
5079
      } else {
5080
        data = (char*)p;
×
5081
      }
5082

5083
      code = colDataSetVal(pDst, (count), data,
16,168✔
5084
                           (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
16,168!
5085
      QUERY_CHECK_CODE(code, lino, _end);
16,132!
5086

5087
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
16,132!
5088
          (data != NULL)) {
5089
        taosMemoryFree(data);
3,433!
5090
      }
5091
    }
5092
  }
5093

5094
_end:
21,521✔
5095
  if (code != TSDB_CODE_SUCCESS) {
21,521!
5096
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5097
    pTaskInfo->code = code;
×
5098
  } else {
5099
    pRes->info.rows++;
21,521✔
5100
  }
5101

5102
  return code;
21,521✔
5103
}
5104

5105
static void tagScanFreeUidTag(void* p) {
68,625✔
5106
  STUidTagInfo* pInfo = p;
68,625✔
5107
  if (pInfo->pTagVal != NULL) {
68,625!
5108
    taosMemoryFree(pInfo->pTagVal);
68,658!
5109
  }
5110
}
68,672✔
5111

5112
static int32_t tagScanCreateResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
15,454✔
5113
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
15,454!
5114
  if (pColumnData == NULL) {
15,468!
5115
    return terrno;
×
5116
  }
5117

5118
  pColumnData->info.type = pType->type;
15,468✔
5119
  pColumnData->info.bytes = pType->bytes;
15,468✔
5120
  pColumnData->info.scale = pType->scale;
15,468✔
5121
  pColumnData->info.precision = pType->precision;
15,468✔
5122

5123
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
15,468✔
5124
  if (code != TSDB_CODE_SUCCESS) {
15,464✔
5125
    terrno = code;
2✔
5126
    taosMemoryFree(pColumnData);
×
5127
    return terrno;
×
5128
  }
5129

5130
  pParam->columnData = pColumnData;
15,462✔
5131
  pParam->colAlloced = true;
15,462✔
5132
  return TSDB_CODE_SUCCESS;
15,462✔
5133
}
5134

5135
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
97,510✔
5136
  int32_t                code = TSDB_CODE_SUCCESS;
97,510✔
5137
  int32_t                lino = 0;
97,510✔
5138
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
97,510✔
5139
  SColumnNode*           pSColumnNode = NULL;
97,510✔
5140
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
97,510✔
5141
    pSColumnNode = *(SColumnNode**)pNode;
27,385✔
5142
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
70,125!
5143
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
×
5144
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
×
5145
      pSColumnNode = NULL;
×
5146
      pCtx->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
×
5147
      if (NULL == pSColumnNode) {
×
5148
        return DEAL_RES_ERROR;
×
5149
      }
5150
      pSColumnNode->colId = -1;
×
5151
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
×
5152
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
×
5153
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
×
5154
      nodesDestroyNode(*pNode);
×
5155
      *pNode = (SNode*)pSColumnNode;
×
5156
    } else {
5157
      return DEAL_RES_CONTINUE;
×
5158
    }
5159
  } else {
5160
    return DEAL_RES_CONTINUE;
70,125✔
5161
  }
5162

5163
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
27,385✔
5164
  if (!data) {
27,367✔
5165
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
15,454✔
5166
    if (code == TSDB_CODE_DUP_KEY) {
15,478!
5167
      code = TSDB_CODE_SUCCESS;
×
5168
    }
5169
    QUERY_CHECK_CODE(code, lino, _end);
15,478!
5170
    pSColumnNode->slotId = pCtx->index++;
15,478✔
5171
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
15,478✔
5172
                         .type = pSColumnNode->node.resType.type,
15,478✔
5173
                         .bytes = pSColumnNode->node.resType.bytes};
15,478✔
5174
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
15,478✔
5175
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
15,480!
5176
  } else {
5177
    SColumnNode* col = *(SColumnNode**)data;
11,913✔
5178
    pSColumnNode->slotId = col->slotId;
11,913✔
5179
  }
5180

5181
_end:
27,393✔
5182
  if (code != TSDB_CODE_SUCCESS) {
27,393!
5183
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5184
    return DEAL_RES_ERROR;
×
5185
  }
5186
  return DEAL_RES_CONTINUE;
27,393✔
5187
}
5188

5189
static int32_t tagScanFilterByTagCond(SArray* aUidTags, SNode* pTagCond, SArray* aFilterIdxs, void* pVnode,
15,467✔
5190
                                      SStorageAPI* pAPI, STagScanInfo* pInfo) {
5191
  int32_t code = TSDB_CODE_SUCCESS;
15,467✔
5192
  int32_t lino = 0;
15,467✔
5193
  int32_t numOfTables = taosArrayGetSize(aUidTags);
15,467✔
5194
  SArray* pBlockList = NULL;
15,468✔
5195

5196
  SSDataBlock* pResBlock = createTagValBlockForFilter(pInfo->filterCtx.cInfoList, numOfTables, aUidTags, pVnode, pAPI);
15,468✔
5197
  QUERY_CHECK_NULL(pResBlock, code, lino, _end, terrno);
15,447!
5198

5199
  pBlockList = taosArrayInit(1, POINTER_BYTES);
15,447✔
5200
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
15,468!
5201

5202
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
15,459✔
5203
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
15,459!
5204

5205
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
15,459✔
5206

5207
  SScalarParam output = {0};
15,459✔
5208
  code = tagScanCreateResultData(&type, numOfTables, &output);
15,459✔
5209
  QUERY_CHECK_CODE(code, lino, _end);
15,463!
5210

5211
  code = scalarCalculate(pTagCond, pBlockList, &output);
15,463✔
5212
  QUERY_CHECK_CODE(code, lino, _end);
15,448!
5213

5214
  bool* result = (bool*)output.columnData->pData;
15,448✔
5215
  for (int32_t i = 0; i < numOfTables; ++i) {
38,662✔
5216
    if (result[i]) {
23,215✔
5217
      void* tmp = taosArrayPush(aFilterIdxs, &i);
22,131✔
5218
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
22,131!
5219
    }
5220
  }
5221

5222
  colDataDestroy(output.columnData);
15,447✔
5223
  taosMemoryFreeClear(output.columnData);
15,465!
5224

5225
_end:
×
5226
  blockDataDestroy(pResBlock);
15,462✔
5227
  taosArrayDestroy(pBlockList);
15,465✔
5228
  if (code != TSDB_CODE_SUCCESS) {
15,466!
5229
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5230
  }
5231
  return code;
15,456✔
5232
}
5233

5234
static int32_t tagScanFillOneCellWithTag(SOperatorInfo* pOperator, const STUidTagInfo* pUidTagInfo,
282,521✔
5235
                                         SExprInfo* pExprInfo, SColumnInfoData* pColInfo, int rowIndex,
5236
                                         const SStorageAPI* pAPI, void* pVnode) {
5237
  int32_t code = TSDB_CODE_SUCCESS;
282,521✔
5238
  int32_t lino = 0;
282,521✔
5239
  if (QUERY_NODE_FUNCTION == pExprInfo->pExpr->nodeType) {
282,521✔
5240
    if (FUNCTION_TYPE_TBNAME == pExprInfo->pExpr->_function.functionType) {  // tbname
135,014!
5241
      char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5242
      STR_TO_VARSTR(str, "ctbidx");
×
5243

5244
      code = colDataSetVal(pColInfo, rowIndex, str, false);
×
5245
      QUERY_CHECK_CODE(code, lino, _end);
×
5246
    } else if (FUNCTION_TYPE_TBUID == pExprInfo->pExpr->_function.functionType) {
135,014✔
5247
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
67,570✔
5248
      QUERY_CHECK_CODE(code, lino, _end);
67,579!
5249
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
67,444!
5250
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
67,561✔
5251
      QUERY_CHECK_CODE(code, lino, _end);
67,595!
5252
    }
5253
  } else {
5254
    STagVal tagVal = {0};
147,507✔
5255
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
147,507✔
5256
    if (pUidTagInfo->pTagVal == NULL) {
147,507!
5257
      colDataSetNULL(pColInfo, rowIndex);
×
5258
    } else {
5259
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
147,507✔
5260

5261
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
147,992!
5262
        colDataSetNULL(pColInfo, rowIndex);
35!
5263
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
147,957!
5264
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
5265
        QUERY_CHECK_CODE(code, lino, _end);
×
5266
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
182,843!
5267
        char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
34,890!
5268
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
34,885!
5269

5270
        varDataSetLen(tmp, tagVal.nData);
34,885✔
5271
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
34,885✔
5272
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
34,885✔
5273
        taosMemoryFree(tmp);
34,895!
5274
        QUERY_CHECK_CODE(code, lino, _end);
34,886!
5275
      } else {
5276
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
113,067✔
5277
        QUERY_CHECK_CODE(code, lino, _end);
113,088!
5278
      }
5279
    }
5280
  }
5281

5282
_end:
283,066✔
5283
  if (code != TSDB_CODE_SUCCESS) {
283,066!
5284
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5285
  }
5286
  return code;
282,690✔
5287
}
5288

5289
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
43,985✔
5290
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
5291
  int32_t       code = TSDB_CODE_SUCCESS;
43,985✔
5292
  int32_t       lino = 0;
43,985✔
5293
  STagScanInfo* pInfo = pOperator->info;
43,985✔
5294
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
43,985✔
5295
  if (!ignoreFilterIdx) {
43,985✔
5296
    size_t szTables = taosArrayGetSize(aFilterIdxs);
15,454✔
5297
    for (int i = 0; i < szTables; ++i) {
37,630✔
5298
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
22,156✔
5299
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
22,154✔
5300
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
22,153!
5301
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
126,430✔
5302
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
104,257✔
5303
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
104,206!
5304
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
104,206✔
5305
        QUERY_CHECK_CODE(code, lino, _end);
104,277!
5306
      }
5307
    }
5308
  } else {
5309
    size_t szTables = taosArrayGetSize(aUidTags);
28,531✔
5310
    for (int i = 0; i < szTables; ++i) {
74,067✔
5311
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
45,471✔
5312
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
45,467✔
5313
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
223,882✔
5314
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
178,380✔
5315
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
178,331!
5316
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
178,331✔
5317
        QUERY_CHECK_CODE(code, lino, _end);
178,416!
5318
      }
5319
    }
5320
  }
5321

5322
_end:
28,596✔
5323
  if (code != TSDB_CODE_SUCCESS) {
44,070!
5324
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5325
  }
5326
  return code;
44,011✔
5327
}
5328

5329
static int32_t doTagScanFromCtbIdxNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
87,428✔
5330
  int32_t        code = TSDB_CODE_SUCCESS;
87,428✔
5331
  int32_t        lino = 0;
87,428✔
5332
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
87,428✔
5333
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
87,428✔
5334

5335
  STagScanInfo* pInfo = pOperator->info;
87,428✔
5336
  SSDataBlock*  pRes = pInfo->pRes;
87,428✔
5337

5338
  QRY_PARAM_CHECK(ppRes);
87,428!
5339

5340
  if (pOperator->status == OP_EXEC_DONE) {
87,428✔
5341
    return TSDB_CODE_SUCCESS;
43,307✔
5342
  }
5343
  blockDataCleanup(pRes);
44,121✔
5344

5345
  if (pInfo->pCtbCursor == NULL) {
44,203!
5346
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
44,207✔
5347
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
44,188!
5348
  } else {
5349
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
5350
    QUERY_CHECK_CODE(code, lino, _end);
×
5351
  }
5352

5353
  SArray* aUidTags = pInfo->aUidTags;
44,189✔
5354
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
44,189✔
5355
  int32_t count = 0;
44,189✔
5356
  bool    ctbCursorFinished = false;
44,189✔
5357
  while (1) {
683✔
5358
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
44,872✔
5359
    taosArrayClear(aFilterIdxs);
44,871✔
5360

5361
    int32_t numTables = 0;
44,815✔
5362
    while (numTables < pOperator->resultInfo.capacity) {
113,458!
5363
      SMCtbCursor* pCur = pInfo->pCtbCursor;
113,461✔
5364
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
113,461✔
5365
      if (uid == 0) {
113,492✔
5366
        ctbCursorFinished = true;
44,903✔
5367
        break;
44,903✔
5368
      }
5369
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
68,589✔
5370
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
68,589!
5371
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
68,674!
5372

5373
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
68,674✔
5374
      void* tmp = taosArrayPush(aUidTags, &info);
68,643✔
5375
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
68,643!
5376
      ++numTables;
68,643✔
5377
    }
5378

5379
    if (numTables == 0) {
44,900✔
5380
      break;
866✔
5381
    }
5382
    bool ignoreFilterIdx = true;
44,034✔
5383
    if (pInfo->pTagCond != NULL) {
44,034✔
5384
      ignoreFilterIdx = false;
15,468✔
5385
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo);
15,468✔
5386
      QUERY_CHECK_CODE(code, lino, _end);
15,447!
5387
    } else {
5388
      ignoreFilterIdx = true;
28,566✔
5389
    }
5390

5391
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
44,013✔
5392
    QUERY_CHECK_CODE(code, lino, _end);
44,013!
5393

5394
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
44,013✔
5395

5396
    if (count != 0) {
43,996✔
5397
      break;
43,313✔
5398
    }
5399
  }
5400

5401
  if (count > 0) {
44,179✔
5402
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
43,319✔
5403
  }
5404
  if (count == 0 || ctbCursorFinished) {
44,205!
5405
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
44,206✔
5406
    pInfo->pCtbCursor = NULL;
44,215✔
5407
    setOperatorCompleted(pOperator);
44,215✔
5408
  }
5409

5410
  pRes->info.rows = count;
44,204✔
5411
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
44,204✔
5412
  if (bLimitReached) {
44,178!
5413
    setOperatorCompleted(pOperator);
×
5414
  }
5415

5416
  pOperator->resultInfo.totalRows += pRes->info.rows;
44,205✔
5417
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
44,205✔
5418

5419
_end:
44,205✔
5420
  if (code != TSDB_CODE_SUCCESS) {
44,205!
5421
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5422
    pTaskInfo->code = code;
×
5423
    T_LONG_JMP(pTaskInfo->env, code);
×
5424
  }
5425

5426
  return code;
44,205✔
5427
}
5428

5429
static int32_t doTagScanFromMetaEntryNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
6,895✔
5430
  int32_t code = TSDB_CODE_SUCCESS;
6,895✔
5431
  int32_t lino = 0;
6,895✔
5432
  if (pOperator->status == OP_EXEC_DONE) {
6,895✔
5433
    (*ppRes) = NULL;
3,417✔
5434
    return code;
3,417✔
5435
  }
5436

5437
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3,478✔
5438
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
3,478✔
5439

5440
  STagScanInfo* pInfo = pOperator->info;
3,478✔
5441
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
3,478✔
5442
  SSDataBlock*  pRes = pInfo->pRes;
3,478✔
5443
  blockDataCleanup(pRes);
3,478✔
5444

5445
  int32_t size = 0;
3,492✔
5446
  code = tableListGetSize(pInfo->pTableListInfo, &size);
3,492✔
5447
  QUERY_CHECK_CODE(code, lino, _end);
3,490!
5448

5449
  if (size == 0) {
3,490✔
5450
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
79✔
5451
    (*ppRes) = NULL;
79✔
5452
    return code;
79✔
5453
  }
5454

5455
  SMetaReader mr = {0};
3,411✔
5456
  pAPI->metaReaderFn.initReader(&mr, pInfo->readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
3,411✔
5457
  pRes->info.rows = 0;
3,419✔
5458

5459
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
24,903!
5460
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
21,486✔
5461
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD && code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
21,478!
5462
      // ignore other error
5463
      code = TSDB_CODE_SUCCESS;
21,485✔
5464
    }
5465
    QUERY_CHECK_CODE(code, lino, _end);
21,478!
5466

5467
    if (++pInfo->curPos >= size) {
21,478✔
5468
      setOperatorCompleted(pOperator);
3,409✔
5469
    }
5470
  }
5471

5472
  pAPI->metaReaderFn.clearReader(&mr);
3,417✔
5473
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
3,417✔
5474
  if (bLimitReached) {
3,414!
5475
    setOperatorCompleted(pOperator);
×
5476
  }
5477

5478
  // qDebug("QInfo:0x%" PRIx64 ", create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
5479
  if (pOperator->status == OP_EXEC_DONE) {
3,414!
5480
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
3,414✔
5481
  }
5482

5483
  pOperator->resultInfo.totalRows += pRes->info.rows;
3,419✔
5484

5485
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
3,419!
5486

5487
_end:
3,419✔
5488
  if (code != TSDB_CODE_SUCCESS) {
3,419!
5489
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5490
    pTaskInfo->code = code;
×
5491
    T_LONG_JMP(pTaskInfo->env, code);
×
5492
  }
5493
  return code;
3,419✔
5494
}
5495

5496
static SSDataBlock* doTagScanFromMetaEntry(SOperatorInfo* pOperator) {
×
5497
  SSDataBlock* pRes = NULL;
×
5498
  int32_t      code = doTagScanFromMetaEntryNext(pOperator, &pRes);
×
5499
  return pRes;
×
5500
}
5501

5502
static void destroyTagScanOperatorInfo(void* param) {
47,550✔
5503
  STagScanInfo* pInfo = (STagScanInfo*)param;
47,550✔
5504
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
47,550!
5505
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
5506
  }
5507
  taosHashCleanup(pInfo->filterCtx.colHash);
47,550✔
5508
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
47,680✔
5509
  taosArrayDestroy(pInfo->aFilterIdxs);
47,694✔
5510
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
47,702✔
5511

5512
  blockDataDestroy(pInfo->pRes);
47,701✔
5513
  taosArrayDestroy(pInfo->matchInfo.pList);
47,711✔
5514
  tableListDestroy(pInfo->pTableListInfo);
47,709✔
5515

5516
  pInfo->pRes = NULL;
47,709✔
5517
  pInfo->pTableListInfo = NULL;
47,709✔
5518
  taosMemoryFreeClear(param);
47,709!
5519
}
47,713✔
5520

5521
int32_t createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pTagScanNode,
47,623✔
5522
                                  STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
5523
                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
5524
  QRY_PARAM_CHECK(pOptrInfo);
47,623!
5525

5526
  int32_t         code = TSDB_CODE_SUCCESS;
47,623✔
5527
  int32_t         lino = 0;
47,623✔
5528
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
47,623✔
5529
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
47,623!
5530
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
47,650!
5531
  if (pInfo == NULL || pOperator == NULL) {
47,672!
5532
    code = terrno;
×
5533
    goto _error;
×
5534
  }
5535

5536
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
47,676✔
5537

5538
  int32_t    numOfExprs = 0;
47,676✔
5539
  SExprInfo* pExprInfo = NULL;
47,676✔
5540

5541
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
47,676✔
5542
  QUERY_CHECK_CODE(code, lino, _error);
47,660!
5543

5544
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
47,660✔
5545
  QUERY_CHECK_CODE(code, lino, _error);
47,581!
5546

5547
  int32_t num = 0;
47,581✔
5548
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
47,581✔
5549
  QUERY_CHECK_CODE(code, lino, _error);
47,645!
5550

5551
  pInfo->pTagCond = pTagCond;
47,645✔
5552
  pInfo->pTagIndexCond = pTagIndexCond;
47,645✔
5553
  pInfo->suid = pPhyNode->suid;
47,645✔
5554
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
47,645✔
5555

5556
  pInfo->pTableListInfo = pTableListInfo;
47,645✔
5557
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
47,645✔
5558
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
47,684!
5559

5560
  pInfo->readHandle = *pReadHandle;
47,684✔
5561
  pInfo->curPos = 0;
47,684✔
5562

5563
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
47,684✔
5564
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
47,650✔
5565
                  pTaskInfo);
5566
  initResultSizeInfo(&pOperator->resultInfo, 4096);
47,566✔
5567
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
47,586✔
5568
  QUERY_CHECK_CODE(code, lino, _error);
47,709!
5569

5570
  if (pTagScanNode->onlyMetaCtbIdx) {
47,709✔
5571
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
44,222✔
5572
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
44,198!
5573

5574
    pInfo->aFilterIdxs = taosArrayInit(pOperator->resultInfo.capacity, sizeof(int32_t));
44,198✔
5575
    QUERY_CHECK_NULL(pInfo->aFilterIdxs, code, lino, _error, terrno);
44,191!
5576

5577
    pInfo->filterCtx.colHash =
44,203✔
5578
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
44,191✔
5579
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
44,203!
5580

5581
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
44,203✔
5582
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
44,218!
5583

5584
    if (pInfo->pTagCond != NULL) {
44,218✔
5585
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
15,481✔
5586
    }
5587
  }
5588
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
47,684✔
5589
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
47,684✔
5590
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
5591
  *pOptrInfo = pOperator;
47,670✔
5592
  return code;
47,670✔
5593

5594
_error:
×
5595
  if (pInfo) {
×
5596
    pInfo->pTableListInfo = NULL;
×
5597
  }
5598

5599
  if (pInfo != NULL) destroyTagScanOperatorInfo(pInfo);
×
5600
  if (pOperator != NULL) {
×
5601
    pOperator->info = NULL;
×
5602
    destroyOperator(pOperator);
×
5603
  }
5604
  return code;
×
5605
}
5606

5607
// table merge scan operator
5608

5609
static int32_t subTblRowCompareTsFn(const void* pLeft, const void* pRight, void* param) {
913,472✔
5610
  int32_t                 left = *(int32_t*)pLeft;
913,472✔
5611
  int32_t                 right = *(int32_t*)pRight;
913,472✔
5612
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
913,472✔
5613

5614
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
913,472✔
5615
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
913,472✔
5616

5617
  if (leftIdx == -1) {
913,472✔
5618
    return 1;
768,121✔
5619
  } else if (rightIdx == -1) {
145,351✔
5620
    return -1;
1,440✔
5621
  }
5622

5623
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
143,911✔
5624
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
143,911✔
5625
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
143,911!
5626
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
143,911!
5627
    ret = -1 * ret;
×
5628
  }
5629
  return ret;
143,911✔
5630
}
5631

5632
static int32_t subTblRowCompareTsPkFn(const void* pLeft, const void* pRight, void* param) {
×
5633
  int32_t                 left = *(int32_t*)pLeft;
×
5634
  int32_t                 right = *(int32_t*)pRight;
×
5635
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
×
5636

5637
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
×
5638
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
×
5639

5640
  if (leftIdx == -1) {
×
5641
    return 1;
×
5642
  } else if (rightIdx == -1) {
×
5643
    return -1;
×
5644
  }
5645

5646
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
×
5647
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
×
5648
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
×
5649
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
×
5650
    ret = -1 * ret;
×
5651
  }
5652
  if (ret == 0 && pInfo->pPkOrderInfo) {
×
5653
    ret = tsortComparBlockCell(pInfo->aInputs[left].pInputBlock, pInfo->aInputs[right].pInputBlock, leftIdx, rightIdx,
×
5654
                               pInfo->pPkOrderInfo);
×
5655
  }
5656
  return ret;
×
5657
}
5658

5659
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
11,166✔
5660
  int32_t code = TSDB_CODE_SUCCESS;
11,166✔
5661
  int32_t lino = 0;
11,166✔
5662
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
11,166✔
5663
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
11,166!
5664
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
11,218!
5665
  for (int i = 0; i < src->numOfCols; i++) {
36,890✔
5666
    dst->colList[i] = src->colList[i];
25,672✔
5667
  }
5668

5669
_end:
11,218✔
5670
  if (code != TSDB_CODE_SUCCESS) {
11,218!
5671
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5672
  }
5673
  return code;
11,212✔
5674
}
5675

5676
static int32_t fetchNextSubTableBlockFromReader(SOperatorInfo* pOperator, STmsSubTableInput* pInput,
16,795✔
5677
                                                bool* pSubTableHasBlock) {
5678
  int32_t code = 0;
16,795✔
5679

5680
  STableMergeScanInfo*    pInfo = pOperator->info;
16,795✔
5681
  SReadHandle*            pHandle = &pInfo->base.readHandle;
16,795✔
5682
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
16,795✔
5683
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
16,795✔
5684
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
16,795✔
5685

5686
  blockDataCleanup(pInput->pReaderBlock);
16,795✔
5687
  if (!pInput->bInMemReader) {
16,798✔
5688
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
8,486✔
5689
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
8,486✔
5690
    if (code != 0) {
8,484!
5691
      return code;
×
5692
    }
5693
  }
5694

5695
  pInfo->base.dataReader = pInput->pReader;
16,796✔
5696

5697
  while (true) {
23✔
5698
    bool hasNext = false;
16,819✔
5699
    code = pAPI->tsdReader.tsdNextDataBlock(pInfo->base.dataReader, &hasNext);
16,819✔
5700
    if (code != 0) {
16,835!
5701
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
5702
      pInfo->base.dataReader = NULL;
×
5703
      return code;
×
5704
    }
5705

5706
    if (!hasNext || isTaskKilled(pTaskInfo)) {
16,835!
5707
      if (isTaskKilled(pTaskInfo)) {
11,220!
5708
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
5709
        pInfo->base.dataReader = NULL;
×
5710
        return code;
×
5711
      }
5712

5713
      *pSubTableHasBlock = false;
11,223✔
5714
      break;
11,223✔
5715
    }
5716

5717
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
5,615✔
5718
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
5,538✔
5719
    } else {
5720
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
77✔
5721
    }
5722

5723
    uint32_t status = 0;
5,615✔
5724
    code = loadDataBlock(pOperator, &pInfo->base, pInput->pReaderBlock, &status);
5,615✔
5725
    if (code != 0) {
5,614!
5726
      pInfo->base.dataReader = NULL;
×
5727
      return code;
×
5728
    }
5729

5730
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
5,614!
5731
      *pSubTableHasBlock = false;
×
5732
      break;
×
5733
    }
5734
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pInput->pReaderBlock->info.rows == 0) {
5,614!
5735
      continue;
23✔
5736
    }
5737

5738
    *pSubTableHasBlock = true;
5,591✔
5739
    break;
5,591✔
5740
  }
5741

5742
  if (*pSubTableHasBlock) {
16,814✔
5743
    pInput->pReaderBlock->info.id.groupId =
11,183✔
5744
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
5,591✔
5745
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
5,592✔
5746
  }
5747
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
16,815✔
5748
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
12,665✔
5749
    pInput->pReader = NULL;
12,665✔
5750
  }
5751

5752
  pInfo->base.dataReader = NULL;
16,815✔
5753
  return TSDB_CODE_SUCCESS;
16,815✔
5754
}
5755

5756
static int32_t setGroupStartEndIndex(STableMergeScanInfo* pInfo) {
6,990✔
5757
  pInfo->bGroupProcessed = false;
6,990✔
5758

5759
  int32_t numOfTables = 0;
6,990✔
5760
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
6,990✔
5761
  if (code != TSDB_CODE_SUCCESS) {
7,038!
5762
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
5763
    return code;
×
5764
  }
5765

5766
  int32_t i = pInfo->tableStartIndex + 1;
7,038✔
5767
  for (; i < numOfTables; ++i) {
11,202✔
5768
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
4,389✔
5769
    if (!tableKeyInfo) {
4,395✔
5770
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
1!
5771
      return terrno;
1✔
5772
    }
5773
    if (tableKeyInfo->groupId != pInfo->groupId) {
4,394✔
5774
      break;
230✔
5775
    }
5776
  }
5777
  pInfo->tableEndIndex = i - 1;
7,043✔
5778
  return TSDB_CODE_SUCCESS;
7,043✔
5779
}
5780

5781
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
7,055✔
5782
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
18,275✔
5783
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
11,221✔
5784
    if (pInput->rowIdx == -1) {
11,221✔
5785
      continue;
5,629✔
5786
    }
5787

5788
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
5,592!
5789
      pInput->rowIdx = 0;
5,592✔
5790
      pInput->pageIdx = -1;
5,592✔
5791
    }
5792

5793
    pInput->pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
5,592!
5794
    SColumnInfoData* col = taosArrayGet(pInput->pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
5,592✔
5795
    if (!col) {
5,591!
5796
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5797
      return terrno;
×
5798
    }
5799
    pInput->aTs = (int64_t*)col->pData;
5,591✔
5800
  }
5801

5802
  __merge_compare_fn_t mergeCompareFn = (!pSubTblsInfo->pPkOrderInfo) ? subTblRowCompareTsFn : subTblRowCompareTsPkFn;
7,054!
5803
  return tMergeTreeCreate(&pSubTblsInfo->pTree, pSubTblsInfo->numSubTables, pSubTblsInfo, mergeCompareFn);
7,054✔
5804
}
5805

5806
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
6,984✔
5807
  int32_t code = setGroupStartEndIndex(pInfo);
6,984✔
5808
  if (code != TSDB_CODE_SUCCESS) {
7,026!
5809
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
5810
    return code;
×
5811
  }
5812
  STmsSubTablesMergeInfo* pSubTblsInfo = taosMemoryCalloc(1, sizeof(STmsSubTablesMergeInfo));
7,026!
5813
  if (pSubTblsInfo == NULL) {
7,045!
5814
    return terrno;
×
5815
  }
5816
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
7,045✔
5817
  if (!pSubTblsInfo->pTsOrderInfo) {
7,042!
5818
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5819
    return terrno;
×
5820
  }
5821
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
7,042✔
5822
    pSubTblsInfo->pPkOrderInfo = taosArrayGet(pInfo->pSortInfo, 1);
16✔
5823
    if (!pSubTblsInfo->pPkOrderInfo) {
×
5824
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5825
      return terrno;
×
5826
    }
5827
  } else {
5828
    pSubTblsInfo->pPkOrderInfo = NULL;
7,021✔
5829
  }
5830
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
7,021✔
5831
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
7,021!
5832
  if (pSubTblsInfo->aInputs == NULL) {
7,036!
5833
    taosMemoryFree(pSubTblsInfo);
×
5834
    return terrno;
×
5835
  }
5836
  int32_t bufPageSize = pInfo->bufPageSize;
7,036✔
5837
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
7,036✔
5838
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
7,036✔
5839
  if (code != TSDB_CODE_SUCCESS) {
7,049✔
5840
    taosMemoryFree(pSubTblsInfo->aInputs);
25!
5841
    taosMemoryFree(pSubTblsInfo);
×
5842
    return code;
×
5843
  }
5844
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
7,024✔
5845
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
7,024✔
5846

5847
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
7,024✔
5848
  return TSDB_CODE_SUCCESS;
7,024✔
5849
}
5850

5851
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
7,016✔
5852
  int32_t        code = TSDB_CODE_SUCCESS;
7,016✔
5853
  int32_t        lino = 0;
7,016✔
5854
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
7,016✔
5855
  SReadHandle*   pHandle = &pInfo->base.readHandle;
7,016✔
5856
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
7,016✔
5857

5858
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
7,016✔
5859

5860
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
18,236✔
5861
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
11,174✔
5862
    pInput->type = SUB_TABLE_MEM_BLOCK;
11,174✔
5863

5864
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
11,174✔
5865
    QUERY_CHECK_CODE(code, lino, _end);
11,213!
5866

5867
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
11,213✔
5868
    QUERY_CHECK_CODE(code, lino, _end);
11,215!
5869

5870
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
11,215✔
5871
    QUERY_CHECK_CODE(code, lino, _end);
11,219!
5872

5873
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
11,219✔
5874
    pInput->pKeyInfo = keyInfo;
11,218✔
5875

5876
    if (isTaskKilled(pTaskInfo)) {
11,218!
5877
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
×
5878
    }
5879

5880
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
11,218✔
5881
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
4,165✔
5882
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
4,165✔
5883
      QUERY_CHECK_CODE(code, lino, _end);
4,163!
5884
      pInput->bInMemReader = true;
4,163✔
5885
    } else {
5886
      pInput->pReader = NULL;
7,053✔
5887
      pInput->bInMemReader = false;
7,053✔
5888
    }
5889
    bool hasNext = true;
11,216✔
5890
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
11,216✔
5891
    QUERY_CHECK_CODE(code, lino, _end);
11,220!
5892
    if (!hasNext) {
11,220✔
5893
      pInput->rowIdx = -1;
5,631✔
5894
      ++pSubTblsInfo->numSubTablesCompleted;
5,631✔
5895
      continue;
5,631✔
5896
    } else {
5897
      pInput->rowIdx = 0;
5,589✔
5898
      pInput->pageIdx = -1;
5,589✔
5899
    }
5900
  }
5901

5902
_end:
7,062✔
5903
  if (code != TSDB_CODE_SUCCESS) {
7,062!
5904
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5905
  }
5906
  return code;
7,057✔
5907
}
5908

5909
static void adjustSubTableFromMemBlock(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
5,591✔
5910
  int32_t              code = TSDB_CODE_SUCCESS;
5,591✔
5911
  int32_t              lino = 0;
5,591✔
5912
  SExecTaskInfo*       pTaskInfo = pOperatorInfo->pTaskInfo;
5,591✔
5913
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
5,591✔
5914
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
5,591✔
5915
  bool                 hasNext = true;
5,591✔
5916
  code = fetchNextSubTableBlockFromReader(pOperatorInfo, pInput, &hasNext);
5,591✔
5917
  QUERY_CHECK_CODE(code, lino, _end);
5,592!
5918

5919
  if (!hasNext) {
5,592!
5920
    pInput->rowIdx = -1;
5,592✔
5921
    ++pSubTblsInfo->numSubTablesCompleted;
5,592✔
5922
  } else {
5923
    pInput->rowIdx = 0;
×
5924
  }
5925

5926
_end:
5,592✔
5927
  if (code != TSDB_CODE_SUCCESS) {
5,592!
5928
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5929
    pTaskInfo->code = code;
×
5930
    T_LONG_JMP(pTaskInfo->env, code);
×
5931
  }
5932
}
5,592✔
5933

5934
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
1,009,787✔
5935
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
1,009,787✔
5936
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,009,787✔
5937

5938
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,009,787✔
5939
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
1,009,787✔
5940
    ++pInput->rowIdx;
1,003,630✔
5941
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
6,157✔
5942
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
5,591!
5943
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
5,591✔
5944
    }
5945
    if (pInput->rowIdx != -1) {
5,592!
5946
      SColumnInfoData* col = taosArrayGet(pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
×
5947
      if (!col) {
×
5948
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5949
        return terrno;
×
5950
      }
5951
      pInput->pInputBlock = pInputBlock;
×
5952
      pInput->aTs = (int64_t*)col->pData;
×
5953
    }
5954
  }
5955

5956
  return tMergeTreeAdjust(pSubTblsInfo->pTree, tMergeTreeGetAdjustIndex(pSubTblsInfo->pTree));
1,009,788✔
5957
}
5958

5959
static int32_t appendChosenRowToDataBlock(STmsSubTablesMergeInfo* pSubTblsInfo, SSDataBlock* pBlock) {
1,007,967✔
5960
  int32_t            code = TSDB_CODE_SUCCESS;
1,007,967✔
5961
  int32_t            lino = 0;
1,007,967✔
5962
  STmsSubTableInput* pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,007,967✔
5963
  SSDataBlock*       pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,007,967!
5964

5965
  for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
4,055,708✔
5966
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
2,992,210✔
5967
    QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
3,032,258!
5968

5969
    SColumnInfoData* pSrcColInfo = taosArrayGet(pInputBlock->pDataBlock, i);
3,032,258✔
5970
    QUERY_CHECK_NULL(pSrcColInfo, code, lino, _end, terrno);
3,016,775!
5971
    bool isNull = colDataIsNull(pSrcColInfo, pInputBlock->info.rows, pInput->rowIdx, NULL);
3,028,809✔
5972

5973
    if (isNull) {
3,028,809✔
5974
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
178,628✔
5975
      QUERY_CHECK_CODE(code, lino, _end);
178,820!
5976
    } else {
5977
      if (pSrcColInfo->pData != NULL) {
2,850,181✔
5978
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
2,847,319!
5979
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
2,847,319✔
5980
        QUERY_CHECK_CODE(code, lino, _end);
2,866,059!
5981
      }
5982
    }
5983
  }
5984
  pBlock->info.dataLoad = 1;
1,010,622✔
5985
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
1,010,622✔
5986
  pBlock->info.rows += 1;
1,010,622✔
5987

5988
_end:
1,010,622✔
5989
  if (code != TSDB_CODE_SUCCESS) {
1,010,622!
5990
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5991
  }
5992
  return code;
1,010,499✔
5993
}
5994

5995
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
11,210✔
5996
                                       SSDataBlock** pResBlock) {
5997
  int32_t                 code = TSDB_CODE_SUCCESS;
11,210✔
5998
  int32_t                 lino = 0;
11,210✔
5999
  STableMergeScanInfo*    pInfo = pOperator->info;
11,210✔
6000
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
11,210✔
6001
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
11,210✔
6002
  bool                    finished = false;
11,210✔
6003

6004
  QRY_PARAM_CHECK(pResBlock);
11,210!
6005

6006
  blockDataCleanup(pBlock);
11,210✔
6007

6008
  while (true) {
×
6009
    while (true) {
6010
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
1,019,359✔
6011
        finished = true;
11,210✔
6012
        break;
11,210✔
6013
      }
6014

6015
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
1,008,149✔
6016
      QUERY_CHECK_CODE(code, lino, _end);
1,010,466!
6017

6018
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
1,010,466✔
6019
      QUERY_CHECK_CODE(code, lino, _end);
1,007,969!
6020

6021
      if (pBlock->info.rows >= capacity) {
1,007,969!
6022
        break;
×
6023
      }
6024
    }
6025

6026
    if (isTaskKilled(pTaskInfo)) {
11,210!
6027
      return pTaskInfo->code;
×
6028
    }
6029

6030
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo);
11,209✔
6031
    if (finished || limitReached || pBlock->info.rows > 0) {
11,209!
6032
      break;
6033
    }
6034
  }
6035

6036
  if (pBlock->info.rows > 0) {
11,209✔
6037
    *pResBlock = pBlock;
4,151✔
6038
  }
6039

6040
_end:
7,058✔
6041
  if (code != TSDB_CODE_SUCCESS) {
11,209!
6042
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6043
    pTaskInfo->code = code;
×
6044
  }
6045
  return code;
11,209✔
6046
}
6047

6048
static int32_t startSubTablesTableMergeScan(SOperatorInfo* pOperator) {
7,000✔
6049
  int32_t              code = TSDB_CODE_SUCCESS;
7,000✔
6050
  int32_t              lino = 0;
7,000✔
6051
  STableMergeScanInfo* pInfo = pOperator->info;
7,000✔
6052

6053
  code = initSubTablesMergeInfo(pInfo);
7,000✔
6054
  QUERY_CHECK_CODE(code, lino, _end);
7,024!
6055

6056
  code = initSubTableInputs(pOperator, pInfo);
7,024✔
6057
  QUERY_CHECK_CODE(code, lino, _end);
7,056!
6058

6059
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
7,056✔
6060
  QUERY_CHECK_CODE(code, lino, _end);
7,058!
6061

6062
_end:
7,058✔
6063
  if (code != TSDB_CODE_SUCCESS) {
7,058!
6064
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6065
  }
6066
  return code;
7,058✔
6067
}
6068

6069
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
2,034,382✔
6070
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
2,034,382✔
6071
  if (pSubTblsInfo != NULL) {
2,034,382✔
6072
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
7,059✔
6073

6074
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
18,281✔
6075
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
11,222✔
6076
      taosMemoryFree(pInput->tblCond.colList);
11,222!
6077
      blockDataDestroy(pInput->pReaderBlock);
11,223✔
6078
      blockDataDestroy(pInput->pPageBlock);
11,224✔
6079
      taosArrayDestroy(pInput->aBlockPages);
11,224✔
6080
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
11,224✔
6081
      pInput->pReader = NULL;
11,224✔
6082
    }
6083

6084
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
7,059✔
6085
    taosMemoryFree(pSubTblsInfo->aInputs);
7,057!
6086

6087
    taosMemoryFree(pSubTblsInfo);
7,059!
6088
    pInfo->pSubTablesMergeInfo = NULL;
7,059✔
6089

6090
    // taosMemoryTrim(0);
6091
  }
6092
}
2,034,382✔
6093

6094
int32_t doTableMergeScanParaSubTablesNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
11,029✔
6095
  QRY_PARAM_CHECK(ppRes);
11,029!
6096

6097
  int32_t lino = 0;
11,029✔
6098
  int32_t tableListSize = 0;
11,029✔
6099
  int64_t st = taosGetTimestampUs();
11,065✔
6100

6101
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
11,065✔
6102
  STableMergeScanInfo* pInfo = pOperator->info;
11,065✔
6103

6104
  if (pOperator->status == OP_EXEC_DONE) {
11,065!
6105
    return TSDB_CODE_SUCCESS;
×
6106
  }
6107

6108
  int32_t code = pOperator->fpSet._openFn(pOperator);
11,065✔
6109
  QUERY_CHECK_CODE(code, lino, _end);
11,039!
6110

6111
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
11,039✔
6112
  QUERY_CHECK_CODE(code, lino, _end);
11,087!
6113

6114
  if (!pInfo->hasGroupId) {
11,087✔
6115
    pInfo->hasGroupId = true;
6,930✔
6116

6117
    if (tableListSize == 0) {
6,930✔
6118
      setOperatorCompleted(pOperator);
121✔
6119
      (*ppRes) = NULL;
121✔
6120
      return code;
121✔
6121
    }
6122

6123
    pInfo->tableStartIndex = 0;
6,809✔
6124
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
6,809✔
6125
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
6,806!
6126
    pInfo->groupId = pTmpGpId->groupId;
6,806✔
6127
    code = startSubTablesTableMergeScan(pOperator);
6,806✔
6128
    QUERY_CHECK_CODE(code, lino, _end);
6,828!
6129
  }
6130

6131
  SSDataBlock* pBlock = NULL;
10,985✔
6132
  while (pInfo->tableStartIndex < tableListSize) {
11,215✔
6133
    if (isTaskKilled(pTaskInfo)) {
11,210!
6134
      break;
×
6135
    }
6136

6137
    code = getSubTablesSortedBlock(pOperator, pInfo->pResBlock, pOperator->resultInfo.capacity, &pBlock);
11,210✔
6138
    QUERY_CHECK_CODE(code, lino, _end);
11,208!
6139

6140
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
11,208!
6141
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
×
6142
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
×
6143

6144
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
×
6145
    }
6146

6147
    if (pBlock != NULL) {
11,208✔
6148
      pBlock->info.id.groupId = pInfo->groupId;
4,151✔
6149
      pOperator->resultInfo.totalRows += pBlock->info.rows;
4,151✔
6150
      pInfo->bGroupProcessed = true;
4,151✔
6151
      break;
4,151✔
6152
    } else {
6153
      // Data of this group are all dumped, let's try the next group
6154
      stopSubTablesTableMergeScan(pInfo);
7,057✔
6155
      if (pInfo->tableEndIndex >= tableListSize - 1) {
7,057✔
6156
        setOperatorCompleted(pOperator);
6,828✔
6157
        break;
6,827✔
6158
      }
6159

6160
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
229✔
6161
      STableKeyInfo* pTmpGpId = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
229✔
6162
      QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
230!
6163

6164
      pInfo->groupId = pTmpGpId->groupId;
230✔
6165
      code = startSubTablesTableMergeScan(pOperator);
230✔
6166
      QUERY_CHECK_CODE(code, lino, _end);
230!
6167
      resetLimitInfoForNextGroup(&pInfo->limitInfo);
230✔
6168
    }
6169
  }
6170

6171
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
10,980✔
6172

6173
_end:
10,980✔
6174
  if (code != TSDB_CODE_SUCCESS) {
10,980!
6175
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6176
    pTaskInfo->code = code;
×
6177
    T_LONG_JMP(pTaskInfo->env, code);
×
6178
  } else {
6179
    (*ppRes) = pBlock;
10,980✔
6180
  }
6181

6182
  return code;
10,980✔
6183
}
6184

6185
static void tableMergeScanDoSkipTable(uint64_t uid, void* pTableMergeOpInfo) {
50,626✔
6186
  int32_t              code = TSDB_CODE_SUCCESS;
50,626✔
6187
  int32_t              lino = 0;
50,626✔
6188
  SOperatorInfo*       pOperator = (SOperatorInfo*)pTableMergeOpInfo;
50,626✔
6189
  STableMergeScanInfo* pInfo = pOperator->info;
50,626✔
6190
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
50,626✔
6191

6192
  if (pInfo->mSkipTables == NULL) {
50,626✔
6193
    pInfo->mSkipTables = taosHashInit(pInfo->tableEndIndex - pInfo->tableStartIndex + 1,
19,399✔
6194
                                      taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_NO_LOCK);
6195
    QUERY_CHECK_NULL(pInfo->mSkipTables, code, lino, _end, terrno);
19,397!
6196
  }
6197
  int bSkip = 1;
50,624✔
6198
  if (pInfo->mSkipTables != NULL) {
50,624!
6199
    code = taosHashPut(pInfo->mSkipTables, &uid, sizeof(uid), &bSkip, sizeof(bSkip));
50,624✔
6200
    if (code == TSDB_CODE_DUP_KEY) {
50,632!
6201
      code = TSDB_CODE_SUCCESS;
×
6202
    }
6203
    QUERY_CHECK_CODE(code, lino, _end);
50,632!
6204
  }
6205

6206
_end:
50,632✔
6207
  if (code != TSDB_CODE_SUCCESS) {
50,632!
6208
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6209
    pTaskInfo->code = code;
×
6210
    T_LONG_JMP(pTaskInfo->env, code);
×
6211
  }
6212
}
50,632✔
6213

6214
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
7,824,105✔
6215
  STableMergeScanInfo* pInfo = pOperator->info;
7,824,105✔
6216
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
7,824,105✔
6217
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
7,824,105✔
6218
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
7,824,105✔
6219
  int32_t              code = 0;
7,824,105✔
6220
  bool                 hasNext = false;
7,824,105✔
6221
  STsdbReader*         reader = pInfo->base.dataReader;
7,824,105✔
6222

6223
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
7,824,105✔
6224
  if (code != 0) {
7,827,215!
6225
    pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
6226
    qError("table merge scan fetch next data block error code: %d, %s", code, GET_TASKID(pTaskInfo));
×
6227
    pTaskInfo->code = code;
×
6228
    return code;
×
6229
  }
6230

6231
  if (!hasNext || isTaskKilled(pTaskInfo)) {
7,827,215!
6232
    if (isTaskKilled(pTaskInfo)) {
1,986,911!
6233
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
×
6234
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
6235
    }
6236
    *pFinished = true;
1,987,263✔
6237
    return code;
1,987,263✔
6238
  }
6239

6240
  uint32_t status = 0;
5,840,247✔
6241
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
5,840,247✔
6242

6243
  if (code != TSDB_CODE_SUCCESS) {
5,839,495!
6244
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
×
6245
    pTaskInfo->code = code;
×
6246
    return code;
×
6247
  }
6248

6249
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
5,839,528!
6250
    *pFinished = true;
×
6251
    return code;
×
6252
  }
6253

6254
  // current block is filter out according to filter condition, continue load the next block
6255
  if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
5,839,528!
6256
    *pSkipped = true;
22,402✔
6257
    return code;
22,402✔
6258
  }
6259

6260
  return code;
5,817,126✔
6261
}
6262

6263
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
10,189,938✔
6264
  STableMergeScanSortSourceParam* source = param;
10,189,938✔
6265

6266
  SOperatorInfo*       pOperator = source->pOperator;
10,189,938✔
6267
  STableMergeScanInfo* pInfo = pOperator->info;
10,189,938✔
6268
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
10,189,938✔
6269
  SSDataBlock*         pBlock = NULL;
10,189,938✔
6270
  int64_t              st = taosGetTimestampUs();
10,190,590✔
6271
  int32_t              code = TSDB_CODE_SUCCESS;
10,190,590✔
6272

6273
  while (true) {
6274
    if (pInfo->rtnNextDurationBlocks) {
12,597,376✔
6275
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
4,772,428✔
6276
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
6277

6278
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
4,772,509✔
6279
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
2,384,679✔
6280
        ++pInfo->nextDurationBlocksIdx;
2,384,679✔
6281
      } else {
6282
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
4,772,584✔
6283
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
2,384,724✔
6284
          pInfo->nextDurationBlocks[i] = NULL;
2,384,754✔
6285
        }
6286

6287
        pInfo->rtnNextDurationBlocks = false;
2,387,860✔
6288
        pInfo->nextDurationBlocksIdx = 0;
2,387,860✔
6289
        pInfo->numNextDurationBlocks = 0;
2,387,860✔
6290
        continue;
2,387,860✔
6291
      }
6292
    } else {
6293
      bool bFinished = false;
7,824,948✔
6294
      bool bSkipped = false;
7,824,948✔
6295

6296
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
7,824,948✔
6297
      if (code != 0) {
7,826,420!
6298
        return code;
2,389,088✔
6299
      }
6300

6301
      pBlock = pInfo->pReaderBlock;
7,826,420✔
6302
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
7,826,420✔
6303
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
6304
      if (bFinished) {
7,826,619✔
6305
        pInfo->bNewFilesetEvent = false;
1,987,222✔
6306
        break;
1,987,222✔
6307
      }
6308

6309
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
5,839,397!
6310
        if (!bSkipped) {
2,389,094✔
6311
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
2,385,998✔
6312
          if (code) {
2,385,992!
6313
            *ppBlock = NULL;
×
6314
            return code;
×
6315
          }
6316

6317
          ++pInfo->numNextDurationBlocks;
2,385,992✔
6318
          if (pInfo->numNextDurationBlocks > 2) {
2,385,992!
6319
            qError("%s table merge scan prefetch %d next duration blocks. end early.", GET_TASKID(pTaskInfo),
×
6320
                   pInfo->numNextDurationBlocks);
6321
            pInfo->bNewFilesetEvent = false;
×
6322
            break;
×
6323
          }
6324
        }
6325

6326
        if (pInfo->bNewFilesetEvent) {
2,389,088!
6327
          pInfo->rtnNextDurationBlocks = true;
2,389,088✔
6328
          *ppBlock = NULL;
2,389,088✔
6329
          return code;
2,389,088✔
6330
        }
6331

6332
        if (pInfo->bNextDurationBlockEvent) {
×
6333
          pInfo->bNextDurationBlockEvent = false;
×
6334
          continue;
18,926✔
6335
        }
6336
      }
6337
      if (bSkipped) continue;
3,450,303✔
6338
    }
6339

6340
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
5,816,056✔
6341

6342
    pOperator->resultInfo.totalRows += pBlock->info.rows;
5,815,553✔
6343
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
5,815,563✔
6344
    *ppBlock = pBlock;
5,815,563✔
6345

6346
    return code;
5,815,563✔
6347
  }
6348

6349
  *ppBlock = NULL;
1,987,222✔
6350
  return code;
1,987,222✔
6351
}
6352

6353
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
2,021,128✔
6354
  int32_t code = TSDB_CODE_SUCCESS;
2,021,128✔
6355
  int32_t lino = 0;
2,021,128✔
6356
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
2,021,128✔
6357
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
2,024,576!
6358
  SBlockOrderInfo biTs = {0};
2,025,349✔
6359
  SBlockOrderInfo biPk = {0};
2,025,349✔
6360

6361
  int32_t tsTargetSlotId = 0;
2,025,349✔
6362
  int32_t pkTargetSlotId = -1;
2,025,349✔
6363
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
12,776,389✔
6364
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
10,756,022✔
6365
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
10,752,736✔
6366
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
10,752,655✔
6367
      tsTargetSlotId = colInfo->dstSlotId;
2,022,858✔
6368
      biTs.order = order;
2,022,858✔
6369
      biTs.slotId = tsTargetSlotId;
2,022,858✔
6370
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
2,022,858✔
6371
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
2,022,858✔
6372
    }
6373
    // TODO: order by just ts
6374
    if (colInfo->isPk) {
10,751,041✔
6375
      pkTargetSlotId = colInfo->dstSlotId;
10,001✔
6376
      biPk.order = order;
10,001✔
6377
      biPk.slotId = pkTargetSlotId;
10,001✔
6378
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
10,001✔
6379
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
10,001✔
6380
    }
6381
  }
6382

6383
  void* tmp = taosArrayPush(pSortInfo, &biTs);
2,023,540✔
6384
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,023,540!
6385
  if (pkTargetSlotId != -1) {
2,023,540✔
6386
    tmp = taosArrayPush(pSortInfo, &biPk);
10,007✔
6387
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
10,007!
6388
  }
6389

6390
  (*ppSortArray) = pSortInfo;
2,023,541✔
6391

6392
_end:
2,023,541✔
6393
  if (code != TSDB_CODE_SUCCESS) {
2,023,541!
6394
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6395
  }
6396
  return code;
2,020,379✔
6397
}
6398

6399
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
3,212,959✔
6400
  STableMergeScanInfo* pTmsInfo = param;
3,212,959✔
6401
  if (type == TSD_READER_NOTIFY_DURATION_START) {
3,212,959!
6402
    pTmsInfo->bNewFilesetEvent = true;
3,213,427✔
6403
  } else if (type == TSD_READER_NOTIFY_NEXT_DURATION_BLOCK) {
×
6404
    pTmsInfo->bNextDurationBlockEvent = true;
×
6405
  }
6406
  qDebug("table merge scan receive notification. type %d, fileset %d", type, info->duration.filesetId);
3,212,959✔
6407

6408
  return;
3,213,616✔
6409
}
6410

6411
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
4,373,770✔
6412
  STableMergeScanInfo* pInfo = pOperator->info;
4,373,770✔
6413
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
4,373,770✔
6414

6415
  int32_t code = TSDB_CODE_SUCCESS;
4,373,770✔
6416
  int32_t lino = 0;
4,373,770✔
6417
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
4,373,770✔
6418

6419
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
4,373,770✔
6420
  pInfo->bNewFilesetEvent = false;
4,373,770✔
6421
  pInfo->bNextDurationBlockEvent = false;
4,373,770✔
6422

6423
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
4,373,770✔
6424
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
4,373,770✔
6425

6426
  pInfo->pSortHandle = NULL;
4,373,770✔
6427
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
4,373,770✔
6428
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
4,373,770✔
6429
  if (code) {
4,373,514!
6430
    return code;
×
6431
  }
6432

6433
  if (pInfo->bSortRowId && numOfTable != 1) {
4,373,514✔
6434
    int32_t memSize = 512 * 1024 * 1024;
1,111,818✔
6435
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
1,111,818✔
6436
    QUERY_CHECK_CODE(code, lino, _end);
1,112,622!
6437
  }
6438

6439
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
4,374,318✔
6440
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
4,373,155✔
6441
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
4,372,810✔
6442

6443
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
4,372,819✔
6444
  QUERY_CHECK_CODE(code, lino, _end);
4,373,141!
6445

6446
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
4,373,141!
6447
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
4,375,851!
6448
  param->pOperator = pOperator;
4,375,851✔
6449

6450
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
4,375,851✔
6451
  if (ps == NULL) {
4,375,924!
6452
    taosMemoryFree(param);
×
6453
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
6454
  }
6455

6456
  ps->param = param;
4,375,924✔
6457
  ps->onlyRef = false;
4,375,924✔
6458
  code = tsortAddSource(pInfo->pSortHandle, ps);
4,375,924✔
6459
  QUERY_CHECK_CODE(code, lino, _end);
4,373,155!
6460

6461
  if (numOfTable == 1) {
4,373,155✔
6462
    tsortSetSingleTableMerge(pInfo->pSortHandle);
1,807,179✔
6463
  } else {
6464
    code = tsortOpen(pInfo->pSortHandle);
2,565,976✔
6465
    QUERY_CHECK_CODE(code, lino, _end);
2,568,633!
6466
  }
6467

6468
_end:
2,568,633✔
6469
  if (code != TSDB_CODE_SUCCESS) {
4,375,756!
6470
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6471
  }
6472
  return code;
4,375,678✔
6473
}
6474

6475
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
4,373,318✔
6476
  STableMergeScanInfo* pInfo = pOperator->info;
4,373,318✔
6477
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
4,373,318✔
6478
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
4,373,318✔
6479

6480
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
4,373,318✔
6481
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
4,373,613✔
6482
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
4,373,613✔
6483
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
4,373,613✔
6484
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
4,373,613✔
6485
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
4,373,613✔
6486

6487
  tsortDestroySortHandle(pInfo->pSortHandle);
4,373,613✔
6488
  pInfo->pSortHandle = NULL;
4,373,786✔
6489
}
4,373,786✔
6490

6491
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
1,978,161✔
6492
  int32_t              code = TSDB_CODE_SUCCESS;
1,978,161✔
6493
  int32_t              lino = 0;
1,978,161✔
6494
  STableMergeScanInfo* pInfo = pOperator->info;
1,978,161✔
6495
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
1,978,161✔
6496
  SReadHandle*         pHandle = &pInfo->base.readHandle;
1,978,161✔
6497
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
1,978,161✔
6498
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
1,978,161✔
6499

6500
  {
6501
    int32_t numOfTables = 0;
1,978,161✔
6502
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
1,978,161✔
6503
    QUERY_CHECK_CODE(code, lino, _end);
1,982,273!
6504

6505
    int32_t i = pInfo->tableStartIndex + 1;
1,982,273✔
6506
    for (; i < numOfTables; ++i) {
5,040,777✔
6507
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
3,119,827✔
6508
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
3,119,988!
6509
      if (tableKeyInfo->groupId != pInfo->groupId) {
3,120,406✔
6510
        break;
61,902✔
6511
      }
6512
    }
6513
    pInfo->tableEndIndex = i - 1;
1,982,852✔
6514
  }
6515
  pInfo->bGroupProcessed = false;
1,982,852✔
6516
  int32_t tableStartIdx = pInfo->tableStartIndex;
1,982,852✔
6517
  int32_t tableEndIdx = pInfo->tableEndIndex;
1,982,852✔
6518

6519
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
1,982,852✔
6520
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
1,982,852✔
6521
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
1,984,261✔
6522
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
1,984,261✔
6523
  QUERY_CHECK_CODE(code, lino, _end);
1,986,081!
6524
  if (pInfo->filesetDelimited) {
1,986,081!
6525
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
1,986,891✔
6526
  }
6527
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
1,983,864✔
6528

6529
  code = startDurationForGroupTableMergeScan(pOperator);
1,987,231✔
6530
  QUERY_CHECK_CODE(code, lino, _end);
1,987,829!
6531

6532
_end:
1,987,829✔
6533
  if (code != TSDB_CODE_SUCCESS) {
1,987,829!
6534
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6535
    pTaskInfo->code = code;
×
6536
    T_LONG_JMP(pTaskInfo->env, code);
×
6537
  }
6538
}
1,987,829✔
6539

6540
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
1,985,629✔
6541
  STableMergeScanInfo* pInfo = pOperator->info;
1,985,629✔
6542
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
1,985,629✔
6543
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
1,985,629✔
6544

6545
  stopDurationForGroupTableMergeScan(pOperator);
1,985,629✔
6546

6547
  if (pInfo->base.dataReader != NULL) {
1,985,953✔
6548
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
1,985,941✔
6549
    pInfo->base.dataReader = NULL;
1,985,896✔
6550
  }
6551
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
1,985,908!
6552
    if (pInfo->nextDurationBlocks[i]) {
×
6553
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
×
6554
      pInfo->nextDurationBlocks[i] = NULL;
×
6555
    }
6556
    pInfo->numNextDurationBlocks = 0;
×
6557
    pInfo->nextDurationBlocksIdx = 0;
×
6558
  }
6559
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
1,985,908✔
6560
  taosHashCleanup(pInfo->mSkipTables);
1,985,742✔
6561
  pInfo->mSkipTables = NULL;
1,985,681✔
6562
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
1,985,681✔
6563
}
1,985,681✔
6564

6565
// all data produced by this function only belongs to one group
6566
// slimit/soffset does not need to be concerned here, since this function only deal with data within one group.
6567
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
11,874,584✔
6568
                                              SOperatorInfo* pOperator) {
6569
  STableMergeScanInfo* pInfo = pOperator->info;
11,874,584✔
6570
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
11,874,584✔
6571
  STupleHandle*        pTupleHandle = NULL;
11,874,584✔
6572

6573
  blockDataCleanup(pResBlock);
11,874,584✔
6574

6575
  while (1) {
×
6576
    while (1) {
2,147,483,647✔
6577
      pTupleHandle = NULL;
2,147,483,647✔
6578
      int32_t code = tsortNextTuple(pHandle, &pTupleHandle);
2,147,483,647✔
6579
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647✔
6580
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
103,213!
6581
        T_LONG_JMP(pOperator->pTaskInfo->env, code);
103,213!
6582
      }
6583
      if (pTupleHandle == NULL) {
2,147,483,647✔
6584
        break;
7,237,226✔
6585
      }
6586

6587
      code = tsortAppendTupleToBlock(pInfo->pSortHandle, pResBlock, pTupleHandle);
2,147,483,647✔
6588
      if (code != TSDB_CODE_SUCCESS) {
2,147,483,647!
6589
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
6590
        T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
6591
      }
6592

6593
      if (pResBlock->info.rows >= capacity) {
2,147,483,647✔
6594
        break;
4,638,569✔
6595
      }
6596
    }
6597

6598
    if (tsortIsClosed(pHandle)) {
11,875,795!
6599
      terrno = TSDB_CODE_TSC_QUERY_CANCELLED;
×
6600
      T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
6601
    }
6602

6603
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
11,876,140✔
6604
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
11,875,815✔
6605
           pInfo->limitInfo.numOfOutputRows);
6606
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
11,875,982!
6607
      break;
6608
    }
6609
  }
6610
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
11,875,982✔
6611
}
6612

6613
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
9,541,987✔
6614
  if (pOperator->status == OP_EXEC_DONE) {
9,541,987✔
6615
    (*ppRes) = NULL;
32,905✔
6616
    return TSDB_CODE_SUCCESS;
32,905✔
6617
  }
6618

6619
  int32_t              code = TSDB_CODE_SUCCESS;
9,509,082✔
6620
  int32_t              lino = 0;
9,509,082✔
6621
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
9,509,082✔
6622
  STableMergeScanInfo* pInfo = pOperator->info;
9,509,082✔
6623

6624
  code = pOperator->fpSet._openFn(pOperator);
9,509,082✔
6625
  QUERY_CHECK_CODE(code, lino, _end);
9,510,875!
6626

6627
  int64_t st = taosGetTimestampUs();
9,517,040✔
6628

6629
  int32_t tableListSize = 0;
9,517,040✔
6630
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
9,517,040✔
6631
  QUERY_CHECK_CODE(code, lino, _end);
9,513,635!
6632

6633
  if (!pInfo->hasGroupId) {
9,513,635✔
6634
    pInfo->hasGroupId = true;
2,014,882✔
6635

6636
    if (tableListSize == 0) {
2,014,882✔
6637
      setOperatorCompleted(pOperator);
93,772✔
6638
      (*ppRes) = NULL;
93,815✔
6639
      return code;
93,815✔
6640
    }
6641
    pInfo->tableStartIndex = 0;
1,921,110✔
6642
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
1,921,110✔
6643
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
1,919,234!
6644
    pInfo->groupId = tmp->groupId;
1,919,234✔
6645
    startGroupTableMergeScan(pOperator);
1,919,234✔
6646
  }
6647

6648
  SSDataBlock* pBlock = NULL;
9,425,349✔
6649
  while (pInfo->tableStartIndex < tableListSize) {
11,874,741!
6650
    if (isTaskKilled(pTaskInfo)) {
11,874,741!
6651
      goto _end;
×
6652
    }
6653

6654
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
11,874,848✔
6655
                                              pOperator);
6656
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
11,875,714✔
6657
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
14,201✔
6658
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
14,195!
6659
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
14,195✔
6660
    }
6661
    if (pBlock != NULL) {
11,876,024✔
6662
      pBlock->info.id.groupId = pInfo->groupId;
7,502,509✔
6663
      pOperator->resultInfo.totalRows += pBlock->info.rows;
7,502,509✔
6664
      pInfo->bGroupProcessed = true;
7,502,509✔
6665
      break;
7,502,509✔
6666
    } else {
6667
      if (pInfo->bNewFilesetEvent) {
4,373,515✔
6668
        stopDurationForGroupTableMergeScan(pOperator);
2,387,851✔
6669
        code = startDurationForGroupTableMergeScan(pOperator);
2,387,849✔
6670
        QUERY_CHECK_CODE(code, lino, _end);
2,387,773!
6671
      } else {
6672
        // Data of this group are all dumped, let's try the next group
6673
        stopGroupTableMergeScan(pOperator);
1,985,664✔
6674
        if (pInfo->tableEndIndex >= tableListSize - 1) {
1,985,712✔
6675
          setOperatorCompleted(pOperator);
1,923,820✔
6676
          break;
1,923,816✔
6677
        }
6678

6679
        pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
61,892✔
6680
        STableKeyInfo* tmp = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
61,892✔
6681
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
61,973!
6682
        pInfo->groupId = tmp->groupId;
61,973✔
6683
        startGroupTableMergeScan(pOperator);
61,973✔
6684
        resetLimitInfoForNextGroup(&pInfo->limitInfo);
61,907✔
6685
      }
6686
    }
6687
  }
6688

6689
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
9,426,287✔
6690

6691
_end:
9,426,287✔
6692
  if (code != TSDB_CODE_SUCCESS) {
9,426,287!
6693
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6694
    pTaskInfo->code = code;
×
6695
    T_LONG_JMP(pTaskInfo->env, code);
×
6696
  } else {
6697
    (*ppRes) = pBlock;
9,426,287✔
6698
  }
6699

6700
  return code;
9,426,287✔
6701
}
6702

6703
static SSDataBlock* doTableMergeScan(SOperatorInfo* pOperator) {
×
6704
  SSDataBlock* pRes = NULL;
×
6705
  int32_t      code = doTableMergeScanNext(pOperator, &pRes);
×
6706
  return pRes;
×
6707
}
6708

6709
void destroyTableMergeScanOperatorInfo(void* param) {
2,027,206✔
6710
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
2,027,206✔
6711

6712
  // start one reader variable
6713
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
2,027,206!
6714
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
2,027,307✔
6715
    pTableScanInfo->base.dataReader = NULL;
2,027,317✔
6716
  }
6717

6718
  for (int32_t i = 0; i < pTableScanInfo->numNextDurationBlocks; ++i) {
2,028,508✔
6719
    if (pTableScanInfo->nextDurationBlocks[i] != NULL) {
1,292!
6720
      blockDataDestroy(pTableScanInfo->nextDurationBlocks[i]);
1,292✔
6721
      pTableScanInfo->nextDurationBlocks[i] = NULL;
1,292✔
6722
    }
6723
  }
6724

6725
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
2,027,216✔
6726
  pTableScanInfo->pSortHandle = NULL;
2,027,258✔
6727
  taosHashCleanup(pTableScanInfo->mSkipTables);
2,027,258✔
6728
  pTableScanInfo->mSkipTables = NULL;
2,027,257✔
6729
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
2,027,257✔
6730
  pTableScanInfo->pSortInputBlock = NULL;
2,027,378✔
6731
  // end one reader variable
6732

6733
  cleanupQueryTableDataCond(&pTableScanInfo->base.cond);
2,027,378✔
6734
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
2,027,438✔
6735

6736
  blockDataDestroy(pTableScanInfo->pResBlock);
2,027,298✔
6737
  pTableScanInfo->pResBlock = NULL;
2,027,439✔
6738

6739
  // remove it from the task->result list
6740
  blockDataDestroy(pTableScanInfo->pReaderBlock);
2,027,439✔
6741
  pTableScanInfo->pReaderBlock = NULL;
2,027,460✔
6742
  taosArrayDestroy(pTableScanInfo->pSortInfo);
2,027,460✔
6743

6744
  stopSubTablesTableMergeScan(pTableScanInfo);
2,027,470✔
6745

6746
  taosMemoryFreeClear(param);
2,027,317!
6747
}
2,027,444✔
6748

6749
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
252,335✔
6750
  if (pOptr == NULL) {
252,335!
6751
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
6752
    return TSDB_CODE_INVALID_PARA;
×
6753
  }
6754
  // TODO: merge these two info into one struct
6755
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
252,335!
6756
  if (!execInfo) {
252,366!
6757
    return terrno;
×
6758
  }
6759
  STableMergeScanInfo* pInfo = pOptr->info;
252,366✔
6760
  execInfo->blockRecorder = pInfo->base.readRecorder;
252,366✔
6761
  execInfo->sortExecInfo = pInfo->sortExecInfo;
252,366✔
6762

6763
  *pOptrExplain = execInfo;
252,366✔
6764
  *len = sizeof(STableMergeScanExecInfo);
252,366✔
6765

6766
  return TSDB_CODE_SUCCESS;
252,366✔
6767
}
6768

6769
int32_t createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
2,018,310✔
6770
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
6771
                                         SOperatorInfo** pOptrInfo) {
6772
  QRY_PARAM_CHECK(pOptrInfo);
2,018,310!
6773

6774
  int32_t              code = TSDB_CODE_SUCCESS;
2,018,310✔
6775
  int32_t              lino = 0;
2,018,310✔
6776
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
2,018,310!
6777
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
2,022,739!
6778
  if (pInfo == NULL || pOperator == NULL) {
2,023,038!
6779
    code = terrno;
×
6780
    goto _error;
×
6781
  }
6782

6783
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
2,023,503✔
6784

6785
  int32_t numOfCols = 0;
2,023,503✔
6786
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
2,023,503✔
6787
                             &pInfo->base.matchInfo);
6788
  QUERY_CHECK_CODE(code, lino, _error);
2,023,097!
6789

6790
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
2,023,097✔
6791
  QUERY_CHECK_CODE(code, lino, _error);
2,026,289!
6792

6793
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
2,026,289✔
6794
    SExprSupp* pSup = &pInfo->base.pseudoSup;
1,164,064✔
6795
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
1,164,064✔
6796
    QUERY_CHECK_CODE(code, lino, _error);
1,162,974!
6797

6798
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
1,162,974✔
6799
                                      &pTaskInfo->storageAPI.functionStore);
6800
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
1,159,784!
6801
  }
6802

6803
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
2,022,009✔
6804

6805
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
2,022,009✔
6806
  QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
2,025,684!
6807

6808
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
2,025,684✔
6809
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
2,025,684✔
6810
  pInfo->base.scanFlag = MAIN_SCAN;
2,025,684✔
6811
  pInfo->base.readHandle = *readHandle;
2,025,684✔
6812

6813
  pInfo->readIdx = -1;
2,025,684✔
6814

6815
  pInfo->base.limitInfo.limit.limit = -1;
2,025,684✔
6816
  pInfo->base.limitInfo.slimit.limit = -1;
2,025,684✔
6817
  pInfo->base.pTableListInfo = pTableListInfo;
2,025,684✔
6818

6819
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
2,025,684✔
6820
  pInfo->sample.seed = taosGetTimestampSec();
2,025,684✔
6821

6822
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
2,025,761✔
6823
  QUERY_CHECK_CODE(code, lino, _error);
2,023,479!
6824

6825
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
2,023,479✔
6826

6827
  pInfo->mergeLimit = -1;
2,021,269✔
6828
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
2,021,269!
6829
  if (hasLimit) {
2,021,269✔
6830
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1 ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset : pInfo->limitInfo.limit.limit;
114,822✔
6831
    pInfo->mSkipTables = NULL;
114,822✔
6832
  }
6833

6834
  initResultSizeInfo(&pOperator->resultInfo, 1024);
2,021,269✔
6835
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
2,020,277✔
6836
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
2,023,090!
6837
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
2,023,090✔
6838
  QUERY_CHECK_CODE(code, lino, _error);
2,026,314!
6839
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
2,026,314!
6840
    pInfo->bSortRowId = true;
1,047,392✔
6841
  } else {
6842
    pInfo->bSortRowId = false;
977,448✔
6843
  }
6844

6845
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
2,024,840✔
6846
  QUERY_CHECK_CODE(code, lino, _error);
2,022,776!
6847

6848
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
2,022,776✔
6849
  QUERY_CHECK_CODE(code, lino, _error);
2,021,530!
6850

6851
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
2,021,530✔
6852
  QUERY_CHECK_CODE(code, lino, _error);
2,023,060!
6853

6854
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
2,023,060✔
6855

6856
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
2,023,060✔
6857
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
2,023,060✔
6858

6859
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
2,022,740✔
6860

6861
  // start one reader variable
6862
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
2,023,672✔
6863
  QUERY_CHECK_CODE(code, lino, _error);
2,025,126!
6864

6865
  if (!tsExperimental) {
2,025,126!
6866
    pInfo->filesetDelimited = false;
×
6867
  } else {
6868
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
2,025,126✔
6869
  }
6870
  // end one reader variable
6871
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
2,025,126✔
6872
                  pInfo, pTaskInfo);
6873
  pOperator->exprSupp.numOfExprs = numOfCols;
2,023,517✔
6874

6875
  pOperator->fpSet = createOperatorFpSet(
2,023,517✔
6876
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
2,023,517✔
6877
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
6878
      NULL);
6879
  pOperator->cost.openCost = 0;
2,022,080✔
6880

6881
  *pOptrInfo = pOperator;
2,022,080✔
6882
  return code;
2,022,080✔
6883

6884
_error:
×
6885
  if (code != TSDB_CODE_SUCCESS) {
×
6886
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6887
  }
6888
  pTaskInfo->code = code;
×
6889
  if (pInfo != NULL) {
×
6890
    pInfo->base.pTableListInfo = NULL;
×
6891
    destroyTableMergeScanOperatorInfo(pInfo);
×
6892
  }
6893
  if (pOperator != NULL) {
×
6894
    pOperator->info = NULL;
×
6895
    destroyOperator(pOperator);
×
6896
  }
6897
  return code;
×
6898
}
6899

6900
// ====================================================================================================================
6901
// TableCountScanOperator
6902
static void        destoryTableCountScanOperator(void* param);
6903
static int32_t     buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
6904
                                                  SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI);
6905
static int32_t     buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
6906
                                                  SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI);
6907
static int32_t     buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6908
                                             STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName);
6909
static int32_t     buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6910
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName);
6911
static int32_t     buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6912
                                          STableCountScanSupp* pSupp, SSDataBlock* pRes);
6913
static void        buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6914
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
6915
                                               size_t perfdbTableNum);
6916
static void        buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
6917
                                              size_t infodbTableNum, size_t perfdbTableNum);
6918
static const char* GROUP_TAG_DB_NAME = "db_name";
6919
static const char* GROUP_TAG_STABLE_NAME = "stable_name";
6920

6921
int32_t tblCountScanGetGroupTagsSlotId(const SNodeList* scanCols, STableCountScanSupp* supp) {
×
6922
  if (scanCols != NULL) {
×
6923
    SNode* pNode = NULL;
×
6924
    FOREACH(pNode, scanCols) {
×
6925
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
6926
        return TSDB_CODE_QRY_SYS_ERROR;
×
6927
      }
6928
      STargetNode* targetNode = (STargetNode*)pNode;
×
6929
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
×
6930
        return TSDB_CODE_QRY_SYS_ERROR;
×
6931
      }
6932
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
×
6933
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
6934
        supp->dbNameSlotId = targetNode->slotId;
×
6935
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
6936
        supp->stbNameSlotId = targetNode->slotId;
×
6937
      }
6938
    }
6939
  }
6940
  return TSDB_CODE_SUCCESS;
×
6941
}
6942

6943
int32_t tblCountScanGetCountSlotId(const SNodeList* pseudoCols, STableCountScanSupp* supp) {
×
6944
  if (pseudoCols != NULL) {
×
6945
    SNode* pNode = NULL;
×
6946
    FOREACH(pNode, pseudoCols) {
×
6947
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
6948
        return TSDB_CODE_QRY_SYS_ERROR;
×
6949
      }
6950
      STargetNode* targetNode = (STargetNode*)pNode;
×
6951
      if (nodeType(targetNode->pExpr) != QUERY_NODE_FUNCTION) {
×
6952
        return TSDB_CODE_QRY_SYS_ERROR;
×
6953
      }
6954
      SFunctionNode* funcNode = (SFunctionNode*)(targetNode->pExpr);
×
6955
      if (funcNode->funcType == FUNCTION_TYPE_TABLE_COUNT) {
×
6956
        supp->tbCountSlotId = targetNode->slotId;
×
6957
      }
6958
    }
6959
  }
6960
  return TSDB_CODE_SUCCESS;
×
6961
}
6962

6963
int32_t tblCountScanGetInputs(SNodeList* groupTags, SName* tableName, STableCountScanSupp* supp) {
×
6964
  if (groupTags != NULL) {
×
6965
    SNode* pNode = NULL;
×
6966
    FOREACH(pNode, groupTags) {
×
6967
      if (nodeType(pNode) != QUERY_NODE_COLUMN) {
×
6968
        return TSDB_CODE_QRY_SYS_ERROR;
×
6969
      }
6970
      SColumnNode* colNode = (SColumnNode*)pNode;
×
6971
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
6972
        supp->groupByDbName = true;
×
6973
      }
6974
      if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
6975
        supp->groupByStbName = true;
×
6976
      }
6977
    }
6978
  } else {
6979
    tstrncpy(supp->dbNameFilter, tNameGetDbNameP(tableName), TSDB_DB_NAME_LEN);
×
6980
    tstrncpy(supp->stbNameFilter, tNameGetTableName(tableName), TSDB_TABLE_NAME_LEN);
×
6981
  }
6982
  return TSDB_CODE_SUCCESS;
×
6983
}
6984

6985
int32_t getTableCountScanSupp(SNodeList* groupTags, SName* tableName, SNodeList* scanCols, SNodeList* pseudoCols,
×
6986
                              STableCountScanSupp* supp, SExecTaskInfo* taskInfo) {
6987
  int32_t code = 0;
×
6988
  code = tblCountScanGetInputs(groupTags, tableName, supp);
×
6989
  if (code != TSDB_CODE_SUCCESS) {
×
6990
    qError("%s get table count scan supp. get inputs error", GET_TASKID(taskInfo));
×
6991
    return code;
×
6992
  }
6993

6994
  supp->dbNameSlotId = -1;
×
6995
  supp->stbNameSlotId = -1;
×
6996
  supp->tbCountSlotId = -1;
×
6997

6998
  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
×
6999
  if (code != TSDB_CODE_SUCCESS) {
×
7000
    qError("%s get table count scan supp. get group tags slot id error", GET_TASKID(taskInfo));
×
7001
    return code;
×
7002
  }
7003

7004
  code = tblCountScanGetCountSlotId(pseudoCols, supp);
×
7005
  if (code != TSDB_CODE_SUCCESS) {
×
7006
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
×
7007
    return code;
×
7008
  }
7009
  return code;
×
7010
}
7011

7012
int32_t createTableCountScanOperatorInfo(SReadHandle* readHandle, STableCountScanPhysiNode* pTblCountScanNode,
×
7013
                                         SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
7014
  QRY_PARAM_CHECK(pOptrInfo);
×
7015

7016
  int32_t                      code = TSDB_CODE_SUCCESS;
×
7017
  int32_t                      lino = 0;
×
7018
  SScanPhysiNode*              pScanNode = &pTblCountScanNode->scan;
×
7019
  STableCountScanOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableCountScanOperatorInfo));
×
7020
  SOperatorInfo*               pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
7021
  if (!pInfo || !pOperator) {
×
7022
    code = terrno;
×
7023
    goto _error;
×
7024
  }
7025

7026
  pInfo->readHandle = *readHandle;
×
7027

7028
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
×
7029
  initResultSizeInfo(&pOperator->resultInfo, 1);
×
7030
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
×
7031
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
×
7032

7033
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
×
7034
  QUERY_CHECK_CODE(code, lino, _error);
×
7035

7036
  code = getTableCountScanSupp(pTblCountScanNode->pGroupTags, &pTblCountScanNode->scan.tableName,
×
7037
                               pTblCountScanNode->scan.pScanCols, pTblCountScanNode->scan.pScanPseudoCols, &pInfo->supp,
7038
                               pTaskInfo);
7039
  QUERY_CHECK_CODE(code, lino, _error);
×
7040

7041
  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
×
7042
                  pInfo, pTaskInfo);
7043
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScanNext, NULL, destoryTableCountScanOperator,
×
7044
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
7045
  *pOptrInfo = pOperator;
×
7046
  return code;
×
7047

7048
_error:
×
7049
  if (pInfo != NULL) {
×
7050
    destoryTableCountScanOperator(pInfo);
×
7051
  }
7052
  if (pOperator != NULL) {
×
7053
    pOperator->info = NULL;
×
7054
    destroyOperator(pOperator);
×
7055
  }
7056
  pTaskInfo->code = code;
×
7057
  return code;
×
7058
}
7059

7060
int32_t fillTableCountScanDataBlock(STableCountScanSupp* pSupp, char* dbName, char* stbName, int64_t count,
×
7061
                                    SSDataBlock* pRes) {
7062
  int32_t code = TSDB_CODE_SUCCESS;
×
7063
  int32_t lino = 0;
×
7064
  if (pSupp->dbNameSlotId != -1) {
×
7065
    QUERY_CHECK_CONDITION((strlen(dbName) > 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
7066
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
×
7067
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
7068

7069
    char varDbName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
7070
    tstrncpy(varDataVal(varDbName), dbName, TSDB_DB_NAME_LEN);
×
7071

7072
    varDataSetLen(varDbName, strlen(dbName));
×
7073
    code = colDataSetVal(colInfoData, 0, varDbName, false);
×
7074
    QUERY_CHECK_CODE(code, lino, _end);
×
7075
  }
7076

7077
  if (pSupp->stbNameSlotId != -1) {
×
7078
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
×
7079
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
7080
    if (strlen(stbName) != 0) {
×
7081
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
7082
      tstrncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
×
7083
      varDataSetLen(varStbName, strlen(stbName));
×
7084
      code = colDataSetVal(colInfoData, 0, varStbName, false);
×
7085
      QUERY_CHECK_CODE(code, lino, _end);
×
7086
    } else {
7087
      colDataSetNULL(colInfoData, 0);
7088
    }
7089
  }
7090

7091
  if (pSupp->tbCountSlotId != -1) {
×
7092
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
×
7093
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
7094
    code = colDataSetVal(colInfoData, 0, (char*)&count, false);
×
7095
    QUERY_CHECK_CODE(code, lino, _end);
×
7096
  }
7097
  pRes->info.rows = 1;
×
7098

7099
_end:
×
7100
  if (code != TSDB_CODE_SUCCESS) {
×
7101
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7102
  }
7103
  return code;
×
7104
}
7105

7106
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
×
7107
  STableCountScanSupp* pSupp = &pInfo->supp;
×
7108
  SSDataBlock*         pRes = pInfo->pRes;
×
7109

7110
  size_t infodbTableNum;
7111
  getInfosDbMeta(NULL, &infodbTableNum);
×
7112
  infodbTableNum -= 1;
×
7113
  size_t perfdbTableNum;
7114
  getPerfDbMeta(NULL, &perfdbTableNum);
×
7115

7116
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
7117
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
7118
    return (pRes->info.rows > 0) ? pRes : NULL;
×
7119
  } else {
7120
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
7121
    return (pRes->info.rows > 0) ? pRes : NULL;
×
7122
  }
7123
}
7124

7125
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
×
7126
                                       size_t infodbTableNum, size_t perfdbTableNum) {
7127
  int32_t        code = TSDB_CODE_SUCCESS;
×
7128
  int32_t        lino = 0;
×
7129
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7130
  if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
×
7131
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
7132
    QUERY_CHECK_CODE(code, lino, _end);
×
7133
  } else if (strcmp(pSupp->dbNameFilter, TSDB_PERFORMANCE_SCHEMA_DB) == 0) {
×
7134
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
7135
    QUERY_CHECK_CODE(code, lino, _end);
×
7136
  } else if (strlen(pSupp->dbNameFilter) == 0) {
×
7137
    code = fillTableCountScanDataBlock(pSupp, "", "", infodbTableNum + perfdbTableNum, pRes);
×
7138
    QUERY_CHECK_CODE(code, lino, _end);
×
7139
  }
7140

7141
_end:
×
7142
  if (code != TSDB_CODE_SUCCESS) {
×
7143
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7144
    pTaskInfo->code = code;
×
7145
    T_LONG_JMP(pTaskInfo->env, code);
×
7146
  }
7147
  setOperatorCompleted(pOperator);
×
7148
}
×
7149

7150
static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7151
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
7152
                                        size_t perfdbTableNum) {
7153
  int32_t        code = TSDB_CODE_SUCCESS;
×
7154
  int32_t        lino = 0;
×
7155
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7156
  if (pInfo->currGrpIdx == 0) {
×
7157
    uint64_t groupId = 0;
×
7158
    if (pSupp->groupByDbName) {
×
7159
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
×
7160
    } else {
7161
      groupId = calcGroupId("", 0);
×
7162
    }
7163

7164
    pRes->info.id.groupId = groupId;
×
7165
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
7166
    QUERY_CHECK_CODE(code, lino, _end);
×
7167
  } else if (pInfo->currGrpIdx == 1) {
×
7168
    uint64_t groupId = 0;
×
7169
    if (pSupp->groupByDbName) {
×
7170
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
×
7171
    } else {
7172
      groupId = calcGroupId("", 0);
×
7173
    }
7174

7175
    pRes->info.id.groupId = groupId;
×
7176
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
7177
    QUERY_CHECK_CODE(code, lino, _end);
×
7178
  } else {
7179
    setOperatorCompleted(pOperator);
×
7180
  }
7181
  pInfo->currGrpIdx++;
×
7182

7183
_end:
×
7184
  if (code != TSDB_CODE_SUCCESS) {
×
7185
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7186
    pTaskInfo->code = code;
×
7187
    T_LONG_JMP(pTaskInfo->env, code);
×
7188
  }
7189
}
×
7190

7191
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
×
7192
  int32_t                      code = TSDB_CODE_SUCCESS;
×
7193
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
×
7194
  STableCountScanOperatorInfo* pInfo = pOperator->info;
×
7195
  STableCountScanSupp*         pSupp = &pInfo->supp;
×
7196
  SSDataBlock*                 pRes = pInfo->pRes;
×
7197

7198
  blockDataCleanup(pRes);
×
7199
  QRY_PARAM_CHECK(ppRes);
×
7200

7201
  if (pOperator->status == OP_EXEC_DONE) {
×
7202
    return code;
×
7203
  }
7204

7205
  if (pInfo->readHandle.mnd != NULL) {
×
7206
    (*ppRes) = buildSysDbTableCount(pOperator, pInfo);
×
7207
    return code;
×
7208
  }
7209

7210
  code = buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
×
7211
  if (code != TSDB_CODE_SUCCESS) {
×
7212
    qError("%s failed since %s", __func__, tstrerror(code));
×
7213
    pTaskInfo->code = code;
×
7214
    T_LONG_JMP(pTaskInfo->env, code);
×
7215
  }
7216
  if (pRes->info.rows > 0) {
×
7217
    *ppRes = pRes;
×
7218
  }
7219

7220
  return code;
×
7221
}
7222

7223
static int32_t buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7224
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes) {
7225
  int32_t        code = TSDB_CODE_SUCCESS;
×
7226
  int32_t        lino = 0;
×
7227
  const char*    db = NULL;
×
7228
  int32_t        vgId = 0;
×
7229
  char           dbName[TSDB_DB_NAME_LEN] = {0};
×
7230
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7231
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
7232

7233
  // get dbname
7234
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, &db, &vgId, NULL, NULL);
×
7235
  SName sn = {0};
×
7236

7237
  code = tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
×
7238
  QUERY_CHECK_CODE(code, lino, _end);
×
7239

7240
  code = tNameGetDbName(&sn, dbName);
×
7241
  QUERY_CHECK_CODE(code, lino, _end);
×
7242

7243
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
7244
    code = buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
×
7245
  } else {
7246
    code = buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
×
7247
  }
7248

7249
_end:
×
7250
  if (code != TSDB_CODE_SUCCESS) {
×
7251
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7252
    pTaskInfo->code = code;
×
7253
  }
7254

7255
  return code;
×
7256
}
7257

7258
static int32_t buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7259
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
7260
  int32_t        code = TSDB_CODE_SUCCESS;
×
7261
  int32_t        lino = 0;
×
7262
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7263
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
7264

7265
  if (pSupp->groupByStbName) {
×
7266
    if (pInfo->stbUidList == NULL) {
×
7267
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
×
7268
      QUERY_CHECK_NULL(pInfo->stbUidList, code, lino, _end, terrno);
×
7269
      code = pAPI->metaFn.storeGetTableList(pInfo->readHandle.vnode, TSDB_SUPER_TABLE, pInfo->stbUidList);
×
7270
      QUERY_CHECK_CODE(code, lino, _end);
×
7271
    }
7272

7273
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
×
7274
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
×
7275
      code = buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid, pAPI);
×
7276
      QUERY_CHECK_CODE(code, lino, _end);
×
7277

7278
      pInfo->currGrpIdx++;
×
7279
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
×
7280
      code = buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName, pAPI);
×
7281
      QUERY_CHECK_CODE(code, lino, _end);
×
7282

7283
      pInfo->currGrpIdx++;
×
7284
    } else {
7285
      setOperatorCompleted(pOperator);
×
7286
    }
7287
  } else {
7288
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
×
7289
    pRes->info.id.groupId = groupId;
×
7290

7291
    int64_t dbTableCount = 0;
×
7292
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &dbTableCount, NULL);
×
7293
    code = fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
×
7294
    QUERY_CHECK_CODE(code, lino, _end);
×
7295
    setOperatorCompleted(pOperator);
×
7296
  }
7297

7298
_end:
×
7299
  if (code != TSDB_CODE_SUCCESS) {
×
7300
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7301
    pTaskInfo->code = code;
×
7302
  }
7303
  return code;
×
7304
}
7305

7306
static int32_t buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7307
                                         STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
7308
  int32_t        code = TSDB_CODE_SUCCESS;
×
7309
  int32_t        lino = 0;
×
7310
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7311
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
7312

7313
  if (strlen(pSupp->dbNameFilter) != 0) {
×
7314
    if (strlen(pSupp->stbNameFilter) != 0) {
×
7315
      uint64_t uid = 0;
×
7316
      code = pAPI->metaFn.getTableUidByName(pInfo->readHandle.vnode, pSupp->stbNameFilter, &uid);
×
7317
      QUERY_CHECK_CODE(code, lino, _end);
×
7318

7319
      int64_t numOfChildTables = 0;
×
7320
      code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, uid, &numOfChildTables, NULL);
×
7321
      QUERY_CHECK_CODE(code, lino, _end);
×
7322

7323
      code = fillTableCountScanDataBlock(pSupp, dbName, pSupp->stbNameFilter, numOfChildTables, pRes);
×
7324
      QUERY_CHECK_CODE(code, lino, _end);
×
7325
    } else {
7326
      int64_t tbNumVnode = 0;
×
7327
      pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
7328
      code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
7329
      QUERY_CHECK_CODE(code, lino, _end);
×
7330
    }
7331
  } else {
7332
    int64_t tbNumVnode = 0;
×
7333
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
7334
    code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
7335
    QUERY_CHECK_CODE(code, lino, _end);
×
7336
  }
7337

7338
_end:
×
7339
  if (code != TSDB_CODE_SUCCESS) {
×
7340
    pTaskInfo->code = code;
×
7341
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7342
  }
7343

7344
  setOperatorCompleted(pOperator);
×
7345
  return code;
×
7346
}
7347

7348
static int32_t buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
7349
                                              SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI) {
7350
  int32_t code = TSDB_CODE_SUCCESS;
×
7351
  int32_t lino = 0;
×
7352
  char    fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
7353
  if (pSupp->groupByDbName) {
×
7354
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
×
7355
  }
7356

7357
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
7358
  pRes->info.id.groupId = groupId;
×
7359

7360
  int64_t numOfTables = 0;
×
7361
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, NULL, &numOfTables);
×
7362

7363
  if (numOfTables != 0) {
×
7364
    code = fillTableCountScanDataBlock(pSupp, dbName, "", numOfTables, pRes);
×
7365
  }
7366

7367
_end:
×
7368
  if (code != TSDB_CODE_SUCCESS) {
×
7369
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7370
  }
7371
  return code;
×
7372
}
7373

7374
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
7375
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
7376
  int32_t code = TSDB_CODE_SUCCESS;
×
7377
  int32_t lino = 0;
×
7378
  char    stbName[TSDB_TABLE_NAME_LEN] = {0};
×
7379
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
×
7380
  QUERY_CHECK_CODE(code, lino, _end);
×
7381

7382
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
7383
  if (pSupp->groupByDbName) {
×
7384
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
×
7385
  } else {
7386
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
×
7387
  }
7388

7389
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
7390
  pRes->info.id.groupId = groupId;
×
7391

7392
  int64_t ctbNum = 0;
×
7393
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL);
×
7394
  QUERY_CHECK_CODE(code, lino, _end);
×
7395
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
×
7396
  QUERY_CHECK_CODE(code, lino, _end);
×
7397

7398
_end:
×
7399
  if (code != TSDB_CODE_SUCCESS) {
×
7400
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7401
  }
7402
  return code;
×
7403
}
7404

7405
static void destoryTableCountScanOperator(void* param) {
×
7406
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
×
7407
  blockDataDestroy(pTableCountScanInfo->pRes);
×
7408

7409
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
×
7410
  taosMemoryFreeClear(param);
×
7411
}
×
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