executorimpl.c 191.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * 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/>.
 */
15

H
Haojun Liao 已提交
16 17
#include "filter.h"
#include "function.h"
18 19
#include "functionMgt.h"
#include "os.h"
H
Haojun Liao 已提交
20
#include "querynodes.h"
21
#include "tfill.h"
dengyihao's avatar
dengyihao 已提交
22
#include "tname.h"
23

H
Haojun Liao 已提交
24
#include "tdatablock.h"
25
#include "tglobal.h"
H
Haojun Liao 已提交
26
#include "tmsg.h"
H
Haojun Liao 已提交
27
#include "tsort.h"
28
#include "ttime.h"
H
Haojun Liao 已提交
29

30
#include "executorimpl.h"
dengyihao's avatar
dengyihao 已提交
31
#include "index.h"
32
#include "query.h"
33 34
#include "tcompare.h"
#include "tcompression.h"
H
Haojun Liao 已提交
35
#include "thash.h"
36
#include "ttypes.h"
dengyihao's avatar
dengyihao 已提交
37
#include "vnode.h"
38

H
Haojun Liao 已提交
39
#define IS_MAIN_SCAN(runtime)          ((runtime)->scanFlag == MAIN_SCAN)
40 41
#define IS_REVERSE_SCAN(runtime)       ((runtime)->scanFlag == REVERSE_SCAN)
#define IS_REPEAT_SCAN(runtime)        ((runtime)->scanFlag == REPEAT_SCAN)
L
Liu Jicong 已提交
42
#define SET_MAIN_SCAN_FLAG(runtime)    ((runtime)->scanFlag = MAIN_SCAN)
43 44
#define SET_REVERSE_SCAN_FLAG(runtime) ((runtime)->scanFlag = REVERSE_SCAN)

L
Liu Jicong 已提交
45 46
#define SDATA_BLOCK_INITIALIZER \
  (SDataBlockInfo) { {0}, 0 }
47 48 49 50

#define GET_FORWARD_DIRECTION_FACTOR(ord) (((ord) == TSDB_ORDER_ASC) ? QUERY_ASC_FORWARD_STEP : QUERY_DESC_FORWARD_STEP)

enum {
L
Liu Jicong 已提交
51 52
  TS_JOIN_TS_EQUAL = 0,
  TS_JOIN_TS_NOT_EQUALS = 1,
53 54 55 56 57
  TS_JOIN_TAG_NOT_EQUALS = 2,
};

#if 0
static UNUSED_FUNC void *u_malloc (size_t __size) {
wafwerar's avatar
wafwerar 已提交
58
  uint32_t v = taosRand();
59 60 61 62

  if (v % 1000 <= 0) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
63
    return taosMemoryMalloc(__size);
64 65 66 67
  }
}

static UNUSED_FUNC void* u_calloc(size_t num, size_t __size) {
wafwerar's avatar
wafwerar 已提交
68
  uint32_t v = taosRand();
69 70 71
  if (v % 1000 <= 0) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
72
    return taosMemoryCalloc(num, __size);
73 74 75 76
  }
}

static UNUSED_FUNC void* u_realloc(void* p, size_t __size) {
wafwerar's avatar
wafwerar 已提交
77
  uint32_t v = taosRand();
78 79 80
  if (v % 5 <= 1) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
81
    return taosMemoryRealloc(p, __size);
82 83 84 85 86 87 88 89
  }
}

#define calloc  u_calloc
#define malloc  u_malloc
#define realloc u_realloc
#endif

dengyihao's avatar
dengyihao 已提交
90
#define CLEAR_QUERY_STATUS(q, st) ((q)->status &= (~(st)))
wmmhello's avatar
wmmhello 已提交
91
//#define GET_NUM_OF_TABLEGROUP(q)    taosArrayGetSize((q)->tableqinfoGroupInfo.pGroupList)
92 93
#define QUERY_IS_INTERVAL_QUERY(_q) ((_q)->interval.interval > 0)

L
Liu Jicong 已提交
94 95 96
int32_t getMaximumIdleDurationSec() { return tsShellActivityTimer * 2; }

static int32_t getExprFunctionId(SExprInfo* pExprInfo) {
97
  assert(pExprInfo != NULL && pExprInfo->pExpr != NULL && pExprInfo->pExpr->nodeType == TEXPR_UNARYEXPR_NODE);
98
  return 0;
99 100 101 102
}

static void doSetTagValueToResultBuf(char* output, const char* val, int16_t type, int16_t bytes);

103
static void setBlockStatisInfo(SqlFunctionCtx* pCtx, SExprInfo* pExpr, SSDataBlock* pSDataBlock);
104

L
Liu Jicong 已提交
105
static void destroyTableQueryInfoImpl(STableQueryInfo* pTableQueryInfo);
106 107 108

static SColumnInfo* extractColumnFilterInfo(SExprInfo* pExpr, int32_t numOfOutput, int32_t* numOfFilterCols);

X
Xiaoyu Wang 已提交
109
static void releaseQueryBuf(size_t numOfTables);
110 111 112 113 114 115

static void destroySFillOperatorInfo(void* param, int32_t numOfOutput);
static void destroyProjectOperatorInfo(void* param, int32_t numOfOutput);
static void destroyTagScanOperatorInfo(void* param, int32_t numOfOutput);
static void destroyOrderOperatorInfo(void* param, int32_t numOfOutput);
static void destroyAggOperatorInfo(void* param, int32_t numOfOutput);
X
Xiaoyu Wang 已提交
116

H
Haojun Liao 已提交
117
static void destroyIntervalOperatorInfo(void* param, int32_t numOfOutput);
H
Haojun Liao 已提交
118 119
static void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput);

120
static void destroyOperatorInfo(SOperatorInfo* pOperator);
121
static void destroySysTableScannerOperatorInfo(void* param, int32_t numOfOutput);
122

123
void doSetOperatorCompleted(SOperatorInfo* pOperator) {
124
  pOperator->status = OP_EXEC_DONE;
125

126
  pOperator->cost.totalCost = (taosGetTimestampUs() - pOperator->pTaskInfo->cost.start * 1000) / 1000.0;
H
Haojun Liao 已提交
127
  if (pOperator->pTaskInfo != NULL) {
128
    setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
129 130
  }
}
131

H
Haojun Liao 已提交
132
int32_t operatorDummyOpenFn(SOperatorInfo* pOperator) {
133
  OPTR_SET_OPENED(pOperator);
134
  pOperator->cost.openCost = 0;
H
Haojun Liao 已提交
135
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
136 137
}

138
SOperatorFpSet createOperatorFpSet(__optr_open_fn_t openFn, __optr_fn_t nextFn, __optr_fn_t streamFn,
L
Liu Jicong 已提交
139
                                   __optr_fn_t cleanup, __optr_close_fn_t closeFn, __optr_encode_fn_t encode,
140
                                   __optr_decode_fn_t decode, __optr_explain_fn_t explain) {
141 142 143 144 145 146 147 148 149 150 151 152 153 154
  SOperatorFpSet fpSet = {
      ._openFn = openFn,
      .getNextFn = nextFn,
      .getStreamResFn = streamFn,
      .cleanupFn = cleanup,
      .closeFn = closeFn,
      .encodeResultRow = encode,
      .decodeResultRow = decode,
      .getExplainFn = explain,
  };

  return fpSet;
}

H
Haojun Liao 已提交
155
void operatorDummyCloseFn(void* param, int32_t numOfCols) {}
H
Haojun Liao 已提交
156

X
Xiaoyu Wang 已提交
157 158 159
static int32_t doCopyToSDataBlock(SExecTaskInfo* taskInfo, SSDataBlock* pBlock, SExprInfo* pExprInfo,
                                  SDiskbasedBuf* pBuf, SGroupResInfo* pGroupResInfo, const int32_t* rowCellOffset,
                                  SqlFunctionCtx* pCtx, int32_t numOfExprs);
H
Haojun Liao 已提交
160

161
static void initCtxOutputBuffer(SqlFunctionCtx* pCtx, int32_t size);
H
Haojun Liao 已提交
162
static void setResultBufSize(STaskAttr* pQueryAttr, SResultInfo* pResultInfo);
dengyihao's avatar
dengyihao 已提交
163 164
static void doSetTableGroupOutputBuf(SAggOperatorInfo* pAggInfo, int32_t numOfOutput, uint64_t groupId,
                                     SExecTaskInfo* pTaskInfo);
165

H
Haojun Liao 已提交
166
SArray* getOrderCheckColumns(STaskAttr* pQuery);
167 168

typedef struct SRowCompSupporter {
L
Liu Jicong 已提交
169 170 171
  STaskRuntimeEnv* pRuntimeEnv;
  int16_t          dataOffset;
  __compar_fn_t    comFunc;
172 173
} SRowCompSupporter;

L
Liu Jicong 已提交
174 175 176
static int compareRowData(const void* a, const void* b, const void* userData) {
  const SResultRow* pRow1 = (const SResultRow*)a;
  const SResultRow* pRow2 = (const SResultRow*)b;
177

L
Liu Jicong 已提交
178 179
  SRowCompSupporter* supporter = (SRowCompSupporter*)userData;
  STaskRuntimeEnv*   pRuntimeEnv = supporter->pRuntimeEnv;
180

L
Liu Jicong 已提交
181 182
  SFilePage* page1 = getBufPage(pRuntimeEnv->pResultBuf, pRow1->pageId);
  SFilePage* page2 = getBufPage(pRuntimeEnv->pResultBuf, pRow2->pageId);
183 184

  int16_t offset = supporter->dataOffset;
185
  return 0;
X
Xiaoyu Wang 已提交
186 187
  //  char*   in1 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page1, pRow1->offset, offset);
  //  char*   in2 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page2, pRow2->offset, offset);
188

X
Xiaoyu Wang 已提交
189
  //  return (in1 != NULL && in2 != NULL) ? supporter->comFunc(in1, in2) : 0;
190 191
}

L
Liu Jicong 已提交
192
// setup the output buffer for each operator
193
SSDataBlock* createResDataBlock(SDataBlockDescNode* pNode) {
H
Haojun Liao 已提交
194
  int32_t numOfCols = LIST_LENGTH(pNode->pSlots);
H
Haojun Liao 已提交
195

wafwerar's avatar
wafwerar 已提交
196
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
H
Haojun Liao 已提交
197 198
  pBlock->pDataBlock = taosArrayInit(numOfCols, sizeof(SColumnInfoData));

H
Haojun Liao 已提交
199
  pBlock->info.blockId = pNode->dataBlockId;
dengyihao's avatar
dengyihao 已提交
200
  pBlock->info.rowSize = pNode->totalRowSize;  // todo ??
5
54liuyao 已提交
201
  pBlock->info.type = STREAM_INVALID;
H
Haojun Liao 已提交
202

L
Liu Jicong 已提交
203
  for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
204
    SColumnInfoData idata = {{0}};
L
Liu Jicong 已提交
205
    SSlotDescNode*  pDescNode = nodesListGetNode(pNode->pSlots, i);
X
Xiaoyu Wang 已提交
206 207 208
    //    if (!pDescNode->output) {  // todo disable it temporarily
    //      continue;
    //    }
H
Haojun Liao 已提交
209

dengyihao's avatar
dengyihao 已提交
210
    idata.info.type = pDescNode->dataType.type;
L
Liu Jicong 已提交
211 212
    idata.info.bytes = pDescNode->dataType.bytes;
    idata.info.scale = pDescNode->dataType.scale;
H
Haojun Liao 已提交
213
    idata.info.slotId = pDescNode->slotId;
H
Haojun Liao 已提交
214 215
    idata.info.precision = pDescNode->dataType.precision;

H
Hongze Cheng 已提交
216 217 218 219
    if (IS_VAR_DATA_TYPE(idata.info.type)) {
      pBlock->info.hasVarCol = true;
    }

H
Haojun Liao 已提交
220
    taosArrayPush(pBlock->pDataBlock, &idata);
H
Haojun Liao 已提交
221
  }
H
Haojun Liao 已提交
222

223
  pBlock->info.numOfCols = taosArrayGetSize(pBlock->pDataBlock);
H
Haojun Liao 已提交
224
  return pBlock;
H
Haojun Liao 已提交
225 226
}

L
Liu Jicong 已提交
227
static bool hasNull(SColumn* pColumn, SColumnDataAgg* pStatis) {
dengyihao's avatar
dengyihao 已提交
228 229
  if (TSDB_COL_IS_TAG(pColumn->flag) || TSDB_COL_IS_UD_COL(pColumn->flag) ||
      pColumn->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
230 231 232 233 234 235 236 237 238 239
    return false;
  }

  if (pStatis != NULL && pStatis->numOfNull == 0) {
    return false;
  }

  return true;
}

L
Liu Jicong 已提交
240 241
static bool chkResultRowFromKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, char* pData,
                                int16_t bytes, bool masterscan, uint64_t uid) {
242 243 244
  bool existed = false;
  SET_RES_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid);

L
Liu Jicong 已提交
245 246
  SResultRow** p1 =
      (SResultRow**)taosHashGet(pRuntimeEnv->pResultRowHashTable, pRuntimeEnv->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
247 248 249 250 251 252 253 254 255 256 257

  // in case of repeat scan/reverse scan, no new time window added.
  if (QUERY_IS_INTERVAL_QUERY(pRuntimeEnv->pQueryAttr)) {
    if (!masterscan) {  // the *p1 may be NULL in case of sliding+offset exists.
      return p1 != NULL;
    }

    if (p1 != NULL) {
      if (pResultRowInfo->size == 0) {
        existed = false;
      } else if (pResultRowInfo->size == 1) {
dengyihao's avatar
dengyihao 已提交
258
        //        existed = (pResultRowInfo->pResult[0] == (*p1));
259 260
      } else {  // check if current pResultRowInfo contains the existed pResultRow
        SET_RES_EXT_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid, pResultRowInfo);
L
Liu Jicong 已提交
261 262
        int64_t* index =
            taosHashGet(pRuntimeEnv->pResultRowListSet, pRuntimeEnv->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes));
263 264 265 266 267 268 269 270 271 272 273 274 275 276
        if (index != NULL) {
          existed = true;
        } else {
          existed = false;
        }
      }
    }

    return existed;
  }

  return p1 != NULL;
}

277
SResultRow* getNewResultRow(SDiskbasedBuf* pResultBuf, int64_t tableGroupId, int32_t interBufSize) {
L
Liu Jicong 已提交
278
  SFilePage* pData = NULL;
279 280 281 282 283 284 285 286 287 288 289 290 291

  // in the first scan, new space needed for results
  int32_t pageId = -1;
  SIDList list = getDataBufPagesIdList(pResultBuf, tableGroupId);

  if (taosArrayGetSize(list) == 0) {
    pData = getNewBufPage(pResultBuf, tableGroupId, &pageId);
    pData->num = sizeof(SFilePage);
  } else {
    SPageInfo* pi = getLastPageInfo(list);
    pData = getBufPage(pResultBuf, getPageId(pi));
    pageId = getPageId(pi);

wmmhello's avatar
wmmhello 已提交
292
    if (pData->num + interBufSize > getBufPageSize(pResultBuf)) {
293 294 295 296 297 298 299 300 301 302 303 304 305 306
      // release current page first, and prepare the next one
      releaseBufPageInfo(pResultBuf, pi);

      pData = getNewBufPage(pResultBuf, tableGroupId, &pageId);
      if (pData != NULL) {
        pData->num = sizeof(SFilePage);
      }
    }
  }

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

307 308
  setBufPageDirty(pData, true);

309 310 311 312 313
  // set the number of rows in current disk page
  SResultRow* pResultRow = (SResultRow*)((char*)pData + pData->num);
  pResultRow->pageId = pageId;
  pResultRow->offset = (int32_t)pData->num;

wmmhello's avatar
wmmhello 已提交
314
  pData->num += interBufSize;
315 316 317 318

  return pResultRow;
}

319 320 321 322 323 324 325
/**
 * the struct of key in hash table
 * +----------+---------------+
 * | group id |   key data    |
 * | 8 bytes  | actual length |
 * +----------+---------------+
 */
326 327 328
SResultRow* doSetResultOutBufByKey(SDiskbasedBuf* pResultBuf, SResultRowInfo* pResultRowInfo, char* pData,
                                   int16_t bytes, bool masterscan, uint64_t groupId, SExecTaskInfo* pTaskInfo,
                                   bool isIntervalQuery, SAggSupporter* pSup) {
329
  SET_RES_WINDOW_KEY(pSup->keyBuf, pData, bytes, groupId);
H
Haojun Liao 已提交
330

dengyihao's avatar
dengyihao 已提交
331 332
  SResultRowPosition* p1 =
      (SResultRowPosition*)taosHashGet(pSup->pResultRowHashTable, pSup->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
H
Haojun Liao 已提交
333

334 335
  SResultRow* pResult = NULL;

H
Haojun Liao 已提交
336 337
  // in case of repeat scan/reverse scan, no new time window added.
  if (isIntervalQuery) {
338 339
    if (masterscan && p1 != NULL) {  // the *p1 may be NULL in case of sliding+offset exists.
      pResult = getResultRowByPos(pResultBuf, p1);
340
      ASSERT(pResult->pageId == p1->pageId && pResult->offset == p1->offset);
H
Haojun Liao 已提交
341 342
    }
  } else {
dengyihao's avatar
dengyihao 已提交
343 344
    // In case of group by column query, the required SResultRow object must be existInCurrentResusltRowInfo in the
    // pResultRowInfo object.
H
Haojun Liao 已提交
345
    if (p1 != NULL) {
346
      // todo
347
      pResult = getResultRowByPos(pResultBuf, p1);
348
      ASSERT(pResult->pageId == p1->pageId && pResult->offset == p1->offset);
H
Haojun Liao 已提交
349 350 351
    }
  }

L
Liu Jicong 已提交
352
  // 1. close current opened time window
353
  if (pResultRowInfo->cur.pageId != -1 && ((pResult == NULL) || (pResult->pageId != pResultRowInfo->cur.pageId &&
L
Liu Jicong 已提交
354
                                                                 pResult->offset != pResultRowInfo->cur.offset))) {
355
    SResultRowPosition pos = pResultRowInfo->cur;
X
Xiaoyu Wang 已提交
356
    SFilePage*         pPage = getBufPage(pResultBuf, pos.pageId);
357 358 359 360 361
    releaseBufPage(pResultBuf, pPage);
  }

  // allocate a new buffer page
  if (pResult == NULL) {
H
Haojun Liao 已提交
362
    ASSERT(pSup->resultRowSize > 0);
363 364
    pResult = getNewResultRow(pResultBuf, groupId, pSup->resultRowSize);

365
    initResultRow(pResult);
H
Haojun Liao 已提交
366

367 368
    // add a new result set for a new group
    SResultRowPosition pos = {.pageId = pResult->pageId, .offset = pResult->offset};
X
Xiaoyu Wang 已提交
369 370
    taosHashPut(pSup->pResultRowHashTable, pSup->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes), &pos,
                sizeof(SResultRowPosition));
H
Haojun Liao 已提交
371 372
  }

373 374 375
  // 2. set the new time window to be the new active time window
  pResultRowInfo->cur = (SResultRowPosition){.pageId = pResult->pageId, .offset = pResult->offset};

H
Haojun Liao 已提交
376
  // too many time window in query
377
  if (taosHashGetSize(pSup->pResultRowHashTable) > MAX_INTERVAL_TIME_WINDOW) {
H
Haojun Liao 已提交
378 379 380
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_TOO_MANY_TIMEWINDOW);
  }

H
Haojun Liao 已提交
381
  return pResult;
H
Haojun Liao 已提交
382 383
}

384
// get the correct time window according to the handled timestamp
L
Liu Jicong 已提交
385
static STimeWindow getCurrentActiveTimeWindow(SResultRowInfo* pResultRowInfo, int64_t ts, STaskAttr* pQueryAttr) {
386
  STimeWindow w = {0};
H
Haojun Liao 已提交
387
#if 0
L
Liu Jicong 已提交
388 389
  if (pResultRowInfo->curPos == -1) {  // the first window, from the previous stored value
                                       //    getInitialStartTimeWindow(pQueryAttr, ts, &w);
390 391

    if (pQueryAttr->interval.intervalUnit == 'n' || pQueryAttr->interval.intervalUnit == 'y') {
L
Liu Jicong 已提交
392 393 394
      w.ekey =
          taosTimeAdd(w.skey, pQueryAttr->interval.interval, pQueryAttr->interval.intervalUnit, pQueryAttr->precision) -
          1;
395 396 397 398
    } else {
      w.ekey = w.skey + pQueryAttr->interval.interval - 1;
    }
  } else {
H
Haojun Liao 已提交
399
    w = pRow->win;
400 401 402 403 404 405 406 407 408
  }

  /*
   * query border check, skey should not be bounded by the query time range, since the value skey will
   * be used as the time window index value. So we only change ekey of time window accordingly.
   */
  if (w.ekey > pQueryAttr->window.ekey && QUERY_IS_ASC_QUERY(pQueryAttr)) {
    w.ekey = pQueryAttr->window.ekey;
  }
H
Haojun Liao 已提交
409
#endif
410 411 412 413 414

  return w;
}

// a new buffer page for each table. Needs to opt this design
L
Liu Jicong 已提交
415
static int32_t addNewWindowResultBuf(SResultRow* pWindowRes, SDiskbasedBuf* pResultBuf, int32_t tid, uint32_t size) {
416 417 418 419
  if (pWindowRes->pageId != -1) {
    return 0;
  }

L
Liu Jicong 已提交
420
  SFilePage* pData = NULL;
421 422 423 424 425 426

  // in the first scan, new space needed for results
  int32_t pageId = -1;
  SIDList list = getDataBufPagesIdList(pResultBuf, tid);

  if (taosArrayGetSize(list) == 0) {
H
Haojun Liao 已提交
427
    pData = getNewBufPage(pResultBuf, tid, &pageId);
428
    pData->num = sizeof(SFilePage);
429 430
  } else {
    SPageInfo* pi = getLastPageInfo(list);
431
    pData = getBufPage(pResultBuf, getPageId(pi));
432
    pageId = getPageId(pi);
433

434
    if (pData->num + size > getBufPageSize(pResultBuf)) {
435
      // release current page first, and prepare the next one
436
      releaseBufPageInfo(pResultBuf, pi);
437

H
Haojun Liao 已提交
438
      pData = getNewBufPage(pResultBuf, tid, &pageId);
439
      if (pData != NULL) {
440
        pData->num = sizeof(SFilePage);
441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460
      }
    }
  }

  if (pData == NULL) {
    return -1;
  }

  // set the number of rows in current disk page
  if (pWindowRes->pageId == -1) {  // not allocated yet, allocate new buffer
    pWindowRes->pageId = pageId;
    pWindowRes->offset = (int32_t)pData->num;

    pData->num += size;
    assert(pWindowRes->pageId >= 0);
  }

  return 0;
}

L
Liu Jicong 已提交
461 462 463
static bool chkWindowOutputBufByKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, STimeWindow* win,
                                    bool masterscan, SResultRow** pResult, int64_t groupId, SqlFunctionCtx* pCtx,
                                    int32_t numOfOutput, int32_t* rowCellInfoOffset) {
464
  assert(win->skey <= win->ekey);
L
Liu Jicong 已提交
465
  return chkResultRowFromKey(pRuntimeEnv, pResultRowInfo, (char*)&win->skey, TSDB_KEYSIZE, masterscan, groupId);
466 467
}

dengyihao's avatar
dengyihao 已提交
468 469
static void doUpdateResultRowIndex(SResultRowInfo* pResultRowInfo, TSKEY lastKey, bool ascQuery,
                                   bool timeWindowInterpo) {
470
  int64_t skey = TSKEY_INITIAL_VAL;
H
Haojun Liao 已提交
471
#if 0
472 473
  int32_t i = 0;
  for (i = pResultRowInfo->size - 1; i >= 0; --i) {
L
Liu Jicong 已提交
474
    SResultRow* pResult = pResultRowInfo->pResult[i];
475 476 477 478 479 480
    if (pResult->closed) {
      break;
    }

    // new closed result rows
    if (timeWindowInterpo) {
L
Liu Jicong 已提交
481 482 483
      if (pResult->endInterp &&
          ((pResult->win.skey <= lastKey && ascQuery) || (pResult->win.skey >= lastKey && !ascQuery))) {
        if (i > 0) {  // the first time window, the startInterp is false.
484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502
          assert(pResult->startInterp);
        }

        closeResultRow(pResultRowInfo, i);
      } else {
        skey = pResult->win.skey;
      }
    } else {
      if ((pResult->win.ekey <= lastKey && ascQuery) || (pResult->win.skey >= lastKey && !ascQuery)) {
        closeResultRow(pResultRowInfo, i);
      } else {
        skey = pResult->win.skey;
      }
    }
  }

  // all result rows are closed, set the last one to be the skey
  if (skey == TSKEY_INITIAL_VAL) {
    if (pResultRowInfo->size == 0) {
L
Liu Jicong 已提交
503
      //      assert(pResultRowInfo->current == NULL);
504 505 506 507 508 509 510
      assert(pResultRowInfo->curPos == -1);
      pResultRowInfo->curPos = -1;
    } else {
      pResultRowInfo->curPos = pResultRowInfo->size - 1;
    }
  } else {
    for (i = pResultRowInfo->size - 1; i >= 0; --i) {
L
Liu Jicong 已提交
511
      SResultRow* pResult = pResultRowInfo->pResult[i];
512 513 514 515 516 517 518 519 520 521 522
      if (pResult->closed) {
        break;
      }
    }

    if (i == pResultRowInfo->size - 1) {
      pResultRowInfo->curPos = i;
    } else {
      pResultRowInfo->curPos = i + 1;  // current not closed result object
    }
  }
H
Haojun Liao 已提交
523
#endif
524
}
525
//
dengyihao's avatar
dengyihao 已提交
526
// static void updateResultRowInfoActiveIndex(SResultRowInfo* pResultRowInfo, const STimeWindow* pWin, TSKEY lastKey,
527 528 529 530 531 532 533 534 535
//                                           bool ascQuery, bool interp) {
//  if ((lastKey > pWin->ekey && ascQuery) || (lastKey < pWin->ekey && (!ascQuery))) {
//    closeAllResultRows(pResultRowInfo);
//    pResultRowInfo->curPos = pResultRowInfo->size - 1;
//  } else {
//    int32_t step = ascQuery ? 1 : -1;
//    doUpdateResultRowIndex(pResultRowInfo, lastKey - step, ascQuery, interp);
//  }
//}
536

537
//  query_range_start, query_range_end, window_duration, window_start, window_end
538
void initExecTimeWindowInfo(SColumnInfoData* pColData, STimeWindow* pQueryWindow) {
539 540 541
  pColData->info.type = TSDB_DATA_TYPE_TIMESTAMP;
  pColData->info.bytes = sizeof(int64_t);

542
  colInfoDataEnsureCapacity(pColData, 0, 5);
543 544 545 546 547 548 549 550 551
  colDataAppendInt64(pColData, 0, &pQueryWindow->skey);
  colDataAppendInt64(pColData, 1, &pQueryWindow->ekey);

  int64_t interval = 0;
  colDataAppendInt64(pColData, 2, &interval);  // this value may be variable in case of 'n' and 'y'.
  colDataAppendInt64(pColData, 3, &pQueryWindow->skey);
  colDataAppendInt64(pColData, 4, &pQueryWindow->ekey);
}

X
Xiaoyu Wang 已提交
552 553 554
void doApplyFunctions(SExecTaskInfo* taskInfo, SqlFunctionCtx* pCtx, STimeWindow* pWin,
                      SColumnInfoData* pTimeWindowData, int32_t offset, int32_t forwardStep, TSKEY* tsCol,
                      int32_t numOfTotal, int32_t numOfOutput, int32_t order) {
555
  for (int32_t k = 0; k < numOfOutput; ++k) {
H
Haojun Liao 已提交
556
    // keep it temporarily
557
    // todo no need this??
dengyihao's avatar
dengyihao 已提交
558 559
    bool    hasAgg = pCtx[k].input.colDataAggIsSet;
    int32_t numOfRows = pCtx[k].input.numOfRows;
H
Haojun Liao 已提交
560
    int32_t startOffset = pCtx[k].input.startRowIndex;
561

562
    pCtx[k].input.startRowIndex = offset;
563
    pCtx[k].input.numOfRows = forwardStep;
564 565

    if (tsCol != NULL) {
H
Haojun Liao 已提交
566
      pCtx[k].ptsList = tsCol;
567 568 569 570
    }

    // not a whole block involved in query processing, statistics data can not be used
    // NOTE: the original value of isSet have been changed here
571 572
    if (pCtx[k].input.colDataAggIsSet && forwardStep < numOfTotal) {
      pCtx[k].input.colDataAggIsSet = false;
573 574
    }

575 576
    if (fmIsWindowPseudoColumnFunc(pCtx[k].functionId)) {
      SResultRowEntryInfo* pEntryInfo = GET_RES_INFO(&pCtx[k]);
577 578

      char* p = GET_ROWCELL_INTERBUF(pEntryInfo);
579

580
      SColumnInfoData idata = {0};
dengyihao's avatar
dengyihao 已提交
581
      idata.info.type = TSDB_DATA_TYPE_BIGINT;
582
      idata.info.bytes = tDataTypes[TSDB_DATA_TYPE_BIGINT].bytes;
dengyihao's avatar
dengyihao 已提交
583
      idata.pData = p;
584 585 586 587

      SScalarParam out = {.columnData = &idata};
      SScalarParam tw = {.numOfRows = 5, .columnData = pTimeWindowData};
      pCtx[k].sfp.process(&tw, 1, &out);
588
      pEntryInfo->numOfRes = 1;
589 590 591 592 593 594 595 596 597 598
    } else {
      int32_t code = TSDB_CODE_SUCCESS;
      if (functionNeedToExecute(&pCtx[k]) && pCtx[k].fpSet.process != NULL) {
        code = pCtx[k].fpSet.process(&pCtx[k]);

        if (code != TSDB_CODE_SUCCESS) {
          qError("%s apply functions error, code: %s", GET_TASKID(taskInfo), tstrerror(code));
          taskInfo->code = code;
          longjmp(taskInfo->env, code);
        }
599
      }
600

601 602 603 604 605
      // restore it
      pCtx[k].input.colDataAggIsSet = hasAgg;
      pCtx[k].input.startRowIndex = startOffset;
      pCtx[k].input.numOfRows = numOfRows;
    }
606 607 608
  }
}

L
Liu Jicong 已提交
609
static FORCE_INLINE TSKEY reviseWindowEkey(STaskAttr* pQueryAttr, STimeWindow* pWindow) {
dengyihao's avatar
dengyihao 已提交
610
  TSKEY   ekey = -1;
611 612
  int32_t order = TSDB_ORDER_ASC;
  if (order == TSDB_ORDER_ASC) {
613 614 615 616 617 618 619 620 621 622 623 624 625 626
    ekey = pWindow->ekey;
    if (ekey > pQueryAttr->window.ekey) {
      ekey = pQueryAttr->window.ekey;
    }
  } else {
    ekey = pWindow->skey;
    if (ekey < pQueryAttr->window.ekey) {
      ekey = pQueryAttr->window.ekey;
    }
  }

  return ekey;
}

dengyihao's avatar
dengyihao 已提交
627
static int32_t doSetInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order,
628
                                   int32_t scanFlag, bool createDummyCol);
629

dengyihao's avatar
dengyihao 已提交
630 631
static void doSetInputDataBlockInfo(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock,
                                    int32_t order) {
632
  for (int32_t i = 0; i < pOperator->numOfExprs; ++i) {
633
    pCtx[i].order = order;
634
    pCtx[i].input.numOfRows = pBlock->info.rows;
635
    setBlockStatisInfo(&pCtx[i], &pOperator->pExpr[i], pBlock);
636 637 638
  }
}

X
Xiaoyu Wang 已提交
639 640
void setInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order,
                       int32_t scanFlag, bool createDummyCol) {
641
  if (pBlock->pBlockAgg != NULL) {
H
Haojun Liao 已提交
642
    doSetInputDataBlockInfo(pOperator, pCtx, pBlock, order);
643
  } else {
644
    doSetInputDataBlock(pOperator, pCtx, pBlock, order, scanFlag, createDummyCol);
H
Haojun Liao 已提交
645
  }
646 647
}

L
Liu Jicong 已提交
648 649
static int32_t doCreateConstantValColumnInfo(SInputColumnInfoData* pInput, SFunctParam* pFuncParam, int32_t paramIndex,
                                             int32_t numOfRows) {
650 651 652 653 654 655 656 657
  SColumnInfoData* pColInfo = NULL;
  if (pInput->pData[paramIndex] == NULL) {
    pColInfo = taosMemoryCalloc(1, sizeof(SColumnInfoData));
    if (pColInfo == NULL) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }

    // Set the correct column info (data type and bytes)
658 659
    pColInfo->info.type = pFuncParam->param.nType;
    pColInfo->info.bytes = pFuncParam->param.nLen;
660 661

    pInput->pData[paramIndex] = pColInfo;
662 663
  } else {
    pColInfo = pInput->pData[paramIndex];
664 665
  }

666
  colInfoDataEnsureCapacity(pColInfo, 0, numOfRows);
667

668
  int8_t type = pFuncParam->param.nType;
