executorimpl.c 200.7 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

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

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

31
#include "executorimpl.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 90 91 92 93
  }
}

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

#define CLEAR_QUERY_STATUS(q, st)   ((q)->status &= (~(st)))
#define GET_NUM_OF_TABLEGROUP(q)    taosArrayGetSize((q)->tableqinfoGroupInfo.pGroupList)
#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
}

static void doSetTagValueToResultBuf(char* output, const char* val, int16_t type, int16_t bytes);
L
Liu Jicong 已提交
102
static bool functionNeedToExecute(SqlFunctionCtx* pCtx);
103

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

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

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

L
Liu Jicong 已提交
110
static void    releaseQueryBuf(size_t numOfTables);
111

H
Haojun Liao 已提交
112
static int32_t getNumOfScanTimes(STaskAttr* pQueryAttr);
113 114 115 116 117 118

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 已提交
119

H
Haojun Liao 已提交
120
static void destroyIntervalOperatorInfo(void* param, int32_t numOfOutput);
H
Haojun Liao 已提交
121 122
static void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput);

123
static void destroyOperatorInfo(SOperatorInfo* pOperator);
124
static void destroySysTableScannerOperatorInfo(void* param, int32_t numOfOutput);
125

126
void doSetOperatorCompleted(SOperatorInfo* pOperator) {
127
  pOperator->status = OP_EXEC_DONE;
H
Haojun Liao 已提交
128
  if (pOperator->pTaskInfo != NULL) {
129
    setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
130 131
  }
}
132

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

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

  return fpSet;
}

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

158 159
static int32_t doCopyToSDataBlock(SExecTaskInfo* taskInfo, SSDataBlock* pBlock, SExprInfo* pExprInfo, SDiskbasedBuf* pBuf, SGroupResInfo* pGroupResInfo,
                                  int32_t orderType, 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;
L
Liu Jicong 已提交
185 186
  char*   in1 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page1, pRow1->offset, offset);
  char*   in2 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page2, pRow2->offset, offset);
187 188 189 190

  return (in1 != NULL && in2 != NULL) ? supporter->comFunc(in1, in2) : 0;
}

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

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

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

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

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

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

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

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

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

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

  return true;
}

H
Haojun Liao 已提交
239
static void prepareResultListBuffer(SResultRowInfo* pResultRowInfo, jmp_buf env) {
240 241
  int64_t newCapacity = 0;

242 243 244 245 246 247 248 249 250 251 252
  // more than the capacity, reallocate the resources
  if (pResultRowInfo->size < pResultRowInfo->capacity) {
    return;
  }

  if (pResultRowInfo->capacity > 10000) {
    newCapacity = (int64_t)(pResultRowInfo->capacity * 1.25);
  } else {
    newCapacity = (int64_t)(pResultRowInfo->capacity * 1.5);
  }

253
  if (newCapacity <= pResultRowInfo->capacity) {
H
Haojun Liao 已提交
254 255 256
    newCapacity += 4;
  }

257 258 259 260 261 262
  char* p = taosMemoryRealloc(pResultRowInfo->pPosition, newCapacity * sizeof(SResultRowPosition));
  if (p == NULL) {
    longjmp(env, TSDB_CODE_OUT_OF_MEMORY);
  }

  pResultRowInfo->pPosition = (SResultRowPosition*)p;
263 264

  int32_t inc = (int32_t)newCapacity - pResultRowInfo->capacity;
wmmhello's avatar
wmmhello 已提交
265
  memset(&pResultRowInfo->pPosition[pResultRowInfo->capacity], 0, sizeof(SResultRowPosition) * inc);
266 267 268
  pResultRowInfo->capacity = (int32_t)newCapacity;
}

L
Liu Jicong 已提交
269 270
static bool chkResultRowFromKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, char* pData,
                                int16_t bytes, bool masterscan, uint64_t uid) {
271 272 273
  bool existed = false;
  SET_RES_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid);

L
Liu Jicong 已提交
274 275
  SResultRow** p1 =
      (SResultRow**)taosHashGet(pRuntimeEnv->pResultRowHashTable, pRuntimeEnv->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
276 277 278 279 280 281 282 283 284 285 286

  // 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 已提交
287
        //        existed = (pResultRowInfo->pResult[0] == (*p1));
288 289
      } else {  // check if current pResultRowInfo contains the existed pResultRow
        SET_RES_EXT_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid, pResultRowInfo);
L
Liu Jicong 已提交
290 291
        int64_t* index =
            taosHashGet(pRuntimeEnv->pResultRowListSet, pRuntimeEnv->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes));
292 293 294 295 296 297 298 299 300 301 302 303 304 305
        if (index != NULL) {
          existed = true;
        } else {
          existed = false;
        }
      }
    }

    return existed;
  }

  return p1 != NULL;
}

306
SResultRow* getNewResultRow_rv(SDiskbasedBuf* pResultBuf, int64_t tableGroupId, int32_t interBufSize) {
L
Liu Jicong 已提交
307
  SFilePage* pData = NULL;
308 309 310 311 312 313 314 315 316 317 318 319 320

  // 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 已提交
321
    if (pData->num + interBufSize > getBufPageSize(pResultBuf)) {
322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
      // 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;
  }

  // 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 已提交
341
  pData->num += interBufSize;
342 343 344 345

  return pResultRow;
}

5
54liuyao 已提交
346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367
void doClearWindow(SIntervalAggOperatorInfo* pInfo, char* pData, int16_t bytes,
    uint64_t groupId, int32_t numOfOutput) {
  SAggSupporter* pSup = &pInfo->aggSup;
  SET_RES_WINDOW_KEY(pSup->keyBuf, pData, bytes, groupId);
  SResultRowPosition* p1 =
      (SResultRowPosition*)taosHashGet(pSup->pResultRowHashTable, pSup->keyBuf,
          GET_RES_WINDOW_KEY_LEN(bytes));
  SResultRow* pResult = getResultRowByPos(pSup->pResultBuf, p1);
  SqlFunctionCtx* pCtx = pInfo->binfo.pCtx;
  for (int32_t i = 0; i < numOfOutput; ++i) {
    pCtx[i].resultInfo = getResultCell(pResult, i, pInfo->binfo.rowCellInfoOffset);
    struct SResultRowEntryInfo* pResInfo = pCtx[i].resultInfo;
    if (fmIsWindowPseudoColumnFunc(pCtx[i].functionId)) {
      continue;
    }
    pResInfo->initialized = false;
    if (pCtx[i].functionId != -1) {
      pCtx[i].fpSet.init(&pCtx[i], pResInfo);
    }
  }
}

368 369 370 371 372 373 374
/**
 * the struct of key in hash table
 * +----------+---------------+
 * | group id |   key data    |
 * | 8 bytes  | actual length |
 * +----------+---------------+
 */
375 376 377
SResultRow* doSetResultOutBufByKey(SDiskbasedBuf* pResultBuf, SResultRowInfo* pResultRowInfo, char* pData,
                                   int16_t bytes, bool masterscan, uint64_t groupId, SExecTaskInfo* pTaskInfo,
                                   bool isIntervalQuery, SAggSupporter* pSup) {
378
  SET_RES_WINDOW_KEY(pSup->keyBuf, pData, bytes, groupId);
H
Haojun Liao 已提交
379

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

383 384
  SResultRow* pResult = NULL;

H
Haojun Liao 已提交
385 386
  // in case of repeat scan/reverse scan, no new time window added.
  if (isIntervalQuery) {
387 388
    if (masterscan && p1 != NULL) {  // the *p1 may be NULL in case of sliding+offset exists.
      pResult = getResultRowByPos(pResultBuf, p1);
H
Haojun Liao 已提交
389 390
    }
  } else {
dengyihao's avatar
dengyihao 已提交
391 392
    // In case of group by column query, the required SResultRow object must be existInCurrentResusltRowInfo in the
    // pResultRowInfo object.
H
Haojun Liao 已提交
393
    if (p1 != NULL) {
394
      pResult = getResultRowByPos(pResultBuf, p1);
H
Haojun Liao 已提交
395 396 397
    }
  }

L
Liu Jicong 已提交
398
  // 1. close current opened time window
399
  if (pResultRowInfo->cur.pageId != -1 && ((pResult == NULL) || (pResult->pageId != pResultRowInfo->cur.pageId &&
L
Liu Jicong 已提交
400
                                                                 pResult->offset != pResultRowInfo->cur.offset))) {
401 402 403 404 405 406 407 408 409 410 411
    // todo extract function
    SResultRowPosition pos = pResultRowInfo->cur;
    SFilePage*         pPage = getBufPage(pResultBuf, pos.pageId);
    SResultRow*        pRow = (SResultRow*)((char*)pPage + pos.offset);
    closeResultRow(pRow);
    releaseBufPage(pResultBuf, pPage);
  }

  // allocate a new buffer page
  prepareResultListBuffer(pResultRowInfo, pTaskInfo->env);
  if (pResult == NULL) {
H
Haojun Liao 已提交
412
    ASSERT(pSup->resultRowSize > 0);
413 414
    pResult = getNewResultRow_rv(pResultBuf, groupId, pSup->resultRowSize);
    initResultRow(pResult);
H
Haojun Liao 已提交
415

416 417
    // add a new result set for a new group
    SResultRowPosition pos = {.pageId = pResult->pageId, .offset = pResult->offset};
L
Liu Jicong 已提交
418 419
    taosHashPut(pSup->pResultRowHashTable, pSup->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes), &pos,
                sizeof(SResultRowPosition));
H
Haojun Liao 已提交
420 421
  }

422 423 424 425 426
  // 2. set the new time window to be the new active time window
  pResultRowInfo->pPosition[pResultRowInfo->size++] =
      (SResultRowPosition){.pageId = pResult->pageId, .offset = pResult->offset};
  pResultRowInfo->cur = (SResultRowPosition){.pageId = pResult->pageId, .offset = pResult->offset};

H
Haojun Liao 已提交
427 428 429 430 431
  // too many time window in query
  if (pResultRowInfo->size > MAX_INTERVAL_TIME_WINDOW) {
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_TOO_MANY_TIMEWINDOW);
  }

H
Haojun Liao 已提交
432
  return pResult;
H
Haojun Liao 已提交
433 434
}

435
// get the correct time window according to the handled timestamp
L
Liu Jicong 已提交
436
static STimeWindow getCurrentActiveTimeWindow(SResultRowInfo* pResultRowInfo, int64_t ts, STaskAttr* pQueryAttr) {
437
  STimeWindow w = {0};
H
Haojun Liao 已提交
438
#if 0
L
Liu Jicong 已提交
439 440
  if (pResultRowInfo->curPos == -1) {  // the first window, from the previous stored value
                                       //    getInitialStartTimeWindow(pQueryAttr, ts, &w);
441 442

    if (pQueryAttr->interval.intervalUnit == 'n' || pQueryAttr->interval.intervalUnit == 'y') {
L
Liu Jicong 已提交
443 444 445
      w.ekey =
          taosTimeAdd(w.skey, pQueryAttr->interval.interval, pQueryAttr->interval.intervalUnit, pQueryAttr->precision) -
          1;
446 447 448 449
    } else {
      w.ekey = w.skey + pQueryAttr->interval.interval - 1;
    }
  } else {
H
Haojun Liao 已提交
450
    w = pRow->win;
451 452 453 454 455 456 457 458 459
  }

  /*
   * 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 已提交
460
#endif
461 462 463 464 465

  return w;
}

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

L
Liu Jicong 已提交
471
  SFilePage* pData = NULL;
472 473 474 475 476 477

  // 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 已提交
478
    pData = getNewBufPage(pResultBuf, tid, &pageId);
479
    pData->num = sizeof(SFilePage);
480 481
  } else {
    SPageInfo* pi = getLastPageInfo(list);
482
    pData = getBufPage(pResultBuf, getPageId(pi));
483
    pageId = getPageId(pi);
484

485
    if (pData->num + size > getBufPageSize(pResultBuf)) {
486
      // release current page first, and prepare the next one
487
      releaseBufPageInfo(pResultBuf, pi);
488

H
Haojun Liao 已提交
489
      pData = getNewBufPage(pResultBuf, tid, &pageId);
490
      if (pData != NULL) {
491
        pData->num = sizeof(SFilePage);
492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511
      }
    }
  }

  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 已提交
512 513 514
static bool chkWindowOutputBufByKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, STimeWindow* win,
                                    bool masterscan, SResultRow** pResult, int64_t groupId, SqlFunctionCtx* pCtx,
                                    int32_t numOfOutput, int32_t* rowCellInfoOffset) {
515
  assert(win->skey <= win->ekey);
L
Liu Jicong 已提交
516
  return chkResultRowFromKey(pRuntimeEnv, pResultRowInfo, (char*)&win->skey, TSDB_KEYSIZE, masterscan, groupId);
517 518
}

dengyihao's avatar
dengyihao 已提交
519 520
static void doUpdateResultRowIndex(SResultRowInfo* pResultRowInfo, TSKEY lastKey, bool ascQuery,
                                   bool timeWindowInterpo) {
521
  int64_t skey = TSKEY_INITIAL_VAL;
H
Haojun Liao 已提交
522
#if 0
523 524
  int32_t i = 0;
  for (i = pResultRowInfo->size - 1; i >= 0; --i) {
L
Liu Jicong 已提交
525
    SResultRow* pResult = pResultRowInfo->pResult[i];
526 527 528 529 530 531
    if (pResult->closed) {
      break;
    }

    // new closed result rows
    if (timeWindowInterpo) {
L
Liu Jicong 已提交
532 533 534
      if (pResult->endInterp &&
          ((pResult->win.skey <= lastKey && ascQuery) || (pResult->win.skey >= lastKey && !ascQuery))) {
        if (i > 0) {  // the first time window, the startInterp is false.
535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553
          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 已提交
554
      //      assert(pResultRowInfo->current == NULL);
555 556 557 558 559 560 561
      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 已提交
562
      SResultRow* pResult = pResultRowInfo->pResult[i];
563 564 565 566 567 568 569 570 571 572 573
      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 已提交
574
#endif
575
}
576
//
dengyihao's avatar
dengyihao 已提交
577
// static void updateResultRowInfoActiveIndex(SResultRowInfo* pResultRowInfo, const STimeWindow* pWin, TSKEY lastKey,
578 579 580 581 582 583 584 585 586
//                                           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);
//  }
//}
587

588
//  query_range_start, query_range_end, window_duration, window_start, window_end
589
void initExecTimeWindowInfo(SColumnInfoData* pColData, STimeWindow* pQueryWindow) {
590 591 592
  pColData->info.type = TSDB_DATA_TYPE_TIMESTAMP;
  pColData->info.bytes = sizeof(int64_t);

593
  colInfoDataEnsureCapacity(pColData, 0, 5);
594 595 596 597 598 599 600 601 602
  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);
}

603
void doApplyFunctions(SExecTaskInfo* taskInfo, SqlFunctionCtx* pCtx, STimeWindow* pWin, SColumnInfoData* pTimeWindowData, int32_t offset,
dengyihao's avatar
dengyihao 已提交
604
                      int32_t forwardStep, TSKEY* tsCol, int32_t numOfTotal, int32_t numOfOutput, int32_t order) {
605 606 607
  for (int32_t k = 0; k < numOfOutput; ++k) {
    pCtx[k].startTs = pWin->skey;

H
Haojun Liao 已提交
608
    // keep it temporarily
dengyihao's avatar
dengyihao 已提交
609 610
    bool    hasAgg = pCtx[k].input.colDataAggIsSet;
    int32_t numOfRows = pCtx[k].input.numOfRows;
H
Haojun Liao 已提交
611
    int32_t startOffset = pCtx[k].input.startRowIndex;
612

H
Haojun Liao 已提交
613
    int32_t pos = (order == TSDB_ORDER_ASC) ? offset : offset - (forwardStep - 1);
614 615
    pCtx[k].input.startRowIndex = pos;
    pCtx[k].input.numOfRows = forwardStep;
616 617

    if (tsCol != NULL) {
H
Haojun Liao 已提交
618
      pCtx[k].ptsList = tsCol;
619 620 621 622 623 624 625 626
    }

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

627 628
    if (fmIsWindowPseudoColumnFunc(pCtx[k].functionId)) {
      SResultRowEntryInfo* pEntryInfo = GET_RES_INFO(&pCtx[k]);
dengyihao's avatar
dengyihao 已提交
629
      char*                p = GET_ROWCELL_INTERBUF(pEntryInfo);
630

631
      SColumnInfoData idata = {0};
dengyihao's avatar
dengyihao 已提交
632
      idata.info.type = TSDB_DATA_TYPE_BIGINT;
633
      idata.info.bytes = tDataTypes[TSDB_DATA_TYPE_BIGINT].bytes;
dengyihao's avatar
dengyihao 已提交
634
      idata.pData = p;
635 636 637 638

      SScalarParam out = {.columnData = &idata};
      SScalarParam tw = {.numOfRows = 5, .columnData = pTimeWindowData};
      pCtx[k].sfp.process(&tw, 1, &out);
639 640 641
      pEntryInfo->numOfRes = 1;
      continue;
    }
642
    int32_t code = TSDB_CODE_SUCCESS;
643
    if (functionNeedToExecute(&pCtx[k]) && pCtx[k].fpSet.process != NULL) {
644 645 646 647 648 649
      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);
      }
650 651 652
    }

    // restore it
653 654 655
    pCtx[k].input.colDataAggIsSet = hasAgg;
    pCtx[k].input.startRowIndex = startOffset;
    pCtx[k].input.numOfRows = numOfRows;
656 657 658
  }
}

L
Liu Jicong 已提交
659
static FORCE_INLINE TSKEY reviseWindowEkey(STaskAttr* pQueryAttr, STimeWindow* pWindow) {
dengyihao's avatar
dengyihao 已提交
660
  TSKEY   ekey = -1;
661 662
  int32_t order = TSDB_ORDER_ASC;
  if (order == TSDB_ORDER_ASC) {
663 664 665 666 667 668 669 670 671 672 673 674 675 676
    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 已提交
677
static int32_t doSetInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order,
678
                                   int32_t scanFlag, bool createDummyCol);
679

dengyihao's avatar
dengyihao 已提交
680 681
static void doSetInputDataBlockInfo(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock,
                                    int32_t order) {
682
  for (int32_t i = 0; i < pOperator->numOfExprs; ++i) {
683
    pCtx[i].order = order;
L
Liu Jicong 已提交
684
    pCtx[i].size = pBlock->info.rows;
685
    setBlockStatisInfo(&pCtx[i], &pOperator->pExpr[i], pBlock);
686 687 688
  }
}

689
void setInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order, int32_t scanFlag,
dengyihao's avatar
dengyihao 已提交
690
                       bool createDummyCol) {
691
  if (pBlock->pBlockAgg != NULL) {
H
Haojun Liao 已提交
692
    doSetInputDataBlockInfo(pOperator, pCtx, pBlock, order);
693
  } else {
694
    doSetInputDataBlock(pOperator, pCtx, pBlock, order, scanFlag, createDummyCol);
H
Haojun Liao 已提交
695
  }
696 697
}

L
Liu Jicong 已提交
698 699
static int32_t doCreateConstantValColumnInfo(SInputColumnInfoData* pInput, SFunctParam* pFuncParam, int32_t paramIndex,
                                             int32_t numOfRows) {
700 701 702 703 704 705 706 707
  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)
708 709
    pColInfo->info.type = pFuncParam->param.nType;
    pColInfo->info.bytes = pFuncParam->param.nLen;
710 711

    pInput->pData[paramIndex] = pColInfo;
712 713
  } else {
    pColInfo = pInput->pData[paramIndex];
714 715
  }

716
  colInfoDataEnsureCapacity(pColInfo, 0, numOfRows);
717

718
  int8_t type = pFuncParam->param.nType;
719 720
  if (type == TSDB_DATA_TYPE_BIGINT || type == TSDB_DATA_TYPE_UBIGINT) {
    int64_t v = pFuncParam->param.i;
dengyihao's avatar
dengyihao 已提交
721
    for (int32_t i = 0; i < numOfRows; ++i) {
722 723 724 725
      colDataAppendInt64(pColInfo, i, &v);
    }
  } else if (type == TSDB_DATA_TYPE_DOUBLE) {
    double v = pFuncParam->param.d;
dengyihao's avatar
dengyihao 已提交
726
    for (int32_t i = 0; i < numOfRows; ++i) {
727 728
      colDataAppendDouble(pColInfo, i, &v);
    }
729
  } else if (type == TSDB_DATA_TYPE_VARCHAR) {
L
Liu Jicong 已提交
730
    char* tmp = taosMemoryMalloc(pFuncParam->param.nLen + VARSTR_HEADER_SIZE);
731
    STR_WITH_SIZE_TO_VARSTR(tmp, pFuncParam->param.pz, pFuncParam->param.nLen);
L
Liu Jicong 已提交
732
    for (int32_t i = 0; i < numOfRows; ++i) {
733 734
      colDataAppend(pColInfo, i, tmp, false);
    }
735 736 737 738 739
  }

  return TSDB_CODE_SUCCESS;
}

dengyihao's avatar
dengyihao 已提交
740
static int32_t doSetInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order,
741
    int32_t scanFlag, bool createDummyCol) {
742 743
  int32_t code = TSDB_CODE_SUCCESS;

744
  for (int32_t i = 0; i < pOperator->numOfExprs; ++i) {
L
Liu Jicong 已提交
745 746 747
    pCtx[i].order = order;
    pCtx[i].size = pBlock->info.rows;
    pCtx[i].pSrcBlock = pBlock;
748
    pCtx[i].scanFlag  = scanFlag;
H
Haojun Liao 已提交
749

750
    SInputColumnInfoData* pInput = &pCtx[i].input;
751
    pInput->uid = pBlock->info.uid;
C
Cary Xu 已提交
752
    pInput->colDataAggIsSet = false;
753

754 755
    SExprInfo* pOneExpr = &pOperator->pExpr[i];
    for (int32_t j = 0; j < pOneExpr->base.numOfParams; ++j) {
dengyihao's avatar
dengyihao 已提交
756
      SFunctParam* pFuncParam = &pOneExpr->base.pParam[j];
G
Ganlin Zhao 已提交
757 758
      if (pFuncParam->type == FUNC_PARAM_TYPE_COLUMN) {
        int32_t slotId = pFuncParam->pCol->slotId;
dengyihao's avatar
dengyihao 已提交
759
        pInput->pData[j] = taosArrayGet(pBlock->pDataBlock, slotId);
760 761 762
        pInput->totalRows = pBlock->info.rows;
        pInput->numOfRows = pBlock->info.rows;
        pInput->startRowIndex = 0;
763

764
        // NOTE: the last parameter is the primary timestamp column
765 766 767
        if (fmIsTimelineFunc(pCtx[i].functionId) && (j == pOneExpr->base.numOfParams - 1)) {
          pInput->pPTS = pInput->pData[j];
        }
768 769
        ASSERT(pInput->pData[j] != NULL);
      } else if (pFuncParam->type == FUNC_PARAM_TYPE_VALUE) {
770 771 772
        // 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) {
773 774 775 776
          pInput->totalRows = pBlock->info.rows;
          pInput->numOfRows = pBlock->info.rows;
          pInput->startRowIndex = 0;

777
          code = doCreateConstantValColumnInfo(pInput, pFuncParam, j, pBlock->info.rows);
778 779 780
          if (code != TSDB_CODE_SUCCESS) {
            return code;
          }
781
        }
G
Ganlin Zhao 已提交
782 783
      }
    }
H
Haojun Liao 已提交
784

H
Haojun Liao 已提交
785
    //    setBlockStatisInfo(&pCtx[i], pBlock, pOperator->pExpr[i].base.pColumns);
H
Haojun Liao 已提交
786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823
    //      uint32_t flag = pOperator->pExpr[i].base.pParam[0].pCol->flag;
    //      if (TSDB_COL_IS_NORMAL_COL(flag) /*|| (pCtx[i].functionId == FUNCTION_BLKINFO) ||
    //          (TSDB_COL_IS_TAG(flag) && pOperator->pRuntimeEnv->scanFlag == MERGE_STAGE)*/) {

    //        SColumn* pCol = pOperator->pExpr[i].base.pParam[0].pCol;
    //        if (pCtx[i].columnIndex == -1) {
    //          for(int32_t j = 0; j < pBlock->info.numOfCols; ++j) {
    //            SColumnInfoData* pColData = taosArrayGet(pBlock->pDataBlock, j);
    //            if (pColData->info.colId == pCol->colId) {
    //              pCtx[i].columnIndex = j;
    //              break;
    //            }
    //          }
    //        }

    //        uint32_t status = aAggs[pCtx[i].functionId].status;
    //        if ((status & (FUNCSTATE_SELECTIVITY | FUNCSTATE_NEED_TS)) != 0) {
    //          SColumnInfoData* tsInfo = taosArrayGet(pBlock->pDataBlock, 0);
    // In case of the top/bottom query again the nest query result, which has no timestamp column
    // don't set the ptsList attribute.
    //          if (tsInfo->info.type == TSDB_DATA_TYPE_TIMESTAMP) {
    //            pCtx[i].ptsList = (int64_t*) tsInfo->pData;
    //          } else {
    //            pCtx[i].ptsList = NULL;
    //          }
    //        }
    //      } else if (TSDB_COL_IS_UD_COL(pCol->flag) && (pOperator->pRuntimeEnv->scanFlag == MERGE_STAGE)) {
    //        SColIndex*       pColIndex = &pOperator->pExpr[i].base.colInfo;
    //        SColumnInfoData* p = taosArrayGet(pBlock->pDataBlock, pColIndex->colIndex);
    //
    //        pCtx[i].pInput = p->pData;
    //        assert(p->info.colId == pColIndex->info.colId && pCtx[i].inputType == p->info.type);
    //        for(int32_t j = 0; j < pBlock->info.rows; ++j) {
    //          char* dst = p->pData + j * p->info.bytes;
    //          taosVariantDump(&pOperator->pExpr[i].base.param[1], dst, p->info.type, true);
    //        }
    //      }
  }
824 825

  return code;
H
Haojun Liao 已提交
826 827
}

828
static int32_t doAggregateImpl(SOperatorInfo* pOperator, TSKEY startTs, SqlFunctionCtx* pCtx) {
829
  for (int32_t k = 0; k < pOperator->numOfExprs; ++k) {
H
Haojun Liao 已提交
830
    if (functionNeedToExecute(&pCtx[k])) {
831 832 833
      pCtx[k].startTs = startTs;
      // todo add a dummy funtion to avoid process check
      if (pCtx[k].fpSet.process != NULL) {
834 835
        int32_t code = pCtx[k].fpSet.process(&pCtx[k]);
        if (code != TSDB_CODE_SUCCESS) {
836 837
          qError("%s aggregate function error happens, code: %s", GET_TASKID(pOperator->pTaskInfo), tstrerror(code));
          return code;
838
        }
839
      }
840 841
    }
  }
842 843

  return TSDB_CODE_SUCCESS;
844 845
}

H
Haojun Liao 已提交
846
static void setPseudoOutputColInfo(SSDataBlock* pResult, SqlFunctionCtx* pCtx, SArray* pPseudoList) {
dengyihao's avatar
dengyihao 已提交
847
  size_t num = (pPseudoList != NULL) ? taosArrayGetSize(pPseudoList) : 0;
H
Haojun Liao 已提交
848 849 850 851 852
  for (int32_t i = 0; i < num; ++i) {
    pCtx[i].pOutput = taosArrayGet(pResult->pDataBlock, i);
  }
}

