scanoperator.c 130.3 KB
Newer Older
H
Haojun Liao 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

16
#include "executorimpl.h"
H
Haojun Liao 已提交
17
#include "filter.h"
18
#include "function.h"
19
#include "functionMgt.h"
L
Liu Jicong 已提交
20
#include "os.h"
H
Haojun Liao 已提交
21
#include "querynodes.h"
22
#include "systable.h"
H
Haojun Liao 已提交
23
#include "tname.h"
24
#include "ttime.h"
H
Haojun Liao 已提交
25 26 27 28 29 30 31 32

#include "tdatablock.h"
#include "tmsg.h"

#include "query.h"
#include "tcompare.h"
#include "thash.h"
#include "ttypes.h"
33 34
#include "operator.h"
#include "querytask.h"
H
Haojun Liao 已提交
35

D
dapan1121 已提交
36 37
int32_t scanDebug = 0;

X
Xiaoyu Wang 已提交
38
#define MULTI_READER_MAX_TABLE_NUM   5000
H
Haojun Liao 已提交
39
#define SET_REVERSE_SCAN_FLAG(_info) ((_info)->scanFlag = REVERSE_SCAN)
40
#define SWITCH_ORDER(n)              (((n) = ((n) == TSDB_ORDER_ASC) ? TSDB_ORDER_DESC : TSDB_ORDER_ASC))
41

H
Haojun Liao 已提交
42 43 44 45 46 47 48 49 50
typedef struct STableMergeScanExecInfo {
  SFileBlockLoadRecorder blockRecorder;
  SSortExecInfo          sortExecInfo;
} STableMergeScanExecInfo;

typedef struct STableMergeScanSortSourceParam {
  SOperatorInfo* pOperator;
  int32_t        readerIdx;
  uint64_t       uid;
X
Xiaoyu Wang 已提交
51
  SSDataBlock*   inputBlock;
D
dapan1121 已提交
52
  bool           multiReader;
53
  STsdbReader*   dataReader;
H
Haojun Liao 已提交
54 55
} STableMergeScanSortSourceParam;

56 57 58 59 60 61 62 63 64 65
typedef struct STableCountScanOperatorInfo {
  SReadHandle  readHandle;
  SSDataBlock* pRes;

  STableCountScanSupp supp;

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

L
Liu Jicong 已提交
66
static bool processBlockWithProbability(const SSampleExecInfo* pInfo);
67

H
Haojun Liao 已提交
68
bool processBlockWithProbability(const SSampleExecInfo* pInfo) {
69 70 71 72 73 74 75 76 77 78 79 80
#if 0
  if (pInfo->sampleRatio == 1) {
    return true;
  }

  uint32_t val = taosRandR((uint32_t*) &pInfo->seed);
  return (val % ((uint32_t)(1/pInfo->sampleRatio))) == 0;
#else
  return true;
#endif
}

81
static void switchCtxOrder(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
H
Haojun Liao 已提交
82 83 84 85 86
  for (int32_t i = 0; i < numOfOutput; ++i) {
    SWITCH_ORDER(pCtx[i].order);
  }
}

87 88 89 90 91 92 93 94 95
static void getNextTimeWindow(SInterval* pInterval, STimeWindow* tw, int32_t order) {
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
  if (pInterval->intervalUnit != 'n' && pInterval->intervalUnit != 'y') {
    tw->skey += pInterval->sliding * factor;
    tw->ekey = tw->skey + pInterval->interval - 1;
    return;
  }

  int64_t key = tw->skey, interval = pInterval->interval;
96
  // convert key to second
97 98 99 100 101 102 103
  key = convertTimePrecision(key, pInterval->precision, TSDB_TIME_PRECISION_MILLI) / 1000;

  if (pInterval->intervalUnit == 'y') {
    interval *= 12;
  }

  struct tm tm;
104
  time_t    t = (time_t)key;
105
  taosLocalTime(&t, &tm, NULL);
106 107 108 109

  int mon = (int)(tm.tm_year * 12 + tm.tm_mon + interval * factor);
  tm.tm_year = mon / 12;
  tm.tm_mon = mon % 12;
wafwerar's avatar
wafwerar 已提交
110
  tw->skey = convertTimePrecision((int64_t)taosMktime(&tm) * 1000LL, TSDB_TIME_PRECISION_MILLI, pInterval->precision);
111 112 113 114

  mon = (int)(mon + interval);
  tm.tm_year = mon / 12;
  tm.tm_mon = mon % 12;
wafwerar's avatar
wafwerar 已提交
115
  tw->ekey = convertTimePrecision((int64_t)taosMktime(&tm) * 1000LL, TSDB_TIME_PRECISION_MILLI, pInterval->precision);
116 117 118 119

  tw->ekey -= 1;
}

120
static bool overlapWithTimeWindow(SInterval* pInterval, SDataBlockInfo* pBlockInfo, int32_t order) {
121 122 123 124 125 126 127
  STimeWindow w = {0};

  // 0 by default, which means it is not a interval operator of the upstream operator.
  if (pInterval->interval == 0) {
    return false;
  }

128
  if (order == TSDB_ORDER_ASC) {
129
    w = getAlignQueryTimeWindow(pInterval, pInterval->precision, pBlockInfo->window.skey);
130
    ASSERT(w.ekey >= pBlockInfo->window.skey);
131

132
    if (w.ekey < pBlockInfo->window.ekey) {
133 134 135
      return true;
    }

136 137
    while (1) {
      getNextTimeWindow(pInterval, &w, order);
138 139 140 141
      if (w.skey > pBlockInfo->window.ekey) {
        break;
      }

142
      ASSERT(w.ekey > pBlockInfo->window.ekey);
143
      if (TMAX(w.skey, pBlockInfo->window.skey) <= pBlockInfo->window.ekey) {
144 145 146 147
        return true;
      }
    }
  } else {
148
    w = getAlignQueryTimeWindow(pInterval, pInterval->precision, pBlockInfo->window.ekey);
149
    ASSERT(w.skey <= pBlockInfo->window.ekey);
150

151
    if (w.skey > pBlockInfo->window.skey) {
152 153 154
      return true;
    }

155
    while (1) {
156 157 158 159 160
      getNextTimeWindow(pInterval, &w, order);
      if (w.ekey < pBlockInfo->window.skey) {
        break;
      }

H
Haojun Liao 已提交
161
      ASSERT(w.skey < pBlockInfo->window.skey);
162
      if (pBlockInfo->window.skey <= TMIN(w.ekey, pBlockInfo->window.ekey)) {
163 164 165
        return true;
      }
    }
166 167 168 169 170
  }

  return false;
}

171 172 173 174 175 176 177 178 179 180 181
// this function is for table scanner to extract temporary results of upstream aggregate results.
static SResultRow* getTableGroupOutputBuf(SOperatorInfo* pOperator, uint64_t groupId, SFilePage** pPage) {
  if (pOperator->operatorType != QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
    return NULL;
  }

  int64_t buf[2] = {0};
  SET_RES_WINDOW_KEY((char*)buf, &groupId, sizeof(groupId), groupId);

  STableScanInfo* pTableScanInfo = pOperator->info;

S
slzhou 已提交
182 183
  SResultRowPosition* p1 = (SResultRowPosition*)tSimpleHashGet(pTableScanInfo->base.pdInfo.pAggSup->pResultRowHashTable,
                                                               buf, GET_RES_WINDOW_KEY_LEN(sizeof(groupId)));
184 185 186 187 188

  if (p1 == NULL) {
    return NULL;
  }

H
Haojun Liao 已提交
189
  *pPage = getBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, p1->pageId);
190 191 192
  if (NULL == *pPage) {
    return NULL;
  }
L
Liu Jicong 已提交
193

194 195 196 197 198 199
  return (SResultRow*)((char*)(*pPage) + p1->offset);
}

static int32_t doDynamicPruneDataBlock(SOperatorInfo* pOperator, SDataBlockInfo* pBlockInfo, uint32_t* status) {
  STableScanInfo* pTableScanInfo = pOperator->info;

H
Haojun Liao 已提交
200
  if (pTableScanInfo->base.pdInfo.pExprSup == NULL) {
201 202 203
    return TSDB_CODE_SUCCESS;
  }

H
Haojun Liao 已提交
204
  SExprSupp* pSup1 = pTableScanInfo->base.pdInfo.pExprSup;
205 206

  SFilePage*  pPage = NULL;
H
Haojun Liao 已提交
207
  SResultRow* pRow = getTableGroupOutputBuf(pOperator, pBlockInfo->id.groupId, &pPage);
208 209 210 211 212 213 214 215 216

  if (pRow == NULL) {
    return TSDB_CODE_SUCCESS;
  }

  bool notLoadBlock = true;
  for (int32_t i = 0; i < pSup1->numOfExprs; ++i) {
    int32_t functionId = pSup1->pCtx[i].functionId;

H
Haojun Liao 已提交
217
    SResultRowEntryInfo* pEntry = getResultEntryInfo(pRow, i, pTableScanInfo->base.pdInfo.pExprSup->rowEntryInfoOffset);
218 219 220 221 222 223 224 225 226

    int32_t reqStatus = fmFuncDynDataRequired(functionId, pEntry, &pBlockInfo->window);
    if (reqStatus != FUNC_DATA_REQUIRED_NOT_LOAD) {
      notLoadBlock = false;
      break;
    }
  }

  // release buffer pages
H
Haojun Liao 已提交
227
  releaseBufPage(pTableScanInfo->base.pdInfo.pAggSup->pResultBuf, pPage);
228 229 230 231 232 233 234 235

  if (notLoadBlock) {
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
  }

  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
236
static bool doFilterByBlockSMA(SFilterInfo* pFilterInfo, SColumnDataAgg** pColsAgg, int32_t numOfCols,
237
                               int32_t numOfRows) {
H
Haojun Liao 已提交
238
  if (pColsAgg == NULL || pFilterInfo == NULL) {
H
Haojun Liao 已提交
239 240 241
    return true;
  }

H
Haojun Liao 已提交
242
  bool keep = filterRangeExecute(pFilterInfo, pColsAgg, numOfCols, numOfRows);
H
Haojun Liao 已提交
243 244 245
  return keep;
}

H
Haojun Liao 已提交
246
static bool doLoadBlockSMA(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
247
  bool    allColumnsHaveAgg = true;
248
  int32_t code = tsdbRetrieveDatablockSMA(pTableScanInfo->dataReader, pBlock, &allColumnsHaveAgg);
H
Haojun Liao 已提交
249
  if (code != TSDB_CODE_SUCCESS) {
250
    T_LONG_JMP(pTaskInfo->env, code);
H
Haojun Liao 已提交
251 252 253 254 255 256 257 258
  }

  if (!allColumnsHaveAgg) {
    return false;
  }
  return true;
}

H
Haojun Liao 已提交
259
static void doSetTagColumnData(STableScanBase* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
260
                               int32_t rows) {
H
Haojun Liao 已提交
261 262 263
  if (pTableScanInfo->pseudoSup.numOfExprs > 0) {
    SExprSupp* pSup = &pTableScanInfo->pseudoSup;

264
    int32_t code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
265
                                          GET_TASKID(pTaskInfo), &pTableScanInfo->metaCache);
H
Haojun Liao 已提交
266
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
H
Haojun Liao 已提交
267
    if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_PAR_TABLE_NOT_EXIST) {
H
Haojun Liao 已提交
268 269
      T_LONG_JMP(pTaskInfo->env, code);
    }
H
Haojun Liao 已提交
270 271 272

    // reset the error code.
    terrno = 0;
H
Haojun Liao 已提交
273 274 275
  }
}

276
bool applyLimitOffset(SLimitInfo* pLimitInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
277
  SLimit*     pLimit = &pLimitInfo->limit;
H
Haojun Liao 已提交
278
  const char* id = GET_TASKID(pTaskInfo);
279

280
  if (pLimitInfo->remainOffset > 0) {
281 282
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
      pLimitInfo->remainOffset -= pBlock->info.rows;
H
Haojun Liao 已提交
283
      blockDataEmpty(pBlock);
H
Haojun Liao 已提交
284
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset, id);
285
      return false;
286
    } else {
287
      blockDataTrimFirstRows(pBlock, pLimitInfo->remainOffset);
288 289 290 291 292 293
      pLimitInfo->remainOffset = 0;
    }
  }

  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
    // limit the output rows
294
    int32_t keep = (int32_t)(pLimit->limit - pLimitInfo->numOfOutputRows);
295
    blockDataKeepFirstNRows(pBlock, keep);
296 297

    pLimitInfo->numOfOutputRows += pBlock->info.rows;
H
Haojun Liao 已提交
298
    qDebug("output limit %" PRId64 " has reached, %s", pLimit->limit, id);
299
    return true;
300
  }
301

302
  pLimitInfo->numOfOutputRows += pBlock->info.rows;
303
  return false;
304 305
}

H
Haojun Liao 已提交
306
static int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanBase* pTableScanInfo, SSDataBlock* pBlock,
L
Liu Jicong 已提交
307
                             uint32_t* status) {
S
slzhou 已提交
308
  SExecTaskInfo*          pTaskInfo = pOperator->pTaskInfo;
309
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
H
Haojun Liao 已提交
310 311

  pCost->totalBlocks += 1;
312
  pCost->totalRows += pBlock->info.rows;
313

H
Haojun Liao 已提交
314
  bool loadSMA = false;
H
Haojun Liao 已提交
315
  *status = pTableScanInfo->dataBlockLoadFlag;
H
Haojun Liao 已提交
316
  if (pOperator->exprSupp.pFilterInfo != NULL ||
317
      overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order)) {
318 319 320 321
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
  }

  SDataBlockInfo* pBlockInfo = &pBlock->info;
322
  taosMemoryFreeClear(pBlock->pBlockAgg);
323 324

  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
X
Xiaoyu Wang 已提交
325
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
326
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
327
    pCost->filterOutBlocks += 1;
328
    pCost->totalRows += pBlock->info.rows;
329
    tsdbReleaseDataBlock(pTableScanInfo->dataReader);
330 331
    return TSDB_CODE_SUCCESS;
  } else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
X
Xiaoyu Wang 已提交
332 333 334
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", uid:%" PRIu64,
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows,
           pBlockInfo->id.uid);
G
Ganlin Zhao 已提交
335
    doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
336
    pCost->skipBlocks += 1;
337
    tsdbReleaseDataBlock(pTableScanInfo->dataReader);
338
    return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
339
  } else if (*status == FUNC_DATA_REQUIRED_SMA_LOAD) {
340
    pCost->loadBlockStatis += 1;
L
Liu Jicong 已提交
341
    loadSMA = true;  // mark the operation of load sma;
H
Haojun Liao 已提交
342
    bool success = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo);
L
Liu Jicong 已提交
343
    if (success) {  // failed to load the block sma data, data block statistics does not exist, load data block instead
X
Xiaoyu Wang 已提交
344
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64, GET_TASKID(pTaskInfo),
345
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
G
Ganlin Zhao 已提交
346
      doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
347
      tsdbReleaseDataBlock(pTableScanInfo->dataReader);
348 349
      return TSDB_CODE_SUCCESS;
    } else {
350
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
351
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
352
    }
H
Haojun Liao 已提交
353
  }
354

H
Haojun Liao 已提交
355
  ASSERT(*status == FUNC_DATA_REQUIRED_DATA_LOAD);
356

H
Haojun Liao 已提交
357
  // try to filter data block according to sma info
H
Haojun Liao 已提交
358
  if (pOperator->exprSupp.pFilterInfo != NULL && (!loadSMA)) {
359 360 361
    bool success = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo);
    if (success) {
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
H
Haojun Liao 已提交
362
      bool   keep = doFilterByBlockSMA(pOperator->exprSupp.pFilterInfo, pBlock->pBlockAgg, size, pBlockInfo->rows);
363
      if (!keep) {
X
Xiaoyu Wang 已提交
364 365
        qDebug("%s data block filter out by block SMA, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
               GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
366 367 368
        pCost->filterOutBlocks += 1;
        (*status) = FUNC_DATA_REQUIRED_FILTEROUT;

369
        tsdbReleaseDataBlock(pTableScanInfo->dataReader);
370 371
        return TSDB_CODE_SUCCESS;
      }
372
    }
H
Haojun Liao 已提交
373
  }
374

375 376 377
  // free the sma info, since it should not be involved in later computing process.
  taosMemoryFreeClear(pBlock->pBlockAgg);

378
  // try to filter data block according to current results
379 380
  doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
  if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
X
Xiaoyu Wang 已提交
381 382
    qDebug("%s data block skipped due to dynamic prune, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64,
           GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
383
    pCost->skipBlocks += 1;
384
    tsdbReleaseDataBlock(pTableScanInfo->dataReader);
385
    *status = FUNC_DATA_REQUIRED_FILTEROUT;
386 387 388
    return TSDB_CODE_SUCCESS;
  }

H
Haojun Liao 已提交
389 390
  pCost->totalCheckedRows += pBlock->info.rows;
  pCost->loadBlocks += 1;
391

H
Haojun Liao 已提交
392 393
  SSDataBlock* p = tsdbRetrieveDataBlock(pTableScanInfo->dataReader, NULL);
  if (p == NULL) {
H
Haojun Liao 已提交
394
    return terrno;
H
Haojun Liao 已提交
395 396
  }

H
Haojun Liao 已提交
397
  ASSERT(p == pBlock);
398
  doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
399

H
Haojun Liao 已提交
400 401
  // restore the previous value
  pCost->totalRows -= pBlock->info.rows;
402

H
Haojun Liao 已提交
403
  if (pOperator->exprSupp.pFilterInfo != NULL) {
404
    int64_t st = taosGetTimestampUs();
H
Haojun Liao 已提交
405
    doFilter(pBlock, pOperator->exprSupp.pFilterInfo, &pTableScanInfo->matchInfo);
406

407 408
    double el = (taosGetTimestampUs() - st) / 1000.0;
    pTableScanInfo->readRecorder.filterTime += el;
409

410 411
    if (pBlock->info.rows == 0) {
      pCost->filterOutBlocks += 1;
D
dapan1121 已提交
412
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%" PRId64 ", elapsed time:%.2f ms",
413 414 415 416
             GET_TASKID(pTaskInfo), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows, el);
    } else {
      qDebug("%s data block filter applied, elapsed time:%.2f ms", GET_TASKID(pTaskInfo), el);
    }
417 418
  }

419
  bool limitReached = applyLimitOffset(&pTableScanInfo->limitInfo, pBlock, pTaskInfo);
X
Xiaoyu Wang 已提交
420
  if (limitReached) {  // set operator flag is done
421 422
    setOperatorCompleted(pOperator);
  }
423

H
Haojun Liao 已提交
424
  pCost->totalRows += pBlock->info.rows;
H
Haojun Liao 已提交
425 426 427
  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
428
static void prepareForDescendingScan(STableScanBase* pTableScanInfo, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
H
Haojun Liao 已提交
429 430 431
  SET_REVERSE_SCAN_FLAG(pTableScanInfo);

  switchCtxOrder(pCtx, numOfOutput);
432
  pTableScanInfo->cond.order = TSDB_ORDER_DESC;
H
Haojun Liao 已提交
433 434
  STimeWindow* pTWindow = &pTableScanInfo->cond.twindows;
  TSWAP(pTWindow->skey, pTWindow->ekey);
H
Haojun Liao 已提交
435 436
}

437 438
typedef struct STableCachedVal {
  const char* pName;
439
  STag*       pTags;
440 441
} STableCachedVal;

442 443 444 445 446 447 448 449 450 451 452
static void freeTableCachedVal(void* param) {
  if (param == NULL) {
    return;
  }

  STableCachedVal* pVal = param;
  taosMemoryFree((void*)pVal->pName);
  taosMemoryFree(pVal->pTags);
  taosMemoryFree(pVal);
}

H
Haojun Liao 已提交
453 454
static STableCachedVal* createTableCacheVal(const SMetaReader* pMetaReader) {
  STableCachedVal* pVal = taosMemoryMalloc(sizeof(STableCachedVal));
455
  pVal->pName = taosStrdup(pMetaReader->me.name);
H
Haojun Liao 已提交
456 457 458 459
  pVal->pTags = NULL;

  // only child table has tag value
  if (pMetaReader->me.type == TSDB_CHILD_TABLE) {
460
    STag* pTag = (STag*)pMetaReader->me.ctbEntry.pTags;
H
Haojun Liao 已提交
461 462 463 464 465 466 467
    pVal->pTags = taosMemoryMalloc(pTag->len);
    memcpy(pVal->pTags, pTag, pTag->len);
  }

  return pVal;
}

468 469
// const void *key, size_t keyLen, void *value
static void freeCachedMetaItem(const void* key, size_t keyLen, void* value) { freeTableCachedVal(value); }
470

471 472 473 474 475
static void doSetNullValue(SSDataBlock* pBlock, const SExprInfo* pExpr, int32_t numOfExpr) {
  for (int32_t j = 0; j < numOfExpr; ++j) {
    int32_t dstSlotId = pExpr[j].base.resSchema.slotId;

    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
476
    colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
477 478 479
  }
}