669 670
  if (type == TSDB_DATA_TYPE_BIGINT || type == TSDB_DATA_TYPE_UBIGINT) {
    int64_t v = pFuncParam->param.i;
dengyihao's avatar
dengyihao 已提交
671
    for (int32_t i = 0; i < numOfRows; ++i) {
672 673 674 675
      colDataAppendInt64(pColInfo, i, &v);
    }
  } else if (type == TSDB_DATA_TYPE_DOUBLE) {
    double v = pFuncParam->param.d;
dengyihao's avatar
dengyihao 已提交
676
    for (int32_t i = 0; i < numOfRows; ++i) {
677 678
      colDataAppendDouble(pColInfo, i, &v);
    }
679
  } else if (type == TSDB_DATA_TYPE_VARCHAR) {
L
Liu Jicong 已提交
680
    char* tmp = taosMemoryMalloc(pFuncParam->param.nLen + VARSTR_HEADER_SIZE);
681
    STR_WITH_SIZE_TO_VARSTR(tmp, pFuncParam->param.pz, pFuncParam->param.nLen);
L
Liu Jicong 已提交
682
    for (int32_t i = 0; i < numOfRows; ++i) {
683 684
      colDataAppend(pColInfo, i, tmp, false);
    }
685 686 687 688 689
  }

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
690
static int32_t doSetInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order,
X
Xiaoyu Wang 已提交
691
                                   int32_t scanFlag, bool createDummyCol) {
692 693
  int32_t code = TSDB_CODE_SUCCESS;

694
  for (int32_t i = 0; i < pOperator->numOfExprs; ++i) {
L
Liu Jicong 已提交
695
    pCtx[i].order = order;
696 697
    pCtx[i].input.numOfRows = pBlock->info.rows;

L
Liu Jicong 已提交
698
    pCtx[i].pSrcBlock = pBlock;
X
Xiaoyu Wang 已提交
699
    pCtx[i].scanFlag = scanFlag;
H
Haojun Liao 已提交
700

701
    SInputColumnInfoData* pInput = &pCtx[i].input;
702
    pInput->uid = pBlock->info.uid;
C
Cary Xu 已提交
703
    pInput->colDataAggIsSet = false;
704

705 706
    SExprInfo* pOneExpr = &pOperator->pExpr[i];
    for (int32_t j = 0; j < pOneExpr->base.numOfParams; ++j) {
dengyihao's avatar
dengyihao 已提交
707
      SFunctParam* pFuncParam = &pOneExpr->base.pParam[j];
G
Ganlin Zhao 已提交
708 709
      if (pFuncParam->type == FUNC_PARAM_TYPE_COLUMN) {
        int32_t slotId = pFuncParam->pCol->slotId;
dengyihao's avatar
dengyihao 已提交
710
        pInput->pData[j] = taosArrayGet(pBlock->pDataBlock, slotId);
711 712 713
        pInput->totalRows = pBlock->info.rows;
        pInput->numOfRows = pBlock->info.rows;
        pInput->startRowIndex = 0;
714

715
        // NOTE: the last parameter is the primary timestamp column
716 717 718
        if (fmIsTimelineFunc(pCtx[i].functionId) && (j == pOneExpr->base.numOfParams - 1)) {
          pInput->pPTS = pInput->pData[j];
        }
719 720
        ASSERT(pInput->pData[j] != NULL);
      } else if (pFuncParam->type == FUNC_PARAM_TYPE_VALUE) {
721 722 723
        // todo avoid case: top(k, 12), 12 is the value parameter.
        // sum(11), 11 is also the value parameter.
        if (createDummyCol && pOneExpr->base.numOfParams == 1) {
724 725 726 727
          pInput->totalRows = pBlock->info.rows;
          pInput->numOfRows = pBlock->info.rows;
          pInput->startRowIndex = 0;

728
          code = doCreateConstantValColumnInfo(pInput, pFuncParam, j, pBlock->info.rows);
729 730 731
          if (code != TSDB_CODE_SUCCESS) {
            return code;
          }
732
        }
G
Ganlin Zhao 已提交
733 734
      }
    }
H
Haojun Liao 已提交
735
  }
736 737

  return code;
H
Haojun Liao 已提交
738 739
}

740
static int32_t doAggregateImpl(SOperatorInfo* pOperator, TSKEY startTs, SqlFunctionCtx* pCtx) {
741
  for (int32_t k = 0; k < pOperator->numOfExprs; ++k) {
H
Haojun Liao 已提交
742
    if (functionNeedToExecute(&pCtx[k])) {
743
      // todo add a dummy funtion to avoid process check
744 745 746 747 748 749 750 751
      if (pCtx[k].fpSet.process == NULL) {
        continue;
      }

      int32_t code = pCtx[k].fpSet.process(&pCtx[k]);
      if (code != TSDB_CODE_SUCCESS) {
        qError("%s aggregate function error happens, code: %s", GET_TASKID(pOperator->pTaskInfo), tstrerror(code));
        return code;
752
      }
753 754
    }
  }
755 756

  return TSDB_CODE_SUCCESS;
757 758
}

H
Haojun Liao 已提交
759
static void setPseudoOutputColInfo(SSDataBlock* pResult, SqlFunctionCtx* pCtx, SArray* pPseudoList) {
dengyihao's avatar
dengyihao 已提交
760
  size_t num = (pPseudoList != NULL) ? taosArrayGetSize(pPseudoList) : 0;
H
Haojun Liao 已提交
761 762 763 764 765
  for (int32_t i = 0; i < num; ++i) {
    pCtx[i].pOutput = taosArrayGet(pResult->pDataBlock, i);
  }
}

766
int32_t projectApplyFunctions(SExprInfo* pExpr, SSDataBlock* pResult, SSDataBlock* pSrcBlock, SqlFunctionCtx* pCtx,
X
Xiaoyu Wang 已提交
767
                              int32_t numOfOutput, SArray* pPseudoList) {
H
Haojun Liao 已提交
768
  setPseudoOutputColInfo(pResult, pCtx, pPseudoList);
H
Haojun Liao 已提交
769
  pResult->info.groupId = pSrcBlock->info.groupId;
H
Haojun Liao 已提交
770

dengyihao's avatar
dengyihao 已提交
771 772
  // if the source equals to the destination, it is to create a new column as the result of scalar function or some
  // operators.
773 774
  bool createNewColModel = (pResult == pSrcBlock);

775 776
  int32_t numOfRows = 0;

777
  for (int32_t k = 0; k < numOfOutput; ++k) {
dengyihao's avatar
dengyihao 已提交
778
    int32_t         outputSlotId = pExpr[k].base.resSchema.slotId;
779 780
    SqlFunctionCtx* pfCtx = &pCtx[k];

L
Liu Jicong 已提交
781
    if (pExpr[k].pExpr->nodeType == QUERY_NODE_COLUMN) {  // it is a project query
782
      SColumnInfoData* pColInfoData = taosArrayGet(pResult->pDataBlock, outputSlotId);
783
      if (pResult->info.rows > 0 && !createNewColModel) {
X
Xiaoyu Wang 已提交
784 785
        colDataMergeCol(pColInfoData, pResult->info.rows, &pResult->info.capacity, pfCtx->input.pData[0],
                        pfCtx->input.numOfRows);
786 787 788
      } else {
        colDataAssign(pColInfoData, pfCtx->input.pData[0], pfCtx->input.numOfRows);
      }
789

790
      numOfRows = pfCtx->input.numOfRows;
791
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_VALUE) {
792
      SColumnInfoData* pColInfoData = taosArrayGet(pResult->pDataBlock, outputSlotId);
793

dengyihao's avatar
dengyihao 已提交
794
      int32_t offset = createNewColModel ? 0 : pResult->info.rows;
795
      for (int32_t i = 0; i < pSrcBlock->info.rows; ++i) {
dengyihao's avatar
dengyihao 已提交
796 797 798
        colDataAppend(pColInfoData, i + offset,
                      taosVariantGet(&pExpr[k].base.pParam[0].param, pExpr[k].base.pParam[0].param.nType),
                      TSDB_DATA_TYPE_NULL == pExpr[k].base.pParam[0].param.nType);
799
      }
800 801

      numOfRows = pSrcBlock->info.rows;
H
Haojun Liao 已提交
802
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_OPERATOR) {
803 804 805
      SArray* pBlockList = taosArrayInit(4, POINTER_BYTES);
      taosArrayPush(pBlockList, &pSrcBlock);

806
      SColumnInfoData* pResColData = taosArrayGet(pResult->pDataBlock, outputSlotId);
807
      SColumnInfoData  idata = {.info = pResColData->info, .hasNull = true};
808

809
      SScalarParam dest = {.columnData = &idata};
X
Xiaoyu Wang 已提交
810
      int32_t      code = scalarCalculate(pExpr[k].pExpr->_optrRoot.pRootNode, pBlockList, &dest);
811 812 813 814
      if (code != TSDB_CODE_SUCCESS) {
        taosArrayDestroy(pBlockList);
        return code;
      }
815

dengyihao's avatar
dengyihao 已提交
816
      int32_t startOffset = createNewColModel ? 0 : pResult->info.rows;
817
      colInfoDataEnsureCapacity(pResColData, startOffset, pResult->info.capacity);
818
      colDataMergeCol(pResColData, startOffset, &pResult->info.capacity, &idata, dest.numOfRows);
819 820

      numOfRows = dest.numOfRows;
821 822
      taosArrayDestroy(pBlockList);
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_FUNCTION) {
823
      ASSERT(!fmIsAggFunc(pfCtx->functionId));
824

825 826
      // _rowts/_c0, not tbname column
      if (fmIsPseudoColumnFunc(pfCtx->functionId) && (!fmIsScanPseudoColumnFunc(pfCtx->functionId))) {
H
Haojun Liao 已提交
827
        // do nothing
X
Xiaoyu Wang 已提交
828
      } else if (fmIsIndefiniteRowsFunc(pfCtx->functionId)) {
dengyihao's avatar
dengyihao 已提交
829
        SResultRowEntryInfo* pResInfo = GET_RES_INFO(&pCtx[k]);
830
        pfCtx->fpSet.init(&pCtx[k], pResInfo);
831

832
        pfCtx->pOutput = taosArrayGet(pResult->pDataBlock, outputSlotId);
dengyihao's avatar
dengyihao 已提交
833
        pfCtx->offset = createNewColModel ? 0 : pResult->info.rows;  // set the start offset
H
Haojun Liao 已提交
834

835
        // set the timestamp(_rowts) output buffer
836 837
        if (taosArrayGetSize(pPseudoList) > 0) {
          int32_t* outputColIndex = taosArrayGet(pPseudoList, 0);
838
          pfCtx->pTsOutput = (SColumnInfoData*)pCtx[*outputColIndex].pOutput;
839
        }
H
Haojun Liao 已提交
840

841
        numOfRows = pfCtx->fpSet.process(pfCtx);
H
Haojun Liao 已提交
842 843 844
      } else {
        SArray* pBlockList = taosArrayInit(4, POINTER_BYTES);
        taosArrayPush(pBlockList, &pSrcBlock);
G
Ganlin Zhao 已提交
845

846
        SColumnInfoData* pResColData = taosArrayGet(pResult->pDataBlock, outputSlotId);
847
        SColumnInfoData  idata = {.info = pResColData->info, .hasNull = true};
H
Haojun Liao 已提交
848

849
        SScalarParam dest = {.columnData = &idata};
X
Xiaoyu Wang 已提交
850
        int32_t      code = scalarCalculate((SNode*)pExpr[k].pExpr->_function.pFunctNode, pBlockList, &dest);
851 852 853 854
        if (code != TSDB_CODE_SUCCESS) {
          taosArrayDestroy(pBlockList);
          return code;
        }
855

dengyihao's avatar
dengyihao 已提交
856
        int32_t startOffset = createNewColModel ? 0 : pResult->info.rows;
857
        colInfoDataEnsureCapacity(pResColData, startOffset, pResult->info.capacity);
858
        colDataMergeCol(pResColData, startOffset, &pResult->info.capacity, &idata, dest.numOfRows);
859 860

        numOfRows = dest.numOfRows;
H
Haojun Liao 已提交
861 862
        taosArrayDestroy(pBlockList);
      }
863
    } else {
864
      ASSERT(0);
865 866
    }
  }
867

868 869 870
  if (!createNewColModel) {
    pResult->info.rows += numOfRows;
  }
871 872

  return TSDB_CODE_SUCCESS;
873 874
}

875 876 877
static void setResultRowKey(SResultRow* pResultRow, char* pData, int16_t type) {
  if (IS_VAR_DATA_TYPE(type)) {
    // todo disable this
878

879 880 881 882 883 884 885 886 887
    //    if (pResultRow->key == NULL) {
    //      pResultRow->key = taosMemoryMalloc(varDataTLen(pData));
    //      varDataCopy(pResultRow->key, pData);
    //    } else {
    //      ASSERT(memcmp(pResultRow->key, pData, varDataTLen(pData)) == 0);
    //    }
  } else {
    int64_t v = -1;
    GET_TYPED_DATA(v, int64_t, type, pData);
888

889 890
    pResultRow->win.skey = v;
    pResultRow->win.ekey = v;
891 892 893
  }
}

894 895 896 897 898
int32_t setGroupResultOutputBuf(SOptrBasicInfo* binfo, int32_t numOfCols, char* pData, int16_t type, int16_t bytes,
                                int32_t groupId, SDiskbasedBuf* pBuf, SExecTaskInfo* pTaskInfo,
                                SAggSupporter* pAggSup) {
  SResultRowInfo* pResultRowInfo = &binfo->resultRowInfo;
  SqlFunctionCtx* pCtx = binfo->pCtx;
899

900 901 902
  SResultRow* pResultRow =
      doSetResultOutBufByKey(pBuf, pResultRowInfo, (char*)pData, bytes, true, groupId, pTaskInfo, false, pAggSup);
  assert(pResultRow != NULL);
903

904
  setResultRowKey(pResultRow, pData, type);
905
  setResultRowInitCtx(pResultRow, pCtx, numOfCols, binfo->rowCellInfoOffset);
906
  return TSDB_CODE_SUCCESS;
907 908
}

5
54liuyao 已提交
909
bool functionNeedToExecute(SqlFunctionCtx* pCtx) {
910
  struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(pCtx);
911

912 913 914 915 916
  // in case of timestamp column, always generated results.
  int32_t functionId = pCtx->functionId;
  if (functionId == -1) {
    return false;
  }
917

918 919
  if (pCtx->scanFlag == REPEAT_SCAN) {
    return fmIsRepeatScanFunc(pCtx->functionId);
920 921
  }

922 923
  if (isRowEntryCompleted(pResInfo)) {
    return false;
924 925
  }

X
Xiaoyu Wang 已提交
926 927 928 929 930 931 932 933
  //  if (functionId == FUNCTION_FIRST_DST || functionId == FUNCTION_FIRST) {
  //    //    return QUERY_IS_ASC_QUERY(pQueryAttr);
  //  }
  //
  //  // denote the order type
  //  if ((functionId == FUNCTION_LAST_DST || functionId == FUNCTION_LAST)) {
  //    //    return pCtx->param[0].i == pQueryAttr->order.order;
  //  }
934

935 936 937 938 939
  // in the reverse table scan, only the following functions need to be executed
  //  if (IS_REVERSE_SCAN(pRuntimeEnv) ||
  //      (pRuntimeEnv->scanFlag == REPEAT_SCAN && functionId != FUNCTION_STDDEV && functionId != FUNCTION_PERCT)) {
  //    return false;
  //  }
940 941 942 943

  return true;
}

944 945 946 947 948 949 950
static int32_t doCreateConstantValColumnAggInfo(SInputColumnInfoData* pInput, SFunctParam* pFuncParam, int32_t type,
                                                int32_t paramIndex, int32_t numOfRows) {
  if (pInput->pData[paramIndex] == NULL) {
    pInput->pData[paramIndex] = taosMemoryCalloc(1, sizeof(SColumnInfoData));
    if (pInput->pData[paramIndex] == NULL) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }
951

952 953 954
    // Set the correct column info (data type and bytes)
    pInput->pData[paramIndex]->info.type = type;
    pInput->pData[paramIndex]->info.bytes = tDataTypes[type].bytes;
955
  }
H
Haojun Liao 已提交
956

957 958 959 960 961 962
  SColumnDataAgg* da = NULL;
  if (pInput->pColumnDataAgg[paramIndex] == NULL) {
    da = taosMemoryCalloc(1, sizeof(SColumnDataAgg));
    pInput->pColumnDataAgg[paramIndex] = da;
    if (da == NULL) {
      return TSDB_CODE_OUT_OF_MEMORY;
963 964
    }
  } else {
965
    da = pInput->pColumnDataAgg[paramIndex];
966 967
  }

968
  ASSERT(!IS_VAR_DATA_TYPE(type));
969

970 971 972 973 974 975
  if (type == TSDB_DATA_TYPE_BIGINT) {
    int64_t v = pFuncParam->param.i;
    *da = (SColumnDataAgg){.numOfNull = 0, .min = v, .max = v, .maxIndex = 0, .minIndex = 0, .sum = v * numOfRows};
  } else if (type == TSDB_DATA_TYPE_DOUBLE) {
    double v = pFuncParam->param.d;
    *da = (SColumnDataAgg){.numOfNull = 0, .maxIndex = 0, .minIndex = 0};
976

977 978 979 980 981 982 983 984 985 986 987 988
    *(double*)&da->min = v;
    *(double*)&da->max = v;
    *(double*)&da->sum = v * numOfRows;
  } else if (type == TSDB_DATA_TYPE_BOOL) {  // todo validate this data type
    bool v = pFuncParam->param.i;

    *da = (SColumnDataAgg){.numOfNull = 0, .maxIndex = 0, .minIndex = 0};
    *(bool*)&da->min = 0;
    *(bool*)&da->max = v;
    *(bool*)&da->sum = v * numOfRows;
  } else if (type == TSDB_DATA_TYPE_TIMESTAMP) {
    // do nothing
989
  } else {
990
    ASSERT(0);
991 992
  }

993 994
  return TSDB_CODE_SUCCESS;
}
995 996 997 998 999 1000 1001 1002 1003 1004 1005

void setBlockStatisInfo(SqlFunctionCtx* pCtx, SExprInfo* pExprInfo, SSDataBlock* pBlock) {
  int32_t numOfRows = pBlock->info.rows;

  SInputColumnInfoData* pInput = &pCtx->input;
  pInput->numOfRows = numOfRows;
  pInput->totalRows = numOfRows;

  if (pBlock->pBlockAgg != NULL) {
    pInput->colDataAggIsSet = true;

1006 1007
    for (int32_t j = 0; j < pExprInfo->base.numOfParams; ++j) {
      SFunctParam* pFuncParam = &pExprInfo->base.pParam[j];
1008

1009 1010
      if (pFuncParam->type == FUNC_PARAM_TYPE_COLUMN) {
        int32_t slotId = pFuncParam->pCol->slotId;
1011 1012 1013 1014
        pInput->pColumnDataAgg[j] = pBlock->pBlockAgg[slotId];
        if (pInput->pColumnDataAgg[j] == NULL) {
          pInput->colDataAggIsSet = false;
        }
1015 1016 1017 1018

        // Here we set the column info data since the data type for each column data is required, but
        // the data in the corresponding SColumnInfoData will not be used.
        pInput->pData[j] = taosArrayGet(pBlock->pDataBlock, slotId);
1019 1020
      } else if (pFuncParam->type == FUNC_PARAM_TYPE_VALUE) {
        doCreateConstantValColumnAggInfo(pInput, pFuncParam, pFuncParam->param.nType, j, pBlock->info.rows);
1021 1022
      }
    }
1023
  } else {
1024
    pInput->colDataAggIsSet = false;
1025 1026 1027
  }

  // set the statistics data for primary time stamp column
1028 1029 1030 1031 1032
  //  if (pCtx->functionId == FUNCTION_SPREAD && pColumn->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
  //    pCtx->isAggSet = true;
  //    pCtx->agg.min = pBlock->info.window.skey;
  //    pCtx->agg.max = pBlock->info.window.ekey;
  //  }
1033 1034 1035
}

// set the output buffer for the selectivity + tag query
1036
static int32_t setSelectValueColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
1037 1038
  int32_t num = 0;

H
Haojun Liao 已提交
1039
  SqlFunctionCtx*  p = NULL;
1040 1041
  SqlFunctionCtx** pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
  if (pValCtx == NULL) {
1042 1043 1044 1045
    return TSDB_CODE_QRY_OUT_OF_MEMORY;
  }

  for (int32_t i = 0; i < numOfOutput; ++i) {
1046 1047
    if (strcmp(pCtx[i].pExpr->pExpr->_function.functionName, "_select_value") == 0) {
      pValCtx[num++] = &pCtx[i];
X
Xiaoyu Wang 已提交
1048
    } else if (fmIsSelectFunc(pCtx[i].functionId)) {
1049
      p = &pCtx[i];
1050
    }
X
Xiaoyu Wang 已提交
1051 1052 1053 1054 1055 1056 1057 1058 1059 1060
    //    if (functionId == FUNCTION_TAG_DUMMY || functionId == FUNCTION_TS_DUMMY) {
    //      tagLen += pCtx[i].resDataInfo.bytes;
    //      pTagCtx[num++] = &pCtx[i];
    //    } else if (functionId == FUNCTION_TS || functionId == FUNCTION_TAG) {
    //      // tag function may be the group by tag column
    //      // ts may be the required primary timestamp column
    //      continue;
    //    } else {
    //      // the column may be the normal column, group by normal_column, the functionId is FUNCTION_PRJ
    //    }
1061
  }
1062

1063
  if (p != NULL) {
1064
    p->subsidiaries.pCtx = pValCtx;
1065
    p->subsidiaries.num = num;
1066
  } else {
1067
    taosMemoryFreeClear(pValCtx);
1068 1069 1070 1071 1072
  }

  return TSDB_CODE_SUCCESS;
}

1073
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowCellInfoOffset) {
L
Liu Jicong 已提交
1074
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
H
Haojun Liao 已提交
1075 1076 1077 1078
  if (pFuncCtx == NULL) {
    return NULL;
  }

wafwerar's avatar
wafwerar 已提交
1079
  *rowCellInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
H
Haojun Liao 已提交
1080
  if (*rowCellInfoOffset == 0) {
wafwerar's avatar
wafwerar 已提交
1081
    taosMemoryFreeClear(pFuncCtx);
H
Haojun Liao 已提交
1082 1083 1084 1085
    return NULL;
  }

  for (int32_t i = 0; i < numOfOutput; ++i) {
H
Haojun Liao 已提交
1086
    SExprInfo* pExpr = &pExprInfo[i];
H
Haojun Liao 已提交
1087

L
Liu Jicong 已提交
1088
    SExprBasicInfo* pFunct = &pExpr->base;
H
Haojun Liao 已提交
1089
    SqlFunctionCtx* pCtx = &pFuncCtx[i];
H
Haojun Liao 已提交
1090

1091
    pCtx->functionId = -1;
1092
    pCtx->curBufPage = -1;
L
Liu Jicong 已提交
1093
    pCtx->pExpr = pExpr;
1094

H
Haojun Liao 已提交
1095
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
1096
      SFuncExecEnv env = {0};
H
Haojun Liao 已提交
1097 1098
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;

X
Xiaoyu Wang 已提交
1099
      if (fmIsAggFunc(pCtx->functionId) || fmIsIndefiniteRowsFunc(pCtx->functionId)) {
1100 1101 1102 1103
        bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
        if (!isUdaf) {
          fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
        } else {
1104
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
1105 1106 1107
          strncpy(pCtx->udfName, udfName, strlen(udfName));
          fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
        }
1108 1109 1110
        pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
      } else {
        fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
1111 1112 1113
        if (pCtx->sfp.getEnv != NULL) {
          pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
        }
1114
      }
H
Haojun Liao 已提交
1115
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
1116 1117
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
H
Haojun Liao 已提交
1118
      // for simple column, the result buffer needs to hold at least one element.
1119
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
H
Haojun Liao 已提交
1120
    }
H
Haojun Liao 已提交
1121

H
Haojun Liao 已提交
1122
    pCtx->input.numOfInputCols = pFunct->numOfParams;
wafwerar's avatar
wafwerar 已提交
1123 1124
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
H
Haojun Liao 已提交
1125

1126
    pCtx->pTsOutput = NULL;
L
Liu Jicong 已提交
1127
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1128
    pCtx->resDataInfo.type = pFunct->resSchema.type;
L
Liu Jicong 已提交
1129
    pCtx->order = TSDB_ORDER_ASC;
1130
    pCtx->start.key = INT64_MIN;
L
Liu Jicong 已提交
1131
    pCtx->end.key = INT64_MIN;
1132
    pCtx->numOfParams = pExpr->base.numOfParams;
H
Haojun Liao 已提交
1133

1134
    pCtx->param = pFunct->pParam;
dengyihao's avatar
dengyihao 已提交
1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168
    //    for (int32_t j = 0; j < pCtx->numOfParams; ++j) {
    //      // set the order information for top/bottom query
    //      int32_t functionId = pCtx->functionId;
    //      if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF) {
    //        int32_t f = getExprFunctionId(&pExpr[0]);
    //        assert(f == FUNCTION_TS || f == FUNCTION_TS_DUMMY);
    //
    //        //      pCtx->param[2].i = pQueryAttr->order.order;
    //        //      pCtx->param[2].nType = TSDB_DATA_TYPE_BIGINT;
    //        //      pCtx->param[3].i = functionId;
    //        //      pCtx->param[3].nType = TSDB_DATA_TYPE_BIGINT;
    //
    //        //      pCtx->param[1].i = pQueryAttr->order.col.info.colId;
    //      } else if (functionId == FUNCTION_INTERP) {
    //        //      pCtx->param[2].i = (int8_t)pQueryAttr->fillType;
    //        //      if (pQueryAttr->fillVal != NULL) {
    //        //        if (isNull((const char *)&pQueryAttr->fillVal[i], pCtx->inputType)) {
    //        //          pCtx->param[1].nType = TSDB_DATA_TYPE_NULL;
    //        //        } else {  // todo refactor, taosVariantCreateFromBinary should handle the NULL value
    //        //          if (pCtx->inputType != TSDB_DATA_TYPE_BINARY && pCtx->inputType != TSDB_DATA_TYPE_NCHAR) {
    //        //            taosVariantCreateFromBinary(&pCtx->param[1], (char *)&pQueryAttr->fillVal[i],
    //        pCtx->inputBytes, pCtx->inputType);
    //        //          }
    //        //        }
    //        //      }
    //      } else if (functionId == FUNCTION_TWA) {
    //        //      pCtx->param[1].i = pQueryAttr->window.skey;
    //        //      pCtx->param[1].nType = TSDB_DATA_TYPE_BIGINT;
    //        //      pCtx->param[2].i = pQueryAttr->window.ekey;
    //        //      pCtx->param[2].nType = TSDB_DATA_TYPE_BIGINT;
    //      } else if (functionId == FUNCTION_ARITHM) {
    //        //      pCtx->param[1].pz = (char*) getScalarFuncSupport(pRuntimeEnv->scalarSup, i);
    //      }
    //    }
H
Haojun Liao 已提交
1169 1170
  }

L
Liu Jicong 已提交
1171 1172 1173
  for (int32_t i = 1; i < numOfOutput; ++i) {
    (*rowCellInfoOffset)[i] =
        (int32_t)((*rowCellInfoOffset)[i - 1] + sizeof(SResultRowEntryInfo) + pFuncCtx[i - 1].resDataInfo.interBufSize);
H
Haojun Liao 已提交
1174
  }
H
Haojun Liao 已提交
1175

1176
  setSelectValueColumnInfo(pFuncCtx, numOfOutput);
H
Haojun Liao 已提交
1177 1178 1179
  return pFuncCtx;
}

1180
static void* destroySqlFunctionCtx(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
1181 1182 1183 1184 1185 1186
  if (pCtx == NULL) {
    return NULL;
  }

  for (int32_t i = 0; i < numOfOutput; ++i) {
    for (int32_t j = 0; j < pCtx[i].numOfParams; ++j) {
1187
      taosVariantDestroy(&pCtx[i].param[j].param);
1188 1189
    }

1190
    taosMemoryFreeClear(pCtx[i].subsidiaries.pCtx);
H
Haojun Liao 已提交
1191 1192
    taosMemoryFree(pCtx[i].input.pData);
    taosMemoryFree(pCtx[i].input.pColumnDataAgg);
1193 1194
  }

wafwerar's avatar
wafwerar 已提交
1195
  taosMemoryFreeClear(pCtx);
1196 1197 1198
  return NULL;
}

L
Liu Jicong 已提交
1199
bool isTaskKilled(SExecTaskInfo* pTaskInfo) {
1200 1201
  // query has been executed more than tsShellActivityTimer, and the retrieve has not arrived
  // abort current query execution.
L
Liu Jicong 已提交
1202 1203
  if (pTaskInfo->owner != 0 &&
      ((taosGetTimestampSec() - pTaskInfo->cost.start / 1000) > 10 * getMaximumIdleDurationSec())
1204 1205
      /*(!needBuildResAfterQueryComplete(pTaskInfo))*/) {
    assert(pTaskInfo->cost.start != 0);
L
Liu Jicong 已提交
1206 1207 1208
    //    qDebug("QInfo:%" PRIu64 " retrieve not arrive beyond %d ms, abort current query execution, start:%" PRId64
    //           ", current:%d", pQInfo->qId, 1, pQInfo->startExecTs, taosGetTimestampSec());
    //    return true;
1209 1210 1211 1212 1213
  }

  return false;
}

L
Liu Jicong 已提交
1214
void setTaskKilled(SExecTaskInfo* pTaskInfo) { pTaskInfo->code = TSDB_CODE_TSC_QUERY_CANCELLED; }
1215

L
Liu Jicong 已提交
1216
static bool isCachedLastQuery(STaskAttr* pQueryAttr) {
1217 1218
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
    int32_t functionId = getExprFunctionId(&pQueryAttr->pExpr1[i]);
X
Xiaoyu Wang 已提交
1219 1220 1221
    //    if (functionId == FUNCTION_LAST || functionId == FUNCTION_LAST_DST) {
    //      continue;
    //    }
1222 1223 1224 1225

    return false;
  }

1226 1227
  int32_t order = TSDB_ORDER_ASC;
  if (order != TSDB_ORDER_DESC || !TSWINDOW_IS_EQUAL(pQueryAttr->window, TSWINDOW_DESC_INITIALIZER)) {
1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246
    return false;
  }

  if (pQueryAttr->groupbyColumn) {
    return false;
  }

  if (pQueryAttr->interval.interval > 0) {
    return false;
  }

  if (pQueryAttr->numOfFilterCols > 0 || pQueryAttr->havingNum > 0) {
    return false;
  }

  return true;
}

/////////////////////////////////////////////////////////////////////////////////////////////
L
Liu Jicong 已提交
1247
// todo refactor : return window
1248
void getAlignQueryTimeWindow(SInterval* pInterval, int32_t precision, int64_t key, STimeWindow* win) {
H
Haojun Liao 已提交
1249
  win->skey = taosTimeTruncate(key, pInterval, precision);
1250 1251

  /*
H
Haojun Liao 已提交
1252
   * if the realSkey > INT64_MAX - pInterval->interval, the query duration between
1253 1254
   * realSkey and realEkey must be less than one interval.Therefore, no need to adjust the query ranges.
   */
1255 1256
  win->ekey = taosTimeAdd(win->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
  if (win->ekey < win->skey) {
1257 1258 1259 1260
    win->ekey = INT64_MAX;
  }
}

L
Liu Jicong 已提交
1261
static int32_t updateBlockLoadStatus(STaskAttr* pQuery, int32_t status) {
1262 1263 1264
  bool hasFirstLastFunc = false;
  bool hasOtherFunc = false;

1265
  if (status == BLK_DATA_DATA_LOAD || status == BLK_DATA_FILTEROUT) {
1266 1267 1268 1269 1270
    return status;
  }

  for (int32_t i = 0; i < pQuery->numOfOutput; ++i) {
    int32_t functionId = getExprFunctionId(&pQuery->pExpr1[i]);
1271
#if 0
1272 1273 1274 1275 1276 1277 1278 1279 1280 1281
    if (functionId == FUNCTION_TS || functionId == FUNCTION_TS_DUMMY || functionId == FUNCTION_TAG ||
        functionId == FUNCTION_TAG_DUMMY) {
      continue;
    }

    if (functionId == FUNCTION_FIRST_DST || functionId == FUNCTION_LAST_DST) {
      hasFirstLastFunc = true;
    } else {
      hasOtherFunc = true;
    }
1282
#endif
1283 1284
  }

1285
  if (hasFirstLastFunc && status == BLK_DATA_NOT_LOAD) {
L
Liu Jicong 已提交
1286
    if (!hasOtherFunc) {
1287
      return BLK_DATA_FILTEROUT;
1288
    } else {
1289
      return BLK_DATA_DATA_LOAD;
1290 1291 1292 1293 1294 1295
    }
  }

  return status;
}

L
Liu Jicong 已提交
1296 1297
// static void updateDataCheckOrder(SQInfo *pQInfo, SQueryTableReq* pQueryMsg, bool stableQuery) {
//   STaskAttr* pQueryAttr = pQInfo->runtimeEnv.pQueryAttr;
H
Haojun Liao 已提交
1298
//
L
Liu Jicong 已提交
1299 1300 1301 1302
//   // in case of point-interpolation query, use asc order scan
//   char msg[] = "QInfo:0x%"PRIx64" scan order changed for %s query, old:%d, new:%d, qrange exchanged, old qrange:%"
//   PRId64
//                "-%" PRId64 ", new qrange:%" PRId64 "-%" PRId64;
H
Haojun Liao 已提交
1303
//
L
Liu Jicong 已提交
1304 1305 1306 1307 1308
//   // todo handle the case the the order irrelevant query type mixed up with order critical query type
//   // descending order query for last_row query
//   if (isFirstLastRowQuery(pQueryAttr)) {
//     //qDebug("QInfo:0x%"PRIx64" scan order changed for last_row query, old:%d, new:%d", pQInfo->qId,
//     pQueryAttr->order.order, TSDB_ORDER_ASC);
H
Haojun Liao 已提交
1309
//
L
Liu Jicong 已提交
1310 1311
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     if (pQueryAttr->window.skey > pQueryAttr->window.ekey) {
wafwerar's avatar
wafwerar 已提交
1312
//       TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
L
Liu Jicong 已提交
1313
//     }
H
Haojun Liao 已提交
1314
//
L
Liu Jicong 已提交
1315 1316 1317
//     pQueryAttr->needReverseScan = false;
//     return;
//   }
H
Haojun Liao 已提交
1318
//
L
Liu Jicong 已提交
1319 1320 1321
//   if (pQueryAttr->groupbyColumn && pQueryAttr->order.order == TSDB_ORDER_DESC) {
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     if (pQueryAttr->window.skey > pQueryAttr->window.ekey) {
wafwerar's avatar
wafwerar 已提交
1322
//       TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
L
Liu Jicong 已提交
1323
//     }
H
Haojun Liao 已提交
1324
//
L
Liu Jicong 已提交
1325 1326 1327 1328
//     pQueryAttr->needReverseScan = false;
//     doUpdateLastKey(pQueryAttr);
//     return;
//   }
H
Haojun Liao 已提交
1329
//
L
Liu Jicong 已提交
1330 1331 1332 1333 1334 1335
//   if (pQueryAttr->pointInterpQuery && pQueryAttr->interval.interval == 0) {
//     if (!QUERY_IS_ASC_QUERY(pQueryAttr)) {
//       //qDebug(msg, pQInfo->qId, "interp", pQueryAttr->order.order, TSDB_ORDER_ASC, pQueryAttr->window.skey,
//       pQueryAttr->window.ekey, pQueryAttr->window.ekey, pQueryAttr->window.skey); TSWAP(pQueryAttr->window.skey,
//       pQueryAttr->window.ekey, TSKEY);
//     }
H
Haojun Liao 已提交
1336
//
L
Liu Jicong 已提交
1337 1338 1339
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     return;
//   }
H
Haojun Liao 已提交
1340
//
L
Liu Jicong 已提交
1341 1342 1343 1344
//   if (pQueryAttr->interval.interval == 0) {
//     if (onlyFirstQuery(pQueryAttr)) {
//       if (!QUERY_IS_ASC_QUERY(pQueryAttr)) {
//         //qDebug(msg, pQInfo->qId, "only-first", pQueryAttr->order.order, TSDB_ORDER_ASC, pQueryAttr->window.skey,
H
Haojun Liao 已提交
1345 1346
////               pQueryAttr->window.ekey, pQueryAttr->window.ekey, pQueryAttr->window.skey);
//
wafwerar's avatar
wafwerar 已提交
1347
//        TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1348 1349 1350 1351 1352 1353 1354 1355 1356 1357
//        doUpdateLastKey(pQueryAttr);
//      }
//
//      pQueryAttr->order.order = TSDB_ORDER_ASC;
//      pQueryAttr->needReverseScan = false;
//    } else if (onlyLastQuery(pQueryAttr) && notContainSessionOrStateWindow(pQueryAttr)) {
//      if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//        //qDebug(msg, pQInfo->qId, "only-last", pQueryAttr->order.order, TSDB_ORDER_DESC, pQueryAttr->window.skey,
////               pQueryAttr->window.ekey, pQueryAttr->window.ekey, pQueryAttr->window.skey);
//
wafwerar's avatar
wafwerar 已提交
1358
//        TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370
//        doUpdateLastKey(pQueryAttr);
//      }
//
//      pQueryAttr->order.order = TSDB_ORDER_DESC;
//      pQueryAttr->needReverseScan = false;
//    }
//
//  } else {  // interval query
//    if (stableQuery) {
//      if (onlyFirstQuery(pQueryAttr)) {
//        if (!QUERY_IS_ASC_QUERY(pQueryAttr)) {
//          //qDebug(msg, pQInfo->qId, "only-first stable", pQueryAttr->order.order, TSDB_ORDER_ASC,
L
Liu Jicong 已提交
1371 1372
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
1373
//
wafwerar's avatar
wafwerar 已提交
1374
//          TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1375 1376 1377 1378 1379 1380 1381 1382
//          doUpdateLastKey(pQueryAttr);
//        }
//
//        pQueryAttr->order.order = TSDB_ORDER_ASC;
//        pQueryAttr->needReverseScan = false;
//      } else if (onlyLastQuery(pQueryAttr)) {
//        if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//          //qDebug(msg, pQInfo->qId, "only-last stable", pQueryAttr->order.order, TSDB_ORDER_DESC,
L
Liu Jicong 已提交
1383 1384
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
1385
//
wafwerar's avatar
wafwerar 已提交
1386
//          TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1387 1388 1389 1390 1391 1392 1393 1394 1395
//          doUpdateLastKey(pQueryAttr);
//        }
//
//        pQueryAttr->order.order = TSDB_ORDER_DESC;
//        pQueryAttr->needReverseScan = false;
//      }
//    }
//  }
//}
1396

L
Liu Jicong 已提交
1397 1398 1399
// static FORCE_INLINE bool doFilterByBlockStatistics(STaskRuntimeEnv* pRuntimeEnv, SDataStatis *pDataStatis,
// SqlFunctionCtx *pCtx, int32_t numOfRows) {
//   STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
1400
//
L
Liu Jicong 已提交
1401 1402 1403
//   if (pDataStatis == NULL || pQueryAttr->pFilters == NULL) {
//     return true;
//   }
1404
//
L
Liu Jicong 已提交
1405 1406
//   return filterRangeExecute(pQueryAttr->pFilters, pDataStatis, pQueryAttr->numOfCols, numOfRows);
// }
1407

H
Haojun Liao 已提交
1408
static bool overlapWithTimeWindow(STaskAttr* pQueryAttr, SDataBlockInfo* pBlockInfo) {
1409 1410
  STimeWindow w = {0};

dengyihao's avatar
dengyihao 已提交
1411 1412
  TSKEY sk = TMIN(pQueryAttr->window.skey, pQueryAttr->window.ekey);
  TSKEY ek = TMAX(pQueryAttr->window.skey, pQueryAttr->window.ekey);
1413

1414
  if (true) {
L
Liu Jicong 已提交
1415
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.skey, sk, ek, &w);
1416 1417 1418 1419 1420 1421
    assert(w.ekey >= pBlockInfo->window.skey);

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

L
Liu Jicong 已提交
1422 1423
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
1424 1425 1426 1427 1428 1429 1430 1431 1432 1433
      if (w.skey > pBlockInfo->window.ekey) {
        break;
      }

      assert(w.ekey > pBlockInfo->window.ekey);
      if (w.skey <= pBlockInfo->window.ekey && w.skey > pBlockInfo->window.skey) {
        return true;
      }
    }
  } else {
L
Liu Jicong 已提交
1434
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.ekey, sk, ek, &w);
1435 1436 1437 1438 1439 1440
    assert(w.skey <= pBlockInfo->window.ekey);

    if (w.skey > pBlockInfo->window.skey) {
      return true;
    }

L
Liu Jicong 已提交
1441 1442
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506
      if (w.ekey < pBlockInfo->window.skey) {
        break;
      }

      assert(w.skey < pBlockInfo->window.skey);
      if (w.ekey < pBlockInfo->window.ekey && w.ekey >= pBlockInfo->window.skey) {
        return true;
      }
    }
  }

  return false;
}

