scanoperator.c 172.4 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
#include "vnode.h"
H
Haojun Liao 已提交
34 35

#define SET_REVERSE_SCAN_FLAG(_info) ((_info)->scanFlag = REVERSE_SCAN)
36
#define SWITCH_ORDER(n)              (((n) = ((n) == TSDB_ORDER_ASC) ? TSDB_ORDER_DESC : TSDB_ORDER_ASC))
37

38
static int32_t buildSysDbTableInfo(const SSysTableScanInfo* pInfo, int32_t capacity);
39 40
static int32_t buildDbTableInfoBlock(bool sysInfo, const SSDataBlock* p, const SSysTableMeta* pSysDbTableMeta,
                                     size_t size, const char* dbName);
41

dengyihao's avatar
dengyihao 已提交
42 43 44
static char* SYSTABLE_IDX_COLUMN[] = {"table_name", "db_name",     "create_time",      "columns",
                                      "ttl",        "stable_name", "vgroup_id', 'uid", "type"};

dengyihao's avatar
opt mem  
dengyihao 已提交
45
static char* SYSTABLE_SPECIAL_COL[] = {"db_name", "vgroup_id"};
dengyihao's avatar
dengyihao 已提交
46

dengyihao's avatar
dengyihao 已提交
47 48 49
typedef int32_t (*__sys_filte)(void* pMeta, SNode* cond, SArray* result);
typedef int32_t (*__sys_check)(SNode* cond);

dengyihao's avatar
dengyihao 已提交
50
typedef struct {
dengyihao's avatar
dengyihao 已提交
51 52 53
  const char* name;
  __sys_check chkFunc;
  __sys_filte fltFunc;
dengyihao's avatar
dengyihao 已提交
54 55
} SSTabFltFuncDef;

dengyihao's avatar
dengyihao 已提交
56 57
typedef struct {
  void* pMeta;
dengyihao's avatar
dengyihao 已提交
58
  void* pVnode;
dengyihao's avatar
dengyihao 已提交
59 60
} SSTabFltArg;

dengyihao's avatar
dengyihao 已提交
61
static int32_t sysChkFilter__Comm(SNode* pNode);
dengyihao's avatar
dengyihao 已提交
62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
static int32_t sysChkFilter__DBName(SNode* pNode);
static int32_t sysChkFilter__VgroupId(SNode* pNode);
static int32_t sysChkFilter__TableName(SNode* pNode);
static int32_t sysChkFilter__CreateTime(SNode* pNode);
static int32_t sysChkFilter__Ncolumn(SNode* pNode);
static int32_t sysChkFilter__Ttl(SNode* pNode);
static int32_t sysChkFilter__STableName(SNode* pNode);
static int32_t sysChkFilter__Uid(SNode* pNode);
static int32_t sysChkFilter__Type(SNode* pNode);

static int32_t sysFilte__DbName(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__VgroupId(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__TableName(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__CreateTime(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__Ncolumn(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__Ttl(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__STableName(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__Uid(void* pMeta, SNode* pNode, SArray* result);
static int32_t sysFilte__Type(void* pMeta, SNode* pNode, SArray* result);

const SSTabFltFuncDef filterDict[] = {
    {.name = "table_name", .chkFunc = sysChkFilter__TableName, .fltFunc = sysFilte__TableName},
    {.name = "db_name", .chkFunc = sysChkFilter__DBName, .fltFunc = sysFilte__DbName},
    {.name = "create_time", .chkFunc = sysChkFilter__CreateTime, .fltFunc = sysFilte__CreateTime},
    {.name = "columns", .chkFunc = sysChkFilter__Ncolumn, .fltFunc = sysFilte__Ncolumn},
    {.name = "ttl", .chkFunc = sysChkFilter__Ttl, .fltFunc = sysFilte__Ttl},
    {.name = "stable_name", .chkFunc = sysChkFilter__STableName, .fltFunc = sysFilte__STableName},
    {.name = "vgroup_id", .chkFunc = sysChkFilter__VgroupId, .fltFunc = sysFilte__VgroupId},
    {.name = "uid", .chkFunc = sysChkFilter__Uid, .fltFunc = sysFilte__Uid},
    {.name = "type", .chkFunc = sysChkFilter__Type, .fltFunc = sysFilte__Type}};
dengyihao's avatar
dengyihao 已提交
92

dengyihao's avatar
dengyihao 已提交
93
#define SYSTAB_FILTER_DICT_SIZE (sizeof(filterDict) / sizeof(filterDict[0]))
dengyihao's avatar
dengyihao 已提交
94

dengyihao's avatar
dengyihao 已提交
95
static int32_t optSysTabFilte(void* arg, SNode* cond, SArray* result);
dengyihao's avatar
dengyihao 已提交
96
static int32_t optSysTabFilteImpl(void* arg, SNode* cond, SArray* result);
dengyihao's avatar
dengyihao 已提交
97
static int32_t optSysCheckOper(SNode* pOpear);
dengyihao's avatar
dengyihao 已提交
98
static int32_t optSysMergeRslt(SArray* multiRslt, SArray* reslt);
dengyihao's avatar
dengyihao 已提交
99

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

102 103 104
static int32_t sysTableUserTagsFillOneTableTags(const SSysTableScanInfo* pInfo, SMetaReader* smrSuperTable,
                                                SMetaReader* smrChildTable, const char* dbname, const char* tableName,
                                                int32_t* pNumOfRows, const SSDataBlock* dataBlock);
105 106 107

static void relocateAndFilterSysTagsScanResult(SSysTableScanInfo* pInfo, int32_t numOfRows, SSDataBlock* dataBlock);
bool        processBlockWithProbability(const SSampleExecInfo* pInfo) {
108 109 110 111 112 113 114 115 116 117 118 119
#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
}

120
static void switchCtxOrder(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
H
Haojun Liao 已提交
121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148
  for (int32_t i = 0; i < numOfOutput; ++i) {
    SWITCH_ORDER(pCtx[i].order);
  }
}

static void setupQueryRangeForReverseScan(STableScanInfo* pTableScanInfo) {
#if 0
  int32_t numOfGroups = (int32_t)(GET_NUM_OF_TABLEGROUP(pRuntimeEnv));
  for(int32_t i = 0; i < numOfGroups; ++i) {
    SArray *group = GET_TABLEGROUP(pRuntimeEnv, i);
    SArray *tableKeyGroup = taosArrayGetP(pQueryAttr->tableGroupInfo.pGroupList, i);

    size_t t = taosArrayGetSize(group);
    for (int32_t j = 0; j < t; ++j) {
      STableQueryInfo *pCheckInfo = taosArrayGetP(group, j);
      updateTableQueryInfoForReverseScan(pCheckInfo);

      // update the last key in tableKeyInfo list, the tableKeyInfo is used to build the tsdbQueryHandle and decide
      // the start check timestamp of tsdbQueryHandle
//      STableKeyInfo *pTableKeyInfo = taosArrayGet(tableKeyGroup, j);
//      pTableKeyInfo->lastKey = pCheckInfo->lastKey;
//
//      assert(pCheckInfo->pTable == pTableKeyInfo->pTable);
    }
  }
#endif
}

149 150 151 152 153 154 155 156 157
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;
158
  // convert key to second
159 160 161 162 163 164 165
  key = convertTimePrecision(key, pInterval->precision, TSDB_TIME_PRECISION_MILLI) / 1000;

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

  struct tm tm;
166
  time_t    t = (time_t)key;
167 168 169 170 171
  taosLocalTime(&t, &tm);

  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 已提交
172
  tw->skey = convertTimePrecision((int64_t)taosMktime(&tm) * 1000LL, TSDB_TIME_PRECISION_MILLI, pInterval->precision);
173 174 175 176

  mon = (int)(mon + interval);
  tm.tm_year = mon / 12;
  tm.tm_mon = mon % 12;
wafwerar's avatar
wafwerar 已提交
177
  tw->ekey = convertTimePrecision((int64_t)taosMktime(&tm) * 1000LL, TSDB_TIME_PRECISION_MILLI, pInterval->precision);
178 179 180 181

  tw->ekey -= 1;
}

182
static bool overlapWithTimeWindow(SInterval* pInterval, SDataBlockInfo* pBlockInfo, int32_t order) {
183 184 185 186 187 188 189
  STimeWindow w = {0};

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

190
  if (order == TSDB_ORDER_ASC) {
191
    w = getAlignQueryTimeWindow(pInterval, pInterval->precision, pBlockInfo->window.skey);
192 193
    assert(w.ekey >= pBlockInfo->window.skey);

S
slzhou 已提交
194
    if (TMAX(w.skey, pBlockInfo->window.skey) <= TMIN(w.ekey, pBlockInfo->window.ekey)) {
195 196 197
      return true;
    }

198 199
    while (1) {
      getNextTimeWindow(pInterval, &w, order);
200 201 202 203 204
      if (w.skey > pBlockInfo->window.ekey) {
        break;
      }

      assert(w.ekey > pBlockInfo->window.ekey);
205
      if (TMAX(w.skey, pBlockInfo->window.skey) <= pBlockInfo->window.ekey) {
206 207 208 209
        return true;
      }
    }
  } else {
210
    w = getAlignQueryTimeWindow(pInterval, pInterval->precision, pBlockInfo->window.ekey);
211 212
    assert(w.skey <= pBlockInfo->window.ekey);

213
    if (TMAX(w.skey, pBlockInfo->window.skey) <= TMIN(w.ekey, pBlockInfo->window.ekey)) {
214 215 216
      return true;
    }

217
    while (1) {
218 219 220 221 222 223
      getNextTimeWindow(pInterval, &w, order);
      if (w.ekey < pBlockInfo->window.skey) {
        break;
      }

      assert(w.skey < pBlockInfo->window.skey);
224
      if (pBlockInfo->window.skey <= TMIN(w.ekey, pBlockInfo->window.ekey)) {
225 226 227
        return true;
      }
    }
228 229 230 231 232
  }

  return false;
}

233 234 235 236 237 238 239 240 241 242 243
// 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;

244 245
  SResultRowPosition* p1 = (SResultRowPosition*)tSimpleHashGet(pTableScanInfo->pdInfo.pAggSup->pResultRowHashTable, buf,
                                                               GET_RES_WINDOW_KEY_LEN(sizeof(groupId)));
246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293

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

  *pPage = getBufPage(pTableScanInfo->pdInfo.pAggSup->pResultBuf, p1->pageId);
  return (SResultRow*)((char*)(*pPage) + p1->offset);
}

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

  if (pTableScanInfo->pdInfo.pExprSup == NULL) {
    return TSDB_CODE_SUCCESS;
  }

  SExprSupp* pSup1 = pTableScanInfo->pdInfo.pExprSup;

  SFilePage*  pPage = NULL;
  SResultRow* pRow = getTableGroupOutputBuf(pOperator, pBlockInfo->groupId, &pPage);

  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;

    SResultRowEntryInfo* pEntry = getResultEntryInfo(pRow, i, pTableScanInfo->pdInfo.pExprSup->rowEntryInfoOffset);

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

  // release buffer pages
  releaseBufPage(pTableScanInfo->pdInfo.pAggSup->pResultBuf, pPage);

  if (notLoadBlock) {
    *status = FUNC_DATA_REQUIRED_NOT_LOAD;
  }

  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
294 295 296 297 298 299 300 301 302 303
static FORCE_INLINE bool doFilterByBlockSMA(const SNode* pFilterNode, SColumnDataAgg** pColsAgg, int32_t numOfCols,
                                            int32_t numOfRows) {
  if (pColsAgg == NULL || pFilterNode == NULL) {
    return true;
  }

  SFilterInfo* filter = NULL;

  // todo move to the initialization function
  int32_t code = filterInitFromNode((SNode*)pFilterNode, &filter, 0);
L
Liu Jicong 已提交
304
  bool    keep = filterRangeExecute(filter, pColsAgg, numOfCols, numOfRows);
H
Haojun Liao 已提交
305 306 307 308 309 310 311 312 313 314 315

  filterFreeInfo(filter);
  return keep;
}

static bool doLoadBlockSMA(STableScanInfo* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
  bool             allColumnsHaveAgg = true;
  SColumnDataAgg** pColAgg = NULL;

  int32_t code = tsdbRetrieveDatablockSMA(pTableScanInfo->dataReader, &pColAgg, &allColumnsHaveAgg);
  if (code != TSDB_CODE_SUCCESS) {
316
    T_LONG_JMP(pTaskInfo->env, code);
H
Haojun Liao 已提交
317 318 319 320 321 322 323 324 325 326 327 328 329
  }

  if (!allColumnsHaveAgg) {
    return false;
  }

  //  if (allColumnsHaveAgg == true) {
  int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);

  // todo create this buffer during creating operator
  if (pBlock->pBlockAgg == NULL) {
    pBlock->pBlockAgg = taosMemoryCalloc(numOfCols, POINTER_BYTES);
    if (pBlock->pBlockAgg == NULL) {
330
      T_LONG_JMP(pTaskInfo->env, TSDB_CODE_OUT_OF_MEMORY);
H
Haojun Liao 已提交
331 332 333
    }
  }

334 335
  size_t num = taosArrayGetSize(pTableScanInfo->matchInfo.pList);
  for (int32_t i = 0; i < num; ++i) {
H
Haojun Liao 已提交
336 337
    SColMatchItem* pColMatchInfo = taosArrayGet(pTableScanInfo->matchInfo.pList, i);
    if (!pColMatchInfo->needOutput) {
H
Haojun Liao 已提交
338 339
      continue;
    }
H
Haojun Liao 已提交
340 341

    pBlock->pBlockAgg[pColMatchInfo->dstSlotId] = pColAgg[i];
H
Haojun Liao 已提交
342 343 344 345 346
  }

  return true;
}

347 348
static void doSetTagColumnData(STableScanInfo* pTableScanInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo,
                               int32_t rows) {
H
Haojun Liao 已提交
349 350 351
  if (pTableScanInfo->pseudoSup.numOfExprs > 0) {
    SExprSupp* pSup = &pTableScanInfo->pseudoSup;

352
    int32_t code = addTagPseudoColumnData(&pTableScanInfo->readHandle, pSup->pExprInfo, pSup->numOfExprs, pBlock, rows,
H
Haojun Liao 已提交
353 354 355 356 357 358 359
                                          GET_TASKID(pTaskInfo));
    if (code != TSDB_CODE_SUCCESS) {
      T_LONG_JMP(pTaskInfo->env, code);
    }
  }
}

360 361
// todo handle the slimit info
void applyLimitOffset(SLimitInfo* pLimitInfo, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo, SOperatorInfo* pOperator) {
362 363 364 365 366 367
  SLimit* pLimit = &pLimitInfo->limit;

  if (pLimit->offset > 0 && pLimitInfo->remainOffset > 0) {
    if (pLimitInfo->remainOffset >= pBlock->info.rows) {
      pLimitInfo->remainOffset -= pBlock->info.rows;
      pBlock->info.rows = 0;
dengyihao's avatar
opt mem  
dengyihao 已提交
368 369
      qDebug("current block ignore due to offset, current:%" PRId64 ", %s", pLimitInfo->remainOffset,
             GET_TASKID(pTaskInfo));
370 371 372 373 374 375 376 377 378 379 380 381
    } else {
      blockDataTrimFirstNRows(pBlock, pLimitInfo->remainOffset);
      pLimitInfo->remainOffset = 0;
    }
  }

  if (pLimit->limit != -1 && pLimit->limit <= (pLimitInfo->numOfOutputRows + pBlock->info.rows)) {
    // limit the output rows
    int32_t overflowRows = pLimitInfo->numOfOutputRows + pBlock->info.rows - pLimit->limit;
    int32_t keep = pBlock->info.rows - overflowRows;

    blockDataKeepFirstNRows(pBlock, keep);
382
    qDebug("output limit %" PRId64 " has reached, %s", pLimit->limit, GET_TASKID(pTaskInfo));
383 384 385 386
    pOperator->status = OP_EXEC_DONE;
  }
}

387 388 389 390 391 392 393 394 395 396 397
static void ensureBlockCapacity(SSDataBlock* pBlock, int32_t capacity) {
  // keep the value of rows temporarily
  int32_t rows = pBlock->info.rows;

  pBlock->info.rows = 0;
  blockDataEnsureCapacity(pBlock, capacity);

  // restore the rows number
  pBlock->info.rows = rows;
}

L
Liu Jicong 已提交
398 399
static int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock,
                             uint32_t* status) {
400
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
401 402
  STableScanInfo* pInfo = pOperator->info;

403
  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;
H
Haojun Liao 已提交
404 405

  pCost->totalBlocks += 1;
406
  pCost->totalRows += pBlock->info.rows;
407

H
Haojun Liao 已提交
408
  bool loadSMA = false;
H
Haojun Liao 已提交
409

410
  *status = pInfo->dataBlockLoadFlag;
411
  if (pTableScanInfo->pFilterNode != NULL ||
412
      overlapWithTimeWindow(&pTableScanInfo->pdInfo.interval, &pBlock->info, pTableScanInfo->cond.order)) {
413 414 415 416
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
  }

  SDataBlockInfo* pBlockInfo = &pBlock->info;
417
  taosMemoryFreeClear(pBlock->pBlockAgg);
418 419

  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
420 421
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
422
    pCost->filterOutBlocks += 1;
423
    pCost->totalRows += pBlock->info.rows;
424 425
    return TSDB_CODE_SUCCESS;
  } else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
426 427
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
H
Haojun Liao 已提交
428

429 430 431
    if (pTableScanInfo->pseudoSup.numOfExprs > 0) {
      ensureBlockCapacity(pBlock, pBlock->info.rows);
    }
432
    doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, 1);
433 434 435 436
    pCost->skipBlocks += 1;
    return TSDB_CODE_SUCCESS;
  } else if (*status == FUNC_DATA_REQUIRED_STATIS_LOAD) {
    pCost->loadBlockStatis += 1;
L
Liu Jicong 已提交
437
    loadSMA = true;  // mark the operation of load sma;
H
Haojun Liao 已提交
438
    bool success = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo);
L
Liu Jicong 已提交
439
    if (success) {  // failed to load the block sma data, data block statistics does not exist, load data block instead
440 441
      qDebug("%s data block SMA loaded, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
             pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
442 443 444 445
      if (pTableScanInfo->pseudoSup.numOfExprs > 0) {
        ensureBlockCapacity(pBlock, pBlock->info.rows);
      }

446
      doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, 1);
447 448
      return TSDB_CODE_SUCCESS;
    } else {
449
      qDebug("%s failed to load SMA, since not all columns have SMA", GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
450
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
451
    }
H
Haojun Liao 已提交
452
  }
453

H
Haojun Liao 已提交
454
  ASSERT(*status == FUNC_DATA_REQUIRED_DATA_LOAD);
455

H
Haojun Liao 已提交
456
  // try to filter data block according to sma info
457 458 459 460 461 462 463 464 465 466 467 468 469
  if (pTableScanInfo->pFilterNode != NULL && (!loadSMA)) {
    bool success = doLoadBlockSMA(pTableScanInfo, pBlock, pTaskInfo);
    if (success) {
      size_t size = taosArrayGetSize(pBlock->pDataBlock);
      bool   keep = doFilterByBlockSMA(pTableScanInfo->pFilterNode, pBlock->pBlockAgg, size, pBlockInfo->rows);
      if (!keep) {
        qDebug("%s data block filter out by block SMA, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
               pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
        pCost->filterOutBlocks += 1;
        (*status) = FUNC_DATA_REQUIRED_FILTEROUT;

        return TSDB_CODE_SUCCESS;
      }
470
    }
H
Haojun Liao 已提交
471
  }
472

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

476
  // try to filter data block according to current results
477 478
  doDynamicPruneDataBlock(pOperator, pBlockInfo, status);
  if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
479
    qDebug("%s data block skipped due to dynamic prune, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
480 481 482
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
    pCost->skipBlocks += 1;

483
    *status = FUNC_DATA_REQUIRED_FILTEROUT;
484 485 486
    return TSDB_CODE_SUCCESS;
  }

H
Haojun Liao 已提交
487 488
  pCost->totalCheckedRows += pBlock->info.rows;
  pCost->loadBlocks += 1;
489

H
Haojun Liao 已提交
490 491 492
  SArray* pCols = tsdbRetrieveDataBlock(pTableScanInfo->dataReader, NULL);
  if (pCols == NULL) {
    return terrno;
H
Haojun Liao 已提交
493 494
  }

495
  ensureBlockCapacity(pBlock, pBlock->info.rows);
H
Haojun Liao 已提交
496
  relocateColumnData(pBlock, pTableScanInfo->matchInfo.pList, pCols, true);
497
  doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, pBlock->info.rows);
498

H
Haojun Liao 已提交
499 500
  // restore the previous value
  pCost->totalRows -= pBlock->info.rows;
501

502 503
  if (pTableScanInfo->pFilterNode != NULL) {
    int64_t st = taosGetTimestampUs();
H
Haojun Liao 已提交
504
    doFilter(pTableScanInfo->pFilterNode, pBlock, &pTableScanInfo->matchInfo, pOperator->exprSupp.pFilterInfo);
505

506 507
    double el = (taosGetTimestampUs() - st) / 1000.0;
    pTableScanInfo->readRecorder.filterTime += el;
508

509 510 511 512 513 514 515
    if (pBlock->info.rows == 0) {
      pCost->filterOutBlocks += 1;
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d, elapsed time:%.2f ms",
             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);
    }
516 517
  }

518
  applyLimitOffset(&pInfo->limitInfo, pBlock, pTaskInfo, pOperator);
519

H
Haojun Liao 已提交
520 521
  pCost->totalRows += pBlock->info.rows;
  pInfo->limitInfo.numOfOutputRows = pCost->totalRows;
H
Haojun Liao 已提交
522 523 524
  return TSDB_CODE_SUCCESS;
}

525
static void prepareForDescendingScan(STableScanInfo* pTableScanInfo, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
H
Haojun Liao 已提交
526 527 528
  SET_REVERSE_SCAN_FLAG(pTableScanInfo);

  switchCtxOrder(pCtx, numOfOutput);
529
  //  setupQueryRangeForReverseScan(pTableScanInfo);
H
Haojun Liao 已提交
530

531
  pTableScanInfo->cond.order = TSDB_ORDER_DESC;
H
Haojun Liao 已提交
532 533
  STimeWindow* pTWindow = &pTableScanInfo->cond.twindows;
  TSWAP(pTWindow->skey, pTWindow->ekey);
H
Haojun Liao 已提交
534 535
}

H
Haojun Liao 已提交
536
int32_t addTagPseudoColumnData(SReadHandle* pHandle, SExprInfo* pPseudoExpr, int32_t numOfPseudoExpr,
537
                               SSDataBlock* pBlock, int32_t rows, const char* idStr) {
538
  // currently only the tbname pseudo column
539
  if (numOfPseudoExpr == 0) {
H
Haojun Liao 已提交
540
    return TSDB_CODE_SUCCESS;
541 542
  }

543 544 545 546
  // backup the rows
  int32_t backupRows = pBlock->info.rows;
  pBlock->info.rows = rows;

547
  SMetaReader mr = {0};
548
  metaReaderInit(&mr, pHandle->meta, 0);
H
Haojun Liao 已提交
549
  int32_t code = metaGetTableEntryByUid(&mr, pBlock->info.uid);
550 551
  metaReaderReleaseLock(&mr);

H
Haojun Liao 已提交
552
  if (code != TSDB_CODE_SUCCESS) {
L
Liu Jicong 已提交
553
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", pBlock->info.uid, tstrerror(terrno), idStr);
H
Haojun Liao 已提交
554 555 556
    metaReaderClear(&mr);
    return terrno;
  }
557

558 559
  for (int32_t j = 0; j < numOfPseudoExpr; ++j) {
    SExprInfo* pExpr = &pPseudoExpr[j];
560
    int32_t    dstSlotId = pExpr->base.resSchema.slotId;
561 562

    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, dstSlotId);
D
dapan1121 已提交
563
    colInfoDataCleanup(pColInfoData, pBlock->info.rows);
564 565 566 567 568

    int32_t functionId = pExpr->pExpr->_function.functionId;

    // this is to handle the tbname
    if (fmIsScanPseudoColumnFunc(functionId)) {
569
      setTbNameColData(pHandle->meta, pBlock, pColInfoData, functionId);
570
    } else {  // these are tags
wmmhello's avatar
wmmhello 已提交
571 572
      STagVal tagVal = {0};
      tagVal.cid = pExpr->base.pParam[0].pCol->colId;
573
      const char* p = metaGetTableTagVal(mr.me.ctbEntry.pTags, pColInfoData->info.type, &tagVal);
wmmhello's avatar
wmmhello 已提交
574

575 576 577 578
      char* data = NULL;
      if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
        data = tTagValToData((const STagVal*)p, false);
      } else {
wmmhello's avatar
wmmhello 已提交
579
        data = (char*)p;
wmmhello's avatar
wmmhello 已提交
580
      }
581

H
Haojun Liao 已提交
582 583 584
      bool isNullVal = (data == NULL) || (pColInfoData->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data));
      if (isNullVal) {
        colDataAppendNNULL(pColInfoData, 0, pBlock->info.rows);
H
Haojun Liao 已提交
585
      } else if (pColInfoData->info.type != TSDB_DATA_TYPE_JSON) {
H
Haojun Liao 已提交
586
        colDataAppendNItems(pColInfoData, 0, data, pBlock->info.rows);
H
Haojun Liao 已提交
587 588 589
        if (IS_VAR_DATA_TYPE(((const STagVal*)p)->type)) {
          taosMemoryFree(data);
        }
L
Liu Jicong 已提交
590
      } else {  // todo opt for json tag
H
Haojun Liao 已提交
591
        for (int32_t i = 0; i < pBlock->info.rows; ++i) {
H
Haojun Liao 已提交
592
          colDataAppend(pColInfoData, i, data, false);
H
Haojun Liao 已提交
593
        }
594 595 596 597 598
      }
    }
  }

  metaReaderClear(&mr);
599 600 601

  // restore the rows
  pBlock->info.rows = backupRows;
H
Haojun Liao 已提交
602
  return TSDB_CODE_SUCCESS;
603 604
}

605 606 607 608
void setTbNameColData(void* pMeta, const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId) {
  struct SScalarFuncExecFuncs fpSet = {0};
  fmGetScalarFuncExecFuncs(functionId, &fpSet);

609 610
  SColumnInfoData infoData = createColumnInfoData(TSDB_DATA_TYPE_BIGINT, sizeof(uint64_t), 1);
  colInfoDataEnsureCapacity(&infoData, 1);
611

612
  colDataAppendInt64(&infoData, 0, (int64_t*)&pBlock->info.uid);
613
  SScalarParam srcParam = {.numOfRows = pBlock->info.rows, .param = pMeta, .columnData = &infoData};
614 615

  SScalarParam param = {.columnData = pColInfoData};
H
Haojun Liao 已提交
616 617 618 619 620 621 622

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

D
dapan1121 已提交
623
  colDataDestroy(&infoData);
624 625
}

626
static SSDataBlock* doTableScanImpl(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
627
  STableScanInfo* pTableScanInfo = pOperator->info;
628
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
L
Liu Jicong 已提交
629
  SSDataBlock*    pBlock = pTableScanInfo->pResBlock;
H
Haojun Liao 已提交
630

631 632
  int64_t st = taosGetTimestampUs();

633
  while (tsdbNextDataBlock(pTableScanInfo->dataReader)) {
634
    if (isTaskKilled(pTaskInfo)) {
635
      T_LONG_JMP(pTaskInfo->env, TSDB_CODE_TSC_QUERY_CANCELLED);
636
    }
H
Haojun Liao 已提交
637

638 639 640 641 642 643
    // process this data block based on the probabilities
    bool processThisBlock = processBlockWithProbability(&pTableScanInfo->sample);
    if (!processThisBlock) {
      continue;
    }

644 645
    blockDataCleanup(pBlock);

H
Haojun Liao 已提交
646
    SDataBlockInfo* pBInfo = &pBlock->info;
H
Haojun Liao 已提交
647
    tsdbRetrieveDataBlockInfo(pTableScanInfo->dataReader, &pBInfo->rows, &pBInfo->uid, &pBInfo->window);
648

H
Haojun Liao 已提交
649
    ASSERT(pBInfo->uid != 0);
H
Haojun Liao 已提交
650
    pBlock->info.groupId = getTableGroupId(pTaskInfo->pTableInfoList, pBlock->info.uid);
651

652 653 654 655
    uint32_t status = 0;
    int32_t  code = loadDataBlock(pOperator, pTableScanInfo, pBlock, &status);
    //    int32_t  code = loadDataBlockOnDemand(pOperator->pRuntimeEnv, pTableScanInfo, pBlock, &status);
    if (code != TSDB_CODE_SUCCESS) {
656
      T_LONG_JMP(pOperator->pTaskInfo->env, code);
657
    }
658

659 660 661
    // 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;
662
    }
663 664 665 666 667

    pOperator->resultInfo.totalRows = pTableScanInfo->readRecorder.totalRows;
    pTableScanInfo->readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;

    pOperator->cost.totalCost = pTableScanInfo->readRecorder.elapsedTime;
668 669

    // todo refactor
L
Liu Jicong 已提交
670 671 672 673 674
    /*pTableScanInfo->lastStatus.uid = pBlock->info.uid;*/
    /*pTableScanInfo->lastStatus.ts = pBlock->info.window.ekey;*/
    pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__SNAPSHOT_DATA;
    pTaskInfo->streamInfo.lastStatus.uid = pBlock->info.uid;
    pTaskInfo->streamInfo.lastStatus.ts = pBlock->info.window.ekey;
675

L
Liu Jicong 已提交
676
    ASSERT(pBlock->info.uid != 0);
677
    return pBlock;
H
Haojun Liao 已提交
678 679 680 681
  }
  return NULL;
}