480 481
int32_t addTagPseudoColumnData(SReadHandle* pHandle, const SExprInfo* pExpr, int32_t numOfExpr, SSDataBlock* pBlock,
                               int32_t rows, const char* idStr, STableMetaCacheInfo* pCache) {
482
  // currently only the tbname pseudo column
483
  if (numOfExpr <= 0) {
H
Haojun Liao 已提交
484
    return TSDB_CODE_SUCCESS;
485 486
  }

487 488
  int32_t code = 0;

489 490 491 492
  // backup the rows
  int32_t backupRows = pBlock->info.rows;
  pBlock->info.rows = rows;

493
  bool            freeReader = false;
494
  STableCachedVal val = {0};
495 496

  SMetaReader mr = {0};
497
  LRUHandle*  h = NULL;
498

499 500 501
  // todo refactor: extract method
  // the handling of the null data should be packed in the extracted method

502
  // 1. check if it is existed in meta cache
503
  if (pCache == NULL) {
504
    metaReaderInit(&mr, pHandle->meta, 0);
H
Haojun Liao 已提交
505
    code = metaGetTableEntryByUidCache(&mr, pBlock->info.id.uid);
506
    if (code != TSDB_CODE_SUCCESS) {
507
      // when encounter the TSDB_CODE_PAR_TABLE_NOT_EXIST error, we proceed.
H
Haojun Liao 已提交
508
      if (terrno == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
S
slzhou 已提交
509 510
        qWarn("failed to get table meta, table may have been dropped, uid:0x%" PRIx64 ", code:%s, %s",
              pBlock->info.id.uid, tstrerror(terrno), idStr);
511 512 513

        // append null value before return to caller, since the caller will ignore this error code and proceed
        doSetNullValue(pBlock, pExpr, numOfExpr);
H
Haojun Liao 已提交
514
      } else {
S
slzhou 已提交
515 516
        qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", pBlock->info.id.uid, tstrerror(terrno),
               idStr);
H
Haojun Liao 已提交
517
      }
518 519 520 521 522
      metaReaderClear(&mr);
      return terrno;
    }

    metaReaderReleaseLock(&mr);
523

524 525
    val.pName = mr.me.name;
    val.pTags = (STag*)mr.me.ctbEntry.pTags;
526 527

    freeReader = true;
528
  } else {
529 530
    pCache->metaFetch += 1;

H
Haojun Liao 已提交
531
    h = taosLRUCacheLookup(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(pBlock->info.id.uid));
532 533
    if (h == NULL) {
      metaReaderInit(&mr, pHandle->meta, 0);
H
Haojun Liao 已提交
534
      code = metaGetTableEntryByUidCache(&mr, pBlock->info.id.uid);
535
      if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
536
        if (terrno == TSDB_CODE_PAR_TABLE_NOT_EXIST) {
537
          qWarn("failed to get table meta, table may have been dropped, uid:0x%" PRIx64 ", code:%s, %s",
H
Haojun Liao 已提交
538
                pBlock->info.id.uid, tstrerror(terrno), idStr);
539 540
          // append null value before return to caller, since the caller will ignore this error code and proceed
          doSetNullValue(pBlock, pExpr, numOfExpr);
H
Haojun Liao 已提交
541
        } else {
H
Haojun Liao 已提交
542
          qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", pBlock->info.id.uid, tstrerror(terrno),
543
                 idStr);
H
Haojun Liao 已提交
544
        }
545 546 547 548 549 550
        metaReaderClear(&mr);
        return terrno;
      }

      metaReaderReleaseLock(&mr);

H
Haojun Liao 已提交
551
      STableCachedVal* pVal = createTableCacheVal(&mr);
552

H
Haojun Liao 已提交
553
      val = *pVal;
554
      freeReader = true;
H
Haojun Liao 已提交
555

H
Haojun Liao 已提交
556
      int32_t ret = taosLRUCacheInsert(pCache->pTableMetaEntryCache, &pBlock->info.id.uid, sizeof(uint64_t), pVal,
557
                                       sizeof(STableCachedVal), freeCachedMetaItem, NULL, TAOS_LRU_PRIORITY_LOW);
558 559 560 561 562 563 564 565
      if (ret != TAOS_LRU_STATUS_OK) {
        qError("failed to put meta into lru cache, code:%d, %s", ret, idStr);
        freeTableCachedVal(pVal);
      }
    } else {
      pCache->cacheHit += 1;
      STableCachedVal* pVal = taosLRUCacheValue(pCache->pTableMetaEntryCache, h);
      val = *pVal;
H
Haojun Liao 已提交
566

H
Haojun Liao 已提交
567
      taosLRUCacheRelease(pCache->pTableMetaEntryCache, h, false);
568
    }
H
Haojun Liao 已提交
569

570 571
    qDebug("retrieve table meta from cache:%" PRIu64 ", hit:%" PRIu64 " miss:%" PRIu64 ", %s", pCache->metaFetch,
           pCache->cacheHit, (pCache->metaFetch - pCache->cacheHit), idStr);
H
Haojun Liao 已提交
572
  }
573

574 575
  for (int32_t j = 0; j < numOfExpr; ++j) {
    const SExprInfo* pExpr1 = &pExpr[j];
576
    int32_t          dstSlotId = pExpr1->base.resSchema.slotId;
577 578

    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
D
dapan1121 已提交
579
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
580

581
    int32_t functionId = pExpr1->pExpr->_function.functionId;
582 583 584

    // this is to handle the tbname
    if (fmIsScanPseudoColumnFunc(functionId)) {
585
      setTbNameColData(pBlock, pColInfoData, functionId, val.pName);
586
    } else {  // these are tags
wmmhello's avatar
wmmhello 已提交
587
      STagVal tagVal = {0};
588 589
      tagVal.cid = pExpr1->base.pParam[0].pCol->colId;
      const char* p = metaGetTableTagVal(val.pTags, pColInfoData->info.type, &tagVal);
wmmhello's avatar
wmmhello 已提交
590

591 592 593 594
      char* data = NULL;
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
        data = tTagValToData((const STagVal*)p, false);
      } else {
wmmhello's avatar
wmmhello 已提交
595
        data = (char*)p;
wmmhello's avatar
wmmhello 已提交
596
      }
597

H
Haojun Liao 已提交
598 599
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
      if (isNullVal) {
600
        colDataSetNNULL(pColInfoData, 0, pBlock->info.rows);
H
Haojun Liao 已提交
601
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
D
dapan1121 已提交
602
        code = colDataSetNItems(pColInfoData, 0, data, pBlock->info.rows, false);
H
Haojun Liao 已提交
603 604 605
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
          taosMemoryFree(data);
        }
D
dapan1121 已提交
606 607 608 609 610 611
        if (code) {
          if (freeReader) {
            metaReaderClear(&mr);
          }
          return code;
        }
L
Liu Jicong 已提交
612
      } else {  // todo opt for json tag
H
Haojun Liao 已提交
613
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
614
          colDataSetVal(pColInfoData, i, data, false);
H
Haojun Liao 已提交
615
        }
616 617 618 619
      }
    }
  }

620 621
  // restore the rows
  pBlock->info.rows = backupRows;
622 623 624 625
  if (freeReader) {
    metaReaderClear(&mr);
  }

H
Haojun Liao 已提交
626
  return TSDB_CODE_SUCCESS;
627 628
}

H
Haojun Liao 已提交
629
void setTbNameColData(const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId, const char* name) {
630 631 632
  struct SScalarFuncExecFuncs fpSet = {0};
  fmGetScalarFuncExecFuncs(functionId, &fpSet);

H
Haojun Liao 已提交
633
  size_t len = TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE;
634
  char   buf[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
H
Haojun Liao 已提交
635 636 637
  STR_TO_VARSTR(buf, name)

  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, len, 1);
638

H
Haojun Liao 已提交
639
  colInfoDataEnsureCapacity(&infoData, 1, false);
640
  colDataSetVal(&infoData, 0, buf, false);
641

H
Haojun Liao 已提交
642
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .columnData = &infoData};
643
  SScalarParam param = {.columnData = pColInfoData};
H
Haojun Liao 已提交
644 645 646 647 648 649 650

  if (fpSet.process != NULL) {
    fpSet.process(&srcParam, 1, &param);
  } else {
    qError("failed to get the corresponding callback function, functionId:%d", functionId);
  }

D
dapan1121 已提交
651
  colDataDestroy(&infoData);
652 653
}

654
static SSDataBlock* doTableScanImpl(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
655
  STableScanInfo* pTableScanInfo = pOperator->info;
656
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
L
Liu Jicong 已提交
657
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
D
dapan1121 已提交
658 659
  bool            hasNext = false;
  int32_t         code = TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
660

661 662
  int64_t st = taosGetTimestampUs();

D
dapan1121 已提交
663 664 665 666 667 668 669 670 671 672
  while (true) {
    code = tsdbNextDataBlock(pTableScanInfo->base.dataReader, &hasNext);
    if (code) {
      tsdbReleaseDataBlock(pTableScanInfo->base.dataReader);
      T_LONG_JMP(pTaskInfo->env, code);
    }

    if (!hasNext) {
      break;
    }
X
Xiaoyu Wang 已提交
673

674
    if (isTaskKilled(pTaskInfo)) {
X
Xiaoyu Wang 已提交
675
      tsdbReleaseDataBlock(pTableScanInfo->base.dataReader);
676
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
677
    }
H
Haojun Liao 已提交
678

679
    if (pOperator->status == OP_EXEC_DONE) {
X
Xiaoyu Wang 已提交
680
      tsdbReleaseDataBlock(pTableScanInfo->base.dataReader);
681 682 683
      break;
    }

684 685 686 687 688 689
    // process this data block based on the probabilities
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
    if (!processThisBlock) {
      continue;
    }

D
dapan1121 已提交
690
    if (pBlock->info.id.uid) {
691
      pBlock->info.id.groupId = getTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
D
dapan1121 已提交
692
    }
693

694
    uint32_t status = 0;
695
    code = loadDataBlock(pOperator, &pTableScanInfo->base, pBlock, &status);
696
    if (code != TSDB_CODE_SUCCESS) {
697
      T_LONG_JMP(pTaskInfo->env, code);
698
    }
699

700 701 702
    // current block is filter out according to filter condition, continue load the next block
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
      continue;
703
    }
704

H
Haojun Liao 已提交
705 706
    pOperator->resultInfo.totalRows = pTableScanInfo->base.readRecorder.totalRows;
    pTableScanInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
707

H
Haojun Liao 已提交
708
    pOperator->cost.totalCost = pTableScanInfo->base.readRecorder.elapsedTime;
709 710

    // todo refactor
H
Haojun Liao 已提交
711
    /*pTableScanInfo->lastStatus.uid = pBlock->info.id.uid;*/
L
Liu Jicong 已提交
712
    /*pTableScanInfo->lastStatus.ts = pBlock->info.window.ekey;*/
X
Xiaoyu Wang 已提交
713 714 715
    //    pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__SNAPSHOT_DATA;
    //    pTaskInfo->streamInfo.lastStatus.uid = pBlock->info.id.uid;
    //    pTaskInfo->streamInfo.lastStatus.ts = pBlock->info.window.ekey;
716

717
    return pBlock;
H
Haojun Liao 已提交
718 719 720 721
  }
  return NULL;
}

H
Haojun Liao 已提交
722
static SSDataBlock* doGroupedTableScan(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
723 724 725 726
  STableScanInfo* pTableScanInfo = pOperator->info;
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;

  // The read handle is not initialized yet, since no qualified tables exists
H
Haojun Liao 已提交
727
  if (pTableScanInfo->base.dataReader == NULL || pOperator->status == OP_EXEC_DONE) {
H
Haojun Liao 已提交
728 729 730
    return NULL;
  }

731 732
  // do the ascending order traverse in the first place.
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
H
Haojun Liao 已提交
733 734 735
    SSDataBlock* p = doTableScanImpl(pOperator);
    if (p != NULL) {
      return p;
H
Haojun Liao 已提交
736 737
    }

738
    pTableScanInfo->scanTimes += 1;
739

740
    if (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
741
      setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
G
Ganlin Zhao 已提交
742 743
      pTableScanInfo->base.scanFlag = MAIN_SCAN;
      pTableScanInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
744
      qDebug("start to repeat ascending order scan data blocks due to query func required, %s", GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
745

746
      // do prepare for the next round table scan operation
H
Haojun Liao 已提交
747
      tsdbReaderReset(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
H
Haojun Liao 已提交
748
    }
749
  }
H
Haojun Liao 已提交
750

751
  int32_t total = pTableScanInfo->scanInfo.numOfAsc + pTableScanInfo->scanInfo.numOfDesc;
752
  if (pTableScanInfo->scanTimes < total) {
H
Haojun Liao 已提交
753 754 755
    if (pTableScanInfo->base.cond.order == TSDB_ORDER_ASC) {
      prepareForDescendingScan(&pTableScanInfo->base, pOperator->exprSupp.pCtx, 0);
      tsdbReaderReset(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
756
      qDebug("%s start to descending order scan data blocks due to query func required", GET_TASKID(pTaskInfo));
757
    }
H
Haojun Liao 已提交
758

759
    while (pTableScanInfo->scanTimes < total) {
H
Haojun Liao 已提交
760 761 762
      SSDataBlock* p = doTableScanImpl(pOperator);
      if (p != NULL) {
        return p;
763
      }
H
Haojun Liao 已提交
764

765
      pTableScanInfo->scanTimes += 1;
H
Haojun Liao 已提交
766

767
      if (pTableScanInfo->scanTimes < total) {
768
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
G
Ganlin Zhao 已提交
769
        pTableScanInfo->base.scanFlag = MAIN_SCAN;
H
Haojun Liao 已提交
770

771
        qDebug("%s start to repeat descending order scan data blocks", GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
772
        tsdbReaderReset(pTableScanInfo->base.dataReader, &pTableScanInfo->base.cond);
773
      }
H
Haojun Liao 已提交
774 775 776
    }
  }

wmmhello's avatar
wmmhello 已提交
777 778 779 780 781 782 783
  return NULL;
}

static SSDataBlock* doTableScan(SOperatorInfo* pOperator) {
  STableScanInfo* pInfo = pOperator->info;
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;

784
  // scan table one by one sequentially
L
Liu Jicong 已提交
785
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
X
Xiaoyu Wang 已提交
786
    int32_t       numOfTables = 0;  // tableListGetSize(pTaskInfo->pTableListInfo);
787
    STableKeyInfo tInfo = {0};
H
Haojun Liao 已提交
788

L
Liu Jicong 已提交
789
    while (1) {
H
Haojun Liao 已提交
790
      SSDataBlock* result = doGroupedTableScan(pOperator);
H
Haojun Liao 已提交
791
      if (result || (pOperator->status == OP_EXEC_DONE) || isTaskKilled(pTaskInfo)) {
L
Liu Jicong 已提交
792 793
        return result;
      }
H
Haojun Liao 已提交
794

L
Liu Jicong 已提交
795 796
      // if no data, switch to next table and continue scan
      pInfo->currentTable++;
797 798

      taosRLockLatch(&pTaskInfo->lock);
799
      numOfTables = tableListGetSize(pInfo->base.pTableListInfo);
800

H
Haojun Liao 已提交
801
      if (pInfo->currentTable >= numOfTables) {
H
Haojun Liao 已提交
802
        qDebug("all table checked in table list, total:%d, return NULL, %s", numOfTables, GET_TASKID(pTaskInfo));
803
        taosRUnLockLatch(&pTaskInfo->lock);
L
Liu Jicong 已提交
804 805
        return NULL;
      }
H
Haojun Liao 已提交
806

X
Xiaoyu Wang 已提交
807
      tInfo = *(STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->currentTable);
808 809 810 811
      taosRUnLockLatch(&pTaskInfo->lock);

      tsdbSetTableList(pInfo->base.dataReader, &tInfo, 1);
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d/%d %s", tInfo.uid, numOfTables,
H
Haojun Liao 已提交
812
             pInfo->currentTable, numOfTables, GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
813

H
Haojun Liao 已提交
814
      tsdbReaderReset(pInfo->base.dataReader, &pInfo->base.cond);
L
Liu Jicong 已提交
815 816
      pInfo->scanTimes = 0;
    }
817 818
  } else {  // scan table group by group sequentially
    if (pInfo->currentGroupId == -1) {
819
      if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
H
Haojun Liao 已提交
820
        setOperatorCompleted(pOperator);
821 822
        return NULL;
      }
823

5
54liuyao 已提交
824
      int32_t        num = 0;
825
      STableKeyInfo* pList = NULL;
826
      tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, &pList, &num);
H
Haojun Liao 已提交
827
      ASSERT(pInfo->base.dataReader == NULL);
828

L
Liu Jicong 已提交
829
      int32_t code = tsdbReaderOpen(pInfo->base.readHandle.vnode, &pInfo->base.cond, pList, num, pInfo->pResBlock,
D
dapan1121 已提交
830
                                    (STsdbReader**)&pInfo->base.dataReader, GET_TASKID(pTaskInfo), pInfo->countOnly);
831 832 833
      if (code != TSDB_CODE_SUCCESS) {
        T_LONG_JMP(pTaskInfo->env, code);
      }
834 835 836 837

      if (pInfo->pResBlock->info.capacity > pOperator->resultInfo.capacity) {
        pOperator->resultInfo.capacity = pInfo->pResBlock->info.capacity;
      }
wmmhello's avatar
wmmhello 已提交
838
    }
H
Haojun Liao 已提交
839

H
Haojun Liao 已提交
840
    SSDataBlock* result = doGroupedTableScan(pOperator);
841 842 843
    if (result != NULL) {
      return result;
    }
H
Haojun Liao 已提交
844

845
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pInfo->base.pTableListInfo)) {
H
Haojun Liao 已提交
846
      setOperatorCompleted(pOperator);
847 848
      return NULL;
    }
wmmhello's avatar
wmmhello 已提交
849

850 851
    // reset value for the next group data output
    pOperator->status = OP_OPENED;
852
    resetLimitInfoForNextGroup(&pInfo->base.limitInfo);
wmmhello's avatar
wmmhello 已提交
853

5
54liuyao 已提交
854
    int32_t        num = 0;
855
    STableKeyInfo* pList = NULL;
856
    tableListGetGroupList(pInfo->base.pTableListInfo, pInfo->currentGroupId, &pList, &num);
wmmhello's avatar
wmmhello 已提交
857

H
Haojun Liao 已提交
858 859
    tsdbSetTableList(pInfo->base.dataReader, pList, num);
    tsdbReaderReset(pInfo->base.dataReader, &pInfo->base.cond);
860
    pInfo->scanTimes = 0;
wmmhello's avatar
wmmhello 已提交
861

H
Haojun Liao 已提交
862
    result = doGroupedTableScan(pOperator);
863 864 865
    if (result != NULL) {
      return result;
    }
866

H
Haojun Liao 已提交
867
    setOperatorCompleted(pOperator);
868 869
    return NULL;
  }
H
Haojun Liao 已提交
870 871
}

872 873
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
874
  STableScanInfo*         pTableScanInfo = pOptr->info;
H
Haojun Liao 已提交
875
  *pRecorder = pTableScanInfo->base.readRecorder;
876 877 878 879 880
  *pOptrExplain = pRecorder;
  *len = sizeof(SFileBlockLoadRecorder);
  return 0;
}

881 882
static void destroyTableScanBase(STableScanBase* pBase) {
  cleanupQueryTableDataCond(&pBase->cond);
H
Haojun Liao 已提交
883

884 885
  tsdbReaderClose(pBase->dataReader);
  pBase->dataReader = NULL;
886

887 888
  if (pBase->matchInfo.pList != NULL) {
    taosArrayDestroy(pBase->matchInfo.pList);
889
  }
L
Liu Jicong 已提交
890

891
  tableListDestroy(pBase->pTableListInfo);
892 893 894 895 896 897 898 899
  taosLRUCacheCleanup(pBase->metaCache.pTableMetaEntryCache);
  cleanupExprSupp(&pBase->pseudoSup);
}

static void destroyTableScanOperatorInfo(void* param) {
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
  blockDataDestroy(pTableScanInfo->pResBlock);
  destroyTableScanBase(&pTableScanInfo->base);
D
dapan1121 已提交
900
  taosMemoryFreeClear(param);
901 902
}

903
SOperatorInfo* createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
904
                                           STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo) {
X
Xiaoyu Wang 已提交
905
  int32_t         code = 0;
H
Haojun Liao 已提交
906 907 908
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
909
    code = TSDB_CODE_OUT_OF_MEMORY;
910
    goto _error;
H
Haojun Liao 已提交
911 912
  }

913
  SScanPhysiNode*     pScanNode = &pTableScanNode->scan;
H
Haojun Liao 已提交
914
  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
915 916

  int32_t numOfCols = 0;
X
Xiaoyu Wang 已提交
917 918
  code =
      extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->base.matchInfo);
919 920 921 922
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

H
Haojun Liao 已提交
923
  initLimitInfo(pScanNode->node.pLimit, pScanNode->node.pSlimit, &pInfo->base.limitInfo);
H
Haojun Liao 已提交
924
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode);
925
  if (code != TSDB_CODE_SUCCESS) {
926
    goto _error;
927 928
  }

H
Haojun Liao 已提交
929
  if (pScanNode->pScanPseudoCols != NULL) {
H
Haojun Liao 已提交
930
    SExprSupp* pSup = &pInfo->base.pseudoSup;
H
Haojun Liao 已提交
931
    pSup->pExprInfo = createExprInfo(pScanNode->pScanPseudoCols, NULL, &pSup->numOfExprs);
932
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset);
933 934
  }

935
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
G
Ganlin Zhao 已提交
936
  pInfo->base.scanFlag = (pInfo->scanInfo.numOfAsc > 1) ? PRE_SCAN : MAIN_SCAN;
H
Haojun Liao 已提交
937

H
Haojun Liao 已提交
938 939
  pInfo->base.pdInfo.interval = extractIntervalInfo(pTableScanNode);
  pInfo->base.readHandle = *readHandle;
H
Haojun Liao 已提交
940 941
  pInfo->base.dataBlockLoadFlag = pTableScanNode->dataRequired;

942 943
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
  pInfo->sample.seed = taosGetTimestampSec();
944

H
Haojun Liao 已提交
945
  initResultSizeInfo(&pOperator->resultInfo, 4096);
H
Haojun Liao 已提交
946
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
X
Xiaoyu Wang 已提交
947
  //  blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);
H
Haojun Liao 已提交
948

H
Haojun Liao 已提交
949 950 951
  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
H
Haojun Liao 已提交
952 953
  }

wmmhello's avatar
wmmhello 已提交
954
  pInfo->currentGroupId = -1;
955
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
956
  pInfo->hasGroupByTag = pTableScanNode->pGroupTags ? true : false;
957

L
Liu Jicong 已提交
958 959
  setOperatorInfo(pOperator, "TableScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
                  pTaskInfo);
960
  pOperator->exprSupp.numOfExprs = numOfCols;
961

962
  pInfo->base.pTableListInfo = pTableListInfo;
H
Haojun Liao 已提交
963 964
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
965 966 967
    code = terrno;
    goto _error;
  }
968

D
dapan1121 已提交
969 970 971 972
  if (scanDebug) {
    pInfo->countOnly = true;
  }

H
Haojun Liao 已提交
973
  taosLRUCacheSetStrictCapacity(pInfo->base.metaCache.pTableMetaEntryCache, false);
974 975
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScan, NULL, destroyTableScanOperatorInfo,
                                         optrDefaultBufFn, getTableScannerExecInfo);
976 977 978

  // for non-blocking operator, the open cost is always 0
  pOperator->cost.openCost = 0;
H
Haojun Liao 已提交
979
  return pOperator;
980

981
_error:
982 983 984
  if (pInfo != NULL) {
    destroyTableScanOperatorInfo(pInfo);
  }
985

986 987
  taosMemoryFreeClear(pOperator);
  pTaskInfo->code = code;
988
  return NULL;
H
Haojun Liao 已提交
989 990
}

991
SOperatorInfo* createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
992
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
L
Liu Jicong 已提交
993
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
994

H
Haojun Liao 已提交
995
  pInfo->base.dataReader = pReadHandle;
L
Liu Jicong 已提交
996
  //  pInfo->prevGroupId       = -1;