void doCompactSDataBlock(SSDataBlock* pBlock, int32_t numOfRows, int8_t* p) {
  int32_t len = 0;
  int32_t start = 0;
  for (int32_t j = 0; j < numOfRows; ++j) {
    if (p[j] == 1) {
      len++;
    } else {
      if (len > 0) {
        int32_t cstart = j - len;
        for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
          SColumnInfoData* pColumnInfoData = taosArrayGet(pBlock->pDataBlock, i);

          int16_t bytes = pColumnInfoData->info.bytes;
          memmove(((char*)pColumnInfoData->pData) + start * bytes, pColumnInfoData->pData + cstart * bytes,
                  len * bytes);
        }

        start += len;
        len = 0;
      }
    }
  }

  if (len > 0) {
    int32_t cstart = numOfRows - len;
    for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
      SColumnInfoData* pColumnInfoData = taosArrayGet(pBlock->pDataBlock, i);

      int16_t bytes = pColumnInfoData->info.bytes;
      memmove(pColumnInfoData->pData + start * bytes, pColumnInfoData->pData + cstart * bytes, len * bytes);
    }

    start += len;
    len = 0;
  }

  pBlock->info.rows = start;
  pBlock->pBlockAgg = NULL;  // clean the block statistics info

  if (start > 0) {
    SColumnInfoData* pColumnInfoData = taosArrayGet(pBlock->pDataBlock, 0);
    if (pColumnInfoData->info.type == TSDB_DATA_TYPE_TIMESTAMP &&
        pColumnInfoData->info.colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
      pBlock->info.window.skey = *(int64_t*)pColumnInfoData->pData;
      pBlock->info.window.ekey = *(int64_t*)(pColumnInfoData->pData + TSDB_KEYSIZE * (start - 1));
    }
  }
}

static uint32_t doFilterByBlockTimeWindow(STableScanInfo* pTableScanInfo, SSDataBlock* pBlock) {
H
Haojun Liao 已提交
1507
  SqlFunctionCtx* pCtx = pTableScanInfo->pCtx;
1508
  uint32_t        status = BLK_DATA_NOT_LOAD;
1509 1510 1511 1512

  int32_t numOfOutput = pTableScanInfo->numOfOutput;
  for (int32_t i = 0; i < numOfOutput; ++i) {
    int32_t functionId = pCtx[i].functionId;
H
Haojun Liao 已提交
1513
    int32_t colId = pTableScanInfo->pExpr[i].base.pParam[0].pCol->colId;
1514 1515 1516

    // group by + first/last should not apply the first/last block filter
    if (functionId < 0) {
1517
      status |= BLK_DATA_DATA_LOAD;
1518 1519
      return status;
    } else {
L
Liu Jicong 已提交
1520
      //      status |= aAggs[functionId].dataReqFunc(&pTableScanInfo->pCtx[i], &pBlock->info.window, colId);
1521
      //      if ((status & BLK_DATA_DATA_LOAD) == BLK_DATA_DATA_LOAD) {
L
Liu Jicong 已提交
1522 1523
      //        return status;
      //      }
1524 1525 1526 1527 1528 1529
    }
  }

  return status;
}

L
Liu Jicong 已提交
1530 1531
int32_t loadDataBlockOnDemand(SExecTaskInfo* pTaskInfo, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock,
                              uint32_t* status) {
1532
  *status = BLK_DATA_NOT_LOAD;
1533

H
Haojun Liao 已提交
1534
  pBlock->pDataBlock = NULL;
L
Liu Jicong 已提交
1535
  pBlock->pBlockAgg = NULL;
H
Haojun Liao 已提交
1536

L
Liu Jicong 已提交
1537 1538
  //  int64_t groupId = pRuntimeEnv->current->groupIndex;
  //  bool    ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
1539

H
Haojun Liao 已提交
1540
  STaskCostInfo* pCost = &pTaskInfo->cost;
1541

1542 1543
//  pCost->totalBlocks += 1;
//  pCost->totalRows += pBlock->info.rows;
H
Haojun Liao 已提交
1544
#if 0
1545 1546 1547
  // Calculate all time windows that are overlapping or contain current data block.
  // If current data block is contained by all possible time window, do not load current data block.
  if (/*pQueryAttr->pFilters || */pQueryAttr->groupbyColumn || pQueryAttr->sw.gap > 0 ||
H
Haojun Liao 已提交
1548
      (QUERY_IS_INTERVAL_QUERY(pQueryAttr) && overlapWithTimeWindow(pTaskInfo, &pBlock->info))) {
1549
    (*status) = BLK_DATA_DATA_LOAD;
1550 1551 1552
  }

  // check if this data block is required to load
1553
  if ((*status) != BLK_DATA_DATA_LOAD) {
1554 1555 1556 1557 1558 1559 1560
    bool needFilter = true;

    // the pCtx[i] result is belonged to previous time window since the outputBuf has not been set yet,
    // the filter result may be incorrect. So in case of interval query, we need to set the correct time output buffer
    if (QUERY_IS_INTERVAL_QUERY(pQueryAttr)) {
      SResultRow* pResult = NULL;

H
Haojun Liao 已提交
1561
      bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584
      TSKEY k = ascQuery? pBlock->info.window.skey : pBlock->info.window.ekey;

      STimeWindow win = getActiveTimeWindow(pTableScanInfo->pResultRowInfo, k, pQueryAttr);
      if (pQueryAttr->pointInterpQuery) {
        needFilter = chkWindowOutputBufByKey(pRuntimeEnv, pTableScanInfo->pResultRowInfo, &win, masterScan, &pResult, groupId,
                                    pTableScanInfo->pCtx, pTableScanInfo->numOfOutput,
                                    pTableScanInfo->rowCellInfoOffset);
      } else {
        if (setResultOutputBufByKey(pRuntimeEnv, pTableScanInfo->pResultRowInfo, pBlock->info.uid, &win, masterScan, &pResult, groupId,
                                    pTableScanInfo->pCtx, pTableScanInfo->numOfOutput,
                                    pTableScanInfo->rowCellInfoOffset) != TSDB_CODE_SUCCESS) {
          longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
        }
      }
    } else if (pQueryAttr->stableQuery && (!pQueryAttr->tsCompQuery) && (!pQueryAttr->diffQuery)) { // stable aggregate, not interval aggregate or normal column aggregate
      doSetTableGroupOutputBuf(pRuntimeEnv, pTableScanInfo->pResultRowInfo, pTableScanInfo->pCtx,
                               pTableScanInfo->rowCellInfoOffset, pTableScanInfo->numOfOutput,
                               pRuntimeEnv->current->groupIndex);
    }

    if (needFilter) {
      (*status) = doFilterByBlockTimeWindow(pTableScanInfo, pBlock);
    } else {
1585
      (*status) = BLK_DATA_DATA_LOAD;
1586 1587 1588 1589
    }
  }

  SDataBlockInfo* pBlockInfo = &pBlock->info;
H
Haojun Liao 已提交
1590
//  *status = updateBlockLoadStatus(pRuntimeEnv->pQueryAttr, *status);
1591

1592
  if ((*status) == BLK_DATA_NOT_LOAD || (*status) == BLK_DATA_FILTEROUT) {
1593 1594
    //qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId, pBlockInfo->window.skey,
//           pBlockInfo->window.ekey, pBlockInfo->rows);
1595
    pCost->skipBlocks += 1;
1596
  } else if ((*status) == BLK_DATA_SMA_LOAD) {
1597 1598
    // this function never returns error?
    pCost->loadBlockStatis += 1;
1599
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
1600 1601

    if (pBlock->pBlockAgg == NULL) {  // data block statistics does not exist, load data block
1602
//      pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
1603 1604 1605
      pCost->totalCheckedRows += pBlock->info.rows;
    }
  } else {
1606
    assert((*status) == BLK_DATA_DATA_LOAD);
1607 1608 1609

    // load the data block statistics to perform further filter
    pCost->loadBlockStatis += 1;
1610
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
1611 1612 1613 1614 1615 1616

    if (pQueryAttr->topBotQuery && pBlock->pBlockAgg != NULL) {
      { // set previous window
        if (QUERY_IS_INTERVAL_QUERY(pQueryAttr)) {
          SResultRow* pResult = NULL;

H
Haojun Liao 已提交
1617
          bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634
          TSKEY k = ascQuery? pBlock->info.window.skey : pBlock->info.window.ekey;

          STimeWindow win = getActiveTimeWindow(pTableScanInfo->pResultRowInfo, k, pQueryAttr);
          if (setResultOutputBufByKey(pRuntimeEnv, pTableScanInfo->pResultRowInfo, pBlock->info.uid, &win, masterScan, &pResult, groupId,
                                      pTableScanInfo->pCtx, pTableScanInfo->numOfOutput,
                                      pTableScanInfo->rowCellInfoOffset) != TSDB_CODE_SUCCESS) {
            longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
          }
        }
      }
      bool load = false;
      for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
        int32_t functionId = pTableScanInfo->pCtx[i].functionId;
        if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM) {
//          load = topbot_datablock_filter(&pTableScanInfo->pCtx[i], (char*)&(pBlock->pBlockAgg[i].min),
//                                         (char*)&(pBlock->pBlockAgg[i].max));
          if (!load) { // current block has been discard due to filter applied
1635
            pCost->skipBlocks += 1;
1636 1637
            //qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId,
//                   pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
1638
            (*status) = BLK_DATA_FILTEROUT;
1639 1640 1641 1642 1643 1644 1645 1646
            return TSDB_CODE_SUCCESS;
          }
        }
      }
    }

    // current block has been discard due to filter applied
//    if (!doFilterByBlockStatistics(pRuntimeEnv, pBlock->pBlockAgg, pTableScanInfo->pCtx, pBlockInfo->rows)) {
1647
//      pCost->skipBlocks += 1;
1648 1649
//      qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId, pBlockInfo->window.skey,
//             pBlockInfo->window.ekey, pBlockInfo->rows);
1650
//      (*status) = BLK_DATA_FILTEROUT;
1651 1652 1653 1654 1655
//      return TSDB_CODE_SUCCESS;
//    }

    pCost->totalCheckedRows += pBlockInfo->rows;
    pCost->loadBlocks += 1;
1656
//    pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
1657 1658 1659 1660 1661 1662 1663
//    if (pBlock->pDataBlock == NULL) {
//      return terrno;
//    }

//    if (pQueryAttr->pFilters != NULL) {
//      filterSetColFieldData(pQueryAttr->pFilters, pBlock->info.numOfCols, pBlock->pDataBlock);
//    }
1664

1665 1666 1667 1668
//    if (pQueryAttr->pFilters != NULL || pRuntimeEnv->pTsBuf != NULL) {
//      filterColRowsInDataBlock(pRuntimeEnv, pBlock, ascQuery);
//    }
  }
H
Haojun Liao 已提交
1669
#endif
1670 1671 1672
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
1673
static void updateTableQueryInfoForReverseScan(STableQueryInfo* pTableQueryInfo) {
1674 1675 1676 1677
  if (pTableQueryInfo == NULL) {
    return;
  }

wafwerar's avatar
wafwerar 已提交
1678
  //  TSWAP(pTableQueryInfo->win.skey, pTableQueryInfo->win.ekey);
L
Liu Jicong 已提交
1679
  //  pTableQueryInfo->lastKey = pTableQueryInfo->win.skey;
1680

L
Liu Jicong 已提交
1681 1682
  //  SWITCH_ORDER(pTableQueryInfo->cur.order);
  //  pTableQueryInfo->cur.vgroupIndex = -1;
1683 1684

  // set the index to be the end slot of result rows array
dengyihao's avatar
dengyihao 已提交
1685 1686 1687 1688 1689 1690
  //  SResultRowInfo* pResultRowInfo = &pTableQueryInfo->resInfo;
  //  if (pResultRowInfo->size > 0) {
  //    pResultRowInfo->curPos = pResultRowInfo->size - 1;
  //  } else {
  //    pResultRowInfo->curPos = -1;
  //  }
1691 1692
}

H
Haojun Liao 已提交
1693
void initResultRow(SResultRow* pResultRow) {
X
Xiaoyu Wang 已提交
1694
  //  pResultRow->pEntryInfo = (struct SResultRowEntryInfo*)((char*)pResultRow + sizeof(SResultRow));
1695 1696 1697 1698 1699
}

/*
 * The start of each column SResultRowEntryInfo is denote by RowCellInfoOffset.
 * Note that in case of top/bottom query, the whole multiple rows of result is treated as only one row of results.
H
Haojun Liao 已提交
1700 1701 1702
 * +------------+-----------------result column 1------------+------------------result column 2-----------+
 * | SResultRow | SResultRowEntryInfo | intermediate buffer1 | SResultRowEntryInfo | intermediate buffer 2|
 * +------------+--------------------------------------------+--------------------------------------------+
1703 1704
 *           offset[0]                                  offset[1]                                   offset[2]
 */
1705
// TODO refactor: some function move away
X
Xiaoyu Wang 已提交
1706 1707
void setFunctionResultOutput(SOptrBasicInfo* pInfo, SAggSupporter* pSup, int32_t stage, int32_t numOfExprs,
                             SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
1708 1709 1710
  SqlFunctionCtx* pCtx = pInfo->pCtx;
  SSDataBlock*    pDataBlock = pInfo->pRes;
  int32_t*        rowCellInfoOffset = pInfo->rowCellInfoOffset;
H
Haojun Liao 已提交
1711

H
Haojun Liao 已提交
1712
  SResultRowInfo* pResultRowInfo = &pInfo->resultRowInfo;
H
Haojun Liao 已提交
1713
  initResultRowInfo(pResultRowInfo, 16);
H
Haojun Liao 已提交
1714

L
Liu Jicong 已提交
1715 1716
  int64_t     tid = 0;
  int64_t     groupId = 0;
1717 1718
  SResultRow* pRow = doSetResultOutBufByKey(pSup->pResultBuf, pResultRowInfo, (char*)&tid, sizeof(tid), true, groupId,
                                            pTaskInfo, false, pSup);
H
Haojun Liao 已提交
1719

1720
  for (int32_t i = 0; i < numOfExprs; ++i) {
H
Haojun Liao 已提交
1721 1722 1723
    struct SResultRowEntryInfo* pEntry = getResultCell(pRow, i, rowCellInfoOffset);
    cleanupResultRowEntry(pEntry);

L
Liu Jicong 已提交
1724
    pCtx[i].resultInfo = pEntry;
1725
    pCtx[i].scanFlag = stage;
H
Haojun Liao 已提交
1726 1727
  }

1728
  initCtxOutputBuffer(pCtx, numOfExprs);
H
Haojun Liao 已提交
1729 1730
}

L
Liu Jicong 已提交
1731
void updateOutputBuf(SOptrBasicInfo* pBInfo, int32_t* bufCapacity, int32_t numOfInputRows) {
1732 1733
  SSDataBlock* pDataBlock = pBInfo->pRes;

L
Liu Jicong 已提交
1734
  int32_t newSize = pDataBlock->info.rows + numOfInputRows + 5;  // extra output buffer
1735
  if ((*bufCapacity) < newSize) {
L
Liu Jicong 已提交
1736 1737
    for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
      SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
1738

wafwerar's avatar
wafwerar 已提交
1739
      char* p = taosMemoryRealloc(pColInfo->pData, newSize * pColInfo->info.bytes);
1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752
      if (p != NULL) {
        pColInfo->pData = p;

        // it starts from the tail of the previously generated results.
        pBInfo->pCtx[i].pOutput = pColInfo->pData;
        (*bufCapacity) = newSize;
      } else {
        // longjmp
      }
    }
  }

  for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
L
Liu Jicong 已提交
1753
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
1754 1755 1756 1757
    pBInfo->pCtx[i].pOutput = pColInfo->pData + pColInfo->info.bytes * pDataBlock->info.rows;

    // set the correct pointer after the memory buffer reallocated.
    int32_t functionId = pBInfo->pCtx[i].functionId;
1758
#if 0
L
Liu Jicong 已提交
1759 1760
    if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF ||
        functionId == FUNCTION_DERIVATIVE) {
dengyihao's avatar
dengyihao 已提交
1761
      //      if (i > 0) pBInfo->pCtx[i].pTsOutput = pBInfo->pCtx[i - 1].pOutput;
1762
    }
1763
#endif
1764 1765 1766
  }
}

H
Haojun Liao 已提交
1767
void initCtxOutputBuffer(SqlFunctionCtx* pCtx, int32_t size) {
1768 1769
  for (int32_t j = 0; j < size; ++j) {
    struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(&pCtx[j]);
dengyihao's avatar
dengyihao 已提交
1770 1771
    if (isRowEntryInitialized(pResInfo) || fmIsPseudoColumnFunc(pCtx[j].functionId) || pCtx[j].functionId == -1 ||
        fmIsScalarFunc(pCtx[j].functionId)) {
1772 1773 1774
      continue;
    }

H
Haojun Liao 已提交
1775
    pCtx[j].fpSet.init(&pCtx[j], pCtx[j].resultInfo);
1776 1777 1778
  }
}

L
Liu Jicong 已提交
1779
void setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status) {
1780
  if (status == TASK_NOT_COMPLETED) {
H
Haojun Liao 已提交
1781
    pTaskInfo->status = status;
1782 1783
  } else {
    // QUERY_NOT_COMPLETED is not compatible with any other status, so clear its position first
1784
    CLEAR_QUERY_STATUS(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
1785
    pTaskInfo->status |= status;
1786 1787 1788
  }
}

L
Liu Jicong 已提交
1789
void destroyTableQueryInfoImpl(STableQueryInfo* pTableQueryInfo) {
1790 1791 1792 1793
  if (pTableQueryInfo == NULL) {
    return;
  }

L
Liu Jicong 已提交
1794
  //  taosVariantDestroy(&pTableQueryInfo->tag);
dengyihao's avatar
dengyihao 已提交
1795
  //  cleanupResultRowInfo(&pTableQueryInfo->resInfo);
1796 1797
}

1798
void setResultRowInitCtx(SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowCellInfoOffset) {
1799 1800 1801 1802 1803 1804 1805
  for (int32_t i = 0; i < numOfOutput; ++i) {
    pCtx[i].resultInfo = getResultCell(pResult, i, rowCellInfoOffset);

    struct SResultRowEntryInfo* pResInfo = pCtx[i].resultInfo;
    if (isRowEntryCompleted(pResInfo) && isRowEntryInitialized(pResInfo)) {
      continue;
    }
1806 1807 1808 1809 1810

    if (fmIsWindowPseudoColumnFunc(pCtx[i].functionId)) {
      continue;
    }

1811 1812 1813 1814 1815 1816
    if (!pResInfo->initialized) {
      if (pCtx[i].functionId != -1) {
        pCtx[i].fpSet.init(&pCtx[i], pResInfo);
      } else {
        pResInfo->initialized = true;
      }
1817 1818 1819 1820
    }
  }
}

1821
static void extractQualifiedTupleByFilterResult(SSDataBlock* pBlock, const int8_t* rowRes, bool keep);
1822 1823

void doFilter(const SNode* pFilterNode, SSDataBlock* pBlock) {
1824 1825 1826 1827 1828
  if (pFilterNode == NULL) {
    return;
  }

  SFilterInfo* filter = NULL;
H
Haojun Liao 已提交
1829

H
Haojun Liao 已提交
1830
  // todo move to the initialization function
H
Haojun Liao 已提交
1831
  int32_t code = filterInitFromNode((SNode*)pFilterNode, &filter, 0);
1832 1833 1834 1835 1836

  SFilterColumnParam param1 = {.numOfCols = pBlock->info.numOfCols, .pDataBlock = pBlock->pDataBlock};
  code = filterSetDataFromSlotId(filter, &param1);

  int8_t* rowRes = NULL;
1837

1838
  // todo the keep seems never to be True??
1839
  bool keep = filterExecute(filter, pBlock, &rowRes, NULL, param1.numOfCols);
D
dapan1121 已提交
1840
  filterFreeInfo(filter);
1841

1842
  extractQualifiedTupleByFilterResult(pBlock, rowRes, keep);
1843
  blockDataUpdateTsWindow(pBlock, 0);
1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
}

void extractQualifiedTupleByFilterResult(SSDataBlock* pBlock, const int8_t* rowRes, bool keep) {
  if (keep) {
    return;
  }

  if (rowRes != NULL) {
    SSDataBlock* px = createOneDataBlock(pBlock, false);
    blockDataEnsureCapacity(px, pBlock->info.rows);

1855 1856
    int32_t totalRows = pBlock->info.rows;

1857 1858 1859 1860
    for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
      SColumnInfoData* pDst = taosArrayGet(px->pDataBlock, i);
      SColumnInfoData* pSrc = taosArrayGet(pBlock->pDataBlock, i);

1861 1862 1863 1864 1865
      // it is a reserved column for scalar function, and no data in this column yet.
      if (pSrc->pData == NULL) {
        continue;
      }

1866
      int32_t numOfRows = 0;
1867
      for (int32_t j = 0; j < totalRows; ++j) {
D
dapan1121 已提交
1868 1869 1870
        if (rowRes[j] == 0) {
          continue;
        }
1871

D
dapan1121 已提交
1872
        if (colDataIsNull_s(pSrc, j)) {
1873
          colDataAppendNULL(pDst, numOfRows);
D
dapan1121 已提交
1874
        } else {
1875
          colDataAppend(pDst, numOfRows, colDataGetData(pSrc, j), false);
D
dapan1121 已提交
1876
        }
1877
        numOfRows += 1;
H
Haojun Liao 已提交
1878
      }
1879

1880 1881 1882 1883 1884 1885
      if (pBlock->info.rows == totalRows) {
        pBlock->info.rows = numOfRows;
      } else {
        ASSERT(pBlock->info.rows == numOfRows);
      }

1886 1887 1888 1889 1890
      *pSrc = *pDst;
    }
  } else {
    // do nothing
    pBlock->info.rows = 0;
1891 1892 1893
  }
}

dengyihao's avatar
dengyihao 已提交
1894 1895
void doSetTableGroupOutputBuf(SAggOperatorInfo* pAggInfo, int32_t numOfOutput, uint64_t groupId,
                              SExecTaskInfo* pTaskInfo) {
1896 1897 1898
  // for simple group by query without interval, all the tables belong to one group result.
  int64_t uid = 0;

1899 1900
  SResultRowInfo* pResultRowInfo = &pAggInfo->binfo.resultRowInfo;
  SqlFunctionCtx* pCtx = pAggInfo->binfo.pCtx;
L
Liu Jicong 已提交
1901
  int32_t*        rowCellInfoOffset = pAggInfo->binfo.rowCellInfoOffset;
1902

1903
  SResultRow* pResultRow = doSetResultOutBufByKey(pAggInfo->aggSup.pResultBuf, pResultRowInfo, (char*)&groupId,
L
Liu Jicong 已提交
1904
                                                  sizeof(groupId), true, groupId, pTaskInfo, false, &pAggInfo->aggSup);
L
Liu Jicong 已提交
1905
  assert(pResultRow != NULL);
1906 1907 1908 1909 1910 1911

  /*
   * not assign result buffer yet, add new result buffer
   * all group belong to one result set, and each group result has different group id so set the id to be one
   */
  if (pResultRow->pageId == -1) {
dengyihao's avatar
dengyihao 已提交
1912 1913
    int32_t ret =
        addNewWindowResultBuf(pResultRow, pAggInfo->aggSup.pResultBuf, groupId, pAggInfo->binfo.pRes->info.rowSize);
1914 1915 1916 1917 1918
    if (ret != TSDB_CODE_SUCCESS) {
      return;
    }
  }

1919
  setResultRowInitCtx(pResultRow, pCtx, numOfOutput, rowCellInfoOffset);
1920 1921
}

H
Haojun Liao 已提交
1922 1923
void setExecutionContext(int32_t numOfOutput, uint64_t groupId, SExecTaskInfo* pTaskInfo, SAggOperatorInfo* pAggInfo) {
  if (pAggInfo->groupId != INT32_MIN && pAggInfo->groupId == groupId) {
1924 1925 1926
    return;
  }

H
Haojun Liao 已提交
1927
  doSetTableGroupOutputBuf(pAggInfo, numOfOutput, groupId, pTaskInfo);
1928 1929

  // record the current active group id
H
Haojun Liao 已提交
1930
  pAggInfo->groupId = groupId;
1931 1932
}

1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945
static void doUpdateNumOfRows(SResultRow* pRow, int32_t numOfExprs, const int32_t* rowCellOffset) {
  for (int32_t j = 0; j < numOfExprs; ++j) {
    struct SResultRowEntryInfo* pResInfo = getResultCell(pRow, j, rowCellOffset);
    if (!isRowEntryInitialized(pResInfo)) {
      continue;
    }

    if (pRow->numOfRows < pResInfo->numOfRes) {
      pRow->numOfRows = pResInfo->numOfRes;
    }
  }
}

X
Xiaoyu Wang 已提交
1946 1947 1948
int32_t doCopyToSDataBlock(SExecTaskInfo* pTaskInfo, SSDataBlock* pBlock, SExprInfo* pExprInfo, SDiskbasedBuf* pBuf,
                           SGroupResInfo* pGroupResInfo, const int32_t* rowCellOffset, SqlFunctionCtx* pCtx,
                           int32_t numOfExprs) {
1949
  int32_t numOfRows = getNumOfTotalRes(pGroupResInfo);
1950
  int32_t start = pGroupResInfo->index;
1951

1952
  for (int32_t i = start; i < numOfRows; i += 1) {
L
Liu Jicong 已提交
1953 1954
    SResKeyPos* pPos = taosArrayGetP(pGroupResInfo->pRows, i);
    SFilePage*  page = getBufPage(pBuf, pPos->pos.pageId);
1955

1956
    SResultRow* pRow = (SResultRow*)((char*)page + pPos->pos.offset);
1957 1958

    doUpdateNumOfRows(pRow, numOfExprs, rowCellOffset);
1959 1960
    if (pRow->numOfRows == 0) {
      pGroupResInfo->index += 1;
1961
      releaseBufPage(pBuf, page);
1962 1963 1964
      continue;
    }

1965 1966 1967 1968 1969
    if (pBlock->info.groupId == 0) {
      pBlock->info.groupId = pPos->groupId;
    } else {
      // current value belongs to different group, it can't be packed into one datablock
      if (pBlock->info.groupId != pPos->groupId) {
1970
        releaseBufPage(pBuf, page);
1971 1972 1973 1974
        break;
      }
    }

1975
    if (pBlock->info.rows + pRow->numOfRows > pBlock->info.capacity) {
1976
      releaseBufPage(pBuf, page);
1977 1978 1979 1980 1981
      break;
    }

    pGroupResInfo->index += 1;

1982
    for (int32_t j = 0; j < numOfExprs; ++j) {
1983 1984
      int32_t slotId = pExprInfo[j].base.resSchema.slotId;

1985
      pCtx[j].resultInfo = getResultCell(pRow, j, rowCellOffset);
1986
      if (pCtx[j].fpSet.finalize) {
1987
        int32_t code = pCtx[j].fpSet.finalize(&pCtx[j], pBlock);
1988
        if (TAOS_FAILED(code)) {
1989 1990
          qError("%s build result data block error, code %s", GET_TASKID(pTaskInfo), tstrerror(code));
          longjmp(pTaskInfo->env, code);
1991
        }
1992 1993
      } else if (strcmp(pCtx[j].pExpr->pExpr->_function.functionName, "_select_value") == 0) {
        // do nothing, todo refactor
1994
      } else {
1995 1996
        // expand the result into multiple rows. E.g., _wstartts, top(k, 20)
        // the _wstartts needs to copy to 20 following rows, since the results of top-k expands to 20 different rows.
X
Xiaoyu Wang 已提交
1997 1998 1999 2000 2001
        SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, slotId);
        char*            in = GET_ROWCELL_INTERBUF(pCtx[j].resultInfo);
        for (int32_t k = 0; k < pRow->numOfRows; ++k) {
          colDataAppend(pColInfoData, pBlock->info.rows + k, in, pCtx[j].resultInfo->isNullRes);
        }
2002
      }
2003 2004
    }

2005
    releaseBufPage(pBuf, page);
2006 2007
    pBlock->info.rows += pRow->numOfRows;
    if (pBlock->info.rows >= pBlock->info.capacity) {  // output buffer is full
2008 2009 2010 2011
      break;
    }
  }

X
Xiaoyu Wang 已提交
2012 2013
  qDebug("%s result generated, rows:%d, groupId:%" PRIu64, GET_TASKID(pTaskInfo), pBlock->info.rows,
         pBlock->info.groupId);
2014
  blockDataUpdateTsWindow(pBlock, 0);
2015 2016 2017
  return 0;
}

X
Xiaoyu Wang 已提交
2018 2019 2020 2021
void doBuildResultDatablock(SOperatorInfo* pOperator, SOptrBasicInfo* pbInfo, SGroupResInfo* pGroupResInfo,
                            SDiskbasedBuf* pBuf) {
  SExprInfo*     pExprInfo = pOperator->pExpr;
  int32_t        numOfExprs = pOperator->numOfExprs;
2022 2023
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;

X
Xiaoyu Wang 已提交
2024 2025 2026
  int32_t*        rowCellOffset = pbInfo->rowCellInfoOffset;
  SSDataBlock*    pBlock = pbInfo->pRes;
  SqlFunctionCtx* pCtx = pbInfo->pCtx;
2027

2028
  blockDataCleanup(pBlock);
2029
  if (!hashRemainDataInGroupInfo(pGroupResInfo)) {
2030 2031 2032
    return;
  }

2033 2034
  // clear the existed group id
  pBlock->info.groupId = 0;
2035
  doCopyToSDataBlock(pTaskInfo, pBlock, pExprInfo, pBuf, pGroupResInfo, rowCellOffset, pCtx, numOfExprs);
2036 2037
}

L
Liu Jicong 已提交
2038 2039
static void updateNumOfRowsInResultRows(SqlFunctionCtx* pCtx, int32_t numOfOutput, SResultRowInfo* pResultRowInfo,
                                        int32_t* rowCellInfoOffset) {
2040
  // update the number of result for each, only update the number of rows for the corresponding window result.
L
Liu Jicong 已提交
2041 2042 2043
  //  if (QUERY_IS_INTERVAL_QUERY(pQueryAttr)) {
  //    return;
  //  }
H
Haojun Liao 已提交
2044
#if 0
2045
  for (int32_t i = 0; i < pResultRowInfo->size; ++i) {
L
Liu Jicong 已提交
2046
    SResultRow* pResult = pResultRowInfo->pResult[i];
2047 2048 2049 2050 2051 2052 2053

    for (int32_t j = 0; j < numOfOutput; ++j) {
      int32_t functionId = pCtx[j].functionId;
      if (functionId == FUNCTION_TS || functionId == FUNCTION_TAG || functionId == FUNCTION_TAGPRJ) {
        continue;
      }

2054 2055
      SResultRowEntryInfo* pCell = getResultCell(pResult, j, rowCellInfoOffset);
      pResult->numOfRows = (uint16_t)(TMAX(pResult->numOfRows, pCell->numOfRes));
2056 2057
    }
  }
H
Haojun Liao 已提交
2058
#endif
2059 2060
}