wmmhello's avatar
wmmhello 已提交
682
static SSDataBlock* doTableScanGroup(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
683 684 685 686
  STableScanInfo* pTableScanInfo = pOperator->info;
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;

  // The read handle is not initialized yet, since no qualified tables exists
687
  if (pTableScanInfo->dataReader == NULL || pOperator->status == OP_EXEC_DONE) {
H
Haojun Liao 已提交
688 689 690
    return NULL;
  }

691 692
  // do the ascending order traverse in the first place.
  while (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
H
Haojun Liao 已提交
693 694 695 696
    SSDataBlock* p = doTableScanImpl(pOperator);
    if (p != NULL) {
      ASSERT(p->info.uid != 0);
      return p;
H
Haojun Liao 已提交
697 698
    }

699
    pTableScanInfo->scanTimes += 1;
700

701
    if (pTableScanInfo->scanTimes < pTableScanInfo->scanInfo.numOfAsc) {
702 703
      setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
      pTableScanInfo->scanFlag = REPEAT_SCAN;
704
      qDebug("start to repeat ascending order scan data blocks due to query func required, %s", GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
705

706
      // do prepare for the next round table scan operation
H
Haojun Liao 已提交
707
      tsdbReaderReset(pTableScanInfo->dataReader, &pTableScanInfo->cond);
H
Haojun Liao 已提交
708
    }
709
  }
H
Haojun Liao 已提交
710

711
  int32_t total = pTableScanInfo->scanInfo.numOfAsc + pTableScanInfo->scanInfo.numOfDesc;
712
  if (pTableScanInfo->scanTimes < total) {
713
    if (pTableScanInfo->cond.order == TSDB_ORDER_ASC) {
H
Haojun Liao 已提交
714 715
      prepareForDescendingScan(pTableScanInfo, pOperator->exprSupp.pCtx, 0);
      tsdbReaderReset(pTableScanInfo->dataReader, &pTableScanInfo->cond);
716
      qDebug("%s start to descending order scan data blocks due to query func required", GET_TASKID(pTaskInfo));
717
    }
H
Haojun Liao 已提交
718

719
    while (pTableScanInfo->scanTimes < total) {
H
Haojun Liao 已提交
720 721 722
      SSDataBlock* p = doTableScanImpl(pOperator);
      if (p != NULL) {
        return p;
723
      }
H
Haojun Liao 已提交
724

725
      pTableScanInfo->scanTimes += 1;
H
Haojun Liao 已提交
726

727
      if (pTableScanInfo->scanTimes < total) {
728 729
        setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
        pTableScanInfo->scanFlag = REPEAT_SCAN;
H
Haojun Liao 已提交
730

731
        qDebug("%s start to repeat descending order scan data blocks", GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
732
        tsdbReaderReset(pTableScanInfo->dataReader, &pTableScanInfo->cond);
733
      }
H
Haojun Liao 已提交
734 735 736
    }
  }

wmmhello's avatar
wmmhello 已提交
737 738 739 740 741 742 743
  return NULL;
}

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

744
  // scan table one by one sequentially
L
Liu Jicong 已提交
745
  if (pInfo->scanMode == TABLE_SCAN__TABLE_ORDER) {
H
Haojun Liao 已提交
746
    int32_t numOfTables = tableListGetSize(pTaskInfo->pTableInfoList);
H
Haojun Liao 已提交
747

L
Liu Jicong 已提交
748
    while (1) {
L
Liu Jicong 已提交
749 750 751 752
      SSDataBlock* result = doTableScanGroup(pOperator);
      if (result) {
        return result;
      }
H
Haojun Liao 已提交
753

L
Liu Jicong 已提交
754 755
      // if no data, switch to next table and continue scan
      pInfo->currentTable++;
H
Haojun Liao 已提交
756
      if (pInfo->currentTable >= numOfTables) {
L
Liu Jicong 已提交
757 758
        return NULL;
      }
H
Haojun Liao 已提交
759

H
Haojun Liao 已提交
760
      STableKeyInfo* pTableInfo = tableListGetInfo(pTaskInfo->pTableInfoList, pInfo->currentTable);
761
      tsdbSetTableList(pInfo->dataReader, pTableInfo, 1);
L
Liu Jicong 已提交
762 763
      qDebug("set uid:%" PRIu64 " into scanner, total tables:%d, index:%d %s", pTableInfo->uid, numOfTables,
             pInfo->currentTable, pTaskInfo->id.str);
H
Haojun Liao 已提交
764

H
Haojun Liao 已提交
765
      tsdbReaderReset(pInfo->dataReader, &pInfo->cond);
L
Liu Jicong 已提交
766 767
      pInfo->scanTimes = 0;
    }
768 769
  } else {  // scan table group by group sequentially
    if (pInfo->currentGroupId == -1) {
H
Haojun Liao 已提交
770
      if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pTaskInfo->pTableInfoList)) {
771 772 773
        doSetOperatorCompleted(pOperator);
        return NULL;
      }
774

5
54liuyao 已提交
775
      int32_t        num = 0;
776
      STableKeyInfo* pList = NULL;
H
Haojun Liao 已提交
777
      tableListGetGroupList(pTaskInfo->pTableInfoList, pInfo->currentGroupId, &pList, &num);
778 779
      ASSERT(pInfo->dataReader == NULL);

5
54liuyao 已提交
780 781
      int32_t code = tsdbReaderOpen(pInfo->readHandle.vnode, &pInfo->cond, pList, num,
                                    (STsdbReader**)&pInfo->dataReader, GET_TASKID(pTaskInfo));
782 783 784
      if (code != TSDB_CODE_SUCCESS) {
        T_LONG_JMP(pTaskInfo->env, code);
      }
wmmhello's avatar
wmmhello 已提交
785
    }
H
Haojun Liao 已提交
786

787 788
    SSDataBlock* result = doTableScanGroup(pOperator);
    if (result != NULL) {
H
Haojun Liao 已提交
789
      ASSERT(result->info.uid != 0);
790 791
      return result;
    }
H
Haojun Liao 已提交
792

H
Haojun Liao 已提交
793
    if ((++pInfo->currentGroupId) >= tableListGetOutputGroups(pTaskInfo->pTableInfoList)) {
794
      doSetOperatorCompleted(pOperator);
795 796
      return NULL;
    }
wmmhello's avatar
wmmhello 已提交
797

798 799 800 801
    // reset value for the next group data output
    pOperator->status = OP_OPENED;
    pInfo->limitInfo.numOfOutputRows = 0;
    pInfo->limitInfo.remainOffset = pInfo->limitInfo.limit.offset;
wmmhello's avatar
wmmhello 已提交
802

5
54liuyao 已提交
803
    int32_t        num = 0;
804
    STableKeyInfo* pList = NULL;
H
Haojun Liao 已提交
805
    tableListGetGroupList(pTaskInfo->pTableInfoList, pInfo->currentGroupId, &pList, &num);
wmmhello's avatar
wmmhello 已提交
806

807 808 809
    tsdbSetTableList(pInfo->dataReader, pList, num);
    tsdbReaderReset(pInfo->dataReader, &pInfo->cond);
    pInfo->scanTimes = 0;
wmmhello's avatar
wmmhello 已提交
810

811 812 813 814
    result = doTableScanGroup(pOperator);
    if (result != NULL) {
      return result;
    }
815

816 817 818
    doSetOperatorCompleted(pOperator);
    return NULL;
  }
H
Haojun Liao 已提交
819 820
}

821 822
static int32_t getTableScannerExecInfo(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
  SFileBlockLoadRecorder* pRecorder = taosMemoryCalloc(1, sizeof(SFileBlockLoadRecorder));
823
  STableScanInfo*         pTableScanInfo = pOptr->info;
824 825 826 827 828 829
  *pRecorder = pTableScanInfo->readRecorder;
  *pOptrExplain = pRecorder;
  *len = sizeof(SFileBlockLoadRecorder);
  return 0;
}

830
static void destroyTableScanOperatorInfo(void* param) {
831
  STableScanInfo* pTableScanInfo = (STableScanInfo*)param;
H
Haojun Liao 已提交
832
  blockDataDestroy(pTableScanInfo->pResBlock);
833
  cleanupQueryTableDataCond(&pTableScanInfo->cond);
H
Haojun Liao 已提交
834

H
refact  
Hongze Cheng 已提交
835
  tsdbReaderClose(pTableScanInfo->dataReader);
836
  pTableScanInfo->dataReader = NULL;
837

H
Haojun Liao 已提交
838 839
  if (pTableScanInfo->matchInfo.pList != NULL) {
    taosArrayDestroy(pTableScanInfo->matchInfo.pList);
840
  }
L
Liu Jicong 已提交
841

842
  cleanupExprSupp(&pTableScanInfo->pseudoSup);
D
dapan1121 已提交
843
  taosMemoryFreeClear(param);
844 845
}

846
SOperatorInfo* createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
847
                                           SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
848 849 850
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
851
    goto _error;
H
Haojun Liao 已提交
852 853
  }

854
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;
855 856

  int32_t numOfCols = 0;
H
Haojun Liao 已提交
857 858
  int32_t code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
                                     &pInfo->matchInfo);
859 860 861 862 863
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
H
Haojun Liao 已提交
864
  code = initQueryTableDataCond(&pInfo->cond, pTableScanNode);
865
  if (code != TSDB_CODE_SUCCESS) {
866
    goto _error;
867 868 869
  }

  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
870
    SExprSupp* pSup = &pInfo->pseudoSup;
871 872
    pSup->pExprInfo = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->numOfExprs);
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset);
873 874
  }

875
  pInfo->scanInfo = (SScanInfo){.numOfAsc = pTableScanNode->scanSeq[0], .numOfDesc = pTableScanNode->scanSeq[1]};
876
  pInfo->pdInfo.interval = extractIntervalInfo(pTableScanNode);
877 878 879
  pInfo->readHandle = *readHandle;
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
  pInfo->sample.seed = taosGetTimestampSec();
880

881
  pInfo->dataBlockLoadFlag = pTableScanNode->dataRequired;
882 883
  pInfo->pResBlock = createResDataBlock(pDescNode);
  pInfo->pFilterNode = pTableScanNode->scan.node.pConditions;
H
Haojun Liao 已提交
884 885 886

  if (pInfo->pFilterNode != NULL) {
    code = filterInitFromNode((SNode*)pInfo->pFilterNode, &pOperator->exprSupp.pFilterInfo, 0);
887
    if (code != TSDB_CODE_SUCCESS) {
888 889
      goto _error;
    }
H
Haojun Liao 已提交
890 891
  }

892
  pInfo->scanFlag = MAIN_SCAN;
wmmhello's avatar
wmmhello 已提交
893
  pInfo->currentGroupId = -1;
894
  pInfo->assignBlockUid = pTableScanNode->assignBlockUid;
895 896

  pOperator->name = "TableScanOperator";  // for debug purpose
L
Liu Jicong 已提交
897
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN;
898 899 900
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
901
  pOperator->exprSupp.numOfExprs = numOfCols;
902
  pOperator->pTaskInfo = pTaskInfo;
903

904
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doTableScan, NULL, NULL, destroyTableScanOperatorInfo,
905
                                         getTableScannerExecInfo);
906 907 908

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

911
_error:
912 913 914
  if (pInfo != NULL) {
    destroyTableScanOperatorInfo(pInfo);
  }
915

916 917
  taosMemoryFreeClear(pOperator);
  pTaskInfo->code = code;
918
  return NULL;
H
Haojun Liao 已提交
919 920
}

921
SOperatorInfo* createTableSeqScanOperatorInfo(void* pReadHandle, SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
922
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
L
Liu Jicong 已提交
923
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
924

L
Liu Jicong 已提交
925 926
  pInfo->dataReader = pReadHandle;
  //  pInfo->prevGroupId       = -1;
H
Haojun Liao 已提交
927

928
  pOperator->name = "TableSeqScanOperator";
H
Haojun Liao 已提交
929
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN;
930 931 932 933
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
934

935
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doTableScanImpl, NULL, NULL, NULL, NULL);
H
Haojun Liao 已提交
936 937 938
  return pOperator;
}

H
Haojun Liao 已提交
939 940
static int32_t doGetTableRowSize(void* pMeta, uint64_t uid, int32_t* rowLen, const char* idstr) {
  *rowLen = 0;
H
Haojun Liao 已提交
941

942
  SMetaReader mr = {0};
943
  metaReaderInit(&mr, pMeta, 0);
H
Haojun Liao 已提交
944 945
  int32_t code = metaGetTableEntryByUid(&mr, uid);
  if (code != TSDB_CODE_SUCCESS) {
L
Liu Jicong 已提交
946
    qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", uid, tstrerror(terrno), idstr);
H
Haojun Liao 已提交
947 948 949 950
    metaReaderClear(&mr);
    return terrno;
  }

951 952
  if (mr.me.type == TSDB_SUPER_TABLE) {
    int32_t numOfCols = mr.me.stbEntry.schemaRow.nCols;
953
    for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
954
      (*rowLen) += mr.me.stbEntry.schemaRow.pSchema[i].bytes;
955 956 957
    }
  } else if (mr.me.type == TSDB_CHILD_TABLE) {
    uint64_t suid = mr.me.ctbEntry.suid;
958
    tDecoderClear(&mr.coder);
H
Haojun Liao 已提交
959 960
    code = metaGetTableEntryByUid(&mr, suid);
    if (code != TSDB_CODE_SUCCESS) {
L
Liu Jicong 已提交
961
      qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", suid, tstrerror(terrno), idstr);
H
Haojun Liao 已提交
962 963 964 965
      metaReaderClear(&mr);
      return terrno;
    }

966 967
    int32_t numOfCols = mr.me.stbEntry.schemaRow.nCols;

968
    for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
969
      (*rowLen) += mr.me.stbEntry.schemaRow.pSchema[i].bytes;
970 971 972
    }
  } else if (mr.me.type == TSDB_NORMAL_TABLE) {
    int32_t numOfCols = mr.me.ntbEntry.schemaRow.nCols;
973
    for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
974
      (*rowLen) += mr.me.ntbEntry.schemaRow.pSchema[i].bytes;
975 976 977 978
    }
  }

  metaReaderClear(&mr);
H
Haojun Liao 已提交
979
  return TSDB_CODE_SUCCESS;
980 981 982 983 984 985 986 987
}

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

  SBlockDistInfo* pBlockScanInfo = pOperator->info;
L
Liu Jicong 已提交
988
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
989 990

  STableBlockDistInfo blockDistInfo = {.minRows = INT_MAX, .maxRows = INT_MIN};
L
Liu Jicong 已提交
991 992
  int32_t code = doGetTableRowSize(pBlockScanInfo->readHandle.meta, pBlockScanInfo->uid, &blockDistInfo.rowSize,
                                   GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
993
  if (code != TSDB_CODE_SUCCESS) {
994
    T_LONG_JMP(pTaskInfo->env, code);
H
Haojun Liao 已提交
995
  }
996 997 998

  tsdbGetFileBlocksDistInfo(pBlockScanInfo->pHandle, &blockDistInfo);
  blockDistInfo.numOfInmemRows = (int32_t)tsdbGetNumOfRowsInMemTable(pBlockScanInfo->pHandle);
H
Haojun Liao 已提交
999

1000
  SSDataBlock* pBlock = pBlockScanInfo->pResBlock;
H
Haojun Liao 已提交
1001

1002
  int32_t          slotId = pOperator->exprSupp.pExprInfo->base.resSchema.slotId;
1003
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, slotId);
H
Haojun Liao 已提交
1004

1005
  int32_t len = tSerializeBlockDistInfo(NULL, 0, &blockDistInfo);
1006
  char*   p = taosMemoryCalloc(1, len + VARSTR_HEADER_SIZE);
1007 1008 1009
  tSerializeBlockDistInfo(varDataVal(p), len, &blockDistInfo);
  varDataSetLen(p, len);

1010
  blockDataEnsureCapacity(pBlock, 1);
1011 1012
  colDataAppend(pColInfo, 0, p, false);
  taosMemoryFree(p);
H
Haojun Liao 已提交
1013

1014 1015
  pBlock->info.rows = 1;

H
Haojun Liao 已提交
1016 1017 1018 1019
  pOperator->status = OP_EXEC_DONE;
  return pBlock;
}

1020
static void destroyBlockDistScanOperatorInfo(void* param) {
1021
  SBlockDistInfo* pDistInfo = (SBlockDistInfo*)param;
1022
  blockDataDestroy(pDistInfo->pResBlock);
H
Hongze Cheng 已提交
1023
  tsdbReaderClose(pDistInfo->pHandle);
D
dapan1121 已提交
1024
  taosMemoryFreeClear(param);
1025 1026
}

1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050
static int32_t initTableblockDistQueryCond(uint64_t uid, SQueryTableDataCond* pCond) {
  memset(pCond, 0, sizeof(SQueryTableDataCond));

  pCond->order = TSDB_ORDER_ASC;
  pCond->numOfCols = 1;
  pCond->colList = taosMemoryCalloc(1, sizeof(SColumnInfo));
  if (pCond->colList == NULL) {
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return terrno;
  }

  pCond->colList->colId = 1;
  pCond->colList->type = TSDB_DATA_TYPE_TIMESTAMP;
  pCond->colList->bytes = sizeof(TSKEY);

  pCond->twindows = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
  pCond->suid = uid;
  pCond->type = TIMEWINDOW_RANGE_CONTAINED;
  pCond->startVersion = -1;
  pCond->endVersion = -1;

  return TSDB_CODE_SUCCESS;
}

5
54liuyao 已提交
1051 1052
SOperatorInfo* createDataBlockInfoScanOperator(SReadHandle* readHandle, SBlockDistScanPhysiNode* pBlockScanNode,
                                               SExecTaskInfo* pTaskInfo) {
1053
  SBlockDistInfo* pInfo = taosMemoryCalloc(1, sizeof(SBlockDistInfo));
1054
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
1055 1056 1057 1058 1059
  if (pInfo == NULL || pOperator == NULL) {
    pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
    goto _error;
  }

1060 1061 1062 1063 1064
  {
    SQueryTableDataCond cond = {0};

    int32_t code = initTableblockDistQueryCond(pBlockScanNode->suid, &cond);
    if (code != TSDB_CODE_SUCCESS) {
dengyihao's avatar
merge  
dengyihao 已提交
1065
      goto _error;
1066 1067 1068
    }

    STableListInfo* pTableListInfo = pTaskInfo->pTableInfoList;
5
54liuyao 已提交
1069 1070
    size_t          num = tableListGetSize(pTableListInfo);
    void*           pList = tableListGetInfo(pTableListInfo, 0);
1071

dengyihao's avatar
dengyihao 已提交
1072
    code = tsdbReaderOpen(readHandle->vnode, &cond, pList, num, &pInfo->pHandle, pTaskInfo->id.str);
1073
    cleanupQueryTableDataCond(&cond);
dengyihao's avatar
dengyihao 已提交
1074 1075 1076
    if (code != 0) {
      goto _error;
    }
1077 1078
  }

1079
  pInfo->readHandle = *readHandle;
1080
  pInfo->uid = pBlockScanNode->suid;
1081
  pInfo->pResBlock = createResDataBlock(pBlockScanNode->node.pOutputDataBlockDesc);
1082

1083
  int32_t    numOfCols = 0;
1084
  SExprInfo* pExprInfo = createExprInfo(pBlockScanNode->pScanPseudoCols, NULL, &numOfCols);
1085
  int32_t    code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfCols);
1086 1087 1088
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
1089

1090
  pOperator->name = "DataBlockDistScanOperator";
1091
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_BLOCK_DIST_SCAN;
1092 1093 1094
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
1095 1096
  pOperator->pTaskInfo = pTaskInfo;

1097 1098
  pOperator->fpSet =
      createOperatorFpSet(operatorDummyOpenFn, doBlockInfoScan, NULL, NULL, destroyBlockDistScanOperatorInfo, NULL);
H
Haojun Liao 已提交
1099 1100
  return pOperator;

1101
_error:
H
Haojun Liao 已提交
1102 1103 1104 1105 1106
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
  return NULL;
}

1107
static FORCE_INLINE void doClearBufferedBlocks(SStreamScanInfo* pInfo) {
L
Liu Jicong 已提交
1108 1109
  taosArrayClear(pInfo->pBlockLists);
  pInfo->validBlockIndex = 0;
H
Haojun Liao 已提交
1110 1111
}

1112
static bool isSessionWindow(SStreamScanInfo* pInfo) {
H
Haojun Liao 已提交
1113
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION;
5
54liuyao 已提交
1114 1115
}

1116
static bool isStateWindow(SStreamScanInfo* pInfo) {
1117
  return pInfo->windowSup.parentType == QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE;
5
54liuyao 已提交
1118
}
5
54liuyao 已提交
1119

L
Liu Jicong 已提交
1120
static bool isIntervalWindow(SStreamScanInfo* pInfo) {
1121 1122 1123
  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 已提交
1124 1125 1126
}

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

1130 1131 1132 1133
static bool isSlidingWindow(SStreamScanInfo* pInfo) {
  return isIntervalWindow(pInfo) && pInfo->interval.interval != pInfo->interval.sliding;
}

1134
static void setGroupId(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t groupColIndex, int32_t rowIndex) {
1135 1136
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, groupColIndex);
  uint64_t*        groupCol = (uint64_t*)pColInfo->pData;
1137
  ASSERT(rowIndex < pBlock->info.rows);
1138
  pInfo->groupId = groupCol[rowIndex];
1139 1140
}

L
Liu Jicong 已提交
1141
void resetTableScanInfo(STableScanInfo* pTableScanInfo, STimeWindow* pWin) {
H
Haojun Liao 已提交
1142
  pTableScanInfo->cond.twindows = *pWin;
L
Liu Jicong 已提交
1143 1144
  pTableScanInfo->scanTimes = 0;
  pTableScanInfo->currentGroupId = -1;
1145 1146
  tsdbReaderClose(pTableScanInfo->dataReader);
  pTableScanInfo->dataReader = NULL;
1147 1148
}

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

1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165
  STableScanInfo*     pTableScanInfo = pTableScanOp->info;
  SQueryTableDataCond cond = pTableScanInfo->cond;

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

  SExecTaskInfo* pTaskInfo = pTableScanOp->pTaskInfo;

  SSDataBlock* pBlock = pTableScanInfo->pResBlock;
  blockDataCleanup(pBlock);

  STsdbReader* pReader = NULL;
5
54liuyao 已提交
1166 1167
  int32_t      code = tsdbReaderOpen(pTableScanInfo->readHandle.vnode, &cond, &tblInfo, 1, (STsdbReader**)&pReader,
                                     GET_TASKID(pTaskInfo));
1168 1169
  if (code != TSDB_CODE_SUCCESS) {
    terrno = code;
dengyihao's avatar
dengyihao 已提交
1170
    T_LONG_JMP(pTaskInfo->env, code);
1171 1172 1173 1174 1175
    return NULL;
  }

  bool hasBlock = tsdbNextDataBlock(pReader);
  if (hasBlock) {
H
Haojun Liao 已提交
1176
    SDataBlockInfo* pBInfo = &pBlock->info;
1177

H
Haojun Liao 已提交
1178 1179
    int32_t rows = 0;
    tsdbRetrieveDataBlockInfo(pReader, &rows, &pBInfo->uid, &pBInfo->window);
1180

H
Haojun Liao 已提交
1181
    SArray* pCols = tsdbRetrieveDataBlock(pReader, NULL);
H
Haojun Liao 已提交
1182 1183
    blockDataEnsureCapacity(pBlock, rows);
    pBlock->info.rows = rows;
1184 1185

    relocateColumnData(pBlock, pTableScanInfo->matchInfo.pList, pCols, true);
1186
    doSetTagColumnData(pTableScanInfo, pBlock, pTaskInfo, rows);
H
Haojun Liao 已提交
1187

H
Haojun Liao 已提交
1188
    pBlock->info.groupId = getTableGroupId(pTaskInfo->pTableInfoList, pBInfo->uid);
1189 1190 1191 1192
  }

  tsdbReaderClose(pReader);
  qDebug("retrieve prev rows:%d, skey:%" PRId64 ", ekey:%" PRId64 " uid:%" PRIu64 ", max ver:%" PRId64
5
54liuyao 已提交
1193 1194
         ", suid:%" PRIu64,
         pBlock->info.rows, startTs, endTs, tbUid, maxVersion, cond.suid);
1195 1196

  return pBlock->info.rows > 0 ? pBlock : NULL;
1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207
}

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 已提交
1208
static uint64_t getGroupIdByUid(SStreamScanInfo* pInfo, uint64_t uid) {
H
Haojun Liao 已提交
1209
  return getTableGroupId(pInfo->pTableScanOp->pTaskInfo->pTableInfoList, uid);
1210 1211
}

5
54liuyao 已提交
1212 1213 1214 1215 1216 1217 1218 1219
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 已提交
1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
static bool prepareRangeScan(SStreamScanInfo* pInfo, SSDataBlock* pBlock, int32_t* pRowIndex) {
  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]};
1231 1232 1233
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
  uint64_t*        gpData = (uint64_t*)pGpCol->pData;
  uint64_t         groupId = gpData[*pRowIndex];
1234 1235 1236 1237 1238 1239

  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 已提交
1240
  setGroupId(pInfo, pBlock, GROUPID_COLUMN_INDEX, *pRowIndex);
1241 1242 1243 1244
  if (isSlidingWindow(pInfo)) {
    pInfo->updateWin.skey = calStartData[*pRowIndex];
    pInfo->updateWin.ekey = calEndData[*pRowIndex];
  }
L
Liu Jicong 已提交
1245 1246 1247
  (*pRowIndex)++;

  for (; *pRowIndex < pBlock->info.rows; (*pRowIndex)++) {
1248
    if (win.skey == startData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
L
Liu Jicong 已提交
1249 1250 1251
      win.ekey = TMAX(win.ekey, endData[*pRowIndex]);
      continue;
    }
1252
    if (win.skey == endData[*pRowIndex] && groupId == gpData[*pRowIndex]) {
L
Liu Jicong 已提交
1253 1254 1255
      win.skey = TMIN(win.skey, startData[*pRowIndex]);
      continue;
    }
1256 1257
    ASSERT(!(win.skey > startData[*pRowIndex] && win.ekey < endData[*pRowIndex]) ||
           !(isInTimeWindow(&win, startData[*pRowIndex], 0) || isInTimeWindow(&win, endData[*pRowIndex], 0)));
L
Liu Jicong 已提交
1258 1259 1260 1261
    break;
  }

  resetTableScanInfo(pInfo->pTableScanOp->info, &win);
1262
  pInfo->pTableScanOp->status = OP_OPENED;
L
Liu Jicong 已提交
1263 1264 1265
  return true;
}

5
54liuyao 已提交
1266
static STimeWindow getSlidingWindow(TSKEY* startTsCol, TSKEY* endTsCol, uint64_t* gpIdCol, SInterval* pInterval,
1267
                                    SDataBlockInfo* pDataBlockInfo, int32_t* pRowIndex, bool hasGroup) {
H
Haojun Liao 已提交
1268
  SResultRowInfo dumyInfo = {0};
5
54liuyao 已提交
1269
  dumyInfo.cur.pageId = -1;
1270
  STimeWindow win = getActiveTimeWindow(NULL, &dumyInfo, startTsCol[*pRowIndex], pInterval, TSDB_ORDER_ASC);
5
54liuyao 已提交
1271 1272
  STimeWindow endWin = win;
  STimeWindow preWin = win;
5
54liuyao 已提交
1273
  uint64_t    groupId = gpIdCol[*pRowIndex];
H
Haojun Liao 已提交
1274

5
54liuyao 已提交
1275
  while (1) {
1276 1277 1278
    if (hasGroup) {
      (*pRowIndex) += 1;
    } else {
5
54liuyao 已提交
1279 1280 1281 1282 1283 1284
      while ((groupId == gpIdCol[(*pRowIndex)] && startTsCol[*pRowIndex] < endWin.ekey)) {
        (*pRowIndex) += 1;
        if ((*pRowIndex) == pDataBlockInfo->rows) {
          break;
        }
      }
1285
    }
5
54liuyao 已提交
1286

5
54liuyao 已提交
1287 1288 1289
    do {
      preWin = endWin;
      getNextTimeWindow(pInterval, &endWin, TSDB_ORDER_ASC);
1290
    } while (endTsCol[(*pRowIndex) - 1] >= endWin.skey);
5
54liuyao 已提交
1291
    endWin = preWin;
5
54liuyao 已提交
1292
    if (win.ekey == endWin.ekey || (*pRowIndex) == pDataBlockInfo->rows || groupId != gpIdCol[*pRowIndex]) {
5
54liuyao 已提交
1293 1294 1295 1296 1297 1298
      win.ekey = endWin.ekey;
      return win;
    }
    win.ekey = endWin.ekey;
  }
}
5
54liuyao 已提交
1299

L
Liu Jicong 已提交
1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310
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 已提交
1311
      pInfo->updateWin = (STimeWindow){.skey = INT64_MIN, .ekey = INT64_MAX};
H
Hongze Cheng 已提交
1312 1313 1314
      STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
      tsdbReaderClose(pTableScanInfo->dataReader);
      pTableScanInfo->dataReader = NULL;
1315 1316
      return NULL;
    }
L
Liu Jicong 已提交
1317

H
Haojun Liao 已提交
1318
    doFilter(pInfo->pCondition, pResult, NULL, NULL);
1319 1320 1321 1322
    if (pResult->info.rows == 0) {
      continue;
    }

1323 1324 1325 1326 1327 1328 1329 1330
    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 已提交
1331 1332
            bool             isNull = colDataIsNull(pSrcCol, tmpBlock->info.rows, i, NULL);
            char*            pSrcData = colDataGetData(pSrcCol, i);
1333 1334 1335 1336 1337
            colDataAppend(pDestCol, pResult->info.rows, pSrcData, isNull);
          }
          pResult->info.rows++;
        }
      }
H
Haojun Liao 已提交
1338 1339 1340

      blockDataDestroy(tmpBlock);

1341 1342 1343 1344 1345
      if (pResult->info.rows > 0) {
        pResult->info.calWin = pInfo->updateWin;
        return pResult;
      }
    } else if (pResult->info.groupId == pInfo->groupId) {
5
54liuyao 已提交
1346
      pResult->info.calWin = pInfo->updateWin;
1347
      return pResult;
5
54liuyao 已提交
1348 1349
    }
  }