H
Haojun Liao 已提交
997

L
Liu Jicong 已提交
998 999
  setOperatorInfo(pOperator, "TableSeqScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN, false, OP_NOT_OPENED,
                  pInfo, pTaskInfo);
1000
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableScanImpl, NULL, NULL, optrDefaultBufFn, NULL);
H
Haojun Liao 已提交
1001 1002 1003
  return pOperator;
}

1004
FORCE_INLINE void doClearBufferedBlocks(SStreamScanInfo* pInfo) {
5
54liuyao 已提交
1005
  qDebug("clear buff blocks:%d", (int32_t)taosArrayGetSize(pInfo->pBlockLists));
L
Liu Jicong 已提交
1006 1007
  taosArrayClear(pInfo->pBlockLists);
  pInfo->validBlockIndex = 0;
H
Haojun Liao 已提交
1008 1009
}

1010
static bool isSessionWindow(SStreamScanInfo* pInfo) {
H
Haojun Liao 已提交
1011
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION;
5
54liuyao 已提交
1012 1013
}

1014
static bool isStateWindow(SStreamScanInfo* pInfo) {
1015
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE;
5
54liuyao 已提交
1016
}
5
54liuyao 已提交
1017

L
Liu Jicong 已提交
1018
static bool isIntervalWindow(SStreamScanInfo* pInfo) {
1019 1020 1021
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL ||
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL ||
         pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL;
5
54liuyao 已提交
1022 1023 1024
}

static bool isSignleIntervalWindow(SStreamScanInfo* pInfo) {
1025
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL;
L
Liu Jicong 已提交
1026 1027
}

1028 1029 1030 1031
static bool isSlidingWindow(SStreamScanInfo* pInfo) {
  return isIntervalWindow(pInfo) && pInfo->interval.interval != pInfo->interval.sliding;
}

1032
static void setGroupId(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t groupColIndex, int32_t rowIndex) {
1033 1034
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, groupColIndex);
  uint64_t*        groupCol = (uint64_t*)pColInfo->pData;
1035
  ASSERT(rowIndex < pBlock->info.rows);
1036
  pInfo->groupId = groupCol[rowIndex];
1037 1038
}

L
Liu Jicong 已提交
1039
void resetTableScanInfo(STableScanInfo* pTableScanInfo, STimeWindow* pWin) {
H
Haojun Liao 已提交
1040
  pTableScanInfo->base.cond.twindows = *pWin;
L
Liu Jicong 已提交
1041 1042
  pTableScanInfo->scanTimes = 0;
  pTableScanInfo->currentGroupId = -1;
H
Haojun Liao 已提交
1043
  tsdbReaderClose(pTableScanInfo->base.dataReader);
H
Haojun Liao 已提交
1044
  qDebug("1");
H
Haojun Liao 已提交
1045
  pTableScanInfo->base.dataReader = NULL;
1046 1047
}

L
Liu Jicong 已提交
1048 1049
static SSDataBlock* readPreVersionData(SOperatorInfo* pTableScanOp, uint64_t tbUid, TSKEY startTs, TSKEY endTs,
                                       int64_t maxVersion) {
1050
  STableKeyInfo tblInfo = {.uid = tbUid, .groupId = 0};
1051

1052
  STableScanInfo*     pTableScanInfo = pTableScanOp->info;
H
Haojun Liao 已提交
1053
  SQueryTableDataCond cond = pTableScanInfo->base.cond;
1054 1055 1056 1057 1058 1059 1060 1061 1062

  cond.startVersion = -1;
  cond.endVersion = maxVersion;
  cond.twindows = (STimeWindow){.skey = startTs, .ekey = endTs};

  SExecTaskInfo* pTaskInfo = pTableScanOp->pTaskInfo;

  SSDataBlock* pBlock = pTableScanInfo->pResBlock;
  STsdbReader* pReader = NULL;
L
Liu Jicong 已提交
1063
  int32_t      code = tsdbReaderOpen(pTableScanInfo->base.readHandle.vnode, &cond, &tblInfo, 1, pBlock,
D
dapan1121 已提交
1064
                                     (STsdbReader**)&pReader, GET_TASKID(pTaskInfo), false);
1065 1066
  if (code != TSDB_CODE_SUCCESS) {
    terrno = code;
dengyihao's avatar
dengyihao 已提交
1067
    T_LONG_JMP(pTaskInfo->env, code);
1068 1069 1070
    return NULL;
  }

D
dapan1121 已提交
1071 1072 1073 1074 1075 1076 1077 1078 1079
  bool hasNext = false;
  code = tsdbNextDataBlock(pReader, &hasNext);
  if (code != TSDB_CODE_SUCCESS) {
    terrno = code;
    T_LONG_JMP(pTaskInfo->env, code);
    return NULL;
  }

  if (hasNext) {
L
Liu Jicong 已提交
1080
    /*SSDataBlock* p = */ tsdbRetrieveDataBlock(pReader, NULL);
H
Haojun Liao 已提交
1081
    doSetTagColumnData(&pTableScanInfo->base, pBlock, pTaskInfo, pBlock->info.rows);
1082
    pBlock->info.id.groupId = getTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
1083 1084 1085
  }

  tsdbReaderClose(pReader);
D
dapan1121 已提交
1086
  qDebug("retrieve prev rows:%" PRId64 ", skey:%" PRId64 ", ekey:%" PRId64 " uid:%" PRIu64 ", max ver:%" PRId64
5
54liuyao 已提交
1087 1088
         ", suid:%" PRIu64,
         pBlock->info.rows, startTs, endTs, tbUid, maxVersion, cond.suid);
1089 1090

  return pBlock->info.rows > 0 ? pBlock : NULL;
1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
}

static uint64_t getGroupIdByCol(SStreamScanInfo* pInfo, uint64_t uid, TSKEY ts, int64_t maxVersion) {
  SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, uid, ts, ts, maxVersion);
  if (!pPreRes || pPreRes->info.rows == 0) {
    return 0;
  }
  ASSERT(pPreRes->info.rows == 1);
  return calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, 0);
}

5
54liuyao 已提交
1102
static uint64_t getGroupIdByUid(SStreamScanInfo* pInfo, uint64_t uid) {
1103
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
1104
  return getTableGroupId(pTableScanInfo->base.pTableListInfo, uid);
1105 1106
}

5
54liuyao 已提交
1107 1108 1109 1110 1111 1112 1113 1114
static uint64_t getGroupIdByData(SStreamScanInfo* pInfo, uint64_t uid, TSKEY ts, int64_t maxVersion) {
  if (pInfo->partitionSup.needCalc) {
    return getGroupIdByCol(pInfo, uid, ts, maxVersion);
  }

  return getGroupIdByUid(pInfo, uid);
}

L
Liu Jicong 已提交
1115
static bool prepareRangeScan(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t* pRowIndex) {
5
54liuyao 已提交
1116 1117 1118
  if (pBlock->info.rows == 0) {
    return false;
  }
L
Liu Jicong 已提交
1119 1120 1121 1122 1123 1124 1125 1126 1127 1128
  if ((*pRowIndex) == pBlock->info.rows) {
    return false;
  }

  ASSERT(taosArrayGetSize(pBlock->pDataBlock) >= 3);
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
  SColumnInfoData* pEndTsCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
  STimeWindow      win = {.skey = startData[*pRowIndex], .ekey = endData[*pRowIndex]};
1129 1130 1131
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
  uint64_t*        gpData = (uint64_t*)pGpCol->pData;
  uint64_t         groupId = gpData[*pRowIndex];
1132 1133 1134 1135 1136 1137

  SColumnInfoData* pCalStartTsCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
  TSKEY*           calStartData = (TSKEY*)pCalStartTsCol->pData;
  SColumnInfoData* pCalEndTsCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
  TSKEY*           calEndData = (TSKEY*)pCalEndTsCol->pData;

L
Liu Jicong 已提交
1138
  setGroupId(pInfo, pBlock, GROUPID_COLUMN_INDEX, *pRowIndex);
1139 1140 1141 1142
  if (isSlidingWindow(pInfo)) {
    pInfo->updateWin.skey = calStartData[*pRowIndex];
    pInfo->updateWin.ekey = calEndData[*pRowIndex];
  }
L
Liu Jicong 已提交
1143 1144 1145
  (*pRowIndex)++;

  for (; *pRowIndex < pBlock->info.rows; (*pRowIndex)++) {
1146
    if (win.skey == startData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
L
Liu Jicong 已提交
1147 1148 1149
      win.ekey = TMAX(win.ekey, endData[*pRowIndex]);
      continue;
    }
1150
    if (win.skey == endData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
L
Liu Jicong 已提交
1151 1152 1153
      win.skey = TMIN(win.skey, startData[*pRowIndex]);
      continue;
    }
1154 1155
    ASSERT(!(win.skey > startData[*pRowIndex] && win.ekey < endData[*pRowIndex]) ||
           !(isInTimeWindow(&win, startData[*pRowIndex], 0) || isInTimeWindow(&win, endData[*pRowIndex], 0)));
L
Liu Jicong 已提交
1156 1157 1158 1159
    break;
  }

  resetTableScanInfo(pInfo->pTableScanOp->info, &win);
1160
  pInfo->pTableScanOp->status = OP_OPENED;
L
Liu Jicong 已提交
1161 1162 1163
  return true;
}

5
54liuyao 已提交
1164
static STimeWindow getSlidingWindow(TSKEY* startTsCol, TSKEY* endTsCol, uint64_t* gpIdCol, SInterval* pInterval,
1165
                                    SDataBlockInfo* pDataBlockInfo, int32_t* pRowIndex, bool hasGroup) {
H
Haojun Liao 已提交
1166
  SResultRowInfo dumyInfo = {0};
5
54liuyao 已提交
1167
  dumyInfo.cur.pageId = -1;
1168
  STimeWindow win = getActiveTimeWindow(NULL, &dumyInfo, startTsCol[*pRowIndex], pInterval, TSDB_ORDER_ASC);
5
54liuyao 已提交
1169 1170
  STimeWindow endWin = win;
  STimeWindow preWin = win;
5
54liuyao 已提交
1171
  uint64_t    groupId = gpIdCol[*pRowIndex];
H
Haojun Liao 已提交
1172

5
54liuyao 已提交
1173
  while (1) {
1174 1175 1176
    if (hasGroup) {
      (*pRowIndex) += 1;
    } else {
5
54liuyao 已提交
1177
      while ((groupId == gpIdCol[(*pRowIndex)] && startTsCol[*pRowIndex] <= endWin.ekey)) {
5
54liuyao 已提交
1178 1179 1180 1181 1182
        (*pRowIndex) += 1;
        if ((*pRowIndex) == pDataBlockInfo->rows) {
          break;
        }
      }
1183
    }
5
54liuyao 已提交
1184

5
54liuyao 已提交
1185 1186 1187
    do {
      preWin = endWin;
      getNextTimeWindow(pInterval, &endWin, TSDB_ORDER_ASC);
1188
    } while (endTsCol[(*pRowIndex) - 1] >= endWin.skey);
5
54liuyao 已提交
1189
    endWin = preWin;
5
54liuyao 已提交
1190
    if (win.ekey == endWin.ekey || (*pRowIndex) == pDataBlockInfo->rows || groupId != gpIdCol[*pRowIndex]) {
5
54liuyao 已提交
1191 1192 1193 1194 1195 1196
      win.ekey = endWin.ekey;
      return win;
    }
    win.ekey = endWin.ekey;
  }
}
5
54liuyao 已提交
1197

L
Liu Jicong 已提交
1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208
static SSDataBlock* doRangeScan(SStreamScanInfo* pInfo, SSDataBlock* pSDB, int32_t tsColIndex, int32_t* pRowIndex) {
  while (1) {
    SSDataBlock* pResult = NULL;
    pResult = doTableScan(pInfo->pTableScanOp);
    if (!pResult && prepareRangeScan(pInfo, pSDB, pRowIndex)) {
      // scan next window data
      pResult = doTableScan(pInfo->pTableScanOp);
    }
    if (!pResult) {
      blockDataCleanup(pSDB);
      *pRowIndex = 0;
5
54liuyao 已提交
1209
      pInfo->updateWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
H
Hongze Cheng 已提交
1210
      STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
H
Haojun Liao 已提交
1211
      tsdbReaderClose(pTableScanInfo->base.dataReader);
H
Haojun Liao 已提交
1212
      qDebug("2");
H
Haojun Liao 已提交
1213
      pTableScanInfo->base.dataReader = NULL;
1214 1215
      return NULL;
    }
L
Liu Jicong 已提交
1216

H
Haojun Liao 已提交
1217
    doFilter(pResult, pInfo->pTableScanOp->exprSupp.pFilterInfo, NULL);
1218 1219 1220 1221
    if (pResult->info.rows == 0) {
      continue;
    }

1222 1223 1224 1225 1226 1227 1228 1229
    if (pInfo->partitionSup.needCalc) {
      SSDataBlock* tmpBlock = createOneDataBlock(pResult, true);
      blockDataCleanup(pResult);
      for (int32_t i = 0; i < tmpBlock->info.rows; i++) {
        if (calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, tmpBlock, i) == pInfo->groupId) {
          for (int32_t j = 0; j < pInfo->pTableScanOp->exprSupp.numOfExprs; j++) {
            SColumnInfoData* pSrcCol = taosArrayGet(tmpBlock->pDataBlock, j);
            SColumnInfoData* pDestCol = taosArrayGet(pResult->pDataBlock, j);
L
Liu Jicong 已提交
1230 1231
            bool             isNull = colDataIsNull(pSrcCol, tmpBlock->info.rows, i, NULL);
            char*            pSrcData = colDataGetData(pSrcCol, i);
1232
            colDataSetVal(pDestCol, pResult->info.rows, pSrcData, isNull);
1233 1234 1235 1236
          }
          pResult->info.rows++;
        }
      }
H
Haojun Liao 已提交
1237 1238 1239

      blockDataDestroy(tmpBlock);

1240 1241 1242 1243
      if (pResult->info.rows > 0) {
        pResult->info.calWin = pInfo->updateWin;
        return pResult;
      }
H
Haojun Liao 已提交
1244
    } else if (pResult->info.id.groupId == pInfo->groupId) {
5
54liuyao 已提交
1245
      pResult->info.calWin = pInfo->updateWin;
1246
      return pResult;
5
54liuyao 已提交
1247 1248
    }
  }
1249
}
1250

1251
static int32_t getPreSessionWindow(SStreamAggSupporter* pAggSup, TSKEY startTs, TSKEY endTs, uint64_t groupId,
X
Xiaoyu Wang 已提交
1252
                                   SSessionKey* pKey) {
1253 1254 1255
  pKey->win.skey = startTs;
  pKey->win.ekey = endTs;
  pKey->groupId = groupId;
X
Xiaoyu Wang 已提交
1256

1257 1258 1259 1260 1261
  SStreamStateCur* pCur = streamStateSessionSeekKeyCurrentPrev(pAggSup->pState, pKey);
  int32_t          code = streamStateSessionGetKVByCur(pCur, pKey, NULL, 0);
  if (code != TSDB_CODE_SUCCESS) {
    SET_SESSION_WIN_KEY_INVALID(pKey);
  }
H
Haojun Liao 已提交
1262 1263

  taosMemoryFree(pCur);
1264 1265 1266
  return code;
}

1267
static int32_t generateSessionScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
5
54liuyao 已提交
1268
  blockDataCleanup(pDestBlock);
1269 1270
  if (pSrcBlock->info.rows == 0) {
    return TSDB_CODE_SUCCESS;
1271
  }
1272
  int32_t code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
1273
  if (code != TSDB_CODE_SUCCESS) {
1274
    return code;
L
Liu Jicong 已提交
1275
  }
1276 1277
  ASSERT(taosArrayGetSize(pSrcBlock->pDataBlock) >= 3);
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
L
Liu Jicong 已提交
1278
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
1279
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
L
Liu Jicong 已提交
1280
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
1281 1282
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
L
Liu Jicong 已提交
1283

1284 1285
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
5
54liuyao 已提交
1286
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
1287
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
5
54liuyao 已提交
1288 1289
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
L
Liu Jicong 已提交
1290
  int64_t          version = pSrcBlock->info.version - 1;
1291
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
1292
    uint64_t groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], version);
L
Liu Jicong 已提交
1293
    // gap must be 0.
5
54liuyao 已提交
1294
    SSessionKey startWin = {0};
1295
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, startData[i], startData[i], groupId, &startWin);
5
54liuyao 已提交
1296
    if (IS_INVALID_SESSION_WIN_KEY(startWin)) {
L
Liu Jicong 已提交
1297 1298 1299
      // window has been closed.
      continue;
    }
5
54liuyao 已提交
1300 1301
    SSessionKey endWin = {0};
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, endData[i], endData[i], groupId, &endWin);
X
Xiaoyu Wang 已提交
1302
    if (IS_INVALID_SESSION_WIN_KEY(endWin)) {
1303 1304 1305 1306
      getPreSessionWindow(pInfo->windowSup.pStreamAggSup, endData[i], endData[i], groupId, &endWin);
    }
    if (IS_INVALID_SESSION_WIN_KEY(startWin)) {
      // window has been closed.
X
Xiaoyu Wang 已提交
1307
      qError("generate session scan range failed. rang start:%" PRIx64 ", end:%" PRIx64, startData[i], endData[i]);
1308 1309
      continue;
    }
1310 1311
    colDataSetVal(pDestStartCol, i, (const char*)&startWin.win.skey, false);
    colDataSetVal(pDestEndCol, i, (const char*)&endWin.win.ekey, false);
5
54liuyao 已提交
1312

1313
    colDataSetNULL(pDestUidCol, i);
1314
    colDataSetVal(pDestGpCol, i, (const char*)&groupId, false);
1315 1316
    colDataSetNULL(pDestCalStartTsCol, i);
    colDataSetNULL(pDestCalEndTsCol, i);
1317
    pDestBlock->info.rows++;
L
Liu Jicong 已提交
1318
  }
1319
  return TSDB_CODE_SUCCESS;
L
Liu Jicong 已提交
1320
}
1321 1322 1323 1324 1325 1326

static int32_t generateIntervalScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
  blockDataCleanup(pDestBlock);
  int32_t rows = pSrcBlock->info.rows;
  if (rows == 0) {
    return TSDB_CODE_SUCCESS;
1327
  }
1328

1329 1330
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
1331 1332
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
5
54liuyao 已提交
1333

L
Liu Jicong 已提交
1334
  uint64_t* srcUidData = (uint64_t*)pSrcUidCol->pData;
1335
  ASSERT(pSrcStartTsCol->info.type == TSDB_DATA_TYPE_TIMESTAMP);
5
54liuyao 已提交
1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371
  TSKEY*  srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
  TSKEY*  srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
  int64_t version = pSrcBlock->info.version - 1;

  if (pInfo->partitionSup.needCalc && srcStartTsCol[0] != srcEndTsCol[0]) {
    uint64_t     srcUid = srcUidData[0];
    TSKEY        startTs = srcStartTsCol[0];
    TSKEY        endTs = srcEndTsCol[0];
    SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, srcUid, startTs, endTs, version);
    printDataBlock(pPreRes, "pre res");
    blockDataCleanup(pSrcBlock);
    int32_t code = blockDataEnsureCapacity(pSrcBlock, pPreRes->info.rows);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }

    SColumnInfoData* pTsCol = (SColumnInfoData*)taosArrayGet(pPreRes->pDataBlock, pInfo->primaryTsIndex);
    rows = pPreRes->info.rows;

    for (int32_t i = 0; i < rows; i++) {
      uint64_t groupId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, i);
      appendOneRowToStreamSpecialBlock(pSrcBlock, ((TSKEY*)pTsCol->pData) + i, ((TSKEY*)pTsCol->pData) + i, &srcUid,
                                       &groupId, NULL);
    }
    printDataBlock(pSrcBlock, "new delete");
  }
  uint64_t* srcGp = (uint64_t*)pSrcGpCol->pData;
  srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
  srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
  srcUidData = (uint64_t*)pSrcUidCol->pData;

  int32_t code = blockDataEnsureCapacity(pDestBlock, rows);
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

1372 1373
  SColumnInfoData* pStartTsCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pEndTsCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
1374
  SColumnInfoData* pDeUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
1375 1376 1377
  SColumnInfoData* pGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
  SColumnInfoData* pCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
  SColumnInfoData* pCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
1378
  for (int32_t i = 0; i < rows;) {
1379
    uint64_t srcUid = srcUidData[i];
5
54liuyao 已提交
1380 1381 1382 1383 1384
    uint64_t groupId = srcGp[i];
    if (groupId == 0) {
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], version);
    }
    TSKEY calStartTs = srcStartTsCol[i];
1385
    colDataSetVal(pCalStartTsCol, pDestBlock->info.rows, (const char*)(&calStartTs), false);
5
54liuyao 已提交
1386
    STimeWindow win = getSlidingWindow(srcStartTsCol, srcEndTsCol, srcGp, &pInfo->interval, &pSrcBlock->info, &i,
1387 1388
                                       pInfo->partitionSup.needCalc);
    TSKEY       calEndTs = srcStartTsCol[i - 1];
1389 1390 1391 1392 1393
    colDataSetVal(pCalEndTsCol, pDestBlock->info.rows, (const char*)(&calEndTs), false);
    colDataSetVal(pDeUidCol, pDestBlock->info.rows, (const char*)(&srcUid), false);
    colDataSetVal(pStartTsCol, pDestBlock->info.rows, (const char*)(&win.skey), false);
    colDataSetVal(pEndTsCol, pDestBlock->info.rows, (const char*)(&win.ekey), false);
    colDataSetVal(pGpCol, pDestBlock->info.rows, (const char*)(&groupId), false);
1394
    pDestBlock->info.rows++;
5
54liuyao 已提交
1395
  }
1396 1397
  return TSDB_CODE_SUCCESS;
}
1398

1399
static int32_t generateDeleteResultBlock(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
5
54liuyao 已提交
1400 1401 1402
  blockDataCleanup(pDestBlock);
  int32_t rows = pSrcBlock->info.rows;
  if (rows == 0) {
1403 1404
    return TSDB_CODE_SUCCESS;
  }
5
54liuyao 已提交
1405
  int32_t code = blockDataEnsureCapacity(pDestBlock, rows);
1406 1407 1408 1409
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

5
54liuyao 已提交
1410 1411 1412 1413 1414 1415 1416 1417 1418 1419
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
  uint64_t*        srcUidData = (uint64_t*)pSrcUidCol->pData;
  SColumnInfoData* pSrcGpCol = taosArrayGet(pSrcBlock->pDataBlock, GROUPID_COLUMN_INDEX);
  uint64_t*        srcGp = (uint64_t*)pSrcGpCol->pData;
  ASSERT(pSrcStartTsCol->info.type == TSDB_DATA_TYPE_TIMESTAMP);
  TSKEY*  srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
  TSKEY*  srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
  int64_t version = pSrcBlock->info.version - 1;
1420
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
5
54liuyao 已提交
1421 1422
    uint64_t srcUid = srcUidData[i];
    uint64_t groupId = srcGp[i];
L
Liu Jicong 已提交
1423
    char*    tbname[VARSTR_HEADER_SIZE + TSDB_TABLE_NAME_LEN] = {0};
5
54liuyao 已提交
1424 1425 1426
    if (groupId == 0) {
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], version);
    }