853
int32_t projectApplyFunctions(SExprInfo* pExpr, SSDataBlock* pResult, SSDataBlock* pSrcBlock, SqlFunctionCtx* pCtx,
X
Xiaoyu Wang 已提交
854
                              int32_t numOfOutput, SArray* pPseudoList) {
H
Haojun Liao 已提交
855
  setPseudoOutputColInfo(pResult, pCtx, pPseudoList);
H
Haojun Liao 已提交
856
  pResult->info.groupId = pSrcBlock->info.groupId;
H
Haojun Liao 已提交
857

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

862 863
  int32_t numOfRows = 0;

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

L
Liu Jicong 已提交
868
    if (pExpr[k].pExpr->nodeType == QUERY_NODE_COLUMN) {  // it is a project query
869
      SColumnInfoData* pColInfoData = taosArrayGet(pResult->pDataBlock, outputSlotId);
870
      if (pResult->info.rows > 0 && !createNewColModel) {
X
Xiaoyu Wang 已提交
871 872
        colDataMergeCol(pColInfoData, pResult->info.rows, &pResult->info.capacity, pfCtx->input.pData[0],
                        pfCtx->input.numOfRows);
873 874 875
      } else {
        colDataAssign(pColInfoData, pfCtx->input.pData[0], pfCtx->input.numOfRows);
      }
876

877
      numOfRows = pfCtx->input.numOfRows;
878
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_VALUE) {
879
      SColumnInfoData* pColInfoData = taosArrayGet(pResult->pDataBlock, outputSlotId);
880

dengyihao's avatar
dengyihao 已提交
881
      int32_t offset = createNewColModel ? 0 : pResult->info.rows;
882
      for (int32_t i = 0; i < pSrcBlock->info.rows; ++i) {
dengyihao's avatar
dengyihao 已提交
883 884 885
        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);
886
      }
887 888

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

893
      SColumnInfoData* pResColData = taosArrayGet(pResult->pDataBlock, outputSlotId);
894
      SColumnInfoData  idata = {.info = pResColData->info, .hasNull = true};
895

896
      SScalarParam dest = {.columnData = &idata};
897 898
      scalarCalculate(pExpr[k].pExpr->_optrRoot.pRootNode, pBlockList, &dest);

dengyihao's avatar
dengyihao 已提交
899
      int32_t startOffset = createNewColModel ? 0 : pResult->info.rows;
900
      colDataMergeCol(pResColData, startOffset, &pResult->info.capacity, &idata, dest.numOfRows);
901 902

      numOfRows = dest.numOfRows;
903 904
      taosArrayDestroy(pBlockList);
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_FUNCTION) {
905
      ASSERT(!fmIsAggFunc(pfCtx->functionId));
906

907 908
      // _rowts/_c0, not tbname column
      if (fmIsPseudoColumnFunc(pfCtx->functionId) && (!fmIsScanPseudoColumnFunc(pfCtx->functionId))) {
H
Haojun Liao 已提交
909
        // do nothing
910
      } else if (fmIsNonstandardSQLFunc(pfCtx->functionId)) {
dengyihao's avatar
dengyihao 已提交
911
        SResultRowEntryInfo* pResInfo = GET_RES_INFO(&pCtx[k]);
912
        pfCtx->fpSet.init(&pCtx[k], pResInfo);
913

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

917
        // set the timestamp(_rowts) output buffer
918 919
        if (taosArrayGetSize(pPseudoList) > 0) {
          int32_t* outputColIndex = taosArrayGet(pPseudoList, 0);
920
          pfCtx->pTsOutput = (SColumnInfoData*)pCtx[*outputColIndex].pOutput;
921
        }
H
Haojun Liao 已提交
922

923
        numOfRows = pfCtx->fpSet.process(pfCtx);
H
Haojun Liao 已提交
924 925 926
      } else {
        SArray* pBlockList = taosArrayInit(4, POINTER_BYTES);
        taosArrayPush(pBlockList, &pSrcBlock);
G
Ganlin Zhao 已提交
927

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

931
        SScalarParam dest = {.columnData = &idata};
X
Xiaoyu Wang 已提交
932
        int32_t      code = scalarCalculate((SNode*)pExpr[k].pExpr->_function.pFunctNode, pBlockList, &dest);
933 934 935 936
        if (code != TSDB_CODE_SUCCESS) {
          taosArrayDestroy(pBlockList);
          return code;
        }
937

dengyihao's avatar
dengyihao 已提交
938
        int32_t startOffset = createNewColModel ? 0 : pResult->info.rows;
939
        colDataMergeCol(pResColData, startOffset, &pResult->info.capacity, &idata, dest.numOfRows);
940 941

        numOfRows = dest.numOfRows;
H
Haojun Liao 已提交
942 943
        taosArrayDestroy(pBlockList);
      }
944
    } else {
945
      ASSERT(0);
946 947
    }
  }
948

949 950 951
  if (!createNewColModel) {
    pResult->info.rows += numOfRows;
  }
952 953

  return TSDB_CODE_SUCCESS;
954 955
}

956 957 958
static void setResultRowKey(SResultRow* pResultRow, char* pData, int16_t type) {
  if (IS_VAR_DATA_TYPE(type)) {
    // todo disable this
959

960 961 962 963 964 965 966 967 968
    //    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);
969

970 971
    pResultRow->win.skey = v;
    pResultRow->win.ekey = v;
972 973 974
  }
}

975 976 977 978 979
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;
980

981 982 983
  SResultRow* pResultRow =
      doSetResultOutBufByKey(pBuf, pResultRowInfo, (char*)pData, bytes, true, groupId, pTaskInfo, false, pAggSup);
  assert(pResultRow != NULL);
984

985
  setResultRowKey(pResultRow, pData, type);
986
  setResultRowInitCtx(pResultRow, pCtx, numOfCols, binfo->rowCellInfoOffset);
987
  return TSDB_CODE_SUCCESS;
988 989
}

990 991
static bool functionNeedToExecute(SqlFunctionCtx* pCtx) {
  struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(pCtx);
992

993 994 995 996 997
  // in case of timestamp column, always generated results.
  int32_t functionId = pCtx->functionId;
  if (functionId == -1) {
    return false;
  }
998

999 1000
  if (pCtx->scanFlag == REPEAT_SCAN) {
    return fmIsRepeatScanFunc(pCtx->functionId);
1001 1002
  }

1003 1004
  if (isRowEntryCompleted(pResInfo)) {
    return false;
1005 1006
  }

1007 1008 1009 1010 1011 1012 1013 1014
//  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;
//  }
1015

1016 1017 1018 1019 1020
  // 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;
  //  }
1021 1022 1023 1024

  return true;
}

1025 1026 1027 1028 1029 1030 1031
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;
    }
1032

1033 1034 1035
    // Set the correct column info (data type and bytes)
    pInput->pData[paramIndex]->info.type = type;
    pInput->pData[paramIndex]->info.bytes = tDataTypes[type].bytes;
1036
  }
H
Haojun Liao 已提交
1037

1038 1039 1040 1041 1042 1043
  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;
1044 1045
    }
  } else {
1046
    da = pInput->pColumnDataAgg[paramIndex];
1047 1048
  }

1049
  ASSERT(!IS_VAR_DATA_TYPE(type));
1050

1051 1052 1053 1054 1055 1056
  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};
1057

1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069
    *(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
1070
  } else {
1071
    ASSERT(0);
1072 1073
  }

1074 1075
  return TSDB_CODE_SUCCESS;
}
1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086

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;

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

1090 1091
      if (pFuncParam->type == FUNC_PARAM_TYPE_COLUMN) {
        int32_t slotId = pFuncParam->pCol->slotId;
1092 1093 1094 1095
        pInput->pColumnDataAgg[j] = pBlock->pBlockAgg[slotId];
        if (pInput->pColumnDataAgg[j] == NULL) {
          pInput->colDataAggIsSet = false;
        }
1096 1097 1098 1099

        // 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);
1100 1101
      } else if (pFuncParam->type == FUNC_PARAM_TYPE_VALUE) {
        doCreateConstantValColumnAggInfo(pInput, pFuncParam, pFuncParam->param.nType, j, pBlock->info.rows);
1102 1103
      }
    }
1104
  } else {
1105
    pInput->colDataAggIsSet = false;
1106 1107 1108
  }

  // set the statistics data for primary time stamp column
1109 1110 1111 1112 1113
  //  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;
  //  }
1114 1115 1116
}

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

H
Haojun Liao 已提交
1120
  SqlFunctionCtx*  p = NULL;
1121 1122
  SqlFunctionCtx** pValCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
  if (pValCtx == NULL) {
1123 1124 1125 1126
    return TSDB_CODE_QRY_OUT_OF_MEMORY;
  }

  for (int32_t i = 0; i < numOfOutput; ++i) {
1127 1128
    if (strcmp(pCtx[i].pExpr->pExpr->_function.functionName, "_select_value") == 0) {
      pValCtx[num++] = &pCtx[i];
1129
    } else if (fmIsAggFunc(pCtx[i].functionId)) {
1130
      p = &pCtx[i];
1131
    }
1132 1133 1134 1135 1136 1137 1138 1139 1140 1141
//    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
//    }
1142
  }
1143

1144
  if (p != NULL) {
1145
    p->subsidiaries.pCtx = pValCtx;
1146
    p->subsidiaries.num = num;
1147
  } else {
1148
    taosMemoryFreeClear(pValCtx);
1149 1150 1151 1152 1153
  }

  return TSDB_CODE_SUCCESS;
}

1154
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowCellInfoOffset) {
L
Liu Jicong 已提交
1155
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
H
Haojun Liao 已提交
1156 1157 1158 1159
  if (pFuncCtx == NULL) {
    return NULL;
  }

wafwerar's avatar
wafwerar 已提交
1160
  *rowCellInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
H
Haojun Liao 已提交
1161
  if (*rowCellInfoOffset == 0) {
wafwerar's avatar
wafwerar 已提交
1162
    taosMemoryFreeClear(pFuncCtx);
H
Haojun Liao 已提交
1163 1164 1165 1166
    return NULL;
  }

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

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

1172
    pCtx->functionId = -1;
1173
    pCtx->curBufPage = -1;
L
Liu Jicong 已提交
1174
    pCtx->pExpr = pExpr;
1175

H
Haojun Liao 已提交
1176
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
1177
      SFuncExecEnv env = {0};
H
Haojun Liao 已提交
1178 1179
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;

H
Haojun Liao 已提交
1180
      if (fmIsAggFunc(pCtx->functionId) || fmIsNonstandardSQLFunc(pCtx->functionId)) {
1181 1182 1183 1184
        bool isUdaf = fmIsUserDefinedFunc(pCtx->functionId);
        if (!isUdaf) {
          fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
        } else {
1185
          char* udfName = pExpr->pExpr->_function.pFunctNode->functionName;
1186 1187 1188
          strncpy(pCtx->udfName, udfName, strlen(udfName));
          fmGetUdafExecFuncs(pCtx->functionId, &pCtx->fpSet);
        }
1189 1190 1191
        pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
      } else {
        fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
1192 1193 1194
        if (pCtx->sfp.getEnv != NULL) {
          pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
        }
1195
      }
H
Haojun Liao 已提交
1196
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
1197 1198
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR ||
               pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
H
Haojun Liao 已提交
1199
      // for simple column, the result buffer needs to hold at least one element.
1200
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes;
H
Haojun Liao 已提交
1201
    }
H
Haojun Liao 已提交
1202

H
Haojun Liao 已提交
1203
    pCtx->input.numOfInputCols = pFunct->numOfParams;
wafwerar's avatar
wafwerar 已提交
1204 1205
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
H
Haojun Liao 已提交
1206

1207
    pCtx->pTsOutput = NULL;
L
Liu Jicong 已提交
1208
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
1209
    pCtx->resDataInfo.type = pFunct->resSchema.type;
L
Liu Jicong 已提交
1210
    pCtx->order = TSDB_ORDER_ASC;
1211
    pCtx->start.key = INT64_MIN;
L
Liu Jicong 已提交
1212
    pCtx->end.key = INT64_MIN;
1213
    pCtx->numOfParams = pExpr->base.numOfParams;
H
Haojun Liao 已提交
1214

1215
    pCtx->param = pFunct->pParam;
dengyihao's avatar
dengyihao 已提交
1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249
    //    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 已提交
1250 1251
  }

L
Liu Jicong 已提交
1252 1253 1254
  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 已提交
1255
  }
H
Haojun Liao 已提交
1256

1257
  setSelectValueColumnInfo(pFuncCtx, numOfOutput);
H
Haojun Liao 已提交
1258 1259 1260
  return pFuncCtx;
}

1261
static void* destroySqlFunctionCtx(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
1262 1263 1264 1265 1266 1267
  if (pCtx == NULL) {
    return NULL;
  }

  for (int32_t i = 0; i < numOfOutput; ++i) {
    for (int32_t j = 0; j < pCtx[i].numOfParams; ++j) {
1268
      taosVariantDestroy(&pCtx[i].param[j].param);
1269 1270 1271
    }

    taosVariantDestroy(&pCtx[i].tag);
1272
    taosMemoryFreeClear(pCtx[i].subsidiaries.pCtx);
H
Haojun Liao 已提交
1273 1274
    taosMemoryFree(pCtx[i].input.pData);
    taosMemoryFree(pCtx[i].input.pColumnDataAgg);
1275 1276
  }

wafwerar's avatar
wafwerar 已提交
1277
  taosMemoryFreeClear(pCtx);
1278 1279 1280
  return NULL;
}

L
Liu Jicong 已提交
1281
bool isTaskKilled(SExecTaskInfo* pTaskInfo) {
1282 1283
  // query has been executed more than tsShellActivityTimer, and the retrieve has not arrived
  // abort current query execution.
L
Liu Jicong 已提交
1284 1285
  if (pTaskInfo->owner != 0 &&
      ((taosGetTimestampSec() - pTaskInfo->cost.start / 1000) > 10 * getMaximumIdleDurationSec())
1286 1287
      /*(!needBuildResAfterQueryComplete(pTaskInfo))*/) {
    assert(pTaskInfo->cost.start != 0);
L
Liu Jicong 已提交
1288 1289 1290
    //    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;
1291 1292 1293 1294 1295
  }

  return false;
}

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

L
Liu Jicong 已提交
1298
static bool isCachedLastQuery(STaskAttr* pQueryAttr) {
1299 1300 1301 1302 1303 1304 1305 1306 1307
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
    int32_t functionId = getExprFunctionId(&pQueryAttr->pExpr1[i]);
    if (functionId == FUNCTION_LAST || functionId == FUNCTION_LAST_DST) {
      continue;
    }

    return false;
  }

1308 1309
  int32_t order = TSDB_ORDER_ASC;
  if (order != TSDB_ORDER_DESC || !TSWINDOW_IS_EQUAL(pQueryAttr->window, TSWINDOW_DESC_INITIALIZER)) {
1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328
    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 已提交
1329
// todo refactor : return window
1330
void getAlignQueryTimeWindow(SInterval* pInterval, int32_t precision, int64_t key, STimeWindow* win) {
H
Haojun Liao 已提交
1331
  win->skey = taosTimeTruncate(key, pInterval, precision);
1332 1333

  /*
H
Haojun Liao 已提交
1334
   * if the realSkey > INT64_MAX - pInterval->interval, the query duration between
1335 1336
   * realSkey and realEkey must be less than one interval.Therefore, no need to adjust the query ranges.
   */
1337 1338
  win->ekey = taosTimeAdd(win->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
  if (win->ekey < win->skey) {
1339 1340 1341 1342
    win->ekey = INT64_MAX;
  }
}

L
Liu Jicong 已提交
1343
static int32_t updateBlockLoadStatus(STaskAttr* pQuery, int32_t status) {
1344 1345 1346
  bool hasFirstLastFunc = false;
  bool hasOtherFunc = false;

1347
  if (status == BLK_DATA_DATA_LOAD || status == BLK_DATA_FILTEROUT) {
1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365
    return status;
  }

  for (int32_t i = 0; i < pQuery->numOfOutput; ++i) {
    int32_t functionId = getExprFunctionId(&pQuery->pExpr1[i]);

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

1366
  if (hasFirstLastFunc && status == BLK_DATA_NOT_LOAD) {
L
Liu Jicong 已提交
1367
    if (!hasOtherFunc) {
1368
      return BLK_DATA_FILTEROUT;
1369
    } else {
1370
      return BLK_DATA_DATA_LOAD;
1371 1372 1373 1374 1375 1376
    }
  }

  return status;
}

L
Liu Jicong 已提交
1377 1378
// static void updateDataCheckOrder(SQInfo *pQInfo, SQueryTableReq* pQueryMsg, bool stableQuery) {
//   STaskAttr* pQueryAttr = pQInfo->runtimeEnv.pQueryAttr;
H
Haojun Liao 已提交
1379
//
L
Liu Jicong 已提交
1380 1381 1382 1383
//   // 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 已提交
1384
//
L
Liu Jicong 已提交
1385 1386 1387 1388 1389
//   // 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 已提交
1390
//
L
Liu Jicong 已提交
1391 1392
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     if (pQueryAttr->window.skey > pQueryAttr->window.ekey) {
wafwerar's avatar
wafwerar 已提交
1393
//       TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
L
Liu Jicong 已提交
1394
//     }
H
Haojun Liao 已提交
1395
//
L
Liu Jicong 已提交
1396 1397 1398
//     pQueryAttr->needReverseScan = false;
//     return;
//   }
H
Haojun Liao 已提交
1399
//
L
Liu Jicong 已提交
1400 1401 1402
//   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 已提交
1403
//       TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
L
Liu Jicong 已提交
1404
//     }
H
Haojun Liao 已提交
1405
//
L
Liu Jicong 已提交
1406 1407 1408 1409
//     pQueryAttr->needReverseScan = false;
//     doUpdateLastKey(pQueryAttr);
//     return;
//   }
H
Haojun Liao 已提交
1410
//
L
Liu Jicong 已提交
1411 1412 1413 1414 1415 1416
//   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 已提交
1417
//
L
Liu Jicong 已提交
1418 1419 1420
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     return;
//   }
H
Haojun Liao 已提交
1421
//
L
Liu Jicong 已提交
1422 1423 1424 1425
//   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 已提交
1426 1427
////               pQueryAttr->window.ekey, pQueryAttr->window.ekey, pQueryAttr->window.skey);
//
wafwerar's avatar
wafwerar 已提交
1428
//        TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1429 1430 1431 1432 1433 1434 1435 1436 1437 1438
//        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 已提交
1439
//        TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451
//        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 已提交
1452 1453
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
1454
//
wafwerar's avatar
wafwerar 已提交
1455
//          TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1456 1457 1458 1459 1460 1461 1462 1463
//          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 已提交
1464 1465
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
1466
//
wafwerar's avatar
wafwerar 已提交
1467
//          TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey);
H
Haojun Liao 已提交
1468 1469 1470 1471 1472 1473 1474 1475 1476
//          doUpdateLastKey(pQueryAttr);
//        }
//
//        pQueryAttr->order.order = TSDB_ORDER_DESC;
//        pQueryAttr->needReverseScan = false;
//      }
//    }
//  }
//}
1477

H
Haojun Liao 已提交
1478 1479
static void getIntermediateBufInfo(STaskRuntimeEnv* pRuntimeEnv, int32_t* ps, int32_t* rowsize) {
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
L
Liu Jicong 已提交
1480
  int32_t    MIN_ROWS_PER_PAGE = 4;
1481

L
Liu Jicong 已提交
1482 1483
  *rowsize = (int32_t)(pQueryAttr->resultRowSize *
                       getRowNumForMultioutput(pQueryAttr, pQueryAttr->topBotQuery, pQueryAttr->stableQuery));
1484 1485 1486 1487
  int32_t overhead = sizeof(SFilePage);

  // one page contains at least two rows
  *ps = DEFAULT_INTERN_BUF_PAGE_SIZE;
L
Liu Jicong 已提交
1488
  while (((*rowsize) * MIN_ROWS_PER_PAGE) > (*ps) - overhead) {
1489 1490 1491 1492
    *ps = ((*ps) << 1u);
  }
}

L
Liu Jicong 已提交
1493 1494 1495
// static FORCE_INLINE bool doFilterByBlockStatistics(STaskRuntimeEnv* pRuntimeEnv, SDataStatis *pDataStatis,
// SqlFunctionCtx *pCtx, int32_t numOfRows) {
//   STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
1496
//
L
Liu Jicong 已提交
1497 1498 1499
//   if (pDataStatis == NULL || pQueryAttr->pFilters == NULL) {
//     return true;
//   }
1500
//
L
Liu Jicong 已提交
1501 1502
//   return filterRangeExecute(pQueryAttr->pFilters, pDataStatis, pQueryAttr->numOfCols, numOfRows);
// }
1503

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

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

1510
  if (true) {
L
Liu Jicong 已提交
1511
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.skey, sk, ek, &w);
1512 1513 1514 1515 1516 1517
    assert(w.ekey >= pBlockInfo->window.skey);

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

L
Liu Jicong 已提交
1518 1519
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
1520 1521 1522 1523 1524 1525 1526 1527 1528 1529
      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 已提交
1530
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.ekey, sk, ek, &w);
1531 1532 1533 1534 1535 1536
    assert(w.skey <= pBlockInfo->window.ekey);

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

L
Liu Jicong 已提交
1537 1538
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602
      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 SColumnInfo* doGetTagColumnInfoById(SColumnInfo* pTagColList, int32_t numOfTags, int16_t colId);
L
Liu Jicong 已提交
1603
static void         doSetTagValueInParam(void* pTable, int32_t tagColId, SVariant* tag, int16_t type, int16_t bytes);
1604 1605

static uint32_t doFilterByBlockTimeWindow(STableScanInfo* pTableScanInfo, SSDataBlock* pBlock) {
H
Haojun Liao 已提交
1606
  SqlFunctionCtx* pCtx = pTableScanInfo->pCtx;
1607
  uint32_t        status = BLK_DATA_NOT_LOAD;
1608 1609 1610 1611

  int32_t numOfOutput = pTableScanInfo->numOfOutput;
  for (int32_t i = 0; i < numOfOutput; ++i) {
    int32_t functionId = pCtx[i].functionId;
H
Haojun Liao 已提交
1612
    int32_t colId = pTableScanInfo->pExpr[i].base.pParam[0].pCol->colId;
1613 1614 1615

    // group by + first/last should not apply the first/last block filter
    if (functionId < 0) {
1616
      status |= BLK_DATA_DATA_LOAD;
1617 1618
      return status;
    } else {
L
Liu Jicong 已提交
1619
      //      status |= aAggs[functionId].dataReqFunc(&pTableScanInfo->pCtx[i], &pBlock->info.window, colId);
1620
      //      if ((status & BLK_DATA_DATA_LOAD) == BLK_DATA_DATA_LOAD) {
L
Liu Jicong 已提交
1621 1622
      //        return status;
      //      }
1623 1624 1625 1626 1627 1628
    }
  }

  return status;
}

L
Liu Jicong 已提交
1629 1630
int32_t loadDataBlockOnDemand(SExecTaskInfo* pTaskInfo, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock,
                              uint32_t* status) {
1631
  *status = BLK_DATA_NOT_LOAD;
1632

H
Haojun Liao 已提交
1633
  pBlock->pDataBlock = NULL;
L
Liu Jicong 已提交
1634
  pBlock->pBlockAgg = NULL;
H
Haojun Liao 已提交
1635

L
Liu Jicong 已提交
1636 1637
  //  int64_t groupId = pRuntimeEnv->current->groupIndex;
  //  bool    ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
1638

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

1641 1642
//  pCost->totalBlocks += 1;
//  pCost->totalRows += pBlock->info.rows;
H
Haojun Liao 已提交
1643
#if 0
1644 1645 1646
  // 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 已提交
1647
      (QUERY_IS_INTERVAL_QUERY(pQueryAttr) && overlapWithTimeWindow(pTaskInfo, &pBlock->info))) {
1648
    (*status) = BLK_DATA_DATA_LOAD;
1649 1650 1651
  }

  // check if this data block is required to load
1652
  if ((*status) != BLK_DATA_DATA_LOAD) {
1653 1654 1655 1656 1657 1658 1659
    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 已提交
1660
      bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683
      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 {
1684
      (*status) = BLK_DATA_DATA_LOAD;
1685 1686 1687 1688
    }
  }

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

1691
  if ((*status) == BLK_DATA_NOT_LOAD || (*status) == BLK_DATA_FILTEROUT) {
1692 1693
    //qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId, pBlockInfo->window.skey,
//           pBlockInfo->window.ekey, pBlockInfo->rows);
1694
    pCost->skipBlocks += 1;
1695
  } else if ((*status) == BLK_DATA_SMA_LOAD) {
1696 1697
    // this function never returns error?
    pCost->loadBlockStatis += 1;
1698
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
1699 1700

    if (pBlock->pBlockAgg == NULL) {  // data block statistics does not exist, load data block
1701
//      pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
1702 1703 1704
      pCost->totalCheckedRows += pBlock->info.rows;
    }
  } else {
1705
    assert((*status) == BLK_DATA_DATA_LOAD);
1706 1707 1708

    // load the data block statistics to perform further filter
    pCost->loadBlockStatis += 1;
1709
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
1710 1711 1712 1713 1714 1715

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

H
Haojun Liao 已提交
1716
          bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733
          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
1734
            pCost->skipBlocks += 1;
1735 1736
            //qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId,
//                   pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
1737
            (*status) = BLK_DATA_FILTEROUT;
1738 1739 1740 1741 1742 1743 1744 1745
            return TSDB_CODE_SUCCESS;
          }
        }
      }
    }

    // current block has been discard due to filter applied
//    if (!doFilterByBlockStatistics(pRuntimeEnv, pBlock->pBlockAgg, pTableScanInfo->pCtx, pBlockInfo->rows)) {
1746
//      pCost->skipBlocks += 1;
1747 1748
//      qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId, pBlockInfo->window.skey,
//             pBlockInfo->window.ekey, pBlockInfo->rows);
1749
//      (*status) = BLK_DATA_FILTEROUT;
1750 1751 1752 1753 1754
//      return TSDB_CODE_SUCCESS;
//    }

    pCost->totalCheckedRows += pBlockInfo->rows;
    pCost->loadBlocks += 1;
1755
//    pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
1756 1757 1758 1759 1760 1761 1762
//    if (pBlock->pDataBlock == NULL) {
//      return terrno;
//    }

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

1764 1765 1766 1767
//    if (pQueryAttr->pFilters != NULL || pRuntimeEnv->pTsBuf != NULL) {
//      filterColRowsInDataBlock(pRuntimeEnv, pBlock, ascQuery);
//    }
  }
H
Haojun Liao 已提交
1768
#endif
1769 1770 1771 1772
  return TSDB_CODE_SUCCESS;
}

/*
H
Haojun Liao 已提交
1773
 * set tag value in SqlFunctionCtx
1774 1775
 * e.g.,tag information into input buffer
 */
L
Liu Jicong 已提交
1776
static void doSetTagValueInParam(void* pTable, int32_t tagColId, SVariant* tag, int16_t type, int16_t bytes) {
1777 1778 1779
  taosVariantDestroy(tag);

  char* val = NULL;
L
Liu Jicong 已提交
1780 1781 1782 1783 1784 1785
  //  if (tagColId == TSDB_TBNAME_COLUMN_INDEX) {
  //    val = tsdbGetTableName(pTable);
  //    assert(val != NULL);
  //  } else {
  //    val = tsdbGetTableTagVal(pTable, tagColId, type, bytes);
  //  }
1786 1787 1788 1789 1790 1791 1792 1793

  if (val == NULL || isNull(val, type)) {
    tag->nType = TSDB_DATA_TYPE_NULL;
    return;
  }

  if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
    int32_t maxLen = bytes - VARSTR_HEADER_SIZE;
L
Liu Jicong 已提交
1794
    int32_t len = (varDataLen(val) > maxLen) ? maxLen : varDataLen(val);
1795
    taosVariantCreateFromBinary(tag, varDataVal(val), len, type);
L
Liu Jicong 已提交
1796
    // taosVariantCreateFromBinary(tag, varDataVal(val), varDataLen(val), type);
1797 1798 1799 1800 1801 1802 1803 1804
  } else {
    taosVariantCreateFromBinary(tag, val, bytes, type);
  }
}

static SColumnInfo* doGetTagColumnInfoById(SColumnInfo* pTagColList, int32_t numOfTags, int16_t colId) {
  assert(pTagColList != NULL && numOfTags > 0);

L
Liu Jicong 已提交
1805
  for (int32_t i = 0; i < numOfTags; ++i) {
1806 1807 1808 1809 1810 1811 1812 1813
    if (pTagColList[i].colId == colId) {
      return &pTagColList[i];
    }
  }

  return NULL;
}