1350
}
1351

1352 1353 1354
static int32_t generateSessionScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
  if (pSrcBlock->info.rows == 0) {
    return TSDB_CODE_SUCCESS;
1355
  }
1356 1357
  blockDataCleanup(pDestBlock);
  int32_t code = blockDataEnsureCapacity(pDestBlock, pSrcBlock->info.rows);
1358
  if (code != TSDB_CODE_SUCCESS) {
1359
    return code;
L
Liu Jicong 已提交
1360
  }
1361 1362
  ASSERT(taosArrayGetSize(pSrcBlock->pDataBlock) >= 3);
  SColumnInfoData* pStartTsCol = taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
L
Liu Jicong 已提交
1363
  TSKEY*           startData = (TSKEY*)pStartTsCol->pData;
1364
  SColumnInfoData* pEndTsCol = taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
L
Liu Jicong 已提交
1365
  TSKEY*           endData = (TSKEY*)pEndTsCol->pData;
1366 1367
  SColumnInfoData* pUidCol = taosArrayGet(pSrcBlock->pDataBlock, UID_COLUMN_INDEX);
  uint64_t*        uidCol = (uint64_t*)pUidCol->pData;
L
Liu Jicong 已提交
1368

1369 1370
  SColumnInfoData* pDestStartCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pDestEndCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
5
54liuyao 已提交
1371
  SColumnInfoData* pDestUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
1372
  SColumnInfoData* pDestGpCol = taosArrayGet(pDestBlock->pDataBlock, GROUPID_COLUMN_INDEX);
5
54liuyao 已提交
1373 1374
  SColumnInfoData* pDestCalStartTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
  SColumnInfoData* pDestCalEndTsCol = taosArrayGet(pDestBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
L
Liu Jicong 已提交
1375
  int64_t          version = pSrcBlock->info.version - 1;
1376
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
1377
    uint64_t groupId = getGroupIdByData(pInfo, uidCol[i], startData[i], version);
L
Liu Jicong 已提交
1378
    // gap must be 0.
5
54liuyao 已提交
1379 1380 1381
    SSessionKey startWin = {0};
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, startData[i], endData[i], groupId, &startWin);
    if (IS_INVALID_SESSION_WIN_KEY(startWin)) {
L
Liu Jicong 已提交
1382 1383 1384
      // window has been closed.
      continue;
    }
5
54liuyao 已提交
1385 1386 1387 1388 1389 1390
    SSessionKey endWin = {0};
    getCurSessionWindow(pInfo->windowSup.pStreamAggSup, endData[i], endData[i], groupId, &endWin);
    ASSERT(!IS_INVALID_SESSION_WIN_KEY(endWin));
    colDataAppend(pDestStartCol, i, (const char*)&startWin.win.skey, false);
    colDataAppend(pDestEndCol, i, (const char*)&endWin.win.ekey, false);

5
54liuyao 已提交
1391
    colDataAppendNULL(pDestUidCol, i);
L
Liu Jicong 已提交
1392
    colDataAppend(pDestGpCol, i, (const char*)&groupId, false);
5
54liuyao 已提交
1393 1394
    colDataAppendNULL(pDestCalStartTsCol, i);
    colDataAppendNULL(pDestCalEndTsCol, i);
1395
    pDestBlock->info.rows++;
L
Liu Jicong 已提交
1396
  }
1397
  return TSDB_CODE_SUCCESS;
L
Liu Jicong 已提交
1398
}
1399 1400 1401 1402 1403 1404

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;
1405
  }
5
54liuyao 已提交
1406
  int32_t code = blockDataEnsureCapacity(pDestBlock, rows);
1407 1408 1409 1410
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

1411 1412
  SColumnInfoData* pSrcStartTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pSrcEndTsCol = (SColumnInfoData*)taosArrayGet(pSrcBlock->pDataBlock, END_TS_COLUMN_INDEX);
1413 1414 1415 1416
  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;
1417 1418 1419
  ASSERT(pSrcStartTsCol->info.type == TSDB_DATA_TYPE_TIMESTAMP);
  TSKEY*           srcStartTsCol = (TSKEY*)pSrcStartTsCol->pData;
  TSKEY*           srcEndTsCol = (TSKEY*)pSrcEndTsCol->pData;
1420 1421
  SColumnInfoData* pStartTsCol = taosArrayGet(pDestBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pEndTsCol = taosArrayGet(pDestBlock->pDataBlock, END_TS_COLUMN_INDEX);
1422
  SColumnInfoData* pDeUidCol = taosArrayGet(pDestBlock->pDataBlock, UID_COLUMN_INDEX);
1423 1424 1425
  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);
L
Liu Jicong 已提交
1426
  int64_t          version = pSrcBlock->info.version - 1;
1427
  for (int32_t i = 0; i < rows;) {
1428
    uint64_t srcUid = srcUidData[i];
5
54liuyao 已提交
1429 1430 1431 1432 1433
    uint64_t groupId = srcGp[i];
    if (groupId == 0) {
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], version);
    }
    TSKEY calStartTs = srcStartTsCol[i];
1434
    colDataAppend(pCalStartTsCol, pDestBlock->info.rows, (const char*)(&calStartTs), false);
5
54liuyao 已提交
1435
    STimeWindow win = getSlidingWindow(srcStartTsCol, srcEndTsCol, srcGp, &pInfo->interval, &pSrcBlock->info, &i,
1436 1437
                                       pInfo->partitionSup.needCalc);
    TSKEY       calEndTs = srcStartTsCol[i - 1];
1438 1439
    colDataAppend(pCalEndTsCol, pDestBlock->info.rows, (const char*)(&calEndTs), false);
    colDataAppend(pDeUidCol, pDestBlock->info.rows, (const char*)(&srcUid), false);
1440 1441 1442 1443
    colDataAppend(pStartTsCol, pDestBlock->info.rows, (const char*)(&win.skey), false);
    colDataAppend(pEndTsCol, pDestBlock->info.rows, (const char*)(&win.ekey), false);
    colDataAppend(pGpCol, pDestBlock->info.rows, (const char*)(&groupId), false);
    pDestBlock->info.rows++;
5
54liuyao 已提交
1444
  }
1445 1446
  return TSDB_CODE_SUCCESS;
}
1447

1448
static int32_t generateDeleteResultBlock(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
5
54liuyao 已提交
1449 1450 1451
  blockDataCleanup(pDestBlock);
  int32_t rows = pSrcBlock->info.rows;
  if (rows == 0) {
1452 1453
    return TSDB_CODE_SUCCESS;
  }
5
54liuyao 已提交
1454
  int32_t code = blockDataEnsureCapacity(pDestBlock, rows);
1455 1456 1457 1458
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

5
54liuyao 已提交
1459 1460 1461 1462 1463 1464 1465 1466 1467 1468
  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;
1469
  for (int32_t i = 0; i < pSrcBlock->info.rows; i++) {
5
54liuyao 已提交
1470 1471 1472 1473 1474 1475
    uint64_t srcUid = srcUidData[i];
    uint64_t groupId = srcGp[i];
    if (groupId == 0) {
      groupId = getGroupIdByData(pInfo, srcUid, srcStartTsCol[i], version);
    }
    appendOneRowToStreamSpecialBlock(pDestBlock, srcStartTsCol + i, srcEndTsCol + i, srcUidData + i, &groupId, NULL);
1476 1477 1478 1479
  }
  return TSDB_CODE_SUCCESS;
}

1480 1481 1482 1483
static int32_t generateScanRange(SStreamScanInfo* pInfo, SSDataBlock* pSrcBlock, SSDataBlock* pDestBlock) {
  int32_t code = TSDB_CODE_SUCCESS;
  if (isIntervalWindow(pInfo)) {
    code = generateIntervalScanRange(pInfo, pSrcBlock, pDestBlock);
1484
  } else if (isSessionWindow(pInfo) || isStateWindow(pInfo)) {
1485
    code = generateSessionScanRange(pInfo, pSrcBlock, pDestBlock);
5
54liuyao 已提交
1486 1487
  } else {
    code = generateDeleteResultBlock(pInfo, pSrcBlock, pDestBlock);
1488
  }
1489
  pDestBlock->info.type = STREAM_CLEAR;
1490
  pDestBlock->info.version = pSrcBlock->info.version;
1491 1492 1493 1494
  blockDataUpdateTsWindow(pDestBlock, 0);
  return code;
}

L
Liu Jicong 已提交
1495 1496 1497 1498 1499 1500 1501 1502 1503
static void calBlockTag(SExprSupp* pTagCalSup, SSDataBlock* pBlock, SSDataBlock* pResBlock) {
  if (pTagCalSup == NULL || pTagCalSup->numOfExprs == 0) return;
  if (pBlock == NULL || pBlock->info.rows == 0) return;

  SSDataBlock* pSrcBlock = blockCopyOneRow(pBlock, 0);
  ASSERT(pSrcBlock->info.rows == 1);

  blockDataEnsureCapacity(pResBlock, 1);

H
Haojun Liao 已提交
1504
  projectApplyFunctions(pTagCalSup->pExprInfo, pResBlock, pSrcBlock, pTagCalSup->pCtx, 1, NULL);
L
Liu Jicong 已提交
1505 1506 1507
  ASSERT(pResBlock->info.rows == 1);

  // build tagArray
1508 1509 1510 1511 1512
  /*SArray* tagArray = taosArrayInit(0, sizeof(void*));*/
  /*STagVal tagVal = {*/
  /*.cid = 0,*/
  /*.type = 0,*/
  /*};*/
L
Liu Jicong 已提交
1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540
  // build STag
  // set STag

  blockDataDestroy(pSrcBlock);
}

static void calBlockTbName(SExprSupp* pTbNameCalSup, SSDataBlock* pBlock) {
  if (pTbNameCalSup == NULL || pTbNameCalSup->numOfExprs == 0) return;
  if (pBlock == NULL || pBlock->info.rows == 0) return;

  SSDataBlock* pSrcBlock = blockCopyOneRow(pBlock, 0);
  ASSERT(pSrcBlock->info.rows == 1);

  SSDataBlock* pResBlock = createDataBlock();
  pResBlock->info.rowSize = VARSTR_HEADER_SIZE + TSDB_TABLE_NAME_LEN;
  SColumnInfoData data = createColumnInfoData(TSDB_DATA_TYPE_VARCHAR, TSDB_TABLE_NAME_LEN, 0);
  taosArrayPush(pResBlock->pDataBlock, &data);
  blockDataEnsureCapacity(pResBlock, 1);

  projectApplyFunctions(pTbNameCalSup->pExprInfo, pResBlock, pSrcBlock, pTbNameCalSup->pCtx, 1, NULL);
  ASSERT(pResBlock->info.rows == 1);
  ASSERT(taosArrayGetSize(pResBlock->pDataBlock) == 1);
  SColumnInfoData* pCol = taosArrayGet(pResBlock->pDataBlock, 0);
  ASSERT(pCol->info.type == TSDB_DATA_TYPE_VARCHAR);

  void* pData = colDataGetData(pCol, 0);
  // TODO check tbname validation
  if (pData != (void*)-1 && pData != NULL) {
L
Liu Jicong 已提交
1541 1542
    memcpy(pBlock->info.parTbName, varDataVal(pData), TMIN(varDataLen(pData), TSDB_TABLE_NAME_LEN));
    pBlock->info.parTbName[TSDB_TABLE_NAME_LEN - 1] = 0;
L
Liu Jicong 已提交
1543 1544 1545 1546 1547 1548 1549 1550
  } else {
    pBlock->info.parTbName[0] = 0;
  }

  blockDataDestroy(pSrcBlock);
  blockDataDestroy(pResBlock);
}

1551 1552
void appendOneRowToStreamSpecialBlock(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, uint64_t* pUid,
                                      uint64_t* pGp, void* pTbName) {
1553 1554
  SColumnInfoData* pStartTsCol = taosArrayGet(pBlock->pDataBlock, START_TS_COLUMN_INDEX);
  SColumnInfoData* pEndTsCol = taosArrayGet(pBlock->pDataBlock, END_TS_COLUMN_INDEX);
1555 1556
  SColumnInfoData* pUidCol = taosArrayGet(pBlock->pDataBlock, UID_COLUMN_INDEX);
  SColumnInfoData* pGpCol = taosArrayGet(pBlock->pDataBlock, GROUPID_COLUMN_INDEX);
1557 1558
  SColumnInfoData* pCalStartCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX);
  SColumnInfoData* pCalEndCol = taosArrayGet(pBlock->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX);
1559
  SColumnInfoData* pTableCol = taosArrayGet(pBlock->pDataBlock, TABLE_NAME_COLUMN_INDEX);
1560 1561
  colDataAppend(pStartTsCol, pBlock->info.rows, (const char*)pStartTs, false);
  colDataAppend(pEndTsCol, pBlock->info.rows, (const char*)pEndTs, false);
1562 1563
  colDataAppend(pUidCol, pBlock->info.rows, (const char*)pUid, false);
  colDataAppend(pGpCol, pBlock->info.rows, (const char*)pGp, false);
1564 1565
  colDataAppend(pCalStartCol, pBlock->info.rows, (const char*)pStartTs, false);
  colDataAppend(pCalEndCol, pBlock->info.rows, (const char*)pEndTs, false);
1566
  colDataAppend(pTableCol, pBlock->info.rows, (const char*)pTbName, pTbName == NULL);
1567
  pBlock->info.rows++;
5
54liuyao 已提交
1568 1569
}

1570
static void checkUpdateData(SStreamScanInfo* pInfo, bool invertible, SSDataBlock* pBlock, bool out) {
1571 1572
  if (out) {
    blockDataCleanup(pInfo->pUpdateDataRes);
5
54liuyao 已提交
1573
    blockDataEnsureCapacity(pInfo->pUpdateDataRes, pBlock->info.rows * 2);
1574
  }
1575 1576
  SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, pInfo->primaryTsIndex);
  ASSERT(pColDataInfo->info.type == TSDB_DATA_TYPE_TIMESTAMP);
5
54liuyao 已提交
1577
  TSKEY* tsCol = (TSKEY*)pColDataInfo->pData;
L
Liu Jicong 已提交
1578
  bool   tableInserted = updateInfoIsTableInserted(pInfo->pUpdateInfo, pBlock->info.uid);
1579
  for (int32_t rowId = 0; rowId < pBlock->info.rows; rowId++) {
5
54liuyao 已提交
1580 1581
    SResultRowInfo dumyInfo;
    dumyInfo.cur.pageId = -1;
L
Liu Jicong 已提交
1582
    bool        isClosed = false;
5
54liuyao 已提交
1583
    STimeWindow win = {.skey = INT64_MIN, .ekey = INT64_MAX};
L
Liu Jicong 已提交
1584
    if (tableInserted && isOverdue(tsCol[rowId], &pInfo->twAggSup)) {
5
54liuyao 已提交
1585 1586 1587
      win = getActiveTimeWindow(NULL, &dumyInfo, tsCol[rowId], &pInfo->interval, TSDB_ORDER_ASC);
      isClosed = isCloseWindow(&win, &pInfo->twAggSup);
    }
5
54liuyao 已提交
1588 1589
    // must check update info first.
    bool update = updateInfoIsUpdated(pInfo->pUpdateInfo, pBlock->info.uid, tsCol[rowId]);
L
Liu Jicong 已提交
1590
    bool closedWin = isClosed && isSignleIntervalWindow(pInfo) &&
1591 1592
                     isDeletedStreamWindow(&win, pBlock->info.groupId,
                                           pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, &pInfo->twAggSup);
L
Liu Jicong 已提交
1593
    if ((update || closedWin) && out) {
L
Liu Jicong 已提交
1594
      qDebug("stream update check not pass, update %d, closedWin %d", update, closedWin);
5
54liuyao 已提交
1595
      uint64_t gpId = 0;
1596 1597
      appendOneRowToStreamSpecialBlock(pInfo->pUpdateDataRes, tsCol + rowId, tsCol + rowId, &pBlock->info.uid, &gpId,
                                       NULL);
5
54liuyao 已提交
1598 1599
      if (closedWin && pInfo->partitionSup.needCalc) {
        gpId = calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pBlock, rowId);
1600 1601
        appendOneRowToStreamSpecialBlock(pInfo->pUpdateDataRes, tsCol + rowId, tsCol + rowId, &pBlock->info.uid, &gpId,
                                         NULL);
5
54liuyao 已提交
1602
      }
1603 1604
    }
  }
1605 1606
  if (out && pInfo->pUpdateDataRes->info.rows > 0) {
    pInfo->pUpdateDataRes->info.version = pBlock->info.version;
1607
    blockDataUpdateTsWindow(pInfo->pUpdateDataRes, 0);
1608
    pInfo->pUpdateDataRes->info.type = pInfo->partitionSup.needCalc ? STREAM_DELETE_DATA : STREAM_CLEAR;
5
54liuyao 已提交
1609 1610
  }
}
L
Liu Jicong 已提交
1611

1612
static int32_t setBlockIntoRes(SStreamScanInfo* pInfo, const SSDataBlock* pBlock, bool filter) {
L
Liu Jicong 已提交
1613 1614
  SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
  SOperatorInfo*  pOperator = pInfo->pStreamScanOp;
L
Liu Jicong 已提交
1615
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
L
Liu Jicong 已提交
1616

1617 1618
  blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);

L
Liu Jicong 已提交
1619 1620 1621
  pInfo->pRes->info.rows = pBlock->info.rows;
  pInfo->pRes->info.uid = pBlock->info.uid;
  pInfo->pRes->info.type = STREAM_NORMAL;
1622
  pInfo->pRes->info.version = pBlock->info.version;
L
Liu Jicong 已提交
1623

H
Haojun Liao 已提交
1624
  pInfo->pRes->info.groupId = getTableGroupId(pTaskInfo->pTableInfoList, pBlock->info.uid);
L
Liu Jicong 已提交
1625 1626

  // todo extract method
H
Haojun Liao 已提交
1627 1628 1629
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->matchInfo.pList); ++i) {
    SColMatchItem* pColMatchInfo = taosArrayGet(pInfo->matchInfo.pList, i);
    if (!pColMatchInfo->needOutput) {
L
Liu Jicong 已提交
1630 1631 1632 1633 1634 1635 1636
      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 已提交
1637
        SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
1638
        colDataAssign(pDst, pResCol, pBlock->info.rows, &pInfo->pRes->info);
L
Liu Jicong 已提交
1639 1640 1641 1642 1643 1644 1645
        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 已提交
1646
      SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->dstSlotId);
L
Liu Jicong 已提交
1647 1648 1649 1650 1651 1652
      colDataAppendNNULL(pDst, 0, pBlockInfo->rows);
    }
  }

  // currently only the tbname pseudo column
  if (pInfo->numOfPseudoExpr > 0) {
L
Liu Jicong 已提交
1653
    int32_t code = addTagPseudoColumnData(&pInfo->readHandle, pInfo->pPseudoExpr, pInfo->numOfPseudoExpr, pInfo->pRes,
1654
                                          pInfo->pRes->info.rows, GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
1655
    if (code != TSDB_CODE_SUCCESS) {
L
Liu Jicong 已提交
1656
      blockDataFreeRes((SSDataBlock*)pBlock);
1657
      T_LONG_JMP(pTaskInfo->env, code);
H
Haojun Liao 已提交
1658
    }
L
Liu Jicong 已提交
1659 1660
  }

1661
  if (filter) {
D
dapan1121 已提交
1662
    doFilter(pInfo->pCondition, pInfo->pRes, NULL, NULL);
1663
  }
1664

L
Liu Jicong 已提交
1665
  blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);
L
Liu Jicong 已提交
1666
  blockDataFreeRes((SSDataBlock*)pBlock);
L
Liu Jicong 已提交
1667

L
Liu Jicong 已提交
1668
  calBlockTbName(&pInfo->tbnameCalSup, pInfo->pRes);
L
Liu Jicong 已提交
1669 1670
  return 0;
}
5
54liuyao 已提交
1671

L
Liu Jicong 已提交
1672
static SSDataBlock* doQueueScan(SOperatorInfo* pOperator) {
1673 1674
  SExecTaskInfo*   pTaskInfo = pOperator->pTaskInfo;
  SStreamScanInfo* pInfo = pOperator->info;
H
Haojun Liao 已提交
1675

L
Liu Jicong 已提交
1676
  qDebug("queue scan called");
L
Liu Jicong 已提交
1677 1678 1679 1680 1681 1682 1683 1684

  if (pTaskInfo->streamInfo.pReq != NULL) {
    if (pInfo->tqReader->pMsg == NULL) {
      pInfo->tqReader->pMsg = pTaskInfo->streamInfo.pReq;
      const SSubmitReq* pSubmit = pInfo->tqReader->pMsg;
      if (tqReaderSetDataMsg(pInfo->tqReader, pSubmit, 0) < 0) {
        qError("submit msg messed up when initing stream submit block %p", pSubmit);
        pInfo->tqReader->pMsg = NULL;
L
Liu Jicong 已提交
1685
        pTaskInfo->streamInfo.pReq = NULL;
L
Liu Jicong 已提交
1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701
        ASSERT(0);
      }
    }

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

    while (tqNextDataBlock(pInfo->tqReader)) {
      SSDataBlock block = {0};

      int32_t code = tqRetrieveDataBlock(&block, pInfo->tqReader);

      if (code != TSDB_CODE_SUCCESS || block.info.rows == 0) {
        continue;
      }

1702
      setBlockIntoRes(pInfo, &block, true);
L
Liu Jicong 已提交
1703 1704 1705 1706 1707 1708 1709 1710

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

    pInfo->tqReader->pMsg = NULL;
    pTaskInfo->streamInfo.pReq = NULL;
L
Liu Jicong 已提交
1711
    return NULL;
L
Liu Jicong 已提交
1712 1713
  }

L
Liu Jicong 已提交
1714 1715 1716
  if (pTaskInfo->streamInfo.prepareStatus.type == TMQ_OFFSET__SNAPSHOT_DATA) {
    SSDataBlock* pResult = doTableScan(pInfo->pTableScanOp);
    if (pResult && pResult->info.rows > 0) {
L
Liu Jicong 已提交
1717
      qDebug("queue scan tsdb return %d rows", pResult->info.rows);
1718
      pTaskInfo->streamInfo.returned = 1;
L
Liu Jicong 已提交
1719 1720
      return pResult;
    } else {
1721 1722 1723 1724 1725
      if (!pTaskInfo->streamInfo.returned) {
        STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
        tsdbReaderClose(pTSInfo->dataReader);
        pTSInfo->dataReader = NULL;
        tqOffsetResetToLog(&pTaskInfo->streamInfo.prepareStatus, pTaskInfo->streamInfo.snapshotVer);
1726
        qDebug("queue scan tsdb over, switch to wal ver %" PRId64 "", pTaskInfo->streamInfo.snapshotVer + 1);
1727 1728 1729 1730 1731
        if (tqSeekVer(pInfo->tqReader, pTaskInfo->streamInfo.snapshotVer + 1) < 0) {
          return NULL;
        }
        ASSERT(pInfo->tqReader->pWalReader->curVersion == pTaskInfo->streamInfo.snapshotVer + 1);
      } else {
L
Liu Jicong 已提交
1732 1733
        return NULL;
      }
1734 1735 1736
    }
  }

L
Liu Jicong 已提交
1737 1738 1739 1740 1741 1742
  if (pTaskInfo->streamInfo.prepareStatus.type == TMQ_OFFSET__LOG) {
    while (1) {
      SFetchRet ret = {0};
      tqNextBlock(pInfo->tqReader, &ret);
      if (ret.fetchType == FETCH_TYPE__DATA) {
        blockDataCleanup(pInfo->pRes);
1743
        if (setBlockIntoRes(pInfo, &ret.data, true) < 0) {
L
Liu Jicong 已提交
1744 1745 1746
          ASSERT(0);
        }
        if (pInfo->pRes->info.rows > 0) {
L
Liu Jicong 已提交
1747
          pOperator->status = OP_EXEC_RECV;
L
Liu Jicong 已提交
1748
          qDebug("queue scan log return %d rows", pInfo->pRes->info.rows);
L
Liu Jicong 已提交
1749 1750 1751 1752
          return pInfo->pRes;
        }
      } else if (ret.fetchType == FETCH_TYPE__META) {
        ASSERT(0);
L
Liu Jicong 已提交
1753 1754 1755
        //        pTaskInfo->streamInfo.lastStatus = ret.offset;
        //        pTaskInfo->streamInfo.metaBlk = ret.meta;
        //        return NULL;
L
Liu Jicong 已提交
1756 1757
      } else if (ret.fetchType == FETCH_TYPE__NONE ||
                 (ret.fetchType == FETCH_TYPE__SEP && pOperator->status == OP_EXEC_RECV)) {
L
Liu Jicong 已提交
1758
        pTaskInfo->streamInfo.lastStatus = ret.offset;
1759 1760 1761 1762
        ASSERT(pTaskInfo->streamInfo.lastStatus.version >= pTaskInfo->streamInfo.prepareStatus.version);
        ASSERT(pTaskInfo->streamInfo.lastStatus.version + 1 == pInfo->tqReader->pWalReader->curVersion);
        char formatBuf[80];
        tFormatOffset(formatBuf, 80, &ret.offset);
L
Liu Jicong 已提交
1763
        qDebug("queue scan log return null, offset %s", formatBuf);
L
Liu Jicong 已提交
1764
        pOperator->status = OP_OPENED;
L
Liu Jicong 已提交
1765 1766 1767
        return NULL;
      }
    }
L
Liu Jicong 已提交
1768
#if 0
1769
    } else if (pTaskInfo->streamInfo.prepareStatus.type == TMQ_OFFSET__SNAPSHOT_DATA) {
L
Liu Jicong 已提交
1770
    SSDataBlock* pResult = doTableScan(pInfo->pTableScanOp);
L
Liu Jicong 已提交
1771 1772 1773 1774 1775 1776
    if (pResult && pResult->info.rows > 0) {
      qDebug("stream scan tsdb return %d rows", pResult->info.rows);
      return pResult;
    }
    qDebug("stream scan tsdb return null");
    return NULL;
L
Liu Jicong 已提交
1777
#endif
L
Liu Jicong 已提交
1778 1779 1780
  } else {
    ASSERT(0);
    return NULL;
H
Haojun Liao 已提交
1781
  }
L
Liu Jicong 已提交
1782 1783
}

L
Liu Jicong 已提交
1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811
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))) {
      colDataAppend(pDstStartCol, j, (const char*)&startCol[i], false);
      colDataAppend(pDstEndCol, j, (const char*)&endCol[i], false);
      colDataAppend(pDstUidCol, j, (const char*)&uidCol[i], false);

      colDataAppendNULL(taosArrayGet(pDst->pDataBlock, GROUPID_COLUMN_INDEX), j);
      colDataAppendNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_START_TS_COLUMN_INDEX), j);
      colDataAppendNULL(taosArrayGet(pDst->pDataBlock, CALCULATE_END_TS_COLUMN_INDEX), j);
      j++;
    }
  }
L
Liu Jicong 已提交
1812
  uint32_t cap = pDst->info.capacity;
L
Liu Jicong 已提交
1813 1814
  pDst->info = pSrc->info;
  pDst->info.rows = j;
L
Liu Jicong 已提交
1815
  pDst->info.capacity = cap;
L
Liu Jicong 已提交
1816 1817 1818 1819

  return 0;
}

5
54liuyao 已提交
1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843
// 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]);
      colDataAppend(pGpCol, i, (const char*)&groupId, false);
    }
  } else {
    // SSDataBlock* pPreRes = readPreVersionData(pInfo->pTableScanOp, uidCol[i], startTsCol, ts, maxVersion);
    // if (!pPreRes || pPreRes->info.rows == 0) {
    //   return 0;
    // }
    // ASSERT(pPreRes->info.rows == 1);
    // return calGroupIdByData(&pInfo->partitionSup, pInfo->pPartScalarSup, pPreRes, 0);
  }
}

L
Liu Jicong 已提交
1844 1845 1846 1847 1848
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 已提交
1849
  qDebug("stream scan called");
L
Liu Jicong 已提交
1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882
#if 0
  SStreamState* pState = pTaskInfo->streamInfo.pState;
  if (pState) {
    printf(">>>>>>>> stream write backend\n");
    SWinKey key = {
        .ts = 1,
        .groupId = 2,
    };
    char tmp[100] = "abcdefg1";
    if (streamStatePut(pState, &key, &tmp, strlen(tmp) + 1) < 0) {
      ASSERT(0);
    }

    key.ts = 2;
    char tmp2[100] = "abcdefg2";
    if (streamStatePut(pState, &key, &tmp2, strlen(tmp2) + 1) < 0) {
      ASSERT(0);
    }

    key.groupId = 5;
    key.ts = 1;
    char tmp3[100] = "abcdefg3";
    if (streamStatePut(pState, &key, &tmp3, strlen(tmp3) + 1) < 0) {
      ASSERT(0);
    }

    char*   val2 = NULL;
    int32_t sz;
    if (streamStateGet(pState, &key, (void**)&val2, &sz) < 0) {
      ASSERT(0);
    }
    printf("stream read %s %d\n", val2, sz);
    streamFreeVal(val2);
H
Haojun Liao 已提交
1883
  }
L
Liu Jicong 已提交
1884
#endif
H
Haojun Liao 已提交
1885