L
Liu Jicong 已提交
1427
    if (pInfo->tbnameCalSup.pExprInfo) {
1428 1429 1430
      void* parTbname = NULL;
      streamStateGetParName(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, groupId, &parTbname);

L
Liu Jicong 已提交
1431 1432
      memcpy(varDataVal(tbname), parTbname, TSDB_TABLE_NAME_LEN);
      varDataSetLen(tbname, strlen(varDataVal(tbname)));
L
Liu Jicong 已提交
1433
      tdbFree(parTbname);
L
Liu Jicong 已提交
1434 1435 1436
    }
    appendOneRowToStreamSpecialBlock(pDestBlock, srcStartTsCol + i, srcEndTsCol + i, srcUidData + i, &groupId,
                                     tbname[0] == 0 ? NULL : tbname);
1437 1438 1439 1440
  }
  return TSDB_CODE_SUCCESS;
}

1441 1442 1443 1444
static int32_t generateScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
  int32_t code = TSDB_CODE_SUCCESS;
  if (isIntervalWindow(pInfo)) {
    code = generateIntervalScanRange(pInfo, pSrcBlock, pDestBlock);
1445
  } else if (isSessionWindow(pInfo) || isStateWindow(pInfo)) {
1446
    code = generateSessionScanRange(pInfo, pSrcBlock, pDestBlock);
5
54liuyao 已提交
1447 1448
  } else {
    code = generateDeleteResultBlock(pInfo, pSrcBlock, pDestBlock);
1449
  }
1450
  pDestBlock->info.type = STREAM_CLEAR;
1451
  pDestBlock->info.version = pSrcBlock->info.version;
1452
  pDestBlock->info.dataLoad = 1;
1453 1454 1455 1456
  blockDataUpdateTsWindow(pDestBlock, 0);
  return code;
}

L
Liu Jicong 已提交
1457 1458 1459
#if 0
void calBlockTag(SStreamScanInfo* pInfo, SSDataBlock* pBlock) {
  SExprSupp*    pTagCalSup = &pInfo->tagCalSup;
1460
  SStreamState* pState = pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState;
L
Liu Jicong 已提交
1461
  if (pTagCalSup == NULL || pTagCalSup->numOfExprs == 0) return;
L
Liu Jicong 已提交
1462
  if (pBlock == NULL || pBlock->info.rows == 0) return;
1463

L
Liu Jicong 已提交
1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479
  void*   tag = NULL;
  int32_t tagLen = 0;
  if (streamStateGetParTag(pState, pBlock->info.id.groupId, &tag, &tagLen) == 0) {
    pBlock->info.tagLen = tagLen;
    void* pTag = taosMemoryRealloc(pBlock->info.pTag, tagLen);
    if (pTag == NULL) {
      tdbFree(tag);
      taosMemoryFree(pBlock->info.pTag);
      pBlock->info.pTag = NULL;
      pBlock->info.tagLen = 0;
      return;
    }
    pBlock->info.pTag = pTag;
    memcpy(pBlock->info.pTag, tag, tagLen);
    tdbFree(tag);
    return;
L
Liu Jicong 已提交
1480
  } else {
L
Liu Jicong 已提交
1481
    pBlock->info.pTag = NULL;
L
Liu Jicong 已提交
1482
  }
L
Liu Jicong 已提交
1483 1484 1485
  tdbFree(tag);
}
#endif
L
Liu Jicong 已提交
1486

5
54liuyao 已提交
1487
static void calBlockTbName(SStreamScanInfo* pInfo, SSDataBlock* pBlock) {
1488 1489
  SExprSupp*    pTbNameCalSup = &pInfo->tbnameCalSup;
  SStreamState* pState = pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState;
5
54liuyao 已提交
1490 1491
  blockDataCleanup(pInfo->pCreateTbRes);
  if (pInfo->tbnameCalSup.numOfExprs == 0 && pInfo->tagCalSup.numOfExprs == 0) {
L
Liu Jicong 已提交
1492
    pBlock->info.parTbName[0] = 0;
L
Liu Jicong 已提交
1493
  } else {
5
54liuyao 已提交
1494 1495
    appendCreateTableRow(pInfo->pStreamScanOp->pTaskInfo->streamInfo.pState, &pInfo->tbnameCalSup, &pInfo->tagCalSup,
                         pBlock->info.id.groupId, pBlock, 0, pInfo->pCreateTbRes);
L
Liu Jicong 已提交
1496
  }
L
Liu Jicong 已提交
1497 1498
}

1499 1500
void appendOneRowToStreamSpecialBlock(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, uint64_t* pUid,
                                      uint64_t* pGp, void* pTbName) {
1501 1502
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pEndTsCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
1503 1504
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
1505 1506
  SColumnInfoData* pCalStartCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
  SColumnInfoData* pCalEndCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
1507
  SColumnInfoData* pTableCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
1508 1509 1510 1511 1512 1513 1514
  colDataSetVal(pStartTsCol, pBlock->info.rows, (const char*)pStartTs, false);
  colDataSetVal(pEndTsCol, pBlock->info.rows, (const char*)pEndTs, false);
  colDataSetVal(pUidCol, pBlock->info.rows, (const char*)pUid, false);
  colDataSetVal(pGpCol, pBlock->info.rows, (const char*)pGp, false);
  colDataSetVal(pCalStartCol, pBlock->info.rows, (const char*)pStartTs, false);
  colDataSetVal(pCalEndCol, pBlock->info.rows, (const char*)pEndTs, false);
  colDataSetVal(pTableCol, pBlock->info.rows, (const char*)pTbName, pTbName == NULL);
1515
  pBlock->info.rows++;
5
54liuyao 已提交
1516 1517
}

1518
static void checkUpdateData(SStreamScanInfo* pInfo, bool invertible, SSDataBlock* pBlock, bool out) {
1519 1520
  if (out) {
    blockDataCleanup(pInfo->pUpdateDataRes);
5
54liuyao 已提交
1521
    blockDataEnsureCapacity(pInfo->pUpdateDataRes, pBlock->info.rows * 2);
1522
  }
1523 1524
  SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, pInfo->primaryTsIndex);
  ASSERT(pColDataInfo->info.type == TSDB_DATA_TYPE_TIMESTAMP);
5
54liuyao 已提交
1525
  TSKEY* tsCol = (TSKEY*)pColDataInfo->pData;
H
Haojun Liao 已提交
1526
  bool   tableInserted = updateInfoIsTableInserted(pInfo->pUpdateInfo, pBlock->info.id.uid);
1527
  for (int32_t rowId = 0; rowId < pBlock->info.rows; rowId++) {
5
54liuyao 已提交
1528 1529
    SResultRowInfo dumyInfo;
    dumyInfo.cur.pageId = -1;
L
Liu Jicong 已提交
1530
    bool        isClosed = false;
5
54liuyao 已提交
1531
    STimeWindow win = {.skey = INT64_MIN, .ekey = INT64_MAX};
X
Xiaoyu Wang 已提交
1532
    bool        overDue = isOverdue(tsCol[rowId], &pInfo->twAggSup);
1533 1534 1535 1536 1537
    if (pInfo->igExpired && overDue) {
      continue;
    }

    if (tableInserted && overDue) {
5
54liuyao 已提交
1538 1539 1540
      win = getActiveTimeWindow(NULL, &dumyInfo, tsCol[rowId], &pInfo->interval, TSDB_ORDER_ASC);
      isClosed = isCloseWindow(&win, &pInfo->twAggSup);
    }
5
54liuyao 已提交
1541
    // must check update info first.
H
Haojun Liao 已提交
1542
    bool update = updateInfoIsUpdated(pInfo->pUpdateInfo, pBlock->info.id.uid, tsCol[rowId]);
L
Liu Jicong 已提交
1543
    bool closedWin = isClosed && isSignleIntervalWindow(pInfo) &&
H
Haojun Liao 已提交
1544
                     isDeletedStreamWindow(&win, pBlock->info.id.groupId,
1545
                                           pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, &pInfo->twAggSup);
L
Liu Jicong 已提交
1546
    if ((update || closedWin) && out) {
L
Liu Jicong 已提交
1547
      qDebug("stream update check not pass, update %d, closedWin %d", update, closedWin);
5
54liuyao 已提交
1548
      uint64_t gpId = 0;
H
Haojun Liao 已提交
1549
      appendOneRowToStreamSpecialBlock(pInfo->pUpdateDataRes, tsCol + rowId, tsCol + rowId, &pBlock->info.id.uid, &gpId,
1550
                                       NULL);
5
54liuyao 已提交
1551 1552
      if (closedWin && pInfo->partitionSup.needCalc) {
        gpId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pBlock, rowId);
S
slzhou 已提交
1553 1554
        appendOneRowToStreamSpecialBlock(pInfo->pUpdateDataRes, tsCol + rowId, tsCol + rowId, &pBlock->info.id.uid,
                                         &gpId, NULL);
5
54liuyao 已提交
1555
      }
1556 1557
    }
  }
1558 1559
  if (out && pInfo->pUpdateDataRes->info.rows > 0) {
    pInfo->pUpdateDataRes->info.version = pBlock->info.version;
1560
    pInfo->pUpdateDataRes->info.dataLoad = 1;
1561
    blockDataUpdateTsWindow(pInfo->pUpdateDataRes, 0);
1562
    pInfo->pUpdateDataRes->info.type = pInfo->partitionSup.needCalc ? STREAM_DELETE_DATA : STREAM_CLEAR;
5
54liuyao 已提交
1563 1564
  }
}
L
Liu Jicong 已提交
1565

1566
static int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock, bool filter) {
L
Liu Jicong 已提交
1567 1568
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
L
Liu Jicong 已提交
1569
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
L
Liu Jicong 已提交
1570

1571 1572
  blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);

L
Liu Jicong 已提交
1573
  pInfo->pRes->info.rows = pBlock->info.rows;
H
Haojun Liao 已提交
1574
  pInfo->pRes->info.id.uid = pBlock->info.id.uid;
L
Liu Jicong 已提交
1575
  pInfo->pRes->info.type = STREAM_NORMAL;
1576
  pInfo->pRes->info.version = pBlock->info.version;
L
Liu Jicong 已提交
1577

1578
  STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
1579
  pInfo->pRes->info.id.groupId = getTableGroupId(pTableScanInfo->base.pTableListInfo, pBlock->info.id.uid);
L
Liu Jicong 已提交
1580 1581

  // todo extract method
H
Haojun Liao 已提交
1582 1583 1584
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
    if (!pColMatchInfo->needOutput) {
L
Liu Jicong 已提交
1585 1586 1587 1588 1589 1590 1591
      continue;
    }

    bool colExists = false;
    for (int32_t j = 0; j < blockDataGetNumOfCols(pBlock); ++j) {
      SColumnInfoData* pResCol = bdGetColumnInfoData(pBlock, j);
      if (pResCol->info.colId == pColMatchInfo->colId) {
H
Haojun Liao 已提交
1592
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
1593
        colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
L
Liu Jicong 已提交
1594 1595 1596 1597 1598 1599 1600
        colExists = true;
        break;
      }
    }

    // the required column does not exists in submit block, let's set it to be all null value
    if (!colExists) {
H
Haojun Liao 已提交
1601
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
1602
      colDataSetNNULL(pDst, 0, pBlockInfo->rows);
L
Liu Jicong 已提交
1603 1604 1605 1606 1607
    }
  }

  // currently only the tbname pseudo column
  if (pInfo->numOfPseudoExpr > 0) {
L
Liu Jicong 已提交
1608
    int32_t code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
1609
                                          pInfo->pRes->info.rows, GET_TASKID(pTaskInfo), NULL);
K
kailixu 已提交
1610 1611
    // ignore the table not exists error, since this table may have been dropped during the scan procedure.
    if (code != TSDB_CODE_SUCCESS && code != TSDB_CODE_PAR_TABLE_NOT_EXIST) {
L
Liu Jicong 已提交
1612
      blockDataFreeRes((SSDataBlock*)pBlock);
1613
      T_LONG_JMP(pTaskInfo->env, code);
H
Haojun Liao 已提交
1614
    }
K
kailixu 已提交
1615 1616 1617

    // reset the error code.
    terrno = 0;
L
Liu Jicong 已提交
1618 1619
  }

1620
  if (filter) {
H
Haojun Liao 已提交
1621
    doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
1622
  }
1623

1624
  pInfo->pRes->info.dataLoad = 1;
L
Liu Jicong 已提交
1625
  blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
L
Liu Jicong 已提交
1626
  blockDataFreeRes((SSDataBlock*)pBlock);
L
Liu Jicong 已提交
1627

L
Liu Jicong 已提交
1628
  calBlockTbName(pInfo, pInfo->pRes);
L
Liu Jicong 已提交
1629 1630
  return 0;
}
5
54liuyao 已提交
1631

L
Liu Jicong 已提交
1632
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
1633 1634
  SExecTaskInfo*   pTaskInfo = pOperator->pTaskInfo;
  SStreamScanInfo* pInfo = pOperator->info;
X
Xiaoyu Wang 已提交
1635
  const char*      id = GET_TASKID(pTaskInfo);
H
Haojun Liao 已提交
1636

1637
  qDebug("start to exec queue scan, %s", id);
L
Liu Jicong 已提交
1638

L
Liu Jicong 已提交
1639
  if (pTaskInfo->streamInfo.submit.msgStr != NULL) {
1640
    if (pInfo->tqReader->msg2.msgStr == NULL) {
L
Liu Jicong 已提交
1641
      SPackedData submit = pTaskInfo->streamInfo.submit;
1642
      if (tqReaderSetSubmitMsg(pInfo->tqReader, submit.msgStr, submit.msgLen, submit.ver) < 0) {
L
Liu Jicong 已提交
1643
        qError("submit msg messed up when initing stream submit block %p", submit.msgStr);
1644
        return NULL;
L
Liu Jicong 已提交
1645 1646 1647 1648 1649 1650
      }
    }

    blockDataCleanup(pInfo->pRes);
    SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;

1651
    while (tqNextBlockImpl(pInfo->tqReader)) {
L
Liu Jicong 已提交
1652 1653
      SSDataBlock block = {0};

1654
      int32_t code = tqRetrieveDataBlock(&block, pInfo->tqReader, NULL);
L
Liu Jicong 已提交
1655 1656 1657 1658
      if (code != TSDB_CODE_SUCCESS || block.info.rows == 0) {
        continue;
      }

1659
      setBlockIntoRes(pInfo, &block, true);
L
Liu Jicong 已提交
1660 1661 1662 1663 1664 1665

      if (pBlockInfo->rows > 0) {
        return pInfo->pRes;
      }
    }

L
Liu Jicong 已提交
1666 1667
    pInfo->tqReader->msg2 = (SPackedData){0};
    pTaskInfo->streamInfo.submit = (SPackedData){0};
L
Liu Jicong 已提交
1668
    return NULL;
L
Liu Jicong 已提交
1669 1670
  }

1671
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
L
Liu Jicong 已提交
1672 1673
    SSDataBlock* pResult = doTableScan(pInfo->pTableScanOp);
    if (pResult && pResult->info.rows > 0) {
X
Xiaoyu Wang 已提交
1674 1675 1676
      qDebug("queue scan tsdb return %" PRId64 " rows min:%" PRId64 " max:%" PRId64 " wal curVersion:%" PRId64,
             pResult->info.rows, pResult->info.window.skey, pResult->info.window.ekey,
             pInfo->tqReader->pWalReader->curVersion);
1677
      tqOffsetResetToData(&pTaskInfo->streamInfo.currentOffset, pResult->info.id.uid, pResult->info.window.ekey);
L
Liu Jicong 已提交
1678
      return pResult;
1679
    }
1680 1681 1682 1683 1684 1685
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
    tsdbReaderClose(pTSInfo->base.dataReader);
    pTSInfo->base.dataReader = NULL;
    qDebug("queue scan tsdb over, switch to wal ver %" PRId64 "", pTaskInfo->streamInfo.snapshotVer + 1);
    if (tqSeekVer(pInfo->tqReader, pTaskInfo->streamInfo.snapshotVer + 1, pTaskInfo->id.str) < 0) {
      return NULL;
1686
    }
wmmhello's avatar
wmmhello 已提交
1687
    tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pTaskInfo->streamInfo.snapshotVer);
1688 1689
  }

1690
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__LOG) {
L
Liu Jicong 已提交
1691
    while (1) {
1692 1693 1694 1695 1696 1697 1698 1699
      SSDataBlock block = {0};
      int32_t type = tqNextBlock(pInfo->tqReader, &block);

      // curVersion move to next, so currentOffset = curVersion - 1
      tqOffsetResetToLog(&pTaskInfo->streamInfo.currentOffset, pInfo->tqReader->pWalReader->curVersion - 1);

      if (type == FETCH_TYPE__DATA) {
        qDebug("doQueueScan get data from log %" PRId64 " rows, version:%" PRId64, block.info.rows,
X
Xiaoyu Wang 已提交
1700
               pTaskInfo->streamInfo.currentOffset.version);
L
Liu Jicong 已提交
1701
        blockDataCleanup(pInfo->pRes);
1702
        setBlockIntoRes(pInfo, &block, true);
L
Liu Jicong 已提交
1703
        if (pInfo->pRes->info.rows > 0) {
X
Xiaoyu Wang 已提交
1704 1705
          qDebug("doQueueScan get data from log %" PRId64 " rows, return, version:%" PRId64, pInfo->pRes->info.rows,
                 pTaskInfo->streamInfo.currentOffset.version);
L
Liu Jicong 已提交
1706 1707
          return pInfo->pRes;
        }
1708
      } else if (type == FETCH_TYPE__NONE) {
wmmhello's avatar
wmmhello 已提交
1709
        qDebug("doQueueScan get none from log, return, version:%" PRId64, pTaskInfo->streamInfo.currentOffset.version);
L
Liu Jicong 已提交
1710 1711 1712
        return NULL;
      }
    }
L
Liu Jicong 已提交
1713
  } else {
1714
    qError("unexpected streamInfo prepare type: %d", pTaskInfo->streamInfo.currentOffset.type);
L
Liu Jicong 已提交
1715
    return NULL;
H
Haojun Liao 已提交
1716
  }
L
Liu Jicong 已提交
1717 1718
}

L
Liu Jicong 已提交
1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736
static int32_t filterDelBlockByUid(SSDataBlock* pDst, const SSDataBlock* pSrc, SStreamScanInfo* pInfo) {
  STqReader* pReader = pInfo->tqReader;
  int32_t    rows = pSrc->info.rows;
  blockDataEnsureCapacity(pDst, rows);

  SColumnInfoData* pSrcStartCol = taosArrayGet(pSrc->pDataBlock, START_TS_COLUMN_INDEX);
  uint64_t*        startCol = (uint64_t*)pSrcStartCol->pData;
  SColumnInfoData* pSrcEndCol = taosArrayGet(pSrc->pDataBlock, END_TS_COLUMN_INDEX);
  uint64_t*        endCol = (uint64_t*)pSrcEndCol->pData;
  SColumnInfoData* pSrcUidCol = taosArrayGet(pSrc->pDataBlock, UID_COLUMN_INDEX);
  uint64_t*        uidCol = (uint64_t*)pSrcUidCol->pData;

  SColumnInfoData* pDstStartCol = taosArrayGet(pDst->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pDstEndCol = taosArrayGet(pDst->pDataBlock, END_TS_COLUMN_INDEX);
  SColumnInfoData* pDstUidCol = taosArrayGet(pDst->pDataBlock, UID_COLUMN_INDEX);
  int32_t          j = 0;
  for (int32_t i = 0; i < rows; i++) {
    if (taosHashGet(pReader->tbIdHash, &uidCol[i], sizeof(uint64_t))) {
1737 1738 1739
      colDataSetVal(pDstStartCol, j, (const char*)&startCol[i], false);
      colDataSetVal(pDstEndCol, j, (const char*)&endCol[i], false);
      colDataSetVal(pDstUidCol, j, (const char*)&uidCol[i], false);
L
Liu Jicong 已提交
1740

1741 1742 1743
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, GROUPID_COLUMN_INDEX), j);
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX), j);
      colDataSetNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX), j);
L
Liu Jicong 已提交
1744 1745 1746
      j++;
    }
  }
L
Liu Jicong 已提交
1747
  uint32_t cap = pDst->info.capacity;
L
Liu Jicong 已提交
1748 1749
  pDst->info = pSrc->info;
  pDst->info.rows = j;
L
Liu Jicong 已提交
1750
  pDst->info.capacity = cap;
L
Liu Jicong 已提交
1751 1752 1753 1754

  return 0;
}

5
54liuyao 已提交
1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766
// for partition by tag
static void setBlockGroupIdByUid(SStreamScanInfo* pInfo, SSDataBlock* pBlock) {
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
  TSKEY*           startTsCol = (TSKEY*)pStartTsCol->pData;
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
  uint64_t*        gpCol = (uint64_t*)pGpCol->pData;
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
  int32_t          rows = pBlock->info.rows;
  if (!pInfo->partitionSup.needCalc) {
    for (int32_t i = 0; i < rows; i++) {
      uint64_t groupId = getGroupIdByUid(pInfo, uidCol[i]);
1767
      colDataSetVal(pGpCol, i, (const char*)&groupId, false);
5
54liuyao 已提交
1768 1769 1770 1771
    }
  }
}

5
54liuyao 已提交
1772
static void doCheckUpdate(SStreamScanInfo* pInfo, TSKEY endKey, SSDataBlock* pBlock) {
5
54liuyao 已提交
1773
  if (pInfo->pUpdateInfo) {
5
54liuyao 已提交
1774
    checkUpdateData(pInfo, true, pBlock, true);
5
54liuyao 已提交
1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785
    pInfo->twAggSup.maxTs = TMAX(pInfo->twAggSup.maxTs, endKey);
    if (pInfo->pUpdateDataRes->info.rows > 0) {
      pInfo->updateResIndex = 0;
      if (pInfo->pUpdateDataRes->info.type == STREAM_CLEAR) {
        pInfo->scanMode = STREAM_SCAN_FROM_UPDATERES;
      } else if (pInfo->pUpdateDataRes->info.type == STREAM_INVERT) {
        pInfo->scanMode = STREAM_SCAN_FROM_RES;
        // return pInfo->pUpdateDataRes;
      } else if (pInfo->pUpdateDataRes->info.type == STREAM_DELETE_DATA) {
        pInfo->scanMode = STREAM_SCAN_FROM_DELETE_DATA;
      }
5
54liuyao 已提交
1786 1787 1788 1789
    }
  }
}