L
Liu Jicong 已提交
1814 1815
void setTagValue(SOperatorInfo* pOperatorInfo, void* pTable, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
  SExprInfo* pExpr = pOperatorInfo->pExpr;
1816
  SExprInfo* pExprInfo = &pExpr[0];
L
Liu Jicong 已提交
1817
  int32_t    functionId = getExprFunctionId(pExprInfo);
1818
#if 0
1819 1820 1821
  if (pQueryAttr->numOfOutput == 1 && functionId == FUNCTION_TS_COMP && pQueryAttr->stableQuery) {
    assert(pExprInfo->base.numOfParams == 1);

L
Liu Jicong 已提交
1822
    //    int16_t      tagColId = (int16_t)pExprInfo->base.param[0].i;
H
Haojun Liao 已提交
1823
    int16_t      tagColId = -1;
1824 1825 1826
    SColumnInfo* pColInfo = doGetTagColumnInfoById(pQueryAttr->tagColList, pQueryAttr->numOfTags, tagColId);

    doSetTagValueInParam(pTable, tagColId, &pCtx[0].tag, pColInfo->type, pColInfo->bytes);
H
Haojun Liao 已提交
1827

1828 1829 1830 1831 1832 1833 1834 1835 1836
  } else {
    // set tag value, by which the results are aggregated.
    int32_t offset = 0;
    memset(pRuntimeEnv->tagVal, 0, pQueryAttr->tagLen);

    for (int32_t idx = 0; idx < numOfOutput; ++idx) {
      SExprInfo* pLocalExprInfo = &pExpr[idx];

      // ts_comp column required the tag value for join filter
H
Haojun Liao 已提交
1837
      if (!TSDB_COL_IS_TAG(pLocalExprInfo->base.pParam[0].pCol->flag)) {
1838 1839 1840 1841
        continue;
      }

      // todo use tag column index to optimize performance
L
Liu Jicong 已提交
1842 1843
      doSetTagValueInParam(pTable, pLocalExprInfo->base.pParam[0].pCol->colId, &pCtx[idx].tag,
                           pLocalExprInfo->base.resSchema.type, pLocalExprInfo->base.resSchema.bytes);
1844

L
Liu Jicong 已提交
1845 1846 1847
      if (IS_NUMERIC_TYPE(pLocalExprInfo->base.resSchema.type) ||
          pLocalExprInfo->base.resSchema.type == TSDB_DATA_TYPE_BOOL ||
          pLocalExprInfo->base.resSchema.type == TSDB_DATA_TYPE_TIMESTAMP) {
1848 1849 1850 1851
        memcpy(pRuntimeEnv->tagVal + offset, &pCtx[idx].tag.i, pLocalExprInfo->base.resSchema.bytes);
      } else {
        if (pCtx[idx].tag.pz != NULL) {
          memcpy(pRuntimeEnv->tagVal + offset, pCtx[idx].tag.pz, pCtx[idx].tag.nLen);
L
Liu Jicong 已提交
1852
        }
1853 1854 1855 1856 1857 1858 1859 1860 1861 1862
      }

      offset += pLocalExprInfo->base.resSchema.bytes;
    }
  }

  // set the tsBuf start position before check each data block
  if (pRuntimeEnv->pTsBuf != NULL) {
    setCtxTagForJoin(pRuntimeEnv, &pCtx[0], pExprInfo, pTable);
  }
1863
#endif
1864 1865
}

H
Haojun Liao 已提交
1866
void copyToSDataBlock(SSDataBlock* pBlock, int32_t* offset, SGroupResInfo* pGroupResInfo, SDiskbasedBuf* pResBuf) {
1867 1868 1869 1870 1871 1872 1873
  pBlock->info.rows = 0;

  int32_t code = TSDB_CODE_SUCCESS;
  while (pGroupResInfo->currentGroup < pGroupResInfo->totalGroup) {
    // all results in current group have been returned to client, try next group
    if ((pGroupResInfo->pRows == NULL) || taosArrayGetSize(pGroupResInfo->pRows) == 0) {
      assert(pGroupResInfo->index == 0);
L
Liu Jicong 已提交
1874 1875 1876
      //      if ((code = mergeIntoGroupResult(&pGroupResInfo, pRuntimeEnv, offset)) != TSDB_CODE_SUCCESS) {
      return;
      //      }
1877 1878
    }

L
Liu Jicong 已提交
1879
    //    doCopyToSDataBlock(pResBuf, pGroupResInfo, TSDB_ORDER_ASC, pBlock, );
1880 1881 1882 1883 1884 1885 1886 1887 1888 1889

    // current data are all dumped to result buffer, clear it
    if (!hasRemainDataInCurrentGroup(pGroupResInfo)) {
      cleanupGroupResInfo(pGroupResInfo);
      if (!incNextGroup(pGroupResInfo)) {
        break;
      }
    }

    // enough results in data buffer, return
L
Liu Jicong 已提交
1890 1891 1892
    //    if (pBlock->info.rows >= threshold) {
    //      break;
    //    }
1893 1894 1895
  }
}

L
Liu Jicong 已提交
1896
static void updateTableQueryInfoForReverseScan(STableQueryInfo* pTableQueryInfo) {
1897 1898 1899 1900
  if (pTableQueryInfo == NULL) {
    return;
  }

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

L
Liu Jicong 已提交
1904 1905
  //  SWITCH_ORDER(pTableQueryInfo->cur.order);
  //  pTableQueryInfo->cur.vgroupIndex = -1;
1906 1907

  // set the index to be the end slot of result rows array
dengyihao's avatar
dengyihao 已提交
1908 1909 1910 1911 1912 1913
  //  SResultRowInfo* pResultRowInfo = &pTableQueryInfo->resInfo;
  //  if (pResultRowInfo->size > 0) {
  //    pResultRowInfo->curPos = pResultRowInfo->size - 1;
  //  } else {
  //    pResultRowInfo->curPos = -1;
  //  }
1914 1915
}

H
Haojun Liao 已提交
1916
void initResultRow(SResultRow* pResultRow) {
X
Xiaoyu Wang 已提交
1917
  //  pResultRow->pEntryInfo = (struct SResultRowEntryInfo*)((char*)pResultRow + sizeof(SResultRow));
1918 1919 1920 1921 1922
}

/*
 * 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 已提交
1923 1924 1925
 * +------------+-----------------result column 1------------+------------------result column 2-----------+
 * | SResultRow | SResultRowEntryInfo | intermediate buffer1 | SResultRowEntryInfo | intermediate buffer 2|
 * +------------+--------------------------------------------+--------------------------------------------+
1926 1927
 *           offset[0]                                  offset[1]                                   offset[2]
 */
1928
// TODO refactor: some function move away
X
Xiaoyu Wang 已提交
1929 1930
void setFunctionResultOutput(SOptrBasicInfo* pInfo, SAggSupporter* pSup, int32_t stage, int32_t numOfExprs,
                             SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
1931 1932 1933
  SqlFunctionCtx* pCtx = pInfo->pCtx;
  SSDataBlock*    pDataBlock = pInfo->pRes;
  int32_t*        rowCellInfoOffset = pInfo->rowCellInfoOffset;
H
Haojun Liao 已提交
1934

H
Haojun Liao 已提交
1935
  SResultRowInfo* pResultRowInfo = &pInfo->resultRowInfo;
H
Haojun Liao 已提交
1936
  initResultRowInfo(pResultRowInfo, 16);
H
Haojun Liao 已提交
1937

L
Liu Jicong 已提交
1938 1939
  int64_t     tid = 0;
  int64_t     groupId = 0;
1940 1941
  SResultRow* pRow = doSetResultOutBufByKey(pSup->pResultBuf, pResultRowInfo, (char*)&tid, sizeof(tid), true, groupId,
                                            pTaskInfo, false, pSup);
H
Haojun Liao 已提交
1942

H
Haojun Liao 已提交
1943
  ASSERT(pDataBlock->info.numOfCols == numOfExprs);
H
Haojun Liao 已提交
1944 1945 1946 1947
  for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
    struct SResultRowEntryInfo* pEntry = getResultCell(pRow, i, rowCellInfoOffset);
    cleanupResultRowEntry(pEntry);

L
Liu Jicong 已提交
1948
    pCtx[i].resultInfo = pEntry;
1949
    pCtx[i].scanFlag = stage;
H
Haojun Liao 已提交
1950 1951

    // set the timestamp output buffer for top/bottom/diff query
L
Liu Jicong 已提交
1952 1953
    //    int32_t fid = pCtx[i].functionId;
    //    if (fid == FUNCTION_TOP || fid == FUNCTION_BOTTOM || fid == FUNCTION_DIFF || fid == FUNCTION_DERIVATIVE) {
H
Haojun Liao 已提交
1954
    //      if (i > 0) pCtx[i].pTsOutput = pCtx[i-1].pOutput;
L
Liu Jicong 已提交
1955
    //    }
H
Haojun Liao 已提交
1956 1957 1958 1959 1960
  }

  initCtxOutputBuffer(pCtx, pDataBlock->info.numOfCols);
}

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

L
Liu Jicong 已提交
1964
  int32_t newSize = pDataBlock->info.rows + numOfInputRows + 5;  // extra output buffer
1965
  if ((*bufCapacity) < newSize) {
L
Liu Jicong 已提交
1966 1967
    for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
      SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
1968

wafwerar's avatar
wafwerar 已提交
1969
      char* p = taosMemoryRealloc(pColInfo->pData, newSize * pColInfo->info.bytes);
1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982
      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 已提交
1983
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
1984 1985 1986 1987 1988
    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;

L
Liu Jicong 已提交
1989 1990
    if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF ||
        functionId == FUNCTION_DERIVATIVE) {
dengyihao's avatar
dengyihao 已提交
1991
      //      if (i > 0) pBInfo->pCtx[i].pTsOutput = pBInfo->pCtx[i - 1].pOutput;
1992 1993 1994 1995
    }
  }
}

H
Haojun Liao 已提交
1996
void copyTsColoum(SSDataBlock* pRes, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
1997 1998
  bool    needCopyTs = false;
  int32_t tsNum = 0;
L
Liu Jicong 已提交
1999
  char*   src = NULL;
2000 2001 2002 2003
  for (int32_t i = 0; i < numOfOutput; i++) {
    int32_t functionId = pCtx[i].functionId;
    if (functionId == FUNCTION_DIFF || functionId == FUNCTION_DERIVATIVE) {
      needCopyTs = true;
L
Liu Jicong 已提交
2004 2005
      if (i > 0 && pCtx[i - 1].functionId == FUNCTION_TS_DUMMY) {
        SColumnInfoData* pColRes = taosArrayGet(pRes->pDataBlock, i - 1);  // find ts data
2006 2007
        src = pColRes->pData;
      }
L
Liu Jicong 已提交
2008
    } else if (functionId == FUNCTION_TS_DUMMY) {
2009 2010 2011 2012 2013 2014 2015 2016 2017 2018
      tsNum++;
    }
  }

  if (!needCopyTs) return;
  if (tsNum < 2) return;
  if (src == NULL) return;

  for (int32_t i = 0; i < numOfOutput; i++) {
    int32_t functionId = pCtx[i].functionId;
L
Liu Jicong 已提交
2019
    if (functionId == FUNCTION_TS_DUMMY) {
2020 2021 2022 2023 2024 2025
      SColumnInfoData* pColRes = taosArrayGet(pRes->pDataBlock, i);
      memcpy(pColRes->pData, src, pColRes->info.bytes * pRes->info.rows);
    }
  }
}

H
Haojun Liao 已提交
2026
void initCtxOutputBuffer(SqlFunctionCtx* pCtx, int32_t size) {
2027 2028
  for (int32_t j = 0; j < size; ++j) {
    struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(&pCtx[j]);
dengyihao's avatar
dengyihao 已提交
2029 2030
    if (isRowEntryInitialized(pResInfo) || fmIsPseudoColumnFunc(pCtx[j].functionId) || pCtx[j].functionId == -1 ||
        fmIsScalarFunc(pCtx[j].functionId)) {
2031 2032 2033
      continue;
    }

H
Haojun Liao 已提交
2034
    pCtx[j].fpSet.init(&pCtx[j], pCtx[j].resultInfo);
2035 2036 2037
  }
}

L
Liu Jicong 已提交
2038
void setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status) {
2039
  if (status == TASK_NOT_COMPLETED) {
H
Haojun Liao 已提交
2040
    pTaskInfo->status = status;
2041 2042
  } else {
    // QUERY_NOT_COMPLETED is not compatible with any other status, so clear its position first
2043
    CLEAR_QUERY_STATUS(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
2044
    pTaskInfo->status |= status;
2045 2046 2047
  }
}

2048
// todo merged with the build group result.
2049 2050
void finalizeMultiTupleQueryResult(int32_t numOfOutput, SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo,
                                   int32_t* rowCellInfoOffset) {
2051 2052 2053
  for (int32_t i = 0; i < pResultRowInfo->size; ++i) {
    SResultRowPosition* pPos = &pResultRowInfo->pPosition[i];

L
Liu Jicong 已提交
2054
    SFilePage*  bufPage = getBufPage(pBuf, pPos->pageId);
2055
    SResultRow* pRow = (SResultRow*)((char*)bufPage + pPos->offset);
H
Haojun Liao 已提交
2056 2057

    // TODO ignore the close status anyway.
dengyihao's avatar
dengyihao 已提交
2058 2059 2060
    //    if (!isResultRowClosed(pRow)) {
    //      continue;
    //    }
2061

2062
    for (int32_t j = 0; j < numOfOutput; ++j) {
2063
      struct SResultRowEntryInfo* pResInfo = getResultCell(pRow, j, rowCellInfoOffset);
H
Haojun Liao 已提交
2064
      if (!isRowEntryInitialized(pResInfo)) {
2065 2066 2067 2068 2069 2070
        continue;
      }

      if (pRow->numOfRows < pResInfo->numOfRes) {
        pRow->numOfRows = pResInfo->numOfRes;
      }
2071
    }
2072 2073 2074

    releaseBufPage(pBuf, bufPage);
  }
2075 2076
}

2077
STableQueryInfo* createTableQueryInfo(void* buf, STimeWindow win) {
L
Liu Jicong 已提交
2078
  STableQueryInfo* pTableQueryInfo = buf;
2079 2080 2081 2082
  pTableQueryInfo->lastKey = win.skey;
  return pTableQueryInfo;
}

L
Liu Jicong 已提交
2083
void destroyTableQueryInfoImpl(STableQueryInfo* pTableQueryInfo) {
2084 2085 2086 2087
  if (pTableQueryInfo == NULL) {
    return;
  }

L
Liu Jicong 已提交
2088
  //  taosVariantDestroy(&pTableQueryInfo->tag);
dengyihao's avatar
dengyihao 已提交
2089
  //  cleanupResultRowInfo(&pTableQueryInfo->resInfo);
2090 2091
}

2092
void setResultRowInitCtx(SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowCellInfoOffset) {
2093 2094 2095 2096 2097 2098 2099
  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;
    }
2100 2101 2102 2103 2104

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

2105 2106 2107 2108 2109 2110
    if (!pResInfo->initialized) {
      if (pCtx[i].functionId != -1) {
        pCtx[i].fpSet.init(&pCtx[i], pResInfo);
      } else {
        pResInfo->initialized = true;
      }
2111 2112 2113 2114
    }
  }
}

2115 2116 2117 2118 2119 2120
void doFilter(const SNode* pFilterNode, SSDataBlock* pBlock) {
  if (pFilterNode == NULL) {
    return;
  }

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

H
Haojun Liao 已提交
2122
  // todo move to the initialization function
H
Haojun Liao 已提交
2123
  int32_t code = filterInitFromNode((SNode*)pFilterNode, &filter, 0);
2124 2125 2126 2127 2128 2129

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

  int8_t* rowRes = NULL;
  bool    keep = filterExecute(filter, pBlock, &rowRes, NULL, param1.numOfCols);
D
dapan1121 已提交
2130
  filterFreeInfo(filter);
2131

2132
  SSDataBlock* px = createOneDataBlock(pBlock, false);
2133 2134
  blockDataEnsureCapacity(px, pBlock->info.rows);

H
Haojun Liao 已提交
2135
  // todo extract method
2136 2137 2138 2139
  int32_t numOfRow = 0;
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
    SColumnInfoData* pDst = taosArrayGet(px->pDataBlock, i);
    SColumnInfoData* pSrc = taosArrayGet(pBlock->pDataBlock, i);
D
dapan1121 已提交
2140 2141 2142 2143 2144 2145 2146 2147 2148
    if (keep) {
      colDataAssign(pDst, pSrc, pBlock->info.rows);
      numOfRow = pBlock->info.rows;
    } else if (NULL != rowRes) {
      numOfRow = 0;
      for (int32_t j = 0; j < pBlock->info.rows; ++j) {
        if (rowRes[j] == 0) {
          continue;
        }
2149

D
dapan1121 已提交
2150 2151 2152 2153 2154 2155
        if (colDataIsNull_s(pSrc, j)) {
          colDataAppendNULL(pDst, numOfRow);
        } else {
          colDataAppend(pDst, numOfRow, colDataGetData(pSrc, j), false);
        }
        numOfRow += 1;
H
Haojun Liao 已提交
2156
      }
D
dapan1121 已提交
2157 2158
    } else {
      numOfRow = 0;
2159 2160 2161 2162 2163 2164
    }

    *pSrc = *pDst;
  }

  pBlock->info.rows = numOfRow;
2165
  blockDataUpdateTsWindow(pBlock);
2166 2167
}

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

2173 2174
  SResultRowInfo* pResultRowInfo = &pAggInfo->binfo.resultRowInfo;
  SqlFunctionCtx* pCtx = pAggInfo->binfo.pCtx;
L
Liu Jicong 已提交
2175
  int32_t*        rowCellInfoOffset = pAggInfo->binfo.rowCellInfoOffset;
2176

2177
  SResultRow* pResultRow = doSetResultOutBufByKey(pAggInfo->aggSup.pResultBuf, pResultRowInfo, (char*)&groupId,
L
Liu Jicong 已提交
2178
                                                  sizeof(groupId), true, groupId, pTaskInfo, false, &pAggInfo->aggSup);
L
Liu Jicong 已提交
2179
  assert(pResultRow != NULL);
2180 2181 2182 2183 2184 2185

  /*
   * 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 已提交
2186 2187
    int32_t ret =
        addNewWindowResultBuf(pResultRow, pAggInfo->aggSup.pResultBuf, groupId, pAggInfo->binfo.pRes->info.rowSize);
2188 2189 2190 2191 2192
    if (ret != TSDB_CODE_SUCCESS) {
      return;
    }
  }

2193
  setResultRowInitCtx(pResultRow, pCtx, numOfOutput, rowCellInfoOffset);
2194 2195
}

H
Haojun Liao 已提交
2196 2197
void setExecutionContext(int32_t numOfOutput, uint64_t groupId, SExecTaskInfo* pTaskInfo, SAggOperatorInfo* pAggInfo) {
  if (pAggInfo->groupId != INT32_MIN && pAggInfo->groupId == groupId) {
2198 2199 2200
    return;
  }

H
Haojun Liao 已提交
2201
  doSetTableGroupOutputBuf(pAggInfo, numOfOutput, groupId, pTaskInfo);
2202 2203

  // record the current active group id
H
Haojun Liao 已提交
2204
  pAggInfo->groupId = groupId;
2205 2206 2207 2208 2209 2210 2211 2212 2213 2214
}

/**
 * For interval query of both super table and table, copy the data in ascending order, since the output results are
 * ordered in SWindowResutl already. While handling the group by query for both table and super table,
 * all group result are completed already.
 *
 * @param pQInfo
 * @param result
 */
2215
int32_t doCopyToSDataBlock(SExecTaskInfo* taskInfo, SSDataBlock* pBlock, SExprInfo* pExprInfo, SDiskbasedBuf* pBuf, SGroupResInfo* pGroupResInfo,
2216
                           int32_t orderType, int32_t* rowCellOffset, SqlFunctionCtx* pCtx, int32_t numOfExprs) {
2217
  int32_t numOfRows = getNumOfTotalRes(pGroupResInfo);
L
Liu Jicong 已提交
2218
  int32_t numOfResult = pBlock->info.rows;  // there are already exists result rows
2219 2220

  int32_t start = 0;
2221
  int32_t step = 1;
2222

L
Liu Jicong 已提交
2223
  // qDebug("QInfo:0x%"PRIx64" start to copy data from windowResInfo to output buf", GET_TASKID(pRuntimeEnv));
2224 2225 2226 2227 2228 2229 2230 2231 2232
  assert(orderType == TSDB_ORDER_ASC || orderType == TSDB_ORDER_DESC);

  if (orderType == TSDB_ORDER_ASC) {
    start = pGroupResInfo->index;
  } else {  // desc order copy all data
    start = numOfRows - pGroupResInfo->index - 1;
  }

  for (int32_t i = start; (i < numOfRows) && (i >= 0); i += step) {
L
Liu Jicong 已提交
2233 2234
    SResKeyPos* pPos = taosArrayGetP(pGroupResInfo->pRows, i);
    SFilePage*  page = getBufPage(pBuf, pPos->pos.pageId);
2235

2236
    SResultRow* pRow = (SResultRow*)((char*)page + pPos->pos.offset);
2237 2238 2239 2240 2241
    if (pRow->numOfRows == 0) {
      pGroupResInfo->index += 1;
      continue;
    }

2242
    // TODO copy multiple rows?
2243
    int32_t numOfRowsToCopy = pRow->numOfRows;
2244
    if (numOfResult + numOfRowsToCopy >= pBlock->info.capacity) {
2245 2246 2247 2248 2249
      break;
    }

    pGroupResInfo->index += 1;

2250
    for (int32_t j = 0; j < numOfExprs; ++j) {
2251 2252
      int32_t slotId = pExprInfo[j].base.resSchema.slotId;

2253
      pCtx[j].resultInfo = getResultCell(pRow, j, rowCellOffset);
2254
      if (pCtx[j].fpSet.finalize) {
2255
        int32_t code = pCtx[j].fpSet.finalize(&pCtx[j], pBlock);
2256
        if (TAOS_FAILED(code)) {
2257 2258 2259 2260
          qError("%s build result data block error, code %s", GET_TASKID(taskInfo), tstrerror(code));
          taskInfo->code = code;
          longjmp(taskInfo->env, code);
        }
2261 2262
      } else if (strcmp(pCtx[j].pExpr->pExpr->_function.functionName, "_select_value") == 0) {
        // do nothing, todo refactor
2263
      } else {
2264 2265 2266 2267 2268 2269 2270
        // 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.
          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);
          }
2271
      }
2272 2273
    }

2274 2275
    releaseBufPage(pBuf, page);

2276 2277
    pBlock->info.rows += pRow->numOfRows;
    if (pBlock->info.rows >= pBlock->info.capacity) {  // output buffer is full
2278 2279 2280 2281
      break;
    }
  }

L
Liu Jicong 已提交
2282
  // qDebug("QInfo:0x%"PRIx64" copy data to query buf completed", GET_TASKID(pRuntimeEnv));
2283
  blockDataUpdateTsWindow(pBlock);
2284 2285 2286
  return 0;
}

2287
void doBuildResultDatablock(SOperatorInfo* pOperator, SOptrBasicInfo* pbInfo, SGroupResInfo* pGroupResInfo, SDiskbasedBuf* pBuf) {
2288 2289
  assert(pGroupResInfo->currentGroup <= pGroupResInfo->totalGroup);

2290 2291 2292 2293
  SExprInfo* pExprInfo = pOperator->pExpr;
  int32_t numOfExprs = pOperator->numOfExprs;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;

X
Xiaoyu Wang 已提交
2294 2295 2296
  int32_t*        rowCellOffset = pbInfo->rowCellInfoOffset;
  SSDataBlock*    pBlock = pbInfo->pRes;
  SqlFunctionCtx* pCtx = pbInfo->pCtx;
2297

2298
  blockDataCleanup(pBlock);
2299 2300 2301 2302
  if (!hasRemainDataInCurrentGroup(pGroupResInfo)) {
    return;
  }

2303
  int32_t orderType = TSDB_ORDER_ASC;
2304
  doCopyToSDataBlock(pTaskInfo, pBlock, pExprInfo, pBuf, pGroupResInfo, orderType, rowCellOffset, pCtx, numOfExprs);
2305

H
Haojun Liao 已提交
2306 2307
  // add condition (pBlock->info.rows >= 1) just to runtime happy
  blockDataUpdateTsWindow(pBlock);
2308 2309
}

L
Liu Jicong 已提交
2310 2311
static void updateNumOfRowsInResultRows(SqlFunctionCtx* pCtx, int32_t numOfOutput, SResultRowInfo* pResultRowInfo,
                                        int32_t* rowCellInfoOffset) {
2312
  // update the number of result for each, only update the number of rows for the corresponding window result.
L
Liu Jicong 已提交
2313 2314 2315
  //  if (QUERY_IS_INTERVAL_QUERY(pQueryAttr)) {
  //    return;
  //  }
H
Haojun Liao 已提交
2316
#if 0
2317
  for (int32_t i = 0; i < pResultRowInfo->size; ++i) {
L
Liu Jicong 已提交
2318
    SResultRow* pResult = pResultRowInfo->pResult[i];
2319 2320 2321 2322 2323 2324 2325

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

2326 2327
      SResultRowEntryInfo* pCell = getResultCell(pResult, j, rowCellInfoOffset);
      pResult->numOfRows = (uint16_t)(TMAX(pResult->numOfRows, pCell->numOfRes));
2328 2329
    }
  }
H
Haojun Liao 已提交
2330
#endif
2331 2332
}

L
Liu Jicong 已提交
2333
static int32_t compressQueryColData(SColumnInfoData* pColRes, int32_t numOfRows, char* data, int8_t compressed) {
2334 2335
  int32_t colSize = pColRes->info.bytes * numOfRows;
  return (*(tDataTypes[pColRes->info.type].compFunc))(pColRes->pData, colSize, numOfRows, data,
L
Liu Jicong 已提交
2336
                                                      colSize + COMP_OVERFLOW_BYTES, compressed, NULL, 0);
2337 2338
}

2339
int32_t doFillTimeIntervalGapsInResults(struct SFillInfo* pFillInfo, SSDataBlock* pBlock, int32_t capacity) {
L
Liu Jicong 已提交
2340 2341 2342 2343
  //  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);
  //  }
2344

2345 2346
  int32_t numOfRows = (int32_t)taosFillResultDataBlock(pFillInfo, pBlock, capacity - pBlock->info.rows);
  pBlock->info.rows += numOfRows;
2347

2348
  return pBlock->info.rows;
2349 2350
}

2351
void publishOperatorProfEvent(SOperatorInfo* pOperator, EQueryProfEventType eventType) {
2352 2353
  SQueryProfEvent event = {0};

L
Liu Jicong 已提交
2354 2355
  event.eventType = eventType;
  event.eventTime = taosGetTimestampUs();
2356
  event.operatorType = pOperator->operatorType;
dengyihao's avatar
dengyihao 已提交
2357 2358 2359
  //    if (pQInfo->summary.queryProfEvents) {
  //      taosArrayPush(pQInfo->summary.queryProfEvents, &event);
  //    }
2360 2361
}

L
Liu Jicong 已提交
2362
void publishQueryAbortEvent(SExecTaskInfo* pTaskInfo, int32_t code) {
2363 2364 2365 2366 2367
  SQueryProfEvent event;
  event.eventType = QUERY_PROF_QUERY_ABORT;
  event.eventTime = taosGetTimestampUs();
  event.abortCode = code;

2368 2369
  if (pTaskInfo->cost.queryProfEvents) {
    taosArrayPush(pTaskInfo->cost.queryProfEvents, &event);
2370 2371 2372
  }
}

L
Liu Jicong 已提交
2373
typedef struct {
2374 2375 2376 2377 2378 2379 2380
  uint8_t operatorType;
  int64_t beginTime;
  int64_t endTime;
  int64_t selfTime;
  int64_t descendantsTime;
} SOperatorStackItem;