1886 1887
  if (pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__PREPARE1 ||
      pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__PREPARE2) {
L
Liu Jicong 已提交
1888 1889
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
    memcpy(&pTSInfo->cond, &pTaskInfo->streamInfo.tableCond, sizeof(SQueryTableDataCond));
1890
    if (pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__PREPARE1) {
1891
      pTSInfo->cond.startVersion = 0;
1892
      pTSInfo->cond.endVersion = pTaskInfo->streamInfo.fillHistoryVer1;
1893 1894
      qDebug("stream recover step 1, from %" PRId64 " to %" PRId64, pTSInfo->cond.startVersion,
             pTSInfo->cond.endVersion);
1895 1896 1897
    } else {
      pTSInfo->cond.startVersion = pTaskInfo->streamInfo.fillHistoryVer1 + 1;
      pTSInfo->cond.endVersion = pTaskInfo->streamInfo.fillHistoryVer2;
1898 1899
      qDebug("stream recover step 2, from %" PRId64 " to %" PRId64, pTSInfo->cond.startVersion,
             pTSInfo->cond.endVersion);
1900
    }
L
Liu Jicong 已提交
1901 1902 1903 1904 1905

    /*resetTableScanInfo(pTSInfo, pWin);*/
    tsdbReaderClose(pTSInfo->dataReader);
    pTSInfo->dataReader = NULL;

L
Liu Jicong 已提交
1906 1907 1908 1909 1910 1911 1912 1913
    pTSInfo->scanTimes = 0;
    pTSInfo->currentGroupId = -1;
    pTaskInfo->streamInfo.recoverStep = STREAM_RECOVER_STEP__SCAN;
  }

  if (pTaskInfo->streamInfo.recoverStep == STREAM_RECOVER_STEP__SCAN) {
    SSDataBlock* pBlock = doTableScan(pInfo->pTableScanOp);
    if (pBlock != NULL) {
L
Liu Jicong 已提交
1914 1915
      calBlockTbName(&pInfo->tbnameCalSup, pBlock);
      updateInfoFillBlockData(pInfo->pUpdateInfo, pBlock, pInfo->primaryTsIndex);
1916
      qDebug("stream recover scan get block, rows %d", pBlock->info.rows);
L
Liu Jicong 已提交
1917 1918 1919
      return pBlock;
    }
    pTaskInfo->streamInfo.recoverStep = STREAM_RECOVER_STEP__NONE;
L
Liu Jicong 已提交
1920
    STableScanInfo* pTSInfo = pInfo->pTableScanOp->info;
1921 1922 1923 1924
    tsdbReaderClose(pTSInfo->dataReader);
    pTSInfo->dataReader = NULL;

    pTSInfo->cond.startVersion = -1;
L
Liu Jicong 已提交
1925 1926
    pTSInfo->cond.endVersion = -1;

L
Liu Jicong 已提交
1927 1928 1929
    return NULL;
  }

5
54liuyao 已提交
1930
  size_t total = taosArrayGetSize(pInfo->pBlockLists);
5
54liuyao 已提交
1931
// TODO: refactor
L
Liu Jicong 已提交
1932
FETCH_NEXT_BLOCK:
L
Liu Jicong 已提交
1933
  if (pInfo->blockType == STREAM_INPUT__DATA_BLOCK) {
H
Haojun Liao 已提交
1934
    if (pInfo->validBlockIndex >= total) {
L
Liu Jicong 已提交
1935
      doClearBufferedBlocks(pInfo);
L
Liu Jicong 已提交
1936
      /*pOperator->status = OP_EXEC_DONE;*/
H
Haojun Liao 已提交
1937 1938 1939
      return NULL;
    }

1940
    int32_t      current = pInfo->validBlockIndex++;
1941
    SSDataBlock* pBlock = taosArrayGetP(pInfo->pBlockLists, current);
1942
    // TODO move into scan
5
54liuyao 已提交
1943 1944
    pBlock->info.calWin.skey = INT64_MIN;
    pBlock->info.calWin.ekey = INT64_MAX;
1945
    blockDataUpdateTsWindow(pBlock, 0);
1946
    switch (pBlock->info.type) {
L
Liu Jicong 已提交
1947 1948 1949
      case STREAM_NORMAL:
      case STREAM_GET_ALL:
        return pBlock;
1950 1951 1952
      case STREAM_RETRIEVE: {
        pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
        pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RETRIEVE;
1953 1954
        copyDataBlock(pInfo->pUpdateRes, pBlock);
        prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
1955 1956 1957
        updateInfoAddCloseWindowSBF(pInfo->pUpdateInfo);
      } break;
      case STREAM_DELETE_DATA: {
1958
        printDataBlock(pBlock, "stream scan delete recv");
L
Liu Jicong 已提交
1959
        SSDataBlock* pDelBlock = NULL;
L
Liu Jicong 已提交
1960
        if (pInfo->tqReader) {
L
Liu Jicong 已提交
1961
          pDelBlock = createSpecialDataBlock(STREAM_DELETE_DATA);
L
Liu Jicong 已提交
1962
          filterDelBlockByUid(pDelBlock, pBlock, pInfo);
L
Liu Jicong 已提交
1963 1964
        } else {
          pDelBlock = pBlock;
L
Liu Jicong 已提交
1965
        }
5
54liuyao 已提交
1966 1967
        setBlockGroupIdByUid(pInfo, pDelBlock);
        printDataBlock(pDelBlock, "stream scan delete recv filtered");
1968
        if (!isIntervalWindow(pInfo) && !isSessionWindow(pInfo) && !isStateWindow(pInfo)) {
L
Liu Jicong 已提交
1969
          generateDeleteResultBlock(pInfo, pDelBlock, pInfo->pDeleteDataRes);
1970
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_RESULT;
L
Liu Jicong 已提交
1971
          printDataBlock(pDelBlock, "stream scan delete result");
H
Haojun Liao 已提交
1972 1973
          blockDataDestroy(pDelBlock);

L
Liu Jicong 已提交
1974 1975 1976 1977 1978
          if (pInfo->pDeleteDataRes->info.rows > 0) {
            return pInfo->pDeleteDataRes;
          } else {
            goto FETCH_NEXT_BLOCK;
          }
1979 1980 1981
        } else {
          pInfo->blockType = STREAM_INPUT__DATA_SUBMIT;
          pInfo->updateResIndex = 0;
L
Liu Jicong 已提交
1982
          generateScanRange(pInfo, pDelBlock, pInfo->pUpdateRes);
1983 1984 1985 1986
          prepareRangeScan(pInfo, pInfo->pUpdateRes, &pInfo->updateResIndex);
          copyDataBlock(pInfo->pDeleteDataRes, pInfo->pUpdateRes);
          pInfo->pDeleteDataRes->info.type = STREAM_DELETE_DATA;
          pInfo->scanMode = STREAM_SCAN_FROM_DATAREADER_RANGE;
L
Liu Jicong 已提交
1987 1988 1989 1990
          printDataBlock(pDelBlock, "stream scan delete data");
          if (pInfo->tqReader) {
            blockDataDestroy(pDelBlock);
          }
L
Liu Jicong 已提交
1991 1992 1993 1994 1995
          if (pInfo->pDeleteDataRes->info.rows > 0) {
            return pInfo->pDeleteDataRes;
          } else {
            goto FETCH_NEXT_BLOCK;
          }
1996
        }
1997 1998 1999
      } break;
      default:
        break;
5
54liuyao 已提交
2000
    }
2001
    // printDataBlock(pBlock, "stream scan recv");
2002
    return pBlock;
L
Liu Jicong 已提交
2003
  } else if (pInfo->blockType == STREAM_INPUT__DATA_SUBMIT) {
L
Liu Jicong 已提交
2004
    qDebug("scan mode %d", pInfo->scanMode);
5
54liuyao 已提交
2005 2006 2007 2008 2009 2010
    switch (pInfo->scanMode) {
      case STREAM_SCAN_FROM_RES: {
        blockDataDestroy(pInfo->pUpdateRes);
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
        return pInfo->pRes;
      } break;
2011
      case STREAM_SCAN_FROM_DELETE_DATA: {
2012 2013 2014 2015 2016 2017 2018
        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 已提交
2019 2020 2021 2022 2023 2024 2025 2026 2027 2028
      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) {
2029
          STableScanInfo* pTableScanInfo = pInfo->pTableScanOp->info;
L
Liu Jicong 已提交
2030 2031
          uint64_t        version = getReaderMaxVersion(pTableScanInfo->dataReader);
          updateInfoSetScanRange(pInfo->pUpdateInfo, &pTableScanInfo->cond.twindows, pInfo->groupId, version);
5
54liuyao 已提交
2032 2033
          pSDB->info.type = pInfo->scanMode == STREAM_SCAN_FROM_DATAREADER_RANGE ? STREAM_NORMAL : STREAM_PULL_DATA;
          checkUpdateData(pInfo, true, pSDB, false);
2034
          // printDataBlock(pSDB, "stream scan update");
L
Liu Jicong 已提交
2035
          calBlockTbName(&pInfo->tbnameCalSup, pSDB);
5
54liuyao 已提交
2036 2037
          return pSDB;
        }
2038
        blockDataCleanup(pInfo->pUpdateDataRes);
5
54liuyao 已提交
2039 2040 2041 2042
        pInfo->scanMode = STREAM_SCAN_FROM_READERHANDLE;
      } break;
      default:
        break;
2043
    }
2044

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

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

2057
    int32_t totBlockNum = taosArrayGetSize(pInfo->pBlockLists);
2058

L
Liu Jicong 已提交
2059
  NEXT_SUBMIT_BLK:
2060 2061 2062
    while (1) {
      if (pInfo->tqReader->pMsg == NULL) {
        if (pInfo->validBlockIndex >= totBlockNum) {
5
54liuyao 已提交
2063
          updateInfoDestoryColseWinSBF(pInfo->pUpdateInfo);
L
Liu Jicong 已提交
2064
          doClearBufferedBlocks(pInfo);
2065 2066
          return NULL;
        }
2067

2068 2069 2070 2071 2072 2073 2074 2075
        int32_t     current = pInfo->validBlockIndex++;
        SSubmitReq* pSubmit = taosArrayGetP(pInfo->pBlockLists, current);
        if (tqReaderSetDataMsg(pInfo->tqReader, pSubmit, 0) < 0) {
          qError("submit msg messed up when initing stream submit block %p, current %d, total %d", pSubmit, current,
                 totBlockNum);
          pInfo->tqReader->pMsg = NULL;
          continue;
        }
H
Haojun Liao 已提交
2076 2077
      }

2078 2079 2080 2081
      blockDataCleanup(pInfo->pRes);

      while (tqNextDataBlock(pInfo->tqReader)) {
        SSDataBlock block = {0};
2082

2083 2084 2085 2086 2087 2088
        int32_t code = tqRetrieveDataBlock(&block, pInfo->tqReader);

        if (code != TSDB_CODE_SUCCESS || block.info.rows == 0) {
          continue;
        }

2089
        setBlockIntoRes(pInfo, &block, false);
2090

L
Liu Jicong 已提交
2091 2092
        if (updateInfoIgnore(pInfo->pUpdateInfo, &pInfo->pRes->info.window, pInfo->pRes->info.groupId,
                             pInfo->pRes->info.version)) {
2093 2094 2095 2096 2097
          printDataBlock(pInfo->pRes, "stream scan ignore");
          blockDataCleanup(pInfo->pRes);
          continue;
        }

2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113
        if (pInfo->pUpdateInfo) {
          checkUpdateData(pInfo, true, pInfo->pRes, true);
          pInfo->twAggSup.maxTs = TMAX(pInfo->twAggSup.maxTs, pBlockInfo->window.ekey);
          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;
            }
          }
        }

H
Haojun Liao 已提交
2114
        doFilter(pInfo->pCondition, pInfo->pRes, NULL, NULL);
2115 2116 2117
        blockDataUpdateTsWindow(pInfo->pRes, pInfo->primaryTsIndex);

        if (pBlockInfo->rows > 0 || pInfo->pUpdateDataRes->info.rows > 0) {
2118 2119 2120
          break;
        }
      }
2121
      if (pBlockInfo->rows > 0 || pInfo->pUpdateDataRes->info.rows > 0) {
5
54liuyao 已提交
2122
        break;
J
jiacy-jcy 已提交
2123 2124
      } else {
        pInfo->tqReader->pMsg = NULL;
2125
        continue;
5
54liuyao 已提交
2126
      }
2127
      /*blockDataCleanup(pInfo->pRes);*/
H
Haojun Liao 已提交
2128 2129 2130 2131
    }

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

L
Liu Jicong 已提交
2135
    qDebug("scan rows: %d", pBlockInfo->rows);
L
Liu Jicong 已提交
2136 2137 2138
    if (pBlockInfo->rows > 0) {
      return pInfo->pRes;
    }
2139 2140 2141 2142 2143 2144

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

    goto NEXT_SUBMIT_BLK;
L
Liu Jicong 已提交
2145 2146 2147
  } else {
    ASSERT(0);
    return NULL;
H
Haojun Liao 已提交
2148 2149 2150
  }
}

H
Haojun Liao 已提交
2151
static SArray* extractTableIdList(const STableListInfo* pTableListInfo) {
2152 2153 2154
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));

  // Transfer the Array of STableKeyInfo into uid list.
H
Haojun Liao 已提交
2155 2156 2157
  size_t size = tableListGetSize(pTableListInfo);
  for (int32_t i = 0; i < size; ++i) {
    STableKeyInfo* pkeyInfo = tableListGetInfo(pTableListInfo, i);
2158 2159 2160 2161 2162 2163
    taosArrayPush(tableIdList, &pkeyInfo->uid);
  }

  return tableIdList;
}

2164
static SSDataBlock* doRawScan(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
2165 2166
  // NOTE: this operator does never check if current status is done or not
  SExecTaskInfo*      pTaskInfo = pOperator->pTaskInfo;
2167
  SStreamRawScanInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
2168
  pTaskInfo->streamInfo.metaRsp.metaRspLen = 0;  // use metaRspLen !=0 to judge if data is meta
wmmhello's avatar
wmmhello 已提交
2169
  pTaskInfo->streamInfo.metaRsp.metaRsp = NULL;
2170

wmmhello's avatar
wmmhello 已提交
2171
  qDebug("tmqsnap doRawScan called");
L
Liu Jicong 已提交
2172
  if (pTaskInfo->streamInfo.prepareStatus.type == TMQ_OFFSET__SNAPSHOT_DATA) {
wmmhello's avatar
wmmhello 已提交
2173
    SSDataBlock* pBlock = &pInfo->pRes;
2174

2175
    if (pInfo->dataReader && tsdbNextDataBlock(pInfo->dataReader)) {
wmmhello's avatar
wmmhello 已提交
2176 2177 2178
      if (isTaskKilled(pTaskInfo)) {
        longjmp(pTaskInfo->env, TSDB_CODE_TSC_QUERY_CANCELLED);
      }
2179

H
Haojun Liao 已提交
2180 2181 2182
      int32_t rows = 0;
      tsdbRetrieveDataBlockInfo(pInfo->dataReader, &rows, &pBlock->info.uid, &pBlock->info.window);
      pBlock->info.rows = rows;
2183

wmmhello's avatar
wmmhello 已提交
2184 2185 2186
      SArray* pCols = tsdbRetrieveDataBlock(pInfo->dataReader, NULL);
      pBlock->pDataBlock = pCols;
      if (pCols == NULL) {
wmmhello's avatar
wmmhello 已提交
2187
        longjmp(pTaskInfo->env, terrno);
wmmhello's avatar
wmmhello 已提交
2188 2189
      }

2190
      qDebug("tmqsnap doRawScan get data uid:%" PRId64 "", pBlock->info.uid);
wmmhello's avatar
wmmhello 已提交
2191 2192 2193 2194 2195
      pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__SNAPSHOT_DATA;
      pTaskInfo->streamInfo.lastStatus.uid = pBlock->info.uid;
      pTaskInfo->streamInfo.lastStatus.ts = pBlock->info.window.ekey;
      return pBlock;
    }
wmmhello's avatar
wmmhello 已提交
2196 2197

    SMetaTableInfo mtInfo = getUidfromSnapShot(pInfo->sContext);
L
Liu Jicong 已提交
2198
    if (mtInfo.uid == 0) {  // read snapshot done, change to get data from wal
wmmhello's avatar
wmmhello 已提交
2199 2200
      qDebug("tmqsnap read snapshot done, change to get data from wal");
      pTaskInfo->streamInfo.prepareStatus.uid = mtInfo.uid;
wmmhello's avatar
wmmhello 已提交
2201 2202
      pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__LOG;
      pTaskInfo->streamInfo.lastStatus.version = pInfo->sContext->snapVersion;
L
Liu Jicong 已提交
2203
    } else {
wmmhello's avatar
wmmhello 已提交
2204 2205
      pTaskInfo->streamInfo.prepareStatus.uid = mtInfo.uid;
      pTaskInfo->streamInfo.prepareStatus.ts = INT64_MIN;
2206
      qDebug("tmqsnap change get data uid:%" PRId64 "", mtInfo.uid);
wmmhello's avatar
wmmhello 已提交
2207 2208
      qStreamPrepareScan(pTaskInfo, &pTaskInfo->streamInfo.prepareStatus, pInfo->sContext->subType);
    }
2209
    tDeleteSSchemaWrapper(mtInfo.schema);
wmmhello's avatar
wmmhello 已提交
2210
    qDebug("tmqsnap stream scan tsdb return null");
wmmhello's avatar
wmmhello 已提交
2211
    return NULL;
L
Liu Jicong 已提交
2212 2213 2214 2215 2216 2217 2218
  } else if (pTaskInfo->streamInfo.prepareStatus.type == TMQ_OFFSET__SNAPSHOT_META) {
    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 已提交
2219
      qError("tmqsnap getMetafromSnapShot error");
wmmhello's avatar
wmmhello 已提交
2220
      taosMemoryFreeClear(data);
2221 2222 2223
      return NULL;
    }

L
Liu Jicong 已提交
2224
    if (!sContext->queryMetaOrData) {  // change to get data next poll request
wmmhello's avatar
wmmhello 已提交
2225 2226 2227 2228
      pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__SNAPSHOT_META;
      pTaskInfo->streamInfo.lastStatus.uid = uid;
      pTaskInfo->streamInfo.metaRsp.rspOffset.type = TMQ_OFFSET__SNAPSHOT_DATA;
      pTaskInfo->streamInfo.metaRsp.rspOffset.uid = 0;
wmmhello's avatar
wmmhello 已提交
2229
      pTaskInfo->streamInfo.metaRsp.rspOffset.ts = INT64_MIN;
L
Liu Jicong 已提交
2230
    } else {
wmmhello's avatar
wmmhello 已提交
2231 2232 2233 2234 2235 2236 2237
      pTaskInfo->streamInfo.lastStatus.type = TMQ_OFFSET__SNAPSHOT_META;
      pTaskInfo->streamInfo.lastStatus.uid = uid;
      pTaskInfo->streamInfo.metaRsp.rspOffset = pTaskInfo->streamInfo.lastStatus;
      pTaskInfo->streamInfo.metaRsp.resMsgType = type;
      pTaskInfo->streamInfo.metaRsp.metaRspLen = dataLen;
      pTaskInfo->streamInfo.metaRsp.metaRsp = data;
    }
2238

wmmhello's avatar
wmmhello 已提交
2239
    return NULL;
2240
  }
L
Liu Jicong 已提交
2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278
  //  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;
  //      }
  //    }
2279 2280 2281
  return NULL;
}

wmmhello's avatar
wmmhello 已提交
2282
static void destroyRawScanOperatorInfo(void* param) {
wmmhello's avatar
wmmhello 已提交
2283 2284 2285 2286 2287 2288
  SStreamRawScanInfo* pRawScan = (SStreamRawScanInfo*)param;
  tsdbReaderClose(pRawScan->dataReader);
  destroySnapContext(pRawScan->sContext);
  taosMemoryFree(pRawScan);
}

L
Liu Jicong 已提交
2289 2290 2291
// for subscribing db or stb (not including column),
// if this scan is used, meta data can be return
// and schemas are decided when scanning
2292
SOperatorInfo* createRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
2293 2294 2295 2296 2297
  // create operator
  // create tb reader
  // create meta reader
  // create tq reader

H
Haojun Liao 已提交
2298 2299
  int32_t code = TSDB_CODE_SUCCESS;

2300
  SStreamRawScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamRawScanInfo));
L
Liu Jicong 已提交
2301
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
2302
  if (pInfo == NULL || pOperator == NULL) {
H
Haojun Liao 已提交
2303 2304
    code = TSDB_CODE_OUT_OF_MEMORY;
    goto _end;
2305 2306
  }

wmmhello's avatar
wmmhello 已提交
2307 2308
  pInfo->vnode = pHandle->vnode;

2309
  pInfo->sContext = pHandle->sContext;
2310
  pOperator->name = "RawScanOperator";
2311 2312 2313
  pOperator->info = pInfo;
  pOperator->pTaskInfo = pTaskInfo;

2314
  pOperator->fpSet = createOperatorFpSet(NULL, doRawScan, NULL, NULL, destroyRawScanOperatorInfo, NULL);
2315
  return pOperator;
H
Haojun Liao 已提交
2316

L
Liu Jicong 已提交
2317
_end:
H
Haojun Liao 已提交
2318 2319 2320 2321
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  pTaskInfo->code = code;
  return NULL;
L
Liu Jicong 已提交
2322 2323
}

2324
static void destroyStreamScanOperatorInfo(void* param) {
2325 2326 2327
  SStreamScanInfo* pStreamScan = (SStreamScanInfo*)param;
  if (pStreamScan->pTableScanOp && pStreamScan->pTableScanOp->info) {
    STableScanInfo* pTableScanInfo = pStreamScan->pTableScanOp->info;
2328
    destroyTableScanOperatorInfo(pTableScanInfo);
5
54liuyao 已提交
2329
    taosMemoryFreeClear(pStreamScan->pTableScanOp);
2330 2331 2332 2333
  }
  if (pStreamScan->tqReader) {
    tqCloseReader(pStreamScan->tqReader);
  }
H
Haojun Liao 已提交
2334 2335
  if (pStreamScan->matchInfo.pList) {
    taosArrayDestroy(pStreamScan->matchInfo.pList);
2336
  }
C
Cary Xu 已提交
2337 2338
  if (pStreamScan->pPseudoExpr) {
    destroyExprInfo(pStreamScan->pPseudoExpr, pStreamScan->numOfPseudoExpr);
L
Liu Jicong 已提交
2339
    taosMemoryFree(pStreamScan->pPseudoExpr);
C
Cary Xu 已提交
2340
  }
C
Cary Xu 已提交
2341

L
Liu Jicong 已提交
2342
  updateInfoDestroy(pStreamScan->pUpdateInfo);
2343 2344 2345 2346
  blockDataDestroy(pStreamScan->pRes);
  blockDataDestroy(pStreamScan->pUpdateRes);
  blockDataDestroy(pStreamScan->pPullDataRes);
  blockDataDestroy(pStreamScan->pDeleteDataRes);
5
54liuyao 已提交
2347
  blockDataDestroy(pStreamScan->pUpdateDataRes);
2348 2349 2350 2351
  taosArrayDestroy(pStreamScan->pBlockLists);
  taosMemoryFree(pStreamScan);
}

2352
SOperatorInfo* createStreamScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode, SNode* pTagCond,
2353
                                            SExecTaskInfo* pTaskInfo) {
2354 2355
  SStreamScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamScanInfo));
  SOperatorInfo*   pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
2356

H
Haojun Liao 已提交
2357 2358
  if (pInfo == NULL || pOperator == NULL) {
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
2359
    goto _error;
H
Haojun Liao 已提交
2360 2361
  }

2362
  SScanPhysiNode*     pScanPhyNode = &pTableScanNode->scan;
2363
  SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
H
Haojun Liao 已提交
2364

2365
  pInfo->pTagCond = pTagCond;
2366
  pInfo->pGroupTags = pTableScanNode->pGroupTags;
2367

2368
  int32_t numOfCols = 0;
2369 2370
  int32_t code =
      extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
H
Haojun Liao 已提交
2371 2372 2373
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
2374

H
Haojun Liao 已提交
2375
  int32_t numOfOutput = taosArrayGetSize(pInfo->matchInfo.pList);
2376
  SArray* pColIds = taosArrayInit(numOfOutput, sizeof(int16_t));
2377
  for (int32_t i = 0; i < numOfOutput; ++i) {
H
Haojun Liao 已提交
2378
    SColMatchItem* id = taosArrayGet(pInfo->matchInfo.pList, i);
2379 2380

    int16_t colId = id->colId;
2381
    taosArrayPush(pColIds, &colId);
2382
    if (id->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
H
Haojun Liao 已提交
2383
      pInfo->primaryTsIndex = id->dstSlotId;
5
54liuyao 已提交
2384
    }
H
Haojun Liao 已提交
2385 2386
  }

L
Liu Jicong 已提交
2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399
  if (pTableScanNode->pSubtable != NULL) {
    SExprInfo* pSubTableExpr = taosMemoryCalloc(1, sizeof(SExprInfo));
    if (pSubTableExpr == NULL) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      goto _error;
    }
    pInfo->tbnameCalSup.pExprInfo = pSubTableExpr;
    createExprFromOneNode(pSubTableExpr, pTableScanNode->pSubtable, 0);
    if (initExprSupp(&pInfo->tbnameCalSup, pSubTableExpr, 1) != 0) {
      goto _error;
    }
  }

2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412
  if (pTableScanNode->pTags != NULL) {
    int32_t    numOfTags;
    SExprInfo* pTagExpr = createExprInfo(pTableScanNode->pTags, NULL, &numOfTags);
    if (pTagExpr == NULL) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      goto _error;
    }
    if (initExprSupp(&pInfo->tagCalSup, pTagExpr, numOfTags) != 0) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      goto _error;
    }
  }

H
Haojun Liao 已提交
2413 2414
  pInfo->pBlockLists = taosArrayInit(4, POINTER_BYTES);
  if (pInfo->pBlockLists == NULL) {
2415 2416
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    goto _error;
H
Haojun Liao 已提交
2417 2418
  }

5
54liuyao 已提交
2419
  if (pHandle->vnode) {
L
Liu Jicong 已提交
2420
    SOperatorInfo*  pTableScanOp = createTableScanOperatorInfo(pTableScanNode, pHandle, pTaskInfo);
L
Liu Jicong 已提交
2421
    STableScanInfo* pTSInfo = (STableScanInfo*)pTableScanOp->info;
2422
    if (pHandle->version > 0) {
L
Liu Jicong 已提交
2423
      pTSInfo->cond.endVersion = pHandle->version;
2424
    }
L
Liu Jicong 已提交
2425

2426
    STableKeyInfo* pList = NULL;
5
54liuyao 已提交
2427
    int32_t        num = 0;
H
Haojun Liao 已提交
2428
    tableListGetGroupList(pTaskInfo->pTableInfoList, 0, &pList, &num);
2429

2430
    if (pHandle->initTableReader) {
L
Liu Jicong 已提交
2431 2432
      pTSInfo->scanMode = TABLE_SCAN__TABLE_ORDER;
      pTSInfo->dataReader = NULL;
dengyihao's avatar
dengyihao 已提交
2433 2434 2435
      int32_t code = tsdbReaderOpen(pHandle->vnode, &pTSInfo->cond, pList, num, &pTSInfo->dataReader, NULL);
      if (code != 0) {
        terrno = code;
H
Haojun Liao 已提交
2436
        destroyTableScanOperatorInfo(pTableScanOp);
2437
        goto _error;
L
Liu Jicong 已提交
2438
      }
L
Liu Jicong 已提交
2439 2440
    }

L
Liu Jicong 已提交
2441 2442 2443 2444
    if (pHandle->initTqReader) {
      ASSERT(pHandle->tqReader == NULL);
      pInfo->tqReader = tqOpenReader(pHandle->vnode);
      ASSERT(pInfo->tqReader);
2445
    } else {
L
Liu Jicong 已提交
2446 2447
      ASSERT(pHandle->tqReader);
      pInfo->tqReader = pHandle->tqReader;
2448 2449
    }

2450
    pInfo->pUpdateInfo = NULL;
2451
    pInfo->pTableScanOp = pTableScanOp;
2452 2453 2454
    if (pInfo->pTableScanOp->pTaskInfo->streamInfo.pState) {
      streamStateSetNumber(pInfo->pTableScanOp->pTaskInfo->streamInfo.pState, -1);
    }
L
Liu Jicong 已提交
2455

L
Liu Jicong 已提交
2456 2457
    pInfo->readHandle = *pHandle;
    pInfo->tableUid = pScanPhyNode->uid;
L
Liu Jicong 已提交
2458
    pTaskInfo->streamInfo.snapshotVer = pHandle->version;
L
Liu Jicong 已提交
2459

L
Liu Jicong 已提交
2460
    // set the extract column id to streamHandle
L
Liu Jicong 已提交
2461
    tqReaderSetColIdList(pInfo->tqReader, pColIds);
H
Haojun Liao 已提交
2462
    SArray* tableIdList = extractTableIdList(pTaskInfo->pTableInfoList);
2463
    code = tqReaderSetTbUidList(pInfo->tqReader, tableIdList);
L
Liu Jicong 已提交
2464 2465 2466 2467 2468
    if (code != 0) {
      taosArrayDestroy(tableIdList);
      goto _error;
    }
    taosArrayDestroy(tableIdList);
L
Liu Jicong 已提交
2469
    memcpy(&pTaskInfo->streamInfo.tableCond, &pTSInfo->cond, sizeof(SQueryTableDataCond));
L
Liu Jicong 已提交
2470 2471
  } else {
    taosArrayDestroy(pColIds);
5
54liuyao 已提交
2472 2473
  }

2474 2475 2476 2477 2478
  // create the pseduo columns info
  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
    pInfo->pPseudoExpr = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pInfo->numOfPseudoExpr);
  }

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

2494
  pOperator->name = "StreamScanOperator";
L
Liu Jicong 已提交
2495
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN;
2496 2497 2498
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
2499
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
2500
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
2501

L
Liu Jicong 已提交
2502
  __optr_fn_t nextFn = pTaskInfo->execModel == OPTR_EXEC_MODEL_STREAM ? doStreamScan : doQueueScan;
2503
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, nextFn, NULL, NULL, destroyStreamScanOperatorInfo, NULL);
2504

H
Haojun Liao 已提交
2505
  return pOperator;
2506

L
Liu Jicong 已提交
2507
_error:
H
Haojun Liao 已提交
2508 2509 2510 2511 2512 2513 2514 2515
  if (pColIds != NULL) {
    taosArrayDestroy(pColIds);
  }

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

