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

taosdata / TDengine / #3981

29 Apr 2025 09:40AM UTC coverage: 62.469% (+0.01%) from 62.459%
#3981

push

travis-ci

web-flow
Merge pull request #30938 from taosdata/fix/taosdCmdArgs

fix: taosd cmd args

155555 of 317399 branches covered (49.01%)

Branch coverage included in aggregate %.

1 of 4 new or added lines in 1 file covered. (25.0%)

376 existing lines in 99 files now uncovered.

240435 of 316504 relevant lines covered (75.97%)

6428957.83 hits per line

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

62.55
/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
#include "function.h"
40

41
int32_t scanDebug = 0;
42

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

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

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

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

66
  STableCountScanSupp supp;
67

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

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

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

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

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

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

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

104
  if (order == TSDB_ORDER_ASC) {
385,422✔
105
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.skey);
361,949✔
106
    if (w.ekey < pBlockInfo->window.skey) {
361,832!
107
      qError("w.ekey:%" PRId64 " < pBlockInfo->window.skey:%" PRId64, w.ekey, pBlockInfo->window.skey);
×
108
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
109
    }
110

111
    if (w.ekey < pBlockInfo->window.ekey) {
361,832✔
112
      *overlap = true;
327,121✔
113
      return code;
327,121✔
114
    }
115

116
    while (1) {
117
      getNextTimeWindow(pInterval, &w, order);
34,725✔
118
      if (w.skey > pBlockInfo->window.ekey) {
34,649✔
119
        break;
34,104✔
120
      }
121

122
      if (w.ekey <= pBlockInfo->window.ekey) {
545!
123
        qError("w.ekey:%" PRId64 " <= pBlockInfo->window.ekey:%" PRId64, w.ekey, pBlockInfo->window.ekey);
×
124
        return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
125
      }
126
      if (TMAX(w.skey, pBlockInfo->window.skey) <= pBlockInfo->window.ekey) {
545✔
127
        *overlap = true;
531✔
128
        return code;
531✔
129
      }
130
    }
131
  } else {
132
    w = getAlignQueryTimeWindow(pInterval, pBlockInfo->window.ekey);
23,473✔
133
    if (w.skey > pBlockInfo->window.ekey) {
24,290!
134
      qError("w.skey:%" PRId64 " > pBlockInfo->window.skey:%" PRId64, w.skey, pBlockInfo->window.ekey);
×
135
      return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
136
    }
137

138
    if (w.skey > pBlockInfo->window.skey) {
24,290✔
139
      *overlap = true;
21,084✔
140
      return code;
21,084✔
141
    }
142

143
    while (1) {
144
      getNextTimeWindow(pInterval, &w, order);
3,207✔
145
      if (w.ekey < pBlockInfo->window.skey) {
3,204✔
146
        break;
3,179✔
147
      }
148

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

160
  *overlap = false;
37,283✔
161
  return code;
37,283✔
162
}
163

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

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

173
  STableScanInfo* pTableScanInfo = pOperator->info;
622,377✔
174

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

178
  if (p1 == NULL) {
623,503✔
179
    return NULL;
273,396✔
180
  }
181

182
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
350,107✔
183
  if (NULL == *pPage) {
349,848!
184
    return NULL;
×
185
  }
186

187
  return (SResultRow*)((char*)(*pPage) + p1->offset);
349,848✔
188
}
189

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

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

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

213
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
3,836,826✔
214
    return TSDB_CODE_SUCCESS;
3,216,452✔
215
  }
216

217
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
620,374✔
218

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

222
  if (pRow == NULL) {
623,191✔
223
    return TSDB_CODE_SUCCESS;
273,413✔
224
  }
225

226
  bool notLoadBlock = true;
349,778✔
227
  for (int32_t i = 0; i < pSup1->numOfExprs; ++i) {
380,832✔
228
    int32_t functionId = pSup1->pCtx[i].functionId;
374,278✔
229

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

232
    EFuncDataRequired reqStatus = fmFuncDynDataRequired(functionId, pEntry, pBlockInfo);
374,129✔
233
    if (reqStatus != FUNC_DATA_REQUIRED_NOT_LOAD) {
374,240✔
234
      notLoadBlock = false;
343,186✔
235
      break;
343,186✔
236
    }
237
  }
238

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

242
  if (notLoadBlock) {
349,596✔
243
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
6,561✔
244
    code = insertTableToScanIgnoreList(pTableScanInfo, pBlockInfo->id.uid);
6,561✔
245
  }
246

247
  return code;
349,653✔
248
}
249

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

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

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

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

275
  if (!allColumnsHaveAgg || hasNullSMA) {
925,573!
276
    *pLoad = false;
880,653✔
277
  } else {
278
    *pLoad = true;
44,920✔
279
  }
280

281
  return code;
925,573✔
282
}
283

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

298
  return code;
4,377,770✔
299
}
300

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

305
  if (pLimitInfo->remainOffset > 0) {
7,359,442✔
306
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
45,560✔
307
      pLimitInfo->remainOffset -= pBlock->info.rows;
21,527✔
308
      blockDataEmpty(pBlock);
21,527✔
309
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
21,527✔
310
      return false;
21,527✔
311
    } else {
312
      int32_t code = blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
24,033✔
313
      if (code != TSDB_CODE_SUCCESS) {
24,033!
314
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
315
        pTaskInfo->code = code;
×
316
        T_LONG_JMP(pTaskInfo->env, code);
×
317
      }
318
      pLimitInfo->remainOffset = 0;
24,033✔
319
    }
320
  }
321

322
  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
7,337,915✔
323
    // limit the output rows
324
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
152,530✔
325
    blockDataKeepFirstNRows(pBlock, keep);
152,530✔
326

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

332
  pLimitInfo->numOfOutputRows += pBlock->info.rows;
7,185,385✔
333
  return false;
7,185,385✔
334
}
335

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

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

348
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
4,018,431✔
349

350
  pCost->totalBlocks += 1;
4,018,431✔
351
  pCost->totalRows += pBlock->info.rows;
4,018,431✔
352
  *status = pTableScanInfo->dataBlockLoadFlag;
4,018,431✔
353

354
  if (pOperator->exprSupp.pFilterInfo != NULL) {
4,018,431✔
355
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
673,317✔
356
  } else {
357
    bool overlap = false;
3,345,114✔
358
    int  ret =
359
        overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order, &overlap);
3,345,114✔
360
    if (ret != TSDB_CODE_SUCCESS) {
3,344,654!
361
      return ret;
×
362
    }
363
    if (overlap) {
3,344,654✔
364
      (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
348,787✔
365
    }
366
  }
367

368
  SDataBlockInfo* pBlockInfo = &pBlock->info;
4,017,971✔
369
  taosMemoryFreeClear(pBlock->pBlockAgg);
4,017,971!
370

371
  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
4,017,971!
372
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
×
373
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
374
    pCost->filterOutBlocks += 1;
×
375
    pCost->totalRows += pBlock->info.rows;
×
376
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
377
    return TSDB_CODE_SUCCESS;
×
378
  } else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
4,017,971✔
379
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
164,127✔
380
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows,
381
           pBlockInfo->id.uid);
382
    code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
164,127✔
383
    pCost->skipBlocks += 1;
164,110✔
384
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
164,110✔
385
    return code;
164,151✔
386
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
3,853,844✔
387
    pCost->loadBlockStatis += 1;
252,848✔
388
    loadSMA = true;  // mark the operation of load sma;
252,848✔
389
    bool success = true;
252,848✔
390
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
252,848✔
391
    if (code) {
252,798!
392
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
393
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
394
      QUERY_CHECK_CODE(code, lino, _end);
×
395
    }
396

397
    if (success) {  // failed to load the block sma data, data block statistics does not exist, load data block instead
252,543✔
398
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
12,759✔
399
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
400
      code = doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
12,759✔
401
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
12,789✔
402
      return code;
16,096✔
403
    } else {
404
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
239,784✔
405
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
239,759✔
406
    }
407
  }
408

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

415
  // try to filter data block according to sma info
416
  if (pOperator->exprSupp.pFilterInfo != NULL && (!loadSMA)) {
3,840,755!
417
    bool success = true;
672,905✔
418
    code = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo, &success);
672,905✔
419
    if (code) {
673,089!
420
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
×
421
      qError("%s failed to retrieve sma info", GET_TASKID(pTaskInfo));
×
422
      QUERY_CHECK_CODE(code, lino, _end);
×
423
    }
424

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

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

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

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

451
  // try to filter data block according to current results
452
  code = doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
3,840,518✔
453
  if (code) {
3,839,439✔
454
    pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->dataReader);
1,885✔
455
    QUERY_CHECK_CODE(code, lino, _end);
×
456
  }
457

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

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

473
  pCost->totalCheckedRows += pBlock->info.rows;
3,830,983✔
474
  pCost->loadBlocks += 1;
3,830,983✔
475

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

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

492
  // restore the previous value
493
  pCost->totalRows -= pBlock->info.rows;
3,832,158✔
494

495
  if (pOperator->exprSupp.pFilterInfo != NULL) {
3,832,158✔
496
    code = doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo);
673,239✔
497
    QUERY_CHECK_CODE(code, lino, _end);
673,216✔
498

499
    int64_t st = taosGetTimestampUs();
673,091✔
500
    double  el = (taosGetTimestampUs() - st) / 1000.0;
673,294✔
501
    pTableScanInfo->readRecorder.filterTime += el;
673,294✔
502

503
    if (pBlock->info.rows == 0) {
673,294✔
504
      pCost->filterOutBlocks += 1;
103,876✔
505
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
103,876✔
506
             GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows, el);
507
    } else {
508
      qDebug("%s data block filter applied, elapsed time:%.2f ms", GET_TASKID(pTaskInfo), el);
569,418✔
509
    }
510
  }
511

512
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
3,832,220✔
513
  if (limitReached) {  // set operator flag is done
3,830,143✔
514
    setOperatorCompleted(pOperator);
26,080✔
515
  }
516

517
  pCost->totalRows += pBlock->info.rows;
3,832,227✔
518

519
_end:
3,832,228✔
520
  if (code != TSDB_CODE_SUCCESS) {
3,832,228✔
521
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1!
522
  }
523
  return code;
3,831,919✔
524
}
525

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

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

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

540
static void freeTableCachedVal(void* param) {
1,038,107✔
541
  if (param == NULL) {
1,038,107!
542
    return;
×
543
  }
544

545
  STableCachedVal* pVal = param;
1,038,107✔
546
  taosMemoryFree((void*)pVal->pName);
1,038,107!
547
  taosMemoryFree(pVal->pTags);
1,039,631!
548
  taosMemoryFree(pVal);
1,039,584!
549
}
550

551
static int32_t createTableCacheVal(const SMetaReader* pMetaReader, STableCachedVal** ppResVal) {
1,038,755✔
552
  int32_t          code = TSDB_CODE_SUCCESS;
1,038,755✔
553
  int32_t          lino = 0;
1,038,755✔
554
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
1,038,755!
555
  QUERY_CHECK_NULL(pVal, code, lino, _end, terrno);
1,039,179!
556

557
  pVal->pTags = NULL;
1,039,179✔
558
  pVal->pName = taosStrdup(pMetaReader->me.name);
1,039,179!
559
  QUERY_CHECK_NULL(pVal->pName, code, lino, _end, terrno);
1,038,920!
560

561
  // only child table has tag value
562
  if (pMetaReader->me.type == TSDB_CHILD_TABLE || pMetaReader->me.type == TSDB_VIRTUAL_CHILD_TABLE) {
1,038,920✔
563
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
1,037,555✔
564
    pVal->pTags = taosMemoryMalloc(pTag->len);
1,037,555✔
565
    QUERY_CHECK_NULL(pVal->pTags, code, lino, _end, terrno);
1,036,792!
566
    memcpy(pVal->pTags, pTag, pTag->len);
1,036,792✔
567
  }
568

569
  (*ppResVal) = pVal;
1,038,157✔
570

571
_end:
1,038,157✔
572
  if (code != TSDB_CODE_SUCCESS) {
1,038,157!
573
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
574
    freeTableCachedVal(pVal);
×
575
  }
576
  return code;
1,038,210✔
577
}
578

579
// const void *key, size_t keyLen, void *value
580
static void freeCachedMetaItem(const void* key, size_t keyLen, void* value, void* ud) {
1,038,057✔
581
  (void)key;
582
  (void)keyLen;
583
  (void)ud;
584
  freeTableCachedVal(value);
1,038,057✔
585
}
1,039,548✔
586

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

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

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

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

613
  // currently only the tbname pseudo column
614
  if (numOfExpr <= 0) {
1,910,453✔
615
    return TSDB_CODE_SUCCESS;
5,980✔
616
  }
617

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

620
  // backup the rows
621
  int32_t backupRows = pBlock->info.rows;
1,904,473✔
622
  pBlock->info.rows = rows;
1,904,473✔
623

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

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

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

646
    pHandle->api.metaReaderFn.readerReleaseLock(&mr);
869✔
647

648
    val.pName = mr.me.name;
870✔
649
    val.pTags = (STag*)mr.me.ctbEntry.pTags;
870✔
650

651
    freeReader = true;
870✔
652
  } else {
653
    pCache->metaFetch += 1;
1,903,603✔
654

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

674
      pHandle->api.metaReaderFn.readerReleaseLock(&mr);
1,038,028✔
675

676
      code = createTableCacheVal(&mr, &pVal);
1,039,130✔
677
      QUERY_CHECK_CODE(code, lino, _end);
1,038,093!
678

679
      val = *pVal;
1,038,093✔
680
    } else {
681
      pCache->cacheHit += 1;
865,515✔
682
      STableCachedVal* pValTmp = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
865,515✔
683
      val = *pValTmp;
865,210✔
684

685
      bool bRes = taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
865,210✔
686
      qTrace("release LRU cache, res %d", bRes);
865,796✔
687
    }
688

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

693
  for (int32_t j = 0; j < numOfExpr; ++j) {
7,436,909✔
694
    const SExprInfo* pExpr1 = &pExpr[j];
5,530,338✔
695
    int32_t          dstSlotId = pExpr1->base.resSchema.slotId;
5,530,338✔
696

697
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
5,530,338✔
698
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
5,529,315✔
699

700
    int32_t functionId = pExpr1->pExpr->_function.functionId;
5,528,471✔
701

702
    // this is to handle the tbname
703
    if (fmIsScanPseudoColumnFunc(functionId)) {
5,528,471✔
704
      int32_t fType = pExpr1->pExpr->_function.functionType;
920,683✔
705
      if (fType == FUNCTION_TYPE_TBNAME) {
920,683✔
706
        code = setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
912,191✔
707
        QUERY_CHECK_CODE(code, lino, _end);
912,527!
708
      } else if (fType == FUNCTION_TYPE_VGID) {
8,492✔
709
        code = setVgIdColData(pBlock, pColInfoData, functionId, pTask->id.vgId);
4,247✔
710
        QUERY_CHECK_CODE(code, lino, _end);
4,246!
711
      } else if (fType == FUNCTION_TYPE_VGVER) {
4,245!
712
        code = setVgVerColData(pBlock, pColInfoData, functionId, pBlock->info.version);
4,245✔
713
        QUERY_CHECK_CODE(code, lino, _end);
4,247!
714
      }
715
    } else {  // these are tags
716
      STagVal tagVal = {0};
4,607,753✔
717
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
4,607,753✔
718
      const char* p = pHandle->api.metaFn.extractTagVal(val.pTags, pColInfoData->info.type, &tagVal);
4,607,753✔
719

720
      char* data = NULL;
4,606,612✔
721
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
4,606,612✔
722
        data = tTagValToData((const STagVal*)p, false);
4,560,675✔
723
      } else {
724
        data = (char*)p;
45,937✔
725
      }
726

727
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
4,607,473✔
728
      if (isNullVal) {
4,607,473✔
729
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
43,534✔
730
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
4,563,939✔
731
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, false);
4,560,493✔
732
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
4,564,333!
733
          char* tmp = taosMemoryCalloc(1, varDataLen(data) + 1);
1,597,766!
734
          if (tmp != NULL){
1,600,217!
735
            memcpy(tmp, varDataVal(data), varDataLen(data));
1,600,223✔
736
            qDebug("get tag value:%s, cid:%d, table name:%s, uid%"PRId64, tmp, tagVal.cid, val.pName, pBlock->info.id.uid);
1,600,223✔
737
            taosMemoryFree(tmp);
1,600,226✔
738
          }
739
          taosMemoryFree(data);
1,600,113!
740
        }
741
        QUERY_CHECK_CODE(code, lino, _end);
4,564,142!
742
      } else {  // todo opt for json tag
743
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
6,330✔
744
          code = colDataSetVal(pColInfoData, i, data, false);
2,884✔
745
          QUERY_CHECK_CODE(code, lino, _end);
2,884!
746
        }
747
      }
748
    }
749
  }
750

751
  // restore the rows
752
  pBlock->info.rows = backupRows;
1,906,571✔
753

754
_end:
1,906,571✔
755

756
  if (NULL != pVal) {
1,906,571✔
757
    insertRet = taosLRUCacheInsert(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(uint64_t), pVal,
1,039,104✔
758
                                   sizeof(STableCachedVal), freeCachedMetaItem, NULL, NULL, TAOS_LRU_PRIORITY_LOW, NULL);
759
    if (insertRet != TAOS_LRU_STATUS_OK) {
1,039,045!
760
      qWarn("failed to put meta into lru cache, code:%d, %s", insertRet, idStr);
×
761
    }
762
  }
763

764
  if (freeReader) {
1,906,512✔
765
    pHandle->api.metaReaderFn.clearReader(&mr);
1,040,038✔
766
  }
767
  if (code != TSDB_CODE_SUCCESS) {
1,906,571!
768
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
769
  }
770
  return code;
1,904,921✔
771
}
772

773
int32_t setTbNameColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId,
912,008✔
774
                         const char* name) {
775
  int32_t                     code = TSDB_CODE_SUCCESS;
912,008✔
776
  int32_t                     lino = 0;
912,008✔
777
  struct SScalarFuncExecFuncs fpSet = {0};
912,008✔
778
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
912,008✔
779
  QUERY_CHECK_CODE(code, lino, _end);
912,008!
780

781
  size_t len = TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE;
912,008✔
782
  char   buf[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
912,008✔
783
  STR_TO_VARSTR(buf, name)
912,008✔
784

785
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
912,008✔
786

787
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
912,913✔
788
  QUERY_CHECK_CODE(code, lino, _end);
912,747!
789

790
  code = colDataSetVal(&infoData, 0, buf, false);
912,747✔
791
  QUERY_CHECK_CODE(code, lino, _end);
912,654!
792

793
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
912,654✔
794
  SScalarParam param = {.columnData = pColInfoData};
912,654✔
795

796
  if (fpSet.process != NULL) {
912,654✔
797
    code = fpSet.process(&srcParam, 1, &param);
912,610✔
798
    QUERY_CHECK_CODE(code, lino, _end);
913,041!
799
  } else {
800
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
44!
801
  }
802

803
  colDataDestroy(&infoData);
913,085✔
804

805
_end:
912,666✔
806
  if (code != TSDB_CODE_SUCCESS) {
912,666!
807
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
808
  }
809
  return code;
912,561✔
810
}
811

812
int32_t setVgIdColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId, int32_t vgId) {
4,247✔
813
  int32_t                     code = TSDB_CODE_SUCCESS;
4,247✔
814
  int32_t                     lino = 0;
4,247✔
815
  struct SScalarFuncExecFuncs fpSet = {0};
4,247✔
816
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
4,247✔
817
  QUERY_CHECK_CODE(code, lino, _end);
4,247!
818

819
  SColumnInfoData infoData = createColumnInfoData(pColInfoData->info.type, pColInfoData->info.bytes, 1);
4,247✔
820

821
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
4,247✔
822
  QUERY_CHECK_CODE(code, lino, _end);
4,247!
823

824
  code = colDataSetVal(&infoData, 0, (const char*)&vgId, false);
4,247✔
825
  QUERY_CHECK_CODE(code, lino, _end);
4,247!
826

827
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
4,247✔
828
  SScalarParam param = {.columnData = pColInfoData};
4,247✔
829

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

837
_end:
×
838
  colDataDestroy(&infoData);
4,247✔
839
  if (code != TSDB_CODE_SUCCESS) {
4,246!
840
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
841
  }
842
  return code;
4,246✔
843
}
844

845
int32_t setVgVerColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId, int64_t vgVer) {
4,245✔
846
  int32_t                     code = TSDB_CODE_SUCCESS;
4,245✔
847
  int32_t                     lino = 0;
4,245✔
848
  struct SScalarFuncExecFuncs fpSet = {0};
4,245✔
849
  code = fmGetScalarFuncExecFuncs(functionId, &fpSet);
4,245✔
850
  QUERY_CHECK_CODE(code, lino, _end);
4,245!
851

852
  SColumnInfoData infoData = createColumnInfoData(pColInfoData->info.type, pColInfoData->info.bytes, 1);
4,245✔
853

854
  code = colInfoDataEnsureCapacity(&infoData, 1, false);
4,247✔
855
  QUERY_CHECK_CODE(code, lino, _end);
4,246!
856

857
  code = colDataSetVal(&infoData, 0, (const char*)&vgVer, false);
4,246✔
858
  QUERY_CHECK_CODE(code, lino, _end);
4,246!
859

860
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
4,246✔
861
  SScalarParam param = {.columnData = pColInfoData};
4,246✔
862

863
  if (fpSet.process != NULL) {
4,246!
864
    code = fpSet.process(&srcParam, 1, &param);
4,247✔
865
    QUERY_CHECK_CODE(code, lino, _end);
4,247!
866
  } else {
UNCOV
867
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
×
868
  }
869

UNCOV
870
_end:
×
871
  colDataDestroy(&infoData);
4,246✔
872
  if (code != TSDB_CODE_SUCCESS) {
4,247!
873
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
874
  }
875
  return code;
4,247✔
876
}
877

878
static int32_t initNextGroupScan(STableScanInfo* pInfo, STableKeyInfo** pKeyInfo, int32_t* size) {
1,129,546✔
879
  int32_t code = TSDB_CODE_SUCCESS;
1,129,546✔
880
  int32_t lino = 0;
1,129,546✔
881
  code = tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, pKeyInfo, size);
1,129,546✔
882
  QUERY_CHECK_CODE(code, lino, _end);
1,131,185!
883

884
  pInfo->tableStartIndex = TARRAY_ELEM_IDX(pInfo->base.pTableListInfo->pTableList, *pKeyInfo);
1,131,185✔
885
  pInfo->tableEndIndex = (pInfo->tableStartIndex + (*size) - 1);
1,131,185✔
886
  pInfo->pResBlock->info.blankFill = false;
1,131,185✔
887
  taosMemoryFreeClear(pInfo->pResBlock->pBlockAgg);
1,131,185!
888

889
  if (!pInfo->needCountEmptyTable) {
1,131,185✔
890
    pInfo->countState = TABLE_COUNT_STATE_END;
1,000,836✔
891
  } else {
892
    pInfo->countState = TABLE_COUNT_STATE_SCAN;
130,349✔
893
  }
894

895
_end:
1,131,185✔
896
  if (code != TSDB_CODE_SUCCESS) {
1,131,185!
897
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
898
  }
899
  return code;
1,130,438✔
900
}
901

902
void markGroupProcessed(STableScanInfo* pInfo, uint64_t groupId) {
3,766,613✔
903
  if (pInfo->countState == TABLE_COUNT_STATE_END) {
3,766,613✔
904
    return;
3,065,433✔
905
  }
906
  if (pInfo->base.pTableListInfo->groupOffset) {
701,180✔
907
    pInfo->countState = TABLE_COUNT_STATE_PROCESSED;
406✔
908
  } else {
909
    int32_t code = taosHashRemove(pInfo->base.pTableListInfo->remainGroups, &groupId, sizeof(groupId));
700,774✔
910
    if (code != TSDB_CODE_SUCCESS) {
702,776✔
911
      qDebug("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
195,807✔
912
    }
913
  }
914
}
915

916
static SSDataBlock* getOneRowResultBlock(SExecTaskInfo* pTaskInfo, STableScanBase* pBase, SSDataBlock* pBlock,
368,323✔
917
                                         const STableKeyInfo* tbInfo) {
918
  blockDataEmpty(pBlock);
368,323✔
919
  pBlock->info.rows = 1;
368,348✔
920
  pBlock->info.id.uid = tbInfo->uid;
368,348✔
921
  pBlock->info.id.groupId = tbInfo->groupId;
368,348✔
922
  pBlock->info.blankFill = true;
368,348✔
923

924
  // only one row: set all col data to null & hasNull
925
  int32_t col_num = blockDataGetNumOfCols(pBlock);
368,348✔
926
  for (int32_t i = 0; i < col_num; ++i) {
1,489,949✔
927
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
1,122,001✔
928
    colDataSetNULL(pColInfoData, 0);
929
  }
930

931
  // set tag/tbname
932
  terrno = doSetTagColumnData(pBase, pBlock, pTaskInfo, 1);
367,948✔
933

934
  return pBlock;
368,204✔
935
}
936

937
static SSDataBlock* getBlockForEmptyTable(SOperatorInfo* pOperator, const STableKeyInfo* tbInfo) {
353,829✔
938
  STableScanInfo* pTableScanInfo = pOperator->info;
353,829✔
939
  SSDataBlock*    pBlock =
940
      getOneRowResultBlock(pOperator->pTaskInfo, &pTableScanInfo->base, pTableScanInfo->pResBlock, tbInfo);
353,829✔
941

942
  pOperator->resultInfo.totalRows++;
353,724✔
943
  return pBlock;
353,724✔
944
}
945

946
static int32_t doTableScanImplNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
4,516,361✔
947
  int32_t         code = TSDB_CODE_SUCCESS;
4,516,361✔
948
  int32_t         lino = 0;
4,516,361✔
949
  STableScanInfo* pTableScanInfo = pOperator->info;
4,516,361✔
950
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
4,516,361✔
951
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
4,516,361✔
952
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
4,516,361✔
953
  bool            hasNext = false;
4,516,361✔
954
  int64_t         st = taosGetTimestampUs();
4,518,019✔
955

956
  QRY_PARAM_CHECK(ppRes);
4,518,019!
957
  pBlock->info.dataLoad = false;
4,518,019✔
958

959
  while (true) {
96,494✔
960
    code = pAPI->tsdReader.tsdNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
4,614,513✔
961
    if (code != TSDB_CODE_SUCCESS) {
4,614,225!
962
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
×
963
      QUERY_CHECK_CODE(code, lino, _end);
1,197!
964
    }
965

966
    if (!hasNext) {
4,615,789✔
967
      break;
1,101,037✔
968
    }
969

970
    if (isTaskKilled(pTaskInfo)) {
3,518,966✔
971
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
440✔
972
      code = pTaskInfo->code;
1,192✔
973
      goto _end;
1,192✔
974
    }
975

976
    if (pOperator->status == OP_EXEC_DONE) {
3,518,999✔
977
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pTableScanInfo->base.dataReader);
1,120✔
978
      break;
1,120✔
979
    }
980

981
    // process this data block based on the probabilities
982
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
3,517,879✔
983
    if (!processThisBlock) {
3,516,556!
984
      continue;
96,494✔
985
    }
986

987
    if (pBlock->info.id.uid) {
3,516,556!
988
      pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
3,517,777✔
989
    }
990

991
    uint32_t status = 0;
3,517,748✔
992
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
3,517,748✔
993
    QUERY_CHECK_CODE(code, lino, _end);
3,515,958✔
994

995
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
3,515,957✔
996
      break;
3,094✔
997
    }
998

999
    // current block is filter out according to filter condition, continue load the next block
1000
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
3,512,863✔
1001
      continue;
96,494✔
1002
    }
1003

1004
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
3,416,369✔
1005
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
3,416,047✔
1006

1007
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
3,416,047✔
1008
    pBlock->info.scanFlag = pTableScanInfo->base.scanFlag;
3,416,047✔
1009

1010
    (*ppRes) = pBlock;
3,416,047✔
1011
    return code;
3,416,047✔
1012
  }
1013

1014
_end:
1,102,234✔
1015
  if (code != TSDB_CODE_SUCCESS) {
1,102,234✔
1016
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
1,197!
1017
    pTaskInfo->code = code;
1,197✔
1018
    T_LONG_JMP(pTaskInfo->env, code);
1,197!
1019
  }
1020
  return code;
1,101,037✔
1021
}
1022

1023
static int32_t doGroupedTableScan(SOperatorInfo* pOperator, SSDataBlock** pBlock) {
4,884,929✔
1024
  int32_t         code = TSDB_CODE_SUCCESS;
4,884,929✔
1025
  int32_t         lino = 0;
4,884,929✔
1026
  STableScanInfo* pTableScanInfo = pOperator->info;
4,884,929✔
1027
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
4,884,929✔
1028
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
4,884,929✔
1029

1030
  QRY_PARAM_CHECK(pBlock);
4,884,929!
1031

1032
  // The read handle is not initialized yet, since no qualified tables exists
1033
  if (pTableScanInfo->base.dataReader == NULL || pOperator->status == OP_EXEC_DONE) {
4,884,929!
1034
    return code;
14,559✔
1035
  }
1036

1037
  // do the ascending order traverse in the first place.
1038
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
5,898,053✔
1039
    SSDataBlock* p = NULL;
4,209,646✔
1040
    code = doTableScanImplNext(pOperator, &p);
4,209,646✔
1041
    QUERY_CHECK_CODE(code, lino, _end);
4,205,656!
1042

1043
    if (p != NULL) {
4,205,656✔
1044
      markGroupProcessed(pTableScanInfo, p->info.id.groupId);
3,179,329✔
1045
      *pBlock = p;
3,179,667✔
1046
      return code;
3,179,667✔
1047
    }
1048

1049
    pTableScanInfo->scanTimes += 1;
1,026,327✔
1050
    taosHashClear(pTableScanInfo->pIgnoreTables);
1,026,327✔
1051

1052
    if (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
1,027,696✔
1053
      setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
916✔
1054
      pTableScanInfo->base.scanFlag = MAIN_SCAN;
903✔
1055
      pTableScanInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
903✔
1056
      qDebug("start to repeat ascending order scan data blocks due to query func required, %s", GET_TASKID(pTaskInfo));
903✔
1057

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

1064
  int32_t total = pTableScanInfo->scanInfo.numOfAsc + pTableScanInfo->scanInfo.numOfDesc;
1,688,407✔
1065
  if (pTableScanInfo->scanTimes < total) {
1,688,407✔
1066
    if (pTableScanInfo->base.cond.order == TSDB_ORDER_ASC) {
308,357✔
1067
      prepareForDescendingScan(&pTableScanInfo->base, pOperator->exprSupp.pCtx, 0);
191✔
1068
      code = pAPI->tsdReader.tsdReaderResetStatus(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
191✔
1069
      QUERY_CHECK_CODE(code, lino, _end);
191!
1070
      qDebug("%s start to descending order scan data blocks due to query func required", GET_TASKID(pTaskInfo));
191✔
1071
    }
1072

1073
    while (pTableScanInfo->scanTimes < total) {
381,462✔
1074
      SSDataBlock* p = NULL;
308,127✔
1075
      code = doTableScanImplNext(pOperator, &p);
308,127✔
1076
      QUERY_CHECK_CODE(code, lino, _end);
308,007!
1077

1078
      if (p != NULL) {
308,007✔
1079
        markGroupProcessed(pTableScanInfo, p->info.id.groupId);
234,908✔
1080
        *pBlock = p;
235,217✔
1081
        return code;
235,217✔
1082
      }
1083

1084
      pTableScanInfo->scanTimes += 1;
73,099✔
1085
      taosHashClear(pTableScanInfo->pIgnoreTables);
73,099✔
1086

1087
      if (pTableScanInfo->scanTimes < total) {
73,078✔
1088
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
7✔
1089
        pTableScanInfo->base.scanFlag = MAIN_SCAN;
×
1090

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

1098
  if (pTableScanInfo->countState < TABLE_COUNT_STATE_END) {
1,453,385✔
1099
    STableListInfo* pTableListInfo = pTableScanInfo->base.pTableListInfo;
479,973✔
1100
    if (pTableListInfo->groupOffset) {  // group by tbname, group by tag + sort
479,973✔
1101
      if (pTableScanInfo->countState < TABLE_COUNT_STATE_PROCESSED) {
198!
1102
        pTableScanInfo->countState = TABLE_COUNT_STATE_PROCESSED;
×
1103
        STableKeyInfo* pStart =
1104
            (STableKeyInfo*)tableListGetInfo(pTableScanInfo->base.pTableListInfo, pTableScanInfo->tableStartIndex);
×
1105

1106
        if (NULL == pStart) {
×
1107
          return code;
×
1108
        }
1109

1110
        *pBlock = getBlockForEmptyTable(pOperator, pStart);
×
1111
        return code;
×
1112
      }
1113
    } else {  // group by tag + no sort
1114
      int32_t numOfTables = 0;
479,775✔
1115
      code = tableListGetSize(pTableListInfo, &numOfTables);
479,775✔
1116
      QUERY_CHECK_CODE(code, lino, _end);
479,838!
1117

1118
      if (pTableScanInfo->tableEndIndex + 1 >= numOfTables) {
479,838✔
1119
        // get empty group, mark processed & rm from hash
1120
        void* pIte = taosHashIterate(pTableListInfo->remainGroups, NULL);
475,910✔
1121
        if (pIte != NULL) {
476,413✔
1122
          size_t        keySize = 0;
353,952✔
1123
          uint64_t*     pGroupId = taosHashGetKey(pIte, &keySize);
353,952✔
1124
          STableKeyInfo info = {.uid = *(uint64_t*)pIte, .groupId = *pGroupId};
353,973✔
1125
          taosHashCancelIterate(pTableListInfo->remainGroups, pIte);
353,973✔
1126
          markGroupProcessed(pTableScanInfo, *pGroupId);
353,975✔
1127
          *pBlock = getBlockForEmptyTable(pOperator, &info);
353,871✔
1128

1129
          return code;
353,709✔
1130
        }
1131
      }
1132
    }
1133
    pTableScanInfo->countState = TABLE_COUNT_STATE_END;
126,587✔
1134
  }
1135

1136
_end:
973,412✔
1137
  if (code != TSDB_CODE_SUCCESS) {
1,099,999!
1138
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1139
    pTaskInfo->code = code;
×
1140
  }
1141

1142
  return code;
1,099,999✔
1143
}
1144

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

1157
  qDebug("vgId:%d add total %d dynamic tables to scan, tableSeq:%d, exist num:%" PRId64 ", operator status:%d",
1,101✔
1158
         pTaskInfo->id.vgId, num, pParam->tableSeq, (int64_t)taosArrayGetSize(pListInfo->pTableList),
1159
         pOperator->status);
1160

1161
  if (pParam->tableSeq) {
1,101✔
1162
    pListInfo->oneTableForEachGroup = true;
48✔
1163
    if (taosArrayGetSize(pListInfo->pTableList) > 0) {
48!
1164
      taosHashClear(pListInfo->map);
×
1165
      taosArrayClear(pListInfo->pTableList);
×
UNCOV
1166
      pOperator->status = OP_EXEC_DONE;
×
1167
    }
1168
  } else {
1169
    pListInfo->oneTableForEachGroup = false;
1,053✔
1170
    pListInfo->numOfOuputGroups = 1;
1,053✔
1171
  }
1172

1173
  STableKeyInfo info = {.groupId = 0};
1,101✔
1174
  int32_t       tableIdx = 0;
1,101✔
1175
  for (int32_t i = 0; i < num; ++i) {
69,907✔
1176
    uint64_t* pUid = taosArrayGet(pParam->pUidList, i);
68,806✔
1177
    if (!pUid) {
68,789!
1178
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1179
      return terrno;
×
1180
    }
1181

1182
    if (taosHashPut(pListInfo->map, pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t))) {
68,789✔
1183
      if (TSDB_CODE_DUP_KEY == terrno) {
65,248✔
1184
        continue;
65,142✔
1185
      }
1186
      return terrno;
13✔
1187
    }
1188

1189
    info.uid = *pUid;
3,790✔
1190
    void* p = taosArrayPush(pListInfo->pTableList, &info);
3,790✔
1191
    if (p == NULL) {
3,824!
1192
      return terrno;
×
1193
    }
1194

1195
    tableIdx++;
3,824✔
1196
    qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
3,824✔
1197
  }
1198

1199
  return code;
1,101✔
1200
}
1201

1202
static int32_t doInitReader(STableScanInfo* pInfo, SExecTaskInfo* pTaskInfo, SStorageAPI* pAPI, int32_t* pNum,
1,129,378✔
1203
                            STableKeyInfo** pList) {
1204
  const char* idStr = GET_TASKID(pTaskInfo);
1,129,378✔
1205
  int32_t     code = initNextGroupScan(pInfo, pList, pNum);
1,129,378✔
1206
  if (code) {
1,129,641!
1207
    qError("%s failed to init groupScan Info, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
×
1208
    return code;
×
1209
  }
1210

1211
  if (pInfo->base.dataReader != NULL) {
1,129,934!
1212
    qError("%s tsdb reader should be null", idStr);
×
1213
    return TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
1214
  }
1215

1216
  code = pAPI->tsdReader.tsdReaderOpen(pInfo->base.readHandle.vnode, &pInfo->base.cond, *pList, *pNum, pInfo->pResBlock,
1,129,934✔
1217
                                       (void**)&pInfo->base.dataReader, idStr, &pInfo->pIgnoreTables);
1,129,934✔
1218
  if (code) {
1,130,714✔
1219
    qError("%s failed to open tsdbReader, code:%s at line:%d", idStr, tstrerror(code), __LINE__);
16!
1220
  }
1221

1222
  return code;
1,130,542✔
1223
}
1224

1225
int compareColIdPair(const void* elem1, const void* elem2) {
×
1226
  SColIdPair* node1 = (SColIdPair*)elem1;
×
1227
  SColIdPair* node2 = (SColIdPair*)elem2;
×
1228

1229
  if (node1->orgColId < node2->orgColId) {
×
1230
    return -1;
×
1231
  }
1232

1233
  return node1->orgColId > node2->orgColId;
×
1234
}
1235

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

1252
  cleanupQueryTableDataCond(&pInfo->base.cond);
×
1253
  if (pAPI->tsdReader.tsdReaderClose) {
×
1254
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
×
1255
  }
1256

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

1276
  pColArray = taosArrayInit(schema->nCols, sizeof(SColIdPair));
×
1277
  QUERY_CHECK_NULL(pColArray, code, lino, _return, terrno);
×
1278
  pBlockColArray = taosArrayInit(schema->nCols, sizeof(int32_t));
×
1279
  QUERY_CHECK_NULL(pBlockColArray, code, lino, _return, terrno);
×
1280

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

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

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

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

1324
  pListInfo->oneTableForEachGroup = true;
×
1325
  taosHashClear(pListInfo->map);
×
1326
  taosArrayClear(pListInfo->pTableList);
×
1327

1328
  uint64_t      pUid = orgTable.me.uid;
×
1329
  STableKeyInfo info = {.groupId = 0, .uid = pUid};
×
1330
  int32_t       tableIdx = 0;
×
1331
  code = taosHashPut(pListInfo->map, &pUid, sizeof(uint64_t), &tableIdx, sizeof(int32_t));
×
1332
  QUERY_CHECK_CODE(code, lino, _return);
×
1333
  QUERY_CHECK_NULL(taosArrayPush(pListInfo->pTableList, &info), code, lino, _return, terrno);
×
1334
  qDebug("add dynamic table scan uid:%" PRIu64 ", %s", info.uid, GET_TASKID(pTaskInfo));
×
1335

1336
  pOperator->status = OP_OPENED;
×
1337

1338
  taosRLockLatch(&pTaskInfo->lock);
×
1339
  code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
×
1340
  taosRUnLockLatch(&pTaskInfo->lock);
×
1341
  QUERY_CHECK_CODE(code, lino, _return);
×
1342

1343
  if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
×
1344
    pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1345
  }
1346

1347
  pInfo->currentGroupId = -1;
×
1348
_return:
×
1349
  if (code) {
×
1350
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1351
  }
1352
  taosArrayDestroy(pColArray);
×
1353
  taosArrayDestroy(pBlockColArray);
×
1354
  pAPI->metaReaderFn.clearReader(&superTable);
×
1355
  pAPI->metaReaderFn.clearReader(&orgTable);
×
1356
  return code;
×
1357
}
1358

1359
static int32_t startNextGroupScan(SOperatorInfo* pOperator, SSDataBlock** pResult) {
1,098,581✔
1360
  int32_t         code = TSDB_CODE_SUCCESS;
1,098,581✔
1361
  int32_t         lino = 0;
1,098,581✔
1362
  STableScanInfo* pInfo = pOperator->info;
1,098,581✔
1363
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
1,098,581✔
1364
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
1,098,581✔
1365
  int32_t         numOfTables = 0;
1,098,581✔
1366

1367
  QRY_PARAM_CHECK(pResult);
1,098,581!
1368

1369
  code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
1,098,581✔
1370
  QUERY_CHECK_CODE(code, lino, _end);
1,099,000!
1371

1372
  if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
1,099,000✔
1373
    setOperatorCompleted(pOperator);
1,097,934✔
1374
    if (pOperator->dynamicTask) {
1,098,020✔
1375
      taosArrayClear(pInfo->base.pTableListInfo->pTableList);
1,091✔
1376
      taosHashClear(pInfo->base.pTableListInfo->map);
1,091✔
1377
    }
1378
    return code;
1,097,987✔
1379
  }
1380

1381
  // reset value for the next group data output
1382
  pOperator->status = OP_OPENED;
836✔
1383
  resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
836✔
1384

1385
  int32_t        num = 0;
936✔
1386
  STableKeyInfo* pList = NULL;
936✔
1387
  code = initNextGroupScan(pInfo, &pList, &num);
936✔
1388
  QUERY_CHECK_CODE(code, lino, _end);
938!
1389

1390
  code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, pList, num);
938✔
1391
  QUERY_CHECK_CODE(code, lino, _end);
939!
1392

1393
  code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
939✔
1394
  QUERY_CHECK_CODE(code, lino, _end);
939!
1395
  pInfo->scanTimes = 0;
939✔
1396

1397
  code = doGroupedTableScan(pOperator, pResult);
939✔
1398
  QUERY_CHECK_CODE(code, lino, _end);
936!
1399

1400
  if (*pResult != NULL) {
936✔
1401
    if (pOperator->dynamicTask) {
830✔
1402
      (*pResult)->info.id.groupId = (*pResult)->info.id.uid;
24✔
1403
    }
1404
    return code;
830✔
1405
  }
1406

1407
_end:
106✔
1408
  if (code != TSDB_CODE_SUCCESS) {
106!
1409
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1410
    pTaskInfo->code = code;
×
1411
  }
1412
  return code;
106✔
1413
}
1414

1415
static int32_t groupSeqTableScan(SOperatorInfo* pOperator, SSDataBlock** pResBlock) {
4,842,241✔
1416
  int32_t         code = TSDB_CODE_SUCCESS;
4,842,241✔
1417
  int32_t         lino = 0;
4,842,241✔
1418
  STableScanInfo* pInfo = pOperator->info;
4,842,241✔
1419
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
4,842,241✔
1420
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
4,842,241✔
1421
  int32_t         num = 0;
4,842,241✔
1422
  STableKeyInfo*  pList = NULL;
4,842,241✔
1423
  SSDataBlock*    pResult = NULL;
4,842,241✔
1424
  const char*     idStr = GET_TASKID(pTaskInfo);
4,842,241✔
1425

1426
  QRY_PARAM_CHECK(pResBlock);
4,842,241!
1427

1428
  if (pInfo->currentGroupId == -1) {
4,842,241✔
1429
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
1,129,540!
1430
      setOperatorCompleted(pOperator);
×
1431
      return code;
×
1432
    }
1433

1434
    taosRLockLatch(&pTaskInfo->lock);
1,129,679✔
1435
    code = doInitReader(pInfo, pTaskInfo, pAPI, &num, &pList);
1,131,035✔
1436
    taosRUnLockLatch(&pTaskInfo->lock);
1,130,541✔
1437
    QUERY_CHECK_CODE(code, lino, _end);
1,131,208✔
1438

1439
    if (pInfo->filesetDelimited) {
1,131,192✔
1440
      pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
1,074✔
1441
    }
1442

1443
    if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
1,131,193!
1444
      pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
×
1445
    }
1446
  }
1447

1448
  code = doGroupedTableScan(pOperator, &pResult);
4,843,894✔
1449
  QUERY_CHECK_CODE(code, lino, _end);
4,842,979!
1450

1451
  if (pResult != NULL) {
4,842,979✔
1452
    if (pOperator->dynamicTask) {
3,746,728✔
1453
      pResult->info.id.groupId = pResult->info.id.uid;
3,419✔
1454
    }
1455

1456
    *pResBlock = pResult;
3,746,728✔
1457
    return code;
3,746,728✔
1458
  }
1459

1460
  while (true) {
1461
    code = startNextGroupScan(pOperator, &pResult);
1,096,358✔
1462
    QUERY_CHECK_CODE(code, lino, _end);
1,098,989!
1463

1464
    if (pResult || pOperator->status == OP_EXEC_DONE) {
1,098,989✔
1465
      *pResBlock = pResult;
1,098,882✔
1466
      return code;
1,098,882✔
1467
    }
1468
  }
1469

1470
_end:
16✔
1471
  if (code != TSDB_CODE_SUCCESS) {
16!
1472
    qError("%s %s failed at line %d since %s", idStr, __func__, lino, tstrerror(code));
16!
1473
    pTaskInfo->code = code;
16✔
1474
  }
1475

1476
  return code;
16✔
1477
}
1478

1479
int32_t doTableScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
4,863,846✔
1480
  int32_t         code = TSDB_CODE_SUCCESS;
4,863,846✔
1481
  int32_t         lino = 0;
4,863,846✔
1482
  STableScanInfo* pInfo = pOperator->info;
4,863,846✔
1483
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
4,863,846✔
1484
  SStorageAPI*    pAPI = &pTaskInfo->storageAPI;
4,863,846✔
1485
  QRY_PARAM_CHECK(ppRes);
4,863,846!
1486
  qTrace("%s call", __FUNCTION__);
4,863,846✔
1487
  if (pOperator->pOperatorGetParam) {
4,863,002✔
1488
    pOperator->dynamicTask = true;
1,101✔
1489
    if (isDynVtbScan(pOperator)) {
1,101!
1490
      code = createVTableScanInfoFromParam(pOperator);
×
1491

1492
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
×
1493
      pOperator->pOperatorGetParam = NULL;
×
1494
      QUERY_CHECK_CODE(code, lino, _end);
×
1495

1496
      SSDataBlock* result = NULL;
×
1497
      while (true) {
1498
        code = startNextGroupScan(pOperator, &result);
×
1499
        QUERY_CHECK_CODE(code, lino, _end);
×
1500

1501
        if (result || pOperator->status == OP_EXEC_DONE) {
×
1502
          SSDataBlock* res = NULL;
×
1503
          if (result) {
×
1504
            code = createOneDataBlockWithTwoBlock(result, pInfo->pOrgBlock, &res);
×
1505
            QUERY_CHECK_CODE(code, lino, _end);
×
1506
            pInfo->pResBlock = res;
×
1507
            blockDataDestroy(result);
×
1508
          }
1509
          (*ppRes) = res;
×
1510
          return code;
×
1511
        }
1512
      }
1513
    } else {
1514
      code = createTableListInfoFromParam(pOperator);
1,101✔
1515
      freeOperatorParam(pOperator->pOperatorGetParam, OP_GET_PARAM);
1,101✔
1516
      pOperator->pOperatorGetParam = NULL;
1,101✔
1517
      QUERY_CHECK_CODE(code, lino, _end);
1,101!
1518

1519
      if (pOperator->status == OP_EXEC_DONE) {
1,101✔
1520
        pInfo->currentGroupId = -1;
24✔
1521
        pOperator->status = OP_OPENED;
24✔
1522
        SSDataBlock* result = NULL;
24✔
1523

1524
        while (true) {
1525
          code = startNextGroupScan(pOperator, &result);
24✔
1526
          QUERY_CHECK_CODE(code, lino, _end);
24!
1527

1528
          if (result || pOperator->status == OP_EXEC_DONE) {
24!
1529
            (*ppRes) = result;
24✔
1530
            return code;
24✔
1531
          }
1532
        }
1533
      }
1534
    }
1535
  }
1536

1537
  // scan table one by one sequentially
1538
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
4,862,978✔
1539
    int32_t       numOfTables = 0;
20,534✔
1540
    STableKeyInfo tInfo = {0};
20,534✔
1541
    pInfo->countState = TABLE_COUNT_STATE_END;
20,534✔
1542

1543
    while (1) {
19,023✔
1544
      SSDataBlock* result = NULL;
39,557✔
1545
      code = doGroupedTableScan(pOperator, &result);
39,557✔
1546
      QUERY_CHECK_CODE(code, lino, _end);
39,542!
1547

1548
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
39,542✔
1549
        (*ppRes) = result;
20,309✔
1550
        return code;
20,510✔
1551
      }
1552

1553
      // if no data, switch to next table and continue scan
1554
      pInfo->currentTable++;
19,232✔
1555

1556
      taosRLockLatch(&pTaskInfo->lock);
19,232✔
1557
      numOfTables = 0;
19,242✔
1558
      code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
19,242✔
1559
      if (code != TSDB_CODE_SUCCESS) {
19,242!
1560
        taosRUnLockLatch(&pTaskInfo->lock);
×
1561
        TSDB_CHECK_CODE(code, lino, _end);
×
1562
      }
1563

1564
      if (pInfo->currentTable >= numOfTables) {
19,242✔
1565
        qDebug("all table checked in table list, total:%d, return NULL, %s", numOfTables, GET_TASKID(pTaskInfo));
201✔
1566
        taosRUnLockLatch(&pTaskInfo->lock);
201✔
1567
        (*ppRes) = NULL;
201✔
1568
        return code;
201✔
1569
      }
1570

1571
      STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
19,041✔
1572
      if (!tmp) {
19,038!
1573
        taosRUnLockLatch(&pTaskInfo->lock);
×
1574
        (*ppRes) = NULL;
×
1575
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
×
1576
      }
1577

1578
      tInfo = *tmp;
19,038✔
1579
      taosRUnLockLatch(&pTaskInfo->lock);
19,038✔
1580

1581
      code = pAPI->tsdReader.tsdSetQueryTableList(pInfo->base.dataReader, &tInfo, 1);
19,042✔
1582
      QUERY_CHECK_CODE(code, lino, _end);
19,042!
1583
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d/%d %s", tInfo.uid, numOfTables,
19,042✔
1584
             pInfo->currentTable, numOfTables, GET_TASKID(pTaskInfo));
1585

1586
      code = pAPI->tsdReader.tsdReaderResetStatus(pInfo->base.dataReader, &pInfo->base.cond);
19,042✔
1587
      QUERY_CHECK_CODE(code, lino, _end);
19,023!
1588
      pInfo->scanTimes = 0;
19,023✔
1589
    }
1590
  } else {  // scan table group by group sequentially
1591
    code = groupSeqTableScan(pOperator, ppRes);
4,842,444✔
1592
    QUERY_CHECK_CODE(code, lino, _end);
4,843,263✔
1593
  }
1594

1595
_end:
4,843,247✔
1596
  if (code != TSDB_CODE_SUCCESS) {
4,843,263✔
1597
    qError("%s %s failed at line %d since %s", GET_TASKID(pTaskInfo), __func__, lino, tstrerror(code));
16!
1598
    pTaskInfo->code = code;
16✔
1599
    T_LONG_JMP(pTaskInfo->env, code);
16!
1600
  }
1601

1602
  return code;
4,843,247✔
1603
}
1604

1605
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
17,874✔
1606
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
17,874!
1607
  if (!pRecorder) {
17,876!
1608
    return terrno;
×
1609
  }
1610
  STableScanInfo* pTableScanInfo = pOptr->info;
17,876✔
1611
  *pRecorder = pTableScanInfo->base.readRecorder;
17,876✔
1612
  *pOptrExplain = pRecorder;
17,876✔
1613
  *len = sizeof(SFileBlockLoadRecorder);
17,876✔
1614
  return 0;
17,876✔
1615
}
1616

1617
static void destroyTableScanBase(STableScanBase* pBase, TsdReader* pAPI) {
1,345,850✔
1618
  cleanupQueryTableDataCond(&pBase->cond);
1,345,850✔
1619
  cleanupQueryTableDataCond(&pBase->orgCond);
1,346,294✔
1620

1621
  if (pAPI->tsdReaderClose) {
1,346,310✔
1622
    pAPI->tsdReaderClose(pBase->dataReader);
1,346,302✔
1623
  }
1624
  pBase->dataReader = NULL;
1,346,325✔
1625

1626
  if (pBase->matchInfo.pList != NULL) {
1,346,325!
1627
    taosArrayDestroy(pBase->matchInfo.pList);
1,346,350✔
1628
  }
1629

1630
  tableListDestroy(pBase->pTableListInfo);
1,346,384✔
1631
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
1,346,301✔
1632
  cleanupExprSupp(&pBase->pseudoSup);
1,346,342✔
1633
}
1,346,406✔
1634

1635
static void destroyTableScanOperatorInfo(void* param) {
1,171,487✔
1636
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
1,171,487✔
1637
  blockDataDestroy(pTableScanInfo->pResBlock);
1,171,487✔
1638
  blockDataDestroy(pTableScanInfo->pOrgBlock);
1,172,007✔
1639
  taosHashCleanup(pTableScanInfo->pIgnoreTables);
1,172,000✔
1640
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
1,171,721✔
1641
  taosMemoryFreeClear(param);
1,171,922!
1642
}
1,171,991✔
1643

1644
static void resetClolumnReserve(SSDataBlock* pBlock, int32_t dataRequireFlag) {
1,169,862✔
1645
  if (pBlock && dataRequireFlag == FUNC_DATA_REQUIRED_NOT_LOAD) {
1,169,862!
1646
    int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
73,001✔
1647
    for (int32_t i = 0; i < numOfCols; ++i) {
167,039✔
1648
      SColumnInfoData* pCol = (SColumnInfoData*)taosArrayGet(pBlock->pDataBlock, i);
94,070✔
1649
      if (pCol) {
94,051!
1650
        pCol->info.noData = true;
94,057✔
1651
      }
1652
    }
1653
  }
1654
}
1,169,830✔
1655

1656
int32_t createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
1,170,643✔
1657
                                    STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
1658
                                    SOperatorInfo** pOptrInfo) {
1659
  QRY_PARAM_CHECK(pOptrInfo);
1,170,643!
1660

1661
  int32_t         code = TSDB_CODE_SUCCESS;
1,170,643✔
1662
  int32_t         lino = 0;
1,170,643✔
1663
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
1,170,643!
1664
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
1,171,187!
1665
  if (pInfo == NULL || pOperator == NULL) {
1,171,317!
1666
    code = terrno;
×
1667
    goto _error;
×
1668
  }
1669

1670
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
1,171,384✔
1671
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
1,171,384✔
1672

1673
  int32_t numOfCols = 0;
1,171,384✔
1674
  code =
1675
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
1,171,384✔
1676
  QUERY_CHECK_CODE(code, lino, _error);
1,170,631!
1677

1678
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
1,170,631✔
1679
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
1,171,297✔
1680
  QUERY_CHECK_CODE(code, lino, _error);
1,171,511!
1681

1682
  if (pScanNode->pScanPseudoCols != NULL) {
1,171,511✔
1683
    SExprSupp* pSup = &pInfo->base.pseudoSup;
342,072✔
1684
    pSup->pExprInfo = NULL;
342,072✔
1685
    code = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
342,072✔
1686
    QUERY_CHECK_CODE(code, lino, _error);
341,903!
1687

1688
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
341,903✔
1689
                                      &pTaskInfo->storageAPI.functionStore);
1690
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
341,651✔
1691
  }
1692

1693
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
1,170,444✔
1694
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
1,170,444✔
1695

1696
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
1,170,444✔
1697
  pInfo->base.readHandle = *readHandle;
1,170,200✔
1698
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;
1,170,200✔
1699

1700
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
1,170,200✔
1701
  pInfo->sample.seed = taosGetTimestampSec();
1,170,200✔
1702

1703
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
1,170,465✔
1704
  initResultSizeInfo(&pOperator->resultInfo, 4096);
1,170,465✔
1705
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
1,169,896✔
1706
  resetClolumnReserve(pInfo->pResBlock, pInfo->base.dataBlockLoadFlag);
1,170,328✔
1707
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
1,170,689!
1708

1709
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
1,170,689✔
1710
  QUERY_CHECK_CODE(code, lino, _error);
1,170,970!
1711

1712
  pInfo->virtualStableScan = pScanNode->virtualStableScan;
1,170,970✔
1713
  if (pScanNode->node.dynamicOp && pScanNode->virtualStableScan) {
1,170,970!
1714
    TSWAP(pInfo->pOrgBlock, pInfo->pResBlock);
×
1715
    pInfo->pResBlock = NULL;
×
1716
    memcpy(&pInfo->base.orgCond, &pInfo->base.cond, sizeof(SQueryTableDataCond));
×
1717
    memset(&pInfo->base.cond, 0, sizeof(SQueryTableDataCond));
×
1718
  }
1719

1720
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
1,170,970✔
1721
  QUERY_CHECK_CODE(code, lino, _error);
1,170,136✔
1722

1723
  pInfo->currentGroupId = -1;
1,169,914✔
1724

1725
  pInfo->tableEndIndex = -1;
1,169,914✔
1726
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
1,169,914✔
1727
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
1,169,914✔
1728

1729
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
1,169,914✔
1730
                  pTaskInfo);
1731
  pOperator->exprSupp.numOfExprs = numOfCols;
1,169,408✔
1732

1733
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
1,169,408✔
1734
  pInfo->ignoreTag = false;
1,169,408✔
1735

1736
  pInfo->base.pTableListInfo = pTableListInfo;
1,169,408✔
1737
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
1,169,408✔
1738
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
1,171,153!
1739
    code = terrno;
×
1740
    QUERY_CHECK_CODE(code, lino, _error);
×
1741
  }
1742

1743
  pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
1,171,153✔
1744

1745
  taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
1,171,153✔
1746
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanNext, NULL, destroyTableScanOperatorInfo,
1,171,659✔
1747
                                         optrDefaultBufFn, getTableScannerExecInfo, optrDefaultGetNextExtFn, NULL);
1748

1749
  // for non-blocking operator, the open cost is always 0
1750
  pOperator->cost.openCost = 0;
1,170,392✔
1751
  *pOptrInfo = pOperator;
1,170,392✔
1752
  return TSDB_CODE_SUCCESS;
1,170,392✔
1753

1754
_error:
222✔
1755
  if (pInfo != NULL) {
222!
1756
    pInfo->base.pTableListInfo = NULL;  // this attribute will be destroy outside of this function
222✔
1757
    destroyTableScanOperatorInfo(pInfo);
222✔
1758
  }
1759

1760
  if (pOperator != NULL) {
222!
1761
    pOperator->info = NULL;
222✔
1762
    destroyOperator(pOperator);
222✔
1763
  }
1764
  pTaskInfo->code = code;
222✔
1765
  return code;
222✔
1766
}
1767

1768
int32_t createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
×
1769
  QRY_PARAM_CHECK(pOptrInfo);
×
1770

1771
  int32_t         code = 0;
×
1772
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
×
1773
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
1774
  if (pInfo == NULL || pOperator == NULL) {
×
1775
    code = terrno;
×
1776
    goto _end;
×
1777
  }
1778

1779
  pInfo->base.dataReader = pReadHandle;
×
1780
  //  pInfo->prevGroupId       = -1;
1781

1782
  setOperatorInfo(pOperator, "TableSeqScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN, false, OP_NOT_OPENED,
×
1783
                  pInfo, pTaskInfo);
1784
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanImplNext, NULL, NULL, optrDefaultBufFn, NULL,
×
1785
                                         optrDefaultGetNextExtFn, NULL);
1786
  *pOptrInfo = pOperator;
×
1787
  return code;
×
1788

1789
_end:
×
1790
  if (pInfo != NULL) {
×
1791
    taosMemoryFree(pInfo);
×
1792
  }
1793

1794
  if (pOperator != NULL) {
×
1795
    taosMemoryFree(pOperator);
×
1796
  }
1797

1798
  pTaskInfo->code = code;
×
1799
  return code;
×
1800
}
1801

1802
FORCE_INLINE void doClearBufferedBlocks(SStreamScanInfo* pInfo) {
×
1803
  qDebug("clear buff blocks:%d", (int32_t)taosArrayGetSize(pInfo->pBlockLists));
17,399!
1804
  taosArrayClear(pInfo->pBlockLists);
17,398✔
1805
  pInfo->validBlockIndex = 0;
17,398✔
1806
}
17,398✔
1807

1808
static bool isSessionWindow(SStreamScanInfo* pInfo) {
980✔
1809
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION ||
1,652✔
1810
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_SESSION ||
1,612✔
1811
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_SESSION;
632!
1812
}
1813

1814
static bool isStateWindow(SStreamScanInfo* pInfo) {
657✔
1815
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE ||
781✔
1816
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_EVENT;
124✔
1817
}
1818

1819
static bool isIntervalWindow(SStreamScanInfo* pInfo) {
8,593✔
1820
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL ||
15,538✔
1821
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL ||
6,945✔
1822
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_MID_INTERVAL ||
5,978!
1823
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL ||
5,978!
1824
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_INTERVAL ||
5,978✔
1825
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_SEMI_INTERVAL ||
19,590!
1826
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_CONTINUE_FINAL_INTERVAL;
4,052!
1827
}
1828

1829
static bool isSignleIntervalWindow(SStreamScanInfo* pInfo) {
1,328✔
1830
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL;
1,328✔
1831
}
1832

1833
static bool isSlidingWindow(SStreamScanInfo* pInfo) {
6,513✔
1834
  return isIntervalWindow(pInfo) && pInfo->interval.interval != pInfo->interval.sliding;
6,513✔
1835
}
1836

1837
static bool isCountSlidingWindow(SStreamScanInfo* pInfo) {
6,195✔
1838
  return pInfo->windowSup.pStreamAggSup &&
10,868✔
1839
         (pInfo->windowSup.pStreamAggSup->windowCount != pInfo->windowSup.pStreamAggSup->windowSliding);
4,673✔
1840
}
1841

1842
static bool isCountWindow(SStreamScanInfo* pInfo) {
99✔
1843
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_COUNT;
99✔
1844
}
1845

1846
static bool isTimeSlice(SStreamScanInfo* pInfo) {
81✔
1847
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERP_FUNC;
81✔
1848
}
1849

1850
static void setGroupId(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t groupColIndex, int32_t rowIndex) {
6,513✔
1851
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, groupColIndex);
6,513✔
1852
  uint64_t*        groupCol = (uint64_t*)pColInfo->pData;
6,513✔
1853
  if (colDataIsNull_s(pColInfo, rowIndex)) {
13,026✔
1854
    pInfo->igCheckGroupId = true;
4,329✔
1855
  } else {
1856
    pInfo->groupId = groupCol[rowIndex];
2,184✔
1857
  }
1858
}
6,513✔
1859

1860
void resetTableScanInfo(STableScanInfo* pTableScanInfo, STimeWindow* pWin, int64_t ver) {
6,514✔
1861
  pTableScanInfo->base.cond.twindows = *pWin;
6,514✔
1862
  pTableScanInfo->base.cond.startVersion = 0;
6,514✔
1863
  pTableScanInfo->base.cond.endVersion = ver;
6,514✔
1864
  pTableScanInfo->scanTimes = 0;
6,514✔
1865
  pTableScanInfo->currentGroupId = -1;
6,514✔
1866
  pTableScanInfo->tableEndIndex = -1;
6,514✔
1867
  pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
6,514✔
1868
  pTableScanInfo->base.dataReader = NULL;
6,514✔
1869
  pTableScanInfo->scanMode = TABLE_SCAN__BLOCK_ORDER;
6,514✔
1870
}
6,514✔
1871

1872
SSDataBlock* readPreVersionData(SOperatorInfo* pTableScanOp, uint64_t tbUid, TSKEY startTs, TSKEY endTs,
408✔
1873
                                int64_t maxVersion) {
1874
  int32_t       code = TSDB_CODE_SUCCESS;
408✔
1875
  int32_t       lino = 0;
408✔
1876
  STableKeyInfo tblInfo = {.uid = tbUid, .groupId = 0};
408✔
1877

1878
  STableScanInfo*     pTableScanInfo = pTableScanOp->info;
408✔
1879
  SQueryTableDataCond cond = pTableScanInfo->base.cond;
408✔
1880

1881
  cond.startVersion = -1;
408✔
1882
  cond.endVersion = maxVersion;
408✔
1883
  cond.twindows = (STimeWindow){.skey = startTs, .ekey = endTs};
408✔
1884

1885
  SExecTaskInfo* pTaskInfo = pTableScanOp->pTaskInfo;
408✔
1886
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
408✔
1887

1888
  SSDataBlock* pBlock = pTableScanInfo->pResBlock;
408✔
1889
  STsdbReader* pReader = NULL;
408✔
1890
  code = pAPI->tsdReader.tsdReaderOpen(pTableScanInfo->base.readHandle.vnode, &cond, &tblInfo, 1, pBlock,
408✔
1891
                                       (void**)&pReader, GET_TASKID(pTaskInfo), NULL);
408✔
1892
  QUERY_CHECK_CODE(code, lino, _end);
408!
1893

1894
  bool hasNext = false;
408✔
1895
  code = pAPI->tsdReader.tsdNextDataBlock(pReader, &hasNext);
408✔
1896
  QUERY_CHECK_CODE(code, lino, _end);
408!
1897

1898
  if (hasNext) {
408✔
1899
    SSDataBlock* p = NULL;
392✔
1900
    code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pReader, &p, NULL);
392✔
1901
    QUERY_CHECK_CODE(code, lino, _end);
392!
1902

1903
    code = doSetTagColumnData(&pTableScanInfo->base, pBlock, pTaskInfo, pBlock->info.rows);
392✔
1904
    QUERY_CHECK_CODE(code, lino, _end);
392!
1905

1906
    pBlock->info.id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
392✔
1907
  }
1908

1909
_end:
16✔
1910
  pAPI->tsdReader.tsdReaderClose(pReader);
408✔
1911
  qDebug("retrieve prev rows:%" PRId64 ", skey:%" PRId64 ", ekey:%" PRId64 " uid:%" PRIu64 ", max ver:%" PRId64
408!
1912
         ", suid:%" PRIu64,
1913
         pBlock->info.rows, startTs, endTs, tbUid, maxVersion, cond.suid);
1914
  if (code != TSDB_CODE_SUCCESS) {
408!
1915
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
1916
    terrno = code;
×
1917
    return NULL;
×
1918
  }
1919

1920
  return pBlock->info.rows > 0 ? pBlock : NULL;
408✔
1921
}
1922

1923
bool comparePrimaryKey(SColumnInfoData* pCol, int32_t rowId, void* pVal) {
2✔
1924
  // coverity scan
1925
  if (!pVal || !pCol) {
2!
1926
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
1927
    return false;
×
1928
  }
1929
  void* pData = colDataGetData(pCol, rowId);
2!
1930
  if (IS_VAR_DATA_TYPE(pCol->info.type)) {
2!
1931
    int32_t colLen = varDataLen(pData);
×
1932
    int32_t keyLen = varDataLen(pVal);
×
1933
    if (pCol->info.type == TSDB_DATA_TYPE_JSON) {
×
1934
      colLen = getJsonValueLen(pData);
×
1935
      keyLen = getJsonValueLen(pVal);
×
1936
    }
1937

1938
    if (colLen == keyLen && memcmp(pData, pVal, colLen) == 0) {
×
1939
      return true;
×
1940
    }
1941
  } else {
1942
    if (memcmp(pData, pVal, pCol->info.bytes) == 0) {
2!
1943
      return true;
2✔
1944
    }
1945
  }
1946
  return false;
×
1947
}
1948

1949
static bool hasPrimaryKeyCol(SStreamScanInfo* pInfo) { return pInfo->primaryKeyIndex != -1; }
10,016,955✔
1950

1951
static uint64_t getGroupIdByCol(SStreamScanInfo* pInfo, uint64_t uid, TSKEY ts, int64_t maxVersion, void* pVal) {
351✔
1952
  SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, uid, ts, ts, maxVersion);
351✔
1953
  if (!pPreRes || pPreRes->info.rows == 0) {
351!
1954
    if (terrno != TSDB_CODE_SUCCESS) {
16!
1955
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
1956
    }
1957
    return 0;
16✔
1958
  }
1959

1960
  int32_t rowId = 0;
335✔
1961
  if (hasPrimaryKeyCol(pInfo)) {
335✔
1962
    SColumnInfoData* pPkCol = taosArrayGet(pPreRes->pDataBlock, pInfo->primaryKeyIndex);
2✔
1963
    for (; rowId < pPreRes->info.rows; rowId++) {
2!
1964
      if (comparePrimaryKey(pPkCol, rowId, pVal)) {
2!
1965
        break;
2✔
1966
      }
1967
    }
1968
  }
1969
  if (rowId >= pPreRes->info.rows) {
335!
1970
    qInfo("===stream===read preversion data of primary key failed. ts:%" PRId64 ",version:%" PRId64, ts, maxVersion);
×
1971
    return 0;
×
1972
  }
1973
  return calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, rowId);
335✔
1974
}
1975

1976
static uint64_t getGroupIdByUid(SStreamScanInfo* pInfo, uint64_t uid) {
1,371✔
1977
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
1,371✔
1978
  return tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, uid);
1,371✔
1979
}
1980

1981
static uint64_t getGroupIdByData(SStreamScanInfo* pInfo, uint64_t uid, TSKEY ts, int64_t maxVersion, void* pVal) {
1,372✔
1982
  if (pInfo->partitionSup.needCalc) {
1,372✔
1983
    return getGroupIdByCol(pInfo, uid, ts, maxVersion, pVal);
351✔
1984
  }
1985

1986
  return getGroupIdByUid(pInfo, uid);
1,021✔
1987
}
1988

1989
void prepareRangeScan(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t* pRowIndex, bool* pRes) {
12,789✔
1990
  int32_t code = TSDB_CODE_SUCCESS;
12,789✔
1991
  int32_t lino = 0;
12,789✔
1992
  if (pBlock->info.rows == 0) {
12,789✔
1993
    if (pRes) {
1!
1994
      (*pRes) = false;
×
1995
    }
1996
    goto _end;
1✔
1997
  }
1998
  if ((*pRowIndex) == pBlock->info.rows) {
12,788✔
1999
    if (pRes) {
6,276!
2000
      (*pRes) = false;
6,276✔
2001
    }
2002
    goto _end;
6,276✔
2003
  }
2004

2005
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
6,512✔
2006
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
6,512✔
2007
  SColumnInfoData* pEndTsCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
6,512✔
2008
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
6,512✔
2009
  STimeWindow      win = {.skey = startData[*pRowIndex], .ekey = endData[*pRowIndex]};
6,512✔
2010
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
6,512✔
2011
  uint64_t*        gpData = (uint64_t*)pGpCol->pData;
6,512✔
2012
  uint64_t         groupId = gpData[*pRowIndex];
6,512✔
2013

2014
  SColumnInfoData* pCalStartTsCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
6,512✔
2015
  TSKEY*           calStartData = (TSKEY*)pCalStartTsCol->pData;
6,513✔
2016
  SColumnInfoData* pCalEndTsCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
6,513✔
2017
  TSKEY*           calEndData = (TSKEY*)pCalEndTsCol->pData;
6,513✔
2018

2019
  setGroupId(pInfo, pBlock, GROUPID_COLUMN_INDEX, *pRowIndex);
6,513✔
2020
  if (isSlidingWindow(pInfo) || isCountSlidingWindow(pInfo)) {
6,513✔
2021
    pInfo->updateWin.skey = calStartData[*pRowIndex];
326✔
2022
    pInfo->updateWin.ekey = calEndData[*pRowIndex];
326✔
2023
  }
2024
  (*pRowIndex)++;
6,513✔
2025

2026
  for (; *pRowIndex < pBlock->info.rows; (*pRowIndex)++) {
6,514✔
2027
    if (win.skey == startData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
239✔
2028
      win.ekey = TMAX(win.ekey, endData[*pRowIndex]);
1✔
2029
      continue;
1✔
2030
    }
2031

2032
    if (win.skey == endData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
238!
2033
      win.skey = TMIN(win.skey, startData[*pRowIndex]);
×
2034
      continue;
×
2035
    }
2036
    break;
238✔
2037
  }
2038

2039
  STableScanInfo* pTScanInfo = pInfo->pTableScanOp->info;
6,513✔
2040

2041
  int64_t maxVer = -1;
6,513✔
2042
  if (!isRecalculateOperator(&pInfo->basic)) {
6,513!
2043
    // coverity scan
2044
    QUERY_CHECK_NULL(pInfo->pUpdateInfo, code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
6,513!
2045

2046
    qDebug("prepare range scan start:%" PRId64 ",end:%" PRId64 ",maxVer:%" PRId64, win.skey, win.ekey,
6,513✔
2047
          pInfo->pUpdateInfo->maxDataVersion);
2048
    maxVer = pInfo->pUpdateInfo->maxDataVersion;
6,514✔
2049
  }
2050
  resetTableScanInfo(pInfo->pTableScanOp->info, &win, maxVer);
6,514✔
2051
  pInfo->pTableScanOp->status = OP_OPENED;
6,514✔
2052
  if (pRes) {
6,514✔
2053
    (*pRes) = true;
238✔
2054
  }
2055

2056
_end:
6,276✔
2057
  if (code != TSDB_CODE_SUCCESS) {
12,791!
2058
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2059
  }
2060
}
12,791✔
2061

2062
STimeWindow getSlidingWindow(TSKEY* startTsCol, TSKEY* endTsCol, uint64_t* gpIdCol, SInterval* pInterval,
941✔
2063
                             SDataBlockInfo* pDataBlockInfo, int32_t* pRowIndex, bool hasGroup) {
2064
  SResultRowInfo dumyInfo = {0};
941✔
2065
  dumyInfo.cur.pageId = -1;
941✔
2066
  STimeWindow win = getActiveTimeWindow(NULL, &dumyInfo, startTsCol[*pRowIndex], pInterval, TSDB_ORDER_ASC);
941✔
2067
  STimeWindow endWin = win;
941✔
2068
  STimeWindow preWin = win;
941✔
2069
  uint64_t    groupId = gpIdCol[*pRowIndex];
941✔
2070

2071
  while (1) {
2072
    if (hasGroup) {
941✔
2073
      (*pRowIndex) += 1;
123✔
2074
    } else {
2075
      while ((groupId == gpIdCol[(*pRowIndex)] && startTsCol[*pRowIndex] <= endWin.ekey)) {
859✔
2076
        (*pRowIndex) += 1;
825✔
2077
        if ((*pRowIndex) == pDataBlockInfo->rows) {
825✔
2078
          break;
784✔
2079
        }
2080
      }
2081
    }
2082

2083
    do {
2084
      preWin = endWin;
1,721✔
2085
      getNextTimeWindow(pInterval, &endWin, TSDB_ORDER_ASC);
1,721✔
2086
    } while (endTsCol[(*pRowIndex) - 1] >= endWin.skey);
1,721✔
2087
    endWin = preWin;
941✔
2088
    if (win.ekey == endWin.ekey || (*pRowIndex) == pDataBlockInfo->rows || groupId != gpIdCol[*pRowIndex]) {
941!
2089
      win.ekey = endWin.ekey;
941✔
2090
      return win;
941✔
2091
    }
2092
    win.ekey = endWin.ekey;
×
2093
  }
2094
}
2095

2096
int32_t doRangeScan(SStreamScanInfo* pInfo, SSDataBlock* pSDB, int32_t tsColIndex, int32_t* pRowIndex,
8,399✔
2097
                    SSDataBlock** ppRes) {
2098
  qDebug("do stream range scan. windows index:%d", *pRowIndex);
8,399✔
2099

2100
  int32_t code = TSDB_CODE_SUCCESS;
8,399✔
2101
  int32_t lino = 0;
8,399✔
2102
  bool    prepareRes = true;
8,399✔
2103

2104
  while (1) {
412✔
2105
    SSDataBlock* pResult = NULL;
8,811✔
2106
    code = doTableScanNext(pInfo->pTableScanOp, &pResult);
8,811✔
2107
    QUERY_CHECK_CODE(code, lino, _end);
17,210!
2108

2109
    if (!pResult) {
8,811✔
2110
      prepareRangeScan(pInfo, pSDB, pRowIndex, &prepareRes);
6,514✔
2111
      // scan next window data
2112
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
6,514✔
2113
      QUERY_CHECK_CODE(code, lino, _end);
6,514!
2114
    }
2115

2116
    if (!pResult) {
8,811✔
2117
      if (prepareRes) {
6,354✔
2118
        continue;
78✔
2119
      }
2120
      blockDataCleanup(pSDB);
6,276✔
2121
      *pRowIndex = 0;
6,276✔
2122
      pInfo->updateWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
6,276✔
2123
      STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
6,276✔
2124
      pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
6,276✔
2125
      pTableScanInfo->base.dataReader = NULL;
6,276✔
2126
      (*ppRes) = NULL;
6,276✔
2127
      goto _end;
6,276✔
2128
    }
2129

2130
    code = doFilter(pResult, pInfo->pTableScanOp->exprSupp.pFilterInfo, NULL);
2,457✔
2131
    QUERY_CHECK_CODE(code, lino, _end);
2,457!
2132
    if (pResult->info.rows == 0) {
2,457!
2133
      continue;
×
2134
    }
2135

2136
    if (pInfo->igCheckGroupId == true) {
2,457✔
2137
      pResult->info.calWin = pInfo->updateWin;
590✔
2138
      (*ppRes) = pResult;
590✔
2139
      goto _end;
590✔
2140
    }
2141

2142
    if (pInfo->partitionSup.needCalc) {
1,867✔
2143
      SSDataBlock* tmpBlock = NULL;
430✔
2144
      code = createOneDataBlock(pResult, true, &tmpBlock);
430✔
2145
      QUERY_CHECK_CODE(code, lino, _end);
706!
2146

2147
      blockDataCleanup(pResult);
430✔
2148
      for (int32_t i = 0; i < tmpBlock->info.rows; i++) {
1,198✔
2149
        if (calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, tmpBlock, i) == pInfo->groupId) {
768✔
2150
          for (int32_t j = 0; j < pInfo->pTableScanOp->exprSupp.numOfExprs; j++) {
4,749✔
2151
            SColumnInfoData* pSrcCol = taosArrayGet(tmpBlock->pDataBlock, j);
4,305✔
2152
            SColumnInfoData* pDestCol = taosArrayGet(pResult->pDataBlock, j);
4,305✔
2153
            bool             isNull = colDataIsNull(pSrcCol, tmpBlock->info.rows, i, NULL);
4,305✔
2154
            char*            pSrcData = NULL;
4,305✔
2155
            if (!isNull) pSrcData = colDataGetData(pSrcCol, i);
4,305!
2156
            code = colDataSetVal(pDestCol, pResult->info.rows, pSrcData, isNull);
4,305✔
2157
            QUERY_CHECK_CODE(code, lino, _end);
4,305!
2158
          }
2159
          pResult->info.rows++;
444✔
2160
        }
2161
      }
2162

2163
      blockDataDestroy(tmpBlock);
430✔
2164

2165
      if (pResult->info.rows > 0) {
430✔
2166
        pResult->info.calWin = pInfo->updateWin;
276✔
2167
        (*ppRes) = pResult;
276✔
2168
        goto _end;
276✔
2169
      }
2170
    } else if (pResult->info.id.groupId == pInfo->groupId) {
1,437✔
2171
      pResult->info.calWin = pInfo->updateWin;
1,257✔
2172
      (*ppRes) = pResult;
1,257✔
2173
      goto _end;
1,257✔
2174
    }
2175
  }
2176

2177
_end:
8,399✔
2178
  if (code != TSDB_CODE_SUCCESS) {
8,399!
2179
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2180
  }
2181
  return code;
8,399✔
2182
}
2183

2184
static void getPreSessionWindow(SStreamAggSupporter* pAggSup, TSKEY startTs, TSKEY endTs, uint64_t groupId,
2✔
2185
                                SSessionKey* pKey) {
2186
  pKey->win.skey = startTs;
2✔
2187
  pKey->win.ekey = endTs;
2✔
2188
  pKey->groupId = groupId;
2✔
2189

2190
  void*   pCur = pAggSup->stateStore.streamStateSessionSeekKeyCurrentPrev(pAggSup->pState, pKey);
2✔
2191
  int32_t code = pAggSup->stateStore.streamStateSessionGetKVByCur(pCur, pKey, NULL, 0);
2✔
2192
  if (code != TSDB_CODE_SUCCESS) {
2!
2193
    SET_SESSION_WIN_KEY_INVALID(pKey);
×
2194
  }
2195

2196
  taosMemoryFree(pCur);
2!
2197
}
2✔
2198

2199
int32_t appendOneRowToSpecialBlockImpl(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, TSKEY* pCalStartTs,
6,714✔
2200
                                       TSKEY* pCalEndTs, uint64_t* pUid, uint64_t* pGp, void* pTbName, void* pPkData) {
2201
  int32_t          code = TSDB_CODE_SUCCESS;
6,714✔
2202
  int32_t          lino = 0;
6,714✔
2203
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
6,714✔
2204
  SColumnInfoData* pEndTsCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
6,714✔
2205
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
6,714✔
2206
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
6,714✔
2207
  SColumnInfoData* pCalStartCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
6,714✔
2208
  SColumnInfoData* pCalEndCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
6,714✔
2209
  SColumnInfoData* pTableCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
6,714✔
2210
  code = colDataSetVal(pStartTsCol, pBlock->info.rows, (const char*)pStartTs, false);
6,714✔
2211
  QUERY_CHECK_CODE(code, lino, _end);
6,713!
2212

2213
  code = colDataSetVal(pEndTsCol, pBlock->info.rows, (const char*)pEndTs, false);
6,713✔
2214
  QUERY_CHECK_CODE(code, lino, _end);
6,713!
2215

2216
  code = colDataSetVal(pUidCol, pBlock->info.rows, (const char*)pUid, pUid == NULL);
6,713✔
2217
  QUERY_CHECK_CODE(code, lino, _end);
6,713!
2218

2219
  code = colDataSetVal(pGpCol, pBlock->info.rows, (const char*)pGp, pGp == NULL);
6,713✔
2220
  QUERY_CHECK_CODE(code, lino, _end);
6,713!
2221

2222
  code = colDataSetVal(pCalStartCol, pBlock->info.rows, (const char*)pCalStartTs, false);
6,713✔
2223
  QUERY_CHECK_CODE(code, lino, _end);
6,713!
2224

2225
  code = colDataSetVal(pCalEndCol, pBlock->info.rows, (const char*)pCalEndTs, false);
6,713✔
2226
  QUERY_CHECK_CODE(code, lino, _end);
6,712!
2227

2228
  code = colDataSetVal(pTableCol, pBlock->info.rows, (const char*)pTbName, pTbName == NULL);
6,712✔
2229
  QUERY_CHECK_CODE(code, lino, _end);
6,712!
2230

2231
  if (taosArrayGetSize(pBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
6,712✔
2232
    SColumnInfoData* pPkCol = taosArrayGet(pBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
3✔
2233
    code = colDataSetVal(pPkCol, pBlock->info.rows, (const char*)pPkData, pPkData == NULL);
3✔
2234
    QUERY_CHECK_CODE(code, lino, _end);
3!
2235
  }
2236
  pBlock->info.rows++;
6,712✔
2237

2238
_end:
6,712✔
2239
  if (code != TSDB_CODE_SUCCESS) {
6,712!
2240
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2241
  }
2242
  return code;
6,712✔
2243
}
2244

2245
int32_t appendPkToSpecialBlock(SSDataBlock* pBlock, TSKEY* pTsArray, SColumnInfoData* pPkCol, int32_t rowId,
1,079✔
2246
                               uint64_t* pUid, uint64_t* pGp, void* pTbName) {
2247
  void* pVal = NULL;
1,079✔
2248
  if (pPkCol) {
1,079✔
2249
    pVal = colDataGetData(pPkCol, rowId);
9!
2250
  }
2251
  return appendOneRowToSpecialBlockImpl(pBlock, pTsArray + rowId, pTsArray + rowId, pTsArray + rowId, pTsArray + rowId,
1,079✔
2252
                                        pUid, pGp, pTbName, pVal);
2253
}
2254

2255
static int32_t getPreVersionDataBlock(uint64_t uid, TSKEY startTs, TSKEY endTs, int64_t version, char* taskIdStr,
7✔
2256
                                      SStreamScanInfo* pInfo, SSDataBlock* pBlock) {
2257
  int32_t      code = TSDB_CODE_SUCCESS;
7✔
2258
  int32_t      lino = 0;
7✔
2259
  SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, uid, startTs, endTs, version);
7✔
2260
  printDataBlock(pPreRes, "pre res", taskIdStr);
7✔
2261
  blockDataCleanup(pBlock);
7✔
2262
  if (!pPreRes) {
7!
2263
    if (terrno != TSDB_CODE_SUCCESS) {
×
2264
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2265
    }
2266
    goto _end;
×
2267
  }
2268
  code = blockDataEnsureCapacity(pBlock, pPreRes->info.rows);
7✔
2269
  QUERY_CHECK_CODE(code, lino, _end);
7!
2270

2271
  SColumnInfoData* pTsCol = (SColumnInfoData*)taosArrayGet(pPreRes->pDataBlock, pInfo->primaryTsIndex);
7✔
2272
  SColumnInfoData* pPkCol = NULL;
7✔
2273
  if (hasPrimaryKeyCol(pInfo)) {
7✔
2274
    pPkCol = (SColumnInfoData*)taosArrayGet(pPreRes->pDataBlock, pInfo->primaryKeyIndex);
2✔
2275
  }
2276
  for (int32_t i = 0; i < pPreRes->info.rows; i++) {
28✔
2277
    uint64_t groupId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, i);
21✔
2278
    code = appendPkToSpecialBlock(pBlock, (TSKEY*)pTsCol->pData, pPkCol, i, &uid, &groupId, NULL);
21✔
2279
    QUERY_CHECK_CODE(code, lino, _end);
21!
2280
  }
2281
  printDataBlock(pBlock, "new delete", taskIdStr);
7✔
2282

2283
_end:
7✔
2284
  if (code != TSDB_CODE_SUCCESS) {
7!
2285
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2286
  }
2287
  return code;
7✔
2288
}
2289

2290
static int32_t generateSessionScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
565✔
2291
                                        EStreamType mode) {
2292
  int32_t code = TSDB_CODE_SUCCESS;
565✔
2293
  int32_t lino = 0;
565✔
2294
  if (pSrcBlock->info.rows == 0) {
565!
2295
    return TSDB_CODE_SUCCESS;
×
2296
  }
2297
  SSHashObj*       pScanRange = tSimpleHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
565✔
2298
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
565✔
2299
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
565✔
2300
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
565✔
2301
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
565✔
2302
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
565✔
2303
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
565✔
2304
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
565✔
2305
  SColumnInfoData* pGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
565✔
2306
  SColumnInfoData* pSrcPkCol = NULL;
565✔
2307
  uint64_t*        pSrcGp = (uint64_t*)pGpCol->pData;
565✔
2308
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
565✔
2309
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
1✔
2310
  }
2311
  int64_t ver = pSrcBlock->info.version - 1;
565✔
2312

2313
  if (pInfo->partitionSup.needCalc &&
565✔
2314
      (startData[0] != endData[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
228✔
2315
    code = getPreVersionDataBlock(uidCol[0], startData[0], endData[0], ver, GET_TASKID(pTaskInfo), pInfo, pSrcBlock);
4✔
2316
    QUERY_CHECK_CODE(code, lino, _end);
4!
2317
    startData = (TSKEY*)pStartTsCol->pData;
4✔
2318
    endData = (TSKEY*)pEndTsCol->pData;
4✔
2319
    uidCol = (uint64_t*)pUidCol->pData;
4✔
2320
    pSrcGp = (uint64_t*)pGpCol->pData;
4✔
2321
  }
2322
  blockDataCleanup(pDestBlock);
565✔
2323
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
565✔
2324
  QUERY_CHECK_CODE(code, lino, _end);
565!
2325

2326
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
565✔
2327
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
565✔
2328
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
565✔
2329
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
565✔
2330
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
565✔
2331
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
565✔
2332
  SColumnInfoData* pDestTableNameInxCol = taosArrayGet(pDestBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
565✔
2333
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
1,176✔
2334
    uint64_t groupId = pSrcGp[i];
611✔
2335
    if (groupId == 0) {
611✔
2336
      void* pVal = NULL;
522✔
2337
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
522!
2338
        pVal = colDataGetData(pSrcPkCol, i);
1!
2339
      }
2340
      groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], ver, pVal);
522✔
2341
    }
2342
    // gap must be 0.
2343
    SSessionKey startWin = {0};
611✔
2344
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, startData[i], startData[i], groupId, &startWin);
611✔
2345
    if (IS_INVALID_SESSION_WIN_KEY(startWin)) {
611✔
2346
      // window has been closed.
2347
      continue;
53✔
2348
    }
2349
    SSessionKey endWin = {0};
563✔
2350
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, endData[i], endData[i], groupId, &endWin);
563✔
2351
    if (IS_INVALID_SESSION_WIN_KEY(endWin)) {
563✔
2352
      getPreSessionWindow(pInfo->windowSup.pStreamAggSup, endData[i], endData[i], groupId, &endWin);
2✔
2353
    }
2354
    if (IS_INVALID_SESSION_WIN_KEY(startWin)) {
563!
2355
      // window has been closed.
2356
      qError("generate session scan range failed. rang start:%" PRIx64 ", end:%" PRIx64, startData[i], endData[i]);
×
2357
      continue;
×
2358
    }
2359

2360
    SSessionKey checkKey = {.groupId = groupId, .win.skey = startWin.win.skey, .win.ekey = endWin.win.ekey};
563✔
2361
    if (tSimpleHashGet(pScanRange, &checkKey, sizeof(SSessionKey)) != NULL) {
563✔
2362
      continue;
5✔
2363
    }
2364
    code = tSimpleHashPut(pScanRange, &checkKey, sizeof(SSessionKey), NULL, 0);
558✔
2365
    QUERY_CHECK_CODE(code, lino, _end);
558!
2366

2367
    code = colDataSetVal(pDestStartCol, i, (const char*)&startWin.win.skey, false);
558✔
2368
    QUERY_CHECK_CODE(code, lino, _end);
558!
2369

2370
    code = colDataSetVal(pDestEndCol, i, (const char*)&endWin.win.ekey, false);
558✔
2371
    QUERY_CHECK_CODE(code, lino, _end);
558!
2372

2373
    colDataSetNULL(pDestUidCol, i);
558!
2374
    code = colDataSetVal(pDestGpCol, i, (const char*)&groupId, false);
558✔
2375
    QUERY_CHECK_CODE(code, lino, _end);
558!
2376

2377
    colDataSetNULL(pDestCalStartTsCol, i);
558!
2378
    colDataSetNULL(pDestTableNameInxCol, i);
558!
2379
    pDestBlock->info.rows++;
558✔
2380
  }
2381

2382
_end:
565✔
2383
  tSimpleHashCleanup(pScanRange);
565✔
2384
  if (code != TSDB_CODE_SUCCESS) {
565!
2385
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2386
  }
2387
  return code;
565✔
2388
}
2389

2390
static int32_t generateCountScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
7✔
2391
                                      EStreamType mode) {
2392
  int32_t code = TSDB_CODE_SUCCESS;
7✔
2393
  int32_t lino = 0;
7✔
2394
  blockDataCleanup(pDestBlock);
7✔
2395
  if (pSrcBlock->info.rows == 0) {
7!
2396
    return TSDB_CODE_SUCCESS;
×
2397
  }
2398
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
7✔
2399
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
7✔
2400
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
7✔
2401
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
7✔
2402
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
7✔
2403
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
7✔
2404
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
7✔
2405
  SColumnInfoData* pGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
7✔
2406
  uint64_t*        pSrcGp = (uint64_t*)pGpCol->pData;
7✔
2407
  SColumnInfoData* pSrcPkCol = NULL;
7✔
2408
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
7!
2409
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
×
2410
  }
2411
  int64_t ver = pSrcBlock->info.version - 1;
7✔
2412

2413
  if (pInfo->partitionSup.needCalc &&
7✔
2414
      (startData[0] != endData[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
1!
2415
    code = getPreVersionDataBlock(uidCol[0], startData[0], endData[0], ver, GET_TASKID(pTaskInfo), pInfo, pSrcBlock);
1✔
2416
    QUERY_CHECK_CODE(code, lino, _end);
1!
2417
    startData = (TSKEY*)pStartTsCol->pData;
1✔
2418
    endData = (TSKEY*)pEndTsCol->pData;
1✔
2419
    uidCol = (uint64_t*)pUidCol->pData;
1✔
2420
    pSrcGp = (uint64_t*)pGpCol->pData;
1✔
2421
  }
2422

2423
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
7✔
2424
  QUERY_CHECK_CODE(code, lino, _end);
7!
2425

2426
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
7✔
2427
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
7✔
2428
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
7✔
2429
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
7✔
2430
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
7✔
2431
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
7✔
2432
  SColumnInfoData* pDestTableNameInxCol = taosArrayGet(pDestBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
7✔
2433
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
16✔
2434
    uint64_t groupId = pSrcGp[i];
9✔
2435
    if (groupId == 0) {
9✔
2436
      void* pVal = NULL;
6✔
2437
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
6!
2438
        pVal = colDataGetData(pSrcPkCol, i);
×
2439
      }
2440
      groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], ver, pVal);
6✔
2441
    }
2442
    SSessionKey startWin = {.win.skey = startData[i], .win.ekey = endData[i], .groupId = groupId};
9✔
2443
    SSessionKey range = {0};
9✔
2444
    getCountWinRange(pInfo->windowSup.pStreamAggSup, &startWin, mode, &range);
9✔
2445
    code = colDataSetVal(pDestStartCol, i, (const char*)&range.win.skey, false);
9✔
2446
    QUERY_CHECK_CODE(code, lino, _end);
9!
2447

2448
    code = colDataSetVal(pDestEndCol, i, (const char*)&range.win.ekey, false);
9✔
2449
    QUERY_CHECK_CODE(code, lino, _end);
9!
2450

2451
    colDataSetNULL(pDestUidCol, i);
9!
2452
    code = colDataSetVal(pDestGpCol, i, (const char*)&groupId, false);
9✔
2453
    QUERY_CHECK_CODE(code, lino, _end);
9!
2454

2455
    code = colDataSetVal(pDestCalStartTsCol, i, (const char*)&range.win.skey, false);
9✔
2456
    QUERY_CHECK_CODE(code, lino, _end);
9!
2457

2458
    code = colDataSetVal(pDestCalEndTsCol, i, (const char*)&range.win.ekey, false);
9✔
2459
    QUERY_CHECK_CODE(code, lino, _end);
9!
2460

2461
    colDataSetNULL(pDestTableNameInxCol, i);
9!
2462

2463
    pDestBlock->info.rows++;
9✔
2464
  }
2465

2466
_end:
7✔
2467
  if (code != TSDB_CODE_SUCCESS) {
7!
2468
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2469
  }
2470
  return code;
7✔
2471
}
2472

2473
static int32_t setDelRangeEndKey(SStreamAggSupporter* pAggSup, SStreamFillSupporter* pFillSup, SWinKey* pEndKey, STimeWindow* pScanRange, bool* pRes) {
24✔
2474
  int32_t     code = TSDB_CODE_SUCCESS;
24✔
2475
  int32_t     lino = 0;
24✔
2476
  SSlicePoint nextPoint = {.key.groupId = pEndKey->groupId};
24✔
2477
  int32_t     vLen = 0;
24✔
2478
  int32_t     winCode = TSDB_CODE_SUCCESS;
24✔
2479
  code = pAggSup->stateStore.streamStateFillGetNext(pAggSup->pState, pEndKey, &nextPoint.key, (void**)&nextPoint.pResPos, &vLen, &winCode);
24✔
2480
  QUERY_CHECK_CODE(code, lino, _end);
24!
2481
  if (winCode == TSDB_CODE_SUCCESS) {
24✔
2482
    setPointBuff(&nextPoint, pFillSup);
16✔
2483
    if (HAS_ROW_DATA(nextPoint.pLeftRow) && pEndKey->ts < nextPoint.pLeftRow->key) {
16!
2484
      pScanRange->ekey = nextPoint.pLeftRow->key;
8✔
2485
      *pRes = true;
8✔
2486
    } else if (pEndKey->ts < nextPoint.pRightRow->key) {
8✔
2487
      pScanRange->ekey = nextPoint.pRightRow->key;
2✔
2488
      *pRes = true;
2✔
2489
    } else {
2490
      *pEndKey = nextPoint.key;
6✔
2491
      pScanRange->ekey = TMAX(nextPoint.pRightRow->key, nextPoint.key.ts);
6✔
2492
      *pRes = false;
6✔
2493
    }
2494
  }
2495

2496
_end:
8✔
2497
  if (code != TSDB_CODE_SUCCESS) {
24!
2498
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2499
  }
2500
  return code;
24✔
2501
}
2502

2503
int32_t getTimeSliceWinRange(SStreamAggSupporter* pAggSup, SStreamFillSupporter* pFillSup, SInterval* pInterval, TSKEY start, TSKEY end,
15✔
2504
                             int64_t groupId, STimeWindow* pScanRange) {
2505
  int32_t        code = TSDB_CODE_SUCCESS;
15✔
2506
  int32_t        lino = 0;
15✔
2507
  int32_t        winCode = TSDB_CODE_SUCCESS;
15✔
2508
  SResultRowInfo dumyInfo = {0};
15✔
2509
  dumyInfo.cur.pageId = -1;
15✔
2510
  STimeWindow sWin = getActiveTimeWindow(NULL, &dumyInfo, start, pInterval, TSDB_ORDER_ASC);
15✔
2511
  SWinKey     startKey = {.groupId = groupId, .ts = sWin.skey};
15✔
2512

2513
  sWin = getActiveTimeWindow(NULL, &dumyInfo, end, pInterval, TSDB_ORDER_ASC);
15✔
2514
  SWinKey endKey = {.groupId = groupId, .ts = sWin.ekey};
15✔
2515

2516
  SSlicePoint prevPoint = {.key.groupId = groupId};
15✔
2517
  SSlicePoint nextPoint = {.key.groupId = groupId};
15✔
2518
  int32_t     vLen = 0;
15✔
2519
  code = pAggSup->stateStore.streamStateFillGetPrev(pAggSup->pState, &startKey, &prevPoint.key, (void**)&prevPoint.pResPos, &vLen, &winCode);
15✔
2520
  QUERY_CHECK_CODE(code, lino, _end);
15!
2521
  if (winCode == TSDB_CODE_SUCCESS) {
15✔
2522
    setPointBuff(&prevPoint, pFillSup);
10✔
2523
    if (HAS_ROW_DATA(prevPoint.pRightRow)) {
10!
2524
      pScanRange->skey = prevPoint.pRightRow->key;
9✔
2525
    } else {
2526
      pScanRange->skey = prevPoint.pLeftRow->key;
1✔
2527
    }
2528
  } else {
2529
    pScanRange->skey = startKey.ts;
5✔
2530
  }
2531

2532
  bool res = false;
15✔
2533
  SWinKey curKey = endKey;
15✔
2534
  code = setDelRangeEndKey(pAggSup, pFillSup, &curKey, pScanRange, &res);
15✔
2535
  QUERY_CHECK_CODE(code, lino, _end);
15!
2536
  if (res == false) {
15✔
2537
    code = setDelRangeEndKey(pAggSup, pFillSup, &curKey, pScanRange, &res);
9✔
2538
    QUERY_CHECK_CODE(code, lino, _end);
9!
2539
  }
2540
  if (res == false) {
15✔
2541
    pScanRange->ekey = TMAX(endKey.ts, pScanRange->ekey);
5✔
2542
  }
2543

2544
_end:
10✔
2545
  if (code != TSDB_CODE_SUCCESS) {
15!
2546
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2547
  }
2548
  return code;
15✔
2549
}
2550

2551
static int32_t generateTimeSliceScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
15✔
2552
                                          EStreamType mode) {
2553
  int32_t code = TSDB_CODE_SUCCESS;
15✔
2554
  int32_t lino = 0;
15✔
2555
  blockDataCleanup(pDestBlock);
15✔
2556
  if (pSrcBlock->info.rows == 0) {
15!
2557
    return TSDB_CODE_SUCCESS;
×
2558
  }
2559
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
15✔
2560
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
15✔
2561
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
15✔
2562
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
15✔
2563
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
15✔
2564
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
15✔
2565
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
15✔
2566
  SColumnInfoData* pGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
15✔
2567
  uint64_t*        pSrcGp = (uint64_t*)pGpCol->pData;
15✔
2568
  SColumnInfoData* pSrcPkCol = NULL;
15✔
2569
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
15!
2570
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
×
2571
  }
2572
  int64_t ver = pSrcBlock->info.version - 1;
15✔
2573

2574
  if (pInfo->partitionSup.needCalc &&
15!
2575
      (startData[0] != endData[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
×
2576
    code = getPreVersionDataBlock(uidCol[0], startData[0], endData[0], ver, GET_TASKID(pTaskInfo), pInfo, pSrcBlock);
×
2577
    QUERY_CHECK_CODE(code, lino, _end);
×
2578
    startData = (TSKEY*)pStartTsCol->pData;
×
2579
    endData = (TSKEY*)pEndTsCol->pData;
×
2580
    uidCol = (uint64_t*)pUidCol->pData;
×
2581
    pSrcGp = (uint64_t*)pGpCol->pData;
×
2582
  }
2583

2584
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
15✔
2585
  QUERY_CHECK_CODE(code, lino, _end);
15!
2586

2587
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
15✔
2588
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
15✔
2589
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
15✔
2590
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
15✔
2591
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
15✔
2592
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
15✔
2593
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
30✔
2594
    uint64_t groupId = pSrcGp[i];
15✔
2595
    if (groupId == 0) {
15!
2596
      void* pVal = NULL;
15✔
2597
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
15!
2598
        pVal = colDataGetData(pSrcPkCol, i);
×
2599
      }
2600
      groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], ver, pVal);
15✔
2601
    }
2602

2603
    STimeWindow scanRange = {0};
15✔
2604
    code = getTimeSliceWinRange(pInfo->windowSup.pStreamAggSup, pInfo->pFillSup, &pInfo->interval, startData[i], endData[i], groupId,
15✔
2605
                                &scanRange);
2606
    QUERY_CHECK_CODE(code, lino, _end);
15!
2607

2608
    code = colDataSetVal(pDestStartCol, i, (const char*)&scanRange.skey, false);
15✔
2609
    QUERY_CHECK_CODE(code, lino, _end);
15!
2610

2611
    code = colDataSetVal(pDestEndCol, i, (const char*)&scanRange.ekey, false);
15✔
2612
    QUERY_CHECK_CODE(code, lino, _end);
15!
2613

2614
    colDataSetNULL(pDestUidCol, i);
15!
2615
    code = colDataSetVal(pDestGpCol, i, (const char*)&groupId, false);
15✔
2616
    QUERY_CHECK_CODE(code, lino, _end);
15!
2617

2618
    code = colDataSetVal(pDestCalStartTsCol, i, (const char*)&scanRange.skey, false);
15✔
2619
    QUERY_CHECK_CODE(code, lino, _end);
15!
2620

2621
    code = colDataSetVal(pDestCalEndTsCol, i, (const char*)&scanRange.ekey, false);
15✔
2622
    QUERY_CHECK_CODE(code, lino, _end);
15!
2623

2624
    pDestBlock->info.rows++;
15✔
2625
  }
2626

2627
_end:
15✔
2628
  if (code != TSDB_CODE_SUCCESS) {
15!
2629
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2630
  }
2631
  return code;
15✔
2632
}
2633

2634
static int32_t generateIntervalScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock,
889✔
2635
                                         EStreamType mode) {
2636
  int32_t code = TSDB_CODE_SUCCESS;
889✔
2637
  int32_t lino = 0;
889✔
2638
  blockDataCleanup(pDestBlock);
889✔
2639
  if (pSrcBlock->info.rows == 0) {
889!
2640
    return TSDB_CODE_SUCCESS;
×
2641
  }
2642
  SSHashObj*       pScanRange = tSimpleHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT));
889✔
2643
  SExecTaskInfo*   pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
889✔
2644
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
889✔
2645
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
889✔
2646
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
889✔
2647
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
889✔
2648
  SColumnInfoData* pSrcPkCol = NULL;
889✔
2649
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
889✔
2650
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
1✔
2651
  }
2652

2653
  uint64_t* srcUidData = (uint64_t*)pSrcUidCol->pData;
889✔
2654
  TSKEY*    srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
889✔
2655
  TSKEY*    srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
889✔
2656
  int64_t   ver = pSrcBlock->info.version - 1;
889✔
2657

2658
  if (pInfo->partitionSup.needCalc &&
889✔
2659
      (srcStartTsCol[0] != srcEndTsCol[0] || (hasPrimaryKeyCol(pInfo) && mode == STREAM_DELETE_DATA))) {
105✔
2660
    code = getPreVersionDataBlock(srcUidData[0], srcStartTsCol[0], srcEndTsCol[0], ver, GET_TASKID(pTaskInfo), pInfo,
2✔
2661
                                  pSrcBlock);
2662
    QUERY_CHECK_CODE(code, lino, _end);
2!
2663

2664
    srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
2✔
2665
    srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
2✔
2666
    srcUidData = (uint64_t*)pSrcUidCol->pData;
2✔
2667
  }
2668

2669
  uint64_t* srcGp = (uint64_t*)pSrcGpCol->pData;
889✔
2670
  code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
889✔
2671
  QUERY_CHECK_CODE(code, lino, _end);
889!
2672

2673
  SColumnInfoData* pStartTsCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
889✔
2674
  SColumnInfoData* pEndTsCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
889✔
2675
  SColumnInfoData* pDeUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
889✔
2676
  SColumnInfoData* pGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
889✔
2677
  SColumnInfoData* pCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
889✔
2678
  SColumnInfoData* pCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
889✔
2679
  SColumnInfoData* pDestTableNameInxCol = taosArrayGet(pDestBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
889✔
2680
  for (int32_t i = 0; i < pSrcBlock->info.rows;) {
1,830✔
2681
    uint64_t srcUid = srcUidData[i];
941✔
2682
    uint64_t groupId = srcGp[i];
941✔
2683
    if (groupId == 0) {
941✔
2684
      void* pVal = NULL;
828✔
2685
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
828!
2686
        pVal = colDataGetData(pSrcPkCol, i);
1!
2687
      }
2688
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], ver, pVal);
828✔
2689
    }
2690
    TSKEY calStartTs = srcStartTsCol[i];
941✔
2691
    code = colDataSetVal(pCalStartTsCol, pDestBlock->info.rows, (const char*)(&calStartTs), false);
941✔
2692
    QUERY_CHECK_CODE(code, lino, _end);
941!
2693

2694
    STimeWindow win = getSlidingWindow(srcStartTsCol, srcEndTsCol, srcGp, &pInfo->interval, &pSrcBlock->info, &i,
941✔
2695
                                       pInfo->partitionSup.needCalc);
941✔
2696
    TSKEY       calEndTs = srcStartTsCol[i - 1];
941✔
2697
    code = colDataSetVal(pCalEndTsCol, pDestBlock->info.rows, (const char*)(&calEndTs), false);
941✔
2698
    QUERY_CHECK_CODE(code, lino, _end);
941!
2699

2700
    code = colDataSetVal(pDeUidCol, pDestBlock->info.rows, (const char*)(&srcUid), false);
941✔
2701
    QUERY_CHECK_CODE(code, lino, _end);
941!
2702

2703
    SSessionKey checkKey = {.groupId = groupId, .win = win};
941✔
2704
    if (tSimpleHashGet(pScanRange, &checkKey, sizeof(SSessionKey)) != NULL) {
941!
2705
      continue;
×
2706
    }
2707
    code = tSimpleHashPut(pScanRange, &checkKey, sizeof(SSessionKey), NULL, 0);
941✔
2708
    QUERY_CHECK_CODE(code, lino, _end);
941!
2709

2710
    code = colDataSetVal(pStartTsCol, pDestBlock->info.rows, (const char*)(&win.skey), false);
941✔
2711
    QUERY_CHECK_CODE(code, lino, _end);
941!
2712

2713
    code = colDataSetVal(pEndTsCol, pDestBlock->info.rows, (const char*)(&win.ekey), false);
941✔
2714
    QUERY_CHECK_CODE(code, lino, _end);
941!
2715

2716
    code = colDataSetVal(pGpCol, pDestBlock->info.rows, (const char*)(&groupId), false);
941✔
2717
    QUERY_CHECK_CODE(code, lino, _end);
941!
2718

2719
    colDataSetNULL(pDestTableNameInxCol,  pDestBlock->info.rows);
941!
2720

2721
    pDestBlock->info.rows++;
941✔
2722
  }
2723

2724
_end:
889✔
2725
  tSimpleHashCleanup(pScanRange);
889✔
2726
  if (code != TSDB_CODE_SUCCESS) {
889!
2727
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2728
  }
2729
  return code;
889✔
2730
}
2731

2732
int32_t calBlockTbName(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t rowId) {
525,264✔
2733
  int32_t        code = TSDB_CODE_SUCCESS;
525,264✔
2734
  int32_t        lino = 0;
525,264✔
2735
  SExecTaskInfo* pTaskInfo = pInfo->pStreamScanOp->pTaskInfo;
525,264✔
2736
  const char*    idStr = GET_TASKID(pTaskInfo);
525,264✔
2737

2738
  blockDataCleanup(pInfo->pCreateTbRes);
525,264✔
2739

2740
  if (pInfo->tbnameCalSup.numOfExprs == 0 && pInfo->tagCalSup.numOfExprs == 0) {
525,289✔
2741
    pBlock->info.parTbName[0] = 0;
517,070✔
2742
    if (pInfo->hasPart == false) {
517,070✔
2743
      pInfo->stateStore.streamStateSetParNameInvalid(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState);
515,118✔
2744
      clearParTbNameHashPtr(pTaskInfo->pRoot, idStr, &pTaskInfo->storageAPI);
515,105✔
2745
    }
2746
  } else {
2747
    code = appendCreateTableRow(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, &pInfo->tbnameCalSup,
8,219✔
2748
                                &pInfo->tagCalSup, pBlock->info.id.groupId, pBlock, rowId, pInfo->pCreateTbRes,
2749
                                &pInfo->stateStore);
2750
    QUERY_CHECK_CODE(code, lino, _end);
8,224!
2751
  }
2752

2753
  qTrace("%s %s child_table_name:%s,groupId:%" PRIu64, idStr, __func__, pBlock->info.parTbName, pBlock->info.id.groupId);
525,318✔
2754
_end:
522,294✔
2755
  if (code != TSDB_CODE_SUCCESS) {
525,318!
2756
    qError("%s %s failed at line %d since %s", idStr, __func__, lino, tstrerror(code));
×
2757
  }
2758
  return code;
525,246✔
2759
}
2760

2761
static int32_t generatePartitionDelResBlock(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
50✔
2762
  int32_t          code = TSDB_CODE_SUCCESS;
50✔
2763
  int32_t          lino = 0;
50✔
2764
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
50✔
2765
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
50✔
2766
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
50✔
2767
  uint64_t*        srcUidData = (uint64_t*)pSrcUidCol->pData;
50✔
2768
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
50✔
2769
  uint64_t*        srcGp = (uint64_t*)pSrcGpCol->pData;
50✔
2770

2771
  TSKEY*  srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
50✔
2772
  TSKEY*  srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
50✔
2773
  int64_t ver = pSrcBlock->info.version - 1;
50✔
2774
  for (int32_t delI = 0; delI < pSrcBlock->info.rows; delI++) {
100✔
2775
    uint64_t     groupId = 0;
50✔
2776
    uint64_t     srcUid = srcUidData[delI];
50✔
2777
    char         tbname[VARSTR_HEADER_SIZE + TSDB_TABLE_NAME_LEN] = {0};
50✔
2778
    SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, srcUid, srcStartTsCol[delI], srcEndTsCol[delI], ver);
50✔
2779
    if (!pPreRes) {
50!
2780
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2781
      continue;
×
2782
    }
2783
    code = blockDataEnsureCapacity(pDestBlock, pDestBlock->info.rows + pPreRes->info.rows);
50✔
2784
    QUERY_CHECK_CODE(code, lino, _end);
50!
2785
    for (int32_t preJ = 0; preJ < pPreRes->info.rows; preJ++) {
105✔
2786
      groupId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, preJ);
55✔
2787
      if (pInfo->pPartTbnameSup) {
55!
2788
        void*   parTbname = NULL;
55✔
2789
        int32_t winCode = TSDB_CODE_SUCCESS;
55✔
2790
        code = pInfo->stateStore.streamStateGetParName(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, groupId,
55✔
2791
                                                       &parTbname, false, &winCode);
2792
        QUERY_CHECK_CODE(code, lino, _end);
55!
2793

2794
        if (winCode != TSDB_CODE_SUCCESS) {
55✔
2795
          code = calBlockTbName(pInfo, pPreRes, preJ);
52✔
2796
          QUERY_CHECK_CODE(code, lino, _end);
52!
2797
          memcpy(varDataVal(tbname), pPreRes->info.parTbName, strlen(pPreRes->info.parTbName));
52✔
2798
        } else {
2799
          memcpy(varDataVal(tbname), parTbname, TSDB_TABLE_NAME_LEN);
3✔
2800
        }
2801
        varDataSetLen(tbname, strlen(varDataVal(tbname)));
55✔
2802
        pInfo->stateStore.streamStateFreeVal(parTbname);
55✔
2803
      }
2804
      code = appendDataToSpecialBlock(pDestBlock, srcStartTsCol + delI, srcEndTsCol + delI, srcUidData + delI, &groupId,
55✔
2805
                                      tbname[0] == 0 ? NULL : tbname);
55✔
2806
      QUERY_CHECK_CODE(code, lino, _end);
55!
2807
    }
2808
  }
2809

2810
_end:
50✔
2811
  if (code != TSDB_CODE_SUCCESS) {
50!
2812
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2813
  }
2814
  return code;
50✔
2815
}
2816

2817
static int32_t generateDeleteResultBlockImpl(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
1✔
2818
  int32_t          code = TSDB_CODE_SUCCESS;
1✔
2819
  int32_t          lino = 0;
1✔
2820
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
1✔
2821
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
1✔
2822
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
1✔
2823
  uint64_t*        srcUidData = (uint64_t*)pSrcUidCol->pData;
1✔
2824
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
1✔
2825
  uint64_t*        srcGp = (uint64_t*)pSrcGpCol->pData;
1✔
2826
  SColumnInfoData* pSrcPkCol = NULL;
1✔
2827
  if (taosArrayGetSize(pSrcBlock->pDataBlock) > PRIMARY_KEY_COLUMN_INDEX) {
1!
2828
    pSrcPkCol = taosArrayGet(pSrcBlock->pDataBlock, PRIMARY_KEY_COLUMN_INDEX);
×
2829
  }
2830

2831
  TSKEY*  srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
1✔
2832
  TSKEY*  srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
1✔
2833
  int64_t ver = pSrcBlock->info.version - 1;
1✔
2834
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
2✔
2835
    uint64_t srcUid = srcUidData[i];
1✔
2836
    uint64_t groupId = srcGp[i];
1✔
2837
    char     tbname[VARSTR_HEADER_SIZE + TSDB_TABLE_NAME_LEN] = {0};
1✔
2838
    if (groupId == 0) {
1!
2839
      void* pVal = NULL;
1✔
2840
      if (hasPrimaryKeyCol(pInfo) && pSrcPkCol) {
1!
2841
        pVal = colDataGetData(pSrcPkCol, i);
×
2842
      }
2843
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], ver, pVal);
1✔
2844
    }
2845
    if (pInfo->tbnameCalSup.pExprInfo) {
1!
2846
      void*   parTbname = NULL;
1✔
2847
      int32_t winCode = TSDB_CODE_SUCCESS;
1✔
2848
      code = pInfo->stateStore.streamStateGetParName(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, groupId,
1✔
2849
                                                     &parTbname, false, &winCode);
2850
      QUERY_CHECK_CODE(code, lino, _end);
1!
2851

2852
      if (winCode != TSDB_CODE_SUCCESS) {
1!
2853
        SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, srcUid, srcStartTsCol[i], srcStartTsCol[i], ver);
×
2854
        if (!pPreRes) {
×
2855
          qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
2856
          continue;
×
2857
        }
2858
        printDataBlock(pPreRes, "pre res", GET_TASKID(pInfo->pStreamScanOp->pTaskInfo));
×
2859
        code = calBlockTbName(pInfo, pPreRes, 0);
×
2860
        QUERY_CHECK_CODE(code, lino, _end);
×
2861
        memcpy(varDataVal(tbname), pPreRes->info.parTbName, strlen(pPreRes->info.parTbName));
×
2862
      } else {
2863
        memcpy(varDataVal(tbname), parTbname, TSDB_TABLE_NAME_LEN);
1✔
2864
      }
2865
      varDataSetLen(tbname, strlen(varDataVal(tbname)));
1✔
2866
      pInfo->stateStore.streamStateFreeVal(parTbname);
1✔
2867
    }
2868
    code = appendDataToSpecialBlock(pDestBlock, srcStartTsCol + i, srcEndTsCol + i, srcUidData + i, &groupId,
1✔
2869
                                    tbname[0] == 0 ? NULL : tbname);
1!
2870
    QUERY_CHECK_CODE(code, lino, _end);
1!
2871
  }
2872

2873
_end:
1✔
2874
  if (code != TSDB_CODE_SUCCESS) {
1!
2875
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2876
  }
2877
  return code;
1✔
2878
}
2879

2880
int32_t generateDeleteResultBlock(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
51✔
2881
  blockDataCleanup(pDestBlock);
51✔
2882
  int32_t rows = pSrcBlock->info.rows;
51✔
2883
  if (rows == 0) {
51!
2884
    return TSDB_CODE_SUCCESS;
×
2885
  }
2886
  int32_t code = blockDataEnsureCapacity(pDestBlock, rows);
51✔
2887
  if (code != TSDB_CODE_SUCCESS) {
51!
2888
    return code;
×
2889
  }
2890
  if (pInfo->partitionSup.needCalc) {
51✔
2891
    return generatePartitionDelResBlock(pInfo, pSrcBlock, pDestBlock);
50✔
2892
  }
2893
  return generateDeleteResultBlockImpl(pInfo, pSrcBlock, pDestBlock);
1✔
2894
}
2895

2896
int32_t generateScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock, EStreamType type) {
1,524✔
2897
  int32_t code = TSDB_CODE_SUCCESS;
1,524✔
2898
  int32_t lino = 0;
1,524✔
2899
  if (isIntervalWindow(pInfo)) {
1,524✔
2900
    code = generateIntervalScanRange(pInfo, pSrcBlock, pDestBlock, type);
889✔
2901
    QUERY_CHECK_CODE(code, lino, _end);
889!
2902
  } else if (isSessionWindow(pInfo) || isStateWindow(pInfo)) {
635✔
2903
    code = generateSessionScanRange(pInfo, pSrcBlock, pDestBlock, type);
565✔
2904
    QUERY_CHECK_CODE(code, lino, _end);
565!
2905
  } else if (isCountWindow(pInfo)) {
70✔
2906
    code = generateCountScanRange(pInfo, pSrcBlock, pDestBlock, type);
7✔
2907
    QUERY_CHECK_CODE(code, lino, _end);
7!
2908
  } else if (isTimeSlice(pInfo)) {
63✔
2909
    code = generateTimeSliceScanRange(pInfo, pSrcBlock, pDestBlock, type);
15✔
2910
    QUERY_CHECK_CODE(code, lino, _end);
15!
2911
  } else {
2912
    code = generateDeleteResultBlock(pInfo, pSrcBlock, pDestBlock);
48✔
2913
    QUERY_CHECK_CODE(code, lino, _end);
48!
2914
  }
2915
  pDestBlock->info.type = STREAM_CLEAR;
1,524✔
2916
  pDestBlock->info.version = pSrcBlock->info.version;
1,524✔
2917
  pDestBlock->info.dataLoad = 1;
1,524✔
2918
  code = blockDataUpdateTsWindow(pDestBlock, 0);
1,524✔
2919
  QUERY_CHECK_CODE(code, lino, _end);
1,524!
2920

2921
_end:
1,524✔
2922
  if (code != TSDB_CODE_SUCCESS) {
1,524!
2923
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
2924
  }
2925
  return code;
1,524✔
2926
}
2927

2928
int32_t appendDataToSpecialBlock(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, uint64_t* pUid, uint64_t* pGp,
5,635✔
2929
                                 void* pTbName) {
2930
  return appendOneRowToSpecialBlockImpl(pBlock, pStartTs, pEndTs, pStartTs, pEndTs, pUid, pGp, pTbName, NULL);
5,635✔
2931
}
2932

2933
bool checkExpiredData(SStateStore* pAPI, SUpdateInfo* pUpdateInfo, STimeWindowAggSupp* pTwSup, uint64_t tableId,
2,080✔
2934
                      TSKEY ts, void* pPkVal, int32_t len) {
2935
  bool isExpired = false;
2,080✔
2936
  bool isInc = pAPI->isIncrementalTimeStamp(pUpdateInfo, tableId, ts, pPkVal, len);
2,080✔
2937
  if (!isInc) {
2,079✔
2938
    isExpired = isOverdue(ts, pTwSup);
128✔
2939
  }
2940
  return isExpired;
2,079✔
2941
}
2942

2943
int32_t checkUpdateData(SStreamScanInfo* pInfo, bool invertible, SSDataBlock* pBlock, bool out) {
12,841✔
2944
  int32_t code = TSDB_CODE_SUCCESS;
12,841✔
2945
  int32_t lino = 0;
12,841✔
2946
  if (out) {
12,841✔
2947
    blockDataCleanup(pInfo->pUpdateDataRes);
11,323✔
2948
    code = blockDataEnsureCapacity(pInfo->pUpdateDataRes, pBlock->info.rows * 2);
11,321✔
2949
    QUERY_CHECK_CODE(code, lino, _end);
11,321!
2950
  }
2951
  SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, pInfo->primaryTsIndex);
12,839✔
2952
  TSKEY*           tsCol = (TSKEY*)pColDataInfo->pData;
12,839✔
2953
  SColumnInfoData* pPkColDataInfo = NULL;
12,839✔
2954
  if (hasPrimaryKeyCol(pInfo)) {
12,839✔
2955
    pPkColDataInfo = taosArrayGet(pBlock->pDataBlock, pInfo->primaryKeyIndex);
32✔
2956
  }
2957

2958
  bool tableInserted = pInfo->stateStore.updateInfoIsTableInserted(pInfo->pUpdateInfo, pBlock->info.id.uid);
12,839✔
2959
  for (int32_t rowId = 0; rowId < pBlock->info.rows; rowId++) {
10,016,263✔
2960
    SResultRowInfo dumyInfo;
2961
    dumyInfo.cur.pageId = -1;
10,001,592✔
2962
    bool        isClosed = false;
10,001,592✔
2963
    STimeWindow win = {.skey = INT64_MIN, .ekey = INT64_MAX};
10,001,592✔
2964
    bool        overDue = isOverdue(tsCol[rowId], &pInfo->twAggSup);
10,001,592✔
2965
    if (pInfo->igExpired && overDue) {
9,999,684✔
2966
      continue;
23✔
2967
    }
2968

2969
    if (tableInserted && overDue) {
9,999,661✔
2970
      win = getActiveTimeWindow(NULL, &dumyInfo, tsCol[rowId], &pInfo->interval, TSDB_ORDER_ASC);
2,965✔
2971
      isClosed = isCloseWindow(&win, &pInfo->twAggSup);
2,965✔
2972
    }
2973
    // must check update info first.
2974
    void*   pPkVal = NULL;
9,999,660✔
2975
    int32_t pkLen = 0;
9,999,660✔
2976
    if (hasPrimaryKeyCol(pInfo)) {
9,999,660✔
2977
      pPkVal = colDataGetData(pPkColDataInfo, rowId);
32!
2978
      pkLen = colDataGetRowLength(pPkColDataInfo, rowId);
32✔
2979
    }
2980
    bool update =
2981
        pInfo->stateStore.updateInfoIsUpdated(pInfo->pUpdateInfo, pBlock->info.id.uid, tsCol[rowId], pPkVal, pkLen);
10,004,414✔
2982
    bool isDeleted =
10,003,401✔
2983
        isClosed && isSignleIntervalWindow(pInfo) &&
9,994,125!
UNCOV
2984
        isDeletedStreamWindow(&win, pBlock->info.id.groupId, pInfo->pState, &pInfo->twAggSup, &pInfo->stateStore);
×
2985
    if ((update || isDeleted) && out) {
10,003,401!
2986
      qDebug("stream update check not pass, update %d, deleted Win %d", update, isDeleted);
1,058✔
2987
      uint64_t gpId = 0;
1,058✔
2988
      code = appendPkToSpecialBlock(pInfo->pUpdateDataRes, tsCol, pPkColDataInfo, rowId, &pBlock->info.id.uid, &gpId,
1,058✔
2989
                                    NULL);
2990
      QUERY_CHECK_CODE(code, lino, _end);
1,058!
2991
      if (isDeleted && pInfo->partitionSup.needCalc) {
1,058!
2992
        gpId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pBlock, rowId);
×
2993
        code = appendPkToSpecialBlock(pInfo->pUpdateDataRes, tsCol, pPkColDataInfo, rowId, &pBlock->info.id.uid, &gpId,
×
2994
                                      NULL);
2995
        QUERY_CHECK_CODE(code, lino, _end);
×
2996
      }
2997
    }
2998
  }
2999
  if (out && pInfo->pUpdateDataRes->info.rows > 0) {
14,671✔
3000
    pInfo->pUpdateDataRes->info.version = pBlock->info.version;
972✔
3001
    pInfo->pUpdateDataRes->info.dataLoad = 1;
972✔
3002
    code = blockDataUpdateTsWindow(pInfo->pUpdateDataRes, 0);
972✔
3003
    QUERY_CHECK_CODE(code, lino, _end);
972!
3004
    pInfo->pUpdateDataRes->info.type = pInfo->partitionSup.needCalc ? STREAM_DELETE_DATA : STREAM_CLEAR;
972✔
3005
  }
3006

3007
_end:
13,699✔
3008
  if (code != TSDB_CODE_SUCCESS) {
14,671!
3009
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3010
  }
3011
  return code;
12,840✔
3012
}
3013

3014
int32_t doBlockDataWindowFilter(SSDataBlock* pBlock, int32_t tsIndex, STimeWindow* pWindow, const char* id) {
522,138✔
3015
  int32_t code = TSDB_CODE_SUCCESS;
522,138✔
3016
  int32_t lino = 0;
522,138✔
3017
  bool*   p = NULL;
522,138✔
3018
  if (pWindow->skey != INT64_MIN || pWindow->ekey != INT64_MAX) {
522,138✔
3019
    p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
878!
3020
    QUERY_CHECK_NULL(p, code, lino, _end, terrno);
939!
3021
    bool hasUnqualified = false;
939✔
3022

3023
    SColumnInfoData* pCol = taosArrayGet(pBlock->pDataBlock, tsIndex);
939✔
3024

3025
    if (pWindow->skey != INT64_MIN) {
939✔
3026
      qDebug("%s filter for additional history window, skey:%" PRId64, id, pWindow->skey);
579!
3027

3028
      QUERY_CHECK_NULL(pCol->pData, code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
579!
3029
      for (int32_t i = 0; i < pBlock->info.rows; ++i) {
1,158✔
3030
        int64_t* ts = (int64_t*)colDataGetData(pCol, i);
579!
3031
        p[i] = (*ts >= pWindow->skey);
579✔
3032

3033
        if (!p[i]) {
579✔
3034
          hasUnqualified = true;
539✔
3035
        }
3036
      }
3037
    } else if (pWindow->ekey != INT64_MAX) {
360!
3038
      qDebug("%s filter for additional history window, ekey:%" PRId64, id, pWindow->ekey);
360!
3039
      for (int32_t i = 0; i < pBlock->info.rows; ++i) {
720✔
3040
        int64_t* ts = (int64_t*)colDataGetData(pCol, i);
360!
3041
        p[i] = (*ts <= pWindow->ekey);
360✔
3042

3043
        if (!p[i]) {
360✔
3044
          hasUnqualified = true;
40✔
3045
        }
3046
      }
3047
    }
3048

3049
    if (hasUnqualified) {
939✔
3050
      code = trimDataBlock(pBlock, pBlock->info.rows, p);
579✔
3051
      QUERY_CHECK_CODE(code, lino, _end);
579!
3052
    }
3053
  }
3054

3055
_end:
522,199✔
3056
  taosMemoryFree(p);
522,199!
3057
  if (code != TSDB_CODE_SUCCESS) {
522,176!
3058
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3059
  }
3060
  return code;
522,160✔
3061
}
3062

3063
static int32_t doBlockDataPrimaryKeyFilter(SSDataBlock* pBlock, STqOffsetVal* offset) {
×
3064
  int32_t code = TSDB_CODE_SUCCESS;
×
3065
  int32_t lino = 0;
×
3066
  if (pBlock->info.window.skey != offset->ts || offset->primaryKey.type == 0) {
×
3067
    return code;
×
3068
  }
3069
  bool* p = taosMemoryCalloc(pBlock->info.rows, sizeof(bool));
×
3070
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
×
3071
  bool hasUnqualified = false;
×
3072

3073
  SColumnInfoData* pColTs = taosArrayGet(pBlock->pDataBlock, 0);
×
3074
  SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
3075

3076
  qDebug("doBlockDataWindowFilter primary key, ts:%" PRId64 " %" PRId64, offset->ts,
×
3077
         VALUE_GET_TRIVIAL_DATUM(&offset->primaryKey));
3078
  QUERY_CHECK_CONDITION((pColPk->info.type == offset->primaryKey.type), code, lino, _end,
×
3079
                        TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
3080

3081
  __compar_fn_t func = getComparFunc(pColPk->info.type, 0);
×
3082
  QUERY_CHECK_NULL(func, code, lino, _end, terrno);
×
3083
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
×
3084
    int64_t* ts = (int64_t*)colDataGetData(pColTs, i);
×
3085
    void*    data = colDataGetData(pColPk, i);
×
3086
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
3087
      void* tmq = taosMemoryMalloc(offset->primaryKey.nData + VARSTR_HEADER_SIZE);
×
3088
      QUERY_CHECK_NULL(tmq, code, lino, _end, terrno);
×
3089
      memcpy(varDataVal(tmq), offset->primaryKey.pData, offset->primaryKey.nData);
×
3090
      varDataLen(tmq) = offset->primaryKey.nData;
×
3091
      p[i] = (*ts > offset->ts) || (func(data, tmq) > 0);
×
3092
      taosMemoryFree(tmq);
×
3093
    } else {
3094
      p[i] = (*ts > offset->ts) || (func(data, VALUE_GET_DATUM(&offset->primaryKey, pColPk->info.type)) > 0);
×
3095
    }
3096

3097
    if (!p[i]) {
×
3098
      hasUnqualified = true;
×
3099
    }
3100
  }
3101

3102
  if (hasUnqualified) {
×
3103
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
×
3104
    QUERY_CHECK_CODE(code, lino, _end);
×
3105
  }
3106

3107
_end:
×
3108
  taosMemoryFree(p);
×
3109
  if (code != TSDB_CODE_SUCCESS) {
×
3110
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3111
  }
3112
  return code;
×
3113
}
3114

3115
// re-build the delete block, ONLY according to the split timestamp
3116
int32_t rebuildDeleteBlockData(SSDataBlock* pBlock, STimeWindow* pWindow, const char* id) {
985✔
3117
  int32_t code = TSDB_CODE_SUCCESS;
985✔
3118
  int32_t lino = 0;
985✔
3119
  int32_t numOfRows = pBlock->info.rows;
985✔
3120
  bool*   p = taosMemoryCalloc(numOfRows, sizeof(bool));
985!
3121
  QUERY_CHECK_NULL(p, code, lino, _end, terrno);
985!
3122

3123
  bool    hasUnqualified = false;
985✔
3124
  int64_t skey = pWindow->skey;
985✔
3125
  int64_t ekey = pWindow->ekey;
985✔
3126

3127
  SColumnInfoData* pSrcStartCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
985✔
3128
  uint64_t*        tsStartCol = (uint64_t*)pSrcStartCol->pData;
985✔
3129
  SColumnInfoData* pSrcEndCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
985✔
3130
  uint64_t*        tsEndCol = (uint64_t*)pSrcEndCol->pData;
985✔
3131

3132
  if (pWindow->skey != INT64_MIN) {
985!
3133
    for (int32_t i = 0; i < numOfRows; i++) {
×
3134
      if (tsStartCol[i] < skey) {
×
3135
        tsStartCol[i] = skey;
×
3136
      }
3137

3138
      if (tsEndCol[i] >= skey) {
×
3139
        p[i] = true;
×
3140
      } else {  // this row should be removed, since it is not in this query time window, which is [skey, INT64_MAX]
3141
        hasUnqualified = true;
×
3142
      }
3143
    }
3144
  } else if (pWindow->ekey != INT64_MAX) {
985!
3145
    for (int32_t i = 0; i < numOfRows; ++i) {
×
3146
      if (tsEndCol[i] > ekey) {
×
3147
        tsEndCol[i] = ekey;
×
3148
      }
3149

3150
      if (tsStartCol[i] <= ekey) {
×
3151
        p[i] = true;
×
3152
      } else {
3153
        hasUnqualified = true;
×
3154
      }
3155
    }
3156
  }
3157

3158
  if (hasUnqualified) {
985!
3159
    code = trimDataBlock(pBlock, pBlock->info.rows, p);
×
3160
    qDebug("%s re-build delete datablock, start key revised to:%" PRId64 ", rows:%" PRId64, id, skey,
×
3161
           pBlock->info.rows);
3162
    QUERY_CHECK_CODE(code, lino, _end);
×
3163
  } else {
3164
    qDebug("%s not update the delete block", id);
985!
3165
  }
3166

3167
_end:
×
3168
  taosMemoryFree(p);
985!
3169
  if (code != TSDB_CODE_SUCCESS) {
985!
3170
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3171
  }
3172
  return code;
985✔
3173
}
3174

3175
int32_t colIdComparFn(const void* param1, const void* param2) {
4,039,430✔
3176
  int32_t p1 = *(int32_t*)param1;
4,039,430✔
3177
  int32_t p2 = *(int32_t*)param2;
4,039,430✔
3178

3179
  if (p1 == p2) {
4,039,430!
3180
    return 0;
×
3181
  } else {
3182
    return (p1 < p2) ? -1 : 1;
4,039,430✔
3183
  }
3184
}
3185

3186
int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock, STimeWindow* pTimeWindow, bool filter) {
522,182✔
3187
  int32_t         code = TSDB_CODE_SUCCESS;
522,182✔
3188
  int32_t         lino = 0;
522,182✔
3189
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
522,182✔
3190
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
522,182✔
3191
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
522,182✔
3192
  const char*     id = GET_TASKID(pTaskInfo);
522,182✔
3193
  bool            isVtableSourceScan = (pTaskInfo->pSubplan->pVTables != NULL);
522,182✔
3194

3195
  code = blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
522,182✔
3196
  QUERY_CHECK_CODE(code, lino, _end);
522,199!
3197

3198
  pBlockInfo->rows = pBlock->info.rows;
522,199✔
3199
  pBlockInfo->id.uid = pBlock->info.id.uid;
522,199✔
3200
  pBlockInfo->type = STREAM_NORMAL;
522,199✔
3201
  pBlockInfo->version = pBlock->info.version;
522,199✔
3202

3203
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
522,199✔
3204
  if (!isVtableSourceScan) {
522,199✔
3205
    pBlockInfo->id.groupId = tableListGetTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
522,187✔
3206
  } else {
3207
    // use original table uid as groupId for vtable
3208
    pBlockInfo->id.groupId = pBlock->info.id.groupId;
12✔
3209
  }
3210

3211
  SArray* pColList = taosArrayInit(4, sizeof(int32_t));
522,230✔
3212
  QUERY_CHECK_NULL(pColList, code, lino, _end, terrno);
522,219✔
3213

3214
  // todo extract method
3215
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
2,692,712✔
3216
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
2,170,578✔
3217
    if (!pColMatchInfo->needOutput) {
2,170,552✔
3218
      continue;
2✔
3219
    }
3220

3221
    bool colExists = false;
2,170,550✔
3222
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
6,252,831✔
3223
      SColumnInfoData* pResCol = NULL;
6,252,838✔
3224
      code = bdGetColumnInfoData(pBlock, j, &pResCol);
6,252,838✔
3225
      QUERY_CHECK_CODE(code, lino, _end);
6,252,560!
3226

3227
      if (pResCol->info.colId == pColMatchInfo->colId) {
6,252,560✔
3228
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
2,170,279✔
3229
        code = colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
2,170,218✔
3230
        QUERY_CHECK_CODE(code, lino, _end);
2,170,420!
3231

3232
        colExists = true;
2,170,420✔
3233
        void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
2,170,420✔
3234
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
2,170,491!
3235
        break;
2,170,491✔
3236
      }
3237
    }
3238

3239
    // the required column does not exists in submit block, let's set it to be all null value
3240
    if (!colExists) {
2,170,494✔
3241
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
4✔
3242
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
4!
3243
      void* tmp = taosArrayPush(pColList, &pColMatchInfo->dstSlotId);
4✔
3244
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
4!
3245
    }
3246
  }
3247

3248
  // currently only the tbname pseudo column
3249
  if (pInfo->numOfPseudoExpr > 0 && !isVtableSourceScan) {
522,057!
3250
    code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
363,426✔
3251
                                  pBlockInfo->rows, pTaskInfo, &pTableScanInfo->base.metaCache);
363,426✔
3252
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
3253
    if (code) {
363,444!
3254
      QUERY_CHECK_CODE(code, lino, _end);
×
3255
    }
3256

3257
    // reset the error code.
3258
    terrno = 0;
363,444✔
3259

3260
    for (int32_t i = 0; i < pInfo->numOfPseudoExpr; ++i) {
1,339,373✔
3261
      void* tmp = taosArrayPush(pColList, &pInfo->pPseudoExpr[i].base.resSchema.slotId);
975,935✔
3262
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
975,932!
3263
    }
3264
  }
3265

3266
  taosArraySort(pColList, colIdComparFn);
522,069✔
3267

3268
  int32_t i = 0, j = 0;
522,208✔
3269
  while (i < taosArrayGetSize(pColList)) {
3,668,755✔
3270
    int32_t slot1 = *(int32_t*)taosArrayGet(pColList, i);
3,146,487✔
3271
    if (slot1 > j) {
3,146,560✔
3272
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
13✔
3273
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
×
3274
      j += 1;
×
3275
    } else {
3276
      i += 1;
3,146,547✔
3277
      j += 1;
3,146,547✔
3278
    }
3279
  }
3280

3281
  while (j < taosArrayGetSize(pInfo->pRes->pDataBlock)) {
522,744✔
3282
    SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, j);
566✔
3283
    colDataSetNNULL(pDst, 0, pBlockInfo->rows);
566✔
3284
    j += 1;
566✔
3285
  }
3286

3287
  taosArrayDestroy(pColList);
522,167✔
3288

3289
  if (filter) {
522,208✔
3290
    code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
495,045✔
3291
    QUERY_CHECK_CODE(code, lino, _end);
494,975!
3292
  }
3293

3294
  // filter the block extracted from WAL files, according to the time window apply additional time window filter
3295
  code = doBlockDataWindowFilter(pInfo->pRes, pInfo->primaryTsIndex, pTimeWindow, id);
522,138✔
3296
  QUERY_CHECK_CODE(code, lino, _end);
522,151!
3297
  pInfo->pRes->info.dataLoad = 1;
522,151✔
3298

3299
  code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
522,151✔
3300
  QUERY_CHECK_CODE(code, lino, _end);
522,194!
3301
  if (pInfo->pRes->info.rows == 0) {
522,194✔
3302
    return 0;
2,909✔
3303
  }
3304

3305
  code = calBlockTbName(pInfo, pInfo->pRes, 0);
519,285✔
3306
  QUERY_CHECK_CODE(code, lino, _end);
519,202!
3307

3308
_end:
519,202✔
3309
  if (code != TSDB_CODE_SUCCESS) {
519,202!
3310
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3311
  }
3312
  return code;
519,200✔
3313
}
3314

3315
static int32_t processPrimaryKey(SSDataBlock* pBlock, bool hasPrimaryKey, STqOffsetVal* offset) {
22,832✔
3316
  int32_t code = TSDB_CODE_SUCCESS;
22,832✔
3317
  int32_t lino = 0;
22,832✔
3318
  SValue  val = {0};
22,832✔
3319
  if (hasPrimaryKey) {
22,832!
3320
    code = doBlockDataPrimaryKeyFilter(pBlock, offset);
×
3321
    if (code != TSDB_CODE_SUCCESS) {
×
3322
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
3323
      return code;
×
3324
    }
3325
    SColumnInfoData* pColPk = taosArrayGet(pBlock->pDataBlock, 1);
×
3326

3327
    if (pBlock->info.rows < 1) {
×
3328
      return code;
×
3329
    }
3330
    void* tmp = colDataGetData(pColPk, pBlock->info.rows - 1);
×
3331
    val.type = pColPk->info.type;
×
3332
    if (IS_VAR_DATA_TYPE(pColPk->info.type)) {
×
3333
      val.pData = taosMemoryMalloc(varDataLen(tmp));
×
3334
      QUERY_CHECK_NULL(val.pData, code, lino, _end, terrno);
×
3335
      val.nData = varDataLen(tmp);
×
3336
      memcpy(val.pData, varDataVal(tmp), varDataLen(tmp));
×
3337
    } else {
3338
      valueSetDatum(&val, pColPk->info.type, tmp, pColPk->info.bytes);
×
3339
    }
3340
  }
3341
  tqOffsetResetToData(offset, pBlock->info.id.uid, pBlock->info.window.ekey, val);
45,678!
3342

3343
_end:
22,846✔
3344
  if (code != TSDB_CODE_SUCCESS) {
22,846!
3345
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3346
  }
3347
  return code;
22,845✔
3348
}
3349

3350
static int32_t doQueueScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
595,531✔
3351
  int32_t        code = TSDB_CODE_SUCCESS;
595,531✔
3352
  int32_t        lino = 0;
595,531✔
3353
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
595,531✔
3354
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
595,531✔
3355

3356
  SStreamScanInfo* pInfo = pOperator->info;
595,531✔
3357
  const char*      id = GET_TASKID(pTaskInfo);
595,531✔
3358

3359
  qDebug("start to exec queue scan, %s", id);
595,531✔
3360

3361
  if (isTaskKilled(pTaskInfo)) {
595,532!
3362
    (*ppRes) = NULL;
×
3363
    return pTaskInfo->code;
×
3364
  }
3365

3366
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
595,544✔
3367
    while (1) {
×
3368
      SSDataBlock* pResult = NULL;
20,534✔
3369
      code = doTableScanNext(pInfo->pTableScanOp, &pResult);
20,534✔
3370
      QUERY_CHECK_CODE(code, lino, _end);
20,511!
3371

3372
      if (pResult && pResult->info.rows > 0) {
20,511!
3373
        bool hasPrimaryKey = pAPI->tqReaderFn.tqGetTablePrimaryKey(pInfo->tqReader);
20,311✔
3374
        code = processPrimaryKey(pResult, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
20,325✔
3375
        QUERY_CHECK_CODE(code, lino, _end);
20,315!
3376
        qDebug("tmqsnap doQueueScan get data utid:%" PRId64, pResult->info.id.uid);
20,315✔
3377
        if (pResult->info.rows > 0) {
20,321!
3378
          (*ppRes) = pResult;
20,321✔
3379
          return code;
20,321✔
3380
        }
3381
      } else {
3382
        break;
3383
      }
3384
    }
3385

3386
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
200✔
3387
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
200✔
3388

3389
    pTSInfo->base.dataReader = NULL;
201✔
3390
    int64_t validVer = pTaskInfo->streamInfo.snapshotVer + 1;
201✔
3391
    qDebug("queue scan tsdb over, switch to wal ver %" PRId64, validVer);
201✔
3392
    if (pAPI->tqReaderFn.tqReaderSeek(pInfo->tqReader, validVer, pTaskInfo->id.str) < 0) {
201✔
3393
      (*ppRes) = NULL;
107✔
3394
      return code;
107✔
3395
    }
3396

3397
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, validVer);
94✔
3398
  }
3399

3400
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
575,104!
3401
    while (1) {
2,329✔
3402
      bool hasResult =
3403
          pAPI->tqReaderFn.tqReaderNextBlockInWal(pInfo->tqReader, id, pTaskInfo->streamInfo.sourceExcluded);
577,433✔
3404

3405
      SSDataBlock*       pRes = pAPI->tqReaderFn.tqGetResultBlock(pInfo->tqReader);
577,415✔
3406
      struct SWalReader* pWalReader = pAPI->tqReaderFn.tqReaderGetWalReader(pInfo->tqReader);
577,412✔
3407

3408
      // curVersion move to next
3409
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pWalReader->curVersion);
577,410✔
3410

3411
      // use ts to pass time when replay, because ts not used if type is log
3412
      pTaskInfo->streamInfo.currentOffset.ts = pAPI->tqReaderFn.tqGetResultBlockTime(pInfo->tqReader);
577,410✔
3413

3414
      if (hasResult) {
577,412✔
3415
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, pRes->info.rows,
495,041✔
3416
               pTaskInfo->streamInfo.currentOffset.version);
3417
        blockDataCleanup(pInfo->pRes);
495,050✔
3418
        STimeWindow defaultWindow = {.skey = INT64_MIN, .ekey = INT64_MAX};
495,000✔
3419
        code = setBlockIntoRes(pInfo, pRes, &defaultWindow, true);
495,000✔
3420
        QUERY_CHECK_CODE(code, lino, _end);
494,914!
3421
        qDebug("doQueueScan after filter get data from log %" PRId64 " rows, version:%" PRId64, pInfo->pRes->info.rows,
494,914✔
3422
               pTaskInfo->streamInfo.currentOffset.version);
3423
        if (pInfo->pRes->info.rows > 0) {
494,942✔
3424
          (*ppRes) = pInfo->pRes;
492,613✔
3425
          return code;
492,613✔
3426
        }
3427
      } else {
3428
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
82,371✔
3429
        (*ppRes) = NULL;
82,371✔
3430
        return code;
82,371✔
3431
      }
3432
    }
3433
  } else {
3434
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
×
3435
    (*ppRes) = NULL;
×
3436
    return code;
×
3437
  }
3438

3439
_end:
×
3440
  if (code != TSDB_CODE_SUCCESS) {
×
3441
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3442
    pTaskInfo->code = code;
×
3443
    T_LONG_JMP(pTaskInfo->env, code);
×
3444
  }
3445
  (*ppRes) = NULL;
×
3446
  return code;
×
3447
}
3448

3449
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
×
3450
  SSDataBlock* pRes = NULL;
×
3451
  int32_t      code = doQueueScanNext(pOperator, &pRes);
×
3452
  return pRes;
×
3453
}
3454

3455
int32_t filterDelBlockByUid(SSDataBlock* pDst, const SSDataBlock* pSrc, STqReader* pReader, SStoreTqReader* pReaderFn) {
985✔
3456
  int32_t    code = TSDB_CODE_SUCCESS;
985✔
3457
  int32_t    lino = 0;
985✔
3458
  int32_t    rows = pSrc->info.rows;
985✔
3459
  code = blockDataEnsureCapacity(pDst, rows);
985✔
3460
  QUERY_CHECK_CODE(code, lino, _end);
985!
3461

3462
  SColumnInfoData* pSrcStartCol = taosArrayGet(pSrc->pDataBlock, START_TS_COLUMN_INDEX);
985✔
3463
  uint64_t*        startCol = (uint64_t*)pSrcStartCol->pData;
985✔
3464
  SColumnInfoData* pSrcEndCol = taosArrayGet(pSrc->pDataBlock, END_TS_COLUMN_INDEX);
985✔
3465
  uint64_t*        endCol = (uint64_t*)pSrcEndCol->pData;
985✔
3466
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrc->pDataBlock, UID_COLUMN_INDEX);
985✔
3467
  uint64_t*        uidCol = (uint64_t*)pSrcUidCol->pData;
985✔
3468

3469
  SColumnInfoData* pDstStartCol = taosArrayGet(pDst->pDataBlock, START_TS_COLUMN_INDEX);
985✔
3470
  SColumnInfoData* pDstEndCol = taosArrayGet(pDst->pDataBlock, END_TS_COLUMN_INDEX);
985✔
3471
  SColumnInfoData* pDstUidCol = taosArrayGet(pDst->pDataBlock, UID_COLUMN_INDEX);
985✔
3472

3473
  int32_t j = 0;
985✔
3474
  for (int32_t i = 0; i < rows; i++) {
1,977✔
3475
    if (pReaderFn->tqReaderIsQueriedTable(pReader, uidCol[i])) {
992✔
3476
      code = colDataSetVal(pDstStartCol, j, (const char*)&startCol[i], false);
563✔
3477
      QUERY_CHECK_CODE(code, lino, _end);
563!
3478

3479
      code = colDataSetVal(pDstEndCol, j, (const char*)&endCol[i], false);
563✔
3480
      QUERY_CHECK_CODE(code, lino, _end);
563!
3481

3482
      code = colDataSetVal(pDstUidCol, j, (const char*)&uidCol[i], false);
563✔
3483
      QUERY_CHECK_CODE(code, lino, _end);
563!
3484

3485
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, GROUPID_COLUMN_INDEX), j);
563✔
3486
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX), j);
563✔
3487
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX), j);
563✔
3488
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, TABLE_NAME_COLUMN_INDEX), j);
563✔
3489
      j++;
563✔
3490
    }
3491
  }
3492

3493
  uint32_t cap = pDst->info.capacity;
985✔
3494
  pDst->info = pSrc->info;
985✔
3495
  pDst->info.rows = j;
985✔
3496
  pDst->info.capacity = cap;
985✔
3497

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

3505
// for partition by tag
3506
int32_t setBlockGroupIdByUid(SStreamScanInfo* pInfo, SSDataBlock* pBlock) {
985✔
3507
  int32_t          code = TSDB_CODE_SUCCESS;
985✔
3508
  int32_t          lino = 0;
985✔
3509
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
985✔
3510
  TSKEY*           startTsCol = (TSKEY*)pStartTsCol->pData;
985✔
3511
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
985✔
3512
  uint64_t*        gpCol = (uint64_t*)pGpCol->pData;
985✔
3513
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
985✔
3514
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
985✔
3515
  int32_t          rows = pBlock->info.rows;
985✔
3516
  if (!pInfo->partitionSup.needCalc) {
985✔
3517
    for (int32_t i = 0; i < rows; i++) {
935✔
3518
      uint64_t groupId = getGroupIdByUid(pInfo, uidCol[i]);
350✔
3519
      code = colDataSetVal(pGpCol, i, (const char*)&groupId, false);
350✔
3520
      QUERY_CHECK_CODE(code, lino, _end);
350!
3521
    }
3522
  }
3523

3524
_end:
985✔
3525
  if (code != TSDB_CODE_SUCCESS) {
985!
3526
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3527
  }
3528
  return code;
985✔
3529
}
3530

3531
int32_t doCheckUpdate(SStreamScanInfo* pInfo, TSKEY endKey, SSDataBlock* pBlock) {
26,586✔
3532
  int32_t code = TSDB_CODE_SUCCESS;
26,586✔
3533
  int32_t lino = 0;
26,586✔
3534
  if (pInfo->pUpdateInfo) {
26,586✔
3535
    pInfo->pUpdateInfo->maxDataVersion = TMAX(pInfo->pUpdateInfo->maxDataVersion, pBlock->info.version);
25,897✔
3536
  }
3537
  if (!pInfo->igCheckUpdate && pInfo->pUpdateInfo) {
26,586✔
3538
    code = checkUpdateData(pInfo, true, pBlock, true);
11,323✔
3539
    QUERY_CHECK_CODE(code, lino, _end);
11,322!
3540
    pInfo->twAggSup.maxTs = TMAX(pInfo->twAggSup.maxTs, endKey);
11,322✔
3541
    if (pInfo->pUpdateDataRes->info.rows > 0) {
11,322✔
3542
      pInfo->updateResIndex = 0;
972✔
3543
      if (pInfo->pUpdateDataRes->info.type == STREAM_CLEAR) {
972✔
3544
        pInfo->scanMode = STREAM_SCAN_FROM_UPDATERES;
797✔
3545
      } else if (pInfo->pUpdateDataRes->info.type == STREAM_INVERT) {
175!
3546
        pInfo->scanMode = STREAM_SCAN_FROM_RES;
×
3547
        // return pInfo->pUpdateDataRes;
3548
      } else if (pInfo->pUpdateDataRes->info.type == STREAM_DELETE_DATA) {
175!
3549
        pInfo->scanMode = STREAM_SCAN_FROM_DELETE_DATA;
175✔
3550
      }
3551
    }
3552
  }
3553

3554
_end:
25,613✔
3555
  if (code != TSDB_CODE_SUCCESS) {
26,585!
3556
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3557
  }
3558
  return code;
26,587✔
3559
}
3560

3561
int32_t streamScanOperatorEncode(SStreamScanInfo* pInfo, void** pBuff, int32_t* pLen) {
36✔
3562
  int32_t   code = TSDB_CODE_SUCCESS;
36✔
3563
  int32_t   lino = 0;
36✔
3564
  int32_t   len = 0;
36✔
3565
  SEncoder* pEnCoder = NULL;
36✔
3566
  SEncoder* pScanEnCoder = NULL;
36✔
3567

3568
  len += encodeSTimeWindowAggSupp(NULL, &pInfo->twAggSup);
36✔
3569
  SEncoder encoder = {0};
36✔
3570
  pEnCoder = &encoder;
36✔
3571
  tEncoderInit(pEnCoder, NULL, 0);
36✔
3572
  if (tStartEncode(pEnCoder) != 0) {
36!
3573
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3574
    QUERY_CHECK_CODE(code, lino, _end);
×
3575
  }
3576
  code = pInfo->stateStore.updateInfoSerialize(pEnCoder, pInfo->pUpdateInfo);
36✔
3577
  QUERY_CHECK_CODE(code, lino, _end);
36!
3578

3579
  if (tEncodeI64(pEnCoder, pInfo->lastScanRange.skey) < 0) {
72!
3580
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3581
    QUERY_CHECK_CODE(code, lino, _end);
×
3582
  }
3583
  if (tEncodeI64(pEnCoder, pInfo->lastScanRange.ekey) < 0) {
72!
3584
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3585
    QUERY_CHECK_CODE(code, lino, _end);
×
3586
  }
3587

3588
  tEndEncode(pEnCoder);
36✔
3589
  len += encoder.pos;
36✔
3590
  tEncoderClear(pEnCoder);
36✔
3591
  pEnCoder = NULL;
36✔
3592

3593
  *pBuff = taosMemoryCalloc(1, len);
36!
3594
  if (!(*pBuff)) {
36!
3595
    code = terrno;
×
3596
    QUERY_CHECK_CODE(code, lino, _end);
×
3597
  }
3598
  void* buf = *pBuff;
36✔
3599
  int32_t stwLen = encodeSTimeWindowAggSupp(&buf, &pInfo->twAggSup);
36✔
3600

3601
  SEncoder scanEncoder = {0};
36✔
3602
  pScanEnCoder = &scanEncoder;
36✔
3603
  tEncoderInit(pScanEnCoder, buf, len - stwLen);
36✔
3604
  if (tStartEncode(pScanEnCoder) != 0) {
36!
3605
    code = TSDB_CODE_FAILED;
×
3606
    QUERY_CHECK_CODE(code, lino, _end);
×
3607
  }
3608
  code = pInfo->stateStore.updateInfoSerialize(pScanEnCoder, pInfo->pUpdateInfo);
36✔
3609
  QUERY_CHECK_CODE(code, lino, _end);
36!
3610

3611
  if (tEncodeI64(pScanEnCoder, pInfo->lastScanRange.skey) < 0) {
72!
3612
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3613
    QUERY_CHECK_CODE(code, lino, _end);
×
3614
  }
3615
  if (tEncodeI64(pScanEnCoder, pInfo->lastScanRange.ekey) < 0) {
72!
3616
    code = TSDB_CODE_STREAM_INTERNAL_ERROR;
×
3617
    QUERY_CHECK_CODE(code, lino, _end);
×
3618
  }
3619

3620
  qDebug("%s last scan range %d. %" PRId64 ",%" PRId64, __func__, __LINE__, pInfo->lastScanRange.skey, pInfo->lastScanRange.ekey);
36✔
3621

3622
  *pLen = len;
36✔
3623

3624
_end:
36✔
3625
  if (code != TSDB_CODE_SUCCESS) {
36!
3626
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3627
  }
3628
  if (pEnCoder != NULL) {
36!
3629
    tEndEncode(pEnCoder);
×
3630
    tEncoderClear(pEnCoder);
×
3631
  }
3632
  if (pScanEnCoder != NULL) {
36!
3633
    tEndEncode(pScanEnCoder);
36✔
3634
    tEncoderClear(pScanEnCoder);
36✔
3635
  }
3636
  return code;
36✔
3637
}
3638

3639
void streamScanOperatorSaveCheckpoint(SStreamScanInfo* pInfo) {
938✔
3640
  int32_t code = TSDB_CODE_SUCCESS;
938✔
3641
  int32_t lino = 0;
938✔
3642
  if (!pInfo->pState) {
938!
3643
    return;
×
3644
  }
3645
  if (needSaveStreamOperatorInfo(&pInfo->basic)) {
938✔
3646
    void*   pBuf = NULL;
36✔
3647
    int32_t len = 0;
36✔
3648
    code = streamScanOperatorEncode(pInfo, &pBuf, &len);
36✔
3649
    QUERY_CHECK_CODE(code, lino, _end);
36!
3650
    pInfo->stateStore.streamStateSaveInfo(pInfo->pState, STREAM_SCAN_OP_CHECKPOINT_NAME,
36✔
3651
                                          strlen(STREAM_SCAN_OP_CHECKPOINT_NAME), pBuf, len);
3652
    taosMemoryFree(pBuf);
36!
3653
    saveStreamOperatorStateComplete(&pInfo->basic);
36✔
3654
  }
3655

3656
_end:
902✔
3657
  if (code != TSDB_CODE_SUCCESS) {
938!
3658
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3659
  }
3660
}
3661

3662
// other properties are recovered from the execution plan
3663
void streamScanOperatorDecode(void* pBuff, int32_t len, SStreamScanInfo* pInfo) {
9✔
3664
  int32_t   code = TSDB_CODE_SUCCESS;
9✔
3665
  int32_t   lino = 0;
9✔
3666
  SDecoder* pDeCoder = NULL;
9✔
3667
  if (!pBuff || len == 0) {
9!
3668
    lino = __LINE__;
×
3669
    goto _end;
×
3670
  }
3671
  void* buf = pBuff;
9✔
3672
  buf = decodeSTimeWindowAggSupp(buf, &pInfo->twAggSup);
9✔
3673
  int32_t tlen = len - encodeSTimeWindowAggSupp(NULL, &pInfo->twAggSup);
9✔
3674
  if (tlen == 0) {
9!
3675
    lino = __LINE__;
×
3676
    goto _end;
×
3677
  }
3678

3679
  SDecoder decoder = {0};
9✔
3680
  pDeCoder = &decoder;
9✔
3681
  tDecoderInit(pDeCoder, buf, tlen);
9✔
3682
  if (tStartDecode(pDeCoder) < 0) {
9!
3683
    lino = __LINE__;
×
3684
    goto _end;
×
3685
  }
3686

3687
  void* pUpInfo = taosMemoryCalloc(1, sizeof(SUpdateInfo));
9!
3688
  if (!pUpInfo) {
9!
3689
    lino = __LINE__;
×
3690
    goto _end;
×
3691
  }
3692
  code = pInfo->stateStore.updateInfoDeserialize(pDeCoder, pUpInfo);
9✔
3693
  if (code == TSDB_CODE_SUCCESS) {
9✔
3694
    pInfo->stateStore.updateInfoDestroy(pInfo->pUpdateInfo);
8✔
3695
    pInfo->pUpdateInfo = pUpInfo;
8✔
3696
    qDebug("%s line:%d. stream scan updateinfo deserialize success", __func__, __LINE__);
8✔
3697
  } else {
3698
    taosMemoryFree(pUpInfo);
1!
3699
    code = TSDB_CODE_SUCCESS;
1✔
3700
    qDebug("%s line:%d. stream scan did not have updateinfo", __func__, __LINE__);
1!
3701
  }
3702

3703
  if (tDecodeIsEnd(pDeCoder)) {
9!
3704
    lino = __LINE__;
×
3705
    goto _end;
×
3706
  }
3707

3708
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.skey);
9✔
3709
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.ekey);
9✔
3710

3711
  if (tDecodeI64(pDeCoder, &pInfo->lastScanRange.skey) < 0) {
18!
3712
    lino = __LINE__;
×
3713
    goto _end;
×
3714
  }
3715

3716
  if (tDecodeI64(pDeCoder, &pInfo->lastScanRange.ekey) < 0) {
18!
3717
    lino = __LINE__;
×
3718
    goto _end;
×
3719
  }
3720
  qDebug("%s last scan range %d. %" PRId64 ",%" PRId64, __func__, __LINE__, pInfo->lastScanRange.skey, pInfo->lastScanRange.ekey);
9✔
3721

3722
_end:
3✔
3723
  if (pDeCoder != NULL) {
9!
3724
    tEndDecode(pDeCoder);
9✔
3725
    tDecoderClear(pDeCoder);
9✔
3726
  }
3727
  qInfo("%s end at line %d", __func__, lino);
9!
3728
}
9✔
3729

3730
bool hasScanRange(SStreamScanInfo* pInfo) {
40,669✔
3731
  SStreamAggSupporter* pSup = pInfo->windowSup.pStreamAggSup;
40,669✔
3732
  return pSup && pSup->pScanBlock->info.rows > 0 && (isStateWindow(pInfo) || isCountWindow(pInfo));
40,669!
3733
}
3734

3735
bool isStreamWindow(SStreamScanInfo* pInfo) {
556✔
3736
  return isIntervalWindow(pInfo) || isSessionWindow(pInfo) || isStateWindow(pInfo) || isCountWindow(pInfo) ||
574✔
3737
         isTimeSlice(pInfo);
18✔
3738
}
3739

3740
int32_t copyGetResultBlock(SSDataBlock* dest, TSKEY start, TSKEY end) {
4,330✔
3741
  int32_t code = blockDataEnsureCapacity(dest, 1);
4,330✔
3742
  if (code != TSDB_CODE_SUCCESS) {
4,330!
3743
    return code;
×
3744
  }
3745
  return appendDataToSpecialBlock(dest, &start, &end, NULL, NULL, NULL);
4,330✔
3746
}
3747

UNCOV
3748
int32_t deletePartName(SStateStore* pStore, SStreamState* pState, SSDataBlock* pBlock, int32_t *deleteNum) {
×
UNCOV
3749
  int32_t code = TSDB_CODE_SUCCESS;
×
UNCOV
3750
  int32_t lino = 0;
×
UNCOV
3751
  for (int32_t i = 0; i < pBlock->info.rows; i++) {
×
3752
    // uid is the same as gid
UNCOV
3753
    SColumnInfoData* pGpIdCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
×
UNCOV
3754
    SColumnInfoData* pTbnameCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
×
UNCOV
3755
    int64_t*         gpIdCol = (int64_t*)pGpIdCol->pData;
×
UNCOV
3756
    void*            pParName = NULL;
×
UNCOV
3757
    int32_t          winCode = 0;
×
UNCOV
3758
    code = pStore->streamStateGetParName(pState, gpIdCol[i],
×
3759
                                                   &pParName, false, &winCode);
UNCOV
3760
    if (TSDB_CODE_SUCCESS == code && winCode != 0) {
×
UNCOV
3761
      qDebug("delete stream part Name for:%"PRId64 " not found", gpIdCol[i]);
×
UNCOV
3762
      colDataSetNULL(pTbnameCol, i);
×
UNCOV
3763
      continue;
×
3764
    }
3765
    (*deleteNum)++;
×
3766
    QUERY_CHECK_CODE(code, lino, _end);
×
3767
    char varTbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE + 1] = {0};
×
3768
    varDataSetLen(varTbName, strlen(pParName));
×
3769
    int64_t len = tsnprintf(varTbName + VARSTR_HEADER_SIZE, TSDB_TABLE_NAME_LEN + 1, "%s", pParName);
×
3770
    code = colDataSetVal(pTbnameCol, i, varTbName, false);
×
3771
    qDebug("delete stream part for:%"PRId64 " res tb: %s", gpIdCol[i], (char*)pParName);
×
3772
    pStore->streamStateFreeVal(pParName);
×
3773
    QUERY_CHECK_CODE(code, lino, _end);
×
3774
    code = pStore->streamStateDeleteParName(pState, gpIdCol[i]);
×
3775
    QUERY_CHECK_CODE(code, lino, _end);
×
3776
    pBlock->info.id.groupId = gpIdCol[i];
×
3777
    // currently, only one valid row in pBlock
3778
    memcpy(pBlock->info.parTbName, varTbName + VARSTR_HEADER_SIZE, TSDB_TABLE_NAME_LEN + 1);
×
3779
  }
3780

UNCOV
3781
_end:
×
UNCOV
3782
  if (code != TSDB_CODE_SUCCESS) {
×
3783
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
3784
  }
UNCOV
3785
  return code;
×
3786
}
3787

3788
static int32_t doStreamScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
64,083✔
3789
  // NOTE: this operator does never check if current status is done or not
3790
  int32_t        code = TSDB_CODE_SUCCESS;
64,083✔
3791
  int32_t        lino = 0;
64,083✔
3792
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
64,083✔
3793
  const char*    id = GET_TASKID(pTaskInfo);
64,083✔
3794

3795
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
64,083✔
3796
  SStreamScanInfo* pInfo = pOperator->info;
64,083✔
3797
  SStreamTaskInfo* pStreamInfo = &pTaskInfo->streamInfo;
64,083✔
3798
  SSHashObj*       pVtableInfos = pTaskInfo->pSubplan->pVTables;
64,083✔
3799

3800
  qDebug("%s stream scan started, %s, recoverStep:%d", __FUNCTION__, id, pStreamInfo->recoverStep);
64,083✔
3801

3802
  if (pVtableInfos != NULL && pStreamInfo->recoverStep != STREAM_RECOVER_STEP__NONE) {
64,084!
3803
    qError("stream vtable source scan should not have recovery step: %d", pStreamInfo->recoverStep);
×
3804
    pStreamInfo->recoverStep = STREAM_RECOVER_STEP__NONE;
×
3805
  }
3806

3807
  if (pVtableInfos != NULL && !pInfo->igCheckUpdate) {
64,084!
3808
    qError("stream vtable source scan should have igCheckUpdate");
×
3809
    pInfo->igCheckUpdate = false;
×
3810
  }
3811

3812
  if (pStreamInfo->recoverStep == STREAM_RECOVER_STEP__PREPARE1 ||
64,084✔
3813
      pStreamInfo->recoverStep == STREAM_RECOVER_STEP__PREPARE2) {
63,193✔
3814
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
2,020✔
3815
    memcpy(&pTSInfo->base.cond, &pStreamInfo->tableCond, sizeof(SQueryTableDataCond));
2,020✔
3816

3817
    if (pStreamInfo->recoverStep == STREAM_RECOVER_STEP__PREPARE1) {
2,020✔
3818
      pTSInfo->base.cond.startVersion = pStreamInfo->fillHistoryVer.minVer;
891✔
3819
      pTSInfo->base.cond.endVersion = pStreamInfo->fillHistoryVer.maxVer;
891✔
3820

3821
      pTSInfo->base.cond.twindows = pStreamInfo->fillHistoryWindow;
891✔
3822
      qDebug("stream scan step1, verRange:%" PRId64 "-%" PRId64 " window:%" PRId64 "-%" PRId64 ", %s",
891✔
3823
             pTSInfo->base.cond.startVersion, pTSInfo->base.cond.endVersion, pTSInfo->base.cond.twindows.skey,
3824
             pTSInfo->base.cond.twindows.ekey, id);
3825
      pStreamInfo->recoverStep = STREAM_RECOVER_STEP__SCAN1;
891✔
3826
      pStreamInfo->recoverScanFinished = false;
891✔
3827
    } else {
3828
      pTSInfo->base.cond.startVersion = pStreamInfo->fillHistoryVer.minVer;
1,129✔
3829
      pTSInfo->base.cond.endVersion = pStreamInfo->fillHistoryVer.maxVer;
1,129✔
3830
      pTSInfo->base.cond.twindows = pStreamInfo->fillHistoryWindow;
1,129✔
3831
      qDebug("stream scan step2 (scan wal), verRange:%" PRId64 " - %" PRId64 ", window:%" PRId64 "-%" PRId64 ", %s",
1,129✔
3832
             pTSInfo->base.cond.startVersion, pTSInfo->base.cond.endVersion, pTSInfo->base.cond.twindows.skey,
3833
             pTSInfo->base.cond.twindows.ekey, id);
3834
      pStreamInfo->recoverStep = STREAM_RECOVER_STEP__NONE;
1,129✔
3835
    }
3836

3837
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
2,020✔
3838

3839
    pTSInfo->base.dataReader = NULL;
2,020✔
3840
    pInfo->pTableScanOp->status = OP_OPENED;
2,020✔
3841

3842
    pTSInfo->scanTimes = 0;
2,020✔
3843
    pTSInfo->currentGroupId = -1;
2,020✔
3844
  }
3845

3846
  if (pStreamInfo->recoverStep == STREAM_RECOVER_STEP__SCAN1) {
64,084✔
3847
    if (isTaskKilled(pTaskInfo)) {
5,861✔
3848
      qInfo("===stream===stream scan is killed. task id:%s, code %s", id, tstrerror(pTaskInfo->code));
1!
3849
      (*ppRes) = NULL;
1✔
3850
      return code;
1✔
3851
    }
3852

3853
    switch (pInfo->scanMode) {
5,860✔
3854
      case STREAM_SCAN_FROM_RES: {
1,087✔
3855
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
1,087✔
3856
        printSpecDataBlock(pInfo->pRecoverRes, getStreamOpName(pOperator->operatorType), "recover",
1,087✔
3857
                           GET_TASKID(pTaskInfo));
1,087✔
3858
        (*ppRes) = pInfo->pRecoverRes;
1,087✔
3859
        return code;
1,087✔
3860
      } break;
3861
      default:
4,773✔
3862
        break;
4,773✔
3863
    }
3864

3865
    code = doTableScanNext(pInfo->pTableScanOp, &pInfo->pRecoverRes);
4,773✔
3866
    QUERY_CHECK_CODE(code, lino, _end);
4,773!
3867

3868
    if (pInfo->pRecoverRes != NULL) {
4,773✔
3869
      code = calBlockTbName(pInfo, pInfo->pRecoverRes, 0);
3,883✔
3870
      QUERY_CHECK_CODE(code, lino, _end);
3,883!
3871
      if (!pInfo->igCheckUpdate && pInfo->pUpdateInfo) {
3,883✔
3872
        TSKEY maxTs = INT64_MIN;
3,073✔
3873
        code = pAPI->stateStore.updateInfoFillBlockData(pInfo->pUpdateInfo, pInfo->pRecoverRes, pInfo->primaryTsIndex,
3,073✔
3874
                                                        pInfo->primaryKeyIndex, &maxTs);
3875
        QUERY_CHECK_CODE(code, lino, _end);
3,073!
3876
        pInfo->twAggSup.maxTs = TMAX(pInfo->twAggSup.maxTs, maxTs);
3,073✔
3877
      }
3878
      if (pInfo->pCreateTbRes->info.rows > 0) {
3,883✔
3879
        pInfo->scanMode = STREAM_SCAN_FROM_RES;
1,087✔
3880
        printSpecDataBlock(pInfo->pCreateTbRes, getStreamOpName(pOperator->operatorType), "recover",
1,087✔
3881
                           GET_TASKID(pTaskInfo));
1,087✔
3882
        (*ppRes) = pInfo->pCreateTbRes;
1,087✔
3883
        return code;
1,087✔
3884
      }
3885

3886
      qDebug("stream recover scan get block, rows %" PRId64, pInfo->pRecoverRes->info.rows);
2,796✔
3887
      printSpecDataBlock(pInfo->pRecoverRes, getStreamOpName(pOperator->operatorType), "recover",
2,796✔
3888
                         GET_TASKID(pTaskInfo));
2,796✔
3889
      (*ppRes) = pInfo->pRecoverRes;
2,796✔
3890
      return code;
2,796✔
3891
    }
3892
    pStreamInfo->recoverStep = STREAM_RECOVER_STEP__NONE;
890✔
3893
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
890✔
3894
    pAPI->tsdReader.tsdReaderClose(pTSInfo->base.dataReader);
890✔
3895

3896
    pTSInfo->base.dataReader = NULL;
890✔
3897

3898
    pTSInfo->base.cond.startVersion = -1;
890✔
3899
    pTSInfo->base.cond.endVersion = -1;
890✔
3900

3901
    pStreamInfo->recoverScanFinished = true;
890✔
3902
    (*ppRes) = NULL;
890✔
3903
    return code;
890✔
3904
  }
3905

3906
  size_t total = taosArrayGetSize(pInfo->pBlockLists);
58,223✔
3907
// TODO: refactor
3908
FETCH_NEXT_BLOCK:
58,657✔
3909
  if (pInfo->blockType == STREAM_INPUT__DATA_BLOCK) {
58,657✔
3910
    if (pVtableInfos != NULL) {
12,301!
3911
      qInfo("stream vtable source scan would ignore all data blocks");
×
3912
      pInfo->validBlockIndex = total;
×
3913
    }
3914
    if (pInfo->validBlockIndex >= total) {
12,301✔
3915
      doClearBufferedBlocks(pInfo);
3916
      (*ppRes) = NULL;
2,401✔
3917
      return code;
2,401✔
3918
    }
3919

3920
    int32_t current = pInfo->validBlockIndex++;
9,900✔
3921
    qDebug("process %d/%d input data blocks, %s", current, (int32_t)total, id);
9,900✔
3922

3923
    SPackedData* pPacked = taosArrayGet(pInfo->pBlockLists, current);
9,900✔
3924
    QUERY_CHECK_NULL(pPacked, code, lino, _end, terrno);
9,900!
3925

3926
    SSDataBlock* pBlock = pPacked->pDataBlock;
9,900✔
3927
    if (pBlock->info.parTbName[0]) {
9,900✔
3928
      code =
3929
          pAPI->stateStore.streamStatePutParName(pStreamInfo->pState, pBlock->info.id.groupId, pBlock->info.parTbName);
57✔
3930
      QUERY_CHECK_CODE(code, lino, _end);
57!
3931
    }
3932

3933
    // TODO move into scan
3934
    pBlock->info.calWin.skey = INT64_MIN;
9,900✔
3935
    pBlock->info.calWin.ekey = INT64_MAX;
9,900✔
3936
    pBlock->info.dataLoad = 1;
9,900✔
3937
    if (pInfo->pUpdateInfo) {
9,900✔
3938
      pInfo->pUpdateInfo->maxDataVersion = TMAX(pInfo->pUpdateInfo->maxDataVersion, pBlock->info.version);
6,694✔
3939
    }
3940

3941
    code = blockDataUpdateTsWindow(pBlock, 0);
9,900✔
3942
    QUERY_CHECK_CODE(code, lino, _end);
9,900!
3943
    switch (pBlock->info.type) {
9,900!
3944
      case STREAM_NORMAL:
2,148✔
3945
      case STREAM_GET_ALL:
3946
        printDataBlock(pBlock, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
2,148✔
3947
        setStreamOperatorState(&pInfo->basic, pBlock->info.type);
2,148✔
3948
        (*ppRes) = pBlock;
2,148✔
3949
        return code;
2,148✔
3950
      case STREAM_RETRIEVE: {
430✔
3951
        pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
430✔
3952
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RETRIEVE;
430✔
3953
        code = copyDataBlock(pInfo->pUpdateRes, pBlock);
430✔
3954
        QUERY_CHECK_CODE(code, lino, _end);
430!
3955
        pInfo->updateResIndex = 0;
430✔
3956
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
430✔
3957
        pAPI->stateStore.updateInfoAddCloseWindowSBF(pInfo->pUpdateInfo);
430✔
3958
      } break;
430✔
3959
      case STREAM_DELETE_DATA: {
985✔
3960
        printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "delete recv", GET_TASKID(pTaskInfo));
985✔
3961
        SSDataBlock* pDelBlock = NULL;
985✔
3962
        if (pInfo->tqReader) {
985!
3963
          code = createSpecialDataBlock(STREAM_DELETE_DATA, &pDelBlock);
985✔
3964
          QUERY_CHECK_CODE(code, lino, _end);
985!
3965

3966
          code = filterDelBlockByUid(pDelBlock, pBlock, pInfo->tqReader, &pInfo->readerFn);
985✔
3967
          QUERY_CHECK_CODE(code, lino, _end);
985!
3968
        } else {
3969
          pDelBlock = pBlock;
×
3970
        }
3971

3972
        code = setBlockGroupIdByUid(pInfo, pDelBlock);
985✔
3973
        QUERY_CHECK_CODE(code, lino, _end);
985!
3974
        code = rebuildDeleteBlockData(pDelBlock, &pStreamInfo->fillHistoryWindow, id);
985✔
3975
        QUERY_CHECK_CODE(code, lino, _end);
985!
3976
        printSpecDataBlock(pDelBlock, getStreamOpName(pOperator->operatorType), "delete recv filtered",
985✔
3977
                           GET_TASKID(pTaskInfo));
985✔
3978
        if (pDelBlock->info.rows == 0) {
985✔
3979
          if (pInfo->tqReader) {
429!
3980
            blockDataDestroy(pDelBlock);
429✔
3981
          }
3982
          goto FETCH_NEXT_BLOCK;
430✔
3983
        }
3984

3985
        if (!isStreamWindow(pInfo)) {
556✔
3986
          code = generateDeleteResultBlock(pInfo, pDelBlock, pInfo->pDeleteDataRes);
3✔
3987
          QUERY_CHECK_CODE(code, lino, _end);
3!
3988
          if (pInfo->partitionSup.needCalc) {
3✔
3989
            pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
2✔
3990
          } else {
3991
            pInfo->pDeleteDataRes->info.type = STREAM_DELETE_RESULT;
1✔
3992
          }
3993
          blockDataDestroy(pDelBlock);
3✔
3994

3995
          if (pInfo->pDeleteDataRes->info.rows > 0) {
3!
3996
            printSpecDataBlock(pInfo->pDeleteDataRes, getStreamOpName(pOperator->operatorType), "delete result",
3✔
3997
                               GET_TASKID(pTaskInfo));
3✔
3998
            setStreamOperatorState(&pInfo->basic, pInfo->pDeleteDataRes->info.type);
3✔
3999
            (*ppRes) = pInfo->pDeleteDataRes;
3✔
4000
            return code;
555✔
4001
          } else {
4002
            goto FETCH_NEXT_BLOCK;
×
4003
          }
4004
        } else {
4005
          pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
553✔
4006
          pInfo->updateResIndex = 0;
553✔
4007
          code = generateScanRange(pInfo, pDelBlock, pInfo->pUpdateRes, STREAM_DELETE_DATA);
553✔
4008
          QUERY_CHECK_CODE(code, lino, _end);
553!
4009
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
553✔
4010
          code = copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
553✔
4011
          QUERY_CHECK_CODE(code, lino, _end);
553!
4012
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
553✔
4013
          if (pInfo->tqReader) {
553!
4014
            blockDataDestroy(pDelBlock);
553✔
4015
          }
4016
          if (pInfo->pDeleteDataRes->info.rows > 0) {
553✔
4017
            pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
552✔
4018
            printSpecDataBlock(pInfo->pDeleteDataRes, getStreamOpName(pOperator->operatorType), "delete result",
552✔
4019
                               GET_TASKID(pTaskInfo));
552✔
4020
            setStreamOperatorState(&pInfo->basic, pInfo->pDeleteDataRes->info.type);
552✔
4021
            (*ppRes) = pInfo->pDeleteDataRes;
552✔
4022
            return code;
552✔
4023
          } else {
4024
            goto FETCH_NEXT_BLOCK;
1✔
4025
          }
4026
        }
4027
      } break;
4028
      case STREAM_GET_RESULT: {
4,330✔
4029
        pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
4,330✔
4030
        pInfo->updateResIndex = 0;
4,330✔
4031
        pInfo->lastScanRange = pBlock->info.window;
4,330✔
4032
        TSKEY endKey = getNextTimeWindowStart(&pInfo->interval, pBlock->info.window.skey, TSDB_ORDER_ASC) - 1;
4,330✔
4033
        if (pInfo->useGetResultRange == true) {
4,330✔
4034
          endKey = pBlock->info.window.ekey;
886✔
4035
        }
4036
        code = copyGetResultBlock(pInfo->pUpdateRes, pBlock->info.window.skey, endKey);
4,330✔
4037
        QUERY_CHECK_CODE(code, lino, _end);
4,328!
4038
        pInfo->pUpdateInfo->maxDataVersion = -1;
4,328✔
4039
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
4,328✔
4040
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
4,330✔
4041
      } break;
4,330✔
UNCOV
4042
      case STREAM_DROP_CHILD_TABLE: {
×
UNCOV
4043
        int32_t deleteNum = 0;
×
UNCOV
4044
        code = deletePartName(&pInfo->stateStore, pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, pBlock, &deleteNum);
×
UNCOV
4045
        QUERY_CHECK_CODE(code, lino, _end);
×
UNCOV
4046
        if (deleteNum == 0) {
×
UNCOV
4047
          printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "block recv", GET_TASKID(pTaskInfo));
×
UNCOV
4048
          qDebug("===stream=== ignore block type 18, delete num is 0");
×
UNCOV
4049
          goto FETCH_NEXT_BLOCK;
×
4050
        }
4051
      } break;
×
4052
      case STREAM_CHECKPOINT: {
×
4053
        qError("stream check point error. msg type: STREAM_INPUT__DATA_BLOCK");
×
4054
      } break;
×
4055
      default:
2,007✔
4056
        break;
2,007✔
4057
    }
4058
    printSpecDataBlock(pBlock, getStreamOpName(pOperator->operatorType), "block recv", GET_TASKID(pTaskInfo));
6,767✔
4059
    setStreamOperatorState(&pInfo->basic, pBlock->info.type);
6,767✔
4060
    (*ppRes) = pBlock;
6,767✔
4061
    return code;
6,767✔
4062
  } else if (pInfo->blockType == STREAM_INPUT__DATA_SUBMIT) {
46,356✔
4063
    qDebug("stream scan mode:%d, %s", pInfo->scanMode, id);
44,477✔
4064
    if (pVtableInfos != NULL && pInfo->scanMode != STREAM_SCAN_FROM_READERHANDLE) {
44,480!
4065
      qError("stream vtable source scan should not have scan mode: %d", pInfo->scanMode);
×
4066
      pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
×
4067
    }
4068
    switch (pInfo->scanMode) {
44,480✔
4069
      case STREAM_SCAN_FROM_RES: {
732✔
4070
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
732✔
4071
        code = doCheckUpdate(pInfo, pInfo->pRes->info.window.ekey, pInfo->pRes);
732✔
4072
        QUERY_CHECK_CODE(code, lino, _end);
732!
4073
        setStreamOperatorState(&pInfo->basic, pInfo->pRes->info.type);
732✔
4074
        code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
732✔
4075
        QUERY_CHECK_CODE(code, lino, _end);
732!
4076
        pInfo->pRes->info.dataLoad = 1;
732✔
4077
        code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
732✔
4078
        QUERY_CHECK_CODE(code, lino, _end);
732!
4079
        if (pInfo->pRes->info.rows > 0) {
732!
4080
          printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
732✔
4081
          (*ppRes) = pInfo->pRes;
732✔
4082
          return code;
732✔
4083
        }
4084
      } break;
×
4085
      case STREAM_SCAN_FROM_DELETE_DATA: {
174✔
4086
        code = generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes, STREAM_PARTITION_DELETE_DATA);
174✔
4087
        QUERY_CHECK_CODE(code, lino, _end);
174!
4088
        if (pInfo->pUpdateRes->info.rows > 0) {
174✔
4089
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
158✔
4090
          pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
158✔
4091
          code = copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
158✔
4092
          QUERY_CHECK_CODE(code, lino, _end);
158!
4093
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
158✔
4094
          (*ppRes) = pInfo->pDeleteDataRes;
158✔
4095
          return code;
158✔
4096
        }
4097
        qError("%s===stream=== %s failed at line %d since pInfo->pUpdateRes is empty", GET_TASKID(pTaskInfo), __func__,
16!
4098
               __LINE__);
4099
        blockDataCleanup(pInfo->pUpdateDataRes);
16✔
4100
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
16✔
4101
      } break;
16✔
4102
      case STREAM_SCAN_FROM_UPDATERES: {
797✔
4103
        code = generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes, STREAM_CLEAR);
797✔
4104
        QUERY_CHECK_CODE(code, lino, _end);
797!
4105
        if (pInfo->pUpdateRes->info.rows > 0) {
797✔
4106
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
781✔
4107
          pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
781✔
4108
          (*ppRes) = pInfo->pUpdateRes;
781✔
4109
          return code;
781✔
4110
        }
4111
        qError("%s===stream=== %s failed at line %d since pInfo->pUpdateRes is empty", GET_TASKID(pTaskInfo), __func__,
16!
4112
               __LINE__);
4113
        blockDataCleanup(pInfo->pUpdateDataRes);
16✔
4114
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
16✔
4115
      } break;
16✔
4116
      case STREAM_SCAN_FROM_DATAREADER_RANGE:
8,416✔
4117
      case STREAM_SCAN_FROM_DATAREADER_RETRIEVE: {
4118
        if (pInfo->pRangeScanRes != NULL) {
8,416✔
4119
          (*ppRes) = pInfo->pRangeScanRes;
17✔
4120
          pInfo->pRangeScanRes = NULL;
17✔
4121
          return code;
2,140✔
4122
        } 
4123
        SSDataBlock* pSDB = NULL;
8,399✔
4124
        code = doRangeScan(pInfo, pInfo->pUpdateRes, pInfo->primaryTsIndex, &pInfo->updateResIndex, &pSDB);
8,399✔
4125
        QUERY_CHECK_CODE(code, lino, _end);
8,399!
4126
        if (pSDB) {
8,399✔
4127
          STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
2,123✔
4128
          pSDB->info.type = pInfo->scanMode == STREAM_SCAN_FROM_DATAREADER_RANGE ? STREAM_NORMAL : STREAM_PULL_DATA;
2,123✔
4129
          if (!pInfo->igCheckUpdate && pInfo->pUpdateInfo) {
2,123!
4130
            code = checkUpdateData(pInfo, true, pSDB, false);
1,518✔
4131
            QUERY_CHECK_CODE(code, lino, _end);
1,518!
4132
          }
4133

4134
          code = calBlockTbName(pInfo, pSDB, 0);
2,123✔
4135
          QUERY_CHECK_CODE(code, lino, _end);
2,123!
4136

4137
          printSpecDataBlock(pSDB, getStreamOpName(pOperator->operatorType), "update", GET_TASKID(pTaskInfo));
2,123✔
4138

4139
          if (pInfo->pCreateTbRes->info.rows > 0) {
2,123✔
4140
            printSpecDataBlock(pInfo->pCreateTbRes, getStreamOpName(pOperator->operatorType), "update",
17✔
4141
                               GET_TASKID(pTaskInfo));
17✔
4142
            (*ppRes) = pInfo->pCreateTbRes;
17✔
4143
            pInfo->pRangeScanRes = pSDB;
17✔
4144
            return code;
17✔
4145
          }
4146

4147
          (*ppRes) = pSDB;
2,106✔
4148
          return code;
2,106✔
4149
        }
4150
        blockDataCleanup(pInfo->pUpdateDataRes);
6,276✔
4151
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
6,276✔
4152
      } break;
6,276✔
4153
      default:
34,361✔
4154
        break;
34,361✔
4155
    }
4156

4157
    if (hasScanRange(pInfo)) {
40,669✔
4158
      pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
25✔
4159
      pInfo->updateResIndex = 0;
25✔
4160
      SStreamAggSupporter* pSup = pInfo->windowSup.pStreamAggSup;
25✔
4161
      code = copyDataBlock(pInfo->pUpdateRes, pSup->pScanBlock);
25✔
4162
      QUERY_CHECK_CODE(code, lino, _end);
25!
4163
      blockDataCleanup(pSup->pScanBlock);
25✔
4164
      prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex, NULL);
25✔
4165
      pInfo->pUpdateRes->info.type = STREAM_DELETE_DATA;
25✔
4166
      printSpecDataBlock(pInfo->pUpdateRes, getStreamOpName(pOperator->operatorType), "rebuild", GET_TASKID(pTaskInfo));
25✔
4167
      (*ppRes) = pInfo->pUpdateRes;
25✔
4168
      return code;
25✔
4169
    }
4170

4171
    SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
40,644✔
4172
    int32_t         totalBlocks = taosArrayGetSize(pInfo->pBlockLists);
40,644✔
4173

4174
  NEXT_SUBMIT_BLK:
40,644✔
4175
    while (1) {
4176
      if (pInfo->readerFn.tqReaderCurrentBlockConsumed(pInfo->tqReader)) {
94,684✔
4177
        if (pInfo->validBlockIndex >= totalBlocks) {
68,103✔
4178
          pAPI->stateStore.updateInfoDestoryColseWinSBF(pInfo->pUpdateInfo);
14,060✔
4179
          doClearBufferedBlocks(pInfo);
4180

4181
          qDebug("stream scan return empty, all %d submit blocks consumed, %s", totalBlocks, id);
14,059✔
4182
          (*ppRes) = NULL;
14,060✔
4183
          return code;
14,060✔
4184
        }
4185

4186
        int32_t      current = pInfo->validBlockIndex++;
54,043✔
4187
        SPackedData* pSubmit = taosArrayGet(pInfo->pBlockLists, current);
54,043✔
4188
        QUERY_CHECK_NULL(pSubmit, code, lino, _end, terrno);
54,043!
4189

4190
        qDebug("set %d/%d as the input submit block, %s", current + 1, totalBlocks, id);
54,043✔
4191
        if (pAPI->tqReaderFn.tqReaderSetSubmitMsg(pInfo->tqReader, pSubmit->msgStr, pSubmit->msgLen, pSubmit->ver, NULL) <
54,042!
4192
            0) {
4193
          qError("submit msg messed up when initializing stream submit block %p, current %d/%d, %s", pSubmit, current,
×
4194
                 totalBlocks, id);
4195
          continue;
×
4196
        }
4197
      }
4198

4199
      blockDataCleanup(pInfo->pRes);
80,623✔
4200

4201
      while (pAPI->tqReaderFn.tqNextBlockImpl(pInfo->tqReader, id)) {
81,205✔
4202
        SSDataBlock* pRes = NULL;
27,167✔
4203

4204
        code = pAPI->tqReaderFn.tqRetrieveBlock(pInfo->tqReader, &pRes, id);
27,167✔
4205
        qDebug("retrieve data from submit completed code:%s rows:%" PRId64 " %s", tstrerror(code), pRes->info.rows, id);
27,166✔
4206

4207
        if (code != TSDB_CODE_SUCCESS || pRes->info.rows == 0) {
27,167!
4208
          qDebug("retrieve data failed, try next block in submit block, %s", id);
×
4209
          continue;
579✔
4210
        }
4211

4212
        code = setBlockIntoRes(pInfo, pRes, &pStreamInfo->fillHistoryWindow, false);
27,167✔
4213
        if (code == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
27,167!
4214
          pInfo->pRes->info.rows = 0;
×
4215
          code = TSDB_CODE_SUCCESS;
×
4216
        } else {
4217
          QUERY_CHECK_CODE(code, lino, _end);
27,167!
4218
        }
4219

4220
        if (pInfo->pRes->info.rows == 0) {
27,167✔
4221
          continue;
579✔
4222
        }
4223

4224
        if (pVtableInfos != NULL && pInfo->pCreateTbRes->info.rows > 0) {
26,588!
4225
          qError("stream vtable source scan should not have create table res");
×
4226
          blockDataCleanup(pInfo->pCreateTbRes);
×
4227
        }
4228

4229
        if (pInfo->pCreateTbRes->info.rows > 0) {
26,588✔
4230
          pInfo->scanMode = STREAM_SCAN_FROM_RES;
732✔
4231
          qDebug("create table res exists, rows:%" PRId64 " return from stream scan, %s",
732!
4232
                 pInfo->pCreateTbRes->info.rows, id);
4233
          (*ppRes) = pInfo->pCreateTbRes;
732✔
4234
          return code;
732✔
4235
        }
4236

4237
        code = doCheckUpdate(pInfo, pBlockInfo->window.ekey, pInfo->pRes);
25,856✔
4238
        QUERY_CHECK_CODE(code, lino, _end);
25,855!
4239
        setStreamOperatorState(&pInfo->basic, pInfo->pRes->info.type);
25,855✔
4240
        if (pVtableInfos == NULL) {
25,851!
4241
          // filter should be applied in merge task for vtables
4242
          code = doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
25,851✔
4243
          QUERY_CHECK_CODE(code, lino, _end);
25,851!
4244
        }
4245

4246
        code = blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
25,851✔
4247
        QUERY_CHECK_CODE(code, lino, _end);
25,852!
4248

4249
        int64_t numOfUpdateRes = pInfo->pUpdateDataRes->info.rows;
25,852✔
4250
        qDebug("%s %" PRId64 " rows in datablock, update res:%" PRId64, id, pBlockInfo->rows, numOfUpdateRes);
25,852✔
4251
        if (pBlockInfo->rows > 0 || numOfUpdateRes > 0) {
25,856✔
4252
          break;
4253
        }
4254
      }
4255

4256
      if (pBlockInfo->rows > 0 || pInfo->pUpdateDataRes->info.rows > 0) {
79,891✔
4257
        break;
4258
      } else {
4259
        continue;
54,040✔
4260
      }
4261
    }
4262

4263
    // record the scan action.
4264
    pInfo->numOfExec++;
25,851✔
4265
    pOperator->resultInfo.totalRows += pBlockInfo->rows;
25,851✔
4266

4267
    qDebug("stream scan completed, and return source rows:%" PRId64 ", %s", pBlockInfo->rows, id);
25,851✔
4268
    if (pBlockInfo->rows > 0) {
25,852✔
4269
      printDataBlock(pInfo->pRes, getStreamOpName(pOperator->operatorType), GET_TASKID(pTaskInfo));
25,848✔
4270
      (*ppRes) = pInfo->pRes;
25,848✔
4271
      return code;
25,848✔
4272
    }
4273

4274
    if (pInfo->pUpdateDataRes->info.rows > 0) {
4!
4275
      goto FETCH_NEXT_BLOCK;
4✔
4276
    }
4277

4278
    goto NEXT_SUBMIT_BLK;
×
4279
  } else if (pInfo->blockType == STREAM_INPUT__CHECKPOINT) {
1,879✔
4280
    if (pInfo->validBlockIndex >= total || pVtableInfos != NULL) {
1,876!
4281
      doClearBufferedBlocks(pInfo);
4282
      (*ppRes) = NULL;
938✔
4283
      return code;
938✔
4284
    }
4285

4286
    int32_t current = pInfo->validBlockIndex++;
938✔
4287
    qDebug("process %d/%d input data blocks, %s", current, (int32_t)total, id);
938✔
4288

4289
    SPackedData* pData = taosArrayGet(pInfo->pBlockLists, current);
938✔
4290
    QUERY_CHECK_NULL(pData, code, lino, _end, terrno);
938!
4291
    SSDataBlock* pBlock = taosArrayGet(pData->pDataBlock, 0);
938✔
4292
    QUERY_CHECK_NULL(pBlock, code, lino, _end, terrno);
938!
4293

4294
    if (pBlock->info.type == STREAM_CHECKPOINT) {
938!
4295
      streamScanOperatorSaveCheckpoint(pInfo);
938✔
4296
    }
4297
    // printDataBlock(pInfo->pCheckpointRes, "stream scan ck", GET_TASKID(pTaskInfo));
4298
    (*ppRes) = pInfo->pCheckpointRes;
938✔
4299
    return code;
938✔
4300
  } else {
4301
    qError("stream scan error, invalid block type %d, %s", pInfo->blockType, id);
3!
4302
    code = TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR;
×
4303
  }
4304

4305
_end:
×
4306
  if (code != TSDB_CODE_SUCCESS) {
×
4307
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4308
    pTaskInfo->code = code;
×
4309
    T_LONG_JMP(pTaskInfo->env, code);
×
4310
  }
4311
  (*ppRes) = NULL;
×
4312
  return code;
×
4313
}
4314

4315
static SSDataBlock* doStreamScan(SOperatorInfo* pOperator) {
×
4316
  SSDataBlock* pRes = NULL;
×
4317
  int32_t      code = doStreamScanNext(pOperator, &pRes);
×
4318
  return pRes;
×
4319
}
4320
int32_t extractTableIdList(const STableListInfo* pTableListInfo, SArray** ppArrayRes) {
4,632✔
4321
  int32_t code = TSDB_CODE_SUCCESS;
4,632✔
4322
  int32_t lino = 0;
4,632✔
4323
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));
4,632✔
4324
  QUERY_CHECK_NULL(tableIdList, code, lino, _end, terrno);
4,647!
4325

4326
  // Transfer the Array of STableKeyInfo into uid list.
4327
  int32_t size = 0;
4,647✔
4328
  code = tableListGetSize(pTableListInfo, &size);
4,647✔
4329
  QUERY_CHECK_CODE(code, lino, _end);
4,647!
4330
  for (int32_t i = 0; i < size; ++i) {
53,410✔
4331
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
48,762✔
4332
    QUERY_CHECK_NULL(pkeyInfo, code, lino, _end, terrno);
48,746!
4333
    void* tmp = taosArrayPush(tableIdList, &pkeyInfo->uid);
48,746✔
4334
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
48,763!
4335
  }
4336

4337
  (*ppArrayRes) = tableIdList;
4,648✔
4338

4339
_end:
4,648✔
4340
  if (code != TSDB_CODE_SUCCESS) {
4,648!
4341
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4342
  }
4343
  return code;
4,646✔
4344
}
4345

4346
static int32_t doRawScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
5,053✔
4347
  int32_t        code = TSDB_CODE_SUCCESS;
5,053✔
4348
  int32_t        lino = 0;
5,053✔
4349
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5,053✔
4350
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
5,053✔
4351

4352
  SStreamRawScanInfo* pInfo = pOperator->info;
5,053✔
4353
  pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = NULL;  // use batchMetaReq != NULL to judge if data is meta
5,053✔
4354
  pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = NULL;
5,053✔
4355

4356
  qDebug("tmqsnap doRawScan called");
5,053✔
4357
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
5,053✔
4358
    bool hasNext = false;
5,036✔
4359
    if (pInfo->dataReader && pInfo->sContext->withMeta != ONLY_META) {
5,036!
4360
      code = pAPI->tsdReader.tsdNextDataBlock(pInfo->dataReader, &hasNext);
5,031✔
4361
      if (code != TSDB_CODE_SUCCESS) {
5,031!
4362
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
4363
        QUERY_CHECK_CODE(code, lino, _end);
×
4364
      }
4365
    }
4366

4367
    if (pInfo->dataReader && hasNext) {
5,036✔
4368
      if (isTaskKilled(pTaskInfo)) {
2,525!
4369
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->dataReader);
×
4370
        return code;
2,524✔
4371
      }
4372

4373
      SSDataBlock* pBlock = NULL;
2,525✔
4374
      code = pAPI->tsdReader.tsdReaderRetrieveDataBlock(pInfo->dataReader, &pBlock, NULL);
2,525✔
4375
      QUERY_CHECK_CODE(code, lino, _end);
2,525!
4376

4377
      if (pBlock && pBlock->info.rows > 0) {
2,525!
4378
        bool hasPrimaryKey = pAPI->snapshotFn.taosXGetTablePrimaryKey(pInfo->sContext);
2,525✔
4379
        code = processPrimaryKey(pBlock, hasPrimaryKey, &pTaskInfo->streamInfo.currentOffset);
2,523✔
4380
        QUERY_CHECK_CODE(code, lino, _end);
2,524!
4381
        qDebug("tmqsnap doRawScan get data uid:%" PRId64, pBlock->info.id.uid);
2,524✔
4382
        (*ppRes) = pBlock;
2,524✔
4383
        return code;
2,524✔
4384
      }
4385
    }
4386

4387
    SMetaTableInfo mtInfo = {0};
2,511✔
4388
    code = pAPI->snapshotFn.getMetaTableInfoFromSnapshot(pInfo->sContext, &mtInfo);
2,511✔
4389
    QUERY_CHECK_CODE(code, lino, _end);
2,509!
4390
    if (code != 0) {
2,509!
4391
      destroyMetaTableInfo(&mtInfo);
4392
      QUERY_CHECK_CODE(code, lino, _end);
×
4393
    }
4394
    STqOffsetVal offset = {0};
2,509✔
4395
    if (mtInfo.uid == 0 || pInfo->sContext->withMeta == ONLY_META) {  // read snapshot done, change to get data from wal
2,509!
4396
      qDebug("tmqsnap read snapshot done, change to get data from wal");
31✔
4397
      tqOffsetResetToLog(&offset, pInfo->sContext->snapVersion + 1);
31✔
4398
    } else {
4399
      SValue val = {0};
2,478✔
4400
      tqOffsetResetToData(&offset, mtInfo.uid, INT64_MIN, val);
2,478✔
4401
      qDebug("tmqsnap change get data uid:%" PRId64, mtInfo.uid);
2,478✔
4402
    }
4403
    destroyMetaTableInfo(&mtInfo);
4404
    code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
2,510✔
4405
    QUERY_CHECK_CODE(code, lino, _end);
2,511!
4406
    (*ppRes) = NULL;
2,511✔
4407
    return code;
2,511✔
4408
  } else if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_META) {
17!
4409
    SSnapContext* sContext = pInfo->sContext;
17✔
4410
    for (int32_t i = 0; i < tmqRowSize; i++) {
148!
4411
      void*   data = NULL;
148✔
4412
      int32_t dataLen = 0;
148✔
4413
      int16_t type = 0;
148✔
4414
      int64_t uid = 0;
148✔
4415
      if (pAPI->snapshotFn.getTableInfoFromSnapshot(sContext, &data, &dataLen, &type, &uid) < 0) {
148!
4416
        qError("tmqsnap getTableInfoFromSnapshot error");
×
4417
        taosMemoryFreeClear(data);
×
4418
        break;
17✔
4419
      }
4420

4421
      if (!sContext->queryMeta) {  // change to get data next poll request
148✔
4422
        STqOffsetVal offset = {0};
17✔
4423
        SValue       val = {0};
17!
4424
        tqOffsetResetToData(&offset, 0, INT64_MIN, val);
17✔
4425
        code = qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
17✔
4426
        QUERY_CHECK_CODE(code, lino, _end);
17!
4427
        break;
17✔
4428
      } else {
4429
        tqOffsetResetToMeta(&pTaskInfo->streamInfo.currentOffset, uid);
131✔
4430
        SMqMetaRsp tmpMetaRsp = {0};
131✔
4431
        tmpMetaRsp.resMsgType = type;
131✔
4432
        tmpMetaRsp.metaRspLen = dataLen;
131✔
4433
        tmpMetaRsp.metaRsp = data;
131✔
4434
        if (!pTaskInfo->streamInfo.btMetaRsp.batchMetaReq) {
131✔
4435
          pTaskInfo->streamInfo.btMetaRsp.batchMetaReq = taosArrayInit(4, POINTER_BYTES);
15✔
4436
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, code, lino, _end, terrno);
15!
4437

4438
          pTaskInfo->streamInfo.btMetaRsp.batchMetaLen = taosArrayInit(4, sizeof(int32_t));
15✔
4439
          QUERY_CHECK_NULL(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, code, lino, _end, terrno);
15!
4440
        }
4441
        int32_t  tempRes = TSDB_CODE_SUCCESS;
131✔
4442
        uint32_t len = 0;
131✔
4443
        tEncodeSize(tEncodeMqMetaRsp, &tmpMetaRsp, len, tempRes);
131!
4444
        if (TSDB_CODE_SUCCESS != tempRes) {
131!
4445
          qError("tmqsnap tEncodeMqMetaRsp error");
×
4446
          taosMemoryFreeClear(data);
×
4447
          break;
×
4448
        }
4449

4450
        int32_t tLen = sizeof(SMqRspHead) + len;
131✔
4451
        void*   tBuf = taosMemoryCalloc(1, tLen);
131!
4452
        QUERY_CHECK_NULL(tBuf, code, lino, _end, terrno);
131!
4453

4454
        void*    metaBuff = POINTER_SHIFT(tBuf, sizeof(SMqRspHead));
131✔
4455
        SEncoder encoder = {0};
131✔
4456
        tEncoderInit(&encoder, metaBuff, len);
131✔
4457
        int32_t tempLen = tEncodeMqMetaRsp(&encoder, &tmpMetaRsp);
131✔
4458
        if (tempLen < 0) {
131!
4459
          qError("tmqsnap tEncodeMqMetaRsp error");
×
4460
          tEncoderClear(&encoder);
×
4461
          taosMemoryFreeClear(tBuf);
×
4462
          taosMemoryFreeClear(data);
×
4463
          break;
×
4464
        }
4465
        taosMemoryFreeClear(data);
131!
4466
        void* tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaReq, &tBuf);
131✔
4467
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
131!
4468

4469
        tmp = taosArrayPush(pTaskInfo->streamInfo.btMetaRsp.batchMetaLen, &tLen);
131✔
4470
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
131!
4471
      }
4472
    }
4473

4474
    (*ppRes) = NULL;
17✔
4475
    return code;
17✔
4476
  }
4477

4478
_end:
×
4479
  if (code != TSDB_CODE_SUCCESS) {
×
4480
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4481
    pTaskInfo->code = code;
×
4482
    T_LONG_JMP(pTaskInfo->env, code);
×
4483
  }
4484

4485
  (*ppRes) = NULL;
×
4486
  return code;
×
4487
}
4488

4489
static void destroyRawScanOperatorInfo(void* param) {
342✔
4490
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
342✔
4491
  pRawScan->pAPI->tsdReader.tsdReaderClose(pRawScan->dataReader);
342✔
4492
  pRawScan->pAPI->snapshotFn.destroySnapshot(pRawScan->sContext);
342✔
4493
  tableListDestroy(pRawScan->pTableListInfo);
342✔
4494
  taosMemoryFree(pRawScan);
342!
4495
}
342✔
4496

4497
// for subscribing db or stb (not including column),
4498
// if this scan is used, meta data can be return
4499
// and schemas are decided when scanning
4500
int32_t createRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
339✔
4501
  // create operator
4502
  // create tb reader
4503
  // create meta reader
4504
  // create tq reader
4505

4506
  QRY_PARAM_CHECK(pOptrInfo);
339!
4507
  int32_t code = TSDB_CODE_SUCCESS;
339✔
4508
  int32_t lino = 0;
339✔
4509

4510
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
339!
4511
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
342!
4512
  if (pInfo == NULL || pOperator == NULL) {
342!
4513
    code = terrno;
×
4514
    lino = __LINE__;
×
4515
    goto _end;
×
4516
  }
4517

4518
  pInfo->pTableListInfo = tableListCreate();
342✔
4519
  QUERY_CHECK_NULL(pInfo->pTableListInfo, code, lino, _end, terrno);
342!
4520
  pInfo->vnode = pHandle->vnode;
342✔
4521
  pInfo->pAPI = &pTaskInfo->storageAPI;
342✔
4522

4523
  pInfo->sContext = pHandle->sContext;
342✔
4524
  setOperatorInfo(pOperator, "RawScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
342✔
4525
                  pTaskInfo);
4526

4527
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScanNext, NULL, destroyRawScanOperatorInfo, optrDefaultBufFn, NULL,
342✔
4528
                                         optrDefaultGetNextExtFn, NULL);
4529
  *pOptrInfo = pOperator;
342✔
4530
  return code;
342✔
4531

4532
_end:
×
4533
  if (code != TSDB_CODE_SUCCESS) {
×
4534
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4535
  }
4536
  taosMemoryFree(pInfo);
×
4537
  taosMemoryFree(pOperator);
×
4538
  pTaskInfo->code = code;
×
4539
  return code;
×
4540
}
4541

4542
void destroyStreamScanOperatorInfo(void* param) {
4,848✔
4543
  if (param == NULL) {
4,848!
4544
    return;
×
4545
  }
4546

4547
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
4,848✔
4548
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
4,848!
4549
    destroyOperator(pStreamScan->pTableScanOp);
4,647✔
4550
  }
4551

4552
  if (pStreamScan->tqReader != NULL && pStreamScan->readerFn.tqReaderClose != NULL) {
4,848!
4553
    pStreamScan->readerFn.tqReaderClose(pStreamScan->tqReader);
4,647✔
4554
  }
4555
  if (pStreamScan->pVtableMergeHandles) {
4,847!
4556
    taosHashCleanup(pStreamScan->pVtableMergeHandles);
×
4557
    pStreamScan->pVtableMergeHandles = NULL;
×
4558
  }
4559
  if (pStreamScan->pVtableMergeBuf) {
4,847!
4560
    destroyDiskbasedBuf(pStreamScan->pVtableMergeBuf);
×
4561
    pStreamScan->pVtableMergeBuf = NULL;
×
4562
  }
4563
  if (pStreamScan->pVtableReadyHandles) {
4,847!
4564
    taosArrayDestroy(pStreamScan->pVtableReadyHandles);
×
4565
    pStreamScan->pVtableReadyHandles = NULL;
×
4566
  }
4567
  if (pStreamScan->pTableListInfo) {
4,847!
4568
    tableListDestroy(pStreamScan->pTableListInfo);
×
4569
    pStreamScan->pTableListInfo = NULL;
×
4570
  }
4571
  if (pStreamScan->matchInfo.pList) {
4,847!
4572
    taosArrayDestroy(pStreamScan->matchInfo.pList);
4,849✔
4573
  }
4574
  if (pStreamScan->pPseudoExpr) {
4,847✔
4575
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
2,366✔
4576
    taosMemoryFree(pStreamScan->pPseudoExpr);
2,366!
4577
  }
4578

4579
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
4,847✔
4580
  cleanupExprSupp(&pStreamScan->tagCalSup);
4,848✔
4581

4582
  if (pStreamScan->stateStore.updateInfoDestroy) {
4,849!
4583
    pStreamScan->stateStore.updateInfoDestroy(pStreamScan->pUpdateInfo);
4,849✔
4584
  }
4585

4586
  blockDataDestroy(pStreamScan->pRes);
4,849✔
4587
  blockDataDestroy(pStreamScan->pUpdateRes);
4,849✔
4588
  blockDataDestroy(pStreamScan->pDeleteDataRes);
4,849✔
4589
  blockDataDestroy(pStreamScan->pUpdateDataRes);
4,849✔
4590
  blockDataDestroy(pStreamScan->pCreateTbRes);
4,849✔
4591
  taosArrayDestroy(pStreamScan->pBlockLists);
4,849✔
4592
  blockDataDestroy(pStreamScan->pCheckpointRes);
4,849✔
4593

4594
  taosMemoryFree(pStreamScan);
4,849!
4595
}
4596

4597
void streamScanReleaseState(SOperatorInfo* pOperator) {
873✔
4598
  int32_t          code = TSDB_CODE_SUCCESS;
873✔
4599
  int32_t          lino = 0;
873✔
4600
  SStreamScanInfo* pInfo = pOperator->info;
873✔
4601
  void*            pBuff = NULL;
873✔
4602
  SEncoder*        pEnCoder = NULL;
873✔
4603
  SEncoder*        pScanEnCoder = NULL;
873✔
4604
  if (!pInfo->pState) {
873!
4605
    return;
211✔
4606
  }
4607
  if (!pInfo->pUpdateInfo) {
873✔
4608
    qDebug("stask:%s streamScanReleaseState cancel", GET_TASKID(pOperator->pTaskInfo));
211✔
4609
    return;
211✔
4610
  }
4611
  int32_t  len = 0;
662✔
4612
  SEncoder encoder = {0};
662✔
4613
  pEnCoder = &encoder;
662✔
4614
  tEncoderInit(pEnCoder, NULL, 0);
662✔
4615
  if (tStartEncode(pEnCoder) != 0) {
662!
4616
    code = TSDB_CODE_FAILED;
×
4617
    QUERY_CHECK_CODE(code, lino, _end);
×
4618
  }
4619
  code = pInfo->stateStore.updateInfoSerialize(pEnCoder, pInfo->pUpdateInfo);
662✔
4620
  QUERY_CHECK_CODE(code, lino, _end);
662!
4621

4622
  tEndEncode(pEnCoder);
662✔
4623
  len += encoder.pos;
662✔
4624
  tEncoderClear(pEnCoder);
662✔
4625
  pEnCoder = NULL;
662✔
4626

4627
  pBuff = taosMemoryCalloc(1, len);
662!
4628
  if (!pBuff) {
662!
4629
    code = terrno;
×
4630
    QUERY_CHECK_CODE(code, lino, _end);
×
4631
  }
4632

4633
  SEncoder scanEncoder = {0};
662✔
4634
  pScanEnCoder = &scanEncoder;
662✔
4635
  tEncoderInit(pScanEnCoder, pBuff, len);
662✔
4636
  if (tStartEncode(pScanEnCoder) != 0) {
662!
4637
    code = TSDB_CODE_FAILED;
×
4638
    QUERY_CHECK_CODE(code, lino, _end);
×
4639
  }
4640
  code = pInfo->stateStore.updateInfoSerialize(pScanEnCoder, pInfo->pUpdateInfo);
662✔
4641
  QUERY_CHECK_CODE(code, lino, _end);
662!
4642

4643
  tEndEncode(pScanEnCoder);
662✔
4644
  tEncoderClear(pScanEnCoder);
662✔
4645
  pScanEnCoder = NULL;
662✔
4646

4647
  pInfo->stateStore.streamStateSaveInfo(pInfo->pState, STREAM_SCAN_OP_STATE_NAME, strlen(STREAM_SCAN_OP_STATE_NAME),
662✔
4648
                                        pBuff, len);
4649
_end:
662✔
4650
  if (code != TSDB_CODE_SUCCESS) {
662!
4651
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4652
  }
4653
  if (pEnCoder != NULL) {
662!
4654
    tEndEncode(pEnCoder);
×
4655
    tEncoderClear(pEnCoder);
×
4656
  }
4657
  if (pScanEnCoder != NULL) {
662!
4658
    tEndEncode(pScanEnCoder);
×
4659
    tEncoderClear(pScanEnCoder);
×
4660
  }
4661
  taosMemoryFree(pBuff);
662!
4662
}
4663

4664
void streamScanReloadState(SOperatorInfo* pOperator) {
873✔
4665
  int32_t          code = TSDB_CODE_SUCCESS;
873✔
4666
  int32_t          lino = 0;
873✔
4667
  SDecoder*        pDeCoder = NULL;
873✔
4668
  SStreamScanInfo* pInfo = pOperator->info;
873✔
4669
  if (!pInfo->pState) {
873!
4670
    return;
211✔
4671
  }
4672
  if (!pInfo->pUpdateInfo) {
873✔
4673
    qDebug("stask:%s streamScanReloadState cancel", GET_TASKID(pOperator->pTaskInfo));
211✔
4674
    return;
211✔
4675
  }
4676
  void*   pBuff = NULL;
662✔
4677
  int32_t len = 0;
662✔
4678
  code = pInfo->stateStore.streamStateGetInfo(pInfo->pState, STREAM_SCAN_OP_STATE_NAME,
662✔
4679
                                              strlen(STREAM_SCAN_OP_STATE_NAME), &pBuff, &len);
4680
  QUERY_CHECK_CODE(code, lino, _end);
662!
4681

4682
  SUpdateInfo* pUpInfo = taosMemoryCalloc(1, sizeof(SUpdateInfo));
662!
4683
  if (!pUpInfo) {
662!
4684
    code = terrno;
×
4685
    QUERY_CHECK_CODE(code, lino, _end);
×
4686
  }
4687

4688
  SDecoder decoder = {0};
662✔
4689
  pDeCoder = &decoder;
662✔
4690
  tDecoderInit(pDeCoder, pBuff, len);
662✔
4691
  if (tStartDecode(pDeCoder) < 0) {
662!
4692
    lino = __LINE__;
×
4693
    goto _end;
×
4694
  }
4695
  int32_t winCode = pInfo->stateStore.updateInfoDeserialize(pDeCoder, pUpInfo);
662✔
4696
  taosMemoryFree(pBuff);
662!
4697
  if (winCode == TSDB_CODE_SUCCESS && pInfo->pUpdateInfo) {
662!
4698
    if (pInfo->pUpdateInfo->minTS < 0) {
662✔
4699
      pInfo->stateStore.updateInfoDestroy(pInfo->pUpdateInfo);
661✔
4700
      pInfo->pUpdateInfo = pUpInfo;
661✔
4701
    } else {
4702
      pInfo->stateStore.windowSBfDelete(pInfo->pUpdateInfo, 1);
1✔
4703
      code = pInfo->stateStore.windowSBfAdd(pInfo->pUpdateInfo, 1);
1✔
4704
      QUERY_CHECK_CODE(code, lino, _end);
1!
4705

4706
      QUERY_CHECK_CONDITION((pInfo->pUpdateInfo->minTS > pUpInfo->minTS), code, lino, _end,
1!
4707
                            TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
4708
      pInfo->pUpdateInfo->maxDataVersion = TMAX(pInfo->pUpdateInfo->maxDataVersion, pUpInfo->maxDataVersion);
1✔
4709
      SHashObj* curMap = pInfo->pUpdateInfo->pMap;
1✔
4710
      void*     pIte = taosHashIterate(curMap, NULL);
1✔
4711
      while (pIte != NULL) {
2✔
4712
        size_t   keySize = 0;
1✔
4713
        int64_t* pUid = taosHashGetKey(pIte, &keySize);
1✔
4714
        code = taosHashPut(pUpInfo->pMap, pUid, sizeof(int64_t), pIte, sizeof(TSKEY));
1✔
4715
        if (code == TSDB_CODE_DUP_KEY) {
1!
4716
          code = TSDB_CODE_SUCCESS;
×
4717
        }
4718
        QUERY_CHECK_CODE(code, lino, _end);
1!
4719

4720
        pIte = taosHashIterate(curMap, pIte);
1✔
4721
      }
4722
      taosHashCleanup(curMap);
1✔
4723
      pInfo->pUpdateInfo->pMap = pUpInfo->pMap;
1✔
4724
      pUpInfo->pMap = NULL;
1✔
4725
      pInfo->stateStore.updateInfoDestroy(pUpInfo);
1✔
4726
    }
4727
  } else {
4728
    pInfo->stateStore.updateInfoDestroy(pUpInfo);
×
4729
  }
4730

4731
_end:
662✔
4732
  if (pDeCoder != NULL) {
662!
4733
    tEndDecode(pDeCoder);
662✔
4734
    tDecoderClear(pDeCoder);
662✔
4735
  }
4736
  if (code != TSDB_CODE_SUCCESS) {
662!
4737
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4738
  }
4739
}
4740

4741
int32_t addPrimaryKeyCol(SSDataBlock* pBlock, uint8_t type, int32_t bytes) {
43✔
4742
  int32_t code = TSDB_CODE_SUCCESS;
43✔
4743
  int32_t lino = 0;
43✔
4744
  pBlock->info.rowSize += bytes;
43✔
4745
  SColumnInfoData infoData = {0};
43✔
4746
  infoData.info.type = type;
43✔
4747
  infoData.info.bytes = bytes;
43✔
4748
  void* tmp = taosArrayPush(pBlock->pDataBlock, &infoData);
43✔
4749
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
43!
4750

4751
_end:
43✔
4752
  if (code != TSDB_CODE_SUCCESS) {
43!
4753
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
4754
  }
4755
  return code;
43✔
4756
}
4757

4758
static SSDataBlock* createStreamVtableBlock(SColMatchInfo *pMatchInfo, const char *idstr) {
×
4759
  int32_t code = TSDB_CODE_SUCCESS;
×
4760
  int32_t lino = 0;
×
4761
  SSDataBlock *pRes = NULL;
×
4762

4763
  QUERY_CHECK_NULL(pMatchInfo, code, lino, _end, TSDB_CODE_INVALID_PARA);
×
4764

4765
  code = createDataBlock(&pRes);
×
4766
  QUERY_CHECK_CODE(code, lino, _end);
×
4767
  int32_t numOfOutput = taosArrayGetSize(pMatchInfo->pList);
×
4768
  for (int32_t i = 0; i < numOfOutput; ++i) {
×
4769
    SColMatchItem*  pItem = taosArrayGet(pMatchInfo->pList, i);
×
4770
    if (!pItem->needOutput) {
×
4771
      continue;
×
4772
    }
4773
    SColumnInfoData colInfo = createColumnInfoData(pItem->dataType.type, pItem->dataType.bytes, pItem->colId);
×
4774
    code = blockDataAppendColInfo(pRes, &colInfo);
×
4775
    QUERY_CHECK_CODE(code, lino, _end);
×
4776
  }
4777

4778
_end:
×
4779
  if (code != TSDB_CODE_SUCCESS) {
×
4780
    qError("%s failed at line %d since %s, id: %s", __func__, lino, tstrerror(code), idstr);
×
4781
    if (pRes != NULL) {
×
4782
      blockDataDestroy(pRes);
×
4783
    }
4784
    pRes = NULL;
×
4785
    terrno = code;
×
4786
  }
4787
  return pRes;
×
4788
}
4789

4790
static int32_t createStreamNormalScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode,
4,844✔
4791
                                                  SNode* pTagCond, STableListInfo* pTableListInfo,
4792
                                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
4793
  QRY_PARAM_CHECK(pOptrInfo);
4,844!
4794

4795
  int32_t          code = TSDB_CODE_SUCCESS;
4,844✔
4796
  int32_t          lino = 0;
4,844✔
4797
  SArray*          pColIds = NULL;
4,844✔
4798
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
4,844!
4799
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
4,848!
4800
  SStorageAPI*     pAPI = &pTaskInfo->storageAPI;
4,849✔
4801
  const char*      idstr = pTaskInfo->id.str;
4,849✔
4802
  SSHashObj*       pVtableInfos = pTaskInfo->pSubplan->pVTables;
4,849✔
4803

4804
  if (pInfo == NULL || pOperator == NULL) {
4,849!
4805
    code = terrno;
1✔
4806
    goto _error;
×
4807
  }
4808

4809
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
4,848✔
4810
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
4,848✔
4811

4812
  pInfo->pTagCond = pTagCond;
4,848✔
4813
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
4,848✔
4814

4815
  int32_t numOfCols = 0;
4,848✔
4816
  code = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
4,848✔
4817
  if (code != TSDB_CODE_SUCCESS) {
4,840!
4818
    goto _error;
×
4819
  }
4820

4821
  SDataType pkType = {0};
4,840✔
4822
  pInfo->primaryKeyIndex = -1;
4,840✔
4823
  pInfo->basic.primaryPkIndex = -1;
4,840✔
4824
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
4,840✔
4825
  pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
4,841✔
4826
  QUERY_CHECK_NULL(pColIds, code, lino, _error, terrno);
4,849!
4827

4828
  for (int32_t i = 0; i < numOfOutput; ++i) {
27,285✔
4829
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
22,441✔
4830
    QUERY_CHECK_NULL(id, code, lino, _error, terrno);
22,442!
4831

4832
    int16_t colId = id->colId;
22,442✔
4833
    void*   tmp = taosArrayPush(pColIds, &colId);
22,436✔
4834
    QUERY_CHECK_NULL(tmp, code, lino, _error, terrno);
22,436!
4835

4836
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
22,436✔
4837
      pInfo->primaryTsIndex = id->dstSlotId;
4,644✔
4838
    }
4839
    if (id->isPk) {
22,436✔
4840
      pInfo->primaryKeyIndex = id->dstSlotId;
43✔
4841
      pInfo->basic.primaryPkIndex = id->dstSlotId;
43✔
4842
      pkType = id->dataType;
43✔
4843
    }
4844
  }
4845

4846
  pInfo->pPartTbnameSup = NULL;
4,844✔
4847
  if (pTableScanNode->pSubtable != NULL) {
4,844✔
4848
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
1,167!
4849
    if (pSubTableExpr == NULL) {
1,167!
4850
      code = terrno;
×
4851
      goto _error;
×
4852
    }
4853

4854
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
1,167✔
4855
    code = createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
1,167✔
4856
    QUERY_CHECK_CODE(code, lino, _error);
1,167!
4857

4858
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1, &pTaskInfo->storageAPI.functionStore) != 0) {
1,167!
4859
      goto _error;
×
4860
    }
4861
  }
4862

4863
  if (pTableScanNode->pTags != NULL) {
4,844✔
4864
    int32_t    numOfTags;
4865
    SExprInfo* pTagExpr = createExpr(pTableScanNode->pTags, &numOfTags);
1,072✔
4866
    if (pTagExpr == NULL) {
1,072!
4867
      goto _error;
×
4868
    }
4869
    code = initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags, &pTaskInfo->storageAPI.functionStore);
1,072✔
4870
    if (code != 0) {
1,072!
4871
      goto _error;
×
4872
    }
4873
  }
4874

4875
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
4,844✔
4876
  TSDB_CHECK_NULL(pInfo->pBlockLists, code, lino, _error, terrno);
4,846!
4877

4878
  if (pHandle->vnode) {
4,847✔
4879
    SOperatorInfo* pTableScanOp = NULL;
4,645✔
4880
    code = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo, &pTableScanOp);
4,645✔
4881
    if (pTableScanOp == NULL || code != 0) {
4,647!
4882
      qError("createTableScanOperatorInfo error, code:%d", pTaskInfo->code);
×
4883
      goto _error;
×
4884
    }
4885

4886
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
4,647✔
4887
    if (pHandle->version > 0) {
4,647✔
4888
      pTSInfo->base.cond.endVersion = pHandle->version;
1,240✔
4889
    }
4890

4891
    STableKeyInfo* pList = NULL;
4,647✔
4892
    int32_t        num = 0;
4,647✔
4893
    code = tableListGetGroupList(pTableListInfo, 0, &pList, &num);
4,647✔
4894
    QUERY_CHECK_CODE(code, lino, _error);
4,645!
4895

4896
    if (pHandle->initTableReader) {
4,645✔
4897
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
1,241✔
4898
      pTSInfo->base.dataReader = NULL;
1,241✔
4899
    }
4900

4901
    if (pHandle->initTqReader) {
4,645!
4902
      pInfo->tqReader = pAPI->tqReaderFn.tqReaderOpen(pHandle->vnode);
4,646✔
4903
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, terrno);
4,647!
4904
    } else {
4905
      pInfo->tqReader = pHandle->tqReader;
×
4906
      QUERY_CHECK_NULL(pInfo->tqReader, code, lino, _error, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
4907
    }
4908

4909
    if (pVtableInfos != NULL) {
4,646!
4910
      // save vtable info into tqReader for vtable source scan
4911
      SSDataBlock* pResBlock = createStreamVtableBlock(&pInfo->matchInfo, idstr);
×
4912
      QUERY_CHECK_CODE(code, lino, _error);
×
4913
      code = pAPI->tqReaderFn.tqReaderSetVtableInfo(pInfo->tqReader, pHandle->vnode, pAPI, pVtableInfos, &pResBlock,
×
4914
                                                    idstr);
4915
      QUERY_CHECK_CODE(code, lino, _error);
×
4916
    }
4917

4918
    pInfo->pUpdateInfo = NULL;
4,646✔
4919
    pInfo->pTableScanOp = pTableScanOp;
4,646✔
4920
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
4,646✔
4921
      pAPI->stateStore.streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1,
3,405✔
4922
                                            pInfo->primaryTsIndex);
4923
    }
4924

4925
    pInfo->readHandle = *pHandle;
4,646✔
4926
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
4,646✔
4927
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
4,646✔
4928
    QUERY_CHECK_NULL(pInfo->pCreateTbRes, code, lino, _error, terrno);
4,647!
4929
    pInfo->hasPart = false;
4,647✔
4930

4931
    code = blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
4,647✔
4932
    QUERY_CHECK_CODE(code, lino, _error);
4,644!
4933

4934
    // set the extract column id to streamHandle
4935
    code = pAPI->tqReaderFn.tqReaderSetColIdList(pInfo->tqReader, pColIds, idstr);
4,644✔
4936
    QUERY_CHECK_CODE(code, lino, _error);
4,646!
4937

4938
    SArray* tableIdList = NULL;
4,646✔
4939
    code = extractTableIdList(((STableScanInfo*)(pInfo->pTableScanOp->info))->base.pTableListInfo, &tableIdList);
4,646✔
4940
    QUERY_CHECK_CODE(code, lino, _error);
4,646!
4941
    code = pAPI->tqReaderFn.tqReaderSetQueryTableList(pInfo->tqReader, tableIdList, idstr);
4,646✔
4942
    QUERY_CHECK_CODE(code, lino, _error);
4,647!
4943
    taosArrayDestroy(tableIdList);
4,647✔
4944
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->base.cond, sizeof(SQueryTableDataCond));
4,647✔
4945
  } else {
4946
    taosArrayDestroy(pColIds);
202✔
4947
    tableListDestroy(pTableListInfo);
202✔
4948
  }
4949

4950
  // clear the local variable to avoid repeatly free
4951
  pColIds = NULL;
4,849✔
4952

4953
  // create the pseduo columns info
4954
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
4,849✔
4955
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->pPseudoExpr, &pInfo->numOfPseudoExpr);
2,366✔
4956
    QUERY_CHECK_CODE(code, lino, _error);
2,366!
4957
  }
4958

4959
  code = filterInitFromNode((SNode*)pScanPhyNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
4,849✔
4960
  QUERY_CHECK_CODE(code, lino, _error);
4,849!
4961

4962
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
4,849✔
4963
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
4,849!
4964
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateRes);
4,849✔
4965
  QUERY_CHECK_CODE(code, lino, _error);
4,849!
4966

4967
  pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
4,849✔
4968
  pInfo->windowSup = (SWindowSupporter){.pStreamAggSup = NULL, .gap = -1, .parentType = QUERY_NODE_PHYSICAL_PLAN};
4,849✔
4969
  pInfo->groupId = 0;
4,849✔
4970
  pInfo->igCheckGroupId = false;
4,849✔
4971
  pInfo->pStreamScanOp = pOperator;
4,849✔
4972
  pInfo->deleteDataIndex = 0;
4,849✔
4973
  code = createSpecialDataBlock(STREAM_DELETE_DATA, &pInfo->pDeleteDataRes);
4,849✔
4974
  QUERY_CHECK_CODE(code, lino, _error);
4,849!
4975

4976
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
4,849✔
4977
  code = createSpecialDataBlock(STREAM_CLEAR, &pInfo->pUpdateDataRes);
4,849✔
4978
  QUERY_CHECK_CODE(code, lino, _error);
4,849!
4979

4980
  if (hasPrimaryKeyCol(pInfo)) {
4,849✔
4981
    code = addPrimaryKeyCol(pInfo->pUpdateDataRes, pkType.type, pkType.bytes);
43✔
4982
    QUERY_CHECK_CODE(code, lino, _error);
43!
4983

4984
    pInfo->pkColType = pkType.type;
43✔
4985
    pInfo->pkColLen = pkType.bytes;
43✔
4986
  }
4987

4988
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
4,849✔
4989
  pInfo->partitionSup.needCalc = false;
4,849✔
4990
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
4,849✔
4991
  pInfo->igExpired = pTableScanNode->igExpired;
4,849✔
4992
  pInfo->twAggSup.maxTs = INT64_MIN;
4,849✔
4993
  pInfo->pState = pTaskInfo->streamInfo.pState;
4,849✔
4994
  pInfo->stateStore = pTaskInfo->storageAPI.stateStore;
4,849✔
4995
  pInfo->readerFn = pTaskInfo->storageAPI.tqReaderFn;
4,849✔
4996
  pInfo->pFillSup = NULL;
4,849✔
4997
  pInfo->useGetResultRange = false;
4,849✔
4998
  pInfo->pRangeScanRes = NULL;
4,849✔
4999

5000
  code = createSpecialDataBlock(STREAM_CHECKPOINT, &pInfo->pCheckpointRes);
4,849✔
5001
  QUERY_CHECK_CODE(code, lino, _error);
4,848!
5002

5003
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.skey);
4,848✔
5004
  SET_WIN_KEY_INVALID(pInfo->lastScanRange.ekey);
4,848✔
5005
  // for stream
5006
  if (pTaskInfo->streamInfo.pState) {
4,848✔
5007
    void*   buff = NULL;
3,607✔
5008
    int32_t len = 0;
3,607✔
5009
    int32_t res = pAPI->stateStore.streamStateGetInfo(pTaskInfo->streamInfo.pState, STREAM_SCAN_OP_CHECKPOINT_NAME,
3,607✔
5010
                                                      strlen(STREAM_SCAN_OP_CHECKPOINT_NAME), &buff, &len);
5011
    if (res == TSDB_CODE_SUCCESS) {
3,607✔
5012
      streamScanOperatorDecode(buff, len, pInfo);
9✔
5013
      taosMemoryFree(buff);
9!
5014
    }
5015
  }
5016

5017
  setOperatorInfo(pOperator, STREAM_SCAN_OP_NAME, QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, false, OP_NOT_OPENED, pInfo,
4,848✔
5018
                  pTaskInfo);
5019
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
4,848✔
5020

5021
  __optr_fn_t nextFn = (pTaskInfo->execModel == OPTR_EXEC_MODEL_STREAM) ? doStreamScanNext : doQueueScanNext;
4,848✔
5022
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, nextFn, NULL, destroyStreamScanOperatorInfo, optrDefaultBufFn,
4,848✔
5023
                                         NULL, optrDefaultGetNextExtFn, NULL);
5024
  setOperatorStreamStateFn(pOperator, streamScanReleaseState, streamScanReloadState);
4,849✔
5025

5026
  *pOptrInfo = pOperator;
4,849✔
5027
  return code;
4,849✔
5028

5029
_error:
×
5030
  if (pColIds != NULL) {
×
5031
    taosArrayDestroy(pColIds);
×
5032
  }
5033

5034
  if (pInfo != NULL && pInfo->pTableScanOp != NULL) {
×
5035
    STableScanInfo* p = (STableScanInfo*)pInfo->pTableScanOp->info;
×
5036
    if (p != NULL) {
×
5037
      p->base.pTableListInfo = NULL;
×
5038
    }
5039
    destroyStreamScanOperatorInfo(pInfo);
×
5040
  }
5041

5042
  if (pOperator != NULL) {
×
5043
    pOperator->info = NULL;
×
5044
    destroyOperator(pOperator);
×
5045
  }
5046
  pTaskInfo->code = code;
×
5047
  return code;
×
5048
}
5049

5050
int32_t createStreamScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode, SNode* pTagCond,
4,834✔
5051
                                     STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
5052
                                     SOperatorInfo** pOptrInfo) {
5053
  if (pTableScanNode->triggerType == STREAM_TRIGGER_CONTINUOUS_WINDOW_CLOSE) {
4,834!
5054
    return createStreamDataScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, pOptrInfo);
×
5055
  } else {
5056
    return createStreamNormalScanOperatorInfo(pHandle, pTableScanNode, pTagCond, pTableListInfo, pTaskInfo, pOptrInfo);
4,834✔
5057
  }
5058
}
5059

5060
static int32_t doTagScanOneTable(SOperatorInfo* pOperator, SSDataBlock* pRes, SMetaReader* mr, SStorageAPI* pAPI) {
12,090✔
5061
  int32_t        code = TSDB_CODE_SUCCESS;
12,090✔
5062
  int32_t        lino = 0;
12,090✔
5063
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
12,090✔
5064
  STagScanInfo*  pInfo = pOperator->info;
12,090✔
5065
  SExprInfo*     pExprInfo = &pOperator->exprSupp.pExprInfo[0];
12,090✔
5066
  int32_t        count = pRes->info.rows;
12,090✔
5067

5068
  STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
12,090✔
5069
  if (!item) {
12,087!
5070
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
5071
           GET_TASKID(pTaskInfo));
5072
    tDecoderClear(&(*mr).coder);
×
5073
    goto _end;
×
5074
  }
5075

5076
  code = pAPI->metaReaderFn.getTableEntryByUid(mr, item->uid);
12,087✔
5077
  tDecoderClear(&(*mr).coder);
12,019✔
5078
  if (code != TSDB_CODE_SUCCESS) {
12,112!
5079
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
×
5080
           GET_TASKID(pTaskInfo));
5081
    goto _end;
×
5082
  }
5083

5084
  char str[512];
5085
  for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
24,854✔
5086
    SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
12,760✔
5087

5088
    // refactor later
5089
    if (FUNCTION_TYPE_TBNAME == pExprInfo[j].pExpr->_function.functionType) {
12,752✔
5090
      STR_TO_VARSTR(str, (*mr).me.name);
12,101✔
5091
      code = colDataSetVal(pDst, (count), str, false);
12,101✔
5092
      QUERY_CHECK_CODE(code, lino, _end);
12,091!
5093
    } else if (FUNCTION_TYPE_TBUID == pExprInfo[j].pExpr->_function.functionType) {
651✔
5094
      code = colDataSetVal(pDst, (count), (char*)&(*mr).me.uid, false);
150✔
5095
      QUERY_CHECK_CODE(code, lino, _end);
150!
5096
    } else if (FUNCTION_TYPE_VGID == pExprInfo[j].pExpr->_function.functionType) {
501✔
5097
      code = colDataSetVal(pDst, (count), (char*)&pTaskInfo->id.vgId, false);
150✔
5098
      QUERY_CHECK_CODE(code, lino, _end);
150!
5099
    } else {  // it is a tag value
5100
      STagVal val = {0};
351✔
5101
      val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
351✔
5102
      const char* p = pAPI->metaFn.extractTagVal((*mr).me.ctbEntry.pTags, pDst->info.type, &val);
351✔
5103

5104
      char* data = NULL;
351✔
5105
      if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
351!
5106
        data = tTagValToData((const STagVal*)p, false);
351✔
5107
      } else {
5108
        data = (char*)p;
×
5109
      }
5110

5111
      code = colDataSetVal(pDst, (count), data,
351✔
5112
                           (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
351!
5113
      QUERY_CHECK_CODE(code, lino, _end);
351!
5114

5115
      if ((pDst->info.type != TSDB_DATA_TYPE_JSON) && (p != NULL) && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
351!
5116
          (data != NULL)) {
5117
        taosMemoryFree(data);
272!
5118
      }
5119
    }
5120
  }
5121

5122
_end:
12,094✔
5123
  if (code != TSDB_CODE_SUCCESS) {
12,094!
5124
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5125
    pTaskInfo->code = code;
×
5126
  } else {
5127
    pRes->info.rows++;
12,094✔
5128
  }
5129

5130
  return code;
12,094✔
5131
}
5132

5133
static void tagScanFreeUidTag(void* p) {
68,644✔
5134
  STUidTagInfo* pInfo = p;
68,644✔
5135
  if (pInfo->pTagVal != NULL) {
68,644!
5136
    taosMemoryFree(pInfo->pTagVal);
68,672!
5137
  }
5138
}
68,682✔
5139

5140
static int32_t tagScanCreateResultData(SDataType* pType, int32_t numOfRows, SScalarParam* pParam) {
15,336✔
5141
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
15,336!
5142
  if (pColumnData == NULL) {
15,350!
5143
    return terrno;
×
5144
  }
5145

5146
  pColumnData->info.type = pType->type;
15,350✔
5147
  pColumnData->info.bytes = pType->bytes;
15,350✔
5148
  pColumnData->info.scale = pType->scale;
15,350✔
5149
  pColumnData->info.precision = pType->precision;
15,350✔
5150

5151
  int32_t code = colInfoDataEnsureCapacity(pColumnData, numOfRows, true);
15,350✔
5152
  if (code != TSDB_CODE_SUCCESS) {
15,350✔
5153
    terrno = code;
3✔
5154
    taosMemoryFree(pColumnData);
×
5155
    return terrno;
×
5156
  }
5157

5158
  pParam->columnData = pColumnData;
15,347✔
5159
  pParam->colAlloced = true;
15,347✔
5160
  return TSDB_CODE_SUCCESS;
15,347✔
5161
}
5162

5163
static EDealRes tagScanRewriteTagColumn(SNode** pNode, void* pContext) {
97,237✔
5164
  int32_t                code = TSDB_CODE_SUCCESS;
97,237✔
5165
  int32_t                lino = 0;
97,237✔
5166
  STagScanFilterContext* pCtx = (STagScanFilterContext*)pContext;
97,237✔
5167
  SColumnNode*           pSColumnNode = NULL;
97,237✔
5168
  if (QUERY_NODE_COLUMN == nodeType((*pNode))) {
97,237✔
5169
    pSColumnNode = *(SColumnNode**)pNode;
27,372✔
5170
  } else if (QUERY_NODE_FUNCTION == nodeType((*pNode))) {
69,865!
5171
    SFunctionNode* pFuncNode = *(SFunctionNode**)(pNode);
×
5172
    if (pFuncNode->funcType == FUNCTION_TYPE_TBNAME) {
×
5173
      pSColumnNode = NULL;
×
5174
      pCtx->code = nodesMakeNode(QUERY_NODE_COLUMN, (SNode**)&pSColumnNode);
×
5175
      if (NULL == pSColumnNode) {
×
5176
        return DEAL_RES_ERROR;
×
5177
      }
5178
      pSColumnNode->colId = -1;
×
5179
      pSColumnNode->colType = COLUMN_TYPE_TBNAME;
×
5180
      pSColumnNode->node.resType.type = TSDB_DATA_TYPE_VARCHAR;
×
5181
      pSColumnNode->node.resType.bytes = TSDB_TABLE_FNAME_LEN - 1 + VARSTR_HEADER_SIZE;
×
5182
      nodesDestroyNode(*pNode);
×
5183
      *pNode = (SNode*)pSColumnNode;
×
5184
    } else {
5185
      return DEAL_RES_CONTINUE;
×
5186
    }
5187
  } else {
5188
    return DEAL_RES_CONTINUE;
69,865✔
5189
  }
5190

5191
  void* data = taosHashGet(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId));
27,372✔
5192
  if (!data) {
27,362✔
5193
    code = taosHashPut(pCtx->colHash, &pSColumnNode->colId, sizeof(pSColumnNode->colId), pNode, sizeof((*pNode)));
15,338✔
5194
    if (code == TSDB_CODE_DUP_KEY) {
15,364!
5195
      code = TSDB_CODE_SUCCESS;
×
5196
    }
5197
    QUERY_CHECK_CODE(code, lino, _end);
15,364!
5198
    pSColumnNode->slotId = pCtx->index++;
15,364✔
5199
    SColumnInfo cInfo = {.colId = pSColumnNode->colId,
15,364✔
5200
                         .type = pSColumnNode->node.resType.type,
15,364✔
5201
                         .bytes = pSColumnNode->node.resType.bytes};
15,364✔
5202
    void*       tmp = taosArrayPush(pCtx->cInfoList, &cInfo);
15,364✔
5203
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
15,359!
5204
  } else {
5205
    SColumnNode* col = *(SColumnNode**)data;
12,024✔
5206
    pSColumnNode->slotId = col->slotId;
12,024✔
5207
  }
5208

5209
_end:
27,383✔
5210
  if (code != TSDB_CODE_SUCCESS) {
27,383!
5211
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5212
    return DEAL_RES_ERROR;
×
5213
  }
5214
  return DEAL_RES_CONTINUE;
27,383✔
5215
}
5216

5217
static int32_t tagScanFilterByTagCond(SArray* aUidTags, SNode* pTagCond, SArray* aFilterIdxs, void* pVnode,
15,343✔
5218
                                      SStorageAPI* pAPI, STagScanInfo* pInfo) {
5219
  int32_t code = TSDB_CODE_SUCCESS;
15,343✔
5220
  int32_t lino = 0;
15,343✔
5221
  int32_t numOfTables = taosArrayGetSize(aUidTags);
15,343✔
5222
  SArray* pBlockList = NULL;
15,347✔
5223

5224
  SSDataBlock* pResBlock = createTagValBlockForFilter(pInfo->filterCtx.cInfoList, numOfTables, aUidTags, pVnode, pAPI);
15,347✔
5225
  QUERY_CHECK_NULL(pResBlock, code, lino, _end, terrno);
15,331!
5226

5227
  pBlockList = taosArrayInit(1, POINTER_BYTES);
15,331✔
5228
  QUERY_CHECK_NULL(pBlockList, code, lino, _end, terrno);
15,349!
5229

5230
  void* tmp = taosArrayPush(pBlockList, &pResBlock);
15,341✔
5231
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
15,341!
5232

5233
  SDataType type = {.type = TSDB_DATA_TYPE_BOOL, .bytes = sizeof(bool)};
15,341✔
5234

5235
  SScalarParam output = {0};
15,341✔
5236
  code = tagScanCreateResultData(&type, numOfTables, &output);
15,341✔
5237
  QUERY_CHECK_CODE(code, lino, _end);
15,348!
5238

5239
  code = scalarCalculate(pTagCond, pBlockList, &output);
15,348✔
5240
  QUERY_CHECK_CODE(code, lino, _end);
15,322!
5241

5242
  bool* result = (bool*)output.columnData->pData;
15,322✔
5243
  for (int32_t i = 0; i < numOfTables; ++i) {
38,364✔
5244
    if (result[i]) {
23,044✔
5245
      void* tmp = taosArrayPush(aFilterIdxs, &i);
22,008✔
5246
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
22,008!
5247
    }
5248
  }
5249

5250
  colDataDestroy(output.columnData);
15,320✔
5251
  taosMemoryFreeClear(output.columnData);
15,346!
5252

5253
_end:
×
5254
  blockDataDestroy(pResBlock);
15,350✔
5255
  taosArrayDestroy(pBlockList);
15,343✔
5256
  if (code != TSDB_CODE_SUCCESS) {
15,353!
5257
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5258
  }
5259
  return code;
15,338✔
5260
}
5261

5262
static int32_t tagScanFillOneCellWithTag(SOperatorInfo* pOperator, const STUidTagInfo* pUidTagInfo,
281,958✔
5263
                                         SExprInfo* pExprInfo, SColumnInfoData* pColInfo, int rowIndex,
5264
                                         const SStorageAPI* pAPI, void* pVnode) {
5265
  int32_t code = TSDB_CODE_SUCCESS;
281,958✔
5266
  int32_t lino = 0;
281,958✔
5267
  if (QUERY_NODE_FUNCTION == pExprInfo->pExpr->nodeType) {
281,958✔
5268
    if (FUNCTION_TYPE_TBNAME == pExprInfo->pExpr->_function.functionType) {  // tbname
135,142!
5269
      char str[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
5270
      STR_TO_VARSTR(str, "ctbidx");
×
5271

5272
      code = colDataSetVal(pColInfo, rowIndex, str, false);
×
5273
      QUERY_CHECK_CODE(code, lino, _end);
×
5274
    } else if (FUNCTION_TYPE_TBUID == pExprInfo->pExpr->_function.functionType) {
135,142✔
5275
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pUidTagInfo->uid, false);
67,611✔
5276
      QUERY_CHECK_CODE(code, lino, _end);
67,640!
5277
    } else if (FUNCTION_TYPE_VGID == pExprInfo->pExpr->_function.functionType) {
67,531!
5278
      code = colDataSetVal(pColInfo, rowIndex, (char*)&pOperator->pTaskInfo->id.vgId, false);
67,625✔
5279
      QUERY_CHECK_CODE(code, lino, _end);
67,627!
5280
    }
5281
  } else {
5282
    STagVal tagVal = {0};
146,816✔
5283
    tagVal.cid = pExprInfo->base.pParam[0].pCol->colId;
146,816✔
5284
    if (pUidTagInfo->pTagVal == NULL) {
146,816!
5285
      colDataSetNULL(pColInfo, rowIndex);
×
5286
    } else {
5287
      const char* p = pAPI->metaFn.extractTagVal(pUidTagInfo->pTagVal, pColInfo->info.type, &tagVal);
146,816✔
5288

5289
      if (p == NULL || (pColInfo->info.type == TSDB_DATA_TYPE_JSON && ((STag*)p)->nTag == 0)) {
147,239!
5290
        colDataSetNULL(pColInfo, rowIndex);
35!
5291
      } else if (pColInfo->info.type == TSDB_DATA_TYPE_JSON) {
147,204!
5292
        code = colDataSetVal(pColInfo, rowIndex, p, false);
×
5293
        QUERY_CHECK_CODE(code, lino, _end);
×
5294
      } else if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
181,745!
5295
        char* tmp = taosMemoryMalloc(tagVal.nData + VARSTR_HEADER_SIZE + 1);
34,535!
5296
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
34,535!
5297

5298
        varDataSetLen(tmp, tagVal.nData);
34,535✔
5299
        memcpy(tmp + VARSTR_HEADER_SIZE, tagVal.pData, tagVal.nData);
34,535✔
5300
        code = colDataSetVal(pColInfo, rowIndex, tmp, false);
34,535✔
5301
        taosMemoryFree(tmp);
34,542!
5302
        QUERY_CHECK_CODE(code, lino, _end);
34,541!
5303
      } else {
5304
        code = colDataSetVal(pColInfo, rowIndex, (const char*)&tagVal.i64, false);
112,669✔
5305
        QUERY_CHECK_CODE(code, lino, _end);
112,690!
5306
      }
5307
    }
5308
  }
5309

5310
_end:
282,439✔
5311
  if (code != TSDB_CODE_SUCCESS) {
282,439!
5312
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5313
  }
5314
  return code;
282,044✔
5315
}
5316

5317
static int32_t tagScanFillResultBlock(SOperatorInfo* pOperator, SSDataBlock* pRes, SArray* aUidTags,
43,990✔
5318
                                      SArray* aFilterIdxs, bool ignoreFilterIdx, SStorageAPI* pAPI) {
5319
  int32_t       code = TSDB_CODE_SUCCESS;
43,990✔
5320
  int32_t       lino = 0;
43,990✔
5321
  STagScanInfo* pInfo = pOperator->info;
43,990✔
5322
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
43,990✔
5323
  if (!ignoreFilterIdx) {
43,990✔
5324
    size_t szTables = taosArrayGetSize(aFilterIdxs);
15,339✔
5325
    for (int i = 0; i < szTables; ++i) {
37,331✔
5326
      int32_t       idx = *(int32_t*)taosArrayGet(aFilterIdxs, i);
22,028✔
5327
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, idx);
22,025✔
5328
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
22,024!
5329
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
125,777✔
5330
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
103,786✔
5331
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
103,736!
5332
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
103,736✔
5333
        QUERY_CHECK_CODE(code, lino, _end);
103,753!
5334
      }
5335
    }
5336
  } else {
5337
    size_t szTables = taosArrayGetSize(aUidTags);
28,651✔
5338
    for (int i = 0; i < szTables; ++i) {
74,303✔
5339
      STUidTagInfo* pUidTagInfo = taosArrayGet(aUidTags, i);
45,641✔
5340
      QUERY_CHECK_NULL(pUidTagInfo, code, lino, _end, terrno);
45,638✔
5341
      for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
223,916✔
5342
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);
178,289✔
5343
        QUERY_CHECK_NULL(pDst, code, lino, _end, terrno);
178,256!
5344
        code = tagScanFillOneCellWithTag(pOperator, pUidTagInfo, &pExprInfo[j], pDst, i, pAPI, pInfo->readHandle.vnode);
178,256✔
5345
        QUERY_CHECK_CODE(code, lino, _end);
178,282!
5346
      }
5347
    }
5348
  }
5349

5350
_end:
28,662✔
5351
  if (code != TSDB_CODE_SUCCESS) {
43,965!
5352
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5353
  }
5354
  return code;
44,013✔
5355
}
5356

5357
static int32_t doTagScanFromCtbIdxNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
87,479✔
5358
  int32_t        code = TSDB_CODE_SUCCESS;
87,479✔
5359
  int32_t        lino = 0;
87,479✔
5360
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
87,479✔
5361
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
87,479✔
5362

5363
  STagScanInfo* pInfo = pOperator->info;
87,479✔
5364
  SSDataBlock*  pRes = pInfo->pRes;
87,479✔
5365

5366
  QRY_PARAM_CHECK(ppRes);
87,479!
5367

5368
  if (pOperator->status == OP_EXEC_DONE) {
87,479✔
5369
    return TSDB_CODE_SUCCESS;
43,340✔
5370
  }
5371
  blockDataCleanup(pRes);
44,139✔
5372

5373
  if (pInfo->pCtbCursor == NULL) {
44,196!
5374
    pInfo->pCtbCursor = pAPI->metaFn.openCtbCursor(pInfo->readHandle.vnode, pInfo->suid, 1);
44,201✔
5375
    QUERY_CHECK_NULL(pInfo->pCtbCursor, code, lino, _end, terrno);
44,194!
5376
  } else {
5377
    code = pAPI->metaFn.resumeCtbCursor(pInfo->pCtbCursor, 0);
×
5378
    QUERY_CHECK_CODE(code, lino, _end);
×
5379
  }
5380

5381
  SArray* aUidTags = pInfo->aUidTags;
44,197✔
5382
  SArray* aFilterIdxs = pInfo->aFilterIdxs;
44,197✔
5383
  int32_t count = 0;
44,197✔
5384
  bool    ctbCursorFinished = false;
44,197✔
5385
  while (1) {
658✔
5386
    taosArrayClearEx(aUidTags, tagScanFreeUidTag);
44,855✔
5387
    taosArrayClear(aFilterIdxs);
44,860✔
5388

5389
    int32_t numTables = 0;
44,796✔
5390
    while (numTables < pOperator->resultInfo.capacity) {
113,448!
5391
      SMCtbCursor* pCur = pInfo->pCtbCursor;
113,451✔
5392
      tb_uid_t     uid = pAPI->metaFn.ctbCursorNext(pInfo->pCtbCursor);
113,451✔
5393
      if (uid == 0) {
113,470✔
5394
        ctbCursorFinished = true;
44,874✔
5395
        break;
44,874✔
5396
      }
5397
      STUidTagInfo info = {.uid = uid, .pTagVal = pCur->pVal};
68,596✔
5398
      info.pTagVal = taosMemoryMalloc(pCur->vLen);
68,596!
5399
      QUERY_CHECK_NULL(info.pTagVal, code, lino, _end, terrno);
68,676!
5400

5401
      memcpy(info.pTagVal, pCur->pVal, pCur->vLen);
68,676✔
5402
      void* tmp = taosArrayPush(aUidTags, &info);
68,652✔
5403
      QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
68,652!
5404
      ++numTables;
68,652✔
5405
    }
5406

5407
    if (numTables == 0) {
44,871✔
5408
      break;
845✔
5409
    }
5410
    bool ignoreFilterIdx = true;
44,026✔
5411
    if (pInfo->pTagCond != NULL) {
44,026✔
5412
      ignoreFilterIdx = false;
15,346✔
5413
      code = tagScanFilterByTagCond(aUidTags, pInfo->pTagCond, aFilterIdxs, pInfo->readHandle.vnode, pAPI, pInfo);
15,346✔
5414
      QUERY_CHECK_CODE(code, lino, _end);
15,331!
5415
    } else {
5416
      ignoreFilterIdx = true;
28,680✔
5417
    }
5418

5419
    code = tagScanFillResultBlock(pOperator, pRes, aUidTags, aFilterIdxs, ignoreFilterIdx, pAPI);
44,011✔
5420
    QUERY_CHECK_CODE(code, lino, _end);
44,008!
5421

5422
    count = ignoreFilterIdx ? taosArrayGetSize(aUidTags) : taosArrayGetSize(aFilterIdxs);
44,008✔
5423

5424
    if (count != 0) {
43,995✔
5425
      break;
43,337✔
5426
    }
5427
  }
5428

5429
  if (count > 0) {
44,182✔
5430
    pAPI->metaFn.pauseCtbCursor(pInfo->pCtbCursor);
43,341✔
5431
  }
5432
  if (count == 0 || ctbCursorFinished) {
44,219!
5433
    pAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
44,219✔
5434
    pInfo->pCtbCursor = NULL;
44,230✔
5435
    setOperatorCompleted(pOperator);
44,230✔
5436
  }
5437

5438
  pRes->info.rows = count;
44,214✔
5439
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
44,214✔
5440
  if (bLimitReached) {
44,185!
5441
    setOperatorCompleted(pOperator);
×
5442
  }
5443

5444
  pOperator->resultInfo.totalRows += pRes->info.rows;
44,195✔
5445
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
44,195✔
5446

5447
_end:
44,195✔
5448
  if (code != TSDB_CODE_SUCCESS) {
44,195!
5449
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5450
    pTaskInfo->code = code;
×
5451
    T_LONG_JMP(pTaskInfo->env, code);
×
5452
  }
5453

5454
  return code;
44,195✔
5455
}
5456

5457
static int32_t doTagScanFromMetaEntryNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
1,218✔
5458
  int32_t code = TSDB_CODE_SUCCESS;
1,218✔
5459
  int32_t lino = 0;
1,218✔
5460
  if (pOperator->status == OP_EXEC_DONE) {
1,218✔
5461
    (*ppRes) = NULL;
563✔
5462
    return code;
563✔
5463
  }
5464

5465
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
655✔
5466
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
655✔
5467

5468
  STagScanInfo* pInfo = pOperator->info;
655✔
5469
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
655✔
5470
  SSDataBlock*  pRes = pInfo->pRes;
655✔
5471
  blockDataCleanup(pRes);
655✔
5472

5473
  int32_t size = 0;
659✔
5474
  code = tableListGetSize(pInfo->pTableListInfo, &size);
659✔
5475
  QUERY_CHECK_CODE(code, lino, _end);
659!
5476

5477
  if (size == 0) {
659✔
5478
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
93✔
5479
    (*ppRes) = NULL;
93✔
5480
    return code;
93✔
5481
  }
5482

5483
  SMetaReader mr = {0};
566✔
5484
  pAPI->metaReaderFn.initReader(&mr, pInfo->readHandle.vnode, META_READER_LOCK, &pAPI->metaFn);
566✔
5485
  pRes->info.rows = 0;
566✔
5486

5487
  while (pInfo->curPos < size && pRes->info.rows < pOperator->resultInfo.capacity) {
12,659✔
5488
    code = doTagScanOneTable(pOperator, pRes, &mr, &pTaskInfo->storageAPI);
12,089✔
5489
    if (code != TSDB_CODE_OUT_OF_MEMORY && code != TSDB_CODE_QRY_REACH_QMEM_THRESHOLD && code != TSDB_CODE_QRY_QUERY_MEM_EXHAUSTED) {
12,093!
5490
      // ignore other error
5491
      code = TSDB_CODE_SUCCESS;
12,094✔
5492
    }
5493
    QUERY_CHECK_CODE(code, lino, _end);
12,093!
5494

5495
    if (++pInfo->curPos >= size) {
12,093✔
5496
      setOperatorCompleted(pOperator);
566✔
5497
    }
5498
  }
5499

5500
  pAPI->metaReaderFn.clearReader(&mr);
570✔
5501
  bool bLimitReached = applyLimitOffset(&pInfo->limitInfo, pRes, pTaskInfo);
566✔
5502
  if (bLimitReached) {
566!
5503
    setOperatorCompleted(pOperator);
×
5504
  }
5505

5506
  // qDebug("QInfo:0x%" PRIx64 ", create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
5507
  if (pOperator->status == OP_EXEC_DONE) {
566!
5508
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
566✔
5509
  }
5510

5511
  pOperator->resultInfo.totalRows += pRes->info.rows;
566✔
5512

5513
  (*ppRes) = (pRes->info.rows == 0) ? NULL : pInfo->pRes;
566!
5514

5515
_end:
566✔
5516
  if (code != TSDB_CODE_SUCCESS) {
566!
5517
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5518
    pTaskInfo->code = code;
×
5519
    T_LONG_JMP(pTaskInfo->env, code);
×
5520
  }
5521
  return code;
566✔
5522
}
5523

5524
static SSDataBlock* doTagScanFromMetaEntry(SOperatorInfo* pOperator) {
×
5525
  SSDataBlock* pRes = NULL;
×
5526
  int32_t      code = doTagScanFromMetaEntryNext(pOperator, &pRes);
×
5527
  return pRes;
×
5528
}
5529

5530
static void destroyTagScanOperatorInfo(void* param) {
44,754✔
5531
  STagScanInfo* pInfo = (STagScanInfo*)param;
44,754✔
5532
  if (pInfo->pCtbCursor != NULL && pInfo->pStorageAPI != NULL) {
44,754!
5533
    pInfo->pStorageAPI->metaFn.closeCtbCursor(pInfo->pCtbCursor);
×
5534
  }
5535
  taosHashCleanup(pInfo->filterCtx.colHash);
44,754✔
5536
  taosArrayDestroy(pInfo->filterCtx.cInfoList);
44,855✔
5537
  taosArrayDestroy(pInfo->aFilterIdxs);
44,852✔
5538
  taosArrayDestroyEx(pInfo->aUidTags, tagScanFreeUidTag);
44,866✔
5539

5540
  blockDataDestroy(pInfo->pRes);
44,869✔
5541
  taosArrayDestroy(pInfo->matchInfo.pList);
44,875✔
5542
  tableListDestroy(pInfo->pTableListInfo);
44,870✔
5543

5544
  pInfo->pRes = NULL;
44,869✔
5545
  pInfo->pTableListInfo = NULL;
44,869✔
5546
  taosMemoryFreeClear(param);
44,869!
5547
}
44,881✔
5548

5549
int32_t createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pTagScanNode,
44,813✔
5550
                                  STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
5551
                                  SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
5552
  QRY_PARAM_CHECK(pOptrInfo);
44,813!
5553

5554
  int32_t         code = TSDB_CODE_SUCCESS;
44,813✔
5555
  int32_t         lino = 0;
44,813✔
5556
  SScanPhysiNode* pPhyNode = (SScanPhysiNode*)pTagScanNode;
44,813✔
5557
  STagScanInfo*   pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
44,813!
5558
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
44,814!
5559
  if (pInfo == NULL || pOperator == NULL) {
44,838!
5560
    code = terrno;
×
5561
    goto _error;
×
5562
  }
5563

5564
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;
44,840✔
5565

5566
  int32_t    numOfExprs = 0;
44,840✔
5567
  SExprInfo* pExprInfo = NULL;
44,840✔
5568

5569
  code = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &pExprInfo, &numOfExprs);
44,840✔
5570
  QUERY_CHECK_CODE(code, lino, _error);
44,839!
5571

5572
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs, &pTaskInfo->storageAPI.functionStore);
44,839✔
5573
  QUERY_CHECK_CODE(code, lino, _error);
44,794!
5574

5575
  int32_t num = 0;
44,794✔
5576
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
44,794✔
5577
  QUERY_CHECK_CODE(code, lino, _error);
44,777!
5578

5579
  pInfo->pTagCond = pTagCond;
44,777✔
5580
  pInfo->pTagIndexCond = pTagIndexCond;
44,777✔
5581
  pInfo->suid = pPhyNode->suid;
44,777✔
5582
  pInfo->pStorageAPI = &pTaskInfo->storageAPI;
44,777✔
5583

5584
  pInfo->pTableListInfo = pTableListInfo;
44,777✔
5585
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
44,777✔
5586
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
44,838!
5587

5588
  pInfo->readHandle = *pReadHandle;
44,838✔
5589
  pInfo->curPos = 0;
44,838✔
5590

5591
  initLimitInfo(pPhyNode->node.pLimit, pPhyNode->node.pSlimit, &pInfo->limitInfo);
44,838✔
5592
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
44,848✔
5593
                  pTaskInfo);
5594
  initResultSizeInfo(&pOperator->resultInfo, 4096);
44,726✔
5595
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
44,728✔
5596
  QUERY_CHECK_CODE(code, lino, _error);
44,867!
5597

5598
  if (pTagScanNode->onlyMetaCtbIdx) {
44,867✔
5599
    pInfo->aUidTags = taosArrayInit(pOperator->resultInfo.capacity, sizeof(STUidTagInfo));
44,214✔
5600
    QUERY_CHECK_NULL(pInfo->aUidTags, code, lino, _error, terrno);
44,198!
5601

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

5605
    pInfo->filterCtx.colHash =
44,194✔
5606
        taosHashInit(4, taosGetDefaultHashFunction(TSDB_DATA_TYPE_SMALLINT), false, HASH_NO_LOCK);
44,191✔
5607
    QUERY_CHECK_NULL(pInfo->filterCtx.colHash, code, lino, _error, terrno);
44,194!
5608

5609
    pInfo->filterCtx.cInfoList = taosArrayInit(4, sizeof(SColumnInfo));
44,194✔
5610
    QUERY_CHECK_NULL(pInfo->filterCtx.cInfoList, code, lino, _error, terrno);
44,219!
5611

5612
    if (pInfo->pTagCond != NULL) {
44,219✔
5613
      nodesRewriteExprPostOrder(&pTagCond, tagScanRewriteTagColumn, (void*)&pInfo->filterCtx);
15,361✔
5614
    }
5615
  }
5616
  __optr_fn_t tagScanNextFn = (pTagScanNode->onlyMetaCtbIdx) ? doTagScanFromCtbIdxNext : doTagScanFromMetaEntryNext;
44,862✔
5617
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, tagScanNextFn, NULL, destroyTagScanOperatorInfo,
44,862✔
5618
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
5619
  *pOptrInfo = pOperator;
44,846✔
5620
  return code;
44,846✔
5621

5622
_error:
×
5623
  if (pInfo) {
×
5624
    pInfo->pTableListInfo = NULL;
×
5625
  }
5626

5627
  if (pInfo != NULL) destroyTagScanOperatorInfo(pInfo);
×
5628
  if (pOperator != NULL) {
×
5629
    pOperator->info = NULL;
×
5630
    destroyOperator(pOperator);
×
5631
  }
5632
  return code;
×
5633
}
5634

5635
// table merge scan operator
5636

5637
static int32_t subTblRowCompareTsFn(const void* pLeft, const void* pRight, void* param) {
911,313✔
5638
  int32_t                 left = *(int32_t*)pLeft;
911,313✔
5639
  int32_t                 right = *(int32_t*)pRight;
911,313✔
5640
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
911,313✔
5641

5642
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
911,313✔
5643
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
911,313✔
5644

5645
  if (leftIdx == -1) {
911,313✔
5646
    return 1;
765,939✔
5647
  } else if (rightIdx == -1) {
145,374✔
5648
    return -1;
1,440✔
5649
  }
5650

5651
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
143,934✔
5652
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
143,934✔
5653
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
143,934!
5654
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
143,934!
5655
    ret = -1 * ret;
×
5656
  }
5657
  return ret;
143,934✔
5658
}
5659

5660
static int32_t subTblRowCompareTsPkFn(const void* pLeft, const void* pRight, void* param) {
×
5661
  int32_t                 left = *(int32_t*)pLeft;
×
5662
  int32_t                 right = *(int32_t*)pRight;
×
5663
  STmsSubTablesMergeInfo* pInfo = (STmsSubTablesMergeInfo*)param;
×
5664

5665
  int32_t leftIdx = pInfo->aInputs[left].rowIdx;
×
5666
  int32_t rightIdx = pInfo->aInputs[right].rowIdx;
×
5667

5668
  if (leftIdx == -1) {
×
5669
    return 1;
×
5670
  } else if (rightIdx == -1) {
×
5671
    return -1;
×
5672
  }
5673

5674
  int64_t leftTs = pInfo->aInputs[left].aTs[leftIdx];
×
5675
  int64_t rightTs = pInfo->aInputs[right].aTs[rightIdx];
×
5676
  int32_t ret = leftTs > rightTs ? 1 : ((leftTs < rightTs) ? -1 : 0);
×
5677
  if (pInfo->pTsOrderInfo->order == TSDB_ORDER_DESC) {
×
5678
    ret = -1 * ret;
×
5679
  }
5680
  if (ret == 0 && pInfo->pPkOrderInfo) {
×
5681
    ret = tsortComparBlockCell(pInfo->aInputs[left].pInputBlock, pInfo->aInputs[right].pInputBlock, leftIdx, rightIdx,
×
5682
                               pInfo->pPkOrderInfo);
×
5683
  }
5684
  return ret;
×
5685
}
5686

5687
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
11,136✔
5688
  int32_t code = TSDB_CODE_SUCCESS;
11,136✔
5689
  int32_t lino = 0;
11,136✔
5690
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
11,136✔
5691
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
11,136!
5692
  QUERY_CHECK_NULL(dst->colList, code, lino, _end, terrno);
11,171!
5693
  for (int i = 0; i < src->numOfCols; i++) {
36,162✔
5694
    dst->colList[i] = src->colList[i];
24,991✔
5695
  }
5696

5697
_end:
11,171✔
5698
  if (code != TSDB_CODE_SUCCESS) {
11,171!
5699
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5700
  }
5701
  return code;
11,163✔
5702
}
5703

5704
static int32_t fetchNextSubTableBlockFromReader(SOperatorInfo* pOperator, STmsSubTableInput* pInput,
16,734✔
5705
                                                bool* pSubTableHasBlock) {
5706
  int32_t code = 0;
16,734✔
5707

5708
  STableMergeScanInfo*    pInfo = pOperator->info;
16,734✔
5709
  SReadHandle*            pHandle = &pInfo->base.readHandle;
16,734✔
5710
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
16,734✔
5711
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
16,734✔
5712
  const SStorageAPI*      pAPI = &pTaskInfo->storageAPI;
16,734✔
5713

5714
  blockDataCleanup(pInput->pReaderBlock);
16,734✔
5715
  if (!pInput->bInMemReader) {
16,741✔
5716
    code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, pInput->pKeyInfo, 1, pInput->pReaderBlock,
8,463✔
5717
                                         (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
8,463✔
5718
    if (code != 0) {
8,449!
5719
      return code;
×
5720
    }
5721
  }
5722

5723
  pInfo->base.dataReader = pInput->pReader;
16,727✔
5724

5725
  while (true) {
20✔
5726
    bool hasNext = false;
16,747✔
5727
    code = pAPI->tsdReader.tsdNextDataBlock(pInfo->base.dataReader, &hasNext);
16,747✔
5728
    if (code != 0) {
16,769!
5729
      pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
5730
      pInfo->base.dataReader = NULL;
×
5731
      return code;
×
5732
    }
5733

5734
    if (!hasNext || isTaskKilled(pTaskInfo)) {
16,769!
5735
      if (isTaskKilled(pTaskInfo)) {
11,177!
5736
        pAPI->tsdReader.tsdReaderReleaseDataBlock(pInfo->base.dataReader);
×
5737
        pInfo->base.dataReader = NULL;
×
5738
        return code;
×
5739
      }
5740

5741
      *pSubTableHasBlock = false;
11,179✔
5742
      break;
11,179✔
5743
    }
5744

5745
    if (pInput->tblCond.order == TSDB_ORDER_ASC) {
5,592✔
5746
      pInput->tblCond.twindows.skey = pInput->pReaderBlock->info.window.ekey + 1;
5,509✔
5747
    } else {
5748
      pInput->tblCond.twindows.ekey = pInput->pReaderBlock->info.window.skey - 1;
83✔
5749
    }
5750

5751
    uint32_t status = 0;
5,592✔
5752
    code = loadDataBlock(pOperator, &pInfo->base, pInput->pReaderBlock, &status);
5,592✔
5753
    if (code != 0) {
5,592!
5754
      pInfo->base.dataReader = NULL;
×
5755
      return code;
×
5756
    }
5757

5758
    if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
5,592!
5759
      *pSubTableHasBlock = false;
×
5760
      break;
×
5761
    }
5762
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pInput->pReaderBlock->info.rows == 0) {
5,592!
5763
      continue;
20✔
5764
    }
5765

5766
    *pSubTableHasBlock = true;
5,572✔
5767
    break;
5,572✔
5768
  }
5769

5770
  if (*pSubTableHasBlock) {
16,751✔
5771
    pInput->pReaderBlock->info.id.groupId =
11,143✔
5772
        tableListGetTableGroupId(pInfo->base.pTableListInfo, pInput->pReaderBlock->info.id.uid);
5,571✔
5773
    pOperator->resultInfo.totalRows += pInput->pReaderBlock->info.rows;
5,572✔
5774
  }
5775
  if (!pInput->bInMemReader || !*pSubTableHasBlock) {
16,752✔
5776
    pAPI->tsdReader.tsdReaderClose(pInput->pReader);
12,616✔
5777
    pInput->pReader = NULL;
12,614✔
5778
  }
5779

5780
  pInfo->base.dataReader = NULL;
16,750✔
5781
  return TSDB_CODE_SUCCESS;
16,750✔
5782
}
5783

5784
static int32_t setGroupStartEndIndex(STableMergeScanInfo* pInfo) {
6,974✔
5785
  pInfo->bGroupProcessed = false;
6,974✔
5786

5787
  int32_t numOfTables = 0;
6,974✔
5788
  int32_t code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
6,974✔
5789
  if (code != TSDB_CODE_SUCCESS) {
7,008!
5790
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
5791
    return code;
×
5792
  }
5793

5794
  int32_t i = pInfo->tableStartIndex + 1;
7,012✔
5795
  for (; i < numOfTables; ++i) {
11,158✔
5796
    STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
4,374✔
5797
    if (!tableKeyInfo) {
4,377✔
5798
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
1!
5799
      return terrno;
1✔
5800
    }
5801
    if (tableKeyInfo->groupId != pInfo->groupId) {
4,376✔
5802
      break;
230✔
5803
    }
5804
  }
5805
  pInfo->tableEndIndex = i - 1;
7,014✔
5806
  return TSDB_CODE_SUCCESS;
7,014✔
5807
}
5808

5809
static int32_t openSubTablesMergeSort(STmsSubTablesMergeInfo* pSubTblsInfo) {
7,028✔
5810
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
18,202✔
5811
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
11,175✔
5812
    if (pInput->rowIdx == -1) {
11,175✔
5813
      continue;
5,603✔
5814
    }
5815

5816
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
5,572!
5817
      pInput->rowIdx = 0;
5,572✔
5818
      pInput->pageIdx = -1;
5,572✔
5819
    }
5820

5821
    pInput->pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
5,572!
5822
    SColumnInfoData* col = taosArrayGet(pInput->pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
5,572✔
5823
    if (!col) {
5,571!
5824
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5825
      return terrno;
×
5826
    }
5827
    pInput->aTs = (int64_t*)col->pData;
5,571✔
5828
  }
5829

5830
  __merge_compare_fn_t mergeCompareFn = (!pSubTblsInfo->pPkOrderInfo) ? subTblRowCompareTsFn : subTblRowCompareTsPkFn;
7,027!
5831
  return tMergeTreeCreate(&pSubTblsInfo->pTree, pSubTblsInfo->numSubTables, pSubTblsInfo, mergeCompareFn);
7,027✔
5832
}
5833

5834
static int32_t initSubTablesMergeInfo(STableMergeScanInfo* pInfo) {
6,978✔
5835
  int32_t code = setGroupStartEndIndex(pInfo);
6,978✔
5836
  if (code != TSDB_CODE_SUCCESS) {
7,020!
5837
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
5838
    return code;
×
5839
  }
5840
  STmsSubTablesMergeInfo* pSubTblsInfo = taosMemoryCalloc(1, sizeof(STmsSubTablesMergeInfo));
7,020!
5841
  if (pSubTblsInfo == NULL) {
7,023!
5842
    return terrno;
×
5843
  }
5844
  pSubTblsInfo->pTsOrderInfo = taosArrayGet(pInfo->pSortInfo, 0);
7,023✔
5845
  if (!pSubTblsInfo->pTsOrderInfo) {
7,014!
5846
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5847
    return terrno;
×
5848
  }
5849
  if (taosArrayGetSize(pInfo->pSortInfo) == 2) {
7,014✔
5850
    pSubTblsInfo->pPkOrderInfo = taosArrayGet(pInfo->pSortInfo, 1);
6✔
5851
    if (!pSubTblsInfo->pPkOrderInfo) {
×
5852
      qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5853
      return terrno;
×
5854
    }
5855
  } else {
5856
    pSubTblsInfo->pPkOrderInfo = NULL;
7,001✔
5857
  }
5858
  pSubTblsInfo->numSubTables = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
7,001✔
5859
  pSubTblsInfo->aInputs = taosMemoryCalloc(pSubTblsInfo->numSubTables, sizeof(STmsSubTableInput));
7,001!
5860
  if (pSubTblsInfo->aInputs == NULL) {
7,013!
5861
    taosMemoryFree(pSubTblsInfo);
×
5862
    return terrno;
×
5863
  }
5864
  int32_t bufPageSize = pInfo->bufPageSize;
7,013✔
5865
  int32_t inMemSize = (pSubTblsInfo->numSubTables - pSubTblsInfo->numTableBlocksInMem) * bufPageSize;
7,013✔
5866
  code = createDiskbasedBuf(&pSubTblsInfo->pBlocksBuf, pInfo->bufPageSize, inMemSize, "blocksExternalBuf", tsTempDir);
7,013✔
5867
  if (code != TSDB_CODE_SUCCESS) {
7,028✔
5868
    taosMemoryFree(pSubTblsInfo->aInputs);
14!
5869
    taosMemoryFree(pSubTblsInfo);
×
5870
    return code;
×
5871
  }
5872
  pSubTblsInfo->numTableBlocksInMem = pSubTblsInfo->numSubTables;
7,014✔
5873
  pSubTblsInfo->numInMemReaders = pSubTblsInfo->numSubTables;
7,014✔
5874

5875
  pInfo->pSubTablesMergeInfo = pSubTblsInfo;
7,014✔
5876
  return TSDB_CODE_SUCCESS;
7,014✔
5877
}
5878

5879
static int32_t initSubTableInputs(SOperatorInfo* pOperator, STableMergeScanInfo* pInfo) {
6,996✔
5880
  int32_t        code = TSDB_CODE_SUCCESS;
6,996✔
5881
  int32_t        lino = 0;
6,996✔
5882
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
6,996✔
5883
  SReadHandle*   pHandle = &pInfo->base.readHandle;
6,996✔
5884
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
6,996✔
5885

5886
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
6,996✔
5887

5888
  for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
18,174✔
5889
    STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
11,140✔
5890
    pInput->type = SUB_TABLE_MEM_BLOCK;
11,140✔
5891

5892
    code = dumpQueryTableCond(&pInfo->base.cond, &pInput->tblCond);
11,140✔
5893
    QUERY_CHECK_CODE(code, lino, _end);
11,165!
5894

5895
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pReaderBlock);
11,165✔
5896
    QUERY_CHECK_CODE(code, lino, _end);
11,176!
5897

5898
    code = createOneDataBlock(pInfo->pResBlock, false, &pInput->pPageBlock);
11,176✔
5899
    QUERY_CHECK_CODE(code, lino, _end);
11,178!
5900

5901
    STableKeyInfo* keyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i + pInfo->tableStartIndex);
11,178✔
5902
    pInput->pKeyInfo = keyInfo;
11,178✔
5903

5904
    if (isTaskKilled(pTaskInfo)) {
11,178!
5905
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
×
5906
    }
5907

5908
    if (i + 1 < pSubTblsInfo->numInMemReaders) {
11,173✔
5909
      code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInput->tblCond, keyInfo, 1, pInput->pReaderBlock,
4,147✔
5910
                                           (void**)&pInput->pReader, GET_TASKID(pTaskInfo), NULL);
4,147✔
5911
      QUERY_CHECK_CODE(code, lino, _end);
4,147!
5912
      pInput->bInMemReader = true;
4,147✔
5913
    } else {
5914
      pInput->pReader = NULL;
7,026✔
5915
      pInput->bInMemReader = false;
7,026✔
5916
    }
5917
    bool hasNext = true;
11,173✔
5918
    code = fetchNextSubTableBlockFromReader(pOperator, pInput, &hasNext);
11,173✔
5919
    QUERY_CHECK_CODE(code, lino, _end);
11,178!
5920
    if (!hasNext) {
11,178✔
5921
      pInput->rowIdx = -1;
5,606✔
5922
      ++pSubTblsInfo->numSubTablesCompleted;
5,606✔
5923
      continue;
5,606✔
5924
    } else {
5925
      pInput->rowIdx = 0;
5,572✔
5926
      pInput->pageIdx = -1;
5,572✔
5927
    }
5928
  }
5929

5930
_end:
7,034✔
5931
  if (code != TSDB_CODE_SUCCESS) {
7,034!
5932
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5933
  }
5934
  return code;
7,030✔
5935
}
5936

5937
static void adjustSubTableFromMemBlock(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
5,572✔
5938
  int32_t              code = TSDB_CODE_SUCCESS;
5,572✔
5939
  int32_t              lino = 0;
5,572✔
5940
  SExecTaskInfo*       pTaskInfo = pOperatorInfo->pTaskInfo;
5,572✔
5941
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
5,572✔
5942
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
5,572✔
5943
  bool                 hasNext = true;
5,572✔
5944
  code = fetchNextSubTableBlockFromReader(pOperatorInfo, pInput, &hasNext);
5,572✔
5945
  QUERY_CHECK_CODE(code, lino, _end);
5,572!
5946

5947
  if (!hasNext) {
5,572!
5948
    pInput->rowIdx = -1;
5,572✔
5949
    ++pSubTblsInfo->numSubTablesCompleted;
5,572✔
5950
  } else {
5951
    pInput->rowIdx = 0;
×
5952
  }
5953

5954
_end:
5,572✔
5955
  if (code != TSDB_CODE_SUCCESS) {
5,572!
5956
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
5957
    pTaskInfo->code = code;
×
5958
    T_LONG_JMP(pTaskInfo->env, code);
×
5959
  }
5960
}
5,572✔
5961

5962
static int32_t adjustSubTableForNextRow(SOperatorInfo* pOperatorInfo, STmsSubTablesMergeInfo* pSubTblsInfo) {
1,004,360✔
5963
  STableMergeScanInfo* pInfo = pOperatorInfo->info;
1,004,360✔
5964
  STmsSubTableInput*   pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,004,360✔
5965

5966
  SSDataBlock* pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,004,360!
5967
  if (pInput->rowIdx < pInputBlock->info.rows - 1) {
1,004,360✔
5968
    ++pInput->rowIdx;
998,883✔
5969
  } else if (pInput->rowIdx == pInputBlock->info.rows - 1) {
5,477!
5970
    if (pInput->type == SUB_TABLE_MEM_BLOCK) {
5,572!
5971
      adjustSubTableFromMemBlock(pOperatorInfo, pSubTblsInfo);
5,572✔
5972
    }
5973
    if (pInput->rowIdx != -1) {
5,572!
5974
      SColumnInfoData* col = taosArrayGet(pInputBlock->pDataBlock, pSubTblsInfo->pTsOrderInfo->slotId);
×
5975
      if (!col) {
×
5976
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(terrno));
×
5977
        return terrno;
×
5978
      }
5979
      pInput->pInputBlock = pInputBlock;
×
5980
      pInput->aTs = (int64_t*)col->pData;
×
5981
    }
5982
  }
5983

5984
  return tMergeTreeAdjust(pSubTblsInfo->pTree, tMergeTreeGetAdjustIndex(pSubTblsInfo->pTree));
1,004,360✔
5985
}
5986

5987
static int32_t appendChosenRowToDataBlock(STmsSubTablesMergeInfo* pSubTblsInfo, SSDataBlock* pBlock) {
1,002,502✔
5988
  int32_t            code = TSDB_CODE_SUCCESS;
1,002,502✔
5989
  int32_t            lino = 0;
1,002,502✔
5990
  STmsSubTableInput* pInput = pSubTblsInfo->aInputs + tMergeTreeGetChosenIndex(pSubTblsInfo->pTree);
1,002,502✔
5991
  SSDataBlock*       pInputBlock = (pInput->type == SUB_TABLE_MEM_BLOCK) ? pInput->pReaderBlock : pInput->pPageBlock;
1,002,502!
5992

5993
  for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
4,010,540✔
5994
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
2,946,578✔
5995
    QUERY_CHECK_NULL(pColInfo, code, lino, _end, terrno);
2,988,840!
5996

5997
    SColumnInfoData* pSrcColInfo = taosArrayGet(pInputBlock->pDataBlock, i);
2,988,840✔
5998
    QUERY_CHECK_NULL(pSrcColInfo, code, lino, _end, terrno);
2,971,777!
5999
    bool isNull = colDataIsNull(pSrcColInfo, pInputBlock->info.rows, pInput->rowIdx, NULL);
2,986,992✔
6000

6001
    if (isNull) {
2,986,992✔
6002
      code = colDataSetVal(pColInfo, pBlock->info.rows, NULL, true);
299,707✔
6003
      QUERY_CHECK_CODE(code, lino, _end);
299,933!
6004
    } else {
6005
      if (pSrcColInfo->pData != NULL) {
2,687,285✔
6006
        char* pData = colDataGetData(pSrcColInfo, pInput->rowIdx);
2,684,843!
6007
        code = colDataSetVal(pColInfo, pBlock->info.rows, pData, false);
2,684,843✔
6008
        QUERY_CHECK_CODE(code, lino, _end);
2,705,663!
6009
      }
6010
    }
6011
  }
6012
  pBlock->info.dataLoad = 1;
1,005,036✔
6013
  pBlock->info.scanFlag = pInputBlock->info.scanFlag;
1,005,036✔
6014
  pBlock->info.rows += 1;
1,005,036✔
6015

6016
_end:
1,005,036✔
6017
  if (code != TSDB_CODE_SUCCESS) {
1,005,036!
6018
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6019
  }
6020
  return code;
1,004,632✔
6021
}
6022

6023
static int32_t getSubTablesSortedBlock(SOperatorInfo* pOperator, SSDataBlock* pBlock, int32_t capacity,
11,155✔
6024
                                       SSDataBlock** pResBlock) {
6025
  int32_t                 code = TSDB_CODE_SUCCESS;
11,155✔
6026
  int32_t                 lino = 0;
11,155✔
6027
  STableMergeScanInfo*    pInfo = pOperator->info;
11,155✔
6028
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
11,155✔
6029
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
11,155✔
6030
  bool                    finished = false;
11,155✔
6031

6032
  QRY_PARAM_CHECK(pResBlock);
11,155!
6033

6034
  blockDataCleanup(pBlock);
11,155✔
6035

6036
  while (true) {
×
6037
    while (true) {
6038
      if (pSubTblsInfo->numSubTablesCompleted >= pSubTblsInfo->numSubTables) {
1,014,085✔
6039
        finished = true;
11,164✔
6040
        break;
11,164✔
6041
      }
6042

6043
      code = appendChosenRowToDataBlock(pSubTblsInfo, pBlock);
1,002,921✔
6044
      QUERY_CHECK_CODE(code, lino, _end);
1,004,436!
6045

6046
      code = adjustSubTableForNextRow(pOperator, pSubTblsInfo);
1,004,436✔
6047
      QUERY_CHECK_CODE(code, lino, _end);
1,002,761!
6048

6049
      if (pBlock->info.rows >= capacity) {
1,002,761!
6050
        break;
×
6051
      }
6052
    }
6053

6054
    if (isTaskKilled(pTaskInfo)) {
11,164!
6055
      return pTaskInfo->code;
×
6056
    }
6057

6058
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo);
11,164✔
6059
    if (finished || limitReached || pBlock->info.rows > 0) {
11,163!
6060
      break;
6061
    }
6062
  }
6063

6064
  if (pBlock->info.rows > 0) {
11,163✔
6065
    *pResBlock = pBlock;
4,132✔
6066
  }
6067

6068
_end:
7,031✔
6069
  if (code != TSDB_CODE_SUCCESS) {
11,163!
6070
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6071
    pTaskInfo->code = code;
×
6072
  }
6073
  return code;
11,163✔
6074
}
6075

6076
static int32_t startSubTablesTableMergeScan(SOperatorInfo* pOperator) {
6,983✔
6077
  int32_t              code = TSDB_CODE_SUCCESS;
6,983✔
6078
  int32_t              lino = 0;
6,983✔
6079
  STableMergeScanInfo* pInfo = pOperator->info;
6,983✔
6080

6081
  code = initSubTablesMergeInfo(pInfo);
6,983✔
6082
  QUERY_CHECK_CODE(code, lino, _end);
7,014!
6083

6084
  code = initSubTableInputs(pOperator, pInfo);
7,014✔
6085
  QUERY_CHECK_CODE(code, lino, _end);
7,030!
6086

6087
  code = openSubTablesMergeSort(pInfo->pSubTablesMergeInfo);
7,030✔
6088
  QUERY_CHECK_CODE(code, lino, _end);
7,028!
6089

6090
_end:
7,028✔
6091
  if (code != TSDB_CODE_SUCCESS) {
7,028!
6092
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6093
  }
6094
  return code;
7,026✔
6095
}
6096

6097
static void stopSubTablesTableMergeScan(STableMergeScanInfo* pInfo) {
181,383✔
6098
  STmsSubTablesMergeInfo* pSubTblsInfo = pInfo->pSubTablesMergeInfo;
181,383✔
6099
  if (pSubTblsInfo != NULL) {
181,383✔
6100
    tMergeTreeDestroy(&pSubTblsInfo->pTree);
7,030✔
6101

6102
    for (int32_t i = 0; i < pSubTblsInfo->numSubTables; ++i) {
18,207✔
6103
      STmsSubTableInput* pInput = pSubTblsInfo->aInputs + i;
11,176✔
6104
      taosMemoryFree(pInput->tblCond.colList);
11,176!
6105
      blockDataDestroy(pInput->pReaderBlock);
11,178✔
6106
      blockDataDestroy(pInput->pPageBlock);
11,178✔
6107
      taosArrayDestroy(pInput->aBlockPages);
11,179✔
6108
      pInfo->base.readerAPI.tsdReaderClose(pInput->pReader);
11,179✔
6109
      pInput->pReader = NULL;
11,177✔
6110
    }
6111

6112
    destroyDiskbasedBuf(pSubTblsInfo->pBlocksBuf);
7,031✔
6113
    taosMemoryFree(pSubTblsInfo->aInputs);
7,031!
6114

6115
    taosMemoryFree(pSubTblsInfo);
7,032!
6116
    pInfo->pSubTablesMergeInfo = NULL;
7,032✔
6117

6118
    // taosMemoryTrim(0);
6119
  }
6120
}
181,385✔
6121

6122
int32_t doTableMergeScanParaSubTablesNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
11,022✔
6123
  QRY_PARAM_CHECK(ppRes);
11,022!
6124

6125
  int32_t lino = 0;
11,022✔
6126
  int32_t tableListSize = 0;
11,022✔
6127
  int64_t st = taosGetTimestampUs();
11,060✔
6128

6129
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
11,060✔
6130
  STableMergeScanInfo* pInfo = pOperator->info;
11,060✔
6131

6132
  if (pOperator->status == OP_EXEC_DONE) {
11,060!
6133
    return TSDB_CODE_SUCCESS;
×
6134
  }
6135

6136
  int32_t code = pOperator->fpSet._openFn(pOperator);
11,060✔
6137
  QUERY_CHECK_CODE(code, lino, _end);
11,024!
6138

6139
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
11,024✔
6140
  QUERY_CHECK_CODE(code, lino, _end);
11,046!
6141

6142
  if (!pInfo->hasGroupId) {
11,046✔
6143
    pInfo->hasGroupId = true;
6,924✔
6144

6145
    if (tableListSize == 0) {
6,924✔
6146
      setOperatorCompleted(pOperator);
147✔
6147
      (*ppRes) = NULL;
147✔
6148
      return code;
147✔
6149
    }
6150

6151
    pInfo->tableStartIndex = 0;
6,777✔
6152
    STableKeyInfo* pTmpGpId = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
6,777✔
6153
    QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
6,777!
6154
    pInfo->groupId = pTmpGpId->groupId;
6,777✔
6155
    code = startSubTablesTableMergeScan(pOperator);
6,777✔
6156
    QUERY_CHECK_CODE(code, lino, _end);
6,796!
6157
  }
6158

6159
  SSDataBlock* pBlock = NULL;
10,918✔
6160
  while (pInfo->tableStartIndex < tableListSize) {
11,148!
6161
    if (isTaskKilled(pTaskInfo)) {
11,155!
6162
      break;
×
6163
    }
6164

6165
    code = getSubTablesSortedBlock(pOperator, pInfo->pResBlock, pOperator->resultInfo.capacity, &pBlock);
11,154✔
6166
    QUERY_CHECK_CODE(code, lino, _end);
11,164!
6167

6168
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
11,164!
6169
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
×
6170
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
×
6171

6172
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
×
6173
    }
6174

6175
    if (pBlock != NULL) {
11,164✔
6176
      pBlock->info.id.groupId = pInfo->groupId;
4,132✔
6177
      pOperator->resultInfo.totalRows += pBlock->info.rows;
4,132✔
6178
      pInfo->bGroupProcessed = true;
4,132✔
6179
      break;
4,132✔
6180
    } else {
6181
      // Data of this group are all dumped, let's try the next group
6182
      stopSubTablesTableMergeScan(pInfo);
7,032✔
6183
      if (pInfo->tableEndIndex >= tableListSize - 1) {
7,032✔
6184
        setOperatorCompleted(pOperator);
6,802✔
6185
        break;
6,801✔
6186
      }
6187

6188
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
230✔
6189
      STableKeyInfo* pTmpGpId = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
230✔
6190
      QUERY_CHECK_NULL(pTmpGpId, code, lino, _end, terrno);
230!
6191

6192
      pInfo->groupId = pTmpGpId->groupId;
230✔
6193
      code = startSubTablesTableMergeScan(pOperator);
230✔
6194
      QUERY_CHECK_CODE(code, lino, _end);
230!
6195
      resetLimitInfoForNextGroup(&pInfo->limitInfo);
230✔
6196
    }
6197
  }
6198

6199
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
10,931✔
6200

6201
_end:
10,931✔
6202
  if (code != TSDB_CODE_SUCCESS) {
10,931!
6203
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6204
    pTaskInfo->code = code;
×
6205
    T_LONG_JMP(pTaskInfo->env, code);
×
6206
  } else {
6207
    (*ppRes) = pBlock;
10,931✔
6208
  }
6209

6210
  return code;
10,931✔
6211
}
6212

6213
static void tableMergeScanDoSkipTable(uint64_t uid, void* pTableMergeOpInfo) {
33,484✔
6214
  int32_t              code = TSDB_CODE_SUCCESS;
33,484✔
6215
  int32_t              lino = 0;
33,484✔
6216
  SOperatorInfo*       pOperator = (SOperatorInfo*)pTableMergeOpInfo;
33,484✔
6217
  STableMergeScanInfo* pInfo = pOperator->info;
33,484✔
6218
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
33,484✔
6219

6220
  if (pInfo->mSkipTables == NULL) {
33,484✔
6221
    pInfo->mSkipTables = taosHashInit(pInfo->tableEndIndex - pInfo->tableStartIndex + 1,
3,313✔
6222
                                      taosGetDefaultHashFunction(TSDB_DATA_TYPE_UBIGINT), false, HASH_NO_LOCK);
6223
    QUERY_CHECK_NULL(pInfo->mSkipTables, code, lino, _end, terrno);
3,322!
6224
  }
6225
  int bSkip = 1;
33,493✔
6226
  if (pInfo->mSkipTables != NULL) {
33,493!
6227
    code = taosHashPut(pInfo->mSkipTables, &uid, sizeof(uid), &bSkip, sizeof(bSkip));
33,493✔
6228
    if (code == TSDB_CODE_DUP_KEY) {
33,513!
6229
      code = TSDB_CODE_SUCCESS;
×
6230
    }
6231
    QUERY_CHECK_CODE(code, lino, _end);
33,513!
6232
  }
6233

6234
_end:
33,513✔
6235
  if (code != TSDB_CODE_SUCCESS) {
33,513!
6236
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6237
    pTaskInfo->code = code;
×
6238
    T_LONG_JMP(pTaskInfo->env, code);
×
6239
  }
6240
}
33,513✔
6241

6242
static int32_t doGetBlockForTableMergeScan(SOperatorInfo* pOperator, bool* pFinished, bool* pSkipped) {
702,633✔
6243
  STableMergeScanInfo* pInfo = pOperator->info;
702,633✔
6244
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
702,633✔
6245
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
702,633✔
6246
  SSDataBlock*         pBlock = pInfo->pReaderBlock;
702,633✔
6247
  int32_t              code = 0;
702,633✔
6248
  bool                 hasNext = false;
702,633✔
6249
  STsdbReader*         reader = pInfo->base.dataReader;
702,633✔
6250

6251
  code = pAPI->tsdReader.tsdNextDataBlock(reader, &hasNext);
702,633✔
6252
  if (code != 0) {
702,981!
6253
    pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
6254
    qError("table merge scan fetch next data block error code: %d, %s", code, GET_TASKID(pTaskInfo));
×
6255
    pTaskInfo->code = code;
×
6256
    return code;
×
6257
  }
6258

6259
  if (!hasNext || isTaskKilled(pTaskInfo)) {
702,981!
6260
    if (isTaskKilled(pTaskInfo)) {
208,354!
6261
      qInfo("table merge scan fetch next data block found task killed. %s", GET_TASKID(pTaskInfo));
×
6262
      pAPI->tsdReader.tsdReaderReleaseDataBlock(reader);
×
6263
    }
6264
    *pFinished = true;
208,489✔
6265
    return code;
208,489✔
6266
  }
6267

6268
  uint32_t status = 0;
494,624✔
6269
  code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
494,624✔
6270

6271
  if (code != TSDB_CODE_SUCCESS) {
494,124!
6272
    qInfo("table merge scan load datablock code %d, %s", code, GET_TASKID(pTaskInfo));
×
6273
    pTaskInfo->code = code;
×
6274
    return code;
×
6275
  }
6276

6277
  if (status == FUNC_DATA_REQUIRED_ALL_FILTEROUT) {
494,159!
6278
    *pFinished = true;
×
6279
    return code;
×
6280
  }
6281

6282
  // current block is filter out according to filter condition, continue load the next block
6283
  if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
494,159!
6284
    *pSkipped = true;
19,440✔
6285
    return code;
19,440✔
6286
  }
6287

6288
  return code;
474,719✔
6289
}
6290

6291
static int32_t getBlockForTableMergeScan(void* param, SSDataBlock** ppBlock) {
720,491✔
6292
  STableMergeScanSortSourceParam* source = param;
720,491✔
6293

6294
  SOperatorInfo*       pOperator = source->pOperator;
720,491✔
6295
  STableMergeScanInfo* pInfo = pOperator->info;
720,491✔
6296
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
720,491✔
6297
  SSDataBlock*         pBlock = NULL;
720,491✔
6298
  int64_t              st = taosGetTimestampUs();
720,720✔
6299
  int32_t              code = TSDB_CODE_SUCCESS;
720,720✔
6300

6301
  while (true) {
6302
    if (pInfo->rtnNextDurationBlocks) {
772,272✔
6303
      qDebug("%s table merge scan return already fetched new duration blocks. index %d num of blocks %d",
69,402✔
6304
             GET_TASKID(pTaskInfo), pInfo->nextDurationBlocksIdx, pInfo->numNextDurationBlocks);
6305

6306
      if (pInfo->nextDurationBlocksIdx < pInfo->numNextDurationBlocks) {
69,408✔
6307
        pBlock = pInfo->nextDurationBlocks[pInfo->nextDurationBlocksIdx];
32,159✔
6308
        ++pInfo->nextDurationBlocksIdx;
32,159✔
6309
      } else {
6310
        for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
69,423✔
6311
          blockDataDestroy(pInfo->nextDurationBlocks[i]);
32,179✔
6312
          pInfo->nextDurationBlocks[i] = NULL;
32,174✔
6313
        }
6314

6315
        pInfo->rtnNextDurationBlocks = false;
37,244✔
6316
        pInfo->nextDurationBlocksIdx = 0;
37,244✔
6317
        pInfo->numNextDurationBlocks = 0;
37,244✔
6318
        continue;
37,244✔
6319
      }
6320
    } else {
6321
      bool bFinished = false;
702,870✔
6322
      bool bSkipped = false;
702,870✔
6323

6324
      code = doGetBlockForTableMergeScan(pOperator, &bFinished, &bSkipped);
702,870✔
6325
      if (code != 0) {
702,446!
6326
        return code;
37,228✔
6327
      }
6328

6329
      pBlock = pInfo->pReaderBlock;
702,446✔
6330
      qDebug("%s table merge scan fetch block. finished %d skipped %d next-duration-block %d new-fileset %d",
702,446✔
6331
             GET_TASKID(pTaskInfo), bFinished, bSkipped, pInfo->bNextDurationBlockEvent, pInfo->bNewFilesetEvent);
6332
      if (bFinished) {
702,508✔
6333
        pInfo->bNewFilesetEvent = false;
208,461✔
6334
        break;
208,461✔
6335
      }
6336

6337
      if (pInfo->bNextDurationBlockEvent || pInfo->bNewFilesetEvent) {
494,047!
6338
        if (!bSkipped) {
37,226✔
6339
          code = createOneDataBlock(pBlock, true, &pInfo->nextDurationBlocks[pInfo->numNextDurationBlocks]);
32,160✔
6340
          if (code) {
32,162!
6341
            *ppBlock = NULL;
×
6342
            return code;
×
6343
          }
6344

6345
          ++pInfo->numNextDurationBlocks;
32,162✔
6346
          if (pInfo->numNextDurationBlocks > 2) {
32,162!
6347
            qError("%s table merge scan prefetch %d next duration blocks. end early.", GET_TASKID(pTaskInfo),
×
6348
                   pInfo->numNextDurationBlocks);
6349
            pInfo->bNewFilesetEvent = false;
×
6350
            break;
×
6351
          }
6352
        }
6353

6354
        if (pInfo->bNewFilesetEvent) {
37,228!
6355
          pInfo->rtnNextDurationBlocks = true;
37,228✔
6356
          *ppBlock = NULL;
37,228✔
6357
          return code;
37,228✔
6358
        }
6359

6360
        if (pInfo->bNextDurationBlockEvent) {
×
6361
          pInfo->bNextDurationBlockEvent = false;
×
6362
          continue;
14,308✔
6363
        }
6364
      }
6365
      if (bSkipped) continue;
456,821✔
6366
    }
6367

6368
    pBlock->info.id.groupId = tableListGetTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
474,672✔
6369

6370
    pOperator->resultInfo.totalRows += pBlock->info.rows;
474,673✔
6371
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
474,714✔
6372
    *ppBlock = pBlock;
474,714✔
6373

6374
    return code;
474,714✔
6375
  }
6376

6377
  *ppBlock = NULL;
208,461✔
6378
  return code;
208,461✔
6379
}
6380

6381
int32_t generateSortByTsPkInfo(SArray* colMatchInfo, int32_t order, SArray** ppSortArray) {
173,669✔
6382
  int32_t code = TSDB_CODE_SUCCESS;
173,669✔
6383
  int32_t lino = 0;
173,669✔
6384
  SArray* pSortInfo = taosArrayInit(1, sizeof(SBlockOrderInfo));
173,669✔
6385
  QUERY_CHECK_NULL(pSortInfo, code, lino, _end, terrno);
174,140!
6386
  SBlockOrderInfo biTs = {0};
174,215✔
6387
  SBlockOrderInfo biPk = {0};
174,215✔
6388

6389
  int32_t tsTargetSlotId = 0;
174,215✔
6390
  int32_t pkTargetSlotId = -1;
174,215✔
6391
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
785,968✔
6392
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
611,963✔
6393
    QUERY_CHECK_NULL(colInfo, code, lino, _end, terrno);
611,877!
6394
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
611,941✔
6395
      tsTargetSlotId = colInfo->dstSlotId;
174,096✔
6396
      biTs.order = order;
174,096✔
6397
      biTs.slotId = tsTargetSlotId;
174,096✔
6398
      biTs.nullFirst = (order == TSDB_ORDER_ASC);
174,096✔
6399
      biTs.compFn = getKeyComparFunc(TSDB_DATA_TYPE_TIMESTAMP, order);
174,096✔
6400
    }
6401
    // TODO: order by just ts
6402
    if (colInfo->isPk) {
611,759✔
6403
      pkTargetSlotId = colInfo->dstSlotId;
11,114✔
6404
      biPk.order = order;
11,114✔
6405
      biPk.slotId = pkTargetSlotId;
11,114✔
6406
      biPk.nullFirst = (order == TSDB_ORDER_ASC);
11,114✔
6407
      biPk.compFn = getKeyComparFunc(colInfo->dataType.type, order);
11,114✔
6408
    }
6409
  }
6410

6411
  void* tmp = taosArrayPush(pSortInfo, &biTs);
174,006✔
6412
  QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
174,006!
6413
  if (pkTargetSlotId != -1) {
174,006✔
6414
    tmp = taosArrayPush(pSortInfo, &biPk);
11,116✔
6415
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
11,116!
6416
  }
6417

6418
  (*ppSortArray) = pSortInfo;
174,009✔
6419

6420
_end:
174,009✔
6421
  if (code != TSDB_CODE_SUCCESS) {
174,009!
6422
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6423
  }
6424
  return code;
173,526✔
6425
}
6426

6427
void tableMergeScanTsdbNotifyCb(ETsdReaderNotifyType type, STsdReaderNotifyInfo* info, void* param) {
123,546✔
6428
  STableMergeScanInfo* pTmsInfo = param;
123,546✔
6429
  if (type == TSD_READER_NOTIFY_DURATION_START) {
123,546!
6430
    pTmsInfo->bNewFilesetEvent = true;
123,621✔
6431
  } else if (type == TSD_READER_NOTIFY_NEXT_DURATION_BLOCK) {
×
6432
    pTmsInfo->bNextDurationBlockEvent = true;
×
6433
  }
6434
  qDebug("table merge scan receive notification. type %d, fileset %d", type, info->duration.filesetId);
123,546✔
6435

6436
  return;
123,655✔
6437
}
6438

6439
int32_t startDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
245,504✔
6440
  STableMergeScanInfo* pInfo = pOperator->info;
245,504✔
6441
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
245,504✔
6442

6443
  int32_t code = TSDB_CODE_SUCCESS;
245,504✔
6444
  int32_t lino = 0;
245,504✔
6445
  int32_t numOfTable = pInfo->tableEndIndex - pInfo->tableStartIndex + 1;
245,504✔
6446

6447
  qDebug("%s table merge scan start duration ", GET_TASKID(pTaskInfo));
245,504✔
6448
  pInfo->bNewFilesetEvent = false;
245,504✔
6449
  pInfo->bNextDurationBlockEvent = false;
245,504✔
6450

6451
  pInfo->sortBufSize = 2048 * pInfo->bufPageSize;
245,504✔
6452
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
245,504✔
6453

6454
  pInfo->pSortHandle = NULL;
245,504✔
6455
  code = tsortCreateSortHandle(pInfo->pSortInfo, SORT_BLOCK_TS_MERGE, pInfo->bufPageSize, numOfBufPage,
245,504✔
6456
                               pInfo->pSortInputBlock, pTaskInfo->id.str, 0, 0, 0, &pInfo->pSortHandle);
245,504✔
6457
  if (code) {
245,508!
6458
    return code;
×
6459
  }
6460

6461
  if (pInfo->bSortRowId && numOfTable != 1) {
245,508✔
6462
    int32_t memSize = 512 * 1024 * 1024;
23,342✔
6463
    code = tsortSetSortByRowId(pInfo->pSortHandle, memSize);
23,342✔
6464
    QUERY_CHECK_CODE(code, lino, _end);
23,351!
6465
  }
6466

6467
  tsortSetMergeLimit(pInfo->pSortHandle, pInfo->mergeLimit);
245,517✔
6468
  tsortSetMergeLimitReachedFp(pInfo->pSortHandle, tableMergeScanDoSkipTable, pOperator);
245,313✔
6469
  tsortSetAbortCheckFn(pInfo->pSortHandle, isTaskKilled, pOperator->pTaskInfo);
245,155✔
6470

6471
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getBlockForTableMergeScan, NULL, NULL);
245,157✔
6472
  QUERY_CHECK_CODE(code, lino, _end);
245,283!
6473

6474
  STableMergeScanSortSourceParam* param = taosMemoryCalloc(1, sizeof(STableMergeScanSortSourceParam));
245,283!
6475
  QUERY_CHECK_NULL(param, code, lino, _end, terrno);
245,697!
6476
  param->pOperator = pOperator;
245,697✔
6477

6478
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
245,697!
6479
  if (ps == NULL) {
245,724!
6480
    taosMemoryFree(param);
×
6481
    QUERY_CHECK_NULL(ps, code, lino, _end, terrno);
×
6482
  }
6483

6484
  ps->param = param;
245,724✔
6485
  ps->onlyRef = false;
245,724✔
6486
  code = tsortAddSource(pInfo->pSortHandle, ps);
245,724✔
6487
  QUERY_CHECK_CODE(code, lino, _end);
245,298!
6488

6489
  if (numOfTable == 1) {
245,298✔
6490
    tsortSetSingleTableMerge(pInfo->pSortHandle);
166,094✔
6491
  } else {
6492
    code = tsortOpen(pInfo->pSortHandle);
79,204✔
6493
    QUERY_CHECK_CODE(code, lino, _end);
79,351!
6494
  }
6495

6496
_end:
79,351✔
6497
  if (code != TSDB_CODE_SUCCESS) {
245,378!
6498
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6499
  }
6500
  return code;
245,426✔
6501
}
6502

6503
void stopDurationForGroupTableMergeScan(SOperatorInfo* pOperator) {
245,452✔
6504
  STableMergeScanInfo* pInfo = pOperator->info;
245,452✔
6505
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
245,452✔
6506
  qDebug("%s table merge scan stop duration ", GET_TASKID(pTaskInfo));
245,452✔
6507

6508
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
245,452✔
6509
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
245,510✔
6510
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
245,510✔
6511
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
245,510✔
6512
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
245,510✔
6513
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;
245,510✔
6514

6515
  tsortDestroySortHandle(pInfo->pSortHandle);
245,510✔
6516
  pInfo->pSortHandle = NULL;
245,576✔
6517
}
245,576✔
6518

6519
void startGroupTableMergeScan(SOperatorInfo* pOperator) {
207,642✔
6520
  int32_t              code = TSDB_CODE_SUCCESS;
207,642✔
6521
  int32_t              lino = 0;
207,642✔
6522
  STableMergeScanInfo* pInfo = pOperator->info;
207,642✔
6523
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
207,642✔
6524
  SReadHandle*         pHandle = &pInfo->base.readHandle;
207,642✔
6525
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
207,642✔
6526
  qDebug("%s table merge scan start group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
207,642✔
6527

6528
  {
6529
    int32_t numOfTables = 0;
207,642✔
6530
    code = tableListGetSize(pInfo->base.pTableListInfo, &numOfTables);
207,642✔
6531
    QUERY_CHECK_CODE(code, lino, _end);
208,214!
6532

6533
    int32_t i = pInfo->tableStartIndex + 1;
208,214✔
6534
    for (; i < numOfTables; ++i) {
564,428✔
6535
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
408,909✔
6536
      QUERY_CHECK_NULL(tableKeyInfo, code, lino, _end, terrno);
408,908!
6537
      if (tableKeyInfo->groupId != pInfo->groupId) {
408,910✔
6538
        break;
52,696✔
6539
      }
6540
    }
6541
    pInfo->tableEndIndex = i - 1;
208,215✔
6542
  }
6543
  pInfo->bGroupProcessed = false;
208,215✔
6544
  int32_t tableStartIdx = pInfo->tableStartIndex;
208,215✔
6545
  int32_t tableEndIdx = pInfo->tableEndIndex;
208,215✔
6546

6547
  int32_t        numOfTable = tableEndIdx - tableStartIdx + 1;
208,215✔
6548
  STableKeyInfo* startKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, tableStartIdx);
208,215✔
6549
  code = pAPI->tsdReader.tsdReaderOpen(pHandle->vnode, &pInfo->base.cond, startKeyInfo, numOfTable, pInfo->pReaderBlock,
208,160✔
6550
                                       (void**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), &pInfo->mSkipTables);
208,160✔
6551
  QUERY_CHECK_CODE(code, lino, _end);
208,228!
6552
  if (pInfo->filesetDelimited) {
208,228!
6553
    pAPI->tsdReader.tsdSetFilesetDelimited(pInfo->base.dataReader);
208,360✔
6554
  }
6555
  pAPI->tsdReader.tsdSetSetNotifyCb(pInfo->base.dataReader, tableMergeScanTsdbNotifyCb, pInfo);
207,957✔
6556

6557
  code = startDurationForGroupTableMergeScan(pOperator);
208,354✔
6558
  QUERY_CHECK_CODE(code, lino, _end);
208,087!
6559

6560
_end:
208,087✔
6561
  if (code != TSDB_CODE_SUCCESS) {
208,087!
6562
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6563
    pTaskInfo->code = code;
×
6564
    T_LONG_JMP(pTaskInfo->env, code);
×
6565
  }
6566
}
208,087✔
6567

6568
void stopGroupTableMergeScan(SOperatorInfo* pOperator) {
208,299✔
6569
  STableMergeScanInfo* pInfo = pOperator->info;
208,299✔
6570
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
208,299✔
6571
  SStorageAPI*         pAPI = &pTaskInfo->storageAPI;
208,299✔
6572

6573
  stopDurationForGroupTableMergeScan(pOperator);
208,299✔
6574

6575
  if (pInfo->base.dataReader != NULL) {
208,331!
6576
    pAPI->tsdReader.tsdReaderClose(pInfo->base.dataReader);
208,331✔
6577
    pInfo->base.dataReader = NULL;
208,340✔
6578
  }
6579
  for (int32_t i = 0; i < pInfo->numNextDurationBlocks; ++i) {
208,340!
6580
    if (pInfo->nextDurationBlocks[i]) {
×
6581
      blockDataDestroy(pInfo->nextDurationBlocks[i]);
×
6582
      pInfo->nextDurationBlocks[i] = NULL;
×
6583
    }
6584
    pInfo->numNextDurationBlocks = 0;
×
6585
    pInfo->nextDurationBlocksIdx = 0;
×
6586
  }
6587
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
208,340✔
6588
  taosHashCleanup(pInfo->mSkipTables);
208,296✔
6589
  pInfo->mSkipTables = NULL;
208,260✔
6590
  qDebug("%s table merge scan stop group %" PRIu64, GET_TASKID(pTaskInfo), pInfo->groupId);
208,260✔
6591
}
208,261✔
6592

6593
// all data produced by this function only belongs to one group
6594
// slimit/soffset does not need to be concerned here, since this function only deal with data within one group.
6595
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
429,101✔
6596
                                              SOperatorInfo* pOperator) {
6597
  STableMergeScanInfo* pInfo = pOperator->info;
429,101✔
6598
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
429,101✔
6599
  STupleHandle*        pTupleHandle = NULL;
429,101✔
6600

6601
  blockDataCleanup(pResBlock);
429,101✔
6602

6603
  while (1) {
×
6604
    while (1) {
57,493,939✔
6605
      pTupleHandle = NULL;
57,923,464✔
6606
      int32_t code = tsortNextTuple(pHandle, &pTupleHandle);
57,923,464✔
6607
      if (code != TSDB_CODE_SUCCESS) {
56,601,252✔
6608
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
32,578!
6609
        T_LONG_JMP(pOperator->pTaskInfo->env, code);
32,578!
6610
      }
6611
      if (pTupleHandle == NULL) {
56,568,674✔
6612
        break;
392,656✔
6613
      }
6614

6615
      code = tsortAppendTupleToBlock(pInfo->pSortHandle, pResBlock, pTupleHandle);
56,176,018✔
6616
      if (code != TSDB_CODE_SUCCESS) {
57,510,551!
6617
        qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(code));
×
6618
        T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
6619
      }
6620

6621
      if (pResBlock->info.rows >= capacity) {
57,530,851✔
6622
        break;
36,912✔
6623
      }
6624
    }
6625

6626
    if (tsortIsClosed(pHandle)) {
429,568!
6627
      terrno = TSDB_CODE_TSC_QUERY_CANCELLED;
×
6628
      T_LONG_JMP(pOperator->pTaskInfo->env, terrno);
×
6629
    }
6630

6631
    bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
429,808✔
6632
    qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
429,689✔
6633
           pInfo->limitInfo.numOfOutputRows);
6634
    if (pTupleHandle == NULL || limitReached || pResBlock->info.rows > 0) {
429,685!
6635
      break;
6636
    }
6637
  }
6638
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
429,685✔
6639
}
6640

6641
int32_t doTableMergeScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
350,509✔
6642
  if (pOperator->status == OP_EXEC_DONE) {
350,509✔
6643
    (*ppRes) = NULL;
11✔
6644
    return TSDB_CODE_SUCCESS;
11✔
6645
  }
6646

6647
  int32_t              code = TSDB_CODE_SUCCESS;
350,498✔
6648
  int32_t              lino = 0;
350,498✔
6649
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
350,498✔
6650
  STableMergeScanInfo* pInfo = pOperator->info;
350,498✔
6651

6652
  code = pOperator->fpSet._openFn(pOperator);
350,498✔
6653
  QUERY_CHECK_CODE(code, lino, _end);
350,742!
6654

6655
  int64_t st = taosGetTimestampUs();
351,348✔
6656

6657
  int32_t tableListSize = 0;
351,348✔
6658
  code = tableListGetSize(pInfo->base.pTableListInfo, &tableListSize);
351,348✔
6659
  QUERY_CHECK_CODE(code, lino, _end);
351,017!
6660

6661
  if (!pInfo->hasGroupId) {
351,017✔
6662
    pInfo->hasGroupId = true;
167,054✔
6663

6664
    if (tableListSize == 0) {
167,054✔
6665
      setOperatorCompleted(pOperator);
11,636✔
6666
      (*ppRes) = NULL;
11,636✔
6667
      return code;
11,636✔
6668
    }
6669
    pInfo->tableStartIndex = 0;
155,418✔
6670
    STableKeyInfo* tmp = (STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
155,418✔
6671
    QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
155,375!
6672
    pInfo->groupId = tmp->groupId;
155,375✔
6673
    startGroupTableMergeScan(pOperator);
155,375✔
6674
  }
6675

6676
  SSDataBlock* pBlock = NULL;
339,468✔
6677
  while (pInfo->tableStartIndex < tableListSize) {
429,227!
6678
    if (isTaskKilled(pTaskInfo)) {
429,227✔
6679
      goto _end;
129✔
6680
    }
6681

6682
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
429,241✔
6683
                                              pOperator);
6684
    if (pBlock == NULL && !pInfo->bGroupProcessed && pInfo->needCountEmptyTable) {
429,638✔
6685
      STableKeyInfo* tbInfo = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
14,471✔
6686
      QUERY_CHECK_NULL(tbInfo, code, lino, _end, terrno);
14,472!
6687
      pBlock = getOneRowResultBlock(pTaskInfo, &pInfo->base, pInfo->pResBlock, tbInfo);
14,472✔
6688
    }
6689
    if (pBlock != NULL) {
429,720✔
6690
      pBlock->info.id.groupId = pInfo->groupId;
184,172✔
6691
      pOperator->resultInfo.totalRows += pBlock->info.rows;
184,172✔
6692
      pInfo->bGroupProcessed = true;
184,172✔
6693
      break;
184,172✔
6694
    } else {
6695
      if (pInfo->bNewFilesetEvent) {
245,548✔
6696
        stopDurationForGroupTableMergeScan(pOperator);
37,238✔
6697
        code = startDurationForGroupTableMergeScan(pOperator);
37,246✔
6698
        QUERY_CHECK_CODE(code, lino, _end);
37,192!
6699
      } else {
6700
        // Data of this group are all dumped, let's try the next group
6701
        stopGroupTableMergeScan(pOperator);
208,310✔
6702
        if (pInfo->tableEndIndex >= tableListSize - 1) {
208,265✔
6703
          setOperatorCompleted(pOperator);
155,582✔
6704
          break;
155,590✔
6705
        }
6706

6707
        pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
52,683✔
6708
        STableKeyInfo* tmp = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex);
52,683✔
6709
        QUERY_CHECK_NULL(tmp, code, lino, _end, terrno);
52,720!
6710
        pInfo->groupId = tmp->groupId;
52,720✔
6711
        startGroupTableMergeScan(pOperator);
52,720✔
6712
        resetLimitInfoForNextGroup(&pInfo->limitInfo);
52,671✔
6713
      }
6714
    }
6715
  }
6716

6717
  pOperator->cost.totalCost += (taosGetTimestampUs() - st) / 1000.0;
339,769✔
6718

6719
_end:
339,898✔
6720
  if (code != TSDB_CODE_SUCCESS) {
339,898!
6721
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6722
    pTaskInfo->code = code;
×
6723
    T_LONG_JMP(pTaskInfo->env, code);
×
6724
  } else {
6725
    (*ppRes) = pBlock;
339,898✔
6726
  }
6727

6728
  return code;
339,898✔
6729
}
6730

6731
static SSDataBlock* doTableMergeScan(SOperatorInfo* pOperator) {
×
6732
  SSDataBlock* pRes = NULL;
×
6733
  int32_t      code = doTableMergeScanNext(pOperator, &pRes);
×
6734
  return pRes;
×
6735
}
6736

6737
void destroyTableMergeScanOperatorInfo(void* param) {
174,339✔
6738
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
174,339✔
6739

6740
  // start one reader variable
6741
  if (pTableScanInfo->base.readerAPI.tsdReaderClose != NULL) {
174,339!
6742
    pTableScanInfo->base.readerAPI.tsdReaderClose(pTableScanInfo->base.dataReader);
174,355✔
6743
    pTableScanInfo->base.dataReader = NULL;
174,362✔
6744
  }
6745

6746
  for (int32_t i = 0; i < pTableScanInfo->numNextDurationBlocks; ++i) {
174,346!
6747
    if (pTableScanInfo->nextDurationBlocks[i] != NULL) {
×
6748
      blockDataDestroy(pTableScanInfo->nextDurationBlocks[i]);
×
6749
      pTableScanInfo->nextDurationBlocks[i] = NULL;
×
6750
    }
6751
  }
6752

6753
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
174,346✔
6754
  pTableScanInfo->pSortHandle = NULL;
174,352✔
6755
  taosHashCleanup(pTableScanInfo->mSkipTables);
174,352✔
6756
  pTableScanInfo->mSkipTables = NULL;
174,356✔
6757
  blockDataDestroy(pTableScanInfo->pSortInputBlock);
174,356✔
6758
  pTableScanInfo->pSortInputBlock = NULL;
174,379✔
6759
  // end one reader variable
6760

6761
  cleanupQueryTableDataCond(&pTableScanInfo->base.cond);
174,379✔
6762
  destroyTableScanBase(&pTableScanInfo->base, &pTableScanInfo->base.readerAPI);
174,377✔
6763

6764
  blockDataDestroy(pTableScanInfo->pResBlock);
174,354✔
6765
  pTableScanInfo->pResBlock = NULL;
174,380✔
6766

6767
  // remove it from the task->result list
6768
  blockDataDestroy(pTableScanInfo->pReaderBlock);
174,380✔
6769
  pTableScanInfo->pReaderBlock = NULL;
174,380✔
6770
  taosArrayDestroy(pTableScanInfo->pSortInfo);
174,380✔
6771

6772
  stopSubTablesTableMergeScan(pTableScanInfo);
174,378✔
6773

6774
  taosMemoryFreeClear(param);
174,359!
6775
}
174,369✔
6776

6777
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
3,240✔
6778
  if (pOptr == NULL) {
3,240!
6779
    qError("%s failed at line %d since %s", __func__, __LINE__, tstrerror(TSDB_CODE_INVALID_PARA));
×
6780
    return TSDB_CODE_INVALID_PARA;
×
6781
  }
6782
  // TODO: merge these two info into one struct
6783
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
3,240!
6784
  if (!execInfo) {
3,241!
6785
    return terrno;
×
6786
  }
6787
  STableMergeScanInfo* pInfo = pOptr->info;
3,241✔
6788
  execInfo->blockRecorder = pInfo->base.readRecorder;
3,241✔
6789
  execInfo->sortExecInfo = pInfo->sortExecInfo;
3,241✔
6790

6791
  *pOptrExplain = execInfo;
3,241✔
6792
  *len = sizeof(STableMergeScanExecInfo);
3,241✔
6793

6794
  return TSDB_CODE_SUCCESS;
3,241✔
6795
}
6796

6797
int32_t createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
173,315✔
6798
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo,
6799
                                         SOperatorInfo** pOptrInfo) {
6800
  QRY_PARAM_CHECK(pOptrInfo);
173,315!
6801

6802
  int32_t              code = TSDB_CODE_SUCCESS;
173,315✔
6803
  int32_t              lino = 0;
173,315✔
6804
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
173,315!
6805
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
174,090!
6806
  if (pInfo == NULL || pOperator == NULL) {
173,771!
6807
    code = terrno;
×
6808
    goto _error;
×
6809
  }
6810

6811
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
173,823✔
6812

6813
  int32_t numOfCols = 0;
173,823✔
6814
  code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
173,823✔
6815
                             &pInfo->base.matchInfo);
6816
  QUERY_CHECK_CODE(code, lino, _error);
173,423!
6817

6818
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode, readHandle);
173,423✔
6819
  QUERY_CHECK_CODE(code, lino, _error);
174,072!
6820

6821
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
174,072✔
6822
    SExprSupp* pSup = &pInfo->base.pseudoSup;
79,900✔
6823
    code = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->pExprInfo, &pSup->numOfExprs);
79,900✔
6824
    QUERY_CHECK_CODE(code, lino, _error);
79,840!
6825

6826
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset,
79,840✔
6827
                                      &pTaskInfo->storageAPI.functionStore);
6828
    QUERY_CHECK_NULL(pSup->pCtx, code, lino, _error, terrno);
79,462!
6829
  }
6830

6831
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
173,634✔
6832

6833
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
173,634✔
6834
  QUERY_CHECK_NULL(pInfo->base.metaCache.pTableMetaEntryCache, code, lino, _error, terrno);
174,038!
6835

6836
  pInfo->base.readerAPI = pTaskInfo->storageAPI.tsdReader;
174,038✔
6837
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
174,038✔
6838
  pInfo->base.scanFlag = MAIN_SCAN;
174,038✔
6839
  pInfo->base.readHandle = *readHandle;
174,038✔
6840

6841
  pInfo->readIdx = -1;
174,038✔
6842

6843
  pInfo->base.limitInfo.limit.limit = -1;
174,038✔
6844
  pInfo->base.limitInfo.slimit.limit = -1;
174,038✔
6845
  pInfo->base.pTableListInfo = pTableListInfo;
174,038✔
6846

6847
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
174,038✔
6848
  pInfo->sample.seed = taosGetTimestampSec();
174,038✔
6849

6850
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
174,128✔
6851
  QUERY_CHECK_CODE(code, lino, _error);
173,963!
6852

6853
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
173,963✔
6854

6855
  pInfo->mergeLimit = -1;
174,072✔
6856
  bool hasLimit = pInfo->limitInfo.limit.limit != -1 || pInfo->limitInfo.limit.offset != -1;
174,072!
6857
  if (hasLimit) {
174,072✔
6858
    pInfo->mergeLimit = pInfo->limitInfo.limit.offset != -1 ? pInfo->limitInfo.limit.limit + pInfo->limitInfo.limit.offset : pInfo->limitInfo.limit.limit;
15,964✔
6859
    pInfo->mSkipTables = NULL;
15,964✔
6860
  }
6861

6862
  initResultSizeInfo(&pOperator->resultInfo, 1024);
174,072✔
6863
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
173,680✔
6864
  QUERY_CHECK_NULL(pInfo->pResBlock, code, lino, _error, terrno);
173,825!
6865
  code = blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
173,825✔
6866
  QUERY_CHECK_CODE(code, lino, _error);
174,207!
6867
  if (!hasLimit && blockDataGetRowSize(pInfo->pResBlock) >= 256 && !pTableScanNode->smallDataTsSort) {
174,207!
6868
    pInfo->bSortRowId = true;
42,463✔
6869
  } else {
6870
    pInfo->bSortRowId = false;
131,617✔
6871
  }
6872

6873
  code = prepareDataBlockBuf(pInfo->pResBlock, &pInfo->base.matchInfo);
174,080✔
6874
  QUERY_CHECK_CODE(code, lino, _error);
174,007!
6875

6876
  code = generateSortByTsPkInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order, &pInfo->pSortInfo);
174,007✔
6877
  QUERY_CHECK_CODE(code, lino, _error);
173,680!
6878

6879
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pReaderBlock);
173,680✔
6880
  QUERY_CHECK_CODE(code, lino, _error);
174,096!
6881

6882
  pInfo->needCountEmptyTable = tsCountAlwaysReturnValue && pTableScanNode->needCountEmptyTable;
174,096✔
6883

6884
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
174,096✔
6885
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
174,096✔
6886

6887
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
173,873✔
6888

6889
  // start one reader variable
6890
  code = createOneDataBlock(pInfo->pResBlock, false, &pInfo->pSortInputBlock);
174,036✔
6891
  QUERY_CHECK_CODE(code, lino, _error);
174,208!
6892

6893
  if (!tsExperimental) {
174,208!
6894
    pInfo->filesetDelimited = false;
×
6895
  } else {
6896
    pInfo->filesetDelimited = pTableScanNode->filesetDelimited;
174,208✔
6897
  }
6898
  // end one reader variable
6899
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
174,208✔
6900
                  pInfo, pTaskInfo);
6901
  pOperator->exprSupp.numOfExprs = numOfCols;
173,966✔
6902

6903
  pOperator->fpSet = createOperatorFpSet(
173,966✔
6904
      optrDummyOpenFn, pTableScanNode->paraTablesSort ? doTableMergeScanParaSubTablesNext : doTableMergeScanNext, NULL,
173,966✔
6905
      destroyTableMergeScanOperatorInfo, optrDefaultBufFn, getTableMergeScanExplainExecInfo, optrDefaultGetNextExtFn,
6906
      NULL);
6907
  pOperator->cost.openCost = 0;
173,871✔
6908

6909
  *pOptrInfo = pOperator;
173,871✔
6910
  return code;
173,871✔
6911

6912
_error:
×
6913
  if (code != TSDB_CODE_SUCCESS) {
×
6914
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
6915
  }
6916
  pTaskInfo->code = code;
×
6917
  if (pInfo != NULL) {
×
6918
    pInfo->base.pTableListInfo = NULL;
×
6919
    destroyTableMergeScanOperatorInfo(pInfo);
×
6920
  }
6921
  if (pOperator != NULL) {
×
6922
    pOperator->info = NULL;
×
6923
    destroyOperator(pOperator);
×
6924
  }
6925
  return code;
×
6926
}
6927

6928
// ====================================================================================================================
6929
// TableCountScanOperator
6930
static void        destoryTableCountScanOperator(void* param);
6931
static int32_t     buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
6932
                                                  SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI);
6933
static int32_t     buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
6934
                                                  SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI);
6935
static int32_t     buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6936
                                             STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName);
6937
static int32_t     buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6938
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName);
6939
static int32_t     buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6940
                                          STableCountScanSupp* pSupp, SSDataBlock* pRes);
6941
static void        buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
6942
                                               STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
6943
                                               size_t perfdbTableNum);
6944
static void        buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
6945
                                              size_t infodbTableNum, size_t perfdbTableNum);
6946
static const char* GROUP_TAG_DB_NAME = "db_name";
6947
static const char* GROUP_TAG_STABLE_NAME = "stable_name";
6948

6949
int32_t tblCountScanGetGroupTagsSlotId(const SNodeList* scanCols, STableCountScanSupp* supp) {
×
6950
  if (scanCols != NULL) {
×
6951
    SNode* pNode = NULL;
×
6952
    FOREACH(pNode, scanCols) {
×
6953
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
6954
        return TSDB_CODE_QRY_SYS_ERROR;
×
6955
      }
6956
      STargetNode* targetNode = (STargetNode*)pNode;
×
6957
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
×
6958
        return TSDB_CODE_QRY_SYS_ERROR;
×
6959
      }
6960
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
×
6961
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
6962
        supp->dbNameSlotId = targetNode->slotId;
×
6963
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
6964
        supp->stbNameSlotId = targetNode->slotId;
×
6965
      }
6966
    }
6967
  }
6968
  return TSDB_CODE_SUCCESS;
×
6969
}
6970

6971
int32_t tblCountScanGetCountSlotId(const SNodeList* pseudoCols, STableCountScanSupp* supp) {
×
6972
  if (pseudoCols != NULL) {
×
6973
    SNode* pNode = NULL;
×
6974
    FOREACH(pNode, pseudoCols) {
×
6975
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
×
6976
        return TSDB_CODE_QRY_SYS_ERROR;
×
6977
      }
6978
      STargetNode* targetNode = (STargetNode*)pNode;
×
6979
      if (nodeType(targetNode->pExpr) != QUERY_NODE_FUNCTION) {
×
6980
        return TSDB_CODE_QRY_SYS_ERROR;
×
6981
      }
6982
      SFunctionNode* funcNode = (SFunctionNode*)(targetNode->pExpr);
×
6983
      if (funcNode->funcType == FUNCTION_TYPE_TABLE_COUNT) {
×
6984
        supp->tbCountSlotId = targetNode->slotId;
×
6985
      }
6986
    }
6987
  }
6988
  return TSDB_CODE_SUCCESS;
×
6989
}
6990

6991
int32_t tblCountScanGetInputs(SNodeList* groupTags, SName* tableName, STableCountScanSupp* supp) {
×
6992
  if (groupTags != NULL) {
×
6993
    SNode* pNode = NULL;
×
6994
    FOREACH(pNode, groupTags) {
×
6995
      if (nodeType(pNode) != QUERY_NODE_COLUMN) {
×
6996
        return TSDB_CODE_QRY_SYS_ERROR;
×
6997
      }
6998
      SColumnNode* colNode = (SColumnNode*)pNode;
×
6999
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
×
7000
        supp->groupByDbName = true;
×
7001
      }
7002
      if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
×
7003
        supp->groupByStbName = true;
×
7004
      }
7005
    }
7006
  } else {
7007
    tstrncpy(supp->dbNameFilter, tNameGetDbNameP(tableName), TSDB_DB_NAME_LEN);
×
7008
    tstrncpy(supp->stbNameFilter, tNameGetTableName(tableName), TSDB_TABLE_NAME_LEN);
×
7009
  }
7010
  return TSDB_CODE_SUCCESS;
×
7011
}
7012

7013
int32_t getTableCountScanSupp(SNodeList* groupTags, SName* tableName, SNodeList* scanCols, SNodeList* pseudoCols,
×
7014
                              STableCountScanSupp* supp, SExecTaskInfo* taskInfo) {
7015
  int32_t code = 0;
×
7016
  code = tblCountScanGetInputs(groupTags, tableName, supp);
×
7017
  if (code != TSDB_CODE_SUCCESS) {
×
7018
    qError("%s get table count scan supp. get inputs error", GET_TASKID(taskInfo));
×
7019
    return code;
×
7020
  }
7021

7022
  supp->dbNameSlotId = -1;
×
7023
  supp->stbNameSlotId = -1;
×
7024
  supp->tbCountSlotId = -1;
×
7025

7026
  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
×
7027
  if (code != TSDB_CODE_SUCCESS) {
×
7028
    qError("%s get table count scan supp. get group tags slot id error", GET_TASKID(taskInfo));
×
7029
    return code;
×
7030
  }
7031

7032
  code = tblCountScanGetCountSlotId(pseudoCols, supp);
×
7033
  if (code != TSDB_CODE_SUCCESS) {
×
7034
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
×
7035
    return code;
×
7036
  }
7037
  return code;
×
7038
}
7039

7040
int32_t createTableCountScanOperatorInfo(SReadHandle* readHandle, STableCountScanPhysiNode* pTblCountScanNode,
×
7041
                                         SExecTaskInfo* pTaskInfo, SOperatorInfo** pOptrInfo) {
7042
  QRY_PARAM_CHECK(pOptrInfo);
×
7043

7044
  int32_t                      code = TSDB_CODE_SUCCESS;
×
7045
  int32_t                      lino = 0;
×
7046
  SScanPhysiNode*              pScanNode = &pTblCountScanNode->scan;
×
7047
  STableCountScanOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableCountScanOperatorInfo));
×
7048
  SOperatorInfo*               pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
×
7049
  if (!pInfo || !pOperator) {
×
7050
    code = terrno;
×
7051
    goto _error;
×
7052
  }
7053

7054
  pInfo->readHandle = *readHandle;
×
7055

7056
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
×
7057
  initResultSizeInfo(&pOperator->resultInfo, 1);
×
7058
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
×
7059
  QUERY_CHECK_NULL(pInfo->pRes, code, lino, _error, terrno);
×
7060

7061
  code = blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
×
7062
  QUERY_CHECK_CODE(code, lino, _error);
×
7063

7064
  code = getTableCountScanSupp(pTblCountScanNode->pGroupTags, &pTblCountScanNode->scan.tableName,
×
7065
                               pTblCountScanNode->scan.pScanCols, pTblCountScanNode->scan.pScanPseudoCols, &pInfo->supp,
7066
                               pTaskInfo);
7067
  QUERY_CHECK_CODE(code, lino, _error);
×
7068

7069
  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
×
7070
                  pInfo, pTaskInfo);
7071
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScanNext, NULL, destoryTableCountScanOperator,
×
7072
                                         optrDefaultBufFn, NULL, optrDefaultGetNextExtFn, NULL);
7073
  *pOptrInfo = pOperator;
×
7074
  return code;
×
7075

7076
_error:
×
7077
  if (pInfo != NULL) {
×
7078
    destoryTableCountScanOperator(pInfo);
×
7079
  }
7080
  if (pOperator != NULL) {
×
7081
    pOperator->info = NULL;
×
7082
    destroyOperator(pOperator);
×
7083
  }
7084
  pTaskInfo->code = code;
×
7085
  return code;
×
7086
}
7087

7088
int32_t fillTableCountScanDataBlock(STableCountScanSupp* pSupp, char* dbName, char* stbName, int64_t count,
×
7089
                                    SSDataBlock* pRes) {
7090
  int32_t code = TSDB_CODE_SUCCESS;
×
7091
  int32_t lino = 0;
×
7092
  if (pSupp->dbNameSlotId != -1) {
×
7093
    QUERY_CHECK_CONDITION((strlen(dbName) > 0), code, lino, _end, TSDB_CODE_QRY_EXECUTOR_INTERNAL_ERROR);
×
7094
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
×
7095
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
7096

7097
    char varDbName[TSDB_DB_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
7098
    tstrncpy(varDataVal(varDbName), dbName, TSDB_DB_NAME_LEN);
×
7099

7100
    varDataSetLen(varDbName, strlen(dbName));
×
7101
    code = colDataSetVal(colInfoData, 0, varDbName, false);
×
7102
    QUERY_CHECK_CODE(code, lino, _end);
×
7103
  }
7104

7105
  if (pSupp->stbNameSlotId != -1) {
×
7106
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
×
7107
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
7108
    if (strlen(stbName) != 0) {
×
7109
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
×
7110
      tstrncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
×
7111
      varDataSetLen(varStbName, strlen(stbName));
×
7112
      code = colDataSetVal(colInfoData, 0, varStbName, false);
×
7113
      QUERY_CHECK_CODE(code, lino, _end);
×
7114
    } else {
7115
      colDataSetNULL(colInfoData, 0);
7116
    }
7117
  }
7118

7119
  if (pSupp->tbCountSlotId != -1) {
×
7120
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
×
7121
    QUERY_CHECK_NULL(colInfoData, code, lino, _end, terrno);
×
7122
    code = colDataSetVal(colInfoData, 0, (char*)&count, false);
×
7123
    QUERY_CHECK_CODE(code, lino, _end);
×
7124
  }
7125
  pRes->info.rows = 1;
×
7126

7127
_end:
×
7128
  if (code != TSDB_CODE_SUCCESS) {
×
7129
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7130
  }
7131
  return code;
×
7132
}
7133

7134
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
×
7135
  STableCountScanSupp* pSupp = &pInfo->supp;
×
7136
  SSDataBlock*         pRes = pInfo->pRes;
×
7137

7138
  size_t infodbTableNum;
7139
  getInfosDbMeta(NULL, &infodbTableNum);
×
7140
  infodbTableNum -= 1;
×
7141
  size_t perfdbTableNum;
7142
  getPerfDbMeta(NULL, &perfdbTableNum);
×
7143

7144
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
7145
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
7146
    return (pRes->info.rows > 0) ? pRes : NULL;
×
7147
  } else {
7148
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
×
7149
    return (pRes->info.rows > 0) ? pRes : NULL;
×
7150
  }
7151
}
7152

7153
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
×
7154
                                       size_t infodbTableNum, size_t perfdbTableNum) {
7155
  int32_t        code = TSDB_CODE_SUCCESS;
×
7156
  int32_t        lino = 0;
×
7157
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7158
  if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
×
7159
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
7160
    QUERY_CHECK_CODE(code, lino, _end);
×
7161
  } else if (strcmp(pSupp->dbNameFilter, TSDB_PERFORMANCE_SCHEMA_DB) == 0) {
×
7162
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
7163
    QUERY_CHECK_CODE(code, lino, _end);
×
7164
  } else if (strlen(pSupp->dbNameFilter) == 0) {
×
7165
    code = fillTableCountScanDataBlock(pSupp, "", "", infodbTableNum + perfdbTableNum, pRes);
×
7166
    QUERY_CHECK_CODE(code, lino, _end);
×
7167
  }
7168

7169
_end:
×
7170
  if (code != TSDB_CODE_SUCCESS) {
×
7171
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7172
    pTaskInfo->code = code;
×
7173
    T_LONG_JMP(pTaskInfo->env, code);
×
7174
  }
7175
  setOperatorCompleted(pOperator);
×
7176
}
×
7177

7178
static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7179
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
7180
                                        size_t perfdbTableNum) {
7181
  int32_t        code = TSDB_CODE_SUCCESS;
×
7182
  int32_t        lino = 0;
×
7183
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7184
  if (pInfo->currGrpIdx == 0) {
×
7185
    uint64_t groupId = 0;
×
7186
    if (pSupp->groupByDbName) {
×
7187
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
×
7188
    } else {
7189
      groupId = calcGroupId("", 0);
×
7190
    }
7191

7192
    pRes->info.id.groupId = groupId;
×
7193
    code = fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
×
7194
    QUERY_CHECK_CODE(code, lino, _end);
×
7195
  } else if (pInfo->currGrpIdx == 1) {
×
7196
    uint64_t groupId = 0;
×
7197
    if (pSupp->groupByDbName) {
×
7198
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
×
7199
    } else {
7200
      groupId = calcGroupId("", 0);
×
7201
    }
7202

7203
    pRes->info.id.groupId = groupId;
×
7204
    code = fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
×
7205
    QUERY_CHECK_CODE(code, lino, _end);
×
7206
  } else {
7207
    setOperatorCompleted(pOperator);
×
7208
  }
7209
  pInfo->currGrpIdx++;
×
7210

7211
_end:
×
7212
  if (code != TSDB_CODE_SUCCESS) {
×
7213
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7214
    pTaskInfo->code = code;
×
7215
    T_LONG_JMP(pTaskInfo->env, code);
×
7216
  }
7217
}
×
7218

7219
static int32_t doTableCountScanNext(SOperatorInfo* pOperator, SSDataBlock** ppRes) {
×
7220
  int32_t                      code = TSDB_CODE_SUCCESS;
×
7221
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
×
7222
  STableCountScanOperatorInfo* pInfo = pOperator->info;
×
7223
  STableCountScanSupp*         pSupp = &pInfo->supp;
×
7224
  SSDataBlock*                 pRes = pInfo->pRes;
×
7225

7226
  blockDataCleanup(pRes);
×
7227
  QRY_PARAM_CHECK(ppRes);
×
7228

7229
  if (pOperator->status == OP_EXEC_DONE) {
×
7230
    return code;
×
7231
  }
7232

7233
  if (pInfo->readHandle.mnd != NULL) {
×
7234
    (*ppRes) = buildSysDbTableCount(pOperator, pInfo);
×
7235
    return code;
×
7236
  }
7237

7238
  code = buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
×
7239
  if (code != TSDB_CODE_SUCCESS) {
×
7240
    qError("%s failed since %s", __func__, tstrerror(code));
×
7241
    pTaskInfo->code = code;
×
7242
    T_LONG_JMP(pTaskInfo->env, code);
×
7243
  }
7244
  if (pRes->info.rows > 0) {
×
7245
    *ppRes = pRes;
×
7246
  }
7247

7248
  return code;
×
7249
}
7250

7251
static int32_t buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7252
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes) {
7253
  int32_t        code = TSDB_CODE_SUCCESS;
×
7254
  int32_t        lino = 0;
×
7255
  const char*    db = NULL;
×
7256
  int32_t        vgId = 0;
×
7257
  char           dbName[TSDB_DB_NAME_LEN] = {0};
×
7258
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7259
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
7260

7261
  // get dbname
7262
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, &db, &vgId, NULL, NULL);
×
7263
  SName sn = {0};
×
7264

7265
  code = tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
×
7266
  QUERY_CHECK_CODE(code, lino, _end);
×
7267

7268
  code = tNameGetDbName(&sn, dbName);
×
7269
  QUERY_CHECK_CODE(code, lino, _end);
×
7270

7271
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
×
7272
    code = buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
×
7273
  } else {
7274
    code = buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
×
7275
  }
7276

7277
_end:
×
7278
  if (code != TSDB_CODE_SUCCESS) {
×
7279
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7280
    pTaskInfo->code = code;
×
7281
  }
7282

7283
  return code;
×
7284
}
7285

7286
static int32_t buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7287
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
7288
  int32_t        code = TSDB_CODE_SUCCESS;
×
7289
  int32_t        lino = 0;
×
7290
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7291
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
7292

7293
  if (pSupp->groupByStbName) {
×
7294
    if (pInfo->stbUidList == NULL) {
×
7295
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
×
7296
      QUERY_CHECK_NULL(pInfo->stbUidList, code, lino, _end, terrno);
×
7297
      code = pAPI->metaFn.storeGetTableList(pInfo->readHandle.vnode, TSDB_SUPER_TABLE, pInfo->stbUidList);
×
7298
      QUERY_CHECK_CODE(code, lino, _end);
×
7299
    }
7300

7301
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
×
7302
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
×
7303
      code = buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid, pAPI);
×
7304
      QUERY_CHECK_CODE(code, lino, _end);
×
7305

7306
      pInfo->currGrpIdx++;
×
7307
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
×
7308
      code = buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName, pAPI);
×
7309
      QUERY_CHECK_CODE(code, lino, _end);
×
7310

7311
      pInfo->currGrpIdx++;
×
7312
    } else {
7313
      setOperatorCompleted(pOperator);
×
7314
    }
7315
  } else {
7316
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
×
7317
    pRes->info.id.groupId = groupId;
×
7318

7319
    int64_t dbTableCount = 0;
×
7320
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &dbTableCount, NULL);
×
7321
    code = fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
×
7322
    QUERY_CHECK_CODE(code, lino, _end);
×
7323
    setOperatorCompleted(pOperator);
×
7324
  }
7325

7326
_end:
×
7327
  if (code != TSDB_CODE_SUCCESS) {
×
7328
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7329
    pTaskInfo->code = code;
×
7330
  }
7331
  return code;
×
7332
}
7333

7334
static int32_t buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
×
7335
                                         STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
7336
  int32_t        code = TSDB_CODE_SUCCESS;
×
7337
  int32_t        lino = 0;
×
7338
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
×
7339
  SStorageAPI*   pAPI = &pTaskInfo->storageAPI;
×
7340

7341
  if (strlen(pSupp->dbNameFilter) != 0) {
×
7342
    if (strlen(pSupp->stbNameFilter) != 0) {
×
7343
      uint64_t uid = 0;
×
7344
      code = pAPI->metaFn.getTableUidByName(pInfo->readHandle.vnode, pSupp->stbNameFilter, &uid);
×
7345
      QUERY_CHECK_CODE(code, lino, _end);
×
7346

7347
      int64_t numOfChildTables = 0;
×
7348
      code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, uid, &numOfChildTables, NULL, NULL);
×
7349
      QUERY_CHECK_CODE(code, lino, _end);
×
7350

7351
      code = fillTableCountScanDataBlock(pSupp, dbName, pSupp->stbNameFilter, numOfChildTables, pRes);
×
7352
      QUERY_CHECK_CODE(code, lino, _end);
×
7353
    } else {
7354
      int64_t tbNumVnode = 0;
×
7355
      pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
7356
      code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
7357
      QUERY_CHECK_CODE(code, lino, _end);
×
7358
    }
7359
  } else {
7360
    int64_t tbNumVnode = 0;
×
7361
    pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, &tbNumVnode, NULL);
×
7362
    code = fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
×
7363
    QUERY_CHECK_CODE(code, lino, _end);
×
7364
  }
7365

7366
_end:
×
7367
  if (code != TSDB_CODE_SUCCESS) {
×
7368
    pTaskInfo->code = code;
×
7369
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7370
  }
7371

7372
  setOperatorCompleted(pOperator);
×
7373
  return code;
×
7374
}
7375

7376
static int32_t buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
7377
                                              SSDataBlock* pRes, char* dbName, SStorageAPI* pAPI) {
7378
  int32_t code = TSDB_CODE_SUCCESS;
×
7379
  int32_t lino = 0;
×
7380
  char    fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
7381
  if (pSupp->groupByDbName) {
×
7382
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
×
7383
  }
7384

7385
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
7386
  pRes->info.id.groupId = groupId;
×
7387

7388
  int64_t numOfTables = 0;
×
7389
  pAPI->metaFn.getBasicInfo(pInfo->readHandle.vnode, NULL, NULL, NULL, &numOfTables);
×
7390

7391
  if (numOfTables != 0) {
×
7392
    code = fillTableCountScanDataBlock(pSupp, dbName, "", numOfTables, pRes);
×
7393
  }
7394

7395
_end:
×
7396
  if (code != TSDB_CODE_SUCCESS) {
×
7397
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7398
  }
7399
  return code;
×
7400
}
7401

7402
static int32_t buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
×
7403
                                              SSDataBlock* pRes, char* dbName, tb_uid_t stbUid, SStorageAPI* pAPI) {
7404
  int32_t code = TSDB_CODE_SUCCESS;
×
7405
  int32_t lino = 0;
×
7406
  char    stbName[TSDB_TABLE_NAME_LEN] = {0};
×
7407
  code = pAPI->metaFn.getTableNameByUid(pInfo->readHandle.vnode, stbUid, stbName);
×
7408
  QUERY_CHECK_CODE(code, lino, _end);
×
7409

7410
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
×
7411
  if (pSupp->groupByDbName) {
×
7412
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, varDataVal(stbName));
×
7413
  } else {
7414
    (void)snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", varDataVal(stbName));
×
7415
  }
7416

7417
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
×
7418
  pRes->info.id.groupId = groupId;
×
7419

7420
  int64_t ctbNum = 0;
×
7421
  code = pAPI->metaFn.getNumOfChildTables(pInfo->readHandle.vnode, stbUid, &ctbNum, NULL, NULL);
×
7422
  QUERY_CHECK_CODE(code, lino, _end);
×
7423
  code = fillTableCountScanDataBlock(pSupp, dbName, varDataVal(stbName), ctbNum, pRes);
×
7424
  QUERY_CHECK_CODE(code, lino, _end);
×
7425

7426
_end:
×
7427
  if (code != TSDB_CODE_SUCCESS) {
×
7428
    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));
×
7429
  }
7430
  return code;
×
7431
}
7432

7433
static void destoryTableCountScanOperator(void* param) {
×
7434
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
×
7435
  blockDataDestroy(pTableCountScanInfo->pRes);
×
7436

7437
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
×
7438
  taosMemoryFreeClear(param);
×
7439
}
×
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