L
Liu Jicong 已提交
2381 2382
static void doOperatorExecProfOnce(SOperatorStackItem* item, SQueryProfEvent* event, SArray* opStack,
                                   SHashObj* profResults) {
2383 2384 2385 2386 2387 2388 2389 2390
  item->endTime = event->eventTime;
  item->selfTime = (item->endTime - item->beginTime) - (item->descendantsTime);

  for (int32_t j = 0; j < taosArrayGetSize(opStack); ++j) {
    SOperatorStackItem* ancestor = taosArrayGet(opStack, j);
    ancestor->descendantsTime += item->selfTime;
  }

L
Liu Jicong 已提交
2391
  uint8_t              operatorType = item->operatorType;
2392 2393 2394 2395 2396 2397 2398 2399 2400
  SOperatorProfResult* result = taosHashGet(profResults, &operatorType, sizeof(operatorType));
  if (result != NULL) {
    result->sumRunTimes++;
    result->sumSelfTime += item->selfTime;
  } else {
    SOperatorProfResult opResult;
    opResult.operatorType = operatorType;
    opResult.sumSelfTime = item->selfTime;
    opResult.sumRunTimes = 1;
L
Liu Jicong 已提交
2401
    taosHashPut(profResults, &(operatorType), sizeof(operatorType), &opResult, sizeof(opResult));
2402 2403 2404
  }
}

2405
void calculateOperatorProfResults(void) {
dengyihao's avatar
dengyihao 已提交
2406 2407 2408 2409 2410 2411 2412 2413 2414
  //  if (pQInfo->summary.queryProfEvents == NULL) {
  //    // qDebug("QInfo:0x%"PRIx64" query prof events array is null", pQInfo->qId);
  //    return;
  //  }
  //
  //  if (pQInfo->summary.operatorProfResults == NULL) {
  //    // qDebug("QInfo:0x%"PRIx64" operator prof results hash is null", pQInfo->qId);
  //    return;
  //  }
2415 2416 2417 2418 2419

  SArray* opStack = taosArrayInit(32, sizeof(SOperatorStackItem));
  if (opStack == NULL) {
    return;
  }
2420
#if 0
L
Liu Jicong 已提交
2421
  size_t    size = taosArrayGetSize(pQInfo->summary.queryProfEvents);
2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442
  SHashObj* profResults = pQInfo->summary.operatorProfResults;

  for (int i = 0; i < size; ++i) {
    SQueryProfEvent* event = taosArrayGet(pQInfo->summary.queryProfEvents, i);
    if (event->eventType == QUERY_PROF_BEFORE_OPERATOR_EXEC) {
      SOperatorStackItem opItem;
      opItem.operatorType = event->operatorType;
      opItem.beginTime = event->eventTime;
      opItem.descendantsTime = 0;
      taosArrayPush(opStack, &opItem);
    } else if (event->eventType == QUERY_PROF_AFTER_OPERATOR_EXEC) {
      SOperatorStackItem* item = taosArrayPop(opStack);
      assert(item->operatorType == event->operatorType);
      doOperatorExecProfOnce(item, event, opStack, profResults);
    } else if (event->eventType == QUERY_PROF_QUERY_ABORT) {
      SOperatorStackItem* item;
      while ((item = taosArrayPop(opStack)) != NULL) {
        doOperatorExecProfOnce(item, event, opStack, profResults);
      }
    }
  }
2443
#endif
2444 2445 2446
  taosArrayDestroy(opStack);
}

L
Liu Jicong 已提交
2447 2448
void queryCostStatis(SExecTaskInfo* pTaskInfo) {
  STaskCostInfo* pSummary = &pTaskInfo->cost;
2449

L
Liu Jicong 已提交
2450 2451 2452
  //  uint64_t hashSize = taosHashGetMemSize(pQInfo->runtimeEnv.pResultRowHashTable);
  //  hashSize += taosHashGetMemSize(pRuntimeEnv->tableqinfoGroupInfo.map);
  //  pSummary->hashSize = hashSize;
2453 2454 2455 2456

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

L
Liu Jicong 已提交
2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467
  //  SResultRowPool* p = pTaskInfo->pool;
  //  if (p != NULL) {
  //    pSummary->winInfoSize = getResultRowPoolMemSize(p);
  //    pSummary->numOfTimeWindows = getNumOfAllocatedResultRows(p);
  //  } else {
  //    pSummary->winInfoSize = 0;
  //    pSummary->numOfTimeWindows = 0;
  //  }
  //
  //  calculateOperatorProfResults(pQInfo);

2468 2469
  SFileBlockLoadRecorder* pRecorder = pSummary->pRecoder;
  if (pSummary->pRecoder != NULL) {
X
Xiaoyu Wang 已提交
2470 2471
    qDebug("%s :cost summary: elapsed time:%" PRId64 " us, first merge:%" PRId64
           " us, total blocks:%d, "
2472 2473 2474 2475
           "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 已提交
2476 2477 2478
  // 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);
2479 2480 2481 2482

  if (pSummary->operatorProfResults) {
    SOperatorProfResult* opRes = taosHashIterate(pSummary->operatorProfResults, NULL);
    while (opRes != NULL) {
L
Liu Jicong 已提交
2483 2484
      // qDebug("QInfo:0x%" PRIx64 " :cost summary: operator : %d, exec times: %" PRId64 ", self time: %" PRId64,
      //             pQInfo->qId, opRes->operatorType, opRes->sumRunTimes, opRes->sumSelfTime);
2485 2486 2487 2488 2489
      opRes = taosHashIterate(pSummary->operatorProfResults, opRes);
    }
  }
}

L
Liu Jicong 已提交
2490 2491 2492
// static void updateOffsetVal(STaskRuntimeEnv *pRuntimeEnv, SDataBlockInfo *pBlockInfo) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
2493
//
L
Liu Jicong 已提交
2494
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
2495
//
L
Liu Jicong 已提交
2496 2497 2498 2499
//   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;
//   }
2500
//
L
Liu Jicong 已提交
2501 2502 2503 2504 2505
//   if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//     pQueryAttr->pos = (int32_t)pQueryAttr->limit.offset;
//   } else {
//     pQueryAttr->pos = pBlockInfo->rows - (int32_t)pQueryAttr->limit.offset - 1;
//   }
2506
//
L
Liu Jicong 已提交
2507
//   assert(pQueryAttr->pos >= 0 && pQueryAttr->pos <= pBlockInfo->rows - 1);
2508
//
L
Liu Jicong 已提交
2509 2510
//   SArray *         pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//   SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
2511
//
L
Liu Jicong 已提交
2512 2513
//   // update the pQueryAttr->limit.offset value, and pQueryAttr->pos value
//   TSKEY *keys = (TSKEY *) pColInfoData->pData;
2514
//
L
Liu Jicong 已提交
2515 2516 2517
//   // update the offset value
//   pTableQueryInfo->lastKey = keys[pQueryAttr->pos];
//   pQueryAttr->limit.offset = 0;
2518
//
L
Liu Jicong 已提交
2519
//   int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
2520
//
L
Liu Jicong 已提交
2521 2522 2523 2524
//   //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);
// }
2525

L
Liu Jicong 已提交
2526 2527
// void skipBlocks(STaskRuntimeEnv *pRuntimeEnv) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
2528
//
L
Liu Jicong 已提交
2529 2530 2531
//   if (pQueryAttr->limit.offset <= 0 || pQueryAttr->numOfFilterCols > 0) {
//     return;
//   }
2532
//
L
Liu Jicong 已提交
2533 2534
//   pQueryAttr->pos = 0;
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
2535
//
L
Liu Jicong 已提交
2536 2537
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
//   TsdbQueryHandleT pTsdbReadHandle = pRuntimeEnv->pTsdbReadHandle;
2538
//
L
Liu Jicong 已提交
2539 2540 2541 2542 2543
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pTsdbReadHandle)) {
//     if (isTaskKilled(pRuntimeEnv->qinfo)) {
//       longjmp(pRuntimeEnv->env, TSDB_CODE_TSC_QUERY_CANCELLED);
//     }
2544
//
L
Liu Jicong 已提交
2545
//     tsdbRetrieveDataBlockInfo(pTsdbReadHandle, &blockInfo);
2546
//
L
Liu Jicong 已提交
2547 2548 2549 2550
//     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;
2551
//
L
Liu Jicong 已提交
2552 2553 2554 2555 2556 2557 2558
//       //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;
//     }
//   }
2559
//
L
Liu Jicong 已提交
2560 2561 2562 2563 2564 2565 2566 2567 2568
//   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;
2569
//
L
Liu Jicong 已提交
2570 2571 2572
//   assert(pQueryAttr->limit.offset == 0);
//   STimeWindow tw = *win;
//   getNextTimeWindow(pQueryAttr, &tw);
2573
//
L
Liu Jicong 已提交
2574 2575
//   if ((tw.skey <= pBlockInfo->window.ekey && QUERY_IS_ASC_QUERY(pQueryAttr)) ||
//       (tw.ekey >= pBlockInfo->window.skey && !QUERY_IS_ASC_QUERY(pQueryAttr))) {
2576
//
L
Liu Jicong 已提交
2577 2578 2579 2580
//     // 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);
2581
//
L
Liu Jicong 已提交
2582 2583 2584 2585
//     tw = *win;
//     int32_t startPos =
//         getNextQualifiedWindow(pQueryAttr, &tw, pBlockInfo, pColInfoData->pData, binarySearchForKey, -1);
//     assert(startPos >= 0);
2586
//
L
Liu Jicong 已提交
2587 2588
//     // set the abort info
//     pQueryAttr->pos = startPos;
2589
//
L
Liu Jicong 已提交
2590 2591 2592 2593
//     // reset the query start timestamp
//     pTableQueryInfo->win.skey = ((TSKEY *)pColInfoData->pData)[startPos];
//     pQueryAttr->window.skey = pTableQueryInfo->win.skey;
//     TSKEY key = pTableQueryInfo->win.skey;
2594
//
L
Liu Jicong 已提交
2595 2596
//     pWindowResInfo->prevSKey = tw.skey;
//     int32_t index = pRuntimeEnv->resultRowInfo.curIndex;
2597
//
L
Liu Jicong 已提交
2598 2599
//     int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
//     pRuntimeEnv->resultRowInfo.curIndex = index;  // restore the window index
2600
//
L
Liu Jicong 已提交
2601 2602 2603 2604
//     //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);
2605
//
L
Liu Jicong 已提交
2606 2607 2608 2609 2610
//     return key;
//   } else {  // do nothing
//     pQueryAttr->window.skey      = tw.skey;
//     pWindowResInfo->prevSKey = tw.skey;
//     pTableQueryInfo->lastKey = tw.skey;
2611
//
L
Liu Jicong 已提交
2612 2613
//     return tw.skey;
//   }
2614
//
L
Liu Jicong 已提交
2615 2616 2617 2618 2619 2620 2621 2622 2623 2624
//   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);
//   }
2625
//
L
Liu Jicong 已提交
2626 2627 2628 2629 2630
//   // 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;
//   }
2631
//
L
Liu Jicong 已提交
2632 2633 2634 2635 2636 2637 2638
//   /*
//    * 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);
2639
//
L
Liu Jicong 已提交
2640 2641
//   STimeWindow w = TSWINDOW_INITIALIZER;
//   bool ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
2642
//
L
Liu Jicong 已提交
2643 2644
//   SResultRowInfo *pWindowResInfo = &pRuntimeEnv->resultRowInfo;
//   STableQueryInfo *pTableQueryInfo = pRuntimeEnv->current;
2645
//
L
Liu Jicong 已提交
2646 2647 2648
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pRuntimeEnv->pTsdbReadHandle)) {
//     tsdbRetrieveDataBlockInfo(pRuntimeEnv->pTsdbReadHandle, &blockInfo);
2649
//
L
Liu Jicong 已提交
2650 2651 2652 2653 2654 2655 2656 2657 2658
//     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;
//     }
2659
//
L
Liu Jicong 已提交
2660 2661
//     // the first time window
//     STimeWindow win = getActiveTimeWindow(pWindowResInfo, pWindowResInfo->prevSKey, pQueryAttr);
2662
//
L
Liu Jicong 已提交
2663 2664
//     while (pQueryAttr->limit.offset > 0) {
//       STimeWindow tw = win;
2665
//
L
Liu Jicong 已提交
2666 2667 2668
//       if ((win.ekey <= blockInfo.window.ekey && ascQuery) || (win.ekey >= blockInfo.window.skey && !ascQuery)) {
//         pQueryAttr->limit.offset -= 1;
//         pWindowResInfo->prevSKey = win.skey;
2669
//
L
Liu Jicong 已提交
2670 2671 2672 2673 2674 2675
//         // 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;
//         }
//       }
2676
//
L
Liu Jicong 已提交
2677 2678 2679 2680
//       if (pQueryAttr->limit.offset == 0) {
//         *start = doSkipIntervalProcess(pRuntimeEnv, &win, &blockInfo, pTableQueryInfo);
//         return true;
//       }
2681
//
L
Liu Jicong 已提交
2682 2683
//       // current window does not ended in current data block, try next data block
//       getNextTimeWindow(pQueryAttr, &tw);
2684
//
L
Liu Jicong 已提交
2685 2686 2687 2688 2689 2690 2691 2692 2693
//       /*
//        * 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)) {
2694
//
L
Liu Jicong 已提交
2695 2696
//         SArray *pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//         SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
2697
//
L
Liu Jicong 已提交
2698 2699 2700
//         if ((win.ekey > blockInfo.window.ekey && ascQuery) || (win.ekey < blockInfo.window.skey && !ascQuery)) {
//           pQueryAttr->limit.offset -= 1;
//         }
2701
//
L
Liu Jicong 已提交
2702 2703 2704 2705 2706 2707 2708 2709
//         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);
2710
//
L
Liu Jicong 已提交
2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721
//           // 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.
//       }
//     }
//   }
2722
//
L
Liu Jicong 已提交
2723 2724 2725 2726
//   // check for error
//   if (terrno != TSDB_CODE_SUCCESS) {
//     longjmp(pRuntimeEnv->env, terrno);
//   }
2727
//
L
Liu Jicong 已提交
2728 2729
//   return true;
// }
2730

2731
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num) {
H
Haojun Liao 已提交
2732
  if (p->pDownstream == NULL) {
H
Haojun Liao 已提交
2733
    assert(p->numOfDownstream == 0);
2734 2735
  }

wafwerar's avatar
wafwerar 已提交
2736
  p->pDownstream = taosMemoryCalloc(1, num * POINTER_BYTES);
2737 2738 2739 2740 2741 2742 2743
  if (p->pDownstream == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  memcpy(p->pDownstream, pDownstream, num * POINTER_BYTES);
  p->numOfDownstream = num;
  return TSDB_CODE_SUCCESS;
2744 2745 2746 2747
}

static void doDestroyTableQueryInfo(STableGroupInfo* pTableqinfoGroupInfo);

2748
static void doTableQueryInfoTimeWindowCheck(SExecTaskInfo* pTaskInfo, STableQueryInfo* pTableQueryInfo, int32_t order) {
H
Haojun Liao 已提交
2749 2750
#if 0
    if (order == TSDB_ORDER_ASC) {
2751 2752
    assert(
        (pTableQueryInfo->win.skey <= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
2753 2754
        (pTableQueryInfo->lastKey >= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey >= pTaskInfo->window.skey && pTableQueryInfo->win.ekey <= pTaskInfo->window.ekey));
2755 2756 2757
  } else {
    assert(
        (pTableQueryInfo->win.skey >= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
2758 2759
        (pTableQueryInfo->lastKey <= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey <= pTaskInfo->window.skey && pTableQueryInfo->win.ekey >= pTaskInfo->window.ekey));
2760
  }
H
Haojun Liao 已提交
2761
#endif
2762 2763
}

L
Liu Jicong 已提交
2764 2765 2766 2767
// 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;
2768
//
L
Liu Jicong 已提交
2769 2770 2771
//   if (pTableQueryInfo->pTable == NULL) {
//     return;
//   }
2772
//
L
Liu Jicong 已提交
2773 2774 2775 2776 2777 2778 2779 2780 2781
//   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));
//   }
// }
2782

2783
int32_t loadRemoteDataCallback(void* param, const SDataBuf* pMsg, int32_t code) {
L
Liu Jicong 已提交
2784
  SSourceDataInfo* pSourceDataInfo = (SSourceDataInfo*)param;
H
Haojun Liao 已提交
2785 2786
  if (code == TSDB_CODE_SUCCESS) {
    pSourceDataInfo->pRsp = pMsg->pData;
2787

H
Haojun Liao 已提交
2788 2789
    SRetrieveTableRsp* pRsp = pSourceDataInfo->pRsp;
    pRsp->numOfRows = htonl(pRsp->numOfRows);
dengyihao's avatar
dengyihao 已提交
2790
    pRsp->compLen = htonl(pRsp->compLen);
2791
    pRsp->numOfCols = htonl(pRsp->numOfCols);
dengyihao's avatar
dengyihao 已提交
2792
    pRsp->useconds = htobe64(pRsp->useconds);
H
Haojun Liao 已提交
2793 2794 2795
  } else {
    pSourceDataInfo->code = code;
  }
H
Haojun Liao 已提交
2796

H
Haojun Liao 已提交
2797
  pSourceDataInfo->status = EX_SOURCE_DATA_READY;
2798
  tsem_post(&pSourceDataInfo->pEx->ready);
wmmhello's avatar
wmmhello 已提交
2799
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
2800 2801 2802 2803
}

static void destroySendMsgInfo(SMsgSendInfo* pMsgBody) {
  assert(pMsgBody != NULL);
wafwerar's avatar
wafwerar 已提交
2804 2805
  taosMemoryFreeClear(pMsgBody->msgInfo.pData);
  taosMemoryFreeClear(pMsgBody);
H
Haojun Liao 已提交
2806 2807
}

S
Shengliang Guan 已提交
2808
void qProcessFetchRsp(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet) {
L
Liu Jicong 已提交
2809
  SMsgSendInfo* pSendInfo = (SMsgSendInfo*)pMsg->ahandle;
H
Haojun Liao 已提交
2810 2811 2812 2813 2814
  assert(pMsg->ahandle != NULL);

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

  if (pMsg->contLen > 0) {
wafwerar's avatar
wafwerar 已提交
2815
    buf.pData = taosMemoryCalloc(1, pMsg->contLen);
H
Haojun Liao 已提交
2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826
    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);
2827 2828
}

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

wafwerar's avatar
wafwerar 已提交
2832
  SResFetchReq* pMsg = taosMemoryCalloc(1, sizeof(SResFetchReq));
2833 2834 2835 2836
  if (NULL == pMsg) {
    pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return pTaskInfo->code;
  }
2837

L
Liu Jicong 已提交
2838 2839
  SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, sourceIndex);
  SSourceDataInfo*       pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, sourceIndex);
2840

L
Liu Jicong 已提交
2841 2842
  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);
2843 2844 2845 2846 2847 2848 2849

  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 已提交
2850
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
2851
  if (NULL == pMsgSendInfo) {
wafwerar's avatar
wafwerar 已提交
2852
    taosMemoryFreeClear(pMsg);
2853 2854 2855
    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 已提交
2856 2857
  }

2858 2859 2860 2861 2862
  pMsgSendInfo->param = pDataInfo;
  pMsgSendInfo->msgInfo.pData = pMsg;
  pMsgSendInfo->msgInfo.len = sizeof(SResFetchReq);
  pMsgSendInfo->msgType = TDMT_VND_FETCH;
  pMsgSendInfo->fp = loadRemoteDataCallback;
2863

2864
  int64_t transporterId = 0;
L
Liu Jicong 已提交
2865
  int32_t code = asyncSendMsgToServer(pExchangeInfo->pTransporter, &pSource->addr.epSet, &transporterId, pMsgSendInfo);
2866 2867 2868
  return TSDB_CODE_SUCCESS;
}

2869
// NOTE: sources columns are more than the destination SSDatablock columns.
2870
void relocateColumnData(SSDataBlock* pBlock, const SArray* pColMatchInfo, SArray* pCols) {
2871 2872 2873
  size_t numOfSrcCols = taosArrayGetSize(pCols);

  int32_t i = 0, j = 0;
L
Liu Jicong 已提交
2874
  while (i < numOfSrcCols && j < taosArrayGetSize(pColMatchInfo)) {
2875 2876 2877 2878 2879 2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893
    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);
    }
  }
}

2894
int32_t setSDataBlockFromFetchRsp(SSDataBlock* pRes, SLoadRemoteDataInfo* pLoadInfo, int32_t numOfRows, char* pData,
dengyihao's avatar
dengyihao 已提交
2895 2896
                                  int32_t compLen, int32_t numOfOutput, int64_t startTs, uint64_t* total,
                                  SArray* pColList) {
H
Haojun Liao 已提交
2897
  if (pColList == NULL) {  // data from other sources
H
Haojun Liao 已提交
2898 2899
    blockDataEnsureCapacity(pRes, numOfRows);

L
Liu Jicong 已提交
2900
    int32_t dataLen = *(int32_t*)pData;
2901 2902
    pData += sizeof(int32_t);

L
Liu Jicong 已提交
2903
    pRes->info.groupId = *(uint64_t*)pData;
2904 2905
    pData += sizeof(uint64_t);

H
Haojun Liao 已提交
2906 2907
    int32_t* colLen = (int32_t*)pData;

2908
    char* pStart = pData + sizeof(int32_t) * numOfOutput;
2909
    for (int32_t i = 0; i < numOfOutput; ++i) {
H
Haojun Liao 已提交
2910
      colLen[i] = htonl(colLen[i]);
2911
      ASSERT(colLen[i] >= 0);
H
Haojun Liao 已提交
2912

2913
      SColumnInfoData* pColInfoData = taosArrayGet(pRes->pDataBlock, i);
H
Haojun Liao 已提交
2914 2915 2916
      if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
        pColInfoData->varmeta.length = colLen[i];
        pColInfoData->varmeta.allocLen = colLen[i];
2917

L
Liu Jicong 已提交
2918 2919
        memcpy(pColInfoData->varmeta.offset, pStart, sizeof(int32_t) * numOfRows);
        pStart += sizeof(int32_t) * numOfRows;
H
Haojun Liao 已提交
2920

2921 2922 2923
        if (colLen[i] > 0) {
          pColInfoData->pData = taosMemoryMalloc(colLen[i]);
        }
H
Haojun Liao 已提交
2924 2925 2926
      } else {
        memcpy(pColInfoData->nullbitmap, pStart, BitmapLen(numOfRows));
        pStart += BitmapLen(numOfRows);
2927
      }
H
Haojun Liao 已提交
2928

2929 2930 2931 2932
      if (colLen[i] > 0) {
        memcpy(pColInfoData->pData, pStart, colLen[i]);
      }

L
Liu Jicong 已提交
2933 2934
      // TODO setting this flag to true temporarily so aggregate function on stable will
      // examine NULL value for non-primary key column
2935
      pColInfoData->hasNull = true;
H
Haojun Liao 已提交
2936
      pStart += colLen[i];
2937
    }
H
Haojun Liao 已提交
2938
  } else {  // extract data according to pColList
2939
    ASSERT(numOfOutput == taosArrayGetSize(pColList));
2940 2941 2942 2943 2944
    char* pStart = pData;

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

2945
    // todo refactor:extract method
2946
    SSysTableSchema* pSchema = (SSysTableSchema*)pStart;
dengyihao's avatar
dengyihao 已提交
2947
    for (int32_t i = 0; i < numOfCols; ++i) {
2948 2949 2950 2951 2952 2953 2954
      SSysTableSchema* p = (SSysTableSchema*)pStart;

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

H
Haojun Liao 已提交
2955 2956 2957 2958
    SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
    pBlock->pDataBlock = taosArrayInit(numOfCols, sizeof(SColumnInfoData));
    pBlock->info.numOfCols = numOfCols;

dengyihao's avatar
dengyihao 已提交
2959
    for (int32_t i = 0; i < numOfCols; ++i) {
2960 2961 2962 2963 2964
      SColumnInfoData idata = {0};
      idata.info.type = pSchema[i].type;
      idata.info.bytes = pSchema[i].bytes;
      idata.info.colId = pSchema[i].colId;

H
Haojun Liao 已提交
2965
      taosArrayPush(pBlock->pDataBlock, &idata);
2966
      if (IS_VAR_DATA_TYPE(idata.info.type)) {
H
Haojun Liao 已提交
2967
        pBlock->info.hasVarCol = true;
2968 2969 2970
      }
    }

H
Haojun Liao 已提交
2971
    blockDataEnsureCapacity(pBlock, numOfRows);
2972

L
Liu Jicong 已提交
2973 2974
    int32_t  dataLen = *(int32_t*)pStart;
    uint64_t groupId = *(uint64_t*)(pStart + sizeof(int32_t));
2975 2976
    pStart += sizeof(int32_t) + sizeof(uint64_t);

L
Liu Jicong 已提交
2977
    int32_t* colLen = (int32_t*)(pStart);
2978 2979 2980 2981
    pStart += sizeof(int32_t) * numOfCols;

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

H
Haojun Liao 已提交
2984
      SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);
2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000
      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 已提交
3001 3002

    // data from mnode
H
Haojun Liao 已提交
3003 3004 3005
    relocateColumnData(pRes, pColList, pBlock->pDataBlock);
    taosArrayDestroy(pBlock->pDataBlock);
    taosMemoryFree(pBlock);
X
Xiaoyu Wang 已提交
3006
    //    blockDataDestroy(pBlock);
3007
  }
3008

H
Haojun Liao 已提交
3009
  pRes->info.rows = numOfRows;
3010
  blockDataUpdateTsWindow(pRes);
3011

3012
  int64_t el = taosGetTimestampUs() - startTs;
3013

H
Haojun Liao 已提交
3014 3015
  pLoadInfo->totalRows += numOfRows;
  pLoadInfo->totalSize += compLen;
3016

H
Haojun Liao 已提交
3017 3018 3019
  if (total != NULL) {
    *total += numOfRows;
  }
3020

H
Haojun Liao 已提交
3021
  pLoadInfo->totalElapsed += el;
3022 3023
  return TSDB_CODE_SUCCESS;
}
3024

L
Liu Jicong 已提交
3025 3026
static void* setAllSourcesCompleted(SOperatorInfo* pOperator, int64_t startTs) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
3027
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
3028

L
Liu Jicong 已提交
3029
  int64_t              el = taosGetTimestampUs() - startTs;
H
Haojun Liao 已提交
3030 3031
  SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
  pLoadInfo->totalElapsed += el;
H
Haojun Liao 已提交
3032

3033
  size_t totalSources = taosArrayGetSize(pExchangeInfo->pSources);
L
Liu Jicong 已提交
3034 3035 3036
  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);
3037 3038 3039 3040 3041

  doSetOperatorCompleted(pOperator);
  return NULL;
}

L
Liu Jicong 已提交
3042 3043
static SSDataBlock* concurrentlyLoadRemoteDataImpl(SOperatorInfo* pOperator, SExchangeInfo* pExchangeInfo,
                                                   SExecTaskInfo* pTaskInfo) {
3044 3045 3046 3047 3048 3049 3050 3051 3052
  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);

3053
      if (pDataInfo->status == EX_SOURCE_DATA_EXHAUSTED) {
3054
        completed += 1;
H
Haojun Liao 已提交
3055 3056
        continue;
      }
3057

3058
      if (pDataInfo->status != EX_SOURCE_DATA_READY) {
3059 3060 3061
        continue;
      }

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

L
Liu Jicong 已提交
3065
      SSDataBlock*         pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
3066
      SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
3067
      if (pRsp->numOfRows == 0) {
L
Liu Jicong 已提交
3068 3069
        qDebug("%s vgId:%d, taskID:0x%" PRIx64 " index:%d completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64
               " try next",
3070
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, i + 1, pDataInfo->totalRows,
H
Haojun Liao 已提交
3071
               pExchangeInfo->loadInfo.totalRows);
3072
        pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
3073 3074 3075
        completed += 1;
        continue;
      }
H
Haojun Liao 已提交
3076

H
Haojun Liao 已提交
3077
      SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
X
Xiaoyu Wang 已提交
3078 3079
      code = setSDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
                                       pTableRsp->compLen, pTableRsp->numOfCols, startTs, &pDataInfo->totalRows, NULL);
3080
      if (code != 0) {
3081 3082 3083
        goto _error;
      }

3084 3085 3086
      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 已提交
3087 3088
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pDataInfo->totalRows,
               pLoadInfo->totalRows, pLoadInfo->totalSize, i + 1, totalSources);
3089
        pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
3090
      } else {
dengyihao's avatar
dengyihao 已提交
3091 3092
        qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, totalRows:%" PRIu64
               ", totalBytes:%" PRIu64,
H
Haojun Liao 已提交
3093 3094
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pLoadInfo->totalRows,
               pLoadInfo->totalSize);