2516 2517
  taosMemoryFreeClear(pOperator);
  return NULL;
H
Haojun Liao 已提交
2518 2519
}

2520
static void destroySysScanOperator(void* param) {
H
Haojun Liao 已提交
2521 2522 2523 2524
  SSysTableScanInfo* pInfo = (SSysTableScanInfo*)param;
  tsem_destroy(&pInfo->ready);
  blockDataDestroy(pInfo->pRes);

2525
  const char* name = tNameGetTableName(&pInfo->name);
D
dapan1121 已提交
2526 2527
  if (strncasecmp(name, TSDB_INS_TABLE_TABLES, TSDB_TABLE_FNAME_LEN) == 0 ||
      strncasecmp(name, TSDB_INS_TABLE_TAGS, TSDB_TABLE_FNAME_LEN) == 0 || pInfo->pCur != NULL) {
H
Haojun Liao 已提交
2528
    metaCloseTbCursor(pInfo->pCur);
2529
    pInfo->pCur = NULL;
H
Haojun Liao 已提交
2530
  }
dengyihao's avatar
dengyihao 已提交
2531 2532 2533 2534 2535
  if (pInfo->pIdx) {
    taosArrayDestroy(pInfo->pIdx->uids);
    taosMemoryFree(pInfo->pIdx);
    pInfo->pIdx = NULL;
  }
H
Haojun Liao 已提交
2536

H
Haojun Liao 已提交
2537
  taosArrayDestroy(pInfo->matchInfo.pList);
2538
  taosMemoryFreeClear(pInfo->pUser);
D
dapan1121 已提交
2539 2540

  taosMemoryFreeClear(param);
H
Haojun Liao 已提交
2541 2542
}

X
Xiaoyu Wang 已提交
2543
static int32_t getSysTableDbNameColId(const char* pTable) {
D
dapan1121 已提交
2544
  // if (0 == strcmp(TSDB_INS_TABLE_INDEXES, pTable)) {
X
Xiaoyu Wang 已提交
2545 2546
  //   return 1;
  // }
X
Xiaoyu Wang 已提交
2547 2548 2549
  return TSDB_INS_USER_STABLES_DBNAME_COLID;
}

H
Haojun Liao 已提交
2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570
EDealRes getDBNameFromConditionWalker(SNode* pNode, void* pContext) {
  int32_t   code = TSDB_CODE_SUCCESS;
  ENodeType nType = nodeType(pNode);

  switch (nType) {
    case QUERY_NODE_OPERATOR: {
      SOperatorNode* node = (SOperatorNode*)pNode;
      if (OP_TYPE_EQUAL == node->opType) {
        *(int32_t*)pContext = 1;
        return DEAL_RES_CONTINUE;
      }

      *(int32_t*)pContext = 0;
      return DEAL_RES_IGNORE_CHILD;
    }
    case QUERY_NODE_COLUMN: {
      if (1 != *(int32_t*)pContext) {
        return DEAL_RES_CONTINUE;
      }

      SColumnNode* node = (SColumnNode*)pNode;
X
Xiaoyu Wang 已提交
2571
      if (getSysTableDbNameColId(node->tableName) == node->colId) {
H
Haojun Liao 已提交
2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587
        *(int32_t*)pContext = 2;
        return DEAL_RES_CONTINUE;
      }

      *(int32_t*)pContext = 0;
      return DEAL_RES_CONTINUE;
    }
    case QUERY_NODE_VALUE: {
      if (2 != *(int32_t*)pContext) {
        return DEAL_RES_CONTINUE;
      }

      SValueNode* node = (SValueNode*)pNode;
      char*       dbName = nodesGetValueFromNode(node);
      strncpy(pContext, varDataVal(dbName), varDataLen(dbName));
      *((char*)pContext + varDataLen(dbName)) = 0;
2588
      return DEAL_RES_END;  // stop walk
H
Haojun Liao 已提交
2589 2590 2591 2592 2593 2594 2595
    }
    default:
      break;
  }
  return DEAL_RES_CONTINUE;
}

2596
static void getDBNameFromCondition(SNode* pCondition, const char* dbName) {
H
Haojun Liao 已提交
2597 2598 2599
  if (NULL == pCondition) {
    return;
  }
L
Liu Jicong 已提交
2600
  nodesWalkExpr(pCondition, getDBNameFromConditionWalker, (char*)dbName);
H
Haojun Liao 已提交
2601 2602
}

D
dapan1121 已提交
2603
static int32_t loadSysTableCallback(void* param, SDataBuf* pMsg, int32_t code) {
H
Haojun Liao 已提交
2604 2605 2606 2607 2608 2609 2610
  SOperatorInfo*     operator=(SOperatorInfo*) param;
  SSysTableScanInfo* pScanResInfo = (SSysTableScanInfo*)operator->info;
  if (TSDB_CODE_SUCCESS == code) {
    pScanResInfo->pRsp = pMsg->pData;

    SRetrieveMetaTableRsp* pRsp = pScanResInfo->pRsp;
    pRsp->numOfRows = htonl(pRsp->numOfRows);
2611 2612 2613
    pRsp->useconds = htobe64(pRsp->useconds);
    pRsp->handle = htobe64(pRsp->handle);
    pRsp->compLen = htonl(pRsp->compLen);
H
Haojun Liao 已提交
2614 2615 2616 2617 2618
  } else {
    operator->pTaskInfo->code = code;
  }

  tsem_post(&pScanResInfo->ready);
wmmhello's avatar
wmmhello 已提交
2619
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
2620 2621 2622 2623 2624 2625 2626
}

static SSDataBlock* doFilterResult(SSysTableScanInfo* pInfo) {
  if (pInfo->pCondition == NULL) {
    return pInfo->pRes->info.rows == 0 ? NULL : pInfo->pRes;
  }

H
Haojun Liao 已提交
2627
  doFilter(pInfo->pCondition, pInfo->pRes, NULL, NULL);
H
Haojun Liao 已提交
2628 2629 2630
  return pInfo->pRes->info.rows == 0 ? NULL : pInfo->pRes;
}

2631
static SSDataBlock* buildInfoSchemaTableMetaBlock(char* tableName) {
L
Liu Jicong 已提交
2632 2633
  size_t               size = 0;
  const SSysTableMeta* pMeta = NULL;
2634 2635 2636
  getInfosDbMeta(&pMeta, &size);

  int32_t index = 0;
L
Liu Jicong 已提交
2637
  for (int32_t i = 0; i < size; ++i) {
2638
    if (strcmp(pMeta[i].name, tableName) == 0) {
2639 2640 2641 2642
      index = i;
      break;
    }
  }
2643

2644
  SSDataBlock* pBlock = createDataBlock();
L
Liu Jicong 已提交
2645
  for (int32_t i = 0; i < pMeta[index].colNum; ++i) {
L
Liu Jicong 已提交
2646 2647
    SColumnInfoData colInfoData =
        createColumnInfoData(pMeta[index].schema[i].type, pMeta[index].schema[i].bytes, i + 1);
2648
    blockDataAppendColInfo(pBlock, &colInfoData);
2649 2650
  }

2651 2652 2653
  return pBlock;
}

2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731 2732 2733 2734
int32_t convertTagDataToStr(char* str, int type, void* buf, int32_t bufSize, int32_t* len) {
  int32_t n = 0;

  switch (type) {
    case TSDB_DATA_TYPE_NULL:
      n = sprintf(str, "null");
      break;

    case TSDB_DATA_TYPE_BOOL:
      n = sprintf(str, (*(int8_t*)buf) ? "true" : "false");
      break;

    case TSDB_DATA_TYPE_TINYINT:
      n = sprintf(str, "%d", *(int8_t*)buf);
      break;

    case TSDB_DATA_TYPE_SMALLINT:
      n = sprintf(str, "%d", *(int16_t*)buf);
      break;

    case TSDB_DATA_TYPE_INT:
      n = sprintf(str, "%d", *(int32_t*)buf);
      break;

    case TSDB_DATA_TYPE_BIGINT:
    case TSDB_DATA_TYPE_TIMESTAMP:
      n = sprintf(str, "%" PRId64, *(int64_t*)buf);
      break;

    case TSDB_DATA_TYPE_FLOAT:
      n = sprintf(str, "%.5f", GET_FLOAT_VAL(buf));
      break;

    case TSDB_DATA_TYPE_DOUBLE:
      n = sprintf(str, "%.9f", GET_DOUBLE_VAL(buf));
      break;

    case TSDB_DATA_TYPE_BINARY:
      if (bufSize < 0) {
        return TSDB_CODE_TSC_INVALID_VALUE;
      }

      memcpy(str, buf, bufSize);
      n = bufSize;
      break;
    case TSDB_DATA_TYPE_NCHAR:
      if (bufSize < 0) {
        return TSDB_CODE_TSC_INVALID_VALUE;
      }

      int32_t length = taosUcs4ToMbs((TdUcs4*)buf, bufSize, str);
      if (length <= 0) {
        return TSDB_CODE_TSC_INVALID_VALUE;
      }
      n = length;
      break;
    case TSDB_DATA_TYPE_UTINYINT:
      n = sprintf(str, "%u", *(uint8_t*)buf);
      break;

    case TSDB_DATA_TYPE_USMALLINT:
      n = sprintf(str, "%u", *(uint16_t*)buf);
      break;

    case TSDB_DATA_TYPE_UINT:
      n = sprintf(str, "%u", *(uint32_t*)buf);
      break;

    case TSDB_DATA_TYPE_UBIGINT:
      n = sprintf(str, "%" PRIu64, *(uint64_t*)buf);
      break;

    default:
      return TSDB_CODE_TSC_INVALID_VALUE;
  }

  if (len) *len = n;

  return TSDB_CODE_SUCCESS;
}

2735 2736 2737 2738 2739 2740 2741
static bool sysTableIsOperatorCondOnOneTable(SNode* pCond, char* condTable) {
  SOperatorNode* node = (SOperatorNode*)pCond;
  if (node->opType == OP_TYPE_EQUAL) {
    if (nodeType(node->pLeft) == QUERY_NODE_COLUMN &&
        strcasecmp(nodesGetNameFromColumnNode(node->pLeft), "table_name") == 0 &&
        nodeType(node->pRight) == QUERY_NODE_VALUE) {
      SValueNode* pValue = (SValueNode*)node->pRight;
2742 2743 2744 2745
      if (pValue->node.resType.type == TSDB_DATA_TYPE_NCHAR || pValue->node.resType.type == TSDB_DATA_TYPE_VARCHAR ||
          pValue->node.resType.type == TSDB_DATA_TYPE_BINARY) {
        char* value = nodesGetValueFromNode(pValue);
        strncpy(condTable, varDataVal(value), TSDB_TABLE_NAME_LEN);
2746 2747 2748 2749 2750 2751 2752 2753
        return true;
      }
    }
  }
  return false;
}

static bool sysTableIsCondOnOneTable(SNode* pCond, char* condTable) {
S
slzhou 已提交
2754 2755 2756
  if (pCond == NULL) {
    return false;
  }
2757 2758 2759
  if (nodeType(pCond) == QUERY_NODE_LOGIC_CONDITION) {
    SLogicConditionNode* node = (SLogicConditionNode*)pCond;
    if (LOGIC_COND_TYPE_AND == node->condType) {
S
slzhou 已提交
2760 2761 2762 2763
      SNode* pChild = NULL;
      FOREACH(pChild, node->pParameterList) {
        if (QUERY_NODE_OPERATOR == nodeType(pChild) && sysTableIsOperatorCondOnOneTable(pChild, condTable)) {
          return true;
2764 2765 2766 2767
        }
      }
    }
  }
S
slzhou 已提交
2768

2769 2770 2771
  if (QUERY_NODE_OPERATOR == nodeType(pCond)) {
    return sysTableIsOperatorCondOnOneTable(pCond, condTable);
  }
S
slzhou 已提交
2772

2773 2774 2775
  return false;
}

S
shenglian zhou 已提交
2776 2777 2778 2779 2780 2781 2782 2783 2784 2785
static SSDataBlock* sysTableScanUserTags(SOperatorInfo* pOperator) {
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
  SSysTableScanInfo* pInfo = pOperator->info;
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  blockDataCleanup(pInfo->pRes);
  int32_t numOfRows = 0;

2786 2787 2788
  SSDataBlock* dataBlock = buildInfoSchemaTableMetaBlock(TSDB_INS_TABLE_TAGS);
  blockDataEnsureCapacity(dataBlock, pOperator->resultInfo.capacity);

S
shenglian zhou 已提交
2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799
  const char* db = NULL;
  int32_t     vgId = 0;
  vnodeGetInfo(pInfo->readHandle.vnode, &db, &vgId);

  SName sn = {0};
  char  dbname[TSDB_DB_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
  tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);

  tNameGetDbName(&sn, varDataVal(dbname));
  varDataSetLen(dbname, strlen(varDataVal(dbname)));

2800
  char condTableName[TSDB_TABLE_NAME_LEN] = {0};
S
slzhou 已提交
2801 2802
  // optimize when sql like where table_name='tablename' and xxx.
  if (pInfo->pCondition && sysTableIsCondOnOneTable(pInfo->pCondition, condTableName)) {
2803 2804 2805
    char tableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
    STR_TO_VARSTR(tableName, condTableName);

2806 2807
    SMetaReader smrChildTable = {0};
    metaReaderInit(&smrChildTable, pInfo->readHandle.meta, 0);
H
Haojun Liao 已提交
2808 2809 2810 2811 2812 2813
    int32_t code = metaGetTableEntryByName(&smrChildTable, condTableName);
    if (code != TSDB_CODE_SUCCESS) {
      // terrno has been set by metaGetTableEntryByName, therefore, return directly
      return NULL;
    }

2814 2815 2816 2817 2818 2819
    if (smrChildTable.me.type != TSDB_CHILD_TABLE) {
      metaReaderClear(&smrChildTable);
      blockDataDestroy(dataBlock);
      pInfo->loadInfo.totalRows = 0;
      return NULL;
    }
H
Haojun Liao 已提交
2820

2821
    SMetaReader smrSuperTable = {0};
2822
    metaReaderInit(&smrSuperTable, pInfo->readHandle.meta, META_READER_NOLOCK);
H
Haojun Liao 已提交
2823 2824 2825 2826 2827 2828
    code = metaGetTableEntryByUid(&smrSuperTable, smrChildTable.me.ctbEntry.suid);
    if (code != TSDB_CODE_SUCCESS) {
      // terrno has been set by metaGetTableEntryByUid
      return NULL;
    }

2829 2830 2831
    sysTableUserTagsFillOneTableTags(pInfo, &smrSuperTable, &smrChildTable, dbname, tableName, &numOfRows, dataBlock);
    metaReaderClear(&smrSuperTable);
    metaReaderClear(&smrChildTable);
2832 2833 2834 2835 2836 2837
    if (numOfRows > 0) {
      relocateAndFilterSysTagsScanResult(pInfo, numOfRows, dataBlock);
      numOfRows = 0;
    }
    blockDataDestroy(dataBlock);
    pInfo->loadInfo.totalRows += pInfo->pRes->info.rows;
2838
    doSetOperatorCompleted(pOperator);
2839 2840
    return (pInfo->pRes->info.rows == 0) ? NULL : pInfo->pRes;
  }
S
shenglian zhou 已提交
2841 2842

  int32_t ret = 0;
2843 2844 2845 2846
  if (pInfo->pCur == NULL) {
    pInfo->pCur = metaOpenTbCursor(pInfo->readHandle.meta);
  }

S
shenglian zhou 已提交
2847
  while ((ret = metaTbCursorNext(pInfo->pCur)) == 0) {
2848 2849 2850
    if (pInfo->pCur->mr.me.type != TSDB_CHILD_TABLE) {
      continue;
    }
S
shenglian zhou 已提交
2851

2852 2853
    char tableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
    STR_TO_VARSTR(tableName, pInfo->pCur->mr.me.name);
S
shenglian zhou 已提交
2854

2855 2856
    SMetaReader smrSuperTable = {0};
    metaReaderInit(&smrSuperTable, pInfo->readHandle.meta, 0);
2857
    uint64_t suid = pInfo->pCur->mr.me.ctbEntry.suid;
2858
    int32_t  code = metaGetTableEntryByUid(&smrSuperTable, suid);
2859 2860 2861
    if (code != TSDB_CODE_SUCCESS) {
      qError("failed to get super table meta, uid:0x%" PRIx64 ", code:%s, %s", suid, tstrerror(terrno),
             GET_TASKID(pTaskInfo));
2862
      metaReaderClear(&smrSuperTable);
2863 2864
      metaCloseTbCursor(pInfo->pCur);
      pInfo->pCur = NULL;
2865
      T_LONG_JMP(pTaskInfo->env, terrno);
2866
    }
S
shenglian zhou 已提交
2867

2868
    sysTableUserTagsFillOneTableTags(pInfo, &smrSuperTable, &pInfo->pCur->mr, dbname, tableName, &numOfRows, dataBlock);
2869

2870
    metaReaderClear(&smrSuperTable);
S
shenglian zhou 已提交
2871

2872
    if (numOfRows >= pOperator->resultInfo.capacity) {
2873
      relocateAndFilterSysTagsScanResult(pInfo, numOfRows, dataBlock);
2874 2875 2876 2877 2878
      numOfRows = 0;

      if (pInfo->pRes->info.rows > 0) {
        break;
      }
S
shenglian zhou 已提交
2879 2880 2881
    }
  }

2882
  if (numOfRows > 0) {
2883
    relocateAndFilterSysTagsScanResult(pInfo, numOfRows, dataBlock);
2884 2885 2886
    numOfRows = 0;
  }

2887
  blockDataDestroy(dataBlock);
S
shenglian zhou 已提交
2888 2889 2890 2891 2892 2893 2894 2895 2896 2897
  if (ret != 0) {
    metaCloseTbCursor(pInfo->pCur);
    pInfo->pCur = NULL;
    doSetOperatorCompleted(pOperator);
  }

  pInfo->loadInfo.totalRows += pInfo->pRes->info.rows;
  return (pInfo->pRes->info.rows == 0) ? NULL : pInfo->pRes;
}

2898 2899 2900 2901
static void relocateAndFilterSysTagsScanResult(SSysTableScanInfo* pInfo, int32_t numOfRows, SSDataBlock* dataBlock) {
  dataBlock->info.rows = numOfRows;
  pInfo->pRes->info.rows = numOfRows;

H
Haojun Liao 已提交
2902
  relocateColumnData(pInfo->pRes, pInfo->matchInfo.pList, dataBlock->pDataBlock, false);
2903 2904 2905 2906 2907
  doFilterResult(pInfo);

  blockDataCleanup(dataBlock);
}

