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

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

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

32
#include "../../function/inc/taggfunction.h"
33 34
#include "executorimpl.h"
#include "function.h"
35
#include "query.h"
36 37
#include "tcompare.h"
#include "tcompression.h"
H
Haojun Liao 已提交
38
#include "thash.h"
39
#include "tsdb.h"
40
#include "ttypes.h"
L
Liu Jicong 已提交
41
#include "vnode.h"
42

H
Haojun Liao 已提交
43
#define IS_MAIN_SCAN(runtime)          ((runtime)->scanFlag == MAIN_SCAN)
44 45
#define IS_REVERSE_SCAN(runtime)       ((runtime)->scanFlag == REVERSE_SCAN)
#define IS_REPEAT_SCAN(runtime)        ((runtime)->scanFlag == REPEAT_SCAN)
L
Liu Jicong 已提交
46
#define SET_MAIN_SCAN_FLAG(runtime)    ((runtime)->scanFlag = MAIN_SCAN)
47 48 49
#define SET_REVERSE_SCAN_FLAG(runtime) ((runtime)->scanFlag = REVERSE_SCAN)

#define TSWINDOW_IS_EQUAL(t1, t2) (((t1).skey == (t2).skey) && ((t1).ekey == (t2).ekey))
L
Liu Jicong 已提交
50
#define SWITCH_ORDER(n)           (((n) = ((n) == TSDB_ORDER_ASC) ? TSDB_ORDER_DESC : TSDB_ORDER_ASC))
51

L
Liu Jicong 已提交
52 53
#define SDATA_BLOCK_INITIALIZER \
  (SDataBlockInfo) { {0}, 0 }
54 55 56

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

L
Liu Jicong 已提交
57
#define MULTI_KEY_DELIM "-"
58 59

enum {
L
Liu Jicong 已提交
60 61
  TS_JOIN_TS_EQUAL = 0,
  TS_JOIN_TS_NOT_EQUALS = 1,
62 63 64 65 66
  TS_JOIN_TAG_NOT_EQUALS = 2,
};

typedef enum SResultTsInterpType {
  RESULT_ROW_START_INTERP = 1,
L
Liu Jicong 已提交
67
  RESULT_ROW_END_INTERP = 2,
68 69
} SResultTsInterpType;

H
Haojun Liao 已提交
70 71 72
typedef struct SColMatchInfo {
  int32_t colId;
  int32_t targetSlotId;
H
Haojun Liao 已提交
73
  bool    output;
H
Haojun Liao 已提交
74 75
} SColMatchInfo;

76 77
#if 0
static UNUSED_FUNC void *u_malloc (size_t __size) {
wafwerar's avatar
wafwerar 已提交
78
  uint32_t v = taosRand();
79 80 81 82

  if (v % 1000 <= 0) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
83
    return taosMemoryMalloc(__size);
84 85 86 87
  }
}

static UNUSED_FUNC void* u_calloc(size_t num, size_t __size) {
wafwerar's avatar
wafwerar 已提交
88
  uint32_t v = taosRand();
89 90 91
  if (v % 1000 <= 0) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
92
    return taosMemoryCalloc(num, __size);
93 94 95 96
  }
}

static UNUSED_FUNC void* u_realloc(void* p, size_t __size) {
wafwerar's avatar
wafwerar 已提交
97
  uint32_t v = taosRand();
98 99 100
  if (v % 5 <= 1) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
101
    return taosMemoryRealloc(p, __size);
102 103 104 105 106 107 108 109 110 111 112 113
  }
}

#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 已提交
114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
#define TSKEY_MAX_ADD(a, b)                      \
  do {                                           \
    if (a < 0) {                                 \
      a = a + b;                                 \
      break;                                     \
    }                                            \
    if (sizeof(a) == sizeof(int32_t)) {          \
      if ((b) > 0 && ((b) >= INT32_MAX - (a))) { \
        a = INT32_MAX;                           \
      } else {                                   \
        a = a + b;                               \
      }                                          \
    } else {                                     \
      if ((b) > 0 && ((b) >= INT64_MAX - (a))) { \
        a = INT64_MAX;                           \
      } else {                                   \
        a = a + b;                               \
      }                                          \
    }                                            \
  } while (0)

#define TSKEY_MIN_SUB(a, b)                      \
  do {                                           \
    if (a >= 0) {                                \
      a = a + b;                                 \
      break;                                     \
    }                                            \
    if (sizeof(a) == sizeof(int32_t)) {          \
      if ((b) < 0 && ((b) <= INT32_MIN - (a))) { \
        a = INT32_MIN;                           \
      } else {                                   \
        a = a + b;                               \
      }                                          \
    } else {                                     \
      if ((b) < 0 && ((b) <= INT64_MIN - (a))) { \
        a = INT64_MIN;                           \
      } else {                                   \
        a = a + b;                               \
      }                                          \
    }                                            \
  } while (0)

int32_t getMaximumIdleDurationSec() { return tsShellActivityTimer * 2; }

static int32_t getExprFunctionId(SExprInfo* pExprInfo) {
159
  assert(pExprInfo != NULL && pExprInfo->pExpr != NULL && pExprInfo->pExpr->nodeType == TEXPR_UNARYEXPR_NODE);
160
  return 0;
161 162
}

H
Haojun Liao 已提交
163 164 165 166 167
static void getNextTimeWindow(SInterval* pInterval, int32_t precision, int32_t order, STimeWindow* tw) {
  int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
  if (pInterval->intervalUnit != 'n' && pInterval->intervalUnit != 'y') {
    tw->skey += pInterval->sliding * factor;
    tw->ekey = tw->skey + pInterval->interval - 1;
168 169 170
    return;
  }

H
Haojun Liao 已提交
171
  int64_t key = tw->skey, interval = pInterval->interval;
L
Liu Jicong 已提交
172
  // convert key to second
H
Haojun Liao 已提交
173
  key = convertTimePrecision(key, precision, TSDB_TIME_PRECISION_MILLI) / 1000;
174

H
Haojun Liao 已提交
175
  if (pInterval->intervalUnit == 'y') {
176 177 178 179
    interval *= 12;
  }

  struct tm tm;
L
Liu Jicong 已提交
180
  time_t    t = (time_t)key;
wafwerar's avatar
wafwerar 已提交
181
  taosLocalTime(&t, &tm);
182 183 184 185

  int mon = (int)(tm.tm_year * 12 + tm.tm_mon + interval * factor);
  tm.tm_year = mon / 12;
  tm.tm_mon = mon % 12;
H
Haojun Liao 已提交
186
  tw->skey = convertTimePrecision((int64_t)mktime(&tm) * 1000L, TSDB_TIME_PRECISION_MILLI, precision);
187 188 189 190

  mon = (int)(mon + interval);
  tm.tm_year = mon / 12;
  tm.tm_mon = mon % 12;
H
Haojun Liao 已提交
191
  tw->ekey = convertTimePrecision((int64_t)mktime(&tm) * 1000L, TSDB_TIME_PRECISION_MILLI, precision);
192 193 194 195 196

  tw->ekey -= 1;
}

static void doSetTagValueToResultBuf(char* output, const char* val, int16_t type, int16_t bytes);
H
Haojun Liao 已提交
197
static void setResultOutputBuf(STaskRuntimeEnv* pRuntimeEnv, SResultRow* pResult, SqlFunctionCtx* pCtx,
198 199
                               int32_t numOfCols, int32_t* rowCellInfoOffset);

L
Liu Jicong 已提交
200 201 202
void        setResultRowOutputBufInitCtx(STaskRuntimeEnv* pRuntimeEnv, SResultRow* pResult, SqlFunctionCtx* pCtx,
                                         int32_t numOfOutput, int32_t* rowCellInfoOffset);
static bool functionNeedToExecute(SqlFunctionCtx* pCtx);
203

L
Liu Jicong 已提交
204
static void setBlockStatisInfo(SqlFunctionCtx* pCtx, SSDataBlock* pSDataBlock, SColumn* pColumn);
205

L
Liu Jicong 已提交
206 207
static void destroyTableQueryInfoImpl(STableQueryInfo* pTableQueryInfo);
static bool hasMainOutput(STaskAttr* pQueryAttr);
208 209 210

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

L
Liu Jicong 已提交
211 212 213 214
static int32_t setTimestampListJoinInfo(STaskRuntimeEnv* pRuntimeEnv, SVariant* pTag, STableQueryInfo* pTableQueryInfo);
static void    releaseQueryBuf(size_t numOfTables);
static int32_t binarySearchForKey(char* pValue, int num, TSKEY key, int order);
// static STsdbQueryCond createTsdbQueryCond(STaskAttr* pQueryAttr, STimeWindow* win);
215 216 217 218
static STableIdInfo createTableIdInfo(STableQueryInfo* pTableQueryInfo);

static void setTableScanFilterOperatorInfo(STableScanInfo* pTableScanInfo, SOperatorInfo* pDownstream);

H
Haojun Liao 已提交
219
static int32_t getNumOfScanTimes(STaskAttr* pQueryAttr);
220 221 222 223 224 225 226 227 228 229

static void destroyBasicOperatorInfo(void* param, int32_t numOfOutput);
static void destroySFillOperatorInfo(void* param, int32_t numOfOutput);
static void destroyGroupbyOperatorInfo(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 destroySWindowOperatorInfo(void* param, int32_t numOfOutput);
static void destroyStateWindowOperatorInfo(void* param, int32_t numOfOutput);
static void destroyAggOperatorInfo(void* param, int32_t numOfOutput);
X
Xiaoyu Wang 已提交
230

H
Haojun Liao 已提交
231
static void destroyIntervalOperatorInfo(void* param, int32_t numOfOutput);
H
Haojun Liao 已提交
232 233 234
static void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput);
static void destroyConditionOperatorInfo(void* param, int32_t numOfOutput);

235
static void destroyOperatorInfo(SOperatorInfo* pOperator);
236
static void destroySysTableScannerOperatorInfo(void* param, int32_t numOfOutput);
237 238 239

static void doSetOperatorCompleted(SOperatorInfo* pOperator) {
  pOperator->status = OP_EXEC_DONE;
H
Haojun Liao 已提交
240
  if (pOperator->pTaskInfo != NULL) {
241
    setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
242 243
  }
}
244 245 246
#define OPTR_IS_OPENED(_optr)  (((_optr)->status & OP_OPENED) == OP_OPENED)
#define OPTR_SET_OPENED(_optr) ((_optr)->status |= OP_OPENED)

L
Liu Jicong 已提交
247
static int32_t operatorDummyOpenFn(SOperatorInfo* pOperator) {
248
  OPTR_SET_OPENED(pOperator);
H
Haojun Liao 已提交
249
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
250 251
}

H
Haojun Liao 已提交
252 253
static void operatorDummyCloseFn(void* param, int32_t numOfCols) {}

L
Liu Jicong 已提交
254 255
static int32_t doCopyToSDataBlock(SDiskbasedBuf* pBuf, SGroupResInfo* pGroupResInfo, int32_t orderType,
                                  SSDataBlock* pBlock, int32_t rowCapacity, int32_t* rowCellOffset);
256

L
Liu Jicong 已提交
257 258 259 260 261 262 263
static int32_t getGroupbyColumnIndex(SGroupbyExpr* pGroupbyExpr, SSDataBlock* pDataBlock);
static int32_t setGroupResultOutputBuf_rv(SOptrBasicInfo* binfo, int32_t numOfCols, char* pData, int16_t type,
                                          int16_t bytes, int32_t groupId, SDiskbasedBuf* pBuf, SExecTaskInfo* pTaskInfo,
                                          SAggSupporter* pAggSup);
static void    initCtxOutputBuffer(SqlFunctionCtx* pCtx, int32_t size);
static void    getAlignQueryTimeWindow(SInterval* pInterval, int32_t precision, int64_t key, int64_t keyFirst,
                                       int64_t keyLast, STimeWindow* win);
H
Haojun Liao 已提交
264

H
Haojun Liao 已提交
265
static void setResultBufSize(STaskAttr* pQueryAttr, SResultInfo* pResultInfo);
H
Haojun Liao 已提交
266
static void setCtxTagForJoin(STaskRuntimeEnv* pRuntimeEnv, SqlFunctionCtx* pCtx, SExprInfo* pExprInfo, void* pTable);
L
Liu Jicong 已提交
267 268 269 270 271 272
static void setParamForStableStddev(STaskRuntimeEnv* pRuntimeEnv, SqlFunctionCtx* pCtx, int32_t numOfOutput,
                                    SExprInfo* pExpr);
static void setParamForStableStddevByColData(STaskRuntimeEnv* pRuntimeEnv, SqlFunctionCtx* pCtx, int32_t numOfOutput,
                                             SExprInfo* pExpr, char* val, int16_t bytes);
static void doSetTableGroupOutputBuf(SAggOperatorInfo* pAggInfo, int32_t numOfOutput, int32_t tableGroupId,
                                     SExecTaskInfo* pTaskInfo);
273

H
Haojun Liao 已提交
274
SArray* getOrderCheckColumns(STaskAttr* pQuery);
275 276

typedef struct SRowCompSupporter {
L
Liu Jicong 已提交
277 278 279
  STaskRuntimeEnv* pRuntimeEnv;
  int16_t          dataOffset;
  __compar_fn_t    comFunc;
280 281
} SRowCompSupporter;

L
Liu Jicong 已提交
282 283 284
static int compareRowData(const void* a, const void* b, const void* userData) {
  const SResultRow* pRow1 = (const SResultRow*)a;
  const SResultRow* pRow2 = (const SResultRow*)b;
285

L
Liu Jicong 已提交
286 287
  SRowCompSupporter* supporter = (SRowCompSupporter*)userData;
  STaskRuntimeEnv*   pRuntimeEnv = supporter->pRuntimeEnv;
288

L
Liu Jicong 已提交
289 290
  SFilePage* page1 = getBufPage(pRuntimeEnv->pResultBuf, pRow1->pageId);
  SFilePage* page2 = getBufPage(pRuntimeEnv->pResultBuf, pRow2->pageId);
291 292

  int16_t offset = supporter->dataOffset;
L
Liu Jicong 已提交
293 294
  char*   in1 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page1, pRow1->offset, offset);
  char*   in2 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page2, pRow2->offset, offset);
295 296 297 298

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

L
Liu Jicong 已提交
299 300 301 302
static void sortGroupResByOrderList(SGroupResInfo* pGroupResInfo, STaskRuntimeEnv* pRuntimeEnv,
                                    SSDataBlock* pDataBlock) {
  SArray* columnOrderList = getOrderCheckColumns(pRuntimeEnv->pQueryAttr);
  size_t  size = taosArrayGetSize(columnOrderList);
303 304 305 306 307 308
  taosArrayDestroy(columnOrderList);

  if (size <= 0) {
    return;
  }

H
Haojun Liao 已提交
309
  int32_t orderId = pRuntimeEnv->pQueryAttr->order.col.colId;
310 311 312 313
  if (orderId <= 0) {
    return;
  }

L
Liu Jicong 已提交
314
  bool    found = false;
315 316 317
  int16_t dataOffset = 0;

  for (int32_t j = 0; j < pDataBlock->info.numOfCols; ++j) {
L
Liu Jicong 已提交
318
    SColumnInfoData* pColInfoData = (SColumnInfoData*)taosArrayGet(pDataBlock->pDataBlock, j);
319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
    if (orderId == j) {
      found = true;
      break;
    }

    dataOffset += pColInfoData->info.bytes;
  }

  if (found == false) {
    return;
  }

  int16_t type = pRuntimeEnv->pQueryAttr->pExpr1[orderId].base.resSchema.type;

  SRowCompSupporter support = {.pRuntimeEnv = pRuntimeEnv, .dataOffset = dataOffset, .comFunc = getComparFunc(type, 0)};
  taosArraySortPWithExt(pGroupResInfo->pRows, compareRowData, &support);
}

L
Liu Jicong 已提交
337
// setup the output buffer for each operator
H
Haojun Liao 已提交
338 339
SSDataBlock* createOutputBuf_rv1(SDataBlockDescNode* pNode) {
  int32_t numOfCols = LIST_LENGTH(pNode->pSlots);
H
Haojun Liao 已提交
340

wafwerar's avatar
wafwerar 已提交
341
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
H
Haojun Liao 已提交
342 343 344
  pBlock->info.numOfCols = numOfCols;
  pBlock->pDataBlock = taosArrayInit(numOfCols, sizeof(SColumnInfoData));

H
Haojun Liao 已提交
345
  pBlock->info.blockId = pNode->dataBlockId;
X
Xiaoyu Wang 已提交
346
  pBlock->info.rowSize = pNode->totalRowSize;   // todo ??
H
Haojun Liao 已提交
347

L
Liu Jicong 已提交
348
  for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
349
    SColumnInfoData idata = {{0}};
L
Liu Jicong 已提交
350
    SSlotDescNode*  pDescNode = nodesListGetNode(pNode->pSlots, i);
H
Haojun Liao 已提交
351 352
    if (!pDescNode->output) {
      continue;
H
Haojun Liao 已提交
353
    }
H
Haojun Liao 已提交
354

L
Liu Jicong 已提交
355 356 357
    idata.info.type = pDescNode->dataType.type;
    idata.info.bytes = pDescNode->dataType.bytes;
    idata.info.scale = pDescNode->dataType.scale;
H
Haojun Liao 已提交
358
    idata.info.slotId = pDescNode->slotId;
H
Haojun Liao 已提交
359 360 361
    idata.info.precision = pDescNode->dataType.precision;

    taosArrayPush(pBlock->pDataBlock, &idata);
H
Haojun Liao 已提交
362
  }
H
Haojun Liao 已提交
363 364

  return pBlock;
H
Haojun Liao 已提交
365 366
}

L
Liu Jicong 已提交
367
static bool isSelectivityWithTagsQuery(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
368
  return true;
L
Liu Jicong 已提交
369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387
  //  bool    hasTags = false;
  //  int32_t numOfSelectivity = 0;
  //
  //  for (int32_t i = 0; i < numOfOutput; ++i) {
  //    int32_t functId = pCtx[i].functionId;
  //    if (functId == FUNCTION_TAG_DUMMY || functId == FUNCTION_TS_DUMMY) {
  //      hasTags = true;
  //      continue;
  //    }
  //
  //    if ((aAggs[functId].status & FUNCSTATE_SELECTIVITY) != 0) {
  //      numOfSelectivity++;
  //    }
  //  }
  //
  //  return (numOfSelectivity > 0 && hasTags);
}

static bool isProjQuery(STaskAttr* pQueryAttr) {
388 389 390 391 392 393 394 395 396 397
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
    int32_t functId = getExprFunctionId(&pQueryAttr->pExpr1[i]);
    if (functId != FUNCTION_PRJ && functId != FUNCTION_TAGPRJ) {
      return false;
    }
  }

  return true;
}

L
Liu Jicong 已提交
398 399 400
static bool hasNull(SColumn* pColumn, SColumnDataAgg* pStatis) {
  if (TSDB_COL_IS_TAG(pColumn->flag) || TSDB_COL_IS_UD_COL(pColumn->flag) ||
      pColumn->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
401 402 403 404 405 406 407 408 409 410
    return false;
  }

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

  return true;
}

H
Haojun Liao 已提交
411
static void prepareResultListBuffer(SResultRowInfo* pResultRowInfo, jmp_buf env) {
412 413
  int64_t newCapacity = 0;

414 415 416 417 418 419 420 421 422 423 424
  // 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);
  }

H
Haojun Liao 已提交
425 426 427 428
  if (newCapacity == pResultRowInfo->capacity) {
    newCapacity += 4;
  }

L
Liu Jicong 已提交
429
  char* t = taosMemoryRealloc(pResultRowInfo->pResult, (size_t)(newCapacity * POINTER_BYTES));
430
  if (t == NULL) {
H
Haojun Liao 已提交
431
    longjmp(env, TSDB_CODE_QRY_OUT_OF_MEMORY);
432 433
  }

wafwerar's avatar
wafwerar 已提交
434
  pResultRowInfo->pPosition = taosMemoryRealloc(pResultRowInfo->pPosition, newCapacity * sizeof(SResultRowPosition));
L
Liu Jicong 已提交
435
  pResultRowInfo->pResult = (SResultRow**)t;
436 437 438

  int32_t inc = (int32_t)newCapacity - pResultRowInfo->capacity;
  memset(&pResultRowInfo->pResult[pResultRowInfo->capacity], 0, POINTER_BYTES * inc);
439
  memset(&pResultRowInfo->pPosition[pResultRowInfo->capacity], 0, sizeof(SResultRowPosition));
440 441 442 443

  pResultRowInfo->capacity = (int32_t)newCapacity;
}

L
Liu Jicong 已提交
444 445
static bool chkResultRowFromKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, char* pData,
                                int16_t bytes, bool masterscan, uint64_t uid) {
446 447 448
  bool existed = false;
  SET_RES_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid);

L
Liu Jicong 已提交
449 450
  SResultRow** p1 =
      (SResultRow**)taosHashGet(pRuntimeEnv->pResultRowHashTable, pRuntimeEnv->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
451 452 453 454 455 456 457 458 459 460 461 462 463 464 465

  // 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;
        assert(pResultRowInfo->curPos == -1);
      } else if (pResultRowInfo->size == 1) {
        existed = (pResultRowInfo->pResult[0] == (*p1));
      } else {  // check if current pResultRowInfo contains the existed pResultRow
        SET_RES_EXT_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid, pResultRowInfo);
L
Liu Jicong 已提交
466 467
        int64_t* index =
            taosHashGet(pRuntimeEnv->pResultRowListSet, pRuntimeEnv->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes));
468 469 470 471 472 473 474 475 476 477 478 479 480 481
        if (index != NULL) {
          existed = true;
        } else {
          existed = false;
        }
      }
    }

    return existed;
  }

  return p1 != NULL;
}

H
Haojun Liao 已提交
482
static SResultRow* doSetResultOutBufByKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, int64_t tid,
483 484 485 486
                                          char* pData, int16_t bytes, bool masterscan, uint64_t tableGroupId) {
  bool existed = false;
  SET_RES_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, tableGroupId);

L
Liu Jicong 已提交
487 488
  SResultRow** p1 =
      (SResultRow**)taosHashGet(pRuntimeEnv->pResultRowHashTable, pRuntimeEnv->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
489 490 491 492

  // 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.
L
Liu Jicong 已提交
493
      return (p1 != NULL) ? *p1 : NULL;
494 495 496 497 498 499 500 501 502 503 504
    }

    if (p1 != NULL) {
      if (pResultRowInfo->size == 0) {
        existed = false;
        assert(pResultRowInfo->curPos == -1);
      } else if (pResultRowInfo->size == 1) {
        existed = (pResultRowInfo->pResult[0] == (*p1));
        pResultRowInfo->curPos = 0;
      } else {  // check if current pResultRowInfo contains the existed pResultRow
        SET_RES_EXT_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, tid, pResultRowInfo);
L
Liu Jicong 已提交
505 506
        int64_t* index =
            taosHashGet(pRuntimeEnv->pResultRowListSet, pRuntimeEnv->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes));
507
        if (index != NULL) {
L
Liu Jicong 已提交
508
          pResultRowInfo->curPos = (int32_t)*index;
509 510 511 512 513 514 515 516 517 518 519 520 521 522
          existed = true;
        } else {
          existed = false;
        }
      }
    }
  } else {
    // In case of group by column query, the required SResultRow object must be existed in the pResultRowInfo object.
    if (p1 != NULL) {
      return *p1;
    }
  }

  if (!existed) {
L
Liu Jicong 已提交
523
    //    prepareResultListBuffer(pResultRowInfo, pRuntimeEnv);
524

L
Liu Jicong 已提交
525
    SResultRow* pResult = NULL;
526 527 528 529 530 531 532 533
    if (p1 == NULL) {
      pResult = getNewResultRow(pRuntimeEnv->pool);
      int32_t ret = initResultRow(pResult);
      if (ret != TSDB_CODE_SUCCESS) {
        longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
      }

      // add a new result set for a new group
L
Liu Jicong 已提交
534 535
      taosHashPut(pRuntimeEnv->pResultRowHashTable, pRuntimeEnv->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes), &pResult,
                  POINTER_BYTES);
536 537 538 539 540 541 542 543 544 545 546
      SResultRowCell cell = {.groupId = tableGroupId, .pRow = pResult};
      taosArrayPush(pRuntimeEnv->pResultRowArrayList, &cell);
    } else {
      pResult = *p1;
    }

    pResultRowInfo->curPos = pResultRowInfo->size;
    pResultRowInfo->pResult[pResultRowInfo->size++] = pResult;

    int64_t index = pResultRowInfo->curPos;
    SET_RES_EXT_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, tid, pResultRowInfo);
L
Liu Jicong 已提交
547 548
    taosHashPut(pRuntimeEnv->pResultRowListSet, pRuntimeEnv->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes), &index,
                POINTER_BYTES);
549 550 551 552 553 554 555 556 557 558
  }

  // too many time window in query
  if (pResultRowInfo->size > MAX_INTERVAL_TIME_WINDOW) {
    longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_TOO_MANY_TIMEWINDOW);
  }

  return pResultRowInfo->pResult[pResultRowInfo->curPos];
}

559
SResultRow* getNewResultRow_rv(SDiskbasedBuf* pResultBuf, int64_t tableGroupId, int32_t interBufSize) {
L
Liu Jicong 已提交
560
  SFilePage* pData = NULL;
561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598

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

    if (pData->num + interBufSize + sizeof(SResultRow) > getBufPageSize(pResultBuf)) {
      // 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;

  pData->num += interBufSize + sizeof(SResultRow);

  return pResultRow;
}

L
Liu Jicong 已提交
599 600 601
static SResultRow* doSetResultOutBufByKey_rv(SDiskbasedBuf* pResultBuf, SResultRowInfo* pResultRowInfo, int64_t tid,
                                             char* pData, int16_t bytes, bool masterscan, uint64_t tableGroupId,
                                             SExecTaskInfo* pTaskInfo, bool isIntervalQuery, SAggSupporter* pSup) {
H
Haojun Liao 已提交
602
  bool existed = false;
603
  SET_RES_WINDOW_KEY(pSup->keyBuf, pData, bytes, tableGroupId);
H
Haojun Liao 已提交
604

L
Liu Jicong 已提交
605
  SResultRow** p1 = (SResultRow**)taosHashGet(pSup->pResultRowHashTable, pSup->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
H
Haojun Liao 已提交
606 607 608 609

  // in case of repeat scan/reverse scan, no new time window added.
  if (isIntervalQuery) {
    if (!masterscan) {  // the *p1 may be NULL in case of sliding+offset exists.
L
Liu Jicong 已提交
610
      return (p1 != NULL) ? *p1 : NULL;
H
Haojun Liao 已提交
611 612 613 614 615 616 617 618 619 620
    }

    if (p1 != NULL) {
      if (pResultRowInfo->size == 0) {
        existed = false;
        assert(pResultRowInfo->curPos == -1);
      } else if (pResultRowInfo->size == 1) {
        existed = (pResultRowInfo->pResult[0] == (*p1));
        pResultRowInfo->curPos = 0;
      } else {  // check if current pResultRowInfo contains the existed pResultRow
621 622
        SET_RES_EXT_WINDOW_KEY(pSup->keyBuf, pData, bytes, tid, pResultRowInfo);
        int64_t* index = taosHashGet(pSup->pResultRowListSet, pSup->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes));
H
Haojun Liao 已提交
623
        if (index != NULL) {
L
Liu Jicong 已提交
624
          pResultRowInfo->curPos = (int32_t)*index;
H
Haojun Liao 已提交
625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640
          existed = true;
        } else {
          existed = false;
        }
      }
    }
  } else {
    // In case of group by column query, the required SResultRow object must be existed in the pResultRowInfo object.
    if (p1 != NULL) {
      return *p1;
    }
  }

  if (!existed) {
    prepareResultListBuffer(pResultRowInfo, pTaskInfo->env);

L
Liu Jicong 已提交
641
    SResultRow* pResult = NULL;
H
Haojun Liao 已提交
642
    if (p1 == NULL) {
643
      pResult = getNewResultRow_rv(pResultBuf, tableGroupId, pSup->resultRowSize);
H
Haojun Liao 已提交
644 645 646 647 648 649
      int32_t ret = initResultRow(pResult);
      if (ret != TSDB_CODE_SUCCESS) {
        longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
      }

      // add a new result set for a new group
650
      taosHashPut(pSup->pResultRowHashTable, pSup->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes), &pResult, POINTER_BYTES);
H
Haojun Liao 已提交
651
      SResultRowCell cell = {.groupId = tableGroupId, .pRow = pResult};
652
      taosArrayPush(pSup->pResultRowArrayList, &cell);
H
Haojun Liao 已提交
653 654 655 656 657
    } else {
      pResult = *p1;
    }

    pResultRowInfo->curPos = pResultRowInfo->size;
L
Liu Jicong 已提交
658 659
    pResultRowInfo->pPosition[pResultRowInfo->size] =
        (SResultRowPosition){.pageId = pResult->pageId, .offset = pResult->offset};
H
Haojun Liao 已提交
660 661 662
    pResultRowInfo->pResult[pResultRowInfo->size++] = pResult;

    int64_t index = pResultRowInfo->curPos;
663 664
    SET_RES_EXT_WINDOW_KEY(pSup->keyBuf, pData, bytes, tid, pResultRowInfo);
    taosHashPut(pSup->pResultRowListSet, pSup->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes), &index, POINTER_BYTES);
H
Haojun Liao 已提交
665 666 667 668 669 670 671 672 673 674
  }

  // too many time window in query
  if (pResultRowInfo->size > MAX_INTERVAL_TIME_WINDOW) {
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_TOO_MANY_TIMEWINDOW);
  }

  return pResultRowInfo->pResult[pResultRowInfo->curPos];
}

L
Liu Jicong 已提交
675 676
static void getInitialStartTimeWindow(SInterval* pInterval, int32_t precision, TSKEY ts, STimeWindow* w, TSKEY ekey,
                                      bool ascQuery) {
H
Haojun Liao 已提交
677 678
  if (ascQuery) {
    getAlignQueryTimeWindow(pInterval, precision, ts, ts, ekey, w);
679 680
  } else {
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
H
Haojun Liao 已提交
681
    getAlignQueryTimeWindow(pInterval, precision, ts, ekey, ts, w);
682 683

    int64_t key = w->skey;
L
Liu Jicong 已提交
684
    while (key < ts) {  // moving towards end
685
      key = taosTimeAdd(key, pInterval->sliding, pInterval->slidingUnit, precision);
686 687 688 689 690 691 692 693 694 695
      if (key >= ts) {
        break;
      }

      w->skey = key;
    }
  }
}

// get the correct time window according to the handled timestamp
L
Liu Jicong 已提交
696 697
static STimeWindow getActiveTimeWindow(SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
                                       int32_t precision, STimeWindow* win) {
698 699
  STimeWindow w = {0};

L
Liu Jicong 已提交
700
  if (pResultRowInfo->curPos == -1) {  // the first window, from the previous stored value
H
Haojun Liao 已提交
701
    getInitialStartTimeWindow(pInterval, precision, ts, &w, win->ekey, true);
702
    w.ekey = taosTimeAdd(w.skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
703 704 705 706 707
  } else {
    w = getResultRow(pResultRowInfo, pResultRowInfo->curPos)->win;
  }

  if (w.skey > ts || w.ekey < ts) {
H
Haojun Liao 已提交
708 709 710
    if (pInterval->intervalUnit == 'n' || pInterval->intervalUnit == 'y') {
      w.skey = taosTimeTruncate(ts, pInterval, precision);
      w.ekey = taosTimeAdd(w.skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
711 712 713 714
    } else {
      int64_t st = w.skey;

      if (st > ts) {
H
Haojun Liao 已提交
715
        st -= ((st - ts + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
716 717
      }

H
Haojun Liao 已提交
718
      int64_t et = st + pInterval->interval - 1;
719
      if (et < ts) {
H
Haojun Liao 已提交
720
        st += ((ts - et + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
721 722 723
      }

      w.skey = st;
724
      w.ekey = taosTimeAdd(w.skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
725 726 727 728 729 730
    }
  }
  return w;
}

// get the correct time window according to the handled timestamp
L
Liu Jicong 已提交
731
static STimeWindow getCurrentActiveTimeWindow(SResultRowInfo* pResultRowInfo, int64_t ts, STaskAttr* pQueryAttr) {
732 733
  STimeWindow w = {0};

L
Liu Jicong 已提交
734 735
  if (pResultRowInfo->curPos == -1) {  // the first window, from the previous stored value
                                       //    getInitialStartTimeWindow(pQueryAttr, ts, &w);
736 737

    if (pQueryAttr->interval.intervalUnit == 'n' || pQueryAttr->interval.intervalUnit == 'y') {
L
Liu Jicong 已提交
738 739 740
      w.ekey =
          taosTimeAdd(w.skey, pQueryAttr->interval.interval, pQueryAttr->interval.intervalUnit, pQueryAttr->precision) -
          1;
741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759
    } else {
      w.ekey = w.skey + pQueryAttr->interval.interval - 1;
    }
  } else {
    w = getResultRow(pResultRowInfo, pResultRowInfo->curPos)->win;
  }

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

  return w;
}

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

L
Liu Jicong 已提交
765
  SFilePage* pData = NULL;
766 767 768 769 770 771

  // 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 已提交
772
    pData = getNewBufPage(pResultBuf, tid, &pageId);
773
    pData->num = sizeof(SFilePage);
774 775
  } else {
    SPageInfo* pi = getLastPageInfo(list);
776
    pData = getBufPage(pResultBuf, getPageId(pi));
777
    pageId = getPageId(pi);
778

779
    if (pData->num + size > getBufPageSize(pResultBuf)) {
780
      // release current page first, and prepare the next one
781
      releaseBufPageInfo(pResultBuf, pi);
782

H
Haojun Liao 已提交
783
      pData = getNewBufPage(pResultBuf, tid, &pageId);
784
      if (pData != NULL) {
785
        pData->num = sizeof(SFilePage);
786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805
      }
    }
  }

  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 已提交
806 807 808
static bool chkWindowOutputBufByKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, STimeWindow* win,
                                    bool masterscan, SResultRow** pResult, int64_t groupId, SqlFunctionCtx* pCtx,
                                    int32_t numOfOutput, int32_t* rowCellInfoOffset) {
809
  assert(win->skey <= win->ekey);
L
Liu Jicong 已提交
810
  return chkResultRowFromKey(pRuntimeEnv, pResultRowInfo, (char*)&win->skey, TSDB_KEYSIZE, masterscan, groupId);
811 812
}

L
Liu Jicong 已提交
813 814 815
static int32_t setResultOutputBufByKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, int64_t tid,
                                       STimeWindow* win, bool masterscan, SResultRow** pResult, int64_t tableGroupId,
                                       SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowCellInfoOffset) {
816
  assert(win->skey <= win->ekey);
L
Liu Jicong 已提交
817
  SDiskbasedBuf* pResultBuf = pRuntimeEnv->pResultBuf;
818

L
Liu Jicong 已提交
819 820
  SResultRow* pResultRow = doSetResultOutBufByKey(pRuntimeEnv, pResultRowInfo, tid, (char*)&win->skey, TSDB_KEYSIZE,
                                                  masterscan, tableGroupId);
821 822 823 824 825 826 827
  if (pResultRow == NULL) {
    *pResult = NULL;
    return TSDB_CODE_SUCCESS;
  }

  // not assign result buffer yet, add new result buffer
  if (pResultRow->pageId == -1) {
L
Liu Jicong 已提交
828 829
    int32_t ret = addNewWindowResultBuf(pResultRow, pResultBuf, (int32_t)tableGroupId,
                                        pRuntimeEnv->pQueryAttr->intermediateResultRowSize);
830 831 832 833 834 835 836 837 838 839 840 841 842
    if (ret != TSDB_CODE_SUCCESS) {
      return -1;
    }
  }

  // set time window for current result
  pResultRow->win = (*win);
  *pResult = pResultRow;
  setResultRowOutputBufInitCtx(pRuntimeEnv, pResultRow, pCtx, numOfOutput, rowCellInfoOffset);

  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
843 844
static void setResultRowOutputBufInitCtx_rv(SDiskbasedBuf* pBuf, SResultRow* pResult, SqlFunctionCtx* pCtx,
                                            int32_t numOfOutput, int32_t* rowCellInfoOffset);
H
Haojun Liao 已提交
845

L
Liu Jicong 已提交
846 847 848 849
static int32_t setResultOutputBufByKey_rv(SResultRowInfo* pResultRowInfo, int64_t id, STimeWindow* win, bool masterscan,
                                          SResultRow** pResult, int64_t tableGroupId, SqlFunctionCtx* pCtx,
                                          int32_t numOfOutput, int32_t* rowCellInfoOffset, SAggSupporter* pAggSup,
                                          SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
850
  assert(win->skey <= win->ekey);
L
Liu Jicong 已提交
851 852
  SResultRow* pResultRow = doSetResultOutBufByKey_rv(pAggSup->pResultBuf, pResultRowInfo, id, (char*)&win->skey,
                                                     TSDB_KEYSIZE, masterscan, tableGroupId, pTaskInfo, true, pAggSup);
H
Haojun Liao 已提交
853 854 855 856 857 858 859 860 861

  if (pResultRow == NULL) {
    *pResult = NULL;
    return TSDB_CODE_SUCCESS;
  }

  // set time window for current result
  pResultRow->win = (*win);
  *pResult = pResultRow;
H
Haojun Liao 已提交
862
  setResultRowOutputBufInitCtx_rv(pAggSup->pResultBuf, pResultRow, pCtx, numOfOutput, rowCellInfoOffset);
H
Haojun Liao 已提交
863 864 865
  return TSDB_CODE_SUCCESS;
}

866 867 868 869 870
static void setResultRowInterpo(SResultRow* pResult, SResultTsInterpType type) {
  assert(pResult != NULL && (type == RESULT_ROW_START_INTERP || type == RESULT_ROW_END_INTERP));
  if (type == RESULT_ROW_START_INTERP) {
    pResult->startInterp = true;
  } else {
L
Liu Jicong 已提交
871
    pResult->endInterp = true;
872 873 874 875 876 877 878 879
  }
}

static bool resultRowInterpolated(SResultRow* pResult, SResultTsInterpType type) {
  assert(pResult != NULL && (type == RESULT_ROW_START_INTERP || type == RESULT_ROW_END_INTERP));
  if (type == RESULT_ROW_START_INTERP) {
    return pResult->startInterp == true;
  } else {
L
Liu Jicong 已提交
880
    return pResult->endInterp == true;
881 882 883
  }
}

L
Liu Jicong 已提交
884 885
static FORCE_INLINE int32_t getForwardStepsInBlock(int32_t numOfRows, __block_search_fn_t searchFn, TSKEY ekey,
                                                   int16_t pos, int16_t order, int64_t* pData) {
886 887 888
  int32_t forwardStep = 0;

  if (order == TSDB_ORDER_ASC) {
L
Liu Jicong 已提交
889
    int32_t end = searchFn((char*)&pData[pos], numOfRows - pos, ekey, order);
890 891 892 893 894 895 896 897
    if (end >= 0) {
      forwardStep = end;

      if (pData[end + pos] == ekey) {
        forwardStep += 1;
      }
    }
  } else {
L
Liu Jicong 已提交
898
    int32_t end = searchFn((char*)pData, pos + 1, ekey, order);
899 900 901 902 903 904 905 906 907 908 909 910 911
    if (end >= 0) {
      forwardStep = pos - end;

      if (pData[end] == ekey) {
        forwardStep += 1;
      }
    }
  }

  assert(forwardStep >= 0);
  return forwardStep;
}

L
Liu Jicong 已提交
912 913
static void doUpdateResultRowIndex(SResultRowInfo* pResultRowInfo, TSKEY lastKey, bool ascQuery,
                                   bool timeWindowInterpo) {
914 915 916
  int64_t skey = TSKEY_INITIAL_VAL;
  int32_t i = 0;
  for (i = pResultRowInfo->size - 1; i >= 0; --i) {
L
Liu Jicong 已提交
917
    SResultRow* pResult = pResultRowInfo->pResult[i];
918 919 920 921 922 923
    if (pResult->closed) {
      break;
    }

    // new closed result rows
    if (timeWindowInterpo) {
L
Liu Jicong 已提交
924 925 926
      if (pResult->endInterp &&
          ((pResult->win.skey <= lastKey && ascQuery) || (pResult->win.skey >= lastKey && !ascQuery))) {
        if (i > 0) {  // the first time window, the startInterp is false.
927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945
          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 已提交
946
      //      assert(pResultRowInfo->current == NULL);
947 948 949 950 951 952 953
      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 已提交
954
      SResultRow* pResult = pResultRowInfo->pResult[i];
955 956 957 958 959 960 961 962 963 964 965 966 967
      if (pResult->closed) {
        break;
      }
    }

    if (i == pResultRowInfo->size - 1) {
      pResultRowInfo->curPos = i;
    } else {
      pResultRowInfo->curPos = i + 1;  // current not closed result object
    }
  }
}

L
Liu Jicong 已提交
968 969
static void updateResultRowInfoActiveIndex(SResultRowInfo* pResultRowInfo, const STimeWindow* pWin, TSKEY lastKey,
                                           bool ascQuery, bool interp) {
H
Haojun Liao 已提交
970
  if ((lastKey > pWin->ekey && ascQuery) || (lastKey < pWin->ekey && (!ascQuery))) {
971 972 973 974
    closeAllResultRows(pResultRowInfo);
    pResultRowInfo->curPos = pResultRowInfo->size - 1;
  } else {
    int32_t step = ascQuery ? 1 : -1;
H
Haojun Liao 已提交
975
    doUpdateResultRowIndex(pResultRowInfo, lastKey - step, ascQuery, interp);
976 977 978
  }
}

L
Liu Jicong 已提交
979 980 981
static int32_t getNumOfRowsInTimeWindow(SDataBlockInfo* pDataBlockInfo, TSKEY* pPrimaryColumn, int32_t startPos,
                                        TSKEY ekey, __block_search_fn_t searchFn, STableQueryInfo* item,
                                        int32_t order) {
982 983
  assert(startPos >= 0 && startPos < pDataBlockInfo->rows);

L
Liu Jicong 已提交
984 985
  int32_t num = -1;
  int32_t step = GET_FORWARD_DIRECTION_FACTOR(order);
986

H
Haojun Liao 已提交
987
  if (order == TSDB_ORDER_ASC) {
988 989
    if (ekey < pDataBlockInfo->window.ekey && pPrimaryColumn) {
      num = getForwardStepsInBlock(pDataBlockInfo->rows, searchFn, ekey, startPos, order, pPrimaryColumn);
H
Haojun Liao 已提交
990
      if (item != NULL) {
991 992 993 994
        item->lastKey = pPrimaryColumn[startPos + (num - 1)] + step;
      }
    } else {
      num = pDataBlockInfo->rows - startPos;
H
Haojun Liao 已提交
995
      if (item != NULL) {
996 997 998 999 1000 1001
        item->lastKey = pDataBlockInfo->window.ekey + step;
      }
    }
  } else {  // desc
    if (ekey > pDataBlockInfo->window.skey && pPrimaryColumn) {
      num = getForwardStepsInBlock(pDataBlockInfo->rows, searchFn, ekey, startPos, order, pPrimaryColumn);
H
Haojun Liao 已提交
1002
      if (item != NULL) {
1003 1004 1005 1006
        item->lastKey = pPrimaryColumn[startPos - (num - 1)] + step;
      }
    } else {
      num = startPos + 1;
H
Haojun Liao 已提交
1007
      if (item != NULL) {
1008 1009 1010 1011 1012 1013 1014 1015 1016
        item->lastKey = pDataBlockInfo->window.skey + step;
      }
    }
  }

  assert(num >= 0);
  return num;
}

H
Haojun Liao 已提交
1017
static void doApplyFunctions(SqlFunctionCtx* pCtx, STimeWindow* pWin, int32_t offset, int32_t forwardStep, TSKEY* tsCol,
1018
                             int32_t numOfTotal, int32_t numOfOutput, int32_t order) {
1019 1020 1021 1022
  for (int32_t k = 0; k < numOfOutput; ++k) {
    pCtx[k].startTs = pWin->skey;

    // keep it temporarialy
L
Liu Jicong 已提交
1023
    bool    hasAgg = pCtx[k].input.colDataAggIsSet;
H
Haojun Liao 已提交
1024
    int32_t startOffset = pCtx[k].input.startRowIndex;
L
Liu Jicong 已提交
1025
    int32_t numOfRows = pCtx[k].input.numOfRows;
1026

H
Haojun Liao 已提交
1027
    int32_t pos = (order == TSDB_ORDER_ASC) ? offset : offset - (forwardStep - 1);
1028 1029
    pCtx[k].input.startRowIndex = pos;
    pCtx[k].input.numOfRows = forwardStep;
1030 1031

    if (tsCol != NULL) {
H
Haojun Liao 已提交
1032
      pCtx[k].ptsList = tsCol;
1033 1034 1035 1036 1037 1038 1039 1040
    }

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

H
Haojun Liao 已提交
1041
    if (functionNeedToExecute(&pCtx[k])) {
1042
      pCtx[k].fpSet.process(&pCtx[k]);
1043 1044 1045
    }

    // restore it
1046 1047 1048
    pCtx[k].input.colDataAggIsSet = hasAgg;
    pCtx[k].input.startRowIndex = startOffset;
    pCtx[k].input.numOfRows = numOfRows;
1049 1050 1051
  }
}

H
Haojun Liao 已提交
1052 1053 1054
static int32_t getNextQualifiedWindow(SInterval* pInterval, STimeWindow* pNext, SDataBlockInfo* pDataBlockInfo,
                                      TSKEY* primaryKeys, int32_t prevPosition, STableIntervalOperatorInfo* pInfo) {
  int32_t order = pInfo->order;
L
Liu Jicong 已提交
1055
  bool    ascQuery = (order == TSDB_ORDER_ASC);
H
Haojun Liao 已提交
1056 1057 1058

  int32_t precision = pInfo->precision;
  getNextTimeWindow(pInterval, precision, order, pNext);
1059 1060

  // next time window is not in current block
H
Haojun Liao 已提交
1061 1062
  if ((pNext->skey > pDataBlockInfo->window.ekey && order == TSDB_ORDER_ASC) ||
      (pNext->ekey < pDataBlockInfo->window.skey && order == TSDB_ORDER_DESC)) {
1063 1064 1065
    return -1;
  }

L
Liu Jicong 已提交
1066
  TSKEY   startKey = ascQuery ? pNext->skey : pNext->ekey;
1067 1068 1069
  int32_t startPos = 0;

  // tumbling time window query, a special case of sliding time window query
H
Haojun Liao 已提交
1070 1071
  if (pInterval->sliding == pInterval->interval && prevPosition != -1) {
    int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
1072 1073
    startPos = prevPosition + factor;
  } else {
H
Haojun Liao 已提交
1074
    if (startKey <= pDataBlockInfo->window.skey && ascQuery) {
1075
      startPos = 0;
H
Haojun Liao 已提交
1076
    } else if (startKey >= pDataBlockInfo->window.ekey && !ascQuery) {
1077 1078
      startPos = pDataBlockInfo->rows - 1;
    } else {
L
Liu Jicong 已提交
1079
      startPos = binarySearchForKey((char*)primaryKeys, pDataBlockInfo->rows, startKey, order);
1080 1081 1082 1083
    }
  }

  /* interp query with fill should not skip time window */
L
Liu Jicong 已提交
1084 1085 1086
  //  if (pQueryAttr->pointInterpQuery && pQueryAttr->fillType != TSDB_FILL_NONE) {
  //    return startPos;
  //  }
1087 1088 1089 1090 1091 1092

  /*
   * This time window does not cover any data, try next time window,
   * this case may happen when the time window is too small
   */
  if (primaryKeys == NULL) {
H
Haojun Liao 已提交
1093
    if (ascQuery) {
1094 1095 1096 1097 1098
      assert(pDataBlockInfo->window.skey <= pNext->ekey);
    } else {
      assert(pDataBlockInfo->window.ekey >= pNext->skey);
    }
  } else {
H
Haojun Liao 已提交
1099
    if (ascQuery && primaryKeys[startPos] > pNext->ekey) {
1100
      TSKEY next = primaryKeys[startPos];
H
Haojun Liao 已提交
1101 1102 1103
      if (pInterval->intervalUnit == 'n' || pInterval->intervalUnit == 'y') {
        pNext->skey = taosTimeTruncate(next, pInterval, precision);
        pNext->ekey = taosTimeAdd(pNext->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
1104
      } else {
L
Liu Jicong 已提交
1105
        pNext->ekey += ((next - pNext->ekey + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
H
Haojun Liao 已提交
1106
        pNext->skey = pNext->ekey - pInterval->interval + 1;
1107
      }
H
Haojun Liao 已提交
1108
    } else if ((!ascQuery) && primaryKeys[startPos] < pNext->skey) {
1109
      TSKEY next = primaryKeys[startPos];
H
Haojun Liao 已提交
1110 1111 1112
      if (pInterval->intervalUnit == 'n' || pInterval->intervalUnit == 'y') {
        pNext->skey = taosTimeTruncate(next, pInterval, precision);
        pNext->ekey = taosTimeAdd(pNext->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
1113
      } else {
H
Haojun Liao 已提交
1114 1115
        pNext->skey -= ((pNext->skey - next + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
        pNext->ekey = pNext->skey + pInterval->interval - 1;
1116 1117 1118 1119 1120 1121 1122
      }
    }
  }

  return startPos;
}

L
Liu Jicong 已提交
1123
static FORCE_INLINE TSKEY reviseWindowEkey(STaskAttr* pQueryAttr, STimeWindow* pWindow) {
1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139
  TSKEY ekey = -1;
  if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
    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;
}

H
Haojun Liao 已提交
1140
static void setNotInterpoWindowKey(SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t type) {
1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151
  if (type == RESULT_ROW_START_INTERP) {
    for (int32_t k = 0; k < numOfOutput; ++k) {
      pCtx[k].start.key = INT64_MIN;
    }
  } else {
    for (int32_t k = 0; k < numOfOutput; ++k) {
      pCtx[k].end.key = INT64_MIN;
    }
  }
}

H
Haojun Liao 已提交
1152
static void saveDataBlockLastRow(char** pRow, SArray* pDataBlock, int32_t rowIndex, int32_t numOfCols) {
1153 1154 1155 1156
  if (pDataBlock == NULL) {
    return;
  }

H
Haojun Liao 已提交
1157
  for (int32_t k = 0; k < numOfCols; ++k) {
L
Liu Jicong 已提交
1158
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock, k);
H
Haojun Liao 已提交
1159
    memcpy(pRow[k], ((char*)pColInfo->pData) + (pColInfo->info.bytes * rowIndex), pColInfo->info.bytes);
1160 1161 1162
  }
}

H
Haojun Liao 已提交
1163
static TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols, int32_t rows, bool ascQuery) {
1164 1165
  TSKEY ts = TSKEY_INITIAL_VAL;
  if (tsCols == NULL) {
L
Liu Jicong 已提交
1166
    ts = ascQuery ? win->skey : win->ekey;
1167
  } else {
L
Liu Jicong 已提交
1168
    int32_t offset = ascQuery ? 0 : rows - 1;
1169 1170 1171 1172 1173 1174
    ts = tsCols[offset];
  }

  return ts;
}

H
Haojun Liao 已提交
1175
static void doSetInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order);
1176

L
Liu Jicong 已提交
1177 1178
static void doSetInputDataBlockInfo(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock,
                                    int32_t order) {
1179 1180
  for (int32_t i = 0; i < pOperator->numOfOutput; ++i) {
    pCtx[i].order = order;
L
Liu Jicong 已提交
1181
    pCtx[i].size = pBlock->info.rows;
1182 1183
    pCtx[i].currentStage = (uint8_t)pOperator->pRuntimeEnv->scanFlag;

L
Liu Jicong 已提交
1184
    setBlockStatisInfo(&pCtx[i], pBlock, NULL /*&pOperator->pExpr[i].base.colInfo*/);
1185 1186 1187
  }
}

H
Haojun Liao 已提交
1188
void setInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order) {
H
Haojun Liao 已提交
1189 1190 1191 1192 1193
  if (pBlock->pDataBlock != NULL) {
    doSetInputDataBlock(pOperator, pCtx, pBlock, order);
  } else {
    doSetInputDataBlockInfo(pOperator, pCtx, pBlock, order);
  }
1194 1195
}

H
Haojun Liao 已提交
1196
static void doSetInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order) {
1197 1198
  for (int32_t i = 0; i < pOperator->numOfOutput; ++i) {
    pCtx[i].order = order;
L
Liu Jicong 已提交
1199
    pCtx[i].size = pBlock->info.rows;
H
Haojun Liao 已提交
1200 1201
    pCtx[i].currentStage = MAIN_SCAN;

G
Ganlin Zhao 已提交
1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213
    SExprInfo expr = pOperator->pExpr[i];
    for (int32_t j = 0; j < expr.base.numOfParams; ++j) {
      SFunctParam *pFuncParam = &expr.base.pParam[j];
      if (pFuncParam->type == FUNC_PARAM_TYPE_COLUMN) {
        int32_t slotId = pFuncParam->pCol->slotId;
        pCtx[i].input.pData[j]  = taosArrayGet(pBlock->pDataBlock, slotId);
        pCtx[i].input.totalRows = pBlock->info.rows;
        pCtx[i].input.numOfRows = pBlock->info.rows;
        pCtx[i].input.startRowIndex = 0;
        ASSERT(pCtx[i].input.pData[j] != NULL);
      }
    }
H
Haojun Liao 已提交
1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231
    //    setBlockStatisInfo(&pCtx[i], pBlock, pOperator->pExpr[i].base.pColumns);

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

    // in case of the block distribution query, the inputBytes is not a constant value.
G
Ganlin Zhao 已提交
1232 1233 1234 1235
    //pCtx[i].input.pData[0]  = taosArrayGet(pBlock->pDataBlock, slotId);
    //pCtx[i].input.totalRows = pBlock->info.rows;
    //pCtx[i].input.numOfRows = pBlock->info.rows;
    //pCtx[i].input.startRowIndex = 0;
H
Haojun Liao 已提交
1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263


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

static void doAggregateImpl(SOperatorInfo* pOperator, TSKEY startTs, SqlFunctionCtx* pCtx) {
1264
  for (int32_t k = 0; k < pOperator->numOfOutput; ++k) {
H
Haojun Liao 已提交
1265
    if (functionNeedToExecute(&pCtx[k])) {
L
Liu Jicong 已提交
1266
      pCtx[k].startTs = startTs;  // this can be set during create the struct
H
Haojun Liao 已提交
1267
      pCtx[k].fpSet.process(&pCtx[k]);
1268 1269 1270 1271
    }
  }
}

L
Liu Jicong 已提交
1272 1273
static void projectApplyFunctions(SExprInfo* pExpr, SSDataBlock* pResult, SSDataBlock* pSrcBlock, SqlFunctionCtx* pCtx,
                                  int32_t numOfOutput) {
1274
  for (int32_t k = 0; k < numOfOutput; ++k) {
L
Liu Jicong 已提交
1275
    if (pExpr[k].pExpr->nodeType == QUERY_NODE_COLUMN) {  // it is a project query
H
Haojun Liao 已提交
1276
      SColumnInfoData* pColInfoData = taosArrayGet(pResult->pDataBlock, k);
1277
      colDataAssign(pColInfoData, pCtx[k].input.pData[0], pCtx[k].input.numOfRows);
1278 1279

      pResult->info.rows = pCtx[0].input.numOfRows;
H
Haojun Liao 已提交
1280
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_OPERATOR) {
1281 1282 1283
      SArray* pBlockList = taosArrayInit(4, POINTER_BYTES);
      taosArrayPush(pBlockList, &pSrcBlock);

1284 1285 1286 1287 1288 1289 1290 1291 1292 1293
      SScalarParam dest = {0};
      dest.columnData = taosArrayGet(pResult->pDataBlock, k);

      scalarCalculate(pExpr[k].pExpr->_optrRoot.pRootNode, pBlockList, &dest);
      pResult->info.rows = dest.numOfRows;

      taosArrayDestroy(pBlockList);
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_FUNCTION) {
      ASSERT(!fmIsAggFunc(pCtx->functionId));

G
Ganlin Zhao 已提交
1294 1295
      SArray* pBlockList = taosArrayInit(4, POINTER_BYTES);
      taosArrayPush(pBlockList, &pSrcBlock);
1296 1297 1298 1299

      SScalarParam dest = {0};
      dest.columnData = taosArrayGet(pResult->pDataBlock, k);

G
Ganlin Zhao 已提交
1300
      scalarCalculate((SNode *)pExpr[k].pExpr->_function.pFunctNode, pBlockList, &dest);
1301
      pResult->info.rows = dest.numOfRows;
G
Ganlin Zhao 已提交
1302 1303 1304

      taosArrayDestroy(pBlockList);

1305
    } else {
1306
      ASSERT(0);
1307 1308 1309 1310 1311 1312
    }
  }
}

void doTimeWindowInterpolation(SOperatorInfo* pOperator, SOptrBasicInfo* pInfo, SArray* pDataBlock, TSKEY prevTs,
                               int32_t prevRowIndex, TSKEY curTs, int32_t curRowIndex, TSKEY windowKey, int32_t type) {
L
Liu Jicong 已提交
1313 1314
  STaskRuntimeEnv* pRuntimeEnv = pOperator->pRuntimeEnv;
  SExprInfo*       pExpr = pOperator->pExpr;
1315

H
Haojun Liao 已提交
1316
  SqlFunctionCtx* pCtx = pInfo->pCtx;
1317 1318 1319 1320 1321 1322 1323 1324

  for (int32_t k = 0; k < pOperator->numOfOutput; ++k) {
    int32_t functionId = pCtx[k].functionId;
    if (functionId != FUNCTION_TWA && functionId != FUNCTION_INTERP) {
      pCtx[k].start.key = INT64_MIN;
      continue;
    }

L
Liu Jicong 已提交
1325
    SColIndex*       pColIndex = NULL /*&pExpr[k].base.colInfo*/;
1326
    int16_t          index = pColIndex->colIndex;
L
Liu Jicong 已提交
1327
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock, index);
1328

L
Liu Jicong 已提交
1329
    //    assert(pColInfo->info.colId == pColIndex->info.colId && curTs != windowKey);
1330 1331 1332
    double v1 = 0, v2 = 0, v = 0;

    if (prevRowIndex == -1) {
L
Liu Jicong 已提交
1333
      GET_TYPED_DATA(v1, double, pColInfo->info.type, (char*)pRuntimeEnv->prevRow[index]);
1334
    } else {
L
Liu Jicong 已提交
1335
      GET_TYPED_DATA(v1, double, pColInfo->info.type, (char*)pColInfo->pData + prevRowIndex * pColInfo->info.bytes);
1336 1337
    }

L
Liu Jicong 已提交
1338
    GET_TYPED_DATA(v2, double, pColInfo->info.type, (char*)pColInfo->pData + curRowIndex * pColInfo->info.bytes);
1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349

    if (functionId == FUNCTION_INTERP) {
      if (type == RESULT_ROW_START_INTERP) {
        pCtx[k].start.key = prevTs;
        pCtx[k].start.val = v1;

        pCtx[k].end.key = curTs;
        pCtx[k].end.val = v2;

        if (pColInfo->info.type == TSDB_DATA_TYPE_BINARY || pColInfo->info.type == TSDB_DATA_TYPE_NCHAR) {
          if (prevRowIndex == -1) {
L
Liu Jicong 已提交
1350
            pCtx[k].start.ptr = (char*)pRuntimeEnv->prevRow[index];
1351
          } else {
L
Liu Jicong 已提交
1352
            pCtx[k].start.ptr = (char*)pColInfo->pData + prevRowIndex * pColInfo->info.bytes;
1353 1354
          }

L
Liu Jicong 已提交
1355
          pCtx[k].end.ptr = (char*)pColInfo->pData + curRowIndex * pColInfo->info.bytes;
1356 1357 1358
        }
      }
    } else if (functionId == FUNCTION_TWA) {
L
Liu Jicong 已提交
1359 1360 1361
      SPoint point1 = (SPoint){.key = prevTs, .val = &v1};
      SPoint point2 = (SPoint){.key = curTs, .val = &v2};
      SPoint point = (SPoint){.key = windowKey, .val = &v};
1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375

      taosGetLinearInterpolationVal(&point, TSDB_DATA_TYPE_DOUBLE, &point1, &point2, TSDB_DATA_TYPE_DOUBLE);

      if (type == RESULT_ROW_START_INTERP) {
        pCtx[k].start.key = point.key;
        pCtx[k].start.val = v;
      } else {
        pCtx[k].end.key = point.key;
        pCtx[k].end.val = v;
      }
    }
  }
}

H
Haojun Liao 已提交
1376
static bool setTimeWindowInterpolationStartTs(SOperatorInfo* pOperatorInfo, SqlFunctionCtx* pCtx, int32_t pos,
L
Liu Jicong 已提交
1377 1378
                                              int32_t numOfRows, SArray* pDataBlock, const TSKEY* tsCols,
                                              STimeWindow* win) {
H
Haojun Liao 已提交
1379
  STaskRuntimeEnv* pRuntimeEnv = pOperatorInfo->pRuntimeEnv;
L
Liu Jicong 已提交
1380
  STaskAttr*       pQueryAttr = pRuntimeEnv->pQueryAttr;
1381 1382 1383

  bool ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);

L
Liu Jicong 已提交
1384 1385
  TSKEY curTs = tsCols[pos];
  TSKEY lastTs = *(TSKEY*)pRuntimeEnv->prevRow[0];
1386 1387 1388

  // lastTs == INT64_MIN and pos == 0 means this is the first time window, interpolation is not needed.
  // start exactly from this point, no need to do interpolation
L
Liu Jicong 已提交
1389
  TSKEY key = ascQuery ? win->skey : win->ekey;
1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400
  if (key == curTs) {
    setNotInterpoWindowKey(pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_START_INTERP);
    return true;
  }

  if (lastTs == INT64_MIN && ((pos == 0 && ascQuery) || (pos == (numOfRows - 1) && !ascQuery))) {
    setNotInterpoWindowKey(pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_START_INTERP);
    return true;
  }

  int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
L
Liu Jicong 已提交
1401
  TSKEY   prevTs = ((pos == 0 && ascQuery) || (pos == (numOfRows - 1) && !ascQuery)) ? lastTs : tsCols[pos - step];
1402

L
Liu Jicong 已提交
1403 1404
  doTimeWindowInterpolation(pOperatorInfo, pOperatorInfo->info, pDataBlock, prevTs, pos - step, curTs, pos, key,
                            RESULT_ROW_START_INTERP);
1405 1406 1407
  return true;
}

L
Liu Jicong 已提交
1408 1409 1410 1411 1412 1413
static bool setTimeWindowInterpolationEndTs(SOperatorInfo* pOperatorInfo, SqlFunctionCtx* pCtx, int32_t endRowIndex,
                                            SArray* pDataBlock, const TSKEY* tsCols, TSKEY blockEkey,
                                            STimeWindow* win) {
  STaskRuntimeEnv* pRuntimeEnv = pOperatorInfo->pRuntimeEnv;
  STaskAttr*       pQueryAttr = pRuntimeEnv->pQueryAttr;
  int32_t          numOfOutput = pOperatorInfo->numOfOutput;
1414

L
Liu Jicong 已提交
1415
  TSKEY actualEndKey = tsCols[endRowIndex];
1416

L
Liu Jicong 已提交
1417
  TSKEY key = QUERY_IS_ASC_QUERY(pQueryAttr) ? win->ekey : win->skey;
1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436

  // not ended in current data block, do not invoke interpolation
  if ((key > blockEkey && QUERY_IS_ASC_QUERY(pQueryAttr)) || (key < blockEkey && !QUERY_IS_ASC_QUERY(pQueryAttr))) {
    setNotInterpoWindowKey(pCtx, numOfOutput, RESULT_ROW_END_INTERP);
    return false;
  }

  // there is actual end point of current time window, no interpolation need
  if (key == actualEndKey) {
    setNotInterpoWindowKey(pCtx, numOfOutput, RESULT_ROW_END_INTERP);
    return true;
  }

  int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
  int32_t nextRowIndex = endRowIndex + step;
  assert(nextRowIndex >= 0);

  TSKEY nextKey = tsCols[nextRowIndex];
  doTimeWindowInterpolation(pOperatorInfo, pOperatorInfo->info, pDataBlock, actualEndKey, endRowIndex, nextKey,
L
Liu Jicong 已提交
1437
                            nextRowIndex, key, RESULT_ROW_END_INTERP);
1438 1439 1440
  return true;
}

H
Haojun Liao 已提交
1441
static void doWindowBorderInterpolation(SOperatorInfo* pOperatorInfo, SSDataBlock* pBlock, SqlFunctionCtx* pCtx,
L
Liu Jicong 已提交
1442 1443
                                        SResultRow* pResult, STimeWindow* win, int32_t startPos, int32_t forwardStep,
                                        int32_t order, bool timeWindowInterpo) {
H
Haojun Liao 已提交
1444
  if (!timeWindowInterpo) {
1445 1446 1447 1448
    return;
  }

  assert(pBlock != NULL);
H
Haojun Liao 已提交
1449
  int32_t step = GET_FORWARD_DIRECTION_FACTOR(order);
1450

L
Liu Jicong 已提交
1451 1452
  if (pBlock->pDataBlock == NULL) {
    //    tscError("pBlock->pDataBlock == NULL");
1453 1454
    return;
  }
H
Haojun Liao 已提交
1455

L
Liu Jicong 已提交
1456
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, 0);
1457

L
Liu Jicong 已提交
1458 1459 1460
  TSKEY* tsCols = (TSKEY*)(pColInfo->pData);
  bool   done = resultRowInterpolated(pResult, RESULT_ROW_START_INTERP);
  if (!done) {  // it is not interpolated, now start to generated the interpolated value
1461
    int32_t startRowIndex = startPos;
L
Liu Jicong 已提交
1462 1463
    bool    interp = setTimeWindowInterpolationStartTs(pOperatorInfo, pCtx, startRowIndex, pBlock->info.rows,
                                                       pBlock->pDataBlock, tsCols, win);
1464 1465 1466 1467
    if (interp) {
      setResultRowInterpo(pResult, RESULT_ROW_START_INTERP);
    }
  } else {
H
Haojun Liao 已提交
1468
    setNotInterpoWindowKey(pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_START_INTERP);
1469 1470 1471
  }

  // point interpolation does not require the end key time window interpolation.
L
Liu Jicong 已提交
1472 1473 1474
  //  if (pointInterpQuery) {
  //    return;
  //  }
1475 1476 1477 1478 1479 1480

  // interpolation query does not generate the time window end interpolation
  done = resultRowInterpolated(pResult, RESULT_ROW_END_INTERP);
  if (!done) {
    int32_t endRowIndex = startPos + (forwardStep - 1) * step;

L
Liu Jicong 已提交
1481 1482 1483
    TSKEY endKey = (order == TSDB_ORDER_ASC) ? pBlock->info.window.ekey : pBlock->info.window.skey;
    bool  interp =
        setTimeWindowInterpolationEndTs(pOperatorInfo, pCtx, endRowIndex, pBlock->pDataBlock, tsCols, endKey, win);
1484 1485 1486 1487
    if (interp) {
      setResultRowInterpo(pResult, RESULT_ROW_END_INTERP);
    }
  } else {
H
Haojun Liao 已提交
1488
    setNotInterpoWindowKey(pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_END_INTERP);
1489 1490 1491
  }
}

L
Liu Jicong 已提交
1492 1493 1494
static void hashIntervalAgg(SOperatorInfo* pOperatorInfo, SResultRowInfo* pResultRowInfo, SSDataBlock* pSDataBlock,
                            int32_t tableGroupId) {
  STableIntervalOperatorInfo* pInfo = (STableIntervalOperatorInfo*)pOperatorInfo->info;
1495

H
Haojun Liao 已提交
1496
  SExecTaskInfo* pTaskInfo = pOperatorInfo->pTaskInfo;
L
Liu Jicong 已提交
1497
  int32_t        numOfOutput = pOperatorInfo->numOfOutput;
1498

H
Haojun Liao 已提交
1499
  int32_t step = 1;
L
Liu Jicong 已提交
1500
  bool    ascQuery = true;
1501 1502 1503 1504 1505 1506

  int32_t prevIndex = pResultRowInfo->curPos;

  TSKEY* tsCols = NULL;
  if (pSDataBlock->pDataBlock != NULL) {
    SColumnInfoData* pColDataInfo = taosArrayGet(pSDataBlock->pDataBlock, 0);
L
Liu Jicong 已提交
1507 1508 1509
    tsCols = (int64_t*)pColDataInfo->pData;
    assert(tsCols[0] == pSDataBlock->info.window.skey &&
           tsCols[pSDataBlock->info.rows - 1] == pSDataBlock->info.window.ekey);
1510 1511
  }

L
Liu Jicong 已提交
1512 1513
  int32_t startPos = ascQuery ? 0 : (pSDataBlock->info.rows - 1);
  TSKEY   ts = getStartTsKey(&pSDataBlock->info.window, tsCols, pSDataBlock->info.rows, ascQuery);
1514

H
Haojun Liao 已提交
1515
  STimeWindow win = getActiveTimeWindow(pResultRowInfo, ts, &pInfo->interval, pInfo->precision, &pInfo->win);
L
Liu Jicong 已提交
1516
  bool        masterScan = true;
1517 1518

  SResultRow* pResult = NULL;
L
Liu Jicong 已提交
1519 1520 1521
  int32_t     ret = setResultOutputBufByKey_rv(pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult,
                                               tableGroupId, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset,
                                               &pInfo->aggSup, pTaskInfo);
1522
  if (ret != TSDB_CODE_SUCCESS || pResult == NULL) {
H
Haojun Liao 已提交
1523
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
1524 1525 1526
  }

  int32_t forwardStep = 0;
H
Haojun Liao 已提交
1527
  TSKEY   ekey = win.ekey;
1528
  forwardStep =
H
Haojun Liao 已提交
1529
      getNumOfRowsInTimeWindow(&pSDataBlock->info, tsCols, startPos, ekey, binarySearchForKey, NULL, TSDB_ORDER_ASC);
1530 1531 1532

  // prev time window not interpolation yet.
  int32_t curIndex = pResultRowInfo->curPos;
H
Haojun Liao 已提交
1533
  if (prevIndex != -1 && prevIndex < curIndex && pInfo->timeWindowInterpo) {
1534 1535 1536
    for (int32_t j = prevIndex; j < curIndex; ++j) {  // previous time window may be all closed already.
      SResultRow* pRes = getResultRow(pResultRowInfo, j);
      if (pRes->closed) {
L
Liu Jicong 已提交
1537 1538
        assert(resultRowInterpolated(pRes, RESULT_ROW_START_INTERP) &&
               resultRowInterpolated(pRes, RESULT_ROW_END_INTERP));
1539 1540 1541
        continue;
      }

L
Liu Jicong 已提交
1542 1543 1544 1545 1546 1547 1548
      STimeWindow w = pRes->win;
      ret = setResultOutputBufByKey_rv(pResultRowInfo, pSDataBlock->info.uid, &w, masterScan, &pResult, tableGroupId,
                                       pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup,
                                       pTaskInfo);
      if (ret != TSDB_CODE_SUCCESS) {
        longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
      }
1549

L
Liu Jicong 已提交
1550
      assert(!resultRowInterpolated(pResult, RESULT_ROW_END_INTERP));
1551

L
Liu Jicong 已提交
1552 1553
      doTimeWindowInterpolation(pOperatorInfo, &pInfo->binfo, pSDataBlock->pDataBlock, *(TSKEY*)pInfo->pRow[0], -1,
                                tsCols[startPos], startPos, w.ekey, RESULT_ROW_END_INTERP);
1554

L
Liu Jicong 已提交
1555 1556
      setResultRowInterpo(pResult, RESULT_ROW_END_INTERP);
      setNotInterpoWindowKey(pInfo->binfo.pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_START_INTERP);
1557

L
Liu Jicong 已提交
1558 1559
      doApplyFunctions(pInfo->binfo.pCtx, &w, startPos, 0, tsCols, pSDataBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);
    }
1560 1561

    // restore current time window
L
Liu Jicong 已提交
1562 1563 1564
    ret = setResultOutputBufByKey_rv(pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult, tableGroupId,
                                     pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup,
                                     pTaskInfo);
1565
    if (ret != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
1566
      longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
1567 1568 1569 1570
    }
  }

  // window start key interpolation
L
Liu Jicong 已提交
1571 1572 1573 1574
  doWindowBorderInterpolation(pOperatorInfo, pSDataBlock, pInfo->binfo.pCtx, pResult, &win, startPos, forwardStep,
                              pInfo->order, false);
  doApplyFunctions(pInfo->binfo.pCtx, &win, startPos, forwardStep, tsCols, pSDataBlock->info.rows, numOfOutput,
                   TSDB_ORDER_ASC);
1575 1576 1577 1578

  STimeWindow nextWin = win;
  while (1) {
    int32_t prevEndPos = (forwardStep - 1) * step + startPos;
H
Haojun Liao 已提交
1579
    startPos = getNextQualifiedWindow(&pInfo->interval, &nextWin, &pSDataBlock->info, tsCols, prevEndPos, pInfo);
1580 1581 1582 1583 1584
    if (startPos < 0) {
      break;
    }

    // null data, failed to allocate more memory buffer
L
Liu Jicong 已提交
1585 1586 1587
    int32_t code = setResultOutputBufByKey_rv(pResultRowInfo, pSDataBlock->info.uid, &nextWin, masterScan, &pResult,
                                              tableGroupId, pInfo->binfo.pCtx, numOfOutput,
                                              pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup, pTaskInfo);
1588
    if (code != TSDB_CODE_SUCCESS || pResult == NULL) {
H
Haojun Liao 已提交
1589
      longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
1590 1591
    }

L
Liu Jicong 已提交
1592 1593 1594
    ekey = nextWin.ekey;  // reviseWindowEkey(pQueryAttr, &nextWin);
    forwardStep =
        getNumOfRowsInTimeWindow(&pSDataBlock->info, tsCols, startPos, ekey, binarySearchForKey, NULL, TSDB_ORDER_ASC);
1595 1596

    // window start(end) key interpolation
L
Liu Jicong 已提交
1597 1598 1599 1600
    doWindowBorderInterpolation(pOperatorInfo, pSDataBlock, pInfo->binfo.pCtx, pResult, &nextWin, startPos, forwardStep,
                                pInfo->order, false);
    doApplyFunctions(pInfo->binfo.pCtx, &nextWin, startPos, forwardStep, tsCols, pSDataBlock->info.rows, numOfOutput,
                     TSDB_ORDER_ASC);
1601 1602
  }

H
Haojun Liao 已提交
1603
  if (pInfo->timeWindowInterpo) {
L
Liu Jicong 已提交
1604
    int32_t rowIndex = ascQuery ? (pSDataBlock->info.rows - 1) : 0;
H
Haojun Liao 已提交
1605
    saveDataBlockLastRow(pInfo->pRow, pSDataBlock->pDataBlock, rowIndex, pSDataBlock->info.numOfCols);
1606 1607
  }

L
Liu Jicong 已提交
1608
  //  updateResultRowInfoActiveIndex(pResultRowInfo, &pInfo->win, pRuntimeEnv->current->lastKey, true, false);
1609 1610
}

L
Liu Jicong 已提交
1611 1612 1613
static void hashAllIntervalAgg(SOperatorInfo* pOperatorInfo, SResultRowInfo* pResultRowInfo, SSDataBlock* pSDataBlock,
                               int32_t tableGroupId) {
  STableIntervalOperatorInfo* pInfo = (STableIntervalOperatorInfo*)pOperatorInfo->info;
1614

H
Haojun Liao 已提交
1615
  STaskRuntimeEnv* pRuntimeEnv = pOperatorInfo->pRuntimeEnv;
L
Liu Jicong 已提交
1616
  int32_t          numOfOutput = pOperatorInfo->numOfOutput;
H
Haojun Liao 已提交
1617
  STaskAttr*       pQueryAttr = pRuntimeEnv->pQueryAttr;
1618 1619

  int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
L
Liu Jicong 已提交
1620
  bool    ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
1621 1622 1623 1624

  TSKEY* tsCols = NULL;
  if (pSDataBlock->pDataBlock != NULL) {
    SColumnInfoData* pColDataInfo = taosArrayGet(pSDataBlock->pDataBlock, 0);
L
Liu Jicong 已提交
1625
    tsCols = (int64_t*)pColDataInfo->pData;
1626 1627 1628 1629
    assert(tsCols[0] == pSDataBlock->info.window.skey &&
           tsCols[pSDataBlock->info.rows - 1] == pSDataBlock->info.window.ekey);
  }

L
Liu Jicong 已提交
1630 1631
  int32_t startPos = ascQuery ? 0 : (pSDataBlock->info.rows - 1);
  TSKEY   ts = getStartTsKey(&pSDataBlock->info.window, tsCols, pSDataBlock->info.rows, ascQuery);
1632 1633

  STimeWindow win = getCurrentActiveTimeWindow(pResultRowInfo, ts, pQueryAttr);
L
Liu Jicong 已提交
1634
  bool        masterScan = IS_MAIN_SCAN(pRuntimeEnv);
1635 1636

  SResultRow* pResult = NULL;
L
Liu Jicong 已提交
1637 1638
  int32_t     forwardStep = 0;
  int32_t     ret = 0;
1639 1640 1641 1642 1643
  STimeWindow preWin = win;

  while (1) {
    // null data, failed to allocate more memory buffer
    ret = setResultOutputBufByKey(pRuntimeEnv, pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult,
H
Haojun Liao 已提交
1644
                                  tableGroupId, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset);
1645 1646 1647 1648
    if (ret != TSDB_CODE_SUCCESS) {
      longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
    }

L
Liu Jicong 已提交
1649 1650 1651
    TSKEY ekey = reviseWindowEkey(pQueryAttr, &win);
    //    forwardStep = getNumOfRowsInTimeWindow(pRuntimeEnv, &pSDataBlock->info, tsCols, startPos, ekey,
    //    binarySearchForKey, true);
1652 1653

    // window start(end) key interpolation
L
Liu Jicong 已提交
1654 1655 1656
    //    doWindowBorderInterpolation(pOperatorInfo, pSDataBlock, pInfo->binfo.pCtx, pResult, &win, startPos,
    //    forwardStep); doApplyFunctions(pRuntimeEnv, pInfo->binfo.pCtx, ascQuery ? &win : &preWin, startPos,
    //    forwardStep, tsCols, pSDataBlock->info.rows, numOfOutput);
1657 1658 1659
    preWin = win;

    int32_t prevEndPos = (forwardStep - 1) * step + startPos;
L
Liu Jicong 已提交
1660 1661
    //    startPos = getNextQualifiedWindow(pQueryAttr, &win, &pSDataBlock->info, tsCols, binarySearchForKey,
    //    prevEndPos);
1662 1663
    if (startPos < 0) {
      if ((ascQuery && win.skey <= pQueryAttr->window.ekey) || ((!ascQuery) && win.ekey >= pQueryAttr->window.ekey)) {
L
Liu Jicong 已提交
1664 1665 1666
        int32_t code =
            setResultOutputBufByKey(pRuntimeEnv, pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult,
                                    tableGroupId, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset);
1667 1668 1669 1670 1671 1672 1673
        if (code != TSDB_CODE_SUCCESS || pResult == NULL) {
          longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
        }

        startPos = pSDataBlock->info.rows - 1;

        // window start(end) key interpolation
L
Liu Jicong 已提交
1674 1675 1676
        //        doWindowBorderInterpolation(pOperatorInfo, pSDataBlock, pInfo->binfo.pCtx, pResult, &win, startPos,
        //        forwardStep); doApplyFunctions(pRuntimeEnv, pInfo->binfo.pCtx, ascQuery ? &win : &preWin, startPos,
        //        forwardStep, tsCols, pSDataBlock->info.rows, numOfOutput);
1677 1678 1679 1680 1681 1682 1683 1684
      }

      break;
    }
    setResultRowInterpo(pResult, RESULT_ROW_END_INTERP);
  }

  if (pQueryAttr->timeWindowInterpo) {
L
Liu Jicong 已提交
1685 1686
    int32_t rowIndex = ascQuery ? (pSDataBlock->info.rows - 1) : 0;
    //    saveDataBlockLastRow(pRuntimeEnv, &pSDataBlock->info, pSDataBlock->pDataBlock, rowIndex);
1687 1688
  }

L
Liu Jicong 已提交
1689
  //  updateResultRowInfoActiveIndex(pResultRowInfo, pQueryAttr, pRuntimeEnv->current->lastKey);
1690 1691
}

L
Liu Jicong 已提交
1692 1693
static bool groupKeyCompare(SGroupbyOperatorInfo* pInfo, SSDataBlock* pBlock, int32_t rowIndex,
                            int32_t numOfGroupCols) {
1694 1695 1696 1697 1698 1699 1700
  SColumnDataAgg* pColAgg = NULL;
  for (int32_t i = 0; i < numOfGroupCols; ++i) {
    SColumn*         pCol = taosArrayGet(pInfo->pGroupCols, i);
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, pCol->slotId);
    if (pBlock->pBlockAgg != NULL) {
      pColAgg = &pBlock->pBlockAgg[pCol->slotId];  // TODO is agg data matched?
    }
1701

1702
    bool isNull = colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg);
1703

1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726
    SGroupKeys* pkey = taosArrayGet(pInfo->pGroupColVals, i);
    if (pkey->isNull && isNull) {
      continue;
    }

    if (isNull || pkey->isNull) {
      return false;
    }

    char* val = colDataGetData(pColInfoData, rowIndex);

    if (IS_VAR_DATA_TYPE(pkey->type)) {
      int32_t len = varDataLen(val);
      if (len == varDataLen(pkey->pData) && memcmp(varDataVal(pkey->pData), varDataVal(val), len) == 0) {
        continue;
      } else {
        return false;
      }
    } else {
      if (memcmp(pkey->pData, val, pkey->bytes) != 0) {
        return false;
      }
    }
1727 1728
  }

1729 1730
  return true;
}
1731

1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760
static void keepGroupKeys(SGroupbyOperatorInfo* pInfo, SSDataBlock* pBlock, int32_t rowIndex, int32_t numOfGroupCols) {
  SColumnDataAgg* pColAgg = NULL;

  for (int32_t i = 0; i < numOfGroupCols; ++i) {
    SColumn*         pCol = taosArrayGet(pInfo->pGroupCols, i);
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, pCol->slotId);

    if (pBlock->pBlockAgg != NULL) {
      pColAgg = &pBlock->pBlockAgg[pCol->slotId];  // TODO is agg data matched?
    }

    SGroupKeys* pkey = taosArrayGet(pInfo->pGroupColVals, i);
    if (colDataIsNull(pColInfoData, pBlock->info.rows, rowIndex, pColAgg)) {
      pkey->isNull = true;
    } else {
      char* val = colDataGetData(pColInfoData, rowIndex);
      if (IS_VAR_DATA_TYPE(pkey->type)) {
        memcpy(pkey->pData, val, varDataTLen(val));
      } else {
        memcpy(pkey->pData, val, pkey->bytes);
      }
    }
  }
}

static int32_t generatedHashKey(void* pKey, int32_t* length, SArray* pGroupColVals) {
  ASSERT(pKey != NULL);
  size_t numOfGroupCols = taosArrayGetSize(pGroupColVals);

L
Liu Jicong 已提交
1761 1762 1763
  char* isNull = (char*)pKey;
  char* pStart = (char*)pKey + sizeof(int8_t) * numOfGroupCols;
  for (int32_t i = 0; i < numOfGroupCols; ++i) {
1764 1765 1766
    SGroupKeys* pkey = taosArrayGet(pGroupColVals, i);
    if (pkey->isNull) {
      isNull[i] = 1;
1767 1768 1769
      continue;
    }

1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780
    isNull[i] = 0;
    if (IS_VAR_DATA_TYPE(pkey->type)) {
      varDataCopy(pStart, pkey->pData);
      pStart += varDataTLen(pkey->pData);
      ASSERT(varDataTLen(pkey->pData) <= pkey->bytes);
    } else {
      memcpy(pStart, pkey->pData, pkey->bytes);
      pStart += pkey->bytes;
    }
  }

L
Liu Jicong 已提交
1781
  *length = (pStart - (char*)pKey);
1782 1783 1784
  return 0;
}

1785 1786
// assign the group keys or user input constant values if required
static void doAssignGroupKeys(SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t totalRows, int32_t rowIndex) {
L
Liu Jicong 已提交
1787
  for (int32_t i = 0; i < numOfOutput; ++i) {
1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804
    if (pCtx[i].functionId == -1) {
      SResultRowEntryInfo* pEntryInfo = GET_RES_INFO(&pCtx[i]);

      SColumnInfoData* pColInfoData = pCtx[i].input.pData[0];
      if (!colDataIsNull(pColInfoData, totalRows, rowIndex, NULL)) {
        char* dest = GET_ROWCELL_INTERBUF(pEntryInfo);
        char* data = colDataGetData(pColInfoData, rowIndex);

        // set result exists, todo refactor
        memcpy(dest, data, pColInfoData->info.bytes);
        pEntryInfo->hasResult = DATA_SET_FLAG;
        pEntryInfo->numOfRes = 1;
      }
    }
  }
}

L
Liu Jicong 已提交
1805 1806 1807
static void doHashGroupbyAgg(SOperatorInfo* pOperator, SSDataBlock* pBlock) {
  SExecTaskInfo*        pTaskInfo = pOperator->pTaskInfo;
  SGroupbyOperatorInfo* pInfo = pOperator->info;
1808

1809
  SqlFunctionCtx* pCtx = pInfo->binfo.pCtx;
L
Liu Jicong 已提交
1810 1811 1812 1813 1814
  int32_t         numOfGroupCols = taosArrayGetSize(pInfo->pGroupCols);
  //  if (type == TSDB_DATA_TYPE_FLOAT || type == TSDB_DATA_TYPE_DOUBLE) {
  // qError("QInfo:0x%"PRIx64" group by not supported on double/float columns, abort", GET_TASKID(pRuntimeEnv));
  //    return;
  //  }
1815

L
Liu Jicong 已提交
1816
  int32_t     len = 0;
1817
  STimeWindow w = TSWINDOW_INITIALIZER;
H
Haojun Liao 已提交
1818

1819 1820
  int32_t num = 0;
  for (int32_t j = 0; j < pBlock->info.rows; ++j) {
1821
    // Compare with the previous row of this column, and do not set the output buffer again if they are identical.
1822 1823 1824
    if (!pInfo->isInit) {
      keepGroupKeys(pInfo, pBlock, j, numOfGroupCols);
      pInfo->isInit = true;
1825 1826 1827 1828
      num++;
      continue;
    }

1829 1830 1831 1832
    bool equal = groupKeyCompare(pInfo, pBlock, j, numOfGroupCols);
    if (equal) {
      num++;
      continue;
1833 1834
    }

L
Liu Jicong 已提交
1835 1836 1837 1838
    /*int32_t ret = */ generatedHashKey(pInfo->keyBuf, &len, pInfo->pGroupColVals);
    int32_t ret =
        setGroupResultOutputBuf_rv(&(pInfo->binfo), pOperator->numOfOutput, pInfo->keyBuf, TSDB_DATA_TYPE_VARCHAR, len,
                                   0, pInfo->aggSup.pResultBuf, pTaskInfo, &pInfo->aggSup);
1839
    if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
H
Haojun Liao 已提交
1840
      longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
1841 1842
    }

1843 1844 1845 1846 1847
    int32_t rowIndex = j - num;
    doApplyFunctions(pCtx, &w, rowIndex, num, NULL, pBlock->info.rows, pOperator->numOfOutput, TSDB_ORDER_ASC);

    // assign the group keys or user input constant values if required
    doAssignGroupKeys(pCtx, pOperator->numOfOutput, pBlock->info.rows, rowIndex);
1848
    keepGroupKeys(pInfo, pBlock, j, numOfGroupCols);
1849 1850 1851 1852
    num = 1;
  }

  if (num > 0) {
L
Liu Jicong 已提交
1853 1854 1855 1856
    /*int32_t ret = */ generatedHashKey(pInfo->keyBuf, &len, pInfo->pGroupColVals);
    int32_t ret =
        setGroupResultOutputBuf_rv(&(pInfo->binfo), pOperator->numOfOutput, pInfo->keyBuf, TSDB_DATA_TYPE_VARCHAR, len,
                                   0, pInfo->aggSup.pResultBuf, pTaskInfo, &pInfo->aggSup);
H
Haojun Liao 已提交
1857 1858
    if (ret != TSDB_CODE_SUCCESS) {
      longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
1859 1860
    }

1861 1862 1863
    int32_t rowIndex = pBlock->info.rows - num;
    doApplyFunctions(pCtx, &w, rowIndex, num, NULL, pBlock->info.rows, pOperator->numOfOutput, TSDB_ORDER_ASC);
    doAssignGroupKeys(pCtx, pOperator->numOfOutput, pBlock->info.rows, rowIndex);
1864 1865 1866
  }
}

1867
// todo handle multiple tables cases.
L
Liu Jicong 已提交
1868
static void doSessionWindowAggImpl(SOperatorInfo* pOperator, SSessionAggOperatorInfo* pInfo, SSDataBlock* pBlock) {
H
Haojun Liao 已提交
1869
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
1870 1871

  // primary timestamp column
1872
  SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, 0);
1873

L
Liu Jicong 已提交
1874
  bool        masterScan = true;
1875
  STimeWindow window = {0};
L
Liu Jicong 已提交
1876 1877
  int32_t     numOfOutput = pOperator->numOfOutput;
  int64_t     gid = pBlock->info.groupId;
1878

H
Haojun Liao 已提交
1879
  int64_t gap = pInfo->gap;
1880
  pInfo->numOfRows = 0;
L
Liu Jicong 已提交
1881
  if (/*IS_REPEAT_SCAN(pRuntimeEnv) && */ !pInfo->reptScan) {
1882 1883 1884 1885 1886
    pInfo->reptScan = true;
    pInfo->prevTs = INT64_MIN;
  }

  TSKEY* tsList = (TSKEY*)pColInfoData->pData;
1887
  for (int32_t j = 0; j < pBlock->info.rows; ++j) {
1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904
    if (pInfo->prevTs == INT64_MIN) {
      pInfo->curWindow.skey = tsList[j];
      pInfo->curWindow.ekey = tsList[j];
      pInfo->prevTs = tsList[j];
      pInfo->numOfRows = 1;
      pInfo->start = j;
    } else if (tsList[j] - pInfo->prevTs <= gap && (tsList[j] - pInfo->prevTs) >= 0) {
      pInfo->curWindow.ekey = tsList[j];
      pInfo->prevTs = tsList[j];
      pInfo->numOfRows += 1;
      if (j == 0 && pInfo->start != 0) {
        pInfo->numOfRows = 1;
        pInfo->start = 0;
      }
    } else {  // start a new session window
      SResultRow* pResult = NULL;
      pInfo->curWindow.ekey = pInfo->curWindow.skey;
L
Liu Jicong 已提交
1905 1906 1907
      int32_t ret = setResultOutputBufByKey_rv(&pInfo->binfo.resultRowInfo, pBlock->info.uid, &window, masterScan,
                                               &pResult, gid, pInfo->binfo.pCtx, numOfOutput,
                                               pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup, pTaskInfo);
1908
      if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
1909
        longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
1910 1911
      }

1912
      // pInfo->numOfRows data belong to the current session window
L
Liu Jicong 已提交
1913 1914
      doApplyFunctions(pInfo->binfo.pCtx, &window, pInfo->start, pInfo->numOfRows, NULL, pBlock->info.rows, numOfOutput,
                       TSDB_ORDER_ASC);
1915 1916 1917 1918 1919 1920 1921 1922 1923 1924 1925 1926

      pInfo->curWindow.skey = tsList[j];
      pInfo->curWindow.ekey = tsList[j];
      pInfo->prevTs = tsList[j];
      pInfo->numOfRows = 1;
      pInfo->start = j;
    }
  }

  SResultRow* pResult = NULL;

  pInfo->curWindow.ekey = pInfo->curWindow.skey;
L
Liu Jicong 已提交
1927 1928 1929
  int32_t ret = setResultOutputBufByKey_rv(&pInfo->binfo.resultRowInfo, pBlock->info.uid, &window, masterScan, &pResult,
                                           gid, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset,
                                           &pInfo->aggSup, pTaskInfo);
1930
  if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
1931
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
1932 1933
  }

L
Liu Jicong 已提交
1934 1935
  doApplyFunctions(pInfo->binfo.pCtx, &window, pInfo->start, pInfo->numOfRows, NULL, pBlock->info.rows, numOfOutput,
                   TSDB_ORDER_ASC);
1936 1937 1938 1939 1940
}

static void setResultRowKey(SResultRow* pResultRow, char* pData, int16_t type) {
  if (IS_VAR_DATA_TYPE(type)) {
    if (pResultRow->key == NULL) {
wafwerar's avatar
wafwerar 已提交
1941
      pResultRow->key = taosMemoryMalloc(varDataTLen(pData));
1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954
      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);

    pResultRow->win.skey = v;
    pResultRow->win.ekey = v;
  }
}

L
Liu Jicong 已提交
1955 1956 1957 1958 1959
static int32_t setGroupResultOutputBuf_rv(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;
1960

L
Liu Jicong 已提交
1961 1962 1963
  SResultRow* pResultRow = doSetResultOutBufByKey_rv(pBuf, pResultRowInfo, groupId, (char*)pData, bytes, true, groupId,
                                                     pTaskInfo, true, pAggSup);
  assert(pResultRow != NULL);
1964 1965 1966

  setResultRowKey(pResultRow, pData, type);

H
Haojun Liao 已提交
1967
  setResultRowOutputBufInitCtx_rv(pBuf, pResultRow, pCtx, numOfCols, binfo->rowCellInfoOffset);
1968 1969 1970
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
1971
static int32_t getGroupbyColumnIndex(SGroupbyExpr* pGroupbyExpr, SSDataBlock* pDataBlock) {
1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992
  size_t num = taosArrayGetSize(pGroupbyExpr->columnInfo);
  for (int32_t k = 0; k < num; ++k) {
    SColIndex* pColIndex = taosArrayGet(pGroupbyExpr->columnInfo, k);
    if (TSDB_COL_IS_TAG(pColIndex->flag)) {
      continue;
    }

    int32_t colId = pColIndex->colId;

    for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
      SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
      if (pColInfo->info.colId == colId) {
        return i;
      }
    }
  }

  assert(0);
  return -1;
}

L
Liu Jicong 已提交
1993 1994
static bool functionNeedToExecute(SqlFunctionCtx* pCtx) {
  struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(pCtx);
1995 1996 1997

  // in case of timestamp column, always generated results.
  int32_t functionId = pCtx->functionId;
H
Haojun Liao 已提交
1998 1999 2000 2001
  if (functionId == -1) {
    return false;
  }

2002 2003 2004 2005 2006 2007 2008 2009 2010
  if (functionId == FUNCTION_TS) {
    return true;
  }

  if (isRowEntryCompleted(pResInfo) || functionId == FUNCTION_TAG_DUMMY || functionId == FUNCTION_TS_DUMMY) {
    return false;
  }

  if (functionId == FUNCTION_FIRST_DST || functionId == FUNCTION_FIRST) {
L
Liu Jicong 已提交
2011
    //    return QUERY_IS_ASC_QUERY(pQueryAttr);
2012 2013 2014 2015
  }

  // denote the order type
  if ((functionId == FUNCTION_LAST_DST || functionId == FUNCTION_LAST)) {
L
Liu Jicong 已提交
2016
    //    return pCtx->param[0].i == pQueryAttr->order.order;
2017 2018 2019
  }

  // in the reverse table scan, only the following functions need to be executed
L
Liu Jicong 已提交
2020 2021 2022 2023
  //  if (IS_REVERSE_SCAN(pRuntimeEnv) ||
  //      (pRuntimeEnv->scanFlag == REPEAT_SCAN && functionId != FUNCTION_STDDEV && functionId != FUNCTION_PERCT)) {
  //    return false;
  //  }
2024 2025 2026 2027

  return true;
}

L
Liu Jicong 已提交
2028 2029
void setBlockStatisInfo(SqlFunctionCtx* pCtx, SSDataBlock* pSDataBlock, SColumn* pColumn) {
  SColumnDataAgg* pAgg = NULL;
2030

H
Haojun Liao 已提交
2031 2032
  if (pSDataBlock->pBlockAgg != NULL && TSDB_COL_IS_NORMAL_COL(pColumn->flag)) {
    pAgg = &pSDataBlock->pBlockAgg[pCtx->columnIndex];
2033 2034

    pCtx->agg = *pAgg;
L
Liu Jicong 已提交
2035
    pCtx->isAggSet = true;
2036 2037 2038 2039 2040
    assert(pCtx->agg.numOfNull <= pSDataBlock->info.rows);
  } else {
    pCtx->isAggSet = false;
  }

H
Haojun Liao 已提交
2041
  pCtx->hasNull = hasNull(pColumn, pAgg);
2042 2043

  // set the statistics data for primary time stamp column
H
Haojun Liao 已提交
2044
  if (pCtx->functionId == FUNCTION_SPREAD && pColumn->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
L
Liu Jicong 已提交
2045
    pCtx->isAggSet = true;
2046 2047 2048 2049 2050 2051
    pCtx->agg.min = pSDataBlock->info.window.skey;
    pCtx->agg.max = pSDataBlock->info.window.ekey;
  }
}

// set the output buffer for the selectivity + tag query
L
Liu Jicong 已提交
2052
static int32_t setCtxTagColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
2053 2054 2055 2056 2057 2058 2059
  if (!isSelectivityWithTagsQuery(pCtx, numOfOutput)) {
    return TSDB_CODE_SUCCESS;
  }

  int32_t num = 0;
  int16_t tagLen = 0;

H
Haojun Liao 已提交
2060
  SqlFunctionCtx*  p = NULL;
wafwerar's avatar
wafwerar 已提交
2061
  SqlFunctionCtx** pTagCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
2062 2063 2064 2065 2066 2067 2068 2069
  if (pTagCtx == NULL) {
    return TSDB_CODE_QRY_OUT_OF_MEMORY;
  }

  for (int32_t i = 0; i < numOfOutput; ++i) {
    int32_t functionId = pCtx[i].functionId;

    if (functionId == FUNCTION_TAG_DUMMY || functionId == FUNCTION_TS_DUMMY) {
H
Haojun Liao 已提交
2070
      tagLen += pCtx[i].resDataInfo.bytes;
2071
      pTagCtx[num++] = &pCtx[i];
L
Liu Jicong 已提交
2072
    } else if (1 /*(aAggs[functionId].status & FUNCSTATE_SELECTIVITY) != 0*/) {
2073 2074 2075 2076 2077 2078 2079 2080 2081 2082
      p = &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
    }
  }
  if (p != NULL) {
2083 2084 2085
    p->subsidiaryRes.pCtx = pTagCtx;
    p->subsidiaryRes.numOfCols = num;
    p->subsidiaryRes.bufLen = tagLen;
2086
  } else {
wafwerar's avatar
wafwerar 已提交
2087
    taosMemoryFreeClear(pTagCtx);
2088 2089 2090 2091 2092
  }

  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
2093
static SqlFunctionCtx* createSqlFunctionCtx_rv(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowCellInfoOffset) {
L
Liu Jicong 已提交
2094
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
H
Haojun Liao 已提交
2095 2096 2097 2098
  if (pFuncCtx == NULL) {
    return NULL;
  }

wafwerar's avatar
wafwerar 已提交
2099
  *rowCellInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
H
Haojun Liao 已提交
2100
  if (*rowCellInfoOffset == 0) {
wafwerar's avatar
wafwerar 已提交
2101
    taosMemoryFreeClear(pFuncCtx);
H
Haojun Liao 已提交
2102 2103 2104 2105
    return NULL;
  }

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

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

2111
    pCtx->functionId = -1;
H
Haojun Liao 已提交
2112
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
2113
      SFuncExecEnv env = {0};
H
Haojun Liao 已提交
2114 2115
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;

2116 2117 2118 2119 2120 2121
      if (fmIsAggFunc(pCtx->functionId)) {
        fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
        pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
      } else {
        fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
      }
H
Haojun Liao 已提交
2122
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
H
Haojun Liao 已提交
2123 2124
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN) {
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR) {
H
Haojun Liao 已提交
2125
    }
H
Haojun Liao 已提交
2126

H
Haojun Liao 已提交
2127
    pCtx->input.numOfInputCols = pFunct->numOfParams;
wafwerar's avatar
wafwerar 已提交
2128 2129
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
H
Haojun Liao 已提交
2130

L
Liu Jicong 已提交
2131 2132 2133 2134 2135 2136
    pCtx->ptsOutputBuf = NULL;
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
    pCtx->resDataInfo.type = pFunct->resSchema.type;
    pCtx->order = TSDB_ORDER_ASC;
    pCtx->start.key = INT64_MIN;
    pCtx->end.key = INT64_MIN;
H
Haojun Liao 已提交
2137
#if 0
H
Haojun Liao 已提交
2138
    for (int32_t j = 0; j < pCtx->numOfParams; ++j) {
H
Haojun Liao 已提交
2139 2140
//      int16_t type = pFunct->param[j].nType;
//      int16_t bytes = pFunct->param[j].nLen;
H
Haojun Liao 已提交
2141

H
Haojun Liao 已提交
2142 2143 2144 2145 2146
//      if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
//        taosVariantCreateFromBinary(&pCtx->param[j], pFunct->param[j].pz, bytes, type);
//      } else {
//        taosVariantCreateFromBinary(&pCtx->param[j], (char *)&pFunct->param[j].i, bytes, type);
//      }
H
Haojun Liao 已提交
2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182
    }

    // 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_TS_COMP) {
//      pCtx->param[0].i = pQueryAttr->vgId;  //TODO this should be the parameter from client
      pCtx->param[0].nType = TSDB_DATA_TYPE_BIGINT;
    } 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 已提交
2183
#endif
H
Haojun Liao 已提交
2184 2185
  }

L
Liu Jicong 已提交
2186 2187 2188
  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 已提交
2189
  }
H
Haojun Liao 已提交
2190 2191 2192 2193 2194

  setCtxTagColumnInfo(pFuncCtx, numOfOutput);
  return pFuncCtx;
}

2195
static void* destroySqlFunctionCtx(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
2196 2197 2198 2199 2200 2201 2202 2203 2204 2205
  if (pCtx == NULL) {
    return NULL;
  }

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

    taosVariantDestroy(&pCtx[i].tag);
wafwerar's avatar
wafwerar 已提交
2206
    taosMemoryFreeClear(pCtx[i].subsidiaryRes.pCtx);
2207 2208
  }

wafwerar's avatar
wafwerar 已提交
2209
  taosMemoryFreeClear(pCtx);
2210 2211 2212
  return NULL;
}

L
Liu Jicong 已提交
2213 2214 2215 2216
static int32_t setupQueryRuntimeEnv(STaskRuntimeEnv* pRuntimeEnv, int32_t numOfTables, SArray* pOperator,
                                    void* merger) {
  // qDebug("QInfo:0x%"PRIx64" setup runtime env", GET_TASKID(pRuntimeEnv));
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
2217 2218 2219 2220

  pRuntimeEnv->prevGroupId = INT32_MIN;
  pRuntimeEnv->pQueryAttr = pQueryAttr;

L
Liu Jicong 已提交
2221 2222 2223 2224 2225 2226
  pRuntimeEnv->pResultRowHashTable =
      taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), true, HASH_NO_LOCK);
  pRuntimeEnv->pResultRowListSet =
      taosHashInit(numOfTables * 10, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
  pRuntimeEnv->keyBuf = taosMemoryMalloc(pQueryAttr->maxTableColumnWidth + sizeof(int64_t) + POINTER_BYTES);
  //  pRuntimeEnv->pool    = initResultRowPool(getResultRowSize(pRuntimeEnv));
2227 2228
  pRuntimeEnv->pResultRowArrayList = taosArrayInit(numOfTables, sizeof(SResultRowCell));

wafwerar's avatar
wafwerar 已提交
2229
  pRuntimeEnv->prevRow = taosMemoryMalloc(POINTER_BYTES * pQueryAttr->numOfCols + pQueryAttr->srcRowSize);
L
Liu Jicong 已提交
2230
  pRuntimeEnv->tagVal = taosMemoryMalloc(pQueryAttr->tagLen);
2231 2232

  // NOTE: pTableCheckInfo need to update the query time range and the lastKey info
L
Liu Jicong 已提交
2233 2234
  pRuntimeEnv->pTableRetrieveTsMap =
      taosHashInit(numOfTables, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), false, HASH_NO_LOCK);
2235

L
Liu Jicong 已提交
2236
  // pRuntimeEnv->scalarSup = createScalarFuncSupport(pQueryAttr->numOfOutput);
2237 2238

  if (pRuntimeEnv->scalarSup == NULL || pRuntimeEnv->pResultRowHashTable == NULL || pRuntimeEnv->keyBuf == NULL ||
L
Liu Jicong 已提交
2239
      pRuntimeEnv->prevRow == NULL || pRuntimeEnv->tagVal == NULL) {
2240 2241 2242 2243
    goto _clean;
  }

  if (pQueryAttr->numOfCols) {
L
Liu Jicong 已提交
2244
    char* start = POINTER_BYTES * pQueryAttr->numOfCols + (char*)pRuntimeEnv->prevRow;
2245
    pRuntimeEnv->prevRow[0] = start;
L
Liu Jicong 已提交
2246 2247
    for (int32_t i = 1; i < pQueryAttr->numOfCols; ++i) {
      pRuntimeEnv->prevRow[i] = pRuntimeEnv->prevRow[i - 1] + pQueryAttr->tableCols[i - 1].bytes;
2248 2249 2250
    }

    if (pQueryAttr->tableCols[0].type == TSDB_DATA_TYPE_TIMESTAMP) {
L
Liu Jicong 已提交
2251
      *(int64_t*)pRuntimeEnv->prevRow[0] = INT64_MIN;
2252 2253 2254
    }
  }

L
Liu Jicong 已提交
2255
  // qDebug("QInfo:0x%"PRIx64" init runtime environment completed", GET_TASKID(pRuntimeEnv));
2256 2257 2258 2259 2260 2261

  // group by normal column, sliding window query, interval query are handled by interval query processor
  // interval (down sampling operation)
  return TSDB_CODE_SUCCESS;

_clean:
L
Liu Jicong 已提交
2262
  // destroyScalarFuncSupport(pRuntimeEnv->scalarSup, pRuntimeEnv->pQueryAttr->numOfOutput);
wafwerar's avatar
wafwerar 已提交
2263 2264 2265 2266
  taosMemoryFreeClear(pRuntimeEnv->pResultRowHashTable);
  taosMemoryFreeClear(pRuntimeEnv->keyBuf);
  taosMemoryFreeClear(pRuntimeEnv->prevRow);
  taosMemoryFreeClear(pRuntimeEnv->tagVal);
2267 2268 2269 2270

  return TSDB_CODE_QRY_OUT_OF_MEMORY;
}

H
Haojun Liao 已提交
2271 2272
static void doFreeQueryHandle(STaskRuntimeEnv* pRuntimeEnv) {
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
2273

L
Liu Jicong 已提交
2274
  //  tsdbCleanupReadHandle(pRuntimeEnv->pTsdbReadHandle);
2275
  pRuntimeEnv->pTsdbReadHandle = NULL;
2276

L
Liu Jicong 已提交
2277 2278
  //  SMemRef* pMemRef = &pQueryAttr->memRef;
  //  assert(pMemRef->ref == 0 && pMemRef->snapshot.imem == NULL && pMemRef->snapshot.mem == NULL);
2279 2280
}

L
Liu Jicong 已提交
2281 2282 2283
static void destroyTsComp(STaskRuntimeEnv* pRuntimeEnv, STaskAttr* pQueryAttr) {
  if (pQueryAttr->tsCompQuery && pRuntimeEnv->outputBuf && pRuntimeEnv->outputBuf->pDataBlock &&
      taosArrayGetSize(pRuntimeEnv->outputBuf->pDataBlock) > 0) {
2284 2285
    SColumnInfoData* pColInfoData = taosArrayGet(pRuntimeEnv->outputBuf->pDataBlock, 0);
    if (pColInfoData) {
L
Liu Jicong 已提交
2286
      TdFilePtr pFile = *(TdFilePtr*)pColInfoData->pData;  // TODO refactor
2287 2288
      if (pFile != NULL) {
        taosCloseFile(&pFile);
L
Liu Jicong 已提交
2289
        *(TdFilePtr*)pColInfoData->pData = NULL;
2290 2291 2292 2293 2294
      }
    }
  }
}

L
Liu Jicong 已提交
2295
bool isTaskKilled(SExecTaskInfo* pTaskInfo) {
2296 2297
  // query has been executed more than tsShellActivityTimer, and the retrieve has not arrived
  // abort current query execution.
L
Liu Jicong 已提交
2298 2299
  if (pTaskInfo->owner != 0 &&
      ((taosGetTimestampSec() - pTaskInfo->cost.start / 1000) > 10 * getMaximumIdleDurationSec())
2300 2301
      /*(!needBuildResAfterQueryComplete(pTaskInfo))*/) {
    assert(pTaskInfo->cost.start != 0);
L
Liu Jicong 已提交
2302 2303 2304
    //    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;
2305 2306 2307 2308 2309
  }

  return false;
}

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

L
Liu Jicong 已提交
2312 2313 2314 2315
// static bool isFixedOutputQuery(STaskAttr* pQueryAttr) {
//   if (QUERY_IS_INTERVAL_QUERY(pQueryAttr)) {
//     return false;
//   }
2316
//
L
Liu Jicong 已提交
2317 2318 2319 2320
//   // Note:top/bottom query is fixed output query
//   if (pQueryAttr->topBotQuery || pQueryAttr->groupbyColumn || pQueryAttr->tsCompQuery) {
//     return true;
//   }
2321
//
L
Liu Jicong 已提交
2322 2323
//   for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
//     SExprBasicInfo *pExpr = &pQueryAttr->pExpr1[i].base;
2324
//
L
Liu Jicong 已提交
2325 2326 2327
//     if (pExpr->functionId == FUNCTION_TS || pExpr->functionId == FUNCTION_TS_DUMMY) {
//       continue;
//     }
2328
//
L
Liu Jicong 已提交
2329 2330 2331 2332
//     if (!IS_MULTIOUTPUT(aAggs[pExpr->functionId].status)) {
//       return true;
//     }
//   }
2333
//
L
Liu Jicong 已提交
2334 2335
//   return false;
// }
2336 2337

// todo refactor with isLastRowQuery
L
Liu Jicong 已提交
2338
// bool isPointInterpoQuery(STaskAttr *pQueryAttr) {
2339 2340 2341 2342 2343 2344 2345 2346 2347 2348
//  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
//    int32_t functionId = pQueryAttr->pExpr1[i].base.functionId;
//    if (functionId == FUNCTION_INTERP) {
//      return true;
//    }
//  }
//
//  return false;
//}

L
Liu Jicong 已提交
2349
static bool isFirstLastRowQuery(STaskAttr* pQueryAttr) {
2350 2351 2352 2353 2354 2355 2356 2357 2358 2359
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
    int32_t functionID = getExprFunctionId(&pQueryAttr->pExpr1[i]);
    if (functionID == FUNCTION_LAST_ROW) {
      return true;
    }
  }

  return false;
}

L
Liu Jicong 已提交
2360
static bool isCachedLastQuery(STaskAttr* pQueryAttr) {
2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389
  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;
  }

  if (pQueryAttr->order.order != TSDB_ORDER_DESC || !TSWINDOW_IS_EQUAL(pQueryAttr->window, TSWINDOW_DESC_INITIALIZER)) {
    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 已提交
2390 2391 2392
// todo refactor : return window
void getAlignQueryTimeWindow(SInterval* pInterval, int32_t precision, int64_t key, int64_t keyFirst, int64_t keyLast,
                             STimeWindow* win) {
2393
  ASSERT(key >= keyFirst && key <= keyLast);
H
Haojun Liao 已提交
2394
  win->skey = taosTimeTruncate(key, pInterval, precision);
2395 2396

  /*
H
Haojun Liao 已提交
2397
   * if the realSkey > INT64_MAX - pInterval->interval, the query duration between
2398 2399
   * realSkey and realEkey must be less than one interval.Therefore, no need to adjust the query ranges.
   */
H
Haojun Liao 已提交
2400 2401
  if (keyFirst > (INT64_MAX - pInterval->interval)) {
    assert(keyLast - keyFirst < pInterval->interval);
2402 2403
    win->ekey = INT64_MAX;
  } else {
2404
    win->ekey = taosTimeAdd(win->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
2405 2406 2407 2408 2409 2410
  }
}

/*
 * todo add more parameters to check soon..
 */
L
Liu Jicong 已提交
2411
bool colIdCheck(STaskAttr* pQueryAttr, uint64_t qId) {
2412 2413 2414
  // load data column information is incorrect
  for (int32_t i = 0; i < pQueryAttr->numOfCols - 1; ++i) {
    if (pQueryAttr->tableCols[i].colId == pQueryAttr->tableCols[i + 1].colId) {
L
Liu Jicong 已提交
2415
      // qError("QInfo:0x%"PRIx64" invalid data load column for query", qId);
2416 2417 2418 2419 2420 2421 2422 2423 2424
      return false;
    }
  }

  return true;
}

// todo ignore the avg/sum/min/max/count/stddev/top/bottom functions, of which
// the scan order is not matter
L
Liu Jicong 已提交
2425
static bool onlyOneQueryType(STaskAttr* pQueryAttr, int32_t functId, int32_t functIdDst) {
2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
    int32_t functionId = getExprFunctionId(&pQueryAttr->pExpr1[i]);

    if (functionId == FUNCTION_TS || functionId == FUNCTION_TS_DUMMY || functionId == FUNCTION_TAG ||
        functionId == FUNCTION_TAG_DUMMY) {
      continue;
    }

    if (functionId != functId && functionId != functIdDst) {
      return false;
    }
  }

  return true;
}

L
Liu Jicong 已提交
2442 2443 2444
static bool onlyFirstQuery(STaskAttr* pQueryAttr) {
  return onlyOneQueryType(pQueryAttr, FUNCTION_FIRST, FUNCTION_FIRST_DST);
}
2445

L
Liu Jicong 已提交
2446 2447 2448
static bool onlyLastQuery(STaskAttr* pQueryAttr) {
  return onlyOneQueryType(pQueryAttr, FUNCTION_LAST, FUNCTION_LAST_DST);
}
2449

L
Liu Jicong 已提交
2450 2451 2452
static bool notContainSessionOrStateWindow(STaskAttr* pQueryAttr) {
  return !(pQueryAttr->sw.gap > 0 || pQueryAttr->stateWindow);
}
2453

L
Liu Jicong 已提交
2454
static int32_t updateBlockLoadStatus(STaskAttr* pQuery, int32_t status) {
2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477
  bool hasFirstLastFunc = false;
  bool hasOtherFunc = false;

  if (status == BLK_DATA_ALL_NEEDED || status == BLK_DATA_DISCARD) {
    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;
    }
  }

  if (hasFirstLastFunc && status == BLK_DATA_NO_NEEDED) {
L
Liu Jicong 已提交
2478
    if (!hasOtherFunc) {
2479 2480 2481 2482 2483 2484 2485 2486 2487
      return BLK_DATA_DISCARD;
    } else {
      return BLK_DATA_ALL_NEEDED;
    }
  }

  return status;
}

H
Haojun Liao 已提交
2488
static void doUpdateLastKey(STaskAttr* pQueryAttr) {
2489 2490 2491
  STimeWindow* win = &pQueryAttr->window;

  size_t num = taosArrayGetSize(pQueryAttr->tableGroupInfo.pGroupList);
L
Liu Jicong 已提交
2492
  for (int32_t i = 0; i < num; ++i) {
2493 2494 2495
    SArray* p1 = taosArrayGetP(pQueryAttr->tableGroupInfo.pGroupList, i);

    size_t len = taosArrayGetSize(p1);
L
Liu Jicong 已提交
2496 2497 2498 2499 2500 2501 2502
    for (int32_t j = 0; j < len; ++j) {
      //      STableKeyInfo* pInfo = taosArrayGet(p1, j);
      //
      //      // update the new lastkey if it is equalled to the value of the old skey
      //      if (pInfo->lastKey == win->ekey) {
      //        pInfo->lastKey = win->skey;
      //      }
2503 2504 2505 2506
    }
  }
}

L
Liu Jicong 已提交
2507 2508
// static void updateDataCheckOrder(SQInfo *pQInfo, SQueryTableReq* pQueryMsg, bool stableQuery) {
//   STaskAttr* pQueryAttr = pQInfo->runtimeEnv.pQueryAttr;
H
Haojun Liao 已提交
2509
//
L
Liu Jicong 已提交
2510 2511 2512 2513
//   // 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 已提交
2514
//
L
Liu Jicong 已提交
2515 2516 2517 2518 2519
//   // 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 已提交
2520
//
L
Liu Jicong 已提交
2521 2522 2523 2524
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     if (pQueryAttr->window.skey > pQueryAttr->window.ekey) {
//       TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//     }
H
Haojun Liao 已提交
2525
//
L
Liu Jicong 已提交
2526 2527 2528
//     pQueryAttr->needReverseScan = false;
//     return;
//   }
H
Haojun Liao 已提交
2529
//
L
Liu Jicong 已提交
2530 2531 2532 2533 2534
//   if (pQueryAttr->groupbyColumn && pQueryAttr->order.order == TSDB_ORDER_DESC) {
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     if (pQueryAttr->window.skey > pQueryAttr->window.ekey) {
//       TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//     }
H
Haojun Liao 已提交
2535
//
L
Liu Jicong 已提交
2536 2537 2538 2539
//     pQueryAttr->needReverseScan = false;
//     doUpdateLastKey(pQueryAttr);
//     return;
//   }
H
Haojun Liao 已提交
2540
//
L
Liu Jicong 已提交
2541 2542 2543 2544 2545 2546
//   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 已提交
2547
//
L
Liu Jicong 已提交
2548 2549 2550
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     return;
//   }
H
Haojun Liao 已提交
2551
//
L
Liu Jicong 已提交
2552 2553 2554 2555
//   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 已提交
2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581
////               pQueryAttr->window.ekey, pQueryAttr->window.ekey, pQueryAttr->window.skey);
//
//        TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//        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);
//
//        TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//        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 已提交
2582 2583
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
2584 2585 2586 2587 2588 2589 2590 2591 2592 2593
//
//          TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//          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 已提交
2594 2595
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606
//
//          TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//          doUpdateLastKey(pQueryAttr);
//        }
//
//        pQueryAttr->order.order = TSDB_ORDER_DESC;
//        pQueryAttr->needReverseScan = false;
//      }
//    }
//  }
//}
2607

H
Haojun Liao 已提交
2608 2609
static void getIntermediateBufInfo(STaskRuntimeEnv* pRuntimeEnv, int32_t* ps, int32_t* rowsize) {
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
L
Liu Jicong 已提交
2610
  int32_t    MIN_ROWS_PER_PAGE = 4;
2611

L
Liu Jicong 已提交
2612 2613
  *rowsize = (int32_t)(pQueryAttr->resultRowSize *
                       getRowNumForMultioutput(pQueryAttr, pQueryAttr->topBotQuery, pQueryAttr->stableQuery));
2614 2615 2616 2617
  int32_t overhead = sizeof(SFilePage);

  // one page contains at least two rows
  *ps = DEFAULT_INTERN_BUF_PAGE_SIZE;
L
Liu Jicong 已提交
2618
  while (((*rowsize) * MIN_ROWS_PER_PAGE) > (*ps) - overhead) {
2619 2620 2621 2622 2623 2624
    *ps = ((*ps) << 1u);
  }
}

#define IS_PREFILTER_TYPE(_t) ((_t) != TSDB_DATA_TYPE_BINARY && (_t) != TSDB_DATA_TYPE_NCHAR)

L
Liu Jicong 已提交
2625 2626 2627
// static FORCE_INLINE bool doFilterByBlockStatistics(STaskRuntimeEnv* pRuntimeEnv, SDataStatis *pDataStatis,
// SqlFunctionCtx *pCtx, int32_t numOfRows) {
//   STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
2628
//
L
Liu Jicong 已提交
2629 2630 2631
//   if (pDataStatis == NULL || pQueryAttr->pFilters == NULL) {
//     return true;
//   }
2632
//
L
Liu Jicong 已提交
2633 2634
//   return filterRangeExecute(pQueryAttr->pFilters, pDataStatis, pQueryAttr->numOfCols, numOfRows);
// }
2635

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

dengyihao's avatar
dengyihao 已提交
2639 2640
  TSKEY sk = TMIN(pQueryAttr->window.skey, pQueryAttr->window.ekey);
  TSKEY ek = TMAX(pQueryAttr->window.skey, pQueryAttr->window.ekey);
2641 2642

  if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
L
Liu Jicong 已提交
2643
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.skey, sk, ek, &w);
2644 2645 2646 2647 2648 2649
    assert(w.ekey >= pBlockInfo->window.skey);

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

L
Liu Jicong 已提交
2650 2651
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
2652 2653 2654 2655 2656 2657 2658 2659 2660 2661
      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 已提交
2662
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.ekey, sk, ek, &w);
2663 2664 2665 2666 2667 2668
    assert(w.skey <= pBlockInfo->window.ekey);

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

L
Liu Jicong 已提交
2669 2670
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681 2682 2683 2684
      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;
}

L
Liu Jicong 已提交
2685
static int32_t doTSJoinFilter(STaskRuntimeEnv* pRuntimeEnv, TSKEY key, bool ascQuery) {
2686 2687 2688
  STSElem elem = tsBufGetElem(pRuntimeEnv->pTsBuf);

#if defined(_DEBUG_VIEW)
L
Liu Jicong 已提交
2689 2690
  printf("elem in comp ts file:%" PRId64 ", key:%" PRId64 ", tag:%" PRIu64
         ", query order:%d, ts order:%d, traverse:%d, index:%d\n",
2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723
         elem.ts, key, elem.tag.i, pQueryAttr->order.order, pRuntimeEnv->pTsBuf->tsOrder,
         pRuntimeEnv->pTsBuf->cur.order, pRuntimeEnv->pTsBuf->cur.tsIndex);
#endif

  if (ascQuery) {
    if (key < elem.ts) {
      return TS_JOIN_TS_NOT_EQUALS;
    } else if (key > elem.ts) {
      longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_INCONSISTAN);
    }
  } else {
    if (key > elem.ts) {
      return TS_JOIN_TS_NOT_EQUALS;
    } else if (key < elem.ts) {
      longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_INCONSISTAN);
    }
  }

  return TS_JOIN_TS_EQUAL;
}

bool doFilterDataBlock(SSingleColumnFilterInfo* pFilterInfo, int32_t numOfFilterCols, int32_t numOfRows, int8_t* p) {
  bool all = true;

  for (int32_t i = 0; i < numOfRows; ++i) {
    bool qualified = false;

    for (int32_t k = 0; k < numOfFilterCols; ++k) {
      char* pElem = (char*)pFilterInfo[k].pData + pFilterInfo[k].info.bytes * i;

      qualified = false;
      for (int32_t j = 0; j < pFilterInfo[k].numOfFilters; ++j) {
        SColumnFilterElem* pFilterElem = NULL;
L
Liu Jicong 已提交
2724
        //        SColumnFilterElem* pFilterElem = &pFilterInfo[k].pFilters[j];
2725 2726 2727

        bool isnull = isNull(pElem, pFilterInfo[k].info.type);
        if (isnull) {
L
Liu Jicong 已提交
2728 2729 2730 2731 2732 2733
          //          if (pFilterElem->fp == isNullOperator) {
          //            qualified = true;
          //            break;
          //          } else {
          //            continue;
          //          }
2734
        } else {
L
Liu Jicong 已提交
2735 2736 2737 2738 2739 2740
          //          if (pFilterElem->fp == notNullOperator) {
          //            qualified = true;
          //            break;
          //          } else if (pFilterElem->fp == isNullOperator) {
          //            continue;
          //          }
2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804 2805 2806 2807 2808 2809 2810 2811
        }

        if (pFilterElem->fp(pFilterElem, pElem, pElem, pFilterInfo[k].info.type)) {
          qualified = true;
          break;
        }
      }

      if (!qualified) {
        break;
      }
    }

    p[i] = qualified ? 1 : 0;
    if (!qualified) {
      all = false;
    }
  }

  return all;
}

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

H
Haojun Liao 已提交
2812
void filterRowsInDataBlock(STaskRuntimeEnv* pRuntimeEnv, SSingleColumnFilterInfo* pFilterInfo, int32_t numOfFilterCols,
2813 2814 2815
                           SSDataBlock* pBlock, bool ascQuery) {
  int32_t numOfRows = pBlock->info.rows;

L
Liu Jicong 已提交
2816
  int8_t* p = taosMemoryCalloc(numOfRows, sizeof(int8_t));
2817
  bool    all = true;
H
Haojun Liao 已提交
2818
#if 0
2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845
  if (pRuntimeEnv->pTsBuf != NULL) {
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, 0);

    TSKEY* k = (TSKEY*) pColInfoData->pData;
    for (int32_t i = 0; i < numOfRows; ++i) {
      int32_t offset = ascQuery? i:(numOfRows - i - 1);
      int32_t ret = doTSJoinFilter(pRuntimeEnv, k[offset], ascQuery);
      if (ret == TS_JOIN_TAG_NOT_EQUALS) {
        break;
      } else if (ret == TS_JOIN_TS_NOT_EQUALS) {
        all = false;
        continue;
      } else {
        assert(ret == TS_JOIN_TS_EQUAL);
        p[offset] = true;
      }

      if (!tsBufNextPos(pRuntimeEnv->pTsBuf)) {
        break;
      }
    }

    // save the cursor status
    pRuntimeEnv->current->cur = tsBufGetCursor(pRuntimeEnv->pTsBuf);
  } else {
    all = doFilterDataBlock(pFilterInfo, numOfFilterCols, numOfRows, p);
  }
H
Haojun Liao 已提交
2846
#endif
2847 2848 2849 2850 2851

  if (!all) {
    doCompactSDataBlock(pBlock, numOfRows, p);
  }

wafwerar's avatar
wafwerar 已提交
2852
  taosMemoryFreeClear(p);
2853 2854
}

H
Haojun Liao 已提交
2855
void filterColRowsInDataBlock(STaskRuntimeEnv* pRuntimeEnv, SSDataBlock* pBlock, bool ascQuery) {
L
Liu Jicong 已提交
2856
  int32_t numOfRows = pBlock->info.rows;
2857

L
Liu Jicong 已提交
2858 2859
  int8_t* p = NULL;
  bool    all = true;
2860

L
Liu Jicong 已提交
2861 2862 2863 2864 2865 2866 2867 2868 2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882
  if (pRuntimeEnv->pTsBuf != NULL) {
    SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, 0);
    p = taosMemoryCalloc(numOfRows, sizeof(int8_t));

    TSKEY* k = (TSKEY*)pColInfoData->pData;
    for (int32_t i = 0; i < numOfRows; ++i) {
      int32_t offset = ascQuery ? i : (numOfRows - i - 1);
      int32_t ret = doTSJoinFilter(pRuntimeEnv, k[offset], ascQuery);
      if (ret == TS_JOIN_TAG_NOT_EQUALS) {
        break;
      } else if (ret == TS_JOIN_TS_NOT_EQUALS) {
        all = false;
        continue;
      } else {
        assert(ret == TS_JOIN_TS_EQUAL);
        p[offset] = true;
      }

      if (!tsBufNextPos(pRuntimeEnv->pTsBuf)) {
        break;
      }
    }
2883

L
Liu Jicong 已提交
2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895 2896 2897 2898
    // save the cursor status
    //   pRuntimeEnv->current->cur = tsBufGetCursor(pRuntimeEnv->pTsBuf);
  } else {
    //   all = filterExecute(pRuntimeEnv->pQueryAttr->pFilters, numOfRows, &p, pBlock->pBlockAgg,
    //   pRuntimeEnv->pQueryAttr->numOfCols);
  }

  if (!all) {
    if (p) {
      doCompactSDataBlock(pBlock, numOfRows, p);
    } else {
      pBlock->info.rows = 0;
      pBlock->pBlockAgg = NULL;  // clean the block statistics info
    }
  }
2899

L
Liu Jicong 已提交
2900
  taosMemoryFreeClear(p);
2901 2902 2903
}

static SColumnInfo* doGetTagColumnInfoById(SColumnInfo* pTagColList, int32_t numOfTags, int16_t colId);
L
Liu Jicong 已提交
2904
static void         doSetTagValueInParam(void* pTable, int32_t tagColId, SVariant* tag, int16_t type, int16_t bytes);
2905 2906

static uint32_t doFilterByBlockTimeWindow(STableScanInfo* pTableScanInfo, SSDataBlock* pBlock) {
H
Haojun Liao 已提交
2907
  SqlFunctionCtx* pCtx = pTableScanInfo->pCtx;
L
Liu Jicong 已提交
2908
  uint32_t        status = BLK_DATA_NO_NEEDED;
2909 2910 2911 2912

  int32_t numOfOutput = pTableScanInfo->numOfOutput;
  for (int32_t i = 0; i < numOfOutput; ++i) {
    int32_t functionId = pCtx[i].functionId;
H
Haojun Liao 已提交
2913
    int32_t colId = pTableScanInfo->pExpr[i].base.pParam[0].pCol->colId;
2914 2915 2916 2917 2918 2919

    // group by + first/last should not apply the first/last block filter
    if (functionId < 0) {
      status |= BLK_DATA_ALL_NEEDED;
      return status;
    } else {
L
Liu Jicong 已提交
2920 2921 2922 2923
      //      status |= aAggs[functionId].dataReqFunc(&pTableScanInfo->pCtx[i], &pBlock->info.window, colId);
      //      if ((status & BLK_DATA_ALL_NEEDED) == BLK_DATA_ALL_NEEDED) {
      //        return status;
      //      }
2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943
    }
  }

  return status;
}

void doSetFilterColumnInfo(SSingleColumnFilterInfo* pFilterInfo, int32_t numOfFilterCols, SSDataBlock* pBlock) {
  // set the initial static data value filter expression
  for (int32_t i = 0; i < numOfFilterCols; ++i) {
    for (int32_t j = 0; j < pBlock->info.numOfCols; ++j) {
      SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, j);

      if (pFilterInfo[i].info.colId == pColInfo->info.colId) {
        pFilterInfo[i].pData = pColInfo->pData;
        break;
      }
    }
  }
}

L
Liu Jicong 已提交
2944
int32_t loadDataBlock(SExecTaskInfo* pTaskInfo, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock, uint32_t* status) {
H
Haojun Liao 已提交
2945 2946 2947 2948 2949 2950 2951 2952
  STaskCostInfo* pCost = &pTaskInfo->cost;

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

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

H
Haojun Liao 已提交
2953 2954
  *status = BLK_DATA_ALL_NEEDED;

H
Haojun Liao 已提交
2955 2956
  SArray* pCols = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
  if (pCols == NULL) {
H
Haojun Liao 已提交
2957 2958
    return terrno;
  }
H
Haojun Liao 已提交
2959

H
Haojun Liao 已提交
2960 2961 2962
  int32_t numOfCols = pBlock->info.numOfCols;
  for (int32_t i = 0; i < numOfCols; ++i) {
    SColumnInfoData* p = taosArrayGet(pCols, i);
L
Liu Jicong 已提交
2963
    SColMatchInfo*   pColMatchInfo = taosArrayGet(pTableScanInfo->pColMatchInfo, i);
H
Haojun Liao 已提交
2964 2965
    if (!pColMatchInfo->output) {
      continue;
H
Haojun Liao 已提交
2966 2967
    }

H
Haojun Liao 已提交
2968 2969 2970 2971 2972
    ASSERT(pColMatchInfo->colId == p->info.colId);
    taosArraySet(pBlock->pDataBlock, pColMatchInfo->targetSlotId, p);
  }

  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
2973 2974
}

L
Liu Jicong 已提交
2975 2976
int32_t loadDataBlockOnDemand(SExecTaskInfo* pTaskInfo, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock,
                              uint32_t* status) {
2977 2978
  *status = BLK_DATA_NO_NEEDED;

H
Haojun Liao 已提交
2979
  pBlock->pDataBlock = NULL;
L
Liu Jicong 已提交
2980
  pBlock->pBlockAgg = NULL;
H
Haojun Liao 已提交
2981

L
Liu Jicong 已提交
2982 2983
  //  int64_t groupId = pRuntimeEnv->current->groupIndex;
  //  bool    ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
2984

H
Haojun Liao 已提交
2985
  STaskCostInfo* pCost = &pTaskInfo->cost;
2986 2987 2988

  pCost->totalBlocks += 1;
  pCost->totalRows += pBlock->info.rows;
H
Haojun Liao 已提交
2989
#if 0
2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013
  if (pRuntimeEnv->pTsBuf != NULL) {
    (*status) = BLK_DATA_ALL_NEEDED;

    if (pQueryAttr->stableQuery) {  // todo refactor
      SExprInfo*   pExprInfo = &pTableScanInfo->pExpr[0];
      int16_t      tagId = (int16_t)pExprInfo->base.param[0].i;
      SColumnInfo* pColInfo = doGetTagColumnInfoById(pQueryAttr->tagColList, pQueryAttr->numOfTags, tagId);

      // compare tag first
      SVariant t = {0};
      doSetTagValueInParam(pRuntimeEnv->current->pTable, tagId, &t, pColInfo->type, pColInfo->bytes);
      setTimestampListJoinInfo(pRuntimeEnv, &t, pRuntimeEnv->current);

      STSElem elem = tsBufGetElem(pRuntimeEnv->pTsBuf);
      if (!tsBufIsValidElem(&elem) || (tsBufIsValidElem(&elem) && (taosVariantCompare(&t, elem.tag) != 0))) {
        (*status) = BLK_DATA_DISCARD;
        return TSDB_CODE_SUCCESS;
      }
    }
  }

  // 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 已提交
3014
      (QUERY_IS_INTERVAL_QUERY(pQueryAttr) && overlapWithTimeWindow(pTaskInfo, &pBlock->info))) {
3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026
    (*status) = BLK_DATA_ALL_NEEDED;
  }

  // check if this data block is required to load
  if ((*status) != BLK_DATA_ALL_NEEDED) {
    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 已提交
3027
      bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055
      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 {
      (*status) = BLK_DATA_ALL_NEEDED;
    }
  }

  SDataBlockInfo* pBlockInfo = &pBlock->info;
H
Haojun Liao 已提交
3056
//  *status = updateBlockLoadStatus(pRuntimeEnv->pQueryAttr, *status);
3057 3058 3059 3060 3061 3062 3063 3064

  if ((*status) == BLK_DATA_NO_NEEDED || (*status) == BLK_DATA_DISCARD) {
    //qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId, pBlockInfo->window.skey,
//           pBlockInfo->window.ekey, pBlockInfo->rows);
    pCost->discardBlocks += 1;
  } else if ((*status) == BLK_DATA_STATIS_NEEDED) {
    // this function never returns error?
    pCost->loadBlockStatis += 1;
3065
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
3066 3067

    if (pBlock->pBlockAgg == NULL) {  // data block statistics does not exist, load data block
3068
//      pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
3069 3070 3071 3072 3073 3074 3075
      pCost->totalCheckedRows += pBlock->info.rows;
    }
  } else {
    assert((*status) == BLK_DATA_ALL_NEEDED);

    // load the data block statistics to perform further filter
    pCost->loadBlockStatis += 1;
3076
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
3077 3078 3079 3080 3081 3082

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

H
Haojun Liao 已提交
3083
          bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121
          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
            pCost->discardBlocks += 1;
            //qDebug("QInfo:0x%"PRIx64" data block discard, brange:%" PRId64 "-%" PRId64 ", rows:%d", pQInfo->qId,
//                   pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
            (*status) = BLK_DATA_DISCARD;
            return TSDB_CODE_SUCCESS;
          }
        }
      }
    }

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

    pCost->totalCheckedRows += pBlockInfo->rows;
    pCost->loadBlocks += 1;
3122
//    pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134
//    if (pBlock->pDataBlock == NULL) {
//      return terrno;
//    }

//    if (pQueryAttr->pFilters != NULL) {
//      filterSetColFieldData(pQueryAttr->pFilters, pBlock->info.numOfCols, pBlock->pDataBlock);
//    }
    
//    if (pQueryAttr->pFilters != NULL || pRuntimeEnv->pTsBuf != NULL) {
//      filterColRowsInDataBlock(pRuntimeEnv, pBlock, ascQuery);
//    }
  }
H
Haojun Liao 已提交
3135
#endif
3136 3137 3138
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
3139
int32_t binarySearchForKey(char* pValue, int num, TSKEY key, int order) {
3140 3141 3142 3143 3144 3145 3146 3147 3148
  int32_t midPos = -1;
  int32_t numOfRows;

  if (num <= 0) {
    return -1;
  }

  assert(order == TSDB_ORDER_ASC || order == TSDB_ORDER_DESC);

L
Liu Jicong 已提交
3149
  TSKEY*  keyList = (TSKEY*)pValue;
3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202
  int32_t firstPos = 0;
  int32_t lastPos = num - 1;

  if (order == TSDB_ORDER_DESC) {
    // find the first position which is smaller than the key
    while (1) {
      if (key >= keyList[lastPos]) return lastPos;
      if (key == keyList[firstPos]) return firstPos;
      if (key < keyList[firstPos]) return firstPos - 1;

      numOfRows = lastPos - firstPos + 1;
      midPos = (numOfRows >> 1) + firstPos;

      if (key < keyList[midPos]) {
        lastPos = midPos - 1;
      } else if (key > keyList[midPos]) {
        firstPos = midPos + 1;
      } else {
        break;
      }
    }

  } else {
    // find the first position which is bigger than the key
    while (1) {
      if (key <= keyList[firstPos]) return firstPos;
      if (key == keyList[lastPos]) return lastPos;

      if (key > keyList[lastPos]) {
        lastPos = lastPos + 1;
        if (lastPos >= num)
          return -1;
        else
          return lastPos;
      }

      numOfRows = lastPos - firstPos + 1;
      midPos = (numOfRows >> 1u) + firstPos;

      if (key < keyList[midPos]) {
        lastPos = midPos - 1;
      } else if (key > keyList[midPos]) {
        firstPos = midPos + 1;
      } else {
        break;
      }
    }
  }

  return midPos;
}

/*
H
Haojun Liao 已提交
3203
 * set tag value in SqlFunctionCtx
3204 3205
 * e.g.,tag information into input buffer
 */
L
Liu Jicong 已提交
3206
static void doSetTagValueInParam(void* pTable, int32_t tagColId, SVariant* tag, int16_t type, int16_t bytes) {
3207 3208 3209
  taosVariantDestroy(tag);

  char* val = NULL;
L
Liu Jicong 已提交
3210 3211 3212 3213 3214 3215
  //  if (tagColId == TSDB_TBNAME_COLUMN_INDEX) {
  //    val = tsdbGetTableName(pTable);
  //    assert(val != NULL);
  //  } else {
  //    val = tsdbGetTableTagVal(pTable, tagColId, type, bytes);
  //  }
3216 3217 3218 3219 3220 3221 3222 3223

  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 已提交
3224
    int32_t len = (varDataLen(val) > maxLen) ? maxLen : varDataLen(val);
3225
    taosVariantCreateFromBinary(tag, varDataVal(val), len, type);
L
Liu Jicong 已提交
3226
    // taosVariantCreateFromBinary(tag, varDataVal(val), varDataLen(val), type);
3227 3228 3229 3230 3231 3232 3233 3234
  } 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 已提交
3235
  for (int32_t i = 0; i < numOfTags; ++i) {
3236 3237 3238 3239 3240 3241 3242 3243
    if (pTagColList[i].colId == colId) {
      return &pTagColList[i];
    }
  }

  return NULL;
}

L
Liu Jicong 已提交
3244
void setTagValue(SOperatorInfo* pOperatorInfo, void* pTable, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
H
Haojun Liao 已提交
3245
  STaskRuntimeEnv* pRuntimeEnv = pOperatorInfo->pRuntimeEnv;
3246

L
Liu Jicong 已提交
3247 3248
  SExprInfo* pExpr = pOperatorInfo->pExpr;
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
3249 3250

  SExprInfo* pExprInfo = &pExpr[0];
L
Liu Jicong 已提交
3251
  int32_t    functionId = getExprFunctionId(pExprInfo);
3252 3253 3254 3255

  if (pQueryAttr->numOfOutput == 1 && functionId == FUNCTION_TS_COMP && pQueryAttr->stableQuery) {
    assert(pExprInfo->base.numOfParams == 1);

L
Liu Jicong 已提交
3256
    //    int16_t      tagColId = (int16_t)pExprInfo->base.param[0].i;
H
Haojun Liao 已提交
3257
    int16_t      tagColId = -1;
3258 3259 3260
    SColumnInfo* pColInfo = doGetTagColumnInfoById(pQueryAttr->tagColList, pQueryAttr->numOfTags, tagColId);

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

3262 3263 3264 3265 3266 3267 3268 3269 3270
  } 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 已提交
3271
      if (!TSDB_COL_IS_TAG(pLocalExprInfo->base.pParam[0].pCol->flag)) {
3272 3273 3274 3275
        continue;
      }

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

L
Liu Jicong 已提交
3279 3280 3281
      if (IS_NUMERIC_TYPE(pLocalExprInfo->base.resSchema.type) ||
          pLocalExprInfo->base.resSchema.type == TSDB_DATA_TYPE_BOOL ||
          pLocalExprInfo->base.resSchema.type == TSDB_DATA_TYPE_TIMESTAMP) {
3282 3283 3284 3285
        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 已提交
3286
        }
3287 3288 3289 3290 3291
      }

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

L
Liu Jicong 已提交
3292
    // todo : use index to avoid iterator all possible output columns
3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303
    if (pQueryAttr->stableQuery && pQueryAttr->stabledev && (pRuntimeEnv->prevResult != NULL)) {
      setParamForStableStddev(pRuntimeEnv, pCtx, numOfOutput, pExprInfo);
    }
  }

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

H
Haojun Liao 已提交
3304
void copyToSDataBlock(SSDataBlock* pBlock, int32_t* offset, SGroupResInfo* pGroupResInfo, SDiskbasedBuf* pResBuf) {
3305 3306 3307 3308 3309 3310 3311
  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 已提交
3312 3313 3314
      //      if ((code = mergeIntoGroupResult(&pGroupResInfo, pRuntimeEnv, offset)) != TSDB_CODE_SUCCESS) {
      return;
      //      }
3315 3316
    }

L
Liu Jicong 已提交
3317
    //    doCopyToSDataBlock(pResBuf, pGroupResInfo, TSDB_ORDER_ASC, pBlock, );
3318 3319 3320 3321 3322 3323 3324 3325 3326 3327

    // 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 已提交
3328 3329 3330
    //    if (pBlock->info.rows >= threshold) {
    //      break;
    //    }
3331 3332 3333
  }
}

L
Liu Jicong 已提交
3334
static void updateTableQueryInfoForReverseScan(STableQueryInfo* pTableQueryInfo) {
3335 3336 3337 3338
  if (pTableQueryInfo == NULL) {
    return;
  }

L
Liu Jicong 已提交
3339 3340
  //  TSWAP(pTableQueryInfo->win.skey, pTableQueryInfo->win.ekey, TSKEY);
  //  pTableQueryInfo->lastKey = pTableQueryInfo->win.skey;
3341

L
Liu Jicong 已提交
3342 3343
  //  SWITCH_ORDER(pTableQueryInfo->cur.order);
  //  pTableQueryInfo->cur.vgroupIndex = -1;
3344 3345 3346 3347 3348 3349 3350 3351 3352 3353

  // set the index to be the end slot of result rows array
  SResultRowInfo* pResultRowInfo = &pTableQueryInfo->resInfo;
  if (pResultRowInfo->size > 0) {
    pResultRowInfo->curPos = pResultRowInfo->size - 1;
  } else {
    pResultRowInfo->curPos = -1;
  }
}

H
Haojun Liao 已提交
3354 3355
static void setupQueryRangeForReverseScan(STableScanInfo* pTableScanInfo) {
#if 0
3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373
  int32_t numOfGroups = (int32_t)(GET_NUM_OF_TABLEGROUP(pRuntimeEnv));
  for(int32_t i = 0; i < numOfGroups; ++i) {
    SArray *group = GET_TABLEGROUP(pRuntimeEnv, i);
    SArray *tableKeyGroup = taosArrayGetP(pQueryAttr->tableGroupInfo.pGroupList, i);

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

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

H
Haojun Liao 已提交
3377
void switchCtxOrder(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
3378 3379 3380 3381 3382
  for (int32_t i = 0; i < numOfOutput; ++i) {
    SWITCH_ORDER(pCtx[i].order);
  }
}

3383
// TODO fix this bug.
L
Liu Jicong 已提交
3384
int32_t initResultRow(SResultRow* pResultRow) {
3385 3386 3387 3388 3389 3390 3391
  pResultRow->pEntryInfo = (struct SResultRowEntryInfo*)((char*)pResultRow + sizeof(SResultRow));
  return TSDB_CODE_SUCCESS;
}

/*
 * 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 已提交
3392 3393 3394
 * +------------+-----------------result column 1------------+------------------result column 2-----------+
 * | SResultRow | SResultRowEntryInfo | intermediate buffer1 | SResultRowEntryInfo | intermediate buffer 2|
 * +------------+--------------------------------------------+--------------------------------------------+
3395 3396
 *           offset[0]                                  offset[1]                                   offset[2]
 */
3397
// TODO refactor: some function move away
H
Haojun Liao 已提交
3398
void setFunctionResultOutput(SOptrBasicInfo* pInfo, SAggSupporter* pSup, int32_t stage, SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
3399 3400 3401
  SqlFunctionCtx* pCtx = pInfo->pCtx;
  SSDataBlock*    pDataBlock = pInfo->pRes;
  int32_t*        rowCellInfoOffset = pInfo->rowCellInfoOffset;
H
Haojun Liao 已提交
3402 3403
  SResultRowInfo* pResultRowInfo = &pInfo->resultRowInfo;

L
Liu Jicong 已提交
3404 3405 3406 3407
  int64_t     tid = 0;
  int64_t     groupId = 0;
  SResultRow* pRow = doSetResultOutBufByKey_rv(pSup->pResultBuf, pResultRowInfo, tid, (char*)&tid, sizeof(tid), true,
                                               groupId, pTaskInfo, false, pSup);
H
Haojun Liao 已提交
3408 3409 3410 3411 3412

  for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
    struct SResultRowEntryInfo* pEntry = getResultCell(pRow, i, rowCellInfoOffset);
    cleanupResultRowEntry(pEntry);

L
Liu Jicong 已提交
3413
    pCtx[i].resultInfo = pEntry;
H
Haojun Liao 已提交
3414 3415 3416
    pCtx[i].currentStage = stage;

    // set the timestamp output buffer for top/bottom/diff query
L
Liu Jicong 已提交
3417 3418 3419 3420
    //    int32_t fid = pCtx[i].functionId;
    //    if (fid == FUNCTION_TOP || fid == FUNCTION_BOTTOM || fid == FUNCTION_DIFF || fid == FUNCTION_DERIVATIVE) {
    //      if (i > 0) pCtx[i].ptsOutputBuf = pCtx[i-1].pOutput;
    //    }
H
Haojun Liao 已提交
3421 3422 3423 3424 3425
  }

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

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

L
Liu Jicong 已提交
3429
  int32_t newSize = pDataBlock->info.rows + numOfInputRows + 5;  // extra output buffer
3430
  if ((*bufCapacity) < newSize) {
L
Liu Jicong 已提交
3431 3432
    for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
      SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
3433

wafwerar's avatar
wafwerar 已提交
3434
      char* p = taosMemoryRealloc(pColInfo->pData, newSize * pColInfo->info.bytes);
3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446 3447
      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 已提交
3448
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
3449 3450 3451 3452 3453
    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 已提交
3454 3455 3456
    if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF ||
        functionId == FUNCTION_DERIVATIVE) {
      if (i > 0) pBInfo->pCtx[i].ptsOutputBuf = pBInfo->pCtx[i - 1].pOutput;
3457 3458 3459 3460
    }
  }
}

H
Haojun Liao 已提交
3461
void copyTsColoum(SSDataBlock* pRes, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
3462 3463
  bool    needCopyTs = false;
  int32_t tsNum = 0;
L
Liu Jicong 已提交
3464
  char*   src = NULL;
3465 3466 3467 3468
  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 已提交
3469 3470
      if (i > 0 && pCtx[i - 1].functionId == FUNCTION_TS_DUMMY) {
        SColumnInfoData* pColRes = taosArrayGet(pRes->pDataBlock, i - 1);  // find ts data
3471 3472
        src = pColRes->pData;
      }
L
Liu Jicong 已提交
3473
    } else if (functionId == FUNCTION_TS_DUMMY) {
3474 3475 3476 3477 3478 3479 3480 3481 3482 3483
      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 已提交
3484
    if (functionId == FUNCTION_TS_DUMMY) {
3485 3486 3487 3488 3489 3490
      SColumnInfoData* pColRes = taosArrayGet(pRes->pDataBlock, i);
      memcpy(pColRes->pData, src, pColRes->info.bytes * pRes->info.rows);
    }
  }
}

L
Liu Jicong 已提交
3491
void clearOutputBuf(SOptrBasicInfo* pBInfo, int32_t* bufCapacity) {
3492 3493 3494
  SSDataBlock* pDataBlock = pBInfo->pRes;

  for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
L
Liu Jicong 已提交
3495
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
3496 3497 3498 3499 3500 3501 3502 3503

    int32_t functionId = pBInfo->pCtx[i].functionId;
    if (functionId < 0) {
      memset(pBInfo->pCtx[i].pOutput, 0, pColInfo->info.bytes * (*bufCapacity));
    }
  }
}

H
Haojun Liao 已提交
3504
void initCtxOutputBuffer(SqlFunctionCtx* pCtx, int32_t size) {
3505 3506
  for (int32_t j = 0; j < size; ++j) {
    struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(&pCtx[j]);
H
Haojun Liao 已提交
3507
    if (isRowEntryInitialized(pResInfo) || pCtx[j].functionId == -1) {
3508 3509 3510
      continue;
    }

H
Haojun Liao 已提交
3511
    pCtx[j].fpSet.init(&pCtx[j], pCtx[j].resultInfo);
3512 3513 3514
  }
}

L
Liu Jicong 已提交
3515
void setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status) {
3516
  if (status == TASK_NOT_COMPLETED) {
H
Haojun Liao 已提交
3517
    pTaskInfo->status = status;
3518 3519
  } else {
    // QUERY_NOT_COMPLETED is not compatible with any other status, so clear its position first
3520
    CLEAR_QUERY_STATUS(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
3521
    pTaskInfo->status |= status;
3522 3523 3524
  }
}

L
Liu Jicong 已提交
3525 3526 3527 3528 3529 3530
static void setupEnvForReverseScan(STableScanInfo* pTableScanInfo, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
  //  if (pRuntimeEnv->pTsBuf) {
  //    SWITCH_ORDER(pRuntimeEnv->pTsBuf->cur.order);
  //    bool ret = tsBufNextPos(pRuntimeEnv->pTsBuf);
  //    assert(ret);
  //  }
3531 3532

  // reverse order time range
H
Haojun Liao 已提交
3533
  SET_REVERSE_SCAN_FLAG(pTableScanInfo);
L
Liu Jicong 已提交
3534
  //  setTaskStatus(pTableScanInfo, QUERY_NOT_COMPLETED);
3535 3536 3537

  switchCtxOrder(pCtx, numOfOutput);

H
Haojun Liao 已提交
3538 3539
  SWITCH_ORDER(pTableScanInfo->order);
  setupQueryRangeForReverseScan(pTableScanInfo);
3540

L
Liu Jicong 已提交
3541 3542
  pTableScanInfo->times = 1;
  pTableScanInfo->current = 0;
3543
  pTableScanInfo->reverseTimes = 0;
3544 3545
}

3546 3547
void finalizeQueryResult(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
  for (int32_t j = 0; j < numOfOutput; ++j) {
H
Haojun Liao 已提交
3548 3549 3550 3551
    if (pCtx[j].functionId == -1) {
      continue;
    }

3552 3553 3554 3555
    pCtx[j].fpSet.finalize(&pCtx[j]);
  }
}

L
Liu Jicong 已提交
3556 3557
void finalizeMultiTupleQueryResult(SqlFunctionCtx* pCtx, int32_t numOfOutput, SDiskbasedBuf* pBuf,
                                   SResultRowInfo* pResultRowInfo, int32_t* rowCellInfoOffset) {
3558 3559 3560
  for (int32_t i = 0; i < pResultRowInfo->size; ++i) {
    SResultRowPosition* pPos = &pResultRowInfo->pPosition[i];

L
Liu Jicong 已提交
3561
    SFilePage*  bufPage = getBufPage(pBuf, pPos->pageId);
3562 3563 3564 3565 3566
    SResultRow* pRow = (SResultRow*)((char*)bufPage + pPos->offset);
    if (!isResultRowClosed(pResultRowInfo, i)) {
      continue;
    }

3567
    for (int32_t j = 0; j < numOfOutput; ++j) {
3568 3569
      pCtx[j].resultInfo = getResultCell(pRow, j, rowCellInfoOffset);

H
Haojun Liao 已提交
3570
      struct SResultRowEntryInfo* pResInfo = pCtx[j].resultInfo;
3571 3572 3573 3574
      if (isRowEntryCompleted(pResInfo) && isRowEntryInitialized(pResInfo)) {
        continue;
      }

L
Liu Jicong 已提交
3575
      if (pCtx[j].fpSet.process) {  // TODO set the dummy function.
H
Haojun Liao 已提交
3576 3577
        pCtx[j].fpSet.finalize(&pCtx[j]);
      }
3578 3579 3580 3581

      if (pRow->numOfRows < pResInfo->numOfRes) {
        pRow->numOfRows = pResInfo->numOfRes;
      }
3582
    }
3583 3584 3585 3586 3587 3588

    releaseBufPage(pBuf, bufPage);
    /*
     * set the number of output results for group by normal columns, the number of output rows usually is 1 except
     * the top and bottom query
     */
L
Liu Jicong 已提交
3589
    //    buf->numOfRows = (uint16_t)getNumOfResult(pCtx, numOfOutput);
3590
  }
3591 3592
}

L
Liu Jicong 已提交
3593
static bool hasMainOutput(STaskAttr* pQueryAttr) {
3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
    int32_t functionId = getExprFunctionId(&pQueryAttr->pExpr1[i]);

    if (functionId != FUNCTION_TS && functionId != FUNCTION_TAG && functionId != FUNCTION_TAGPRJ) {
      return true;
    }
  }

  return false;
}

L
Liu Jicong 已提交
3605 3606
STableQueryInfo* createTableQueryInfo(void* buf, bool groupbyColumn, STimeWindow win) {
  STableQueryInfo* pTableQueryInfo = buf;
3607 3608 3609
  pTableQueryInfo->lastKey = win.skey;

  // set more initial size of interval/groupby query
L
Liu Jicong 已提交
3610 3611 3612 3613 3614 3615 3616 3617
  //  if (/*QUERY_IS_INTERVAL_QUERY(pQueryAttr) || */groupbyColumn) {
  int32_t initialSize = 128;
  int32_t code = initResultRowInfo(&pTableQueryInfo->resInfo, initialSize);
  if (code != TSDB_CODE_SUCCESS) {
    return NULL;
  }
  //  } else { // in other aggregate query, do not initialize the windowResInfo
  //  }
3618 3619 3620 3621 3622

  return pTableQueryInfo;
}

STableQueryInfo* createTmpTableQueryInfo(STimeWindow win) {
wafwerar's avatar
wafwerar 已提交
3623
  STableQueryInfo* pTableQueryInfo = taosMemoryCalloc(1, sizeof(STableQueryInfo));
3624

L
Liu Jicong 已提交
3625
  //  pTableQueryInfo->win = win;
3626 3627 3628 3629
  pTableQueryInfo->lastKey = win.skey;

  // set more initial size of interval/groupby query
  int32_t initialSize = 16;
3630
  int32_t code = initResultRowInfo(&pTableQueryInfo->resInfo, initialSize);
3631
  if (code != TSDB_CODE_SUCCESS) {
wafwerar's avatar
wafwerar 已提交
3632
    taosMemoryFreeClear(pTableQueryInfo);
3633 3634 3635 3636 3637 3638
    return NULL;
  }

  return pTableQueryInfo;
}

L
Liu Jicong 已提交
3639
void destroyTableQueryInfoImpl(STableQueryInfo* pTableQueryInfo) {
3640 3641 3642 3643
  if (pTableQueryInfo == NULL) {
    return;
  }

L
Liu Jicong 已提交
3644
  //  taosVariantDestroy(&pTableQueryInfo->tag);
3645 3646 3647
  cleanupResultRowInfo(&pTableQueryInfo->resInfo);
}

L
Liu Jicong 已提交
3648 3649
void setResultRowOutputBufInitCtx(STaskRuntimeEnv* pRuntimeEnv, SResultRow* pResult, SqlFunctionCtx* pCtx,
                                  int32_t numOfOutput, int32_t* rowCellInfoOffset) {
3650
  // Note: pResult->pos[i]->num == 0, there is only fixed number of results for each group
3651
  SFilePage* bufPage = getBufPage(pRuntimeEnv->pResultBuf, pResult->pageId);
3652 3653 3654 3655 3656 3657 3658

  int32_t offset = 0;
  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)) {
H
Haojun Liao 已提交
3659
      offset += pCtx[i].resDataInfo.bytes;
3660 3661 3662 3663
      continue;
    }

    pCtx[i].pOutput = getPosInResultPage(pRuntimeEnv->pQueryAttr, bufPage, pResult->offset, offset);
H
Haojun Liao 已提交
3664
    offset += pCtx[i].resDataInfo.bytes;
3665 3666 3667 3668 3669 3670 3671

    int32_t functionId = pCtx[i].functionId;
    if (functionId < 0) {
      continue;
    }

    if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF) {
L
Liu Jicong 已提交
3672
      if (i > 0) pCtx[i].ptsOutputBuf = pCtx[i - 1].pOutput;
3673 3674
    }

L
Liu Jicong 已提交
3675 3676 3677
    //    if (!pResInfo->initialized) {
    //      aAggs[functionId].init(&pCtx[i], pResInfo);
    //    }
3678 3679 3680
  }
}

L
Liu Jicong 已提交
3681 3682
void setResultRowOutputBufInitCtx_rv(SDiskbasedBuf* pBuf, SResultRow* pResult, SqlFunctionCtx* pCtx,
                                     int32_t numOfOutput, int32_t* rowCellInfoOffset) {
3683 3684 3685 3686 3687 3688 3689 3690
  // Note: pResult->pos[i]->num == 0, there is only fixed number of results for each group
  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;
    }
L
Liu Jicong 已提交
3691 3692 3693 3694 3695 3696 3697
    //    int32_t functionId = pCtx[i].functionId;
    //    if (functionId < 0) {
    //      continue;
    //    }
    //    if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF) {
    //      if (i > 0) pCtx[i].ptsOutputBuf = pCtx[i - 1].pOutput;
    //    }
3698

H
Haojun Liao 已提交
3699
    if (!pResInfo->initialized && pCtx[i].functionId != -1) {
3700
      pCtx[i].fpSet.init(&pCtx[i], pResInfo);
3701 3702 3703 3704
    }
  }
}

L
Liu Jicong 已提交
3705 3706
void doSetTableGroupOutputBuf(SAggOperatorInfo* pAggInfo, int32_t numOfOutput, int32_t tableGroupId,
                              SExecTaskInfo* pTaskInfo) {
3707 3708 3709 3710
  // for simple group by query without interval, all the tables belong to one group result.
  int64_t uid = 0;
  int64_t tid = 0;

3711 3712
  SResultRowInfo* pResultRowInfo = &pAggInfo->binfo.resultRowInfo;
  SqlFunctionCtx* pCtx = pAggInfo->binfo.pCtx;
L
Liu Jicong 已提交
3713
  int32_t*        rowCellInfoOffset = pAggInfo->binfo.rowCellInfoOffset;
3714

3715
  SResultRow* pResultRow =
L
Liu Jicong 已提交
3716 3717 3718
      doSetResultOutBufByKey_rv(pAggInfo->pResultBuf, pResultRowInfo, tid, (char*)&tableGroupId, sizeof(tableGroupId),
                                true, uid, pTaskInfo, false, &pAggInfo->aggSup);
  assert(pResultRow != NULL);
3719 3720 3721 3722 3723 3724

  /*
   * 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) {
L
Liu Jicong 已提交
3725 3726
    int32_t ret =
        addNewWindowResultBuf(pResultRow, pAggInfo->pResultBuf, tableGroupId, pAggInfo->binfo.pRes->info.rowSize);
3727 3728 3729 3730 3731
    if (ret != TSDB_CODE_SUCCESS) {
      return;
    }
  }

3732
  setResultRowOutputBufInitCtx_rv(pAggInfo->pResultBuf, pResultRow, pCtx, numOfOutput, rowCellInfoOffset);
3733 3734
}

L
Liu Jicong 已提交
3735 3736
void setExecutionContext(int32_t numOfOutput, int32_t tableGroupId, TSKEY nextKey, SExecTaskInfo* pTaskInfo,
                         STableQueryInfo* pTableQueryInfo, SAggOperatorInfo* pAggInfo) {
3737 3738
  // lastKey needs to be updated
  pTableQueryInfo->lastKey = nextKey;
3739
  if (pAggInfo->groupId != INT32_MIN && pAggInfo->groupId == tableGroupId) {
3740 3741 3742
    return;
  }

3743
  doSetTableGroupOutputBuf(pAggInfo, numOfOutput, tableGroupId, pTaskInfo);
3744 3745

  // record the current active group id
3746
  pAggInfo->groupId = tableGroupId;
3747 3748
}

L
Liu Jicong 已提交
3749 3750
void setResultOutputBuf(STaskRuntimeEnv* pRuntimeEnv, SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfCols,
                        int32_t* rowCellInfoOffset) {
3751
  // Note: pResult->pos[i]->num == 0, there is only fixed number of results for each group
L
Liu Jicong 已提交
3752
  SFilePage* page = getBufPage(pRuntimeEnv->pResultBuf, pResult->pageId);
3753 3754 3755 3756

  int16_t offset = 0;
  for (int32_t i = 0; i < numOfCols; ++i) {
    pCtx[i].pOutput = getPosInResultPage(pRuntimeEnv->pQueryAttr, page, pResult->offset, offset);
H
Haojun Liao 已提交
3757
    offset += pCtx[i].resDataInfo.bytes;
3758 3759

    int32_t functionId = pCtx[i].functionId;
L
Liu Jicong 已提交
3760 3761 3762
    if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF ||
        functionId == FUNCTION_DERIVATIVE) {
      if (i > 0) pCtx[i].ptsOutputBuf = pCtx[i - 1].pOutput;
3763 3764 3765 3766 3767 3768 3769 3770 3771 3772
    }

    /*
     * set the output buffer information and intermediate buffer,
     * not all queries require the interResultBuf, such as COUNT
     */
    pCtx[i].resultInfo = getResultCell(pResult, i, rowCellInfoOffset);
  }
}

H
Haojun Liao 已提交
3773
void setCtxTagForJoin(STaskRuntimeEnv* pRuntimeEnv, SqlFunctionCtx* pCtx, SExprInfo* pExprInfo, void* pTable) {
H
Haojun Liao 已提交
3774
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
3775

H
Haojun Liao 已提交
3776
  SExprBasicInfo* pExpr = &pExprInfo->base;
L
Liu Jicong 已提交
3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797 3798 3799 3800
  //  if (pQueryAttr->stableQuery && (pRuntimeEnv->pTsBuf != NULL) &&
  //      (pExpr->functionId == FUNCTION_TS || pExpr->functionId == FUNCTION_PRJ) &&
  //      (pExpr->colInfo.colIndex == PRIMARYKEY_TIMESTAMP_COL_ID)) {
  //    assert(pExpr->numOfParams == 1);
  //
  //    int16_t      tagColId = (int16_t)pExprInfo->base.param[0].i;
  //    SColumnInfo* pColInfo = doGetTagColumnInfoById(pQueryAttr->tagColList, pQueryAttr->numOfTags, tagColId);
  //
  //    doSetTagValueInParam(pTable, tagColId, &pCtx->tag, pColInfo->type, pColInfo->bytes);
  //
  //    int16_t tagType = pCtx[0].tag.nType;
  //    if (tagType == TSDB_DATA_TYPE_BINARY || tagType == TSDB_DATA_TYPE_NCHAR) {
  //      //qDebug("QInfo:0x%"PRIx64" set tag value for join comparison, colId:%" PRId64 ", val:%s",
  //      GET_TASKID(pRuntimeEnv),
  ////             pExprInfo->base.param[0].i, pCtx[0].tag.pz);
  //    } else {
  //      //qDebug("QInfo:0x%"PRIx64" set tag value for join comparison, colId:%" PRId64 ", val:%" PRId64,
  //      GET_TASKID(pRuntimeEnv),
  ////             pExprInfo->base.param[0].i, pCtx[0].tag.i);
  //    }
  //  }
}

int32_t setTimestampListJoinInfo(STaskRuntimeEnv* pRuntimeEnv, SVariant* pTag, STableQueryInfo* pTableQueryInfo) {
H
Haojun Liao 已提交
3801
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
3802 3803

  assert(pRuntimeEnv->pTsBuf != NULL);
H
Haojun Liao 已提交
3804
#if 0
3805 3806 3807 3808 3809 3810 3811 3812 3813
  // both the master and supplement scan needs to set the correct ts comp start position
  if (pTableQueryInfo->cur.vgroupIndex == -1) {
    taosVariantAssign(&pTableQueryInfo->tag, pTag);

    STSElem elem = tsBufGetElemStartPos(pRuntimeEnv->pTsBuf, pQueryAttr->vgId, &pTableQueryInfo->tag);

    // failed to find data with the specified tag value and vnodeId
    if (!tsBufIsValidElem(&elem)) {
      if (pTag->nType == TSDB_DATA_TYPE_BINARY || pTag->nType == TSDB_DATA_TYPE_NCHAR) {
3814
        //qError("QInfo:0x%"PRIx64" failed to find tag:%s in ts_comp", GET_TASKID(pRuntimeEnv), pTag->pz);
3815
      } else {
3816
        //qError("QInfo:0x%"PRIx64" failed to find tag:%" PRId64 " in ts_comp", GET_TASKID(pRuntimeEnv), pTag->i);
3817 3818 3819 3820 3821 3822 3823 3824
      }

      return -1;
    }

    // Keep the cursor info of current table
    pTableQueryInfo->cur = tsBufGetCursor(pRuntimeEnv->pTsBuf);
    if (pTag->nType == TSDB_DATA_TYPE_BINARY || pTag->nType == TSDB_DATA_TYPE_NCHAR) {
3825
      //qDebug("QInfo:0x%"PRIx64" find tag:%s start pos in ts_comp, blockIndex:%d, tsIndex:%d", GET_TASKID(pRuntimeEnv), pTag->pz, pTableQueryInfo->cur.blockIndex, pTableQueryInfo->cur.tsIndex);
3826
    } else {
3827
      //qDebug("QInfo:0x%"PRIx64" find tag:%"PRId64" start pos in ts_comp, blockIndex:%d, tsIndex:%d", GET_TASKID(pRuntimeEnv), pTag->i, pTableQueryInfo->cur.blockIndex, pTableQueryInfo->cur.tsIndex);
3828 3829 3830 3831 3832
    }

  } else {
    tsBufSetCursor(pRuntimeEnv->pTsBuf, &pTableQueryInfo->cur);
    if (pTag->nType == TSDB_DATA_TYPE_BINARY || pTag->nType == TSDB_DATA_TYPE_NCHAR) {
3833
      //qDebug("QInfo:0x%"PRIx64" find tag:%s start pos in ts_comp, blockIndex:%d, tsIndex:%d", GET_TASKID(pRuntimeEnv), pTag->pz, pTableQueryInfo->cur.blockIndex, pTableQueryInfo->cur.tsIndex);
3834
    } else {
3835
      //qDebug("QInfo:0x%"PRIx64" find tag:%"PRId64" start pos in ts_comp, blockIndex:%d, tsIndex:%d", GET_TASKID(pRuntimeEnv), pTag->i, pTableQueryInfo->cur.blockIndex, pTableQueryInfo->cur.tsIndex);
3836 3837
    }
  }
H
Haojun Liao 已提交
3838
#endif
3839 3840 3841 3842
  return 0;
}

// TODO refactor: this funciton should be merged with setparamForStableStddevColumnData function.
L
Liu Jicong 已提交
3843 3844
void setParamForStableStddev(STaskRuntimeEnv* pRuntimeEnv, SqlFunctionCtx* pCtx, int32_t numOfOutput,
                             SExprInfo* pExprInfo) {
3845
#if 0
H
Haojun Liao 已提交
3846
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
3847 3848 3849 3850 3851 3852 3853 3854

  int32_t numOfExprs = pQueryAttr->numOfOutput;
  for(int32_t i = 0; i < numOfExprs; ++i) {
    SExprInfo* pExprInfo1 = &(pExprInfo[i]);
    if (pExprInfo1->base.functionId != FUNCTION_STDDEV_DST) {
      continue;
    }

H
Haojun Liao 已提交
3855
    SExprBasicInfo* pExpr = &pExprInfo1->base;
3856 3857 3858 3859 3860 3861 3862 3863 3864 3865 3866 3867

    pCtx[i].param[0].arr = NULL;
    pCtx[i].param[0].nType = TSDB_DATA_TYPE_INT;  // avoid freeing the memory by setting the type to be int

    // TODO use hash to speedup this loop
    int32_t numOfGroup = (int32_t)taosArrayGetSize(pRuntimeEnv->prevResult);
    for (int32_t j = 0; j < numOfGroup; ++j) {
      SInterResult* p = taosArrayGet(pRuntimeEnv->prevResult, j);
      if (pQueryAttr->tagLen == 0 || memcmp(p->tags, pRuntimeEnv->tagVal, pQueryAttr->tagLen) == 0) {
        int32_t numOfCols = (int32_t)taosArrayGetSize(p->pResult);
        for (int32_t k = 0; k < numOfCols; ++k) {
          SStddevInterResult* pres = taosArrayGet(p->pResult, k);
3868
          if (pres->info.colId == pExpr->colInfo.colId) {
3869 3870 3871 3872 3873 3874 3875 3876 3877 3878
            pCtx[i].param[0].arr = pres->pResult;
            break;
          }
        }
      }
    }
  }
#endif
}

L
Liu Jicong 已提交
3879 3880
void setParamForStableStddevByColData(STaskRuntimeEnv* pRuntimeEnv, SqlFunctionCtx* pCtx, int32_t numOfOutput,
                                      SExprInfo* pExpr, char* val, int16_t bytes) {
H
Haojun Liao 已提交
3881
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
3882 3883 3884
#if 0
  int32_t numOfExprs = pQueryAttr->numOfOutput;
  for(int32_t i = 0; i < numOfExprs; ++i) {
H
Haojun Liao 已提交
3885
    SExprBasicInfo* pExpr1 = &pExpr[i].base;
3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900
    if (pExpr1->functionId != FUNCTION_STDDEV_DST) {
      continue;
    }

    pCtx[i].param[0].arr = NULL;
    pCtx[i].param[0].nType = TSDB_DATA_TYPE_INT;  // avoid freeing the memory by setting the type to be int

    // TODO use hash to speedup this loop
    int32_t numOfGroup = (int32_t)taosArrayGetSize(pRuntimeEnv->prevResult);
    for (int32_t j = 0; j < numOfGroup; ++j) {
      SInterResult* p = taosArrayGet(pRuntimeEnv->prevResult, j);
      if (bytes == 0 || memcmp(p->tags, val, bytes) == 0) {
        int32_t numOfCols = (int32_t)taosArrayGetSize(p->pResult);
        for (int32_t k = 0; k < numOfCols; ++k) {
          SStddevInterResult* pres = taosArrayGet(p->pResult, k);
3901
          if (pres->info.colId == pExpr1->colInfo.colId) {
3902 3903 3904 3905 3906 3907 3908 3909 3910 3911 3912 3913 3914
            pCtx[i].param[0].arr = pres->pResult;
            break;
          }
        }
      }
    }
  }
#endif
}

/*
 * There are two cases to handle:
 *
L
Liu Jicong 已提交
3915 3916
 * 1. Query range is not set yet (queryRangeSet = 0). we need to set the query range info, including
 * pQueryAttr->lastKey, pQueryAttr->window.skey, and pQueryAttr->eKey.
3917 3918 3919 3920
 * 2. Query range is set and query is in progress. There may be another result with the same query ranges to be
 *    merged during merge stage. In this case, we need the pTableQueryInfo->lastResRows to decide if there
 *    is a previous result generated or not.
 */
L
Liu Jicong 已提交
3921 3922 3923 3924
void setIntervalQueryRange(STaskRuntimeEnv* pRuntimeEnv, TSKEY key) {
  STaskAttr*       pQueryAttr = pRuntimeEnv->pQueryAttr;
  STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
  SResultRowInfo*  pResultRowInfo = &pTableQueryInfo->resInfo;
3925 3926 3927 3928 3929

  if (pResultRowInfo->curPos != -1) {
    return;
  }

L
Liu Jicong 已提交
3930
  //  pTableQueryInfo->win.skey = key;
3931 3932 3933 3934 3935 3936 3937 3938 3939 3940
  STimeWindow win = {.skey = key, .ekey = pQueryAttr->window.ekey};

  /**
   * In handling the both ascending and descending order super table query, we need to find the first qualified
   * timestamp of this table, and then set the first qualified start timestamp.
   * In ascending query, the key is the first qualified timestamp. However, in the descending order query, additional
   * operations involve.
   */
  STimeWindow w = TSWINDOW_INITIALIZER;

dengyihao's avatar
dengyihao 已提交
3941 3942
  TSKEY sk = TMIN(win.skey, win.ekey);
  TSKEY ek = TMAX(win.skey, win.ekey);
L
Liu Jicong 已提交
3943
  //  getAlignQueryTimeWindow(pQueryAttr, win.skey, sk, ek, &w);
3944

L
Liu Jicong 已提交
3945 3946 3947 3948 3949 3950 3951
  //  if (pResultRowInfo->prevSKey == TSKEY_INITIAL_VAL) {
  //    if (!QUERY_IS_ASC_QUERY(pQueryAttr)) {
  //      assert(win.ekey == pQueryAttr->window.ekey);
  //    }
  //
  //    pResultRowInfo->prevSKey = w.skey;
  //  }
3952

L
Liu Jicong 已提交
3953
  //  pTableQueryInfo->lastKey = pTableQueryInfo->win.skey;
3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964
}

/**
 * copyToOutputBuf support copy data in ascending/descending order
 * 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
 */
L
Liu Jicong 已提交
3965 3966
static int32_t doCopyToSDataBlock(SDiskbasedBuf* pBuf, SGroupResInfo* pGroupResInfo, int32_t orderType,
                                  SSDataBlock* pBlock, int32_t rowCapacity, int32_t* rowCellOffset) {
3967
  int32_t numOfRows = getNumOfTotalRes(pGroupResInfo);
L
Liu Jicong 已提交
3968
  int32_t numOfResult = pBlock->info.rows;  // there are already exists result rows
3969 3970 3971 3972

  int32_t start = 0;
  int32_t step = -1;

L
Liu Jicong 已提交
3973
  // qDebug("QInfo:0x%"PRIx64" start to copy data from windowResInfo to output buf", GET_TASKID(pRuntimeEnv));
3974 3975 3976 3977 3978 3979 3980 3981 3982 3983
  assert(orderType == TSDB_ORDER_ASC || orderType == TSDB_ORDER_DESC);

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

3984 3985
  int32_t nrows = pBlock->info.rows;

3986
  for (int32_t i = start; (i < numOfRows) && (i >= 0); i += step) {
3987
    SResultRowPosition* pPos = taosArrayGet(pGroupResInfo->pRows, i);
L
Liu Jicong 已提交
3988
    SFilePage*          page = getBufPage(pBuf, pPos->pageId);
3989 3990

    SResultRow* pRow = (SResultRow*)((char*)page + pPos->offset);
3991 3992 3993 3994 3995
    if (pRow->numOfRows == 0) {
      pGroupResInfo->index += 1;
      continue;
    }

3996
    // TODO copy multiple rows?
3997
    int32_t numOfRowsToCopy = pRow->numOfRows;
L
Liu Jicong 已提交
3998
    if (numOfResult + numOfRowsToCopy >= rowCapacity) {
3999 4000 4001 4002 4003 4004
      break;
    }

    pGroupResInfo->index += 1;

    for (int32_t j = 0; j < pBlock->info.numOfCols; ++j) {
L
Liu Jicong 已提交
4005
      SColumnInfoData*     pColInfoData = taosArrayGet(pBlock->pDataBlock, j);
4006
      SResultRowEntryInfo* pEntryInfo = getResultCell(pRow, j, rowCellOffset);
4007

4008 4009
      char* in = GET_ROWCELL_INTERBUF(pEntryInfo);
      colDataAppend(pColInfoData, nrows, in, pEntryInfo->numOfRes == 0);
4010 4011
    }

4012 4013 4014
    releaseBufPage(pBuf, page);
    nrows += 1;

4015
    numOfResult += numOfRowsToCopy;
H
Haojun Liao 已提交
4016
    if (numOfResult == rowCapacity) {  // output buffer is full
4017 4018 4019 4020
      break;
    }
  }

L
Liu Jicong 已提交
4021
  // qDebug("QInfo:0x%"PRIx64" copy data to query buf completed", GET_TASKID(pRuntimeEnv));
4022 4023 4024 4025
  pBlock->info.rows = numOfResult;
  return 0;
}

L
Liu Jicong 已提交
4026 4027
static void toSDatablock(SGroupResInfo* pGroupResInfo, SDiskbasedBuf* pBuf, SSDataBlock* pBlock, int32_t rowCapacity,
                         int32_t* rowCellOffset) {
4028 4029
  assert(pGroupResInfo->currentGroup <= pGroupResInfo->totalGroup);

4030
  blockDataCleanup(pBlock);
4031 4032 4033 4034
  if (!hasRemainDataInCurrentGroup(pGroupResInfo)) {
    return;
  }

L
Liu Jicong 已提交
4035 4036
  int32_t orderType =
      TSDB_ORDER_ASC;  //(pQueryAttr->pGroupbyExpr != NULL) ? pQueryAttr->pGroupbyExpr->orderType : TSDB_ORDER_ASC;
4037
  doCopyToSDataBlock(pBuf, pGroupResInfo, orderType, pBlock, rowCapacity, rowCellOffset);
4038

H
Haojun Liao 已提交
4039 4040
  // add condition (pBlock->info.rows >= 1) just to runtime happy
  blockDataUpdateTsWindow(pBlock);
4041 4042
}

L
Liu Jicong 已提交
4043 4044
static void updateNumOfRowsInResultRows(SqlFunctionCtx* pCtx, int32_t numOfOutput, SResultRowInfo* pResultRowInfo,
                                        int32_t* rowCellInfoOffset) {
4045
  // update the number of result for each, only update the number of rows for the corresponding window result.
L
Liu Jicong 已提交
4046 4047 4048
  //  if (QUERY_IS_INTERVAL_QUERY(pQueryAttr)) {
  //    return;
  //  }
4049 4050

  for (int32_t i = 0; i < pResultRowInfo->size; ++i) {
L
Liu Jicong 已提交
4051
    SResultRow* pResult = pResultRowInfo->pResult[i];
4052 4053 4054 4055 4056 4057 4058

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

4059 4060
      SResultRowEntryInfo* pCell = getResultCell(pResult, j, rowCellInfoOffset);
      pResult->numOfRows = (uint16_t)(TMAX(pResult->numOfRows, pCell->numOfRes));
4061 4062 4063 4064
    }
  }
}

L
Liu Jicong 已提交
4065
static int32_t compressQueryColData(SColumnInfoData* pColRes, int32_t numOfRows, char* data, int8_t compressed) {
4066 4067
  int32_t colSize = pColRes->info.bytes * numOfRows;
  return (*(tDataTypes[pColRes->info.type].compFunc))(pColRes->pData, colSize, numOfRows, data,
L
Liu Jicong 已提交
4068
                                                      colSize + COMP_OVERFLOW_BYTES, compressed, NULL, 0);
4069 4070
}

L
Liu Jicong 已提交
4071 4072 4073 4074 4075
int32_t doFillTimeIntervalGapsInResults(struct SFillInfo* pFillInfo, SSDataBlock* pOutput, int32_t capacity, void** p) {
  //  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);
  //  }
4076 4077 4078 4079 4080 4081 4082 4083 4084 4085

  int32_t numOfRows = (int32_t)taosFillResultDataBlock(pFillInfo, p, capacity - pOutput->info.rows);
  pOutput->info.rows += numOfRows;

  return pOutput->info.rows;
}

void publishOperatorProfEvent(SOperatorInfo* operatorInfo, EQueryProfEventType eventType) {
  SQueryProfEvent event = {0};

L
Liu Jicong 已提交
4086 4087
  event.eventType = eventType;
  event.eventTime = taosGetTimestampUs();
4088 4089 4090
  event.operatorType = operatorInfo->operatorType;

  if (operatorInfo->pRuntimeEnv) {
L
Liu Jicong 已提交
4091 4092 4093 4094
    //    SQInfo* pQInfo = operatorInfo->pRuntimeEnv->qinfo;
    //    if (pQInfo->summary.queryProfEvents) {
    //      taosArrayPush(pQInfo->summary.queryProfEvents, &event);
    //    }
4095 4096 4097
  }
}

L
Liu Jicong 已提交
4098
void publishQueryAbortEvent(SExecTaskInfo* pTaskInfo, int32_t code) {
4099 4100 4101 4102 4103
  SQueryProfEvent event;
  event.eventType = QUERY_PROF_QUERY_ABORT;
  event.eventTime = taosGetTimestampUs();
  event.abortCode = code;

4104 4105
  if (pTaskInfo->cost.queryProfEvents) {
    taosArrayPush(pTaskInfo->cost.queryProfEvents, &event);
4106 4107 4108
  }
}

L
Liu Jicong 已提交
4109
typedef struct {
4110 4111 4112 4113 4114 4115 4116
  uint8_t operatorType;
  int64_t beginTime;
  int64_t endTime;
  int64_t selfTime;
  int64_t descendantsTime;
} SOperatorStackItem;

L
Liu Jicong 已提交
4117 4118
static void doOperatorExecProfOnce(SOperatorStackItem* item, SQueryProfEvent* event, SArray* opStack,
                                   SHashObj* profResults) {
4119 4120 4121 4122 4123 4124 4125 4126
  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 已提交
4127
  uint8_t              operatorType = item->operatorType;
4128 4129 4130 4131 4132 4133 4134 4135 4136
  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 已提交
4137
    taosHashPut(profResults, &(operatorType), sizeof(operatorType), &opResult, sizeof(opResult));
4138 4139 4140 4141 4142
  }
}

void calculateOperatorProfResults(SQInfo* pQInfo) {
  if (pQInfo->summary.queryProfEvents == NULL) {
L
Liu Jicong 已提交
4143
    // qDebug("QInfo:0x%"PRIx64" query prof events array is null", pQInfo->qId);
4144 4145 4146 4147
    return;
  }

  if (pQInfo->summary.operatorProfResults == NULL) {
L
Liu Jicong 已提交
4148
    // qDebug("QInfo:0x%"PRIx64" operator prof results hash is null", pQInfo->qId);
4149 4150 4151 4152 4153 4154 4155 4156
    return;
  }

  SArray* opStack = taosArrayInit(32, sizeof(SOperatorStackItem));
  if (opStack == NULL) {
    return;
  }

L
Liu Jicong 已提交
4157
  size_t    size = taosArrayGetSize(pQInfo->summary.queryProfEvents);
4158 4159 4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171 4172 4173 4174 4175 4176 4177 4178 4179 4180 4181 4182
  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);
      }
    }
  }

  taosArrayDestroy(opStack);
}

L
Liu Jicong 已提交
4183 4184
void queryCostStatis(SExecTaskInfo* pTaskInfo) {
  STaskCostInfo* pSummary = &pTaskInfo->cost;
4185

L
Liu Jicong 已提交
4186 4187 4188
  //  uint64_t hashSize = taosHashGetMemSize(pQInfo->runtimeEnv.pResultRowHashTable);
  //  hashSize += taosHashGetMemSize(pRuntimeEnv->tableqinfoGroupInfo.map);
  //  pSummary->hashSize = hashSize;
4189 4190 4191 4192

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

L
Liu Jicong 已提交
4193 4194 4195 4196 4197 4198 4199 4200 4201 4202 4203 4204 4205 4206 4207 4208 4209 4210 4211 4212
  //  SResultRowPool* p = pTaskInfo->pool;
  //  if (p != NULL) {
  //    pSummary->winInfoSize = getResultRowPoolMemSize(p);
  //    pSummary->numOfTimeWindows = getNumOfAllocatedResultRows(p);
  //  } else {
  //    pSummary->winInfoSize = 0;
  //    pSummary->numOfTimeWindows = 0;
  //  }
  //
  //  calculateOperatorProfResults(pQInfo);

  qDebug("%s :cost summary: elapsed time:%" PRId64 " us, first merge:%" PRId64
         " us, total blocks:%d, "
         "load block statis:%d, load data block:%d, total rows:%" PRId64 ", check rows:%" PRId64,
         GET_TASKID(pTaskInfo), pSummary->elapsedTime, pSummary->firstStageMergeTime, pSummary->totalBlocks,
         pSummary->loadBlockStatis, pSummary->loadBlocks, pSummary->totalRows, pSummary->totalCheckedRows);
  //
  // 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);
4213 4214 4215 4216

  if (pSummary->operatorProfResults) {
    SOperatorProfResult* opRes = taosHashIterate(pSummary->operatorProfResults, NULL);
    while (opRes != NULL) {
L
Liu Jicong 已提交
4217 4218
      // qDebug("QInfo:0x%" PRIx64 " :cost summary: operator : %d, exec times: %" PRId64 ", self time: %" PRId64,
      //             pQInfo->qId, opRes->operatorType, opRes->sumRunTimes, opRes->sumSelfTime);
4219 4220 4221 4222 4223
      opRes = taosHashIterate(pSummary->operatorProfResults, opRes);
    }
  }
}

L
Liu Jicong 已提交
4224 4225 4226
// static void updateOffsetVal(STaskRuntimeEnv *pRuntimeEnv, SDataBlockInfo *pBlockInfo) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
4227
//
L
Liu Jicong 已提交
4228
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
4229
//
L
Liu Jicong 已提交
4230 4231 4232 4233
//   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;
//   }
4234
//
L
Liu Jicong 已提交
4235 4236 4237 4238 4239
//   if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//     pQueryAttr->pos = (int32_t)pQueryAttr->limit.offset;
//   } else {
//     pQueryAttr->pos = pBlockInfo->rows - (int32_t)pQueryAttr->limit.offset - 1;
//   }
4240
//
L
Liu Jicong 已提交
4241
//   assert(pQueryAttr->pos >= 0 && pQueryAttr->pos <= pBlockInfo->rows - 1);
4242
//
L
Liu Jicong 已提交
4243 4244
//   SArray *         pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//   SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
4245
//
L
Liu Jicong 已提交
4246 4247
//   // update the pQueryAttr->limit.offset value, and pQueryAttr->pos value
//   TSKEY *keys = (TSKEY *) pColInfoData->pData;
4248
//
L
Liu Jicong 已提交
4249 4250 4251
//   // update the offset value
//   pTableQueryInfo->lastKey = keys[pQueryAttr->pos];
//   pQueryAttr->limit.offset = 0;
4252
//
L
Liu Jicong 已提交
4253
//   int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
4254
//
L
Liu Jicong 已提交
4255 4256 4257 4258
//   //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);
// }
4259

L
Liu Jicong 已提交
4260 4261
// void skipBlocks(STaskRuntimeEnv *pRuntimeEnv) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
4262
//
L
Liu Jicong 已提交
4263 4264 4265
//   if (pQueryAttr->limit.offset <= 0 || pQueryAttr->numOfFilterCols > 0) {
//     return;
//   }
4266
//
L
Liu Jicong 已提交
4267 4268
//   pQueryAttr->pos = 0;
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
4269
//
L
Liu Jicong 已提交
4270 4271
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
//   TsdbQueryHandleT pTsdbReadHandle = pRuntimeEnv->pTsdbReadHandle;
4272
//
L
Liu Jicong 已提交
4273 4274 4275 4276 4277
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pTsdbReadHandle)) {
//     if (isTaskKilled(pRuntimeEnv->qinfo)) {
//       longjmp(pRuntimeEnv->env, TSDB_CODE_TSC_QUERY_CANCELLED);
//     }
4278
//
L
Liu Jicong 已提交
4279
//     tsdbRetrieveDataBlockInfo(pTsdbReadHandle, &blockInfo);
4280
//
L
Liu Jicong 已提交
4281 4282 4283 4284
//     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;
4285
//
L
Liu Jicong 已提交
4286 4287 4288 4289 4290 4291 4292
//       //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;
//     }
//   }
4293
//
L
Liu Jicong 已提交
4294 4295 4296 4297 4298 4299 4300 4301 4302
//   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;
4303
//
L
Liu Jicong 已提交
4304 4305 4306
//   assert(pQueryAttr->limit.offset == 0);
//   STimeWindow tw = *win;
//   getNextTimeWindow(pQueryAttr, &tw);
4307
//
L
Liu Jicong 已提交
4308 4309
//   if ((tw.skey <= pBlockInfo->window.ekey && QUERY_IS_ASC_QUERY(pQueryAttr)) ||
//       (tw.ekey >= pBlockInfo->window.skey && !QUERY_IS_ASC_QUERY(pQueryAttr))) {
4310
//
L
Liu Jicong 已提交
4311 4312 4313 4314
//     // 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);
4315
//
L
Liu Jicong 已提交
4316 4317 4318 4319
//     tw = *win;
//     int32_t startPos =
//         getNextQualifiedWindow(pQueryAttr, &tw, pBlockInfo, pColInfoData->pData, binarySearchForKey, -1);
//     assert(startPos >= 0);
4320
//
L
Liu Jicong 已提交
4321 4322
//     // set the abort info
//     pQueryAttr->pos = startPos;
4323
//
L
Liu Jicong 已提交
4324 4325 4326 4327
//     // reset the query start timestamp
//     pTableQueryInfo->win.skey = ((TSKEY *)pColInfoData->pData)[startPos];
//     pQueryAttr->window.skey = pTableQueryInfo->win.skey;
//     TSKEY key = pTableQueryInfo->win.skey;
4328
//
L
Liu Jicong 已提交
4329 4330
//     pWindowResInfo->prevSKey = tw.skey;
//     int32_t index = pRuntimeEnv->resultRowInfo.curIndex;
4331
//
L
Liu Jicong 已提交
4332 4333
//     int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
//     pRuntimeEnv->resultRowInfo.curIndex = index;  // restore the window index
4334
//
L
Liu Jicong 已提交
4335 4336 4337 4338
//     //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);
4339
//
L
Liu Jicong 已提交
4340 4341 4342 4343 4344
//     return key;
//   } else {  // do nothing
//     pQueryAttr->window.skey      = tw.skey;
//     pWindowResInfo->prevSKey = tw.skey;
//     pTableQueryInfo->lastKey = tw.skey;
4345
//
L
Liu Jicong 已提交
4346 4347
//     return tw.skey;
//   }
4348
//
L
Liu Jicong 已提交
4349 4350 4351 4352 4353 4354 4355 4356 4357 4358
//   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);
//   }
4359
//
L
Liu Jicong 已提交
4360 4361 4362 4363 4364
//   // 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;
//   }
4365
//
L
Liu Jicong 已提交
4366 4367 4368 4369 4370 4371 4372
//   /*
//    * 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);
4373
//
L
Liu Jicong 已提交
4374 4375
//   STimeWindow w = TSWINDOW_INITIALIZER;
//   bool ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
4376
//
L
Liu Jicong 已提交
4377 4378
//   SResultRowInfo *pWindowResInfo = &pRuntimeEnv->resultRowInfo;
//   STableQueryInfo *pTableQueryInfo = pRuntimeEnv->current;
4379
//
L
Liu Jicong 已提交
4380 4381 4382
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pRuntimeEnv->pTsdbReadHandle)) {
//     tsdbRetrieveDataBlockInfo(pRuntimeEnv->pTsdbReadHandle, &blockInfo);
4383
//
L
Liu Jicong 已提交
4384 4385 4386 4387 4388 4389 4390 4391 4392
//     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;
//     }
4393
//
L
Liu Jicong 已提交
4394 4395
//     // the first time window
//     STimeWindow win = getActiveTimeWindow(pWindowResInfo, pWindowResInfo->prevSKey, pQueryAttr);
4396
//
L
Liu Jicong 已提交
4397 4398
//     while (pQueryAttr->limit.offset > 0) {
//       STimeWindow tw = win;
4399
//
L
Liu Jicong 已提交
4400 4401 4402
//       if ((win.ekey <= blockInfo.window.ekey && ascQuery) || (win.ekey >= blockInfo.window.skey && !ascQuery)) {
//         pQueryAttr->limit.offset -= 1;
//         pWindowResInfo->prevSKey = win.skey;
4403
//
L
Liu Jicong 已提交
4404 4405 4406 4407 4408 4409
//         // 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;
//         }
//       }
4410
//
L
Liu Jicong 已提交
4411 4412 4413 4414
//       if (pQueryAttr->limit.offset == 0) {
//         *start = doSkipIntervalProcess(pRuntimeEnv, &win, &blockInfo, pTableQueryInfo);
//         return true;
//       }
4415
//
L
Liu Jicong 已提交
4416 4417
//       // current window does not ended in current data block, try next data block
//       getNextTimeWindow(pQueryAttr, &tw);
4418
//
L
Liu Jicong 已提交
4419 4420 4421 4422 4423 4424 4425 4426 4427
//       /*
//        * 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)) {
4428
//
L
Liu Jicong 已提交
4429 4430
//         SArray *pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//         SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
4431
//
L
Liu Jicong 已提交
4432 4433 4434
//         if ((win.ekey > blockInfo.window.ekey && ascQuery) || (win.ekey < blockInfo.window.skey && !ascQuery)) {
//           pQueryAttr->limit.offset -= 1;
//         }
4435
//
L
Liu Jicong 已提交
4436 4437 4438 4439 4440 4441 4442 4443
//         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);
4444
//
L
Liu Jicong 已提交
4445 4446 4447 4448 4449 4450 4451 4452 4453 4454 4455
//           // 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.
//       }
//     }
//   }
4456
//
L
Liu Jicong 已提交
4457 4458 4459 4460
//   // check for error
//   if (terrno != TSDB_CODE_SUCCESS) {
//     longjmp(pRuntimeEnv->env, terrno);
//   }
4461
//
L
Liu Jicong 已提交
4462 4463
//   return true;
// }
4464

4465
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num) {
H
Haojun Liao 已提交
4466
  if (p->pDownstream == NULL) {
H
Haojun Liao 已提交
4467
    assert(p->numOfDownstream == 0);
4468 4469
  }

wafwerar's avatar
wafwerar 已提交
4470
  p->pDownstream = taosMemoryCalloc(1, num * POINTER_BYTES);
4471 4472 4473 4474 4475 4476 4477
  if (p->pDownstream == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  memcpy(p->pDownstream, pDownstream, num * POINTER_BYTES);
  p->numOfDownstream = num;
  return TSDB_CODE_SUCCESS;
4478 4479 4480 4481
}

static void doDestroyTableQueryInfo(STableGroupInfo* pTableqinfoGroupInfo);

H
Haojun Liao 已提交
4482
static int32_t setupQueryHandle(void* tsdb, STaskRuntimeEnv* pRuntimeEnv, int64_t qId, bool isSTableQuery) {
L
Liu Jicong 已提交
4483
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
4484 4485 4486 4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508
#if 0
  // TODO set the tags scan handle
  if (onlyQueryTags(pQueryAttr)) {
    return TSDB_CODE_SUCCESS;
  }

  STsdbQueryCond cond = createTsdbQueryCond(pQueryAttr, &pQueryAttr->window);
  if (pQueryAttr->tsCompQuery || pQueryAttr->pointInterpQuery) {
    cond.type = BLOCK_LOAD_TABLE_SEQ_ORDER;
  }

  if (!isSTableQuery
    && (pRuntimeEnv->tableqinfoGroupInfo.numOfTables == 1)
    && (cond.order == TSDB_ORDER_ASC)
    && (!QUERY_IS_INTERVAL_QUERY(pQueryAttr))
    && (!pQueryAttr->groupbyColumn)
    && (!pQueryAttr->simpleAgg)
  ) {
    SArray* pa = GET_TABLEGROUP(pRuntimeEnv, 0);
    STableQueryInfo* pCheckInfo = taosArrayGetP(pa, 0);
    cond.twindow = pCheckInfo->win;
  }

  terrno = TSDB_CODE_SUCCESS;
  if (isFirstLastRowQuery(pQueryAttr)) {
4509
    pRuntimeEnv->pTsdbReadHandle = tsdbQueryLastRow(tsdb, &cond, &pQueryAttr->tableGroupInfo, qId, &pQueryAttr->memRef);
4510 4511 4512 4513 4514 4515 4516 4517 4518 4519 4520 4521 4522 4523 4524 4525 4526 4527 4528 4529

    // update the query time window
    pQueryAttr->window = cond.twindow;
    if (pQueryAttr->tableGroupInfo.numOfTables == 0) {
      pRuntimeEnv->tableqinfoGroupInfo.numOfTables = 0;
    } else {
      size_t numOfGroups = GET_NUM_OF_TABLEGROUP(pRuntimeEnv);
      for(int32_t i = 0; i < numOfGroups; ++i) {
        SArray *group = GET_TABLEGROUP(pRuntimeEnv, i);

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

          pCheckInfo->win = pQueryAttr->window;
          pCheckInfo->lastKey = pCheckInfo->win.skey;
        }
      }
    }
  } else if (isCachedLastQuery(pQueryAttr)) {
4530
    pRuntimeEnv->pTsdbReadHandle = tsdbQueryCacheLast(tsdb, &cond, &pQueryAttr->tableGroupInfo, qId, &pQueryAttr->memRef);
4531
  } else if (pQueryAttr->pointInterpQuery) {
4532
    pRuntimeEnv->pTsdbReadHandle = tsdbQueryRowsInExternalWindow(tsdb, &cond, &pQueryAttr->tableGroupInfo, qId, &pQueryAttr->memRef);
4533
  } else {
4534
    pRuntimeEnv->pTsdbReadHandle = tsdbQueryTables(tsdb, &cond, &pQueryAttr->tableGroupInfo, qId, &pQueryAttr->memRef);
4535 4536 4537 4538 4539 4540
  }
#endif
  return terrno;
}

int32_t doInitQInfo(SQInfo* pQInfo, STSBuf* pTsBuf, void* tsdb, void* sourceOptr, int32_t tbScanner, SArray* pOperator,
L
Liu Jicong 已提交
4541 4542
                    void* param) {
  STaskRuntimeEnv* pRuntimeEnv = &pQInfo->runtimeEnv;
4543

L
Liu Jicong 已提交
4544
  STaskAttr* pQueryAttr = pQInfo->runtimeEnv.pQueryAttr;
4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555
  pQueryAttr->tsdb = tsdb;

  if (tsdb != NULL) {
    int32_t code = setupQueryHandle(tsdb, pRuntimeEnv, pQInfo->qId, pQueryAttr->stableQuery);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }
  }

  pQueryAttr->interBufSize = getOutputInterResultBufSize(pQueryAttr);

L
Liu Jicong 已提交
4556
  pRuntimeEnv->groupResInfo.totalGroup = (int32_t)(pQueryAttr->stableQuery ? GET_NUM_OF_TABLEGROUP(pRuntimeEnv) : 0);
4557 4558 4559 4560 4561 4562 4563 4564 4565 4566 4567 4568 4569 4570 4571 4572 4573
  pRuntimeEnv->enableGroupData = false;

  pRuntimeEnv->pQueryAttr = pQueryAttr;
  pRuntimeEnv->pTsBuf = pTsBuf;
  pRuntimeEnv->cur.vgroupIndex = -1;
  setResultBufSize(pQueryAttr, &pRuntimeEnv->resultInfo);

  if (sourceOptr != NULL) {
    assert(pRuntimeEnv->proot == NULL);
    pRuntimeEnv->proot = sourceOptr;
  }

  if (pTsBuf != NULL) {
    int16_t order = (pQueryAttr->order.order == pRuntimeEnv->pTsBuf->tsOrder) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
    tsBufSetTraverseOrder(pRuntimeEnv->pTsBuf, order);
  }

H
Haojun Liao 已提交
4574
  int32_t ps = 4096;
4575 4576
  getIntermediateBufInfo(pRuntimeEnv, &ps, &pQueryAttr->intermediateResultRowSize);

L
Liu Jicong 已提交
4577
  int32_t TENMB = 1024 * 1024 * 10;
H
Haojun Liao 已提交
4578
  int32_t code = createDiskbasedBuf(&pRuntimeEnv->pResultBuf, ps, TENMB, "", "/tmp");
4579 4580 4581 4582 4583 4584 4585 4586 4587
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

  // create runtime environment
  int32_t numOfTables = (int32_t)pQueryAttr->tableGroupInfo.numOfTables;
  pQInfo->summary.tableInfoSize += (numOfTables * sizeof(STableQueryInfo));
  pQInfo->summary.queryProfEvents = taosArrayInit(512, sizeof(SQueryProfEvent));
  if (pQInfo->summary.queryProfEvents == NULL) {
L
Liu Jicong 已提交
4588
    // qDebug("QInfo:0x%"PRIx64" failed to allocate query prof events array", pQInfo->qId);
4589 4590 4591 4592 4593 4594
  }

  pQInfo->summary.operatorProfResults =
      taosHashInit(8, taosGetDefaultHashFunction(TSDB_DATA_TYPE_TINYINT), true, HASH_NO_LOCK);

  if (pQInfo->summary.operatorProfResults == NULL) {
L
Liu Jicong 已提交
4595
    // qDebug("QInfo:0x%"PRIx64" failed to allocate operator prof results hash", pQInfo->qId);
4596 4597
  }

L
Liu Jicong 已提交
4598
  code = setupQueryRuntimeEnv(pRuntimeEnv, (int32_t)pQueryAttr->tableGroupInfo.numOfTables, pOperator, param);
4599 4600 4601 4602
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

L
Liu Jicong 已提交
4603
  //  setTaskStatus(pOperator->pTaskInfo, QUERY_NOT_COMPLETED);
4604 4605 4606
  return TSDB_CODE_SUCCESS;
}

4607
static void doTableQueryInfoTimeWindowCheck(SExecTaskInfo* pTaskInfo, STableQueryInfo* pTableQueryInfo, int32_t order) {
H
Haojun Liao 已提交
4608 4609
#if 0
    if (order == TSDB_ORDER_ASC) {
4610 4611
    assert(
        (pTableQueryInfo->win.skey <= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
4612 4613
        (pTableQueryInfo->lastKey >= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey >= pTaskInfo->window.skey && pTableQueryInfo->win.ekey <= pTaskInfo->window.ekey));
4614 4615 4616
  } else {
    assert(
        (pTableQueryInfo->win.skey >= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
4617 4618
        (pTableQueryInfo->lastKey <= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey <= pTaskInfo->window.skey && pTableQueryInfo->win.ekey >= pTaskInfo->window.ekey));
4619
  }
H
Haojun Liao 已提交
4620
#endif
4621 4622
}

L
Liu Jicong 已提交
4623 4624 4625 4626 4627 4628 4629 4630
// STsdbQueryCond createTsdbQueryCond(STaskAttr* pQueryAttr, STimeWindow* win) {
//   STsdbQueryCond cond = {
//       .colList   = pQueryAttr->tableCols,
//       .order     = pQueryAttr->order.order,
//       .numOfCols = pQueryAttr->numOfCols,
//       .type      = BLOCK_LOAD_OFFSET_SEQ_ORDER,
//       .loadExternalRows = false,
//   };
4631
//
L
Liu Jicong 已提交
4632 4633 4634
//   TIME_WINDOW_COPY(cond.twindow, *win);
//   return cond;
// }
4635 4636 4637

static STableIdInfo createTableIdInfo(STableQueryInfo* pTableQueryInfo) {
  STableIdInfo tidInfo;
L
Liu Jicong 已提交
4638 4639 4640 4641 4642
  //  STableId* id = TSDB_TABLEID(pTableQueryInfo->pTable);
  //
  //  tidInfo.uid = id->uid;
  //  tidInfo.tid = id->tid;
  //  tidInfo.key = pTableQueryInfo->lastKey;
4643 4644 4645 4646

  return tidInfo;
}

L
Liu Jicong 已提交
4647 4648 4649 4650
// 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;
4651
//
L
Liu Jicong 已提交
4652 4653 4654
//   if (pTableQueryInfo->pTable == NULL) {
//     return;
//   }
4655
//
L
Liu Jicong 已提交
4656 4657 4658 4659 4660 4661 4662 4663 4664
//   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));
//   }
// }
4665

H
Haojun Liao 已提交
4666
static void doCloseAllTimeWindow(STaskRuntimeEnv* pRuntimeEnv) {
4667 4668 4669 4670 4671 4672 4673 4674 4675 4676 4677 4678
  size_t numOfGroup = GET_NUM_OF_TABLEGROUP(pRuntimeEnv);
  for (int32_t i = 0; i < numOfGroup; ++i) {
    SArray* group = GET_TABLEGROUP(pRuntimeEnv, i);

    size_t num = taosArrayGetSize(group);
    for (int32_t j = 0; j < num; ++j) {
      STableQueryInfo* item = taosArrayGetP(group, j);
      closeAllResultRows(&item->resInfo);
    }
  }
}

L
Liu Jicong 已提交
4679 4680 4681
static SSDataBlock* doTableScanImpl(SOperatorInfo* pOperator, bool* newgroup) {
  STableScanInfo* pTableScanInfo = pOperator->info;
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
4682

L
Liu Jicong 已提交
4683 4684
  SSDataBlock*     pBlock = &pTableScanInfo->block;
  STableGroupInfo* pTableGroupInfo = &pOperator->pTaskInfo->tableqinfoGroupInfo;
4685 4686

  *newgroup = false;
H
Haojun Liao 已提交
4687

4688 4689 4690
  while (tsdbNextDataBlock(pTableScanInfo->pTsdbReadHandle)) {
    if (isTaskKilled(pOperator->pTaskInfo)) {
      longjmp(pOperator->pTaskInfo->env, TSDB_CODE_TSC_QUERY_CANCELLED);
4691 4692 4693
    }

    pTableScanInfo->numOfBlocks += 1;
4694
    tsdbRetrieveDataBlockInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->info);
4695 4696

    // todo opt
L
Liu Jicong 已提交
4697 4698 4699 4700 4701 4702 4703 4704 4705 4706
    //    if (pTableGroupInfo->numOfTables > 1 || (pRuntimeEnv->current == NULL && pTableGroupInfo->numOfTables == 1)) {
    //      STableQueryInfo** pTableQueryInfo =
    //          (STableQueryInfo**)taosHashGet(pTableGroupInfo->map, &pBlock->info.uid, sizeof(pBlock->info.uid));
    //      if (pTableQueryInfo == NULL) {
    //        break;
    //      }
    //
    //      pRuntimeEnv->current = *pTableQueryInfo;
    //      doTableQueryInfoTimeWindowCheck(pTaskInfo, *pTableQueryInfo, pTableScanInfo->order);
    //    }
4707 4708

    // this function never returns error?
H
Haojun Liao 已提交
4709
    uint32_t status = BLK_DATA_ALL_NEEDED;
L
Liu Jicong 已提交
4710 4711
    int32_t  code = loadDataBlock(pTaskInfo, pTableScanInfo, pBlock, &status);
    //    int32_t  code = loadDataBlockOnDemand(pOperator->pRuntimeEnv, pTableScanInfo, pBlock, &status);
4712
    if (code != TSDB_CODE_SUCCESS) {
4713
      longjmp(pOperator->pTaskInfo->env, code);
4714 4715 4716 4717 4718 4719 4720 4721 4722 4723 4724 4725 4726
    }

    // current block is ignored according to filter result by block statistics data, continue load the next block
    if (status == BLK_DATA_DISCARD || pBlock->info.rows == 0) {
      continue;
    }

    return pBlock;
  }

  return NULL;
}

L
Liu Jicong 已提交
4727 4728 4729
static SSDataBlock* doTableScan(SOperatorInfo* pOperator, bool* newgroup) {
  STableScanInfo* pTableScanInfo = pOperator->info;
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
4730

H
Haojun Liao 已提交
4731
  // The read handle is not initialized yet, since no qualified tables exists
H
Haojun Liao 已提交
4732 4733 4734 4735
  if (pTableScanInfo->pTsdbReadHandle == NULL) {
    return NULL;
  }

4736 4737 4738 4739 4740 4741 4742 4743 4744 4745
  SResultRowInfo* pResultRowInfo = pTableScanInfo->pResultRowInfo;
  *newgroup = false;

  while (pTableScanInfo->current < pTableScanInfo->times) {
    SSDataBlock* p = doTableScanImpl(pOperator, newgroup);
    if (p != NULL) {
      return p;
    }

    if (++pTableScanInfo->current >= pTableScanInfo->times) {
L
Liu Jicong 已提交
4746
      if (pTableScanInfo->reverseTimes <= 0 /* || isTsdbCacheLastRow(pTableScanInfo->pTsdbReadHandle)*/) {
4747 4748 4749 4750 4751 4752 4753
        return NULL;
      } else {
        break;
      }
    }

    // do prepare for the next round table scan operation
L
Liu Jicong 已提交
4754 4755
    //    STsdbQueryCond cond = createTsdbQueryCond(pQueryAttr, &pQueryAttr->window);
    //    tsdbResetQueryHandle(pTableScanInfo->pTsdbReadHandle, &cond);
4756

4757
    setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
4758
    pTableScanInfo->scanFlag = REPEAT_SCAN;
4759

L
Liu Jicong 已提交
4760 4761 4762 4763 4764
    //    if (pTaskInfo->pTsBuf) {
    //      bool ret = tsBufNextPos(pRuntimeEnv->pTsBuf);
    //      assert(ret);
    //    }
    //
4765 4766 4767 4768
    if (pResultRowInfo->size > 0) {
      pResultRowInfo->curPos = 0;
    }

H
Haojun Liao 已提交
4769
    qDebug("%s start to repeat scan data blocks due to query func required, qrange:%" PRId64 "-%" PRId64,
4770
           GET_TASKID(pTaskInfo), pTaskInfo->window.skey, pTaskInfo->window.ekey);
4771 4772
  }

L
Liu Jicong 已提交
4773
  SSDataBlock* p = NULL;
H
Haojun Liao 已提交
4774
  // todo refactor
4775
  if (pTableScanInfo->reverseTimes > 0) {
H
Haojun Liao 已提交
4776
    setupEnvForReverseScan(pTableScanInfo, pTableScanInfo->pCtx, pTableScanInfo->numOfOutput);
L
Liu Jicong 已提交
4777 4778
    //    STsdbQueryCond cond = createTsdbQueryCond(pQueryAttr, &pQueryAttr->window);
    //    tsdbResetQueryHandle(pTableScanInfo->pTsdbReadHandle, &cond);
4779

H
Haojun Liao 已提交
4780
    qDebug("%s start to reverse scan data blocks due to query func required, qrange:%" PRId64 "-%" PRId64,
4781
           GET_TASKID(pTaskInfo), pTaskInfo->window.skey, pTaskInfo->window.ekey);
4782 4783 4784 4785 4786 4787 4788 4789 4790 4791 4792

    if (pResultRowInfo->size > 0) {
      pResultRowInfo->curPos = pResultRowInfo->size - 1;
    }

    p = doTableScanImpl(pOperator, newgroup);
  }

  return p;
}

L
Liu Jicong 已提交
4793
static SSDataBlock* doBlockInfoScan(SOperatorInfo* pOperator, bool* newgroup) {
4794 4795 4796 4797
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

L
Liu Jicong 已提交
4798
  STableScanInfo* pTableScanInfo = pOperator->info;
4799 4800 4801 4802 4803
  *newgroup = false;
#if 0
  STableBlockDist tableBlockDist = {0};
  tableBlockDist.numOfTables     = (int32_t)pOperator->pRuntimeEnv->tableqinfoGroupInfo.numOfTables;

S
Shengliang Guan 已提交
4804 4805
  int32_t numRowSteps = TSDB_DEFAULT_MAX_ROW_FBLOCK / TSDB_BLOCK_DIST_STEP_ROWS;
  if (TSDB_DEFAULT_MAX_ROW_FBLOCK % TSDB_BLOCK_DIST_STEP_ROWS != 0) {
4806 4807 4808 4809 4810 4811 4812
    ++numRowSteps;
  }
  tableBlockDist.dataBlockInfos  = taosArrayInit(numRowSteps, sizeof(SFileBlockInfo));
  taosArraySetSize(tableBlockDist.dataBlockInfos, numRowSteps);
  tableBlockDist.maxRows = INT_MIN;
  tableBlockDist.minRows = INT_MAX;

4813 4814
  tsdbGetFileBlocksDistInfo(pTableScanInfo->pTsdbReadHandle, &tableBlockDist);
  tableBlockDist.numOfRowsInMemTable = (int32_t) tsdbGetNumOfRowsInMemTable(pTableScanInfo->pTsdbReadHandle);
4815 4816 4817 4818 4819 4820 4821 4822 4823 4824

  SSDataBlock* pBlock = &pTableScanInfo->block;
  pBlock->info.rows   = 1;
  pBlock->info.numOfCols = 1;

  SBufferWriter bw = tbufInitWriter(NULL, false);
  blockDistInfoToBinary(&tableBlockDist, &bw);
  SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, 0);

  int32_t len = (int32_t) tbufTell(&bw);
wafwerar's avatar
wafwerar 已提交
4825
  pColInfo->pData = taosMemoryMalloc(len + sizeof(int32_t));
4826 4827 4828 4829 4830 4831 4832 4833 4834 4835 4836 4837

  *(int32_t*) pColInfo->pData = len;
  memcpy(pColInfo->pData + sizeof(int32_t), tbufGetData(&bw, false), len);

  tbufCloseWriter(&bw);

  SArray* g = GET_TABLEGROUP(pOperator->pRuntimeEnv, 0);
  pOperator->pRuntimeEnv->current = taosArrayGetP(g, 0);

  pOperator->status = OP_EXEC_DONE;
  return pBlock;
#endif
4838 4839
}

H
Haojun Liao 已提交
4840 4841 4842 4843
static void doClearBufferedBlocks(SStreamBlockScanInfo* pInfo) {
  size_t total = taosArrayGetSize(pInfo->pBlockLists);

  pInfo->validBlockIndex = 0;
L
Liu Jicong 已提交
4844 4845
  for (int32_t i = 0; i < total; ++i) {
    SSDataBlock* p = taosArrayGetP(pInfo->pBlockLists, i);
H
Haojun Liao 已提交
4846 4847 4848 4849 4850
    blockDataDestroy(p);
  }
  taosArrayClear(pInfo->pBlockLists);
}

L
Liu Jicong 已提交
4851
static SSDataBlock* doStreamBlockScan(SOperatorInfo* pOperator, bool* newgroup) {
4852
  // NOTE: this operator does never check if current status is done or not
L
Liu Jicong 已提交
4853
  SExecTaskInfo*        pTaskInfo = pOperator->pTaskInfo;
4854 4855
  SStreamBlockScanInfo* pInfo = pOperator->info;

4856 4857 4858 4859 4860
  pTaskInfo->code = pOperator->_openFn(pOperator);
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
    return NULL;
  }

H
Haojun Liao 已提交
4861
  if (pInfo->blockType == STREAM_DATA_TYPE_SSDATA_BLOCK) {
H
Haojun Liao 已提交
4862 4863 4864
    size_t total = taosArrayGetSize(pInfo->pBlockLists);
    if (pInfo->validBlockIndex >= total) {
      doClearBufferedBlocks(pInfo);
4865 4866 4867
      return NULL;
    }

H
Haojun Liao 已提交
4868
    int32_t current = pInfo->validBlockIndex++;
L
Liu Jicong 已提交
4869
    return taosArrayGetP(pInfo->pBlockLists, current);
H
Haojun Liao 已提交
4870 4871 4872
  } else {
    SDataBlockInfo* pBlockInfo = &pInfo->pRes->info;
    blockDataCleanup(pInfo->pRes);
H
Haojun Liao 已提交
4873

H
Haojun Liao 已提交
4874 4875 4876 4877 4878 4879
    while (tqNextDataBlock(pInfo->readerHandle)) {
      pTaskInfo->code = tqRetrieveDataBlockInfo(pInfo->readerHandle, pBlockInfo);
      if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
        terrno = pTaskInfo->code;
        return NULL;
      }
4880

H
Haojun Liao 已提交
4881 4882 4883
      if (pBlockInfo->rows == 0) {
        return NULL;
      }
4884

H
Haojun Liao 已提交
4885 4886 4887 4888 4889 4890
      pInfo->pRes->pDataBlock = tqRetrieveDataBlock(pInfo->readerHandle);
      if (pInfo->pRes->pDataBlock == NULL) {
        // TODO add log
        pTaskInfo->code = terrno;
        return NULL;
      }
4891

H
Haojun Liao 已提交
4892 4893
      break;
    }
4894

H
Haojun Liao 已提交
4895 4896 4897
    // record the scan action.
    pInfo->numOfExec++;
    pInfo->numOfRows += pBlockInfo->rows;
4898

H
Haojun Liao 已提交
4899 4900
    return (pBlockInfo->rows == 0) ? NULL : pInfo->pRes;
  }
4901 4902
}

4903
int32_t loadRemoteDataCallback(void* param, const SDataBuf* pMsg, int32_t code) {
L
Liu Jicong 已提交
4904
  SSourceDataInfo* pSourceDataInfo = (SSourceDataInfo*)param;
4905
  pSourceDataInfo->pRsp = pMsg->pData;
4906

4907 4908
  SRetrieveTableRsp* pRsp = pSourceDataInfo->pRsp;
  pRsp->numOfRows = htonl(pRsp->numOfRows);
L
Liu Jicong 已提交
4909 4910
  pRsp->useconds = htobe64(pRsp->useconds);
  pRsp->compLen = htonl(pRsp->compLen);
H
Haojun Liao 已提交
4911

H
Haojun Liao 已提交
4912
  pSourceDataInfo->status = EX_SOURCE_DATA_READY;
4913
  tsem_post(&pSourceDataInfo->pEx->ready);
H
Haojun Liao 已提交
4914 4915 4916 4917
}

static void destroySendMsgInfo(SMsgSendInfo* pMsgBody) {
  assert(pMsgBody != NULL);
wafwerar's avatar
wafwerar 已提交
4918 4919
  taosMemoryFreeClear(pMsgBody->msgInfo.pData);
  taosMemoryFreeClear(pMsgBody);
H
Haojun Liao 已提交
4920 4921
}

S
Shengliang Guan 已提交
4922
void qProcessFetchRsp(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet) {
L
Liu Jicong 已提交
4923
  SMsgSendInfo* pSendInfo = (SMsgSendInfo*)pMsg->ahandle;
H
Haojun Liao 已提交
4924 4925 4926 4927 4928
  assert(pMsg->ahandle != NULL);

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

  if (pMsg->contLen > 0) {
wafwerar's avatar
wafwerar 已提交
4929
    buf.pData = taosMemoryCalloc(1, pMsg->contLen);
H
Haojun Liao 已提交
4930 4931 4932 4933 4934 4935 4936 4937 4938 4939 4940
    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);
4941 4942
}

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

wafwerar's avatar
wafwerar 已提交
4946
  SResFetchReq* pMsg = taosMemoryCalloc(1, sizeof(SResFetchReq));
4947 4948 4949 4950
  if (NULL == pMsg) {
    pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return pTaskInfo->code;
  }
4951

L
Liu Jicong 已提交
4952 4953
  SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, sourceIndex);
  SSourceDataInfo*       pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, sourceIndex);
4954

L
Liu Jicong 已提交
4955 4956
  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);
4957 4958 4959 4960 4961 4962 4963

  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 已提交
4964
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
4965
  if (NULL == pMsgSendInfo) {
wafwerar's avatar
wafwerar 已提交
4966
    taosMemoryFreeClear(pMsg);
4967 4968 4969
    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 已提交
4970 4971
  }

4972 4973 4974 4975 4976
  pMsgSendInfo->param = pDataInfo;
  pMsgSendInfo->msgInfo.pData = pMsg;
  pMsgSendInfo->msgInfo.len = sizeof(SResFetchReq);
  pMsgSendInfo->msgType = TDMT_VND_FETCH;
  pMsgSendInfo->fp = loadRemoteDataCallback;
4977

4978
  int64_t transporterId = 0;
L
Liu Jicong 已提交
4979
  int32_t code = asyncSendMsgToServer(pExchangeInfo->pTransporter, &pSource->addr.epSet, &transporterId, pMsgSendInfo);
4980 4981 4982
  return TSDB_CODE_SUCCESS;
}

4983
// TODO if only one or two columnss required, how to extract data?
L
Liu Jicong 已提交
4984 4985 4986
static int32_t setSDataBlockFromFetchRsp(SSDataBlock* pRes, SLoadRemoteDataInfo* pLoadInfo, int32_t numOfRows,
                                         char* pData, int32_t compLen, int32_t numOfOutput, int64_t startTs,
                                         uint64_t* total, SArray* pColList) {
H
Haojun Liao 已提交
4987
  blockDataEnsureCapacity(pRes, numOfRows);
4988

H
Haojun Liao 已提交
4989 4990
  if (pColList == NULL) {  // data from other sources
    int32_t* colLen = (int32_t*)pData;
L
Liu Jicong 已提交
4991
    char*    pStart = pData + sizeof(int32_t) * numOfOutput;
H
Haojun Liao 已提交
4992

4993
    for (int32_t i = 0; i < numOfOutput; ++i) {
H
Haojun Liao 已提交
4994 4995 4996
      colLen[i] = htonl(colLen[i]);
      ASSERT(colLen[i] > 0);

4997
      SColumnInfoData* pColInfoData = taosArrayGet(pRes->pDataBlock, i);
H
Haojun Liao 已提交
4998 4999 5000
      if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
        pColInfoData->varmeta.length = colLen[i];
        pColInfoData->varmeta.allocLen = colLen[i];
5001

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

wafwerar's avatar
wafwerar 已提交
5005
        pColInfoData->pData = taosMemoryMalloc(colLen[i]);
H
Haojun Liao 已提交
5006 5007 5008
      } else {
        memcpy(pColInfoData->nullbitmap, pStart, BitmapLen(numOfRows));
        pStart += BitmapLen(numOfRows);
5009
      }
H
Haojun Liao 已提交
5010 5011 5012

      memcpy(pColInfoData->pData, pStart, colLen[i]);
      pStart += colLen[i];
5013
    }
H
Haojun Liao 已提交
5014
  } else {  // extract data according to pColList
5015
    ASSERT(numOfOutput == taosArrayGetSize(pColList));
H
Haojun Liao 已提交
5016 5017

    // data from mnode
L
Liu Jicong 已提交
5018 5019 5020
    for (int32_t i = 0; i < numOfOutput; ++i) {
      for (int32_t j = 0; j < numOfOutput; ++j) {
        int16_t colIndex = *(int16_t*)taosArrayGet(pColList, j);
5021 5022
        if (colIndex - 1 == i) {
          SColumnInfoData* pColInfoData = taosArrayGet(pRes->pDataBlock, j);
5023

5024 5025 5026 5027 5028 5029 5030
          for (int32_t k = 0; k < numOfRows; ++k) {
            colDataAppend(pColInfoData, k, pData, false);
            pData += pColInfoData->info.bytes;
          }
          break;
        }
      }
5031
    }
5032
  }
5033

H
Haojun Liao 已提交
5034
  pRes->info.rows = numOfRows;
5035

5036
  int64_t el = taosGetTimestampUs() - startTs;
5037

H
Haojun Liao 已提交
5038 5039
  pLoadInfo->totalRows += numOfRows;
  pLoadInfo->totalSize += compLen;
5040

H
Haojun Liao 已提交
5041 5042 5043
  if (total != NULL) {
    *total += numOfRows;
  }
5044

H
Haojun Liao 已提交
5045
  pLoadInfo->totalElapsed += el;
5046 5047
  return TSDB_CODE_SUCCESS;
}
5048

L
Liu Jicong 已提交
5049 5050
static void* setAllSourcesCompleted(SOperatorInfo* pOperator, int64_t startTs) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
5051
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
5052

L
Liu Jicong 已提交
5053
  int64_t              el = taosGetTimestampUs() - startTs;
H
Haojun Liao 已提交
5054 5055
  SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
  pLoadInfo->totalElapsed += el;
H
Haojun Liao 已提交
5056

5057
  size_t totalSources = taosArrayGetSize(pExchangeInfo->pSources);
L
Liu Jicong 已提交
5058 5059 5060
  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);
5061 5062 5063 5064 5065

  doSetOperatorCompleted(pOperator);
  return NULL;
}

L
Liu Jicong 已提交
5066 5067
static SSDataBlock* concurrentlyLoadRemoteDataImpl(SOperatorInfo* pOperator, SExchangeInfo* pExchangeInfo,
                                                   SExecTaskInfo* pTaskInfo) {
5068 5069 5070 5071 5072 5073 5074 5075 5076
  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);

H
Haojun Liao 已提交
5077
      if (pDataInfo->status == DATA_EXHAUSTED) {
5078
        completed += 1;
H
Haojun Liao 已提交
5079 5080
        continue;
      }
5081

H
Haojun Liao 已提交
5082
      if (pDataInfo->status != DATA_READY) {
5083 5084 5085
        continue;
      }

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

L
Liu Jicong 已提交
5089
      SSDataBlock*         pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
5090
      SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
5091
      if (pRsp->numOfRows == 0) {
L
Liu Jicong 已提交
5092 5093
        qDebug("%s vgId:%d, taskID:0x%" PRIx64 " index:%d completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64
               " try next",
5094
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, i + 1, pDataInfo->totalRows,
H
Haojun Liao 已提交
5095
               pExchangeInfo->loadInfo.totalRows);
H
Haojun Liao 已提交
5096
        pDataInfo->status = DATA_EXHAUSTED;
5097 5098 5099
        completed += 1;
        continue;
      }
H
Haojun Liao 已提交
5100

H
Haojun Liao 已提交
5101
      SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
L
Liu Jicong 已提交
5102 5103 5104
      code =
          setSDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
                                    pTableRsp->compLen, pOperator->numOfOutput, startTs, &pDataInfo->totalRows, NULL);
5105
      if (code != 0) {
5106 5107 5108
        goto _error;
      }

5109 5110 5111
      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 已提交
5112 5113
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pDataInfo->totalRows,
               pLoadInfo->totalRows, pLoadInfo->totalSize, i + 1, totalSources);
H
Haojun Liao 已提交
5114
        pDataInfo->status = DATA_EXHAUSTED;
5115
      } else {
L
Liu Jicong 已提交
5116 5117
        qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, totalRows:%" PRIu64
               ", totalBytes:%" PRIu64,
H
Haojun Liao 已提交
5118 5119
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pLoadInfo->totalRows,
               pLoadInfo->totalSize);
5120 5121
      }

H
Haojun Liao 已提交
5122 5123
      if (pDataInfo->status != DATA_EXHAUSTED) {
        pDataInfo->status = DATA_NOT_READY;
5124 5125 5126 5127 5128 5129 5130 5131 5132 5133 5134 5135 5136 5137 5138 5139 5140 5141 5142
        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 已提交
5143 5144 5145
static SSDataBlock* concurrentlyLoadRemoteData(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
5146

H
Haojun Liao 已提交
5147 5148 5149
  if (pOperator->status == OP_RES_TO_RETURN) {
    return concurrentlyLoadRemoteDataImpl(pOperator, pExchangeInfo, pTaskInfo);
  }
5150

L
Liu Jicong 已提交
5151
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
5152 5153 5154
  int64_t startTs = taosGetTimestampUs();

  // Asynchronously send all fetch requests to all sources.
L
Liu Jicong 已提交
5155
  for (int32_t i = 0; i < totalSources; ++i) {
5156 5157
    int32_t code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
5158
      return NULL;
5159 5160 5161 5162
    }
  }

  int64_t endTs = taosGetTimestampUs();
L
Liu Jicong 已提交
5163 5164
  qDebug("%s send all fetch request to %" PRIzu " sources completed, elapsed:%" PRId64, GET_TASKID(pTaskInfo),
         totalSources, endTs - startTs);
5165 5166 5167

  tsem_wait(&pExchangeInfo->ready);

H
Haojun Liao 已提交
5168 5169
  pOperator->status = OP_RES_TO_RETURN;
  return concurrentlyLoadRemoteDataImpl(pOperator, pExchangeInfo, pTaskInfo);
5170 5171
}

L
Liu Jicong 已提交
5172 5173 5174
static int32_t prepareConcurrentlyLoad(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5175

L
Liu Jicong 已提交
5176
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
5177 5178 5179
  int64_t startTs = taosGetTimestampUs();

  // Asynchronously send all fetch requests to all sources.
L
Liu Jicong 已提交
5180
  for (int32_t i = 0; i < totalSources; ++i) {
5181 5182
    int32_t code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
5183 5184
      pTaskInfo->code = code;
      return code;
5185 5186 5187 5188
    }
  }

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

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

H
Haojun Liao 已提交
5195
  return TSDB_CODE_SUCCESS;
5196 5197
}

L
Liu Jicong 已提交
5198 5199 5200
static SSDataBlock* seqLoadRemoteData(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5201

L
Liu Jicong 已提交
5202
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
5203
  int64_t startTs = taosGetTimestampUs();
5204

L
Liu Jicong 已提交
5205
  while (1) {
5206 5207
    if (pExchangeInfo->current >= totalSources) {
      return setAllSourcesCompleted(pOperator, startTs);
5208
    }
5209

5210 5211 5212 5213
    doSendFetchDataRequest(pExchangeInfo, pTaskInfo, pExchangeInfo->current);

    tsem_wait(&pExchangeInfo->ready);

L
Liu Jicong 已提交
5214
    SSourceDataInfo*       pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, pExchangeInfo->current);
X
Xiaoyu Wang 已提交
5215
    SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, pExchangeInfo->current);
5216

L
Liu Jicong 已提交
5217
    SRetrieveTableRsp*   pRsp = pDataInfo->pRsp;
H
Haojun Liao 已提交
5218 5219
    SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;

5220
    if (pRsp->numOfRows == 0) {
L
Liu Jicong 已提交
5221 5222
      qDebug("%s vgId:%d, taskID:0x%" PRIx64 " %d of total completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64
             " try next",
5223
             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pExchangeInfo->current + 1,
H
Haojun Liao 已提交
5224
             pDataInfo->totalRows, pLoadInfo->totalRows);
H
Haojun Liao 已提交
5225

H
Haojun Liao 已提交
5226
      pDataInfo->status = DATA_EXHAUSTED;
5227 5228 5229
      pExchangeInfo->current += 1;
      continue;
    }
H
Haojun Liao 已提交
5230

L
Liu Jicong 已提交
5231
    SSDataBlock*       pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
5232
    SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
L
Liu Jicong 已提交
5233 5234 5235
    int32_t            code =
        setSDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
                                  pTableRsp->compLen, pOperator->numOfOutput, startTs, &pDataInfo->totalRows, NULL);
5236 5237

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

H
Haojun Liao 已提交
5243
      pDataInfo->status = DATA_EXHAUSTED;
5244 5245
      pExchangeInfo->current += 1;
    } else {
L
Liu Jicong 已提交
5246 5247 5248 5249
      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);
5250 5251 5252 5253
    }

    return pExchangeInfo->pResult;
  }
5254 5255
}

L
Liu Jicong 已提交
5256
static int32_t prepareLoadRemoteData(SOperatorInfo* pOperator) {
5257
  if (OPTR_IS_OPENED(pOperator)) {
H
Haojun Liao 已提交
5258 5259 5260
    return TSDB_CODE_SUCCESS;
  }

L
Liu Jicong 已提交
5261
  SExchangeInfo* pExchangeInfo = pOperator->info;
H
Haojun Liao 已提交
5262 5263 5264 5265 5266 5267 5268 5269 5270
  if (pExchangeInfo->seqLoadData) {
    // do nothing for sequentially load data
  } else {
    int32_t code = prepareConcurrentlyLoad(pOperator);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }
  }

5271
  OPTR_SET_OPENED(pOperator);
H
Haojun Liao 已提交
5272 5273 5274
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
5275 5276 5277
static SSDataBlock* doLoadRemoteData(SOperatorInfo* pOperator, bool* newgroup) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5278

5279 5280
  pTaskInfo->code = pOperator->_openFn(pOperator);
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
5281 5282
    return NULL;
  }
5283

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

5287
  if (pOperator->status == OP_EXEC_DONE) {
L
Liu Jicong 已提交
5288 5289 5290
    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);
5291 5292 5293 5294
    return NULL;
  }

  *newgroup = false;
H
Haojun Liao 已提交
5295

5296 5297 5298
  if (pExchangeInfo->seqLoadData) {
    return seqLoadRemoteData(pOperator);
  } else {
H
Haojun Liao 已提交
5299
    return concurrentlyLoadRemoteData(pOperator);
5300
  }
H
Haojun Liao 已提交
5301

5302
#if 0
H
Haojun Liao 已提交
5303
  _error:
wafwerar's avatar
wafwerar 已提交
5304 5305
  taosMemoryFreeClear(pMsg);
  taosMemoryFreeClear(pMsgSendInfo);
H
Haojun Liao 已提交
5306 5307 5308

  terrno = pTaskInfo->code;
  return NULL;
5309
#endif
H
Haojun Liao 已提交
5310
}
5311

H
Haojun Liao 已提交
5312
static int32_t initDataSource(int32_t numOfSources, SExchangeInfo* pInfo) {
5313
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
H
Haojun Liao 已提交
5314 5315
  if (pInfo->pSourceDataInfo == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
5316 5317
  }

L
Liu Jicong 已提交
5318
  for (int32_t i = 0; i < numOfSources; ++i) {
5319
    SSourceDataInfo dataInfo = {0};
H
Haojun Liao 已提交
5320
    dataInfo.status = EX_SOURCE_DATA_NOT_READY;
L
Liu Jicong 已提交
5321 5322
    dataInfo.pEx = pInfo;
    dataInfo.index = i;
5323

H
Haojun Liao 已提交
5324 5325 5326 5327 5328 5329 5330 5331 5332 5333
    void* ret = taosArrayPush(pInfo->pSourceDataInfo, &dataInfo);
    if (ret == NULL) {
      taosArrayDestroy(pInfo->pSourceDataInfo);
      return TSDB_CODE_OUT_OF_MEMORY;
    }
  }

  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
5334
SOperatorInfo* createExchangeOperatorInfo(const SNodeList* pSources, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
5335
  SExchangeInfo* pInfo = taosMemoryCalloc(1, sizeof(SExchangeInfo));
wafwerar's avatar
wafwerar 已提交
5336
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
5337 5338

  if (pInfo == NULL || pOperator == NULL) {
wafwerar's avatar
wafwerar 已提交
5339 5340
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
5341 5342
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
H
Haojun Liao 已提交
5343 5344
  }

H
Haojun Liao 已提交
5345
  size_t numOfSources = LIST_LENGTH(pSources);
H
Haojun Liao 已提交
5346
  pInfo->pSources = taosArrayInit(numOfSources, sizeof(SDownstreamSourceNode));
H
Haojun Liao 已提交
5347 5348 5349
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
  if (pInfo->pSourceDataInfo == NULL || pInfo->pSources == NULL) {
    goto _error;
H
Haojun Liao 已提交
5350 5351
  }

L
Liu Jicong 已提交
5352 5353
  for (int32_t i = 0; i < numOfSources; ++i) {
    SNodeListNode* pNode = nodesListGetNode((SNodeList*)pSources, i);
H
Haojun Liao 已提交
5354 5355
    taosArrayPush(pInfo->pSources, pNode);
  }
5356

H
Haojun Liao 已提交
5357 5358 5359
  int32_t code = initDataSource(numOfSources, pInfo);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
5360
  }
H
Haojun Liao 已提交
5361

L
Liu Jicong 已提交
5362 5363
  size_t size = pBlock->info.numOfCols;
  pInfo->pResult = pBlock;
5364 5365 5366
  pInfo->seqLoadData = true;

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

L
Liu Jicong 已提交
5368
  pOperator->name = "ExchangeOperator";
X
Xiaoyu Wang 已提交
5369
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
5370
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
5371 5372 5373 5374 5375 5376 5377
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfOutput = size;
  pOperator->pTaskInfo = pTaskInfo;
  pOperator->_openFn = prepareLoadRemoteData;  // assign a dummy function.
  pOperator->getNextFn = doLoadRemoteData;
  pOperator->closeFn = destroyExchangeOperatorInfo;
5378

S
Shengliang Guan 已提交
5379
#if 1
L
Liu Jicong 已提交
5380
  {  // todo refactor
H
Haojun Liao 已提交
5381 5382 5383
    SRpcInit rpcInit;
    memset(&rpcInit, 0, sizeof(rpcInit));
    rpcInit.localPort = 0;
H
Haojun Liao 已提交
5384
    rpcInit.label = "EX";
H
Haojun Liao 已提交
5385
    rpcInit.numOfThreads = 1;
S
Shengliang Guan 已提交
5386
    rpcInit.cfp = qProcessFetchRsp;
S
Shengliang Guan 已提交
5387
    rpcInit.sessions = tsMaxConnections;
H
Haojun Liao 已提交
5388
    rpcInit.connType = TAOS_CONN_CLIENT;
L
Liu Jicong 已提交
5389
    rpcInit.user = (char*)"root";
S
Shengliang Guan 已提交
5390
    rpcInit.idleTime = tsShellActivityTimer * 1000;
H
Haojun Liao 已提交
5391
    rpcInit.ckey = "key";
S
Shengliang Guan 已提交
5392
    rpcInit.spi = 1;
L
Liu Jicong 已提交
5393
    rpcInit.secret = (char*)"dcc5bed04851fec854c035b2e40263b6";
H
Haojun Liao 已提交
5394 5395 5396

    pInfo->pTransporter = rpcOpen(&rpcInit);
    if (pInfo->pTransporter == NULL) {
L
Liu Jicong 已提交
5397
      return NULL;  // todo
H
Haojun Liao 已提交
5398 5399
    }
  }
S
Shengliang 已提交
5400
#endif
5401

5402
  return pOperator;
H
Haojun Liao 已提交
5403

L
Liu Jicong 已提交
5404
_error:
H
Haojun Liao 已提交
5405
  if (pInfo != NULL) {
H
Haojun Liao 已提交
5406
    destroyExchangeOperatorInfo(pInfo, numOfSources);
H
Haojun Liao 已提交
5407 5408
  }

wafwerar's avatar
wafwerar 已提交
5409 5410
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
5411
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
H
Haojun Liao 已提交
5412
  return NULL;
5413 5414
}

H
Haojun Liao 已提交
5415
SSDataBlock* createResultDataBlock(const SArray* pExprInfo) {
wafwerar's avatar
wafwerar 已提交
5416
  SSDataBlock* pResBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
H
Haojun Liao 已提交
5417 5418 5419 5420 5421 5422 5423 5424
  if (pResBlock == NULL) {
    return NULL;
  }

  size_t numOfCols = taosArrayGetSize(pExprInfo);
  pResBlock->pDataBlock = taosArrayInit(numOfCols, sizeof(SColumnInfoData));

  SArray* pResult = pResBlock->pDataBlock;
L
Liu Jicong 已提交
5425
  for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
5426
    SColumnInfoData colInfoData = {0};
L
Liu Jicong 已提交
5427
    SExprInfo*      p = taosArrayGetP(pExprInfo, i);
H
Haojun Liao 已提交
5428

H
Haojun Liao 已提交
5429
    SResSchema* pSchema = &p->base.resSchema;
L
Liu Jicong 已提交
5430
    colInfoData.info.type = pSchema->type;
H
Haojun Liao 已提交
5431 5432
    colInfoData.info.colId = pSchema->colId;
    colInfoData.info.bytes = pSchema->bytes;
H
Haojun Liao 已提交
5433 5434
    colInfoData.info.scale = pSchema->scale;
    colInfoData.info.precision = pSchema->precision;
H
Haojun Liao 已提交
5435 5436 5437 5438 5439 5440
    taosArrayPush(pResult, &colInfoData);
  }

  return pResBlock;
}

L
Liu Jicong 已提交
5441 5442 5443
SOperatorInfo* createTableScanOperatorInfo(void* pTsdbReadHandle, int32_t order, int32_t numOfOutput,
                                           int32_t repeatTime, int32_t reverseTime, SArray* pColMatchInfo,
                                           SExecTaskInfo* pTaskInfo) {
5444 5445
  assert(repeatTime > 0);

L
Liu Jicong 已提交
5446 5447
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
  SOperatorInfo*  pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
5448
  if (pInfo == NULL || pOperator == NULL) {
wafwerar's avatar
wafwerar 已提交
5449 5450
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
5451

H
Haojun Liao 已提交
5452
    pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
5453 5454 5455
    return NULL;
  }

H
Haojun Liao 已提交
5456
  pInfo->block.pDataBlock = taosArrayInit(numOfOutput, sizeof(SColumnInfoData));
L
Liu Jicong 已提交
5457
  for (int32_t i = 0; i < numOfOutput; ++i) {
H
Haojun Liao 已提交
5458 5459 5460 5461
    SColumnInfoData idata = {0};
    taosArrayPush(pInfo->block.pDataBlock, &idata);
  }

H
Haojun Liao 已提交
5462
  pInfo->pTsdbReadHandle = pTsdbReadHandle;
L
Liu Jicong 已提交
5463 5464 5465 5466 5467 5468 5469 5470 5471 5472 5473 5474 5475 5476
  pInfo->times = repeatTime;
  pInfo->reverseTimes = reverseTime;
  pInfo->order = order;
  pInfo->current = 0;
  pInfo->scanFlag = MAIN_SCAN;
  pInfo->pColMatchInfo = pColMatchInfo;
  pOperator->name = "TableScanOperator";
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN;
  pOperator->blockingOptr = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfOutput = numOfOutput;
  pOperator->getNextFn = doTableScan;
  pOperator->pTaskInfo = pTaskInfo;
5477 5478 5479 5480

  return pOperator;
}

H
Haojun Liao 已提交
5481
SOperatorInfo* createTableSeqScanOperatorInfo(void* pTsdbReadHandle, STaskRuntimeEnv* pRuntimeEnv) {
wafwerar's avatar
wafwerar 已提交
5482
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
5483

L
Liu Jicong 已提交
5484 5485 5486 5487 5488 5489
  pInfo->pTsdbReadHandle = pTsdbReadHandle;
  pInfo->times = 1;
  pInfo->reverseTimes = 0;
  pInfo->order = pRuntimeEnv->pQueryAttr->order.order;
  pInfo->current = 0;
  pInfo->prevGroupId = -1;
5490 5491
  pRuntimeEnv->enableGroupData = true;

wafwerar's avatar
wafwerar 已提交
5492
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
5493
  pOperator->name = "TableSeqScanOperator";
X
Xiaoyu Wang 已提交
5494
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TABLE_SEQ_SCAN;
5495
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
5496 5497 5498 5499 5500
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfOutput = pRuntimeEnv->pQueryAttr->numOfCols;
  pOperator->pRuntimeEnv = pRuntimeEnv;
  pOperator->getNextFn = doTableScanImpl;
5501 5502 5503 5504

  return pOperator;
}

H
Haojun Liao 已提交
5505
SOperatorInfo* createTableBlockInfoScanOperator(void* pTsdbReadHandle, STaskRuntimeEnv* pRuntimeEnv) {
wafwerar's avatar
wafwerar 已提交
5506
  STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
5507

L
Liu Jicong 已提交
5508
  pInfo->pTsdbReadHandle = pTsdbReadHandle;
5509 5510 5511 5512 5513 5514 5515 5516
  pInfo->block.pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));

  SColumnInfoData infoData = {{0}};
  infoData.info.type = TSDB_DATA_TYPE_BINARY;
  infoData.info.bytes = 1024;
  infoData.info.colId = 0;
  taosArrayPush(pInfo->block.pDataBlock, &infoData);

wafwerar's avatar
wafwerar 已提交
5517
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
5518 5519
  pOperator->name = "TableBlockInfoScanOperator";
  //  pOperator->operatorType = OP_TableBlockInfoScan;
5520
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
5521 5522 5523
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  //  pOperator->numOfOutput  = pRuntimeEnv->pQueryAttr->numOfCols;
H
Haojun Liao 已提交
5524
  pOperator->getNextFn = doBlockInfoScan;
5525 5526 5527 5528

  return pOperator;
}

L
Liu Jicong 已提交
5529 5530
SOperatorInfo* createStreamScanOperatorInfo(void* streamReadHandle, SSDataBlock* pResBlock, SArray* pColList,
                                            SArray* pTableIdList, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
5531
  SStreamBlockScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamBlockScanInfo));
L
Liu Jicong 已提交
5532
  SOperatorInfo*        pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
5533
  if (pInfo == NULL || pOperator == NULL) {
wafwerar's avatar
wafwerar 已提交
5534 5535
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
5536 5537 5538 5539
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
5540
  // set the extract column id to streamHandle
L
Liu Jicong 已提交
5541
  tqReadHandleSetColIdList((STqReadHandle*)streamReadHandle, pColList);
5542 5543
  int32_t code = tqReadHandleSetTbUidList(streamReadHandle, pTableIdList);
  if (code != 0) {
wafwerar's avatar
wafwerar 已提交
5544 5545
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
5546 5547
    return NULL;
  }
H
Haojun Liao 已提交
5548

L
Liu Jicong 已提交
5549
  pInfo->pBlockLists = taosArrayInit(4, POINTER_BYTES);
H
Haojun Liao 已提交
5550 5551 5552 5553 5554 5555
  if (pInfo->pBlockLists == NULL) {
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
    return NULL;
  }

L
Liu Jicong 已提交
5556 5557
  pInfo->readerHandle = streamReadHandle;
  pInfo->pRes = pResBlock;
H
Haojun Liao 已提交
5558

L
Liu Jicong 已提交
5559 5560 5561 5562 5563 5564 5565 5566 5567 5568
  pOperator->name = "StreamBlockScanOperator";
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN;
  pOperator->blockingOptr = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfOutput = pResBlock->info.numOfCols;
  pOperator->_openFn = operatorDummyOpenFn;
  pOperator->getNextFn = doStreamBlockScan;
  pOperator->closeFn = operatorDummyCloseFn;
  pOperator->pTaskInfo = pTaskInfo;
L
Liu Jicong 已提交
5569
  return pOperator;
H
Haojun Liao 已提交
5570 5571
}

5572
static int32_t loadSysTableContentCb(void* param, const SDataBuf* pMsg, int32_t code) {
L
Liu Jicong 已提交
5573 5574
  SOperatorInfo*     operator=(SOperatorInfo*) param;
  SSysTableScanInfo* pScanResInfo = (SSysTableScanInfo*)operator->info;
X
Xiaoyu Wang 已提交
5575 5576
  if (TSDB_CODE_SUCCESS == code) {
    pScanResInfo->pRsp = pMsg->pData;
5577

X
Xiaoyu Wang 已提交
5578 5579
    SRetrieveMetaTableRsp* pRsp = pScanResInfo->pRsp;
    pRsp->numOfRows = htonl(pRsp->numOfRows);
L
Liu Jicong 已提交
5580 5581 5582
    pRsp->useconds = htobe64(pRsp->useconds);
    pRsp->handle = htobe64(pRsp->handle);
    pRsp->compLen = htonl(pRsp->compLen);
D
dapan1121 已提交
5583 5584
  } else {
    operator->pTaskInfo->code = code;
X
Xiaoyu Wang 已提交
5585
  }
5586

H
Haojun Liao 已提交
5587
  tsem_post(&pScanResInfo->ready);
5588 5589
}

H
Haojun Liao 已提交
5590 5591
static SSDataBlock* doFilterResult(SSysTableScanInfo* pInfo) {
  if (pInfo->pCondition == NULL) {
L
Liu Jicong 已提交
5592
    return pInfo->pRes->info.rows == 0 ? NULL : pInfo->pRes;
H
Haojun Liao 已提交
5593 5594 5595 5596 5597 5598 5599 5600 5601 5602 5603 5604 5605 5606
  }

  SFilterInfo* filter = NULL;
  int32_t      code = filterInitFromNode(pInfo->pCondition, &filter, 0);

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

  int8_t* rowRes = NULL;
  bool    keep = filterExecute(filter, pInfo->pRes, &rowRes, NULL, param1.numOfCols);

  SSDataBlock* px = createOneDataBlock(pInfo->pRes);
  blockDataEnsureCapacity(px, pInfo->pRes->info.rows);

L
Liu Jicong 已提交
5607
  // TODO refactor
H
Haojun Liao 已提交
5608 5609 5610 5611 5612 5613 5614 5615 5616 5617 5618 5619 5620 5621 5622 5623 5624 5625 5626
  int32_t numOfRow = 0;
  for (int32_t i = 0; i < pInfo->pRes->info.numOfCols; ++i) {
    SColumnInfoData* pDest = taosArrayGet(px->pDataBlock, i);
    SColumnInfoData* pSrc = taosArrayGet(pInfo->pRes->pDataBlock, i);

    numOfRow = 0;
    for (int32_t j = 0; j < pInfo->pRes->info.rows; ++j) {
      if (rowRes[j] == 0) {
        continue;
      }

      colDataAppend(pDest, numOfRow, colDataGetData(pSrc, j), false);
      numOfRow += 1;
    }
  }

  px->info.rows = numOfRow;
  pInfo->pRes = px;

L
Liu Jicong 已提交
5627
  return pInfo->pRes->info.rows == 0 ? NULL : pInfo->pRes;
5628 5629
}

D
dapan1121 已提交
5630
EDealRes getDBNameFromConditionWalker(SNode* pNode, void* pContext) {
L
Liu Jicong 已提交
5631
  int32_t   code = TSDB_CODE_SUCCESS;
D
dapan1121 已提交
5632 5633 5634 5635
  ENodeType nType = nodeType(pNode);

  switch (nType) {
    case QUERY_NODE_OPERATOR: {
L
Liu Jicong 已提交
5636
      SOperatorNode* node = (SOperatorNode*)pNode;
D
dapan1121 已提交
5637 5638

      if (OP_TYPE_EQUAL == node->opType) {
L
Liu Jicong 已提交
5639
        *(int32_t*)pContext = 1;
D
dapan1121 已提交
5640 5641 5642
        return DEAL_RES_CONTINUE;
      }

L
Liu Jicong 已提交
5643
      *(int32_t*)pContext = 0;
D
dapan1121 已提交
5644 5645 5646 5647

      return DEAL_RES_IGNORE_CHILD;
    }
    case QUERY_NODE_COLUMN: {
L
Liu Jicong 已提交
5648
      if (1 != *(int32_t*)pContext) {
D
dapan1121 已提交
5649 5650
        return DEAL_RES_CONTINUE;
      }
H
Haojun Liao 已提交
5651

L
Liu Jicong 已提交
5652
      SColumnNode* node = (SColumnNode*)pNode;
D
dapan1121 已提交
5653
      if (TSDB_INS_USER_STABLES_DBNAME_COLID == node->colId) {
L
Liu Jicong 已提交
5654
        *(int32_t*)pContext = 2;
D
dapan1121 已提交
5655 5656 5657
        return DEAL_RES_CONTINUE;
      }

L
Liu Jicong 已提交
5658
      *(int32_t*)pContext = 0;
D
dapan1121 已提交
5659 5660 5661
      return DEAL_RES_CONTINUE;
    }
    case QUERY_NODE_VALUE: {
L
Liu Jicong 已提交
5662
      if (2 != *(int32_t*)pContext) {
D
dapan1121 已提交
5663 5664 5665
        return DEAL_RES_CONTINUE;
      }

L
Liu Jicong 已提交
5666 5667
      SValueNode* node = (SValueNode*)pNode;
      char*       dbName = nodesGetValueFromNode(node);
H
Haojun Liao 已提交
5668
      strncpy(pContext, varDataVal(dbName), varDataLen(dbName));
L
Liu Jicong 已提交
5669
      *((char*)pContext + varDataLen(dbName)) = 0;
D
dapan1121 已提交
5670 5671 5672 5673 5674 5675 5676 5677 5678
      return DEAL_RES_ERROR;  // stop walk
    }
    default:
      break;
  }

  return DEAL_RES_CONTINUE;
}

L
Liu Jicong 已提交
5679
void getDBNameFromCondition(SNode* pCondition, char* dbName) {
D
dapan1121 已提交
5680 5681 5682 5683 5684 5685 5686
  if (NULL == pCondition) {
    return;
  }

  nodesWalkNode(pCondition, getDBNameFromConditionWalker, dbName);
}

L
Liu Jicong 已提交
5687 5688 5689
static SSDataBlock* doSysTableScan(SOperatorInfo* pOperator, bool* newgroup) {
  // build message and send to mnode to fetch the content of system tables.
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
5690 5691
  SSysTableScanInfo* pInfo = pOperator->info;

5692 5693 5694 5695 5696
  // retrieve local table list info from vnode
  if (pInfo->type == TSDB_MGMT_TABLE_TABLE) {
    if (pInfo->pCur == NULL) {
      pInfo->pCur = metaOpenTbCursor(pInfo->readHandle);
    }
5697

H
Haojun Liao 已提交
5698
    blockDataCleanup(pInfo->pRes);
5699

L
Liu Jicong 已提交
5700
    int32_t          tableNameSlotId = 1;
H
Haojun Liao 已提交
5701
    SColumnInfoData* pTableNameCol = taosArrayGet(pInfo->pRes->pDataBlock, tableNameSlotId);
5702

L
Liu Jicong 已提交
5703
    char*   name = NULL;
5704
    int32_t numOfRows = 0;
5705 5706

    char n[TSDB_TABLE_NAME_LEN] = {0};
5707
    while ((name = metaTbCursorNext(pInfo->pCur)) != NULL) {
5708 5709
      STR_TO_VARSTR(n, name);
      colDataAppend(pTableNameCol, numOfRows, n, false);
5710 5711 5712 5713
      numOfRows += 1;
      if (numOfRows >= pInfo->capacity) {
        break;
      }
H
Haojun Liao 已提交
5714

L
Liu Jicong 已提交
5715
      for (int32_t i = 0; i < pInfo->pRes->info.numOfCols; ++i) {
H
Haojun Liao 已提交
5716
        if (i == tableNameSlotId) {
H
Haojun Liao 已提交
5717 5718 5719 5720
          continue;
        }

        SColumnInfoData* pColInfoData = taosArrayGet(pInfo->pRes->pDataBlock, i);
L
Liu Jicong 已提交
5721 5722
        int64_t          tmp = 0;
        char             t[10] = {0};
H
Haojun Liao 已提交
5723 5724 5725 5726
        STR_TO_VARSTR(t, "_");
        if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
          colDataAppend(pColInfoData, numOfRows, t, false);
        } else {
L
Liu Jicong 已提交
5727
          colDataAppend(pColInfoData, numOfRows, (char*)&tmp, false);
H
Haojun Liao 已提交
5728 5729
        }
      }
5730
    }
5731

H
Haojun Liao 已提交
5732
    pInfo->loadInfo.totalRows += numOfRows;
5733
    pInfo->pRes->info.rows = numOfRows;
5734

L
Liu Jicong 已提交
5735 5736 5737
    //    pInfo->elapsedTime;
    //    pInfo->totalBytes;
    return (pInfo->pRes->info.rows == 0) ? NULL : pInfo->pRes;
5738
  } else {  // load the meta from mnode of the given epset
H
Haojun Liao 已提交
5739
    int64_t startTs = taosGetTimestampUs();
5740

H
Haojun Liao 已提交
5741
    pInfo->req.type = pInfo->type;
H
Haojun Liao 已提交
5742
    strncpy(pInfo->req.tb, tNameGetTableName(&pInfo->name), tListLen(pInfo->req.tb));
D
dapan1121 已提交
5743 5744 5745 5746 5747
    if (pInfo->showRewrite) {
      char dbName[TSDB_DB_NAME_LEN] = {0};
      getDBNameFromCondition(pInfo->pCondition, dbName);
      sprintf(pInfo->req.db, "%d.%s", pInfo->accountId, dbName);
    }
H
Haojun Liao 已提交
5748

H
Haojun Liao 已提交
5749
    int32_t contLen = tSerializeSRetrieveTableReq(NULL, 0, &pInfo->req);
L
Liu Jicong 已提交
5750
    char*   buf1 = taosMemoryCalloc(1, contLen);
H
Haojun Liao 已提交
5751
    tSerializeSRetrieveTableReq(buf1, contLen, &pInfo->req);
5752

5753
    // send the fetch remote task result reques
wafwerar's avatar
wafwerar 已提交
5754
    SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
5755 5756 5757 5758 5759 5760
    if (NULL == pMsgSendInfo) {
      qError("%s prepare message %d failed", GET_TASKID(pTaskInfo), (int32_t)sizeof(SMsgSendInfo));
      pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
      return NULL;
    }

D
dapan1121 已提交
5761
    pMsgSendInfo->param = pOperator;
H
Haojun Liao 已提交
5762 5763
    pMsgSendInfo->msgInfo.pData = buf1;
    pMsgSendInfo->msgInfo.len = contLen;
5764
    pMsgSendInfo->msgType = TDMT_MND_SYSTABLE_RETRIEVE;
H
Haojun Liao 已提交
5765
    pMsgSendInfo->fp = loadSysTableContentCb;
5766 5767 5768 5769

    int64_t transporterId = 0;
    int32_t code = asyncSendMsgToServer(pInfo->pTransporter, &pInfo->epSet, &transporterId, pMsgSendInfo);
    tsem_wait(&pInfo->ready);
H
Haojun Liao 已提交
5770

D
dapan1121 已提交
5771 5772 5773 5774
    if (pTaskInfo->code) {
      return NULL;
    }

H
Haojun Liao 已提交
5775 5776 5777 5778
    SRetrieveMetaTableRsp* pRsp = pInfo->pRsp;
    pInfo->req.showId = pRsp->handle;

    if (pRsp->numOfRows == 0) {
L
Liu Jicong 已提交
5779 5780 5781 5782
      //      qDebug("%s vgId:%d, taskID:0x%"PRIx64" %d of total completed, rowsOfSource:%"PRIu64", totalRows:%"PRIu64"
      //      try next",
      //             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pExchangeInfo->current + 1,
      //             pDataInfo->totalRows, pExchangeInfo->totalRows);
H
Haojun Liao 已提交
5783 5784 5785 5786
      return NULL;
    }

    SRetrieveMetaTableRsp* pTableRsp = pInfo->pRsp;
L
Liu Jicong 已提交
5787 5788
    setSDataBlockFromFetchRsp(pInfo->pRes, &pInfo->loadInfo, pTableRsp->numOfRows, pTableRsp->data, pTableRsp->compLen,
                              pOperator->numOfOutput, startTs, NULL, pInfo->scanCols);
H
Haojun Liao 已提交
5789

H
Haojun Liao 已提交
5790
    return doFilterResult(pInfo);
5791
  }
5792 5793 5794 5795

  return NULL;
}

H
Haojun Liao 已提交
5796
SOperatorInfo* createSysTableScanOperatorInfo(void* pSysTableReadHandle, SSDataBlock* pResBlock, const SName* pName,
L
Liu Jicong 已提交
5797 5798
                                              SNode* pCondition, SEpSet epset, SArray* colList,
                                              SExecTaskInfo* pTaskInfo, bool showRewrite, int32_t accountId) {
wafwerar's avatar
wafwerar 已提交
5799
  SSysTableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SSysTableScanInfo));
L
Liu Jicong 已提交
5800
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
5801
  if (pInfo == NULL || pOperator == NULL) {
wafwerar's avatar
wafwerar 已提交
5802 5803
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
5804 5805 5806 5807
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
5808
  pInfo->accountId = accountId;
D
dapan1121 已提交
5809
  pInfo->showRewrite = showRewrite;
L
Liu Jicong 已提交
5810 5811 5812 5813
  pInfo->pRes = pResBlock;
  pInfo->capacity = 4096;
  pInfo->pCondition = pCondition;
  pInfo->scanCols = colList;
H
Haojun Liao 已提交
5814

H
Haojun Liao 已提交
5815
  // TODO remove it
L
Liu Jicong 已提交
5816
  int32_t     tableType = 0;
H
Haojun Liao 已提交
5817 5818 5819 5820 5821 5822 5823 5824 5825 5826 5827 5828 5829 5830 5831 5832
  const char* name = tNameGetTableName(pName);
  if (strncasecmp(name, TSDB_INS_TABLE_USER_DATABASES, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_DB;
  } else if (strncasecmp(name, TSDB_INS_TABLE_USER_USERS, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_USER;
  } else if (strncasecmp(name, TSDB_INS_TABLE_DNODES, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_DNODE;
  } else if (strncasecmp(name, TSDB_INS_TABLE_MNODES, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_MNODE;
  } else if (strncasecmp(name, TSDB_INS_TABLE_MODULES, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_MODULE;
  } else if (strncasecmp(name, TSDB_INS_TABLE_QNODES, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_QNODE;
  } else if (strncasecmp(name, TSDB_INS_TABLE_USER_FUNCTIONS, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_FUNC;
  } else if (strncasecmp(name, TSDB_INS_TABLE_USER_INDEXES, tListLen(pName->tname)) == 0) {
L
Liu Jicong 已提交
5833
    //    tableType = TSDB_MGMT_TABLE_INDEX;
H
Haojun Liao 已提交
5834 5835 5836 5837 5838 5839 5840 5841 5842
  } else if (strncasecmp(name, TSDB_INS_TABLE_USER_STABLES, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_STB;
  } else if (strncasecmp(name, TSDB_INS_TABLE_USER_STREAMS, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_STREAMS;
  } else if (strncasecmp(name, TSDB_INS_TABLE_USER_TABLES, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_TABLE;
  } else if (strncasecmp(name, TSDB_INS_TABLE_VGROUPS, tListLen(pName->tname)) == 0) {
    tableType = TSDB_MGMT_TABLE_VGROUP;
  } else if (strncasecmp(name, TSDB_INS_TABLE_USER_TABLE_DISTRIBUTED, tListLen(pName->tname)) == 0) {
L
Liu Jicong 已提交
5843
    //    tableType = TSDB_MGMT_TABLE_DIST;
H
Haojun Liao 已提交
5844 5845 5846 5847 5848 5849
  } else {
    ASSERT(0);
  }

  tNameAssign(&pInfo->name, pName);
  pInfo->type = tableType;
5850 5851 5852 5853 5854 5855 5856
  if (pInfo->type == TSDB_MGMT_TABLE_TABLE) {
    pInfo->readHandle = pSysTableReadHandle;
    blockDataEnsureCapacity(pInfo->pRes, pInfo->capacity);
  } else {
    tsem_init(&pInfo->ready, 0, 0);
    pInfo->epSet = epset;

5857
#if 1
L
Liu Jicong 已提交
5858
    {  // todo refactor
5859 5860 5861 5862 5863 5864 5865 5866
      SRpcInit rpcInit;
      memset(&rpcInit, 0, sizeof(rpcInit));
      rpcInit.localPort = 0;
      rpcInit.label = "DB-META";
      rpcInit.numOfThreads = 1;
      rpcInit.cfp = qProcessFetchRsp;
      rpcInit.sessions = tsMaxConnections;
      rpcInit.connType = TAOS_CONN_CLIENT;
L
Liu Jicong 已提交
5867
      rpcInit.user = (char*)"root";
5868 5869 5870
      rpcInit.idleTime = tsShellActivityTimer * 1000;
      rpcInit.ckey = "key";
      rpcInit.spi = 1;
L
Liu Jicong 已提交
5871
      rpcInit.secret = (char*)"dcc5bed04851fec854c035b2e40263b6";
5872 5873 5874

      pInfo->pTransporter = rpcOpen(&rpcInit);
      if (pInfo->pTransporter == NULL) {
L
Liu Jicong 已提交
5875
        return NULL;  // todo
5876 5877 5878
      }
    }
#endif
5879 5880
  }

L
Liu Jicong 已提交
5881 5882 5883 5884 5885 5886 5887 5888 5889
  pOperator->name = "SysTableScanOperator";
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN;
  pOperator->blockingOptr = false;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfOutput = pResBlock->info.numOfCols;
  pOperator->getNextFn = doSysTableScan;
  pOperator->closeFn = destroySysTableScannerOperatorInfo;
  pOperator->pTaskInfo = pTaskInfo;
5890 5891 5892 5893

  return pOperator;
}

H
Haojun Liao 已提交
5894
SArray* getOrderCheckColumns(STaskAttr* pQuery) {
L
Liu Jicong 已提交
5895
  int32_t numOfCols = (pQuery->pGroupbyExpr == NULL) ? 0 : taosArrayGetSize(pQuery->pGroupbyExpr->columnInfo);
5896 5897 5898 5899 5900 5901 5902 5903 5904 5905 5906 5907 5908 5909 5910 5911 5912 5913

  SArray* pOrderColumns = NULL;
  if (numOfCols > 0) {
    pOrderColumns = taosArrayDup(pQuery->pGroupbyExpr->columnInfo);
  } else {
    pOrderColumns = taosArrayInit(4, sizeof(SColIndex));
  }

  if (pQuery->interval.interval > 0) {
    if (pOrderColumns == NULL) {
      pOrderColumns = taosArrayInit(1, sizeof(SColIndex));
    }

    SColIndex colIndex = {.colIndex = 0, .colId = 0, .flag = TSDB_COL_NORMAL};
    taosArrayPush(pOrderColumns, &colIndex);
  }

  {
L
Liu Jicong 已提交
5914 5915
    numOfCols = (int32_t)taosArrayGetSize(pOrderColumns);
    for (int32_t i = 0; i < numOfCols; ++i) {
5916
      SColIndex* index = taosArrayGet(pOrderColumns, i);
L
Liu Jicong 已提交
5917
      for (int32_t j = 0; j < pQuery->numOfOutput; ++j) {
H
Haojun Liao 已提交
5918
        SExprBasicInfo* pExpr = &pQuery->pExpr1[j].base;
L
Liu Jicong 已提交
5919
        int32_t         functionId = getExprFunctionId(&pQuery->pExpr1[j]);
5920

H
Haojun Liao 已提交
5921
        if (index->colId == pExpr->pParam[0].pCol->colId &&
5922 5923 5924 5925 5926 5927 5928 5929 5930 5931 5932
            (functionId == FUNCTION_PRJ || functionId == FUNCTION_TAG || functionId == FUNCTION_TS)) {
          index->colIndex = j;
          index->colId = pExpr->resSchema.colId;
        }
      }
    }
  }

  return pOrderColumns;
}

H
Haojun Liao 已提交
5933
SArray* getResultGroupCheckColumns(STaskAttr* pQuery) {
L
Liu Jicong 已提交
5934
  int32_t numOfCols = (pQuery->pGroupbyExpr == NULL) ? 0 : taosArrayGetSize(pQuery->pGroupbyExpr->columnInfo);
5935 5936 5937 5938 5939 5940 5941 5942

  SArray* pOrderColumns = NULL;
  if (numOfCols > 0) {
    pOrderColumns = taosArrayDup(pQuery->pGroupbyExpr->columnInfo);
  } else {
    pOrderColumns = taosArrayInit(4, sizeof(SColIndex));
  }

L
Liu Jicong 已提交
5943
  for (int32_t i = 0; i < numOfCols; ++i) {
5944 5945 5946
    SColIndex* index = taosArrayGet(pOrderColumns, i);

    bool found = false;
L
Liu Jicong 已提交
5947
    for (int32_t j = 0; j < pQuery->numOfOutput; ++j) {
H
Haojun Liao 已提交
5948
      SExprBasicInfo* pExpr = &pQuery->pExpr1[j].base;
L
Liu Jicong 已提交
5949
      int32_t         functionId = getExprFunctionId(&pQuery->pExpr1[j]);
5950 5951

      // FUNCTION_TAG_DUMMY function needs to be ignored
L
Liu Jicong 已提交
5952 5953 5954 5955 5956 5957 5958 5959
      //      if (index->colId == pExpr->pColumns->info.colId &&
      //          ((TSDB_COL_IS_TAG(pExpr->pColumns->flag) && functionId == FUNCTION_TAG) ||
      //           (TSDB_COL_IS_NORMAL_COL(pExpr->pColumns->flag) && functionId == FUNCTION_PRJ))) {
      //        index->colIndex = j;
      //        index->colId = pExpr->resSchema.colId;
      //        found = true;
      //        break;
      //      }
5960 5961 5962 5963 5964 5965 5966 5967
    }

    assert(found && index->colIndex >= 0 && index->colIndex < pQuery->numOfOutput);
  }

  return pOrderColumns;
}

L
Liu Jicong 已提交
5968 5969
static int32_t doInitAggInfoSup(SAggSupporter* pAggSup, SqlFunctionCtx* pCtx, int32_t numOfOutput, const char* pKey);
static void    cleanupAggSup(SAggSupporter* pAggSup);
5970

5971
static void destroySortedMergeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5972
  SSortedMergeOperatorInfo* pInfo = (SSortedMergeOperatorInfo*)param;
H
Haojun Liao 已提交
5973
  taosArrayDestroy(pInfo->pSortInfo);
5974 5975 5976
  taosArrayDestroy(pInfo->groupInfo);

  if (pInfo->pSortHandle != NULL) {
H
Haojun Liao 已提交
5977
    tsortDestroySortHandle(pInfo->pSortHandle);
5978 5979
  }

H
Haojun Liao 已提交
5980
  blockDataDestroy(pInfo->binfo.pRes);
H
Haojun Liao 已提交
5981
  cleanupAggSup(&pInfo->aggSup);
5982
}
H
Haojun Liao 已提交
5983

5984
static void destroySlimitOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5985
  SSLimitOperatorInfo* pInfo = (SSLimitOperatorInfo*)param;
5986
  taosArrayDestroy(pInfo->orderColumnList);
H
Haojun Liao 已提交
5987
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
wafwerar's avatar
wafwerar 已提交
5988
  taosMemoryFreeClear(pInfo->prevRow);
5989 5990
}

X
Xiaoyu Wang 已提交
5991 5992
static void assignExprInfo(SExprInfo* dst, const SExprInfo* src) {
  assert(dst != NULL && src != NULL);
5993

X
Xiaoyu Wang 已提交
5994
  *dst = *src;
5995

X
Xiaoyu Wang 已提交
5996
  dst->pExpr = exprdup(src->pExpr);
wafwerar's avatar
wafwerar 已提交
5997
  dst->base.pParam = taosMemoryCalloc(src->base.numOfParams, sizeof(SColumn));
H
Haojun Liao 已提交
5998
  memcpy(dst->base.pParam, src->base.pParam, sizeof(SColumn) * src->base.numOfParams);
5999

L
Liu Jicong 已提交
6000 6001 6002 6003
  //  memset(dst->base.param, 0, sizeof(SVariant) * tListLen(dst->base.param));
  //  for (int32_t j = 0; j < src->base.numOfParams; ++j) {
  //    taosVariantAssign(&dst->base.param[j], &src->base.param[j]);
  //  }
X
Xiaoyu Wang 已提交
6004
}
6005

6006 6007
static SExprInfo* exprArrayDup(SArray* pExprList) {
  size_t numOfOutput = taosArrayGetSize(pExprList);
6008

wafwerar's avatar
wafwerar 已提交
6009
  SExprInfo* p = taosMemoryCalloc(numOfOutput, sizeof(SExprInfo));
6010 6011
  for (int32_t i = 0; i < numOfOutput; ++i) {
    SExprInfo* pExpr = taosArrayGetP(pExprList, i);
H
Haojun Liao 已提交
6012
    assignExprInfo(&p[i], pExpr);
6013 6014
  }

H
Haojun Liao 已提交
6015 6016
  return p;
}
6017

H
Haojun Liao 已提交
6018
// TODO merge aggregate super table
L
Liu Jicong 已提交
6019
static void appendOneRowToDataBlock(SSDataBlock* pBlock, STupleHandle* pTupleHandle) {
6020 6021
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
6022

H
Haojun Liao 已提交
6023
    bool isNull = tsortIsNullVal(pTupleHandle, i);
H
Haojun Liao 已提交
6024 6025 6026
    if (isNull) {
      colDataAppend(pColInfo, pBlock->info.rows, NULL, true);
    } else {
H
Haojun Liao 已提交
6027
      char* pData = tsortGetValue(pTupleHandle, i);
H
Haojun Liao 已提交
6028 6029
      colDataAppend(pColInfo, pBlock->info.rows, pData, false);
    }
6030 6031
  }

6032 6033
  pBlock->info.rows += 1;
}
6034

L
Liu Jicong 已提交
6035 6036
static SSDataBlock* getSortedBlockData(SSortHandle* pHandle, SSDataBlock* pDataBlock, bool hasVarCol,
                                       int32_t capacity) {
6037
  blockDataCleanup(pDataBlock);
6038

L
Liu Jicong 已提交
6039
  while (1) {
H
Haojun Liao 已提交
6040
    STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
6041 6042 6043
    if (pTupleHandle == NULL) {
      break;
    }
6044

6045 6046 6047
    appendOneRowToDataBlock(pDataBlock, pTupleHandle);
    if (pDataBlock->info.rows >= capacity) {
      return pDataBlock;
H
Haojun Liao 已提交
6048 6049
    }
  }
6050

L
Liu Jicong 已提交
6051
  return (pDataBlock->info.rows > 0) ? pDataBlock : NULL;
6052
}
6053

6054
SSDataBlock* loadNextDataBlock(void* param) {
L
Liu Jicong 已提交
6055 6056
  SOperatorInfo* pOperator = (SOperatorInfo*)param;
  bool           newgroup = false;
6057

H
Haojun Liao 已提交
6058
  return pOperator->getNextFn(pOperator, &newgroup);
6059 6060
}

L
Liu Jicong 已提交
6061
static bool needToMerge(SSDataBlock* pBlock, SArray* groupInfo, char** buf, int32_t rowIndex) {
6062 6063 6064 6065
  size_t size = taosArrayGetSize(groupInfo);
  if (size == 0) {
    return true;
  }
6066

6067 6068
  for (int32_t i = 0; i < size; ++i) {
    int32_t* index = taosArrayGet(groupInfo, i);
6069

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

6073 6074 6075
    if ((isNull && buf[i] != NULL) || (!isNull && buf[i] == NULL)) {
      return false;
    }
6076

6077 6078 6079 6080 6081 6082 6083 6084 6085 6086 6087 6088 6089
    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;
      }
6090 6091 6092
    }
  }

6093
  return 0;
6094 6095
}

L
Liu Jicong 已提交
6096 6097 6098
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
6099
    pCtx[j].startRow = rowIndex;
6100 6101
  }

6102 6103
  for (int32_t j = 0; j < numOfExpr; ++j) {
    int32_t functionId = pCtx[j].functionId;
L
Liu Jicong 已提交
6104 6105 6106 6107 6108 6109 6110 6111 6112
    //    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]);
    //    }
6113
  }
6114
}
6115

L
Liu Jicong 已提交
6116 6117
static void doFinalizeResultImpl(SqlFunctionCtx* pCtx, int32_t numOfExpr) {
  for (int32_t j = 0; j < numOfExpr; ++j) {
6118 6119 6120 6121
    int32_t functionId = pCtx[j].functionId;
    //    if (functionId == FUNC_TAG_DUMMY || functionId == FUNC_TS_DUMMY) {
    //      continue;
    //    }
6122

6123 6124 6125 6126
    //    if (functionId < 0) {
    //      SUdfInfo* pUdfInfo = taosArrayGet(pInfo->udfInfo, -1 * functionId - 1);
    //      doInvokeUdf(pUdfInfo, &pCtx[j], 0, TSDB_UDF_FUNC_FINALIZE);
    //    } else {
H
Haojun Liao 已提交
6127
    pCtx[j].fpSet.finalize(&pCtx[j]);
6128 6129
  }
}
6130

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

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

6138 6139 6140
    char* data = colDataGetData(pColInfo, rowIndex);
    memcpy(rowColData[i], data, colDataGetLength(pColInfo, rowIndex));
  }
6141

6142 6143
  return true;
}
6144

6145 6146
static void doMergeImpl(SOperatorInfo* pOperator, int32_t numOfExpr, SSDataBlock* pBlock) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
6147

6148
  SqlFunctionCtx* pCtx = pInfo->binfo.pCtx;
L
Liu Jicong 已提交
6149
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
6150 6151
    pCtx[i].size = 1;
  }
6152

L
Liu Jicong 已提交
6153
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
6154 6155 6156 6157 6158 6159 6160 6161 6162
    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);
H
Haojun Liao 已提交
6163
        int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfOutput, NULL);
6164
        //        setTagValueForMultipleRows(pCtx, pOperator->numOfOutput, numOfRows);
6165

6166
        // TODO check for available buffer;
H
Haojun Liao 已提交
6167

6168 6169 6170 6171 6172
        // next group info data
        pInfo->binfo.pRes->info.rows += numOfRows;
        for (int32_t j = 0; j < numOfExpr; ++j) {
          if (pCtx[j].functionId < 0) {
            continue;
6173
          }
6174

H
Haojun Liao 已提交
6175
          pCtx[j].fpSet.process(&pCtx[j]);
6176
        }
6177 6178 6179

        doMergeResultImpl(pInfo, pCtx, numOfExpr, i);
        pInfo->hasGroupVal = saveCurrentTuple(pInfo->groupVal, pInfo->groupInfo, pBlock, i);
H
Haojun Liao 已提交
6180
      }
6181 6182 6183 6184
    }
  }
}

6185 6186
static SSDataBlock* doMerge(SOperatorInfo* pOperator) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
6187
  SSortHandle*              pHandle = pInfo->pSortHandle;
6188

6189 6190 6191
  SSDataBlock* pDataBlock = createOneDataBlock(pInfo->binfo.pRes);
  blockDataEnsureCapacity(pDataBlock, pInfo->binfo.capacity);

L
Liu Jicong 已提交
6192
  while (1) {
6193
    blockDataCleanup(pDataBlock);
6194
    while (1) {
H
Haojun Liao 已提交
6195
      STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
6196 6197
      if (pTupleHandle == NULL) {
        break;
6198
      }
6199

6200 6201 6202 6203 6204
      // build datablock for merge for one group
      appendOneRowToDataBlock(pDataBlock, pTupleHandle);
      if (pDataBlock->info.rows >= pInfo->binfo.capacity) {
        break;
      }
6205
    }
6206

6207 6208 6209
    if (pDataBlock->info.rows == 0) {
      break;
    }
6210

6211
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pDataBlock, TSDB_ORDER_ASC);
L
Liu Jicong 已提交
6212 6213
    //  updateOutputBuf(&pInfo->binfo, &pAggInfo->bufCapacity, pBlock->info.rows * pAggInfo->resultRowFactor,
    //  pOperator->pRuntimeEnv, true);
6214 6215 6216
    doMergeImpl(pOperator, pOperator->numOfOutput, pDataBlock);
    // flush to tuple store, and after all data have been handled, return to upstream node or sink node
  }
6217

6218
  doFinalizeResultImpl(pInfo->binfo.pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
6219
  int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfOutput, NULL);
6220
  //        setTagValueForMultipleRows(pCtx, pOperator->numOfOutput, numOfRows);
6221

6222
  // TODO check for available buffer;
6223

6224 6225
  // next group info data
  pInfo->binfo.pRes->info.rows += numOfRows;
L
Liu Jicong 已提交
6226
  return (pInfo->binfo.pRes->info.rows > 0) ? pInfo->binfo.pRes : NULL;
6227
}
6228

L
Liu Jicong 已提交
6229
static SSDataBlock* doSortedMerge(SOperatorInfo* pOperator, bool* newgroup) {
6230 6231
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
6232 6233
  }

L
Liu Jicong 已提交
6234
  SExecTaskInfo*            pTaskInfo = pOperator->pTaskInfo;
6235
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
H
Haojun Liao 已提交
6236
  if (pOperator->status == OP_RES_TO_RETURN) {
6237
    return getSortedBlockData(pInfo->pSortHandle, pInfo->binfo.pRes, pInfo->hasVarCol, pInfo->binfo.capacity);
6238 6239
  }

6240
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
L
Liu Jicong 已提交
6241 6242
  pInfo->pSortHandle = tsortCreateSortHandle(pInfo->pSortInfo, SORT_MULTISOURCE_MERGE, pInfo->bufPageSize, numOfBufPage,
                                             pInfo->binfo.pRes, "GET_TASKID(pTaskInfo)");
H
Haojun Liao 已提交
6243

H
Haojun Liao 已提交
6244
  tsortSetFetchRawDataFp(pInfo->pSortHandle, loadNextDataBlock);
6245

L
Liu Jicong 已提交
6246
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
wafwerar's avatar
wafwerar 已提交
6247
    SGenericSource* ps = taosMemoryCalloc(1, sizeof(SGenericSource));
H
Haojun Liao 已提交
6248
    ps->param = pOperator->pDownstream[i];
H
Haojun Liao 已提交
6249
    tsortAddSource(pInfo->pSortHandle, ps);
6250 6251
  }

H
Haojun Liao 已提交
6252
  int32_t code = tsortOpen(pInfo->pSortHandle);
6253
  if (code != TSDB_CODE_SUCCESS) {
6254
    longjmp(pTaskInfo->env, terrno);
6255 6256
  }

H
Haojun Liao 已提交
6257
  pOperator->status = OP_RES_TO_RETURN;
6258
  return doMerge(pOperator);
6259
}
6260

L
Liu Jicong 已提交
6261 6262
static int32_t initGroupCol(SExprInfo* pExprInfo, int32_t numOfCols, SArray* pGroupInfo,
                            SSortedMergeOperatorInfo* pInfo) {
6263 6264
  if (pGroupInfo == NULL || taosArrayGetSize(pGroupInfo) == 0) {
    return 0;
H
Haojun Liao 已提交
6265 6266
  }

6267 6268 6269 6270 6271 6272 6273 6274
  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 已提交
6275 6276
  size_t numOfGroupCol = taosArrayGetSize(pInfo->groupInfo);
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
6277
    SColumn* pCol = taosArrayGet(pGroupInfo, i);
L
Liu Jicong 已提交
6278
    for (int32_t j = 0; j < numOfCols; ++j) {
H
Haojun Liao 已提交
6279
      SExprInfo* pe = &pExprInfo[j];
H
Haojun Liao 已提交
6280
      if (pe->base.resSchema.colId == pCol->colId) {
6281 6282
        taosArrayPush(plist, pCol);
        taosArrayPush(pInfo->groupInfo, &j);
H
Haojun Liao 已提交
6283
        len += pCol->bytes;
6284 6285
        break;
      }
H
Haojun Liao 已提交
6286 6287 6288
    }
  }

6289
  ASSERT(taosArrayGetSize(pGroupInfo) == taosArrayGetSize(plist));
H
Haojun Liao 已提交
6290

wafwerar's avatar
wafwerar 已提交
6291
  pInfo->groupVal = taosMemoryCalloc(1, (POINTER_BYTES * numOfGroupCol + len));
6292 6293 6294 6295
  if (pInfo->groupVal == NULL) {
    taosArrayDestroy(plist);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
H
Haojun Liao 已提交
6296

6297
  int32_t offset = 0;
L
Liu Jicong 已提交
6298 6299
  char*   start = (char*)(pInfo->groupVal + (POINTER_BYTES * numOfGroupCol));
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
6300 6301
    pInfo->groupVal[i] = start + offset;
    SColumn* pCol = taosArrayGet(plist, i);
H
Haojun Liao 已提交
6302
    offset += pCol->bytes;
6303
  }
H
Haojun Liao 已提交
6304

6305
  taosArrayDestroy(plist);
H
Haojun Liao 已提交
6306

6307 6308
  return TSDB_CODE_SUCCESS;
}
H
Haojun Liao 已提交
6309

L
Liu Jicong 已提交
6310 6311 6312
SOperatorInfo* createSortedMergeOperatorInfo(SOperatorInfo** downstream, int32_t numOfDownstream, SExprInfo* pExprInfo,
                                             int32_t num, SArray* pSortInfo, SArray* pGroupInfo,
                                             SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
6313
  SSortedMergeOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSortedMergeOperatorInfo));
L
Liu Jicong 已提交
6314
  SOperatorInfo*            pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
6315
  if (pInfo == NULL || pOperator == NULL) {
6316
    goto _error;
6317
  }
H
Haojun Liao 已提交
6318

L
Liu Jicong 已提交
6319
  pInfo->binfo.pCtx = createSqlFunctionCtx_rv(pExprInfo, num, &pInfo->binfo.rowCellInfoOffset);
6320
  initResultRowInfo(&pInfo->binfo.resultRowInfo, (int32_t)1);
H
Haojun Liao 已提交
6321

6322 6323 6324
  if (pInfo->binfo.pCtx == NULL || pInfo->binfo.pRes == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
6325

H
Haojun Liao 已提交
6326
  int32_t code = doInitAggInfoSup(&pInfo->aggSup, pInfo->binfo.pCtx, num, pTaskInfo->id.str);
6327 6328 6329
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
6330

H
Haojun Liao 已提交
6331
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, pTaskInfo);
H
Haojun Liao 已提交
6332
  code = initGroupCol(pExprInfo, num, pGroupInfo, pInfo);
6333 6334 6335
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
6336

L
Liu Jicong 已提交
6337 6338 6339 6340 6341
  //  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 已提交
6342

6343
  pInfo->binfo.capacity = blockDataGetCapacityInRow(pInfo->binfo.pRes, pInfo->bufPageSize);
H
Haojun Liao 已提交
6344

L
Liu Jicong 已提交
6345
  pOperator->name = "SortedMerge";
X
Xiaoyu Wang 已提交
6346
  // pOperator->operatorType = OP_SortedMerge;
6347
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
6348 6349 6350 6351
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfOutput = num;
  pOperator->pExpr = pExprInfo;
H
Haojun Liao 已提交
6352

L
Liu Jicong 已提交
6353 6354 6355
  pOperator->pTaskInfo = pTaskInfo;
  pOperator->getNextFn = doSortedMerge;
  pOperator->closeFn = destroySortedMergeOperatorInfo;
H
Haojun Liao 已提交
6356

6357 6358 6359
  code = appendDownstream(pOperator, downstream, numOfDownstream);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
6360
  }
H
Haojun Liao 已提交
6361

6362
  return pOperator;
H
Haojun Liao 已提交
6363

L
Liu Jicong 已提交
6364
_error:
6365
  if (pInfo != NULL) {
H
Haojun Liao 已提交
6366
    destroySortedMergeOperatorInfo(pInfo, num);
H
Haojun Liao 已提交
6367 6368
  }

wafwerar's avatar
wafwerar 已提交
6369 6370
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
6371 6372
  terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
  return NULL;
H
Haojun Liao 已提交
6373 6374
}

L
Liu Jicong 已提交
6375
static SSDataBlock* doSort(SOperatorInfo* pOperator, bool* newgroup) {
6376 6377 6378 6379
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

L
Liu Jicong 已提交
6380
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
6381
  SSortOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
6382
  bool               hasVarCol = pInfo->pDataBlock->info.hasVarCol;
H
Haojun Liao 已提交
6383

H
Haojun Liao 已提交
6384
  if (pOperator->status == OP_RES_TO_RETURN) {
H
Haojun Liao 已提交
6385
    return getSortedBlockData(pInfo->pSortHandle, pInfo->pDataBlock, hasVarCol, pInfo->numOfRowsInRes);
H
Haojun Liao 已提交
6386 6387
  }

6388
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
L
Liu Jicong 已提交
6389 6390
  pInfo->pSortHandle = tsortCreateSortHandle(pInfo->pSortInfo, SORT_SINGLESOURCE_SORT, pInfo->bufPageSize, numOfBufPage,
                                             pInfo->pDataBlock, "GET_TASKID(pTaskInfo)");
6391

H
Haojun Liao 已提交
6392
  tsortSetFetchRawDataFp(pInfo->pSortHandle, loadNextDataBlock);
H
Haojun Liao 已提交
6393

wafwerar's avatar
wafwerar 已提交
6394
  SGenericSource* ps = taosMemoryCalloc(1, sizeof(SGenericSource));
H
Haojun Liao 已提交
6395
  ps->param = pOperator->pDownstream[0];
H
Haojun Liao 已提交
6396
  tsortAddSource(pInfo->pSortHandle, ps);
6397

H
Haojun Liao 已提交
6398
  int32_t code = tsortOpen(pInfo->pSortHandle);
6399
  if (code != TSDB_CODE_SUCCESS) {
6400
    longjmp(pTaskInfo->env, terrno);
6401
  }
6402

H
Haojun Liao 已提交
6403
  pOperator->status = OP_RES_TO_RETURN;
H
Haojun Liao 已提交
6404
  return getSortedBlockData(pInfo->pSortHandle, pInfo->pDataBlock, hasVarCol, pInfo->numOfRowsInRes);
6405 6406
}

L
Liu Jicong 已提交
6407 6408
SOperatorInfo* createSortOperatorInfo(SOperatorInfo* downstream, SSDataBlock* pResBlock, SArray* pSortInfo,
                                      SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
6409
  SSortOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSortOperatorInfo));
L
Liu Jicong 已提交
6410
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
6411
  if (pInfo == NULL || pOperator == NULL) {
wafwerar's avatar
wafwerar 已提交
6412 6413
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
6414 6415 6416
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
  }
6417

L
Liu Jicong 已提交
6418 6419 6420 6421 6422
  pInfo->sortBufSize = 1024 * 16;  // 1MB, TODO dynamic set the available sort buffer
  pInfo->bufPageSize = 1024;
  pInfo->numOfRowsInRes = 1024;
  pInfo->pDataBlock = pResBlock;
  pInfo->pSortInfo = pSortInfo;
H
Haojun Liao 已提交
6423

L
Liu Jicong 已提交
6424 6425 6426 6427 6428
  pOperator->name = "Sort";
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SORT;
  pOperator->blockingOptr = true;
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
6429

L
Liu Jicong 已提交
6430 6431 6432
  pOperator->pTaskInfo = pTaskInfo;
  pOperator->getNextFn = doSort;
  pOperator->closeFn = destroyOrderOperatorInfo;
6433

6434
  int32_t code = appendDownstream(pOperator, &downstream, 1);
6435 6436 6437
  return pOperator;
}

L
Liu Jicong 已提交
6438
static int32_t getTableScanOrder(STableScanInfo* pTableScanInfo) { return pTableScanInfo->order; }
6439 6440

// this is a blocking operator
L
Liu Jicong 已提交
6441
static int32_t doOpenAggregateOptr(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
6442 6443
  if (OPTR_IS_OPENED(pOperator)) {
    return TSDB_CODE_SUCCESS;
6444 6445 6446
  }

  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
6447
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;
6448

H
Haojun Liao 已提交
6449
  int32_t        order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
6450
  SOperatorInfo* downstream = pOperator->pDownstream[0];
6451

H
Haojun Liao 已提交
6452 6453
  bool newgroup = true;
  while (1) {
H
Haojun Liao 已提交
6454
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
6455
    SSDataBlock* pBlock = downstream->getNextFn(downstream, &newgroup);
H
Haojun Liao 已提交
6456
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
6457 6458 6459 6460 6461

    if (pBlock == NULL) {
      break;
    }

H
Haojun Liao 已提交
6462 6463 6464
    //    if (pAggInfo->current != NULL) {
    //      setTagValue(pOperator, pAggInfo->current->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    }
6465 6466 6467

    // the pDataBlock are always the same one, no need to call this again
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order);
H
Haojun Liao 已提交
6468
    doAggregateImpl(pOperator, 0, pInfo->pCtx);
6469 6470
  }

6471
  finalizeQueryResult(pInfo->pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
6472 6473 6474 6475 6476

  OPTR_SET_OPENED(pOperator);
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
6477 6478
static SSDataBlock* getAggregateResult(SOperatorInfo* pOperator, bool* newgroup) {
  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
6479 6480 6481 6482 6483 6484
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;

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

L
Liu Jicong 已提交
6485
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
6486 6487 6488 6489 6490
  pTaskInfo->code = pOperator->_openFn(pOperator);
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
    return NULL;
  }

H
Haojun Liao 已提交
6491
  getNumOfResult(pInfo->pCtx, pOperator->numOfOutput, pInfo->pRes);
H
Haojun Liao 已提交
6492
  doSetOperatorCompleted(pOperator);
6493

L
Liu Jicong 已提交
6494
  return (blockDataGetNumOfRows(pInfo->pRes) != 0) ? pInfo->pRes : NULL;
6495 6496
}

L
Liu Jicong 已提交
6497 6498 6499
static void aggEncodeResultRow(SOperatorInfo* pOperator, char** result, int32_t* length) {
  SAggOperatorInfo* pAggInfo = pOperator->info;
  SAggSupporter*    pSup = &pAggInfo->aggSup;
wmmhello's avatar
wmmhello 已提交
6500 6501

  int32_t size = taosHashGetSize(pSup->pResultRowHashTable);
L
Liu Jicong 已提交
6502
  size_t  keyLen = POINTER_BYTES;  // estimate the key length
wmmhello's avatar
wmmhello 已提交
6503
  int32_t totalSize = sizeof(int32_t) + size * (sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize);
wafwerar's avatar
wafwerar 已提交
6504
  *result = taosMemoryCalloc(1, totalSize);
L
Liu Jicong 已提交
6505
  if (*result == NULL) {
wmmhello's avatar
wmmhello 已提交
6506 6507 6508 6509 6510
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return;
  }
  *(int32_t*)(*result) = size;
  int32_t offset = sizeof(int32_t);
L
Liu Jicong 已提交
6511
  void*   pIter = taosHashIterate(pSup->pResultRowHashTable, NULL);
wmmhello's avatar
wmmhello 已提交
6512
  while (pIter) {
L
Liu Jicong 已提交
6513 6514
    void*        key = taosHashGetKey(pIter, &keyLen);
    SResultRow** p1 = (SResultRow**)pIter;
wmmhello's avatar
wmmhello 已提交
6515 6516 6517

    // recalculate the result size
    int32_t realTotalSize = offset + sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize;
L
Liu Jicong 已提交
6518 6519 6520
    if (realTotalSize > totalSize) {
      char* tmp = taosMemoryRealloc(*result, realTotalSize);
      if (tmp == NULL) {
wmmhello's avatar
wmmhello 已提交
6521
        terrno = TSDB_CODE_OUT_OF_MEMORY;
wafwerar's avatar
wafwerar 已提交
6522
        taosMemoryFree(*result);
wmmhello's avatar
wmmhello 已提交
6523 6524
        *result = NULL;
        return;
L
Liu Jicong 已提交
6525
      } else {
wmmhello's avatar
wmmhello 已提交
6526 6527 6528 6529 6530 6531 6532 6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543
        *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);
    memcpy(*result + offset, *p1, pSup->resultRowSize);
    offset += pSup->resultRowSize;

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

L
Liu Jicong 已提交
6544
  if (length) {
wmmhello's avatar
wmmhello 已提交
6545 6546 6547 6548 6549
    *length = offset;
  }
  return;
}

L
Liu Jicong 已提交
6550 6551
static bool aggDecodeResultRow(SOperatorInfo* pOperator, char* result, int32_t length) {
  if (!result || length <= 0) {
wmmhello's avatar
wmmhello 已提交
6552 6553 6554
    return false;
  }

L
Liu Jicong 已提交
6555 6556 6557
  SAggOperatorInfo* pAggInfo = pOperator->info;
  SAggSupporter*    pSup = &pAggInfo->aggSup;
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;
wmmhello's avatar
wmmhello 已提交
6558 6559 6560 6561 6562

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

  int32_t offset = sizeof(int32_t);
L
Liu Jicong 已提交
6563
  while (count-- > 0 && length > offset) {
wmmhello's avatar
wmmhello 已提交
6564 6565 6566
    int32_t keyLen = *(int32_t*)(result + offset);
    offset += sizeof(int32_t);

L
Liu Jicong 已提交
6567 6568 6569
    uint64_t    tableGroupId = *(uint64_t*)(result + offset);
    SResultRow* resultRow = getNewResultRow_rv(pSup->pResultBuf, tableGroupId, pSup->resultRowSize);
    if (!resultRow) {
wmmhello's avatar
wmmhello 已提交
6570 6571 6572 6573 6574 6575 6576 6577
      terrno = TSDB_CODE_TSC_INVALID_INPUT;
      return false;
    }
    // add a new result set for a new group
    taosHashPut(pSup->pResultRowHashTable, result + offset, keyLen, &resultRow, POINTER_BYTES);

    offset += keyLen;
    int32_t valueLen = *(int32_t*)(result + offset);
L
Liu Jicong 已提交
6578
    if (valueLen != pSup->resultRowSize) {
wmmhello's avatar
wmmhello 已提交
6579 6580 6581 6582 6583 6584 6585 6586 6587 6588 6589 6590
      terrno = TSDB_CODE_TSC_INVALID_INPUT;
      return false;
    }
    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);
L
Liu Jicong 已提交
6591 6592
    pInfo->resultRowInfo.pPosition[pInfo->resultRowInfo.size++] =
        (SResultRowPosition){.pageId = resultRow->pageId, .offset = resultRow->offset};
wmmhello's avatar
wmmhello 已提交
6593 6594
  }

L
Liu Jicong 已提交
6595
  if (offset != length) {
wmmhello's avatar
wmmhello 已提交
6596 6597 6598 6599 6600 6601
    terrno = TSDB_CODE_TSC_INVALID_INPUT;
    return false;
  }
  return true;
}

L
Liu Jicong 已提交
6602
static SSDataBlock* doMultiTableAggregate(SOperatorInfo* pOperator, bool* newgroup) {
6603 6604 6605 6606 6607
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  SAggOperatorInfo* pAggInfo = pOperator->info;
L
Liu Jicong 已提交
6608 6609
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;
  SExecTaskInfo*    pTaskInfo = pOperator->pTaskInfo;
6610 6611

  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
6612 6613
    toSDatablock(&pAggInfo->groupResInfo, pAggInfo->pResultBuf, pInfo->pRes, pAggInfo->binfo.capacity,
                 pAggInfo->binfo.rowCellInfoOffset);
6614

H
Haojun Liao 已提交
6615
    if (pInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pAggInfo->groupResInfo)) {
6616 6617 6618 6619 6620 6621
      pOperator->status = OP_EXEC_DONE;
    }

    return pInfo->pRes;
  }

6622
  // table scan order
L
Liu Jicong 已提交
6623
  int32_t        order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
6624
  SOperatorInfo* downstream = pOperator->pDownstream[0];
6625

L
Liu Jicong 已提交
6626
  while (1) {
H
Haojun Liao 已提交
6627
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
6628
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
6629
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
6630 6631 6632 6633 6634

    if (pBlock == NULL) {
      break;
    }

L
Liu Jicong 已提交
6635 6636 6637 6638 6639
    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    if (downstream->operatorType == OP_TableScan) {
    //      STableScanInfo* pScanInfo = downstream->info;
    //      order = getTableScanOrder(pScanInfo);
    //    }
6640 6641 6642 6643 6644

    // the pDataBlock are always the same one, no need to call this again
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order);

    TSKEY key = 0;
6645
    if (order == TSDB_ORDER_ASC) {
6646 6647 6648 6649 6650 6651
      key = pBlock->info.window.ekey;
      TSKEY_MAX_ADD(key, 1);
    } else {
      key = pBlock->info.window.skey;
      TSKEY_MIN_SUB(key, -1);
    }
L
Liu Jicong 已提交
6652 6653 6654

    setExecutionContext(pOperator->numOfOutput, pAggInfo->current->groupIndex, key, pTaskInfo, pAggInfo->current,
                        pAggInfo);
H
Haojun Liao 已提交
6655
    doAggregateImpl(pOperator, 0, pInfo->pCtx);
6656 6657 6658 6659
  }

  pOperator->status = OP_RES_TO_RETURN;
  closeAllResultRows(&pInfo->resultRowInfo);
6660
  updateNumOfRowsInResultRows(pInfo->pCtx, pOperator->numOfOutput, &pInfo->resultRowInfo, pInfo->rowCellInfoOffset);
6661

H
Haojun Liao 已提交
6662
  initGroupResInfo(&pAggInfo->groupResInfo, &pInfo->resultRowInfo);
L
Liu Jicong 已提交
6663 6664
  toSDatablock(&pAggInfo->groupResInfo, pAggInfo->pResultBuf, pInfo->pRes, pAggInfo->binfo.capacity,
               pAggInfo->binfo.rowCellInfoOffset);
6665

H
Haojun Liao 已提交
6666
  if (pInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pAggInfo->groupResInfo)) {
6667 6668 6669 6670 6671 6672
    doSetOperatorCompleted(pOperator);
  }

  return pInfo->pRes;
}

L
Liu Jicong 已提交
6673
static SSDataBlock* doProjectOperation(SOperatorInfo* pOperator, bool* newgroup) {
6674
  SProjectOperatorInfo* pProjectInfo = pOperator->info;
L
Liu Jicong 已提交
6675
  SOptrBasicInfo*       pInfo = &pProjectInfo->binfo;
6676 6677

  SSDataBlock* pRes = pInfo->pRes;
6678
  blockDataCleanup(pRes);
6679 6680 6681 6682 6683 6684 6685

  if (pProjectInfo->existDataBlock) {  // TODO refactor
    SSDataBlock* pBlock = pProjectInfo->existDataBlock;
    pProjectInfo->existDataBlock = NULL;
    *newgroup = true;

    // todo dynamic set tags
L
Liu Jicong 已提交
6686 6687 6688
    //    if (pTableQueryInfo != NULL) {
    //      setTagValue(pOperator, pTableQueryInfo->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    }
6689 6690

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

H
Haojun Liao 已提交
6693
    blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
6694
    projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfOutput);
L
Liu Jicong 已提交
6695
    if (pRes->info.rows >= pProjectInfo->binfo.capacity * 0.8) {
6696 6697 6698 6699 6700 6701
      copyTsColoum(pRes, pInfo->pCtx, pOperator->numOfOutput);
      resetResultRowEntryResult(pInfo->pCtx, pOperator->numOfOutput);
      return pRes;
    }
  }

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

L
Liu Jicong 已提交
6704
  while (1) {
6705 6706
    bool prevVal = *newgroup;

H
Haojun Liao 已提交
6707
    // The downstream exec may change the value of the newgroup, so use a local variable instead.
H
Haojun Liao 已提交
6708 6709 6710
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
6711 6712 6713 6714

    if (pBlock == NULL) {
      assert(*newgroup == false);
      *newgroup = prevVal;
6715
      setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
6716 6717 6718 6719 6720 6721 6722 6723
      break;
    }

    // Return result of the previous group in the firstly.
    if (*newgroup) {
      if (pRes->info.rows > 0) {
        pProjectInfo->existDataBlock = pBlock;
        break;
L
Liu Jicong 已提交
6724
      } else {  // init output buffer for a new group data
6725 6726 6727 6728 6729
        initCtxOutputBuffer(pInfo->pCtx, pOperator->numOfOutput);
      }
    }

    // todo dynamic set tags
H
Haojun Liao 已提交
6730 6731 6732 6733
    //    STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
    //    if (pTableQueryInfo != NULL) {
    //      setTagValue(pOperator, pTableQueryInfo->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    }
6734 6735

    // the pDataBlock are always the same one, no need to call this again
H
Haojun Liao 已提交
6736
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, TSDB_ORDER_ASC);
6737 6738
    blockDataEnsureCapacity(pInfo->pRes, pInfo->pRes->info.rows + pBlock->info.rows);

6739 6740
    projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfOutput);
    if (pRes->info.rows >= pOperator->resultInfo.threshold) {
6741 6742 6743
      break;
    }
  }
H
Haojun Liao 已提交
6744

L
Liu Jicong 已提交
6745 6746
  //  copyTsColoum(pRes, pInfo->pCtx, pOperator->numOfOutput);
  return (pInfo->pRes->info.rows > 0) ? pInfo->pRes : NULL;
6747 6748
}

L
Liu Jicong 已提交
6749
static SSDataBlock* doLimit(SOperatorInfo* pOperator, bool* newgroup) {
6750 6751 6752 6753 6754 6755
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  SLimitOperatorInfo* pInfo = pOperator->info;

L
Liu Jicong 已提交
6756
  SSDataBlock*   pBlock = NULL;
H
Haojun Liao 已提交
6757 6758
  SOperatorInfo* pDownstream = pOperator->pDownstream[0];

6759
  while (1) {
H
Haojun Liao 已提交
6760 6761 6762
    publishOperatorProfEvent(pDownstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
    pBlock = pDownstream->getNextFn(pDownstream, newgroup);
    publishOperatorProfEvent(pDownstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
6763 6764 6765 6766 6767 6768

    if (pBlock == NULL) {
      doSetOperatorCompleted(pOperator);
      return NULL;
    }

H
Haojun Liao 已提交
6769
    if (pInfo->currentOffset == 0) {
6770
      break;
H
Haojun Liao 已提交
6771 6772
    } else if (pInfo->currentOffset >= pBlock->info.rows) {
      pInfo->currentOffset -= pBlock->info.rows;
L
Liu Jicong 已提交
6773
    } else {  // TODO handle the data movement
H
Haojun Liao 已提交
6774
      int32_t remain = (int32_t)(pBlock->info.rows - pInfo->currentOffset);
6775 6776 6777 6778 6779 6780
      pBlock->info.rows = remain;

      for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
        SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, i);

        int16_t bytes = pColInfoData->info.bytes;
H
Haojun Liao 已提交
6781
        memmove(pColInfoData->pData, pColInfoData->pData + bytes * pInfo->currentOffset, remain * bytes);
6782 6783
      }

H
Haojun Liao 已提交
6784
      pInfo->currentOffset = 0;
6785 6786 6787 6788
      break;
    }
  }

H
Haojun Liao 已提交
6789 6790 6791
  if (pInfo->currentRows + pBlock->info.rows >= pInfo->limit.limit) {
    pBlock->info.rows = (int32_t)(pInfo->limit.limit - pInfo->currentRows);
    pInfo->currentRows = pInfo->limit.limit;
6792 6793 6794

    doSetOperatorCompleted(pOperator);
  } else {
H
Haojun Liao 已提交
6795
    pInfo->currentRows += pBlock->info.rows;
6796 6797 6798 6799 6800 6801
  }

  return pBlock;
}

static SSDataBlock* doFilter(void* param, bool* newgroup) {
L
Liu Jicong 已提交
6802
  SOperatorInfo* pOperator = (SOperatorInfo*)param;
6803 6804 6805 6806 6807
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  SFilterOperatorInfo* pCondInfo = pOperator->info;
L
Liu Jicong 已提交
6808
  STaskRuntimeEnv*     pRuntimeEnv = pOperator->pRuntimeEnv;
6809 6810

  while (1) {
H
Haojun Liao 已提交
6811
    publishOperatorProfEvent(pOperator->pDownstream[0], QUERY_PROF_BEFORE_OPERATOR_EXEC);
L
Liu Jicong 已提交
6812
    SSDataBlock* pBlock = pOperator->pDownstream[0]->getNextFn(pOperator->pDownstream[0], newgroup);
H
Haojun Liao 已提交
6813
    publishOperatorProfEvent(pOperator->pDownstream[0], QUERY_PROF_AFTER_OPERATOR_EXEC);
6814 6815 6816 6817 6818 6819 6820 6821 6822 6823 6824 6825 6826 6827 6828 6829 6830 6831

    if (pBlock == NULL) {
      break;
    }

    doSetFilterColumnInfo(pCondInfo->pFilterInfo, pCondInfo->numOfFilterCols, pBlock);
    assert(pRuntimeEnv->pTsBuf == NULL);
    filterRowsInDataBlock(pRuntimeEnv, pCondInfo->pFilterInfo, pCondInfo->numOfFilterCols, pBlock, true);

    if (pBlock->info.rows > 0) {
      return pBlock;
    }
  }

  doSetOperatorCompleted(pOperator);
  return NULL;
}

L
Liu Jicong 已提交
6832
static int32_t doOpenIntervalAgg(SOperatorInfo* pOperator) {
6833 6834
  if (OPTR_IS_OPENED(pOperator)) {
    return TSDB_CODE_SUCCESS;
6835 6836
  }

H
Haojun Liao 已提交
6837
  STableIntervalOperatorInfo* pInfo = pOperator->info;
6838

6839
  int32_t order = TSDB_ORDER_ASC;
6840
  //  STimeWindow win = pQueryAttr->window;
L
Liu Jicong 已提交
6841
  bool           newgroup = false;
H
Haojun Liao 已提交
6842
  SOperatorInfo* downstream = pOperator->pDownstream[0];
6843

6844
  while (1) {
H
Haojun Liao 已提交
6845
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
6846
    SSDataBlock* pBlock = downstream->getNextFn(downstream, &newgroup);
H
Haojun Liao 已提交
6847
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
6848 6849 6850 6851 6852

    if (pBlock == NULL) {
      break;
    }

6853
    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pInfo->pCtx, pOperator->numOfOutput);
6854 6855

    // the pDataBlock are always the same one, no need to call this again
6856
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pBlock, order);
H
Haojun Liao 已提交
6857
    hashIntervalAgg(pOperator, &pInfo->binfo.resultRowInfo, pBlock, 0);
6858 6859
  }

H
Haojun Liao 已提交
6860
  closeAllResultRows(&pInfo->binfo.resultRowInfo);
L
Liu Jicong 已提交
6861 6862
  finalizeMultiTupleQueryResult(pInfo->binfo.pCtx, pOperator->numOfOutput, pInfo->aggSup.pResultBuf,
                                &pInfo->binfo.resultRowInfo, pInfo->binfo.rowCellInfoOffset);
6863

H
Haojun Liao 已提交
6864
  initGroupResInfo(&pInfo->groupResInfo, &pInfo->binfo.resultRowInfo);
6865 6866 6867 6868
  OPTR_SET_OPENED(pOperator);
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
6869
static SSDataBlock* doBuildIntervalResult(SOperatorInfo* pOperator, bool* newgroup) {
6870
  STableIntervalOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
6871
  SExecTaskInfo*              pTaskInfo = pOperator->pTaskInfo;
6872 6873 6874 6875 6876 6877 6878 6879 6880 6881 6882

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

  pTaskInfo->code = pOperator->_openFn(pOperator);
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
    return NULL;
  }

  blockDataEnsureCapacity(pInfo->binfo.pRes, pInfo->binfo.capacity);
L
Liu Jicong 已提交
6883 6884
  toSDatablock(&pInfo->groupResInfo, pInfo->aggSup.pResultBuf, pInfo->binfo.pRes, pInfo->binfo.capacity,
               pInfo->binfo.rowCellInfoOffset);
6885

H
Haojun Liao 已提交
6886
  if (pInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
6887 6888 6889
    doSetOperatorCompleted(pOperator);
  }

L
Liu Jicong 已提交
6890
  return pInfo->binfo.pRes->info.rows == 0 ? NULL : pInfo->binfo.pRes;
6891 6892
}

L
Liu Jicong 已提交
6893
static SSDataBlock* doAllIntervalAgg(SOperatorInfo* pOperator, bool* newgroup) {
6894 6895 6896 6897 6898 6899
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  STableIntervalOperatorInfo* pIntervalInfo = pOperator->info;

H
Haojun Liao 已提交
6900
  STaskRuntimeEnv* pRuntimeEnv = pOperator->pRuntimeEnv;
6901
  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
6902
    //    toSDatablock(&pRuntimeEnv->groupResInfo, pRuntimeEnv, pIntervalInfo->pRes);
6903

H
Haojun Liao 已提交
6904
    if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pRuntimeEnv->groupResInfo)) {
6905 6906 6907
      doSetOperatorCompleted(pOperator);
    }

H
Haojun Liao 已提交
6908
    return pIntervalInfo->binfo.pRes;
6909 6910
  }

L
Liu Jicong 已提交
6911 6912
  STaskAttr*  pQueryAttr = pRuntimeEnv->pQueryAttr;
  int32_t     order = pQueryAttr->order.order;
6913 6914
  STimeWindow win = pQueryAttr->window;

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

L
Liu Jicong 已提交
6917
  while (1) {
H
Haojun Liao 已提交
6918
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
6919
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
6920
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
6921 6922 6923 6924 6925

    if (pBlock == NULL) {
      break;
    }

L
Liu Jicong 已提交
6926
    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pIntervalInfo->pCtx, pOperator->numOfOutput);
6927 6928

    // the pDataBlock are always the same one, no need to call this again
H
Haojun Liao 已提交
6929 6930
    setInputDataBlock(pOperator, pIntervalInfo->binfo.pCtx, pBlock, pQueryAttr->order.order);
    hashAllIntervalAgg(pOperator, &pIntervalInfo->binfo.resultRowInfo, pBlock, 0);
6931 6932 6933 6934 6935 6936 6937
  }

  // restore the value
  pQueryAttr->order.order = order;
  pQueryAttr->window = win;

  pOperator->status = OP_RES_TO_RETURN;
H
Haojun Liao 已提交
6938
  closeAllResultRows(&pIntervalInfo->binfo.resultRowInfo);
6939
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
6940
  finalizeQueryResult(pIntervalInfo->binfo.pCtx, pOperator->numOfOutput);
6941

H
Haojun Liao 已提交
6942
  initGroupResInfo(&pRuntimeEnv->groupResInfo, &pIntervalInfo->binfo.resultRowInfo);
L
Liu Jicong 已提交
6943
  //  toSDatablock(&pRuntimeEnv->groupResInfo, pRuntimeEnv, pIntervalInfo->pRes);
6944

H
Haojun Liao 已提交
6945
  if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pRuntimeEnv->groupResInfo)) {
6946 6947 6948
    pOperator->status = OP_EXEC_DONE;
  }

L
Liu Jicong 已提交
6949
  return pIntervalInfo->binfo.pRes->info.rows == 0 ? NULL : pIntervalInfo->binfo.pRes;
6950 6951
}

L
Liu Jicong 已提交
6952
static SSDataBlock* doSTableIntervalAgg(SOperatorInfo* pOperator, bool* newgroup) {
6953 6954 6955 6956 6957
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  STableIntervalOperatorInfo* pIntervalInfo = pOperator->info;
L
Liu Jicong 已提交
6958
  STaskRuntimeEnv*            pRuntimeEnv = pOperator->pRuntimeEnv;
6959 6960 6961 6962

  if (pOperator->status == OP_RES_TO_RETURN) {
    int64_t st = taosGetTimestampUs();

L
Liu Jicong 已提交
6963
    //    copyToSDataBlock(NULL, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
6964
    if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
6965 6966
      doSetOperatorCompleted(pOperator);
    }
H
Haojun Liao 已提交
6967
    return pIntervalInfo->binfo.pRes;
6968 6969
  }

H
Haojun Liao 已提交
6970
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
L
Liu Jicong 已提交
6971
  int32_t    order = pQueryAttr->order.order;
6972

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

L
Liu Jicong 已提交
6975
  while (1) {
H
Haojun Liao 已提交
6976
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
6977
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
6978
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
6979 6980 6981 6982 6983 6984 6985 6986

    if (pBlock == NULL) {
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
    STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;

L
Liu Jicong 已提交
6987
    //    setTagValue(pOperator, pTableQueryInfo->pTable, pIntervalInfo->pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
6988
    setInputDataBlock(pOperator, pIntervalInfo->binfo.pCtx, pBlock, pQueryAttr->order.order);
6989 6990 6991 6992 6993 6994
    setIntervalQueryRange(pRuntimeEnv, pBlock->info.window.skey);

    hashIntervalAgg(pOperator, &pTableQueryInfo->resInfo, pBlock, pTableQueryInfo->groupIndex);
  }

  pOperator->status = OP_RES_TO_RETURN;
L
Liu Jicong 已提交
6995
  pQueryAttr->order.order = order;  // TODO : restore the order
6996
  doCloseAllTimeWindow(pRuntimeEnv);
6997
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
6998

L
Liu Jicong 已提交
6999
  //  copyToSDataBlock(pRuntimeEnv, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
7000
  if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
7001 7002 7003
    pOperator->status = OP_EXEC_DONE;
  }

H
Haojun Liao 已提交
7004
  return pIntervalInfo->binfo.pRes;
7005 7006
}

L
Liu Jicong 已提交
7007
static SSDataBlock* doAllSTableIntervalAgg(SOperatorInfo* pOperator, bool* newgroup) {
7008 7009 7010 7011 7012
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  STableIntervalOperatorInfo* pIntervalInfo = pOperator->info;
L
Liu Jicong 已提交
7013
  STaskRuntimeEnv*            pRuntimeEnv = pOperator->pRuntimeEnv;
7014 7015

  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
7016
    //    copyToSDataBlock(pRuntimeEnv, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
7017
    if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
7018 7019 7020
      pOperator->status = OP_EXEC_DONE;
    }

H
Haojun Liao 已提交
7021
    return pIntervalInfo->binfo.pRes;
7022 7023
  }

H
Haojun Liao 已提交
7024
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
L
Liu Jicong 已提交
7025
  int32_t    order = pQueryAttr->order.order;
7026

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

L
Liu Jicong 已提交
7029
  while (1) {
H
Haojun Liao 已提交
7030
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
7031
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
7032
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
7033 7034 7035 7036 7037 7038 7039 7040

    if (pBlock == NULL) {
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
    STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;

L
Liu Jicong 已提交
7041
    //    setTagValue(pOperator, pTableQueryInfo->pTable, pIntervalInfo->pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
7042
    setInputDataBlock(pOperator, pIntervalInfo->binfo.pCtx, pBlock, pQueryAttr->order.order);
7043 7044 7045 7046 7047 7048
    setIntervalQueryRange(pRuntimeEnv, pBlock->info.window.skey);

    hashAllIntervalAgg(pOperator, &pTableQueryInfo->resInfo, pBlock, pTableQueryInfo->groupIndex);
  }

  pOperator->status = OP_RES_TO_RETURN;
L
Liu Jicong 已提交
7049
  pQueryAttr->order.order = order;  // TODO : restore the order
7050
  doCloseAllTimeWindow(pRuntimeEnv);
7051
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
7052 7053

  int64_t st = taosGetTimestampUs();
L
Liu Jicong 已提交
7054
  //  copyToSDataBlock(pRuntimeEnv, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
7055
  if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
7056 7057 7058
    pOperator->status = OP_EXEC_DONE;
  }

L
Liu Jicong 已提交
7059 7060
  //  SQInfo* pQInfo = pRuntimeEnv->qinfo;
  //  pQInfo->summary.firstStageMergeTime += (taosGetTimestampUs() - st);
7061

H
Haojun Liao 已提交
7062
  return pIntervalInfo->binfo.pRes;
7063 7064
}

L
Liu Jicong 已提交
7065
static void doStateWindowAggImpl(SOperatorInfo* pOperator, SStateWindowOperatorInfo* pInfo, SSDataBlock* pSDataBlock) {
H
Haojun Liao 已提交
7066
  STaskRuntimeEnv* pRuntimeEnv = pOperator->pRuntimeEnv;
L
Liu Jicong 已提交
7067
  STableQueryInfo* item = pRuntimeEnv->current;
7068 7069 7070 7071
  SColumnInfoData* pColInfoData = taosArrayGet(pSDataBlock->pDataBlock, pInfo->colIndex);

  SOptrBasicInfo* pBInfo = &pInfo->binfo;

H
Haojun Liao 已提交
7072
  bool    masterScan = IS_MAIN_SCAN(pRuntimeEnv);
L
Liu Jicong 已提交
7073 7074
  int16_t bytes = pColInfoData->info.bytes;
  int16_t type = pColInfoData->info.type;
7075 7076

  SColumnInfoData* pTsColInfoData = taosArrayGet(pSDataBlock->pDataBlock, 0);
L
Liu Jicong 已提交
7077
  TSKEY*           tsList = (TSKEY*)pTsColInfoData->pData;
7078 7079
  if (IS_REPEAT_SCAN(pRuntimeEnv) && !pInfo->reptScan) {
    pInfo->reptScan = true;
wafwerar's avatar
wafwerar 已提交
7080
    taosMemoryFreeClear(pInfo->prevData);
7081 7082 7083 7084 7085 7086 7087 7088 7089
  }

  pInfo->numOfRows = 0;
  for (int32_t j = 0; j < pSDataBlock->info.rows; ++j) {
    char* val = ((char*)pColInfoData->pData) + bytes * j;
    if (isNull(val, type)) {
      continue;
    }
    if (pInfo->prevData == NULL) {
wafwerar's avatar
wafwerar 已提交
7090
      pInfo->prevData = taosMemoryMalloc(bytes);
7091 7092 7093 7094 7095 7096 7097 7098 7099
      memcpy(pInfo->prevData, val, bytes);
      pInfo->numOfRows = 1;
      pInfo->curWindow.skey = tsList[j];
      pInfo->curWindow.ekey = tsList[j];
      pInfo->start = j;

    } else if (memcmp(pInfo->prevData, val, bytes) == 0) {
      pInfo->curWindow.ekey = tsList[j];
      pInfo->numOfRows += 1;
L
Liu Jicong 已提交
7100
      // pInfo->start = j;
7101 7102 7103 7104 7105 7106 7107
      if (j == 0 && pInfo->start != 0) {
        pInfo->numOfRows = 1;
        pInfo->start = 0;
      }
    } else {
      SResultRow* pResult = NULL;
      pInfo->curWindow.ekey = pInfo->curWindow.skey;
L
Liu Jicong 已提交
7108 7109 7110
      int32_t ret = setResultOutputBufByKey(pRuntimeEnv, &pBInfo->resultRowInfo, pSDataBlock->info.uid,
                                            &pInfo->curWindow, masterScan, &pResult, item->groupIndex, pBInfo->pCtx,
                                            pOperator->numOfOutput, pBInfo->rowCellInfoOffset);
7111 7112 7113
      if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
        longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_APP_ERROR);
      }
L
Liu Jicong 已提交
7114 7115
      //      doApplyFunctions(pRuntimeEnv, pBInfo->pCtx, &pInfo->curWindow, pInfo->start, pInfo->numOfRows, tsList,
      //                       pSDataBlock->info.rows, pOperator->numOfOutput);
7116 7117 7118 7119 7120 7121 7122 7123 7124 7125 7126 7127

      pInfo->curWindow.skey = tsList[j];
      pInfo->curWindow.ekey = tsList[j];
      memcpy(pInfo->prevData, val, bytes);
      pInfo->numOfRows = 1;
      pInfo->start = j;
    }
  }

  SResultRow* pResult = NULL;

  pInfo->curWindow.ekey = pInfo->curWindow.skey;
L
Liu Jicong 已提交
7128 7129
  int32_t ret = setResultOutputBufByKey(pRuntimeEnv, &pBInfo->resultRowInfo, pSDataBlock->info.uid, &pInfo->curWindow,
                                        masterScan, &pResult, item->groupIndex, pBInfo->pCtx, pOperator->numOfOutput,
7130 7131 7132 7133 7134
                                        pBInfo->rowCellInfoOffset);
  if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
    longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_APP_ERROR);
  }

L
Liu Jicong 已提交
7135 7136
  //  doApplyFunctions(pRuntimeEnv, pBInfo->pCtx, &pInfo->curWindow, pInfo->start, pInfo->numOfRows, tsList,
  //                   pSDataBlock->info.rows, pOperator->numOfOutput);
7137 7138
}

L
Liu Jicong 已提交
7139
static SSDataBlock* doStateWindowAgg(SOperatorInfo* pOperator, bool* newgroup) {
7140 7141 7142 7143 7144
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  SStateWindowOperatorInfo* pWindowInfo = pOperator->info;
L
Liu Jicong 已提交
7145
  SOptrBasicInfo*           pBInfo = &pWindowInfo->binfo;
7146

H
Haojun Liao 已提交
7147
  STaskRuntimeEnv* pRuntimeEnv = pOperator->pRuntimeEnv;
7148
  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
7149
    //    toSDatablock(&pRuntimeEnv->groupResInfo, pRuntimeEnv, pBInfo->pRes);
7150 7151 7152 7153 7154 7155 7156 7157

    if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pRuntimeEnv->groupResInfo)) {
      pOperator->status = OP_EXEC_DONE;
    }

    return pBInfo->pRes;
  }

L
Liu Jicong 已提交
7158 7159 7160
  STaskAttr*     pQueryAttr = pRuntimeEnv->pQueryAttr;
  int32_t        order = pQueryAttr->order.order;
  STimeWindow    win = pQueryAttr->window;
H
Haojun Liao 已提交
7161
  SOperatorInfo* downstream = pOperator->pDownstream[0];
7162
  while (1) {
H
Haojun Liao 已提交
7163
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
7164
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
7165
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
7166 7167 7168 7169 7170 7171 7172 7173

    if (pBlock == NULL) {
      break;
    }
    setInputDataBlock(pOperator, pBInfo->pCtx, pBlock, pQueryAttr->order.order);
    if (pWindowInfo->colIndex == -1) {
      pWindowInfo->colIndex = getGroupbyColumnIndex(pRuntimeEnv->pQueryAttr->pGroupbyExpr, pBlock);
    }
L
Liu Jicong 已提交
7174
    doStateWindowAggImpl(pOperator, pWindowInfo, pBlock);
7175 7176 7177 7178 7179 7180 7181 7182
  }

  // restore the value
  pQueryAttr->order.order = order;
  pQueryAttr->window = win;

  pOperator->status = OP_RES_TO_RETURN;
  closeAllResultRows(&pBInfo->resultRowInfo);
7183
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
7184
  finalizeQueryResult(pBInfo->pCtx, pOperator->numOfOutput);
7185 7186

  initGroupResInfo(&pRuntimeEnv->groupResInfo, &pBInfo->resultRowInfo);
L
Liu Jicong 已提交
7187
  //  toSDatablock(&pRuntimeEnv->groupResInfo, pRuntimeEnv, pBInfo->pRes);
7188 7189 7190 7191 7192

  if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pRuntimeEnv->groupResInfo)) {
    pOperator->status = OP_EXEC_DONE;
  }

L
Liu Jicong 已提交
7193
  return pBInfo->pRes->info.rows == 0 ? NULL : pBInfo->pRes;
7194 7195
}

L
Liu Jicong 已提交
7196
static SSDataBlock* doSessionWindowAgg(SOperatorInfo* pOperator, bool* newgroup) {
7197 7198 7199 7200
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

7201
  SSessionAggOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
7202
  SOptrBasicInfo*          pBInfo = &pInfo->binfo;
7203 7204

  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
7205 7206
    toSDatablock(&pInfo->groupResInfo, pInfo->aggSup.pResultBuf, pBInfo->pRes, pBInfo->capacity,
                 pBInfo->rowCellInfoOffset);
7207 7208 7209
    if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
      doSetOperatorCompleted(pOperator);
      return NULL;
7210 7211 7212 7213 7214
    }

    return pBInfo->pRes;
  }

L
Liu Jicong 已提交
7215
  int32_t        order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
7216
  SOperatorInfo* downstream = pOperator->pDownstream[0];
7217

L
Liu Jicong 已提交
7218
  while (1) {
H
Haojun Liao 已提交
7219
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
7220
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
7221
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
7222 7223 7224 7225 7226
    if (pBlock == NULL) {
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
H
Haojun Liao 已提交
7227
    setInputDataBlock(pOperator, pBInfo->pCtx, pBlock, order);
7228
    doSessionWindowAggImpl(pOperator, pInfo, pBlock);
7229 7230 7231 7232 7233
  }

  // restore the value
  pOperator->status = OP_RES_TO_RETURN;
  closeAllResultRows(&pBInfo->resultRowInfo);
L
Liu Jicong 已提交
7234 7235
  finalizeMultiTupleQueryResult(pBInfo->pCtx, pOperator->numOfOutput, pInfo->aggSup.pResultBuf, &pBInfo->resultRowInfo,
                                pBInfo->rowCellInfoOffset);
7236

7237 7238 7239
  initGroupResInfo(&pInfo->groupResInfo, &pBInfo->resultRowInfo);

  blockDataEnsureCapacity(pBInfo->pRes, pBInfo->capacity);
L
Liu Jicong 已提交
7240 7241
  toSDatablock(&pInfo->groupResInfo, pInfo->aggSup.pResultBuf, pBInfo->pRes, pBInfo->capacity,
               pBInfo->rowCellInfoOffset);
7242 7243
  if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
    doSetOperatorCompleted(pOperator);
7244 7245
  }

L
Liu Jicong 已提交
7246
  return pBInfo->pRes->info.rows == 0 ? NULL : pBInfo->pRes;
7247 7248
}

L
Liu Jicong 已提交
7249
static SSDataBlock* hashGroupbyAggregate(SOperatorInfo* pOperator, bool* newgroup) {
7250 7251 7252 7253
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

L
Liu Jicong 已提交
7254
  SGroupbyOperatorInfo* pInfo = pOperator->info;
7255
  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
7256 7257 7258
    toSDatablock(&pInfo->groupResInfo, pInfo->aggSup.pResultBuf, pInfo->binfo.pRes, pInfo->binfo.capacity,
                 pInfo->binfo.rowCellInfoOffset);
    if (pInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
7259 7260 7261 7262 7263
      pOperator->status = OP_EXEC_DONE;
    }
    return pInfo->binfo.pRes;
  }

L
Liu Jicong 已提交
7264
  int32_t        order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
7265
  SOperatorInfo* downstream = pOperator->pDownstream[0];
7266

L
Liu Jicong 已提交
7267
  while (1) {
H
Haojun Liao 已提交
7268
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
7269
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
7270
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
7271 7272 7273 7274 7275
    if (pBlock == NULL) {
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
H
Haojun Liao 已提交
7276
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pBlock, order);
L
Liu Jicong 已提交
7277
    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pInfo->binfo.pCtx, pOperator->numOfOutput);
7278
    doHashGroupbyAgg(pOperator, pBlock);
7279 7280 7281 7282 7283
  }

  pOperator->status = OP_RES_TO_RETURN;
  closeAllResultRows(&pInfo->binfo.resultRowInfo);

L
Liu Jicong 已提交
7284 7285 7286 7287 7288 7289 7290 7291
  finalizeMultiTupleQueryResult(pInfo->binfo.pCtx, pOperator->numOfOutput, pInfo->aggSup.pResultBuf,
                                &pInfo->binfo.resultRowInfo, pInfo->binfo.rowCellInfoOffset);
  //  if (!pRuntimeEnv->pQueryAttr->stableQuery) { // finalize include the update of result rows
  //    finalizeQueryResult(pInfo->binfo.pCtx, pOperator->numOfOutput);
  //  } else {
  //    updateNumOfRowsInResultRows(pInfo->binfo.pCtx, pOperator->numOfOutput, &pInfo->binfo.resultRowInfo,
  //    pInfo->binfo.rowCellInfoOffset);
  //  }
7292

H
Haojun Liao 已提交
7293
  blockDataEnsureCapacity(pInfo->binfo.pRes, pInfo->binfo.capacity);
H
Haojun Liao 已提交
7294
  initGroupResInfo(&pInfo->groupResInfo, &pInfo->binfo.resultRowInfo);
L
Liu Jicong 已提交
7295 7296
  toSDatablock(&pInfo->groupResInfo, pInfo->aggSup.pResultBuf, pInfo->binfo.pRes, pInfo->binfo.capacity,
               pInfo->binfo.rowCellInfoOffset);
H
Haojun Liao 已提交
7297
  if (pInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
7298 7299 7300 7301 7302 7303
    pOperator->status = OP_EXEC_DONE;
  }

  return pInfo->binfo.pRes;
}

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

L
Liu Jicong 已提交
7308 7309
  int64_t ekey = Q_STATUS_EQUAL(pTaskInfo->status, TASK_COMPLETED) ? pTaskInfo->window.ekey
                                                                   : pInfo->existNewGroupBlock->info.window.ekey;
7310 7311
  taosResetFillInfo(pInfo->pFillInfo, getFillInfoStart(pInfo->pFillInfo));

7312
  taosFillSetStartInfo(pInfo->pFillInfo, pInfo->existNewGroupBlock->info.rows, ekey);
7313 7314
  taosFillSetInputDataBlock(pInfo->pFillInfo, pInfo->existNewGroupBlock);

H
Haojun Liao 已提交
7315
  doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, pResultInfo->capacity, pInfo->p);
7316 7317 7318 7319
  pInfo->existNewGroupBlock = NULL;
  *newgroup = true;
}

L
Liu Jicong 已提交
7320 7321
static void doHandleRemainBlockFromNewGroup(SFillOperatorInfo* pInfo, SResultInfo* pResultInfo, bool* newgroup,
                                            SExecTaskInfo* pTaskInfo) {
7322 7323
  if (taosFillHasMoreResults(pInfo->pFillInfo)) {
    *newgroup = false;
H
Haojun Liao 已提交
7324 7325
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, (int32_t)pResultInfo->capacity, pInfo->p);
    if (pInfo->pRes->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult)) {
7326 7327 7328 7329 7330 7331
      return;
    }
  }

  // handle the cached new group data block
  if (pInfo->existNewGroupBlock) {
7332
    doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
7333 7334 7335
  }
}

L
Liu Jicong 已提交
7336 7337 7338
static SSDataBlock* doFill(SOperatorInfo* pOperator, bool* newgroup) {
  SFillOperatorInfo* pInfo = pOperator->info;
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
7339

H
Haojun Liao 已提交
7340 7341
  SResultInfo* pResultInfo = &pOperator->resultInfo;
  blockDataCleanup(pInfo->pRes);
7342 7343 7344 7345
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

7346
  doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
H
Haojun Liao 已提交
7347 7348 7349
  if (pInfo->pRes->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult && pInfo->pRes->info.rows > 0)) {
    return pInfo->pRes;
  }
7350

H
Haojun Liao 已提交
7351
  SOperatorInfo* pDownstream = pOperator->pDownstream[0];
L
Liu Jicong 已提交
7352
  while (1) {
H
Haojun Liao 已提交
7353 7354 7355
    publishOperatorProfEvent(pDownstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
    SSDataBlock* pBlock = pDownstream->getNextFn(pDownstream, newgroup);
    publishOperatorProfEvent(pDownstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
7356 7357 7358 7359 7360 7361 7362 7363 7364 7365 7366

    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
7367
      taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
7368 7369 7370 7371 7372 7373 7374
    } else {
      if (pBlock == NULL) {
        if (pInfo->totalInputRows == 0) {
          pOperator->status = OP_EXEC_DONE;
          return NULL;
        }

7375
        taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
7376 7377 7378 7379 7380 7381 7382
      } else {
        pInfo->totalInputRows += pBlock->info.rows;
        taosFillSetStartInfo(pInfo->pFillInfo, pBlock->info.rows, pBlock->info.window.ekey);
        taosFillSetInputDataBlock(pInfo->pFillInfo, pBlock);
      }
    }

H
Haojun Liao 已提交
7383
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, pInfo->capacity, pInfo->p);
7384 7385 7386 7387 7388

    // current group has no more result to return
    if (pInfo->pRes->info.rows > 0) {
      // 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
H
Haojun Liao 已提交
7389
      if (pInfo->pRes->info.rows > pResultInfo->threshold || pBlock == NULL || (!pInfo->multigroupResult)) {
7390 7391 7392
        return pInfo->pRes;
      }

7393
      doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
H
Haojun Liao 已提交
7394
      if (pInfo->pRes->info.rows > pOperator->resultInfo.threshold || pBlock == NULL) {
7395 7396 7397 7398
        return pInfo->pRes;
      }
    } else if (pInfo->existNewGroupBlock) {  // try next group
      assert(pBlock != NULL);
7399
      doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
H
Haojun Liao 已提交
7400
      if (pInfo->pRes->info.rows > pResultInfo->threshold) {
7401 7402 7403 7404 7405 7406 7407 7408 7409
        return pInfo->pRes;
      }
    } else {
      return NULL;
    }
  }
}

// todo set the attribute of query scan count
H
Haojun Liao 已提交
7410
static int32_t getNumOfScanTimes(STaskAttr* pQueryAttr) {
L
Liu Jicong 已提交
7411
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
7412 7413 7414 7415 7416 7417 7418 7419 7420 7421 7422 7423 7424 7425
    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;
  }

H
Haojun Liao 已提交
7426 7427
  if (pOperator->closeFn != NULL) {
    pOperator->closeFn(pOperator->info, pOperator->numOfOutput);
7428 7429
  }

H
Haojun Liao 已提交
7430
  if (pOperator->pDownstream != NULL) {
L
Liu Jicong 已提交
7431
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
H
Haojun Liao 已提交
7432
      destroyOperatorInfo(pOperator->pDownstream[i]);
7433 7434
    }

wafwerar's avatar
wafwerar 已提交
7435
    taosMemoryFreeClear(pOperator->pDownstream);
H
Haojun Liao 已提交
7436
    pOperator->numOfDownstream = 0;
7437 7438
  }

wafwerar's avatar
wafwerar 已提交
7439 7440
  taosMemoryFreeClear(pOperator->info);
  taosMemoryFreeClear(pOperator);
7441 7442
}

L
Liu Jicong 已提交
7443
int32_t doInitAggInfoSup(SAggSupporter* pAggSup, SqlFunctionCtx* pCtx, int32_t numOfOutput, const char* pKey) {
7444 7445
  _hash_fn_t hashFn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY);

L
Liu Jicong 已提交
7446 7447
  pAggSup->resultRowSize = getResultRowSize(pCtx, numOfOutput);
  pAggSup->keyBuf = taosMemoryCalloc(1, sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES);
7448
  pAggSup->pResultRowHashTable = taosHashInit(10, hashFn, true, HASH_NO_LOCK);
L
Liu Jicong 已提交
7449
  pAggSup->pResultRowListSet = taosHashInit(100, hashFn, false, HASH_NO_LOCK);
7450 7451 7452
  pAggSup->pResultRowArrayList = taosArrayInit(10, sizeof(SResultRowCell));

  if (pAggSup->keyBuf == NULL || pAggSup->pResultRowArrayList == NULL || pAggSup->pResultRowListSet == NULL ||
7453
      pAggSup->pResultRowHashTable == NULL) {
7454 7455 7456
    return TSDB_CODE_OUT_OF_MEMORY;
  }

H
Haojun Liao 已提交
7457 7458 7459 7460 7461
  int32_t code = createDiskbasedBuf(&pAggSup->pResultBuf, 4096, 4096 * 256, pKey, "/tmp/");
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

7462 7463 7464
  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
7465
static void cleanupAggSup(SAggSupporter* pAggSup) {
wafwerar's avatar
wafwerar 已提交
7466
  taosMemoryFreeClear(pAggSup->keyBuf);
7467 7468 7469
  taosHashCleanup(pAggSup->pResultRowHashTable);
  taosHashCleanup(pAggSup->pResultRowListSet);
  taosArrayDestroy(pAggSup->pResultRowArrayList);
H
Haojun Liao 已提交
7470
  destroyDiskbasedBuf(pAggSup->pResultBuf);
7471 7472
}

H
Haojun Liao 已提交
7473
static int32_t initAggInfo(SOptrBasicInfo* pBasicInfo, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols,
H
Haojun Liao 已提交
7474
                           int32_t numOfRows, SSDataBlock* pResultBlock, const char* pkey) {
H
Haojun Liao 已提交
7475 7476 7477 7478
  pBasicInfo->pCtx = createSqlFunctionCtx_rv(pExprInfo, numOfCols, &pBasicInfo->rowCellInfoOffset);
  pBasicInfo->pRes = pResultBlock;
  pBasicInfo->capacity = numOfRows;

H
Haojun Liao 已提交
7479
  doInitAggInfoSup(pAggSup, pBasicInfo->pCtx, numOfCols, pkey);
H
Haojun Liao 已提交
7480 7481 7482
}

static STableQueryInfo* initTableQueryInfo(const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
7483
  STableQueryInfo* pTableQueryInfo = taosMemoryCalloc(pTableGroupInfo->numOfTables, sizeof(STableQueryInfo));
H
Haojun Liao 已提交
7484 7485
  if (pTableQueryInfo == NULL) {
    return NULL;
H
Haojun Liao 已提交
7486
  }
H
Haojun Liao 已提交
7487 7488

  int32_t index = 0;
L
Liu Jicong 已提交
7489
  for (int32_t i = 0; i < taosArrayGetSize(pTableGroupInfo->pGroupList); ++i) {
H
Haojun Liao 已提交
7490
    SArray* pa = taosArrayGetP(pTableGroupInfo->pGroupList, i);
L
Liu Jicong 已提交
7491
    for (int32_t j = 0; j < taosArrayGetSize(pa); ++j) {
H
Haojun Liao 已提交
7492 7493
      STableKeyInfo* pk = taosArrayGet(pa, j);

H
Haojun Liao 已提交
7494
      STableQueryInfo* pTQueryInfo = &pTableQueryInfo[index++];
L
Liu Jicong 已提交
7495 7496
      pTQueryInfo->uid = pk->uid;
      pTQueryInfo->lastKey = pk->lastKey;
H
Haojun Liao 已提交
7497 7498 7499
      pTQueryInfo->groupIndex = i;
    }
  }
H
Haojun Liao 已提交
7500 7501

  STimeWindow win = {0, INT64_MAX};
H
Haojun Liao 已提交
7502 7503
  createTableQueryInfo(pTableQueryInfo, false, win);
  return pTableQueryInfo;
H
Haojun Liao 已提交
7504 7505
}

L
Liu Jicong 已提交
7506 7507 7508
SOperatorInfo* createAggregateOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                           SSDataBlock* pResultBlock, SExecTaskInfo* pTaskInfo,
                                           const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
7509
  SAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SAggOperatorInfo));
L
Liu Jicong 已提交
7510
  SOperatorInfo*    pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
7511 7512 7513
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
7514 7515

  //(int32_t)(getRowNumForMultioutput(pQueryAttr, pQueryAttr->topBotQuery, pQueryAttr->stableQuery));
H
Haojun Liao 已提交
7516
  int32_t numOfRows = 1;
L
Liu Jicong 已提交
7517 7518
  int32_t code =
      initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResultBlock, pTaskInfo->id.str);
H
Haojun Liao 已提交
7519 7520
  pInfo->pTableQueryInfo = initTableQueryInfo(pTableGroupInfo);
  if (code != TSDB_CODE_SUCCESS || pInfo->pTableQueryInfo == NULL) {
H
Haojun Liao 已提交
7521 7522
    goto _error;
  }
H
Haojun Liao 已提交
7523

H
Haojun Liao 已提交
7524
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, pTaskInfo);
7525

L
Liu Jicong 已提交
7526
  pOperator->name = "TableAggregate";
X
Xiaoyu Wang 已提交
7527
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_AGG;
7528
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7529 7530 7531 7532 7533 7534 7535 7536 7537
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfOutput = numOfCols;

  pOperator->pTaskInfo = pTaskInfo;
  pOperator->_openFn = doOpenAggregateOptr;
  pOperator->getNextFn = getAggregateResult;
  pOperator->closeFn = destroyAggOperatorInfo;
wmmhello's avatar
wmmhello 已提交
7538 7539
  pOperator->encodeResultRow = aggEncodeResultRow;
  pOperator->decodeResultRow = aggDecodeResultRow;
H
Haojun Liao 已提交
7540 7541 7542 7543 7544

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

  return pOperator;
L
Liu Jicong 已提交
7547
_error:
H
Haojun Liao 已提交
7548
  destroyAggOperatorInfo(pInfo, numOfCols);
wafwerar's avatar
wafwerar 已提交
7549 7550
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
7551 7552
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
7553 7554 7555 7556 7557
}

static void doDestroyBasicInfo(SOptrBasicInfo* pInfo, int32_t numOfOutput) {
  assert(pInfo != NULL);

7558
  destroySqlFunctionCtx(pInfo->pCtx, numOfOutput);
wafwerar's avatar
wafwerar 已提交
7559
  taosMemoryFreeClear(pInfo->rowCellInfoOffset);
7560 7561

  cleanupResultRowInfo(&pInfo->resultRowInfo);
H
Haojun Liao 已提交
7562
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
7563 7564
}

H
Haojun Liao 已提交
7565
void destroyBasicOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7566
  SOptrBasicInfo* pInfo = (SOptrBasicInfo*)param;
7567 7568
  doDestroyBasicInfo(pInfo, numOfOutput);
}
H
Haojun Liao 已提交
7569 7570

void destroyStateWindowOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7571
  SStateWindowOperatorInfo* pInfo = (SStateWindowOperatorInfo*)param;
7572
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
wafwerar's avatar
wafwerar 已提交
7573
  taosMemoryFreeClear(pInfo->prevData);
7574
}
H
Haojun Liao 已提交
7575 7576

void destroyAggOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7577
  SAggOperatorInfo* pInfo = (SAggOperatorInfo*)param;
7578 7579
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}
7580

H
Haojun Liao 已提交
7581
void destroyIntervalOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7582
  STableIntervalOperatorInfo* pInfo = (STableIntervalOperatorInfo*)param;
H
Haojun Liao 已提交
7583 7584 7585 7586 7587
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
  cleanupAggSup(&pInfo->aggSup);
}

void destroySWindowOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7588
  SSessionAggOperatorInfo* pInfo = (SSessionAggOperatorInfo*)param;
7589 7590 7591
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}

H
Haojun Liao 已提交
7592
void destroySFillOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7593
  SFillOperatorInfo* pInfo = (SFillOperatorInfo*)param;
7594
  pInfo->pFillInfo = taosDestroyFillInfo(pInfo->pFillInfo);
H
Haojun Liao 已提交
7595
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
wafwerar's avatar
wafwerar 已提交
7596
  taosMemoryFreeClear(pInfo->p);
7597 7598
}

H
Haojun Liao 已提交
7599
void destroyGroupbyOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7600
  SGroupbyOperatorInfo* pInfo = (SGroupbyOperatorInfo*)param;
7601
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
wafwerar's avatar
wafwerar 已提交
7602
  taosMemoryFreeClear(pInfo->keyBuf);
7603 7604
  taosArrayDestroy(pInfo->pGroupCols);
  taosArrayDestroy(pInfo->pGroupColVals);
7605 7606
}

H
Haojun Liao 已提交
7607
static void destroyProjectOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7608
  SProjectOperatorInfo* pInfo = (SProjectOperatorInfo*)param;
7609 7610 7611
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}

H
Haojun Liao 已提交
7612
static void destroyTagScanOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7613
  STagScanInfo* pInfo = (STagScanInfo*)param;
H
Haojun Liao 已提交
7614
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
7615 7616
}

H
Haojun Liao 已提交
7617
static void destroyOrderOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7618
  SSortOperatorInfo* pInfo = (SSortOperatorInfo*)param;
H
Haojun Liao 已提交
7619 7620
  pInfo->pDataBlock = blockDataDestroy(pInfo->pDataBlock);

H
Haojun Liao 已提交
7621
  taosArrayDestroy(pInfo->pSortInfo);
7622 7623 7624
}

static void destroyConditionOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7625
  SFilterOperatorInfo* pInfo = (SFilterOperatorInfo*)param;
7626 7627 7628 7629
  doDestroyFilterInfo(pInfo->pFilterInfo, pInfo->numOfFilterCols);
}

static void destroyDistinctOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7630
  SDistinctOperatorInfo* pInfo = (SDistinctOperatorInfo*)param;
7631
  taosHashCleanup(pInfo->pSet);
wafwerar's avatar
wafwerar 已提交
7632
  taosMemoryFreeClear(pInfo->buf);
7633
  taosArrayDestroy(pInfo->pDistinctDataInfo);
H
Haojun Liao 已提交
7634
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
7635 7636
}

7637
static void destroySysTableScannerOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7638
  SSysTableScanInfo* pInfo = (SSysTableScanInfo*)param;
7639 7640 7641 7642 7643 7644 7645 7646
  tsem_destroy(&pInfo->ready);
  blockDataDestroy(pInfo->pRes);

  if (pInfo->type == TSDB_MGMT_TABLE_TABLE) {
    metaCloseTbCursor(pInfo->pCur);
  }
}

H
Haojun Liao 已提交
7647
void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
7648
  SExchangeInfo* pExInfo = (SExchangeInfo*)param;
H
Haojun Liao 已提交
7649 7650 7651 7652 7653 7654 7655 7656 7657
  taosArrayDestroy(pExInfo->pSources);
  taosArrayDestroy(pExInfo->pSourceDataInfo);
  if (pExInfo->pResult != NULL) {
    blockDataDestroy(pExInfo->pResult);
  }

  tsem_destroy(&pExInfo->ready);
}

L
Liu Jicong 已提交
7658 7659 7660
SOperatorInfo* createMultiTableAggOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                               SSDataBlock* pResBlock, SExecTaskInfo* pTaskInfo,
                                               const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
7661
  SAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SAggOperatorInfo));
7662

H
Haojun Liao 已提交
7663
  int32_t numOfRows = 1;
L
Liu Jicong 已提交
7664 7665
  int32_t code =
      initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResBlock, pTaskInfo->id.str);
H
Haojun Liao 已提交
7666 7667 7668 7669
  pInfo->pTableQueryInfo = initTableQueryInfo(pTableGroupInfo);
  if (code != TSDB_CODE_SUCCESS || pInfo->pTableQueryInfo == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
7670

L
Liu Jicong 已提交
7671
  size_t tableGroup = taosArrayGetSize(pTableGroupInfo->pGroupList);
7672
  initResultRowInfo(&pInfo->binfo.resultRowInfo, (int32_t)tableGroup);
7673

wafwerar's avatar
wafwerar 已提交
7674
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
7675
  pOperator->name = "MultiTableAggregate";
X
Xiaoyu Wang 已提交
7676
  // pOperator->operatorType = OP_MultiTableAggregate;
7677
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7678 7679 7680 7681 7682
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfOutput = numOfCols;
  pOperator->pTaskInfo = pTaskInfo;
7683

H
Haojun Liao 已提交
7684
  pOperator->getNextFn = doMultiTableAggregate;
H
Haojun Liao 已提交
7685
  pOperator->closeFn = destroyAggOperatorInfo;
H
Haojun Liao 已提交
7686 7687 7688 7689
  code = appendDownstream(pOperator, &downstream, 1);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
7690 7691

  return pOperator;
H
Haojun Liao 已提交
7692 7693 7694

_error:
  return NULL;
7695 7696
}

L
Liu Jicong 已提交
7697 7698
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t num,
                                         SSDataBlock* pResBlock, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
7699
  SProjectOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SProjectOperatorInfo));
L
Liu Jicong 已提交
7700
  SOperatorInfo*        pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
7701 7702 7703
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
7704

H
Haojun Liao 已提交
7705 7706
  pInfo->binfo.pRes = pResBlock;
  pInfo->binfo.pCtx = createSqlFunctionCtx_rv(pExprInfo, num, &pInfo->binfo.rowCellInfoOffset);
H
Haojun Liao 已提交
7707 7708 7709
  if (pInfo->binfo.pCtx == NULL) {
    goto _error;
  }
7710

L
Liu Jicong 已提交
7711 7712
  //  initResultRowInfo(&pBInfo->resultRowInfo, 8);
  //  setFunctionResultOutput(pBInfo, MAIN_SCAN);
7713

L
Liu Jicong 已提交
7714
  pOperator->name = "ProjectOperator";
H
Haojun Liao 已提交
7715
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_PROJECT;
7716
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
7717 7718 7719 7720 7721 7722 7723 7724 7725
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfOutput = num;
  pOperator->_openFn = operatorDummyOpenFn;
  pOperator->getNextFn = doProjectOperation;
  pOperator->closeFn = destroyProjectOperatorInfo;

  pOperator->pTaskInfo = pTaskInfo;
7726
  int32_t code = appendDownstream(pOperator, &downstream, 1);
H
Haojun Liao 已提交
7727
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
7728 7729
    goto _error;
  }
7730 7731

  return pOperator;
H
Haojun Liao 已提交
7732

L
Liu Jicong 已提交
7733
_error:
H
Haojun Liao 已提交
7734 7735
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
7736 7737 7738 7739
}

SColumnInfo* extractColumnFilterInfo(SExprInfo* pExpr, int32_t numOfOutput, int32_t* numOfFilterCols) {
#if 0
wafwerar's avatar
wafwerar 已提交
7740
  SColumnInfo* pCols = taosMemoryCalloc(numOfOutput, sizeof(SColumnInfo));
7741 7742 7743 7744 7745 7746 7747 7748 7749 7750 7751 7752 7753

  int32_t numOfFilter = 0;
  for(int32_t i = 0; i < numOfOutput; ++i) {
    if (pExpr[i].base.flist.numOfFilters > 0) {
      numOfFilter += 1;
    }

    pCols[i].type  = pExpr[i].base.resSchema.type;
    pCols[i].bytes = pExpr[i].base.resSchema.bytes;
    pCols[i].colId = pExpr[i].base.resSchema.colId;

    pCols[i].flist.numOfFilters = pExpr[i].base.flist.numOfFilters;
    if (pCols[i].flist.numOfFilters != 0) { 
wafwerar's avatar
wafwerar 已提交
7754
      pCols[i].flist.filterInfo   = taosMemoryCalloc(pCols[i].flist.numOfFilters, sizeof(SColumnFilterInfo));
7755 7756 7757 7758 7759 7760 7761 7762 7763 7764 7765 7766 7767 7768 7769 7770
      memcpy(pCols[i].flist.filterInfo, pExpr[i].base.flist.filterInfo, pCols[i].flist.numOfFilters * sizeof(SColumnFilterInfo));
    } else {
      // avoid runtime error
      pCols[i].flist.filterInfo   = NULL; 
    }
  }

  assert(numOfFilter > 0);

  *numOfFilterCols = numOfFilter;
  return pCols;
#endif

  return 0;
}

H
Haojun Liao 已提交
7771
SOperatorInfo* createLimitOperatorInfo(SOperatorInfo* downstream, SLimit* pLimit, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
7772
  SLimitOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SLimitOperatorInfo));
L
Liu Jicong 已提交
7773
  SOperatorInfo*      pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
7774 7775 7776 7777
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }

L
Liu Jicong 已提交
7778 7779
  pInfo->limit = *pLimit;
  pInfo->currentOffset = pLimit->offset;
7780

L
Liu Jicong 已提交
7781 7782
  pOperator->name = "LimitOperator";
  //  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_LIMIT;
7783
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
7784 7785 7786 7787 7788
  pOperator->status = OP_NOT_OPENED;
  pOperator->_openFn = operatorDummyOpenFn;
  pOperator->getNextFn = doLimit;
  pOperator->info = pInfo;
  pOperator->pTaskInfo = pTaskInfo;
H
Haojun Liao 已提交
7789
  int32_t code = appendDownstream(pOperator, &downstream, 1);
7790 7791

  return pOperator;
L
Liu Jicong 已提交
7792
_error:
wafwerar's avatar
wafwerar 已提交
7793 7794
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
7795 7796
  terrno = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
7797 7798
}

L
Liu Jicong 已提交
7799 7800
SOperatorInfo* createIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                          SSDataBlock* pResBlock, SInterval* pInterval,
7801
                                          const STableGroupInfo* pTableGroupInfo, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
7802
  STableIntervalOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableIntervalOperatorInfo));
L
Liu Jicong 已提交
7803
  SOperatorInfo*              pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
7804 7805 7806
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
7807

L
Liu Jicong 已提交
7808
  pInfo->order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
7809
  pInfo->precision = TSDB_TIME_PRECISION_MILLI;
L
Liu Jicong 已提交
7810 7811
  pInfo->win = pTaskInfo->window;
  pInfo->interval = *pInterval;
H
Haojun Liao 已提交
7812

7813 7814 7815 7816
  pInfo->win.skey = INT64_MIN;
  pInfo->win.ekey = INT64_MAX;

  int32_t numOfRows = 4096;
L
Liu Jicong 已提交
7817 7818 7819 7820
  int32_t code =
      initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResBlock, pTaskInfo->id.str);
  //  pInfo->pTableQueryInfo = initTableQueryInfo(pTableGroupInfo);
  if (code != TSDB_CODE_SUCCESS /* || pInfo->pTableQueryInfo == NULL*/) {
H
Haojun Liao 已提交
7821 7822
    goto _error;
  }
H
Haojun Liao 已提交
7823 7824

  initResultRowInfo(&pInfo->binfo.resultRowInfo, (int32_t)1);
7825

L
Liu Jicong 已提交
7826
  pOperator->name = "TimeIntervalAggOperator";
H
Haojun Liao 已提交
7827
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_INTERVAL;
7828
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7829 7830 7831 7832 7833 7834 7835 7836
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExprInfo;
  pOperator->pTaskInfo = pTaskInfo;
  pOperator->numOfOutput = numOfCols;
  pOperator->info = pInfo;
  pOperator->_openFn = doOpenIntervalAgg;
  pOperator->getNextFn = doBuildIntervalResult;
  pOperator->closeFn = destroyIntervalOperatorInfo;
7837

H
Haojun Liao 已提交
7838
  code = appendDownstream(pOperator, &downstream, 1);
H
Haojun Liao 已提交
7839 7840 7841 7842
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

7843
  return pOperator;
H
Haojun Liao 已提交
7844

L
Liu Jicong 已提交
7845
_error:
H
Haojun Liao 已提交
7846
  destroyIntervalOperatorInfo(pInfo, numOfCols);
wafwerar's avatar
wafwerar 已提交
7847 7848
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
7849 7850
  pTaskInfo->code = code;
  return NULL;
7851 7852
}

L
Liu Jicong 已提交
7853 7854
SOperatorInfo* createAllTimeIntervalOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream,
                                                 SExprInfo* pExpr, int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
7855
  STableIntervalOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableIntervalOperatorInfo));
7856

L
Liu Jicong 已提交
7857 7858
  //  pInfo->binfo.pCtx = createSqlFunctionCtx(pRuntimeEnv, pExpr, numOfOutput, &pInfo->binfo.rowCellInfoOffset);
  //  pInfo->binfo.pRes = createOutputBuf(pExpr, numOfOutput, pResultInfo->capacity);
H
Haojun Liao 已提交
7859
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
7860

wafwerar's avatar
wafwerar 已提交
7861
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
7862

L
Liu Jicong 已提交
7863 7864
  pOperator->name = "AllTimeIntervalAggOperator";
  //  pOperator->operatorType = OP_AllTimeWindow;
7865
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7866 7867 7868 7869 7870
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->info = pInfo;
  pOperator->pRuntimeEnv = pRuntimeEnv;
H
Haojun Liao 已提交
7871
  pOperator->getNextFn = doAllIntervalAgg;
H
Haojun Liao 已提交
7872
  pOperator->closeFn = destroyBasicOperatorInfo;
7873

L
Liu Jicong 已提交
7874
  int32_t code = appendDownstream(pOperator, &downstream, 1);
7875 7876 7877
  return pOperator;
}

L
Liu Jicong 已提交
7878 7879
SOperatorInfo* createStatewindowOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream, SExprInfo* pExpr,
                                             int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
7880
  SStateWindowOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SStateWindowOperatorInfo));
L
Liu Jicong 已提交
7881 7882 7883 7884
  pInfo->colIndex = -1;
  pInfo->reptScan = false;
  //  pInfo->binfo.pCtx = createSqlFunctionCtx(pRuntimeEnv, pExpr, numOfOutput, &pInfo->binfo.rowCellInfoOffset);
  //  pInfo->binfo.pRes = createOutputBuf(pExpr, numOfOutput, pResultInfo->capacity);
7885
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
7886

wafwerar's avatar
wafwerar 已提交
7887
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
7888 7889
  pOperator->name = "StateWindowOperator";
  //  pOperator->operatorType = OP_StateWindow;
7890
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7891 7892 7893 7894 7895
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->info = pInfo;
  pOperator->pRuntimeEnv = pRuntimeEnv;
H
Haojun Liao 已提交
7896
  pOperator->getNextFn = doStateWindowAgg;
H
Haojun Liao 已提交
7897
  pOperator->closeFn = destroyStateWindowOperatorInfo;
7898

H
Haojun Liao 已提交
7899
  int32_t code = appendDownstream(pOperator, &downstream, 1);
7900 7901 7902
  return pOperator;
}

L
Liu Jicong 已提交
7903 7904
SOperatorInfo* createSessionAggOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                            SSDataBlock* pResBlock, int64_t gap, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
7905
  SSessionAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSessionAggOperatorInfo));
L
Liu Jicong 已提交
7906
  SOperatorInfo*           pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
7907 7908 7909
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
7910

H
Haojun Liao 已提交
7911
  int32_t numOfRows = 4096;
L
Liu Jicong 已提交
7912 7913
  int32_t code =
      initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResBlock, pTaskInfo->id.str);
H
Haojun Liao 已提交
7914 7915 7916
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
7917

H
Haojun Liao 已提交
7918
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
7919

L
Liu Jicong 已提交
7920 7921 7922 7923 7924
  pInfo->gap = gap;
  pInfo->binfo.pRes = pResBlock;
  pInfo->prevTs = INT64_MIN;
  pInfo->reptScan = false;
  pOperator->name = "SessionWindowAggOperator";
H
Haojun Liao 已提交
7925
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SESSION_WINDOW;
7926
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7927 7928 7929 7930 7931 7932 7933
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfOutput = numOfCols;
  pOperator->info = pInfo;
  pOperator->getNextFn = doSessionWindowAgg;
  pOperator->closeFn = destroySWindowOperatorInfo;
  pOperator->pTaskInfo = pTaskInfo;
7934

H
Haojun Liao 已提交
7935
  code = appendDownstream(pOperator, &downstream, 1);
7936
  return pOperator;
H
Haojun Liao 已提交
7937

L
Liu Jicong 已提交
7938
_error:
H
Haojun Liao 已提交
7939 7940 7941 7942
  if (pInfo != NULL) {
    destroySWindowOperatorInfo(pInfo, numOfCols);
  }

wafwerar's avatar
wafwerar 已提交
7943 7944
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
7945 7946
  pTaskInfo->code = code;
  return NULL;
7947 7948
}

L
Liu Jicong 已提交
7949 7950
SOperatorInfo* createMultiTableTimeIntervalOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream,
                                                        SExprInfo* pExpr, int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
7951
  STableIntervalOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableIntervalOperatorInfo));
7952

L
Liu Jicong 已提交
7953 7954
  //  pInfo->binfo.pCtx = createSqlFunctionCtx(pRuntimeEnv, pExpr, numOfOutput, &pInfo->binfo.rowCellInfoOffset);
  //  pInfo->binfo.pRes = createOutputBuf(pExpr, numOfOutput, pResultInfo->capacity);
H
Haojun Liao 已提交
7955
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
7956

wafwerar's avatar
wafwerar 已提交
7957
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
7958 7959
  pOperator->name = "MultiTableTimeIntervalOperator";
  //  pOperator->operatorType = OP_MultiTableTimeInterval;
7960
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7961 7962 7963 7964 7965
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->info = pInfo;
  pOperator->pRuntimeEnv = pRuntimeEnv;
7966

H
Haojun Liao 已提交
7967
  pOperator->getNextFn = doSTableIntervalAgg;
H
Haojun Liao 已提交
7968
  pOperator->closeFn = destroyBasicOperatorInfo;
7969

L
Liu Jicong 已提交
7970
  int32_t code = appendDownstream(pOperator, &downstream, 1);
7971 7972 7973
  return pOperator;
}

L
Liu Jicong 已提交
7974 7975
SOperatorInfo* createAllMultiTableTimeIntervalOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream,
                                                           SExprInfo* pExpr, int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
7976
  STableIntervalOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableIntervalOperatorInfo));
7977

L
Liu Jicong 已提交
7978 7979
  //  pInfo->binfo.pCtx = createSqlFunctionCtx(pRuntimeEnv, pExpr, numOfOutput, &pInfo->binfo.rowCellInfoOffset);
  //  pInfo->binfo.pRes = createOutputBuf(pExpr, numOfOutput, pResultInfo->capacity);
H
Haojun Liao 已提交
7980
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
7981

wafwerar's avatar
wafwerar 已提交
7982
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
7983 7984
  pOperator->name = "AllMultiTableTimeIntervalOperator";
  //  pOperator->operatorType = OP_AllMultiTableTimeInterval;
7985
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
7986 7987 7988 7989 7990
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->info = pInfo;
  pOperator->pRuntimeEnv = pRuntimeEnv;
7991

H
Haojun Liao 已提交
7992
  pOperator->getNextFn = doAllSTableIntervalAgg;
H
Haojun Liao 已提交
7993
  pOperator->closeFn = destroyBasicOperatorInfo;
7994

L
Liu Jicong 已提交
7995
  int32_t code = appendDownstream(pOperator, &downstream, 1);
7996 7997 7998 7999

  return pOperator;
}

L
Liu Jicong 已提交
8000
static int32_t initGroupOptrInfo(SGroupbyOperatorInfo* pInfo, SArray* pGroupColList) {
8001 8002 8003 8004 8005 8006
  pInfo->pGroupColVals = taosArrayInit(4, sizeof(SGroupKeys));
  if (pInfo->pGroupColVals == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  int32_t numOfGroupCols = taosArrayGetSize(pGroupColList);
L
Liu Jicong 已提交
8007
  for (int32_t i = 0; i < numOfGroupCols; ++i) {
8008 8009 8010 8011
    SColumn* pCol = taosArrayGet(pGroupColList, i);
    pInfo->groupKeyLen += pCol->bytes;

    struct SGroupKeys key = {0};
L
Liu Jicong 已提交
8012 8013
    key.bytes = pCol->bytes;
    key.type = pCol->type;
8014
    key.isNull = false;
L
Liu Jicong 已提交
8015
    key.pData = taosMemoryCalloc(1, pCol->bytes);
8016 8017 8018 8019 8020 8021 8022 8023
    if (key.pData == NULL) {
      return TSDB_CODE_OUT_OF_MEMORY;
    }

    taosArrayPush(pInfo->pGroupColVals, &key);
  }

  int32_t nullFlagSize = sizeof(int8_t) * numOfGroupCols;
wafwerar's avatar
wafwerar 已提交
8024
  pInfo->keyBuf = taosMemoryCalloc(1, pInfo->groupKeyLen + nullFlagSize);
8025 8026 8027 8028 8029 8030 8031 8032

  if (pInfo->keyBuf == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
8033 8034 8035
SOperatorInfo* createGroupOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                       SSDataBlock* pResultBlock, SArray* pGroupColList, SExecTaskInfo* pTaskInfo,
                                       const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
8036
  SGroupbyOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SGroupbyOperatorInfo));
L
Liu Jicong 已提交
8037
  SOperatorInfo*        pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
8038 8039 8040
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
8041

H
Haojun Liao 已提交
8042 8043
  pInfo->pGroupCols = pGroupColList;
  initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, 4096, pResultBlock, pTaskInfo->id.str);
8044
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
8045

8046 8047 8048 8049 8050
  int32_t code = initGroupOptrInfo(pInfo, pGroupColList);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

L
Liu Jicong 已提交
8051
  pOperator->name = "GroupbyAggOperator";
8052
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
8053 8054 8055 8056 8057 8058 8059 8060
  pOperator->status = OP_NOT_OPENED;
  //  pOperator->operatorType = OP_Groupby;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfOutput = numOfCols;
  pOperator->info = pInfo;
  pOperator->_openFn = operatorDummyOpenFn;
  pOperator->getNextFn = hashGroupbyAggregate;
  pOperator->closeFn = destroyGroupbyOperatorInfo;
8061

8062
  code = appendDownstream(pOperator, &downstream, 1);
8063
  return pOperator;
H
Haojun Liao 已提交
8064

L
Liu Jicong 已提交
8065
_error:
wafwerar's avatar
wafwerar 已提交
8066 8067
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
8068
  return NULL;
8069 8070
}

H
Haojun Liao 已提交
8071
static int32_t initFillInfo(SFillOperatorInfo* pInfo, SExprInfo* pExpr, int32_t numOfCols, int64_t* fillVal,
L
Liu Jicong 已提交
8072
                            STimeWindow win, int32_t capacity, const char* id, SInterval* pInterval, int32_t fillType) {
H
Haojun Liao 已提交
8073 8074 8075 8076 8077 8078 8079 8080 8081 8082
  struct SFillColInfo* pColInfo = createFillColInfo(pExpr, numOfCols, (int64_t*)fillVal);

  TSKEY sk = TMIN(win.skey, win.ekey);
  TSKEY ek = TMAX(win.skey, win.ekey);

  // TODO set correct time precision
  STimeWindow w = TSWINDOW_INITIALIZER;
  getAlignQueryTimeWindow(pInterval, TSDB_TIME_PRECISION_MILLI, win.skey, sk, ek, &w);

  int32_t order = TSDB_ORDER_ASC;
L
Liu Jicong 已提交
8083 8084
  pInfo->pFillInfo = taosCreateFillInfo(order, w.skey, 0, capacity, numOfCols, pInterval->sliding,
                                        pInterval->slidingUnit, (int8_t)pInterval->precision, fillType, pColInfo, id);
H
Haojun Liao 已提交
8085

wafwerar's avatar
wafwerar 已提交
8086
  pInfo->p = taosMemoryCalloc(numOfCols, POINTER_BYTES);
H
Haojun Liao 已提交
8087 8088 8089 8090 8091 8092 8093 8094

  if (pInfo->pFillInfo == NULL || pInfo->p == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  } else {
    return TSDB_CODE_SUCCESS;
  }
}

L
Liu Jicong 已提交
8095 8096 8097
SOperatorInfo* createFillOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExpr, int32_t numOfCols,
                                      SInterval* pInterval, SSDataBlock* pResBlock, int32_t fillType, char* fillVal,
                                      bool multigroupResult, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
8098
  SFillOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SFillOperatorInfo));
L
Liu Jicong 已提交
8099
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
8100

L
Liu Jicong 已提交
8101
  pInfo->pRes = pResBlock;
8102
  pInfo->multigroupResult = multigroupResult;
L
Liu Jicong 已提交
8103
  pInfo->intervalInfo = *pInterval;
8104

H
Haojun Liao 已提交
8105
  SResultInfo* pResultInfo = &pOperator->resultInfo;
L
Liu Jicong 已提交
8106 8107
  int32_t      code = initFillInfo(pInfo, pExpr, numOfCols, (int64_t*)fillVal, pTaskInfo->window, pResultInfo->capacity,
                                   pTaskInfo->id.str, pInterval, fillType);
8108 8109 8110
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
8111

L
Liu Jicong 已提交
8112
  pOperator->name = "FillOperator";
8113
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
8114 8115 8116 8117 8118 8119 8120 8121
  pOperator->status = OP_NOT_OPENED;
  //  pOperator->operatorType = OP_Fill;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfCols;
  pOperator->info = pInfo;
  pOperator->_openFn = operatorDummyOpenFn;
  pOperator->getNextFn = doFill;
  pOperator->pTaskInfo = pTaskInfo;
8122

L
Liu Jicong 已提交
8123
  pOperator->closeFn = destroySFillOperatorInfo;
H
Haojun Liao 已提交
8124

8125
  code = appendDownstream(pOperator, &downstream, 1);
8126
  return pOperator;
H
Haojun Liao 已提交
8127

L
Liu Jicong 已提交
8128
_error:
wafwerar's avatar
wafwerar 已提交
8129 8130
  taosMemoryFreeClear(pOperator);
  taosMemoryFreeClear(pInfo);
H
Haojun Liao 已提交
8131
  return NULL;
8132 8133
}

L
Liu Jicong 已提交
8134 8135
SOperatorInfo* createSLimitOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream, SExprInfo* pExpr,
                                        int32_t numOfOutput, void* pMerger, bool multigroupResult) {
wafwerar's avatar
wafwerar 已提交
8136
  SSLimitOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSLimitOperatorInfo));
L
Liu Jicong 已提交
8137 8138 8139 8140 8141 8142 8143 8144 8145 8146
  SOperatorInfo*       pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));

  //  pInfo->orderColumnList = getResultGroupCheckColumns(pQueryAttr);
  //  pInfo->slimit          = pQueryAttr->slimit;
  //  pInfo->limit           = pQueryAttr->limit;
  //  pInfo->capacity        = pResultInfo->capacity;
  //  pInfo->threshold       = (int64_t)(pInfo->capacity * 0.8);
  //  pInfo->currentOffset   = pQueryAttr->limit.offset;
  //  pInfo->currentGroupOffset = pQueryAttr->slimit.offset;
  pInfo->multigroupResult = multigroupResult;
8147 8148 8149

  // TODO refactor
  int32_t len = 0;
L
Liu Jicong 已提交
8150
  for (int32_t i = 0; i < numOfOutput; ++i) {
8151 8152 8153
    len += pExpr[i].base.resSchema.bytes;
  }

L
Liu Jicong 已提交
8154
  int32_t numOfCols = (pInfo->orderColumnList != NULL) ? (int32_t)taosArrayGetSize(pInfo->orderColumnList) : 0;
wafwerar's avatar
wafwerar 已提交
8155
  pInfo->prevRow = taosMemoryCalloc(1, (POINTER_BYTES * numOfCols + len));
8156 8157

  int32_t offset = POINTER_BYTES * numOfCols;
L
Liu Jicong 已提交
8158
  for (int32_t i = 0; i < numOfCols; ++i) {
8159 8160 8161 8162 8163 8164
    pInfo->prevRow[i] = (char*)pInfo->prevRow + offset;

    SColIndex* index = taosArrayGet(pInfo->orderColumnList, i);
    offset += pExpr[index->colIndex].base.resSchema.bytes;
  }

L
Liu Jicong 已提交
8165
  //  pInfo->pRes = createOutputBuf(pExpr, numOfOutput, pOperator->resultInfo.capacity);
8166

L
Liu Jicong 已提交
8167
  pOperator->name = "SLimitOperator";
X
Xiaoyu Wang 已提交
8168
  // pOperator->operatorType = OP_SLimit;
8169
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
8170 8171 8172 8173
  pOperator->status = OP_NOT_OPENED;
  //  pOperator->exec         = doSLimit;
  pOperator->info = pInfo;
  pOperator->pRuntimeEnv = pRuntimeEnv;
H
Haojun Liao 已提交
8174
  pOperator->closeFn = destroySlimitOperatorInfo;
8175

L
Liu Jicong 已提交
8176
  int32_t code = appendDownstream(pOperator, &downstream, 1);
8177 8178 8179
  return pOperator;
}

L
Liu Jicong 已提交
8180
static SSDataBlock* doTagScan(SOperatorInfo* pOperator, bool* newgroup) {
8181 8182 8183 8184 8185 8186
#if 0
  SOperatorInfo* pOperator = (SOperatorInfo*) param;
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

H
Haojun Liao 已提交
8187
  STaskRuntimeEnv* pRuntimeEnv = pOperator->pRuntimeEnv;
H
Haojun Liao 已提交
8188
  int32_t maxNumOfTables = (int32_t)pResultInfo->capacity;
8189 8190 8191 8192 8193 8194 8195 8196 8197 8198

  STagScanInfo *pInfo = pOperator->info;
  SSDataBlock  *pRes = pInfo->pRes;
  *newgroup = false;

  int32_t count = 0;
  SArray* pa = GET_TABLEGROUP(pRuntimeEnv, 0);

  int32_t functionId = getExprFunctionId(&pOperator->pExpr[0]);
  if (functionId == FUNCTION_TID_TAG) { // return the tags & table Id
H
Haojun Liao 已提交
8199
    STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
8200 8201 8202 8203 8204 8205 8206 8207 8208 8209 8210
    assert(pQueryAttr->numOfOutput == 1);

    SExprInfo* pExprInfo = &pOperator->pExpr[0];
    int32_t rsize = pExprInfo->base.resSchema.bytes;

    count = 0;

    int16_t bytes = pExprInfo->base.resSchema.bytes;
    int16_t type  = pExprInfo->base.resSchema.type;

    for(int32_t i = 0; i < pQueryAttr->numOfTags; ++i) {
8211
      if (pQueryAttr->tagColList[i].colId == pExprInfo->base.pColumns->info.colId) {
8212 8213 8214 8215 8216 8217 8218 8219 8220 8221 8222 8223 8224 8225 8226 8227 8228 8229 8230 8231 8232 8233 8234 8235 8236 8237 8238 8239 8240 8241 8242
        bytes = pQueryAttr->tagColList[i].bytes;
        type = pQueryAttr->tagColList[i].type;
        break;
      }
    }

    SColumnInfoData* pColInfo = taosArrayGet(pRes->pDataBlock, 0);

    while(pInfo->curPos < pInfo->totalTables && count < maxNumOfTables) {
      int32_t i = pInfo->curPos++;
      STableQueryInfo *item = taosArrayGetP(pa, i);

      char *output = pColInfo->pData + count * rsize;
      varDataSetLen(output, rsize - VARSTR_HEADER_SIZE);

      output = varDataVal(output);
      STableId* id = TSDB_TABLEID(item->pTable);

      *(int16_t *)output = 0;
      output += sizeof(int16_t);

      *(int64_t *)output = id->uid;  // memory align problem, todo serialize
      output += sizeof(id->uid);

      *(int32_t *)output = id->tid;
      output += sizeof(id->tid);

      *(int32_t *)output = pQueryAttr->vgId;
      output += sizeof(pQueryAttr->vgId);

      char* data = NULL;
8243
      if (pExprInfo->base.pColumns->info.colId == TSDB_TBNAME_COLUMN_INDEX) {
8244 8245
        data = tsdbGetTableName(item->pTable);
      } else {
8246
        data = tsdbGetTableTagVal(item->pTable, pExprInfo->base.pColumns->info.colId, type, bytes);
8247 8248 8249 8250 8251 8252
      }

      doSetTagValueToResultBuf(output, data, type, bytes);
      count += 1;
    }

8253
    //qDebug("QInfo:0x%"PRIx64" create (tableId, tag) info completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
8254 8255 8256 8257 8258 8259
  } else if (functionId == FUNCTION_COUNT) {// handle the "count(tbname)" query
    SColumnInfoData* pColInfo = taosArrayGet(pRes->pDataBlock, 0);
    *(int64_t*)pColInfo->pData = pInfo->totalTables;
    count = 1;

    pOperator->status = OP_EXEC_DONE;
8260
    //qDebug("QInfo:0x%"PRIx64" create count(tbname) query, res:%d rows:1", GET_TASKID(pRuntimeEnv), count);
8261 8262 8263 8264 8265 8266 8267 8268 8269 8270 8271 8272 8273
  } else {  // return only the tags|table name etc.
    SExprInfo* pExprInfo = &pOperator->pExpr[0];  // todo use the column list instead of exprinfo

    count = 0;
    while(pInfo->curPos < pInfo->totalTables && count < maxNumOfTables) {
      int32_t i = pInfo->curPos++;

      STableQueryInfo* item = taosArrayGetP(pa, i);

      char *data = NULL, *dst = NULL;
      int16_t type = 0, bytes = 0;
      for(int32_t j = 0; j < pOperator->numOfOutput; ++j) {
        // not assign value in case of user defined constant output column
8274
        if (TSDB_COL_IS_UD_COL(pExprInfo[j].base.pColumns->flag)) {
8275 8276 8277 8278 8279 8280 8281
          continue;
        }

        SColumnInfoData* pColInfo = taosArrayGet(pRes->pDataBlock, j);
        type  = pExprInfo[j].base.resSchema.type;
        bytes = pExprInfo[j].base.resSchema.bytes;

8282
        if (pExprInfo[j].base.pColumns->info.colId == TSDB_TBNAME_COLUMN_INDEX) {
8283 8284
          data = tsdbGetTableName(item->pTable);
        } else {
8285
          data = tsdbGetTableTagVal(item->pTable, pExprInfo[j].base.pColumns->info.colId, type, bytes);
8286 8287 8288 8289 8290 8291 8292 8293 8294 8295 8296 8297 8298
        }

        dst  = pColInfo->pData + count * pExprInfo[j].base.resSchema.bytes;
        doSetTagValueToResultBuf(dst, data, type, bytes);
      }

      count += 1;
    }

    if (pInfo->curPos >= pInfo->totalTables) {
      pOperator->status = OP_EXEC_DONE;
    }

8299
    //qDebug("QInfo:0x%"PRIx64" create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
8300 8301 8302
  }

  if (pOperator->status == OP_EXEC_DONE) {
8303
    setTaskStatus(pOperator->pRuntimeEnv, TASK_COMPLETED);
8304 8305 8306 8307 8308 8309 8310 8311
  }

  pRes->info.rows = count;
  return (pRes->info.rows == 0)? NULL:pInfo->pRes;

#endif
}

H
Haojun Liao 已提交
8312
SOperatorInfo* createTagScanOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SExprInfo* pExpr, int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
8313
  STagScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
L
Liu Jicong 已提交
8314
  //  pInfo->pRes = createOutputBuf(pExpr, numOfOutput, pResultInfo->capacity);
8315 8316 8317 8318 8319 8320 8321

  size_t numOfGroup = GET_NUM_OF_TABLEGROUP(pRuntimeEnv);
  assert(numOfGroup == 0 || numOfGroup == 1);

  pInfo->totalTables = pRuntimeEnv->tableqinfoGroupInfo.numOfTables;
  pInfo->curPos = 0;

wafwerar's avatar
wafwerar 已提交
8322
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
8323
  pOperator->name = "SeqTableTagScan";
X
Xiaoyu Wang 已提交
8324
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN;
8325
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
8326 8327
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
H
Haojun Liao 已提交
8328
  pOperator->getNextFn = doTagScan;
L
Liu Jicong 已提交
8329 8330 8331
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->pRuntimeEnv = pRuntimeEnv;
H
Haojun Liao 已提交
8332
  pOperator->closeFn = destroyTagScanOperatorInfo;
8333 8334 8335

  return pOperator;
}
L
Liu Jicong 已提交
8336
static bool initMultiDistinctInfo(SDistinctOperatorInfo* pInfo, SOperatorInfo* pOperator, SSDataBlock* pBlock) {
8337
  if (taosArrayGetSize(pInfo->pDistinctDataInfo) == pOperator->numOfOutput) {
L
Liu Jicong 已提交
8338
    // distinct info already inited
8339 8340 8341
    return true;
  }
  for (int i = 0; i < pOperator->numOfOutput; i++) {
L
Liu Jicong 已提交
8342
    //    pInfo->totalBytes += pOperator->pExpr[i].base.colBytes;
8343 8344 8345 8346 8347 8348 8349 8350 8351 8352 8353 8354 8355
  }
  for (int i = 0; i < pOperator->numOfOutput; i++) {
    int numOfBlock = (int)(taosArrayGetSize(pBlock->pDataBlock));
    assert(i < numOfBlock);
    for (int j = 0; j < numOfBlock; j++) {
      SColumnInfoData* pColDataInfo = taosArrayGet(pBlock->pDataBlock, j);
      if (pColDataInfo->info.colId == pOperator->pExpr[i].base.resSchema.colId) {
        SDistinctDataInfo item = {.index = j, .type = pColDataInfo->info.type, .bytes = pColDataInfo->info.bytes};
        taosArrayInsert(pInfo->pDistinctDataInfo, i, &item);
      }
    }
  }
  pInfo->totalBytes += (int32_t)strlen(MULTI_KEY_DELIM) * (pOperator->numOfOutput);
L
Liu Jicong 已提交
8356 8357
  pInfo->buf = taosMemoryCalloc(1, pInfo->totalBytes);
  return taosArrayGetSize(pInfo->pDistinctDataInfo) == pOperator->numOfOutput ? true : false;
8358 8359
}

L
Liu Jicong 已提交
8360 8361 8362
static void buildMultiDistinctKey(SDistinctOperatorInfo* pInfo, SSDataBlock* pBlock, int32_t rowId) {
  char* p = pInfo->buf;
  memset(p, 0, pInfo->totalBytes);
8363 8364

  for (int i = 0; i < taosArrayGetSize(pInfo->pDistinctDataInfo); i++) {
L
Liu Jicong 已提交
8365
    SDistinctDataInfo* pDistDataInfo = (SDistinctDataInfo*)taosArrayGet(pInfo->pDistinctDataInfo, i);
8366
    SColumnInfoData*   pColDataInfo = taosArrayGet(pBlock->pDataBlock, pDistDataInfo->index);
L
Liu Jicong 已提交
8367 8368 8369
    char*              val = ((char*)pColDataInfo->pData) + pColDataInfo->info.bytes * rowId;
    if (isNull(val, pDistDataInfo->type)) {
      p += pDistDataInfo->bytes;
8370 8371 8372 8373 8374 8375 8376 8377 8378 8379 8380 8381 8382 8383
      continue;
    }
    if (IS_VAR_DATA_TYPE(pDistDataInfo->type)) {
      memcpy(p, varDataVal(val), varDataLen(val));
      p += varDataLen(val);
    } else {
      memcpy(p, val, pDistDataInfo->bytes);
      p += pDistDataInfo->bytes;
    }
    memcpy(p, MULTI_KEY_DELIM, strlen(MULTI_KEY_DELIM));
    p += strlen(MULTI_KEY_DELIM);
  }
}

L
Liu Jicong 已提交
8384
static SSDataBlock* hashDistinct(SOperatorInfo* pOperator, bool* newgroup) {
8385 8386 8387 8388 8389
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  SDistinctOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
8390
  SSDataBlock*           pRes = pInfo->pRes;
8391 8392 8393

  pRes->info.rows = 0;
  SSDataBlock* pBlock = NULL;
L
Liu Jicong 已提交
8394 8395

  while (1) {
H
Haojun Liao 已提交
8396
    publishOperatorProfEvent(pOperator->pDownstream[0], QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
8397
    pBlock = pOperator->pDownstream[0]->getNextFn(pOperator->pDownstream[0], newgroup);
H
Haojun Liao 已提交
8398
    publishOperatorProfEvent(pOperator->pDownstream[0], QUERY_PROF_AFTER_OPERATOR_EXEC);
8399 8400 8401 8402 8403 8404 8405 8406 8407

    if (pBlock == NULL) {
      doSetOperatorCompleted(pOperator);
      break;
    }
    if (!initMultiDistinctInfo(pInfo, pOperator, pBlock)) {
      doSetOperatorCompleted(pOperator);
      break;
    }
L
Liu Jicong 已提交
8408
    // ensure result output buf
8409 8410 8411 8412
    if (pRes->info.rows + pBlock->info.rows > pInfo->outputCapacity) {
      int32_t newSize = pRes->info.rows + pBlock->info.rows;
      for (int i = 0; i < taosArrayGetSize(pRes->pDataBlock); i++) {
        SColumnInfoData*   pResultColInfoData = taosArrayGet(pRes->pDataBlock, i);
L
Liu Jicong 已提交
8413 8414
        SDistinctDataInfo* pDistDataInfo = taosArrayGet(pInfo->pDistinctDataInfo, i);
        char*              tmp = taosMemoryRealloc(pResultColInfoData->pData, newSize * pDistDataInfo->bytes);
8415 8416 8417 8418 8419 8420 8421 8422 8423 8424 8425 8426 8427 8428 8429 8430
        if (tmp == NULL) {
          return NULL;
        } else {
          pResultColInfoData->pData = tmp;
        }
      }
      pInfo->outputCapacity = newSize;
    }

    for (int32_t i = 0; i < pBlock->info.rows; i++) {
      buildMultiDistinctKey(pInfo, pBlock, i);
      if (taosHashGet(pInfo->pSet, pInfo->buf, pInfo->totalBytes) == NULL) {
        int32_t dummy;
        taosHashPut(pInfo->pSet, pInfo->buf, pInfo->totalBytes, &dummy, sizeof(dummy));
        for (int j = 0; j < taosArrayGetSize(pRes->pDataBlock); j++) {
          SDistinctDataInfo* pDistDataInfo = taosArrayGet(pInfo->pDistinctDataInfo, j);  // distinct meta info
L
Liu Jicong 已提交
8431 8432
          SColumnInfoData*   pColInfoData = taosArrayGet(pBlock->pDataBlock, pDistDataInfo->index);  // src
          SColumnInfoData*   pResultColInfoData = taosArrayGet(pRes->pDataBlock, j);                 // dist
8433 8434

          char* val = ((char*)pColInfoData->pData) + pDistDataInfo->bytes * i;
L
Liu Jicong 已提交
8435
          char* start = pResultColInfoData->pData + pDistDataInfo->bytes * pInfo->pRes->info.rows;
8436 8437 8438
          memcpy(start, val, pDistDataInfo->bytes);
        }
        pRes->info.rows += 1;
L
Liu Jicong 已提交
8439
      }
8440 8441 8442 8443 8444 8445
    }

    if (pRes->info.rows >= pInfo->threshold) {
      break;
    }
  }
L
Liu Jicong 已提交
8446
  return (pInfo->pRes->info.rows > 0) ? pInfo->pRes : NULL;
8447 8448
}

L
Liu Jicong 已提交
8449 8450
SOperatorInfo* createDistinctOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream, SExprInfo* pExpr,
                                          int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
8451
  SDistinctOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SDistinctOperatorInfo));
L
Liu Jicong 已提交
8452 8453 8454 8455 8456
  pInfo->totalBytes = 0;
  pInfo->buf = NULL;
  pInfo->threshold = tsMaxNumOfDistinctResults;  // distinct result threshold
  pInfo->outputCapacity = 4096;
  pInfo->pDistinctDataInfo = taosArrayInit(numOfOutput, sizeof(SDistinctDataInfo));
8457
  pInfo->pSet = taosHashInit(64, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY), false, HASH_NO_LOCK);
L
Liu Jicong 已提交
8458
  //  pInfo->pRes = createOutputBuf(pExpr, numOfOutput, (int32_t) pInfo->outputCapacity);
8459

wafwerar's avatar
wafwerar 已提交
8460
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
8461
  pOperator->name = "DistinctOperator";
8462
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
8463 8464 8465 8466 8467 8468
  pOperator->status = OP_NOT_OPENED;
  //  pOperator->operatorType = OP_Distinct;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->info = pInfo;
  pOperator->pRuntimeEnv = pRuntimeEnv;
H
Haojun Liao 已提交
8469
  pOperator->getNextFn = hashDistinct;
L
Liu Jicong 已提交
8470
  pOperator->pExpr = pExpr;
H
Haojun Liao 已提交
8471
  pOperator->closeFn = destroyDistinctOperatorInfo;
8472

L
Liu Jicong 已提交
8473
  int32_t code = appendDownstream(pOperator, &downstream, 1);
8474 8475 8476
  return pOperator;
}

L
Liu Jicong 已提交
8477
static int32_t getColumnIndexInSource(SQueriedTableInfo* pTableInfo, SExprBasicInfo* pExpr, SColumnInfo* pTagCols) {
8478 8479
  int32_t j = 0;

H
Haojun Liao 已提交
8480 8481
  if (TSDB_COL_IS_TAG(pExpr->pParam[0].pCol->type)) {
    if (pExpr->pParam[0].pCol->colId == TSDB_TBNAME_COLUMN_INDEX) {
8482 8483 8484
      return TSDB_TBNAME_COLUMN_INDEX;
    }

L
Liu Jicong 已提交
8485
    while (j < pTableInfo->numOfTags) {
H
Haojun Liao 已提交
8486
      if (pExpr->pParam[0].pCol->colId == pTagCols[j].colId) {
8487 8488 8489 8490 8491 8492
        return j;
      }

      j += 1;
    }

8493
  } /*else if (TSDB_COL_IS_UD_COL(pExpr->colInfo.flag)) {  // user specified column data
8494 8495 8496 8497 8498 8499 8500 8501 8502
    return TSDB_UD_COLUMN_INDEX;
  } else {
    while (j < pTableInfo->numOfCols) {
      if (pExpr->colInfo.colId == pTableInfo->colList[j].colId) {
        return j;
      }

      j += 1;
    }
8503
  }*/
8504 8505 8506 8507

  return INT32_MIN;  // return a less than TSDB_TBNAME_COLUMN_INDEX value
}

L
Liu Jicong 已提交
8508
bool validateExprColumnInfo(SQueriedTableInfo* pTableInfo, SExprBasicInfo* pExpr, SColumnInfo* pTagCols) {
8509 8510 8511 8512 8513 8514
  int32_t j = getColumnIndexInSource(pTableInfo, pExpr, pTagCols);
  return j != INT32_MIN;
}

static int32_t deserializeColFilterInfo(SColumnFilterInfo* pColFilters, int16_t numOfFilters, char** pMsg) {
  for (int32_t f = 0; f < numOfFilters; ++f) {
L
Liu Jicong 已提交
8515
    SColumnFilterInfo* pFilterMsg = (SColumnFilterInfo*)(*pMsg);
8516

L
Liu Jicong 已提交
8517
    SColumnFilterInfo* pColFilter = &pColFilters[f];
8518 8519 8520 8521 8522 8523 8524
    pColFilter->filterstr = htons(pFilterMsg->filterstr);

    (*pMsg) += sizeof(SColumnFilterInfo);

    if (pColFilter->filterstr) {
      pColFilter->len = htobe64(pFilterMsg->len);

L
Liu Jicong 已提交
8525 8526
      pColFilter->pz =
          (int64_t)taosMemoryCalloc(1, (size_t)(pColFilter->len + 1 * TSDB_NCHAR_SIZE));  // note: null-terminator
8527 8528 8529 8530
      if (pColFilter->pz == 0) {
        return TSDB_CODE_QRY_OUT_OF_MEMORY;
      }

L
Liu Jicong 已提交
8531
      memcpy((void*)pColFilter->pz, (*pMsg), (size_t)pColFilter->len);
8532 8533 8534 8535 8536 8537 8538 8539 8540 8541 8542 8543 8544
      (*pMsg) += (pColFilter->len + 1);
    } else {
      pColFilter->lowerBndi = htobe64(pFilterMsg->lowerBndi);
      pColFilter->upperBndi = htobe64(pFilterMsg->upperBndi);
    }

    pColFilter->lowerRelOptr = htons(pFilterMsg->lowerRelOptr);
    pColFilter->upperRelOptr = htons(pFilterMsg->upperRelOptr);
  }

  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
8545 8546
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
                                  const char* name) {
H
Haojun Liao 已提交
8547 8548
  SResSchema s = {0};
  s.scale = scale;
L
Liu Jicong 已提交
8549
  s.type = type;
H
Haojun Liao 已提交
8550 8551
  s.bytes = bytes;
  s.colId = slotId;
H
Haojun Liao 已提交
8552
  s.precision = precision;
H
Haojun Liao 已提交
8553 8554 8555 8556
  strncpy(s.name, name, tListLen(s.name));

  return s;
}
H
Haojun Liao 已提交
8557

H
Haojun Liao 已提交
8558
SExprInfo* createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, int32_t* numOfExprs) {
H
Haojun Liao 已提交
8559
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
8560 8561 8562 8563
  int32_t numOfGroupKeys = 0;
  if (pGroupKeys != NULL) {
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
  }
H
Haojun Liao 已提交
8564

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

L
Liu Jicong 已提交
8568
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
H
Haojun Liao 已提交
8569 8570 8571 8572 8573 8574
    STargetNode* pTargetNode = NULL;
    if (i < numOfFuncs) {
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    } else {
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
    }
H
Haojun Liao 已提交
8575

H
Haojun Liao 已提交
8576
    SExprInfo* pExp = &pExprs[pTargetNode->slotId];
H
Haojun Liao 已提交
8577

wafwerar's avatar
wafwerar 已提交
8578
    pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
H
Haojun Liao 已提交
8579
    pExp->pExpr->_function.num = 1;
H
Haojun Liao 已提交
8580
    pExp->pExpr->_function.functionId = -1;
H
Haojun Liao 已提交
8581

H
Haojun Liao 已提交
8582
    // it is a project query, or group by column
H
Haojun Liao 已提交
8583
    if (nodeType(pTargetNode->pExpr) == QUERY_NODE_COLUMN) {
H
Haojun Liao 已提交
8584
      pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
L
Liu Jicong 已提交
8585
      SColumnNode* pColNode = (SColumnNode*)pTargetNode->pExpr;
H
Haojun Liao 已提交
8586

G
Ganlin Zhao 已提交
8587 8588 8589 8590 8591
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;
      pExp->base.pParam[0].pCol = taosMemoryCalloc(1, sizeof(SColumn));
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;

H
Haojun Liao 已提交
8592 8593
      SDataType* pType = &pColNode->node.resType;
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale, pType->precision, pColNode->colName);
G
Ganlin Zhao 已提交
8594 8595

      SColumn* pCol = pExp->base.pParam[0].pCol;
8596
      pCol->slotId    = pColNode->slotId;   // TODO refactor
H
Haojun Liao 已提交
8597 8598 8599 8600
      pCol->bytes     = pType->bytes;
      pCol->type      = pType->type;
      pCol->scale     = pType->scale;
      pCol->precision = pType->precision;
8601
    } else if (nodeType(pTargetNode->pExpr) == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
8602
      pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
H
Haojun Liao 已提交
8603 8604 8605
      SFunctionNode* pFuncNode = (SFunctionNode*)pTargetNode->pExpr;

      SDataType* pType = &pFuncNode->node.resType;
L
Liu Jicong 已提交
8606 8607
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pFuncNode->node.aliasName);
H
Haojun Liao 已提交
8608

H
Haojun Liao 已提交
8609
      pExp->pExpr->_function.functionId = pFuncNode->funcId;
H
Haojun Liao 已提交
8610
      pExp->pExpr->_function.pFunctNode = pFuncNode;
L
Liu Jicong 已提交
8611 8612
      strncpy(pExp->pExpr->_function.functionName, pFuncNode->functionName,
              tListLen(pExp->pExpr->_function.functionName));
H
Haojun Liao 已提交
8613 8614 8615

      // TODO: value parameter needs to be handled
      int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
G
Ganlin Zhao 已提交
8616 8617 8618 8619

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

H
Haojun Liao 已提交
8620 8621
      for (int32_t j = 0; j < numOfParam; ++j) {
        SNode*       p1 = nodesListGetNode(pFuncNode->pParameterList, j);
G
Ganlin Zhao 已提交
8622 8623 8624 8625 8626 8627 8628 8629 8630 8631 8632 8633 8634 8635 8636 8637 8638 8639
        if (p1->type == QUERY_NODE_COLUMN) {
          SColumnNode* pcn = (SColumnNode*)p1;  // TODO refactor

          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
          pExp->base.pParam[j].pCol = taosMemoryCalloc(1, sizeof(SColumn));
          SColumn* pCol = pExp->base.pParam[j].pCol;

          pCol->slotId      = pcn->slotId;
          pCol->bytes       = pcn->node.resType.bytes;
          pCol->type        = pcn->node.resType.type;
          pCol->scale       = pcn->node.resType.scale;
          pCol->precision   = pcn->node.resType.precision;
          pCol->dataBlockId = pcn->dataBlockId;
        } else if (p1->type == QUERY_NODE_VALUE) {
          SValueNode* pvn = (SValueNode*)p1;

          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
        }
H
Haojun Liao 已提交
8640
      }
8641
    } else if (nodeType(pTargetNode->pExpr) == QUERY_NODE_OPERATOR) {
H
Haojun Liao 已提交
8642
      pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
L
Liu Jicong 已提交
8643
      SOperatorNode* pNode = (SOperatorNode*)pTargetNode->pExpr;
8644

G
Ganlin Zhao 已提交
8645 8646 8647 8648 8649
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;
      pExp->base.pParam[0].pCol = taosMemoryCalloc(1, sizeof(SColumn));
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;

8650
      SDataType* pType = &pNode->node.resType;
L
Liu Jicong 已提交
8651 8652
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pNode->node.aliasName);
8653 8654 8655

      pExp->pExpr->_optrRoot.pRootNode = pTargetNode->pExpr;

G
Ganlin Zhao 已提交
8656
      SColumn* pCol   = pExp->base.pParam[0].pCol;
8657 8658 8659 8660 8661 8662 8663
      pCol->slotId    = pTargetNode->slotId;   // TODO refactor
      pCol->bytes     = pType->bytes;
      pCol->type      = pType->type;
      pCol->scale     = pType->scale;
      pCol->precision = pType->precision;
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
8664 8665 8666
    }
  }

H
Haojun Liao 已提交
8667
  return pExprs;
H
Haojun Liao 已提交
8668 8669
}

H
Haojun Liao 已提交
8670
static SExecTaskInfo* createExecTaskInfo(uint64_t queryId, uint64_t taskId) {
wafwerar's avatar
wafwerar 已提交
8671
  SExecTaskInfo* pTaskInfo = taosMemoryCalloc(1, sizeof(SExecTaskInfo));
8672
  setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
8673

8674
  pTaskInfo->cost.created = taosGetTimestampMs();
H
Haojun Liao 已提交
8675
  pTaskInfo->id.queryId = queryId;
H
Haojun Liao 已提交
8676

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

8681 8682
  return pTaskInfo;
}
H
Haojun Liao 已提交
8683

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

L
Liu Jicong 已提交
8687 8688
static int32_t doCreateTableGroup(void* metaHandle, int32_t tableType, uint64_t tableUid, STableGroupInfo* pGroupInfo,
                                  uint64_t queryId, uint64_t taskId);
8689 8690
static SArray* extractTableIdList(const STableGroupInfo* pTableGroupInfo);
static SArray* extractScanColumnId(SNodeList* pNodeList);
H
Haojun Liao 已提交
8691
static SArray* extractColumnInfo(SNodeList* pNodeList);
H
Haojun Liao 已提交
8692
static SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols);
H
Haojun Liao 已提交
8693
static SArray* createSortInfo(SNodeList* pNodeList);
H
Haojun Liao 已提交
8694

L
Liu Jicong 已提交
8695 8696
SOperatorInfo* doCreateOperatorTreeNode(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, SReadHandle* pHandle,
                                        uint64_t queryId, uint64_t taskId, STableGroupInfo* pTableGroupInfo) {
X
Xiaoyu Wang 已提交
8697 8698 8699
  if (pPhyNode->pChildren == NULL || LIST_LENGTH(pPhyNode->pChildren) == 0) {
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == nodeType(pPhyNode)) {
      SScanPhysiNode* pScanPhyNode = (SScanPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
8700

H
Haojun Liao 已提交
8701
      int32_t     numOfCols = 0;
L
Liu Jicong 已提交
8702 8703 8704 8705
      tsdbReaderT pDataReader =
          doCreateDataReader((STableScanPhysiNode*)pPhyNode, pHandle, pTableGroupInfo, (uint64_t)queryId, taskId);
      SArray* pColList =
          extractColMatchInfo(pScanPhyNode->pScanCols, pScanPhyNode->node.pOutputDataBlockDesc, &numOfCols);
H
Haojun Liao 已提交
8706

L
Liu Jicong 已提交
8707 8708
      return createTableScanOperatorInfo(pDataReader, pScanPhyNode->order, numOfCols, pScanPhyNode->count,
                                         pScanPhyNode->reverse, pColList, pTaskInfo);
X
Xiaoyu Wang 已提交
8709
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == nodeType(pPhyNode)) {
H
Haojun Liao 已提交
8710
      SExchangePhysiNode* pExchange = (SExchangePhysiNode*)pPhyNode;
L
Liu Jicong 已提交
8711
      SSDataBlock*        pResBlock = createOutputBuf_rv1(pExchange->node.pOutputDataBlockDesc);
H
Haojun Liao 已提交
8712
      return createExchangeOperatorInfo(pExchange->pSrcEndPoints, pResBlock, pTaskInfo);
X
Xiaoyu Wang 已提交
8713
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == nodeType(pPhyNode)) {
H
Haojun Liao 已提交
8714
      SScanPhysiNode* pScanPhyNode = (SScanPhysiNode*)pPhyNode;  // simple child table.
8715

L
Liu Jicong 已提交
8716 8717
      int32_t code = doCreateTableGroup(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableGroupInfo,
                                        queryId, taskId);
H
Haojun Liao 已提交
8718
      SArray* tableIdList = extractTableIdList(pTableGroupInfo);
H
Haojun Liao 已提交
8719

8720
      SSDataBlock* pResBlock = createOutputBuf_rv1(pScanPhyNode->node.pOutputDataBlockDesc);
L
Liu Jicong 已提交
8721
      SArray*      colList = extractScanColumnId(pScanPhyNode->pScanCols);
H
Haojun Liao 已提交
8722

L
Liu Jicong 已提交
8723 8724
      SOperatorInfo* pOperator =
          createStreamScanOperatorInfo(pHandle->reader, pResBlock, colList, tableIdList, pTaskInfo);
8725
      taosArrayDestroy(tableIdList);
H
Haojun Liao 已提交
8726
      return pOperator;
H
Haojun Liao 已提交
8727
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == nodeType(pPhyNode)) {
L
Liu Jicong 已提交
8728 8729
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
      SSDataBlock*               pResBlock = createOutputBuf_rv1(pSysScanPhyNode->scan.node.pOutputDataBlockDesc);
H
Haojun Liao 已提交
8730

8731
      struct SScanPhysiNode* pScanNode = &pSysScanPhyNode->scan;
L
Liu Jicong 已提交
8732
      SArray*                colList = extractScanColumnId(pScanNode->pScanCols);
8733

L
Liu Jicong 已提交
8734 8735 8736
      SOperatorInfo* pOperator = createSysTableScanOperatorInfo(
          pHandle->meta, pResBlock, &pScanNode->tableName, pScanNode->node.pConditions, pSysScanPhyNode->mgmtEpSet,
          colList, pTaskInfo, pSysScanPhyNode->showRewrite, pSysScanPhyNode->accountId);
H
Haojun Liao 已提交
8737 8738 8739
      return pOperator;
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
8740 8741 8742
    }
  }

H
Haojun Liao 已提交
8743 8744 8745 8746
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == nodeType(pPhyNode)) {
    size_t size = LIST_LENGTH(pPhyNode->pChildren);
    assert(size == 1);

H
Haojun Liao 已提交
8747 8748
    SPhysiNode*    pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, 0);
    SOperatorInfo* op = doCreateOperatorTreeNode(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);
H
Haojun Liao 已提交
8749

L
Liu Jicong 已提交
8750 8751
    int32_t      num = 0;
    SExprInfo*   pExprInfo = createExprInfo(((SProjectPhysiNode*)pPhyNode)->pProjections, NULL, &num);
H
Haojun Liao 已提交
8752 8753
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
    return createProjectOperatorInfo(op, pExprInfo, num, pResBlock, pTaskInfo);
H
Haojun Liao 已提交
8754
  } else if (QUERY_NODE_PHYSICAL_PLAN_AGG == nodeType(pPhyNode)) {
X
Xiaoyu Wang 已提交
8755
    size_t size = LIST_LENGTH(pPhyNode->pChildren);
H
Haojun Liao 已提交
8756 8757
    assert(size == 1);

H
Haojun Liao 已提交
8758
    for (int32_t i = 0; i < size; ++i) {
H
Haojun Liao 已提交
8759
      SPhysiNode*    pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
H
Haojun Liao 已提交
8760
      SOperatorInfo* op = doCreateOperatorTreeNode(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);
H
Haojun Liao 已提交
8761

H
Haojun Liao 已提交
8762
      int32_t num = 0;
H
Haojun Liao 已提交
8763 8764

      SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
L
Liu Jicong 已提交
8765 8766
      SExprInfo*     pExprInfo = createExprInfo(pAggNode->pAggFuncs, pAggNode->pGroupKeys, &num);
      SSDataBlock*   pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
H
Haojun Liao 已提交
8767 8768 8769 8770 8771 8772 8773

      if (pAggNode->pGroupKeys != NULL) {
        SArray* pColList = extractColumnInfo(pAggNode->pGroupKeys);
        return createGroupOperatorInfo(op, pExprInfo, num, pResBlock, pColList, pTaskInfo, NULL);
      } else {
        return createAggregateOperatorInfo(op, pExprInfo, num, pResBlock, pTaskInfo, pTableGroupInfo);
      }
H
Haojun Liao 已提交
8774
    }
H
Haojun Liao 已提交
8775 8776 8777 8778 8779 8780 8781 8782
  } else if (QUERY_NODE_PHYSICAL_PLAN_INTERVAL == nodeType(pPhyNode)) {
    size_t size = LIST_LENGTH(pPhyNode->pChildren);
    assert(size == 1);

    for (int32_t i = 0; i < size; ++i) {
      SPhysiNode*    pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, i);
      SOperatorInfo* op = doCreateOperatorTreeNode(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);

8783 8784
      SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;

L
Liu Jicong 已提交
8785 8786 8787
      // todo: set the correct primary timestamp key column
      int32_t      num = 0;
      SExprInfo*   pExprInfo = createExprInfo(pIntervalPhyNode->window.pFuncs, NULL, &num);
H
Haojun Liao 已提交
8788
      SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
8789

H
Haojun Liao 已提交
8790 8791
      SInterval interval = {.interval = pIntervalPhyNode->interval,
                            .sliding = pIntervalPhyNode->sliding,
H
Haojun Liao 已提交
8792
                            .intervalUnit = pIntervalPhyNode->intervalUnit,
H
Haojun Liao 已提交
8793 8794
                            .slidingUnit = pIntervalPhyNode->slidingUnit,
                            .offset = pIntervalPhyNode->offset};
8795
      return createIntervalOperatorInfo(op, pExprInfo, num, pResBlock, &interval, pTableGroupInfo, pTaskInfo);
H
Haojun Liao 已提交
8796
    }
H
Haojun Liao 已提交
8797 8798 8799 8800
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == nodeType(pPhyNode)) {
    size_t size = LIST_LENGTH(pPhyNode->pChildren);
    assert(size == 1);

H
Haojun Liao 已提交
8801 8802
    SPhysiNode*    pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, 0);
    SOperatorInfo* op = doCreateOperatorTreeNode(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);
H
Haojun Liao 已提交
8803

H
Haojun Liao 已提交
8804
    SSortPhysiNode* pSortPhyNode = (SSortPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
8805

H
Haojun Liao 已提交
8806 8807
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
    SArray*      info = createSortInfo(pSortPhyNode->pSortKeys);
H
Haojun Liao 已提交
8808
    return createSortOperatorInfo(op, pResBlock, info, pTaskInfo);
H
Haojun Liao 已提交
8809 8810 8811 8812 8813 8814 8815 8816 8817 8818 8819 8820 8821
  } else if (QUERY_NODE_PHYSICAL_PLAN_SESSION_WINDOW == nodeType(pPhyNode)) {
    size_t size = LIST_LENGTH(pPhyNode->pChildren);
    assert(size == 1);

    SPhysiNode*    pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, 0);
    SOperatorInfo* op = doCreateOperatorTreeNode(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);

    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;

    int32_t      num = 0;
    SExprInfo*   pExprInfo = createExprInfo(pSessionNode->window.pFuncs, NULL, &num);
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
    return createSessionAggOperatorInfo(op, pExprInfo, num, pResBlock, pSessionNode->gap, pTaskInfo);
H
Haojun Liao 已提交
8822 8823
  } else {
    ASSERT(0);
L
Liu Jicong 已提交
8824 8825 8826 8827 8828 8829 8830 8831 8832 8833
  } /*else if (pPhyNode->info.type == OP_MultiTableAggregate) {
     size_t size = taosArrayGetSize(pPhyNode->pChildren);
     assert(size == 1);

     for (int32_t i = 0; i < size; ++i) {
       SPhysiNode*      pChildNode = taosArrayGetP(pPhyNode->pChildren, i);
       SOperatorInfo* op = doCreateOperatorTreeNode(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);
       return createMultiTableAggOperatorInfo(op, pPhyNode->pTargets, pTaskInfo, pTableGroupInfo);
     }
   }*/
8834
}
H
Haojun Liao 已提交
8835

L
Liu Jicong 已提交
8836 8837
static tsdbReaderT createDataReaderImpl(STableScanPhysiNode* pTableScanNode, STableGroupInfo* pGroupInfo,
                                        void* readHandle, uint64_t queryId, uint64_t taskId) {
H
Haojun Liao 已提交
8838
  STsdbQueryCond cond = {.loadExternalRows = false};
H
Haojun Liao 已提交
8839

L
Liu Jicong 已提交
8840
  cond.order = pTableScanNode->scan.order;
X
Xiaoyu Wang 已提交
8841
  cond.numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
L
Liu Jicong 已提交
8842
  cond.colList = taosMemoryCalloc(cond.numOfCols, sizeof(SColumnInfo));
H
Haojun Liao 已提交
8843 8844 8845 8846 8847
  if (cond.colList == NULL) {
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
  }

X
Xiaoyu Wang 已提交
8848
  cond.twindow = pTableScanNode->scanRange;
H
Haojun Liao 已提交
8849
  cond.type = BLOCK_LOAD_OFFSET_SEQ_ORDER;
L
Liu Jicong 已提交
8850
  //  cond.type = pTableScanNode->scanFlag;
H
Haojun Liao 已提交
8851

H
Haojun Liao 已提交
8852
  int32_t j = 0;
H
Haojun Liao 已提交
8853
  for (int32_t i = 0; i < cond.numOfCols; ++i) {
H
Haojun Liao 已提交
8854 8855
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
8856 8857 8858
    if (pColNode->colType == COLUMN_TYPE_TAG) {
      continue;
    }
H
Haojun Liao 已提交
8859

L
Liu Jicong 已提交
8860
    cond.colList[j].type = pColNode->node.resType.type;
H
Haojun Liao 已提交
8861 8862 8863
    cond.colList[j].bytes = pColNode->node.resType.bytes;
    cond.colList[j].colId = pColNode->colId;
    j += 1;
H
Haojun Liao 已提交
8864 8865
  }

H
Haojun Liao 已提交
8866
  cond.numOfCols = j;
H
Haojun Liao 已提交
8867
  return tsdbQueryTables(readHandle, &cond, pGroupInfo, queryId, taskId);
H
Haojun Liao 已提交
8868
}
H
Haojun Liao 已提交
8869

8870
SArray* extractScanColumnId(SNodeList* pNodeList) {
L
Liu Jicong 已提交
8871
  size_t  numOfCols = LIST_LENGTH(pNodeList);
8872 8873 8874 8875 8876 8877
  SArray* pList = taosArrayInit(numOfCols, sizeof(int16_t));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
8878
  for (int32_t i = 0; i < numOfCols; ++i) {
X
Xiaoyu Wang 已提交
8879
    for (int32_t j = 0; j < numOfCols; ++j) {
L
Liu Jicong 已提交
8880
      STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, j);
X
Xiaoyu Wang 已提交
8881
      if (pNode->slotId == i) {
L
Liu Jicong 已提交
8882
        SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
X
Xiaoyu Wang 已提交
8883 8884 8885 8886
        taosArrayPush(pList, &pColNode->colId);
        break;
      }
    }
8887 8888 8889 8890 8891
  }

  return pList;
}

H
Haojun Liao 已提交
8892
SArray* extractColumnInfo(SNodeList* pNodeList) {
L
Liu Jicong 已提交
8893
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
8894 8895 8896 8897 8898 8899
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
8900 8901 8902
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
8903 8904 8905

    SColumn c = {0};
    c.slotId = pColNode->slotId;
L
Liu Jicong 已提交
8906 8907 8908 8909 8910
    c.colId = pColNode->colId;
    c.type = pColNode->node.resType.type;
    c.bytes = pColNode->node.resType.bytes;
    c.precision = pColNode->node.resType.precision;
    c.scale = pColNode->node.resType.scale;
H
Haojun Liao 已提交
8911 8912 8913 8914 8915 8916 8917

    taosArrayPush(pList, &c);
  }

  return pList;
}

H
Haojun Liao 已提交
8918
SArray* createSortInfo(SNodeList* pNodeList) {
L
Liu Jicong 已提交
8919
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
8920 8921 8922 8923 8924 8925
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return pList;
  }

L
Liu Jicong 已提交
8926 8927 8928 8929 8930
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode*      pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)pNode->pExpr;
    SBlockOrderInfo   bi = {0};
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
H
Haojun Liao 已提交
8931 8932 8933 8934
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);

    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;
    bi.slotId = pColNode->slotId;
L
Liu Jicong 已提交
8935 8936 8937 8938 8939 8940 8941 8942
    //    pColNode->order;
    //    SColumn c = {0};
    //    c.slotId = pColNode->slotId;
    //    c.colId  = pColNode->colId;
    //    c.type   = pColNode->node.resType.type;
    //    c.bytes  = pColNode->node.resType.bytes;
    //    c.precision  = pColNode->node.resType.precision;
    //    c.scale  = pColNode->node.resType.scale;
H
Haojun Liao 已提交
8943 8944 8945 8946 8947 8948 8949

    taosArrayPush(pList, &bi);
  }

  return pList;
}

H
Haojun Liao 已提交
8950
SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols) {
L
Liu Jicong 已提交
8951
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
8952 8953 8954 8955 8956 8957
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
8958 8959 8960
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
8961 8962

    SColMatchInfo c = {0};
L
Liu Jicong 已提交
8963
    c.colId = pColNode->colId;
H
Haojun Liao 已提交
8964
    c.targetSlotId = pNode->slotId;
L
Liu Jicong 已提交
8965
    c.output = true;
H
Haojun Liao 已提交
8966 8967 8968
    taosArrayPush(pList, &c);
  }

H
Haojun Liao 已提交
8969 8970 8971
  *numOfOutputCols = 0;

  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
L
Liu Jicong 已提交
8972 8973
  for (int32_t i = 0; i < num; ++i) {
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
H
Haojun Liao 已提交
8974
    SColMatchInfo* info = taosArrayGet(pList, pNode->slotId);
L
Liu Jicong 已提交
8975 8976 8977 8978 8979
    //    if (pNode->output) {
    (*numOfOutputCols) += 1;
    //    } else {
    //      info->output = false;
    //    }
H
Haojun Liao 已提交
8980 8981
  }

H
Haojun Liao 已提交
8982 8983 8984
  return pList;
}

L
Liu Jicong 已提交
8985 8986
int32_t doCreateTableGroup(void* metaHandle, int32_t tableType, uint64_t tableUid, STableGroupInfo* pGroupInfo,
                           uint64_t queryId, uint64_t taskId) {
8987
  int32_t code = 0;
H
Haojun Liao 已提交
8988
  if (tableType == TSDB_SUPER_TABLE) {
H
Haojun Liao 已提交
8989
    code = tsdbQuerySTableByTagCond(metaHandle, tableUid, 0, NULL, 0, 0, NULL, pGroupInfo, NULL, 0, queryId, taskId);
H
Haojun Liao 已提交
8990
  } else {  // Create one table group.
H
Haojun Liao 已提交
8991
    code = tsdbGetOneTableGroup(metaHandle, tableUid, 0, pGroupInfo);
8992 8993 8994 8995
  }

  return code;
}
H
Haojun Liao 已提交
8996

8997 8998 8999 9000 9001 9002 9003 9004 9005 9006 9007 9008 9009 9010 9011 9012 9013 9014
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 已提交
9015 9016
tsdbReaderT doCreateDataReader(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle,
                               STableGroupInfo* pTableGroupInfo, uint64_t queryId, uint64_t taskId) {
9017
  uint64_t uid = pTableScanNode->scan.uid;
L
Liu Jicong 已提交
9018 9019
  int32_t  code =
      doCreateTableGroup(pHandle->meta, pTableScanNode->scan.tableType, uid, pTableGroupInfo, queryId, taskId);
9020 9021
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
H
Haojun Liao 已提交
9022
  }
H
Haojun Liao 已提交
9023

H
Haojun Liao 已提交
9024
  if (pTableGroupInfo->numOfTables == 0) {
H
Haojun Liao 已提交
9025
    code = 0;
L
Liu Jicong 已提交
9026
    qDebug("no table qualified for query, TID:0x%" PRIx64 ", QID:0x%" PRIx64, taskId, queryId);
H
Haojun Liao 已提交
9027 9028
    goto _error;
  }
H
Haojun Liao 已提交
9029

H
Haojun Liao 已提交
9030
  return createDataReaderImpl(pTableScanNode, pTableGroupInfo, pHandle->reader, queryId, taskId);
H
Haojun Liao 已提交
9031

L
Liu Jicong 已提交
9032
_error:
H
Haojun Liao 已提交
9033 9034 9035 9036
  terrno = code;
  return NULL;
}

H
Haojun Liao 已提交
9037
int32_t createExecTaskInfoImpl(SSubplan* pPlan, SExecTaskInfo** pTaskInfo, SReadHandle* pHandle, uint64_t taskId) {
H
Haojun Liao 已提交
9038 9039
  uint64_t queryId = pPlan->id.queryId;

H
Haojun Liao 已提交
9040
  int32_t code = TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
9041
  *pTaskInfo = createExecTaskInfo(queryId, taskId);
H
Haojun Liao 已提交
9042 9043 9044 9045
  if (*pTaskInfo == NULL) {
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    goto _complete;
  }
H
Haojun Liao 已提交
9046

H
Haojun Liao 已提交
9047 9048
  STableGroupInfo group = {0};
  (*pTaskInfo)->pRoot = doCreateOperatorTreeNode(pPlan->pNode, *pTaskInfo, pHandle, queryId, taskId, &group);
D
dapan1121 已提交
9049
  if (NULL == (*pTaskInfo)->pRoot) {
L
Liu Jicong 已提交
9050
    code = terrno;
D
dapan1121 已提交
9051 9052
    goto _complete;
  }
H
Haojun Liao 已提交
9053

9054
  if ((*pTaskInfo)->pRoot == NULL) {
H
Haojun Liao 已提交
9055
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
H
Haojun Liao 已提交
9056
    goto _complete;
9057 9058
  }

H
Haojun Liao 已提交
9059 9060
  return code;

H
Haojun Liao 已提交
9061
_complete:
wafwerar's avatar
wafwerar 已提交
9062
  taosMemoryFreeClear(*pTaskInfo);
H
Haojun Liao 已提交
9063 9064 9065

  terrno = code;
  return code;
H
Haojun Liao 已提交
9066 9067
}

L
Liu Jicong 已提交
9068
int32_t cloneExprFilterInfo(SColumnFilterInfo** dst, SColumnFilterInfo* src, int32_t filterNum) {
9069 9070 9071 9072
  if (filterNum <= 0) {
    return TSDB_CODE_SUCCESS;
  }

wafwerar's avatar
wafwerar 已提交
9073
  *dst = taosMemoryCalloc(filterNum, sizeof(*src));
9074 9075 9076 9077 9078 9079 9080 9081
  if (*dst == NULL) {
    return TSDB_CODE_QRY_OUT_OF_MEMORY;
  }

  memcpy(*dst, src, sizeof(*src) * filterNum);

  for (int32_t i = 0; i < filterNum; i++) {
    if ((*dst)[i].filterstr && dst[i]->len > 0) {
L
Liu Jicong 已提交
9082
      void* pz = taosMemoryCalloc(1, (size_t)(*dst)[i].len + 1);
9083 9084 9085

      if (pz == NULL) {
        if (i == 0) {
wafwerar's avatar
wafwerar 已提交
9086
          taosMemoryFree(*dst);
9087 9088 9089 9090 9091 9092 9093
        } else {
          freeColumnFilterInfo(*dst, i);
        }

        return TSDB_CODE_QRY_OUT_OF_MEMORY;
      }

L
Liu Jicong 已提交
9094
      memcpy(pz, (void*)src->pz, (size_t)src->len + 1);
9095 9096 9097 9098 9099 9100 9101 9102

      (*dst)[i].pz = (int64_t)pz;
    }
  }

  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
9103 9104
static int32_t updateOutputBufForTopBotQuery(SQueriedTableInfo* pTableInfo, SColumnInfo* pTagCols, SExprInfo* pExprs,
                                             int32_t numOfOutput, int32_t tagLen, bool superTable) {
9105 9106 9107 9108 9109 9110 9111 9112 9113
  for (int32_t i = 0; i < numOfOutput; ++i) {
    int16_t functId = getExprFunctionId(&pExprs[i]);

    if (functId == FUNCTION_TOP || functId == FUNCTION_BOTTOM) {
      int32_t j = getColumnIndexInSource(pTableInfo, &pExprs[i].base, pTagCols);
      if (j < 0 || j >= pTableInfo->numOfCols) {
        return TSDB_CODE_QRY_INVALID_MSG;
      } else {
        SColumnInfo* pCol = &pTableInfo->colList[j];
L
Liu Jicong 已提交
9114 9115 9116 9117
        //        int32_t ret = getResultDataInfo(pCol->type, pCol->bytes, functId, (int32_t)pExprs[i].base.param[0].i,
        //                                        &pExprs[i].base.resSchema.type, &pExprs[i].base.resSchema.bytes,
        //                                        &pExprs[i].base.interBytes, tagLen, superTable, NULL);
        //        assert(ret == TSDB_CODE_SUCCESS);
9118 9119 9120 9121 9122 9123 9124 9125
      }
    }
  }

  return TSDB_CODE_SUCCESS;
}

// TODO tag length should be passed from client, refactor
L
Liu Jicong 已提交
9126
int32_t createQueryFilter(char* data, uint16_t len, SFilterInfo** pFilters) {
9127
  tExprNode* expr = NULL;
L
Liu Jicong 已提交
9128 9129 9130

  TRY(TSDB_MAX_TAG_CONDITIONS) { expr = exprTreeFromBinary(data, len); }
  CATCH(code) {
9131 9132
    CLEANUP_EXECUTE();
    return code;
L
Liu Jicong 已提交
9133 9134
  }
  END_TRY
9135 9136

  if (expr == NULL) {
L
Liu Jicong 已提交
9137
    // qError("failed to create expr tree");
9138 9139 9140
    return TSDB_CODE_QRY_APP_ERROR;
  }

L
Liu Jicong 已提交
9141 9142
  //  int32_t ret = filterInitFromTree(expr, pFilters, 0);
  //  tExprTreeDestroy(expr, NULL);
9143

L
Liu Jicong 已提交
9144
  //  return ret;
9145 9146
}

L
Liu Jicong 已提交
9147 9148 9149 9150 9151 9152
// int32_t doCreateFilterInfo(SColumnInfo* pCols, int32_t numOfCols, int32_t numOfFilterCols, SSingleColumnFilterInfo**
// pFilterInfo, uint64_t qId) {
//   *pFilterInfo = taosMemoryCalloc(1, sizeof(SSingleColumnFilterInfo) * numOfFilterCols);
//   if (*pFilterInfo == NULL) {
//     return TSDB_CODE_QRY_OUT_OF_MEMORY;
//   }
9153
//
L
Liu Jicong 已提交
9154 9155 9156
//   for (int32_t i = 0, j = 0; i < numOfCols; ++i) {
//     if (pCols[i].flist.numOfFilters > 0) {
//       SSingleColumnFilterInfo* pFilter = &((*pFilterInfo)[j]);
9157
//
L
Liu Jicong 已提交
9158 9159
//       memcpy(&pFilter->info, &pCols[i], sizeof(SColumnInfo));
//       pFilter->info = pCols[i];
9160
//
L
Liu Jicong 已提交
9161 9162 9163 9164 9165
//       pFilter->numOfFilters = pCols[i].flist.numOfFilters;
//       pFilter->pFilters = taosMemoryCalloc(pFilter->numOfFilters, sizeof(SColumnFilterElem));
//       if (pFilter->pFilters == NULL) {
//         return TSDB_CODE_QRY_OUT_OF_MEMORY;
//       }
9166
//
L
Liu Jicong 已提交
9167 9168 9169
//       for (int32_t f = 0; f < pFilter->numOfFilters; ++f) {
//         SColumnFilterElem* pSingleColFilter = &pFilter->pFilters[f];
//         pSingleColFilter->filterInfo = pCols[i].flist.filterInfo[f];
9170
//
L
Liu Jicong 已提交
9171 9172 9173 9174 9175 9176
//         int32_t lower = pSingleColFilter->filterInfo.lowerRelOptr;
//         int32_t upper = pSingleColFilter->filterInfo.upperRelOptr;
//         if (lower == TSDB_RELATION_INVALID && upper == TSDB_RELATION_INVALID) {
//           //qError("QInfo:0x%"PRIx64" invalid filter info", qId);
//           return TSDB_CODE_QRY_INVALID_MSG;
//         }
9177
//
L
Liu Jicong 已提交
9178 9179 9180 9181 9182
//         pSingleColFilter->fp = getFilterOperator(lower, upper);
//         if (pSingleColFilter->fp == NULL) {
//           //qError("QInfo:0x%"PRIx64" invalid filter info", qId);
//           return TSDB_CODE_QRY_INVALID_MSG;
//         }
9183
//
L
Liu Jicong 已提交
9184
//         pSingleColFilter->bytes = pCols[i].bytes;
9185
//
L
Liu Jicong 已提交
9186 9187 9188
//         if (lower == TSDB_RELATION_IN) {
////          buildFilterSetFromBinary(&pSingleColFilter->q, (char *)(pSingleColFilter->filterInfo.pz),
///(int32_t)(pSingleColFilter->filterInfo.len));
9189 9190 9191 9192 9193 9194 9195 9196 9197 9198 9199
//        }
//      }
//
//      j++;
//    }
//  }
//
//  return TSDB_CODE_SUCCESS;
//}

void* doDestroyFilterInfo(SSingleColumnFilterInfo* pFilterInfo, int32_t numOfFilterCols) {
L
Liu Jicong 已提交
9200 9201 9202 9203 9204 9205 9206 9207 9208 9209
  //  for (int32_t i = 0; i < numOfFilterCols; ++i) {
  //    if (pFilterInfo[i].numOfFilters > 0) {
  //      if (pFilterInfo[i].pFilters->filterInfo.lowerRelOptr == TSDB_RELATION_IN) {
  //        taosHashCleanup((SHashObj *)(pFilterInfo[i].pFilters->q));
  //      }
  //      taosMemoryFreeClear(pFilterInfo[i].pFilters);
  //    }
  //  }
  //
  //  taosMemoryFreeClear(pFilterInfo);
9210 9211 9212
  return NULL;
}

H
Haojun Liao 已提交
9213
int32_t createFilterInfo(STaskAttr* pQueryAttr, uint64_t qId) {
9214
  for (int32_t i = 0; i < pQueryAttr->numOfCols; ++i) {
L
Liu Jicong 已提交
9215 9216 9217
    //    if (pQueryAttr->tableCols[i].flist.numOfFilters > 0 && pQueryAttr->tableCols[i].flist.filterInfo != NULL) {
    //      pQueryAttr->numOfFilterCols++;
    //    }
9218 9219 9220 9221 9222 9223
  }

  if (pQueryAttr->numOfFilterCols == 0) {
    return TSDB_CODE_SUCCESS;
  }

L
Liu Jicong 已提交
9224 9225
  //  doCreateFilterInfo(pQueryAttr->tableCols, pQueryAttr->numOfCols, pQueryAttr->numOfFilterCols,
  //                     &pQueryAttr->pFilterInfo, qId);
9226 9227 9228 9229 9230 9231

  pQueryAttr->createFilterOperator = true;

  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
9232
static void doUpdateExprColumnIndex(STaskAttr* pQueryAttr) {
9233 9234 9235
  assert(pQueryAttr->pExpr1 != NULL && pQueryAttr != NULL);

  for (int32_t k = 0; k < pQueryAttr->numOfOutput; ++k) {
L
Liu Jicong 已提交
9236 9237 9238 9239
    SExprBasicInfo* pSqlExprMsg = &pQueryAttr->pExpr1[k].base;
    //    if (pSqlExprMsg->functionId == FUNCTION_ARITHM) {
    //      continue;
    //    }
9240 9241

    // todo opt performance
L
Liu Jicong 已提交
9242
    SColIndex* pColIndex = NULL; /*&pSqlExprMsg->colInfo;*/
9243 9244 9245 9246 9247 9248 9249 9250 9251 9252 9253 9254 9255 9256 9257 9258 9259 9260 9261 9262 9263 9264 9265 9266 9267 9268
    if (TSDB_COL_IS_NORMAL_COL(pColIndex->flag)) {
      int32_t f = 0;
      for (f = 0; f < pQueryAttr->numOfCols; ++f) {
        if (pColIndex->colId == pQueryAttr->tableCols[f].colId) {
          pColIndex->colIndex = f;
          break;
        }
      }

      assert(f < pQueryAttr->numOfCols);
    } else if (pColIndex->colId <= TSDB_UD_COLUMN_INDEX) {
      // do nothing for user-defined constant value result columns
    } else {
      int32_t f = 0;
      for (f = 0; f < pQueryAttr->numOfTags; ++f) {
        if (pColIndex->colId == pQueryAttr->tagColList[f].colId) {
          pColIndex->colIndex = f;
          break;
        }
      }

      assert(f < pQueryAttr->numOfTags || pColIndex->colId == TSDB_TBNAME_COLUMN_INDEX);
    }
  }
}

H
Haojun Liao 已提交
9269
void setResultBufSize(STaskAttr* pQueryAttr, SResultInfo* pResultInfo) {
9270 9271 9272 9273 9274 9275 9276 9277 9278 9279 9280 9281 9282 9283
  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;

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

L
Liu Jicong 已提交
9284
    pResultInfo->capacity = numOfRes;
9285 9286 9287 9288 9289
  } else {  // in case of non-prj query, a smaller output buffer will be used.
    pResultInfo->capacity = DEFAULT_MIN_ROWS;
  }

  pResultInfo->threshold = (int32_t)(pResultInfo->capacity * THRESHOLD_RATIO);
H
Haojun Liao 已提交
9290
  pResultInfo->totalRows = 0;
9291 9292
}

L
Liu Jicong 已提交
9293
// TODO refactor
9294
void freeColumnFilterInfo(SColumnFilterInfo* pFilter, int32_t numOfFilters) {
L
Liu Jicong 已提交
9295 9296 9297
  if (pFilter == NULL || numOfFilters == 0) {
    return;
  }
9298

L
Liu Jicong 已提交
9299 9300 9301
  for (int32_t i = 0; i < numOfFilters; i++) {
    if (pFilter[i].filterstr && pFilter[i].pz) {
      taosMemoryFree((void*)(pFilter[i].pz));
9302
    }
L
Liu Jicong 已提交
9303
  }
9304

L
Liu Jicong 已提交
9305
  taosMemoryFree(pFilter);
9306 9307 9308 9309
}

static void doDestroyTableQueryInfo(STableGroupInfo* pTableqinfoGroupInfo) {
  if (pTableqinfoGroupInfo->pGroupList != NULL) {
L
Liu Jicong 已提交
9310
    int32_t numOfGroups = (int32_t)taosArrayGetSize(pTableqinfoGroupInfo->pGroupList);
9311
    for (int32_t i = 0; i < numOfGroups; ++i) {
L
Liu Jicong 已提交
9312
      SArray* p = taosArrayGetP(pTableqinfoGroupInfo->pGroupList, i);
9313 9314

      size_t num = taosArrayGetSize(p);
L
Liu Jicong 已提交
9315
      for (int32_t j = 0; j < num; ++j) {
9316 9317 9318 9319 9320 9321 9322 9323 9324 9325 9326 9327 9328 9329 9330 9331
        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 已提交
9332
void doDestroyTask(SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
9333 9334
  qDebug("%s execTask is freed", GET_TASKID(pTaskInfo));

H
Haojun Liao 已提交
9335
  doDestroyTableQueryInfo(&pTaskInfo->tableqinfoGroupInfo);
L
Liu Jicong 已提交
9336 9337
  //  taosArrayDestroy(pTaskInfo->summary.queryProfEvents);
  //  taosHashCleanup(pTaskInfo->summary.operatorProfResults);
9338

wafwerar's avatar
wafwerar 已提交
9339 9340 9341
  taosMemoryFreeClear(pTaskInfo->sql);
  taosMemoryFreeClear(pTaskInfo->id.str);
  taosMemoryFreeClear(pTaskInfo);
9342 9343 9344 9345 9346 9347 9348 9349 9350 9351 9352 9353
}

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 已提交
9354
      int32_t len = (varDataLen(val) > maxLen) ? maxLen : varDataLen(val);
9355 9356 9357 9358 9359 9360 9361 9362 9363 9364 9365 9366
      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 已提交
9367 9368
  //  size_t s3 = sizeof(STableCheckInfo);  buffer consumption in tsdb
  return (int64_t)(s1 * 1.5 * numOfTables);
9369 9370 9371 9372 9373 9374 9375
}

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 已提交
9376
    while (1) {
9377 9378 9379 9380 9381 9382 9383 9384 9385 9386 9387 9388 9389 9390 9391 9392 9393 9394 9395 9396 9397 9398 9399 9400 9401 9402
      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);
}