3095 3096
      }

3097 3098
      if (pDataInfo->status != EX_SOURCE_DATA_EXHAUSTED) {
        pDataInfo->status = EX_SOURCE_DATA_NOT_READY;
3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117
        code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
        if (code != TSDB_CODE_SUCCESS) {
          goto _error;
        }
      }

      return pExchangeInfo->pResult;
    }

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

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

L
Liu Jicong 已提交
3118 3119 3120
static SSDataBlock* concurrentlyLoadRemoteData(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
3121

H
Haojun Liao 已提交
3122 3123 3124
  if (pOperator->status == OP_RES_TO_RETURN) {
    return concurrentlyLoadRemoteDataImpl(pOperator, pExchangeInfo, pTaskInfo);
  }
3125

L
Liu Jicong 已提交
3126
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
3127 3128 3129
  int64_t startTs = taosGetTimestampUs();

  // Asynchronously send all fetch requests to all sources.
L
Liu Jicong 已提交
3130
  for (int32_t i = 0; i < totalSources; ++i) {
3131 3132
    int32_t code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
3133
      return NULL;
3134 3135 3136 3137
    }
  }

  int64_t endTs = taosGetTimestampUs();
L
Liu Jicong 已提交
3138 3139
  qDebug("%s send all fetch request to %" PRIzu " sources completed, elapsed:%" PRId64, GET_TASKID(pTaskInfo),
         totalSources, endTs - startTs);
3140 3141

  tsem_wait(&pExchangeInfo->ready);
H
Haojun Liao 已提交
3142 3143
  pOperator->status = OP_RES_TO_RETURN;
  return concurrentlyLoadRemoteDataImpl(pOperator, pExchangeInfo, pTaskInfo);
3144 3145
}

L
Liu Jicong 已提交
3146 3147 3148
static int32_t prepareConcurrentlyLoad(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3149

L
Liu Jicong 已提交
3150
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
3151 3152 3153
  int64_t startTs = taosGetTimestampUs();

  // Asynchronously send all fetch requests to all sources.
L
Liu Jicong 已提交
3154
  for (int32_t i = 0; i < totalSources; ++i) {
3155 3156
    int32_t code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
3157 3158
      pTaskInfo->code = code;
      return code;
3159 3160 3161 3162
    }
  }

  int64_t endTs = taosGetTimestampUs();
L
Liu Jicong 已提交
3163 3164
  qDebug("%s send all fetch request to %" PRIzu " sources completed, elapsed:%" PRId64, GET_TASKID(pTaskInfo),
         totalSources, endTs - startTs);
3165 3166

  tsem_wait(&pExchangeInfo->ready);
H
Haojun Liao 已提交
3167
  pOperator->cost.openCost = taosGetTimestampUs() - startTs;
3168

H
Haojun Liao 已提交
3169
  return TSDB_CODE_SUCCESS;
3170 3171
}

L
Liu Jicong 已提交
3172 3173 3174
static SSDataBlock* seqLoadRemoteData(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3175

L
Liu Jicong 已提交
3176
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
3177
  int64_t startTs = taosGetTimestampUs();
3178

L
Liu Jicong 已提交
3179
  while (1) {
3180 3181
    if (pExchangeInfo->current >= totalSources) {
      return setAllSourcesCompleted(pOperator, startTs);
3182
    }
3183

3184 3185 3186
    doSendFetchDataRequest(pExchangeInfo, pTaskInfo, pExchangeInfo->current);
    tsem_wait(&pExchangeInfo->ready);

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

H
Haojun Liao 已提交
3190
    if (pDataInfo->code != TSDB_CODE_SUCCESS) {
dengyihao's avatar
dengyihao 已提交
3191 3192
      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 已提交
3193 3194 3195 3196
      pOperator->pTaskInfo->code = pDataInfo->code;
      return NULL;
    }

L
Liu Jicong 已提交
3197
    SRetrieveTableRsp*   pRsp = pDataInfo->pRsp;
H
Haojun Liao 已提交
3198
    SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
3199
    if (pRsp->numOfRows == 0) {
dengyihao's avatar
dengyihao 已提交
3200 3201
      qDebug("%s vgId:%d, taskID:0x%" PRIx64 " %d of total completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64
             " try next",
3202
             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pExchangeInfo->current + 1,
H
Haojun Liao 已提交
3203
             pDataInfo->totalRows, pLoadInfo->totalRows);
H
Haojun Liao 已提交
3204

3205
      pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
3206 3207 3208
      pExchangeInfo->current += 1;
      continue;
    }
H
Haojun Liao 已提交
3209

L
Liu Jicong 已提交
3210
    SSDataBlock*       pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
3211
    SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
L
Liu Jicong 已提交
3212 3213
    int32_t            code =
        setSDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
3214
                                  pTableRsp->compLen, pTableRsp->numOfCols, startTs, &pDataInfo->totalRows, NULL);
3215 3216

    if (pRsp->completed == 1) {
H
Haojun Liao 已提交
3217
      qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, rowsOfSource:%" PRIu64
L
Liu Jicong 已提交
3218 3219 3220
             ", 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);
3221

3222
      pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
3223 3224
      pExchangeInfo->current += 1;
    } else {
L
Liu Jicong 已提交
3225 3226 3227 3228
      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);
3229 3230 3231 3232
    }

    return pExchangeInfo->pResult;
  }
3233 3234
}

L
Liu Jicong 已提交
3235
static int32_t prepareLoadRemoteData(SOperatorInfo* pOperator) {
3236
  if (OPTR_IS_OPENED(pOperator)) {
H
Haojun Liao 已提交
3237 3238 3239
    return TSDB_CODE_SUCCESS;
  }

L
Liu Jicong 已提交
3240
  SExchangeInfo* pExchangeInfo = pOperator->info;
H
Haojun Liao 已提交
3241 3242 3243 3244 3245 3246 3247 3248 3249
  if (pExchangeInfo->seqLoadData) {
    // do nothing for sequentially load data
  } else {
    int32_t code = prepareConcurrentlyLoad(pOperator);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }
  }

3250
  OPTR_SET_OPENED(pOperator);
H
Haojun Liao 已提交
3251 3252 3253
  return TSDB_CODE_SUCCESS;
}

3254
static SSDataBlock* doLoadRemoteData(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
3255 3256
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3257

3258
  pTaskInfo->code = pOperator->fpSet._openFn(pOperator);
3259
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
3260 3261
    return NULL;
  }
3262

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

3266
  if (pOperator->status == OP_EXEC_DONE) {
L
Liu Jicong 已提交
3267 3268 3269
    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);
3270 3271 3272 3273 3274 3275
    return NULL;
  }

  if (pExchangeInfo->seqLoadData) {
    return seqLoadRemoteData(pOperator);
  } else {
H
Haojun Liao 已提交
3276
    return concurrentlyLoadRemoteData(pOperator);
3277
  }
H
Haojun Liao 已提交
3278

3279
#if 0
H
Haojun Liao 已提交
3280
  _error:
wafwerar's avatar
wafwerar 已提交
3281 3282
  taosMemoryFreeClear(pMsg);
  taosMemoryFreeClear(pMsgSendInfo);
H
Haojun Liao 已提交
3283 3284 3285

  terrno = pTaskInfo->code;
  return NULL;
3286
#endif
H
Haojun Liao 已提交
3287
}
3288

H
Haojun Liao 已提交
3289
static int32_t initDataSource(int32_t numOfSources, SExchangeInfo* pInfo) {
3290
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
H
Haojun Liao 已提交
3291 3292
  if (pInfo->pSourceDataInfo == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
3293 3294
  }

L
Liu Jicong 已提交
3295
  for (int32_t i = 0; i < numOfSources; ++i) {
3296
    SSourceDataInfo dataInfo = {0};
H
Haojun Liao 已提交
3297
    dataInfo.status = EX_SOURCE_DATA_NOT_READY;
L
Liu Jicong 已提交
3298 3299
    dataInfo.pEx = pInfo;
    dataInfo.index = i;
3300

H
Haojun Liao 已提交
3301 3302 3303 3304 3305 3306 3307 3308 3309 3310
    void* ret = taosArrayPush(pInfo->pSourceDataInfo, &dataInfo);
    if (ret == NULL) {
      taosArrayDestroy(pInfo->pSourceDataInfo);
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  return TSDB_CODE_SUCCESS;
}

3311
SOperatorInfo* createExchangeOperatorInfo(void *pTransporter, const SNodeList* pSources, SSDataBlock* pBlock,
3312
                                          SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
3313
  SExchangeInfo* pInfo = taosMemoryCalloc(1, sizeof(SExchangeInfo));
wafwerar's avatar
wafwerar 已提交
3314
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
3315 3316

  if (pInfo == NULL || pOperator == NULL) {
wafwerar's avatar
wafwerar 已提交
3317 3318
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
3319 3320
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
H
Haojun Liao 已提交
3321 3322
  }

H
Haojun Liao 已提交
3323
  size_t numOfSources = LIST_LENGTH(pSources);
H
Haojun Liao 已提交
3324
  pInfo->pSources = taosArrayInit(numOfSources, sizeof(SDownstreamSourceNode));
H
Haojun Liao 已提交
3325 3326 3327
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
  if (pInfo->pSourceDataInfo == NULL || pInfo->pSources == NULL) {
    goto _error;
H
Haojun Liao 已提交
3328 3329
  }

L
Liu Jicong 已提交
3330 3331
  for (int32_t i = 0; i < numOfSources; ++i) {
    SNodeListNode* pNode = nodesListGetNode((SNodeList*)pSources, i);
H
Haojun Liao 已提交
3332 3333
    taosArrayPush(pInfo->pSources, pNode);
  }
3334

H
Haojun Liao 已提交
3335 3336 3337
  int32_t code = initDataSource(numOfSources, pInfo);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
3338
  }
H
Haojun Liao 已提交
3339

dengyihao's avatar
dengyihao 已提交
3340
  pInfo->pResult = pBlock;
3341 3342 3343
  pInfo->seqLoadData = true;

  tsem_init(&pInfo->ready, 0, 0);
3344

dengyihao's avatar
dengyihao 已提交
3345
  pOperator->name = "ExchangeOperator";
X
Xiaoyu Wang 已提交
3346
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
3347
  pOperator->blocking = false;
dengyihao's avatar
dengyihao 已提交
3348 3349
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
3350
  pOperator->numOfExprs = pBlock->info.numOfCols;
dengyihao's avatar
dengyihao 已提交
3351
  pOperator->pTaskInfo = pTaskInfo;
3352

L
Liu Jicong 已提交
3353 3354
  pOperator->fpSet = createOperatorFpSet(prepareLoadRemoteData, doLoadRemoteData, NULL, NULL,
                                         destroyExchangeOperatorInfo, NULL, NULL, NULL);
3355
  pInfo->pTransporter = pTransporter;
3356

3357
  return pOperator;
H
Haojun Liao 已提交
3358

L
Liu Jicong 已提交
3359
_error:
H
Haojun Liao 已提交
3360
  if (pInfo != NULL) {
H
Haojun Liao 已提交
3361
    destroyExchangeOperatorInfo(pInfo, numOfSources);
H
Haojun Liao 已提交
3362 3363
  }

wafwerar's avatar
wafwerar 已提交
3364 3365
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
3366
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
H
Haojun Liao 已提交
3367
  return NULL;
3368 3369
}

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

3373
static void destroySortedMergeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
3374
  SSortedMergeOperatorInfo* pInfo = (SSortedMergeOperatorInfo*)param;
H
Haojun Liao 已提交
3375
  taosArrayDestroy(pInfo->pSortInfo);
3376 3377 3378
  taosArrayDestroy(pInfo->groupInfo);

  if (pInfo->pSortHandle != NULL) {
H
Haojun Liao 已提交
3379
    tsortDestroySortHandle(pInfo->pSortHandle);
3380 3381
  }

H
Haojun Liao 已提交
3382
  blockDataDestroy(pInfo->binfo.pRes);
H
Haojun Liao 已提交
3383
  cleanupAggSup(&pInfo->aggSup);
3384
}
H
Haojun Liao 已提交
3385

L
Liu Jicong 已提交
3386
static bool needToMerge(SSDataBlock* pBlock, SArray* groupInfo, char** buf, int32_t rowIndex) {
3387 3388 3389 3390
  size_t size = taosArrayGetSize(groupInfo);
  if (size == 0) {
    return true;
  }
3391

3392 3393
  for (int32_t i = 0; i < size; ++i) {
    int32_t* index = taosArrayGet(groupInfo, i);
3394

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

3398 3399 3400
    if ((isNull && buf[i] != NULL) || (!isNull && buf[i] == NULL)) {
      return false;
    }
3401

3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414
    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;
      }
3415 3416 3417
    }
  }

3418
  return 0;
3419 3420
}

L
Liu Jicong 已提交
3421 3422 3423
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
3424
    pCtx[j].startRow = rowIndex;
3425 3426
  }

3427 3428
  for (int32_t j = 0; j < numOfExpr; ++j) {
    int32_t functionId = pCtx[j].functionId;
L
Liu Jicong 已提交
3429 3430 3431 3432 3433 3434 3435 3436 3437
    //    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]);
    //    }
3438
  }
3439
}
3440

L
Liu Jicong 已提交
3441 3442
static void doFinalizeResultImpl(SqlFunctionCtx* pCtx, int32_t numOfExpr) {
  for (int32_t j = 0; j < numOfExpr; ++j) {
3443 3444 3445 3446
    int32_t functionId = pCtx[j].functionId;
    //    if (functionId == FUNC_TAG_DUMMY || functionId == FUNC_TS_DUMMY) {
    //      continue;
    //    }
3447

3448 3449 3450 3451
    //    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 已提交
3452
    //    pCtx[j].fpSet.finalize(&pCtx[j]);
3453 3454
  }
}
3455

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

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

3463 3464 3465
    char* data = colDataGetData(pColInfo, rowIndex);
    memcpy(rowColData[i], data, colDataGetLength(pColInfo, rowIndex));
  }
3466

3467 3468
  return true;
}
3469

3470 3471
static void doMergeImpl(SOperatorInfo* pOperator, int32_t numOfExpr, SSDataBlock* pBlock) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
3472

3473
  SqlFunctionCtx* pCtx = pInfo->binfo.pCtx;
L
Liu Jicong 已提交
3474
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
3475 3476
    pCtx[i].size = 1;
  }
3477

L
Liu Jicong 已提交
3478
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
3479 3480 3481 3482 3483 3484 3485 3486 3487
    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);
3488 3489
        int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfExprs, NULL);
        //        setTagValueForMultipleRows(pCtx, pOperator->numOfExprs, numOfRows);
3490

3491
        // TODO check for available buffer;
H
Haojun Liao 已提交
3492

3493 3494 3495 3496 3497
        // next group info data
        pInfo->binfo.pRes->info.rows += numOfRows;
        for (int32_t j = 0; j < numOfExpr; ++j) {
          if (pCtx[j].functionId < 0) {
            continue;
3498
          }
3499

H
Haojun Liao 已提交
3500
          pCtx[j].fpSet.process(&pCtx[j]);
3501
        }
3502 3503 3504

        doMergeResultImpl(pInfo, pCtx, numOfExpr, i);
        pInfo->hasGroupVal = saveCurrentTuple(pInfo->groupVal, pInfo->groupInfo, pBlock, i);
H
Haojun Liao 已提交
3505
      }
3506 3507 3508 3509
    }
  }
}

3510 3511
static SSDataBlock* doMerge(SOperatorInfo* pOperator) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
3512
  SSortHandle*              pHandle = pInfo->pSortHandle;
3513

3514
  SSDataBlock* pDataBlock = createOneDataBlock(pInfo->binfo.pRes, false);
3515
  blockDataEnsureCapacity(pDataBlock, pOperator->resultInfo.capacity);
3516

L
Liu Jicong 已提交
3517
  while (1) {
3518
    blockDataCleanup(pDataBlock);
3519
    while (1) {
H
Haojun Liao 已提交
3520
      STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
3521 3522
      if (pTupleHandle == NULL) {
        break;
3523
      }
3524

3525 3526
      // build datablock for merge for one group
      appendOneRowToDataBlock(pDataBlock, pTupleHandle);
3527
      if (pDataBlock->info.rows >= pOperator->resultInfo.capacity) {
3528 3529
        break;
      }
3530
    }
3531

3532 3533 3534
    if (pDataBlock->info.rows == 0) {
      break;
    }
3535

3536
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pDataBlock, TSDB_ORDER_ASC, MAIN_SCAN, true);
L
Liu Jicong 已提交
3537 3538
    //  updateOutputBuf(&pInfo->binfo, &pAggInfo->bufCapacity, pBlock->info.rows * pAggInfo->resultRowFactor,
    //  pOperator->pRuntimeEnv, true);
3539
    doMergeImpl(pOperator, pOperator->numOfExprs, pDataBlock);
3540 3541
    // flush to tuple store, and after all data have been handled, return to upstream node or sink node
  }
3542

3543 3544 3545
  doFinalizeResultImpl(pInfo->binfo.pCtx, pOperator->numOfExprs);
  int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfExprs, NULL);
  //        setTagValueForMultipleRows(pCtx, pOperator->numOfExprs, numOfRows);
3546

3547
  // TODO check for available buffer;
3548

3549 3550
  // next group info data
  pInfo->binfo.pRes->info.rows += numOfRows;
L
Liu Jicong 已提交
3551
  return (pInfo->binfo.pRes->info.rows > 0) ? pInfo->binfo.pRes : NULL;
3552
}
3553

3554
static SSDataBlock* doSortedMerge(SOperatorInfo* pOperator) {
3555 3556
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
3557 3558
  }

L
Liu Jicong 已提交
3559
  SExecTaskInfo*            pTaskInfo = pOperator->pTaskInfo;
3560
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
H
Haojun Liao 已提交
3561
  if (pOperator->status == OP_RES_TO_RETURN) {
3562
    return getSortedBlockData(pInfo->pSortHandle, pInfo->binfo.pRes, pOperator->resultInfo.capacity, NULL);
3563 3564
  }

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

3569
  tsortSetFetchRawDataFp(pInfo->pSortHandle, loadNextDataBlock, NULL, NULL);
3570

L
Liu Jicong 已提交
3571
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
wmmhello's avatar
wmmhello 已提交
3572
    SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
H
Haojun Liao 已提交
3573
    ps->param = pOperator->pDownstream[i];
H
Haojun Liao 已提交
3574
    tsortAddSource(pInfo->pSortHandle, ps);
3575 3576
  }

H
Haojun Liao 已提交
3577
  int32_t code = tsortOpen(pInfo->pSortHandle);
3578
  if (code != TSDB_CODE_SUCCESS) {
3579
    longjmp(pTaskInfo->env, terrno);
3580 3581
  }

H
Haojun Liao 已提交
3582
  pOperator->status = OP_RES_TO_RETURN;
3583
  return doMerge(pOperator);
3584
}
3585

L
Liu Jicong 已提交
3586 3587
static int32_t initGroupCol(SExprInfo* pExprInfo, int32_t numOfCols, SArray* pGroupInfo,
                            SSortedMergeOperatorInfo* pInfo) {
3588 3589
  if (pGroupInfo == NULL || taosArrayGetSize(pGroupInfo) == 0) {
    return 0;
H
Haojun Liao 已提交
3590 3591
  }

3592 3593 3594 3595 3596 3597 3598 3599
  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 已提交
3600 3601
  size_t numOfGroupCol = taosArrayGetSize(pInfo->groupInfo);
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
3602
    SColumn* pCol = taosArrayGet(pGroupInfo, i);
L
Liu Jicong 已提交
3603
    for (int32_t j = 0; j < numOfCols; ++j) {
H
Haojun Liao 已提交
3604
      SExprInfo* pe = &pExprInfo[j];
3605
      if (pe->base.resSchema.slotId == pCol->colId) {
3606 3607
        taosArrayPush(plist, pCol);
        taosArrayPush(pInfo->groupInfo, &j);
H
Haojun Liao 已提交
3608
        len += pCol->bytes;
3609 3610
        break;
      }
H
Haojun Liao 已提交
3611 3612 3613
    }
  }

3614
  ASSERT(taosArrayGetSize(pGroupInfo) == taosArrayGetSize(plist));
H
Haojun Liao 已提交
3615

wafwerar's avatar
wafwerar 已提交
3616
  pInfo->groupVal = taosMemoryCalloc(1, (POINTER_BYTES * numOfGroupCol + len));
3617 3618 3619 3620
  if (pInfo->groupVal == NULL) {
    taosArrayDestroy(plist);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
H
Haojun Liao 已提交
3621

3622
  int32_t offset = 0;
L
Liu Jicong 已提交
3623 3624
  char*   start = (char*)(pInfo->groupVal + (POINTER_BYTES * numOfGroupCol));
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
3625 3626
    pInfo->groupVal[i] = start + offset;
    SColumn* pCol = taosArrayGet(plist, i);
H
Haojun Liao 已提交
3627
    offset += pCol->bytes;
3628
  }
H
Haojun Liao 已提交
3629

3630
  taosArrayDestroy(plist);
H
Haojun Liao 已提交
3631

3632 3633
  return TSDB_CODE_SUCCESS;
}
H
Haojun Liao 已提交
3634

L
Liu Jicong 已提交
3635 3636 3637
SOperatorInfo* createSortedMergeOperatorInfo(SOperatorInfo** downstream, int32_t numOfDownstream, SExprInfo* pExprInfo,
                                             int32_t num, SArray* pSortInfo, SArray* pGroupInfo,
                                             SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
3638
  SSortedMergeOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSortedMergeOperatorInfo));
L
Liu Jicong 已提交
3639
  SOperatorInfo*            pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
3640
  if (pInfo == NULL || pOperator == NULL) {
3641
    goto _error;
3642
  }
H
Haojun Liao 已提交
3643

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

3647 3648 3649
  if (pInfo->binfo.pCtx == NULL || pInfo->binfo.pRes == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
3650

dengyihao's avatar
dengyihao 已提交
3651
  size_t  keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
3652
  int32_t code = doInitAggInfoSup(&pInfo->aggSup, pInfo->binfo.pCtx, num, keyBufSize, pTaskInfo->id.str);
3653 3654 3655
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
3656

H
Haojun Liao 已提交
3657
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, num, pTaskInfo);
H
Haojun Liao 已提交
3658
  code = initGroupCol(pExprInfo, num, pGroupInfo, pInfo);
3659 3660 3661
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
3662

L
Liu Jicong 已提交
3663 3664 3665 3666 3667
  //  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 已提交
3668

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

L
Liu Jicong 已提交
3671
  pOperator->name = "SortedMerge";
X
Xiaoyu Wang 已提交
3672
  // pOperator->operatorType = OP_SortedMerge;
3673
  pOperator->blocking = true;
L
Liu Jicong 已提交
3674 3675
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
3676
  pOperator->numOfExprs = num;
L
Liu Jicong 已提交
3677
  pOperator->pExpr = pExprInfo;
H
Haojun Liao 已提交
3678

L
Liu Jicong 已提交
3679
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
3680

3681 3682
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doSortedMerge, NULL, NULL, destroySortedMergeOperatorInfo,
                                         NULL, NULL, NULL);
3683 3684 3685
  code = appendDownstream(pOperator, downstream, numOfDownstream);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
3686
  }
H
Haojun Liao 已提交
3687

3688
  return pOperator;
H
Haojun Liao 已提交
3689

L
Liu Jicong 已提交
3690
_error:
3691
  if (pInfo != NULL) {
H
Haojun Liao 已提交
3692
    destroySortedMergeOperatorInfo(pInfo, num);
H
Haojun Liao 已提交
3693 3694
  }

wafwerar's avatar
wafwerar 已提交
3695 3696
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
3697 3698
  terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
  return NULL;
H
Haojun Liao 已提交
3699 3700
}

3701
int32_t getTableScanInfo(SOperatorInfo* pOperator, int32_t *order, int32_t* scanFlag) {
3702 3703 3704 3705 3706 3707 3708 3709 3710 3711 3712
  // todo add more information about exchange operation
  if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_EXCHANGE) {
    *order = TSDB_ORDER_ASC;
    *scanFlag = MAIN_SCAN;
    return TSDB_CODE_SUCCESS;
  } else if (pOperator->operatorType == QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN) {
    STableScanInfo* pTableScanInfo = pOperator->info;
    *order = pTableScanInfo->cond.order;
    *scanFlag = pTableScanInfo->scanFlag;
    return TSDB_CODE_SUCCESS;
  } else {
H
Haojun Liao 已提交
3713
    if (pOperator->pDownstream == NULL || pOperator->pDownstream[0] == NULL) {
3714
      return TSDB_CODE_INVALID_PARA;
H
Haojun Liao 已提交
3715
    } else {
3716
      return getTableScanInfo(pOperator->pDownstream[0], order, scanFlag);
3717 3718 3719
    }
  }
}
3720 3721

// this is a blocking operator
L
Liu Jicong 已提交
3722
static int32_t doOpenAggregateOptr(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
3723 3724
  if (OPTR_IS_OPENED(pOperator)) {
    return TSDB_CODE_SUCCESS;
3725 3726
  }

H
Haojun Liao 已提交
3727
  SExecTaskInfo*    pTaskInfo = pOperator->pTaskInfo;
3728
  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
3729

dengyihao's avatar
dengyihao 已提交
3730
  SOptrBasicInfo* pInfo = &pAggInfo->binfo;
H
Haojun Liao 已提交
3731
  SOperatorInfo* downstream = pOperator->pDownstream[0];
3732

3733 3734 3735
  int32_t order = TSDB_ORDER_ASC;
  int32_t scanFlag = MAIN_SCAN;

H
Haojun Liao 已提交
3736
  while (1) {
H
Haojun Liao 已提交
3737
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
3738
    SSDataBlock* pBlock = downstream->fpSet.getNextFn(downstream);
H
Haojun Liao 已提交
3739
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
3740 3741 3742 3743 3744

    if (pBlock == NULL) {
      break;
    }

3745 3746 3747 3748
    int32_t code = getTableScanInfo(pOperator, &order, &scanFlag);
    if (code != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, code);
    }
3749

3750 3751
    // there is an scalar expression that needs to be calculated before apply the group aggregation.
    if (pAggInfo->pScalarExprInfo != NULL) {
3752
      code = projectApplyFunctions(pAggInfo->pScalarExprInfo, pBlock, pBlock, pAggInfo->pScalarCtx,
3753
                                   pAggInfo->numOfScalarExpr, NULL);
3754
      if (code != TSDB_CODE_SUCCESS) {
3755
        longjmp(pTaskInfo->env, code);
3756
      }
3757 3758
    }

3759
    // the pDataBlock are always the same one, no need to call this again
3760
    setExecutionContext(pOperator->numOfExprs, pBlock->info.groupId, pTaskInfo, pAggInfo);
3761
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order, scanFlag, true);
3762 3763 3764 3765
    code = doAggregateImpl(pOperator, 0, pInfo->pCtx);
    if (code != 0) {
      longjmp(pTaskInfo->env, code);
    }
3766

dengyihao's avatar
dengyihao 已提交
3767
#if 0  // test for encode/decode result info
3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778
    if(pOperator->encodeResultRow){
      char *result = NULL;
      int32_t length = 0;
      SAggSupporter   *pSup = &pAggInfo->aggSup;
      pOperator->encodeResultRow(pOperator, pSup, pInfo, &result, &length);
      taosHashClear(pSup->pResultRowHashTable);
      pInfo->resultRowInfo.size = 0;
      pOperator->decodeResultRow(pOperator, pSup, pInfo, result, length);
      if(result){
        taosMemoryFree(result);
      }
3779
    }
3780
#endif
3781 3782
  }

H
Haojun Liao 已提交
3783
  closeAllResultRows(&pAggInfo->binfo.resultRowInfo);
3784 3785
  finalizeMultiTupleQueryResult(pOperator->numOfExprs, pAggInfo->aggSup.pResultBuf, &pAggInfo->binfo.resultRowInfo,
                                pAggInfo->binfo.rowCellInfoOffset);
H
Haojun Liao 已提交
3786

3787
  initGroupedResultInfo(&pAggInfo->groupResInfo, pAggInfo->aggSup.pResultRowHashTable, false);