L
Liu Jicong 已提交
2061
static int32_t compressQueryColData(SColumnInfoData* pColRes, int32_t numOfRows, char* data, int8_t compressed) {
2062 2063
  int32_t colSize = pColRes->info.bytes * numOfRows;
  return (*(tDataTypes[pColRes->info.type].compFunc))(pColRes->pData, colSize, numOfRows, data,
L
Liu Jicong 已提交
2064
                                                      colSize + COMP_OVERFLOW_BYTES, compressed, NULL, 0);
2065 2066
}

2067
int32_t doFillTimeIntervalGapsInResults(struct SFillInfo* pFillInfo, SSDataBlock* pBlock, int32_t capacity) {
L
Liu Jicong 已提交
2068 2069 2070 2071
  //  for(int32_t i = 0; i < pFillInfo->numOfCols; ++i) {
  //    SColumnInfoData* pColInfoData = taosArrayGet(pOutput->pDataBlock, i);
  //    p[i] = pColInfoData->pData + (pColInfoData->info.bytes * pOutput->info.rows);
  //  }
2072

2073 2074
  int32_t numOfRows = (int32_t)taosFillResultDataBlock(pFillInfo, pBlock, capacity - pBlock->info.rows);
  pBlock->info.rows += numOfRows;
2075

2076
  return pBlock->info.rows;
2077 2078
}

L
Liu Jicong 已提交
2079 2080
void queryCostStatis(SExecTaskInfo* pTaskInfo) {
  STaskCostInfo* pSummary = &pTaskInfo->cost;
2081

L
Liu Jicong 已提交
2082 2083 2084
  //  uint64_t hashSize = taosHashGetMemSize(pQInfo->runtimeEnv.pResultRowHashTable);
  //  hashSize += taosHashGetMemSize(pRuntimeEnv->tableqinfoGroupInfo.map);
  //  pSummary->hashSize = hashSize;
2085 2086 2087 2088

  // add the merge time
  pSummary->elapsedTime += pSummary->firstStageMergeTime;

L
Liu Jicong 已提交
2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099
  //  SResultRowPool* p = pTaskInfo->pool;
  //  if (p != NULL) {
  //    pSummary->winInfoSize = getResultRowPoolMemSize(p);
  //    pSummary->numOfTimeWindows = getNumOfAllocatedResultRows(p);
  //  } else {
  //    pSummary->winInfoSize = 0;
  //    pSummary->numOfTimeWindows = 0;
  //  }
  //
  //  calculateOperatorProfResults(pQInfo);

2100 2101
  SFileBlockLoadRecorder* pRecorder = pSummary->pRecoder;
  if (pSummary->pRecoder != NULL) {
X
Xiaoyu Wang 已提交
2102 2103
    qDebug("%s :cost summary: elapsed time:%" PRId64 " us, first merge:%" PRId64
           " us, total blocks:%d, "
2104 2105 2106 2107
           "load block statis:%d, load data block:%d, total rows:%" PRId64 ", check rows:%" PRId64,
           GET_TASKID(pTaskInfo), pSummary->elapsedTime, pSummary->firstStageMergeTime, pRecorder->totalBlocks,
           pRecorder->loadBlockStatis, pRecorder->loadBlocks, pRecorder->totalRows, pRecorder->totalCheckedRows);
  }
L
Liu Jicong 已提交
2108 2109 2110
  // qDebug("QInfo:0x%"PRIx64" :cost summary: winResPool size:%.2f Kb, numOfWin:%"PRId64", tableInfoSize:%.2f Kb,
  // hashTable:%.2f Kb", pQInfo->qId, pSummary->winInfoSize/1024.0,
  //      pSummary->numOfTimeWindows, pSummary->tableInfoSize/1024.0, pSummary->hashSize/1024.0);
2111 2112
}

L
Liu Jicong 已提交
2113 2114 2115
// static void updateOffsetVal(STaskRuntimeEnv *pRuntimeEnv, SDataBlockInfo *pBlockInfo) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
2116
//
L
Liu Jicong 已提交
2117
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
2118
//
L
Liu Jicong 已提交
2119 2120 2121 2122
//   if (pQueryAttr->limit.offset == pBlockInfo->rows) {  // current block will ignore completed
//     pTableQueryInfo->lastKey = QUERY_IS_ASC_QUERY(pQueryAttr) ? pBlockInfo->window.ekey + step :
//     pBlockInfo->window.skey + step; pQueryAttr->limit.offset = 0; return;
//   }
2123
//
L
Liu Jicong 已提交
2124 2125 2126 2127 2128
//   if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//     pQueryAttr->pos = (int32_t)pQueryAttr->limit.offset;
//   } else {
//     pQueryAttr->pos = pBlockInfo->rows - (int32_t)pQueryAttr->limit.offset - 1;
//   }
2129
//
L
Liu Jicong 已提交
2130
//   assert(pQueryAttr->pos >= 0 && pQueryAttr->pos <= pBlockInfo->rows - 1);
2131
//
L
Liu Jicong 已提交
2132 2133
//   SArray *         pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//   SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
2134
//
L
Liu Jicong 已提交
2135 2136
//   // update the pQueryAttr->limit.offset value, and pQueryAttr->pos value
//   TSKEY *keys = (TSKEY *) pColInfoData->pData;
2137
//
L
Liu Jicong 已提交
2138 2139 2140
//   // update the offset value
//   pTableQueryInfo->lastKey = keys[pQueryAttr->pos];
//   pQueryAttr->limit.offset = 0;
2141
//
L
Liu Jicong 已提交
2142
//   int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
2143
//
L
Liu Jicong 已提交
2144 2145 2146 2147
//   //qDebug("QInfo:0x%"PRIx64" check data block, brange:%" PRId64 "-%" PRId64 ", numBlocksOfStep:%d, numOfRes:%d,
//   lastKey:%"PRId64, GET_TASKID(pRuntimeEnv),
//          pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows, numOfRes, pQuery->current->lastKey);
// }
2148

L
Liu Jicong 已提交
2149 2150
// void skipBlocks(STaskRuntimeEnv *pRuntimeEnv) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
2151
//
L
Liu Jicong 已提交
2152 2153 2154
//   if (pQueryAttr->limit.offset <= 0 || pQueryAttr->numOfFilterCols > 0) {
//     return;
//   }
2155
//
L
Liu Jicong 已提交
2156 2157
//   pQueryAttr->pos = 0;
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
2158
//
L
Liu Jicong 已提交
2159 2160
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
//   TsdbQueryHandleT pTsdbReadHandle = pRuntimeEnv->pTsdbReadHandle;
2161
//
L
Liu Jicong 已提交
2162 2163 2164 2165 2166
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pTsdbReadHandle)) {
//     if (isTaskKilled(pRuntimeEnv->qinfo)) {
//       longjmp(pRuntimeEnv->env, TSDB_CODE_TSC_QUERY_CANCELLED);
//     }
2167
//
L
Liu Jicong 已提交
2168
//     tsdbRetrieveDataBlockInfo(pTsdbReadHandle, &blockInfo);
2169
//
L
Liu Jicong 已提交
2170 2171 2172 2173
//     if (pQueryAttr->limit.offset > blockInfo.rows) {
//       pQueryAttr->limit.offset -= blockInfo.rows;
//       pTableQueryInfo->lastKey = (QUERY_IS_ASC_QUERY(pQueryAttr)) ? blockInfo.window.ekey : blockInfo.window.skey;
//       pTableQueryInfo->lastKey += step;
2174
//
L
Liu Jicong 已提交
2175 2176 2177 2178 2179 2180 2181
//       //qDebug("QInfo:0x%"PRIx64" skip rows:%d, offset:%" PRId64, GET_TASKID(pRuntimeEnv), blockInfo.rows,
//              pQuery->limit.offset);
//     } else {  // find the appropriated start position in current block
//       updateOffsetVal(pRuntimeEnv, &blockInfo);
//       break;
//     }
//   }
2182
//
L
Liu Jicong 已提交
2183 2184 2185 2186 2187 2188 2189 2190 2191
//   if (terrno != TSDB_CODE_SUCCESS) {
//     longjmp(pRuntimeEnv->env, terrno);
//   }
// }

// static TSKEY doSkipIntervalProcess(STaskRuntimeEnv* pRuntimeEnv, STimeWindow* win, SDataBlockInfo* pBlockInfo,
// STableQueryInfo* pTableQueryInfo) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
//   SResultRowInfo *pWindowResInfo = &pRuntimeEnv->resultRowInfo;
2192
//
L
Liu Jicong 已提交
2193 2194 2195
//   assert(pQueryAttr->limit.offset == 0);
//   STimeWindow tw = *win;
//   getNextTimeWindow(pQueryAttr, &tw);
2196
//
L
Liu Jicong 已提交
2197 2198
//   if ((tw.skey <= pBlockInfo->window.ekey && QUERY_IS_ASC_QUERY(pQueryAttr)) ||
//       (tw.ekey >= pBlockInfo->window.skey && !QUERY_IS_ASC_QUERY(pQueryAttr))) {
2199
//
L
Liu Jicong 已提交
2200 2201 2202 2203
//     // load the data block and check data remaining in current data block
//     // TODO optimize performance
//     SArray *         pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//     SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
2204
//
L
Liu Jicong 已提交
2205 2206 2207 2208
//     tw = *win;
//     int32_t startPos =
//         getNextQualifiedWindow(pQueryAttr, &tw, pBlockInfo, pColInfoData->pData, binarySearchForKey, -1);
//     assert(startPos >= 0);
2209
//
L
Liu Jicong 已提交
2210 2211
//     // set the abort info
//     pQueryAttr->pos = startPos;
2212
//
L
Liu Jicong 已提交
2213 2214 2215 2216
//     // reset the query start timestamp
//     pTableQueryInfo->win.skey = ((TSKEY *)pColInfoData->pData)[startPos];
//     pQueryAttr->window.skey = pTableQueryInfo->win.skey;
//     TSKEY key = pTableQueryInfo->win.skey;
2217
//
L
Liu Jicong 已提交
2218 2219
//     pWindowResInfo->prevSKey = tw.skey;
//     int32_t index = pRuntimeEnv->resultRowInfo.curIndex;
2220
//
L
Liu Jicong 已提交
2221 2222
//     int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
//     pRuntimeEnv->resultRowInfo.curIndex = index;  // restore the window index
2223
//
L
Liu Jicong 已提交
2224 2225 2226 2227
//     //qDebug("QInfo:0x%"PRIx64" check data block, brange:%" PRId64 "-%" PRId64 ", numOfRows:%d, numOfRes:%d,
//     lastKey:%" PRId64,
//            GET_TASKID(pRuntimeEnv), pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows, numOfRes,
//            pQueryAttr->current->lastKey);
2228
//
L
Liu Jicong 已提交
2229 2230 2231 2232 2233
//     return key;
//   } else {  // do nothing
//     pQueryAttr->window.skey      = tw.skey;
//     pWindowResInfo->prevSKey = tw.skey;
//     pTableQueryInfo->lastKey = tw.skey;
2234
//
L
Liu Jicong 已提交
2235 2236
//     return tw.skey;
//   }
2237
//
L
Liu Jicong 已提交
2238 2239 2240 2241 2242 2243 2244 2245 2246 2247
//   return true;
// }

// static bool skipTimeInterval(STaskRuntimeEnv *pRuntimeEnv, TSKEY* start) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
//   if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//     assert(*start <= pRuntimeEnv->current->lastKey);
//   } else {
//     assert(*start >= pRuntimeEnv->current->lastKey);
//   }
2248
//
L
Liu Jicong 已提交
2249 2250 2251 2252 2253
//   // if queried with value filter, do NOT forward query start position
//   if (pQueryAttr->limit.offset <= 0 || pQueryAttr->numOfFilterCols > 0 || pRuntimeEnv->pTsBuf != NULL ||
//   pRuntimeEnv->pFillInfo != NULL) {
//     return true;
//   }
2254
//
L
Liu Jicong 已提交
2255 2256 2257 2258 2259 2260 2261
//   /*
//    * 1. for interval without interpolation query we forward pQueryAttr->interval.interval at a time for
//    *    pQueryAttr->limit.offset times. Since hole exists, pQueryAttr->interval.interval*pQueryAttr->limit.offset
//    value is
//    *    not valid. otherwise, we only forward pQueryAttr->limit.offset number of points
//    */
//   assert(pRuntimeEnv->resultRowInfo.prevSKey == TSKEY_INITIAL_VAL);
2262
//
L
Liu Jicong 已提交
2263 2264
//   STimeWindow w = TSWINDOW_INITIALIZER;
//   bool ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
2265
//
L
Liu Jicong 已提交
2266 2267
//   SResultRowInfo *pWindowResInfo = &pRuntimeEnv->resultRowInfo;
//   STableQueryInfo *pTableQueryInfo = pRuntimeEnv->current;
2268
//
L
Liu Jicong 已提交
2269 2270 2271
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pRuntimeEnv->pTsdbReadHandle)) {
//     tsdbRetrieveDataBlockInfo(pRuntimeEnv->pTsdbReadHandle, &blockInfo);
2272
//
L
Liu Jicong 已提交
2273 2274 2275 2276 2277 2278 2279 2280 2281
//     if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//       if (pWindowResInfo->prevSKey == TSKEY_INITIAL_VAL) {
//         getAlignQueryTimeWindow(pQueryAttr, blockInfo.window.skey, blockInfo.window.skey, pQueryAttr->window.ekey,
//         &w); pWindowResInfo->prevSKey = w.skey;
//       }
//     } else {
//       getAlignQueryTimeWindow(pQueryAttr, blockInfo.window.ekey, pQueryAttr->window.ekey, blockInfo.window.ekey, &w);
//       pWindowResInfo->prevSKey = w.skey;
//     }
2282
//
L
Liu Jicong 已提交
2283 2284
//     // the first time window
//     STimeWindow win = getActiveTimeWindow(pWindowResInfo, pWindowResInfo->prevSKey, pQueryAttr);
2285
//
L
Liu Jicong 已提交
2286 2287
//     while (pQueryAttr->limit.offset > 0) {
//       STimeWindow tw = win;
2288
//
L
Liu Jicong 已提交
2289 2290 2291
//       if ((win.ekey <= blockInfo.window.ekey && ascQuery) || (win.ekey >= blockInfo.window.skey && !ascQuery)) {
//         pQueryAttr->limit.offset -= 1;
//         pWindowResInfo->prevSKey = win.skey;
2292
//
L
Liu Jicong 已提交
2293 2294 2295 2296 2297 2298
//         // current time window is aligned with blockInfo.window.ekey
//         // restart it from next data block by set prevSKey to be TSKEY_INITIAL_VAL;
//         if ((win.ekey == blockInfo.window.ekey && ascQuery) || (win.ekey == blockInfo.window.skey && !ascQuery)) {
//           pWindowResInfo->prevSKey = TSKEY_INITIAL_VAL;
//         }
//       }
2299
//
L
Liu Jicong 已提交
2300 2301 2302 2303
//       if (pQueryAttr->limit.offset == 0) {
//         *start = doSkipIntervalProcess(pRuntimeEnv, &win, &blockInfo, pTableQueryInfo);
//         return true;
//       }
2304
//
L
Liu Jicong 已提交
2305 2306
//       // current window does not ended in current data block, try next data block
//       getNextTimeWindow(pQueryAttr, &tw);
2307
//
L
Liu Jicong 已提交
2308 2309 2310 2311 2312 2313 2314 2315 2316
//       /*
//        * If the next time window still starts from current data block,
//        * load the primary timestamp column first, and then find the start position for the next queried time window.
//        * Note that only the primary timestamp column is required.
//        * TODO: Optimize for this cases. All data blocks are not needed to be loaded, only if the first actually
//        required
//        * time window resides in current data block.
//        */
//       if ((tw.skey <= blockInfo.window.ekey && ascQuery) || (tw.ekey >= blockInfo.window.skey && !ascQuery)) {
2317
//
L
Liu Jicong 已提交
2318 2319
//         SArray *pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//         SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
2320
//
L
Liu Jicong 已提交
2321 2322 2323
//         if ((win.ekey > blockInfo.window.ekey && ascQuery) || (win.ekey < blockInfo.window.skey && !ascQuery)) {
//           pQueryAttr->limit.offset -= 1;
//         }
2324
//
L
Liu Jicong 已提交
2325 2326 2327 2328 2329 2330 2331 2332
//         if (pQueryAttr->limit.offset == 0) {
//           *start = doSkipIntervalProcess(pRuntimeEnv, &win, &blockInfo, pTableQueryInfo);
//           return true;
//         } else {
//           tw = win;
//           int32_t startPos =
//               getNextQualifiedWindow(pQueryAttr, &tw, &blockInfo, pColInfoData->pData, binarySearchForKey, -1);
//           assert(startPos >= 0);
2333
//
L
Liu Jicong 已提交
2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344
//           // set the abort info
//           pQueryAttr->pos = startPos;
//           pTableQueryInfo->lastKey = ((TSKEY *)pColInfoData->pData)[startPos];
//           pWindowResInfo->prevSKey = tw.skey;
//           win = tw;
//         }
//       } else {
//         break;  // offset is not 0, and next time window begins or ends in the next block.
//       }
//     }
//   }
2345
//
L
Liu Jicong 已提交
2346 2347 2348 2349
//   // check for error
//   if (terrno != TSDB_CODE_SUCCESS) {
//     longjmp(pRuntimeEnv->env, terrno);
//   }
2350
//
L
Liu Jicong 已提交
2351 2352
//   return true;
// }
2353

2354
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num) {
H
Haojun Liao 已提交
2355
  if (p->pDownstream == NULL) {
H
Haojun Liao 已提交
2356
    assert(p->numOfDownstream == 0);
2357 2358
  }

wafwerar's avatar
wafwerar 已提交
2359
  p->pDownstream = taosMemoryCalloc(1, num * POINTER_BYTES);
2360 2361 2362 2363 2364 2365 2366
  if (p->pDownstream == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  memcpy(p->pDownstream, pDownstream, num * POINTER_BYTES);
  p->numOfDownstream = num;
  return TSDB_CODE_SUCCESS;
2367 2368
}

wmmhello's avatar
wmmhello 已提交
2369
static void doDestroyTableList(STableListInfo* pTableqinfoList);
2370

2371
static void doTableQueryInfoTimeWindowCheck(SExecTaskInfo* pTaskInfo, STableQueryInfo* pTableQueryInfo, int32_t order) {
H
Haojun Liao 已提交
2372 2373
#if 0
    if (order == TSDB_ORDER_ASC) {
2374 2375
    assert(
        (pTableQueryInfo->win.skey <= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
2376 2377
        (pTableQueryInfo->lastKey >= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey >= pTaskInfo->window.skey && pTableQueryInfo->win.ekey <= pTaskInfo->window.ekey));
2378 2379 2380
  } else {
    assert(
        (pTableQueryInfo->win.skey >= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
2381 2382
        (pTableQueryInfo->lastKey <= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey <= pTaskInfo->window.skey && pTableQueryInfo->win.ekey >= pTaskInfo->window.ekey));
2383
  }
H
Haojun Liao 已提交
2384
#endif
2385 2386
}

L
Liu Jicong 已提交
2387 2388 2389 2390
// static void updateTableIdInfo(STableQueryInfo* pTableQueryInfo, SSDataBlock* pBlock, SHashObj* pTableIdInfo, int32_t
// order) {
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(order);
//   pTableQueryInfo->lastKey = ((order == TSDB_ORDER_ASC)? pBlock->info.window.ekey:pBlock->info.window.skey) + step;
2391
//
L
Liu Jicong 已提交
2392 2393 2394
//   if (pTableQueryInfo->pTable == NULL) {
//     return;
//   }
2395
//
L
Liu Jicong 已提交
2396 2397 2398 2399 2400 2401 2402 2403 2404
//   STableIdInfo tidInfo = createTableIdInfo(pTableQueryInfo);
//   STableIdInfo *idinfo = taosHashGet(pTableIdInfo, &tidInfo.tid, sizeof(tidInfo.tid));
//   if (idinfo != NULL) {
//     assert(idinfo->tid == tidInfo.tid && idinfo->uid == tidInfo.uid);
//     idinfo->key = tidInfo.key;
//   } else {
//     taosHashPut(pTableIdInfo, &tidInfo.tid, sizeof(tidInfo.tid), &tidInfo, sizeof(STableIdInfo));
//   }
// }
2405

2406
int32_t loadRemoteDataCallback(void* param, const SDataBuf* pMsg, int32_t code) {
L
Liu Jicong 已提交
2407
  SSourceDataInfo* pSourceDataInfo = (SSourceDataInfo*)param;
H
Haojun Liao 已提交
2408 2409
  if (code == TSDB_CODE_SUCCESS) {
    pSourceDataInfo->pRsp = pMsg->pData;
2410

H
Haojun Liao 已提交
2411 2412
    SRetrieveTableRsp* pRsp = pSourceDataInfo->pRsp;
    pRsp->numOfRows = htonl(pRsp->numOfRows);
dengyihao's avatar
dengyihao 已提交
2413
    pRsp->compLen = htonl(pRsp->compLen);
2414
    pRsp->numOfCols = htonl(pRsp->numOfCols);
dengyihao's avatar
dengyihao 已提交
2415
    pRsp->useconds = htobe64(pRsp->useconds);
2416 2417 2418

    ASSERT(pSourceDataInfo->pRsp != NULL);
    qDebug("fetch rsp received, index:%d, rows:%d", pSourceDataInfo->index, pRsp->numOfRows);
H
Haojun Liao 已提交
2419 2420 2421
  } else {
    pSourceDataInfo->code = code;
  }
H
Haojun Liao 已提交
2422

H
Haojun Liao 已提交
2423
  pSourceDataInfo->status = EX_SOURCE_DATA_READY;
2424
  tsem_post(&pSourceDataInfo->pEx->ready);
wmmhello's avatar
wmmhello 已提交
2425
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
2426 2427 2428 2429
}

static void destroySendMsgInfo(SMsgSendInfo* pMsgBody) {
  assert(pMsgBody != NULL);
wafwerar's avatar
wafwerar 已提交
2430 2431
  taosMemoryFreeClear(pMsgBody->msgInfo.pData);
  taosMemoryFreeClear(pMsgBody);
H
Haojun Liao 已提交
2432 2433
}

S
Shengliang Guan 已提交
2434
void qProcessFetchRsp(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet) {
S
Shengliang Guan 已提交
2435 2436
  SMsgSendInfo* pSendInfo = (SMsgSendInfo*)pMsg->info.ahandle;
  assert(pMsg->info.ahandle != NULL);
H
Haojun Liao 已提交
2437 2438 2439 2440

  SDataBuf buf = {.len = pMsg->contLen, .pData = NULL};

  if (pMsg->contLen > 0) {
wafwerar's avatar
wafwerar 已提交
2441
    buf.pData = taosMemoryCalloc(1, pMsg->contLen);
H
Haojun Liao 已提交
2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452
    if (buf.pData == NULL) {
      terrno = TSDB_CODE_OUT_OF_MEMORY;
      pMsg->code = TSDB_CODE_OUT_OF_MEMORY;
    } else {
      memcpy(buf.pData, pMsg->pCont, pMsg->contLen);
    }
  }

  pSendInfo->fp(pSendInfo->param, &buf, pMsg->code);
  rpcFreeCont(pMsg->pCont);
  destroySendMsgInfo(pSendInfo);
2453 2454
}

L
Liu Jicong 已提交
2455
static int32_t doSendFetchDataRequest(SExchangeInfo* pExchangeInfo, SExecTaskInfo* pTaskInfo, int32_t sourceIndex) {
2456
  size_t totalSources = taosArrayGetSize(pExchangeInfo->pSources);
2457

wafwerar's avatar
wafwerar 已提交
2458
  SResFetchReq* pMsg = taosMemoryCalloc(1, sizeof(SResFetchReq));
2459 2460 2461 2462
  if (NULL == pMsg) {
    pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return pTaskInfo->code;
  }
2463

L
Liu Jicong 已提交
2464 2465
  SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, sourceIndex);
  SSourceDataInfo*       pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, sourceIndex);
2466

2467 2468
  ASSERT(pDataInfo->status == EX_SOURCE_DATA_NOT_READY);

L
Liu Jicong 已提交
2469 2470
  qDebug("%s build fetch msg and send to vgId:%d, ep:%s, taskId:0x%" PRIx64 ", %d/%" PRIzu, GET_TASKID(pTaskInfo),
         pSource->addr.nodeId, pSource->addr.epSet.eps[0].fqdn, pSource->taskId, sourceIndex, totalSources);
2471 2472 2473 2474 2475 2476 2477

  pMsg->header.vgId = htonl(pSource->addr.nodeId);
  pMsg->sId = htobe64(pSource->schedId);
  pMsg->taskId = htobe64(pSource->taskId);
  pMsg->queryId = htobe64(pTaskInfo->id.queryId);

  // send the fetch remote task result reques
wafwerar's avatar
wafwerar 已提交
2478
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
2479
  if (NULL == pMsgSendInfo) {
wafwerar's avatar
wafwerar 已提交
2480
    taosMemoryFreeClear(pMsg);
2481 2482 2483
    qError("%s prepare message %d failed", GET_TASKID(pTaskInfo), (int32_t)sizeof(SMsgSendInfo));
    pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return pTaskInfo->code;
H
Haojun Liao 已提交
2484 2485
  }

2486 2487 2488 2489 2490
  pMsgSendInfo->param = pDataInfo;
  pMsgSendInfo->msgInfo.pData = pMsg;
  pMsgSendInfo->msgInfo.len = sizeof(SResFetchReq);
  pMsgSendInfo->msgType = TDMT_VND_FETCH;
  pMsgSendInfo->fp = loadRemoteDataCallback;
2491

2492
  int64_t transporterId = 0;
L
Liu Jicong 已提交
2493
  int32_t code = asyncSendMsgToServer(pExchangeInfo->pTransporter, &pSource->addr.epSet, &transporterId, pMsgSendInfo);
2494 2495 2496
  return TSDB_CODE_SUCCESS;
}

2497
// NOTE: sources columns are more than the destination SSDatablock columns.
2498
void relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols) {
2499 2500 2501
  size_t numOfSrcCols = taosArrayGetSize(pCols);

  int32_t i = 0, j = 0;
L
Liu Jicong 已提交
2502
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521
    SColumnInfoData* p = taosArrayGet(pCols, i);
    SColMatchInfo*   pmInfo = taosArrayGet(pColMatchInfo, j);
    if (!pmInfo->output) {
      j++;
      continue;
    }

    if (p->info.colId == pmInfo->colId) {
      taosArraySet(pBlock->pDataBlock, pmInfo->targetSlotId, p);
      i++;
      j++;
    } else if (p->info.colId < pmInfo->colId) {
      i++;
    } else {
      ASSERT(0);
    }
  }
}

2522
int32_t setDataBlockFromFetchRsp(SSDataBlock* pRes, SLoadRemoteDataInfo* pLoadInfo, int32_t numOfRows, char* pData,
X
Xiaoyu Wang 已提交
2523 2524
                                 int32_t compLen, int32_t numOfOutput, int64_t startTs, uint64_t* total,
                                 SArray* pColList) {
H
Haojun Liao 已提交
2525
  if (pColList == NULL) {  // data from other sources
2526
    blockCompressDecode(pRes, numOfOutput, numOfRows, pData);
H
Haojun Liao 已提交
2527
  } else {  // extract data according to pColList
2528
    ASSERT(numOfOutput == taosArrayGetSize(pColList));
2529 2530 2531 2532 2533
    char* pStart = pData;

    int32_t numOfCols = htonl(*(int32_t*)pStart);
    pStart += sizeof(int32_t);

2534
    // todo refactor:extract method
2535
    SSysTableSchema* pSchema = (SSysTableSchema*)pStart;
dengyihao's avatar
dengyihao 已提交
2536
    for (int32_t i = 0; i < numOfCols; ++i) {
2537 2538 2539 2540 2541 2542 2543
      SSysTableSchema* p = (SSysTableSchema*)pStart;

      p->colId = htons(p->colId);
      p->bytes = htonl(p->bytes);
      pStart += sizeof(SSysTableSchema);
    }

H
Haojun Liao 已提交
2544 2545 2546 2547
    SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
    pBlock->pDataBlock = taosArrayInit(numOfCols, sizeof(SColumnInfoData));
    pBlock->info.numOfCols = numOfCols;

dengyihao's avatar
dengyihao 已提交
2548
    for (int32_t i = 0; i < numOfCols; ++i) {
2549
      SColumnInfoData idata = {0};
2550

X
Xiaoyu Wang 已提交
2551
      idata.info.type = pSchema[i].type;
2552 2553
      idata.info.bytes = pSchema[i].bytes;
      idata.info.colId = pSchema[i].colId;
X
Xiaoyu Wang 已提交
2554
      idata.hasNull = true;
2555

H
Haojun Liao 已提交
2556
      taosArrayPush(pBlock->pDataBlock, &idata);
2557
      if (IS_VAR_DATA_TYPE(idata.info.type)) {
H
Haojun Liao 已提交
2558
        pBlock->info.hasVarCol = true;
2559 2560 2561
      }
    }

H
Haojun Liao 已提交
2562
    blockDataEnsureCapacity(pBlock, numOfRows);
2563

L
Liu Jicong 已提交
2564 2565
    int32_t  dataLen = *(int32_t*)pStart;
    uint64_t groupId = *(uint64_t*)(pStart + sizeof(int32_t));
2566 2567
    pStart += sizeof(int32_t) + sizeof(uint64_t);

L
Liu Jicong 已提交
2568
    int32_t* colLen = (int32_t*)(pStart);
2569 2570 2571 2572
    pStart += sizeof(int32_t) * numOfCols;

    for (int32_t i = 0; i < numOfCols; ++i) {
      colLen[i] = htonl(colLen[i]);
2573
      ASSERT(colLen[i] >= 0);
2574

H
Haojun Liao 已提交
2575
      SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591
      if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
        pColInfoData->varmeta.length = colLen[i];
        pColInfoData->varmeta.allocLen = colLen[i];

        memcpy(pColInfoData->varmeta.offset, pStart, sizeof(int32_t) * numOfRows);
        pStart += sizeof(int32_t) * numOfRows;

        pColInfoData->pData = taosMemoryMalloc(colLen[i]);
      } else {
        memcpy(pColInfoData->nullbitmap, pStart, BitmapLen(numOfRows));
        pStart += BitmapLen(numOfRows);
      }

      memcpy(pColInfoData->pData, pStart, colLen[i]);
      pStart += colLen[i];
    }
H
Haojun Liao 已提交
2592 2593

    // data from mnode
H
Haojun Liao 已提交
2594 2595 2596
    relocateColumnData(pRes, pColList, pBlock->pDataBlock);
    taosArrayDestroy(pBlock->pDataBlock);
    taosMemoryFree(pBlock);
X
Xiaoyu Wang 已提交
2597
    //    blockDataDestroy(pBlock);
2598
  }
2599

H
Haojun Liao 已提交
2600
  pRes->info.rows = numOfRows;
2601 2602 2603

  // todo move this to time window aggregator, since the primary timestamp may not be known by exchange operator.
  blockDataUpdateTsWindow(pRes, 0);
2604

2605
  int64_t el = taosGetTimestampUs() - startTs;
2606

H
Haojun Liao 已提交
2607 2608
  pLoadInfo->totalRows += numOfRows;
  pLoadInfo->totalSize += compLen;
2609

H
Haojun Liao 已提交
2610 2611 2612
  if (total != NULL) {
    *total += numOfRows;
  }
2613

H
Haojun Liao 已提交
2614
  pLoadInfo->totalElapsed += el;
2615 2616
  return TSDB_CODE_SUCCESS;
}
2617

L
Liu Jicong 已提交
2618 2619
static void* setAllSourcesCompleted(SOperatorInfo* pOperator, int64_t startTs) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
2620
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
2621

2622
  int64_t              el = taosGetTimestampUs() - startTs;
H
Haojun Liao 已提交
2623
  SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
2624

H
Haojun Liao 已提交
2625
  pLoadInfo->totalElapsed += el;
H
Haojun Liao 已提交
2626

2627
  size_t totalSources = taosArrayGetSize(pExchangeInfo->pSources);
L
Liu Jicong 已提交
2628 2629 2630
  qDebug("%s all %" PRIzu " sources are exhausted, total rows: %" PRIu64 " bytes:%" PRIu64 ", elapsed:%.2f ms",
         GET_TASKID(pTaskInfo), totalSources, pLoadInfo->totalRows, pLoadInfo->totalSize,
         pLoadInfo->totalElapsed / 1000.0);
2631 2632 2633 2634 2635

  doSetOperatorCompleted(pOperator);
  return NULL;
}

L
Liu Jicong 已提交
2636 2637
static SSDataBlock* concurrentlyLoadRemoteDataImpl(SOperatorInfo* pOperator, SExchangeInfo* pExchangeInfo,
                                                   SExecTaskInfo* pTaskInfo) {
2638 2639 2640 2641 2642 2643 2644 2645
  int32_t code = 0;
  int64_t startTs = taosGetTimestampUs();
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);

  while (1) {
    int32_t completed = 0;
    for (int32_t i = 0; i < totalSources; ++i) {
      SSourceDataInfo* pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, i);
2646
      if (pDataInfo->status == EX_SOURCE_DATA_EXHAUSTED) {
2647
        completed += 1;
H
Haojun Liao 已提交
2648 2649
        continue;
      }
2650

2651
      if (pDataInfo->status != EX_SOURCE_DATA_READY) {
2652 2653 2654
        continue;
      }

2655 2656 2657 2658 2659
      if (pDataInfo->code != TSDB_CODE_SUCCESS) {
        code = pDataInfo->code;
        goto _error;
      }

L
Liu Jicong 已提交
2660
      SRetrieveTableRsp*     pRsp = pDataInfo->pRsp;
X
Xiaoyu Wang 已提交
2661
      SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, i);
2662

L
Liu Jicong 已提交
2663
      SSDataBlock*         pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
2664
      SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
2665
      if (pRsp->numOfRows == 0) {
L
Liu Jicong 已提交
2666 2667
        qDebug("%s vgId:%d, taskID:0x%" PRIx64 " index:%d completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64
               " try next",
2668
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, i + 1, pDataInfo->totalRows,
H
Haojun Liao 已提交
2669
               pExchangeInfo->loadInfo.totalRows);
2670
        pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
2671
        completed += 1;
D
dapan1121 已提交
2672
        taosMemoryFreeClear(pDataInfo->pRsp);
2673 2674
        continue;
      }
H
Haojun Liao 已提交
2675

H
Haojun Liao 已提交
2676
      SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
2677
      code = setDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
X
Xiaoyu Wang 已提交
2678
                                      pTableRsp->compLen, pTableRsp->numOfCols, startTs, &pDataInfo->totalRows, NULL);
2679
      if (code != 0) {
2680
        taosMemoryFreeClear(pDataInfo->pRsp);
2681 2682 2683
        goto _error;
      }