L
Liu Jicong 已提交
1790 1791 1792 1793 1794
static SSDataBlock* doStreamScan(SOperatorInfo* pOperator) {
  // NOTE: this operator does never check if current status is done or not
  SExecTaskInfo*   pTaskInfo = pOperator->pTaskInfo;
  SStreamScanInfo* pInfo = pOperator->info;

L
Liu Jicong 已提交
1795
  qDebug("stream scan called");
H
Haojun Liao 已提交
1796

1797 1798
  if (pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__PREPARE1 ||
      pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__PREPARE2) {
L
Liu Jicong 已提交
1799
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
H
Haojun Liao 已提交
1800
    memcpy(&pTSInfo->base.cond, &pTaskInfo->streamInfo.tableCond, sizeof(SQueryTableDataCond));
1801
    if (pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__PREPARE1) {
H
Haojun Liao 已提交
1802 1803 1804 1805
      pTSInfo->base.cond.startVersion = 0;
      pTSInfo->base.cond.endVersion = pTaskInfo->streamInfo.fillHistoryVer1;
      qDebug("stream recover step 1, from %" PRId64 " to %" PRId64, pTSInfo->base.cond.startVersion,
             pTSInfo->base.cond.endVersion);
5
54liuyao 已提交
1806
      pTaskInfo->streamInfo.recoverStep = STREAM_RECOVER_STEP__SCAN1;
1807
    } else {
H
Haojun Liao 已提交
1808 1809 1810 1811
      pTSInfo->base.cond.startVersion = pTaskInfo->streamInfo.fillHistoryVer1 + 1;
      pTSInfo->base.cond.endVersion = pTaskInfo->streamInfo.fillHistoryVer2;
      qDebug("stream recover step 2, from %" PRId64 " to %" PRId64, pTSInfo->base.cond.startVersion,
             pTSInfo->base.cond.endVersion);
5
54liuyao 已提交
1812
      pTaskInfo->streamInfo.recoverStep = STREAM_RECOVER_STEP__SCAN2;
1813
    }
L
Liu Jicong 已提交
1814 1815

    /*resetTableScanInfo(pTSInfo, pWin);*/
H
Haojun Liao 已提交
1816
    tsdbReaderClose(pTSInfo->base.dataReader);
H
Haojun Liao 已提交
1817

H
Haojun Liao 已提交
1818
    pTSInfo->base.dataReader = NULL;
L
Liu Jicong 已提交
1819
    pInfo->pTableScanOp->status = OP_OPENED;
L
Liu Jicong 已提交
1820

L
Liu Jicong 已提交
1821 1822
    pTSInfo->scanTimes = 0;
    pTSInfo->currentGroupId = -1;
L
Liu Jicong 已提交
1823
    pTaskInfo->streamInfo.recoverScanFinished = false;
L
Liu Jicong 已提交
1824 1825
  }

5
54liuyao 已提交
1826 1827
  if (pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__SCAN1 ||
      pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__SCAN2) {
L
Liu Jicong 已提交
1828 1829 1830 1831 1832
    if (pInfo->blockRecoverContiCnt > 100) {
      pInfo->blockRecoverTotCnt += pInfo->blockRecoverContiCnt;
      pInfo->blockRecoverContiCnt = 0;
      return NULL;
    }
5
54liuyao 已提交
1833 1834 1835 1836 1837 1838 1839

    switch (pInfo->scanMode) {
      case STREAM_SCAN_FROM_RES: {
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
        printDataBlock(pInfo->pRecoverRes, "scan recover");
        return pInfo->pRecoverRes;
      } break;
5
54liuyao 已提交
1840 1841 1842 1843
      case STREAM_SCAN_FROM_UPDATERES: {
        generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes);
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
1844
        printDataBlock(pInfo->pUpdateRes, "recover update");
5
54liuyao 已提交
1845 1846
        return pInfo->pUpdateRes;
      } break;
1847 1848 1849 1850 1851 1852 1853 1854 1855
      case STREAM_SCAN_FROM_DELETE_DATA: {
        generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes);
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
        copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
        pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
        printDataBlock(pInfo->pDeleteDataRes, "recover delete");
        return pInfo->pDeleteDataRes;
      } break;
5
54liuyao 已提交
1856 1857 1858 1859
      case STREAM_SCAN_FROM_DATAREADER_RANGE: {
        SSDataBlock* pSDB = doRangeScan(pInfo, pInfo->pUpdateRes, pInfo->primaryTsIndex, &pInfo->updateResIndex);
        if (pSDB) {
          STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
1860
          uint64_t        version = tsdbGetReaderMaxVersion(pTableScanInfo->base.dataReader);
5
54liuyao 已提交
1861 1862 1863
          updateInfoSetScanRange(pInfo->pUpdateInfo, &pTableScanInfo->base.cond.twindows, pInfo->groupId, version);
          pSDB->info.type = pInfo->scanMode == STREAM_SCAN_FROM_DATAREADER_RANGE ? STREAM_NORMAL : STREAM_PULL_DATA;
          checkUpdateData(pInfo, true, pSDB, false);
1864
          printDataBlock(pSDB, "scan recover update");
5
54liuyao 已提交
1865 1866 1867 1868 1869 1870
          calBlockTbName(pInfo, pSDB);
          return pSDB;
        }
        blockDataCleanup(pInfo->pUpdateDataRes);
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
      } break;
5
54liuyao 已提交
1871 1872 1873 1874 1875 1876
      default:
        break;
    }

    pInfo->pRecoverRes = doTableScan(pInfo->pTableScanOp);
    if (pInfo->pRecoverRes != NULL) {
L
Liu Jicong 已提交
1877
      pInfo->blockRecoverContiCnt++;
5
54liuyao 已提交
1878
      calBlockTbName(pInfo, pInfo->pRecoverRes);
1879
      if (pInfo->pUpdateInfo) {
5
54liuyao 已提交
1880 1881 1882 1883 1884 1885
        if (pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__SCAN1) {
          TSKEY maxTs = updateInfoFillBlockData(pInfo->pUpdateInfo, pInfo->pRecoverRes, pInfo->primaryTsIndex);
          pInfo->twAggSup.maxTs = TMAX(pInfo->twAggSup.maxTs, maxTs);
        } else {
          doCheckUpdate(pInfo, pInfo->pRecoverRes->info.window.ekey, pInfo->pRecoverRes);
        }
1886
      }
5
54liuyao 已提交
1887 1888
      if (pInfo->pCreateTbRes->info.rows > 0) {
        pInfo->scanMode = STREAM_SCAN_FROM_RES;
1889
        printDataBlock(pInfo->pCreateTbRes, "recover createTbl");
5
54liuyao 已提交
1890 1891
        return pInfo->pCreateTbRes;
      }
X
Xiaoyu Wang 已提交
1892
      qDebug("stream recover scan get block, rows %" PRId64, pInfo->pRecoverRes->info.rows);
5
54liuyao 已提交
1893 1894
      printDataBlock(pInfo->pRecoverRes, "scan recover");
      return pInfo->pRecoverRes;
L
Liu Jicong 已提交
1895 1896
    }
    pTaskInfo->streamInfo.recoverStep = STREAM_RECOVER_STEP__NONE;
L
Liu Jicong 已提交
1897
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
H
Haojun Liao 已提交
1898
    tsdbReaderClose(pTSInfo->base.dataReader);
H
Haojun Liao 已提交
1899

H
Haojun Liao 已提交
1900
    pTSInfo->base.dataReader = NULL;
1901

H
Haojun Liao 已提交
1902 1903
    pTSInfo->base.cond.startVersion = -1;
    pTSInfo->base.cond.endVersion = -1;
L
Liu Jicong 已提交
1904

L
Liu Jicong 已提交
1905
    pTaskInfo->streamInfo.recoverScanFinished = true;
L
Liu Jicong 已提交
1906 1907 1908
    return NULL;
  }

5
54liuyao 已提交
1909
  size_t total = taosArrayGetSize(pInfo->pBlockLists);
5
54liuyao 已提交
1910
// TODO: refactor
L
Liu Jicong 已提交
1911
FETCH_NEXT_BLOCK:
L
Liu Jicong 已提交
1912
  if (pInfo->blockType == STREAM_INPUT__DATA_BLOCK) {
H
Haojun Liao 已提交
1913
    if (pInfo->validBlockIndex >= total) {
L
Liu Jicong 已提交
1914
      doClearBufferedBlocks(pInfo);
L
Liu Jicong 已提交
1915
      /*pOperator->status = OP_EXEC_DONE;*/
H
Haojun Liao 已提交
1916 1917 1918
      return NULL;
    }

1919
    int32_t      current = pInfo->validBlockIndex++;
L
Liu Jicong 已提交
1920 1921
    SPackedData* pPacked = taosArrayGet(pInfo->pBlockLists, current);
    SSDataBlock* pBlock = pPacked->pDataBlock;
5
54liuyao 已提交
1922
    if (pBlock->info.parTbName[0]) {
H
Haojun Liao 已提交
1923
      streamStatePutParName(pTaskInfo->streamInfo.pState, pBlock->info.id.groupId, pBlock->info.parTbName);
1924
    }
1925

1926
    // TODO move into scan
5
54liuyao 已提交
1927 1928
    pBlock->info.calWin.skey = INT64_MIN;
    pBlock->info.calWin.ekey = INT64_MAX;
1929
    pBlock->info.dataLoad = 1;
1930
    blockDataUpdateTsWindow(pBlock, 0);
1931
    switch (pBlock->info.type) {
L
Liu Jicong 已提交
1932 1933 1934
      case STREAM_NORMAL:
      case STREAM_GET_ALL:
        return pBlock;
1935 1936 1937
      case STREAM_RETRIEVE: {
        pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RETRIEVE;
1938 1939
        copyDataBlock(pInfo->pUpdateRes, pBlock);
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
1940 1941 1942
        updateInfoAddCloseWindowSBF(pInfo->pUpdateInfo);
      } break;
      case STREAM_DELETE_DATA: {
1943
        printDataBlock(pBlock, "stream scan delete recv");
L
Liu Jicong 已提交
1944
        SSDataBlock* pDelBlock = NULL;
L
Liu Jicong 已提交
1945
        if (pInfo->tqReader) {
L
Liu Jicong 已提交
1946
          pDelBlock = createSpecialDataBlock(STREAM_DELETE_DATA);
L
Liu Jicong 已提交
1947
          filterDelBlockByUid(pDelBlock, pBlock, pInfo);
L
Liu Jicong 已提交
1948 1949
        } else {
          pDelBlock = pBlock;
L
Liu Jicong 已提交
1950
        }
5
54liuyao 已提交
1951 1952
        setBlockGroupIdByUid(pInfo, pDelBlock);
        printDataBlock(pDelBlock, "stream scan delete recv filtered");
5
54liuyao 已提交
1953 1954 1955 1956 1957 1958
        if (pDelBlock->info.rows == 0) {
          if (pInfo->tqReader) {
            blockDataDestroy(pDelBlock);
          }
          goto FETCH_NEXT_BLOCK;
        }
1959
        if (!isIntervalWindow(pInfo) && !isSessionWindow(pInfo) && !isStateWindow(pInfo)) {
L
Liu Jicong 已提交
1960
          generateDeleteResultBlock(pInfo, pDelBlock, pInfo->pDeleteDataRes);
1961
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_RESULT;
L
Liu Jicong 已提交
1962
          printDataBlock(pDelBlock, "stream scan delete result");
H
Haojun Liao 已提交
1963 1964
          blockDataDestroy(pDelBlock);

L
Liu Jicong 已提交
1965 1966 1967 1968 1969
          if (pInfo->pDeleteDataRes->info.rows > 0) {
            return pInfo->pDeleteDataRes;
          } else {
            goto FETCH_NEXT_BLOCK;
          }
1970 1971 1972
        } else {
          pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
          pInfo->updateResIndex = 0;
L
Liu Jicong 已提交
1973
          generateScanRange(pInfo, pDelBlock, pInfo->pUpdateRes);
1974 1975 1976
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
          copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
L
Liu Jicong 已提交
1977 1978 1979 1980
          printDataBlock(pDelBlock, "stream scan delete data");
          if (pInfo->tqReader) {
            blockDataDestroy(pDelBlock);
          }
L
Liu Jicong 已提交
1981
          if (pInfo->pDeleteDataRes->info.rows > 0) {
5
54liuyao 已提交
1982
            pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
L
Liu Jicong 已提交
1983 1984 1985 1986
            return pInfo->pDeleteDataRes;
          } else {
            goto FETCH_NEXT_BLOCK;
          }
1987
        }
1988 1989 1990
      } break;
      default:
        break;
5
54liuyao 已提交
1991
    }
1992
    // printDataBlock(pBlock, "stream scan recv");
1993
    return pBlock;
L
Liu Jicong 已提交
1994
  } else if (pInfo->blockType == STREAM_INPUT__DATA_SUBMIT) {
L
Liu Jicong 已提交
1995
    qDebug("scan mode %d", pInfo->scanMode);
5
54liuyao 已提交
1996 1997 1998
    switch (pInfo->scanMode) {
      case STREAM_SCAN_FROM_RES: {
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
5
54liuyao 已提交
1999
        doCheckUpdate(pInfo, pInfo->pRes->info.window.ekey, pInfo->pRes);
5
54liuyao 已提交
2000 2001 2002
        doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
        pInfo->pRes->info.dataLoad = 1;
        blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
5
54liuyao 已提交
2003 2004 2005
        if (pInfo->pRes->info.rows > 0) {
          return pInfo->pRes;
        }
5
54liuyao 已提交
2006
      } break;
2007
      case STREAM_SCAN_FROM_DELETE_DATA: {
2008 2009 2010 2011 2012 2013 2014
        generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes);
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
        copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
        pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
        return pInfo->pDeleteDataRes;
      } break;
5
54liuyao 已提交
2015 2016 2017 2018 2019 2020 2021 2022 2023 2024
      case STREAM_SCAN_FROM_UPDATERES: {
        generateScanRange(pInfo, pInfo->pUpdateDataRes, pInfo->pUpdateRes);
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
        return pInfo->pUpdateRes;
      } break;
      case STREAM_SCAN_FROM_DATAREADER_RANGE:
      case STREAM_SCAN_FROM_DATAREADER_RETRIEVE: {
        SSDataBlock* pSDB = doRangeScan(pInfo, pInfo->pUpdateRes, pInfo->primaryTsIndex, &pInfo->updateResIndex);
        if (pSDB) {
2025
          STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
2026
          uint64_t        version = tsdbGetReaderMaxVersion(pTableScanInfo->base.dataReader);
H
Haojun Liao 已提交
2027
          updateInfoSetScanRange(pInfo->pUpdateInfo, &pTableScanInfo->base.cond.twindows, pInfo->groupId, version);
5
54liuyao 已提交
2028 2029
          pSDB->info.type = pInfo->scanMode == STREAM_SCAN_FROM_DATAREADER_RANGE ? STREAM_NORMAL : STREAM_PULL_DATA;
          checkUpdateData(pInfo, true, pSDB, false);
2030
          // printDataBlock(pSDB, "stream scan update");
L
Liu Jicong 已提交
2031
          calBlockTbName(pInfo, pSDB);
5
54liuyao 已提交
2032 2033
          return pSDB;
        }
2034
        blockDataCleanup(pInfo->pUpdateDataRes);
5
54liuyao 已提交
2035 2036 2037 2038
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
      } break;
      default:
        break;
2039
    }
2040

2041
    SStreamAggSupporter* pSup = pInfo->windowSup.pStreamAggSup;
5
54liuyao 已提交
2042
    if (isStateWindow(pInfo) && pSup->pScanBlock->info.rows > 0) {
2043 2044
      pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
      pInfo->updateResIndex = 0;
5
54liuyao 已提交
2045 2046
      copyDataBlock(pInfo->pUpdateRes, pSup->pScanBlock);
      blockDataCleanup(pSup->pScanBlock);
2047
      prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
2048
      pInfo->pUpdateRes->info.type = STREAM_DELETE_DATA;
2049
      return pInfo->pUpdateRes;
5
54liuyao 已提交
2050
    }
5
54liuyao 已提交
2051

H
Haojun Liao 已提交
2052 2053
    SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;

2054
    int32_t totBlockNum = taosArrayGetSize(pInfo->pBlockLists);
2055

L
Liu Jicong 已提交
2056
  NEXT_SUBMIT_BLK:
2057
    while (1) {
L
Liu Jicong 已提交
2058
      if (pInfo->tqReader->msg2.msgStr == NULL) {
2059
        if (pInfo->validBlockIndex >= totBlockNum) {
5
54liuyao 已提交
2060
          updateInfoDestoryColseWinSBF(pInfo->pUpdateInfo);
L
Liu Jicong 已提交
2061
          doClearBufferedBlocks(pInfo);
L
Liu Jicong 已提交
2062
          qDebug("stream scan return empty, consume block %d", totBlockNum);
2063 2064
          return NULL;
        }
2065

L
Liu Jicong 已提交
2066 2067
        int32_t      current = pInfo->validBlockIndex++;
        SPackedData* pSubmit = taosArrayGet(pInfo->pBlockLists, current);
2068
        if (tqReaderSetSubmitMsg(pInfo->tqReader, pSubmit->msgStr, pSubmit->msgLen, pSubmit->ver) < 0) {
2069 2070 2071 2072
          qError("submit msg messed up when initing stream submit block %p, current %d, total %d", pSubmit, current,
                 totBlockNum);
          continue;
        }
H
Haojun Liao 已提交
2073 2074
      }

2075 2076
      blockDataCleanup(pInfo->pRes);

2077
      while (tqNextBlockImpl(pInfo->tqReader)) {
2078
        SSDataBlock block = {0};
2079

2080
        int32_t code = tqRetrieveDataBlock(&block, pInfo->tqReader, NULL);
2081 2082 2083 2084
        if (code != TSDB_CODE_SUCCESS || block.info.rows == 0) {
          continue;
        }

2085
        setBlockIntoRes(pInfo, &block, false);
2086

H
Haojun Liao 已提交
2087
        if (updateInfoIgnore(pInfo->pUpdateInfo, &pInfo->pRes->info.window, pInfo->pRes->info.id.groupId,
L
Liu Jicong 已提交
2088
                             pInfo->pRes->info.version)) {
2089 2090 2091 2092 2093
          printDataBlock(pInfo->pRes, "stream scan ignore");
          blockDataCleanup(pInfo->pRes);
          continue;
        }

5
54liuyao 已提交
2094 2095 2096
        if (pInfo->pCreateTbRes->info.rows > 0) {
          pInfo->scanMode = STREAM_SCAN_FROM_RES;
          return pInfo->pCreateTbRes;
2097 2098
        }

5
54liuyao 已提交
2099
        doCheckUpdate(pInfo, pBlockInfo->window.ekey, pInfo->pRes);
H
Haojun Liao 已提交
2100
        doFilter(pInfo->pRes, pOperator->exprSupp.pFilterInfo, NULL);
2101
        pInfo->pRes->info.dataLoad = 1;
2102 2103 2104
        blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);

        if (pBlockInfo->rows > 0 || pInfo->pUpdateDataRes->info.rows > 0) {
2105 2106 2107
          break;
        }
      }
2108
      if (pBlockInfo->rows > 0 || pInfo->pUpdateDataRes->info.rows > 0) {
5
54liuyao 已提交
2109
        break;
J
jiacy-jcy 已提交
2110
      } else {
2111
        continue;
5
54liuyao 已提交
2112
      }
H
Haojun Liao 已提交
2113 2114 2115 2116
    }

    // record the scan action.
    pInfo->numOfExec++;
2117
    pOperator->resultInfo.totalRows += pBlockInfo->rows;
2118
    // printDataBlock(pInfo->pRes, "stream scan");
H
Haojun Liao 已提交
2119

X
Xiaoyu Wang 已提交
2120
    qDebug("scan rows: %" PRId64, pBlockInfo->rows);
L
Liu Jicong 已提交
2121 2122 2123
    if (pBlockInfo->rows > 0) {
      return pInfo->pRes;
    }
2124 2125 2126 2127 2128 2129

    if (pInfo->pUpdateDataRes->info.rows > 0) {
      goto FETCH_NEXT_BLOCK;
    }

    goto NEXT_SUBMIT_BLK;
L
Liu Jicong 已提交
2130 2131 2132
  } else {
    ASSERT(0);
    return NULL;
H
Haojun Liao 已提交
2133 2134 2135
  }
}

H
Haojun Liao 已提交
2136
static SArray* extractTableIdList(const STableListInfo* pTableListInfo) {
2137 2138 2139
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));

  // Transfer the Array of STableKeyInfo into uid list.
H
Haojun Liao 已提交
2140 2141 2142
  size_t size = tableListGetSize(pTableListInfo);
  for (int32_t i = 0; i < size; ++i) {
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
2143 2144 2145 2146 2147 2148
    taosArrayPush(tableIdList, &pkeyInfo->uid);
  }

  return tableIdList;
}