H
Haojun Liao 已提交
3788 3789 3790 3791
  OPTR_SET_OPENED(pOperator);
  return TSDB_CODE_SUCCESS;
}

3792
static SSDataBlock* getAggregateResult(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
3793
  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
3794 3795 3796 3797 3798 3799
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;

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

L
Liu Jicong 已提交
3800
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
3801
  pTaskInfo->code = pOperator->fpSet._openFn(pOperator);
H
Haojun Liao 已提交
3802 3803 3804 3805
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
    return NULL;
  }

H
Haojun Liao 已提交
3806
  blockDataEnsureCapacity(pInfo->pRes, pOperator->resultInfo.capacity);
3807
  doBuildResultDatablock(pOperator, pInfo, &pAggInfo->groupResInfo, pAggInfo->aggSup.pResultBuf);
H
Haojun Liao 已提交
3808 3809 3810
  if (pInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pAggInfo->groupResInfo)) {
    doSetOperatorCompleted(pOperator);
  }
3811

H
Haojun Liao 已提交
3812
  doSetOperatorCompleted(pOperator);
L
Liu Jicong 已提交
3813
  return (blockDataGetNumOfRows(pInfo->pRes) != 0) ? pInfo->pRes : NULL;
3814 3815
}

dengyihao's avatar
dengyihao 已提交
3816 3817
void aggEncodeResultRow(SOperatorInfo* pOperator, SAggSupporter* pSup, SOptrBasicInfo* pInfo, char** result,
                        int32_t* length) {
wmmhello's avatar
wmmhello 已提交
3818
  int32_t size = taosHashGetSize(pSup->pResultRowHashTable);
3819
  size_t  keyLen = sizeof(uint64_t) * 2;  // estimate the key length
wmmhello's avatar
wmmhello 已提交
3820
  int32_t totalSize = sizeof(int32_t) + size * (sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize);
wafwerar's avatar
wafwerar 已提交
3821
  *result = taosMemoryCalloc(1, totalSize);
L
Liu Jicong 已提交
3822
  if (*result == NULL) {
3823
    longjmp(pOperator->pTaskInfo->env, TSDB_CODE_OUT_OF_MEMORY);
wmmhello's avatar
wmmhello 已提交
3824 3825 3826
  }
  *(int32_t*)(*result) = size;
  int32_t offset = sizeof(int32_t);
3827 3828

  // prepare memory
3829
  SResultRowPosition* pos = &pInfo->resultRowInfo.cur;
dengyihao's avatar
dengyihao 已提交
3830 3831
  void*               pPage = getBufPage(pSup->pResultBuf, pos->pageId);
  SResultRow*         pRow = (SResultRow*)((char*)pPage + pos->offset);
3832 3833 3834
  setBufPageDirty(pPage, true);
  releaseBufPage(pSup->pResultBuf, pPage);

dengyihao's avatar
dengyihao 已提交
3835
  void* pIter = taosHashIterate(pSup->pResultRowHashTable, NULL);
wmmhello's avatar
wmmhello 已提交
3836
  while (pIter) {
dengyihao's avatar
dengyihao 已提交
3837
    void*               key = taosHashGetKey(pIter, &keyLen);
3838
    SResultRowPosition* p1 = (SResultRowPosition*)pIter;
3839

dengyihao's avatar
dengyihao 已提交
3840
    pPage = (SFilePage*)getBufPage(pSup->pResultBuf, p1->pageId);
3841
    pRow = (SResultRow*)((char*)pPage + p1->offset);
3842 3843
    setBufPageDirty(pPage, true);
    releaseBufPage(pSup->pResultBuf, pPage);
wmmhello's avatar
wmmhello 已提交
3844 3845 3846

    // recalculate the result size
    int32_t realTotalSize = offset + sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize;
L
Liu Jicong 已提交
3847 3848 3849
    if (realTotalSize > totalSize) {
      char* tmp = taosMemoryRealloc(*result, realTotalSize);
      if (tmp == NULL) {
wmmhello's avatar
wmmhello 已提交
3850
        terrno = TSDB_CODE_OUT_OF_MEMORY;
wafwerar's avatar
wafwerar 已提交
3851
        taosMemoryFree(*result);
wmmhello's avatar
wmmhello 已提交
3852
        *result = NULL;
3853
        longjmp(pOperator->pTaskInfo->env, TSDB_CODE_OUT_OF_MEMORY);
L
Liu Jicong 已提交
3854
      } else {
wmmhello's avatar
wmmhello 已提交
3855 3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866
        *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);
3867
    memcpy(*result + offset, pRow, pSup->resultRowSize);
wmmhello's avatar
wmmhello 已提交
3868 3869 3870 3871 3872
    offset += pSup->resultRowSize;

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

L
Liu Jicong 已提交
3873
  if (length) {
wmmhello's avatar
wmmhello 已提交
3874 3875 3876 3877 3878
    *length = offset;
  }
  return;
}

dengyihao's avatar
dengyihao 已提交
3879 3880
bool aggDecodeResultRow(SOperatorInfo* pOperator, SAggSupporter* pSup, SOptrBasicInfo* pInfo, char* result,
                        int32_t length) {
L
Liu Jicong 已提交
3881
  if (!result || length <= 0) {
wmmhello's avatar
wmmhello 已提交
3882 3883 3884 3885 3886 3887 3888
    return false;
  }

  //  int32_t size = taosHashGetSize(pSup->pResultRowHashTable);
  int32_t count = *(int32_t*)(result);

  int32_t offset = sizeof(int32_t);
L
Liu Jicong 已提交
3889
  while (count-- > 0 && length > offset) {
wmmhello's avatar
wmmhello 已提交
3890 3891 3892
    int32_t keyLen = *(int32_t*)(result + offset);
    offset += sizeof(int32_t);

L
Liu Jicong 已提交
3893 3894 3895
    uint64_t    tableGroupId = *(uint64_t*)(result + offset);
    SResultRow* resultRow = getNewResultRow_rv(pSup->pResultBuf, tableGroupId, pSup->resultRowSize);
    if (!resultRow) {
3896
      longjmp(pOperator->pTaskInfo->env, TSDB_CODE_TSC_INVALID_INPUT);
wmmhello's avatar
wmmhello 已提交
3897
    }
3898

wmmhello's avatar
wmmhello 已提交
3899
    // add a new result set for a new group
3900 3901
    SResultRowPosition pos = {.pageId = resultRow->pageId, .offset = resultRow->offset};
    taosHashPut(pSup->pResultRowHashTable, result + offset, keyLen, &pos, sizeof(SResultRowPosition));
wmmhello's avatar
wmmhello 已提交
3902 3903 3904

    offset += keyLen;
    int32_t valueLen = *(int32_t*)(result + offset);
L
Liu Jicong 已提交
3905
    if (valueLen != pSup->resultRowSize) {
3906
      longjmp(pOperator->pTaskInfo->env, TSDB_CODE_TSC_INVALID_INPUT);
wmmhello's avatar
wmmhello 已提交
3907 3908 3909 3910 3911 3912 3913 3914 3915 3916
    }
    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);
3917
    prepareResultListBuffer(&pInfo->resultRowInfo, pOperator->pTaskInfo->env);
dengyihao's avatar
dengyihao 已提交
3918
    //    pInfo->resultRowInfo.cur = pInfo->resultRowInfo.size;
L
Liu Jicong 已提交
3919 3920
    //    pInfo->resultRowInfo.pPosition[pInfo->resultRowInfo.size++] =
    //        (SResultRowPosition){.pageId = resultRow->pageId, .offset = resultRow->offset};
dengyihao's avatar
dengyihao 已提交
3921
    pInfo->resultRowInfo.cur = (SResultRowPosition){.pageId = resultRow->pageId, .offset = resultRow->offset};
wmmhello's avatar
wmmhello 已提交
3922 3923
  }

L
Liu Jicong 已提交
3924
  if (offset != length) {
3925
    longjmp(pOperator->pTaskInfo->env, TSDB_CODE_TSC_INVALID_INPUT);
wmmhello's avatar
wmmhello 已提交
3926 3927 3928 3929
  }
  return true;
}

3930 3931
enum {
  PROJECT_RETRIEVE_CONTINUE = 0x1,
L
Liu Jicong 已提交
3932
  PROJECT_RETRIEVE_DONE = 0x2,
3933 3934 3935 3936 3937
};

static int32_t handleLimitOffset(SOperatorInfo* pOperator, SSDataBlock* pBlock) {
  SProjectOperatorInfo* pProjectInfo = pOperator->info;
  SOptrBasicInfo*       pInfo = &pProjectInfo->binfo;
L
Liu Jicong 已提交
3938
  SSDataBlock*          pRes = pInfo->pRes;
3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986

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

3987 3988 3989
  // 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);
3990 3991 3992 3993 3994

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

3995
    return PROJECT_RETRIEVE_DONE;
3996
  }
3997

3998
  // todo optimize performance
3999 4000
  // 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 已提交
4001 4002
  if (pRes->info.rows >= pOperator->resultInfo.threshold || pProjectInfo->slimit.offset != -1 ||
      pProjectInfo->slimit.limit != -1) {
4003
    return PROJECT_RETRIEVE_DONE;
L
Liu Jicong 已提交
4004
  } else {  // not full enough, continue to accumulate the output data in the buffer.
4005 4006 4007 4008
    return PROJECT_RETRIEVE_CONTINUE;
  }
}

4009
static SSDataBlock* doProjectOperation(SOperatorInfo* pOperator) {
4010
  SProjectOperatorInfo* pProjectInfo = pOperator->info;
L
Liu Jicong 已提交
4011
  SOptrBasicInfo*       pInfo = &pProjectInfo->binfo;
4012 4013

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

4016
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4017 4018 4019
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }
dengyihao's avatar
dengyihao 已提交
4020

H
Haojun Liao 已提交
4021
#if 0
4022 4023 4024 4025 4026 4027
  if (pProjectInfo->existDataBlock) {  // TODO refactor
    SSDataBlock* pBlock = pProjectInfo->existDataBlock;
    pProjectInfo->existDataBlock = NULL;
    *newgroup = true;

    // todo dynamic set tags
L
Liu Jicong 已提交
4028
    //    if (pTableQueryInfo != NULL) {
4029
    //      setTagValue(pOperator, pTableQueryInfo->pTable, pInfo->pCtx, pOperator->numOfExprs);
L
Liu Jicong 已提交
4030
    //    }
4031 4032

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

H
Haojun Liao 已提交
4035
    blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
4036
    projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfExprs);
L
Liu Jicong 已提交
4037
    if (pRes->info.rows >= pProjectInfo->binfo.capacity * 0.8) {
4038 4039
      copyTsColoum(pRes, pInfo->pCtx, pOperator->numOfExprs);
      resetResultRowEntryResult(pInfo->pCtx, pOperator->numOfExprs);
4040 4041 4042
      return pRes;
    }
  }
H
Haojun Liao 已提交
4043
#endif
4044

4045 4046 4047
  int32_t order = 0;
  int32_t scanFlag = 0;

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

L
Liu Jicong 已提交
4050
  while (1) {
H
Haojun Liao 已提交
4051
    // The downstream exec may change the value of the newgroup, so use a local variable instead.
H
Haojun Liao 已提交
4052
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
4053
    SSDataBlock* pBlock = downstream->fpSet.getNextFn(downstream);
H
Haojun Liao 已提交
4054
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
4055 4056

    if (pBlock == NULL) {
4057
      setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
4058 4059 4060 4061
      break;
    }

    // Return result of the previous group in the firstly.
4062
    if (false) {
4063 4064 4065
      if (pRes->info.rows > 0) {
        pProjectInfo->existDataBlock = pBlock;
        break;
L
Liu Jicong 已提交
4066
      } else {  // init output buffer for a new group data
4067
        initCtxOutputBuffer(pInfo->pCtx, pOperator->numOfExprs);
4068 4069 4070
      }
    }

4071 4072
    // todo set tags

H
Haojun Liao 已提交
4073 4074
    //    STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
    //    if (pTableQueryInfo != NULL) {
4075
    //      setTagValue(pOperator, pTableQueryInfo->pTable, pInfo->pCtx, pOperator->numOfExprs);
H
Haojun Liao 已提交
4076
    //    }
4077 4078

    // the pDataBlock are always the same one, no need to call this again
4079
    int32_t code = getTableScanInfo(pOperator->pDownstream[0], &order, &scanFlag);
4080

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

4084 4085 4086
    code = projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfExprs, pProjectInfo->pPseudoColInfo);
    if (code != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, code);
4087 4088
    }

4089 4090
    int32_t status = handleLimitOffset(pOperator, pBlock);
    if (status == PROJECT_RETRIEVE_CONTINUE) {
H
Haojun Liao 已提交
4091
      continue;
L
Liu Jicong 已提交
4092
    } else if (status == PROJECT_RETRIEVE_DONE) {
4093 4094 4095
      break;
    }
  }
dengyihao's avatar
dengyihao 已提交
4096

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

4099
  //  copyTsColoum(pRes, pInfo->pCtx, pOperator->numOfExprs);
L
Liu Jicong 已提交
4100
  return (pInfo->pRes->info.rows > 0) ? pInfo->pRes : NULL;
4101 4102
}

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

L
Liu Jicong 已提交
4107 4108
  int64_t ekey = Q_STATUS_EQUAL(pTaskInfo->status, TASK_COMPLETED) ? pTaskInfo->window.ekey
                                                                   : pInfo->existNewGroupBlock->info.window.ekey;
4109 4110
  taosResetFillInfo(pInfo->pFillInfo, getFillInfoStart(pInfo->pFillInfo));

4111
  taosFillSetStartInfo(pInfo->pFillInfo, pInfo->existNewGroupBlock->info.rows, ekey);
4112 4113
  taosFillSetInputDataBlock(pInfo->pFillInfo, pInfo->existNewGroupBlock);

4114
  doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, pResultInfo->capacity);
4115 4116 4117 4118
  pInfo->existNewGroupBlock = NULL;
  *newgroup = true;
}

L
Liu Jicong 已提交
4119 4120
static void doHandleRemainBlockFromNewGroup(SFillOperatorInfo* pInfo, SResultInfo* pResultInfo, bool* newgroup,
                                            SExecTaskInfo* pTaskInfo) {
4121 4122
  if (taosFillHasMoreResults(pInfo->pFillInfo)) {
    *newgroup = false;
4123
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, (int32_t)pResultInfo->capacity);
H
Haojun Liao 已提交
4124
    if (pInfo->pRes->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult)) {
4125 4126 4127 4128 4129 4130
      return;
    }
  }

  // handle the cached new group data block
  if (pInfo->existNewGroupBlock) {
4131
    doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
4132 4133 4134
  }
}

4135
static SSDataBlock* doFill(SOperatorInfo* pOperator) {
L
Liu Jicong 已提交
4136 4137
  SFillOperatorInfo* pInfo = pOperator->info;
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
4138

H
Haojun Liao 已提交
4139
  SResultInfo* pResultInfo = &pOperator->resultInfo;
4140 4141 4142
  SSDataBlock* pResBlock = pInfo->pRes;

  blockDataCleanup(pResBlock);
4143 4144 4145 4146
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

4147
  // todo handle different group data interpolation
X
Xiaoyu Wang 已提交
4148 4149
  bool  n = false;
  bool* newgroup = &n;
4150
  doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
4151 4152
  if (pResBlock->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult && pResBlock->info.rows > 0)) {
    return pResBlock;
H
Haojun Liao 已提交
4153
  }
4154

H
Haojun Liao 已提交
4155
  SOperatorInfo* pDownstream = pOperator->pDownstream[0];
L
Liu Jicong 已提交
4156
  while (1) {
H
Haojun Liao 已提交
4157
    publishOperatorProfEvent(pDownstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
4158
    SSDataBlock* pBlock = pDownstream->fpSet.getNextFn(pDownstream);
H
Haojun Liao 已提交
4159
    publishOperatorProfEvent(pDownstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170

    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
4171
      taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
4172 4173 4174 4175 4176 4177 4178
    } else {
      if (pBlock == NULL) {
        if (pInfo->totalInputRows == 0) {
          pOperator->status = OP_EXEC_DONE;
          return NULL;
        }

4179
        taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
4180 4181 4182 4183 4184 4185 4186
      } else {
        pInfo->totalInputRows += pBlock->info.rows;
        taosFillSetStartInfo(pInfo->pFillInfo, pBlock->info.rows, pBlock->info.window.ekey);
        taosFillSetInputDataBlock(pInfo->pFillInfo, pBlock);
      }
    }

4187 4188
    blockDataEnsureCapacity(pResBlock, pOperator->resultInfo.capacity);
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pResBlock, pOperator->resultInfo.capacity);
4189 4190

    // current group has no more result to return
4191
    if (pResBlock->info.rows > 0) {
4192 4193
      // 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
4194 4195
      if (pResBlock->info.rows > pResultInfo->threshold || pBlock == NULL || (!pInfo->multigroupResult)) {
        return pResBlock;
4196 4197
      }

4198
      doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
4199 4200
      if (pResBlock->info.rows > pOperator->resultInfo.threshold || pBlock == NULL) {
        return pResBlock;
4201 4202 4203
      }
    } else if (pInfo->existNewGroupBlock) {  // try next group
      assert(pBlock != NULL);
4204
      doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
4205 4206
      if (pResBlock->info.rows > pResultInfo->threshold) {
        return pResBlock;
4207 4208 4209 4210 4211 4212 4213 4214
      }
    } else {
      return NULL;
    }
  }
}

// todo set the attribute of query scan count
H
Haojun Liao 已提交
4215
static int32_t getNumOfScanTimes(STaskAttr* pQueryAttr) {
L
Liu Jicong 已提交
4216
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
4217 4218 4219 4220 4221 4222 4223 4224 4225 4226 4227 4228 4229 4230
    int32_t functionId = getExprFunctionId(&pQueryAttr->pExpr1[i]);
    if (functionId == FUNCTION_STDDEV || functionId == FUNCTION_PERCT) {
      return 2;
    }
  }

  return 1;
}

static void destroyOperatorInfo(SOperatorInfo* pOperator) {
  if (pOperator == NULL) {
    return;
  }

4231
  if (pOperator->fpSet.closeFn != NULL) {
4232
    pOperator->fpSet.closeFn(pOperator->info, pOperator->numOfExprs);
4233 4234
  }

H
Haojun Liao 已提交
4235
  if (pOperator->pDownstream != NULL) {
L
Liu Jicong 已提交
4236
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
H
Haojun Liao 已提交
4237
      destroyOperatorInfo(pOperator->pDownstream[i]);
4238 4239
    }

wafwerar's avatar
wafwerar 已提交
4240
    taosMemoryFreeClear(pOperator->pDownstream);
H
Haojun Liao 已提交
4241
    pOperator->numOfDownstream = 0;
4242 4243
  }

H
Haojun Liao 已提交
4244 4245 4246 4247 4248 4249 4250 4251 4252 4253 4254
  if (pOperator->pExpr != NULL) {
    for (int32_t i = 0; i < pOperator->numOfExprs; ++i) {
      SExprInfo* pExprInfo = &pOperator->pExpr[i];
      if (pExprInfo->pExpr->nodeType == QUERY_NODE_COLUMN) {
        taosMemoryFree(pExprInfo->base.pParam[0].pCol);
      }
      taosMemoryFree(pExprInfo->base.pParam);
      taosMemoryFree(pExprInfo->pExpr);
    }
  }

D
fix bug  
dapan 已提交
4255
  taosMemoryFreeClear(pOperator->pExpr);
wafwerar's avatar
wafwerar 已提交
4256 4257
  taosMemoryFreeClear(pOperator->info);
  taosMemoryFreeClear(pOperator);
4258 4259
}

dengyihao's avatar
dengyihao 已提交
4260 4261
int32_t doInitAggInfoSup(SAggSupporter* pAggSup, SqlFunctionCtx* pCtx, int32_t numOfOutput, size_t keyBufSize,
                         const char* pKey) {
4262 4263
  _hash_fn_t hashFn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY);

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

H
Haojun Liao 已提交
4268
  if (pAggSup->keyBuf == NULL || pAggSup->pResultRowHashTable == NULL) {
4269 4270 4271
    return TSDB_CODE_OUT_OF_MEMORY;
  }

H
Haojun Liao 已提交
4272
  uint32_t defaultPgsz = 4096;
X
Xiaoyu Wang 已提交
4273
  while (defaultPgsz < pAggSup->resultRowSize * 4) {
H
Haojun Liao 已提交
4274 4275 4276 4277 4278 4279 4280 4281 4282 4283
    defaultPgsz <<= 1u;
  }

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

  int32_t code = createDiskbasedBuf(&pAggSup->pResultBuf, defaultPgsz, defaultBufsz, pKey, "/tmp/");
H
Haojun Liao 已提交
4284 4285 4286 4287
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

4288 4289 4290
  return TSDB_CODE_SUCCESS;
}

4291
void cleanupAggSup(SAggSupporter* pAggSup) {
wafwerar's avatar
wafwerar 已提交
4292
  taosMemoryFreeClear(pAggSup->keyBuf);
4293
  taosHashCleanup(pAggSup->pResultRowHashTable);
H
Haojun Liao 已提交
4294
  destroyDiskbasedBuf(pAggSup->pResultBuf);
4295 4296
}

H
Haojun Liao 已提交
4297
int32_t initAggInfo(SOptrBasicInfo* pBasicInfo, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols,
4298
                    SSDataBlock* pResultBlock, size_t keyBufSize, const char* pkey) {
4299
  pBasicInfo->pCtx = createSqlFunctionCtx(pExprInfo, numOfCols, &pBasicInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
4300 4301
  pBasicInfo->pRes = pResultBlock;

4302
  doInitAggInfoSup(pAggSup, pBasicInfo->pCtx, numOfCols, keyBufSize, pkey);
4303

L
Liu Jicong 已提交
4304
  for (int32_t i = 0; i < numOfCols; ++i) {
4305 4306 4307
    pBasicInfo->pCtx[i].pBuf = pAggSup->pResultBuf;
  }

4308
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
4309 4310
}

4311 4312 4313 4314 4315 4316 4317 4318 4319
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;
  }
}

H
Haojun Liao 已提交
4320
static STableQueryInfo* initTableQueryInfo(const STableGroupInfo* pTableGroupInfo) {
L
Liu Jicong 已提交
4321 4322 4323 4324
  if (pTableGroupInfo->numOfTables == 0) {
    return NULL;
  }

wafwerar's avatar
wafwerar 已提交
4325
  STableQueryInfo* pTableQueryInfo = taosMemoryCalloc(pTableGroupInfo->numOfTables, sizeof(STableQueryInfo));
H
Haojun Liao 已提交
4326 4327
  if (pTableQueryInfo == NULL) {
    return NULL;
H
Haojun Liao 已提交
4328
  }
H
Haojun Liao 已提交
4329 4330

  int32_t index = 0;
L
Liu Jicong 已提交
4331
  for (int32_t i = 0; i < taosArrayGetSize(pTableGroupInfo->pGroupList); ++i) {
H
Haojun Liao 已提交
4332
    SArray* pa = taosArrayGetP(pTableGroupInfo->pGroupList, i);
L
Liu Jicong 已提交
4333
    for (int32_t j = 0; j < taosArrayGetSize(pa); ++j) {
H
Haojun Liao 已提交
4334
      STableKeyInfo* pk = taosArrayGet(pa, j);
H
Haojun Liao 已提交
4335
      STableQueryInfo* pTQueryInfo = &pTableQueryInfo[index++];
L
Liu Jicong 已提交
4336
      pTQueryInfo->lastKey = pk->lastKey;
H
Haojun Liao 已提交
4337 4338
    }
  }
H
Haojun Liao 已提交
4339 4340

  STimeWindow win = {0, INT64_MAX};
4341
  createTableQueryInfo(pTableQueryInfo, win);
H
Haojun Liao 已提交
4342
  return pTableQueryInfo;
H
Haojun Liao 已提交
4343 4344
}

L
Liu Jicong 已提交
4345
SOperatorInfo* createAggregateOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
4346
                                           SSDataBlock* pResultBlock, SExprInfo* pScalarExprInfo,
L
Liu Jicong 已提交
4347 4348
                                           int32_t numOfScalarExpr, SExecTaskInfo* pTaskInfo,
                                           const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
4349
  SAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SAggOperatorInfo));
L
Liu Jicong 已提交
4350
  SOperatorInfo*    pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
4351 4352 4353
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
4354

4355
  int32_t numOfRows = 10;
dengyihao's avatar
dengyihao 已提交
4356
  size_t  keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
4357 4358

  initResultSizeInfo(pOperator, numOfRows);
dengyihao's avatar
dengyihao 已提交
4359 4360
  int32_t code =
      initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, pResultBlock, keyBufSize, pTaskInfo->id.str);
H
Haojun Liao 已提交
4361
  pInfo->pTableQueryInfo = initTableQueryInfo(pTableGroupInfo);
L
Liu Jicong 已提交
4362
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
4363 4364
    goto _error;
  }
H
Haojun Liao 已提交
4365

H
Haojun Liao 已提交
4366 4367 4368 4369
  int32_t numOfGroup = 10;  // todo replaced with true value
  pInfo->groupId = INT32_MIN;
  initResultRowInfo(&pInfo->binfo.resultRowInfo, numOfGroup);

4370 4371
  pInfo->pScalarExprInfo = pScalarExprInfo;
  pInfo->numOfScalarExpr = numOfScalarExpr;
4372 4373 4374
  if (pInfo->pScalarExprInfo != NULL) {
    pInfo->pScalarCtx = createSqlFunctionCtx(pScalarExprInfo, numOfCols, &pInfo->rowCellInfoOffset);
  }
4375

dengyihao's avatar
dengyihao 已提交
4376
  pOperator->name = "TableAggregate";
X
Xiaoyu Wang 已提交
4377
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_AGG;
4378
  pOperator->blocking = true;
dengyihao's avatar
dengyihao 已提交
4379 4380 4381
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
4382
  pOperator->numOfExprs = numOfCols;
dengyihao's avatar
dengyihao 已提交
4383
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
4384

4385 4386
  pOperator->fpSet = createOperatorFpSet(doOpenAggregateOptr, getAggregateResult, NULL, NULL, destroyAggOperatorInfo,
                                         aggEncodeResultRow, aggDecodeResultRow, NULL);
H
Haojun Liao 已提交
4387 4388 4389 4390 4391

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

  return pOperator;
L
Liu Jicong 已提交
4394
_error:
H
Haojun Liao 已提交
4395
  destroyAggOperatorInfo(pInfo, numOfCols);
wafwerar's avatar
wafwerar 已提交
4396 4397
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
4398 4399
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
4400 4401
}

H
Haojun Liao 已提交
4402
void doDestroyBasicInfo(SOptrBasicInfo* pInfo, int32_t numOfOutput) {
4403 4404
  assert(pInfo != NULL);

4405
  destroySqlFunctionCtx(pInfo->pCtx, numOfOutput);
wafwerar's avatar
wafwerar 已提交
4406
  taosMemoryFreeClear(pInfo->rowCellInfoOffset);
4407 4408

  cleanupResultRowInfo(&pInfo->resultRowInfo);
H
Haojun Liao 已提交
4409
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
4410 4411
}

H
Haojun Liao 已提交
4412
void destroyBasicOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
4413
  SOptrBasicInfo* pInfo = (SOptrBasicInfo*)param;
4414 4415
  doDestroyBasicInfo(pInfo, numOfOutput);
}
H
Haojun Liao 已提交
4416

H
Haojun Liao 已提交
4417
void destroyMergeJoinOperator(void* param, int32_t numOfOutput) {
X
Xiaoyu Wang 已提交
4418
  SJoinOperatorInfo* pJoinOperator = (SJoinOperatorInfo*)param;
H
Haojun Liao 已提交
4419 4420
}

H
Haojun Liao 已提交
4421
void destroyAggOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
4422
  SAggOperatorInfo* pInfo = (SAggOperatorInfo*)param;
4423 4424
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}
4425

H
Haojun Liao 已提交
4426
void destroySFillOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
4427
  SFillOperatorInfo* pInfo = (SFillOperatorInfo*)param;
4428
  pInfo->pFillInfo = taosDestroyFillInfo(pInfo->pFillInfo);