2908 2909 2910
static int32_t sysTableUserTagsFillOneTableTags(const SSysTableScanInfo* pInfo, SMetaReader* smrSuperTable,
                                                SMetaReader* smrChildTable, const char* dbname, const char* tableName,
                                                int32_t* pNumOfRows, const SSDataBlock* dataBlock) {
2911
  char stableName[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
2912
  STR_TO_VARSTR(stableName, (*smrSuperTable).me.name);
2913 2914 2915

  int32_t numOfRows = *pNumOfRows;

2916
  int32_t numOfTags = (*smrSuperTable).me.stbEntry.schemaTag.nCols;
2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933
  for (int32_t i = 0; i < numOfTags; ++i) {
    SColumnInfoData* pColInfoData = NULL;

    // table name
    pColInfoData = taosArrayGet(dataBlock->pDataBlock, 0);
    colDataAppend(pColInfoData, numOfRows, tableName, false);

    // database name
    pColInfoData = taosArrayGet(dataBlock->pDataBlock, 1);
    colDataAppend(pColInfoData, numOfRows, dbname, false);

    // super table name
    pColInfoData = taosArrayGet(dataBlock->pDataBlock, 2);
    colDataAppend(pColInfoData, numOfRows, stableName, false);

    // tag name
    char tagName[TSDB_COL_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
2934
    STR_TO_VARSTR(tagName, (*smrSuperTable).me.stbEntry.schemaTag.pSchema[i].name);
2935 2936 2937 2938
    pColInfoData = taosArrayGet(dataBlock->pDataBlock, 3);
    colDataAppend(pColInfoData, numOfRows, tagName, false);

    // tag type
2939
    int8_t tagType = (*smrSuperTable).me.stbEntry.schemaTag.pSchema[i].type;
2940 2941 2942 2943 2944
    pColInfoData = taosArrayGet(dataBlock->pDataBlock, 4);
    char tagTypeStr[VARSTR_HEADER_SIZE + 32];
    int  tagTypeLen = sprintf(varDataVal(tagTypeStr), "%s", tDataTypes[tagType].name);
    if (tagType == TSDB_DATA_TYPE_VARCHAR) {
      tagTypeLen += sprintf(varDataVal(tagTypeStr) + tagTypeLen, "(%d)",
2945
                            (int32_t)((*smrSuperTable).me.stbEntry.schemaTag.pSchema[i].bytes - VARSTR_HEADER_SIZE));
2946
    } else if (tagType == TSDB_DATA_TYPE_NCHAR) {
2947 2948 2949
      tagTypeLen += sprintf(
          varDataVal(tagTypeStr) + tagTypeLen, "(%d)",
          (int32_t)(((*smrSuperTable).me.stbEntry.schemaTag.pSchema[i].bytes - VARSTR_HEADER_SIZE) / TSDB_NCHAR_SIZE));
2950 2951 2952 2953 2954
    }
    varDataSetLen(tagTypeStr, tagTypeLen);
    colDataAppend(pColInfoData, numOfRows, (char*)tagTypeStr, false);

    STagVal tagVal = {0};
2955
    tagVal.cid = (*smrSuperTable).me.stbEntry.schemaTag.pSchema[i].colId;
2956 2957 2958 2959
    char*    tagData = NULL;
    uint32_t tagLen = 0;

    if (tagType == TSDB_DATA_TYPE_JSON) {
2960
      tagData = (char*)smrChildTable->me.ctbEntry.pTags;
2961
    } else {
2962
      bool exist = tTagGet((STag*)smrChildTable->me.ctbEntry.pTags, &tagVal);
2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002
      if (exist) {
        if (IS_VAR_DATA_TYPE(tagType)) {
          tagData = (char*)tagVal.pData;
          tagLen = tagVal.nData;
        } else {
          tagData = (char*)&tagVal.i64;
          tagLen = tDataTypes[tagType].bytes;
        }
      }
    }

    char* tagVarChar = NULL;
    if (tagData != NULL) {
      if (tagType == TSDB_DATA_TYPE_JSON) {
        char* tagJson = parseTagDatatoJson(tagData);
        tagVarChar = taosMemoryMalloc(strlen(tagJson) + VARSTR_HEADER_SIZE);
        memcpy(varDataVal(tagVarChar), tagJson, strlen(tagJson));
        varDataSetLen(tagVarChar, strlen(tagJson));
        taosMemoryFree(tagJson);
      } else {
        int32_t bufSize = IS_VAR_DATA_TYPE(tagType) ? (tagLen + VARSTR_HEADER_SIZE)
                                                    : (3 + DBL_MANT_DIG - DBL_MIN_EXP + VARSTR_HEADER_SIZE);
        tagVarChar = taosMemoryMalloc(bufSize);
        int32_t len = -1;
        convertTagDataToStr(varDataVal(tagVarChar), tagType, tagData, tagLen, &len);
        varDataSetLen(tagVarChar, len);
      }
    }
    pColInfoData = taosArrayGet(dataBlock->pDataBlock, 5);
    colDataAppend(pColInfoData, numOfRows, tagVarChar,
                  (tagData == NULL) || (tagType == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(tagData)));
    taosMemoryFree(tagVarChar);
    ++numOfRows;
  }

  *pNumOfRows = numOfRows;

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028
typedef int (*__optSysFilter)(void* a, void* b, int16_t dtype);

int optSysDoCompare(__compar_fn_t func, int8_t comparType, void* a, void* b) {
  int32_t cmp = func(a, b);
  switch (comparType) {
    case OP_TYPE_LOWER_THAN:
      if (cmp < 0) return 0;
      break;
    case OP_TYPE_LOWER_EQUAL: {
      if (cmp <= 0) return 0;
      break;
    }
    case OP_TYPE_GREATER_THAN: {
      if (cmp > 0) return 0;
      break;
    }
    case OP_TYPE_GREATER_EQUAL: {
      if (cmp >= 0) return 0;
      break;
    }
    case OP_TYPE_EQUAL: {
      if (cmp == 0) return 0;
      break;
    }
    default:
      return -1;
3029
  }
dengyihao's avatar
dengyihao 已提交
3030
  return cmp;
dengyihao's avatar
dengyihao 已提交
3031
}
H
Haojun Liao 已提交
3032

dengyihao's avatar
dengyihao 已提交
3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052
static int optSysFilterFuncImpl__LowerThan(void* a, void* b, int16_t dtype) {
  __compar_fn_t func = getComparFunc(dtype, 0);
  return optSysDoCompare(func, OP_TYPE_LOWER_THAN, a, b);
}
static int optSysFilterFuncImpl__LowerEqual(void* a, void* b, int16_t dtype) {
  __compar_fn_t func = getComparFunc(dtype, 0);
  return optSysDoCompare(func, OP_TYPE_LOWER_EQUAL, a, b);
}
static int optSysFilterFuncImpl__GreaterThan(void* a, void* b, int16_t dtype) {
  __compar_fn_t func = getComparFunc(dtype, 0);
  return optSysDoCompare(func, OP_TYPE_GREATER_THAN, a, b);
}
static int optSysFilterFuncImpl__GreaterEqual(void* a, void* b, int16_t dtype) {
  __compar_fn_t func = getComparFunc(dtype, 0);
  return optSysDoCompare(func, OP_TYPE_GREATER_EQUAL, a, b);
}
static int optSysFilterFuncImpl__Equal(void* a, void* b, int16_t dtype) {
  __compar_fn_t func = getComparFunc(dtype, 0);
  return optSysDoCompare(func, OP_TYPE_EQUAL, a, b);
}
3053

dengyihao's avatar
dengyihao 已提交
3054 3055 3056 3057
static int optSysFilterFuncImpl__NoEqual(void* a, void* b, int16_t dtype) {
  __compar_fn_t func = getComparFunc(dtype, 0);
  return optSysDoCompare(func, OP_TYPE_NOT_EQUAL, a, b);
}
dengyihao's avatar
dengyihao 已提交
3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071
static __optSysFilter optSysGetFilterFunc(int32_t ctype, bool* reverse) {
  if (ctype == OP_TYPE_LOWER_EQUAL || ctype == OP_TYPE_LOWER_THAN) {
    *reverse = true;
  }
  if (ctype == OP_TYPE_LOWER_THAN)
    return optSysFilterFuncImpl__LowerThan;
  else if (ctype == OP_TYPE_LOWER_EQUAL)
    return optSysFilterFuncImpl__LowerEqual;
  else if (ctype == OP_TYPE_GREATER_THAN)
    return optSysFilterFuncImpl__GreaterThan;
  else if (ctype == OP_TYPE_GREATER_EQUAL)
    return optSysFilterFuncImpl__GreaterEqual;
  else if (ctype == OP_TYPE_EQUAL)
    return optSysFilterFuncImpl__Equal;
dengyihao's avatar
dengyihao 已提交
3072 3073
  else if (ctype == OP_TYPE_NOT_EQUAL)
    return optSysFilterFuncImpl__NoEqual;
dengyihao's avatar
dengyihao 已提交
3074 3075
  return NULL;
}
dengyihao's avatar
dengyihao 已提交
3076 3077
static int32_t sysFilte__DbName(void* arg, SNode* pNode, SArray* result) {
  void* pVnode = ((SSTabFltArg*)arg)->pVnode;
3078

dengyihao's avatar
dengyihao 已提交
3079 3080
  const char* db = NULL;
  vnodeGetInfo(pVnode, &db, NULL);
dengyihao's avatar
dengyihao 已提交
3081

dengyihao's avatar
dengyihao 已提交
3082 3083 3084 3085 3086 3087
  SName sn = {0};
  char  dbname[TSDB_DB_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
  tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);

  tNameGetDbName(&sn, varDataVal(dbname));
  varDataSetLen(dbname, strlen(varDataVal(dbname)));
dengyihao's avatar
dengyihao 已提交
3088

dengyihao's avatar
dengyihao 已提交
3089
  SOperatorNode* pOper = (SOperatorNode*)pNode;
dengyihao's avatar
dengyihao 已提交
3090
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
dengyihao's avatar
dengyihao 已提交
3091

dengyihao's avatar
dengyihao 已提交
3092 3093
  bool           reverse = false;
  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
dengyihao's avatar
dengyihao 已提交
3094 3095 3096 3097 3098
  if (func == NULL) return -1;

  int ret = func(dbname, pVal->datum.p, TSDB_DATA_TYPE_VARCHAR);
  if (ret == 0) return 0;

dengyihao's avatar
dengyihao 已提交
3099
  return -2;
dengyihao's avatar
dengyihao 已提交
3100
}
dengyihao's avatar
dengyihao 已提交
3101 3102
static int32_t sysFilte__VgroupId(void* arg, SNode* pNode, SArray* result) {
  void* pVnode = ((SSTabFltArg*)arg)->pVnode;
dengyihao's avatar
dengyihao 已提交
3103

dengyihao's avatar
dengyihao 已提交
3104 3105
  int64_t vgId = 0;
  vnodeGetInfo(pVnode, NULL, (int32_t*)&vgId);
dengyihao's avatar
dengyihao 已提交
3106

dengyihao's avatar
dengyihao 已提交
3107
  SOperatorNode* pOper = (SOperatorNode*)pNode;
dengyihao's avatar
dengyihao 已提交
3108 3109
  SValueNode*    pVal = (SValueNode*)pOper->pRight;

dengyihao's avatar
dengyihao 已提交
3110 3111 3112
  bool reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
dengyihao's avatar
dengyihao 已提交
3113 3114 3115 3116
  if (func == NULL) return -1;

  int ret = func(&vgId, &pVal->datum.i, TSDB_DATA_TYPE_BIGINT);
  if (ret == 0) return 0;
dengyihao's avatar
dengyihao 已提交
3117

dengyihao's avatar
dengyihao 已提交
3118
  return -1;
dengyihao's avatar
dengyihao 已提交
3119
}
dengyihao's avatar
dengyihao 已提交
3120 3121
static int32_t sysFilte__TableName(void* arg, SNode* pNode, SArray* result) {
  void* pMeta = ((SSTabFltArg*)arg)->pMeta;
dengyihao's avatar
dengyihao 已提交
3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135

  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
  bool           reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
  if (func == NULL) return -1;

  SMetaFltParam param = {.suid = 0,
                         .cid = 0,
                         .type = TSDB_DATA_TYPE_VARCHAR,
                         .val = pVal->datum.p,
                         .reverse = reverse,
                         .filterFunc = func};
dengyihao's avatar
dengyihao 已提交
3136
  return -1;
dengyihao's avatar
dengyihao 已提交
3137 3138
}

dengyihao's avatar
dengyihao 已提交
3139 3140
static int32_t sysFilte__CreateTime(void* arg, SNode* pNode, SArray* result) {
  void* pMeta = ((SSTabFltArg*)arg)->pMeta;
dengyihao's avatar
dengyihao 已提交
3141 3142 3143 3144 3145 3146 3147

  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
  bool           reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
  if (func == NULL) return -1;
dengyihao's avatar
dengyihao 已提交
3148 3149 3150 3151 3152 3153 3154 3155 3156 3157

  SMetaFltParam param = {.suid = 0,
                         .cid = 0,
                         .type = TSDB_DATA_TYPE_BIGINT,
                         .val = &pVal->datum.i,
                         .reverse = reverse,
                         .filterFunc = func};

  int32_t ret = metaFilterCreateTime(pMeta, &param, result);
  return ret;
dengyihao's avatar
dengyihao 已提交
3158
}
dengyihao's avatar
dengyihao 已提交
3159 3160
static int32_t sysFilte__Ncolumn(void* arg, SNode* pNode, SArray* result) {
  void* pMeta = ((SSTabFltArg*)arg)->pMeta;
dengyihao's avatar
dengyihao 已提交
3161 3162 3163 3164 3165 3166

  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
  bool           reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
dengyihao's avatar
dengyihao 已提交
3167 3168
  if (func == NULL) return -1;
  return -1;
dengyihao's avatar
dengyihao 已提交
3169 3170
}

dengyihao's avatar
dengyihao 已提交
3171 3172
static int32_t sysFilte__Ttl(void* arg, SNode* pNode, SArray* result) {
  void* pMeta = ((SSTabFltArg*)arg)->pMeta;
dengyihao's avatar
dengyihao 已提交
3173 3174 3175 3176 3177 3178

  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
  bool           reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
dengyihao's avatar
dengyihao 已提交
3179 3180
  if (func == NULL) return -1;
  return -1;
dengyihao's avatar
dengyihao 已提交
3181
}
dengyihao's avatar
dengyihao 已提交
3182 3183
static int32_t sysFilte__STableName(void* arg, SNode* pNode, SArray* result) {
  void* pMeta = ((SSTabFltArg*)arg)->pMeta;
dengyihao's avatar
dengyihao 已提交
3184 3185 3186 3187 3188 3189

  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
  bool           reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
dengyihao's avatar
dengyihao 已提交
3190 3191
  if (func == NULL) return -1;
  return -1;
dengyihao's avatar
dengyihao 已提交
3192
}
dengyihao's avatar
dengyihao 已提交
3193 3194
static int32_t sysFilte__Uid(void* arg, SNode* pNode, SArray* result) {
  void* pMeta = ((SSTabFltArg*)arg)->pMeta;
dengyihao's avatar
dengyihao 已提交
3195 3196 3197 3198 3199 3200

  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
  bool           reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
dengyihao's avatar
dengyihao 已提交
3201 3202
  if (func == NULL) return -1;
  return -1;
dengyihao's avatar
dengyihao 已提交
3203
}
dengyihao's avatar
dengyihao 已提交
3204 3205
static int32_t sysFilte__Type(void* arg, SNode* pNode, SArray* result) {
  void* pMeta = ((SSTabFltArg*)arg)->pMeta;
dengyihao's avatar
dengyihao 已提交
3206 3207 3208 3209 3210 3211

  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
  bool           reverse = false;

  __optSysFilter func = optSysGetFilterFunc(pOper->opType, &reverse);
dengyihao's avatar
dengyihao 已提交
3212
  if (func == NULL) return -1;
dengyihao's avatar
dengyihao 已提交
3213
  return -1;
dengyihao's avatar
dengyihao 已提交
3214
}
dengyihao's avatar
dengyihao 已提交
3215 3216 3217 3218 3219
static int32_t sysChkFilter__Comm(SNode* pNode) {
  // impl
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  EOperatorType  opType = pOper->opType;
  if (opType != OP_TYPE_EQUAL && opType != OP_TYPE_LOWER_EQUAL && opType != OP_TYPE_LOWER_THAN &&
dengyihao's avatar
dengyihao 已提交
3220
      opType != OP_TYPE_GREATER_EQUAL && opType != OP_TYPE_GREATER_THAN) {
dengyihao's avatar
dengyihao 已提交
3221
    return -1;
dengyihao's avatar
dengyihao 已提交
3222 3223 3224 3225
  }
  return 0;
}

dengyihao's avatar
dengyihao 已提交
3226 3227
static int32_t sysChkFilter__DBName(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
dengyihao's avatar
dengyihao 已提交
3228 3229 3230 3231 3232 3233

  if (pOper->opType != OP_TYPE_EQUAL && pOper->opType != OP_TYPE_NOT_EQUAL) {
    return -1;
  }

  SValueNode* pVal = (SValueNode*)pOper->pRight;
dengyihao's avatar
dengyihao 已提交
3234
  if (!IS_STR_DATA_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3235 3236
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3237

dengyihao's avatar
dengyihao 已提交
3238 3239 3240 3241 3242
  return 0;
}
static int32_t sysChkFilter__VgroupId(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
dengyihao's avatar
dengyihao 已提交
3243
  if (!IS_INTEGER_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3244 3245
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3246
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3247 3248 3249 3250
}
static int32_t sysChkFilter__TableName(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
dengyihao's avatar
dengyihao 已提交
3251
  if (!IS_STR_DATA_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3252 3253
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3254
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3255 3256 3257 3258 3259
}
static int32_t sysChkFilter__CreateTime(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;

dengyihao's avatar
dengyihao 已提交
3260
  if (!IS_TIMESTAMP_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3261 3262
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3263
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3264 3265 3266 3267 3268 3269
}

static int32_t sysChkFilter__Ncolumn(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;

dengyihao's avatar
dengyihao 已提交
3270
  if (!IS_INTEGER_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3271 3272
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3273
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3274 3275 3276 3277 3278
}
static int32_t sysChkFilter__Ttl(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;

dengyihao's avatar
dengyihao 已提交
3279
  if (!IS_INTEGER_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3280 3281
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3282
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3283 3284 3285 3286
}
static int32_t sysChkFilter__STableName(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
dengyihao's avatar
dengyihao 已提交
3287
  if (!IS_STR_DATA_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3288 3289
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3290
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3291 3292 3293 3294
}
static int32_t sysChkFilter__Uid(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
dengyihao's avatar
dengyihao 已提交
3295
  if (!IS_INTEGER_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3296 3297
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3298
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3299 3300 3301 3302
}
static int32_t sysChkFilter__Type(SNode* pNode) {
  SOperatorNode* pOper = (SOperatorNode*)pNode;
  SValueNode*    pVal = (SValueNode*)pOper->pRight;
dengyihao's avatar
dengyihao 已提交
3303
  if (!IS_INTEGER_TYPE(pVal->node.resType.type)) {
dengyihao's avatar
dengyihao 已提交
3304 3305
    return -1;
  }
dengyihao's avatar
dengyihao 已提交
3306
  return sysChkFilter__Comm(pNode);
dengyihao's avatar
dengyihao 已提交
3307
}
dengyihao's avatar
dengyihao 已提交
3308
static int32_t optSysTabFilteImpl(void* arg, SNode* cond, SArray* result) {
dengyihao's avatar
dengyihao 已提交
3309
  if (optSysCheckOper(cond) != 0) return -1;
dengyihao's avatar
dengyihao 已提交
3310 3311 3312

  SOperatorNode* pNode = (SOperatorNode*)cond;

dengyihao's avatar
dengyihao 已提交
3313 3314
  int8_t i = 0;
  for (; i < SYSTAB_FILTER_DICT_SIZE; i++) {
dengyihao's avatar
dengyihao 已提交
3315 3316
    if (strcmp(filterDict[i].name, ((SColumnNode*)(pNode->pLeft))->colName) == 0) {
      break;
3317
    }
dengyihao's avatar
dengyihao 已提交
3318 3319
  }
  if (i >= SYSTAB_FILTER_DICT_SIZE) return -1;
3320

dengyihao's avatar
dengyihao 已提交
3321
  if (filterDict[i].chkFunc(cond) != 0) return -1;
3322

dengyihao's avatar
dengyihao 已提交
3323 3324
  return filterDict[i].fltFunc(arg, cond, result);
}
3325

dengyihao's avatar
dengyihao 已提交
3326 3327
static int32_t optSysCheckOper(SNode* pOpear) {
  if (nodeType(pOpear) != QUERY_NODE_OPERATOR) return -1;
3328

dengyihao's avatar
dengyihao 已提交
3329
  SOperatorNode* pOper = (SOperatorNode*)pOpear;
dengyihao's avatar
dengyihao 已提交
3330 3331 3332
  if (pOper->opType < OP_TYPE_GREATER_THAN || pOper->opType > OP_TYPE_NOT_EQUAL) {
    return -1;
  }
3333

dengyihao's avatar
dengyihao 已提交
3334 3335 3336 3337
  if (nodeType(pOper->pLeft) != QUERY_NODE_COLUMN || nodeType(pOper->pRight) != QUERY_NODE_VALUE) {
    return -1;
  }
  return 0;
dengyihao's avatar
dengyihao 已提交
3338
}
3339

dengyihao's avatar
dengyihao 已提交
3340 3341 3342 3343 3344 3345 3346 3347
static int tableUidCompare(const void* a, const void* b) {
  int64_t u1 = *(int64_t*)a;
  int64_t u2 = *(int64_t*)b;
  if (u1 == u2) {
    return 0;
  }
  return u1 < u2 ? -1 : 1;
}
dengyihao's avatar
opt mem  
dengyihao 已提交
3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402

typedef struct MergeIndex {
  int idx;
  int len;
} MergeIndex;

static FORCE_INLINE int optSysBinarySearch(SArray* arr, int s, int e, uint64_t k) {
  uint64_t v;
  int32_t  m;
  while (s <= e) {
    m = s + (e - s) / 2;
    v = *(uint64_t*)taosArrayGet(arr, m);
    if (v >= k) {
      e = m - 1;
    } else {
      s = m + 1;
    }
  }
  return s;
}

void optSysIntersection(SArray* in, SArray* out) {
  int32_t sz = (int32_t)taosArrayGetSize(in);
  if (sz <= 0) {
    return;
  }
  MergeIndex* mi = taosMemoryCalloc(sz, sizeof(MergeIndex));
  for (int i = 0; i < sz; i++) {
    SArray* t = taosArrayGetP(in, i);
    mi[i].len = (int32_t)taosArrayGetSize(t);
    mi[i].idx = 0;
  }

  SArray* base = taosArrayGetP(in, 0);
  for (int i = 0; i < taosArrayGetSize(base); i++) {
    uint64_t tgt = *(uint64_t*)taosArrayGet(base, i);
    bool     has = true;
    for (int j = 1; j < taosArrayGetSize(in); j++) {
      SArray* oth = taosArrayGetP(in, j);
      int     mid = optSysBinarySearch(oth, mi[j].idx, mi[j].len - 1, tgt);
      if (mid >= 0 && mid < mi[j].len) {
        uint64_t val = *(uint64_t*)taosArrayGet(oth, mid);
        has = (val == tgt ? true : false);
        mi[j].idx = mid;
      } else {
        has = false;
      }
    }
    if (has == true) {
      taosArrayPush(out, &tgt);
    }
  }
  taosMemoryFreeClear(mi);
}

dengyihao's avatar
dengyihao 已提交
3403 3404 3405
static int32_t optSysMergeRslt(SArray* mRslt, SArray* rslt) {
  // TODO, find comm mem from mRslt
  for (int i = 0; i < taosArrayGetSize(mRslt); i++) {
dengyihao's avatar
opt mem  
dengyihao 已提交
3406 3407 3408 3409 3410 3411
    SArray* arslt = taosArrayGetP(mRslt, i);
    taosArraySort(arslt, tableUidCompare);
  }
  optSysIntersection(mRslt, rslt);
  return 0;
}
3412

dengyihao's avatar
opt mem  
dengyihao 已提交
3413 3414 3415 3416 3417 3418 3419
static int32_t optSysSpecialColumn(SNode* cond) {
  SOperatorNode* pOper = (SOperatorNode*)cond;
  SColumnNode*   pCol = (SColumnNode*)pOper->pLeft;
  for (int i = 0; i < sizeof(SYSTABLE_SPECIAL_COL) / sizeof(SYSTABLE_SPECIAL_COL[0]); i++) {
    if (0 == strcmp(pCol->colName, SYSTABLE_SPECIAL_COL[i])) {
      return 1;
    }
dengyihao's avatar
dengyihao 已提交
3420 3421 3422
  }
  return 0;
}
3423

dengyihao's avatar
dengyihao 已提交
3424 3425 3426 3427
static int32_t optSysTabFilte(void* arg, SNode* cond, SArray* result) {
  int ret = -1;
  if (nodeType(cond) == QUERY_NODE_OPERATOR) {
    ret = optSysTabFilteImpl(arg, cond, result);
dengyihao's avatar
dengyihao 已提交
3428 3429 3430
    if (ret == 0) {
      SOperatorNode* pOper = (SOperatorNode*)cond;
      SColumnNode*   pCol = (SColumnNode*)pOper->pLeft;
dengyihao's avatar
dengyihao 已提交
3431
      if (0 == strcmp(pCol->colName, "create_time")) {
dengyihao's avatar
dengyihao 已提交
3432 3433 3434 3435
        return 0;
      }
      return -1;
    }
dengyihao's avatar
dengyihao 已提交
3436 3437
    return ret;
  }
3438

dengyihao's avatar
dengyihao 已提交
3439 3440 3441
  if (nodeType(cond) != QUERY_NODE_LOGIC_CONDITION || ((SLogicConditionNode*)cond)->condType != LOGIC_COND_TYPE_AND) {
    return ret;
  }
3442

dengyihao's avatar
dengyihao 已提交
3443 3444
  SLogicConditionNode* pNode = (SLogicConditionNode*)cond;
  SNodeList*           pList = (SNodeList*)pNode->pParameterList;
3445

dengyihao's avatar
dengyihao 已提交
3446
  int32_t len = LIST_LENGTH(pList);
3447

dengyihao's avatar
dengyihao 已提交
3448
  bool    hasIdx = false;
dengyihao's avatar
dengyihao 已提交
3449
  bool    hasRslt = true;
dengyihao's avatar
dengyihao 已提交
3450
  SArray* mRslt = taosArrayInit(len, POINTER_BYTES);
3451

dengyihao's avatar
dengyihao 已提交
3452 3453 3454
  SListCell* cell = pList->pHead;
  for (int i = 0; i < len; i++) {
    if (cell == NULL) break;
3455

dengyihao's avatar
dengyihao 已提交
3456
    SArray* aRslt = taosArrayInit(16, sizeof(int64_t));
3457

dengyihao's avatar
dengyihao 已提交
3458 3459
    ret = optSysTabFilteImpl(arg, cell->pNode, aRslt);
    if (ret == 0) {
dengyihao's avatar
dengyihao 已提交
3460
      // has index
dengyihao's avatar
dengyihao 已提交
3461
      hasIdx = true;
dengyihao's avatar
opt mem  
dengyihao 已提交
3462 3463 3464 3465 3466 3467
      if (optSysSpecialColumn(cell->pNode) == 0) {
        taosArrayPush(mRslt, &aRslt);
      } else {
        // db_name/vgroup not result
        taosArrayDestroy(aRslt);
      }
dengyihao's avatar
dengyihao 已提交
3468
    } else if (ret == -2) {
dengyihao's avatar
dengyihao 已提交
3469
      // current vg
dengyihao's avatar
dengyihao 已提交
3470 3471 3472 3473 3474 3475
      hasIdx = true;
      hasRslt = false;
      taosArrayDestroy(aRslt);
      break;
    } else {
      taosArrayDestroy(aRslt);
dengyihao's avatar
dengyihao 已提交
3476 3477 3478
    }
    cell = cell->pNext;
  }
dengyihao's avatar
dengyihao 已提交
3479 3480 3481
  if (hasRslt && hasIdx) {
    optSysMergeRslt(mRslt, result);
  }
3482

dengyihao's avatar
dengyihao 已提交
3483 3484 3485
  for (int i = 0; i < taosArrayGetSize(mRslt); i++) {
    SArray* aRslt = taosArrayGetP(mRslt, i);
    taosArrayDestroy(aRslt);
dengyihao's avatar
dengyihao 已提交
3486
  }
dengyihao's avatar
dengyihao 已提交
3487
  taosArrayDestroy(mRslt);
dengyihao's avatar
dengyihao 已提交
3488 3489 3490 3491
  if (hasRslt == false) {
    return -2;
  }
  if (hasRslt && hasIdx) {
dengyihao's avatar
dengyihao 已提交
3492 3493 3494 3495 3496
    cell = pList->pHead;
    for (int i = 0; i < len; i++) {
      if (cell == NULL) break;
      SOperatorNode* pOper = (SOperatorNode*)cell->pNode;
      SColumnNode*   pCol = (SColumnNode*)pOper->pLeft;
dengyihao's avatar
dengyihao 已提交
3497
      if (0 == strcmp(pCol->colName, "create_time")) {
dengyihao's avatar
dengyihao 已提交
3498 3499 3500 3501 3502
        return 0;
      }
      cell = cell->pNext;
    }
    return -1;
dengyihao's avatar
dengyihao 已提交
3503 3504
  }
  return -1;
dengyihao's avatar
dengyihao 已提交
3505
}
3506

dengyihao's avatar
dengyihao 已提交
3507
static SSDataBlock* sysTableBuildUserTablesByUids(SOperatorInfo* pOperator) {
dengyihao's avatar
dengyihao 已提交
3508 3509
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
  SSysTableScanInfo* pInfo = pOperator->info;
3510

dengyihao's avatar
dengyihao 已提交
3511
  SSysTableIndex* pIdx = pInfo->pIdx;
dengyihao's avatar
dengyihao 已提交
3512 3513
  blockDataCleanup(pInfo->pRes);
  int32_t numOfRows = 0;
3514

dengyihao's avatar
dengyihao 已提交
3515
  int ret = 0;
3516

dengyihao's avatar
dengyihao 已提交
3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530
  const char* db = NULL;
  int32_t     vgId = 0;
  vnodeGetInfo(pInfo->readHandle.vnode, &db, &vgId);

  SName sn = {0};
  char  dbname[TSDB_DB_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
  tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);

  tNameGetDbName(&sn, varDataVal(dbname));
  varDataSetLen(dbname, strlen(varDataVal(dbname)));

  SSDataBlock* p = buildInfoSchemaTableMetaBlock(TSDB_INS_TABLE_TABLES);
  blockDataEnsureCapacity(p, pOperator->resultInfo.capacity);

dengyihao's avatar
dengyihao 已提交
3531 3532 3533 3534
  char    n[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
  int32_t i = pIdx->lastIdx;
  for (; i < taosArrayGetSize(pIdx->uids); i++) {
    tb_uid_t* uid = taosArrayGet(pIdx->uids, i);
3535

dengyihao's avatar
dengyihao 已提交
3536
    SMetaReader mr = {0};
dengyihao's avatar
dengyihao 已提交
3537
    metaReaderInit(&mr, pInfo->readHandle.meta, 0);
H
Haojun Liao 已提交
3538
    ret = metaGetTableEntryByUid(&mr, *uid);
dengyihao's avatar
dengyihao 已提交
3539 3540 3541 3542 3543 3544 3545 3546 3547
    if (ret < 0) {
      metaReaderClear(&mr);
      continue;
    }
    STR_TO_VARSTR(n, mr.me.name);

    // table name
    SColumnInfoData* pColInfoData = taosArrayGet(p->pDataBlock, 0);
    colDataAppend(pColInfoData, numOfRows, n, false);
3548

dengyihao's avatar
dengyihao 已提交
3549 3550 3551
    // database name
    pColInfoData = taosArrayGet(p->pDataBlock, 1);
    colDataAppend(pColInfoData, numOfRows, dbname, false);
3552

dengyihao's avatar
dengyihao 已提交
3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565 3566
    // vgId
    pColInfoData = taosArrayGet(p->pDataBlock, 6);
    colDataAppend(pColInfoData, numOfRows, (char*)&vgId, false);

    int32_t tableType = mr.me.type;
    if (tableType == TSDB_CHILD_TABLE) {
      // create time
      int64_t ts = mr.me.ctbEntry.ctime;
      pColInfoData = taosArrayGet(p->pDataBlock, 2);
      colDataAppend(pColInfoData, numOfRows, (char*)&ts, false);

      SMetaReader mr1 = {0};
      metaReaderInit(&mr1, pInfo->readHandle.meta, META_READER_NOLOCK);

dengyihao's avatar
dengyihao 已提交
3567 3568
      int64_t suid = mr.me.ctbEntry.suid;
      int32_t code = metaGetTableEntryByUid(&mr1, suid);
dengyihao's avatar
dengyihao 已提交
3569 3570 3571 3572
      if (code != TSDB_CODE_SUCCESS) {
        qError("failed to get super table meta, cname:%s, suid:0x%" PRIx64 ", code:%s, %s", pInfo->pCur->mr.me.name,
               suid, tstrerror(terrno), GET_TASKID(pTaskInfo));
        metaReaderClear(&mr1);
dengyihao's avatar
dengyihao 已提交
3573
        metaReaderClear(&mr);
dengyihao's avatar
dengyihao 已提交
3574
        T_LONG_JMP(pTaskInfo->env, terrno);
H
Haojun Liao 已提交
3575
      }
dengyihao's avatar
dengyihao 已提交
3576
      pColInfoData = taosArrayGet(p->pDataBlock, 3);
dengyihao's avatar
dengyihao 已提交
3577
      colDataAppend(pColInfoData, numOfRows, (char*)&mr1.me.stbEntry.schemaRow.nCols, false);
H
Haojun Liao 已提交
3578

dengyihao's avatar
dengyihao 已提交
3579 3580 3581
      // super table name
      STR_TO_VARSTR(n, mr1.me.name);
      pColInfoData = taosArrayGet(p->pDataBlock, 4);
3582
      colDataAppend(pColInfoData, numOfRows, n, false);
dengyihao's avatar
dengyihao 已提交
3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597
      metaReaderClear(&mr1);

      // table comment
      pColInfoData = taosArrayGet(p->pDataBlock, 8);
      if (mr.me.ctbEntry.commentLen > 0) {
        char comment[TSDB_TB_COMMENT_LEN + VARSTR_HEADER_SIZE] = {0};
        STR_TO_VARSTR(comment, mr.me.ctbEntry.comment);
        colDataAppend(pColInfoData, numOfRows, comment, false);
      } else if (mr.me.ctbEntry.commentLen == 0) {
        char comment[VARSTR_HEADER_SIZE + VARSTR_HEADER_SIZE] = {0};
        STR_TO_VARSTR(comment, "");
        colDataAppend(pColInfoData, numOfRows, comment, false);
      } else {
        colDataAppendNULL(pColInfoData, numOfRows);
      }
3598

dengyihao's avatar
dengyihao 已提交
3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623
      // uid
      pColInfoData = taosArrayGet(p->pDataBlock, 5);
      colDataAppend(pColInfoData, numOfRows, (char*)&mr.me.uid, false);

      // ttl
      pColInfoData = taosArrayGet(p->pDataBlock, 7);
      colDataAppend(pColInfoData, numOfRows, (char*)&mr.me.ctbEntry.ttlDays, false);

      STR_TO_VARSTR(n, "CHILD_TABLE");

    } else if (tableType == TSDB_NORMAL_TABLE) {
      // create time
      pColInfoData = taosArrayGet(p->pDataBlock, 2);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.ntbEntry.ctime, false);

      // number of columns
      pColInfoData = taosArrayGet(p->pDataBlock, 3);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.ntbEntry.schemaRow.nCols, false);

      // super table name
      pColInfoData = taosArrayGet(p->pDataBlock, 4);
      colDataAppendNULL(pColInfoData, numOfRows);

      // table comment
      pColInfoData = taosArrayGet(p->pDataBlock, 8);
dengyihao's avatar
dengyihao 已提交
3624
      if (mr.me.ntbEntry.commentLen > 0) {
dengyihao's avatar
dengyihao 已提交
3625
        char comment[TSDB_TB_COMMENT_LEN + VARSTR_HEADER_SIZE] = {0};
dengyihao's avatar
dengyihao 已提交
3626
        STR_TO_VARSTR(comment, mr.me.ntbEntry.comment);
dengyihao's avatar
dengyihao 已提交
3627
        colDataAppend(pColInfoData, numOfRows, comment, false);
dengyihao's avatar
dengyihao 已提交
3628
      } else if (mr.me.ntbEntry.commentLen == 0) {
dengyihao's avatar
dengyihao 已提交
3629 3630 3631 3632 3633 3634
        char comment[VARSTR_HEADER_SIZE + VARSTR_HEADER_SIZE] = {0};
        STR_TO_VARSTR(comment, "");
        colDataAppend(pColInfoData, numOfRows, comment, false);
      } else {
        colDataAppendNULL(pColInfoData, numOfRows);
      }
3635

dengyihao's avatar
dengyihao 已提交
3636 3637
      // uid
      pColInfoData = taosArrayGet(p->pDataBlock, 5);
dengyihao's avatar
dengyihao 已提交
3638
      colDataAppend(pColInfoData, numOfRows, (char*)&mr.me.uid, false);
3639

dengyihao's avatar
dengyihao 已提交
3640 3641
      // ttl
      pColInfoData = taosArrayGet(p->pDataBlock, 7);
dengyihao's avatar
dengyihao 已提交
3642
      colDataAppend(pColInfoData, numOfRows, (char*)&mr.me.ntbEntry.ttlDays, false);
3643

dengyihao's avatar
dengyihao 已提交
3644 3645
      STR_TO_VARSTR(n, "NORMAL_TABLE");
      // impl later
3646
    }
3647

dengyihao's avatar
dengyihao 已提交
3648 3649
    metaReaderClear(&mr);

dengyihao's avatar
dengyihao 已提交
3650 3651 3652 3653
    pColInfoData = taosArrayGet(p->pDataBlock, 9);
    colDataAppend(pColInfoData, numOfRows, n, false);

    if (++numOfRows >= pOperator->resultInfo.capacity) {
3654 3655 3656
      p->info.rows = numOfRows;
      pInfo->pRes->info.rows = numOfRows;

H
Haojun Liao 已提交
3657
      relocateColumnData(pInfo->pRes, pInfo->matchInfo.pList, p->pDataBlock, false);
3658 3659 3660 3661
      doFilterResult(pInfo);

      blockDataCleanup(p);
      numOfRows = 0;
dengyihao's avatar
dengyihao 已提交
3662 3663 3664 3665

      if (pInfo->pRes->info.rows > 0) {
        break;
      }
3666
    }
dengyihao's avatar
dengyihao 已提交
3667
  }
3668

dengyihao's avatar
dengyihao 已提交
3669 3670 3671
  if (numOfRows > 0) {
    p->info.rows = numOfRows;
    pInfo->pRes->info.rows = numOfRows;
3672

H
Haojun Liao 已提交
3673
    relocateColumnData(pInfo->pRes, pInfo->matchInfo.pList, p->pDataBlock, false);
dengyihao's avatar
dengyihao 已提交
3674 3675 3676 3677 3678 3679
    doFilterResult(pInfo);

    blockDataCleanup(p);
    numOfRows = 0;
  }

dengyihao's avatar
dengyihao 已提交
3680 3681 3682
  if (i >= taosArrayGetSize(pIdx->uids)) {
    doSetOperatorCompleted(pOperator);
  } else {
dengyihao's avatar
dengyihao 已提交
3683
    pIdx->lastIdx = i + 1;
dengyihao's avatar
dengyihao 已提交
3684
  }
dengyihao's avatar
dengyihao 已提交
3685

dengyihao's avatar
dengyihao 已提交
3686
  blockDataDestroy(p);
dengyihao's avatar
dengyihao 已提交
3687 3688 3689 3690

  pInfo->loadInfo.totalRows += pInfo->pRes->info.rows;
  return (pInfo->pRes->info.rows == 0) ? NULL : pInfo->pRes;
}
dengyihao's avatar
dengyihao 已提交
3691
static SSDataBlock* sysTableBuildUserTables(SOperatorInfo* pOperator) {
dengyihao's avatar
dengyihao 已提交
3692 3693
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;

H
Haojun Liao 已提交
3694
  SSysTableScanInfo* pInfo = pOperator->info;
dengyihao's avatar
dengyihao 已提交
3695 3696
  if (pInfo->pCur == NULL) {
    pInfo->pCur = metaOpenTbCursor(pInfo->readHandle.meta);
3697
  }
H
Haojun Liao 已提交
3698

dengyihao's avatar
dengyihao 已提交
3699 3700
  blockDataCleanup(pInfo->pRes);
  int32_t numOfRows = 0;
3701

dengyihao's avatar
dengyihao 已提交
3702 3703 3704
  const char* db = NULL;
  int32_t     vgId = 0;
  vnodeGetInfo(pInfo->readHandle.vnode, &db, &vgId);
3705

dengyihao's avatar
dengyihao 已提交
3706 3707 3708
  SName sn = {0};
  char  dbname[TSDB_DB_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
  tNameFromString(&sn, db, T_NAME_ACCT | T_NAME_DB);
3709

dengyihao's avatar
dengyihao 已提交
3710 3711
  tNameGetDbName(&sn, varDataVal(dbname));
  varDataSetLen(dbname, strlen(varDataVal(dbname)));
3712

dengyihao's avatar
dengyihao 已提交
3713 3714
  SSDataBlock* p = buildInfoSchemaTableMetaBlock(TSDB_INS_TABLE_TABLES);
  blockDataEnsureCapacity(p, pOperator->resultInfo.capacity);
3715

dengyihao's avatar
dengyihao 已提交
3716
  char n[TSDB_TABLE_NAME_LEN + VARSTR_HEADER_SIZE] = {0};
3717

dengyihao's avatar
dengyihao 已提交
3718 3719 3720
  int32_t ret = 0;
  while ((ret = metaTbCursorNext(pInfo->pCur)) == 0) {
    STR_TO_VARSTR(n, pInfo->pCur->mr.me.name);
3721

dengyihao's avatar
dengyihao 已提交
3722 3723 3724
    // table name
    SColumnInfoData* pColInfoData = taosArrayGet(p->pDataBlock, 0);
    colDataAppend(pColInfoData, numOfRows, n, false);
3725

dengyihao's avatar
dengyihao 已提交
3726 3727 3728
    // database name
    pColInfoData = taosArrayGet(p->pDataBlock, 1);
    colDataAppend(pColInfoData, numOfRows, dbname, false);
3729

dengyihao's avatar
dengyihao 已提交
3730 3731 3732 3733 3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748
    // vgId
    pColInfoData = taosArrayGet(p->pDataBlock, 6);
    colDataAppend(pColInfoData, numOfRows, (char*)&vgId, false);

    int32_t tableType = pInfo->pCur->mr.me.type;
    if (tableType == TSDB_CHILD_TABLE) {
      // create time
      int64_t ts = pInfo->pCur->mr.me.ctbEntry.ctime;
      pColInfoData = taosArrayGet(p->pDataBlock, 2);
      colDataAppend(pColInfoData, numOfRows, (char*)&ts, false);

      SMetaReader mr = {0};
      metaReaderInit(&mr, pInfo->readHandle.meta, META_READER_NOLOCK);

      uint64_t suid = pInfo->pCur->mr.me.ctbEntry.suid;
      int32_t  code = metaGetTableEntryByUid(&mr, suid);
      if (code != TSDB_CODE_SUCCESS) {
        qError("failed to get super table meta, cname:%s, suid:0x%" PRIx64 ", code:%s, %s", pInfo->pCur->mr.me.name,
               suid, tstrerror(terrno), GET_TASKID(pTaskInfo));
3749
        metaReaderClear(&mr);
dengyihao's avatar
dengyihao 已提交
3750 3751 3752 3753
        metaCloseTbCursor(pInfo->pCur);
        pInfo->pCur = NULL;
        T_LONG_JMP(pTaskInfo->env, terrno);
      }
3754

dengyihao's avatar
dengyihao 已提交
3755 3756 3757
      // number of columns
      pColInfoData = taosArrayGet(p->pDataBlock, 3);
      colDataAppend(pColInfoData, numOfRows, (char*)&mr.me.stbEntry.schemaRow.nCols, false);
3758

dengyihao's avatar
dengyihao 已提交
3759 3760 3761 3762 3763
      // super table name
      STR_TO_VARSTR(n, mr.me.name);
      pColInfoData = taosArrayGet(p->pDataBlock, 4);
      colDataAppend(pColInfoData, numOfRows, n, false);
      metaReaderClear(&mr);
3764

dengyihao's avatar
dengyihao 已提交
3765 3766 3767 3768 3769 3770 3771 3772 3773 3774 3775
      // table comment
      pColInfoData = taosArrayGet(p->pDataBlock, 8);
      if (pInfo->pCur->mr.me.ctbEntry.commentLen > 0) {
        char comment[TSDB_TB_COMMENT_LEN + VARSTR_HEADER_SIZE] = {0};
        STR_TO_VARSTR(comment, pInfo->pCur->mr.me.ctbEntry.comment);
        colDataAppend(pColInfoData, numOfRows, comment, false);
      } else if (pInfo->pCur->mr.me.ctbEntry.commentLen == 0) {
        char comment[VARSTR_HEADER_SIZE + VARSTR_HEADER_SIZE] = {0};
        STR_TO_VARSTR(comment, "");
        colDataAppend(pColInfoData, numOfRows, comment, false);
      } else {
3776
        colDataAppendNULL(pColInfoData, numOfRows);
H
Haojun Liao 已提交
3777 3778
      }

dengyihao's avatar
dengyihao 已提交
3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797 3798 3799 3800 3801 3802 3803 3804 3805 3806 3807 3808 3809 3810 3811 3812 3813
      // uid
      pColInfoData = taosArrayGet(p->pDataBlock, 5);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.uid, false);

      // ttl
      pColInfoData = taosArrayGet(p->pDataBlock, 7);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.ctbEntry.ttlDays, false);

      STR_TO_VARSTR(n, "CHILD_TABLE");
    } else if (tableType == TSDB_NORMAL_TABLE) {
      // create time
      pColInfoData = taosArrayGet(p->pDataBlock, 2);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.ntbEntry.ctime, false);

      // number of columns
      pColInfoData = taosArrayGet(p->pDataBlock, 3);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.ntbEntry.schemaRow.nCols, false);

      // super table name
      pColInfoData = taosArrayGet(p->pDataBlock, 4);
      colDataAppendNULL(pColInfoData, numOfRows);

      // table comment
      pColInfoData = taosArrayGet(p->pDataBlock, 8);
      if (pInfo->pCur->mr.me.ntbEntry.commentLen > 0) {
        char comment[TSDB_TB_COMMENT_LEN + VARSTR_HEADER_SIZE] = {0};
        STR_TO_VARSTR(comment, pInfo->pCur->mr.me.ntbEntry.comment);
        colDataAppend(pColInfoData, numOfRows, comment, false);
      } else if (pInfo->pCur->mr.me.ntbEntry.commentLen == 0) {
        char comment[VARSTR_HEADER_SIZE + VARSTR_HEADER_SIZE] = {0};
        STR_TO_VARSTR(comment, "");
        colDataAppend(pColInfoData, numOfRows, comment, false);
      } else {
        colDataAppendNULL(pColInfoData, numOfRows);
      }
3814

dengyihao's avatar
dengyihao 已提交
3815 3816 3817
      // uid
      pColInfoData = taosArrayGet(p->pDataBlock, 5);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.uid, false);
3818

dengyihao's avatar
dengyihao 已提交
3819 3820 3821
      // ttl
      pColInfoData = taosArrayGet(p->pDataBlock, 7);
      colDataAppend(pColInfoData, numOfRows, (char*)&pInfo->pCur->mr.me.ntbEntry.ttlDays, false);
3822

dengyihao's avatar
dengyihao 已提交
3823
      STR_TO_VARSTR(n, "NORMAL_TABLE");
3824
    }
H
Haojun Liao 已提交
3825

dengyihao's avatar
dengyihao 已提交
3826 3827 3828 3829
    pColInfoData = taosArrayGet(p->pDataBlock, 9);
    colDataAppend(pColInfoData, numOfRows, n, false);

    if (++numOfRows >= pOperator->resultInfo.capacity) {
3830 3831 3832
      p->info.rows = numOfRows;
      pInfo->pRes->info.rows = numOfRows;

H
Haojun Liao 已提交
3833
      relocateColumnData(pInfo->pRes, pInfo->matchInfo.pList, p->pDataBlock, false);
3834 3835 3836 3837
      doFilterResult(pInfo);

      blockDataCleanup(p);
      numOfRows = 0;
dengyihao's avatar
dengyihao 已提交
3838 3839 3840 3841

      if (pInfo->pRes->info.rows > 0) {
        break;
      }
3842
    }
dengyihao's avatar
dengyihao 已提交
3843
  }
3844

dengyihao's avatar
dengyihao 已提交
3845 3846 3847
  if (numOfRows > 0) {
    p->info.rows = numOfRows;
    pInfo->pRes->info.rows = numOfRows;
3848

H
Haojun Liao 已提交
3849
    relocateColumnData(pInfo->pRes, pInfo->matchInfo.pList, p->pDataBlock, false);
dengyihao's avatar
dengyihao 已提交
3850
    doFilterResult(pInfo);
H
Haojun Liao 已提交
3851

dengyihao's avatar
dengyihao 已提交
3852 3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866 3867
    blockDataCleanup(p);
    numOfRows = 0;
  }

  blockDataDestroy(p);

  // todo temporarily free the cursor here, the true reason why the free is not valid needs to be found
  if (ret != 0) {
    metaCloseTbCursor(pInfo->pCur);
    pInfo->pCur = NULL;
    doSetOperatorCompleted(pOperator);
  }

  pInfo->loadInfo.totalRows += pInfo->pRes->info.rows;
  return (pInfo->pRes->info.rows == 0) ? NULL : pInfo->pRes;
}
dengyihao's avatar
dengyihao 已提交
3868

dengyihao's avatar
dengyihao 已提交
3869 3870 3871 3872 3873 3874 3875 3876 3877 3878 3879 3880 3881 3882
static SSDataBlock* sysTableScanUserTables(SOperatorInfo* pOperator) {
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
  SSysTableScanInfo* pInfo = pOperator->info;

  SNode* pCondition = pInfo->pCondition;
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  // the retrieve is executed on the mnode, so return tables that belongs to the information schema database.
  if (pInfo->readHandle.mnd != NULL) {
    buildSysDbTableInfo(pInfo, pOperator->resultInfo.capacity);

    doFilterResult(pInfo);
3883
    pInfo->loadInfo.totalRows += pInfo->pRes->info.rows;
dengyihao's avatar
dengyihao 已提交
3884 3885

    doSetOperatorCompleted(pOperator);
3886
    return (pInfo->pRes->info.rows == 0) ? NULL : pInfo->pRes;
dengyihao's avatar
dengyihao 已提交
3887
  } else {
dengyihao's avatar
dengyihao 已提交
3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907 3908 3909 3910 3911
    if (pInfo->showRewrite == false) {
      if (pCondition != NULL && pInfo->pIdx == NULL) {
        SSTabFltArg arg = {.pMeta = pInfo->readHandle.meta, .pVnode = pInfo->readHandle.vnode};

        SSysTableIndex* idx = taosMemoryMalloc(sizeof(SSysTableIndex));
        idx->init = 0;
        idx->uids = taosArrayInit(128, sizeof(int64_t));
        idx->lastIdx = 0;

        pInfo->pIdx = idx;  // set idx arg

        int flt = optSysTabFilte(&arg, pCondition, idx->uids);
        if (flt == 0) {
          pInfo->pIdx->init = 1;
          SSDataBlock* blk = sysTableBuildUserTablesByUids(pOperator);
          return blk;
        } else if (flt == -2) {
          qDebug("%s failed to get sys table info by idx, empty result", GET_TASKID(pTaskInfo));
          return NULL;
        } else if (flt == -1) {
          // not idx
          qDebug("%s failed to get sys table info by idx, scan sys table one by one", GET_TASKID(pTaskInfo));
        }
      } else if (pCondition != NULL && (pInfo->pIdx != NULL && pInfo->pIdx->init == 1)) {
dengyihao's avatar
dengyihao 已提交
3912
        SSDataBlock* blk = sysTableBuildUserTablesByUids(pOperator);
dengyihao's avatar
dengyihao 已提交
3913 3914
        return blk;
      }
dengyihao's avatar
dengyihao 已提交
3915
    }
dengyihao's avatar
dengyihao 已提交
3916

dengyihao's avatar
dengyihao 已提交
3917
    return sysTableBuildUserTables(pOperator);
3918
  }
dengyihao's avatar
dengyihao 已提交
3919
  return NULL;
3920 3921
}

3922 3923 3924 3925 3926 3927 3928 3929
static SSDataBlock* sysTableScanUserSTables(SOperatorInfo* pOperator) {
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
  SSysTableScanInfo* pInfo = pOperator->info;
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  pInfo->pRes->info.rows = 0;
D
dapan1121 已提交
3930
  pOperator->status = OP_EXEC_DONE;
3931 3932 3933 3934 3935

  pInfo->loadInfo.totalRows += pInfo->pRes->info.rows;
  return (pInfo->pRes->info.rows == 0) ? NULL : pInfo->pRes;
}

3936 3937 3938 3939
static SSDataBlock* doSysTableScan(SOperatorInfo* pOperator) {
  // build message and send to mnode to fetch the content of system tables.
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
  SSysTableScanInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
3940
  char               dbName[TSDB_DB_NAME_LEN] = {0};
3941 3942

  const char* name = tNameGetTableName(&pInfo->name);
D
dapan1121 已提交
3943 3944 3945 3946
  if (pInfo->showRewrite) {
    getDBNameFromCondition(pInfo->pCondition, dbName);
    sprintf(pInfo->req.db, "%d.%s", pInfo->accountId, dbName);
  }
H
Hongze Cheng 已提交
3947

D
dapan1121 已提交
3948
  if (strncasecmp(name, TSDB_INS_TABLE_TABLES, TSDB_TABLE_FNAME_LEN) == 0) {
3949
    return sysTableScanUserTables(pOperator);
D
dapan1121 已提交
3950
  } else if (strncasecmp(name, TSDB_INS_TABLE_TAGS, TSDB_TABLE_FNAME_LEN) == 0) {
3951
    return sysTableScanUserTags(pOperator);
L
Liu Jicong 已提交
3952 3953
  } else if (strncasecmp(name, TSDB_INS_TABLE_STABLES, TSDB_TABLE_FNAME_LEN) == 0 && pInfo->showRewrite &&
             IS_SYS_DBNAME(dbName)) {
3954
    return sysTableScanUserSTables(pOperator);
H
Haojun Liao 已提交
3955 3956 3957 3958 3959
  } else {  // load the meta from mnode of the given epset
    if (pOperator->status == OP_EXEC_DONE) {
      return NULL;
    }

3960 3961
    while (1) {
      int64_t startTs = taosGetTimestampUs();
H
Haojun Liao 已提交
3962
      tstrncpy(pInfo->req.tb, tNameGetTableName(&pInfo->name), tListLen(pInfo->req.tb));
H
Haojun Liao 已提交
3963
      tstrncpy(pInfo->req.user, pInfo->pUser, tListLen(pInfo->req.user));
H
Haojun Liao 已提交
3964

3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975
      int32_t contLen = tSerializeSRetrieveTableReq(NULL, 0, &pInfo->req);
      char*   buf1 = taosMemoryCalloc(1, contLen);
      tSerializeSRetrieveTableReq(buf1, contLen, &pInfo->req);

      // send the fetch remote task result reques
      SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
      if (NULL == pMsgSendInfo) {
        qError("%s prepare message %d failed", GET_TASKID(pTaskInfo), (int32_t)sizeof(SMsgSendInfo));
        pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
        return NULL;
      }
H
Haojun Liao 已提交
3976

L
Liu Jicong 已提交
3977 3978
      int32_t msgType = (strcasecmp(name, TSDB_INS_TABLE_DNODE_VARIABLES) == 0) ? TDMT_DND_SYSTABLE_RETRIEVE
                                                                                : TDMT_MND_SYSTABLE_RETRIEVE;
D
dapan1121 已提交
3979

3980 3981 3982
      pMsgSendInfo->param = pOperator;
      pMsgSendInfo->msgInfo.pData = buf1;
      pMsgSendInfo->msgInfo.len = contLen;
D
dapan1121 已提交
3983
      pMsgSendInfo->msgType = msgType;
3984
      pMsgSendInfo->fp = loadSysTableCallback;
D
dapan1121 已提交
3985
      pMsgSendInfo->requestId = pTaskInfo->id.queryId;
H
Haojun Liao 已提交
3986

3987
      int64_t transporterId = 0;
3988 3989
      int32_t code =
          asyncSendMsgToServer(pInfo->readHandle.pMsgCb->clientRpc, &pInfo->epSet, &transporterId, pMsgSendInfo);
3990
      tsem_wait(&pInfo->ready);
H
Haojun Liao 已提交
3991

3992 3993 3994 3995 3996
      if (pTaskInfo->code) {
        qDebug("%s load meta data from mnode failed, totalRows:%" PRIu64 ", code:%s", GET_TASKID(pTaskInfo),
               pInfo->loadInfo.totalRows, tstrerror(pTaskInfo->code));
        return NULL;
      }
H
Haojun Liao 已提交
3997

3998 3999
      SRetrieveMetaTableRsp* pRsp = pInfo->pRsp;
      pInfo->req.showId = pRsp->handle;
H
Haojun Liao 已提交
4000

4001 4002
      if (pRsp->numOfRows == 0 || pRsp->completed) {
        pOperator->status = OP_EXEC_DONE;
4003
        qDebug("%s load meta data from mnode completed, rowsOfSource:%d, totalRows:%" PRIu64, GET_TASKID(pTaskInfo),
4004
               pRsp->numOfRows, pInfo->loadInfo.totalRows);
H
Haojun Liao 已提交
4005

4006
        if (pRsp->numOfRows == 0) {
H
Haojun Liao 已提交
4007
          taosMemoryFree(pRsp);
4008 4009 4010
          return NULL;
        }
      }
H
Haojun Liao 已提交
4011

4012
      char* pStart = pRsp->data;
H
Haojun Liao 已提交
4013
      extractDataBlockFromFetchRsp(pInfo->pRes, pRsp->data, pInfo->matchInfo.pList, &pStart);
4014
      updateLoadRemoteInfo(&pInfo->loadInfo, pRsp->numOfRows, pRsp->compLen, startTs, pOperator);
H
Haojun Liao 已提交
4015

4016 4017
      // todo log the filter info
      doFilterResult(pInfo);
H
Haojun Liao 已提交
4018
      taosMemoryFree(pRsp);
4019 4020
      if (pInfo->pRes->info.rows > 0) {
        return pInfo->pRes;
D
dapan1121 已提交
4021 4022
      } else if (pOperator->status == OP_EXEC_DONE) {
        return NULL;
4023
      }
4024
    }
H
Haojun Liao 已提交
4025 4026 4027
  }
}

4028
int32_t buildSysDbTableInfo(const SSysTableScanInfo* pInfo, int32_t capacity) {
D
dapan1121 已提交
4029
  SSDataBlock* p = buildInfoSchemaTableMetaBlock(TSDB_INS_TABLE_TABLES);
4030
  blockDataEnsureCapacity(p, capacity);
4031

L
Liu Jicong 已提交
4032
  size_t               size = 0;
4033 4034 4035
  const SSysTableMeta* pSysDbTableMeta = NULL;

  getInfosDbMeta(&pSysDbTableMeta, &size);
4036
  p->info.rows = buildDbTableInfoBlock(pInfo->sysInfo, p, pSysDbTableMeta, size, TSDB_INFORMATION_SCHEMA_DB);
4037 4038

  getPerfDbMeta(&pSysDbTableMeta, &size);
4039
  p->info.rows = buildDbTableInfoBlock(pInfo->sysInfo, p, pSysDbTableMeta, size, TSDB_PERFORMANCE_SCHEMA_DB);
4040 4041

  pInfo->pRes->info.rows = p->info.rows;
H
Haojun Liao 已提交
4042
  relocateColumnData(pInfo->pRes, pInfo->matchInfo.pList, p->pDataBlock, false);
4043 4044 4045
  blockDataDestroy(p);

  return pInfo->pRes->info.rows;
4046 4047
}

4048
int32_t buildDbTableInfoBlock(bool sysInfo, const SSDataBlock* p, const SSysTableMeta* pSysDbTableMeta, size_t size,
L
Liu Jicong 已提交
4049 4050
                              const char* dbName) {
  char    n[TSDB_TABLE_FNAME_LEN + VARSTR_HEADER_SIZE] = {0};
4051 4052
  int32_t numOfRows = p->info.rows;

L
Liu Jicong 已提交
4053
  for (int32_t i = 0; i < size; ++i) {
4054
    const SSysTableMeta* pm = &pSysDbTableMeta[i];
4055 4056 4057
    if (!sysInfo && pm->sysInfo) {
      continue;
    }
4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076

    SColumnInfoData* pColInfoData = taosArrayGet(p->pDataBlock, 0);

    STR_TO_VARSTR(n, pm->name);
    colDataAppend(pColInfoData, numOfRows, n, false);

    // database name
    STR_TO_VARSTR(n, dbName);
    pColInfoData = taosArrayGet(p->pDataBlock, 1);
    colDataAppend(pColInfoData, numOfRows, n, false);

    // create time
    pColInfoData = taosArrayGet(p->pDataBlock, 2);
    colDataAppendNULL(pColInfoData, numOfRows);

    // number of columns
    pColInfoData = taosArrayGet(p->pDataBlock, 3);
    colDataAppend(pColInfoData, numOfRows, (char*)&pm->colNum, false);

L
Liu Jicong 已提交
4077
    for (int32_t j = 4; j <= 8; ++j) {
4078 4079 4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091 4092
      pColInfoData = taosArrayGet(p->pDataBlock, j);
      colDataAppendNULL(pColInfoData, numOfRows);
    }

    STR_TO_VARSTR(n, "SYSTEM_TABLE");

    pColInfoData = taosArrayGet(p->pDataBlock, 9);
    colDataAppend(pColInfoData, numOfRows, n, false);

    numOfRows += 1;
  }

  return numOfRows;
}

4093
SOperatorInfo* createSysTableScanOperatorInfo(void* readHandle, SSystemTableScanPhysiNode* pScanPhyNode,
4094
                                              const char* pUser, SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
4095 4096 4097
  SSysTableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SSysTableScanInfo));
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
4098
    goto _error;
H
Haojun Liao 已提交
4099 4100
  }

4101 4102 4103 4104 4105
  SScanPhysiNode* pScanNode = &pScanPhyNode->scan;

  SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;

  int32_t num = 0;
H
Haojun Liao 已提交
4106
  int32_t code = extractColMatchInfo(pScanNode->pScanCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
4107

4108 4109
  pInfo->accountId = pScanPhyNode->accountId;
  pInfo->pUser = taosMemoryStrDup((void*)pUser);
4110
  pInfo->sysInfo = pScanPhyNode->sysInfo;
4111
  pInfo->showRewrite = pScanPhyNode->showRewrite;
H
Haojun Liao 已提交
4112
  pInfo->pRes = createResDataBlock(pDescNode);
4113
  pInfo->pCondition = pScanNode->node.pConditions;
4114

4115
  initResultSizeInfo(&pOperator->resultInfo, 4096);
H
Haojun Liao 已提交
4116

4117
  tNameAssign(&pInfo->name, &pScanNode->tableName);
4118
  const char* name = tNameGetTableName(&pInfo->name);
4119

D
dapan1121 已提交
4120 4121
  if (strncasecmp(name, TSDB_INS_TABLE_TABLES, TSDB_TABLE_FNAME_LEN) == 0 ||
      strncasecmp(name, TSDB_INS_TABLE_TAGS, TSDB_TABLE_FNAME_LEN) == 0) {
L
Liu Jicong 已提交
4122
    pInfo->readHandle = *(SReadHandle*)readHandle;
4123
    blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
H
Haojun Liao 已提交
4124 4125
  } else {
    tsem_init(&pInfo->ready, 0, 0);
4126
    pInfo->epSet = pScanPhyNode->mgmtEpSet;
4127
    pInfo->readHandle = *(SReadHandle*)readHandle;
H
Haojun Liao 已提交
4128 4129
  }

4130
  pOperator->name = "SysTableScanOperator";
H
Haojun Liao 已提交
4131
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN;
4132 4133 4134
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
H
Haojun Liao 已提交
4135
  pOperator->exprSupp.numOfExprs = taosArrayGetSize(pInfo->pRes->pDataBlock);
4136
  pOperator->pTaskInfo = pTaskInfo;
4137

4138
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doSysTableScan, NULL, NULL, destroySysScanOperator, NULL);
H
Haojun Liao 已提交
4139 4140

  return pOperator;
4141

4142
_error:
4143 4144 4145 4146
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
  terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
  return NULL;
H
Haojun Liao 已提交
4147
}
H
Haojun Liao 已提交
4148

4149
static SSDataBlock* doTagScan(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
4150 4151 4152 4153
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

4154 4155 4156
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;

  STagScanInfo* pInfo = pOperator->info;
4157
  SExprInfo*    pExprInfo = &pOperator->exprSupp.pExprInfo[0];
4158
  SSDataBlock*  pRes = pInfo->pRes;
4159
  blockDataCleanup(pRes);
H
Haojun Liao 已提交
4160

H
Haojun Liao 已提交
4161
  int32_t size = tableListGetSize(pTaskInfo->pTableInfoList);
wmmhello's avatar
wmmhello 已提交
4162
  if (size == 0) {
H
Haojun Liao 已提交
4163 4164 4165 4166
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
    return NULL;
  }

4167 4168 4169
  char        str[512] = {0};
  int32_t     count = 0;
  SMetaReader mr = {0};
4170
  metaReaderInit(&mr, pInfo->readHandle.meta, 0);
H
Haojun Liao 已提交
4171

wmmhello's avatar
wmmhello 已提交
4172
  while (pInfo->curPos < size && count < pOperator->resultInfo.capacity) {
H
Haojun Liao 已提交
4173
    STableKeyInfo* item = tableListGetInfo(pInfo->pTableList, pInfo->curPos);
L
Liu Jicong 已提交
4174
    int32_t        code = metaGetTableEntryByUid(&mr, item->uid);
4175
    tDecoderClear(&mr.coder);
H
Haojun Liao 已提交
4176
    if (code != TSDB_CODE_SUCCESS) {
L
Liu Jicong 已提交
4177 4178
      qError("failed to get table meta, uid:0x%" PRIx64 ", code:%s, %s", item->uid, tstrerror(terrno),
             GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
4179
      metaReaderClear(&mr);
4180
      T_LONG_JMP(pTaskInfo->env, terrno);
H
Haojun Liao 已提交
4181
    }
H
Haojun Liao 已提交
4182

4183
    for (int32_t j = 0; j < pOperator->exprSupp.numOfExprs; ++j) {
4184 4185 4186 4187 4188 4189
      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);
        colDataAppend(pDst, count, str, false);
4190
      } else {  // it is a tag value
wmmhello's avatar
wmmhello 已提交
4191 4192
        STagVal val = {0};
        val.cid = pExprInfo[j].base.pParam[0].pCol->colId;
4193
        const char* p = metaGetTableTagVal(mr.me.ctbEntry.pTags, pDst->info.type, &val);
wmmhello's avatar
wmmhello 已提交
4194

4195 4196 4197 4198
        char* data = NULL;
        if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL) {
          data = tTagValToData((const STagVal*)p, false);
        } else {
wmmhello's avatar
wmmhello 已提交
4199 4200
          data = (char*)p;
        }
L
Liu Jicong 已提交
4201 4202
        colDataAppend(pDst, count, data,
                      (data == NULL) || (pDst->info.type == TSDB_DATA_TYPE_JSON && tTagIsJsonNull(data)));
4203

4204 4205
        if (pDst->info.type != TSDB_DATA_TYPE_JSON && p != NULL && IS_VAR_DATA_TYPE(((const STagVal*)p)->type) &&
            data != NULL) {
wmmhello's avatar
wmmhello 已提交
4206
          taosMemoryFree(data);
wmmhello's avatar
wmmhello 已提交
4207
        }
H
Haojun Liao 已提交
4208 4209 4210
      }
    }

4211
    count += 1;
wmmhello's avatar
wmmhello 已提交
4212
    if (++pInfo->curPos >= size) {
4213
      doSetOperatorCompleted(pOperator);
H
Haojun Liao 已提交
4214 4215 4216
    }
  }

4217 4218
  metaReaderClear(&mr);

4219
  // qDebug("QInfo:0x%"PRIx64" create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
H
Haojun Liao 已提交
4220
  if (pOperator->status == OP_EXEC_DONE) {
4221
    setTaskStatus(pTaskInfo, TASK_COMPLETED);
H
Haojun Liao 已提交
4222 4223 4224
  }

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

4227
  return (pRes->info.rows == 0) ? NULL : pInfo->pRes;
H
Haojun Liao 已提交
4228 4229
}

4230
static void destroyTagScanOperatorInfo(void* param) {
H
Haojun Liao 已提交
4231 4232
  STagScanInfo* pInfo = (STagScanInfo*)param;
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
H
Haojun Liao 已提交
4233
  taosArrayDestroy(pInfo->matchInfo.pList);
D
dapan1121 已提交
4234
  taosMemoryFreeClear(param);
H
Haojun Liao 已提交
4235 4236
}

4237 4238
SOperatorInfo* createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pPhyNode,
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo) {
4239
  STagScanInfo*  pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
H
Haojun Liao 已提交
4240 4241 4242 4243 4244
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }

4245 4246
  SDataBlockDescNode* pDescNode = pPhyNode->node.pOutputDataBlockDesc;

4247
  int32_t    num = 0;
4248 4249
  int32_t    numOfExprs = 0;
  SExprInfo* pExprInfo = createExprInfo(pPhyNode->pScanPseudoCols, NULL, &numOfExprs);
4250 4251
  int32_t    code =
      extractColMatchInfo(pPhyNode->pScanPseudoCols, pDescNode, &num, COL_MATCH_FROM_COL_ID, &pInfo->matchInfo);
4252 4253 4254
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
4255

H
Haojun Liao 已提交
4256
  code = initExprSupp(&pOperator->exprSupp, pExprInfo, numOfExprs);
4257 4258 4259
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
4260

4261 4262 4263 4264
  pInfo->pTableList = pTableListInfo;
  pInfo->pRes = createResDataBlock(pDescNode);
  pInfo->readHandle = *pReadHandle;
  pInfo->curPos = 0;
4265

4266
  pOperator->name = "TagScanOperator";
4267
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN;
4268

4269 4270 4271 4272
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pTaskInfo = pTaskInfo;
4273

4274
  initResultSizeInfo(&pOperator->resultInfo, 4096);
4275 4276
  blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);

4277
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doTagScan, NULL, NULL, destroyTagScanOperatorInfo, NULL);
H
Haojun Liao 已提交
4278 4279

  return pOperator;