2684 2685 2686
      if (pRsp->completed == 1) {
        qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, rowsOfSource:%" PRIu64
               ", totalRows:%" PRIu64 ", totalBytes:%" PRIu64 " try next %d/%" PRIzu,
L
Liu Jicong 已提交
2687 2688
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pDataInfo->totalRows,
               pLoadInfo->totalRows, pLoadInfo->totalSize, i + 1, totalSources);
2689
        pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
2690
      } else {
dengyihao's avatar
dengyihao 已提交
2691 2692
        qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, totalRows:%" PRIu64
               ", totalBytes:%" PRIu64,
H
Haojun Liao 已提交
2693 2694
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pLoadInfo->totalRows,
               pLoadInfo->totalSize);
2695 2696
      }

2697 2698
      taosMemoryFreeClear(pDataInfo->pRsp);

2699 2700
      if (pDataInfo->status != EX_SOURCE_DATA_EXHAUSTED) {
        pDataInfo->status = EX_SOURCE_DATA_NOT_READY;
2701 2702
        code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
        if (code != TSDB_CODE_SUCCESS) {
2703
          taosMemoryFreeClear(pDataInfo->pRsp);
2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720
          goto _error;
        }
      }

      return pExchangeInfo->pResult;
    }

    if (completed == totalSources) {
      return setAllSourcesCompleted(pOperator, startTs);
    }
  }

_error:
  pTaskInfo->code = code;
  return NULL;
}

L
Liu Jicong 已提交
2721 2722 2723
static int32_t prepareConcurrentlyLoad(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
2724

L
Liu Jicong 已提交
2725
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
2726 2727 2728
  int64_t startTs = taosGetTimestampUs();

  // Asynchronously send all fetch requests to all sources.
L
Liu Jicong 已提交
2729
  for (int32_t i = 0; i < totalSources; ++i) {
2730 2731
    int32_t code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
2732 2733
      pTaskInfo->code = code;
      return code;
2734 2735 2736 2737
    }
  }

  int64_t endTs = taosGetTimestampUs();
2738
  qDebug("%s send all fetch requests to %" PRIzu " sources completed, elapsed:%" PRId64, GET_TASKID(pTaskInfo),
L
Liu Jicong 已提交
2739
         totalSources, endTs - startTs);
2740 2741

  tsem_wait(&pExchangeInfo->ready);
2742
  pOperator->status = OP_RES_TO_RETURN;
H
Haojun Liao 已提交
2743
  pOperator->cost.openCost = taosGetTimestampUs() - startTs;
2744

H
Haojun Liao 已提交
2745
  return TSDB_CODE_SUCCESS;
2746 2747
}

L
Liu Jicong 已提交
2748 2749 2750
static SSDataBlock* seqLoadRemoteData(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
2751

L
Liu Jicong 已提交
2752
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
2753
  int64_t startTs = taosGetTimestampUs();
2754

L
Liu Jicong 已提交
2755
  while (1) {
2756 2757
    if (pExchangeInfo->current >= totalSources) {
      return setAllSourcesCompleted(pOperator, startTs);
2758
    }
2759

2760 2761 2762
    doSendFetchDataRequest(pExchangeInfo, pTaskInfo, pExchangeInfo->current);
    tsem_wait(&pExchangeInfo->ready);

dengyihao's avatar
dengyihao 已提交
2763
    SSourceDataInfo*       pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, pExchangeInfo->current);
X
Xiaoyu Wang 已提交
2764
    SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, pExchangeInfo->current);
2765

H
Haojun Liao 已提交
2766
    if (pDataInfo->code != TSDB_CODE_SUCCESS) {
dengyihao's avatar
dengyihao 已提交
2767 2768
      qError("%s vgId:%d, taskID:0x%" PRIx64 " error happens, code:%s", GET_TASKID(pTaskInfo), pSource->addr.nodeId,
             pSource->taskId, tstrerror(pDataInfo->code));
H
Haojun Liao 已提交
2769 2770 2771 2772
      pOperator->pTaskInfo->code = pDataInfo->code;
      return NULL;
    }

L
Liu Jicong 已提交
2773
    SRetrieveTableRsp*   pRsp = pDataInfo->pRsp;
H
Haojun Liao 已提交
2774
    SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
2775
    if (pRsp->numOfRows == 0) {
dengyihao's avatar
dengyihao 已提交
2776 2777
      qDebug("%s vgId:%d, taskID:0x%" PRIx64 " %d of total completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64
             " try next",
2778
             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pExchangeInfo->current + 1,
H
Haojun Liao 已提交
2779
             pDataInfo->totalRows, pLoadInfo->totalRows);
H
Haojun Liao 已提交
2780

2781
      pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
2782
      pExchangeInfo->current += 1;
D
dapan1121 已提交
2783
      taosMemoryFreeClear(pDataInfo->pRsp);
2784 2785
      continue;
    }
H
Haojun Liao 已提交
2786

L
Liu Jicong 已提交
2787
    SSDataBlock*       pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
2788
    SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
L
Liu Jicong 已提交
2789
    int32_t            code =
2790
        setDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
X
Xiaoyu Wang 已提交
2791
                                 pTableRsp->compLen, pTableRsp->numOfCols, startTs, &pDataInfo->totalRows, NULL);
2792 2793

    if (pRsp->completed == 1) {
H
Haojun Liao 已提交
2794
      qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, rowsOfSource:%" PRIu64
L
Liu Jicong 已提交
2795 2796 2797
             ", totalRows:%" PRIu64 ", totalBytes:%" PRIu64 " try next %d/%" PRIzu,
             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pDataInfo->totalRows,
             pLoadInfo->totalRows, pLoadInfo->totalSize, pExchangeInfo->current + 1, totalSources);
2798

2799
      pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
2800 2801
      pExchangeInfo->current += 1;
    } else {
L
Liu Jicong 已提交
2802 2803 2804 2805
      qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, totalRows:%" PRIu64
             ", totalBytes:%" PRIu64,
             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pLoadInfo->totalRows,
             pLoadInfo->totalSize);
2806 2807
    }

2808
    pOperator->resultInfo.totalRows += pRes->info.rows;
2809
    taosMemoryFreeClear(pDataInfo->pRsp);
2810 2811
    return pExchangeInfo->pResult;
  }
2812 2813
}

L
Liu Jicong 已提交
2814
static int32_t prepareLoadRemoteData(SOperatorInfo* pOperator) {
2815
  if (OPTR_IS_OPENED(pOperator)) {
H
Haojun Liao 已提交
2816 2817 2818
    return TSDB_CODE_SUCCESS;
  }

2819 2820
  int64_t st = taosGetTimestampUs();

L
Liu Jicong 已提交
2821
  SExchangeInfo* pExchangeInfo = pOperator->info;
2822
  if (!pExchangeInfo->seqLoadData) {
H
Haojun Liao 已提交
2823 2824 2825 2826 2827 2828
    int32_t code = prepareConcurrentlyLoad(pOperator);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }
  }

2829
  OPTR_SET_OPENED(pOperator);
2830
  pOperator->cost.openCost = (taosGetTimestampUs() - st) / 1000.0;
H
Haojun Liao 已提交
2831 2832 2833
  return TSDB_CODE_SUCCESS;
}

2834
static SSDataBlock* doLoadRemoteData(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
2835 2836
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
2837

2838
  pTaskInfo->code = pOperator->fpSet._openFn(pOperator);
2839
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
2840 2841
    return NULL;
  }
2842

L
Liu Jicong 已提交
2843
  size_t               totalSources = taosArrayGetSize(pExchangeInfo->pSources);
H
Haojun Liao 已提交
2844
  SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
H
Haojun Liao 已提交
2845

2846
  if (pOperator->status == OP_EXEC_DONE) {
L
Liu Jicong 已提交
2847 2848 2849
    qDebug("%s all %" PRIzu " source(s) are exhausted, total rows:%" PRIu64 " bytes:%" PRIu64 ", elapsed:%.2f ms",
           GET_TASKID(pTaskInfo), totalSources, pLoadInfo->totalRows, pLoadInfo->totalSize,
           pLoadInfo->totalElapsed / 1000.0);
2850 2851 2852 2853 2854 2855
    return NULL;
  }

  if (pExchangeInfo->seqLoadData) {
    return seqLoadRemoteData(pOperator);
  } else {
2856
    return concurrentlyLoadRemoteDataImpl(pOperator, pExchangeInfo, pTaskInfo);
X
Xiaoyu Wang 已提交
2857
    //    return concurrentlyLoadRemoteData(pOperator);
2858
  }
H
Haojun Liao 已提交
2859
}
2860

H
Haojun Liao 已提交
2861
static int32_t initDataSource(int32_t numOfSources, SExchangeInfo* pInfo) {
2862
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
H
Haojun Liao 已提交
2863 2864
  if (pInfo->pSourceDataInfo == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
2865 2866
  }

L
Liu Jicong 已提交
2867
  for (int32_t i = 0; i < numOfSources; ++i) {
2868
    SSourceDataInfo dataInfo = {0};
H
Haojun Liao 已提交
2869
    dataInfo.status = EX_SOURCE_DATA_NOT_READY;
L
Liu Jicong 已提交
2870 2871
    dataInfo.pEx = pInfo;
    dataInfo.index = i;
2872

H
Haojun Liao 已提交
2873 2874 2875 2876 2877 2878 2879 2880 2881 2882
    void* ret = taosArrayPush(pInfo->pSourceDataInfo, &dataInfo);
    if (ret == NULL) {
      taosArrayDestroy(pInfo->pSourceDataInfo);
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  return TSDB_CODE_SUCCESS;
}

2883 2884
static int32_t initExchangeOperator(SExchangePhysiNode* pExNode, SExchangeInfo* pInfo) {
  size_t numOfSources = LIST_LENGTH(pExNode->pSrcEndPoints);
H
Haojun Liao 已提交
2885

H
Haojun Liao 已提交
2886
  pInfo->pSources = taosArrayInit(numOfSources, sizeof(SDownstreamSourceNode));
H
Haojun Liao 已提交
2887 2888
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
  if (pInfo->pSourceDataInfo == NULL || pInfo->pSources == NULL) {
2889
    return TSDB_CODE_OUT_OF_MEMORY;
H
Haojun Liao 已提交
2890 2891
  }

L
Liu Jicong 已提交
2892
  for (int32_t i = 0; i < numOfSources; ++i) {
2893
    SNodeListNode* pNode = nodesListGetNode((SNodeList*)pExNode->pSrcEndPoints, i);
H
Haojun Liao 已提交
2894 2895
    taosArrayPush(pInfo->pSources, pNode);
  }
2896

2897 2898 2899 2900 2901 2902 2903
  return initDataSource(numOfSources, pInfo);
}

SOperatorInfo* createExchangeOperatorInfo(void* pTransporter, SExchangePhysiNode* pExNode, SExecTaskInfo* pTaskInfo) {
  SExchangeInfo* pInfo = taosMemoryCalloc(1, sizeof(SExchangeInfo));
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
H
Haojun Liao 已提交
2904
    goto _error;
2905
  }
H
Haojun Liao 已提交
2906

2907 2908 2909 2910
  int32_t code = initExchangeOperator(pExNode, pInfo);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
2911

X
Xiaoyu Wang 已提交
2912
  pInfo->seqLoadData = false;
2913
  pInfo->pTransporter = pTransporter;
X
Xiaoyu Wang 已提交
2914
  pInfo->pResult = createResDataBlock(pExNode->node.pOutputDataBlockDesc);
2915
  tsem_init(&pInfo->ready, 0, 0);
2916

X
Xiaoyu Wang 已提交
2917
  pOperator->name = "ExchangeOperator";
X
Xiaoyu Wang 已提交
2918
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
X
Xiaoyu Wang 已提交
2919 2920 2921 2922 2923
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfExprs = pInfo->pResult->info.numOfCols;
  pOperator->pTaskInfo = pTaskInfo;
2924

L
Liu Jicong 已提交
2925 2926
  pOperator->fpSet = createOperatorFpSet(prepareLoadRemoteData, doLoadRemoteData, NULL, NULL,
                                         destroyExchangeOperatorInfo, NULL, NULL, NULL);
2927
  return pOperator;
H
Haojun Liao 已提交
2928

L
Liu Jicong 已提交
2929
_error:
H
Haojun Liao 已提交
2930
  if (pInfo != NULL) {
2931
    destroyExchangeOperatorInfo(pInfo, LIST_LENGTH(pExNode->pSrcEndPoints));
H
Haojun Liao 已提交
2932 2933
  }

wafwerar's avatar
wafwerar 已提交
2934 2935
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
2936
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
H
Haojun Liao 已提交
2937
  return NULL;
2938 2939
}

dengyihao's avatar
dengyihao 已提交
2940 2941
static int32_t doInitAggInfoSup(SAggSupporter* pAggSup, SqlFunctionCtx* pCtx, int32_t numOfOutput, size_t keyBufSize,
                                const char* pKey);
2942

2943
static void destroySortedMergeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
2944
  SSortedMergeOperatorInfo* pInfo = (SSortedMergeOperatorInfo*)param;
H
Haojun Liao 已提交
2945
  taosArrayDestroy(pInfo->pSortInfo);
2946 2947 2948
  taosArrayDestroy(pInfo->groupInfo);

  if (pInfo->pSortHandle != NULL) {
H
Haojun Liao 已提交
2949
    tsortDestroySortHandle(pInfo->pSortHandle);
2950 2951
  }

H
Haojun Liao 已提交
2952
  blockDataDestroy(pInfo->binfo.pRes);
H
Haojun Liao 已提交
2953
  cleanupAggSup(&pInfo->aggSup);
2954
}
H
Haojun Liao 已提交
2955

L
Liu Jicong 已提交
2956
static bool needToMerge(SSDataBlock* pBlock, SArray* groupInfo, char** buf, int32_t rowIndex) {
2957 2958 2959 2960
  size_t size = taosArrayGetSize(groupInfo);
  if (size == 0) {
    return true;
  }
2961

2962 2963
  for (int32_t i = 0; i < size; ++i) {
    int32_t* index = taosArrayGet(groupInfo, i);
2964

2965
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, *index);
L
Liu Jicong 已提交
2966
    bool             isNull = colDataIsNull(pColInfo, rowIndex, pBlock->info.rows, NULL);
2967

2968 2969 2970
    if ((isNull && buf[i] != NULL) || (!isNull && buf[i] == NULL)) {
      return false;
    }
2971

2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984
    char* pCell = colDataGetData(pColInfo, rowIndex);
    if (IS_VAR_DATA_TYPE(pColInfo->info.type)) {
      if (varDataLen(pCell) != varDataLen(buf[i])) {
        return false;
      } else {
        if (memcmp(varDataVal(pCell), varDataVal(buf[i]), varDataLen(pCell)) != 0) {
          return false;
        }
      }
    } else {
      if (memcmp(pCell, buf[i], pColInfo->info.bytes) != 0) {
        return false;
      }
2985 2986 2987
    }
  }

2988
  return 0;
2989 2990
}

L
Liu Jicong 已提交
2991 2992 2993
static void doMergeResultImpl(SSortedMergeOperatorInfo* pInfo, SqlFunctionCtx* pCtx, int32_t numOfExpr,
                              int32_t rowIndex) {
  for (int32_t j = 0; j < numOfExpr; ++j) {  // TODO set row index
X
Xiaoyu Wang 已提交
2994
                                             //    pCtx[j].startRow = rowIndex;
2995 2996
  }

2997 2998
  for (int32_t j = 0; j < numOfExpr; ++j) {
    int32_t functionId = pCtx[j].functionId;
L
Liu Jicong 已提交
2999 3000 3001 3002 3003 3004 3005 3006 3007
    //    pCtx[j].fpSet->addInput(&pCtx[j]);

    //    if (functionId < 0) {
    //      SUdfInfo* pUdfInfo = taosArrayGet(pInfo->udfInfo, -1 * functionId - 1);
    //      doInvokeUdf(pUdfInfo, &pCtx[j], 0, TSDB_UDF_FUNC_MERGE);
    //    } else {
    //      assert(!TSDB_FUNC_IS_SCALAR(functionId));
    //      aAggs[functionId].mergeFunc(&pCtx[j]);
    //    }
3008
  }
3009
}
3010

L
Liu Jicong 已提交
3011 3012
static void doFinalizeResultImpl(SqlFunctionCtx* pCtx, int32_t numOfExpr) {
  for (int32_t j = 0; j < numOfExpr; ++j) {
3013 3014 3015 3016
    int32_t functionId = pCtx[j].functionId;
    //    if (functionId == FUNC_TAG_DUMMY || functionId == FUNC_TS_DUMMY) {
    //      continue;
    //    }
3017

3018 3019 3020 3021
    //    if (functionId < 0) {
    //      SUdfInfo* pUdfInfo = taosArrayGet(pInfo->udfInfo, -1 * functionId - 1);
    //      doInvokeUdf(pUdfInfo, &pCtx[j], 0, TSDB_UDF_FUNC_FINALIZE);
    //    } else {
dengyihao's avatar
dengyihao 已提交
3022
    //    pCtx[j].fpSet.finalize(&pCtx[j]);
3023 3024
  }
}
3025

3026
static bool saveCurrentTuple(char** rowColData, SArray* pColumnList, SSDataBlock* pBlock, int32_t rowIndex) {
L
Liu Jicong 已提交
3027
  int32_t size = (int32_t)taosArrayGetSize(pColumnList);
3028

L
Liu Jicong 已提交
3029 3030
  for (int32_t i = 0; i < size; ++i) {
    int32_t*         index = taosArrayGet(pColumnList, i);
3031
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, *index);
H
Haojun Liao 已提交
3032

3033 3034 3035
    char* data = colDataGetData(pColInfo, rowIndex);
    memcpy(rowColData[i], data, colDataGetLength(pColInfo, rowIndex));
  }
3036

3037 3038
  return true;
}
3039

3040 3041
static void doMergeImpl(SOperatorInfo* pOperator, int32_t numOfExpr, SSDataBlock* pBlock) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
3042

3043
  SqlFunctionCtx* pCtx = pInfo->binfo.pCtx;
L
Liu Jicong 已提交
3044
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
X
Xiaoyu Wang 已提交
3045
    //    pCtx[i].size = 1;
3046
  }
3047

L
Liu Jicong 已提交
3048
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
3049 3050 3051 3052 3053 3054 3055 3056 3057
    if (!pInfo->hasGroupVal) {
      ASSERT(i == 0);
      doMergeResultImpl(pInfo, pCtx, numOfExpr, i);
      pInfo->hasGroupVal = saveCurrentTuple(pInfo->groupVal, pInfo->groupInfo, pBlock, i);
    } else {
      if (needToMerge(pBlock, pInfo->groupInfo, pInfo->groupVal, i)) {
        doMergeResultImpl(pInfo, pCtx, numOfExpr, i);
      } else {
        doFinalizeResultImpl(pCtx, numOfExpr);
3058 3059
        int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfExprs, NULL);
        //        setTagValueForMultipleRows(pCtx, pOperator->numOfExprs, numOfRows);
3060

3061
        // TODO check for available buffer;
H
Haojun Liao 已提交
3062

3063 3064 3065 3066 3067
        // next group info data
        pInfo->binfo.pRes->info.rows += numOfRows;
        for (int32_t j = 0; j < numOfExpr; ++j) {
          if (pCtx[j].functionId < 0) {
            continue;
3068
          }
3069

H
Haojun Liao 已提交
3070
          pCtx[j].fpSet.process(&pCtx[j]);
3071
        }
3072 3073 3074

        doMergeResultImpl(pInfo, pCtx, numOfExpr, i);
        pInfo->hasGroupVal = saveCurrentTuple(pInfo->groupVal, pInfo->groupInfo, pBlock, i);
H
Haojun Liao 已提交
3075
      }
3076 3077 3078 3079
    }
  }
}

3080 3081
static SSDataBlock* doMerge(SOperatorInfo* pOperator) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
3082
  SSortHandle*              pHandle = pInfo->pSortHandle;
3083

3084
  SSDataBlock* pDataBlock = createOneDataBlock(pInfo->binfo.pRes, false);
3085
  blockDataEnsureCapacity(pDataBlock, pOperator->resultInfo.capacity);
3086

L
Liu Jicong 已提交
3087
  while (1) {
3088
    blockDataCleanup(pDataBlock);
3089
    while (1) {
H
Haojun Liao 已提交
3090
      STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
3091 3092
      if (pTupleHandle == NULL) {
        break;
3093
      }
3094

3095 3096
      // build datablock for merge for one group
      appendOneRowToDataBlock(pDataBlock, pTupleHandle);
3097
      if (pDataBlock->info.rows >= pOperator->resultInfo.capacity) {
3098 3099
        break;
      }
3100
    }
3101

3102 3103 3104
    if (pDataBlock->info.rows == 0) {
      break;
    }
3105

3106
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pDataBlock, TSDB_ORDER_ASC, MAIN_SCAN, true);
L
Liu Jicong 已提交
3107 3108
    //  updateOutputBuf(&pInfo->binfo, &pAggInfo->bufCapacity, pBlock->info.rows * pAggInfo->resultRowFactor,
    //  pOperator->pRuntimeEnv, true);
3109
    doMergeImpl(pOperator, pOperator->numOfExprs, pDataBlock);
3110 3111
    // flush to tuple store, and after all data have been handled, return to upstream node or sink node
  }
3112

3113 3114 3115
  doFinalizeResultImpl(pInfo->binfo.pCtx, pOperator->numOfExprs);
  int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfExprs, NULL);
  //        setTagValueForMultipleRows(pCtx, pOperator->numOfExprs, numOfRows);
3116

3117
  // TODO check for available buffer;
3118

3119 3120
  // next group info data
  pInfo->binfo.pRes->info.rows += numOfRows;
L
Liu Jicong 已提交
3121
  return (pInfo->binfo.pRes->info.rows > 0) ? pInfo->binfo.pRes : NULL;
3122
}
3123

3124
static SSDataBlock* doSortedMerge(SOperatorInfo* pOperator) {
3125 3126
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
3127 3128
  }

L
Liu Jicong 已提交
3129
  SExecTaskInfo*            pTaskInfo = pOperator->pTaskInfo;
3130
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
H
Haojun Liao 已提交
3131
  if (pOperator->status == OP_RES_TO_RETURN) {
3132
    return getSortedBlockData(pInfo->pSortHandle, pInfo->binfo.pRes, pOperator->resultInfo.capacity, NULL);
3133 3134
  }

3135
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
dengyihao's avatar
dengyihao 已提交
3136 3137
  pInfo->pSortHandle = tsortCreateSortHandle(pInfo->pSortInfo, NULL, SORT_MULTISOURCE_MERGE, pInfo->bufPageSize,
                                             numOfBufPage, pInfo->binfo.pRes, "GET_TASKID(pTaskInfo)");
H
Haojun Liao 已提交
3138

3139
  tsortSetFetchRawDataFp(pInfo->pSortHandle, loadNextDataBlock, NULL, NULL);
3140

L
Liu Jicong 已提交
3141
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
wmmhello's avatar
wmmhello 已提交
3142
    SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
H
Haojun Liao 已提交
3143
    ps->param = pOperator->pDownstream[i];
H
Haojun Liao 已提交
3144
    tsortAddSource(pInfo->pSortHandle, ps);
3145 3146
  }

H
Haojun Liao 已提交
3147
  int32_t code = tsortOpen(pInfo->pSortHandle);
3148
  if (code != TSDB_CODE_SUCCESS) {
3149
    longjmp(pTaskInfo->env, terrno);
3150 3151
  }

H
Haojun Liao 已提交
3152
  pOperator->status = OP_RES_TO_RETURN;
3153
  return doMerge(pOperator);
3154
}
3155

L
Liu Jicong 已提交
3156 3157
static int32_t initGroupCol(SExprInfo* pExprInfo, int32_t numOfCols, SArray* pGroupInfo,
                            SSortedMergeOperatorInfo* pInfo) {
3158 3159
  if (pGroupInfo == NULL || taosArrayGetSize(pGroupInfo) == 0) {
    return 0;
H
Haojun Liao 已提交
3160 3161
  }

3162 3163 3164 3165 3166 3167 3168 3169
  int32_t len = 0;
  SArray* plist = taosArrayInit(3, sizeof(SColumn));
  pInfo->groupInfo = taosArrayInit(3, sizeof(int32_t));

  if (plist == NULL || pInfo->groupInfo == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

L
Liu Jicong 已提交
3170 3171
  size_t numOfGroupCol = taosArrayGetSize(pInfo->groupInfo);
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
3172
    SColumn* pCol = taosArrayGet(pGroupInfo, i);
L
Liu Jicong 已提交
3173
    for (int32_t j = 0; j < numOfCols; ++j) {
H
Haojun Liao 已提交
3174
      SExprInfo* pe = &pExprInfo[j];
3175
      if (pe->base.resSchema.slotId == pCol->colId) {
3176 3177
        taosArrayPush(plist, pCol);
        taosArrayPush(pInfo->groupInfo, &j);
H
Haojun Liao 已提交
3178
        len += pCol->bytes;
3179 3180
        break;
      }
H
Haojun Liao 已提交
3181 3182 3183
    }
  }

3184
  ASSERT(taosArrayGetSize(pGroupInfo) == taosArrayGetSize(plist));
H
Haojun Liao 已提交
3185

wafwerar's avatar
wafwerar 已提交
3186
  pInfo->groupVal = taosMemoryCalloc(1, (POINTER_BYTES * numOfGroupCol + len));
3187 3188 3189 3190
  if (pInfo->groupVal == NULL) {
    taosArrayDestroy(plist);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
H
Haojun Liao 已提交
3191

3192
  int32_t offset = 0;
L
Liu Jicong 已提交
3193 3194
  char*   start = (char*)(pInfo->groupVal + (POINTER_BYTES * numOfGroupCol));
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
3195 3196
    pInfo->groupVal[i] = start + offset;
    SColumn* pCol = taosArrayGet(plist, i);
H
Haojun Liao 已提交
3197
    offset += pCol->bytes;
3198
  }
H
Haojun Liao 已提交
3199

3200
  taosArrayDestroy(plist);
H
Haojun Liao 已提交
3201

3202 3203
  return TSDB_CODE_SUCCESS;
}
H
Haojun Liao 已提交
3204

L
Liu Jicong 已提交
3205 3206 3207
SOperatorInfo* createSortedMergeOperatorInfo(SOperatorInfo** downstream, int32_t numOfDownstream, SExprInfo* pExprInfo,
                                             int32_t num, SArray* pSortInfo, SArray* pGroupInfo,
                                             SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
3208
  SSortedMergeOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSortedMergeOperatorInfo));
L
Liu Jicong 已提交
3209
  SOperatorInfo*            pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
3210
  if (pInfo == NULL || pOperator == NULL) {
3211
    goto _error;
3212
  }
H
Haojun Liao 已提交
3213

3214
  pInfo->binfo.pCtx = createSqlFunctionCtx(pExprInfo, num, &pInfo->binfo.rowCellInfoOffset);
3215
  initResultRowInfo(&pInfo->binfo.resultRowInfo, (int32_t)1);
H
Haojun Liao 已提交
3216

3217 3218 3219
  if (pInfo->binfo.pCtx == NULL || pInfo->binfo.pRes == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
3220

dengyihao's avatar
dengyihao 已提交
3221
  size_t  keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
3222
  int32_t code = doInitAggInfoSup(&pInfo->aggSup, pInfo->binfo.pCtx, num, keyBufSize, pTaskInfo->id.str);
3223 3224 3225
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
3226

H
Haojun Liao 已提交
3227
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, num, pTaskInfo);
H
Haojun Liao 已提交
3228
  code = initGroupCol(pExprInfo, num, pGroupInfo, pInfo);
3229 3230 3231
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
3232

L
Liu Jicong 已提交
3233 3234 3235 3236 3237
  //  pInfo->resultRowFactor = (int32_t)(getRowNumForMultioutput(pRuntimeEnv->pQueryAttr,
  //      pRuntimeEnv->pQueryAttr->topBotQuery, false));
  pInfo->sortBufSize = 1024 * 16;  // 1MB
  pInfo->bufPageSize = 1024;
  pInfo->pSortInfo = pSortInfo;
H
Haojun Liao 已提交
3238

3239
  pOperator->resultInfo.capacity = blockDataGetCapacityInRow(pInfo->binfo.pRes, pInfo->bufPageSize);
H
Haojun Liao 已提交
3240

L
Liu Jicong 已提交
3241
  pOperator->name = "SortedMerge";
X
Xiaoyu Wang 已提交
3242
  // pOperator->operatorType = OP_SortedMerge;
3243
  pOperator->blocking = true;
L
Liu Jicong 已提交
3244 3245
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
3246
  pOperator->numOfExprs = num;
L
Liu Jicong 已提交
3247
  pOperator->pExpr = pExprInfo;
H
Haojun Liao 已提交
3248

L
Liu Jicong 已提交
3249
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
3250

3251 3252
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doSortedMerge, NULL, NULL, destroySortedMergeOperatorInfo,
                                         NULL, NULL, NULL);
3253 3254 3255
  code = appendDownstream(pOperator, downstream, numOfDownstream);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
3256
  }
H
Haojun Liao 已提交
3257

3258
  return pOperator;
H
Haojun Liao 已提交
3259

L
Liu Jicong 已提交
3260
_error:
3261
  if (pInfo != NULL) {
H
Haojun Liao 已提交
3262
    destroySortedMergeOperatorInfo(pInfo, num);
H
Haojun Liao 已提交
3263 3264
  }

wafwerar's avatar
wafwerar 已提交
3265 3266
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
3267 3268
  terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
  return NULL;
H
Haojun Liao 已提交
3269 3270
}

X
Xiaoyu Wang 已提交
3271
int32_t getTableScanInfo(SOperatorInfo* pOperator, int32_t* order, int32_t* scanFlag) {
3272
  // todo add more information about exchange operation
3273
  int32_t type = pOperator->operatorType;
X
Xiaoyu Wang 已提交
3274
  if (type == QUERY_NODE_PHYSICAL_PLAN_EXCHANGE || type == QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN ||
3275
      type == QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN || type == QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN) {
3276 3277 3278
    *order = TSDB_ORDER_ASC;
    *scanFlag = MAIN_SCAN;
    return TSDB_CODE_SUCCESS;
3279
  } else if (type == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
3280 3281 3282 3283 3284
    STableScanInfo* pTableScanInfo = pOperator->info;
    *order = pTableScanInfo->cond.order;
    *scanFlag = pTableScanInfo->scanFlag;
    return TSDB_CODE_SUCCESS;
  } else {
H
Haojun Liao 已提交
3285
    if (pOperator->pDownstream == NULL || pOperator->pDownstream[0] == NULL) {
3286
      return TSDB_CODE_INVALID_PARA;
H
Haojun Liao 已提交
3287
    } else {
3288
      return getTableScanInfo(pOperator->pDownstream[0], order, scanFlag);
3289 3290 3291
    }
  }
}
3292 3293

// this is a blocking operator
L
Liu Jicong 已提交
3294
static int32_t doOpenAggregateOptr(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
3295 3296
  if (OPTR_IS_OPENED(pOperator)) {
    return TSDB_CODE_SUCCESS;
3297 3298
  }

H
Haojun Liao 已提交
3299
  SExecTaskInfo*    pTaskInfo = pOperator->pTaskInfo;
3300
  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
3301

dengyihao's avatar
dengyihao 已提交
3302
  SOptrBasicInfo* pInfo = &pAggInfo->binfo;
X
Xiaoyu Wang 已提交
3303
  SOperatorInfo*  downstream = pOperator->pDownstream[0];
3304

3305 3306
  int64_t st = taosGetTimestampUs();

3307 3308 3309
  int32_t order = TSDB_ORDER_ASC;
  int32_t scanFlag = MAIN_SCAN;

H
Haojun Liao 已提交
3310
  while (1) {
3311
    SSDataBlock* pBlock = downstream->fpSet.getNextFn(downstream);
3312 3313 3314 3315
    if (pBlock == NULL) {
      break;
    }

3316 3317 3318 3319
    int32_t code = getTableScanInfo(pOperator, &order, &scanFlag);
    if (code != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, code);
    }
3320

3321 3322
    // there is an scalar expression that needs to be calculated before apply the group aggregation.
    if (pAggInfo->pScalarExprInfo != NULL) {
3323
      code = projectApplyFunctions(pAggInfo->pScalarExprInfo, pBlock, pBlock, pAggInfo->pScalarCtx,
3324
                                   pAggInfo->numOfScalarExpr, NULL);
3325
      if (code != TSDB_CODE_SUCCESS) {
3326
        longjmp(pTaskInfo->env, code);
3327
      }
3328 3329
    }

3330
    // the pDataBlock are always the same one, no need to call this again
3331
    setExecutionContext(pOperator->numOfExprs, pBlock->info.groupId, pTaskInfo, pAggInfo);
3332
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order, scanFlag, true);
3333 3334 3335 3336
    code = doAggregateImpl(pOperator, 0, pInfo->pCtx);
    if (code != 0) {
      longjmp(pTaskInfo->env, code);
    }
3337

dengyihao's avatar
dengyihao 已提交
3338
#if 0  // test for encode/decode result info
3339
    if(pOperator->fpSet.encodeResultRow){
3340 3341
      char *result = NULL;
      int32_t length = 0;
3342 3343
      pOperator->fpSet.encodeResultRow(pOperator, &result, &length);
      SAggSupporter* pSup = &pAggInfo->aggSup;
3344 3345
      taosHashClear(pSup->pResultRowHashTable);
      pInfo->resultRowInfo.size = 0;
3346
      pOperator->fpSet.decodeResultRow(pOperator, result);
3347 3348 3349
      if(result){
        taosMemoryFree(result);
      }
3350
    }
3351
#endif
3352 3353
  }

H
Haojun Liao 已提交
3354
  closeAllResultRows(&pAggInfo->binfo.resultRowInfo);
3355
  initGroupedResultInfo(&pAggInfo->groupResInfo, pAggInfo->aggSup.pResultRowHashTable, 0);
H
Haojun Liao 已提交
3356
  OPTR_SET_OPENED(pOperator);
3357

3358
  pOperator->cost.openCost = (taosGetTimestampUs() - st) / 1000.0;
H
Haojun Liao 已提交
3359 3360 3361
  return TSDB_CODE_SUCCESS;
}

3362
static SSDataBlock* getAggregateResult(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
3363
  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
3364 3365 3366 3367 3368 3369
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;

  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

L
Liu Jicong 已提交
3370
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3371
  pTaskInfo->code = pOperator->fpSet._openFn(pOperator);
H
Haojun Liao 已提交
3372
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
3373
    doSetOperatorCompleted(pOperator);
H
Haojun Liao 已提交
3374 3375 3376
    return NULL;
  }

H
Haojun Liao 已提交
3377
  blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
3378
  doBuildResultDatablock(pOperator, pInfo, &pAggInfo->groupResInfo, pAggInfo->aggSup.pResultBuf);
3379
  if (pInfo->pRes->info.rows == 0 || !hashRemainDataInGroupInfo(&pAggInfo->groupResInfo)) {
H
Haojun Liao 已提交
3380 3381
    doSetOperatorCompleted(pOperator);
  }
3382

3383
  size_t rows = blockDataGetNumOfRows(pInfo->pRes);  // pInfo->pRes : NULL;