H
Haojun Liao 已提交
4429
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
wafwerar's avatar
wafwerar 已提交
4430
  taosMemoryFreeClear(pInfo->p);
4431 4432
}

H
Haojun Liao 已提交
4433
static void destroyProjectOperatorInfo(void* param, int32_t numOfOutput) {
D
fix bug  
dapan 已提交
4434 4435 4436
  if (NULL == param) {
    return;
  }
L
Liu Jicong 已提交
4437
  SProjectOperatorInfo* pInfo = (SProjectOperatorInfo*)param;
4438
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
H
Haojun Liao 已提交
4439
  cleanupAggSup(&pInfo->aggSup);
H
Haojun Liao 已提交
4440
  taosArrayDestroy(pInfo->pPseudoColInfo);
4441 4442
}

H
Haojun Liao 已提交
4443
void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
4444
  SExchangeInfo* pExInfo = (SExchangeInfo*)param;
H
Haojun Liao 已提交
4445 4446 4447 4448 4449 4450 4451 4452 4453
  taosArrayDestroy(pExInfo->pSources);
  taosArrayDestroy(pExInfo->pSourceDataInfo);
  if (pExInfo->pResult != NULL) {
    blockDataDestroy(pExInfo->pResult);
  }

  tsem_destroy(&pExInfo->ready);
}

H
Haojun Liao 已提交
4454 4455
static SArray* setRowTsColumnOutputInfo(SqlFunctionCtx* pCtx, int32_t numOfCols) {
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
dengyihao's avatar
dengyihao 已提交
4456
  for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
4457 4458 4459 4460 4461 4462 4463 4464
    if (fmIsPseudoColumnFunc(pCtx[i].functionId)) {
      taosArrayPush(pList, &i);
    }
  }

  return pList;
}

L
Liu Jicong 已提交
4465
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t num,
dengyihao's avatar
dengyihao 已提交
4466 4467
                                         SSDataBlock* pResBlock, SLimit* pLimit, SLimit* pSlimit,
                                         SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
4468
  SProjectOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SProjectOperatorInfo));
L
Liu Jicong 已提交
4469
  SOperatorInfo*        pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
4470 4471 4472
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
4473

dengyihao's avatar
dengyihao 已提交
4474 4475 4476
  pInfo->limit = *pLimit;
  pInfo->slimit = *pSlimit;
  pInfo->curOffset = pLimit->offset;
H
Haojun Liao 已提交
4477 4478
  pInfo->curSOffset = pSlimit->offset;

H
Haojun Liao 已提交
4479
  pInfo->binfo.pRes = pResBlock;
H
Haojun Liao 已提交
4480 4481 4482

  int32_t numOfCols = num;
  int32_t numOfRows = 4096;
dengyihao's avatar
dengyihao 已提交
4483
  size_t  keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
4484 4485 4486

  initResultSizeInfo(pOperator, numOfRows);
  initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, pResBlock, keyBufSize, pTaskInfo->id.str);
H
Haojun Liao 已提交
4487
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, numOfCols, pTaskInfo);
H
Haojun Liao 已提交
4488
  pInfo->pPseudoColInfo = setRowTsColumnOutputInfo(pInfo->binfo.pCtx, numOfCols);
4489

dengyihao's avatar
dengyihao 已提交
4490
  pOperator->name = "ProjectOperator";
H
Haojun Liao 已提交
4491
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_PROJECT;
4492
  pOperator->blocking = false;
dengyihao's avatar
dengyihao 已提交
4493 4494 4495
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
4496
  pOperator->numOfExprs = num;
4497

L
Liu Jicong 已提交
4498 4499
  pOperator->fpSet = createOperatorFpSet(operatorDummyOpenFn, doProjectOperation, NULL, NULL,
                                         destroyProjectOperatorInfo, NULL, NULL, NULL);
L
Liu Jicong 已提交
4500 4501

  pOperator->pTaskInfo = pTaskInfo;
4502
  int32_t code = appendDownstream(pOperator, &downstream, 1);
H
Haojun Liao 已提交
4503
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
4504 4505
    goto _error;
  }
4506 4507

  return pOperator;
H
Haojun Liao 已提交
4508

L
Liu Jicong 已提交
4509
_error:
H
Haojun Liao 已提交
4510 4511
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
4512 4513
}

4514
static int32_t initFillInfo(SFillOperatorInfo* pInfo, SExprInfo* pExpr, int32_t numOfCols, SNodeListNode* pValNode,
L
Liu Jicong 已提交
4515
                            STimeWindow win, int32_t capacity, const char* id, SInterval* pInterval, int32_t fillType) {
4516
  SFillColInfo* pColInfo = createFillColInfo(pExpr, numOfCols, pValNode);
H
Haojun Liao 已提交
4517 4518

  STimeWindow w = TSWINDOW_INITIALIZER;
4519
  getAlignQueryTimeWindow(pInterval, pInterval->precision, win.skey, &w);
H
Haojun Liao 已提交
4520 4521

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

wafwerar's avatar
wafwerar 已提交
4524
  pInfo->p = taosMemoryCalloc(numOfCols, POINTER_BYTES);
H
Haojun Liao 已提交
4525 4526 4527 4528 4529 4530 4531
  if (pInfo->pFillInfo == NULL || pInfo->p == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  } else {
    return TSDB_CODE_SUCCESS;
  }
}

L
Liu Jicong 已提交
4532
SOperatorInfo* createFillOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExpr, int32_t numOfCols,
X
Xiaoyu Wang 已提交
4533 4534 4535
                                      SInterval* pInterval, STimeWindow* pWindow, SSDataBlock* pResBlock,
                                      int32_t fillType, SNodeListNode* pValueNode, bool multigroupResult,
                                      SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
4536
  SFillOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SFillOperatorInfo));
L
Liu Jicong 已提交
4537
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
4538

L
Liu Jicong 已提交
4539
  pInfo->pRes = pResBlock;
4540 4541
  pInfo->multigroupResult = multigroupResult;

4542 4543
  int32_t type = TSDB_FILL_NONE;
  switch (fillType) {
dengyihao's avatar
dengyihao 已提交
4544 4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558 4559 4560 4561
    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;
4562 4563 4564 4565
    default:
      type = TSDB_FILL_NONE;
  }

H
Haojun Liao 已提交
4566
  SResultInfo* pResultInfo = &pOperator->resultInfo;
4567 4568
  initResultSizeInfo(pOperator, 4096);

X
Xiaoyu Wang 已提交
4569 4570
  int32_t code = initFillInfo(pInfo, pExpr, numOfCols, pValueNode, *pWindow, pResultInfo->capacity, pTaskInfo->id.str,
                              pInterval, type);
4571 4572 4573
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
4574

dengyihao's avatar
dengyihao 已提交
4575
  pOperator->name = "FillOperator";
4576
  pOperator->blocking = false;
dengyihao's avatar
dengyihao 已提交
4577
  pOperator->status = OP_NOT_OPENED;
4578
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_FILL;
dengyihao's avatar
dengyihao 已提交
4579
  pOperator->pExpr = pExpr;
4580
  pOperator->numOfExprs = numOfCols;
dengyihao's avatar
dengyihao 已提交
4581
  pOperator->info = pInfo;
H
Haojun Liao 已提交
4582

L
Liu Jicong 已提交
4583 4584
  pOperator->fpSet =
      createOperatorFpSet(operatorDummyOpenFn, doFill, NULL, NULL, destroySFillOperatorInfo, NULL, NULL, NULL);
4585
  pOperator->pTaskInfo = pTaskInfo;
4586
  code = appendDownstream(pOperator, &downstream, 1);
4587
  return pOperator;
H
Haojun Liao 已提交
4588

L
Liu Jicong 已提交
4589
_error:
wafwerar's avatar
wafwerar 已提交
4590 4591
  taosMemoryFreeClear(pOperator);
  taosMemoryFreeClear(pInfo);
H
Haojun Liao 已提交
4592
  return NULL;
4593 4594
}

L
Liu Jicong 已提交
4595 4596
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
                                  const char* name) {
H
Haojun Liao 已提交
4597
  SResSchema s = {0};
dengyihao's avatar
dengyihao 已提交
4598 4599 4600 4601
  s.scale = scale;
  s.type = type;
  s.bytes = bytes;
  s.slotId = slotId;
H
Haojun Liao 已提交
4602
  s.precision = precision;
H
Haojun Liao 已提交
4603 4604 4605 4606
  strncpy(s.name, name, tListLen(s.name));

  return s;
}
H
Haojun Liao 已提交
4607

4608
static SColumn* createColumn(int32_t blockId, int32_t slotId, int32_t colId, SDataType* pType) {
H
Haojun Liao 已提交
4609 4610 4611 4612 4613 4614
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
  if (pCol == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

dengyihao's avatar
dengyihao 已提交
4615
  pCol->slotId = slotId;
4616 4617 4618 4619
  pCol->colId  = colId;
  pCol->bytes  = pType->bytes;
  pCol->type   = pType->type;
  pCol->scale  = pType->scale;
dengyihao's avatar
dengyihao 已提交
4620
  pCol->precision = pType->precision;
H
Haojun Liao 已提交
4621 4622 4623 4624 4625
  pCol->dataBlockId = blockId;

  return pCol;
}

H
Haojun Liao 已提交
4626
SExprInfo* createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, int32_t* numOfExprs) {
H
Haojun Liao 已提交
4627
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
4628 4629 4630 4631
  int32_t numOfGroupKeys = 0;
  if (pGroupKeys != NULL) {
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
  }
H
Haojun Liao 已提交
4632

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

L
Liu Jicong 已提交
4636
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
H
Haojun Liao 已提交
4637 4638 4639 4640 4641 4642
    STargetNode* pTargetNode = NULL;
    if (i < numOfFuncs) {
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    } else {
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
    }
H
Haojun Liao 已提交
4643

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

wafwerar's avatar
wafwerar 已提交
4646
    pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
H
Haojun Liao 已提交
4647
    pExp->pExpr->_function.num = 1;
H
Haojun Liao 已提交
4648
    pExp->pExpr->_function.functionId = -1;
H
Haojun Liao 已提交
4649

4650
    int32_t type = nodeType(pTargetNode->pExpr);
H
Haojun Liao 已提交
4651
    // it is a project query, or group by column
4652
    if (type == QUERY_NODE_COLUMN) {
H
Haojun Liao 已提交
4653
      pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
L
Liu Jicong 已提交
4654
      SColumnNode* pColNode = (SColumnNode*)pTargetNode->pExpr;
H
Haojun Liao 已提交
4655

G
Ganlin Zhao 已提交
4656 4657 4658
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;

H
Haojun Liao 已提交
4659
      SDataType* pType = &pColNode->node.resType;
dengyihao's avatar
dengyihao 已提交
4660 4661
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pColNode->colName);
4662
      pExp->base.pParam[0].pCol = createColumn(pColNode->dataBlockId, pColNode->slotId, pColNode->colId, pType);
H
Haojun Liao 已提交
4663
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
4664
    } else if (type == QUERY_NODE_VALUE) {
4665 4666 4667 4668 4669 4670 4671
      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 已提交
4672 4673
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pValNode->node.aliasName);
4674 4675
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
      valueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
4676
    } else if (type == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
4677
      pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
H
Haojun Liao 已提交
4678 4679 4680
      SFunctionNode* pFuncNode = (SFunctionNode*)pTargetNode->pExpr;

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

H
Haojun Liao 已提交
4684
      pExp->pExpr->_function.functionId = pFuncNode->funcId;
H
Haojun Liao 已提交
4685
      pExp->pExpr->_function.pFunctNode = pFuncNode;
4686

dengyihao's avatar
dengyihao 已提交
4687 4688
      strncpy(pExp->pExpr->_function.functionName, pFuncNode->functionName,
              tListLen(pExp->pExpr->_function.functionName));
4689 4690 4691 4692 4693 4694 4695 4696 4697 4698 4699 4700 4701
#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);
        SValueNode *res = (SValueNode *)nodesMakeNode(QUERY_NODE_VALUE);
        if (NULL == res) { // todo handle error
        } else {
          res->node.resType = (SDataType) {.bytes = sizeof(int64_t), .type = TSDB_DATA_TYPE_BIGINT};
          nodesListAppend(pFuncNode->pParameterList, res);
        }
      }
#endif
H
Haojun Liao 已提交
4702 4703

      int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
G
Ganlin Zhao 已提交
4704 4705 4706 4707

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

H
Haojun Liao 已提交
4708
      for (int32_t j = 0; j < numOfParam; ++j) {
4709
        SNode* p1 = nodesListGetNode(pFuncNode->pParameterList, j);
G
Ganlin Zhao 已提交
4710
        if (p1->type == QUERY_NODE_COLUMN) {
dengyihao's avatar
dengyihao 已提交
4711
          SColumnNode* pcn = (SColumnNode*)p1;
G
Ganlin Zhao 已提交
4712 4713

          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
4714
          pExp->base.pParam[j].pCol = createColumn(pcn->dataBlockId, pcn->slotId, pcn->colId, &pcn->node.resType);
G
Ganlin Zhao 已提交
4715 4716 4717
        } else if (p1->type == QUERY_NODE_VALUE) {
          SValueNode* pvn = (SValueNode*)p1;
          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
4718
          valueNodeToVariant(pvn, &pExp->base.pParam[j].param);
G
Ganlin Zhao 已提交
4719
        }
H
Haojun Liao 已提交
4720
      }
4721
    } else if (type == QUERY_NODE_OPERATOR) {
H
Haojun Liao 已提交
4722
      pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
L
Liu Jicong 已提交
4723
      SOperatorNode* pNode = (SOperatorNode*)pTargetNode->pExpr;
4724

G
Ganlin Zhao 已提交
4725 4726 4727
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;

4728
      SDataType* pType = &pNode->node.resType;
dengyihao's avatar
dengyihao 已提交
4729 4730
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pNode->node.aliasName);
4731 4732 4733
      pExp->pExpr->_optrRoot.pRootNode = pTargetNode->pExpr;
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
4734 4735 4736
    }
  }

H
Haojun Liao 已提交
4737
  return pExprs;
H
Haojun Liao 已提交
4738 4739
}

4740
static SExecTaskInfo* createExecTaskInfo(uint64_t queryId, uint64_t taskId, EOPTR_EXEC_MODEL model) {
wafwerar's avatar
wafwerar 已提交
4741
  SExecTaskInfo* pTaskInfo = taosMemoryCalloc(1, sizeof(SExecTaskInfo));
4742
  setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
4743

4744
  pTaskInfo->cost.created = taosGetTimestampMs();
H
Haojun Liao 已提交
4745
  pTaskInfo->id.queryId = queryId;
dengyihao's avatar
dengyihao 已提交
4746
  pTaskInfo->execModel = model;
H
Haojun Liao 已提交
4747

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

4752 4753
  return pTaskInfo;
}
H
Haojun Liao 已提交
4754

L
Liu Jicong 已提交
4755 4756
static tsdbReaderT doCreateDataReader(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle,
                                      STableGroupInfo* pTableGroupInfo, uint64_t queryId, uint64_t taskId);
H
Haojun Liao 已提交
4757

L
Liu Jicong 已提交
4758 4759
static int32_t doCreateTableGroup(void* metaHandle, int32_t tableType, uint64_t tableUid, STableGroupInfo* pGroupInfo,
                                  uint64_t queryId, uint64_t taskId);
4760
static SArray* extractTableIdList(const STableGroupInfo* pTableGroupInfo);
H
Haojun Liao 已提交
4761
static SArray* extractColumnInfo(SNodeList* pNodeList);
4762

4763
static SArray* createSortInfo(SNodeList* pNodeList);
4764
static SArray* extractPartitionColInfo(SNodeList* pNodeList);
4765
static void    setJoinColumnInfo(SColumnInfo* pColumn, const SColumnNode* pColumnNode);
4766

H
Haojun Liao 已提交
4767
SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, SReadHandle* pHandle,
4768
                                  uint64_t queryId, uint64_t taskId, STableGroupInfo* pTableGroupInfo) {
4769 4770
  int32_t type = nodeType(pPhyNode);

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

H
Haojun Liao 已提交
4775
      tsdbReaderT pDataReader = doCreateDataReader(pTableScanNode, pHandle, pTableGroupInfo, (uint64_t)queryId, taskId);
4776
      if (pDataReader == NULL && terrno != 0) {
4777 4778
        return NULL;
      }
4779

4780
      SOperatorInfo* pOperator = createTableScanOperatorInfo(pTableScanNode, pDataReader, pHandle, pTaskInfo);
4781

4782 4783
      STableScanInfo* pScanInfo = pOperator->info;
      pTaskInfo->cost.pRecoder = &pScanInfo->readRecorder;
4784

4785
      return pOperator;
H
Haojun Liao 已提交
4786
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == type) {
H
Haojun Liao 已提交
4787
      SExchangePhysiNode* pExchange = (SExchangePhysiNode*)pPhyNode;
4788
      SSDataBlock*        pResBlock = createResDataBlock(pExchange->node.pOutputDataBlockDesc);
4789
      return createExchangeOperatorInfo(pHandle->pMsgCb->clientRpc, pExchange->pSrcEndPoints, pResBlock, pTaskInfo);
H
Haojun Liao 已提交
4790
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == type) {
H
Haojun Liao 已提交
4791
      SScanPhysiNode* pScanPhyNode = (SScanPhysiNode*)pPhyNode;  // simple child table.
5
54liuyao 已提交
4792
      STableScanPhysiNode* pTableScanNode = (STableScanPhysiNode*)pPhyNode;
4793

4794
      int32_t numOfCols = 0;
5
54liuyao 已提交
4795 4796 4797 4798 4799 4800 4801 4802

      tsdbReaderT pDataReader = NULL;
      if (pHandle->vnode) {
        pDataReader = doCreateDataReader(pTableScanNode, pHandle, pTableGroupInfo, (uint64_t)queryId, taskId);
      } else {
        doCreateTableGroup(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableGroupInfo, 
            queryId, taskId);
      }
4803

5
54liuyao 已提交
4804 4805 4806 4807 4808 4809
      if (pDataReader == NULL && terrno != 0) {
        qDebug("pDataReader is     NULL");
        // return NULL;
      } else {
        qDebug("pDataReader is not NULL");
      }
H
Haojun Liao 已提交
4810

4811
      SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;
4812
      SOperatorInfo* pOperatorDumy = createTableScanOperatorInfo(pTableScanNode, pDataReader, pHandle, pTaskInfo);
5
54liuyao 已提交
4813 4814

      SArray* tableIdList = extractTableIdList(pTableGroupInfo);
X
Xiaoyu Wang 已提交
4815
      SSDataBlock*        pResBlock = createResDataBlock(pDescNode);
4816

X
Xiaoyu Wang 已提交
4817
      SArray*        pCols = extractColMatchInfo(pScanPhyNode->pScanCols, pDescNode, &numOfCols, COL_MATCH_FROM_COL_ID);
5
54liuyao 已提交
4818
      SOperatorInfo* pOperator = createStreamScanOperatorInfo(pHandle->reader, pDataReader, pResBlock, pCols, tableIdList, pTaskInfo,
5
54liuyao 已提交
4819
                                                              pScanPhyNode->node.pConditions, pOperatorDumy);
4820
      taosArrayDestroy(tableIdList);
H
Haojun Liao 已提交
4821
      return pOperator;
H
Haojun Liao 已提交
4822
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == type) {
L
Liu Jicong 已提交
4823
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
L
Liu Jicong 已提交
4824
      SScanPhysiNode*            pScanNode = &pSysScanPhyNode->scan;
H
Haojun Liao 已提交
4825

4826 4827 4828
      SDataBlockDescNode* pDescNode = pScanNode->node.pOutputDataBlockDesc;

      SSDataBlock* pResBlock = createResDataBlock(pDescNode);
4829

4830
      int32_t numOfOutputCols = 0;
4831
      SArray* colList = extractColMatchInfo(pScanNode->pScanCols, pDescNode, &numOfOutputCols, COL_MATCH_FROM_COL_ID);
L
Liu Jicong 已提交
4832
      SOperatorInfo* pOperator = createSysTableScanOperatorInfo(
L
Liu Jicong 已提交
4833 4834
          pHandle, pResBlock, &pScanNode->tableName, pScanNode->node.pConditions, pSysScanPhyNode->mgmtEpSet, colList,
          pTaskInfo, pSysScanPhyNode->showRewrite, pSysScanPhyNode->accountId);
H
Haojun Liao 已提交
4835
      return pOperator;
4836
    } else if (QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN == type) {
X
Xiaoyu Wang 已提交
4837
      STagScanPhysiNode* pScanPhyNode = (STagScanPhysiNode*)pPhyNode;
4838 4839 4840 4841

      SDataBlockDescNode* pDescNode = pScanPhyNode->node.pOutputDataBlockDesc;

      SSDataBlock* pResBlock = createResDataBlock(pDescNode);
4842

X
Xiaoyu Wang 已提交
4843 4844
      int32_t code = doCreateTableGroup(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableGroupInfo,
                                        queryId, taskId);
4845 4846 4847 4848
      if (code != TSDB_CODE_SUCCESS) {
        return NULL;
      }

X
Xiaoyu Wang 已提交
4849
      int32_t    num = 0;
4850 4851 4852
      SExprInfo* pExprInfo = createExprInfo(pScanPhyNode->pScanPseudoCols, NULL, &num);

      int32_t numOfOutputCols = 0;
X
Xiaoyu Wang 已提交
4853 4854
      SArray* colList =
          extractColMatchInfo(pScanPhyNode->pScanPseudoCols, pDescNode, &numOfOutputCols, COL_MATCH_FROM_COL_ID);
4855

X
Xiaoyu Wang 已提交
4856 4857
      SOperatorInfo* pOperator =
          createTagScanOperatorInfo(pHandle, pExprInfo, num, pResBlock, colList, pTableGroupInfo, pTaskInfo);
4858
      return pOperator;
H
Haojun Liao 已提交
4859 4860
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
4861 4862 4863
    }
  }

4864 4865
  int32_t num = 0;
  size_t  size = LIST_LENGTH(pPhyNode->pChildren);
H
Haojun Liao 已提交
4866

4867
  SOperatorInfo** ops = taosMemoryCalloc(size, POINTER_BYTES);
dengyihao's avatar
dengyihao 已提交
4868
  for (int32_t i = 0; i < size; ++i) {
4869 4870
    SPhysiNode* pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
    ops[i] = createOperatorTree(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);
4871 4872 4873
    if (ops[i] == NULL) {
      return NULL;
    }
4874
  }
H
Haojun Liao 已提交
4875

4876
  SOperatorInfo* pOptr = NULL;
H
Haojun Liao 已提交
4877
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
dengyihao's avatar
dengyihao 已提交
4878 4879
    SProjectPhysiNode* pProjPhyNode = (SProjectPhysiNode*)pPhyNode;
    SExprInfo*         pExprInfo = createExprInfo(pProjPhyNode->pProjections, NULL, &num);
H
Haojun Liao 已提交
4880

4881
    SSDataBlock* pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
dengyihao's avatar
dengyihao 已提交
4882 4883
    SLimit       limit = {.limit = pProjPhyNode->limit, .offset = pProjPhyNode->offset};
    SLimit       slimit = {.limit = pProjPhyNode->slimit, .offset = pProjPhyNode->soffset};
4884
    pOptr = createProjectOperatorInfo(ops[0], pExprInfo, num, pResBlock, &limit, &slimit, pTaskInfo);
H
Haojun Liao 已提交
4885 4886 4887
  } else if (QUERY_NODE_PHYSICAL_PLAN_AGG == type) {
    SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
    SExprInfo*     pExprInfo = createExprInfo(pAggNode->pAggFuncs, pAggNode->pGroupKeys, &num);
4888
    SSDataBlock*   pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
H
Haojun Liao 已提交
4889

dengyihao's avatar
dengyihao 已提交
4890
    int32_t    numOfScalarExpr = 0;
4891 4892 4893 4894 4895
    SExprInfo* pScalarExprInfo = NULL;
    if (pAggNode->pExprs != NULL) {
      pScalarExprInfo = createExprInfo(pAggNode->pExprs, NULL, &numOfScalarExpr);
    }

H
Haojun Liao 已提交
4896 4897
    if (pAggNode->pGroupKeys != NULL) {
      SArray* pColList = extractColumnInfo(pAggNode->pGroupKeys);
dengyihao's avatar
dengyihao 已提交
4898 4899
      pOptr = createGroupOperatorInfo(ops[0], pExprInfo, num, pResBlock, pColList, pAggNode->node.pConditions,
                                      pScalarExprInfo, numOfScalarExpr, pTaskInfo, NULL);
H
Haojun Liao 已提交
4900
    } else {
dengyihao's avatar
dengyihao 已提交
4901 4902
      pOptr = createAggregateOperatorInfo(ops[0], pExprInfo, num, pResBlock, pScalarExprInfo, numOfScalarExpr,
                                          pTaskInfo, pTableGroupInfo);
H
Haojun Liao 已提交
4903
    }
X
Xiaoyu Wang 已提交
4904
  } else if (QUERY_NODE_PHYSICAL_PLAN_INTERVAL == type || QUERY_NODE_PHYSICAL_PLAN_STREAM_INTERVAL == type) {
H
Haojun Liao 已提交
4905
    SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
4906

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

dengyihao's avatar
dengyihao 已提交
4910 4911 4912 4913 4914 4915
    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 已提交
4916

dengyihao's avatar
dengyihao 已提交
4917 4918
    STimeWindowAggSupp as = {.waterMark = pIntervalPhyNode->window.watermark,
                             .calTrigger = pIntervalPhyNode->window.triggerType};
4919

4920 4921 4922 4923
    int32_t tsSlotId = ((SColumnNode*)pIntervalPhyNode->window.pTspk)->slotId;
    pOptr = createIntervalOperatorInfo(ops[0], pExprInfo, num, pResBlock, &interval, tsSlotId, &as, pTableGroupInfo,
                                       pTaskInfo);

H
Haojun Liao 已提交
4924
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) {
H
Haojun Liao 已提交
4925
    SSortPhysiNode* pSortPhyNode = (SSortPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
4926

4927 4928 4929
    SDataBlockDescNode* pDescNode = pPhyNode->pOutputDataBlockDesc;

    SSDataBlock* pResBlock = createResDataBlock(pDescNode);
4930
    SArray*      info = createSortInfo(pSortPhyNode->pSortKeys);
4931 4932

    int32_t    numOfCols = 0;
4933 4934 4935
    SExprInfo* pExprInfo = createExprInfo(pSortPhyNode->pExprs, NULL, &numOfCols);

    int32_t numOfOutputCols = 0;
4936
    SArray* pColList = extractColMatchInfo(pSortPhyNode->pTargets, pDescNode, &numOfOutputCols, COL_MATCH_FROM_SLOT_ID);
4937

4938
    pOptr = createSortOperatorInfo(ops[0], pResBlock, info, pExprInfo, numOfCols, pColList, pTaskInfo);
H
Haojun Liao 已提交
4939
  } else if (QUERY_NODE_PHYSICAL_PLAN_SESSION_WINDOW == type) {
H
Haojun Liao 已提交
4940 4941
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;

X
Xiaoyu Wang 已提交
4942 4943
    STimeWindowAggSupp as = {.waterMark = pSessionNode->window.watermark,
                             .calTrigger = pSessionNode->window.triggerType};
4944

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

X
Xiaoyu Wang 已提交
4949 4950
    pOptr =
        createSessionAggOperatorInfo(ops[0], pExprInfo, num, pResBlock, pSessionNode->gap, tsSlotId, &as, pTaskInfo);
H
Haojun Liao 已提交
4951
  } else if (QUERY_NODE_PHYSICAL_PLAN_PARTITION == type) {
dengyihao's avatar
dengyihao 已提交
4952 4953 4954
    SPartitionPhysiNode* pPartNode = (SPartitionPhysiNode*)pPhyNode;
    SArray*              pColList = extractPartitionColInfo(pPartNode->pPartitionKeys);
    SSDataBlock*         pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
4955 4956

    SExprInfo* pExprInfo = createExprInfo(pPartNode->pTargets, NULL, &num);
4957
    pOptr = createPartitionOperatorInfo(ops[0], pExprInfo, num, pResBlock, pColList, pTaskInfo, NULL);
4958
  } else if (QUERY_NODE_PHYSICAL_PLAN_STATE_WINDOW == type) {
dengyihao's avatar
dengyihao 已提交
4959
    SStateWinodwPhysiNode* pStateNode = (SStateWinodwPhysiNode*)pPhyNode;
4960

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

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

4967 4968 4969
    SColumnNode* pColNode = (SColumnNode*)((STargetNode*)pStateNode->pStateKey)->pExpr;
    SColumn col = extractColumnFromColumnNode(pColNode);
    pOptr = createStatewindowOperatorInfo(ops[0], pExprInfo, num, pResBlock, &as, tsSlotId, &col, pTaskInfo);
4970
  } else if (QUERY_NODE_PHYSICAL_PLAN_JOIN == type) {
dengyihao's avatar
dengyihao 已提交
4971 4972
    SJoinPhysiNode* pJoinNode = (SJoinPhysiNode*)pPhyNode;
    SSDataBlock*    pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
4973 4974

    SExprInfo* pExprInfo = createExprInfo(pJoinNode->pTargets, NULL, &num);
H
Haojun Liao 已提交
4975
    pOptr = createMergeJoinOperatorInfo(ops, size, pExprInfo, num, pResBlock, pJoinNode->pOnConditions, pTaskInfo);
4976 4977
  } else if (QUERY_NODE_PHYSICAL_PLAN_FILL == type) {
    SFillPhysiNode* pFillNode = (SFillPhysiNode*)pPhyNode;
X
Xiaoyu Wang 已提交
4978 4979
    SSDataBlock*    pResBlock = createResDataBlock(pPhyNode->pOutputDataBlockDesc);
    SExprInfo*      pExprInfo = createExprInfo(pFillNode->pTargets, NULL, &num);
4980

4981
    SInterval* pInterval = &((SIntervalAggOperatorInfo*)ops[0]->info)->interval;
X
Xiaoyu Wang 已提交
4982 4983
    pOptr = createFillOperatorInfo(ops[0], pExprInfo, num, pInterval, &pFillNode->timeRange, pResBlock, pFillNode->mode,
                                   (SNodeListNode*)pFillNode->pValues, false, pTaskInfo);
H
Haojun Liao 已提交
4984 4985
  } else {
    ASSERT(0);
H
Haojun Liao 已提交
4986
  }
4987 4988 4989

  taosMemoryFree(ops);
  return pOptr;
4990
}
H
Haojun Liao 已提交
4991