2149
static SSDataBlock* doRawScan(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
2150 2151
  // NOTE: this operator does never check if current status is done or not
  SExecTaskInfo*      pTaskInfo = pOperator->pTaskInfo;
2152
  SStreamRawScanInfo* pInfo = pOperator->info;
D
dapan1121 已提交
2153
  int32_t             code = TSDB_CODE_SUCCESS;
L
Liu Jicong 已提交
2154
  pTaskInfo->streamInfo.metaRsp.metaRspLen = 0;  // use metaRspLen !=0 to judge if data is meta
wmmhello's avatar
wmmhello 已提交
2155
  pTaskInfo->streamInfo.metaRsp.metaRsp = NULL;
2156

wmmhello's avatar
wmmhello 已提交
2157
  qDebug("tmqsnap doRawScan called");
2158
  if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_DATA) {
D
dapan1121 已提交
2159 2160 2161 2162 2163
    bool hasNext = false;
    if (pInfo->dataReader) {
      code = tsdbNextDataBlock(pInfo->dataReader, &hasNext);
      if (code) {
        tsdbReleaseDataBlock(pInfo->dataReader);
2164
        T_LONG_JMP(pTaskInfo->env, code);
D
dapan1121 已提交
2165 2166
      }
    }
X
Xiaoyu Wang 已提交
2167

D
dapan1121 已提交
2168
    if (pInfo->dataReader && hasNext) {
wmmhello's avatar
wmmhello 已提交
2169
      if (isTaskKilled(pTaskInfo)) {
X
Xiaoyu Wang 已提交
2170
        tsdbReleaseDataBlock(pInfo->dataReader);
2171
        T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
wmmhello's avatar
wmmhello 已提交
2172
      }
2173

H
Haojun Liao 已提交
2174 2175
      SSDataBlock* pBlock = tsdbRetrieveDataBlock(pInfo->dataReader, NULL);
      if (pBlock == NULL) {
2176
        T_LONG_JMP(pTaskInfo->env, terrno);
wmmhello's avatar
wmmhello 已提交
2177 2178
      }

H
Haojun Liao 已提交
2179
      qDebug("tmqsnap doRawScan get data uid:%" PRId64 "", pBlock->info.id.uid);
2180
      tqOffsetResetToData(&pTaskInfo->streamInfo.currentOffset, pBlock->info.id.uid, pBlock->info.window.ekey);
wmmhello's avatar
wmmhello 已提交
2181 2182
      return pBlock;
    }
wmmhello's avatar
wmmhello 已提交
2183 2184

    SMetaTableInfo mtInfo = getUidfromSnapShot(pInfo->sContext);
X
Xiaoyu Wang 已提交
2185
    STqOffsetVal   offset = {0};
L
Liu Jicong 已提交
2186
    if (mtInfo.uid == 0) {  // read snapshot done, change to get data from wal
wmmhello's avatar
wmmhello 已提交
2187
      qDebug("tmqsnap read snapshot done, change to get data from wal");
2188
      tqOffsetResetToLog(&offset, pInfo->sContext->snapVersion);
L
Liu Jicong 已提交
2189
    } else {
2190
      tqOffsetResetToData(&offset, mtInfo.uid, INT64_MIN);
2191
      qDebug("tmqsnap change get data uid:%" PRId64 "", mtInfo.uid);
wmmhello's avatar
wmmhello 已提交
2192
    }
2193
    qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
2194
    tDeleteSSchemaWrapper(mtInfo.schema);
wmmhello's avatar
wmmhello 已提交
2195
    return NULL;
2196
  } else if (pTaskInfo->streamInfo.currentOffset.type == TMQ_OFFSET__SNAPSHOT_META) {
L
Liu Jicong 已提交
2197 2198 2199 2200 2201 2202
    SSnapContext* sContext = pInfo->sContext;
    void*         data = NULL;
    int32_t       dataLen = 0;
    int16_t       type = 0;
    int64_t       uid = 0;
    if (getMetafromSnapShot(sContext, &data, &dataLen, &type, &uid) < 0) {
wmmhello's avatar
wmmhello 已提交
2203
      qError("tmqsnap getMetafromSnapShot error");
wmmhello's avatar
wmmhello 已提交
2204
      taosMemoryFreeClear(data);
2205 2206 2207
      return NULL;
    }

2208
    if (!sContext->queryMeta) {  // change to get data next poll request
2209 2210 2211
      STqOffsetVal offset = {0};
      tqOffsetResetToData(&offset, 0, INT64_MIN);
      qStreamPrepareScan(pTaskInfo, &offset, pInfo->sContext->subType);
L
Liu Jicong 已提交
2212
    } else {
2213
      tqOffsetResetToMeta(&pTaskInfo->streamInfo.currentOffset, uid);
wmmhello's avatar
wmmhello 已提交
2214 2215 2216 2217
      pTaskInfo->streamInfo.metaRsp.resMsgType = type;
      pTaskInfo->streamInfo.metaRsp.metaRspLen = dataLen;
      pTaskInfo->streamInfo.metaRsp.metaRsp = data;
    }
2218

wmmhello's avatar
wmmhello 已提交
2219
    return NULL;
2220
  }
L
Liu Jicong 已提交
2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258
  //  else if (pTaskInfo->streamInfo.prepareStatus.type == TMQ_OFFSET__LOG) {
  //    int64_t fetchVer = pTaskInfo->streamInfo.prepareStatus.version + 1;
  //
  //    while(1){
  //      if (tqFetchLog(pInfo->tqReader->pWalReader, pInfo->sContext->withMeta, &fetchVer, &pInfo->pCkHead) < 0) {
  //        qDebug("tmqsnap tmq poll: consumer log end. offset %" PRId64, fetchVer);
  //        pTaskInfo->streamInfo.lastStatus.version = fetchVer;
  //        pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__LOG;
  //        return NULL;
  //      }
  //      SWalCont* pHead = &pInfo->pCkHead->head;
  //      qDebug("tmqsnap tmq poll: consumer log offset %" PRId64 " msgType %d", fetchVer, pHead->msgType);
  //
  //      if (pHead->msgType == TDMT_VND_SUBMIT) {
  //        SSubmitReq* pCont = (SSubmitReq*)&pHead->body;
  //        tqReaderSetDataMsg(pInfo->tqReader, pCont, 0);
  //        SSDataBlock* block = tqLogScanExec(pInfo->sContext->subType, pInfo->tqReader, pInfo->pFilterOutTbUid,
  //        &pInfo->pRes); if(block){
  //          pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__LOG;
  //          pTaskInfo->streamInfo.lastStatus.version = fetchVer;
  //          qDebug("tmqsnap fetch data msg, ver:%" PRId64 ", type:%d", pHead->version, pHead->msgType);
  //          return block;
  //        }else{
  //          fetchVer++;
  //        }
  //      } else{
  //        ASSERT(pInfo->sContext->withMeta);
  //        ASSERT(IS_META_MSG(pHead->msgType));
  //        qDebug("tmqsnap fetch meta msg, ver:%" PRId64 ", type:%d", pHead->version, pHead->msgType);
  //        pTaskInfo->streamInfo.metaRsp.rspOffset.version = fetchVer;
  //        pTaskInfo->streamInfo.metaRsp.rspOffset.type = TMQ_OFFSET__LOG;
  //        pTaskInfo->streamInfo.metaRsp.resMsgType = pHead->msgType;
  //        pTaskInfo->streamInfo.metaRsp.metaRspLen = pHead->bodyLen;
  //        pTaskInfo->streamInfo.metaRsp.metaRsp = taosMemoryMalloc(pHead->bodyLen);
  //        memcpy(pTaskInfo->streamInfo.metaRsp.metaRsp, pHead->body, pHead->bodyLen);
  //        return NULL;
  //      }
  //    }
2259 2260 2261
  return NULL;
}

wmmhello's avatar
wmmhello 已提交
2262
static void destroyRawScanOperatorInfo(void* param) {
wmmhello's avatar
wmmhello 已提交
2263 2264 2265
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
  tsdbReaderClose(pRawScan->dataReader);
  destroySnapContext(pRawScan->sContext);
2266
  tableListDestroy(pRawScan->pTableListInfo);
wmmhello's avatar
wmmhello 已提交
2267 2268 2269
  taosMemoryFree(pRawScan);
}

L
Liu Jicong 已提交
2270 2271 2272
// for subscribing db or stb (not including column),
// if this scan is used, meta data can be return
// and schemas are decided when scanning
2273
SOperatorInfo* createRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
2274 2275 2276 2277 2278
  // create operator
  // create tb reader
  // create meta reader
  // create tq reader

H
Haojun Liao 已提交
2279 2280
  int32_t code = TSDB_CODE_SUCCESS;

2281
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
L
Liu Jicong 已提交
2282
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
2283
  if (pInfo == NULL || pOperator == NULL) {
H
Haojun Liao 已提交
2284 2285
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _end;
2286 2287
  }

2288
  pInfo->pTableListInfo = tableListCreate();
wmmhello's avatar
wmmhello 已提交
2289 2290
  pInfo->vnode = pHandle->vnode;

2291
  pInfo->sContext = pHandle->sContext;
L
Liu Jicong 已提交
2292 2293
  setOperatorInfo(pOperator, "RawScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN, false, OP_NOT_OPENED, pInfo,
                  pTaskInfo);
2294

2295
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScan, NULL, destroyRawScanOperatorInfo, optrDefaultBufFn, NULL);
2296
  return pOperator;
H
Haojun Liao 已提交
2297

L
Liu Jicong 已提交
2298
_end:
H
Haojun Liao 已提交
2299 2300 2301 2302
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  pTaskInfo->code = code;
  return NULL;
L
Liu Jicong 已提交
2303 2304
}

2305
static void destroyStreamScanOperatorInfo(void* param) {
2306
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
2307

2308
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
2309
    destroyOperator(pStreamScan->pTableScanOp);
2310
  }
2311

2312 2313 2314
  if (pStreamScan->tqReader) {
    tqCloseReader(pStreamScan->tqReader);
  }
H
Haojun Liao 已提交
2315 2316
  if (pStreamScan->matchInfo.pList) {
    taosArrayDestroy(pStreamScan->matchInfo.pList);
2317
  }
C
Cary Xu 已提交
2318 2319
  if (pStreamScan->pPseudoExpr) {
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
L
Liu Jicong 已提交
2320
    taosMemoryFree(pStreamScan->pPseudoExpr);
C
Cary Xu 已提交
2321
  }
C
Cary Xu 已提交
2322

L
Liu Jicong 已提交
2323
  cleanupExprSupp(&pStreamScan->tbnameCalSup);
5
54liuyao 已提交
2324
  cleanupExprSupp(&pStreamScan->tagCalSup);
L
Liu Jicong 已提交
2325

L
Liu Jicong 已提交
2326
  updateInfoDestroy(pStreamScan->pUpdateInfo);
2327 2328 2329 2330
  blockDataDestroy(pStreamScan->pRes);
  blockDataDestroy(pStreamScan->pUpdateRes);
  blockDataDestroy(pStreamScan->pPullDataRes);
  blockDataDestroy(pStreamScan->pDeleteDataRes);
5
54liuyao 已提交
2331
  blockDataDestroy(pStreamScan->pUpdateDataRes);
5
54liuyao 已提交
2332
  blockDataDestroy(pStreamScan->pCreateTbRes);
2333 2334 2335 2336
  taosArrayDestroy(pStreamScan->pBlockLists);
  taosMemoryFree(pStreamScan);
}

2337
SOperatorInfo* createStreamScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode, SNode* pTagCond,
2338
                                            STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
2339
  SArray*          pColIds = NULL;
2340 2341
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
2342

H
Haojun Liao 已提交
2343
  if (pInfo == NULL || pOperator == NULL) {
S
Shengliang Guan 已提交
2344
    terrno = TSDB_CODE_OUT_OF_MEMORY;
2345
    tableListDestroy(pTableListInfo);
2346
    goto _error;
H
Haojun Liao 已提交
2347 2348
  }

2349
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
2350
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
H
Haojun Liao 已提交
2351

2352
  pInfo->pTagCond = pTagCond;
2353
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
2354

2355
  int32_t numOfCols = 0;
2356 2357
  int32_t code =
      extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
H
Haojun Liao 已提交
2358
  if (code != TSDB_CODE_SUCCESS) {
2359
    tableListDestroy(pTableListInfo);
H
Haojun Liao 已提交
2360 2361
    goto _error;
  }
2362

H
Haojun Liao 已提交
2363
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
H
Haojun Liao 已提交
2364
  pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
2365
  for (int32_t i = 0; i < numOfOutput; ++i) {
H
Haojun Liao 已提交
2366
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
2367 2368

    int16_t colId = id->colId;
2369
    taosArrayPush(pColIds, &colId);
2370
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
H
Haojun Liao 已提交
2371
      pInfo->primaryTsIndex = id->dstSlotId;
5
54liuyao 已提交
2372
    }
H
Haojun Liao 已提交
2373 2374
  }

L
Liu Jicong 已提交
2375 2376 2377 2378
  if (pTableScanNode->pSubtable != NULL) {
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
    if (pSubTableExpr == NULL) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
2379
      tableListDestroy(pTableListInfo);
L
Liu Jicong 已提交
2380 2381
      goto _error;
    }
2382

L
Liu Jicong 已提交
2383 2384 2385
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
    createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1) != 0) {
2386
      tableListDestroy(pTableListInfo);
L
Liu Jicong 已提交
2387 2388 2389 2390
      goto _error;
    }
  }

2391 2392
  if (pTableScanNode->pTags != NULL) {
    int32_t    numOfTags;
5
54liuyao 已提交
2393
    SExprInfo* pTagExpr = createExpr(pTableScanNode->pTags, &numOfTags);
2394 2395
    if (pTagExpr == NULL) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
2396
      tableListDestroy(pTableListInfo);
2397 2398 2399 2400
      goto _error;
    }
    if (initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags) != 0) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
2401
      tableListDestroy(pTableListInfo);
2402 2403 2404 2405
      goto _error;
    }
  }

L
Liu Jicong 已提交
2406
  pInfo->pBlockLists = taosArrayInit(4, sizeof(SPackedData));
H
Haojun Liao 已提交
2407
  if (pInfo->pBlockLists == NULL) {
2408
    terrno = TSDB_CODE_OUT_OF_MEMORY;
2409
    tableListDestroy(pTableListInfo);
2410
    goto _error;
H
Haojun Liao 已提交
2411 2412
  }

5
54liuyao 已提交
2413
  if (pHandle->vnode) {
2414
    SOperatorInfo*  pTableScanOp = createTableScanOperatorInfo(pTableScanNode, pHandle, pTableListInfo, pTaskInfo);
L
Liu Jicong 已提交
2415
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
2416
    if (pHandle->version > 0) {
H
Haojun Liao 已提交
2417
      pTSInfo->base.cond.endVersion = pHandle->version;
2418
    }
L
Liu Jicong 已提交
2419

2420
    STableKeyInfo* pList = NULL;
5
54liuyao 已提交
2421
    int32_t        num = 0;
2422
    tableListGetGroupList(pTableListInfo, 0, &pList, &num);
2423

2424
    if (pHandle->initTableReader) {
L
Liu Jicong 已提交
2425
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
H
Haojun Liao 已提交
2426
      pTSInfo->base.dataReader = NULL;
L
Liu Jicong 已提交
2427 2428
    }

L
Liu Jicong 已提交
2429 2430
    if (pHandle->initTqReader) {
      ASSERT(pHandle->tqReader == NULL);
2431
      pInfo->tqReader = tqReaderOpen(pHandle->vnode);
L
Liu Jicong 已提交
2432
      ASSERT(pInfo->tqReader);
2433
    } else {
L
Liu Jicong 已提交
2434 2435
      ASSERT(pHandle->tqReader);
      pInfo->tqReader = pHandle->tqReader;
2436 2437
    }

2438
    pInfo->pUpdateInfo = NULL;
2439
    pInfo->pTableScanOp = pTableScanOp;
2440 2441 2442
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
      streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1);
    }
L
Liu Jicong 已提交
2443

L
Liu Jicong 已提交
2444
    pInfo->readHandle = *pHandle;
L
Liu Jicong 已提交
2445
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
5
54liuyao 已提交
2446 2447
    pInfo->pCreateTbRes = buildCreateTableBlock(&pInfo->tbnameCalSup, &pInfo->tagCalSup);
    blockDataEnsureCapacity(pInfo->pCreateTbRes, 8);
L
Liu Jicong 已提交
2448

L
Liu Jicong 已提交
2449
    // set the extract column id to streamHandle
L
Liu Jicong 已提交
2450
    tqReaderSetColIdList(pInfo->tqReader, pColIds);
2451
    SArray* tableIdList = extractTableIdList(((STableScanInfo*)(pInfo->pTableScanOp->info))->base.pTableListInfo);
2452
    code = tqReaderSetTbUidList(pInfo->tqReader, tableIdList);
L
Liu Jicong 已提交
2453 2454 2455 2456
    if (code != 0) {
      taosArrayDestroy(tableIdList);
      goto _error;
    }
2457

L
Liu Jicong 已提交
2458
    taosArrayDestroy(tableIdList);
H
Haojun Liao 已提交
2459
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->base.cond, sizeof(SQueryTableDataCond));
L
Liu Jicong 已提交
2460 2461
  } else {
    taosArrayDestroy(pColIds);
2462
    tableListDestroy(pTableListInfo);
H
Haojun Liao 已提交
2463
    pColIds = NULL;
5
54liuyao 已提交
2464 2465
  }

2466 2467 2468 2469 2470
  // create the pseduo columns info
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
    pInfo->pPseudoExpr = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->numOfPseudoExpr);
  }

H
Haojun Liao 已提交
2471 2472 2473 2474 2475
  code = filterInitFromNode((SNode*)pScanPhyNode->node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

H
Haojun Liao 已提交
2476
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
2477
  pInfo->pUpdateRes = createSpecialDataBlock(STREAM_CLEAR);
2478
  pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
L
Liu Jicong 已提交
2479
  pInfo->windowSup = (SWindowSupporter){.pStreamAggSup = NULL, .gap = -1, .parentType = QUERY_NODE_PHYSICAL_PLAN};
2480
  pInfo->groupId = 0;
2481
  pInfo->pPullDataRes = createSpecialDataBlock(STREAM_RETRIEVE);
2482
  pInfo->pStreamScanOp = pOperator;
2483
  pInfo->deleteDataIndex = 0;
2484
  pInfo->pDeleteDataRes = createSpecialDataBlock(STREAM_DELETE_DATA);
5
54liuyao 已提交
2485
  pInfo->updateWin = (STimeWindow){.skey = INT64_MAX, .ekey = INT64_MAX};
2486
  pInfo->pUpdateDataRes = createSpecialDataBlock(STREAM_CLEAR);
X
Xiaoyu Wang 已提交
2487
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
2488
  pInfo->partitionSup.needCalc = false;
5
54liuyao 已提交
2489 2490
  pInfo->igCheckUpdate = pTableScanNode->igCheckUpdate;
  pInfo->igExpired = pTableScanNode->igExpired;
2491
  pInfo->twAggSup.maxTs = INT64_MIN;
L
Liu Jicong 已提交
2492

L
Liu Jicong 已提交
2493 2494
  setOperatorInfo(pOperator, "StreamScanOperator", QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN, false, OP_NOT_OPENED, pInfo,
                  pTaskInfo);
2495
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
H
Haojun Liao 已提交
2496

2497
  __optr_fn_t nextFn = (pTaskInfo->execModel == OPTR_EXEC_MODEL_STREAM) ? doStreamScan : doQueueScan;
L
Liu Jicong 已提交
2498 2499
  pOperator->fpSet =
      createOperatorFpSet(optrDummyOpenFn, nextFn, NULL, destroyStreamScanOperatorInfo, optrDefaultBufFn, NULL);
2500

H
Haojun Liao 已提交
2501
  return pOperator;
2502

L
Liu Jicong 已提交
2503
_error:
H
Haojun Liao 已提交
2504 2505 2506 2507 2508 2509 2510 2511
  if (pColIds != NULL) {
    taosArrayDestroy(pColIds);
  }

  if (pInfo != NULL) {
    destroyStreamScanOperatorInfo(pInfo);
  }

2512 2513
  taosMemoryFreeClear(pOperator);
  return NULL;
H
Haojun Liao 已提交
2514 2515
}

2516
static SSDataBlock* doTagScan(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
2517 2518 2519 2520
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

2521 2522 2523
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;

  STagScanInfo* pInfo = pOperator->info;
2524
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
2525
  SSDataBlock*  pRes = pInfo->pRes;
2526
  blockDataCleanup(pRes);
H
Haojun Liao 已提交
2527

2528
  int32_t size = tableListGetSize(pInfo->pTableListInfo);
wmmhello's avatar
wmmhello 已提交
2529
  if (size == 0) {
H
Haojun Liao 已提交
2530 2531 2532 2533
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
    return NULL;
  }

2534 2535 2536
  char        str[512] = {0};
  int32_t     count = 0;
  SMetaReader mr = {0};
2537
  metaReaderInit(&mr, pInfo->readHandle.meta, 0);
H
Haojun Liao 已提交
2538

wmmhello's avatar
wmmhello 已提交
2539
  while (pInfo->curPos < size && count < pOperator->resultInfo.capacity) {
2540
    STableKeyInfo* item = tableListGetInfo(pInfo->pTableListInfo, pInfo->curPos);
L
Liu Jicong 已提交
2541
    int32_t        code = metaGetTableEntryByUid(&mr, item->uid);
2542
    tDecoderClear(&mr.coder);
H
Haojun Liao 已提交
2543
    if (code != TSDB_CODE_SUCCESS) {
L
Liu Jicong 已提交
2544 2545
      qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
             GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
2546
      metaReaderClear(&mr);
2547
      T_LONG_JMP(pTaskInfo->env, terrno);
H
Haojun Liao 已提交
2548
    }
H
Haojun Liao 已提交
2549

2550
    for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
2551 2552 2553 2554 2555
      SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, pExprInfo[j].base.resSchema.slotId);

      // refactor later
      if (fmIsScanPseudoColumnFunc(pExprInfo[j].pExpr->_function.functionId)) {
        STR_TO_VARSTR(str, mr.me.name);
2556
        colDataSetVal(pDst, count, str, false);
2557
      } else {  // it is a tag value
wmmhello's avatar
wmmhello 已提交
2558 2559
        STagVal val = {0};
        val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
2560
        const char* p = metaGetTableTagVal(mr.me.ctbEntry.pTags, pDst->info.type, &val);
wmmhello's avatar
wmmhello 已提交
2561

2562 2563 2564 2565
        char* data = NULL;
        if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
          data = tTagValToData((const STagVal*)p, false);
        } else {
wmmhello's avatar
wmmhello 已提交
2566 2567
          data = (char*)p;
        }
2568
        colDataSetVal(pDst, count, data,
L
Liu Jicong 已提交
2569
                      (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
2570

2571 2572
        if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
            data != NULL) {
wmmhello's avatar
wmmhello 已提交
2573
          taosMemoryFree(data);
wmmhello's avatar
wmmhello 已提交
2574
        }
H
Haojun Liao 已提交
2575 2576 2577
      }
    }

2578
    count += 1;
wmmhello's avatar
wmmhello 已提交
2579
    if (++pInfo->curPos >= size) {
H
Haojun Liao 已提交
2580
      setOperatorCompleted(pOperator);
H
Haojun Liao 已提交
2581 2582 2583
    }
  }

2584 2585
  metaReaderClear(&mr);

2586
  // qDebug("QInfo:0x%"PRIx64" create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
H
Haojun Liao 已提交
2587
  if (pOperator->status == OP_EXEC_DONE) {
2588
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
H
Haojun Liao 已提交
2589 2590 2591
  }

  pRes->info.rows = count;
wmmhello's avatar
wmmhello 已提交
2592
  pOperator->resultInfo.totalRows += count;
2593