3384 3385
  pOperator->resultInfo.totalRows += rows;

3386
  return (rows == 0) ? NULL : pInfo->pRes;
3387 3388
}

wmmhello's avatar
wmmhello 已提交
3389
int32_t aggEncodeResultRow(SOperatorInfo* pOperator, char** result, int32_t* length) {
3390
  if (result == NULL || length == NULL) {
wmmhello's avatar
wmmhello 已提交
3391 3392 3393
    return TSDB_CODE_TSC_INVALID_INPUT;
  }
  SOptrBasicInfo* pInfo = (SOptrBasicInfo*)(pOperator->info);
3394 3395 3396 3397 3398
  SAggSupporter*  pSup = (SAggSupporter*)POINTER_SHIFT(pOperator->info, sizeof(SOptrBasicInfo));
  int32_t         size = taosHashGetSize(pSup->pResultRowHashTable);
  size_t          keyLen = sizeof(uint64_t) * 2;  // estimate the key length
  int32_t         totalSize =
      sizeof(int32_t) + sizeof(int32_t) + size * (sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize);
wmmhello's avatar
wmmhello 已提交
3399

wmmhello's avatar
wmmhello 已提交
3400
  *result = (char*)taosMemoryCalloc(1, totalSize);
L
Liu Jicong 已提交
3401
  if (*result == NULL) {
wmmhello's avatar
wmmhello 已提交
3402
    return TSDB_CODE_OUT_OF_MEMORY;
wmmhello's avatar
wmmhello 已提交
3403
  }
wmmhello's avatar
wmmhello 已提交
3404

wmmhello's avatar
wmmhello 已提交
3405
  int32_t offset = sizeof(int32_t);
wmmhello's avatar
wmmhello 已提交
3406 3407
  *(int32_t*)(*result + offset) = size;
  offset += sizeof(int32_t);
3408 3409

  // prepare memory
3410
  SResultRowPosition* pos = &pInfo->resultRowInfo.cur;
dengyihao's avatar
dengyihao 已提交
3411 3412
  void*               pPage = getBufPage(pSup->pResultBuf, pos->pageId);
  SResultRow*         pRow = (SResultRow*)((char*)pPage + pos->offset);
3413 3414 3415
  setBufPageDirty(pPage, true);
  releaseBufPage(pSup->pResultBuf, pPage);

dengyihao's avatar
dengyihao 已提交
3416
  void* pIter = taosHashIterate(pSup->pResultRowHashTable, NULL);
wmmhello's avatar
wmmhello 已提交
3417
  while (pIter) {
dengyihao's avatar
dengyihao 已提交
3418
    void*               key = taosHashGetKey(pIter, &keyLen);
3419
    SResultRowPosition* p1 = (SResultRowPosition*)pIter;
3420

dengyihao's avatar
dengyihao 已提交
3421
    pPage = (SFilePage*)getBufPage(pSup->pResultBuf, p1->pageId);
3422
    pRow = (SResultRow*)((char*)pPage + p1->offset);
3423 3424
    setBufPageDirty(pPage, true);
    releaseBufPage(pSup->pResultBuf, pPage);
wmmhello's avatar
wmmhello 已提交
3425 3426 3427

    // recalculate the result size
    int32_t realTotalSize = offset + sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize;
L
Liu Jicong 已提交
3428
    if (realTotalSize > totalSize) {
wmmhello's avatar
wmmhello 已提交
3429
      char* tmp = (char*)taosMemoryRealloc(*result, realTotalSize);
L
Liu Jicong 已提交
3430
      if (tmp == NULL) {
wafwerar's avatar
wafwerar 已提交
3431
        taosMemoryFree(*result);
wmmhello's avatar
wmmhello 已提交
3432
        *result = NULL;
wmmhello's avatar
wmmhello 已提交
3433
        return TSDB_CODE_OUT_OF_MEMORY;
L
Liu Jicong 已提交
3434
      } else {
wmmhello's avatar
wmmhello 已提交
3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446
        *result = tmp;
      }
    }
    // save key
    *(int32_t*)(*result + offset) = keyLen;
    offset += sizeof(int32_t);
    memcpy(*result + offset, key, keyLen);
    offset += keyLen;

    // save value
    *(int32_t*)(*result + offset) = pSup->resultRowSize;
    offset += sizeof(int32_t);
3447
    memcpy(*result + offset, pRow, pSup->resultRowSize);
wmmhello's avatar
wmmhello 已提交
3448 3449 3450 3451 3452
    offset += pSup->resultRowSize;

    pIter = taosHashIterate(pSup->pResultRowHashTable, pIter);
  }

wmmhello's avatar
wmmhello 已提交
3453 3454 3455 3456
  *(int32_t*)(*result) = offset;
  *length = offset;

  return TDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
3457 3458
}

3459
int32_t aggDecodeResultRow(SOperatorInfo* pOperator, char* result) {
3460
  if (result == NULL) {
wmmhello's avatar
wmmhello 已提交
3461
    return TSDB_CODE_TSC_INVALID_INPUT;
wmmhello's avatar
wmmhello 已提交
3462
  }
wmmhello's avatar
wmmhello 已提交
3463
  SOptrBasicInfo* pInfo = (SOptrBasicInfo*)(pOperator->info);
3464
  SAggSupporter*  pSup = (SAggSupporter*)POINTER_SHIFT(pOperator->info, sizeof(SOptrBasicInfo));
wmmhello's avatar
wmmhello 已提交
3465 3466

  //  int32_t size = taosHashGetSize(pSup->pResultRowHashTable);
3467
  int32_t length = *(int32_t*)(result);
wmmhello's avatar
wmmhello 已提交
3468
  int32_t offset = sizeof(int32_t);
3469 3470 3471 3472

  int32_t count = *(int32_t*)(result + offset);
  offset += sizeof(int32_t);

L
Liu Jicong 已提交
3473
  while (count-- > 0 && length > offset) {
wmmhello's avatar
wmmhello 已提交
3474 3475 3476
    int32_t keyLen = *(int32_t*)(result + offset);
    offset += sizeof(int32_t);

L
Liu Jicong 已提交
3477
    uint64_t    tableGroupId = *(uint64_t*)(result + offset);
3478
    SResultRow* resultRow = getNewResultRow(pSup->pResultBuf, tableGroupId, pSup->resultRowSize);
L
Liu Jicong 已提交
3479
    if (!resultRow) {
wmmhello's avatar
wmmhello 已提交
3480
      return TSDB_CODE_TSC_INVALID_INPUT;
wmmhello's avatar
wmmhello 已提交
3481
    }
3482

wmmhello's avatar
wmmhello 已提交
3483
    // add a new result set for a new group
3484 3485
    SResultRowPosition pos = {.pageId = resultRow->pageId, .offset = resultRow->offset};
    taosHashPut(pSup->pResultRowHashTable, result + offset, keyLen, &pos, sizeof(SResultRowPosition));
wmmhello's avatar
wmmhello 已提交
3486 3487 3488

    offset += keyLen;
    int32_t valueLen = *(int32_t*)(result + offset);
L
Liu Jicong 已提交
3489
    if (valueLen != pSup->resultRowSize) {
wmmhello's avatar
wmmhello 已提交
3490
      return TSDB_CODE_TSC_INVALID_INPUT;
wmmhello's avatar
wmmhello 已提交
3491 3492 3493 3494 3495 3496 3497 3498 3499 3500
    }
    offset += sizeof(int32_t);
    int32_t pageId = resultRow->pageId;
    int32_t pOffset = resultRow->offset;
    memcpy(resultRow, result + offset, valueLen);
    resultRow->pageId = pageId;
    resultRow->offset = pOffset;
    offset += valueLen;

    initResultRow(resultRow);
dengyihao's avatar
dengyihao 已提交
3501
    pInfo->resultRowInfo.cur = (SResultRowPosition){.pageId = resultRow->pageId, .offset = resultRow->offset};
wmmhello's avatar
wmmhello 已提交
3502 3503
  }

L
Liu Jicong 已提交
3504
  if (offset != length) {
wmmhello's avatar
wmmhello 已提交
3505
    return TSDB_CODE_TSC_INVALID_INPUT;
wmmhello's avatar
wmmhello 已提交
3506
  }
wmmhello's avatar
wmmhello 已提交
3507
  return TDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
3508 3509
}

3510 3511
enum {
  PROJECT_RETRIEVE_CONTINUE = 0x1,
L
Liu Jicong 已提交
3512
  PROJECT_RETRIEVE_DONE = 0x2,
3513 3514 3515 3516 3517
};

static int32_t handleLimitOffset(SOperatorInfo* pOperator, SSDataBlock* pBlock) {
  SProjectOperatorInfo* pProjectInfo = pOperator->info;
  SOptrBasicInfo*       pInfo = &pProjectInfo->binfo;
L
Liu Jicong 已提交
3518
  SSDataBlock*          pRes = pInfo->pRes;
3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565 3566

  if (pProjectInfo->curSOffset > 0) {
    if (pProjectInfo->groupId == 0) {  // it is the first group
      pProjectInfo->groupId = pBlock->info.groupId;
      blockDataCleanup(pInfo->pRes);
      return PROJECT_RETRIEVE_CONTINUE;
    } else if (pProjectInfo->groupId != pBlock->info.groupId) {
      pProjectInfo->curSOffset -= 1;

      // ignore data block in current group
      if (pProjectInfo->curSOffset > 0) {
        blockDataCleanup(pInfo->pRes);
        return PROJECT_RETRIEVE_CONTINUE;
      }
    }

    // set current group id of the project operator
    pProjectInfo->groupId = pBlock->info.groupId;
  }

  if (pProjectInfo->groupId != 0 && pProjectInfo->groupId != pBlock->info.groupId) {
    pProjectInfo->curGroupOutput += 1;
    if ((pProjectInfo->slimit.limit > 0) && (pProjectInfo->slimit.limit <= pProjectInfo->curGroupOutput)) {
      pOperator->status = OP_EXEC_DONE;
      blockDataCleanup(pRes);

      return PROJECT_RETRIEVE_DONE;
    }

    // reset the value for a new group data
    pProjectInfo->curOffset = 0;
    pProjectInfo->curOutput = 0;
  }

  // here we reach the start position, according to the limit/offset requirements.

  // set current group id
  pProjectInfo->groupId = pBlock->info.groupId;

  if (pProjectInfo->curOffset >= pRes->info.rows) {
    pProjectInfo->curOffset -= pRes->info.rows;
    blockDataCleanup(pRes);
    return PROJECT_RETRIEVE_CONTINUE;
  } else if (pProjectInfo->curOffset < pRes->info.rows && pProjectInfo->curOffset > 0) {
    blockDataTrimFirstNRows(pRes, pProjectInfo->curOffset);
    pProjectInfo->curOffset = 0;
  }

3567 3568 3569
  // check for the limitation in each group
  if (pProjectInfo->limit.limit > 0 && pProjectInfo->curOutput + pRes->info.rows >= pProjectInfo->limit.limit) {
    pRes->info.rows = (int32_t)(pProjectInfo->limit.limit - pProjectInfo->curOutput);
3570 3571 3572 3573 3574

    if (pProjectInfo->slimit.limit == -1 || pProjectInfo->slimit.limit <= pProjectInfo->curGroupOutput) {
      pOperator->status = OP_EXEC_DONE;
    }

3575
    return PROJECT_RETRIEVE_DONE;
3576
  }
3577

3578
  // todo optimize performance
3579 3580
  // If there are slimit/soffset value exists, multi-round result can not be packed into one group, since the
  // they may not belong to the same group the limit/offset value is not valid in this case.
L
Liu Jicong 已提交
3581 3582
  if (pRes->info.rows >= pOperator->resultInfo.threshold || pProjectInfo->slimit.offset != -1 ||
      pProjectInfo->slimit.limit != -1) {
3583
    return PROJECT_RETRIEVE_DONE;
L
Liu Jicong 已提交
3584
  } else {  // not full enough, continue to accumulate the output data in the buffer.
3585 3586 3587 3588
    return PROJECT_RETRIEVE_CONTINUE;
  }
}

3589
static SSDataBlock* doProjectOperation(SOperatorInfo* pOperator) {
3590
  SProjectOperatorInfo* pProjectInfo = pOperator->info;
L
Liu Jicong 已提交
3591
  SOptrBasicInfo*       pInfo = &pProjectInfo->binfo;
3592 3593

  SSDataBlock* pRes = pInfo->pRes;
3594
  blockDataCleanup(pRes);
3595

3596
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3597 3598 3599
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
3600

H
Haojun Liao 已提交
3601
#if 0
3602 3603 3604 3605 3606
  if (pProjectInfo->existDataBlock) {  // TODO refactor
    SSDataBlock* pBlock = pProjectInfo->existDataBlock;
    pProjectInfo->existDataBlock = NULL;

    // the pDataBlock are always the same one, no need to call this again
H
Haojun Liao 已提交
3607
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, TSDB_ORDER_ASC);
3608

H
Haojun Liao 已提交
3609
    blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
3610
    projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfExprs);
L
Liu Jicong 已提交
3611
    if (pRes->info.rows >= pProjectInfo->binfo.capacity * 0.8) {
3612 3613
      copyTsColoum(pRes, pInfo->pCtx, pOperator->numOfExprs);
      resetResultRowEntryResult(pInfo->pCtx, pOperator->numOfExprs);
3614 3615 3616
      return pRes;
    }
  }
H
Haojun Liao 已提交
3617
#endif
3618

3619
  int64_t st = 0;
3620 3621 3622
  int32_t order = 0;
  int32_t scanFlag = 0;

3623 3624 3625 3626
  if (pOperator->cost.openCost == 0) {
    st = taosGetTimestampUs();
  }

H
Haojun Liao 已提交
3627 3628
  SOperatorInfo* downstream = pOperator->pDownstream[0];

L
Liu Jicong 已提交
3629
  while (1) {
H
Haojun Liao 已提交
3630
    // The downstream exec may change the value of the newgroup, so use a local variable instead.
3631
    SSDataBlock* pBlock = downstream->fpSet.getNextFn(downstream);
3632
    if (pBlock == NULL) {
3633
      doSetOperatorCompleted(pOperator);
3634 3635 3636 3637
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
3638
    int32_t code = getTableScanInfo(pOperator->pDownstream[0], &order, &scanFlag);
3639 3640 3641
    if (code != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, code);
    }
3642

3643 3644
    doFilter(pProjectInfo->pFilterNode, pBlock);

3645
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order, scanFlag, false);
3646 3647
    blockDataEnsureCapacity(pInfo->pRes, pInfo->pRes->info.rows + pBlock->info.rows);

X
Xiaoyu Wang 已提交
3648 3649
    code = projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfExprs,
                                 pProjectInfo->pPseudoColInfo);
3650 3651
    if (code != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, code);
3652 3653
    }

3654 3655
    int32_t status = handleLimitOffset(pOperator, pBlock);
    if (status == PROJECT_RETRIEVE_CONTINUE) {
H
Haojun Liao 已提交
3656
      continue;
L
Liu Jicong 已提交
3657
    } else if (status == PROJECT_RETRIEVE_DONE) {
3658 3659 3660
      break;
    }
  }
dengyihao's avatar
dengyihao 已提交
3661

H
Haojun Liao 已提交
3662
  pProjectInfo->curOutput += pInfo->pRes->info.rows;
H
Haojun Liao 已提交
3663

3664 3665 3666 3667
  size_t rows = pInfo->pRes->info.rows;
  pOperator->resultInfo.totalRows += rows;

  if (pOperator->cost.openCost == 0) {
3668
    pOperator->cost.openCost = (taosGetTimestampUs() - st) / 1000.0;
3669 3670
  }

3671
  return (rows > 0) ? pInfo->pRes : NULL;
3672 3673
}

L
Liu Jicong 已提交
3674 3675
static void doHandleRemainBlockForNewGroupImpl(SFillOperatorInfo* pInfo, SResultInfo* pResultInfo, bool* newgroup,
                                               SExecTaskInfo* pTaskInfo) {
3676
  pInfo->totalInputRows = pInfo->existNewGroupBlock->info.rows;
H
Haojun Liao 已提交
3677

L
Liu Jicong 已提交
3678 3679
  int64_t ekey = Q_STATUS_EQUAL(pTaskInfo->status, TASK_COMPLETED) ? pTaskInfo->window.ekey
                                                                   : pInfo->existNewGroupBlock->info.window.ekey;
3680 3681
  taosResetFillInfo(pInfo->pFillInfo, getFillInfoStart(pInfo->pFillInfo));

3682
  taosFillSetStartInfo(pInfo->pFillInfo, pInfo->existNewGroupBlock->info.rows, ekey);
3683 3684
  taosFillSetInputDataBlock(pInfo->pFillInfo, pInfo->existNewGroupBlock);

3685
  doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, pResultInfo->capacity);
3686 3687 3688 3689
  pInfo->existNewGroupBlock = NULL;
  *newgroup = true;
}

L
Liu Jicong 已提交
3690 3691
static void doHandleRemainBlockFromNewGroup(SFillOperatorInfo* pInfo, SResultInfo* pResultInfo, bool* newgroup,
                                            SExecTaskInfo* pTaskInfo) {
3692 3693
  if (taosFillHasMoreResults(pInfo->pFillInfo)) {
    *newgroup = false;
3694
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, (int32_t)pResultInfo->capacity);
H
Haojun Liao 已提交
3695
    if (pInfo->pRes->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult)) {
3696 3697 3698 3699 3700 3701
      return;
    }
  }

  // handle the cached new group data block
  if (pInfo->existNewGroupBlock) {
3702
    doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
3703 3704 3705
  }
}

3706
static SSDataBlock* doFill(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
3707 3708
  SFillOperatorInfo* pInfo = pOperator->info;
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
3709

H
Haojun Liao 已提交
3710
  SResultInfo* pResultInfo = &pOperator->resultInfo;
3711 3712 3713
  SSDataBlock* pResBlock = pInfo->pRes;

  blockDataCleanup(pResBlock);
3714 3715 3716 3717
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

3718
  // todo handle different group data interpolation
X
Xiaoyu Wang 已提交
3719 3720
  bool  n = false;
  bool* newgroup = &n;
3721
  doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
3722 3723
  if (pResBlock->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult && pResBlock->info.rows > 0)) {
    return pResBlock;
H
Haojun Liao 已提交
3724
  }
3725

H
Haojun Liao 已提交
3726
  SOperatorInfo* pDownstream = pOperator->pDownstream[0];
L
Liu Jicong 已提交
3727
  while (1) {
3728
    SSDataBlock* pBlock = pDownstream->fpSet.getNextFn(pDownstream);
3729 3730 3731 3732 3733 3734 3735 3736 3737 3738
    if (*newgroup) {
      assert(pBlock != NULL);
    }

    if (*newgroup && pInfo->totalInputRows > 0) {  // there are already processed current group data block
      pInfo->existNewGroupBlock = pBlock;
      *newgroup = false;

      // Fill the previous group data block, before handle the data block of new group.
      // Close the fill operation for previous group data block
3739
      taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
3740 3741 3742 3743 3744 3745 3746
    } else {
      if (pBlock == NULL) {
        if (pInfo->totalInputRows == 0) {
          pOperator->status = OP_EXEC_DONE;
          return NULL;
        }

3747
        taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
3748 3749 3750 3751 3752 3753 3754
      } else {
        pInfo->totalInputRows += pBlock->info.rows;
        taosFillSetStartInfo(pInfo->pFillInfo, pBlock->info.rows, pBlock->info.window.ekey);
        taosFillSetInputDataBlock(pInfo->pFillInfo, pBlock);
      }
    }

3755 3756
    blockDataEnsureCapacity(pResBlock, pOperator->resultInfo.capacity);
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pResBlock, pOperator->resultInfo.capacity);
3757 3758

    // current group has no more result to return
3759
    if (pResBlock->info.rows > 0) {
3760 3761
      // 1. The result in current group not reach the threshold of output result, continue
      // 2. If multiple group results existing in one SSDataBlock is not allowed, return immediately
3762 3763
      if (pResBlock->info.rows > pResultInfo->threshold || pBlock == NULL || (!pInfo->multigroupResult)) {
        return pResBlock;
3764 3765
      }

3766
      doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
3767 3768
      if (pResBlock->info.rows > pOperator->resultInfo.threshold || pBlock == NULL) {
        return pResBlock;
3769 3770 3771
      }
    } else if (pInfo->existNewGroupBlock) {  // try next group
      assert(pBlock != NULL);
3772
      doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
3773 3774
      if (pResBlock->info.rows > pResultInfo->threshold) {
        return pResBlock;
3775 3776 3777 3778 3779 3780 3781
      }
    } else {
      return NULL;
    }
  }
}

H
Haojun Liao 已提交
3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792
static void destroyExprInfo(SExprInfo* pExpr, int32_t numOfExprs) {
  for (int32_t i = 0; i < numOfExprs; ++i) {
    SExprInfo* pExprInfo = &pExpr[i];
    if (pExprInfo->pExpr->nodeType == QUERY_NODE_COLUMN) {
      taosMemoryFree(pExprInfo->base.pParam[0].pCol);
    }
    taosMemoryFree(pExprInfo->base.pParam);
    taosMemoryFree(pExprInfo->pExpr);
  }
}

3793 3794 3795 3796 3797
static void destroyOperatorInfo(SOperatorInfo* pOperator) {
  if (pOperator == NULL) {
    return;
  }

3798
  if (pOperator->fpSet.closeFn != NULL) {
3799
    pOperator->fpSet.closeFn(pOperator->info, pOperator->numOfExprs);
3800 3801
  }

H
Haojun Liao 已提交
3802
  if (pOperator->pDownstream != NULL) {
L
Liu Jicong 已提交
3803
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
H
Haojun Liao 已提交
3804
      destroyOperatorInfo(pOperator->pDownstream[i]);
3805 3806
    }

wafwerar's avatar
wafwerar 已提交
3807
    taosMemoryFreeClear(pOperator->pDownstream);
H
Haojun Liao 已提交
3808
    pOperator->numOfDownstream = 0;
3809 3810
  }

H
Haojun Liao 已提交
3811
  if (pOperator->pExpr != NULL) {
H
Haojun Liao 已提交
3812
    destroyExprInfo(pOperator->pExpr, pOperator->numOfExprs);
H
Haojun Liao 已提交
3813 3814
  }

D
fix bug  
dapan 已提交
3815
  taosMemoryFreeClear(pOperator->pExpr);
wafwerar's avatar
wafwerar 已提交
3816 3817
  taosMemoryFreeClear(pOperator->info);
  taosMemoryFreeClear(pOperator);
3818 3819
}

3820 3821 3822 3823 3824 3825 3826 3827 3828 3829 3830 3831 3832 3833 3834
int32_t getBufferPgSize(int32_t rowSize, uint32_t* defaultPgsz, uint32_t* defaultBufsz) {
  *defaultPgsz = 4096;
  while (*defaultPgsz < rowSize * 4) {
    *defaultPgsz <<= 1u;
  }

  // at least four pages need to be in buffer
  *defaultBufsz = 4096 * 256;
  if ((*defaultBufsz) <= (*defaultPgsz)) {
    (*defaultBufsz) = (*defaultPgsz) * 4;
  }

  return 0;
}

dengyihao's avatar
dengyihao 已提交
3835 3836
int32_t doInitAggInfoSup(SAggSupporter* pAggSup, SqlFunctionCtx* pCtx, int32_t numOfOutput, size_t keyBufSize,
                         const char* pKey) {
3837 3838
  _hash_fn_t hashFn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY);

dengyihao's avatar
dengyihao 已提交
3839 3840
  pAggSup->resultRowSize = getResultRowSize(pCtx, numOfOutput);
  pAggSup->keyBuf = taosMemoryCalloc(1, keyBufSize + POINTER_BYTES + sizeof(int64_t));
3841 3842
  pAggSup->pResultRowHashTable = taosHashInit(10, hashFn, true, HASH_NO_LOCK);

H
Haojun Liao 已提交
3843
  if (pAggSup->keyBuf == NULL || pAggSup->pResultRowHashTable == NULL) {
3844 3845 3846
    return TSDB_CODE_OUT_OF_MEMORY;
  }

dengyihao's avatar
dengyihao 已提交
3847
  uint32_t defaultPgsz = 0;
3848 3849
  uint32_t defaultBufsz = 0;
  getBufferPgSize(pAggSup->resultRowSize, &defaultPgsz, &defaultBufsz);
H
Haojun Liao 已提交
3850

3851
  int32_t code = createDiskbasedBuf(&pAggSup->pResultBuf, defaultPgsz, defaultBufsz, pKey, TD_TMP_DIR_PATH);
H
Haojun Liao 已提交
3852 3853 3854 3855
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

3856 3857 3858
  return TSDB_CODE_SUCCESS;
}

3859
void cleanupAggSup(SAggSupporter* pAggSup) {
wafwerar's avatar
wafwerar 已提交
3860
  taosMemoryFreeClear(pAggSup->keyBuf);
3861
  taosHashCleanup(pAggSup->pResultRowHashTable);
H
Haojun Liao 已提交
3862
  destroyDiskbasedBuf(pAggSup->pResultBuf);
3863 3864
}

H
Haojun Liao 已提交
3865
int32_t initAggInfo(SOptrBasicInfo* pBasicInfo, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols,
3866
                    SSDataBlock* pResultBlock, size_t keyBufSize, const char* pkey) {
3867
  pBasicInfo->pCtx = createSqlFunctionCtx(pExprInfo, numOfCols, &pBasicInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
3868 3869
  pBasicInfo->pRes = pResultBlock;

3870
  doInitAggInfoSup(pAggSup, pBasicInfo->pCtx, numOfCols, keyBufSize, pkey);
3871

L
Liu Jicong 已提交
3872
  for (int32_t i = 0; i < numOfCols; ++i) {
3873 3874 3875
    pBasicInfo->pCtx[i].pBuf = pAggSup->pResultBuf;
  }

3876
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
3877 3878
}

3879 3880 3881 3882 3883 3884 3885 3886 3887
void initResultSizeInfo(SOperatorInfo* pOperator, int32_t numOfRows) {
  pOperator->resultInfo.capacity = numOfRows;
  pOperator->resultInfo.threshold = numOfRows * 0.75;

  if (pOperator->resultInfo.threshold == 0) {
    pOperator->resultInfo.capacity = numOfRows;
  }
}

dengyihao's avatar
dengyihao 已提交
3888
// static STableQueryInfo* initTableQueryInfo(const STableListInfo* pTableListInfo) {
wmmhello's avatar
wmmhello 已提交
3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907
//  int32_t size = taosArrayGetSize(pTableListInfo->pTableList);
//  if (size == 0) {
//    return NULL;
//  }
//
//  STableQueryInfo* pTableQueryInfo = taosMemoryCalloc(size, sizeof(STableQueryInfo));
//  if (pTableQueryInfo == NULL) {
//    return NULL;
//  }
//
//  for (int32_t j = 0; j < size; ++j) {
//    STableKeyInfo*   pk = taosArrayGet(pTableListInfo->pTableList, j);
//    STableQueryInfo* pTQueryInfo = &pTableQueryInfo[j];
//    pTQueryInfo->lastKey = pk->lastKey;
//  }
//
//  pTableQueryInfo->lastKey = 0;
//  return pTableQueryInfo;
//}
H
Haojun Liao 已提交
3908

L
Liu Jicong 已提交
3909
SOperatorInfo* createAggregateOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
3910
                                           SSDataBlock* pResultBlock, SExprInfo* pScalarExprInfo,
wmmhello's avatar
wmmhello 已提交
3911
                                           int32_t numOfScalarExpr, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
3912
  SAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SAggOperatorInfo));
L
Liu Jicong 已提交
3913
  SOperatorInfo*    pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
3914 3915 3916
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
3917

3918
  int32_t numOfRows = 1024;
dengyihao's avatar
dengyihao 已提交
3919
  size_t  keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
3920 3921

  initResultSizeInfo(pOperator, numOfRows);
dengyihao's avatar
dengyihao 已提交
3922 3923
  int32_t code =
      initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, pResultBlock, keyBufSize, pTaskInfo->id.str);
L
Liu Jicong 已提交
3924
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
3925 3926
    goto _error;
  }
H
Haojun Liao 已提交
3927

H
Haojun Liao 已提交
3928 3929 3930 3931
  int32_t numOfGroup = 10;  // todo replaced with true value
  pInfo->groupId = INT32_MIN;
  initResultRowInfo(&pInfo->binfo.resultRowInfo, numOfGroup);

3932 3933
  pInfo->pScalarExprInfo = pScalarExprInfo;
  pInfo->numOfScalarExpr = numOfScalarExpr;
3934
  if (pInfo->pScalarExprInfo != NULL) {
3935
    pInfo->pScalarCtx = createSqlFunctionCtx(pScalarExprInfo, numOfScalarExpr, &pInfo->rowCellInfoOffset);
3936
  }
3937

dengyihao's avatar
dengyihao 已提交
3938
  pOperator->name = "TableAggregate";
3939
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_HASH_AGG;
3940
  pOperator->blocking = true;
dengyihao's avatar
dengyihao 已提交
3941 3942 3943
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
3944
  pOperator->numOfExprs = numOfCols;
dengyihao's avatar
dengyihao 已提交
3945
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
3946

3947 3948
  pOperator->fpSet = createOperatorFpSet(doOpenAggregateOptr, getAggregateResult, NULL, NULL, destroyAggOperatorInfo,
                                         aggEncodeResultRow, aggDecodeResultRow, NULL);
H
Haojun Liao 已提交
3949 3950 3951 3952 3953

  code = appendDownstream(pOperator, &downstream, 1);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
3954 3955

  return pOperator;
L
Liu Jicong 已提交
3956
_error:
H
Haojun Liao 已提交
3957
  destroyAggOperatorInfo(pInfo, numOfCols);
wafwerar's avatar
wafwerar 已提交
3958 3959
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
3960 3961
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
3962 3963
}

H
Haojun Liao 已提交
3964
void doDestroyBasicInfo(SOptrBasicInfo* pInfo, int32_t numOfOutput) {
3965 3966
  assert(pInfo != NULL);

3967
  destroySqlFunctionCtx(pInfo->pCtx, numOfOutput);
wafwerar's avatar
wafwerar 已提交
3968
  taosMemoryFreeClear(pInfo->rowCellInfoOffset);
3969 3970

  cleanupResultRowInfo(&pInfo->resultRowInfo);
H
Haojun Liao 已提交
3971
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
3972 3973
}

H
Haojun Liao 已提交
3974
void destroyBasicOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
3975
  SOptrBasicInfo* pInfo = (SOptrBasicInfo*)param;
3976 3977
  doDestroyBasicInfo(pInfo, numOfOutput);
}
H
Haojun Liao 已提交
3978 3979

void destroyAggOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
3980
  SAggOperatorInfo* pInfo = (SAggOperatorInfo*)param;
3981 3982
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}
3983

H
Haojun Liao 已提交
3984
void destroySFillOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
3985
  SFillOperatorInfo* pInfo = (SFillOperatorInfo*)param;
3986
  pInfo->pFillInfo = taosDestroyFillInfo(pInfo->pFillInfo);
H
Haojun Liao 已提交
3987
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
wafwerar's avatar
wafwerar 已提交
3988
  taosMemoryFreeClear(pInfo->p);
3989 3990
}

H
Haojun Liao 已提交
3991
static void destroyProjectOperatorInfo(void* param, int32_t numOfOutput) {
D
fix bug  
dapan 已提交
3992 3993 3994
  if (NULL == param) {
    return;
  }
L
Liu Jicong 已提交
3995
  SProjectOperatorInfo* pInfo = (SProjectOperatorInfo*)param;
3996
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
H
Haojun Liao 已提交
3997
  cleanupAggSup(&pInfo->aggSup);
H
Haojun Liao 已提交
3998
  taosArrayDestroy(pInfo->pPseudoColInfo);
3999 4000
}

H
Haojun Liao 已提交
4001
static void destroyIndefinitOperatorInfo(void* param, int32_t numOfOutput) {
4002
  SIndefOperatorInfo* pInfo = (SIndefOperatorInfo*)param;
H
Haojun Liao 已提交
4003 4004 4005 4006 4007 4008 4009 4010 4011 4012 4013
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);

  taosArrayDestroy(pInfo->pPseudoColInfo);
  cleanupAggSup(&pInfo->aggSup);

  destroySqlFunctionCtx(pInfo->pScalarCtx, numOfOutput);
  destroyExprInfo(pInfo->pScalarExpr, pInfo->numOfScalarExpr);

  taosMemoryFree(pInfo->rowCellInfoOffset);
}

H
Haojun Liao 已提交
4014
void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
4015
  SExchangeInfo* pExInfo = (SExchangeInfo*)param;
H
Haojun Liao 已提交
4016 4017 4018 4019 4020 4021 4022 4023 4024
  taosArrayDestroy(pExInfo->pSources);
  taosArrayDestroy(pExInfo->pSourceDataInfo);
  if (pExInfo->pResult != NULL) {
    blockDataDestroy(pExInfo->pResult);
  }

  tsem_destroy(&pExInfo->ready);
}

H
Haojun Liao 已提交
4025 4026
static SArray* setRowTsColumnOutputInfo(SqlFunctionCtx* pCtx, int32_t numOfCols) {
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
dengyihao's avatar
dengyihao 已提交
4027
  for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
4028 4029 4030 4031 4032 4033 4034 4035
    if (fmIsPseudoColumnFunc(pCtx[i].functionId)) {
      taosArrayPush(pList, &i);
    }
  }

  return pList;
}

L
Liu Jicong 已提交
4036
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t num,
4037
                                         SSDataBlock* pResBlock, SLimit* pLimit, SLimit* pSlimit, SNode* pCondition,
dengyihao's avatar
dengyihao 已提交
4038
                                         SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
4039
  SProjectOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SProjectOperatorInfo));
L
Liu Jicong 已提交
4040
  SOperatorInfo*        pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
4041 4042 4043
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
4044

X
Xiaoyu Wang 已提交
4045 4046 4047
  pInfo->limit = *pLimit;
  pInfo->slimit = *pSlimit;
  pInfo->curOffset = pLimit->offset;
H
Haojun Liao 已提交
4048
  pInfo->curSOffset = pSlimit->offset;
H
Haojun Liao 已提交
4049
  pInfo->binfo.pRes = pResBlock;
X
Xiaoyu Wang 已提交
4050
  pInfo->pFilterNode = pCondition;
H
Haojun Liao 已提交
4051 4052 4053

  int32_t numOfCols = num;
  int32_t numOfRows = 4096;
dengyihao's avatar
dengyihao 已提交
4054
  size_t  keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
4055

4056 4057 4058 4059 4060
  // Make sure the size of SSDataBlock will never exceed the size of 2MB.
  int32_t TWOMB = 2 * 1024 * 1024;
  if (numOfRows * pResBlock->info.rowSize > TWOMB) {
    numOfRows = TWOMB / pResBlock->info.rowSize;
  }
4061
  initResultSizeInfo(pOperator, numOfRows);
4062

4063
  initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, pResBlock, keyBufSize, pTaskInfo->id.str);
H
Haojun Liao 已提交
4064
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, numOfCols, pTaskInfo);
4065