4992
int32_t initQueryTableDataCond(SQueryTableDataCond* pCond, const STableScanPhysiNode* pTableScanNode) {
4993 4994 4995 4996 4997 4998
  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 已提交
4999
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
5000 5001 5002 5003 5004 5005
    return terrno;
  }

  pCond->twindow = pTableScanNode->scanRange;

#if 1
L
Liu Jicong 已提交
5006
  // todo work around a problem, remove it later
5007 5008
  if ((pCond->order == TSDB_ORDER_ASC && pCond->twindow.skey > pCond->twindow.ekey) ||
      (pCond->order == TSDB_ORDER_DESC && pCond->twindow.skey < pCond->twindow.ekey)) {
wafwerar's avatar
wafwerar 已提交
5009
    TSWAP(pCond->twindow.skey, pCond->twindow.ekey);
H
Haojun Liao 已提交
5010
  }
5011
#endif
H
Haojun Liao 已提交
5012

5013 5014
  pCond->type = BLOCK_LOAD_OFFSET_SEQ_ORDER;
  //  pCond->type = pTableScanNode->scanFlag;
H
Haojun Liao 已提交
5015

H
Haojun Liao 已提交
5016
  int32_t j = 0;
5017
  for (int32_t i = 0; i < pCond->numOfCols; ++i) {
H
Haojun Liao 已提交
5018 5019
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
5020 5021 5022
    if (pColNode->colType == COLUMN_TYPE_TAG) {
      continue;
    }
H
Haojun Liao 已提交
5023

5024 5025 5026
    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 已提交
5027
    j += 1;
H
Haojun Liao 已提交
5028 5029
  }

5030 5031
  pCond->numOfCols = j;
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
5032
}
H
Haojun Liao 已提交
5033

5034 5035 5036 5037 5038 5039 5040 5041 5042 5043 5044
SColumn extractColumnFromColumnNode(SColumnNode* pColNode) {
  SColumn c = {0};
  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;
  c.precision = pColNode->node.resType.precision;
  return c;
}

H
Haojun Liao 已提交
5045
SArray* extractColumnInfo(SNodeList* pNodeList) {
L
Liu Jicong 已提交
5046
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
5047 5048 5049 5050 5051 5052
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

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

5056 5057 5058
    if (nodeType(pNode->pExpr) == QUERY_NODE_COLUMN) {
      SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;

5059
      SColumn c = extractColumnFromColumnNode(pColNode);
5060 5061
      taosArrayPush(pList, &c);
    } else if (nodeType(pNode->pExpr) == QUERY_NODE_VALUE) {
L
Liu Jicong 已提交
5062 5063
      SValueNode* pValNode = (SValueNode*)pNode->pExpr;
      SColumn     c = {0};
5064
      c.slotId = pNode->slotId;
L
Liu Jicong 已提交
5065 5066 5067 5068
      c.colId = pNode->slotId;
      c.type = pValNode->node.type;
      c.bytes = pValNode->node.resType.bytes;
      c.scale = pValNode->node.resType.scale;
5069 5070 5071 5072
      c.precision = pValNode->node.resType.precision;

      taosArrayPush(pList, &c);
    }
H
Haojun Liao 已提交
5073 5074 5075 5076 5077
  }

  return pList;
}

5078 5079 5080 5081 5082 5083 5084 5085 5086 5087 5088 5089 5090 5091
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 已提交
5092 5093 5094
    c.colId = pColNode->colId;
    c.type = pColNode->node.resType.type;
    c.bytes = pColNode->node.resType.bytes;
5095 5096 5097 5098 5099 5100 5101 5102 5103
    c.precision = pColNode->node.resType.precision;
    c.scale = pColNode->node.resType.scale;

    taosArrayPush(pList, &c);
  }

  return pList;
}

5104
SArray* createSortInfo(SNodeList* pNodeList) {
L
Liu Jicong 已提交
5105
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
5106 5107 5108 5109 5110 5111
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return pList;
  }

L
Liu Jicong 已提交
5112
  for (int32_t i = 0; i < numOfCols; ++i) {
5113
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
L
Liu Jicong 已提交
5114 5115
    SBlockOrderInfo   bi = {0};
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
H
Haojun Liao 已提交
5116 5117 5118
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);

    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
5119
    bi.slotId = pColNode->slotId;
H
Haojun Liao 已提交
5120 5121 5122 5123 5124 5125
    taosArrayPush(pList, &bi);
  }

  return pList;
}

X
Xiaoyu Wang 已提交
5126 5127
SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols,
                            int32_t type) {
L
Liu Jicong 已提交
5128
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
5129 5130 5131 5132 5133 5134
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
5135 5136 5137
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
5138 5139

    SColMatchInfo c = {0};
X
Xiaoyu Wang 已提交
5140 5141 5142 5143
    c.output = true;
    c.colId = pColNode->colId;
    c.srcSlotId = pColNode->slotId;
    c.matchType = type;
H
Haojun Liao 已提交
5144 5145 5146 5147
    c.targetSlotId = pNode->slotId;
    taosArrayPush(pList, &c);
  }

H
Haojun Liao 已提交
5148 5149
  *numOfOutputCols = 0;
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
L
Liu Jicong 已提交
5150 5151
  for (int32_t i = 0; i < num; ++i) {
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
5152

5153
    // todo: add reserve flag check
5154 5155
    // it is a column reserved for the arithmetic expression calculation
    if (pNode->slotId >= numOfCols) {
5156 5157 5158 5159
      (*numOfOutputCols) += 1;
      continue;
    }

H
Haojun Liao 已提交
5160
    SColMatchInfo* info = taosArrayGet(pList, pNode->slotId);
H
Haojun Liao 已提交
5161 5162 5163 5164 5165
    if (pNode->output) {
      (*numOfOutputCols) += 1;
    } else {
      info->output = false;
    }
H
Haojun Liao 已提交
5166 5167
  }

H
Haojun Liao 已提交
5168 5169 5170
  return pList;
}

dengyihao's avatar
dengyihao 已提交
5171 5172
int32_t doCreateTableGroup(void* metaHandle, int32_t tableType, uint64_t tableUid, STableGroupInfo* pGroupInfo,
                           uint64_t queryId, uint64_t taskId) {
5173
  int32_t code = 0;
H
Haojun Liao 已提交
5174
  if (tableType == TSDB_SUPER_TABLE) {
H
Haojun Liao 已提交
5175
    code = tsdbQuerySTableByTagCond(metaHandle, tableUid, 0, NULL, 0, 0, NULL, pGroupInfo, NULL, 0, queryId, taskId);
H
Haojun Liao 已提交
5176
  } else {  // Create one table group.
H
Haojun Liao 已提交
5177
    code = tsdbGetOneTableGroup(metaHandle, tableUid, 0, pGroupInfo);
5178 5179 5180 5181
  }

  return code;
}
H
Haojun Liao 已提交
5182

5183 5184 5185 5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198 5199 5200
SArray* extractTableIdList(const STableGroupInfo* pTableGroupInfo) {
  SArray* tableIdList = taosArrayInit(4, sizeof(uint64_t));

  if (pTableGroupInfo->numOfTables > 0) {
    SArray* pa = taosArrayGetP(pTableGroupInfo->pGroupList, 0);
    ASSERT(taosArrayGetSize(pTableGroupInfo->pGroupList) == 1);

    // Transfer the Array of STableKeyInfo into uid list.
    size_t numOfTables = taosArrayGetSize(pa);
    for (int32_t i = 0; i < numOfTables; ++i) {
      STableKeyInfo* pkeyInfo = taosArrayGet(pa, i);
      taosArrayPush(tableIdList, &pkeyInfo->uid);
    }
  }

  return tableIdList;
}

L
Liu Jicong 已提交
5201 5202
tsdbReaderT doCreateDataReader(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle,
                               STableGroupInfo* pTableGroupInfo, uint64_t queryId, uint64_t taskId) {
5203
  uint64_t uid = pTableScanNode->scan.uid;
L
Liu Jicong 已提交
5204 5205
  int32_t  code =
      doCreateTableGroup(pHandle->meta, pTableScanNode->scan.tableType, uid, pTableGroupInfo, queryId, taskId);
5206 5207
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
H
Haojun Liao 已提交
5208
  }
H
Haojun Liao 已提交
5209

H
Haojun Liao 已提交
5210
  if (pTableGroupInfo->numOfTables == 0) {
H
Haojun Liao 已提交
5211
    code = 0;
L
Liu Jicong 已提交
5212
    qDebug("no table qualified for query, TID:0x%" PRIx64 ", QID:0x%" PRIx64, taskId, queryId);
H
Haojun Liao 已提交
5213 5214
    goto _error;
  }
H
Haojun Liao 已提交
5215

5216 5217 5218 5219
  SQueryTableDataCond cond = {0};
  code = initQueryTableDataCond(&cond, pTableScanNode);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
X
Xiaoyu Wang 已提交
5220
  }
5221

5222
  return tsdbQueryTables(pHandle->vnode, &cond, pTableGroupInfo, queryId, taskId);
H
Haojun Liao 已提交
5223

L
Liu Jicong 已提交
5224
_error:
H
Haojun Liao 已提交
5225 5226 5227 5228
  terrno = code;
  return NULL;
}

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

H
Haojun Liao 已提交
5233
  int32_t code = TSDB_CODE_SUCCESS;
5234
  *pTaskInfo = createExecTaskInfo(queryId, taskId, model);
H
Haojun Liao 已提交
5235 5236 5237 5238
  if (*pTaskInfo == NULL) {
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    goto _complete;
  }
H
Haojun Liao 已提交
5239

L
Liu Jicong 已提交
5240 5241
  (*pTaskInfo)->pRoot =
      createOperatorTree(pPlan->pNode, *pTaskInfo, pHandle, queryId, taskId, &(*pTaskInfo)->tableqinfoGroupInfo);
D
dapan1121 已提交
5242
  if (NULL == (*pTaskInfo)->pRoot) {
L
Liu Jicong 已提交
5243
    code = terrno;
D
dapan1121 已提交
5244 5245
    goto _complete;
  }
H
Haojun Liao 已提交
5246

5247
  if ((*pTaskInfo)->pRoot == NULL) {
H
Haojun Liao 已提交
5248
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
H
Haojun Liao 已提交
5249
    goto _complete;
5250 5251
  }

H
Haojun Liao 已提交
5252 5253
  return code;

H
Haojun Liao 已提交
5254
_complete:
wafwerar's avatar
wafwerar 已提交
5255
  taosMemoryFreeClear(*pTaskInfo);
H
Haojun Liao 已提交
5256 5257 5258

  terrno = code;
  return code;
H
Haojun Liao 已提交
5259 5260
}

H
Haojun Liao 已提交
5261
void setResultBufSize(STaskAttr* pQueryAttr, SResultInfo* pResultInfo) {
5262 5263 5264 5265 5266 5267 5268 5269
  const int32_t DEFAULT_RESULT_MSG_SIZE = 1024 * (1024 + 512);

  // the minimum number of rows for projection query
  const int32_t MIN_ROWS_FOR_PRJ_QUERY = 8192;
  const int32_t DEFAULT_MIN_ROWS = 4096;

  const float THRESHOLD_RATIO = 0.85f;

dengyihao's avatar
dengyihao 已提交
5270 5271 5272 5273 5274 5275 5276 5277 5278 5279
  //  if (isProjQuery(pQueryAttr)) {
  //    int32_t numOfRes = DEFAULT_RESULT_MSG_SIZE / pQueryAttr->resultRowSize;
  //    if (numOfRes < MIN_ROWS_FOR_PRJ_QUERY) {
  //      numOfRes = MIN_ROWS_FOR_PRJ_QUERY;
  //    }
  //
  //    pResultInfo->capacity = numOfRes;
  //  } else {  // in case of non-prj query, a smaller output buffer will be used.
  //    pResultInfo->capacity = DEFAULT_MIN_ROWS;
  //  }
5280 5281

  pResultInfo->threshold = (int32_t)(pResultInfo->capacity * THRESHOLD_RATIO);
H
Haojun Liao 已提交
5282
  pResultInfo->totalRows = 0;
5283 5284
}

L
Liu Jicong 已提交
5285
// TODO refactor
5286
void freeColumnFilterInfo(SColumnFilterInfo* pFilter, int32_t numOfFilters) {
L
Liu Jicong 已提交
5287 5288 5289
  if (pFilter == NULL || numOfFilters == 0) {
    return;
  }
5290

L
Liu Jicong 已提交
5291 5292 5293
  for (int32_t i = 0; i < numOfFilters; i++) {
    if (pFilter[i].filterstr && pFilter[i].pz) {
      taosMemoryFree((void*)(pFilter[i].pz));
5294
    }
L
Liu Jicong 已提交
5295
  }
5296

L
Liu Jicong 已提交
5297
  taosMemoryFree(pFilter);
5298 5299 5300 5301
}

static void doDestroyTableQueryInfo(STableGroupInfo* pTableqinfoGroupInfo) {
  if (pTableqinfoGroupInfo->pGroupList != NULL) {
L
Liu Jicong 已提交
5302
    int32_t numOfGroups = (int32_t)taosArrayGetSize(pTableqinfoGroupInfo->pGroupList);
5303
    for (int32_t i = 0; i < numOfGroups; ++i) {
L
Liu Jicong 已提交
5304
      SArray* p = taosArrayGetP(pTableqinfoGroupInfo->pGroupList, i);
5305 5306

      size_t num = taosArrayGetSize(p);
L
Liu Jicong 已提交
5307
      for (int32_t j = 0; j < num; ++j) {
5308 5309 5310 5311 5312 5313 5314 5315 5316 5317 5318 5319 5320 5321 5322 5323
        STableQueryInfo* item = taosArrayGetP(p, j);
        destroyTableQueryInfoImpl(item);
      }

      taosArrayDestroy(p);
    }
  }

  taosArrayDestroy(pTableqinfoGroupInfo->pGroupList);
  taosHashCleanup(pTableqinfoGroupInfo->map);

  pTableqinfoGroupInfo->pGroupList = NULL;
  pTableqinfoGroupInfo->map = NULL;
  pTableqinfoGroupInfo->numOfTables = 0;
}

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

H
Haojun Liao 已提交
5327
  doDestroyTableQueryInfo(&pTaskInfo->tableqinfoGroupInfo);
H
Haojun Liao 已提交
5328
  destroyOperatorInfo(pTaskInfo->pRoot);
L
Liu Jicong 已提交
5329 5330
  //  taosArrayDestroy(pTaskInfo->summary.queryProfEvents);
  //  taosHashCleanup(pTaskInfo->summary.operatorProfResults);
5331

wafwerar's avatar
wafwerar 已提交
5332 5333 5334
  taosMemoryFreeClear(pTaskInfo->sql);
  taosMemoryFreeClear(pTaskInfo->id.str);
  taosMemoryFreeClear(pTaskInfo);
5335 5336 5337 5338 5339 5340 5341 5342 5343 5344 5345 5346
}

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 已提交
5347
      int32_t len = (varDataLen(val) > maxLen) ? maxLen : varDataLen(val);
5348 5349 5350 5351 5352 5353 5354 5355 5356 5357 5358 5359
      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 已提交
5360 5361
  //  size_t s3 = sizeof(STableCheckInfo);  buffer consumption in tsdb
  return (int64_t)(s1 * 1.5 * numOfTables);
5362 5363 5364 5365 5366 5367 5368
}

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 已提交
5369
    while (1) {
5370 5371 5372 5373 5374 5375 5376 5377 5378 5379 5380 5381 5382 5383 5384 5385 5386 5387 5388 5389 5390 5391 5392 5393 5394 5395
      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 已提交
5396

dengyihao's avatar
dengyihao 已提交
5397 5398
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SExplainExecInfo** pRes, int32_t* capacity,
                                   int32_t* resNum) {
D
dapan1121 已提交
5399 5400
  if (*resNum >= *capacity) {
    *capacity += 10;
dengyihao's avatar
dengyihao 已提交
5401

D
dapan1121 已提交
5402 5403
    *pRes = taosMemoryRealloc(*pRes, (*capacity) * sizeof(SExplainExecInfo));
    if (NULL == *pRes) {
D
dapan1121 已提交
5404
      qError("malloc %d failed", (*capacity) * (int32_t)sizeof(SExplainExecInfo));
D
dapan1121 已提交
5405 5406 5407 5408
      return TSDB_CODE_QRY_OUT_OF_MEMORY;
    }
  }

D
dapan1121 已提交
5409 5410 5411 5412
  (*pRes)[*resNum].numOfRows = operatorInfo->resultInfo.totalRows;
  (*pRes)[*resNum].startupCost = operatorInfo->cost.openCost;
  (*pRes)[*resNum].totalCost = operatorInfo->cost.totalCost;

5413 5414
  if (operatorInfo->fpSet.getExplainFn) {
    int32_t code = (*operatorInfo->fpSet.getExplainFn)(operatorInfo, &(*pRes)->verboseInfo);
D
dapan1121 已提交
5415 5416 5417 5418 5419
    if (code) {
      qError("operator getExplainFn failed, error:%s", tstrerror(code));
      return code;
    }
  }
dengyihao's avatar
dengyihao 已提交
5420

D
dapan1121 已提交
5421
  ++(*resNum);
dengyihao's avatar
dengyihao 已提交
5422

D
dapan1121 已提交
5423
  int32_t code = 0;
D
dapan1121 已提交
5424 5425
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pRes, capacity, resNum);
D
dapan1121 已提交
5426 5427 5428 5429 5430 5431 5432
    if (code) {
      taosMemoryFreeClear(*pRes);
      return TSDB_CODE_QRY_OUT_OF_MEMORY;
    }
  }

  return TSDB_CODE_SUCCESS;
D
dapan1121 已提交
5433 5434
}

5435
static SSDataBlock* doMergeJoin(struct SOperatorInfo* pOperator) {
5436 5437 5438 5439 5440 5441 5442 5443 5444 5445 5446 5447
  SJoinOperatorInfo* pJoinInfo = pOperator->info;

  SSDataBlock* pRes = pJoinInfo->pRes;
  blockDataCleanup(pRes);
  blockDataEnsureCapacity(pRes, 4096);

  int32_t nrows = 0;

  while (1) {
    if (pJoinInfo->pLeft == NULL || pJoinInfo->leftPos >= pJoinInfo->pLeft->info.rows) {
      SOperatorInfo* ds1 = pOperator->pDownstream[0];
      publishOperatorProfEvent(ds1, QUERY_PROF_BEFORE_OPERATOR_EXEC);
5448
      pJoinInfo->pLeft = ds1->fpSet.getNextFn(ds1);
5449 5450 5451 5452 5453 5454 5455 5456 5457 5458 5459 5460
      publishOperatorProfEvent(ds1, QUERY_PROF_AFTER_OPERATOR_EXEC);

      pJoinInfo->leftPos = 0;
      if (pJoinInfo->pLeft == NULL) {
        setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
        break;
      }
    }

    if (pJoinInfo->pRight == NULL || pJoinInfo->rightPos >= pJoinInfo->pRight->info.rows) {
      SOperatorInfo* ds2 = pOperator->pDownstream[1];
      publishOperatorProfEvent(ds2, QUERY_PROF_BEFORE_OPERATOR_EXEC);
5461
      pJoinInfo->pRight = ds2->fpSet.getNextFn(ds2);
5462 5463 5464 5465 5466 5467 5468 5469 5470 5471
      publishOperatorProfEvent(ds2, QUERY_PROF_AFTER_OPERATOR_EXEC);

      pJoinInfo->rightPos = 0;
      if (pJoinInfo->pRight == NULL) {
        setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
        break;
      }
    }

    SColumnInfoData* pLeftCol = taosArrayGet(pJoinInfo->pLeft->pDataBlock, pJoinInfo->leftCol.slotId);
dengyihao's avatar
dengyihao 已提交
5472
    char*            pLeftVal = colDataGetData(pLeftCol, pJoinInfo->leftPos);
5473 5474

    SColumnInfoData* pRightCol = taosArrayGet(pJoinInfo->pRight->pDataBlock, pJoinInfo->rightCol.slotId);
dengyihao's avatar
dengyihao 已提交
5475
    char*            pRightVal = colDataGetData(pRightCol, pJoinInfo->rightPos);
5476 5477

    // only the timestamp match support for ordinary table
dengyihao's avatar
dengyihao 已提交
5478 5479
    ASSERT(pLeftCol->info.type == TSDB_DATA_TYPE_TIMESTAMP);
    if (*(int64_t*)pLeftVal == *(int64_t*)pRightVal) {
5480
      for (int32_t i = 0; i < pOperator->numOfExprs; ++i) {
dengyihao's avatar
dengyihao 已提交
5481
        SColumnInfoData* pDst = taosArrayGet(pRes->pDataBlock, i);
5482

dengyihao's avatar
dengyihao 已提交
5483
        SExprInfo* pExprInfo = &pOperator->pExpr[i];
5484

dengyihao's avatar
dengyihao 已提交
5485 5486
        int32_t blockId = pExprInfo->base.pParam[0].pCol->dataBlockId;
        int32_t slotId = pExprInfo->base.pParam[0].pCol->slotId;
5487

dengyihao's avatar
dengyihao 已提交
5488 5489 5490 5491 5492 5493
        SColumnInfoData* pSrc = NULL;
        if (pJoinInfo->pLeft->info.blockId == blockId) {
          pSrc = taosArrayGet(pJoinInfo->pLeft->pDataBlock, slotId);
        } else {
          pSrc = taosArrayGet(pJoinInfo->pRight->pDataBlock, slotId);
        }
5494

dengyihao's avatar
dengyihao 已提交
5495 5496 5497 5498 5499
        if (colDataIsNull_s(pSrc, pJoinInfo->leftPos)) {
          colDataAppendNULL(pDst, nrows);
        } else {
          char* p = colDataGetData(pSrc, pJoinInfo->leftPos);
          colDataAppend(pDst, nrows, p, false);
5500
        }
dengyihao's avatar
dengyihao 已提交
5501
      }
5502

dengyihao's avatar
dengyihao 已提交
5503 5504
      pJoinInfo->leftPos += 1;
      pJoinInfo->rightPos += 1;
5505

dengyihao's avatar
dengyihao 已提交
5506 5507 5508
      nrows += 1;
    } else if (*(int64_t*)pLeftVal < *(int64_t*)pRightVal) {
      pJoinInfo->leftPos += 1;
D
dapan1121 已提交
5509

dengyihao's avatar
dengyihao 已提交
5510 5511
      if (pJoinInfo->leftPos >= pJoinInfo->pLeft->info.rows) {
        continue;
5512
      }
dengyihao's avatar
dengyihao 已提交
5513 5514 5515 5516
    } else if (*(int64_t*)pLeftVal > *(int64_t*)pRightVal) {
      pJoinInfo->rightPos += 1;
      if (pJoinInfo->rightPos >= pJoinInfo->pRight->info.rows) {
        continue;
5517
      }
dengyihao's avatar
dengyihao 已提交
5518
    }
5519 5520 5521 5522 5523 5524 5525 5526 5527 5528 5529

    // the pDataBlock are always the same one, no need to call this again
    pRes->info.rows = nrows;
    if (pRes->info.rows >= pOperator->resultInfo.threshold) {
      break;
    }
  }

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

H
Haojun Liao 已提交
5530
SOperatorInfo* createMergeJoinOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream, SExprInfo* pExprInfo,
X
Xiaoyu Wang 已提交
5531 5532
                                           int32_t numOfCols, SSDataBlock* pResBlock, SNode* pOnCondition,
                                           SExecTaskInfo* pTaskInfo) {
5533
  SJoinOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SJoinOperatorInfo));
dengyihao's avatar
dengyihao 已提交
5534
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
5535 5536 5537 5538
  if (pOperator == NULL || pInfo == NULL) {
    goto _error;
  }

5539
  initResultSizeInfo(pOperator, 4096);
5540

X
Xiaoyu Wang 已提交
5541 5542
  pInfo->pRes = pResBlock;
  pOperator->name = "MergeJoinOperator";
5543
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_JOIN;
X
Xiaoyu Wang 已提交
5544 5545 5546
  pOperator->blocking = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExprInfo;
5547
  pOperator->numOfExprs = numOfCols;
X
Xiaoyu Wang 已提交
5548
  pOperator->info = pInfo;
dengyihao's avatar
dengyihao 已提交
5549
  pOperator->pTaskInfo = pTaskInfo;
5550

5551 5552 5553 5554
  SOperatorNode* pNode = (SOperatorNode*)pOnCondition;
  setJoinColumnInfo(&pInfo->leftCol, (SColumnNode*)pNode->pLeft);
  setJoinColumnInfo(&pInfo->rightCol, (SColumnNode*)pNode->pRight);

L
Liu Jicong 已提交
5555
  pOperator->fpSet =
H
Haojun Liao 已提交
5556
      createOperatorFpSet(operatorDummyOpenFn, doMergeJoin, NULL, NULL, destroyMergeJoinOperator, NULL, NULL, NULL);
5557
  int32_t code = appendDownstream(pOperator, pDownstream, numOfDownstream);
5558 5559 5560 5561
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

5562 5563
  return pOperator;

dengyihao's avatar
dengyihao 已提交
5564
_error:
5565 5566 5567 5568
  taosMemoryFree(pInfo);
  taosMemoryFree(pOperator);
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
dengyihao's avatar
dengyihao 已提交
5569
}
5570 5571

void setJoinColumnInfo(SColumnInfo* pColumn, const SColumnNode* pColumnNode) {
X
Xiaoyu Wang 已提交
5572 5573 5574
  pColumn->slotId = pColumnNode->slotId;
  pColumn->type = pColumnNode->node.resType.type;
  pColumn->bytes = pColumnNode->node.resType.bytes;
5575
  pColumn->precision = pColumnNode->node.resType.precision;
X
Xiaoyu Wang 已提交
5576
  pColumn->scale = pColumnNode->node.resType.scale;
5577
}