4280

4281
_error:
H
Haojun Liao 已提交
4282 4283 4284 4285 4286
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  terrno = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
}
4287

S
slzhou 已提交
4288
int32_t createMultipleDataReaders(SQueryTableDataCond* pQueryCond, SReadHandle* pHandle, STableListInfo* pTableListInfo,
H
Haojun Liao 已提交
4289
                                  int32_t tableStartIdx, int32_t tableEndIdx, SArray* arrayReader, const char* idstr) {
S
slzhou 已提交
4290
  for (int32_t i = tableStartIdx; i <= tableEndIdx; ++i) {
H
Haojun Liao 已提交
4291
    STableKeyInfo* pList = tableListGetInfo(pTableListInfo, i);
4292 4293
    STsdbReader*   pReader = NULL;
    tsdbReaderOpen(pHandle->vnode, pQueryCond, pList, 1, &pReader, idstr);
S
slzhou 已提交
4294 4295 4296 4297 4298 4299
    taosArrayPush(arrayReader, &pReader);
  }

  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
4300
// todo refactor
4301 4302
static int32_t loadDataBlockFromOneTable(SOperatorInfo* pOperator, STableMergeScanInfo* pTableScanInfo,
                                         int32_t readerIdx, SSDataBlock* pBlock, uint32_t* status) {
L
Liu Jicong 已提交
4303
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;
S
shenglian zhou 已提交
4304
  STableMergeScanInfo* pInfo = pOperator->info;
4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330

  SFileBlockLoadRecorder* pCost = &pTableScanInfo->readRecorder;

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

  *status = pInfo->dataBlockLoadFlag;
  if (pTableScanInfo->pFilterNode != NULL ||
      overlapWithTimeWindow(&pTableScanInfo->interval, &pBlock->info, pTableScanInfo->cond.order)) {
    (*status) = FUNC_DATA_REQUIRED_DATA_LOAD;
  }

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

  if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
    pCost->filterOutBlocks += 1;
    return TSDB_CODE_SUCCESS;
  } else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
    qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
           pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
    pCost->skipBlocks += 1;

    // clear all data in pBlock that are set when handing the previous block
4331
    for (int32_t i = 0; i < taosArrayGetSize(pBlock->pDataBlock); ++i) {
4332 4333 4334 4335 4336 4337 4338 4339 4340 4341
      SColumnInfoData* pcol = taosArrayGet(pBlock->pDataBlock, i);
      pcol->pData = NULL;
    }

    return TSDB_CODE_SUCCESS;
  } else if (*status == FUNC_DATA_REQUIRED_STATIS_LOAD) {
    pCost->loadBlockStatis += 1;

    bool             allColumnsHaveAgg = true;
    SColumnDataAgg** pColAgg = NULL;
5
54liuyao 已提交
4342
    //    STsdbReader* reader = pTableScanInfo->pReader; // taosArrayGetP(pTableScanInfo->dataReaders, readerIdx);
4343 4344

    if (allColumnsHaveAgg == true) {
4345
      int32_t numOfCols = taosArrayGetSize(pBlock->pDataBlock);
4346 4347 4348 4349 4350 4351 4352

      // todo create this buffer during creating operator
      if (pBlock->pBlockAgg == NULL) {
        pBlock->pBlockAgg = taosMemoryCalloc(numOfCols, POINTER_BYTES);
      }

      for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
4353 4354
        SColMatchItem* pColMatchInfo = taosArrayGet(pTableScanInfo->matchInfo.pList, i);
        if (!pColMatchInfo->needOutput) {
4355 4356
          continue;
        }
H
Haojun Liao 已提交
4357
        pBlock->pBlockAgg[pColMatchInfo->dstSlotId] = pColAgg[i];
4358 4359 4360 4361 4362 4363 4364 4365 4366 4367 4368 4369
      }

      return TSDB_CODE_SUCCESS;
    } else {  // failed to load the block sma data, data block statistics does not exist, load data block instead
      *status = FUNC_DATA_REQUIRED_DATA_LOAD;
    }
  }

  ASSERT(*status == FUNC_DATA_REQUIRED_DATA_LOAD);

  // todo filter data block according to the block sma data firstly
#if 0
H
Haojun Liao 已提交
4370
  if (!doFilterByBlockSMA(pBlock->pBlockStatis, pTableScanInfo->pCtx, pBlockInfo->rows)) {
4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381
    pCost->filterOutBlocks += 1;
    qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo), pBlockInfo->window.skey,
           pBlockInfo->window.ekey, pBlockInfo->rows);
    (*status) = FUNC_DATA_REQUIRED_FILTEROUT;
    return TSDB_CODE_SUCCESS;
  }
#endif

  pCost->totalCheckedRows += pBlock->info.rows;
  pCost->loadBlocks += 1;

4382
  STsdbReader* reader = pTableScanInfo->pReader;
4383 4384 4385 4386 4387
  SArray*      pCols = tsdbRetrieveDataBlock(reader, NULL);
  if (pCols == NULL) {
    return terrno;
  }

H
Haojun Liao 已提交
4388
  relocateColumnData(pBlock, pTableScanInfo->matchInfo.pList, pCols, true);
4389 4390

  // currently only the tbname pseudo column
S
shenglian zhou 已提交
4391
  if (pTableScanInfo->pseudoSup.numOfExprs > 0) {
4392 4393 4394
    int32_t code =
        addTagPseudoColumnData(&pTableScanInfo->readHandle, pTableScanInfo->pseudoSup.pExprInfo,
                               pTableScanInfo->pseudoSup.numOfExprs, pBlock, pBlock->info.rows, GET_TASKID(pTaskInfo));
H
Haojun Liao 已提交
4395
    if (code != TSDB_CODE_SUCCESS) {
4396
      T_LONG_JMP(pTaskInfo->env, code);
H
Haojun Liao 已提交
4397
    }
4398 4399
  }

4400 4401
  if (pTableScanInfo->pFilterNode != NULL) {
    int64_t st = taosGetTimestampMs();
H
Haojun Liao 已提交
4402
    doFilter(pTableScanInfo->pFilterNode, pBlock, &pTableScanInfo->matchInfo, NULL);
4403

4404 4405
    double el = (taosGetTimestampUs() - st) / 1000.0;
    pTableScanInfo->readRecorder.filterTime += el;
4406

4407 4408 4409 4410 4411 4412 4413
    if (pBlock->info.rows == 0) {
      pCost->filterOutBlocks += 1;
      qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d, elapsed time:%.2f ms",
             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);
    }
4414 4415 4416 4417 4418 4419 4420 4421
  }

  return TSDB_CODE_SUCCESS;
}

typedef struct STableMergeScanSortSourceParam {
  SOperatorInfo* pOperator;
  int32_t        readerIdx;
H
Haojun Liao 已提交
4422
  uint64_t       uid;
4423
  SSDataBlock*   inputBlock;
4424 4425
} STableMergeScanSortSourceParam;

dengyihao's avatar
dengyihao 已提交
4426
static SSDataBlock* getTableDataBlockImpl(void* param) {
dengyihao's avatar
opt mem  
dengyihao 已提交
4427 4428 4429 4430 4431 4432 4433 4434 4435 4436 4437 4438 4439 4440
  STableMergeScanSortSourceParam* source = param;
  SOperatorInfo*                  pOperator = source->pOperator;
  STableMergeScanInfo*            pInfo = pOperator->info;
  SExecTaskInfo*                  pTaskInfo = pOperator->pTaskInfo;
  int32_t                         readIdx = source->readerIdx;
  SSDataBlock*                    pBlock = source->inputBlock;
  STableMergeScanInfo*            pTableScanInfo = pOperator->info;

  SQueryTableDataCond* pQueryCond = taosArrayGet(pTableScanInfo->queryConds, readIdx);

  blockDataCleanup(pBlock);

  int64_t st = taosGetTimestampUs();

5
54liuyao 已提交
4441
  void*        p = tableListGetInfo(pInfo->tableListInfo, readIdx + pInfo->tableStartIndex);
dengyihao's avatar
opt mem  
dengyihao 已提交
4442
  SReadHandle* pHandle = &pInfo->readHandle;
dengyihao's avatar
dengyihao 已提交
4443 4444 4445 4446 4447

  int32_t code = tsdbReaderOpen(pHandle->vnode, pQueryCond, p, 1, &pInfo->pReader, GET_TASKID(pTaskInfo));
  if (code != 0) {
    T_LONG_JMP(pOperator->pTaskInfo->env, code);
  }
dengyihao's avatar
opt mem  
dengyihao 已提交
4448 4449 4450 4451 4452 4453 4454 4455 4456 4457 4458 4459 4460 4461 4462

  STsdbReader* reader = pInfo->pReader;
  while (tsdbNextDataBlock(reader)) {
    if (isTaskKilled(pOperator->pTaskInfo)) {
      T_LONG_JMP(pOperator->pTaskInfo->env, TSDB_CODE_TSC_QUERY_CANCELLED);
    }

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

    blockDataCleanup(pBlock);

H
Haojun Liao 已提交
4463 4464 4465 4466
    int32_t rows = 0;
    tsdbRetrieveDataBlockInfo(reader, &rows, &pBlock->info.uid, &pBlock->info.window);
    blockDataEnsureCapacity(pBlock, rows);
    pBlock->info.rows = rows;
dengyihao's avatar
opt mem  
dengyihao 已提交
4467

H
Haojun Liao 已提交
4468
    if (pQueryCond->order == TSDB_ORDER_ASC) {
dengyihao's avatar
opt mem  
dengyihao 已提交
4469 4470 4471 4472
      pQueryCond->twindows.skey = pBlock->info.window.ekey + 1;
    } else {
      pQueryCond->twindows.ekey = pBlock->info.window.skey - 1;
    }
dengyihao's avatar
opt mem  
dengyihao 已提交
4473 4474 4475 4476 4477 4478 4479 4480 4481 4482 4483 4484

    uint32_t status = 0;
    int32_t  code = loadDataBlockFromOneTable(pOperator, pTableScanInfo, readIdx, pBlock, &status);
    if (code != TSDB_CODE_SUCCESS) {
      T_LONG_JMP(pOperator->pTaskInfo->env, code);
    }

    // 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;
    }

H
Haojun Liao 已提交
4485
    pBlock->info.groupId = getTableGroupId(pOperator->pTaskInfo->pTableInfoList, pBlock->info.uid);
dengyihao's avatar
opt mem  
dengyihao 已提交
4486

dengyihao's avatar
opt mem  
dengyihao 已提交
4487
    pOperator->resultInfo.totalRows += pBlock->info.rows;  // pTableScanInfo->readRecorder.totalRows;
dengyihao's avatar
opt mem  
dengyihao 已提交
4488 4489 4490
    pTableScanInfo->readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;

    tsdbReaderClose(pInfo->pReader);
dengyihao's avatar
opt mem  
dengyihao 已提交
4491
    pInfo->pReader = NULL;
dengyihao's avatar
opt mem  
dengyihao 已提交
4492 4493 4494
    return pBlock;
  }
  tsdbReaderClose(pInfo->pReader);
dengyihao's avatar
opt mem  
dengyihao 已提交
4495
  pInfo->pReader = NULL;
dengyihao's avatar
opt mem  
dengyihao 已提交
4496 4497 4498
  return NULL;
}

4499 4500 4501 4502
static SSDataBlock* getTableDataBlock(void* param) {
  STableMergeScanSortSourceParam* source = param;
  SOperatorInfo*                  pOperator = source->pOperator;
  int32_t                         readerIdx = source->readerIdx;
4503
  SSDataBlock*                    pBlock = source->inputBlock;
4504 4505 4506 4507
  STableMergeScanInfo*            pTableScanInfo = pOperator->info;

  int64_t st = taosGetTimestampUs();

4508 4509
  blockDataCleanup(pBlock);

H
Hongze Cheng 已提交
4510
  STsdbReader* reader = taosArrayGetP(pTableScanInfo->dataReaders, readerIdx);
4511 4512
  while (tsdbNextDataBlock(reader)) {
    if (isTaskKilled(pOperator->pTaskInfo)) {
4513
      T_LONG_JMP(pOperator->pTaskInfo->env, TSDB_CODE_TSC_QUERY_CANCELLED);
4514 4515 4516 4517 4518 4519 4520 4521
    }

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

4522
    blockDataCleanup(pBlock);
H
Haojun Liao 已提交
4523 4524 4525 4526 4527

    int32_t rows = 0;
    tsdbRetrieveDataBlockInfo(reader, &rows, &pBlock->info.uid, &pBlock->info.window);
    blockDataEnsureCapacity(pBlock, rows);
    pBlock->info.rows = rows;
4528 4529 4530

    uint32_t status = 0;
    int32_t  code = loadDataBlockFromOneTable(pOperator, pTableScanInfo, readerIdx, pBlock, &status);
4531
    //    int32_t  code = loadDataBlockOnDemand(pOperator->pRuntimeEnv, pTableScanInfo, pBlock, &status);
4532
    if (code != TSDB_CODE_SUCCESS) {
4533
      T_LONG_JMP(pOperator->pTaskInfo->env, code);
4534 4535 4536 4537 4538 4539 4540
    }

    // 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;
    }

H
Haojun Liao 已提交
4541
    pBlock->info.groupId = getTableGroupId(pOperator->pTaskInfo->pTableInfoList, pBlock->info.uid);
4542 4543 4544 4545 4546 4547 4548 4549
    pOperator->resultInfo.totalRows = pTableScanInfo->readRecorder.totalRows;
    pTableScanInfo->readRecorder.elapsedTime += (taosGetTimestampUs() - st) / 1000.0;

    return pBlock;
  }
  return NULL;
}

4550 4551 4552
SArray* generateSortByTsInfo(SArray* colMatchInfo, int32_t order) {
  int32_t tsTargetSlotId = 0;
  for (int32_t i = 0; i < taosArrayGetSize(colMatchInfo); ++i) {
H
Haojun Liao 已提交
4553
    SColMatchItem* colInfo = taosArrayGet(colMatchInfo, i);
4554
    if (colInfo->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
H
Haojun Liao 已提交
4555
      tsTargetSlotId = colInfo->dstSlotId;
4556 4557 4558
    }
  }

4559 4560 4561
  SArray*         pList = taosArrayInit(1, sizeof(SBlockOrderInfo));
  SBlockOrderInfo bi = {0};
  bi.order = order;
4562
  bi.slotId = tsTargetSlotId;
4563 4564 4565 4566 4567 4568 4569
  bi.nullFirst = NULL_ORDER_FIRST;

  taosArrayPush(pList, &bi);

  return pList;
}

dengyihao's avatar
opt mem  
dengyihao 已提交
4570 4571 4572 4573 4574 4575 4576 4577
int32_t dumpSQueryTableCond(const SQueryTableDataCond* src, SQueryTableDataCond* dst) {
  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 已提交
4578

4579
int32_t startGroupTableMergeScan(SOperatorInfo* pOperator) {
4580 4581 4582
  STableMergeScanInfo* pInfo = pOperator->info;
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;

S
slzhou 已提交
4583
  {
H
Haojun Liao 已提交
4584
    size_t  numOfTables = tableListGetSize(pInfo->tableListInfo);
S
slzhou 已提交
4585
    int32_t i = pInfo->tableStartIndex + 1;
H
Haojun Liao 已提交
4586 4587
    for (; i < numOfTables; ++i) {
      STableKeyInfo* tableKeyInfo = tableListGetInfo(pInfo->tableListInfo, i);
S
slzhou 已提交
4588 4589 4590 4591 4592 4593
      if (tableKeyInfo->groupId != pInfo->groupId) {
        break;
      }
    }
    pInfo->tableEndIndex = i - 1;
  }
4594

S
slzhou 已提交
4595 4596
  int32_t tableStartIdx = pInfo->tableStartIndex;
  int32_t tableEndIdx = pInfo->tableEndIndex;
4597

dengyihao's avatar
opt mem  
dengyihao 已提交
4598
  pInfo->pReader = NULL;
4599

4600 4601
  // 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 已提交
4602
  pInfo->sortBufSize = pInfo->bufPageSize * (tableEndIdx - tableStartIdx + 1 + 1);
4603
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
L
Liu Jicong 已提交
4604 4605
  pInfo->pSortHandle = tsortCreateSortHandle(pInfo->pSortInfo, SORT_MULTISOURCE_MERGE, pInfo->bufPageSize, numOfBufPage,
                                             pInfo->pSortInputBlock, pTaskInfo->id.str);
4606

dengyihao's avatar
dengyihao 已提交
4607
  tsortSetFetchRawDataFp(pInfo->pSortHandle, getTableDataBlockImpl, NULL, NULL);
dengyihao's avatar
opt mem  
dengyihao 已提交
4608 4609 4610 4611 4612 4613

  // 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) {
4614 4615 4616 4617 4618
    STableMergeScanSortSourceParam param = {0};
    param.readerIdx = i;
    param.pOperator = pOperator;
    param.inputBlock = createOneDataBlock(pInfo->pResBlock, false);
    taosArrayPush(pInfo->sortSourceParams, &param);
dengyihao's avatar
opt mem  
dengyihao 已提交
4619 4620 4621 4622

    SQueryTableDataCond cond;
    dumpSQueryTableCond(&pInfo->cond, &cond);
    taosArrayPush(pInfo->queryConds, &cond);
4623 4624
  }

dengyihao's avatar
opt mem  
dengyihao 已提交
4625
  for (int32_t i = 0; i < numOfTable; ++i) {
4626
    SSortSource*                    ps = taosMemoryCalloc(1, sizeof(SSortSource));
4627
    STableMergeScanSortSourceParam* param = taosArrayGet(pInfo->sortSourceParams, i);
4628 4629 4630 4631 4632 4633 4634
    ps->param = param;
    tsortAddSource(pInfo->pSortHandle, ps);
  }

  int32_t code = tsortOpen(pInfo->pSortHandle);

  if (code != TSDB_CODE_SUCCESS) {
4635
    T_LONG_JMP(pTaskInfo->env, terrno);
4636 4637
  }

4638 4639 4640 4641 4642 4643 4644
  return TSDB_CODE_SUCCESS;
}

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

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

4647 4648 4649 4650 4651 4652 4653
  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 已提交
4654
  for (int32_t i = 0; i < numOfTable; ++i) {
4655 4656 4657
    STableMergeScanSortSourceParam* param = taosArrayGet(pInfo->sortSourceParams, i);
    blockDataDestroy(param->inputBlock);
  }
4658 4659
  taosArrayClear(pInfo->sortSourceParams);

4660 4661
  tsortDestroySortHandle(pInfo->pSortHandle);

dengyihao's avatar
opt mem  
dengyihao 已提交
4662 4663 4664
  for (int32_t i = 0; i < taosArrayGetSize(pInfo->queryConds); i++) {
    SQueryTableDataCond* cond = taosArrayGet(pInfo->queryConds, i);
    taosMemoryFree(cond->colList);
4665
  }
dengyihao's avatar
opt mem  
dengyihao 已提交
4666 4667 4668
  taosArrayDestroy(pInfo->queryConds);
  pInfo->queryConds = NULL;

4669 4670 4671
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
4672 4673
SSDataBlock* getSortedTableMergeScanBlockData(SSortHandle* pHandle, SSDataBlock* pResBlock, int32_t capacity,
                                              SOperatorInfo* pOperator) {
4674 4675 4676
  STableMergeScanInfo* pInfo = pOperator->info;
  SExecTaskInfo*       pTaskInfo = pOperator->pTaskInfo;

4677 4678
  blockDataCleanup(pResBlock);
  blockDataEnsureCapacity(pResBlock, capacity);
4679 4680

  while (1) {
4681
    STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
4682 4683 4684 4685
    if (pTupleHandle == NULL) {
      break;
    }

4686 4687
    appendOneRowToDataBlock(pResBlock, pTupleHandle);
    if (pResBlock->info.rows >= capacity) {
4688 4689 4690 4691
      break;
    }
  }

4692
  qDebug("%s get sorted row blocks, rows:%d", GET_TASKID(pTaskInfo), pResBlock->info.rows);
4693 4694 4695
  applyLimitOffset(&pInfo->limitInfo, pResBlock, pTaskInfo, pOperator);
  pInfo->limitInfo.numOfOutputRows += pResBlock->info.rows;

4696
  return (pResBlock->info.rows > 0) ? pResBlock : NULL;
4697 4698 4699 4700 4701 4702 4703 4704 4705 4706 4707 4708
}

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) {
4709
    T_LONG_JMP(pTaskInfo->env, code);
4710
  }
4711

H
Haojun Liao 已提交
4712
  size_t tableListSize = tableListGetSize(pInfo->tableListInfo);
S
slzhou 已提交
4713 4714
  if (!pInfo->hasGroupId) {
    pInfo->hasGroupId = true;
4715

S
slzhou 已提交
4716
    if (tableListSize == 0) {
4717 4718 4719
      doSetOperatorCompleted(pOperator);
      return NULL;
    }
S
slzhou 已提交
4720
    pInfo->tableStartIndex = 0;
H
Haojun Liao 已提交
4721
    pInfo->groupId = ((STableKeyInfo*)tableListGetInfo(pInfo->tableListInfo, pInfo->tableStartIndex))->groupId;
4722 4723
    startGroupTableMergeScan(pOperator);
  }
4724

S
slzhou 已提交
4725 4726
  SSDataBlock* pBlock = NULL;
  while (pInfo->tableStartIndex < tableListSize) {
L
Liu Jicong 已提交
4727 4728
    pBlock = getSortedTableMergeScanBlockData(pInfo->pSortHandle, pInfo->pResBlock, pOperator->resultInfo.capacity,
                                              pOperator);
S
slzhou 已提交
4729 4730 4731 4732 4733 4734 4735 4736 4737 4738 4739
    if (pBlock != NULL) {
      pBlock->info.groupId = pInfo->groupId;
      pOperator->resultInfo.totalRows += pBlock->info.rows;
      return pBlock;
    } else {
      stopGroupTableMergeScan(pOperator);
      if (pInfo->tableEndIndex >= tableListSize - 1) {
        doSetOperatorCompleted(pOperator);
        break;
      }
      pInfo->tableStartIndex = pInfo->tableEndIndex + 1;
H
Haojun Liao 已提交
4740
      pInfo->groupId = tableListGetInfo(pInfo->tableListInfo, pInfo->tableStartIndex)->groupId;
S
slzhou 已提交
4741 4742
      startGroupTableMergeScan(pOperator);
    }
wmmhello's avatar
wmmhello 已提交
4743 4744
  }

4745 4746 4747
  return pBlock;
}

4748
void destroyTableMergeScanOperatorInfo(void* param) {
4749
  STableMergeScanInfo* pTableScanInfo = (STableMergeScanInfo*)param;
4750
  cleanupQueryTableDataCond(&pTableScanInfo->cond);
4751

dengyihao's avatar
dengyihao 已提交
4752 4753 4754
  int32_t numOfTable = taosArrayGetSize(pTableScanInfo->queryConds);

  for (int32_t i = 0; i < numOfTable; i++) {
H
Haojun Liao 已提交
4755 4756
    STableMergeScanSortSourceParam* p = taosArrayGet(pTableScanInfo->sortSourceParams, i);
    blockDataDestroy(p->inputBlock);
4757
  }
H
Haojun Liao 已提交
4758

4759
  taosArrayDestroy(pTableScanInfo->sortSourceParams);
4760

dengyihao's avatar
opt mem  
dengyihao 已提交
4761
  tsdbReaderClose(pTableScanInfo->pReader);
dengyihao's avatar
dengyihao 已提交
4762
  pTableScanInfo->pReader = NULL;
dengyihao's avatar
opt mem  
dengyihao 已提交
4763

dengyihao's avatar
opt mem  
dengyihao 已提交
4764 4765 4766
  for (int i = 0; i < taosArrayGetSize(pTableScanInfo->queryConds); i++) {
    SQueryTableDataCond* pCond = taosArrayGet(pTableScanInfo->queryConds, i);
    taosMemoryFree(pCond->colList);
4767
  }
dengyihao's avatar
opt mem  
dengyihao 已提交
4768
  taosArrayDestroy(pTableScanInfo->queryConds);
4769

H
Haojun Liao 已提交
4770 4771
  if (pTableScanInfo->matchInfo.pList != NULL) {
    taosArrayDestroy(pTableScanInfo->matchInfo.pList);
4772 4773 4774 4775 4776 4777
  }

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

  taosArrayDestroy(pTableScanInfo->pSortInfo);
4778
  cleanupExprSupp(&pTableScanInfo->pseudoSup);
L
Liu Jicong 已提交
4779

4780
  taosMemoryFreeClear(pTableScanInfo->rowEntryInfoOffset);
D
dapan1121 已提交
4781
  taosMemoryFreeClear(param);
4782 4783
}

4784 4785
typedef struct STableMergeScanExecInfo {
  SFileBlockLoadRecorder blockRecorder;
L
Liu Jicong 已提交
4786
  SSortExecInfo          sortExecInfo;
4787 4788
} STableMergeScanExecInfo;

4789 4790
int32_t getTableMergeScanExplainExecInfo(SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len) {
  ASSERT(pOptr != NULL);
4791 4792
  // TODO: merge these two info into one struct
  STableMergeScanExecInfo* execInfo = taosMemoryCalloc(1, sizeof(STableMergeScanExecInfo));
L
Liu Jicong 已提交
4793
  STableMergeScanInfo*     pInfo = pOptr->info;
4794
  execInfo->blockRecorder = pInfo->readRecorder;
4795
  execInfo->sortExecInfo = pInfo->sortExecInfo;
4796 4797 4798

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

4800 4801 4802
  return TSDB_CODE_SUCCESS;
}

S
slzhou 已提交
4803 4804 4805
int32_t compareTableKeyInfoByGid(const void* p1, const void* p2) {
  const STableKeyInfo* info1 = p1;
  const STableKeyInfo* info2 = p2;
4806 4807 4808 4809 4810 4811 4812
  if (info1->groupId < info2->groupId) {
    return -1;
  } else if (info1->groupId > info2->groupId) {
    return 1;
  } else {
    return 0;
  }
S
slzhou 已提交
4813 4814
}

4815
SOperatorInfo* createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, STableListInfo* pTableListInfo,
4816
                                                SReadHandle* readHandle, SExecTaskInfo* pTaskInfo) {
4817 4818 4819 4820 4821
  STableMergeScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableMergeScanInfo));
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
4822

4823 4824 4825
  SDataBlockDescNode* pDescNode = pTableScanNode->scan.node.pOutputDataBlockDesc;

  int32_t numOfCols = 0;
4826 4827
  int32_t code = extractColMatchInfo(pTableScanNode->scan.pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID,
                                     &pInfo->matchInfo);
4828

H
Haojun Liao 已提交
4829
  code = initQueryTableDataCond(&pInfo->cond, pTableScanNode);
4830
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
4831
    taosArrayDestroy(pInfo->matchInfo.pList);
4832 4833 4834 4835
    goto _error;
  }

  if (pTableScanNode->scan.pScanPseudoCols != NULL) {
4836 4837 4838
    SExprSupp* pSup = &pInfo->pseudoSup;
    pSup->pExprInfo = createExprInfo(pTableScanNode->scan.pScanPseudoCols, NULL, &pSup->numOfExprs);
    pSup->pCtx = createSqlFunctionCtx(pSup->pExprInfo, pSup->numOfExprs, &pSup->rowEntryInfoOffset);
4839 4840 4841 4842
  }

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

L
Liu Jicong 已提交
4843 4844
  pInfo->readHandle = *readHandle;
  pInfo->interval = extractIntervalInfo(pTableScanNode);
4845
  pInfo->sample.sampleRatio = pTableScanNode->ratio;
L
Liu Jicong 已提交
4846 4847 4848
  pInfo->sample.seed = taosGetTimestampSec();
  pInfo->dataBlockLoadFlag = pTableScanNode->dataRequired;
  pInfo->pFilterNode = pTableScanNode->scan.node.pConditions;
4849
  pInfo->tableListInfo = pTableListInfo;
L
Liu Jicong 已提交
4850
  pInfo->scanFlag = MAIN_SCAN;
4851 4852

  pInfo->pResBlock = createResDataBlock(pDescNode);
4853
  pInfo->sortSourceParams = taosArrayInit(64, sizeof(STableMergeScanSortSourceParam));
4854

H
Haojun Liao 已提交
4855
  pInfo->pSortInfo = generateSortByTsInfo(pInfo->matchInfo.pList, pInfo->cond.order);
4856
  pInfo->pSortInputBlock = createOneDataBlock(pInfo->pResBlock, false);
4857
  initLimitInfo(pTableScanNode->scan.node.pLimit, pTableScanNode->scan.node.pSlimit, &pInfo->limitInfo);
4858

4859
  int32_t rowSize = pInfo->pResBlock->info.rowSize;
L
Liu Jicong 已提交
4860
  pInfo->bufPageSize = getProperSortPageSize(rowSize);
4861

L
Liu Jicong 已提交
4862
  pOperator->name = "TableMergeScanOperator";
4863
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TABLE_MERGE_SCAN;
L
Liu Jicong 已提交
4864 4865 4866
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
L
Liu Jicong 已提交
4867
  pOperator->exprSupp.numOfExprs = numOfCols;
L
Liu Jicong 已提交
4868
  pOperator->pTaskInfo = pTaskInfo;
4869
  initResultSizeInfo(&pOperator->resultInfo, 1024);
4870

4871 4872
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doTableMergeScan, NULL, NULL,
                                         destroyTableMergeScanOperatorInfo, getTableMergeScanExplainExecInfo);
4873 4874 4875 4876 4877 4878 4879 4880 4881
  pOperator->cost.openCost = 0;
  return pOperator;

_error:
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  return NULL;
}