X
Xiaoyu Wang 已提交
4066 4067
  pInfo->pPseudoColInfo = setRowTsColumnOutputInfo(pInfo->binfo.pCtx, numOfCols);
  pOperator->name = "ProjectOperator";
H
Haojun Liao 已提交
4068
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_PROJECT;
X
Xiaoyu Wang 已提交
4069 4070 4071 4072 4073 4074
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfExprs = num;
  pOperator->pTaskInfo = pTaskInfo;
4075

L
Liu Jicong 已提交
4076 4077
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doProjectOperation, NULL, NULL,
                                         destroyProjectOperatorInfo, NULL, NULL, NULL);
L
Liu Jicong 已提交
4078

4079
  int32_t code = appendDownstream(pOperator, &downstream, 1);
H
Haojun Liao 已提交
4080
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
4081 4082
    goto _error;
  }
4083 4084

  return pOperator;
H
Haojun Liao 已提交
4085

L
Liu Jicong 已提交
4086
_error:
H
Haojun Liao 已提交
4087 4088
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
4089 4090
}

H
Haojun Liao 已提交
4091 4092
static SSDataBlock* doApplyIndefinitFunction(SOperatorInfo* pOperator) {
  SIndefOperatorInfo* pIndefInfo = pOperator->info;
4093
  SOptrBasicInfo*     pInfo = &pIndefInfo->binfo;
H
Haojun Liao 已提交
4094 4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138

  SSDataBlock* pRes = pInfo->pRes;
  blockDataCleanup(pRes);

  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  int64_t st = 0;
  int32_t order = 0;
  int32_t scanFlag = 0;

  if (pOperator->cost.openCost == 0) {
    st = taosGetTimestampUs();
  }

  SOperatorInfo* downstream = pOperator->pDownstream[0];

  while (1) {
    // The downstream exec may change the value of the newgroup, so use a local variable instead.
    SSDataBlock* pBlock = downstream->fpSet.getNextFn(downstream);
    if (pBlock == NULL) {
      doSetOperatorCompleted(pOperator);
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
    int32_t code = getTableScanInfo(pOperator->pDownstream[0], &order, &scanFlag);
    if (code != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, code);
    }

    // there is an scalar expression that needs to be calculated before apply the group aggregation.
    if (pIndefInfo->pScalarExpr != NULL) {
      code = projectApplyFunctions(pIndefInfo->pScalarExpr, pBlock, pBlock, pIndefInfo->pScalarCtx,
                                   pIndefInfo->numOfScalarExpr, pIndefInfo->pPseudoColInfo);
      if (code != TSDB_CODE_SUCCESS) {
        longjmp(pTaskInfo->env, code);
      }
    }

    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order, scanFlag, false);
    blockDataEnsureCapacity(pInfo->pRes, pInfo->pRes->info.rows + pBlock->info.rows);

4139 4140
    code = projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfExprs,
                                 pIndefInfo->pPseudoColInfo);
H
Haojun Liao 已提交
4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155
    if (code != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, code);
    }
  }

  size_t rows = pInfo->pRes->info.rows;
  pOperator->resultInfo.totalRows += rows;

  if (pOperator->cost.openCost == 0) {
    pOperator->cost.openCost = (taosGetTimestampUs() - st) / 1000.0;
  }

  return (rows > 0) ? pInfo->pRes : NULL;
}

4156 4157
SOperatorInfo* createIndefinitOutputOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pNode,
                                                 SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
4158
  SIndefOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SIndefOperatorInfo));
4159
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }

  SIndefRowsFuncPhysiNode* pPhyNode = (SIndefRowsFuncPhysiNode*)pNode;

  int32_t    numOfExpr = 0;
  SExprInfo* pExprInfo = createExprInfo(pPhyNode->pVectorFuncs, NULL, &numOfExpr);

  int32_t numOfScalarExpr = 0;
  if (pPhyNode->pExprs != NULL) {
    pInfo->pScalarExpr = createExprInfo(pPhyNode->pExprs, NULL, &numOfScalarExpr);
4172
    pInfo->pScalarCtx = createSqlFunctionCtx(pInfo->pScalarExpr, numOfScalarExpr, &pInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
4173 4174
  }

4175 4176
  SSDataBlock* pResBlock = createResDataBlock(pPhyNode->node.pOutputDataBlockDesc);
  ;
H
Haojun Liao 已提交
4177 4178 4179 4180 4181 4182 4183 4184 4185 4186 4187 4188 4189 4190

  int32_t numOfRows = 4096;
  size_t  keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;

  // Make sure the size of SSDataBlock will never exceed the size of 2MB.
  int32_t TWOMB = 2 * 1024 * 1024;
  if (numOfRows * pResBlock->info.rowSize > TWOMB) {
    numOfRows = TWOMB / pResBlock->info.rowSize;
  }
  initResultSizeInfo(pOperator, numOfRows);

  initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfExpr, pResBlock, keyBufSize, pTaskInfo->id.str);
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, numOfExpr, pTaskInfo);

4191 4192 4193
  pInfo->binfo.pRes = pResBlock;
  pInfo->numOfScalarExpr = numOfScalarExpr;
  pInfo->pPseudoColInfo = setRowTsColumnOutputInfo(pInfo->binfo.pCtx, numOfExpr);
H
Haojun Liao 已提交
4194

4195
  pOperator->name = "IndefinitOperator";
H
Haojun Liao 已提交
4196
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_PROJECT;
4197 4198 4199 4200 4201 4202
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfExprs = numOfExpr;
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
4203 4204 4205 4206 4207 4208 4209 4210 4211 4212 4213

  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doApplyIndefinitFunction, NULL, NULL,
                                         destroyIndefinitOperatorInfo, NULL, NULL, NULL);

  int32_t code = appendDownstream(pOperator, &downstream, 1);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

  return pOperator;

4214
_error:
H
Haojun Liao 已提交
4215 4216 4217 4218 4219 4220
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
}

4221
static int32_t initFillInfo(SFillOperatorInfo* pInfo, SExprInfo* pExpr, int32_t numOfCols, SNodeListNode* pValNode,
L
Liu Jicong 已提交
4222
                            STimeWindow win, int32_t capacity, const char* id, SInterval* pInterval, int32_t fillType) {
4223
  SFillColInfo* pColInfo = createFillColInfo(pExpr, numOfCols, pValNode);
H
Haojun Liao 已提交
4224 4225

  STimeWindow w = TSWINDOW_INITIALIZER;
4226
  getAlignQueryTimeWindow(pInterval, pInterval->precision, win.skey, &w);
H
Haojun Liao 已提交
4227 4228

  int32_t order = TSDB_ORDER_ASC;
4229
  pInfo->pFillInfo = taosCreateFillInfo(order, w.skey, 0, capacity, numOfCols, pInterval, fillType, pColInfo, id);
H
Haojun Liao 已提交
4230

wafwerar's avatar
wafwerar 已提交
4231
  pInfo->p = taosMemoryCalloc(numOfCols, POINTER_BYTES);
H
Haojun Liao 已提交
4232 4233 4234 4235 4236 4237 4238
  if (pInfo->pFillInfo == NULL || pInfo->p == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  } else {
    return TSDB_CODE_SUCCESS;
  }
}

L
Liu Jicong 已提交
4239
SOperatorInfo* createFillOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExpr, int32_t numOfCols,
X
Xiaoyu Wang 已提交
4240 4241 4242
                                      SInterval* pInterval, STimeWindow* pWindow, SSDataBlock* pResBlock,
                                      int32_t fillType, SNodeListNode* pValueNode, bool multigroupResult,
                                      SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
4243
  SFillOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SFillOperatorInfo));
L
Liu Jicong 已提交
4244
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
4245

L
Liu Jicong 已提交
4246
  pInfo->pRes = pResBlock;
4247 4248
  pInfo->multigroupResult = multigroupResult;

4249 4250
  int32_t type = TSDB_FILL_NONE;
  switch (fillType) {
dengyihao's avatar
dengyihao 已提交
4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268
    case FILL_MODE_PREV:
      type = TSDB_FILL_PREV;
      break;
    case FILL_MODE_NONE:
      type = TSDB_FILL_NONE;
      break;
    case FILL_MODE_NULL:
      type = TSDB_FILL_NULL;
      break;
    case FILL_MODE_NEXT:
      type = TSDB_FILL_NEXT;
      break;
    case FILL_MODE_VALUE:
      type = TSDB_FILL_SET_VALUE;
      break;
    case FILL_MODE_LINEAR:
      type = TSDB_FILL_LINEAR;
      break;
4269 4270 4271 4272
    default:
      type = TSDB_FILL_NONE;
  }

H
Haojun Liao 已提交
4273
  SResultInfo* pResultInfo = &pOperator->resultInfo;
4274 4275
  initResultSizeInfo(pOperator, 4096);

X
Xiaoyu Wang 已提交
4276 4277
  int32_t code = initFillInfo(pInfo, pExpr, numOfCols, pValueNode, *pWindow, pResultInfo->capacity, pTaskInfo->id.str,
                              pInterval, type);
4278 4279 4280
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
4281

dengyihao's avatar
dengyihao 已提交
4282
  pOperator->name = "FillOperator";
4283
  pOperator->blocking = false;
dengyihao's avatar
dengyihao 已提交
4284
  pOperator->status = OP_NOT_OPENED;
4285
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_FILL;
dengyihao's avatar
dengyihao 已提交
4286
  pOperator->pExpr = pExpr;
4287
  pOperator->numOfExprs = numOfCols;
dengyihao's avatar
dengyihao 已提交
4288
  pOperator->info = pInfo;
H
Haojun Liao 已提交
4289

L
Liu Jicong 已提交
4290 4291
  pOperator->fpSet =
      createOperatorFpSet(operatorDummyOpenFn, doFill, NULL, NULL, destroySFillOperatorInfo, NULL, NULL, NULL);
4292
  pOperator->pTaskInfo = pTaskInfo;
4293
  code = appendDownstream(pOperator, &downstream, 1);
4294
  return pOperator;
H
Haojun Liao 已提交
4295

L
Liu Jicong 已提交
4296
_error:
wafwerar's avatar
wafwerar 已提交
4297 4298
  taosMemoryFreeClear(pOperator);
  taosMemoryFreeClear(pInfo);
H
Haojun Liao 已提交
4299
  return NULL;
4300 4301
}

L
Liu Jicong 已提交
4302 4303
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
                                  const char* name) {
H
Haojun Liao 已提交
4304
  SResSchema s = {0};
dengyihao's avatar
dengyihao 已提交
4305 4306 4307 4308
  s.scale = scale;
  s.type = type;
  s.bytes = bytes;
  s.slotId = slotId;
H
Haojun Liao 已提交
4309
  s.precision = precision;
H
Haojun Liao 已提交
4310 4311 4312 4313
  strncpy(s.name, name, tListLen(s.name));

  return s;
}
H
Haojun Liao 已提交
4314

4315
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType) {
H
Haojun Liao 已提交
4316 4317 4318 4319 4320 4321
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
  if (pCol == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

dengyihao's avatar
dengyihao 已提交
4322
  pCol->slotId = slotId;
X
Xiaoyu Wang 已提交
4323 4324 4325 4326
  pCol->colId = colId;
  pCol->bytes = pType->bytes;
  pCol->type = pType->type;
  pCol->scale = pType->scale;
dengyihao's avatar
dengyihao 已提交
4327
  pCol->precision = pType->precision;
H
Haojun Liao 已提交
4328 4329 4330 4331 4332
  pCol->dataBlockId = blockId;

  return pCol;
}

H
Haojun Liao 已提交
4333
SExprInfo* createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, int32_t* numOfExprs) {
H
Haojun Liao 已提交
4334
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
4335 4336 4337 4338
  int32_t numOfGroupKeys = 0;
  if (pGroupKeys != NULL) {
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
  }
H
Haojun Liao 已提交
4339

H
Haojun Liao 已提交
4340
  *numOfExprs = numOfFuncs + numOfGroupKeys;
wafwerar's avatar
wafwerar 已提交
4341
  SExprInfo* pExprs = taosMemoryCalloc(*numOfExprs, sizeof(SExprInfo));
H
Haojun Liao 已提交
4342

L
Liu Jicong 已提交
4343
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
H
Haojun Liao 已提交
4344 4345 4346 4347 4348 4349
    STargetNode* pTargetNode = NULL;
    if (i < numOfFuncs) {
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    } else {
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
    }
H
Haojun Liao 已提交
4350

4351
    SExprInfo* pExp = &pExprs[i];
H
Haojun Liao 已提交
4352

wafwerar's avatar
wafwerar 已提交
4353
    pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
H
Haojun Liao 已提交
4354
    pExp->pExpr->_function.num = 1;
H
Haojun Liao 已提交
4355
    pExp->pExpr->_function.functionId = -1;
H
Haojun Liao 已提交
4356

4357
    int32_t type = nodeType(pTargetNode->pExpr);
H
Haojun Liao 已提交
4358
    // it is a project query, or group by column
4359
    if (type == QUERY_NODE_COLUMN) {
H
Haojun Liao 已提交
4360
      pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
L
Liu Jicong 已提交
4361
      SColumnNode* pColNode = (SColumnNode*)pTargetNode->pExpr;
H
Haojun Liao 已提交
4362

G
Ganlin Zhao 已提交
4363 4364 4365
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;

H
Haojun Liao 已提交
4366
      SDataType* pType = &pColNode->node.resType;
dengyihao's avatar
dengyihao 已提交
4367 4368
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pColNode->colName);
4369
      pExp->base.pParam[0].pCol = createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType);
H
Haojun Liao 已提交
4370
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
4371
    } else if (type == QUERY_NODE_VALUE) {
4372 4373 4374 4375 4376 4377 4378
      pExp->pExpr->nodeType = QUERY_NODE_VALUE;
      SValueNode* pValNode = (SValueNode*)pTargetNode->pExpr;

      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;

      SDataType* pType = &pValNode->node.resType;
dengyihao's avatar
dengyihao 已提交
4379 4380
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pValNode->node.aliasName);
4381
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
X
Xiaoyu Wang 已提交
4382
      nodesValueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
4383
    } else if (type == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
4384
      pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
H
Haojun Liao 已提交
4385 4386 4387
      SFunctionNode* pFuncNode = (SFunctionNode*)pTargetNode->pExpr;

      SDataType* pType = &pFuncNode->node.resType;
dengyihao's avatar
dengyihao 已提交
4388 4389
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pFuncNode->node.aliasName);
H
Haojun Liao 已提交
4390

H
Haojun Liao 已提交
4391
      pExp->pExpr->_function.functionId = pFuncNode->funcId;
H
Haojun Liao 已提交
4392
      pExp->pExpr->_function.pFunctNode = pFuncNode;
4393

dengyihao's avatar
dengyihao 已提交
4394 4395
      strncpy(pExp->pExpr->_function.functionName, pFuncNode->functionName,
              tListLen(pExp->pExpr->_function.functionName));
4396 4397 4398 4399 4400
#if 1
      // todo refactor: add the parameter for tbname function
      if (strcmp(pExp->pExpr->_function.functionName, "tbname") == 0) {
        pFuncNode->pParameterList = nodesMakeList();
        ASSERT(LIST_LENGTH(pFuncNode->pParameterList) == 0);
X
Xiaoyu Wang 已提交
4401 4402
        SValueNode* res = (SValueNode*)nodesMakeNode(QUERY_NODE_VALUE);
        if (NULL == res) {  // todo handle error
4403
        } else {
X
Xiaoyu Wang 已提交
4404
          res->node.resType = (SDataType){.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
4405 4406 4407 4408
          nodesListAppend(pFuncNode->pParameterList, res);
        }
      }
#endif
H
Haojun Liao 已提交
4409 4410

      int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
G
Ganlin Zhao 已提交
4411 4412 4413 4414

      pExp->base.pParam = taosMemoryCalloc(numOfParam, sizeof(SFunctParam));
      pExp->base.numOfParams = numOfParam;

H
Haojun Liao 已提交
4415
      for (int32_t j = 0; j < numOfParam; ++j) {
4416
        SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
G
Ganlin Zhao 已提交
4417
        if (p1->type == QUERY_NODE_COLUMN) {
dengyihao's avatar
dengyihao 已提交
4418
          SColumnNode* pcn = (SColumnNode*)p1;
G
Ganlin Zhao 已提交
4419 4420

          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
4421
          pExp->base.pParam[j].pCol = createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType);
G
Ganlin Zhao 已提交
4422 4423 4424
        } else if (p1->type == QUERY_NODE_VALUE) {
          SValueNode* pvn = (SValueNode*)p1;
          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
X
Xiaoyu Wang 已提交
4425
          nodesValueNodeToVariant(pvn, &pExp->base.pParam[j].param);
G
Ganlin Zhao 已提交
4426
        }
H
Haojun Liao 已提交
4427
      }
4428
    } else if (type == QUERY_NODE_OPERATOR) {
H
Haojun Liao 已提交
4429
      pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
L
Liu Jicong 已提交
4430
      SOperatorNode* pNode = (SOperatorNode*)pTargetNode->pExpr;
4431

G
Ganlin Zhao 已提交
4432 4433 4434
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;

4435
      SDataType* pType = &pNode->node.resType;
dengyihao's avatar
dengyihao 已提交
4436 4437
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pNode->node.aliasName);
4438 4439 4440
      pExp->pExpr->_optrRoot.pRootNode = pTargetNode->pExpr;
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
4441 4442 4443
    }
  }

H
Haojun Liao 已提交
4444
  return pExprs;
H
Haojun Liao 已提交
4445 4446
}

D
dapan1121 已提交
4447
static SExecTaskInfo* createExecTaskInfo(uint64_t queryId, uint64_t taskId, EOPTR_EXEC_MODEL model, char* dbFName) {
wafwerar's avatar
wafwerar 已提交
4448
  SExecTaskInfo* pTaskInfo = taosMemoryCalloc(1, sizeof(SExecTaskInfo));
4449
  setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
4450

D
dapan1121 已提交
4451
  pTaskInfo->schemaVer.dbname = strdup(dbFName);
4452
  pTaskInfo->cost.created = taosGetTimestampMs();
H
Haojun Liao 已提交
4453
  pTaskInfo->id.queryId = queryId;
dengyihao's avatar
dengyihao 已提交
4454
  pTaskInfo->execModel = model;
H
Haojun Liao 已提交
4455

wafwerar's avatar
wafwerar 已提交
4456
  char* p = taosMemoryCalloc(1, 128);
L
Liu Jicong 已提交
4457
  snprintf(p, 128, "TID:0x%" PRIx64 " QID:0x%" PRIx64, taskId, queryId);
H
Haojun Liao 已提交
4458
  pTaskInfo->id.str = p;
H
Haojun Liao 已提交
4459

4460 4461
  return pTaskInfo;
}
H
Haojun Liao 已提交
4462

L
Liu Jicong 已提交
4463
static tsdbReaderT doCreateDataReader(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle,
dengyihao's avatar
dengyihao 已提交
4464 4465
                                      STableListInfo* pTableGroupInfo, uint64_t queryId, uint64_t taskId,
                                      SNode* pTagCond);
H
Haojun Liao 已提交
4466

dengyihao's avatar
dengyihao 已提交
4467 4468
static int32_t getTableList(void* metaHandle, int32_t tableType, uint64_t tableUid, STableListInfo* pListInfo,
                            SNode* pTagCond);
wmmhello's avatar
wmmhello 已提交
4469
static SArray* extractTableIdList(const STableListInfo* pTableGroupInfo);
H
Haojun Liao 已提交
4470
static SArray* extractColumnInfo(SNodeList* pNodeList);
4471

4472
static SArray* createSortInfo(SNodeList* pNodeList);
4473
static SArray* extractPartitionColInfo(SNodeList* pNodeList);
4474

X
Xiaoyu Wang 已提交
4475
void extractTableSchemaVersion(SReadHandle* pHandle, uint64_t uid, SExecTaskInfo* pTaskInfo) {
4476 4477 4478 4479 4480 4481 4482
  SMetaReader mr = {0};
  metaReaderInit(&mr, pHandle->meta, 0);
  metaGetTableEntryByUid(&mr, uid);

  pTaskInfo->schemaVer.tablename = strdup(mr.me.name);

  if (mr.me.type == TSDB_SUPER_TABLE) {
4483 4484
    pTaskInfo->schemaVer.sversion = mr.me.stbEntry.schemaRow.version;
    pTaskInfo->schemaVer.tversion = mr.me.stbEntry.schemaTag.version;
4485 4486 4487
  } else if (mr.me.type == TSDB_CHILD_TABLE) {
    tb_uid_t suid = mr.me.ctbEntry.suid;
    metaGetTableEntryByUid(&mr, suid);
4488 4489
    pTaskInfo->schemaVer.sversion = mr.me.stbEntry.schemaRow.version;
    pTaskInfo->schemaVer.tversion = mr.me.stbEntry.schemaTag.version;
4490
  } else {
4491
    pTaskInfo->schemaVer.sversion = mr.me.ntbEntry.schemaRow.version;
4492
  }
4493 4494

  metaReaderClear(&mr);
4495 4496
}

H
Haojun Liao 已提交
4497
SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, SReadHandle* pHandle,
wmmhello's avatar
wmmhello 已提交
4498
                                  uint64_t queryId, uint64_t taskId, STableListInfo* pTableListInfo, SNode* pTagCond) {
4499 4500
  int32_t type = nodeType(pPhyNode);

X
Xiaoyu Wang 已提交
4501
  if (pPhyNode->pChildren == NULL || LIST_LENGTH(pPhyNode->pChildren) == 0) {
H
Haojun Liao 已提交
4502
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == type) {
dengyihao's avatar
dengyihao 已提交
4503
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
4504

dengyihao's avatar
dengyihao 已提交
4505
      tsdbReaderT pDataReader =
dengyihao's avatar
dengyihao 已提交
4506
          doCreateDataReader(pTableScanNode, pHandle, pTableListInfo, (uint64_t)queryId, taskId, pTagCond);
4507
      if (pDataReader == NULL && terrno != 0) {
4508 4509
        return NULL;
      }
wmmhello's avatar
wmmhello 已提交
4510
      SArray* groupKyes = extractPartitionColInfo(pTableScanNode->pPartitionKeys);
4511
      extractTableSchemaVersion(pHandle, pTableScanNode->scan.uid, pTaskInfo);
X
Xiaoyu Wang 已提交
4512 4513
      SOperatorInfo* pOperator =
          createTableScanOperatorInfo(pTableScanNode, pDataReader, pHandle, groupKyes, pTaskInfo);
4514

4515 4516
      STableScanInfo* pScanInfo = pOperator->info;
      pTaskInfo->cost.pRecoder = &pScanInfo->readRecorder;
4517

4518
      return pOperator;
H
Haojun Liao 已提交
4519
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == type) {
4520
      return createExchangeOperatorInfo(pHandle->pMsgCb->clientRpc, (SExchangePhysiNode*)pPhyNode, pTaskInfo);
H
Haojun Liao 已提交
4521
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == type) {
X
Xiaoyu Wang 已提交
4522
      SScanPhysiNode*      pScanPhyNode = (SScanPhysiNode*)pPhyNode;  // simple child table.
5
54liuyao 已提交
4523
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
4524
      STimeWindowAggSupp   twSup = {
X
Xiaoyu Wang 已提交
4525
            .waterMark = pTableScanNode->watermark, .calTrigger = pTableScanNode->triggerType, .maxTs = INT64_MIN};
5
54liuyao 已提交
4526 4527
      tsdbReaderT pDataReader = NULL;
      if (pHandle->vnode) {
wmmhello's avatar
wmmhello 已提交
4528
        pDataReader = doCreateDataReader(pTableScanNode, pHandle, pTableListInfo, (uint64_t)queryId, taskId, pTagCond);
wmmhello's avatar
wmmhello 已提交
4529 4530
      } else {
        getTableList(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableListInfo, pTagCond);
5
54liuyao 已提交
4531
      }
4532

5
54liuyao 已提交
4533
      if (pDataReader == NULL && terrno != 0) {
4534
        qDebug("%s pDataReader is NULL", GET_TASKID(pTaskInfo));
5
54liuyao 已提交
4535 4536
        // return NULL;
      } else {
4537
        qDebug("%s pDataReader is not NULL", GET_TASKID(pTaskInfo));
5
54liuyao 已提交
4538
      }
X
Xiaoyu Wang 已提交
4539
      SArray* tableIdList = extractTableIdList(pTableListInfo);
4540

X
Xiaoyu Wang 已提交
4541 4542
      SOperatorInfo* pOperator =
          createStreamScanOperatorInfo(pDataReader, pHandle, tableIdList, pTableScanNode, pTaskInfo, &twSup);
4543

4544
      taosArrayDestroy(tableIdList);
H
Haojun Liao 已提交
4545
      return pOperator;
H
Haojun Liao 已提交
4546
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == type) {
L
Liu Jicong 已提交
4547
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
4548
      return createSysTableScanOperatorInfo(pHandle, pSysScanPhyNode, pTaskInfo);
4549
    } else if (QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN == type) {
X
Xiaoyu Wang 已提交
4550
      STagScanPhysiNode* pScanPhyNode = (STagScanPhysiNode*)pPhyNode;
4551

wmmhello's avatar
wmmhello 已提交
4552
      int32_t code = getTableList(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableListInfo, pTagCond);
4553 4554 4555 4556
      if (code != TSDB_CODE_SUCCESS) {
        return NULL;
      }

4557
      return createTagScanOperatorInfo(pHandle, pScanPhyNode, pTableListInfo, pTaskInfo);
H
Haojun Liao 已提交
4558 4559
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
4560 4561 4562
    }
  }

4563 4564
  int32_t num = 0;
  size_t  size = LIST_LENGTH(pPhyNode->pChildren);
H
Haojun Liao 已提交
4565

4566
  SOperatorInfo** ops = taosMemoryCalloc(size, POINTER_BYTES);
dengyihao's avatar
dengyihao 已提交
4567
  for (int32_t i = 0; i < size; ++i) {
4568
    SPhysiNode* pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
wmmhello's avatar
wmmhello 已提交
4569
    ops[i] = createOperatorTree(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableListInfo, pTagCond);
4570 4571 4572
    if (ops[i] == NULL) {
      return NULL;
    }
4573
  }
H
Haojun Liao 已提交
4574

4575
  SOperatorInfo* pOptr = NULL;
H
Haojun Liao 已提交
4576
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
dengyihao's avatar
dengyihao 已提交
4577 4578
    SProjectPhysiNode* pProjPhyNode = (SProjectPhysiNode*)pPhyNode;
    SExprInfo*         pExprInfo = createExprInfo(pProjPhyNode->pProjections, NULL, &num);
H
Haojun Liao 已提交
4579

4580
    SSDataBlock* pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
dengyihao's avatar
dengyihao 已提交
4581 4582
    SLimit       limit = {.limit = pProjPhyNode->limit, .offset = pProjPhyNode->offset};
    SLimit       slimit = {.limit = pProjPhyNode->slimit, .offset = pProjPhyNode->soffset};
X
Xiaoyu Wang 已提交
4583 4584
    pOptr = createProjectOperatorInfo(ops[0], pExprInfo, num, pResBlock, &limit, &slimit,
                                      pProjPhyNode->node.pConditions, pTaskInfo);
4585
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_AGG == type) {
H
Haojun Liao 已提交
4586 4587
    SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
    SExprInfo*     pExprInfo = createExprInfo(pAggNode->pAggFuncs, pAggNode->pGroupKeys, &num);
4588
    SSDataBlock*   pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
H
Haojun Liao 已提交
4589

dengyihao's avatar
dengyihao 已提交
4590
    int32_t    numOfScalarExpr = 0;
4591 4592 4593 4594 4595
    SExprInfo* pScalarExprInfo = NULL;
    if (pAggNode->pExprs != NULL) {
      pScalarExprInfo = createExprInfo(pAggNode->pExprs, NULL, &numOfScalarExpr);
    }

H
Haojun Liao 已提交
4596 4597
    if (pAggNode->pGroupKeys != NULL) {
      SArray* pColList = extractColumnInfo(pAggNode->pGroupKeys);
dengyihao's avatar
dengyihao 已提交
4598
      pOptr = createGroupOperatorInfo(ops[0], pExprInfo, num, pResBlock, pColList, pAggNode->node.pConditions,
wmmhello's avatar
wmmhello 已提交
4599
                                      pScalarExprInfo, numOfScalarExpr, pTaskInfo);
H
Haojun Liao 已提交
4600
    } else {
dengyihao's avatar
dengyihao 已提交
4601 4602
      pOptr =
          createAggregateOperatorInfo(ops[0], pExprInfo, num, pResBlock, pScalarExprInfo, numOfScalarExpr, pTaskInfo);
H
Haojun Liao 已提交
4603
    }
X
Xiaoyu Wang 已提交
4604
  } else if (QUERY_NODE_PHYSICAL_PLAN_HASH_INTERVAL == type || QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL == type) {
H
Haojun Liao 已提交
4605
    SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
4606

H
Haojun Liao 已提交
4607
    SExprInfo*   pExprInfo = createExprInfo(pIntervalPhyNode->window.pFuncs, NULL, &num);
4608
    SSDataBlock* pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
H
Haojun Liao 已提交
4609

dengyihao's avatar
dengyihao 已提交
4610 4611 4612 4613 4614 4615
    SInterval interval = {.interval = pIntervalPhyNode->interval,
                          .sliding = pIntervalPhyNode->sliding,
                          .intervalUnit = pIntervalPhyNode->intervalUnit,
                          .slidingUnit = pIntervalPhyNode->slidingUnit,
                          .offset = pIntervalPhyNode->offset,
                          .precision = ((SColumnNode*)pIntervalPhyNode->window.pTspk)->node.resType.precision};
H
Haojun Liao 已提交
4616

X
Xiaoyu Wang 已提交
4617 4618 4619 4620 4621 4622
    STimeWindowAggSupp as = {
        .waterMark = pIntervalPhyNode->window.watermark,
        .calTrigger = pIntervalPhyNode->window.triggerType,
        .maxTs = INT64_MIN,
        .winMap = NULL,
    };
5
54liuyao 已提交
4623 4624 4625 4626 4627 4628
    if (isSmaStream(pIntervalPhyNode->window.triggerType)) {
      if (FLT_LESS(pIntervalPhyNode->window.filesFactor, 1.000000)) {
        as.calTrigger = STREAM_TRIGGER_AT_ONCE_SMA;
      } else {
        _hash_fn_t hashFn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_TIMESTAMP);
        as.winMap = taosHashInit(64, hashFn, true, HASH_NO_LOCK);
X
Xiaoyu Wang 已提交
4629
        as.waterMark = getSmaWaterMark(interval.interval, pIntervalPhyNode->window.filesFactor);
5
54liuyao 已提交
4630 4631 4632
        as.calTrigger = STREAM_TRIGGER_WINDOW_CLOSE_SMA;
      }
    }
4633

4634
    int32_t tsSlotId = ((SColumnNode*)pIntervalPhyNode->window.pTspk)->slotId;
wmmhello's avatar
wmmhello 已提交
4635
    pOptr = createIntervalOperatorInfo(ops[0], pExprInfo, num, pResBlock, &interval, tsSlotId, &as, pTaskInfo);
4636

5
54liuyao 已提交
4637 4638 4639 4640 4641 4642
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SEMI_INTERVAL == type) {
    int32_t children = 8;
    pOptr = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children);
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_FINAL_INTERVAL == type) {
    int32_t children = 0;
    pOptr = createStreamFinalIntervalOperatorInfo(ops[0], pPhyNode, pTaskInfo, children);
H
Haojun Liao 已提交
4643
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) {
H
Haojun Liao 已提交
4644
    SSortPhysiNode* pSortPhyNode = (SSortPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
4645

4646 4647 4648
    SDataBlockDescNode* pDescNode = pPhyNode->pOutputDataBlockDesc;

    SSDataBlock* pResBlock = createResDataBlock(pDescNode);
4649
    SArray*      info = createSortInfo(pSortPhyNode->pSortKeys);
4650 4651

    int32_t    numOfCols = 0;
4652 4653 4654
    SExprInfo* pExprInfo = createExprInfo(pSortPhyNode->pExprs, NULL, &numOfCols);

    int32_t numOfOutputCols = 0;
X
Xiaoyu Wang 已提交
4655 4656
    SArray* pColList =
        extractColMatchInfo(pSortPhyNode->pTargets, pDescNode, &numOfOutputCols, pTaskInfo, COL_MATCH_FROM_SLOT_ID);
4657

4658
    pOptr = createSortOperatorInfo(ops[0], pResBlock, info, pExprInfo, numOfCols, pColList, pTaskInfo);
X
Xiaoyu Wang 已提交
4659
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE == type) {
4660 4661 4662
    SMergePhysiNode* pMergePhyNode = (SMergePhysiNode*)pPhyNode;

    SDataBlockDescNode* pDescNode = pPhyNode->pOutputDataBlockDesc;
X
Xiaoyu Wang 已提交
4663
    SSDataBlock*        pResBlock = createResDataBlock(pDescNode);
4664

X
Xiaoyu Wang 已提交
4665
    SArray* sortInfo = createSortInfo(pMergePhyNode->pMergeKeys);
4666
    int32_t numOfOutputCols = 0;
X
Xiaoyu Wang 已提交
4667 4668
    SArray* pColList =
        extractColMatchInfo(pMergePhyNode->pTargets, pDescNode, &numOfOutputCols, pTaskInfo, COL_MATCH_FROM_SLOT_ID);
4669

4670
    pOptr = createMultiwaySortMergeOperatorInfo(ops, size, pResBlock, sortInfo, pColList, pTaskInfo);
4671
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_SESSION == type) {
H
Haojun Liao 已提交
4672 4673
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;

X
Xiaoyu Wang 已提交
4674 4675
    STimeWindowAggSupp as = {.waterMark = pSessionNode->window.watermark,
                             .calTrigger = pSessionNode->window.triggerType};
4676

H
Haojun Liao 已提交
4677
    SExprInfo*   pExprInfo = createExprInfo(pSessionNode->window.pFuncs, NULL, &num);
4678
    SSDataBlock* pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
4679 4680
    int32_t      tsSlotId = ((SColumnNode*)pSessionNode->window.pTspk)->slotId;

X
Xiaoyu Wang 已提交
4681 4682
    pOptr =
        createSessionAggOperatorInfo(ops[0], pExprInfo, num, pResBlock, pSessionNode->gap, tsSlotId, &as, pTaskInfo);
4683
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SESSION == type) {
5
54liuyao 已提交
4684 4685 4686 4687 4688 4689 4690 4691 4692
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;

    STimeWindowAggSupp as = {.waterMark = pSessionNode->window.watermark,
                             .calTrigger = pSessionNode->window.triggerType};

    SExprInfo*   pExprInfo = createExprInfo(pSessionNode->window.pFuncs, NULL, &num);
    SSDataBlock* pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
    int32_t      tsSlotId = ((SColumnNode*)pSessionNode->window.pTspk)->slotId;

4693 4694
    pOptr = createStreamSessionAggOperatorInfo(ops[0], pExprInfo, num, pResBlock, pSessionNode->gap, tsSlotId, &as,
                                               pTaskInfo);
5
54liuyao 已提交
4695

H
Haojun Liao 已提交
4696
  } else if (QUERY_NODE_PHYSICAL_PLAN_PARTITION == type) {
dengyihao's avatar
dengyihao 已提交
4697 4698 4699
    SPartitionPhysiNode* pPartNode = (SPartitionPhysiNode*)pPhyNode;
    SArray*              pColList = extractPartitionColInfo(pPartNode->pPartitionKeys);
    SSDataBlock*         pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
4700 4701

    SExprInfo* pExprInfo = createExprInfo(pPartNode->pTargets, NULL, &num);
wmmhello's avatar
wmmhello 已提交
4702
    pOptr = createPartitionOperatorInfo(ops[0], pExprInfo, num, pResBlock, pColList, pTaskInfo);
4703
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_STATE == type) {
dengyihao's avatar
dengyihao 已提交
4704
    SStateWinodwPhysiNode* pStateNode = (SStateWinodwPhysiNode*)pPhyNode;
4705

4706 4707
    STimeWindowAggSupp as = {.waterMark = pStateNode->window.watermark, .calTrigger = pStateNode->window.triggerType};

dengyihao's avatar
dengyihao 已提交
4708
    SExprInfo*   pExprInfo = createExprInfo(pStateNode->window.pFuncs, NULL, &num);
4709
    SSDataBlock* pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
4710 4711
    int32_t      tsSlotId = ((SColumnNode*)pStateNode->window.pTspk)->slotId;

4712
    SColumnNode* pColNode = (SColumnNode*)((STargetNode*)pStateNode->pStateKey)->pExpr;
X
Xiaoyu Wang 已提交
4713
    SColumn      col = extractColumnFromColumnNode(pColNode);
4714
    pOptr = createStatewindowOperatorInfo(ops[0], pExprInfo, num, pResBlock, &as, tsSlotId, &col, pTaskInfo);
4715
  } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_STATE == type) {
5
54liuyao 已提交
4716
    pOptr = createStreamStateAggOperatorInfo(ops[0], pPhyNode, pTaskInfo);
4717
  } else if (QUERY_NODE_PHYSICAL_PLAN_MERGE_JOIN == type) {
dengyihao's avatar
dengyihao 已提交
4718 4719
    SJoinPhysiNode* pJoinNode = (SJoinPhysiNode*)pPhyNode;
    SSDataBlock*    pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
4720 4721

    SExprInfo* pExprInfo = createExprInfo(pJoinNode->pTargets, NULL, &num);
H
Haojun Liao 已提交
4722
    pOptr = createMergeJoinOperatorInfo(ops, size, pExprInfo, num, pResBlock, pJoinNode->pOnConditions, pTaskInfo);
4723 4724
  } else if (QUERY_NODE_PHYSICAL_PLAN_FILL == type) {
    SFillPhysiNode* pFillNode = (SFillPhysiNode*)pPhyNode;
X
Xiaoyu Wang 已提交
4725 4726
    SSDataBlock*    pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
    SExprInfo*      pExprInfo = createExprInfo(pFillNode->pTargets, NULL, &num);
4727

4728
    SInterval* pInterval = &((SIntervalAggOperatorInfo*)ops[0]->info)->interval;
X
Xiaoyu Wang 已提交
4729 4730
    pOptr = createFillOperatorInfo(ops[0], pExprInfo, num, pInterval, &pFillNode->timeRange, pResBlock, pFillNode->mode,
                                   (SNodeListNode*)pFillNode->pValues, false, pTaskInfo);
H
Haojun Liao 已提交
4731 4732
  } else if (QUERY_NODE_PHYSICAL_PLAN_INDEF_ROWS_FUNC == type) {
    pOptr = createIndefinitOutputOperatorInfo(ops[0], pPhyNode, pTaskInfo);
H
Haojun Liao 已提交
4733 4734
  } else {
    ASSERT(0);
H
Haojun Liao 已提交
4735
  }