2594
  return (pRes->info.rows == 0) ? NULL : pInfo->pRes;
H
Haojun Liao 已提交
2595 2596
}

2597
static void destroyTagScanOperatorInfo(void* param) {
H
Haojun Liao 已提交
2598 2599
  STagScanInfo* pInfo = (STagScanInfo*)param;
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
H
Haojun Liao 已提交
2600
  taosArrayDestroy(pInfo->matchInfo.pList);
2601
  pInfo->pTableListInfo = tableListDestroy(pInfo->pTableListInfo);
D
dapan1121 已提交
2602
  taosMemoryFreeClear(param);
H
Haojun Liao 已提交
2603 2604
}

S
slzhou 已提交
2605
SOperatorInfo* createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pPhyNode,
X
Xiaoyu Wang 已提交
2606
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo) {
2607
  STagScanInfo*  pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
H
Haojun Liao 已提交
2608 2609 2610 2611 2612
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }

2613 2614 2615 2616
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;

  int32_t    numOfExprs = 0;
  SExprInfo* pExprInfo = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &numOfExprs);
2617
  int32_t    code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs);
2618 2619 2620
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
2621

H
Haojun Liao 已提交
2622 2623
  int32_t num = 0;
  code = extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
2624 2625 2626
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
2627

2628
  pInfo->pTableListInfo = pTableListInfo;
H
Haojun Liao 已提交
2629
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
2630 2631
  pInfo->readHandle = *pReadHandle;
  pInfo->curPos = 0;
2632

L
Liu Jicong 已提交
2633 2634
  setOperatorInfo(pOperator, "TagScanOperator", QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN, false, OP_NOT_OPENED, pInfo,
                  pTaskInfo);
2635
  initResultSizeInfo(&pOperator->resultInfo, 4096);
2636 2637
  blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);

L
Liu Jicong 已提交
2638 2639
  pOperator->fpSet =
      createOperatorFpSet(optrDummyOpenFn, doTagScan, NULL, destroyTagScanOperatorInfo, optrDefaultBufFn, NULL);
H
Haojun Liao 已提交
2640 2641

  return pOperator;
2642

2643
_error:
H
Haojun Liao 已提交
2644 2645 2646 2647 2648
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  terrno = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
}
2649

dengyihao's avatar
dengyihao 已提交
2650
static SSDataBlock* getTableDataBlockImpl(void* param) {
dengyihao's avatar
opt mem  
dengyihao 已提交
2651 2652 2653 2654 2655 2656
  STableMergeScanSortSourceParam* source = param;
  SOperatorInfo*                  pOperator = source->pOperator;
  STableMergeScanInfo*            pInfo = pOperator->info;
  SExecTaskInfo*                  pTaskInfo = pOperator->pTaskInfo;
  int32_t                         readIdx = source->readerIdx;
  SSDataBlock*                    pBlock = source->inputBlock;
2657
  int32_t                         code = 0;
dengyihao's avatar
opt mem  
dengyihao 已提交
2658

H
Haojun Liao 已提交
2659
  SQueryTableDataCond* pQueryCond = taosArrayGet(pInfo->queryConds, readIdx);
dengyihao's avatar
opt mem  
dengyihao 已提交
2660

L
Liu Jicong 已提交
2661
  int64_t      st = taosGetTimestampUs();
2662
  void*        p = tableListGetInfo(pInfo->base.pTableListInfo, readIdx + pInfo->tableStartIndex);
H
Haojun Liao 已提交
2663
  SReadHandle* pHandle = &pInfo->base.readHandle;
dengyihao's avatar
dengyihao 已提交
2664

D
dapan1121 已提交
2665
  if (NULL == source->dataReader || !source->multiReader) {
D
dapan1121 已提交
2666
    code = tsdbReaderOpen(pHandle->vnode, pQueryCond, p, 1, pBlock, &source->dataReader, GET_TASKID(pTaskInfo), false);
2667 2668 2669
    if (code != 0) {
      T_LONG_JMP(pTaskInfo->env, code);
    }
dengyihao's avatar
dengyihao 已提交
2670
  }
X
Xiaoyu Wang 已提交
2671

2672
  pInfo->base.dataReader = source->dataReader;
H
Haojun Liao 已提交
2673
  STsdbReader* reader = pInfo->base.dataReader;
X
Xiaoyu Wang 已提交
2674
  bool         hasNext = false;
2675
  qTrace("tsdb/read-table-data: %p, enter next reader", reader);
D
dapan1121 已提交
2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687

  while (true) {
    code = tsdbNextDataBlock(reader, &hasNext);
    if (code != 0) {
      tsdbReleaseDataBlock(reader);
      pInfo->base.dataReader = NULL;
      T_LONG_JMP(pTaskInfo->env, code);
    }

    if (!hasNext) {
      break;
    }
X
Xiaoyu Wang 已提交
2688

H
Haojun Liao 已提交
2689
    if (isTaskKilled(pTaskInfo)) {
X
Xiaoyu Wang 已提交
2690
      tsdbReleaseDataBlock(reader);
D
dapan1121 已提交
2691
      pInfo->base.dataReader = NULL;
2692
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
dengyihao's avatar
opt mem  
dengyihao 已提交
2693 2694 2695
    }

    // process this data block based on the probabilities
H
Haojun Liao 已提交
2696
    bool processThisBlock = processBlockWithProbability(&pInfo->sample);
dengyihao's avatar
opt mem  
dengyihao 已提交
2697 2698 2699 2700
    if (!processThisBlock) {
      continue;
    }

H
Haojun Liao 已提交
2701
    if (pQueryCond->order == TSDB_ORDER_ASC) {
dengyihao's avatar
opt mem  
dengyihao 已提交
2702 2703 2704 2705
      pQueryCond->twindows.skey = pBlock->info.window.ekey + 1;
    } else {
      pQueryCond->twindows.ekey = pBlock->info.window.skey - 1;
    }
dengyihao's avatar
opt mem  
dengyihao 已提交
2706 2707

    uint32_t status = 0;
2708
    code = loadDataBlock(pOperator, &pInfo->base, pBlock, &status);
S
slzhou 已提交
2709
    //    code = loadDataBlockFromOneTable(pOperator, pTableScanInfo, pBlock, &status);
dengyihao's avatar
opt mem  
dengyihao 已提交
2710
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
2711
      T_LONG_JMP(pTaskInfo->env, code);
dengyihao's avatar
opt mem  
dengyihao 已提交
2712 2713 2714 2715 2716 2717 2718
    }

    // current block is filter out according to filter condition, continue load the next block
    if (status == FUNC_DATA_REQUIRED_FILTEROUT || pBlock->info.rows == 0) {
      continue;
    }

2719
    pBlock->info.id.groupId = getTableGroupId(pInfo->base.pTableListInfo, pBlock->info.id.uid);
dengyihao's avatar
opt mem  
dengyihao 已提交
2720

H
Haojun Liao 已提交
2721
    pOperator->resultInfo.totalRows += pBlock->info.rows;
H
Haojun Liao 已提交
2722
    pInfo->base.readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;
dengyihao's avatar
opt mem  
dengyihao 已提交
2723

2724
    qTrace("tsdb/read-table-data: %p, close reader", reader);
D
dapan1121 已提交
2725 2726 2727 2728
    if (!source->multiReader) {
      tsdbReaderClose(pInfo->base.dataReader);
      source->dataReader = NULL;
    }
H
Haojun Liao 已提交
2729
    pInfo->base.dataReader = NULL;
dengyihao's avatar
opt mem  
dengyihao 已提交
2730 2731
    return pBlock;
  }
H
Haojun Liao 已提交
2732

D
dapan1121 已提交
2733 2734 2735 2736
  if (!source->multiReader) {
    tsdbReaderClose(pInfo->base.dataReader);
    source->dataReader = NULL;
  }
H
Haojun Liao 已提交
2737
  pInfo->base.dataReader = NULL;
dengyihao's avatar
opt mem  
dengyihao 已提交
2738 2739 2740
  return NULL;
}

2741 2742 2743
SArray* generateSortByTsInfo(SArray* colMatchInfo, int32_t order) {
  int32_t tsTargetSlotId = 0;
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
H
Haojun Liao 已提交
2744
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
2745
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
H
Haojun Liao 已提交
2746
      tsTargetSlotId = colInfo->dstSlotId;
2747 2748 2749
    }
  }

2750 2751 2752
  SArray*         pList = taosArrayInit(1, sizeof(SBlockOrderInfo));
  SBlockOrderInfo bi = {0};
  bi.order = order;
2753
  bi.slotId = tsTargetSlotId;
2754 2755 2756 2757 2758 2759 2760
  bi.nullFirst = NULL_ORDER_FIRST;

  taosArrayPush(pList, &bi);

  return pList;
}

H
Haojun Liao 已提交
2761
int32_t dumpQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
dengyihao's avatar
opt mem  
dengyihao 已提交
2762 2763 2764 2765 2766 2767 2768
  memcpy((void*)dst, (void*)src, sizeof(SQueryTableDataCond));
  dst->colList = taosMemoryCalloc(src->numOfCols, sizeof(SColumnInfo));
  for (int i = 0; i < src->numOfCols; i++) {
    dst->colList[i] = src->colList[i];
  }
  return 0;
}
H
Haojun Liao 已提交
2769

2770
int32_t startGroupTableMergeScan(SOperatorInfo* pOperator) {
2771 2772 2773
  STableMergeScanInfo* pInfo = pOperator->info;
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;

S
slzhou 已提交
2774
  {
2775
    size_t  numOfTables = tableListGetSize(pInfo->base.pTableListInfo);
S
slzhou 已提交
2776
    int32_t i = pInfo->tableStartIndex + 1;
H
Haojun Liao 已提交
2777
    for (; i < numOfTables; ++i) {
2778
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->base.pTableListInfo, i);
S
slzhou 已提交
2779 2780 2781 2782 2783 2784
      if (tableKeyInfo->groupId != pInfo->groupId) {
        break;
      }
    }
    pInfo->tableEndIndex = i - 1;
  }
2785

S
slzhou 已提交
2786 2787
  int32_t tableStartIdx = pInfo->tableStartIndex;
  int32_t tableEndIdx = pInfo->tableEndIndex;
2788

H
Haojun Liao 已提交
2789
  pInfo->base.dataReader = NULL;
2790

2791 2792
  // todo the total available buffer should be determined by total capacity of buffer of this task.
  // the additional one is reserved for merge result
S
slzhou 已提交
2793
  pInfo->sortBufSize = pInfo->bufPageSize * (tableEndIdx - tableStartIdx + 1 + 1);
2794
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
L
Liu Jicong 已提交
2795 2796
  pInfo->pSortHandle = tsortCreateSortHandle(pInfo->pSortInfo, SORT_MULTISOURCE_MERGE, pInfo->bufPageSize, numOfBufPage,
                                             pInfo->pSortInputBlock, pTaskInfo->id.str);
2797

dengyihao's avatar
dengyihao 已提交
2798
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getTableDataBlockImpl, NULL, NULL);
dengyihao's avatar
opt mem  
dengyihao 已提交
2799 2800 2801 2802 2803 2804

  // one table has one data block
  int32_t numOfTable = tableEndIdx - tableStartIdx + 1;
  pInfo->queryConds = taosArrayInit(numOfTable, sizeof(SQueryTableDataCond));

  for (int32_t i = 0; i < numOfTable; ++i) {
2805 2806 2807
    STableMergeScanSortSourceParam param = {0};
    param.readerIdx = i;
    param.pOperator = pOperator;
D
dapan1121 已提交
2808
    param.multiReader = (numOfTable <= MULTI_READER_MAX_TABLE_NUM) ? true : false;
2809
    param.inputBlock = createOneDataBlock(pInfo->pResBlock, false);
H
Haojun Liao 已提交
2810 2811
    blockDataEnsureCapacity(param.inputBlock, pOperator->resultInfo.capacity);

2812
    taosArrayPush(pInfo->sortSourceParams, &param);
dengyihao's avatar
opt mem  
dengyihao 已提交
2813 2814

    SQueryTableDataCond cond;
H
Haojun Liao 已提交
2815
    dumpQueryTableCond(&pInfo->base.cond, &cond);
dengyihao's avatar
opt mem  
dengyihao 已提交
2816
    taosArrayPush(pInfo->queryConds, &cond);
2817 2818
  }

dengyihao's avatar
opt mem  
dengyihao 已提交
2819
  for (int32_t i = 0; i < numOfTable; ++i) {
2820
    SSortSource*                    ps = taosMemoryCalloc(1, sizeof(SSortSource));
2821
    STableMergeScanSortSourceParam* param = taosArrayGet(pInfo->sortSourceParams, i);
2822
    ps->param = param;
2823
    ps->onlyRef = true;
2824 2825 2826 2827 2828 2829
    tsortAddSource(pInfo->pSortHandle, ps);
  }

  int32_t code = tsortOpen(pInfo->pSortHandle);

  if (code != TSDB_CODE_SUCCESS) {
2830
    T_LONG_JMP(pTaskInfo->env, terrno);
2831 2832
  }

2833 2834 2835 2836 2837 2838 2839
  return TSDB_CODE_SUCCESS;
}

int32_t stopGroupTableMergeScan(SOperatorInfo* pOperator) {
  STableMergeScanInfo* pInfo = pOperator->info;
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;

dengyihao's avatar
dengyihao 已提交
2840
  int32_t numOfTable = taosArrayGetSize(pInfo->queryConds);
2841

2842 2843 2844 2845 2846 2847 2848
  SSortExecInfo sortExecInfo = tsortGetSortExecInfo(pInfo->pSortHandle);
  pInfo->sortExecInfo.sortMethod = sortExecInfo.sortMethod;
  pInfo->sortExecInfo.sortBuffer = sortExecInfo.sortBuffer;
  pInfo->sortExecInfo.loops += sortExecInfo.loops;
  pInfo->sortExecInfo.readBytes += sortExecInfo.readBytes;
  pInfo->sortExecInfo.writeBytes += sortExecInfo.writeBytes;

dengyihao's avatar
dengyihao 已提交
2849
  for (int32_t i = 0; i < numOfTable; ++i) {
2850 2851
    STableMergeScanSortSourceParam* param = taosArrayGet(pInfo->sortSourceParams, i);
    blockDataDestroy(param->inputBlock);
2852 2853
    tsdbReaderClose(param->dataReader);
    param->dataReader = NULL;
2854
  }
2855 2856
  taosArrayClear(pInfo->sortSourceParams);

2857
  tsortDestroySortHandle(pInfo->pSortHandle);
dengyihao's avatar
dengyihao 已提交
2858
  pInfo->pSortHandle = NULL;
2859

dengyihao's avatar
opt mem  
dengyihao 已提交
2860 2861 2862
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->queryConds); i++) {
    SQueryTableDataCond* cond = taosArrayGet(pInfo->queryConds, i);
    taosMemoryFree(cond->colList);
2863
  }
dengyihao's avatar
opt mem  
dengyihao 已提交
2864 2865 2866
  taosArrayDestroy(pInfo->queryConds);
  pInfo->queryConds = NULL;

2867
  resetLimitInfoForNextGroup(&pInfo->limitInfo);
2868 2869 2870
  return TSDB_CODE_SUCCESS;
}

2871 2872
// all data produced by this function only belongs to one group
// slimit/soffset does not need to be concerned here, since this function only deal with data within one group.
L
Liu Jicong 已提交
2873 2874
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
                                              SOperatorInfo* pOperator) {
2875 2876 2877
  STableMergeScanInfo* pInfo = pOperator->info;
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;

2878
  blockDataCleanup(pResBlock);
2879 2880

  while (1) {
2881
    STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
2882 2883 2884 2885
    if (pTupleHandle == NULL) {
      break;
    }

2886 2887
    appendOneRowToDataBlock(pResBlock, pTupleHandle);
    if (pResBlock->info.rows >= capacity) {
2888 2889 2890 2891
      break;
    }
  }

2892
  bool limitReached = applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo);
D
dapan1121 已提交
2893
  qDebug("%s get sorted row block, rows:%" PRId64 ", limit:%" PRId64, GET_TASKID(pTaskInfo), pResBlock->info.rows,
2894
         pInfo->limitInfo.numOfOutputRows);
2895

2896
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908
}

SSDataBlock* doTableMergeScan(SOperatorInfo* pOperator) {
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
  STableMergeScanInfo* pInfo = pOperator->info;

  int32_t code = pOperator->fpSet._openFn(pOperator);
  if (code != TSDB_CODE_SUCCESS) {
2909
    T_LONG_JMP(pTaskInfo->env, code);
2910
  }
2911

2912
  size_t tableListSize = tableListGetSize(pInfo->base.pTableListInfo);
S
slzhou 已提交
2913 2914
  if (!pInfo->hasGroupId) {
    pInfo->hasGroupId = true;
2915

S
slzhou 已提交
2916
    if (tableListSize == 0) {
H
Haojun Liao 已提交
2917
      setOperatorCompleted(pOperator);
2918 2919
      return NULL;
    }
S
slzhou 已提交
2920
    pInfo->tableStartIndex = 0;
2921
    pInfo->groupId = ((STableKeyInfo*)tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex))->groupId;
2922 2923
    startGroupTableMergeScan(pOperator);
  }
2924

S
slzhou 已提交
2925 2926
  SSDataBlock* pBlock = NULL;
  while (pInfo->tableStartIndex < tableListSize) {
2927 2928 2929 2930
    if (isTaskKilled(pTaskInfo)) {
      T_LONG_JMP(pTaskInfo->env, pTaskInfo->code);
    }

L
Liu Jicong 已提交
2931 2932
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
                                              pOperator);
S
slzhou 已提交
2933
    if (pBlock != NULL) {
H
Haojun Liao 已提交
2934
      pBlock->info.id.groupId = pInfo->groupId;
S
slzhou 已提交
2935 2936 2937
      pOperator->resultInfo.totalRows += pBlock->info.rows;
      return pBlock;
    } else {
2938
      // Data of this group are all dumped, let's try the next group
S
slzhou 已提交
2939 2940
      stopGroupTableMergeScan(pOperator);
      if (pInfo->tableEndIndex >= tableListSize - 1) {
H
Haojun Liao 已提交
2941
        setOperatorCompleted(pOperator);
S
slzhou 已提交
2942 2943
        break;
      }
2944

S
slzhou 已提交
2945
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
2946
      pInfo->groupId = tableListGetInfo(pInfo->base.pTableListInfo, pInfo->tableStartIndex)->groupId;
S
slzhou 已提交
2947
      startGroupTableMergeScan(pOperator);
X
Xiaoyu Wang 已提交
2948
      resetLimitInfoForNextGroup(&pInfo->limitInfo);
S
slzhou 已提交
2949
    }
wmmhello's avatar
wmmhello 已提交
2950 2951
  }

2952 2953 2954
  return pBlock;
}

2955
void destroyTableMergeScanOperatorInfo(void* param) {
2956
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
H
Haojun Liao 已提交
2957
  cleanupQueryTableDataCond(&pTableScanInfo->base.cond);
2958

dengyihao's avatar
dengyihao 已提交
2959 2960 2961
  int32_t numOfTable = taosArrayGetSize(pTableScanInfo->queryConds);

  for (int32_t i = 0; i < numOfTable; i++) {
H
Haojun Liao 已提交
2962 2963
    STableMergeScanSortSourceParam* p = taosArrayGet(pTableScanInfo->sortSourceParams, i);
    blockDataDestroy(p->inputBlock);
2964 2965
    tsdbReaderClose(p->dataReader);
    p->dataReader = NULL;
2966
  }
H
Haojun Liao 已提交
2967

D
dapan1121 已提交
2968 2969 2970
  tsdbReaderClose(pTableScanInfo->base.dataReader);
  pTableScanInfo->base.dataReader = NULL;

2971
  taosArrayDestroy(pTableScanInfo->sortSourceParams);
dengyihao's avatar
dengyihao 已提交
2972 2973
  tsortDestroySortHandle(pTableScanInfo->pSortHandle);
  pTableScanInfo->pSortHandle = NULL;
2974

dengyihao's avatar
opt mem  
dengyihao 已提交
2975 2976 2977
  for (int i = 0; i < taosArrayGetSize(pTableScanInfo->queryConds); i++) {
    SQueryTableDataCond* pCond = taosArrayGet(pTableScanInfo->queryConds, i);
    taosMemoryFree(pCond->colList);
2978 2979
  }

2980 2981
  taosArrayDestroy(pTableScanInfo->queryConds);
  destroyTableScanBase(&pTableScanInfo->base);
2982 2983 2984 2985 2986

  pTableScanInfo->pResBlock = blockDataDestroy(pTableScanInfo->pResBlock);
  pTableScanInfo->pSortInputBlock = blockDataDestroy(pTableScanInfo->pSortInputBlock);

  taosArrayDestroy(pTableScanInfo->pSortInfo);
D
dapan1121 已提交
2987
  taosMemoryFreeClear(param);
2988 2989 2990 2991
}

int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
  ASSERT(pOptr != NULL);
2992 2993
  // TODO: merge these two info into one struct
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
L
Liu Jicong 已提交
2994
  STableMergeScanInfo*     pInfo = pOptr->info;
H
Haojun Liao 已提交
2995
  execInfo->blockRecorder = pInfo->base.readRecorder;
2996
  execInfo->sortExecInfo = pInfo->sortExecInfo;
2997 2998 2999

  *pOptrExplain = execInfo;
  *len = sizeof(STableMergeScanExecInfo);
L
Liu Jicong 已提交
3000

3001 3002 3003
  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
3004
SOperatorInfo* createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
3005
                                                STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo) {
3006 3007 3008 3009 3010
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
3011

3012 3013 3014
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;

  int32_t numOfCols = 0;
3015
  int32_t code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
H
Haojun Liao 已提交
3016
                                     &pInfo->base.matchInfo);
H
Haojun Liao 已提交
3017 3018 3019
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
3020

H
Haojun Liao 已提交
3021
  code = initQueryTableDataCond(&pInfo->base.cond, pTableScanNode);
3022
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
3023
    taosArrayDestroy(pInfo->base.matchInfo.pList);
3024 3025 3026 3027
    goto _error;
  }

  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
H
Haojun Liao 已提交
3028
    SExprSupp* pSup = &pInfo->base.pseudoSup;
3029 3030
    pSup->pExprInfo = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->numOfExprs);
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset);
3031 3032 3033 3034
  }

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

H
Haojun Liao 已提交
3035 3036 3037 3038 3039 3040
  pInfo->base.metaCache.pTableMetaEntryCache = taosLRUCacheInit(1024 * 128, -1, .5);
  if (pInfo->base.metaCache.pTableMetaEntryCache == NULL) {
    code = terrno;
    goto _error;
  }

H
Haojun Liao 已提交
3041 3042
  pInfo->base.dataBlockLoadFlag = FUNC_DATA_REQUIRED_DATA_LOAD;
  pInfo->base.scanFlag = MAIN_SCAN;
H
Haojun Liao 已提交
3043
  pInfo->base.readHandle = *readHandle;
3044 3045 3046

  pInfo->base.limitInfo.limit.limit = -1;
  pInfo->base.limitInfo.slimit.limit = -1;
3047
  pInfo->base.pTableListInfo = pTableListInfo;
H
Haojun Liao 已提交
3048

3049
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
L
Liu Jicong 已提交
3050
  pInfo->sample.seed = taosGetTimestampSec();
H
Haojun Liao 已提交
3051 3052 3053 3054 3055 3056

  code = filterInitFromNode((SNode*)pTableScanNode->scan.node.pConditions, &pOperator->exprSupp.pFilterInfo, 0);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

H
Haojun Liao 已提交
3057
  initResultSizeInfo(&pOperator->resultInfo, 1024);
H
Haojun Liao 已提交
3058
  pInfo->pResBlock = createDataBlockFromDescNode(pDescNode);
H
Haojun Liao 已提交
3059 3060
  blockDataEnsureCapacity(pInfo->pResBlock, pOperator->resultInfo.capacity);

3061
  pInfo->sortSourceParams = taosArrayInit(64, sizeof(STableMergeScanSortSourceParam));
3062

H
Haojun Liao 已提交
3063
  pInfo->pSortInfo = generateSortByTsInfo(pInfo->base.matchInfo.pList, pInfo->base.cond.order);
3064
  pInfo->pSortInputBlock = createOneDataBlock(pInfo->pResBlock, false);
3065
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
3066

dengyihao's avatar
dengyihao 已提交
3067
  int32_t  rowSize = pInfo->pResBlock->info.rowSize;
A
Alex Duan 已提交
3068 3069
  uint32_t nCols = taosArrayGetSize(pInfo->pResBlock->pDataBlock);
  pInfo->bufPageSize = getProperSortPageSize(rowSize, nCols);
3070

L
Liu Jicong 已提交
3071 3072
  setOperatorInfo(pOperator, "TableMergeScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN, false, OP_NOT_OPENED,
                  pInfo, pTaskInfo);
L
Liu Jicong 已提交
3073
  pOperator->exprSupp.numOfExprs = numOfCols;
3074

3075 3076
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableMergeScan, NULL, destroyTableMergeScanOperatorInfo,
                                         optrDefaultBufFn, getTableMergeScanExplainExecInfo);
3077 3078 3079 3080 3081 3082 3083 3084 3085
  pOperator->cost.openCost = 0;
  return pOperator;

_error:
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  return NULL;
}
S
shenglian zhou 已提交
3086 3087 3088 3089

// ====================================================================================================================
// TableCountScanOperator
static SSDataBlock* doTableCountScan(SOperatorInfo* pOperator);
S
slzhou 已提交
3090
static void         destoryTableCountScanOperator(void* param);
S
slzhou 已提交
3091 3092 3093 3094 3095 3096
static void         buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
                                                   SSDataBlock* pRes, char* dbName, tb_uid_t stbUid);
static void         buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
                                                   SSDataBlock* pRes, char* dbName);
static void         buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                              STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName);
L
Liu Jicong 已提交
3097 3098
static void         buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                                STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName);
S
slzhou 已提交
3099 3100 3101 3102 3103 3104 3105
static SSDataBlock* buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes);
static void         buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                                STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
                                                size_t perfdbTableNum);
static void         buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
                                               size_t infodbTableNum, size_t perfdbTableNum);
S
slzhou 已提交
3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166
static const char*  GROUP_TAG_DB_NAME = "db_name";
static const char*  GROUP_TAG_STABLE_NAME = "stable_name";

int32_t tblCountScanGetGroupTagsSlotId(const SNodeList* scanCols, STableCountScanSupp* supp) {
  if (scanCols != NULL) {
    SNode* pNode = NULL;
    FOREACH(pNode, scanCols) {
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
        return TSDB_CODE_QRY_SYS_ERROR;
      }
      STargetNode* targetNode = (STargetNode*)pNode;
      if (nodeType(targetNode->pExpr) != QUERY_NODE_COLUMN) {
        return TSDB_CODE_QRY_SYS_ERROR;
      }
      SColumnNode* colNode = (SColumnNode*)(targetNode->pExpr);
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
        supp->dbNameSlotId = targetNode->slotId;
      } else if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
        supp->stbNameSlotId = targetNode->slotId;
      }
    }
  }
  return TSDB_CODE_SUCCESS;
}

int32_t tblCountScanGetCountSlotId(const SNodeList* pseudoCols, STableCountScanSupp* supp) {
  if (pseudoCols != NULL) {
    SNode* pNode = NULL;
    FOREACH(pNode, pseudoCols) {
      if (nodeType(pNode) != QUERY_NODE_TARGET) {
        return TSDB_CODE_QRY_SYS_ERROR;
      }
      STargetNode* targetNode = (STargetNode*)pNode;
      if (nodeType(targetNode->pExpr) != QUERY_NODE_FUNCTION) {
        return TSDB_CODE_QRY_SYS_ERROR;
      }
      SFunctionNode* funcNode = (SFunctionNode*)(targetNode->pExpr);
      if (funcNode->funcType == FUNCTION_TYPE_TABLE_COUNT) {
        supp->tbCountSlotId = targetNode->slotId;
      }
    }
  }
  return TSDB_CODE_SUCCESS;
}

int32_t tblCountScanGetInputs(SNodeList* groupTags, SName* tableName, STableCountScanSupp* supp) {
  if (groupTags != NULL) {
    SNode* pNode = NULL;
    FOREACH(pNode, groupTags) {
      if (nodeType(pNode) != QUERY_NODE_COLUMN) {
        return TSDB_CODE_QRY_SYS_ERROR;
      }
      SColumnNode* colNode = (SColumnNode*)pNode;
      if (strcmp(colNode->colName, GROUP_TAG_DB_NAME) == 0) {
        supp->groupByDbName = true;
      }
      if (strcmp(colNode->colName, GROUP_TAG_STABLE_NAME) == 0) {
        supp->groupByStbName = true;
      }
    }
  } else {
H
Haojun Liao 已提交
3167 3168
    tstrncpy(supp->dbNameFilter, tNameGetDbNameP(tableName), TSDB_DB_NAME_LEN);
    tstrncpy(supp->stbNameFilter, tNameGetTableName(tableName), TSDB_TABLE_NAME_LEN);
S
slzhou 已提交
3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196
  }
  return TSDB_CODE_SUCCESS;
}

int32_t getTableCountScanSupp(SNodeList* groupTags, SName* tableName, SNodeList* scanCols, SNodeList* pseudoCols,
                              STableCountScanSupp* supp, SExecTaskInfo* taskInfo) {
  int32_t code = 0;
  code = tblCountScanGetInputs(groupTags, tableName, supp);
  if (code != TSDB_CODE_SUCCESS) {
    qError("%s get table count scan supp. get inputs error", GET_TASKID(taskInfo));
    return code;
  }
  supp->dbNameSlotId = -1;
  supp->stbNameSlotId = -1;
  supp->tbCountSlotId = -1;

  code = tblCountScanGetGroupTagsSlotId(scanCols, supp);
  if (code != TSDB_CODE_SUCCESS) {
    qError("%s get table count scan supp. get group tags slot id error", GET_TASKID(taskInfo));
    return code;
  }
  code = tblCountScanGetCountSlotId(pseudoCols, supp);
  if (code != TSDB_CODE_SUCCESS) {
    qError("%s get table count scan supp. get count error", GET_TASKID(taskInfo));
    return code;
  }
  return code;
}
S
shenglian zhou 已提交
3197

S
slzhou 已提交
3198
SOperatorInfo* createTableCountScanOperatorInfo(SReadHandle* readHandle, STableCountScanPhysiNode* pTblCountScanNode,
S
shenglian zhou 已提交
3199 3200 3201
                                                SExecTaskInfo* pTaskInfo) {
  int32_t code = TSDB_CODE_SUCCESS;

S
slzhou 已提交
3202
  SScanPhysiNode*              pScanNode = &pTblCountScanNode->scan;
S
slzhou 已提交
3203
  STableCountScanOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableCountScanOperatorInfo));
S
slzhou 已提交
3204
  SOperatorInfo*               pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
S
shenglian zhou 已提交
3205 3206 3207 3208 3209 3210 3211 3212 3213

  if (!pInfo || !pOperator) {
    goto _error;
  }

  pInfo->readHandle = *readHandle;

  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;
  initResultSizeInfo(&pOperator->resultInfo, 1);
3214
  pInfo->pRes = createDataBlockFromDescNode(pDescNode);
S
shenglian zhou 已提交
3215 3216
  blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);

S
slzhou 已提交
3217 3218 3219
  getTableCountScanSupp(pTblCountScanNode->pGroupTags, &pTblCountScanNode->scan.tableName,
                        pTblCountScanNode->scan.pScanCols, pTblCountScanNode->scan.pScanPseudoCols, &pInfo->supp,
                        pTaskInfo);
S
shenglian zhou 已提交
3220 3221 3222

  setOperatorInfo(pOperator, "TableCountScanOperator", QUERY_NODE_PHYSICAL_PLAN_TABLE_COUNT_SCAN, false, OP_NOT_OPENED,
                  pInfo, pTaskInfo);
L
Liu Jicong 已提交
3223 3224
  pOperator->fpSet = createOperatorFpSet(optrDummyOpenFn, doTableCountScan, NULL, destoryTableCountScanOperator,
                                         optrDefaultBufFn, NULL);
S
shenglian zhou 已提交
3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235
  return pOperator;

_error:
  if (pInfo != NULL) {
    destoryTableCountScanOperator(pInfo);
  }
  taosMemoryFreeClear(pOperator);
  pTaskInfo->code = code;
  return NULL;
}

S
slzhou 已提交
3236 3237 3238
void fillTableCountScanDataBlock(STableCountScanSupp* pSupp, char* dbName, char* stbName, int64_t count,
                                 SSDataBlock* pRes) {
  if (pSupp->dbNameSlotId != -1) {
3239
    ASSERT(strlen(dbName));
S
slzhou 已提交
3240
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->dbNameSlotId);
H
Haojun Liao 已提交
3241 3242 3243 3244

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

S
slzhou 已提交
3245
    varDataSetLen(varDbName, strlen(dbName));
3246
    colDataSetVal(colInfoData, 0, varDbName, false);
S
slzhou 已提交
3247 3248 3249 3250
  }

  if (pSupp->stbNameSlotId != -1) {
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->stbNameSlotId);
3251
    if (strlen(stbName) != 0) {
S
slzhou 已提交
3252
      char varStbName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
H
Haojun Liao 已提交
3253
      strncpy(varDataVal(varStbName), stbName, TSDB_TABLE_NAME_LEN);
3254
      varDataSetLen(varStbName, strlen(stbName));
3255
      colDataSetVal(colInfoData, 0, varStbName, false);
3256
    } else {
3257
      colDataSetNULL(colInfoData, 0);
3258
    }
S
slzhou 已提交
3259 3260 3261
  }

  if (pSupp->tbCountSlotId != -1) {
S
slzhou 已提交
3262
    SColumnInfoData* colInfoData = taosArrayGet(pRes->pDataBlock, pSupp->tbCountSlotId);
3263
    colDataSetVal(colInfoData, 0, (char*)&count, false);
S
slzhou 已提交
3264 3265 3266 3267
  }
  pRes->info.rows = 1;
}

S
slzhou 已提交
3268
static SSDataBlock* buildSysDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo) {
S
slzhou 已提交
3269 3270 3271
  STableCountScanSupp* pSupp = &pInfo->supp;
  SSDataBlock*         pRes = pInfo->pRes;

S
slzhou 已提交
3272
  size_t infodbTableNum;
S
slzhou 已提交
3273
  getInfosDbMeta(NULL, &infodbTableNum);
S
slzhou 已提交
3274
  size_t perfdbTableNum;
S
slzhou 已提交
3275 3276
  getPerfDbMeta(NULL, &perfdbTableNum);

3277
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
S
slzhou 已提交
3278
    buildSysDbGroupedTableCount(pOperator, pInfo, pSupp, pRes, infodbTableNum, perfdbTableNum);
S
slzhou 已提交
3279 3280
    return (pRes->info.rows > 0) ? pRes : NULL;
  } else {
S
slzhou 已提交
3281
    buildSysDbFilterTableCount(pOperator, pSupp, pRes, infodbTableNum, perfdbTableNum);
S
slzhou 已提交
3282 3283 3284 3285
    return (pRes->info.rows > 0) ? pRes : NULL;
  }
}

S
slzhou 已提交
3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297 3298 3299 3300 3301
static void buildSysDbFilterTableCount(SOperatorInfo* pOperator, STableCountScanSupp* pSupp, SSDataBlock* pRes,
                                       size_t infodbTableNum, size_t perfdbTableNum) {
  if (strcmp(pSupp->dbNameFilter, TSDB_INFORMATION_SCHEMA_DB) == 0) {
    fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
  } else if (strcmp(pSupp->dbNameFilter, TSDB_PERFORMANCE_SCHEMA_DB) == 0) {
    fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
  } else if (strlen(pSupp->dbNameFilter) == 0) {
    fillTableCountScanDataBlock(pSupp, "", "", infodbTableNum + perfdbTableNum, pRes);
  }
  setOperatorCompleted(pOperator);
}

static void buildSysDbGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, size_t infodbTableNum,
                                        size_t perfdbTableNum) {
  if (pInfo->currGrpIdx == 0) {
3302 3303 3304 3305 3306 3307
    uint64_t groupId = 0;
    if (pSupp->groupByDbName) {
      groupId = calcGroupId(TSDB_INFORMATION_SCHEMA_DB, strlen(TSDB_INFORMATION_SCHEMA_DB));
    } else {
      groupId = calcGroupId("", 0);
    }
X
Xiaoyu Wang 已提交
3308

S
slzhou 已提交
3309 3310 3311
    pRes->info.id.groupId = groupId;
    fillTableCountScanDataBlock(pSupp, TSDB_INFORMATION_SCHEMA_DB, "", infodbTableNum, pRes);
  } else if (pInfo->currGrpIdx == 1) {
3312 3313 3314 3315 3316 3317 3318
    uint64_t groupId = 0;
    if (pSupp->groupByDbName) {
      groupId = calcGroupId(TSDB_PERFORMANCE_SCHEMA_DB, strlen(TSDB_PERFORMANCE_SCHEMA_DB));
    } else {
      groupId = calcGroupId("", 0);
    }

S
slzhou 已提交
3319 3320 3321 3322 3323 3324 3325 3326
    pRes->info.id.groupId = groupId;
    fillTableCountScanDataBlock(pSupp, TSDB_PERFORMANCE_SCHEMA_DB, "", perfdbTableNum, pRes);
  } else {
    setOperatorCompleted(pOperator);
  }
  pInfo->currGrpIdx++;
}

S
shenglian zhou 已提交
3327
static SSDataBlock* doTableCountScan(SOperatorInfo* pOperator) {
S
slzhou 已提交
3328 3329 3330 3331
  SExecTaskInfo*               pTaskInfo = pOperator->pTaskInfo;
  STableCountScanOperatorInfo* pInfo = pOperator->info;
  STableCountScanSupp*         pSupp = &pInfo->supp;
  SSDataBlock*                 pRes = pInfo->pRes;
S
slzhou 已提交
3332
  blockDataCleanup(pRes);
3333

S
slzhou 已提交
3334 3335 3336
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }
S
slzhou 已提交
3337
  if (pInfo->readHandle.mnd != NULL) {
S
slzhou 已提交
3338
    return buildSysDbTableCount(pOperator, pInfo);
S
slzhou 已提交
3339
  }
S
slzhou 已提交
3340

S
slzhou 已提交
3341 3342 3343 3344 3345
  return buildVnodeDbTableCount(pOperator, pInfo, pSupp, pRes);
}

static SSDataBlock* buildVnodeDbTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                           STableCountScanSupp* pSupp, SSDataBlock* pRes) {
S
slzhou 已提交
3346 3347
  const char* db = NULL;
  int32_t     vgId = 0;
S
slzhou 已提交
3348
  char        dbName[TSDB_DB_NAME_LEN] = {0};
S
slzhou 已提交
3349

S
slzhou 已提交
3350 3351 3352 3353 3354 3355
  // get dbname
  vnodeGetInfo(pInfo->readHandle.vnode, &db, &vgId);
  SName sn = {0};
  tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
  tNameGetDbName(&sn, dbName);

3356
  if (pSupp->groupByDbName || pSupp->groupByStbName) {
S
slzhou 已提交
3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370
    buildVnodeGroupedTableCount(pOperator, pInfo, pSupp, pRes, vgId, dbName);
  } else {
    buildVnodeFilteredTbCount(pOperator, pInfo, pSupp, pRes, dbName);
  }
  return pRes->info.rows > 0 ? pRes : NULL;
}

static void buildVnodeGroupedTableCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                        STableCountScanSupp* pSupp, SSDataBlock* pRes, int32_t vgId, char* dbName) {
  if (pSupp->groupByStbName) {
    if (pInfo->stbUidList == NULL) {
      pInfo->stbUidList = taosArrayInit(16, sizeof(tb_uid_t));
      if (vnodeGetStbIdList(pInfo->readHandle.vnode, 0, pInfo->stbUidList) < 0) {
        qError("vgId:%d, failed to get stb id list error: %s", vgId, terrstr());
S
slzhou 已提交
3371
      }
S
slzhou 已提交
3372 3373 3374 3375 3376 3377 3378 3379 3380 3381
    }
    if (pInfo->currGrpIdx < taosArrayGetSize(pInfo->stbUidList)) {
      tb_uid_t stbUid = *(tb_uid_t*)taosArrayGet(pInfo->stbUidList, pInfo->currGrpIdx);
      buildVnodeGroupedStbTableCount(pInfo, pSupp, pRes, dbName, stbUid);

      pInfo->currGrpIdx++;
    } else if (pInfo->currGrpIdx == taosArrayGetSize(pInfo->stbUidList)) {
      buildVnodeGroupedNtbTableCount(pInfo, pSupp, pRes, dbName);

      pInfo->currGrpIdx++;
S
slzhou 已提交
3382
    } else {
S
slzhou 已提交
3383
      setOperatorCompleted(pOperator);
S
slzhou 已提交
3384 3385
    }
  } else {
S
slzhou 已提交
3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402
    uint64_t groupId = calcGroupId(dbName, strlen(dbName));
    pRes->info.id.groupId = groupId;
    int64_t dbTableCount = metaGetTbNum(pInfo->readHandle.meta);
    fillTableCountScanDataBlock(pSupp, dbName, "", dbTableCount, pRes);
    setOperatorCompleted(pOperator);
  }
}

static void buildVnodeFilteredTbCount(SOperatorInfo* pOperator, STableCountScanOperatorInfo* pInfo,
                                      STableCountScanSupp* pSupp, SSDataBlock* pRes, char* dbName) {
  if (strlen(pSupp->dbNameFilter) != 0) {
    if (strlen(pSupp->stbNameFilter) != 0) {
      tb_uid_t      uid = metaGetTableEntryUidByName(pInfo->readHandle.meta, pSupp->stbNameFilter);
      SMetaStbStats stats = {0};
      metaGetStbStats(pInfo->readHandle.meta, uid, &stats);
      int64_t ctbNum = stats.ctbNum;
      fillTableCountScanDataBlock(pSupp, dbName, pSupp->stbNameFilter, ctbNum, pRes);
S
slzhou 已提交
3403 3404 3405
    } else {
      int64_t tbNumVnode = metaGetTbNum(pInfo->readHandle.meta);
      fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
S
slzhou 已提交
3406
    }
S
slzhou 已提交
3407 3408 3409
  } else {
    int64_t tbNumVnode = metaGetTbNum(pInfo->readHandle.meta);
    fillTableCountScanDataBlock(pSupp, dbName, "", tbNumVnode, pRes);
S
slzhou 已提交
3410
  }
S
slzhou 已提交
3411 3412 3413 3414 3415 3416
  setOperatorCompleted(pOperator);
}

static void buildVnodeGroupedNtbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
                                           SSDataBlock* pRes, char* dbName) {
  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
3417 3418 3419
  if (pSupp->groupByDbName) {
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, "");
  }
X
Xiaoyu Wang 已提交
3420

S
slzhou 已提交
3421 3422 3423
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
  pRes->info.id.groupId = groupId;
  int64_t ntbNum = metaGetNtbNum(pInfo->readHandle.meta);
3424 3425 3426
  if (ntbNum != 0) {
    fillTableCountScanDataBlock(pSupp, dbName, "", ntbNum, pRes);
  }
S
slzhou 已提交
3427 3428 3429 3430 3431 3432 3433 3434
}

static void buildVnodeGroupedStbTableCount(STableCountScanOperatorInfo* pInfo, STableCountScanSupp* pSupp,
                                           SSDataBlock* pRes, char* dbName, tb_uid_t stbUid) {
  char stbName[TSDB_TABLE_NAME_LEN] = {0};
  metaGetTableSzNameByUid(pInfo->readHandle.meta, stbUid, stbName);

  char fullStbName[TSDB_TABLE_FNAME_LEN] = {0};
3435 3436 3437 3438 3439
  if (pSupp->groupByDbName) {
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s.%s", dbName, stbName);
  } else {
    snprintf(fullStbName, TSDB_TABLE_FNAME_LEN, "%s", stbName);
  }
X
Xiaoyu Wang 已提交
3440

S
slzhou 已提交
3441 3442 3443 3444 3445 3446 3447 3448
  uint64_t groupId = calcGroupId(fullStbName, strlen(fullStbName));
  pRes->info.id.groupId = groupId;

  SMetaStbStats stats = {0};
  metaGetStbStats(pInfo->readHandle.meta, stbUid, &stats);
  int64_t ctbNum = stats.ctbNum;

  fillTableCountScanDataBlock(pSupp, dbName, stbName, ctbNum, pRes);
S
shenglian zhou 已提交
3449 3450 3451
}

static void destoryTableCountScanOperator(void* param) {
S
slzhou 已提交
3452
  STableCountScanOperatorInfo* pTableCountScanInfo = param;
S
shenglian zhou 已提交
3453 3454
  blockDataDestroy(pTableCountScanInfo->pRes);

S
slzhou 已提交
3455
  taosArrayDestroy(pTableCountScanInfo->stbUidList);
S
shenglian zhou 已提交
3456 3457
  taosMemoryFreeClear(param);
}