4736 4737 4738

  taosMemoryFree(ops);
  return pOptr;
4739
}
H
Haojun Liao 已提交
4740

4741
int32_t compareTimeWindow(const void* p1, const void* p2, const void* param) {
dengyihao's avatar
dengyihao 已提交
4742 4743 4744
  const SQueryTableDataCond* pCond = param;
  const STimeWindow*         pWin1 = p1;
  const STimeWindow*         pWin2 = p2;
4745 4746 4747 4748 4749 4750 4751 4752
  if (pCond->order == TSDB_ORDER_ASC) {
    return pWin1->skey - pWin2->skey;
  } else if (pCond->order == TSDB_ORDER_DESC) {
    return pWin2->skey - pWin1->skey;
  }
  return 0;
}

4753
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, const STableScanPhysiNode* pTableScanNode) {
4754 4755 4756 4757 4758 4759
  pCond->loadExternalRows = false;

  pCond->order = pTableScanNode->scanSeq[0] > 0 ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
  pCond->numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
  pCond->colList = taosMemoryCalloc(pCond->numOfCols, sizeof(SColumnInfo));
  if (pCond->colList == NULL) {
H
Haojun Liao 已提交
4760
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
4761 4762 4763
    return terrno;
  }

dengyihao's avatar
dengyihao 已提交
4764 4765
  // pCond->twindow = pTableScanNode->scanRange;
  // TODO: get it from stable scan node
4766 4767 4768
  pCond->numOfTWindows = 1;
  pCond->twindows = taosMemoryCalloc(pCond->numOfTWindows, sizeof(STimeWindow));
  pCond->twindows[0] = pTableScanNode->scanRange;
H
more  
Hongze Cheng 已提交
4769
  pCond->suid = pTableScanNode->scan.suid;
4770 4771

#if 1
L
Liu Jicong 已提交
4772
  // todo work around a problem, remove it later
4773 4774 4775 4776 4777
  for (int32_t i = 0; i < pCond->numOfTWindows; ++i) {
    if ((pCond->order == TSDB_ORDER_ASC && pCond->twindows[i].skey > pCond->twindows[i].ekey) ||
        (pCond->order == TSDB_ORDER_DESC && pCond->twindows[i].skey < pCond->twindows[i].ekey)) {
      TSWAP(pCond->twindows[i].skey, pCond->twindows[i].ekey);
    }
H
Haojun Liao 已提交
4778
  }
4779
#endif
H
Haojun Liao 已提交
4780

4781 4782 4783 4784 4785 4786
  for (int32_t i = 0; i < pCond->numOfTWindows; ++i) {
    if ((pCond->order == TSDB_ORDER_ASC && pCond->twindows[i].skey > pCond->twindows[i].ekey) ||
        (pCond->order == TSDB_ORDER_DESC && pCond->twindows[i].skey < pCond->twindows[i].ekey)) {
      TSWAP(pCond->twindows[i].skey, pCond->twindows[i].ekey);
    }
  }
dengyihao's avatar
dengyihao 已提交
4787
  taosqsort(pCond->twindows, pCond->numOfTWindows, sizeof(STimeWindow), pCond, compareTimeWindow);
4788

4789 4790
  pCond->type = BLOCK_LOAD_OFFSET_SEQ_ORDER;
  //  pCond->type = pTableScanNode->scanFlag;
H
Haojun Liao 已提交
4791

H
Haojun Liao 已提交
4792
  int32_t j = 0;
4793
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
H
Haojun Liao 已提交
4794 4795
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
4796 4797 4798
    if (pColNode->colType == COLUMN_TYPE_TAG) {
      continue;
    }
H
Haojun Liao 已提交
4799

4800 4801 4802
    pCond->colList[j].type = pColNode->node.resType.type;
    pCond->colList[j].bytes = pColNode->node.resType.bytes;
    pCond->colList[j].colId = pColNode->colId;
H
Haojun Liao 已提交
4803
    j += 1;
H
Haojun Liao 已提交
4804 4805
  }

4806 4807
  pCond->numOfCols = j;
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
4808
}
H
Haojun Liao 已提交
4809

4810 4811
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
  SColumn c = {0};
X
Xiaoyu Wang 已提交
4812 4813 4814 4815 4816
  c.slotId = pColNode->slotId;
  c.colId = pColNode->colId;
  c.type = pColNode->node.resType.type;
  c.bytes = pColNode->node.resType.bytes;
  c.scale = pColNode->node.resType.scale;
4817 4818 4819 4820
  c.precision = pColNode->node.resType.precision;
  return c;
}

H
Haojun Liao 已提交
4821
SArray* extractColumnInfo(SNodeList* pNodeList) {
L
Liu Jicong 已提交
4822
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
4823 4824 4825 4826 4827 4828
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
4829 4830
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
H
Haojun Liao 已提交
4831

4832 4833 4834
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;

4835
      SColumn c = extractColumnFromColumnNode(pColNode);
4836 4837
      taosArrayPush(pList, &c);
    } else if (nodeType(pNode->pExpr) == QUERY_NODE_VALUE) {
L
Liu Jicong 已提交
4838 4839
      SValueNode* pValNode = (SValueNode*)pNode->pExpr;
      SColumn     c = {0};
4840
      c.slotId = pNode->slotId;
L
Liu Jicong 已提交
4841 4842 4843 4844
      c.colId = pNode->slotId;
      c.type = pValNode->node.type;
      c.bytes = pValNode->node.resType.bytes;
      c.scale = pValNode->node.resType.scale;
4845 4846 4847 4848
      c.precision = pValNode->node.resType.precision;

      taosArrayPush(pList, &c);
    }
H
Haojun Liao 已提交
4849 4850 4851 4852 4853
  }

  return pList;
}

4854 4855 4856 4857 4858 4859 4860 4861 4862 4863 4864 4865 4866 4867
SArray* extractPartitionColInfo(SNodeList* pNodeList) {
  size_t  numOfCols = LIST_LENGTH(pNodeList);
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

  for (int32_t i = 0; i < numOfCols; ++i) {
    SColumnNode* pColNode = (SColumnNode*)nodesListGetNode(pNodeList, i);

    // todo extract method
    SColumn c = {0};
    c.slotId = pColNode->slotId;
dengyihao's avatar
dengyihao 已提交
4868 4869 4870
    c.colId = pColNode->colId;
    c.type = pColNode->node.resType.type;
    c.bytes = pColNode->node.resType.bytes;
4871 4872 4873 4874 4875 4876 4877 4878 4879
    c.precision = pColNode->node.resType.precision;
    c.scale = pColNode->node.resType.scale;

    taosArrayPush(pList, &c);
  }

  return pList;
}

4880
SArray* createSortInfo(SNodeList* pNodeList) {
L
Liu Jicong 已提交
4881
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
4882 4883 4884 4885 4886 4887
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return pList;
  }

L
Liu Jicong 已提交
4888
  for (int32_t i = 0; i < numOfCols; ++i) {
4889
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
L
Liu Jicong 已提交
4890 4891
    SBlockOrderInfo   bi = {0};
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
H
Haojun Liao 已提交
4892 4893 4894
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);

    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
4895
    bi.slotId = pColNode->slotId;
H
Haojun Liao 已提交
4896 4897 4898 4899 4900 4901
    taosArrayPush(pList, &bi);
  }

  return pList;
}

X
Xiaoyu Wang 已提交
4902
SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
4903
                            SExecTaskInfo* pTaskInfo, int32_t type) {
L
Liu Jicong 已提交
4904
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
4905 4906 4907 4908 4909 4910
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
4911 4912 4913
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
4914 4915

    SColMatchInfo c = {0};
X
Xiaoyu Wang 已提交
4916 4917 4918 4919
    c.output = true;
    c.colId = pColNode->colId;
    c.srcSlotId = pColNode->slotId;
    c.matchType = type;
H
Haojun Liao 已提交
4920 4921 4922 4923
    c.targetSlotId = pNode->slotId;
    taosArrayPush(pList, &c);
  }

H
Haojun Liao 已提交
4924 4925
  *numOfOutputCols = 0;
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
L
Liu Jicong 已提交
4926 4927
  for (int32_t i = 0; i < num; ++i) {
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
4928

4929
    // todo: add reserve flag check
4930 4931
    // it is a column reserved for the arithmetic expression calculation
    if (pNode->slotId >= numOfCols) {
4932 4933 4934 4935
      (*numOfOutputCols) += 1;
      continue;
    }

H
Haojun Liao 已提交
4936
    SColMatchInfo* info = taosArrayGet(pList, pNode->slotId);
H
Haojun Liao 已提交
4937 4938 4939 4940 4941
    if (pNode->output) {
      (*numOfOutputCols) += 1;
    } else {
      info->output = false;
    }
H
Haojun Liao 已提交
4942 4943
  }

H
Haojun Liao 已提交
4944 4945 4946
  return pList;
}

dengyihao's avatar
dengyihao 已提交
4947 4948
int32_t getTableList(void* metaHandle, int32_t tableType, uint64_t tableUid, STableListInfo* pListInfo,
                     SNode* pTagCond) {
wmmhello's avatar
wmmhello 已提交
4949
  int32_t code = TSDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
4950
  pListInfo->pTableList = taosArrayInit(8, sizeof(STableKeyInfo));
wmmhello's avatar
wmmhello 已提交
4951

wmmhello's avatar
wmmhello 已提交
4952
  if (tableType == TSDB_SUPER_TABLE) {
dengyihao's avatar
dengyihao 已提交
4953
    if (pTagCond) {
dengyihao's avatar
dengyihao 已提交
4954 4955
      SIndexMetaArg metaArg = {
          .metaEx = metaHandle, .idx = tsdbGetIdx(metaHandle), .ivtIdx = tsdbGetIvtIdx(metaHandle), .suid = tableUid};
dengyihao's avatar
dengyihao 已提交
4956

wmmhello's avatar
wmmhello 已提交
4957
      SArray* res = taosArrayInit(8, sizeof(uint64_t));
dengyihao's avatar
dengyihao 已提交
4958
      code = doFilterTag(pTagCond, &metaArg, res);
wmmhello's avatar
wmmhello 已提交
4959
      if (code != TSDB_CODE_SUCCESS) {
dengyihao's avatar
dengyihao 已提交
4960
        qError("failed  to  get tableIds, reason: %s, suid: %" PRIu64 "", tstrerror(code), tableUid);
wmmhello's avatar
wmmhello 已提交
4961
        taosArrayDestroy(res);
wmmhello's avatar
wmmhello 已提交
4962
        terrno = code;
wmmhello's avatar
wmmhello 已提交
4963
        return code;
dengyihao's avatar
dengyihao 已提交
4964
      } else {
dengyihao's avatar
dengyihao 已提交
4965
        qDebug("sucess to  get tableIds, size: %d, suid: %" PRIu64 "", (int)taosArrayGetSize(res), tableUid);
wmmhello's avatar
wmmhello 已提交
4966
      }
dengyihao's avatar
dengyihao 已提交
4967
      for (int i = 0; i < taosArrayGetSize(res); i++) {
wmmhello's avatar
wmmhello 已提交
4968 4969 4970 4971
        STableKeyInfo info = {.lastKey = TSKEY_INITIAL_VAL, .uid = *(uint64_t*)taosArrayGet(res, i)};
        taosArrayPush(pListInfo->pTableList, &info);
      }
      taosArrayDestroy(res);
dengyihao's avatar
dengyihao 已提交
4972
    } else {
wmmhello's avatar
wmmhello 已提交
4973
      code = tsdbGetAllTableList(metaHandle, tableUid, pListInfo->pTableList);
wmmhello's avatar
wmmhello 已提交
4974
    }
H
Haojun Liao 已提交
4975
  } else {  // Create one table group.
wmmhello's avatar
wmmhello 已提交
4976 4977
    STableKeyInfo info = {.lastKey = 0, .uid = tableUid};
    taosArrayPush(pListInfo->pTableList, &info);
4978 4979 4980 4981
  }

  return code;
}
H
Haojun Liao 已提交
4982

wmmhello's avatar
wmmhello 已提交
4983
SArray* extractTableIdList(const STableListInfo* pTableGroupInfo) {
4984 4985
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));

wmmhello's avatar
wmmhello 已提交
4986 4987 4988 4989
  // Transfer the Array of STableKeyInfo into uid list.
  for (int32_t i = 0; i < taosArrayGetSize(pTableGroupInfo->pTableList); ++i) {
    STableKeyInfo* pkeyInfo = taosArrayGet(pTableGroupInfo->pTableList, i);
    taosArrayPush(tableIdList, &pkeyInfo->uid);
4990 4991 4992 4993 4994
  }

  return tableIdList;
}

L
Liu Jicong 已提交
4995
tsdbReaderT doCreateDataReader(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle,
wmmhello's avatar
wmmhello 已提交
4996
                               STableListInfo* pTableListInfo, uint64_t queryId, uint64_t taskId, SNode* pTagCond) {
dengyihao's avatar
dengyihao 已提交
4997 4998
  int32_t code =
      getTableList(pHandle->meta, pTableScanNode->scan.tableType, pTableScanNode->scan.uid, pTableListInfo, pTagCond);
wmmhello's avatar
wmmhello 已提交
4999 5000 5001 5002 5003 5004 5005 5006 5007 5008
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

  if (taosArrayGetSize(pTableListInfo->pTableList) == 0) {
    code = 0;
    qDebug("no table qualified for query, TID:0x%" PRIx64 ", QID:0x%" PRIx64, taskId, queryId);
    goto _error;
  }

5009
  SQueryTableDataCond cond = {0};
wmmhello's avatar
wmmhello 已提交
5010
  code = initQueryTableDataCond(&cond, pTableScanNode);
5011
  if (code != TSDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
5012
    goto _error;
X
Xiaoyu Wang 已提交
5013
  }
5014

wmmhello's avatar
wmmhello 已提交
5015
  return tsdbQueryTables(pHandle->vnode, &cond, pTableListInfo, queryId, taskId);
wmmhello's avatar
wmmhello 已提交
5016 5017 5018 5019

_error:
  terrno = code;
  return NULL;
H
Haojun Liao 已提交
5020 5021
}

5022
int32_t encodeOperator(SOperatorInfo* ops, char** result, int32_t* length) {
wmmhello's avatar
wmmhello 已提交
5023
  int32_t code = TDB_CODE_SUCCESS;
5024
  char*   pCurrent = NULL;
wmmhello's avatar
wmmhello 已提交
5025
  int32_t currLength = 0;
5026 5027
  if (ops->fpSet.encodeResultRow) {
    if (result == NULL || length == NULL) {
wmmhello's avatar
wmmhello 已提交
5028 5029 5030
      return TSDB_CODE_TSC_INVALID_INPUT;
    }
    code = ops->fpSet.encodeResultRow(ops, &pCurrent, &currLength);
wmmhello's avatar
wmmhello 已提交
5031

5032 5033
    if (code != TDB_CODE_SUCCESS) {
      if (*result != NULL) {
wmmhello's avatar
wmmhello 已提交
5034 5035 5036 5037 5038
        taosMemoryFree(*result);
        *result = NULL;
      }
      return code;
    }
wmmhello's avatar
wmmhello 已提交
5039

5040
    if (*result == NULL) {
wmmhello's avatar
wmmhello 已提交
5041
      *result = (char*)taosMemoryCalloc(1, currLength + sizeof(int32_t));
wmmhello's avatar
wmmhello 已提交
5042 5043 5044 5045 5046 5047
      if (*result == NULL) {
        taosMemoryFree(pCurrent);
        return TSDB_CODE_OUT_OF_MEMORY;
      }
      memcpy(*result + sizeof(int32_t), pCurrent, currLength);
      *(int32_t*)(*result) = currLength + sizeof(int32_t);
5048
    } else {
wmmhello's avatar
wmmhello 已提交
5049
      int32_t sizePre = *(int32_t*)(*result);
5050
      char*   tmp = (char*)taosMemoryRealloc(*result, sizePre + currLength);
wmmhello's avatar
wmmhello 已提交
5051 5052 5053 5054 5055 5056 5057 5058 5059 5060 5061 5062
      if (tmp == NULL) {
        taosMemoryFree(pCurrent);
        taosMemoryFree(*result);
        *result = NULL;
        return TSDB_CODE_OUT_OF_MEMORY;
      }
      *result = tmp;
      memcpy(*result + sizePre, pCurrent, currLength);
      *(int32_t*)(*result) += currLength;
    }
    taosMemoryFree(pCurrent);
    *length = *(int32_t*)(*result);
wmmhello's avatar
wmmhello 已提交
5063 5064
  }

wmmhello's avatar
wmmhello 已提交
5065 5066
  for (int32_t i = 0; i < ops->numOfDownstream; ++i) {
    code = encodeOperator(ops->pDownstream[i], result, length);
5067
    if (code != TDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
5068
      return code;
wmmhello's avatar
wmmhello 已提交
5069 5070
    }
  }
wmmhello's avatar
wmmhello 已提交
5071
  return TDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
5072 5073
}

5074
int32_t decodeOperator(SOperatorInfo* ops, char* result, int32_t length) {
wmmhello's avatar
wmmhello 已提交
5075
  int32_t code = TDB_CODE_SUCCESS;
5076 5077
  if (ops->fpSet.decodeResultRow) {
    if (result == NULL) {
wmmhello's avatar
wmmhello 已提交
5078 5079
      return TSDB_CODE_TSC_INVALID_INPUT;
    }
5080 5081 5082
    ASSERT(length == *(int32_t*)result);
    char* data = result + sizeof(int32_t);
    code = ops->fpSet.decodeResultRow(ops, data);
5083
    if (code != TDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
5084 5085
      return code;
    }
wmmhello's avatar
wmmhello 已提交
5086

wmmhello's avatar
wmmhello 已提交
5087
    int32_t totalLength = *(int32_t*)result;
5088 5089
    int32_t dataLength = *(int32_t*)data;

5090
    if (totalLength == dataLength + sizeof(int32_t)) {  // the last data
wmmhello's avatar
wmmhello 已提交
5091 5092
      result = NULL;
      length = 0;
5093
    } else {
wmmhello's avatar
wmmhello 已提交
5094 5095 5096 5097
      result += dataLength;
      *(int32_t*)(result) = totalLength - dataLength;
      length = totalLength - dataLength;
    }
wmmhello's avatar
wmmhello 已提交
5098 5099
  }

wmmhello's avatar
wmmhello 已提交
5100 5101
  for (int32_t i = 0; i < ops->numOfDownstream; ++i) {
    code = decodeOperator(ops->pDownstream[i], result, length);
5102
    if (code != TDB_CODE_SUCCESS) {
wmmhello's avatar
wmmhello 已提交
5103
      return code;
wmmhello's avatar
wmmhello 已提交
5104 5105
    }
  }
wmmhello's avatar
wmmhello 已提交
5106
  return TDB_CODE_SUCCESS;
wmmhello's avatar
wmmhello 已提交
5107 5108
}

5109
int32_t createDataSinkParam(SDataSinkNode* pNode, void** pParam, qTaskInfo_t* pTaskInfo) {
D
dapan1121 已提交
5110
  SExecTaskInfo* pTask = *(SExecTaskInfo**)pTaskInfo;
5111

D
dapan1121 已提交
5112 5113
  switch (pNode->type) {
    case QUERY_NODE_PHYSICAL_PLAN_DELETE: {
5114
      SDeleterParam* pDeleterParam = taosMemoryCalloc(1, sizeof(SDeleterParam));
D
dapan1121 已提交
5115 5116 5117 5118 5119 5120 5121 5122 5123 5124
      if (NULL == pDeleterParam) {
        return TSDB_CODE_OUT_OF_MEMORY;
      }
      int32_t tbNum = taosArrayGetSize(pTask->tableqinfoList.pTableList);
      pDeleterParam->pUidList = taosArrayInit(tbNum, sizeof(uint64_t));
      if (NULL == pDeleterParam->pUidList) {
        taosMemoryFree(pDeleterParam);
        return TSDB_CODE_OUT_OF_MEMORY;
      }
      for (int32_t i = 0; i < tbNum; ++i) {
5125
        STableKeyInfo* pTable = taosArrayGet(pTask->tableqinfoList.pTableList, i);
D
dapan1121 已提交
5126 5127 5128 5129 5130 5131 5132 5133 5134 5135 5136 5137 5138
        taosArrayPush(pDeleterParam->pUidList, &pTable->uid);
      }

      *pParam = pDeleterParam;
      break;
    }
    default:
      break;
  }

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
5139 5140
int32_t createExecTaskInfoImpl(SSubplan* pPlan, SExecTaskInfo** pTaskInfo, SReadHandle* pHandle, uint64_t taskId,
                               EOPTR_EXEC_MODEL model) {
H
Haojun Liao 已提交
5141 5142
  uint64_t queryId = pPlan->id.queryId;

H
Haojun Liao 已提交
5143
  int32_t code = TSDB_CODE_SUCCESS;
D
dapan1121 已提交
5144
  *pTaskInfo = createExecTaskInfo(queryId, taskId, model, pPlan->dbFName);
H
Haojun Liao 已提交
5145 5146 5147 5148
  if (*pTaskInfo == NULL) {
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    goto _complete;
  }
H
Haojun Liao 已提交
5149

dengyihao's avatar
dengyihao 已提交
5150
  (*pTaskInfo)->pRoot = createOperatorTree(pPlan->pNode, *pTaskInfo, pHandle, queryId, taskId,
dengyihao's avatar
dengyihao 已提交
5151
                                           &(*pTaskInfo)->tableqinfoList, pPlan->pTagCond);
D
dapan1121 已提交
5152
  if (NULL == (*pTaskInfo)->pRoot) {
L
Liu Jicong 已提交
5153
    code = terrno;
D
dapan1121 已提交
5154 5155
    goto _complete;
  }
H
Haojun Liao 已提交
5156

5157
  if ((*pTaskInfo)->pRoot == NULL) {
H
Haojun Liao 已提交
5158
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
H
Haojun Liao 已提交
5159
    goto _complete;
5160 5161
  }

H
Haojun Liao 已提交
5162 5163
  return code;

H
Haojun Liao 已提交
5164
_complete:
wafwerar's avatar
wafwerar 已提交
5165
  taosMemoryFreeClear(*pTaskInfo);
H
Haojun Liao 已提交
5166 5167 5168

  terrno = code;
  return code;
H
Haojun Liao 已提交
5169 5170
}

wmmhello's avatar
wmmhello 已提交
5171 5172 5173
static void doDestroyTableList(STableListInfo* pTableqinfoList) {
  taosArrayDestroy(pTableqinfoList->pTableList);
  taosHashCleanup(pTableqinfoList->map);
5174

wmmhello's avatar
wmmhello 已提交
5175 5176
  pTableqinfoList->pTableList = NULL;
  pTableqinfoList->map = NULL;
5177 5178
}

L
Liu Jicong 已提交
5179
void doDestroyTask(SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
5180 5181
  qDebug("%s execTask is freed", GET_TASKID(pTaskInfo));

wmmhello's avatar
wmmhello 已提交
5182
  doDestroyTableList(&pTaskInfo->tableqinfoList);
H
Haojun Liao 已提交
5183
  destroyOperatorInfo(pTaskInfo->pRoot);
L
Liu Jicong 已提交
5184 5185
  //  taosArrayDestroy(pTaskInfo->summary.queryProfEvents);
  //  taosHashCleanup(pTaskInfo->summary.operatorProfResults);
5186

5187 5188
  taosMemoryFree(pTaskInfo->schemaVer.dbname);
  taosMemoryFree(pTaskInfo->schemaVer.tablename);
wafwerar's avatar
wafwerar 已提交
5189 5190 5191
  taosMemoryFreeClear(pTaskInfo->sql);
  taosMemoryFreeClear(pTaskInfo->id.str);
  taosMemoryFreeClear(pTaskInfo);
5192 5193 5194 5195 5196 5197 5198 5199 5200 5201 5202 5203
}

static void doSetTagValueToResultBuf(char* output, const char* val, int16_t type, int16_t bytes) {
  if (val == NULL) {
    setNull(output, type, bytes);
    return;
  }

  if (IS_VAR_DATA_TYPE(type)) {
    // Binary data overflows for sort of unknown reasons. Let trim the overflow data
    if (varDataTLen(val) > bytes) {
      int32_t maxLen = bytes - VARSTR_HEADER_SIZE;
L
Liu Jicong 已提交
5204
      int32_t len = (varDataLen(val) > maxLen) ? maxLen : varDataLen(val);
5205 5206 5207 5208 5209 5210 5211 5212 5213 5214 5215 5216
      memcpy(varDataVal(output), varDataVal(val), len);
      varDataSetLen(output, len);
    } else {
      varDataCopy(output, val);
    }
  } else {
    memcpy(output, val, bytes);
  }
}

static int64_t getQuerySupportBufSize(size_t numOfTables) {
  size_t s1 = sizeof(STableQueryInfo);
L
Liu Jicong 已提交
5217 5218
  //  size_t s3 = sizeof(STableCheckInfo);  buffer consumption in tsdb
  return (int64_t)(s1 * 1.5 * numOfTables);
5219 5220 5221 5222 5223 5224 5225
}

int32_t checkForQueryBuf(size_t numOfTables) {
  int64_t t = getQuerySupportBufSize(numOfTables);
  if (tsQueryBufferSizeBytes < 0) {
    return TSDB_CODE_SUCCESS;
  } else if (tsQueryBufferSizeBytes > 0) {
L
Liu Jicong 已提交
5226
    while (1) {
5227 5228 5229 5230 5231 5232 5233 5234 5235 5236 5237 5238 5239 5240 5241 5242 5243 5244 5245 5246 5247 5248 5249 5250 5251 5252
      int64_t s = tsQueryBufferSizeBytes;
      int64_t remain = s - t;
      if (remain >= 0) {
        if (atomic_val_compare_exchange_64(&tsQueryBufferSizeBytes, s, remain) == s) {
          return TSDB_CODE_SUCCESS;
        }
      } else {
        return TSDB_CODE_QRY_NOT_ENOUGH_BUFFER;
      }
    }
  }

  // disable query processing if the value of tsQueryBufferSize is zero.
  return TSDB_CODE_QRY_NOT_ENOUGH_BUFFER;
}

void releaseQueryBuf(size_t numOfTables) {
  if (tsQueryBufferSizeBytes < 0) {
    return;
  }

  int64_t t = getQuerySupportBufSize(numOfTables);

  // restore value is not enough buffer available
  atomic_add_fetch_64(&tsQueryBufferSizeBytes, t);
}
D
dapan1121 已提交
5253

dengyihao's avatar
dengyihao 已提交
5254 5255
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SExplainExecInfo** pRes, int32_t* capacity,
                                   int32_t* resNum) {
D
dapan1121 已提交
5256 5257
  if (*resNum >= *capacity) {
    *capacity += 10;
dengyihao's avatar
dengyihao 已提交
5258

D
dapan1121 已提交
5259 5260
    *pRes = taosMemoryRealloc(*pRes, (*capacity) * sizeof(SExplainExecInfo));
    if (NULL == *pRes) {
D
dapan1121 已提交
5261
      qError("malloc %d failed", (*capacity) * (int32_t)sizeof(SExplainExecInfo));
D
dapan1121 已提交
5262 5263 5264 5265
      return TSDB_CODE_QRY_OUT_OF_MEMORY;
    }
  }

5266 5267 5268 5269 5270
  SExplainExecInfo* pInfo = &(*pRes)[*resNum];

  pInfo->numOfRows = operatorInfo->resultInfo.totalRows;
  pInfo->startupCost = operatorInfo->cost.openCost;
  pInfo->totalCost = operatorInfo->cost.totalCost;
D
dapan1121 已提交
5271

5272
  if (operatorInfo->fpSet.getExplainFn) {
5273
    int32_t code = operatorInfo->fpSet.getExplainFn(operatorInfo, &pInfo->verboseInfo, &pInfo->verboseLen);
D
dapan1121 已提交
5274
    if (code) {
5275
      qError("%s operator getExplainFn failed, code:%s", GET_TASKID(operatorInfo->pTaskInfo), tstrerror(code));
D
dapan1121 已提交
5276 5277
      return code;
    }
5278 5279 5280
  } else {
    pInfo->verboseLen = 0;
    pInfo->verboseInfo = NULL;
D
dapan1121 已提交
5281
  }
dengyihao's avatar
dengyihao 已提交
5282

D
dapan1121 已提交
5283
  ++(*resNum);
dengyihao's avatar
dengyihao 已提交
5284

D
dapan1121 已提交
5285
  int32_t code = 0;
D
dapan1121 已提交
5286 5287
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pRes, capacity, resNum);
D
dapan1121 已提交
5288 5289 5290 5291 5292 5293 5294
    if (code) {
      taosMemoryFreeClear(*pRes);
      return TSDB_CODE_QRY_OUT_OF_MEMORY;
    }
  }

  return TSDB_CODE_SUCCESS;
D
dapan1121 已提交
5295
}
5
54liuyao 已提交
5296

X
Xiaoyu Wang 已提交
5297
int32_t initStreamAggSupporter(SStreamAggSupporter* pSup, const char* pKey, size_t size) {
5
54liuyao 已提交
5298 5299
  pSup->keySize = sizeof(int64_t) + sizeof(TSKEY);
  pSup->pKeyBuf = taosMemoryCalloc(1, pSup->keySize);
5
54liuyao 已提交
5300
  pSup->pResultRows = taosArrayInit(1024, size);
5
54liuyao 已提交
5301 5302 5303 5304
  if (pSup->pKeyBuf == NULL || pSup->pResultRows == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

5
54liuyao 已提交
5305 5306
  pSup->pScanWindow = taosArrayInit(4, sizeof(STimeWindow));

5
54liuyao 已提交
5307 5308 5309 5310 5311 5312 5313 5314 5315
  int32_t pageSize = 4096;
  while (pageSize < pSup->resultRowSize * 4) {
    pageSize <<= 1u;
  }
  // at least four pages need to be in buffer
  int32_t bufSize = 4096 * 256;
  if (bufSize <= pageSize) {
    bufSize = pageSize * 4;
  }
wafwerar's avatar
wafwerar 已提交
5316
  return createDiskbasedBuf(&pSup->pResultBuf, pageSize, bufSize, pKey, TD_TMP_DIR_PATH);
5
54liuyao 已提交
5317
}
5
54liuyao 已提交
5318

5
54liuyao 已提交
5319 5320 5321 5322 5323 5324 5325 5326
int32_t initSessionAggSupporter(SStreamAggSupporter* pSup, const char* pKey) {
  return initStreamAggSupporter(pSup, pKey, sizeof(SResultWindowInfo));
}

int32_t initStateAggSupporter(SStreamAggSupporter* pSup, const char* pKey) {
  return initStreamAggSupporter(pSup, pKey, sizeof(SStateWindowInfo));
}

5
54liuyao 已提交
5327 5328
int64_t getSmaWaterMark(int64_t interval, double filesFactor) {
  int64_t waterMark = 0;
X
Xiaoyu Wang 已提交
5329
  ASSERT(FLT_GREATEREQUAL(filesFactor, 0.000000));
5
54liuyao 已提交
5330 5331 5332 5333 5334
  waterMark = -1 * filesFactor;
  return waterMark;
}

bool isSmaStream(int8_t triggerType) {
X
Xiaoyu Wang 已提交
5335
  if (triggerType == STREAM_TRIGGER_AT_ONCE || triggerType == STREAM_TRIGGER_WINDOW_CLOSE) {
5
54liuyao 已提交
5336 5337 5338 5339
    return false;
  }
  return true;
}