executorimpl.c 269.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 17 18 19 20
#include "filter.h"
#include "functionMgt.h"
#include "function.h"
#include "querynodes.h"
#include "tname.h"
21
#include "os.h"
22

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

31 32
#include "executorimpl.h"
#include "function.h"
33
#include "query.h"
34 35
#include "tcompare.h"
#include "tcompression.h"
H
Haojun Liao 已提交
36
#include "thash.h"
H
Hongze Cheng 已提交
37
#include "vnode.h"
38
#include "ttypes.h"
L
Liu Jicong 已提交
39
#include "vnode.h"
40

H
Haojun Liao 已提交
41
#define IS_MAIN_SCAN(runtime)          ((runtime)->scanFlag == MAIN_SCAN)
42 43
#define IS_REVERSE_SCAN(runtime)       ((runtime)->scanFlag == REVERSE_SCAN)
#define IS_REPEAT_SCAN(runtime)        ((runtime)->scanFlag == REPEAT_SCAN)
L
Liu Jicong 已提交
44
#define SET_MAIN_SCAN_FLAG(runtime)    ((runtime)->scanFlag = MAIN_SCAN)
45 46 47 48
#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 已提交
49 50
#define SDATA_BLOCK_INITIALIZER \
  (SDataBlockInfo) { {0}, 0 }
51 52 53 54

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

enum {
L
Liu Jicong 已提交
55 56
  TS_JOIN_TS_EQUAL = 0,
  TS_JOIN_TS_NOT_EQUALS = 1,
57 58 59 60 61
  TS_JOIN_TAG_NOT_EQUALS = 2,
};

typedef enum SResultTsInterpType {
  RESULT_ROW_START_INTERP = 1,
L
Liu Jicong 已提交
62
  RESULT_ROW_END_INTERP = 2,
63 64
} SResultTsInterpType;

H
Haojun Liao 已提交
65

66 67
#if 0
static UNUSED_FUNC void *u_malloc (size_t __size) {
wafwerar's avatar
wafwerar 已提交
68
  uint32_t v = taosRand();
69 70 71 72

  if (v % 1000 <= 0) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
73
    return taosMemoryMalloc(__size);
74 75 76 77
  }
}

static UNUSED_FUNC void* u_calloc(size_t num, size_t __size) {
wafwerar's avatar
wafwerar 已提交
78
  uint32_t v = taosRand();
79 80 81
  if (v % 1000 <= 0) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
82
    return taosMemoryCalloc(num, __size);
83 84 85 86
  }
}

static UNUSED_FUNC void* u_realloc(void* p, size_t __size) {
wafwerar's avatar
wafwerar 已提交
87
  uint32_t v = taosRand();
88 89 90
  if (v % 5 <= 1) {
    return NULL;
  } else {
wafwerar's avatar
wafwerar 已提交
91
    return taosMemoryRealloc(p, __size);
92 93 94 95 96 97 98 99 100 101 102 103
  }
}

#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 已提交
104 105 106 107 108 109 110 111 112 113 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
#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) {
149
  assert(pExprInfo != NULL && pExprInfo->pExpr != NULL && pExprInfo->pExpr->nodeType == TEXPR_UNARYEXPR_NODE);
150
  return 0;
151 152
}

H
Haojun Liao 已提交
153 154 155 156 157
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;
158 159 160
    return;
  }

H
Haojun Liao 已提交
161
  int64_t key = tw->skey, interval = pInterval->interval;
L
Liu Jicong 已提交
162
  // convert key to second
H
Haojun Liao 已提交
163
  key = convertTimePrecision(key, precision, TSDB_TIME_PRECISION_MILLI) / 1000;
164

H
Haojun Liao 已提交
165
  if (pInterval->intervalUnit == 'y') {
166 167 168 169
    interval *= 12;
  }

  struct tm tm;
L
Liu Jicong 已提交
170
  time_t    t = (time_t)key;
wafwerar's avatar
wafwerar 已提交
171
  taosLocalTime(&t, &tm);
172 173 174 175

  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 已提交
176
  tw->skey = convertTimePrecision((int64_t)mktime(&tm) * 1000L, TSDB_TIME_PRECISION_MILLI, precision);
177 178 179 180

  mon = (int)(mon + interval);
  tm.tm_year = mon / 12;
  tm.tm_mon = mon % 12;
H
Haojun Liao 已提交
181
  tw->ekey = convertTimePrecision((int64_t)mktime(&tm) * 1000L, TSDB_TIME_PRECISION_MILLI, precision);
182 183 184 185 186

  tw->ekey -= 1;
}

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

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

L
Liu Jicong 已提交
191 192
static void destroyTableQueryInfoImpl(STableQueryInfo* pTableQueryInfo);
static bool hasMainOutput(STaskAttr* pQueryAttr);
193 194 195

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

L
Liu Jicong 已提交
196 197 198 199
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);
200 201
static STableIdInfo createTableIdInfo(STableQueryInfo* pTableQueryInfo);

H
Haojun Liao 已提交
202
static int32_t getNumOfScanTimes(STaskAttr* pQueryAttr);
203 204 205 206 207 208 209 210 211

static void destroyBasicOperatorInfo(void* param, int32_t numOfOutput);
static void destroySFillOperatorInfo(void* param, int32_t numOfOutput);
static void destroyProjectOperatorInfo(void* param, int32_t numOfOutput);
static void destroyTagScanOperatorInfo(void* param, int32_t numOfOutput);
static void destroyOrderOperatorInfo(void* param, int32_t numOfOutput);
static void 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 已提交
212

H
Haojun Liao 已提交
213
static void destroyIntervalOperatorInfo(void* param, int32_t numOfOutput);
H
Haojun Liao 已提交
214 215 216
static void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput);
static void destroyConditionOperatorInfo(void* param, int32_t numOfOutput);

217
static void destroyOperatorInfo(SOperatorInfo* pOperator);
218
static void destroySysTableScannerOperatorInfo(void* param, int32_t numOfOutput);
219

220
void doSetOperatorCompleted(SOperatorInfo* pOperator) {
221
  pOperator->status = OP_EXEC_DONE;
H
Haojun Liao 已提交
222
  if (pOperator->pTaskInfo != NULL) {
223
    setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
224 225
  }
}
226

227 228 229
#define OPTR_IS_OPENED(_optr)  (((_optr)->status & OP_OPENED) == OP_OPENED)
#define OPTR_SET_OPENED(_optr) ((_optr)->status |= OP_OPENED)

H
Haojun Liao 已提交
230
int32_t operatorDummyOpenFn(SOperatorInfo* pOperator) {
231
  OPTR_SET_OPENED(pOperator);
H
Haojun Liao 已提交
232
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
233 234
}

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

237 238
static int32_t doCopyToSDataBlock(SSDataBlock* pBlock, int32_t rowCapacity, SExprInfo* pExprInfo, SDiskbasedBuf* pBuf, SGroupResInfo* pGroupResInfo,
                                  int32_t orderType, int32_t* rowCellOffset);
L
Liu Jicong 已提交
239 240 241
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 已提交
242

H
Haojun Liao 已提交
243
static void setResultBufSize(STaskAttr* pQueryAttr, SResultInfo* pResultInfo);
H
Haojun Liao 已提交
244
static void setCtxTagForJoin(STaskRuntimeEnv* pRuntimeEnv, SqlFunctionCtx* pCtx, SExprInfo* pExprInfo, void* pTable);
L
Liu Jicong 已提交
245 246
static void doSetTableGroupOutputBuf(SAggOperatorInfo* pAggInfo, int32_t numOfOutput, int32_t tableGroupId,
                                     SExecTaskInfo* pTaskInfo);
247

H
Haojun Liao 已提交
248
SArray* getOrderCheckColumns(STaskAttr* pQuery);
249 250

typedef struct SRowCompSupporter {
L
Liu Jicong 已提交
251 252 253
  STaskRuntimeEnv* pRuntimeEnv;
  int16_t          dataOffset;
  __compar_fn_t    comFunc;
254 255
} SRowCompSupporter;

L
Liu Jicong 已提交
256 257 258
static int compareRowData(const void* a, const void* b, const void* userData) {
  const SResultRow* pRow1 = (const SResultRow*)a;
  const SResultRow* pRow2 = (const SResultRow*)b;
259

L
Liu Jicong 已提交
260 261
  SRowCompSupporter* supporter = (SRowCompSupporter*)userData;
  STaskRuntimeEnv*   pRuntimeEnv = supporter->pRuntimeEnv;
262

L
Liu Jicong 已提交
263 264
  SFilePage* page1 = getBufPage(pRuntimeEnv->pResultBuf, pRow1->pageId);
  SFilePage* page2 = getBufPage(pRuntimeEnv->pResultBuf, pRow2->pageId);
265 266

  int16_t offset = supporter->dataOffset;
L
Liu Jicong 已提交
267 268
  char*   in1 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page1, pRow1->offset, offset);
  char*   in2 = getPosInResultPage(pRuntimeEnv->pQueryAttr, page2, pRow2->offset, offset);
269 270 271 272

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

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

wafwerar's avatar
wafwerar 已提交
277
  SSDataBlock* pBlock = taosMemoryCalloc(1, sizeof(SSDataBlock));
H
Haojun Liao 已提交
278 279 280
  pBlock->info.numOfCols = numOfCols;
  pBlock->pDataBlock = taosArrayInit(numOfCols, sizeof(SColumnInfoData));

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

L
Liu Jicong 已提交
284
  for (int32_t i = 0; i < numOfCols; ++i) {
H
Haojun Liao 已提交
285
    SColumnInfoData idata = {{0}};
L
Liu Jicong 已提交
286
    SSlotDescNode*  pDescNode = nodesListGetNode(pNode->pSlots, i);
H
Haojun Liao 已提交
287 288
    if (!pDescNode->output) {
      continue;
H
Haojun Liao 已提交
289
    }
H
Haojun Liao 已提交
290

291
    idata.info.type  = pDescNode->dataType.type;
L
Liu Jicong 已提交
292 293
    idata.info.bytes = pDescNode->dataType.bytes;
    idata.info.scale = pDescNode->dataType.scale;
H
Haojun Liao 已提交
294
    idata.info.slotId = pDescNode->slotId;
H
Haojun Liao 已提交
295 296
    idata.info.precision = pDescNode->dataType.precision;

H
Hongze Cheng 已提交
297 298 299 300
    if (IS_VAR_DATA_TYPE(idata.info.type)) {
      pBlock->info.hasVarCol = true;
    }

H
Haojun Liao 已提交
301
    taosArrayPush(pBlock->pDataBlock, &idata);
H
Haojun Liao 已提交
302
  }
H
Haojun Liao 已提交
303 304

  return pBlock;
H
Haojun Liao 已提交
305 306
}

L
Liu Jicong 已提交
307
static bool isSelectivityWithTagsQuery(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
308
  return true;
L
Liu Jicong 已提交
309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329
  //  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 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) {
330 331 332 333 334 335 336 337 338 339
    return false;
  }

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

  return true;
}

H
Haojun Liao 已提交
340
static void prepareResultListBuffer(SResultRowInfo* pResultRowInfo, jmp_buf env) {
341 342
  int64_t newCapacity = 0;

343 344 345 346 347 348 349 350 351 352 353
  // 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 已提交
354 355 356 357
  if (newCapacity == pResultRowInfo->capacity) {
    newCapacity += 4;
  }

wafwerar's avatar
wafwerar 已提交
358
  pResultRowInfo->pPosition = taosMemoryRealloc(pResultRowInfo->pPosition, newCapacity * sizeof(SResultRowPosition));
359 360

  int32_t inc = (int32_t)newCapacity - pResultRowInfo->capacity;
wmmhello's avatar
wmmhello 已提交
361
  memset(&pResultRowInfo->pPosition[pResultRowInfo->capacity], 0, sizeof(SResultRowPosition) * inc);
362 363 364
  pResultRowInfo->capacity = (int32_t)newCapacity;
}

L
Liu Jicong 已提交
365 366
static bool chkResultRowFromKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, char* pData,
                                int16_t bytes, bool masterscan, uint64_t uid) {
367 368 369
  bool existed = false;
  SET_RES_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid);

L
Liu Jicong 已提交
370 371
  SResultRow** p1 =
      (SResultRow**)taosHashGet(pRuntimeEnv->pResultRowHashTable, pRuntimeEnv->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
372 373 374 375 376 377 378 379 380 381 382

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

    if (p1 != NULL) {
      if (pResultRowInfo->size == 0) {
        existed = false;
      } else if (pResultRowInfo->size == 1) {
H
Haojun Liao 已提交
383
//        existed = (pResultRowInfo->pResult[0] == (*p1));
384 385
      } else {  // check if current pResultRowInfo contains the existed pResultRow
        SET_RES_EXT_WINDOW_KEY(pRuntimeEnv->keyBuf, pData, bytes, uid, pResultRowInfo);
L
Liu Jicong 已提交
386 387
        int64_t* index =
            taosHashGet(pRuntimeEnv->pResultRowListSet, pRuntimeEnv->keyBuf, GET_RES_EXT_WINDOW_KEY_LEN(bytes));
388 389 390 391 392 393 394 395 396 397 398 399 400 401
        if (index != NULL) {
          existed = true;
        } else {
          existed = false;
        }
      }
    }

    return existed;
  }

  return p1 != NULL;
}

402
SResultRow* getNewResultRow_rv(SDiskbasedBuf* pResultBuf, int64_t tableGroupId, int32_t interBufSize) {
L
Liu Jicong 已提交
403
  SFilePage* pData = NULL;
404 405 406 407 408 409 410 411 412 413 414 415 416

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

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

wmmhello's avatar
wmmhello 已提交
417
    if (pData->num + interBufSize > getBufPageSize(pResultBuf)) {
418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436
      // release current page first, and prepare the next one
      releaseBufPageInfo(pResultBuf, pi);

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

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

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

wmmhello's avatar
wmmhello 已提交
437
  pData->num += interBufSize;
438 439 440 441

  return pResultRow;
}

L
Liu Jicong 已提交
442 443 444
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 已提交
445
  bool existInCurrentResusltRowInfo = false;
446
  SET_RES_WINDOW_KEY(pSup->keyBuf, pData, bytes, tableGroupId);
H
Haojun Liao 已提交
447

H
Haojun Liao 已提交
448
  SResultRowPosition* p1 = (SResultRowPosition*)taosHashGet(pSup->pResultRowHashTable, pSup->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes));
H
Haojun Liao 已提交
449 450 451 452

  // 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.
H
Haojun Liao 已提交
453 454 455 456 457
      if (p1 != NULL) {
        return getResultRowByPos(pResultBuf, p1);
      } else {
        return NULL;
      }
H
Haojun Liao 已提交
458 459 460 461
    }

    if (p1 != NULL) {
      if (pResultRowInfo->size == 0) {
H
Haojun Liao 已提交
462
        existInCurrentResusltRowInfo = false;  // this time window created by other timestamp that does not belongs to current table.
H
Haojun Liao 已提交
463 464
        assert(pResultRowInfo->curPos == -1);
      } else if (pResultRowInfo->size == 1) {
C
Cary Xu 已提交
465 466 467
        // ASSERT(0);
        SResultRowPosition* p = &pResultRowInfo->pPosition[0];
        existInCurrentResusltRowInfo = (p->pageId == p1->pageId && p->offset == p1->offset);
H
Haojun Liao 已提交
468
      } else {  // check if current pResultRowInfo contains the existInCurrentResusltRowInfo pResultRow
469 470
        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 已提交
471
        if (index != NULL) {
H
Haojun Liao 已提交
472 473 474
          // TODO check the scan order for current opened time window
//          pResultRowInfo->curPos = (int32_t)*index;
          existInCurrentResusltRowInfo = true;
H
Haojun Liao 已提交
475
        } else {
H
Haojun Liao 已提交
476
          existInCurrentResusltRowInfo = false;
H
Haojun Liao 已提交
477 478 479 480
        }
      }
    }
  } else {
H
Haojun Liao 已提交
481
    // In case of group by column query, the required SResultRow object must be existInCurrentResusltRowInfo in the pResultRowInfo object.
H
Haojun Liao 已提交
482
    if (p1 != NULL) {
H
Haojun Liao 已提交
483
      return getResultRowByPos(pResultBuf, p1);
H
Haojun Liao 已提交
484 485 486
    }
  }

H
Haojun Liao 已提交
487 488 489 490 491 492 493 494 495 496
  SResultRow* pResult = NULL;
  if (!existInCurrentResusltRowInfo) {
    // 1. close current opened time window
    if (pResultRowInfo->curPos != -1) {  // todo extract function
      SResultRowPosition* pos = &pResultRowInfo->pPosition[pResultRowInfo->curPos];
      SFilePage* pPage = getBufPage(pResultBuf, pos->pageId);
      SResultRow* pRow = (SResultRow*)((char*)pPage + pos->offset);
      closeResultRow(pRow);
      releaseBufPage(pResultBuf, pPage);
    }
H
Haojun Liao 已提交
497

H
Haojun Liao 已提交
498
    prepareResultListBuffer(pResultRowInfo, pTaskInfo->env);
H
Haojun Liao 已提交
499
    if (p1 == NULL) {
500
      pResult = getNewResultRow_rv(pResultBuf, tableGroupId, pSup->resultRowSize);
H
Haojun Liao 已提交
501
      initResultRow(pResult);
H
Haojun Liao 已提交
502 503

      // add a new result set for a new group
H
Haojun Liao 已提交
504
      SResultRowPosition pos = {.pageId = pResult->pageId, .offset = pResult->offset};
wmmhello's avatar
wmmhello 已提交
505
      taosHashPut(pSup->pResultRowHashTable, pSup->keyBuf, GET_RES_WINDOW_KEY_LEN(bytes), &pos, sizeof(SResultRowPosition));
H
Haojun Liao 已提交
506
      SResultRowCell cell = {.groupId = tableGroupId, .pos = pos};
507
      taosArrayPush(pSup->pResultRowArrayList, &cell);
H
Haojun Liao 已提交
508
    } else {
H
Haojun Liao 已提交
509
      pResult = getResultRowByPos(pResultBuf, p1);
H
Haojun Liao 已提交
510 511
    }

H
Haojun Liao 已提交
512
    // 2. set the new time window to be the new active time window
H
Haojun Liao 已提交
513
    pResultRowInfo->curPos = pResultRowInfo->size;
H
Haojun Liao 已提交
514
    pResultRowInfo->pPosition[pResultRowInfo->size++] = (SResultRowPosition){.pageId = pResult->pageId, .offset = pResult->offset};
H
Haojun Liao 已提交
515 516

    int64_t index = pResultRowInfo->curPos;
517 518
    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);
C
Cary Xu 已提交
519 520
  } else {
    pResult = getResultRowByPos(pResultBuf, p1);
H
Haojun Liao 已提交
521 522 523 524 525 526 527
  }

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

H
Haojun Liao 已提交
528
  return pResult;
H
Haojun Liao 已提交
529 530
}

L
Liu Jicong 已提交
531 532
static void getInitialStartTimeWindow(SInterval* pInterval, int32_t precision, TSKEY ts, STimeWindow* w, TSKEY ekey,
                                      bool ascQuery) {
H
Haojun Liao 已提交
533 534
  if (ascQuery) {
    getAlignQueryTimeWindow(pInterval, precision, ts, ts, ekey, w);
535 536
  } else {
    // the start position of the first time window in the endpoint that spreads beyond the queried last timestamp
H
Haojun Liao 已提交
537
    getAlignQueryTimeWindow(pInterval, precision, ts, ekey, ts, w);
538 539

    int64_t key = w->skey;
L
Liu Jicong 已提交
540
    while (key < ts) {  // moving towards end
541
      key = taosTimeAdd(key, pInterval->sliding, pInterval->slidingUnit, precision);
542 543 544 545 546 547 548 549 550 551
      if (key >= ts) {
        break;
      }

      w->skey = key;
    }
  }
}

// get the correct time window according to the handled timestamp
H
Haojun Liao 已提交
552
static STimeWindow getActiveTimeWindow(SDiskbasedBuf * pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
L
Liu Jicong 已提交
553
                                       int32_t precision, STimeWindow* win) {
554 555
  STimeWindow w = {0};

L
Liu Jicong 已提交
556
  if (pResultRowInfo->curPos == -1) {  // the first window, from the previous stored value
H
Haojun Liao 已提交
557
    getInitialStartTimeWindow(pInterval, precision, ts, &w, win->ekey, true);
558
    w.ekey = taosTimeAdd(w.skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
559
  } else {
H
Haojun Liao 已提交
560
    w = getResultRow(pBuf, pResultRowInfo, pResultRowInfo->curPos)->win;
561 562 563
  }

  if (w.skey > ts || w.ekey < ts) {
H
Haojun Liao 已提交
564 565 566
    if (pInterval->intervalUnit == 'n' || pInterval->intervalUnit == 'y') {
      w.skey = taosTimeTruncate(ts, pInterval, precision);
      w.ekey = taosTimeAdd(w.skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
567 568 569 570
    } else {
      int64_t st = w.skey;

      if (st > ts) {
H
Haojun Liao 已提交
571
        st -= ((st - ts + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
572 573
      }

H
Haojun Liao 已提交
574
      int64_t et = st + pInterval->interval - 1;
575
      if (et < ts) {
H
Haojun Liao 已提交
576
        st += ((ts - et + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
577 578 579
      }

      w.skey = st;
580
      w.ekey = taosTimeAdd(w.skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
581 582 583 584 585 586
    }
  }
  return w;
}

// get the correct time window according to the handled timestamp
L
Liu Jicong 已提交
587
static STimeWindow getCurrentActiveTimeWindow(SResultRowInfo* pResultRowInfo, int64_t ts, STaskAttr* pQueryAttr) {
588
  STimeWindow w = {0};
H
Haojun Liao 已提交
589
#if 0
L
Liu Jicong 已提交
590 591
  if (pResultRowInfo->curPos == -1) {  // the first window, from the previous stored value
                                       //    getInitialStartTimeWindow(pQueryAttr, ts, &w);
592 593

    if (pQueryAttr->interval.intervalUnit == 'n' || pQueryAttr->interval.intervalUnit == 'y') {
L
Liu Jicong 已提交
594 595 596
      w.ekey =
          taosTimeAdd(w.skey, pQueryAttr->interval.interval, pQueryAttr->interval.intervalUnit, pQueryAttr->precision) -
          1;
597 598 599 600
    } else {
      w.ekey = w.skey + pQueryAttr->interval.interval - 1;
    }
  } else {
H
Haojun Liao 已提交
601
    w = pRow->win;
602 603 604 605 606 607 608 609 610
  }

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

  return w;
}

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

L
Liu Jicong 已提交
622
  SFilePage* pData = NULL;
623 624 625 626 627 628

  // 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 已提交
629
    pData = getNewBufPage(pResultBuf, tid, &pageId);
630
    pData->num = sizeof(SFilePage);
631 632
  } else {
    SPageInfo* pi = getLastPageInfo(list);
633
    pData = getBufPage(pResultBuf, getPageId(pi));
634
    pageId = getPageId(pi);
635

636
    if (pData->num + size > getBufPageSize(pResultBuf)) {
637
      // release current page first, and prepare the next one
638
      releaseBufPageInfo(pResultBuf, pi);
639

H
Haojun Liao 已提交
640
      pData = getNewBufPage(pResultBuf, tid, &pageId);
641
      if (pData != NULL) {
642
        pData->num = sizeof(SFilePage);
643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662
      }
    }
  }

  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 已提交
663 664 665
static bool chkWindowOutputBufByKey(STaskRuntimeEnv* pRuntimeEnv, SResultRowInfo* pResultRowInfo, STimeWindow* win,
                                    bool masterscan, SResultRow** pResult, int64_t groupId, SqlFunctionCtx* pCtx,
                                    int32_t numOfOutput, int32_t* rowCellInfoOffset) {
666
  assert(win->skey <= win->ekey);
L
Liu Jicong 已提交
667
  return chkResultRowFromKey(pRuntimeEnv, pResultRowInfo, (char*)&win->skey, TSDB_KEYSIZE, masterscan, groupId);
668 669
}

H
Haojun Liao 已提交
670
static void setResultRowOutputBufInitCtx_rv(SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowCellInfoOffset);
H
Haojun Liao 已提交
671

L
Liu Jicong 已提交
672 673 674 675
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 已提交
676
  assert(win->skey <= win->ekey);
L
Liu Jicong 已提交
677 678
  SResultRow* pResultRow = doSetResultOutBufByKey_rv(pAggSup->pResultBuf, pResultRowInfo, id, (char*)&win->skey,
                                                     TSDB_KEYSIZE, masterscan, tableGroupId, pTaskInfo, true, pAggSup);
H
Haojun Liao 已提交
679 680 681 682 683 684 685 686 687

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

  // set time window for current result
  pResultRow->win = (*win);
  *pResult = pResultRow;
H
Haojun Liao 已提交
688
  setResultRowOutputBufInitCtx_rv(pResultRow, pCtx, numOfOutput, rowCellInfoOffset);
H
Haojun Liao 已提交
689 690 691
  return TSDB_CODE_SUCCESS;
}

692 693 694 695 696
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 已提交
697
    pResult->endInterp = true;
698 699 700 701 702 703 704 705
  }
}

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 已提交
706
    return pResult->endInterp == true;
707 708 709
  }
}

L
Liu Jicong 已提交
710 711
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) {
712 713 714
  int32_t forwardStep = 0;

  if (order == TSDB_ORDER_ASC) {
L
Liu Jicong 已提交
715
    int32_t end = searchFn((char*)&pData[pos], numOfRows - pos, ekey, order);
716 717 718 719 720 721 722 723
    if (end >= 0) {
      forwardStep = end;

      if (pData[end + pos] == ekey) {
        forwardStep += 1;
      }
    }
  } else {
L
Liu Jicong 已提交
724
    int32_t end = searchFn((char*)pData, pos + 1, ekey, order);
725 726 727 728 729 730 731 732 733 734 735 736 737
    if (end >= 0) {
      forwardStep = pos - end;

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

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

H
Haojun Liao 已提交
738
static void doUpdateResultRowIndex(SResultRowInfo* pResultRowInfo, TSKEY lastKey, bool ascQuery, bool timeWindowInterpo) {
739
  int64_t skey = TSKEY_INITIAL_VAL;
H
Haojun Liao 已提交
740
#if 0
741 742
  int32_t i = 0;
  for (i = pResultRowInfo->size - 1; i >= 0; --i) {
L
Liu Jicong 已提交
743
    SResultRow* pResult = pResultRowInfo->pResult[i];
744 745 746 747 748 749
    if (pResult->closed) {
      break;
    }

    // new closed result rows
    if (timeWindowInterpo) {
L
Liu Jicong 已提交
750 751 752
      if (pResult->endInterp &&
          ((pResult->win.skey <= lastKey && ascQuery) || (pResult->win.skey >= lastKey && !ascQuery))) {
        if (i > 0) {  // the first time window, the startInterp is false.
753 754 755 756 757 758 759 760 761 762 763 764 765 766 767 768 769 770 771
          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 已提交
772
      //      assert(pResultRowInfo->current == NULL);
773 774 775 776 777 778 779
      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 已提交
780
      SResultRow* pResult = pResultRowInfo->pResult[i];
781 782 783 784 785 786 787 788 789 790 791
      if (pResult->closed) {
        break;
      }
    }

    if (i == pResultRowInfo->size - 1) {
      pResultRowInfo->curPos = i;
    } else {
      pResultRowInfo->curPos = i + 1;  // current not closed result object
    }
  }
H
Haojun Liao 已提交
792
#endif
793 794
}

L
Liu Jicong 已提交
795 796
static void updateResultRowInfoActiveIndex(SResultRowInfo* pResultRowInfo, const STimeWindow* pWin, TSKEY lastKey,
                                           bool ascQuery, bool interp) {
H
Haojun Liao 已提交
797
  if ((lastKey > pWin->ekey && ascQuery) || (lastKey < pWin->ekey && (!ascQuery))) {
798 799 800 801
    closeAllResultRows(pResultRowInfo);
    pResultRowInfo->curPos = pResultRowInfo->size - 1;
  } else {
    int32_t step = ascQuery ? 1 : -1;
H
Haojun Liao 已提交
802
    doUpdateResultRowIndex(pResultRowInfo, lastKey - step, ascQuery, interp);
803 804 805
  }
}

L
Liu Jicong 已提交
806 807 808
static int32_t getNumOfRowsInTimeWindow(SDataBlockInfo* pDataBlockInfo, TSKEY* pPrimaryColumn, int32_t startPos,
                                        TSKEY ekey, __block_search_fn_t searchFn, STableQueryInfo* item,
                                        int32_t order) {
809 810
  assert(startPos >= 0 && startPos < pDataBlockInfo->rows);

L
Liu Jicong 已提交
811 812
  int32_t num = -1;
  int32_t step = GET_FORWARD_DIRECTION_FACTOR(order);
813

H
Haojun Liao 已提交
814
  if (order == TSDB_ORDER_ASC) {
815 816
    if (ekey < pDataBlockInfo->window.ekey && pPrimaryColumn) {
      num = getForwardStepsInBlock(pDataBlockInfo->rows, searchFn, ekey, startPos, order, pPrimaryColumn);
H
Haojun Liao 已提交
817
      if (item != NULL) {
818 819 820 821
        item->lastKey = pPrimaryColumn[startPos + (num - 1)] + step;
      }
    } else {
      num = pDataBlockInfo->rows - startPos;
H
Haojun Liao 已提交
822
      if (item != NULL) {
823 824 825 826 827 828
        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 已提交
829
      if (item != NULL) {
830 831 832 833
        item->lastKey = pPrimaryColumn[startPos - (num - 1)] + step;
      }
    } else {
      num = startPos + 1;
H
Haojun Liao 已提交
834
      if (item != NULL) {
835 836 837 838 839 840 841 842 843
        item->lastKey = pDataBlockInfo->window.skey + step;
      }
    }
  }

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

844 845 846 847 848
//  query_range_start, query_range_end, window_duration, window_start, window_end
static void initExecTimeWindowInfo(SColumnInfoData* pColData, STimeWindow* pQueryWindow) {
  pColData->info.type = TSDB_DATA_TYPE_TIMESTAMP;
  pColData->info.bytes = sizeof(int64_t);

H
Haojun Liao 已提交
849
  colInfoDataEnsureCapacity(pColData, 5);
850 851 852 853 854 855 856 857 858
  colDataAppendInt64(pColData, 0, &pQueryWindow->skey);
  colDataAppendInt64(pColData, 1, &pQueryWindow->ekey);

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

859
static void updateTimeWindowInfo(SColumnInfoData* pColData, STimeWindow* pWin, bool includeEndpoint) {
860
  int64_t* ts = (int64_t*)pColData->pData;
861
  int32_t delta = includeEndpoint? 1:0;
862

863
  int64_t duration = pWin->ekey - pWin->skey + delta;
864 865
  ts[2] = duration;       // set the duration
  ts[3] = pWin->skey;     // window start key
866
  ts[4] = pWin->ekey + delta; // window end key
867 868
}

H
Haojun Liao 已提交
869
void doApplyFunctions(SqlFunctionCtx* pCtx, STimeWindow* pWin, SColumnInfoData* pTimeWindowData, int32_t offset, int32_t forwardStep, TSKEY* tsCol,
870
                             int32_t numOfTotal, int32_t numOfOutput, int32_t order) {
871 872 873
  for (int32_t k = 0; k < numOfOutput; ++k) {
    pCtx[k].startTs = pWin->skey;

H
Haojun Liao 已提交
874
    // keep it temporarily
875 876
    bool    hasAgg      = pCtx[k].input.colDataAggIsSet;
    int32_t numOfRows   = pCtx[k].input.numOfRows;
H
Haojun Liao 已提交
877
    int32_t startOffset = pCtx[k].input.startRowIndex;
878

H
Haojun Liao 已提交
879
    int32_t pos = (order == TSDB_ORDER_ASC) ? offset : offset - (forwardStep - 1);
880 881
    pCtx[k].input.startRowIndex = pos;
    pCtx[k].input.numOfRows = forwardStep;
882 883

    if (tsCol != NULL) {
H
Haojun Liao 已提交
884
      pCtx[k].ptsList = tsCol;
885 886 887 888 889 890 891 892
    }

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

893 894 895 896
    if (fmIsWindowPseudoColumnFunc(pCtx[k].functionId)) {
      SResultRowEntryInfo* pEntryInfo = GET_RES_INFO(&pCtx[k]);
      char* p = GET_ROWCELL_INTERBUF(pEntryInfo);

897 898 899 900 901 902 903 904
      SColumnInfoData idata = {0};
      idata.info.type  = TSDB_DATA_TYPE_BIGINT;
      idata.info.bytes = tDataTypes[TSDB_DATA_TYPE_BIGINT].bytes;
      idata.pData      = p;

      SScalarParam out = {.columnData = &idata};
      SScalarParam tw = {.numOfRows = 5, .columnData = pTimeWindowData};
      pCtx[k].sfp.process(&tw, 1, &out);
905 906 907 908
      pEntryInfo->numOfRes = 1;
      continue;
    }

H
Haojun Liao 已提交
909
    if (functionNeedToExecute(&pCtx[k])) {
910
      pCtx[k].fpSet.process(&pCtx[k]);
911 912 913
    }

    // restore it
914 915 916
    pCtx[k].input.colDataAggIsSet = hasAgg;
    pCtx[k].input.startRowIndex = startOffset;
    pCtx[k].input.numOfRows = numOfRows;
917 918 919
  }
}

H
Haojun Liao 已提交
920 921 922
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 已提交
923
  bool    ascQuery = (order == TSDB_ORDER_ASC);
H
Haojun Liao 已提交
924

925
  int32_t precision = pInterval->precision;
H
Haojun Liao 已提交
926
  getNextTimeWindow(pInterval, precision, order, pNext);
927 928

  // next time window is not in current block
H
Haojun Liao 已提交
929 930
  if ((pNext->skey > pDataBlockInfo->window.ekey && order == TSDB_ORDER_ASC) ||
      (pNext->ekey < pDataBlockInfo->window.skey && order == TSDB_ORDER_DESC)) {
931 932 933
    return -1;
  }

L
Liu Jicong 已提交
934
  TSKEY   startKey = ascQuery ? pNext->skey : pNext->ekey;
935 936 937
  int32_t startPos = 0;

  // tumbling time window query, a special case of sliding time window query
H
Haojun Liao 已提交
938 939
  if (pInterval->sliding == pInterval->interval && prevPosition != -1) {
    int32_t factor = GET_FORWARD_DIRECTION_FACTOR(order);
940 941
    startPos = prevPosition + factor;
  } else {
H
Haojun Liao 已提交
942
    if (startKey <= pDataBlockInfo->window.skey && ascQuery) {
943
      startPos = 0;
H
Haojun Liao 已提交
944
    } else if (startKey >= pDataBlockInfo->window.ekey && !ascQuery) {
945 946
      startPos = pDataBlockInfo->rows - 1;
    } else {
L
Liu Jicong 已提交
947
      startPos = binarySearchForKey((char*)primaryKeys, pDataBlockInfo->rows, startKey, order);
948 949 950 951
    }
  }

  /* interp query with fill should not skip time window */
L
Liu Jicong 已提交
952 953 954
  //  if (pQueryAttr->pointInterpQuery && pQueryAttr->fillType != TSDB_FILL_NONE) {
  //    return startPos;
  //  }
955 956 957 958 959 960

  /*
   * 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 已提交
961
    if (ascQuery) {
962 963 964 965 966
      assert(pDataBlockInfo->window.skey <= pNext->ekey);
    } else {
      assert(pDataBlockInfo->window.ekey >= pNext->skey);
    }
  } else {
H
Haojun Liao 已提交
967
    if (ascQuery && primaryKeys[startPos] > pNext->ekey) {
968
      TSKEY next = primaryKeys[startPos];
H
Haojun Liao 已提交
969 970 971
      if (pInterval->intervalUnit == 'n' || pInterval->intervalUnit == 'y') {
        pNext->skey = taosTimeTruncate(next, pInterval, precision);
        pNext->ekey = taosTimeAdd(pNext->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
972
      } else {
L
Liu Jicong 已提交
973
        pNext->ekey += ((next - pNext->ekey + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
H
Haojun Liao 已提交
974
        pNext->skey = pNext->ekey - pInterval->interval + 1;
975
      }
H
Haojun Liao 已提交
976
    } else if ((!ascQuery) && primaryKeys[startPos] < pNext->skey) {
977
      TSKEY next = primaryKeys[startPos];
H
Haojun Liao 已提交
978 979 980
      if (pInterval->intervalUnit == 'n' || pInterval->intervalUnit == 'y') {
        pNext->skey = taosTimeTruncate(next, pInterval, precision);
        pNext->ekey = taosTimeAdd(pNext->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
981
      } else {
H
Haojun Liao 已提交
982 983
        pNext->skey -= ((pNext->skey - next + pInterval->sliding - 1) / pInterval->sliding) * pInterval->sliding;
        pNext->ekey = pNext->skey + pInterval->interval - 1;
984 985 986 987 988 989 990
      }
    }
  }

  return startPos;
}

L
Liu Jicong 已提交
991
static FORCE_INLINE TSKEY reviseWindowEkey(STaskAttr* pQueryAttr, STimeWindow* pWindow) {
992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007
  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 已提交
1008
static void setNotInterpoWindowKey(SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t type) {
1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019
  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 已提交
1020
static void saveDataBlockLastRow(char** pRow, SArray* pDataBlock, int32_t rowIndex, int32_t numOfCols) {
1021 1022 1023 1024
  if (pDataBlock == NULL) {
    return;
  }

H
Haojun Liao 已提交
1025
  for (int32_t k = 0; k < numOfCols; ++k) {
L
Liu Jicong 已提交
1026
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock, k);
H
Haojun Liao 已提交
1027
    memcpy(pRow[k], ((char*)pColInfo->pData) + (pColInfo->info.bytes * rowIndex), pColInfo->info.bytes);
1028 1029 1030
  }
}

H
Haojun Liao 已提交
1031
static TSKEY getStartTsKey(STimeWindow* win, const TSKEY* tsCols, int32_t rows, bool ascQuery) {
1032 1033
  TSKEY ts = TSKEY_INITIAL_VAL;
  if (tsCols == NULL) {
L
Liu Jicong 已提交
1034
    ts = ascQuery ? win->skey : win->ekey;
1035
  } else {
L
Liu Jicong 已提交
1036
    int32_t offset = ascQuery ? 0 : rows - 1;
1037 1038 1039 1040 1041 1042
    ts = tsCols[offset];
  }

  return ts;
}

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

L
Liu Jicong 已提交
1045 1046
static void doSetInputDataBlockInfo(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock,
                                    int32_t order) {
1047 1048
  for (int32_t i = 0; i < pOperator->numOfOutput; ++i) {
    pCtx[i].order = order;
L
Liu Jicong 已提交
1049
    pCtx[i].size = pBlock->info.rows;
1050 1051
    pCtx[i].currentStage = (uint8_t)pOperator->pRuntimeEnv->scanFlag;

L
Liu Jicong 已提交
1052
    setBlockStatisInfo(&pCtx[i], pBlock, NULL /*&pOperator->pExpr[i].base.colInfo*/);
1053 1054 1055
  }
}

H
Haojun Liao 已提交
1056
void setInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order) {
H
Haojun Liao 已提交
1057 1058 1059 1060 1061
  if (pBlock->pDataBlock != NULL) {
    doSetInputDataBlock(pOperator, pCtx, pBlock, order);
  } else {
    doSetInputDataBlockInfo(pOperator, pCtx, pBlock, order);
  }
1062 1063
}

H
Haojun Liao 已提交
1064
static void doSetInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order) {
1065 1066
  for (int32_t i = 0; i < pOperator->numOfOutput; ++i) {
    pCtx[i].order = order;
L
Liu Jicong 已提交
1067
    pCtx[i].size = pBlock->info.rows;
H
Haojun Liao 已提交
1068 1069
    pCtx[i].currentStage = MAIN_SCAN;

1070 1071 1072
    SExprInfo* pOneExpr = &pOperator->pExpr[i];
    for (int32_t j = 0; j < pOneExpr->base.numOfParams; ++j) {
      SFunctParam *pFuncParam = &pOneExpr->base.pParam[j];
G
Ganlin Zhao 已提交
1073 1074 1075 1076 1077 1078 1079 1080 1081
      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 已提交
1082

H
Haojun Liao 已提交
1083
    //    setBlockStatisInfo(&pCtx[i], pBlock, pOperator->pExpr[i].base.pColumns);
H
Haojun Liao 已提交
1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099
    //      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 已提交
1100 1101 1102 1103
    //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 已提交
1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131


    //        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) {
1132
  for (int32_t k = 0; k < pOperator->numOfOutput; ++k) {
H
Haojun Liao 已提交
1133
    if (functionNeedToExecute(&pCtx[k])) {
L
Liu Jicong 已提交
1134
      pCtx[k].startTs = startTs;  // this can be set during create the struct
H
Haojun Liao 已提交
1135
      pCtx[k].fpSet.process(&pCtx[k]);
1136 1137 1138 1139
    }
  }
}

H
Haojun Liao 已提交
1140
static void setPseudoOutputColInfo(SSDataBlock* pResult, SqlFunctionCtx* pCtx, SArray* pPseudoList) {
1141
  size_t num = (pPseudoList != NULL)? taosArrayGetSize(pPseudoList):0;
H
Haojun Liao 已提交
1142 1143 1144 1145 1146
  for (int32_t i = 0; i < num; ++i) {
    pCtx[i].pOutput = taosArrayGet(pResult->pDataBlock, i);
  }
}

1147
void projectApplyFunctions(SExprInfo* pExpr, SSDataBlock* pResult, SSDataBlock* pSrcBlock, SqlFunctionCtx* pCtx, int32_t numOfOutput, SArray* pPseudoList) {
H
Haojun Liao 已提交
1148
  setPseudoOutputColInfo(pResult, pCtx, pPseudoList);
H
Haojun Liao 已提交
1149
  pResult->info.groupId = pSrcBlock->info.groupId;
H
Haojun Liao 已提交
1150

1151
  for (int32_t k = 0; k < numOfOutput; ++k) {
1152 1153 1154
    int32_t outputSlotId = pExpr[k].base.resSchema.slotId;
    SqlFunctionCtx* pfCtx = &pCtx[k];

L
Liu Jicong 已提交
1155
    if (pExpr[k].pExpr->nodeType == QUERY_NODE_COLUMN) {  // it is a project query
1156 1157
      SColumnInfoData* pColInfoData = taosArrayGet(pResult->pDataBlock, outputSlotId);
      colDataAssign(pColInfoData, pfCtx->input.pData[0], pfCtx->input.numOfRows);
1158

1159
      pResult->info.rows = pSrcBlock->info.rows;
1160
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_VALUE) {
1161
      SColumnInfoData* pColInfoData = taosArrayGet(pResult->pDataBlock, outputSlotId);
1162
      for (int32_t i = 0; i < pSrcBlock->info.rows; ++i) {
1163
        colDataAppend(pColInfoData, i, taosVariantGet(&pExpr[k].base.pParam[0].param, pExpr[k].base.pParam[0].param.nType), TSDB_DATA_TYPE_NULL == pExpr[k].base.pParam[0].param.nType);
1164 1165
      }
      pResult->info.rows = pSrcBlock->info.rows;
H
Haojun Liao 已提交
1166
    } else if (pExpr[k].pExpr->nodeType == QUERY_NODE_OPERATOR) {
1167 1168 1169
      SArray* pBlockList = taosArrayInit(4, POINTER_BYTES);
      taosArrayPush(pBlockList, &pSrcBlock);

1170
      SScalarParam dest = {0};
1171
      dest.columnData = taosArrayGet(pResult->pDataBlock, outputSlotId);
1172 1173 1174 1175 1176 1177

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

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

1180
      if (fmIsPseudoColumnFunc(pfCtx->functionId)) {
H
Haojun Liao 已提交
1181
        // do nothing
1182
      } else if (fmIsNonstandardSQLFunc(pfCtx->functionId)) {
H
Haojun Liao 已提交
1183
        // todo set the correct timestamp column
1184
        pfCtx->input.pPTS = taosArrayGet(pSrcBlock->pDataBlock, 1);
1185

H
Haojun Liao 已提交
1186
        SResultRowEntryInfo *pResInfo = GET_RES_INFO(&pCtx[k]);
1187
        pfCtx->fpSet.init(&pCtx[k], pResInfo);
1188

1189 1190
        pfCtx->pOutput = taosArrayGet(pResult->pDataBlock, outputSlotId);
        pfCtx->offset  = pResult->info.rows;  // set the start offset
H
Haojun Liao 已提交
1191

1192 1193
        if (taosArrayGetSize(pPseudoList) > 0) {
          int32_t* outputColIndex = taosArrayGet(pPseudoList, 0);
1194
          pfCtx->pTsOutput = (SColumnInfoData*)pCtx[*outputColIndex].pOutput;
1195
        }
H
Haojun Liao 已提交
1196

1197
        int32_t numOfRows = pfCtx->fpSet.process(pfCtx);
H
Haojun Liao 已提交
1198
        pResult->info.rows += numOfRows;
H
Haojun Liao 已提交
1199 1200 1201
      } else {
        SArray* pBlockList = taosArrayInit(4, POINTER_BYTES);
        taosArrayPush(pBlockList, &pSrcBlock);
G
Ganlin Zhao 已提交
1202

H
Haojun Liao 已提交
1203
        SScalarParam dest = {0};
1204
        dest.columnData = taosArrayGet(pResult->pDataBlock, outputSlotId);
H
Haojun Liao 已提交
1205 1206 1207 1208 1209

        scalarCalculate((SNode*)pExpr[k].pExpr->_function.pFunctNode, pBlockList, &dest);
        pResult->info.rows = dest.numOfRows;
        taosArrayDestroy(pBlockList);
      }
1210
    } else {
1211
      ASSERT(0);
1212 1213 1214 1215 1216 1217
    }
  }
}

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 已提交
1218 1219
  STaskRuntimeEnv* pRuntimeEnv = pOperator->pRuntimeEnv;
  SExprInfo*       pExpr = pOperator->pExpr;
1220

H
Haojun Liao 已提交
1221
  SqlFunctionCtx* pCtx = pInfo->pCtx;
1222 1223 1224 1225 1226 1227 1228 1229

  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 已提交
1230
    SColIndex*       pColIndex = NULL /*&pExpr[k].base.colInfo*/;
1231
    int16_t          index = pColIndex->colIndex;
L
Liu Jicong 已提交
1232
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock, index);
1233

L
Liu Jicong 已提交
1234
    //    assert(pColInfo->info.colId == pColIndex->info.colId && curTs != windowKey);
1235 1236 1237
    double v1 = 0, v2 = 0, v = 0;

    if (prevRowIndex == -1) {
L
Liu Jicong 已提交
1238
      GET_TYPED_DATA(v1, double, pColInfo->info.type, (char*)pRuntimeEnv->prevRow[index]);
1239
    } else {
L
Liu Jicong 已提交
1240
      GET_TYPED_DATA(v1, double, pColInfo->info.type, (char*)pColInfo->pData + prevRowIndex * pColInfo->info.bytes);
1241 1242
    }

L
Liu Jicong 已提交
1243
    GET_TYPED_DATA(v2, double, pColInfo->info.type, (char*)pColInfo->pData + curRowIndex * pColInfo->info.bytes);
1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254

    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 已提交
1255
            pCtx[k].start.ptr = (char*)pRuntimeEnv->prevRow[index];
1256
          } else {
L
Liu Jicong 已提交
1257
            pCtx[k].start.ptr = (char*)pColInfo->pData + prevRowIndex * pColInfo->info.bytes;
1258 1259
          }

L
Liu Jicong 已提交
1260
          pCtx[k].end.ptr = (char*)pColInfo->pData + curRowIndex * pColInfo->info.bytes;
1261 1262 1263
        }
      }
    } else if (functionId == FUNCTION_TWA) {
L
Liu Jicong 已提交
1264 1265 1266
      SPoint point1 = (SPoint){.key = prevTs, .val = &v1};
      SPoint point2 = (SPoint){.key = curTs, .val = &v2};
      SPoint point = (SPoint){.key = windowKey, .val = &v};
1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280

      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 已提交
1281
static bool setTimeWindowInterpolationStartTs(SOperatorInfo* pOperatorInfo, SqlFunctionCtx* pCtx, int32_t pos,
H
Haojun Liao 已提交
1282 1283
                                              int32_t numOfRows, SArray* pDataBlock, const TSKEY* tsCols, STimeWindow* win) {
  bool  ascQuery = true;
L
Liu Jicong 已提交
1284
  TSKEY curTs = tsCols[pos];
H
Haojun Liao 已提交
1285
  TSKEY lastTs = 0;//*(TSKEY*)pRuntimeEnv->prevRow[0];
1286 1287 1288

  // 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 已提交
1289
  TSKEY key = ascQuery ? win->skey : win->ekey;
1290 1291 1292 1293 1294 1295 1296 1297 1298 1299
  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;
  }

H
Haojun Liao 已提交
1300
  int32_t step = 1;//GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
L
Liu Jicong 已提交
1301
  TSKEY   prevTs = ((pos == 0 && ascQuery) || (pos == (numOfRows - 1) && !ascQuery)) ? lastTs : tsCols[pos - step];
1302

H
Haojun Liao 已提交
1303
  doTimeWindowInterpolation(pOperatorInfo, pOperatorInfo->info, pDataBlock, prevTs, pos - step, curTs, pos, key, RESULT_ROW_START_INTERP);
1304 1305 1306
  return true;
}

L
Liu Jicong 已提交
1307 1308 1309
static bool setTimeWindowInterpolationEndTs(SOperatorInfo* pOperatorInfo, SqlFunctionCtx* pCtx, int32_t endRowIndex,
                                            SArray* pDataBlock, const TSKEY* tsCols, TSKEY blockEkey,
                                            STimeWindow* win) {
H
Haojun Liao 已提交
1310 1311
  int32_t order = TSDB_ORDER_ASC;
  int32_t numOfOutput = pOperatorInfo->numOfOutput;
1312

L
Liu Jicong 已提交
1313
  TSKEY actualEndKey = tsCols[endRowIndex];
H
Haojun Liao 已提交
1314
  TSKEY key = order ? win->ekey : win->skey;
1315 1316

  // not ended in current data block, do not invoke interpolation
H
Haojun Liao 已提交
1317
  if ((key > blockEkey /*&& QUERY_IS_ASC_QUERY(pQueryAttr)*/) || (key < blockEkey /*&& !QUERY_IS_ASC_QUERY(pQueryAttr)*/)) {
1318 1319 1320 1321 1322 1323 1324 1325 1326 1327
    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;
  }

H
Haojun Liao 已提交
1328
  int32_t step = GET_FORWARD_DIRECTION_FACTOR(order);
1329 1330 1331 1332 1333
  int32_t nextRowIndex = endRowIndex + step;
  assert(nextRowIndex >= 0);

  TSKEY nextKey = tsCols[nextRowIndex];
  doTimeWindowInterpolation(pOperatorInfo, pOperatorInfo->info, pDataBlock, actualEndKey, endRowIndex, nextKey,
L
Liu Jicong 已提交
1334
                            nextRowIndex, key, RESULT_ROW_END_INTERP);
1335 1336 1337
  return true;
}

H
Haojun Liao 已提交
1338
static void doWindowBorderInterpolation(SOperatorInfo* pOperatorInfo, SSDataBlock* pBlock, SqlFunctionCtx* pCtx,
L
Liu Jicong 已提交
1339 1340
                                        SResultRow* pResult, STimeWindow* win, int32_t startPos, int32_t forwardStep,
                                        int32_t order, bool timeWindowInterpo) {
H
Haojun Liao 已提交
1341
  if (!timeWindowInterpo) {
1342 1343 1344 1345
    return;
  }

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

L
Liu Jicong 已提交
1348 1349
  if (pBlock->pDataBlock == NULL) {
    //    tscError("pBlock->pDataBlock == NULL");
1350 1351
    return;
  }
H
Haojun Liao 已提交
1352

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

L
Liu Jicong 已提交
1355 1356 1357
  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
1358
    int32_t startRowIndex = startPos;
L
Liu Jicong 已提交
1359 1360
    bool    interp = setTimeWindowInterpolationStartTs(pOperatorInfo, pCtx, startRowIndex, pBlock->info.rows,
                                                       pBlock->pDataBlock, tsCols, win);
1361 1362 1363 1364
    if (interp) {
      setResultRowInterpo(pResult, RESULT_ROW_START_INTERP);
    }
  } else {
H
Haojun Liao 已提交
1365
    setNotInterpoWindowKey(pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_START_INTERP);
1366 1367 1368
  }

  // point interpolation does not require the end key time window interpolation.
L
Liu Jicong 已提交
1369 1370 1371
  //  if (pointInterpQuery) {
  //    return;
  //  }
1372 1373 1374 1375 1376 1377

  // 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 已提交
1378 1379 1380
    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);
1381 1382 1383 1384
    if (interp) {
      setResultRowInterpo(pResult, RESULT_ROW_END_INTERP);
    }
  } else {
H
Haojun Liao 已提交
1385
    setNotInterpoWindowKey(pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_END_INTERP);
1386 1387 1388
  }
}

1389
static SArray* hashIntervalAgg(SOperatorInfo* pOperatorInfo, SResultRowInfo* pResultRowInfo, SSDataBlock* pSDataBlock, int32_t tableGroupId) {
L
Liu Jicong 已提交
1390
  STableIntervalOperatorInfo* pInfo = (STableIntervalOperatorInfo*)pOperatorInfo->info;
1391

H
Haojun Liao 已提交
1392
  SExecTaskInfo* pTaskInfo = pOperatorInfo->pTaskInfo;
L
Liu Jicong 已提交
1393
  int32_t        numOfOutput = pOperatorInfo->numOfOutput;
1394

1395 1396 1397 1398 1399
  SArray* pUpdated = NULL;
  if (pInfo->execModel == OPTR_EXEC_MODEL_STREAM) {
    pUpdated = taosArrayInit(4, sizeof(SResultRowPosition));
  }

H
Haojun Liao 已提交
1400
  int32_t step = 1;
1401
  bool    ascScan = true;
1402 1403 1404 1405 1406

  int32_t prevIndex = pResultRowInfo->curPos;

  TSKEY* tsCols = NULL;
  if (pSDataBlock->pDataBlock != NULL) {
1407
    SColumnInfoData* pColDataInfo = taosArrayGet(pSDataBlock->pDataBlock, pInfo->primaryTsIndex);
L
Liu Jicong 已提交
1408
    tsCols = (int64_t*)pColDataInfo->pData;
H
Haojun Liao 已提交
1409 1410
//    assert(tsCols[0] == pSDataBlock->info.window.skey && tsCols[pSDataBlock->info.rows - 1] ==
//           pSDataBlock->info.window.ekey);
1411 1412
  }

1413 1414
  int32_t startPos = ascScan? 0 : (pSDataBlock->info.rows - 1);
  TSKEY   ts = getStartTsKey(&pSDataBlock->info.window, tsCols, pSDataBlock->info.rows, ascScan);
1415

H
Haojun Liao 已提交
1416
  STimeWindow win = getActiveTimeWindow(pInfo->aggSup.pResultBuf, pResultRowInfo, ts, &pInfo->interval, pInfo->interval.precision, &pInfo->win);
L
Liu Jicong 已提交
1417
  bool        masterScan = true;
1418 1419

  SResultRow* pResult = NULL;
L
Liu Jicong 已提交
1420 1421 1422
  int32_t     ret = setResultOutputBufByKey_rv(pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult,
                                               tableGroupId, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset,
                                               &pInfo->aggSup, pTaskInfo);
1423
  if (ret != TSDB_CODE_SUCCESS || pResult == NULL) {
H
Haojun Liao 已提交
1424
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
1425 1426
  }

1427 1428 1429 1430 1431
  if (pInfo->execModel == OPTR_EXEC_MODEL_STREAM) {
    SResultRowPosition pos = {.pageId = pResult->pageId, .offset = pResult->offset};
    taosArrayPush(pUpdated, &pos);
  }

1432
  int32_t forwardStep = 0;
H
Haojun Liao 已提交
1433
  TSKEY   ekey = win.ekey;
1434
  forwardStep =
H
Haojun Liao 已提交
1435
      getNumOfRowsInTimeWindow(&pSDataBlock->info, tsCols, startPos, ekey, binarySearchForKey, NULL, TSDB_ORDER_ASC);
1436 1437 1438

  // prev time window not interpolation yet.
  int32_t curIndex = pResultRowInfo->curPos;
H
Haojun Liao 已提交
1439 1440

#if 0
H
Haojun Liao 已提交
1441
  if (prevIndex != -1 && prevIndex < curIndex && pInfo->timeWindowInterpo) {
1442 1443 1444
    for (int32_t j = prevIndex; j < curIndex; ++j) {  // previous time window may be all closed already.
      SResultRow* pRes = getResultRow(pResultRowInfo, j);
      if (pRes->closed) {
1445
        assert(resultRowInterpolated(pRes, RESULT_ROW_START_INTERP) && resultRowInterpolated(pRes, RESULT_ROW_END_INTERP));
1446 1447 1448
        continue;
      }

L
Liu Jicong 已提交
1449 1450 1451 1452 1453 1454 1455
      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);
      }
1456

L
Liu Jicong 已提交
1457 1458 1459
      assert(!resultRowInterpolated(pResult, RESULT_ROW_END_INTERP));
      doTimeWindowInterpolation(pOperatorInfo, &pInfo->binfo, pSDataBlock->pDataBlock, *(TSKEY*)pInfo->pRow[0], -1,
                                tsCols[startPos], startPos, w.ekey, RESULT_ROW_END_INTERP);
1460

L
Liu Jicong 已提交
1461 1462
      setResultRowInterpo(pResult, RESULT_ROW_END_INTERP);
      setNotInterpoWindowKey(pInfo->binfo.pCtx, pOperatorInfo->numOfOutput, RESULT_ROW_START_INTERP);
1463

1464
      doApplyFunctions(pInfo->binfo.pCtx, &w, &pInfo->timeWindowData, startPos, 0, tsCols, pSDataBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);
L
Liu Jicong 已提交
1465
    }
1466 1467

    // restore current time window
L
Liu Jicong 已提交
1468 1469 1470
    ret = setResultOutputBufByKey_rv(pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult, tableGroupId,
                                     pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup,
                                     pTaskInfo);
1471
    if (ret != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
1472
      longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
1473 1474
    }
  }
H
Haojun Liao 已提交
1475
#endif
1476 1477

  // window start key interpolation
1478 1479 1480
  doWindowBorderInterpolation(pOperatorInfo, pSDataBlock, pInfo->binfo.pCtx, pResult, &win, startPos, forwardStep, pInfo->order, false);

  updateTimeWindowInfo(&pInfo->timeWindowData, &win, true);
1481
  doApplyFunctions(pInfo->binfo.pCtx, &win, &pInfo->timeWindowData, startPos, forwardStep, tsCols, pSDataBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);
1482 1483 1484 1485

  STimeWindow nextWin = win;
  while (1) {
    int32_t prevEndPos = (forwardStep - 1) * step + startPos;
H
Haojun Liao 已提交
1486
    startPos = getNextQualifiedWindow(&pInfo->interval, &nextWin, &pSDataBlock->info, tsCols, prevEndPos, pInfo);
1487 1488 1489 1490 1491
    if (startPos < 0) {
      break;
    }

    // null data, failed to allocate more memory buffer
L
Liu Jicong 已提交
1492 1493 1494
    int32_t code = setResultOutputBufByKey_rv(pResultRowInfo, pSDataBlock->info.uid, &nextWin, masterScan, &pResult,
                                              tableGroupId, pInfo->binfo.pCtx, numOfOutput,
                                              pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup, pTaskInfo);
1495
    if (code != TSDB_CODE_SUCCESS || pResult == NULL) {
H
Haojun Liao 已提交
1496
      longjmp(pTaskInfo->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
1497 1498
    }

1499 1500 1501 1502 1503
    if (pInfo->execModel == OPTR_EXEC_MODEL_STREAM) {
      SResultRowPosition pos = {.pageId = pResult->pageId, .offset = pResult->offset};
      taosArrayPush(pUpdated, &pos);
    }

L
Liu Jicong 已提交
1504 1505 1506
    ekey = nextWin.ekey;  // reviseWindowEkey(pQueryAttr, &nextWin);
    forwardStep =
        getNumOfRowsInTimeWindow(&pSDataBlock->info, tsCols, startPos, ekey, binarySearchForKey, NULL, TSDB_ORDER_ASC);
1507 1508

    // window start(end) key interpolation
C
Cary Xu 已提交
1509 1510
    doWindowBorderInterpolation(pOperatorInfo, pSDataBlock, pInfo->binfo.pCtx, pResult, &nextWin, startPos, forwardStep,
                                pInfo->order, false);
1511

H
Haojun Liao 已提交
1512
    updateTimeWindowInfo(&pInfo->timeWindowData, &nextWin, true);
1513
    doApplyFunctions(pInfo->binfo.pCtx, &nextWin, &pInfo->timeWindowData, startPos, forwardStep, tsCols, pSDataBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);
1514 1515
  }

H
Haojun Liao 已提交
1516
  if (pInfo->timeWindowInterpo) {
1517
    int32_t rowIndex = ascScan ? (pSDataBlock->info.rows - 1) : 0;
H
Haojun Liao 已提交
1518
    saveDataBlockLastRow(pInfo->pRow, pSDataBlock->pDataBlock, rowIndex, pSDataBlock->info.numOfCols);
1519 1520
  }

1521
  return pUpdated;
L
Liu Jicong 已提交
1522
  //  updateResultRowInfoActiveIndex(pResultRowInfo, &pInfo->win, pRuntimeEnv->current->lastKey, true, false);
1523 1524
}

L
Liu Jicong 已提交
1525 1526 1527
static void hashAllIntervalAgg(SOperatorInfo* pOperatorInfo, SResultRowInfo* pResultRowInfo, SSDataBlock* pSDataBlock,
                               int32_t tableGroupId) {
  STableIntervalOperatorInfo* pInfo = (STableIntervalOperatorInfo*)pOperatorInfo->info;
1528

H
Haojun Liao 已提交
1529
  STaskRuntimeEnv* pRuntimeEnv = pOperatorInfo->pRuntimeEnv;
L
Liu Jicong 已提交
1530
  int32_t          numOfOutput = pOperatorInfo->numOfOutput;
1531

H
Haojun Liao 已提交
1532 1533
  int32_t step = 1;//GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
  bool    ascQuery = true;
1534 1535 1536 1537

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

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

H
Haojun Liao 已提交
1546
  STimeWindow win = {0};//getCurrentActiveTimeWindow(pResultRowInfo, ts, pQueryAttr);
L
Liu Jicong 已提交
1547
  bool        masterScan = IS_MAIN_SCAN(pRuntimeEnv);
1548 1549

  SResultRow* pResult = NULL;
L
Liu Jicong 已提交
1550 1551
  int32_t     forwardStep = 0;
  int32_t     ret = 0;
1552 1553 1554 1555
  STimeWindow preWin = win;

  while (1) {
    // null data, failed to allocate more memory buffer
1556 1557
//    ret = setResultOutputBufByKey(pRuntimeEnv, pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult,
//                                  tableGroupId, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset);
1558 1559 1560 1561
    if (ret != TSDB_CODE_SUCCESS) {
      longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
    }

H
Haojun Liao 已提交
1562
    TSKEY ekey = 0;//reviseWindowEkey(pQueryAttr, &win);
L
Liu Jicong 已提交
1563 1564
    //    forwardStep = getNumOfRowsInTimeWindow(pRuntimeEnv, &pSDataBlock->info, tsCols, startPos, ekey,
    //    binarySearchForKey, true);
1565 1566

    // window start(end) key interpolation
L
Liu Jicong 已提交
1567 1568 1569
    //    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);
1570 1571 1572
    preWin = win;

    int32_t prevEndPos = (forwardStep - 1) * step + startPos;
L
Liu Jicong 已提交
1573 1574
    //    startPos = getNextQualifiedWindow(pQueryAttr, &win, &pSDataBlock->info, tsCols, binarySearchForKey,
    //    prevEndPos);
1575
    if (startPos < 0) {
H
Haojun Liao 已提交
1576 1577 1578 1579 1580 1581 1582 1583 1584
//      if ((ascQuery && win.skey <= pQueryAttr->window.ekey) || ((!ascQuery) && win.ekey >= pQueryAttr->window.ekey)) {
//        int32_t code =
//            setResultOutputBufByKey(pRuntimeEnv, pResultRowInfo, pSDataBlock->info.uid, &win, masterScan, &pResult,
//                                    tableGroupId, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset);
//        if (code != TSDB_CODE_SUCCESS || pResult == NULL) {
//          longjmp(pRuntimeEnv->env, TSDB_CODE_QRY_OUT_OF_MEMORY);
//        }
//
//        startPos = pSDataBlock->info.rows - 1;
1585 1586

        // window start(end) key interpolation
L
Liu Jicong 已提交
1587 1588 1589
        //        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);
H
Haojun Liao 已提交
1590
//      }
1591 1592 1593 1594 1595 1596

      break;
    }
    setResultRowInterpo(pResult, RESULT_ROW_END_INTERP);
  }

H
Haojun Liao 已提交
1597 1598
//  if (pQueryAttr->timeWindowInterpo) {
//    int32_t rowIndex = ascQuery ? (pSDataBlock->info.rows - 1) : 0;
L
Liu Jicong 已提交
1599
    //    saveDataBlockLastRow(pRuntimeEnv, &pSDataBlock->info, pSDataBlock->pDataBlock, rowIndex);
H
Haojun Liao 已提交
1600
//  }
1601

L
Liu Jicong 已提交
1602
  //  updateResultRowInfoActiveIndex(pResultRowInfo, pQueryAttr, pRuntimeEnv->current->lastKey);
1603 1604
}

1605 1606 1607 1608
static void doKeepTuple(SWindowRowsSup* pRowSup, int64_t ts) {
  pRowSup->win.ekey  = ts;
  pRowSup->prevTs    = ts;
  pRowSup->numOfRows += 1;
1609 1610
}

1611 1612 1613 1614
static void doKeepNewWindowStartInfo(SWindowRowsSup* pRowSup, const int64_t* tsList, int32_t rowIndex) {
  pRowSup->startRowIndex = rowIndex;
  pRowSup->numOfRows     = 0;
  pRowSup->win.skey      = tsList[rowIndex];
1615 1616
}

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

1621
  // todo find the correct time stamp column slot
1622
  SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, 0);
1623

1624 1625 1626
  bool    masterScan = true;
  int32_t numOfOutput = pOperator->numOfOutput;
  int64_t gid = pBlock->info.groupId;
1627

H
Haojun Liao 已提交
1628
  int64_t gap = pInfo->gap;
1629

1630
  if (!pInfo->reptScan) {
1631
    pInfo->reptScan = true;
1632
    pInfo->winSup.prevTs = INT64_MIN;
1633 1634
  }

1635 1636 1637
  SWindowRowsSup* pRowSup = &pInfo->winSup;
  pRowSup->numOfRows = 0;

1638
  // In case of ascending or descending order scan data, only one time window needs to be kepted for each table.
1639
  TSKEY* tsList = (TSKEY*)pColInfoData->pData;
1640
  for (int32_t j = 0; j < pBlock->info.rows; ++j) {
1641 1642 1643 1644
    if (pInfo->winSup.prevTs == INT64_MIN) {
      doKeepNewWindowStartInfo(pRowSup, tsList, j);
      doKeepTuple(pRowSup, tsList[j]);
    } else if (tsList[j] - pRowSup->prevTs <= gap && (tsList[j] - pRowSup->prevTs) >= 0) {
1645
      // The gap is less than the threshold, so it belongs to current session window that has been opened already.
1646 1647 1648
      doKeepTuple(pRowSup, tsList[j]);
      if (j == 0 && pRowSup->startRowIndex != 0) {
        pRowSup->startRowIndex = 0;
1649 1650 1651
      }
    } else {  // start a new session window
      SResultRow* pResult = NULL;
1652 1653

      // keep the time window for the closed time window.
1654
      STimeWindow window = pRowSup->win;
1655

1656
      pRowSup->win.ekey = pRowSup->win.skey;
L
Liu Jicong 已提交
1657
      int32_t ret = setResultOutputBufByKey_rv(&pInfo->binfo.resultRowInfo, pBlock->info.uid, &window, masterScan,
1658
                                               &pResult, gid, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup, pTaskInfo);
1659
      if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
1660
        longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
1661 1662
      }

1663
      // pInfo->numOfRows data belong to the current session window
1664
      updateTimeWindowInfo(&pInfo->timeWindowData, &window, false);
1665
      doApplyFunctions(pInfo->binfo.pCtx, &window, &pInfo->timeWindowData, pRowSup->startRowIndex, pRowSup->numOfRows, NULL, pBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);
1666

1667
      // here we start a new session window
1668 1669
      doKeepNewWindowStartInfo(pRowSup, tsList, j);
      doKeepTuple(pRowSup, tsList[j]);
1670 1671 1672 1673
    }
  }

  SResultRow* pResult = NULL;
1674 1675
  pRowSup->win.ekey = tsList[pBlock->info.rows - 1];
  int32_t ret = setResultOutputBufByKey_rv(&pInfo->binfo.resultRowInfo, pBlock->info.uid, &pRowSup->win, masterScan, &pResult,
1676
                                           gid, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup, pTaskInfo);
1677
  if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
1678
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
1679 1680
  }

1681 1682
  updateTimeWindowInfo(&pInfo->timeWindowData, &pRowSup->win, false);
  doApplyFunctions(pInfo->binfo.pCtx, &pRowSup->win, &pInfo->timeWindowData, pRowSup->startRowIndex, pRowSup->numOfRows, NULL, pBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);
1683 1684 1685 1686
}

static void setResultRowKey(SResultRow* pResultRow, char* pData, int16_t type) {
  if (IS_VAR_DATA_TYPE(type)) {
1687 1688 1689 1690 1691 1692 1693
    // todo disable this
//    if (pResultRow->key == NULL) {
//      pResultRow->key = taosMemoryMalloc(varDataTLen(pData));
//      varDataCopy(pResultRow->key, pData);
//    } else {
//      ASSERT(memcmp(pResultRow->key, pData, varDataTLen(pData)) == 0);
//    }
1694 1695 1696 1697 1698 1699 1700 1701 1702
  } else {
    int64_t v = -1;
    GET_TYPED_DATA(v, int64_t, type, pData);

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

H
Haojun Liao 已提交
1703
int32_t setGroupResultOutputBuf_rv(SOptrBasicInfo* binfo, int32_t numOfCols, char* pData, int16_t type,
L
Liu Jicong 已提交
1704 1705 1706 1707
                                          int16_t bytes, int32_t groupId, SDiskbasedBuf* pBuf, SExecTaskInfo* pTaskInfo,
                                          SAggSupporter* pAggSup) {
  SResultRowInfo* pResultRowInfo = &binfo->resultRowInfo;
  SqlFunctionCtx* pCtx = binfo->pCtx;
1708

L
Liu Jicong 已提交
1709
  SResultRow* pResultRow = doSetResultOutBufByKey_rv(pBuf, pResultRowInfo, groupId, (char*)pData, bytes, true, groupId,
H
Haojun Liao 已提交
1710
                                                     pTaskInfo, false, pAggSup);
L
Liu Jicong 已提交
1711
  assert(pResultRow != NULL);
1712 1713

  setResultRowKey(pResultRow, pData, type);
H
Haojun Liao 已提交
1714
  setResultRowOutputBufInitCtx_rv(pResultRow, pCtx, numOfCols, binfo->rowCellInfoOffset);
1715 1716 1717
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
1718 1719
static bool functionNeedToExecute(SqlFunctionCtx* pCtx) {
  struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(pCtx);
1720 1721 1722

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

1727
  if (isRowEntryCompleted(pResInfo)) {
1728 1729 1730 1731
    return false;
  }

  if (functionId == FUNCTION_FIRST_DST || functionId == FUNCTION_FIRST) {
L
Liu Jicong 已提交
1732
    //    return QUERY_IS_ASC_QUERY(pQueryAttr);
1733 1734 1735 1736
  }

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

  // in the reverse table scan, only the following functions need to be executed
L
Liu Jicong 已提交
1741 1742 1743 1744
  //  if (IS_REVERSE_SCAN(pRuntimeEnv) ||
  //      (pRuntimeEnv->scanFlag == REPEAT_SCAN && functionId != FUNCTION_STDDEV && functionId != FUNCTION_PERCT)) {
  //    return false;
  //  }
1745 1746 1747 1748

  return true;
}

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

H
Haojun Liao 已提交
1752 1753
  if (pSDataBlock->pBlockAgg != NULL && TSDB_COL_IS_NORMAL_COL(pColumn->flag)) {
    pAgg = &pSDataBlock->pBlockAgg[pCtx->columnIndex];
1754 1755

    pCtx->agg = *pAgg;
L
Liu Jicong 已提交
1756
    pCtx->isAggSet = true;
1757 1758 1759 1760 1761
    assert(pCtx->agg.numOfNull <= pSDataBlock->info.rows);
  } else {
    pCtx->isAggSet = false;
  }

H
Haojun Liao 已提交
1762
  pCtx->hasNull = hasNull(pColumn, pAgg);
1763 1764

  // set the statistics data for primary time stamp column
H
Haojun Liao 已提交
1765
  if (pCtx->functionId == FUNCTION_SPREAD && pColumn->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
L
Liu Jicong 已提交
1766
    pCtx->isAggSet = true;
1767 1768 1769 1770 1771 1772
    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 已提交
1773
static int32_t setCtxTagColumnInfo(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
1774 1775 1776 1777 1778 1779 1780
  if (!isSelectivityWithTagsQuery(pCtx, numOfOutput)) {
    return TSDB_CODE_SUCCESS;
  }

  int32_t num = 0;
  int16_t tagLen = 0;

H
Haojun Liao 已提交
1781
  SqlFunctionCtx*  p = NULL;
wafwerar's avatar
wafwerar 已提交
1782
  SqlFunctionCtx** pTagCtx = taosMemoryCalloc(numOfOutput, POINTER_BYTES);
1783 1784 1785 1786 1787 1788 1789 1790
  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 已提交
1791
      tagLen += pCtx[i].resDataInfo.bytes;
1792
      pTagCtx[num++] = &pCtx[i];
L
Liu Jicong 已提交
1793
    } else if (1 /*(aAggs[functionId].status & FUNCSTATE_SELECTIVITY) != 0*/) {
1794 1795 1796 1797 1798 1799 1800 1801 1802 1803
      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) {
1804 1805 1806
    p->subsidiaryRes.pCtx = pTagCtx;
    p->subsidiaryRes.numOfCols = num;
    p->subsidiaryRes.bufLen = tagLen;
1807
  } else {
wafwerar's avatar
wafwerar 已提交
1808
    taosMemoryFreeClear(pTagCtx);
1809 1810 1811 1812 1813
  }

  return TSDB_CODE_SUCCESS;
}

1814
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowCellInfoOffset) {
L
Liu Jicong 已提交
1815
  SqlFunctionCtx* pFuncCtx = (SqlFunctionCtx*)taosMemoryCalloc(numOfOutput, sizeof(SqlFunctionCtx));
H
Haojun Liao 已提交
1816 1817 1818 1819
  if (pFuncCtx == NULL) {
    return NULL;
  }

wafwerar's avatar
wafwerar 已提交
1820
  *rowCellInfoOffset = taosMemoryCalloc(numOfOutput, sizeof(int32_t));
H
Haojun Liao 已提交
1821
  if (*rowCellInfoOffset == 0) {
wafwerar's avatar
wafwerar 已提交
1822
    taosMemoryFreeClear(pFuncCtx);
H
Haojun Liao 已提交
1823 1824 1825 1826
    return NULL;
  }

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

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

1832
    pCtx->functionId = -1;
H
Haojun Liao 已提交
1833
    if (pExpr->pExpr->nodeType == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
1834
      SFuncExecEnv env = {0};
H
Haojun Liao 已提交
1835 1836
      pCtx->functionId = pExpr->pExpr->_function.pFunctNode->funcId;

H
Haojun Liao 已提交
1837
      if (fmIsAggFunc(pCtx->functionId) || fmIsNonstandardSQLFunc(pCtx->functionId)) {
1838 1839 1840 1841
        fmGetFuncExecFuncs(pCtx->functionId, &pCtx->fpSet);
        pCtx->fpSet.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
      } else {
        fmGetScalarFuncExecFuncs(pCtx->functionId, &pCtx->sfp);
1842 1843 1844
        if (pCtx->sfp.getEnv != NULL) {
          pCtx->sfp.getEnv(pExpr->pExpr->_function.pFunctNode, &env);
        }
1845
      }
H
Haojun Liao 已提交
1846
      pCtx->resDataInfo.interBufSize = env.calcMemSize;
D
dapan1121 已提交
1847
    } else if (pExpr->pExpr->nodeType == QUERY_NODE_COLUMN || pExpr->pExpr->nodeType == QUERY_NODE_OPERATOR || pExpr->pExpr->nodeType == QUERY_NODE_VALUE) {
H
Haojun Liao 已提交
1848
      pCtx->resDataInfo.interBufSize = pFunct->resSchema.bytes; // for simple column, the intermediate buffer needs to hold one element.
H
Haojun Liao 已提交
1849
    }
H
Haojun Liao 已提交
1850

H
Haojun Liao 已提交
1851
    pCtx->input.numOfInputCols = pFunct->numOfParams;
wafwerar's avatar
wafwerar 已提交
1852 1853
    pCtx->input.pData = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
    pCtx->input.pColumnDataAgg = taosMemoryCalloc(pFunct->numOfParams, POINTER_BYTES);
H
Haojun Liao 已提交
1854

1855
    pCtx->pTsOutput         = NULL;
L
Liu Jicong 已提交
1856
    pCtx->resDataInfo.bytes = pFunct->resSchema.bytes;
H
Haojun Liao 已提交
1857 1858
    pCtx->resDataInfo.type  = pFunct->resSchema.type;
    pCtx->order             = TSDB_ORDER_ASC;
1859 1860
    pCtx->start.key         = INT64_MIN;
    pCtx->end.key           = INT64_MIN;
H
Haojun Liao 已提交
1861
#if 0
H
Haojun Liao 已提交
1862
    for (int32_t j = 0; j < pCtx->numOfParams; ++j) {
H
Haojun Liao 已提交
1863 1864
//      int16_t type = pFunct->param[j].nType;
//      int16_t bytes = pFunct->param[j].nLen;
H
Haojun Liao 已提交
1865

H
Haojun Liao 已提交
1866 1867 1868 1869 1870
//      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 已提交
1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906
    }

    // 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 已提交
1907
#endif
H
Haojun Liao 已提交
1908 1909
  }

L
Liu Jicong 已提交
1910 1911 1912
  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 已提交
1913
  }
H
Haojun Liao 已提交
1914 1915 1916 1917 1918

  setCtxTagColumnInfo(pFuncCtx, numOfOutput);
  return pFuncCtx;
}

1919
static void* destroySqlFunctionCtx(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
1920 1921 1922 1923 1924 1925 1926 1927 1928 1929
  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 已提交
1930
    taosMemoryFreeClear(pCtx[i].subsidiaryRes.pCtx);
1931 1932
  }

wafwerar's avatar
wafwerar 已提交
1933
  taosMemoryFreeClear(pCtx);
1934 1935 1936
  return NULL;
}

L
Liu Jicong 已提交
1937
bool isTaskKilled(SExecTaskInfo* pTaskInfo) {
1938 1939
  // query has been executed more than tsShellActivityTimer, and the retrieve has not arrived
  // abort current query execution.
L
Liu Jicong 已提交
1940 1941
  if (pTaskInfo->owner != 0 &&
      ((taosGetTimestampSec() - pTaskInfo->cost.start / 1000) > 10 * getMaximumIdleDurationSec())
1942 1943
      /*(!needBuildResAfterQueryComplete(pTaskInfo))*/) {
    assert(pTaskInfo->cost.start != 0);
L
Liu Jicong 已提交
1944 1945 1946
    //    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;
1947 1948 1949 1950 1951
  }

  return false;
}

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

L
Liu Jicong 已提交
1954
static bool isCachedLastQuery(STaskAttr* pQueryAttr) {
1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983
  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 已提交
1984 1985 1986
// todo refactor : return window
void getAlignQueryTimeWindow(SInterval* pInterval, int32_t precision, int64_t key, int64_t keyFirst, int64_t keyLast,
                             STimeWindow* win) {
1987
  ASSERT(key >= keyFirst && key <= keyLast);
H
Haojun Liao 已提交
1988
  win->skey = taosTimeTruncate(key, pInterval, precision);
1989 1990

  /*
H
Haojun Liao 已提交
1991
   * if the realSkey > INT64_MAX - pInterval->interval, the query duration between
1992 1993
   * realSkey and realEkey must be less than one interval.Therefore, no need to adjust the query ranges.
   */
H
Haojun Liao 已提交
1994 1995
  if (keyFirst > (INT64_MAX - pInterval->interval)) {
    assert(keyLast - keyFirst < pInterval->interval);
1996 1997
    win->ekey = INT64_MAX;
  } else {
1998
    win->ekey = taosTimeAdd(win->skey, pInterval->interval, pInterval->intervalUnit, precision) - 1;
1999 2000 2001
  }
}

L
Liu Jicong 已提交
2002
static int32_t updateBlockLoadStatus(STaskAttr* pQuery, int32_t status) {
2003 2004 2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025
  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 已提交
2026
    if (!hasOtherFunc) {
2027 2028 2029 2030 2031 2032 2033 2034 2035
      return BLK_DATA_DISCARD;
    } else {
      return BLK_DATA_ALL_NEEDED;
    }
  }

  return status;
}

H
Haojun Liao 已提交
2036
static void doUpdateLastKey(STaskAttr* pQueryAttr) {
2037 2038 2039
  STimeWindow* win = &pQueryAttr->window;

  size_t num = taosArrayGetSize(pQueryAttr->tableGroupInfo.pGroupList);
L
Liu Jicong 已提交
2040
  for (int32_t i = 0; i < num; ++i) {
2041 2042 2043
    SArray* p1 = taosArrayGetP(pQueryAttr->tableGroupInfo.pGroupList, i);

    size_t len = taosArrayGetSize(p1);
L
Liu Jicong 已提交
2044 2045 2046 2047 2048 2049 2050
    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;
      //      }
2051 2052 2053 2054
    }
  }
}

L
Liu Jicong 已提交
2055 2056
// static void updateDataCheckOrder(SQInfo *pQInfo, SQueryTableReq* pQueryMsg, bool stableQuery) {
//   STaskAttr* pQueryAttr = pQInfo->runtimeEnv.pQueryAttr;
H
Haojun Liao 已提交
2057
//
L
Liu Jicong 已提交
2058 2059 2060 2061
//   // 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 已提交
2062
//
L
Liu Jicong 已提交
2063 2064 2065 2066 2067
//   // 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 已提交
2068
//
L
Liu Jicong 已提交
2069 2070 2071 2072
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     if (pQueryAttr->window.skey > pQueryAttr->window.ekey) {
//       TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//     }
H
Haojun Liao 已提交
2073
//
L
Liu Jicong 已提交
2074 2075 2076
//     pQueryAttr->needReverseScan = false;
//     return;
//   }
H
Haojun Liao 已提交
2077
//
L
Liu Jicong 已提交
2078 2079 2080 2081 2082
//   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 已提交
2083
//
L
Liu Jicong 已提交
2084 2085 2086 2087
//     pQueryAttr->needReverseScan = false;
//     doUpdateLastKey(pQueryAttr);
//     return;
//   }
H
Haojun Liao 已提交
2088
//
L
Liu Jicong 已提交
2089 2090 2091 2092 2093 2094
//   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 已提交
2095
//
L
Liu Jicong 已提交
2096 2097 2098
//     pQueryAttr->order.order = TSDB_ORDER_ASC;
//     return;
//   }
H
Haojun Liao 已提交
2099
//
L
Liu Jicong 已提交
2100 2101 2102 2103
//   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 已提交
2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129
////               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 已提交
2130 2131
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
2132 2133 2134 2135 2136 2137 2138 2139 2140 2141
//
//          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 已提交
2142 2143
////                 pQueryAttr->window.skey, pQueryAttr->window.ekey, pQueryAttr->window.ekey,
/// pQueryAttr->window.skey);
H
Haojun Liao 已提交
2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154
//
//          TSWAP(pQueryAttr->window.skey, pQueryAttr->window.ekey, TSKEY);
//          doUpdateLastKey(pQueryAttr);
//        }
//
//        pQueryAttr->order.order = TSDB_ORDER_DESC;
//        pQueryAttr->needReverseScan = false;
//      }
//    }
//  }
//}
2155

H
Haojun Liao 已提交
2156 2157
static void getIntermediateBufInfo(STaskRuntimeEnv* pRuntimeEnv, int32_t* ps, int32_t* rowsize) {
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
L
Liu Jicong 已提交
2158
  int32_t    MIN_ROWS_PER_PAGE = 4;
2159

L
Liu Jicong 已提交
2160 2161
  *rowsize = (int32_t)(pQueryAttr->resultRowSize *
                       getRowNumForMultioutput(pQueryAttr, pQueryAttr->topBotQuery, pQueryAttr->stableQuery));
2162 2163 2164 2165
  int32_t overhead = sizeof(SFilePage);

  // one page contains at least two rows
  *ps = DEFAULT_INTERN_BUF_PAGE_SIZE;
L
Liu Jicong 已提交
2166
  while (((*rowsize) * MIN_ROWS_PER_PAGE) > (*ps) - overhead) {
2167 2168 2169 2170 2171 2172
    *ps = ((*ps) << 1u);
  }
}

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

L
Liu Jicong 已提交
2173 2174 2175
// static FORCE_INLINE bool doFilterByBlockStatistics(STaskRuntimeEnv* pRuntimeEnv, SDataStatis *pDataStatis,
// SqlFunctionCtx *pCtx, int32_t numOfRows) {
//   STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
2176
//
L
Liu Jicong 已提交
2177 2178 2179
//   if (pDataStatis == NULL || pQueryAttr->pFilters == NULL) {
//     return true;
//   }
2180
//
L
Liu Jicong 已提交
2181 2182
//   return filterRangeExecute(pQueryAttr->pFilters, pDataStatis, pQueryAttr->numOfCols, numOfRows);
// }
2183

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

dengyihao's avatar
dengyihao 已提交
2187 2188
  TSKEY sk = TMIN(pQueryAttr->window.skey, pQueryAttr->window.ekey);
  TSKEY ek = TMAX(pQueryAttr->window.skey, pQueryAttr->window.ekey);
2189 2190

  if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
L
Liu Jicong 已提交
2191
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.skey, sk, ek, &w);
2192 2193 2194 2195 2196 2197
    assert(w.ekey >= pBlockInfo->window.skey);

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

L
Liu Jicong 已提交
2198 2199
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
2200 2201 2202 2203 2204 2205 2206 2207 2208 2209
      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 已提交
2210
    //    getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.ekey, sk, ek, &w);
2211 2212 2213 2214 2215 2216
    assert(w.skey <= pBlockInfo->window.ekey);

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

L
Liu Jicong 已提交
2217 2218
    while (1) {
      //      getNextTimeWindow(pQueryAttr, &w);
2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232
      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 已提交
2233
static int32_t doTSJoinFilter(STaskRuntimeEnv* pRuntimeEnv, TSKEY key, bool ascQuery) {
2234 2235 2236
  STSElem elem = tsBufGetElem(pRuntimeEnv->pTsBuf);

#if defined(_DEBUG_VIEW)
L
Liu Jicong 已提交
2237 2238
  printf("elem in comp ts file:%" PRId64 ", key:%" PRId64 ", tag:%" PRIu64
         ", query order:%d, ts order:%d, traverse:%d, index:%d\n",
2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308
         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;
}

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 已提交
2309
void filterColRowsInDataBlock(STaskRuntimeEnv* pRuntimeEnv, SSDataBlock* pBlock, bool ascQuery) {
L
Liu Jicong 已提交
2310
  int32_t numOfRows = pBlock->info.rows;
2311

L
Liu Jicong 已提交
2312 2313
  int8_t* p = NULL;
  bool    all = true;
2314

L
Liu Jicong 已提交
2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336
  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;
      }
    }
2337

L
Liu Jicong 已提交
2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352
    // 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
    }
  }
2353

L
Liu Jicong 已提交
2354
  taosMemoryFreeClear(p);
2355 2356 2357
}

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

static uint32_t doFilterByBlockTimeWindow(STableScanInfo* pTableScanInfo, SSDataBlock* pBlock) {
H
Haojun Liao 已提交
2361
  SqlFunctionCtx* pCtx = pTableScanInfo->pCtx;
L
Liu Jicong 已提交
2362
  uint32_t        status = BLK_DATA_NO_NEEDED;
2363 2364 2365 2366

  int32_t numOfOutput = pTableScanInfo->numOfOutput;
  for (int32_t i = 0; i < numOfOutput; ++i) {
    int32_t functionId = pCtx[i].functionId;
H
Haojun Liao 已提交
2367
    int32_t colId = pTableScanInfo->pExpr[i].base.pParam[0].pCol->colId;
2368 2369 2370 2371 2372 2373

    // 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 已提交
2374 2375 2376 2377
      //      status |= aAggs[functionId].dataReqFunc(&pTableScanInfo->pCtx[i], &pBlock->info.window, colId);
      //      if ((status & BLK_DATA_ALL_NEEDED) == BLK_DATA_ALL_NEEDED) {
      //        return status;
      //      }
2378 2379 2380 2381 2382 2383
    }
  }

  return status;
}

L
Liu Jicong 已提交
2384 2385
int32_t loadDataBlockOnDemand(SExecTaskInfo* pTaskInfo, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock,
                              uint32_t* status) {
2386 2387
  *status = BLK_DATA_NO_NEEDED;

H
Haojun Liao 已提交
2388
  pBlock->pDataBlock = NULL;
L
Liu Jicong 已提交
2389
  pBlock->pBlockAgg = NULL;
H
Haojun Liao 已提交
2390

L
Liu Jicong 已提交
2391 2392
  //  int64_t groupId = pRuntimeEnv->current->groupIndex;
  //  bool    ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
2393

H
Haojun Liao 已提交
2394
  STaskCostInfo* pCost = &pTaskInfo->cost;
2395 2396 2397

  pCost->totalBlocks += 1;
  pCost->totalRows += pBlock->info.rows;
H
Haojun Liao 已提交
2398
#if 0
2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422
  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 已提交
2423
      (QUERY_IS_INTERVAL_QUERY(pQueryAttr) && overlapWithTimeWindow(pTaskInfo, &pBlock->info))) {
2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435
    (*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 已提交
2436
      bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464
      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 已提交
2465
//  *status = updateBlockLoadStatus(pRuntimeEnv->pQueryAttr, *status);
2466 2467 2468 2469 2470 2471 2472 2473

  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;
2474
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
2475 2476

    if (pBlock->pBlockAgg == NULL) {  // data block statistics does not exist, load data block
2477
//      pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
2478 2479 2480 2481 2482 2483 2484
      pCost->totalCheckedRows += pBlock->info.rows;
    }
  } else {
    assert((*status) == BLK_DATA_ALL_NEEDED);

    // load the data block statistics to perform further filter
    pCost->loadBlockStatis += 1;
2485
//    tsdbRetrieveDataBlockStatisInfo(pTableScanInfo->pTsdbReadHandle, &pBlock->pBlockAgg);
2486 2487 2488 2489 2490 2491

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

H
Haojun Liao 已提交
2492
          bool  masterScan = IS_MAIN_SCAN(pRuntimeEnv);
2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530
          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;
2531
//    pBlock->pDataBlock = tsdbRetrieveDataBlock(pTableScanInfo->pTsdbReadHandle, NULL);
2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543
//    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 已提交
2544
#endif
2545 2546 2547
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
2548
int32_t binarySearchForKey(char* pValue, int num, TSKEY key, int order) {
2549 2550 2551 2552 2553 2554 2555 2556 2557
  int32_t midPos = -1;
  int32_t numOfRows;

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

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

L
Liu Jicong 已提交
2558
  TSKEY*  keyList = (TSKEY*)pValue;
2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608 2609 2610 2611
  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 已提交
2612
 * set tag value in SqlFunctionCtx
2613 2614
 * e.g.,tag information into input buffer
 */
L
Liu Jicong 已提交
2615
static void doSetTagValueInParam(void* pTable, int32_t tagColId, SVariant* tag, int16_t type, int16_t bytes) {
2616 2617 2618
  taosVariantDestroy(tag);

  char* val = NULL;
L
Liu Jicong 已提交
2619 2620 2621 2622 2623 2624
  //  if (tagColId == TSDB_TBNAME_COLUMN_INDEX) {
  //    val = tsdbGetTableName(pTable);
  //    assert(val != NULL);
  //  } else {
  //    val = tsdbGetTableTagVal(pTable, tagColId, type, bytes);
  //  }
2625 2626 2627 2628 2629 2630 2631 2632

  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 已提交
2633
    int32_t len = (varDataLen(val) > maxLen) ? maxLen : varDataLen(val);
2634
    taosVariantCreateFromBinary(tag, varDataVal(val), len, type);
L
Liu Jicong 已提交
2635
    // taosVariantCreateFromBinary(tag, varDataVal(val), varDataLen(val), type);
2636 2637 2638 2639 2640 2641 2642 2643
  } 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 已提交
2644
  for (int32_t i = 0; i < numOfTags; ++i) {
2645 2646 2647 2648 2649 2650 2651 2652
    if (pTagColList[i].colId == colId) {
      return &pTagColList[i];
    }
  }

  return NULL;
}

L
Liu Jicong 已提交
2653 2654
void setTagValue(SOperatorInfo* pOperatorInfo, void* pTable, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
  SExprInfo* pExpr = pOperatorInfo->pExpr;
2655
  SExprInfo* pExprInfo = &pExpr[0];
L
Liu Jicong 已提交
2656
  int32_t    functionId = getExprFunctionId(pExprInfo);
2657
#if 0
2658 2659 2660
  if (pQueryAttr->numOfOutput == 1 && functionId == FUNCTION_TS_COMP && pQueryAttr->stableQuery) {
    assert(pExprInfo->base.numOfParams == 1);

L
Liu Jicong 已提交
2661
    //    int16_t      tagColId = (int16_t)pExprInfo->base.param[0].i;
H
Haojun Liao 已提交
2662
    int16_t      tagColId = -1;
2663 2664 2665
    SColumnInfo* pColInfo = doGetTagColumnInfoById(pQueryAttr->tagColList, pQueryAttr->numOfTags, tagColId);

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

2667 2668 2669 2670 2671 2672 2673 2674 2675
  } 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 已提交
2676
      if (!TSDB_COL_IS_TAG(pLocalExprInfo->base.pParam[0].pCol->flag)) {
2677 2678 2679 2680
        continue;
      }

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

L
Liu Jicong 已提交
2684 2685 2686
      if (IS_NUMERIC_TYPE(pLocalExprInfo->base.resSchema.type) ||
          pLocalExprInfo->base.resSchema.type == TSDB_DATA_TYPE_BOOL ||
          pLocalExprInfo->base.resSchema.type == TSDB_DATA_TYPE_TIMESTAMP) {
2687 2688 2689 2690
        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 已提交
2691
        }
2692 2693 2694 2695 2696 2697 2698 2699 2700 2701
      }

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

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

2704 2705
}

H
Haojun Liao 已提交
2706
void copyToSDataBlock(SSDataBlock* pBlock, int32_t* offset, SGroupResInfo* pGroupResInfo, SDiskbasedBuf* pResBuf) {
2707 2708 2709 2710 2711 2712 2713
  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 已提交
2714 2715 2716
      //      if ((code = mergeIntoGroupResult(&pGroupResInfo, pRuntimeEnv, offset)) != TSDB_CODE_SUCCESS) {
      return;
      //      }
2717 2718
    }

L
Liu Jicong 已提交
2719
    //    doCopyToSDataBlock(pResBuf, pGroupResInfo, TSDB_ORDER_ASC, pBlock, );
2720 2721 2722 2723 2724 2725 2726 2727 2728 2729

    // 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 已提交
2730 2731 2732
    //    if (pBlock->info.rows >= threshold) {
    //      break;
    //    }
2733 2734 2735
  }
}

L
Liu Jicong 已提交
2736
static void updateTableQueryInfoForReverseScan(STableQueryInfo* pTableQueryInfo) {
2737 2738 2739 2740
  if (pTableQueryInfo == NULL) {
    return;
  }

L
Liu Jicong 已提交
2741 2742
  //  TSWAP(pTableQueryInfo->win.skey, pTableQueryInfo->win.ekey, TSKEY);
  //  pTableQueryInfo->lastKey = pTableQueryInfo->win.skey;
2743

L
Liu Jicong 已提交
2744 2745
  //  SWITCH_ORDER(pTableQueryInfo->cur.order);
  //  pTableQueryInfo->cur.vgroupIndex = -1;
2746 2747 2748 2749 2750 2751 2752 2753 2754 2755

  // 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 已提交
2756
void initResultRow(SResultRow* pResultRow) {
2757 2758 2759 2760 2761 2762
  pResultRow->pEntryInfo = (struct SResultRowEntryInfo*)((char*)pResultRow + sizeof(SResultRow));
}

/*
 * 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 已提交
2763 2764 2765
 * +------------+-----------------result column 1------------+------------------result column 2-----------+
 * | SResultRow | SResultRowEntryInfo | intermediate buffer1 | SResultRowEntryInfo | intermediate buffer 2|
 * +------------+--------------------------------------------+--------------------------------------------+
2766 2767
 *           offset[0]                                  offset[1]                                   offset[2]
 */
2768
// TODO refactor: some function move away
H
Haojun Liao 已提交
2769
void setFunctionResultOutput(SOptrBasicInfo* pInfo, SAggSupporter* pSup, int32_t stage, SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
2770 2771 2772
  SqlFunctionCtx* pCtx = pInfo->pCtx;
  SSDataBlock*    pDataBlock = pInfo->pRes;
  int32_t*        rowCellInfoOffset = pInfo->rowCellInfoOffset;
H
Haojun Liao 已提交
2773

H
Haojun Liao 已提交
2774
  SResultRowInfo* pResultRowInfo = &pInfo->resultRowInfo;
H
Haojun Liao 已提交
2775
  initResultRowInfo(pResultRowInfo, 16);
H
Haojun Liao 已提交
2776

L
Liu Jicong 已提交
2777 2778 2779 2780
  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 已提交
2781 2782 2783 2784 2785

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

L
Liu Jicong 已提交
2786
    pCtx[i].resultInfo = pEntry;
H
Haojun Liao 已提交
2787 2788 2789
    pCtx[i].currentStage = stage;

    // set the timestamp output buffer for top/bottom/diff query
L
Liu Jicong 已提交
2790 2791
    //    int32_t fid = pCtx[i].functionId;
    //    if (fid == FUNCTION_TOP || fid == FUNCTION_BOTTOM || fid == FUNCTION_DIFF || fid == FUNCTION_DERIVATIVE) {
H
Haojun Liao 已提交
2792
    //      if (i > 0) pCtx[i].pTsOutput = pCtx[i-1].pOutput;
L
Liu Jicong 已提交
2793
    //    }
H
Haojun Liao 已提交
2794 2795 2796 2797 2798
  }

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

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

L
Liu Jicong 已提交
2802
  int32_t newSize = pDataBlock->info.rows + numOfInputRows + 5;  // extra output buffer
2803
  if ((*bufCapacity) < newSize) {
L
Liu Jicong 已提交
2804 2805
    for (int32_t i = 0; i < pDataBlock->info.numOfCols; ++i) {
      SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
2806

wafwerar's avatar
wafwerar 已提交
2807
      char* p = taosMemoryRealloc(pColInfo->pData, newSize * pColInfo->info.bytes);
2808 2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820
      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 已提交
2821
    SColumnInfoData* pColInfo = taosArrayGet(pDataBlock->pDataBlock, i);
2822 2823 2824 2825 2826
    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 已提交
2827 2828
    if (functionId == FUNCTION_TOP || functionId == FUNCTION_BOTTOM || functionId == FUNCTION_DIFF ||
        functionId == FUNCTION_DERIVATIVE) {
H
Haojun Liao 已提交
2829
//      if (i > 0) pBInfo->pCtx[i].pTsOutput = pBInfo->pCtx[i - 1].pOutput;
2830 2831 2832 2833
    }
  }
}

H
Haojun Liao 已提交
2834
void copyTsColoum(SSDataBlock* pRes, SqlFunctionCtx* pCtx, int32_t numOfOutput) {
2835 2836
  bool    needCopyTs = false;
  int32_t tsNum = 0;
L
Liu Jicong 已提交
2837
  char*   src = NULL;
2838 2839 2840 2841
  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 已提交
2842 2843
      if (i > 0 && pCtx[i - 1].functionId == FUNCTION_TS_DUMMY) {
        SColumnInfoData* pColRes = taosArrayGet(pRes->pDataBlock, i - 1);  // find ts data
2844 2845
        src = pColRes->pData;
      }
L
Liu Jicong 已提交
2846
    } else if (functionId == FUNCTION_TS_DUMMY) {
2847 2848 2849 2850 2851 2852 2853 2854 2855 2856
      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 已提交
2857
    if (functionId == FUNCTION_TS_DUMMY) {
2858 2859 2860 2861 2862 2863
      SColumnInfoData* pColRes = taosArrayGet(pRes->pDataBlock, i);
      memcpy(pColRes->pData, src, pColRes->info.bytes * pRes->info.rows);
    }
  }
}

H
Haojun Liao 已提交
2864
void initCtxOutputBuffer(SqlFunctionCtx* pCtx, int32_t size) {
2865 2866
  for (int32_t j = 0; j < size; ++j) {
    struct SResultRowEntryInfo* pResInfo = GET_RES_INFO(&pCtx[j]);
H
Haojun Liao 已提交
2867
    if (isRowEntryInitialized(pResInfo) || fmIsPseudoColumnFunc(pCtx[j].functionId) || pCtx[j].functionId == -1 || fmIsScalarFunc(pCtx[j].functionId)) {
2868 2869 2870
      continue;
    }

H
Haojun Liao 已提交
2871
    pCtx[j].fpSet.init(&pCtx[j], pCtx[j].resultInfo);
2872 2873 2874
  }
}

L
Liu Jicong 已提交
2875
void setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status) {
2876
  if (status == TASK_NOT_COMPLETED) {
H
Haojun Liao 已提交
2877
    pTaskInfo->status = status;
2878 2879
  } else {
    // QUERY_NOT_COMPLETED is not compatible with any other status, so clear its position first
2880
    CLEAR_QUERY_STATUS(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
2881
    pTaskInfo->status |= status;
2882 2883 2884
  }
}

2885 2886
void finalizeQueryResult(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
  for (int32_t j = 0; j < numOfOutput; ++j) {
H
Haojun Liao 已提交
2887 2888 2889 2890
    if (pCtx[j].functionId == -1) {
      continue;
    }

2891 2892 2893 2894
    pCtx[j].fpSet.finalize(&pCtx[j]);
  }
}

L
Liu Jicong 已提交
2895 2896
void finalizeMultiTupleQueryResult(SqlFunctionCtx* pCtx, int32_t numOfOutput, SDiskbasedBuf* pBuf,
                                   SResultRowInfo* pResultRowInfo, int32_t* rowCellInfoOffset) {
2897 2898 2899
  for (int32_t i = 0; i < pResultRowInfo->size; ++i) {
    SResultRowPosition* pPos = &pResultRowInfo->pPosition[i];

L
Liu Jicong 已提交
2900
    SFilePage*  bufPage = getBufPage(pBuf, pPos->pageId);
2901
    SResultRow* pRow = (SResultRow*)((char*)bufPage + pPos->offset);
H
Haojun Liao 已提交
2902 2903 2904 2905 2906

    // TODO ignore the close status anyway.
//    if (!isResultRowClosed(pRow)) {
//      continue;
//    }
2907

2908
    for (int32_t j = 0; j < numOfOutput; ++j) {
2909 2910
      pCtx[j].resultInfo = getResultCell(pRow, j, rowCellInfoOffset);

H
Haojun Liao 已提交
2911
      struct SResultRowEntryInfo* pResInfo = pCtx[j].resultInfo;
2912 2913 2914 2915
      if (isRowEntryCompleted(pResInfo) && isRowEntryInitialized(pResInfo)) {
        continue;
      }

H
Haojun Liao 已提交
2916
      if (pCtx[j].fpSet.process) {  // TODO set the dummy function, to avoid the check for null ptr.
H
Haojun Liao 已提交
2917 2918
        pCtx[j].fpSet.finalize(&pCtx[j]);
      }
2919 2920 2921 2922

      if (pRow->numOfRows < pResInfo->numOfRes) {
        pRow->numOfRows = pResInfo->numOfRes;
      }
2923
    }
2924 2925 2926

    releaseBufPage(pBuf, bufPage);
  }
2927 2928
}

2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948
void finalizeUpdatedResult(SqlFunctionCtx* pCtx, int32_t numOfOutput, SDiskbasedBuf* pBuf, SArray* pUpdateList,
                           int32_t* rowCellInfoOffset) {
  size_t num = taosArrayGetSize(pUpdateList);

  for (int32_t i = 0; i < num; ++i) {
    SResultRowPosition* pPos = taosArrayGet(pUpdateList, i);

    SFilePage*  bufPage = getBufPage(pBuf, pPos->pageId);
    SResultRow* pRow = (SResultRow*)((char*)bufPage + pPos->offset);

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

      struct SResultRowEntryInfo* pResInfo = pCtx[j].resultInfo;
      if (isRowEntryCompleted(pResInfo) && isRowEntryInitialized(pResInfo)) {
        continue;
      }

      if (pCtx[j].fpSet.process) {  // TODO set the dummy function.
        pCtx[j].fpSet.finalize(&pCtx[j]);
2949
        pResInfo->initialized = true;
2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961 2962 2963 2964 2965
      }

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

    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
     */
    //    buf->numOfRows = (uint16_t)getNumOfResult(pCtx, numOfOutput);
  }
}

L
Liu Jicong 已提交
2966
static bool hasMainOutput(STaskAttr* pQueryAttr) {
2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977
  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 已提交
2978 2979
STableQueryInfo* createTableQueryInfo(void* buf, bool groupbyColumn, STimeWindow win) {
  STableQueryInfo* pTableQueryInfo = buf;
2980 2981 2982
  pTableQueryInfo->lastKey = win.skey;

  // set more initial size of interval/groupby query
L
Liu Jicong 已提交
2983 2984 2985 2986 2987 2988 2989 2990
  //  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
  //  }
2991 2992 2993 2994

  return pTableQueryInfo;
}

L
Liu Jicong 已提交
2995
void destroyTableQueryInfoImpl(STableQueryInfo* pTableQueryInfo) {
2996 2997 2998 2999
  if (pTableQueryInfo == NULL) {
    return;
  }

L
Liu Jicong 已提交
3000
  //  taosVariantDestroy(&pTableQueryInfo->tag);
3001 3002 3003
  cleanupResultRowInfo(&pTableQueryInfo->resInfo);
}

H
Haojun Liao 已提交
3004
void setResultRowOutputBufInitCtx_rv(SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowCellInfoOffset) {
3005 3006 3007 3008 3009 3010 3011
  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;
    }
3012 3013 3014 3015 3016

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

H
Haojun Liao 已提交
3017
    if (!pResInfo->initialized && pCtx[i].functionId != -1) {
3018
      pCtx[i].fpSet.init(&pCtx[i], pResInfo);
3019 3020 3021 3022
    }
  }
}

3023 3024 3025 3026 3027 3028
void doFilter(const SNode* pFilterNode, SSDataBlock* pBlock) {
  if (pFilterNode == NULL) {
    return;
  }

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

H
Haojun Liao 已提交
3030
  // todo move to the initialization function
H
Haojun Liao 已提交
3031
  int32_t code = filterInitFromNode((SNode*)pFilterNode, &filter, 0);
3032 3033 3034 3035 3036 3037

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

  int8_t* rowRes = NULL;
  bool    keep = filterExecute(filter, pBlock, &rowRes, NULL, param1.numOfCols);
D
dapan1121 已提交
3038
  filterFreeInfo(filter);
3039 3040 3041 3042

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

H
Haojun Liao 已提交
3043
  // todo extract method
3044 3045 3046 3047
  int32_t numOfRow = 0;
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
    SColumnInfoData* pDst = taosArrayGet(px->pDataBlock, i);
    SColumnInfoData* pSrc = taosArrayGet(pBlock->pDataBlock, i);
D
dapan1121 已提交
3048 3049 3050 3051 3052 3053 3054 3055 3056
    if (keep) {
      colDataAssign(pDst, pSrc, pBlock->info.rows);
      numOfRow = pBlock->info.rows;
    } else if (NULL != rowRes) {
      numOfRow = 0;
      for (int32_t j = 0; j < pBlock->info.rows; ++j) {
        if (rowRes[j] == 0) {
          continue;
        }
3057

D
dapan1121 已提交
3058 3059 3060 3061 3062 3063
        if (colDataIsNull_s(pSrc, j)) {
          colDataAppendNULL(pDst, numOfRow);
        } else {
          colDataAppend(pDst, numOfRow, colDataGetData(pSrc, j), false);
        }
        numOfRow += 1;
H
Haojun Liao 已提交
3064
      }
D
dapan1121 已提交
3065 3066
    } else {
      numOfRow = 0;
3067 3068 3069 3070 3071 3072
    }

    *pSrc = *pDst;
  }

  pBlock->info.rows = numOfRow;
3073
  blockDataUpdateTsWindow(pBlock);
3074 3075
}

H
Haojun Liao 已提交
3076
void doSetTableGroupOutputBuf(SAggOperatorInfo* pAggInfo, int32_t numOfOutput, int32_t tableGroupId, SExecTaskInfo* pTaskInfo) {
3077 3078 3079 3080
  // for simple group by query without interval, all the tables belong to one group result.
  int64_t uid = 0;
  int64_t tid = 0;

3081 3082
  SResultRowInfo* pResultRowInfo = &pAggInfo->binfo.resultRowInfo;
  SqlFunctionCtx* pCtx = pAggInfo->binfo.pCtx;
L
Liu Jicong 已提交
3083
  int32_t*        rowCellInfoOffset = pAggInfo->binfo.rowCellInfoOffset;
3084

3085
  SResultRow* pResultRow =
L
Liu Jicong 已提交
3086 3087 3088
      doSetResultOutBufByKey_rv(pAggInfo->pResultBuf, pResultRowInfo, tid, (char*)&tableGroupId, sizeof(tableGroupId),
                                true, uid, pTaskInfo, false, &pAggInfo->aggSup);
  assert(pResultRow != NULL);
3089 3090 3091 3092 3093 3094

  /*
   * 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) {
H
Haojun Liao 已提交
3095
    int32_t ret = addNewWindowResultBuf(pResultRow, pAggInfo->pResultBuf, tableGroupId, pAggInfo->binfo.pRes->info.rowSize);
3096 3097 3098 3099 3100
    if (ret != TSDB_CODE_SUCCESS) {
      return;
    }
  }

H
Haojun Liao 已提交
3101
  setResultRowOutputBufInitCtx_rv(pResultRow, pCtx, numOfOutput, rowCellInfoOffset);
3102 3103
}

L
Liu Jicong 已提交
3104 3105
void setExecutionContext(int32_t numOfOutput, int32_t tableGroupId, TSKEY nextKey, SExecTaskInfo* pTaskInfo,
                         STableQueryInfo* pTableQueryInfo, SAggOperatorInfo* pAggInfo) {
3106 3107
  // lastKey needs to be updated
  pTableQueryInfo->lastKey = nextKey;
3108
  if (pAggInfo->groupId != INT32_MIN && pAggInfo->groupId == tableGroupId) {
3109 3110 3111
    return;
  }

3112
  doSetTableGroupOutputBuf(pAggInfo, numOfOutput, tableGroupId, pTaskInfo);
3113 3114

  // record the current active group id
3115
  pAggInfo->groupId = tableGroupId;
3116 3117
}

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

H
Haojun Liao 已提交
3121
  SExprBasicInfo* pExpr = &pExprInfo->base;
L
Liu Jicong 已提交
3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145
  //  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 已提交
3146
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
3147 3148

  assert(pRuntimeEnv->pTsBuf != NULL);
H
Haojun Liao 已提交
3149
#if 0
3150 3151 3152 3153 3154 3155 3156 3157 3158
  // 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) {
3159
        //qError("QInfo:0x%"PRIx64" failed to find tag:%s in ts_comp", GET_TASKID(pRuntimeEnv), pTag->pz);
3160
      } else {
3161
        //qError("QInfo:0x%"PRIx64" failed to find tag:%" PRId64 " in ts_comp", GET_TASKID(pRuntimeEnv), pTag->i);
3162 3163 3164 3165 3166 3167 3168 3169
      }

      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) {
3170
      //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);
3171
    } else {
3172
      //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);
3173 3174 3175 3176 3177
    }

  } else {
    tsBufSetCursor(pRuntimeEnv->pTsBuf, &pTableQueryInfo->cur);
    if (pTag->nType == TSDB_DATA_TYPE_BINARY || pTag->nType == TSDB_DATA_TYPE_NCHAR) {
3178
      //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);
3179
    } else {
3180
      //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);
3181 3182
    }
  }
H
Haojun Liao 已提交
3183
#endif
3184 3185 3186 3187 3188 3189
  return 0;
}

/*
 * There are two cases to handle:
 *
L
Liu Jicong 已提交
3190 3191
 * 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.
3192 3193 3194 3195
 * 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 已提交
3196 3197 3198 3199
void setIntervalQueryRange(STaskRuntimeEnv* pRuntimeEnv, TSKEY key) {
  STaskAttr*       pQueryAttr = pRuntimeEnv->pQueryAttr;
  STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
  SResultRowInfo*  pResultRowInfo = &pTableQueryInfo->resInfo;
3200 3201 3202 3203 3204

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

L
Liu Jicong 已提交
3205
  //  pTableQueryInfo->win.skey = key;
3206 3207 3208 3209 3210 3211 3212 3213 3214 3215
  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 已提交
3216 3217
  TSKEY sk = TMIN(win.skey, win.ekey);
  TSKEY ek = TMAX(win.skey, win.ekey);
L
Liu Jicong 已提交
3218
  //  getAlignQueryTimeWindow(pQueryAttr, win.skey, sk, ek, &w);
3219

L
Liu Jicong 已提交
3220 3221 3222 3223 3224 3225 3226
  //  if (pResultRowInfo->prevSKey == TSKEY_INITIAL_VAL) {
  //    if (!QUERY_IS_ASC_QUERY(pQueryAttr)) {
  //      assert(win.ekey == pQueryAttr->window.ekey);
  //    }
  //
  //    pResultRowInfo->prevSKey = w.skey;
  //  }
3227

L
Liu Jicong 已提交
3228
  //  pTableQueryInfo->lastKey = pTableQueryInfo->win.skey;
3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239
}

/**
 * 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
 */
3240
int32_t doCopyToSDataBlock(SSDataBlock* pBlock, int32_t rowCapacity, SExprInfo* pExprInfo, SDiskbasedBuf* pBuf, SGroupResInfo* pGroupResInfo, int32_t orderType, int32_t* rowCellOffset) {
3241
  int32_t numOfRows = getNumOfTotalRes(pGroupResInfo);
L
Liu Jicong 已提交
3242
  int32_t numOfResult = pBlock->info.rows;  // there are already exists result rows
3243 3244 3245 3246

  int32_t start = 0;
  int32_t step = -1;

L
Liu Jicong 已提交
3247
  // qDebug("QInfo:0x%"PRIx64" start to copy data from windowResInfo to output buf", GET_TASKID(pRuntimeEnv));
3248 3249 3250 3251 3252 3253 3254 3255 3256 3257
  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;
  }

3258 3259
  int32_t nrows = pBlock->info.rows;

3260
  for (int32_t i = start; (i < numOfRows) && (i >= 0); i += step) {
3261
    SResultRowPosition* pPos = taosArrayGet(pGroupResInfo->pRows, i);
L
Liu Jicong 已提交
3262
    SFilePage*          page = getBufPage(pBuf, pPos->pageId);
3263 3264

    SResultRow* pRow = (SResultRow*)((char*)page + pPos->offset);
3265 3266 3267 3268 3269
    if (pRow->numOfRows == 0) {
      pGroupResInfo->index += 1;
      continue;
    }

3270
    // TODO copy multiple rows?
3271
    int32_t numOfRowsToCopy = pRow->numOfRows;
L
Liu Jicong 已提交
3272
    if (numOfResult + numOfRowsToCopy >= rowCapacity) {
3273 3274 3275 3276 3277 3278
      break;
    }

    pGroupResInfo->index += 1;

    for (int32_t j = 0; j < pBlock->info.numOfCols; ++j) {
3279 3280 3281
      int32_t slotId = pExprInfo[j].base.resSchema.slotId;

      SColumnInfoData*     pColInfoData = taosArrayGet(pBlock->pDataBlock, slotId);
3282
      SResultRowEntryInfo* pEntryInfo = getResultCell(pRow, j, rowCellOffset);
3283

3284
      char* in = GET_ROWCELL_INTERBUF(pEntryInfo);
H
Haojun Liao 已提交
3285
      colDataAppend(pColInfoData, nrows, in, pEntryInfo->isNullRes);
3286 3287
    }

3288 3289 3290
    releaseBufPage(pBuf, page);
    nrows += 1;

3291
    numOfResult += numOfRowsToCopy;
H
Haojun Liao 已提交
3292
    if (numOfResult == rowCapacity) {  // output buffer is full
3293 3294 3295 3296
      break;
    }
  }

L
Liu Jicong 已提交
3297
  // qDebug("QInfo:0x%"PRIx64" copy data to query buf completed", GET_TASKID(pRuntimeEnv));
3298 3299 3300 3301
  pBlock->info.rows = numOfResult;
  return 0;
}

3302 3303
void toSDatablock(SSDataBlock* pBlock, int32_t rowCapacity, SGroupResInfo* pGroupResInfo, SExprInfo* pExprInfo, SDiskbasedBuf* pBuf,
                  int32_t* rowCellOffset) {
3304 3305
  assert(pGroupResInfo->currentGroup <= pGroupResInfo->totalGroup);

3306
  blockDataCleanup(pBlock);
3307 3308 3309 3310
  if (!hasRemainDataInCurrentGroup(pGroupResInfo)) {
    return;
  }

3311
  int32_t orderType = TSDB_ORDER_ASC;
3312
  doCopyToSDataBlock(pBlock, rowCapacity, pExprInfo, pBuf, pGroupResInfo, orderType, rowCellOffset);
3313

H
Haojun Liao 已提交
3314 3315
  // add condition (pBlock->info.rows >= 1) just to runtime happy
  blockDataUpdateTsWindow(pBlock);
3316 3317
}

L
Liu Jicong 已提交
3318 3319
static void updateNumOfRowsInResultRows(SqlFunctionCtx* pCtx, int32_t numOfOutput, SResultRowInfo* pResultRowInfo,
                                        int32_t* rowCellInfoOffset) {
3320
  // update the number of result for each, only update the number of rows for the corresponding window result.
L
Liu Jicong 已提交
3321 3322 3323
  //  if (QUERY_IS_INTERVAL_QUERY(pQueryAttr)) {
  //    return;
  //  }
H
Haojun Liao 已提交
3324
#if 0
3325
  for (int32_t i = 0; i < pResultRowInfo->size; ++i) {
L
Liu Jicong 已提交
3326
    SResultRow* pResult = pResultRowInfo->pResult[i];
3327 3328 3329 3330 3331 3332 3333

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

3334 3335
      SResultRowEntryInfo* pCell = getResultCell(pResult, j, rowCellInfoOffset);
      pResult->numOfRows = (uint16_t)(TMAX(pResult->numOfRows, pCell->numOfRes));
3336 3337
    }
  }
H
Haojun Liao 已提交
3338 3339
#endif

3340 3341
}

L
Liu Jicong 已提交
3342
static int32_t compressQueryColData(SColumnInfoData* pColRes, int32_t numOfRows, char* data, int8_t compressed) {
3343 3344
  int32_t colSize = pColRes->info.bytes * numOfRows;
  return (*(tDataTypes[pColRes->info.type].compFunc))(pColRes->pData, colSize, numOfRows, data,
L
Liu Jicong 已提交
3345
                                                      colSize + COMP_OVERFLOW_BYTES, compressed, NULL, 0);
3346 3347
}

L
Liu Jicong 已提交
3348 3349 3350 3351 3352
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);
  //  }
3353 3354 3355 3356 3357 3358 3359 3360 3361 3362

  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 已提交
3363 3364
  event.eventType = eventType;
  event.eventTime = taosGetTimestampUs();
3365 3366 3367
  event.operatorType = operatorInfo->operatorType;

  if (operatorInfo->pRuntimeEnv) {
L
Liu Jicong 已提交
3368 3369 3370 3371
    //    SQInfo* pQInfo = operatorInfo->pRuntimeEnv->qinfo;
    //    if (pQInfo->summary.queryProfEvents) {
    //      taosArrayPush(pQInfo->summary.queryProfEvents, &event);
    //    }
3372 3373 3374
  }
}

L
Liu Jicong 已提交
3375
void publishQueryAbortEvent(SExecTaskInfo* pTaskInfo, int32_t code) {
3376 3377 3378 3379 3380
  SQueryProfEvent event;
  event.eventType = QUERY_PROF_QUERY_ABORT;
  event.eventTime = taosGetTimestampUs();
  event.abortCode = code;

3381 3382
  if (pTaskInfo->cost.queryProfEvents) {
    taosArrayPush(pTaskInfo->cost.queryProfEvents, &event);
3383 3384 3385
  }
}

L
Liu Jicong 已提交
3386
typedef struct {
3387 3388 3389 3390 3391 3392 3393
  uint8_t operatorType;
  int64_t beginTime;
  int64_t endTime;
  int64_t selfTime;
  int64_t descendantsTime;
} SOperatorStackItem;

L
Liu Jicong 已提交
3394 3395
static void doOperatorExecProfOnce(SOperatorStackItem* item, SQueryProfEvent* event, SArray* opStack,
                                   SHashObj* profResults) {
3396 3397 3398 3399 3400 3401 3402 3403
  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 已提交
3404
  uint8_t              operatorType = item->operatorType;
3405 3406 3407 3408 3409 3410 3411 3412 3413
  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 已提交
3414
    taosHashPut(profResults, &(operatorType), sizeof(operatorType), &opResult, sizeof(opResult));
3415 3416 3417
  }
}

3418 3419 3420 3421 3422 3423 3424 3425 3426 3427
void calculateOperatorProfResults(void) {
//  if (pQInfo->summary.queryProfEvents == NULL) {
//    // qDebug("QInfo:0x%"PRIx64" query prof events array is null", pQInfo->qId);
//    return;
//  }
//
//  if (pQInfo->summary.operatorProfResults == NULL) {
//    // qDebug("QInfo:0x%"PRIx64" operator prof results hash is null", pQInfo->qId);
//    return;
//  }
3428 3429 3430 3431 3432

  SArray* opStack = taosArrayInit(32, sizeof(SOperatorStackItem));
  if (opStack == NULL) {
    return;
  }
3433
#if 0
L
Liu Jicong 已提交
3434
  size_t    size = taosArrayGetSize(pQInfo->summary.queryProfEvents);
3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446 3447 3448 3449 3450 3451 3452 3453 3454 3455
  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);
      }
    }
  }
3456
#endif
3457 3458 3459
  taosArrayDestroy(opStack);
}

L
Liu Jicong 已提交
3460 3461
void queryCostStatis(SExecTaskInfo* pTaskInfo) {
  STaskCostInfo* pSummary = &pTaskInfo->cost;
3462

L
Liu Jicong 已提交
3463 3464 3465
  //  uint64_t hashSize = taosHashGetMemSize(pQInfo->runtimeEnv.pResultRowHashTable);
  //  hashSize += taosHashGetMemSize(pRuntimeEnv->tableqinfoGroupInfo.map);
  //  pSummary->hashSize = hashSize;
3466 3467 3468 3469

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

L
Liu Jicong 已提交
3470 3471 3472 3473 3474 3475 3476 3477 3478 3479 3480 3481 3482 3483 3484 3485 3486 3487 3488 3489
  //  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);
3490 3491 3492 3493

  if (pSummary->operatorProfResults) {
    SOperatorProfResult* opRes = taosHashIterate(pSummary->operatorProfResults, NULL);
    while (opRes != NULL) {
L
Liu Jicong 已提交
3494 3495
      // qDebug("QInfo:0x%" PRIx64 " :cost summary: operator : %d, exec times: %" PRId64 ", self time: %" PRId64,
      //             pQInfo->qId, opRes->operatorType, opRes->sumRunTimes, opRes->sumSelfTime);
3496 3497 3498 3499 3500
      opRes = taosHashIterate(pSummary->operatorProfResults, opRes);
    }
  }
}

L
Liu Jicong 已提交
3501 3502 3503
// static void updateOffsetVal(STaskRuntimeEnv *pRuntimeEnv, SDataBlockInfo *pBlockInfo) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
3504
//
L
Liu Jicong 已提交
3505
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
3506
//
L
Liu Jicong 已提交
3507 3508 3509 3510
//   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;
//   }
3511
//
L
Liu Jicong 已提交
3512 3513 3514 3515 3516
//   if (QUERY_IS_ASC_QUERY(pQueryAttr)) {
//     pQueryAttr->pos = (int32_t)pQueryAttr->limit.offset;
//   } else {
//     pQueryAttr->pos = pBlockInfo->rows - (int32_t)pQueryAttr->limit.offset - 1;
//   }
3517
//
L
Liu Jicong 已提交
3518
//   assert(pQueryAttr->pos >= 0 && pQueryAttr->pos <= pBlockInfo->rows - 1);
3519
//
L
Liu Jicong 已提交
3520 3521
//   SArray *         pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//   SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
3522
//
L
Liu Jicong 已提交
3523 3524
//   // update the pQueryAttr->limit.offset value, and pQueryAttr->pos value
//   TSKEY *keys = (TSKEY *) pColInfoData->pData;
3525
//
L
Liu Jicong 已提交
3526 3527 3528
//   // update the offset value
//   pTableQueryInfo->lastKey = keys[pQueryAttr->pos];
//   pQueryAttr->limit.offset = 0;
3529
//
L
Liu Jicong 已提交
3530
//   int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
3531
//
L
Liu Jicong 已提交
3532 3533 3534 3535
//   //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);
// }
3536

L
Liu Jicong 已提交
3537 3538
// void skipBlocks(STaskRuntimeEnv *pRuntimeEnv) {
//   STaskAttr *pQueryAttr = pRuntimeEnv->pQueryAttr;
3539
//
L
Liu Jicong 已提交
3540 3541 3542
//   if (pQueryAttr->limit.offset <= 0 || pQueryAttr->numOfFilterCols > 0) {
//     return;
//   }
3543
//
L
Liu Jicong 已提交
3544 3545
//   pQueryAttr->pos = 0;
//   int32_t step = GET_FORWARD_DIRECTION_FACTOR(pQueryAttr->order.order);
3546
//
L
Liu Jicong 已提交
3547 3548
//   STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
//   TsdbQueryHandleT pTsdbReadHandle = pRuntimeEnv->pTsdbReadHandle;
3549
//
L
Liu Jicong 已提交
3550 3551 3552 3553 3554
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pTsdbReadHandle)) {
//     if (isTaskKilled(pRuntimeEnv->qinfo)) {
//       longjmp(pRuntimeEnv->env, TSDB_CODE_TSC_QUERY_CANCELLED);
//     }
3555
//
L
Liu Jicong 已提交
3556
//     tsdbRetrieveDataBlockInfo(pTsdbReadHandle, &blockInfo);
3557
//
L
Liu Jicong 已提交
3558 3559 3560 3561
//     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;
3562
//
L
Liu Jicong 已提交
3563 3564 3565 3566 3567 3568 3569
//       //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;
//     }
//   }
3570
//
L
Liu Jicong 已提交
3571 3572 3573 3574 3575 3576 3577 3578 3579
//   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;
3580
//
L
Liu Jicong 已提交
3581 3582 3583
//   assert(pQueryAttr->limit.offset == 0);
//   STimeWindow tw = *win;
//   getNextTimeWindow(pQueryAttr, &tw);
3584
//
L
Liu Jicong 已提交
3585 3586
//   if ((tw.skey <= pBlockInfo->window.ekey && QUERY_IS_ASC_QUERY(pQueryAttr)) ||
//       (tw.ekey >= pBlockInfo->window.skey && !QUERY_IS_ASC_QUERY(pQueryAttr))) {
3587
//
L
Liu Jicong 已提交
3588 3589 3590 3591
//     // 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);
3592
//
L
Liu Jicong 已提交
3593 3594 3595 3596
//     tw = *win;
//     int32_t startPos =
//         getNextQualifiedWindow(pQueryAttr, &tw, pBlockInfo, pColInfoData->pData, binarySearchForKey, -1);
//     assert(startPos >= 0);
3597
//
L
Liu Jicong 已提交
3598 3599
//     // set the abort info
//     pQueryAttr->pos = startPos;
3600
//
L
Liu Jicong 已提交
3601 3602 3603 3604
//     // reset the query start timestamp
//     pTableQueryInfo->win.skey = ((TSKEY *)pColInfoData->pData)[startPos];
//     pQueryAttr->window.skey = pTableQueryInfo->win.skey;
//     TSKEY key = pTableQueryInfo->win.skey;
3605
//
L
Liu Jicong 已提交
3606 3607
//     pWindowResInfo->prevSKey = tw.skey;
//     int32_t index = pRuntimeEnv->resultRowInfo.curIndex;
3608
//
L
Liu Jicong 已提交
3609 3610
//     int32_t numOfRes = tableApplyFunctionsOnBlock(pRuntimeEnv, pBlockInfo, NULL, binarySearchForKey, pDataBlock);
//     pRuntimeEnv->resultRowInfo.curIndex = index;  // restore the window index
3611
//
L
Liu Jicong 已提交
3612 3613 3614 3615
//     //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);
3616
//
L
Liu Jicong 已提交
3617 3618 3619 3620 3621
//     return key;
//   } else {  // do nothing
//     pQueryAttr->window.skey      = tw.skey;
//     pWindowResInfo->prevSKey = tw.skey;
//     pTableQueryInfo->lastKey = tw.skey;
3622
//
L
Liu Jicong 已提交
3623 3624
//     return tw.skey;
//   }
3625
//
L
Liu Jicong 已提交
3626 3627 3628 3629 3630 3631 3632 3633 3634 3635
//   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);
//   }
3636
//
L
Liu Jicong 已提交
3637 3638 3639 3640 3641
//   // 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;
//   }
3642
//
L
Liu Jicong 已提交
3643 3644 3645 3646 3647 3648 3649
//   /*
//    * 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);
3650
//
L
Liu Jicong 已提交
3651 3652
//   STimeWindow w = TSWINDOW_INITIALIZER;
//   bool ascQuery = QUERY_IS_ASC_QUERY(pQueryAttr);
3653
//
L
Liu Jicong 已提交
3654 3655
//   SResultRowInfo *pWindowResInfo = &pRuntimeEnv->resultRowInfo;
//   STableQueryInfo *pTableQueryInfo = pRuntimeEnv->current;
3656
//
L
Liu Jicong 已提交
3657 3658 3659
//   SDataBlockInfo blockInfo = SDATA_BLOCK_INITIALIZER;
//   while (tsdbNextDataBlock(pRuntimeEnv->pTsdbReadHandle)) {
//     tsdbRetrieveDataBlockInfo(pRuntimeEnv->pTsdbReadHandle, &blockInfo);
3660
//
L
Liu Jicong 已提交
3661 3662 3663 3664 3665 3666 3667 3668 3669
//     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;
//     }
3670
//
L
Liu Jicong 已提交
3671 3672
//     // the first time window
//     STimeWindow win = getActiveTimeWindow(pWindowResInfo, pWindowResInfo->prevSKey, pQueryAttr);
3673
//
L
Liu Jicong 已提交
3674 3675
//     while (pQueryAttr->limit.offset > 0) {
//       STimeWindow tw = win;
3676
//
L
Liu Jicong 已提交
3677 3678 3679
//       if ((win.ekey <= blockInfo.window.ekey && ascQuery) || (win.ekey >= blockInfo.window.skey && !ascQuery)) {
//         pQueryAttr->limit.offset -= 1;
//         pWindowResInfo->prevSKey = win.skey;
3680
//
L
Liu Jicong 已提交
3681 3682 3683 3684 3685 3686
//         // 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;
//         }
//       }
3687
//
L
Liu Jicong 已提交
3688 3689 3690 3691
//       if (pQueryAttr->limit.offset == 0) {
//         *start = doSkipIntervalProcess(pRuntimeEnv, &win, &blockInfo, pTableQueryInfo);
//         return true;
//       }
3692
//
L
Liu Jicong 已提交
3693 3694
//       // current window does not ended in current data block, try next data block
//       getNextTimeWindow(pQueryAttr, &tw);
3695
//
L
Liu Jicong 已提交
3696 3697 3698 3699 3700 3701 3702 3703 3704
//       /*
//        * 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)) {
3705
//
L
Liu Jicong 已提交
3706 3707
//         SArray *pDataBlock = tsdbRetrieveDataBlock(pRuntimeEnv->pTsdbReadHandle, NULL);
//         SColumnInfoData *pColInfoData = taosArrayGet(pDataBlock, 0);
3708
//
L
Liu Jicong 已提交
3709 3710 3711
//         if ((win.ekey > blockInfo.window.ekey && ascQuery) || (win.ekey < blockInfo.window.skey && !ascQuery)) {
//           pQueryAttr->limit.offset -= 1;
//         }
3712
//
L
Liu Jicong 已提交
3713 3714 3715 3716 3717 3718 3719 3720
//         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);
3721
//
L
Liu Jicong 已提交
3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732
//           // 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.
//       }
//     }
//   }
3733
//
L
Liu Jicong 已提交
3734 3735 3736 3737
//   // check for error
//   if (terrno != TSDB_CODE_SUCCESS) {
//     longjmp(pRuntimeEnv->env, terrno);
//   }
3738
//
L
Liu Jicong 已提交
3739 3740
//   return true;
// }
3741

3742
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num) {
H
Haojun Liao 已提交
3743
  if (p->pDownstream == NULL) {
H
Haojun Liao 已提交
3744
    assert(p->numOfDownstream == 0);
3745 3746
  }

wafwerar's avatar
wafwerar 已提交
3747
  p->pDownstream = taosMemoryCalloc(1, num * POINTER_BYTES);
3748 3749 3750 3751 3752 3753 3754
  if (p->pDownstream == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  }

  memcpy(p->pDownstream, pDownstream, num * POINTER_BYTES);
  p->numOfDownstream = num;
  return TSDB_CODE_SUCCESS;
3755 3756 3757 3758
}

static void doDestroyTableQueryInfo(STableGroupInfo* pTableqinfoGroupInfo);

3759
static void doTableQueryInfoTimeWindowCheck(SExecTaskInfo* pTaskInfo, STableQueryInfo* pTableQueryInfo, int32_t order) {
H
Haojun Liao 已提交
3760 3761
#if 0
    if (order == TSDB_ORDER_ASC) {
3762 3763
    assert(
        (pTableQueryInfo->win.skey <= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
3764 3765
        (pTableQueryInfo->lastKey >= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey >= pTaskInfo->window.skey && pTableQueryInfo->win.ekey <= pTaskInfo->window.ekey));
3766 3767 3768
  } else {
    assert(
        (pTableQueryInfo->win.skey >= pTableQueryInfo->win.ekey) &&
H
Haojun Liao 已提交
3769 3770
        (pTableQueryInfo->lastKey <= pTaskInfo->window.skey) &&
        (pTableQueryInfo->win.skey <= pTaskInfo->window.skey && pTableQueryInfo->win.ekey >= pTaskInfo->window.ekey));
3771
  }
H
Haojun Liao 已提交
3772
#endif
3773 3774
}

L
Liu Jicong 已提交
3775 3776 3777 3778 3779 3780 3781 3782
// 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,
//   };
3783
//
L
Liu Jicong 已提交
3784 3785 3786
//   TIME_WINDOW_COPY(cond.twindow, *win);
//   return cond;
// }
3787 3788 3789

static STableIdInfo createTableIdInfo(STableQueryInfo* pTableQueryInfo) {
  STableIdInfo tidInfo;
L
Liu Jicong 已提交
3790 3791 3792 3793 3794
  //  STableId* id = TSDB_TABLEID(pTableQueryInfo->pTable);
  //
  //  tidInfo.uid = id->uid;
  //  tidInfo.tid = id->tid;
  //  tidInfo.key = pTableQueryInfo->lastKey;
3795 3796 3797 3798

  return tidInfo;
}

L
Liu Jicong 已提交
3799 3800 3801 3802
// 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;
3803
//
L
Liu Jicong 已提交
3804 3805 3806
//   if (pTableQueryInfo->pTable == NULL) {
//     return;
//   }
3807
//
L
Liu Jicong 已提交
3808 3809 3810 3811 3812 3813 3814 3815 3816
//   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));
//   }
// }
3817

H
Haojun Liao 已提交
3818
static void doCloseAllTimeWindow(STaskRuntimeEnv* pRuntimeEnv) {
3819 3820 3821 3822 3823 3824 3825 3826 3827 3828 3829 3830
  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);
    }
  }
}

3831
int32_t loadRemoteDataCallback(void* param, const SDataBuf* pMsg, int32_t code) {
L
Liu Jicong 已提交
3832
  SSourceDataInfo* pSourceDataInfo = (SSourceDataInfo*)param;
H
Haojun Liao 已提交
3833 3834
  if (code == TSDB_CODE_SUCCESS) {
    pSourceDataInfo->pRsp = pMsg->pData;
3835

H
Haojun Liao 已提交
3836 3837 3838 3839 3840 3841 3842
    SRetrieveTableRsp* pRsp = pSourceDataInfo->pRsp;
    pRsp->numOfRows = htonl(pRsp->numOfRows);
    pRsp->compLen   = htonl(pRsp->compLen);
    pRsp->useconds  = htobe64(pRsp->useconds);
  } else {
    pSourceDataInfo->code = code;
  }
H
Haojun Liao 已提交
3843

H
Haojun Liao 已提交
3844
  pSourceDataInfo->status = EX_SOURCE_DATA_READY;
3845
  tsem_post(&pSourceDataInfo->pEx->ready);
wmmhello's avatar
wmmhello 已提交
3846
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
3847 3848 3849 3850
}

static void destroySendMsgInfo(SMsgSendInfo* pMsgBody) {
  assert(pMsgBody != NULL);
wafwerar's avatar
wafwerar 已提交
3851 3852
  taosMemoryFreeClear(pMsgBody->msgInfo.pData);
  taosMemoryFreeClear(pMsgBody);
H
Haojun Liao 已提交
3853 3854
}

S
Shengliang Guan 已提交
3855
void qProcessFetchRsp(void* parent, SRpcMsg* pMsg, SEpSet* pEpSet) {
L
Liu Jicong 已提交
3856
  SMsgSendInfo* pSendInfo = (SMsgSendInfo*)pMsg->ahandle;
H
Haojun Liao 已提交
3857 3858 3859 3860 3861
  assert(pMsg->ahandle != NULL);

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

  if (pMsg->contLen > 0) {
wafwerar's avatar
wafwerar 已提交
3862
    buf.pData = taosMemoryCalloc(1, pMsg->contLen);
H
Haojun Liao 已提交
3863 3864 3865 3866 3867 3868 3869 3870 3871 3872 3873
    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);
3874 3875
}

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

wafwerar's avatar
wafwerar 已提交
3879
  SResFetchReq* pMsg = taosMemoryCalloc(1, sizeof(SResFetchReq));
3880 3881 3882 3883
  if (NULL == pMsg) {
    pTaskInfo->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return pTaskInfo->code;
  }
3884

L
Liu Jicong 已提交
3885 3886
  SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, sourceIndex);
  SSourceDataInfo*       pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, sourceIndex);
3887

L
Liu Jicong 已提交
3888 3889
  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);
3890 3891 3892 3893 3894 3895 3896

  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 已提交
3897
  SMsgSendInfo* pMsgSendInfo = taosMemoryCalloc(1, sizeof(SMsgSendInfo));
3898
  if (NULL == pMsgSendInfo) {
wafwerar's avatar
wafwerar 已提交
3899
    taosMemoryFreeClear(pMsg);
3900 3901 3902
    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 已提交
3903 3904
  }

3905 3906 3907 3908 3909
  pMsgSendInfo->param = pDataInfo;
  pMsgSendInfo->msgInfo.pData = pMsg;
  pMsgSendInfo->msgInfo.len = sizeof(SResFetchReq);
  pMsgSendInfo->msgType = TDMT_VND_FETCH;
  pMsgSendInfo->fp = loadRemoteDataCallback;
3910

3911
  int64_t transporterId = 0;
L
Liu Jicong 已提交
3912
  int32_t code = asyncSendMsgToServer(pExchangeInfo->pTransporter, &pSource->addr.epSet, &transporterId, pMsgSendInfo);
3913 3914 3915
  return TSDB_CODE_SUCCESS;
}

3916 3917 3918
// TODO if only one or two columns required, how to extract data?
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 已提交
3919
  blockDataEnsureCapacity(pRes, numOfRows);
3920

H
Haojun Liao 已提交
3921 3922
  if (pColList == NULL) {  // data from other sources
    int32_t* colLen = (int32_t*)pData;
L
Liu Jicong 已提交
3923
    char*    pStart = pData + sizeof(int32_t) * numOfOutput;
H
Haojun Liao 已提交
3924

3925
    for (int32_t i = 0; i < numOfOutput; ++i) {
H
Haojun Liao 已提交
3926 3927 3928
      colLen[i] = htonl(colLen[i]);
      ASSERT(colLen[i] > 0);

3929
      SColumnInfoData* pColInfoData = taosArrayGet(pRes->pDataBlock, i);
H
Haojun Liao 已提交
3930 3931 3932
      if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
        pColInfoData->varmeta.length = colLen[i];
        pColInfoData->varmeta.allocLen = colLen[i];
3933

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

wafwerar's avatar
wafwerar 已提交
3937
        pColInfoData->pData = taosMemoryMalloc(colLen[i]);
H
Haojun Liao 已提交
3938 3939 3940
      } else {
        memcpy(pColInfoData->nullbitmap, pStart, BitmapLen(numOfRows));
        pStart += BitmapLen(numOfRows);
3941
      }
H
Haojun Liao 已提交
3942 3943 3944

      memcpy(pColInfoData->pData, pStart, colLen[i]);
      pStart += colLen[i];
3945
    }
H
Haojun Liao 已提交
3946
  } else {  // extract data according to pColList
3947
    ASSERT(numOfOutput == taosArrayGetSize(pColList));
3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981
    char* pStart = pData;

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

    SSysTableSchema* pSchema = (SSysTableSchema*)pStart;
    for(int32_t i = 0; i < numOfCols; ++i) {
      SSysTableSchema* p = (SSysTableSchema*)pStart;

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

    SSDataBlock block = {.pDataBlock = taosArrayInit(numOfCols, sizeof(SColumnInfoData)), .info.numOfCols = numOfCols};
    for(int32_t i = 0; i < numOfCols; ++i) {
      SColumnInfoData idata = {0};
      idata.info.type = pSchema[i].type;
      idata.info.bytes = pSchema[i].bytes;
      idata.info.colId = pSchema[i].colId;

      taosArrayPush(block.pDataBlock, &idata);
      if (IS_VAR_DATA_TYPE(idata.info.type)) {
        block.info.hasVarCol = true;
      }
    }

    blockDataEnsureCapacity(&block, numOfRows);

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

    for (int32_t i = 0; i < numOfCols; ++i) {
      colLen[i] = htonl(colLen[i]);
3982
      ASSERT(colLen[i] >= 0);
3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000

      SColumnInfoData* pColInfoData = taosArrayGet(block.pDataBlock, i);
      if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
        pColInfoData->varmeta.length = colLen[i];
        pColInfoData->varmeta.allocLen = colLen[i];

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

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

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

    // data from mnode
4003 4004 4005
    for (int32_t i = 0; i < numOfCols; ++i) {
      SColumnInfoData* pSrc = taosArrayGet(block.pDataBlock, i);

L
Liu Jicong 已提交
4006 4007
      for (int32_t j = 0; j < numOfOutput; ++j) {
        int16_t colIndex = *(int16_t*)taosArrayGet(pColList, j);
4008

4009 4010
        if (colIndex - 1 == i) {
          SColumnInfoData* pColInfoData = taosArrayGet(pRes->pDataBlock, j);
4011
          colDataAssign(pColInfoData, pSrc, numOfRows);
4012 4013 4014
          break;
        }
      }
4015
    }
4016
  }
4017

H
Haojun Liao 已提交
4018
  pRes->info.rows = numOfRows;
4019

4020
  int64_t el = taosGetTimestampUs() - startTs;
4021

H
Haojun Liao 已提交
4022 4023
  pLoadInfo->totalRows += numOfRows;
  pLoadInfo->totalSize += compLen;
4024

H
Haojun Liao 已提交
4025 4026 4027
  if (total != NULL) {
    *total += numOfRows;
  }
4028

H
Haojun Liao 已提交
4029
  pLoadInfo->totalElapsed += el;
4030 4031
  return TSDB_CODE_SUCCESS;
}
4032

L
Liu Jicong 已提交
4033 4034
static void* setAllSourcesCompleted(SOperatorInfo* pOperator, int64_t startTs) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
4035
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
4036

L
Liu Jicong 已提交
4037
  int64_t              el = taosGetTimestampUs() - startTs;
H
Haojun Liao 已提交
4038 4039
  SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
  pLoadInfo->totalElapsed += el;
H
Haojun Liao 已提交
4040

4041
  size_t totalSources = taosArrayGetSize(pExchangeInfo->pSources);
L
Liu Jicong 已提交
4042 4043 4044
  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);
4045 4046 4047 4048 4049

  doSetOperatorCompleted(pOperator);
  return NULL;
}

L
Liu Jicong 已提交
4050 4051
static SSDataBlock* concurrentlyLoadRemoteDataImpl(SOperatorInfo* pOperator, SExchangeInfo* pExchangeInfo,
                                                   SExecTaskInfo* pTaskInfo) {
4052 4053 4054 4055 4056 4057 4058 4059 4060
  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);

4061
      if (pDataInfo->status == EX_SOURCE_DATA_EXHAUSTED) {
4062
        completed += 1;
H
Haojun Liao 已提交
4063 4064
        continue;
      }
4065

4066
      if (pDataInfo->status != EX_SOURCE_DATA_READY) {
4067 4068 4069
        continue;
      }

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

L
Liu Jicong 已提交
4073
      SSDataBlock*         pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
4074
      SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
4075
      if (pRsp->numOfRows == 0) {
L
Liu Jicong 已提交
4076 4077
        qDebug("%s vgId:%d, taskID:0x%" PRIx64 " index:%d completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64
               " try next",
4078
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, i + 1, pDataInfo->totalRows,
H
Haojun Liao 已提交
4079
               pExchangeInfo->loadInfo.totalRows);
4080
        pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
4081 4082 4083
        completed += 1;
        continue;
      }
H
Haojun Liao 已提交
4084

H
Haojun Liao 已提交
4085
      SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
L
Liu Jicong 已提交
4086 4087 4088
      code =
          setSDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
                                    pTableRsp->compLen, pOperator->numOfOutput, startTs, &pDataInfo->totalRows, NULL);
4089
      if (code != 0) {
4090 4091 4092
        goto _error;
      }

4093 4094 4095
      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 已提交
4096 4097
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pDataInfo->totalRows,
               pLoadInfo->totalRows, pLoadInfo->totalSize, i + 1, totalSources);
4098
        pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
4099
      } else {
4100
        qDebug("%s fetch msg rsp from vgId:%d, taskId:0x%" PRIx64 " numOfRows:%d, totalRows:%" PRIu64 ", totalBytes:%" PRIu64,
H
Haojun Liao 已提交
4101 4102
               GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pRes->info.rows, pLoadInfo->totalRows,
               pLoadInfo->totalSize);
4103 4104
      }

4105 4106
      if (pDataInfo->status != EX_SOURCE_DATA_EXHAUSTED) {
        pDataInfo->status = EX_SOURCE_DATA_NOT_READY;
4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125
        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 已提交
4126 4127 4128
static SSDataBlock* concurrentlyLoadRemoteData(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
4129

H
Haojun Liao 已提交
4130 4131 4132
  if (pOperator->status == OP_RES_TO_RETURN) {
    return concurrentlyLoadRemoteDataImpl(pOperator, pExchangeInfo, pTaskInfo);
  }
4133

L
Liu Jicong 已提交
4134
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
4135 4136 4137
  int64_t startTs = taosGetTimestampUs();

  // Asynchronously send all fetch requests to all sources.
L
Liu Jicong 已提交
4138
  for (int32_t i = 0; i < totalSources; ++i) {
4139 4140
    int32_t code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
4141
      return NULL;
4142 4143 4144 4145
    }
  }

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

  tsem_wait(&pExchangeInfo->ready);
H
Haojun Liao 已提交
4150 4151
  pOperator->status = OP_RES_TO_RETURN;
  return concurrentlyLoadRemoteDataImpl(pOperator, pExchangeInfo, pTaskInfo);
4152 4153
}

L
Liu Jicong 已提交
4154 4155 4156
static int32_t prepareConcurrentlyLoad(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4157

L
Liu Jicong 已提交
4158
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
4159 4160 4161
  int64_t startTs = taosGetTimestampUs();

  // Asynchronously send all fetch requests to all sources.
L
Liu Jicong 已提交
4162
  for (int32_t i = 0; i < totalSources; ++i) {
4163 4164
    int32_t code = doSendFetchDataRequest(pExchangeInfo, pTaskInfo, i);
    if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
4165 4166
      pTaskInfo->code = code;
      return code;
4167 4168 4169 4170
    }
  }

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

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

H
Haojun Liao 已提交
4177
  return TSDB_CODE_SUCCESS;
4178 4179
}

L
Liu Jicong 已提交
4180 4181 4182
static SSDataBlock* seqLoadRemoteData(SOperatorInfo* pOperator) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4183

L
Liu Jicong 已提交
4184
  size_t  totalSources = taosArrayGetSize(pExchangeInfo->pSources);
4185
  int64_t startTs = taosGetTimestampUs();
4186

L
Liu Jicong 已提交
4187
  while (1) {
4188 4189
    if (pExchangeInfo->current >= totalSources) {
      return setAllSourcesCompleted(pOperator, startTs);
4190
    }
4191

4192 4193 4194
    doSendFetchDataRequest(pExchangeInfo, pTaskInfo, pExchangeInfo->current);
    tsem_wait(&pExchangeInfo->ready);

H
Haojun Liao 已提交
4195
    SSourceDataInfo* pDataInfo = taosArrayGet(pExchangeInfo->pSourceDataInfo, pExchangeInfo->current);
X
Xiaoyu Wang 已提交
4196
    SDownstreamSourceNode* pSource = taosArrayGet(pExchangeInfo->pSources, pExchangeInfo->current);
4197

H
Haojun Liao 已提交
4198 4199 4200 4201 4202 4203 4204
    if (pDataInfo->code != TSDB_CODE_SUCCESS) {
      qError("%s vgId:%d, taskID:0x%" PRIx64 " error happens, code:%s",
             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, tstrerror(pDataInfo->code));
      pOperator->pTaskInfo->code = pDataInfo->code;
      return NULL;
    }

L
Liu Jicong 已提交
4205
    SRetrieveTableRsp*   pRsp = pDataInfo->pRsp;
H
Haojun Liao 已提交
4206
    SLoadRemoteDataInfo* pLoadInfo = &pExchangeInfo->loadInfo;
4207
    if (pRsp->numOfRows == 0) {
H
Haojun Liao 已提交
4208
      qDebug("%s vgId:%d, taskID:0x%" PRIx64 " %d of total completed, rowsOfSource:%" PRIu64 ", totalRows:%" PRIu64 " try next",
4209
             GET_TASKID(pTaskInfo), pSource->addr.nodeId, pSource->taskId, pExchangeInfo->current + 1,
H
Haojun Liao 已提交
4210
             pDataInfo->totalRows, pLoadInfo->totalRows);
H
Haojun Liao 已提交
4211

4212
      pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
4213 4214 4215
      pExchangeInfo->current += 1;
      continue;
    }
H
Haojun Liao 已提交
4216

L
Liu Jicong 已提交
4217
    SSDataBlock*       pRes = pExchangeInfo->pResult;
H
Haojun Liao 已提交
4218
    SRetrieveTableRsp* pTableRsp = pDataInfo->pRsp;
L
Liu Jicong 已提交
4219 4220 4221
    int32_t            code =
        setSDataBlockFromFetchRsp(pExchangeInfo->pResult, pLoadInfo, pTableRsp->numOfRows, pTableRsp->data,
                                  pTableRsp->compLen, pOperator->numOfOutput, startTs, &pDataInfo->totalRows, NULL);
4222 4223

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

4229
      pDataInfo->status = EX_SOURCE_DATA_EXHAUSTED;
4230 4231
      pExchangeInfo->current += 1;
    } else {
L
Liu Jicong 已提交
4232 4233 4234 4235
      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);
4236 4237 4238 4239
    }

    return pExchangeInfo->pResult;
  }
4240 4241
}

L
Liu Jicong 已提交
4242
static int32_t prepareLoadRemoteData(SOperatorInfo* pOperator) {
4243
  if (OPTR_IS_OPENED(pOperator)) {
H
Haojun Liao 已提交
4244 4245 4246
    return TSDB_CODE_SUCCESS;
  }

L
Liu Jicong 已提交
4247
  SExchangeInfo* pExchangeInfo = pOperator->info;
H
Haojun Liao 已提交
4248 4249 4250 4251 4252 4253 4254 4255 4256
  if (pExchangeInfo->seqLoadData) {
    // do nothing for sequentially load data
  } else {
    int32_t code = prepareConcurrentlyLoad(pOperator);
    if (code != TSDB_CODE_SUCCESS) {
      return code;
    }
  }

4257
  OPTR_SET_OPENED(pOperator);
H
Haojun Liao 已提交
4258 4259 4260
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
4261 4262 4263
static SSDataBlock* doLoadRemoteData(SOperatorInfo* pOperator, bool* newgroup) {
  SExchangeInfo* pExchangeInfo = pOperator->info;
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
4264

4265 4266
  pTaskInfo->code = pOperator->_openFn(pOperator);
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
4267 4268
    return NULL;
  }
4269

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

4273
  if (pOperator->status == OP_EXEC_DONE) {
L
Liu Jicong 已提交
4274 4275 4276
    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);
4277 4278 4279 4280
    return NULL;
  }

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

4282 4283 4284
  if (pExchangeInfo->seqLoadData) {
    return seqLoadRemoteData(pOperator);
  } else {
H
Haojun Liao 已提交
4285
    return concurrentlyLoadRemoteData(pOperator);
4286
  }
H
Haojun Liao 已提交
4287

4288
#if 0
H
Haojun Liao 已提交
4289
  _error:
wafwerar's avatar
wafwerar 已提交
4290 4291
  taosMemoryFreeClear(pMsg);
  taosMemoryFreeClear(pMsgSendInfo);
H
Haojun Liao 已提交
4292 4293 4294

  terrno = pTaskInfo->code;
  return NULL;
4295
#endif
H
Haojun Liao 已提交
4296
}
4297

H
Haojun Liao 已提交
4298
static int32_t initDataSource(int32_t numOfSources, SExchangeInfo* pInfo) {
4299
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
H
Haojun Liao 已提交
4300 4301
  if (pInfo->pSourceDataInfo == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
4302 4303
  }

L
Liu Jicong 已提交
4304
  for (int32_t i = 0; i < numOfSources; ++i) {
4305
    SSourceDataInfo dataInfo = {0};
H
Haojun Liao 已提交
4306
    dataInfo.status = EX_SOURCE_DATA_NOT_READY;
L
Liu Jicong 已提交
4307 4308
    dataInfo.pEx = pInfo;
    dataInfo.index = i;
4309

H
Haojun Liao 已提交
4310 4311 4312 4313 4314 4315 4316 4317 4318 4319
    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 已提交
4320
SOperatorInfo* createExchangeOperatorInfo(const SNodeList* pSources, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo) {
L
Liu Jicong 已提交
4321
  SExchangeInfo* pInfo = taosMemoryCalloc(1, sizeof(SExchangeInfo));
wafwerar's avatar
wafwerar 已提交
4322
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
4323 4324

  if (pInfo == NULL || pOperator == NULL) {
wafwerar's avatar
wafwerar 已提交
4325 4326
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
4327 4328
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
H
Haojun Liao 已提交
4329 4330
  }

H
Haojun Liao 已提交
4331
  size_t numOfSources = LIST_LENGTH(pSources);
H
Haojun Liao 已提交
4332
  pInfo->pSources = taosArrayInit(numOfSources, sizeof(SDownstreamSourceNode));
H
Haojun Liao 已提交
4333 4334 4335
  pInfo->pSourceDataInfo = taosArrayInit(numOfSources, sizeof(SSourceDataInfo));
  if (pInfo->pSourceDataInfo == NULL || pInfo->pSources == NULL) {
    goto _error;
H
Haojun Liao 已提交
4336 4337
  }

L
Liu Jicong 已提交
4338 4339
  for (int32_t i = 0; i < numOfSources; ++i) {
    SNodeListNode* pNode = nodesListGetNode((SNodeList*)pSources, i);
H
Haojun Liao 已提交
4340 4341
    taosArrayPush(pInfo->pSources, pNode);
  }
4342

H
Haojun Liao 已提交
4343 4344 4345
  int32_t code = initDataSource(numOfSources, pInfo);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
4346
  }
H
Haojun Liao 已提交
4347

4348
  pInfo->pResult     = pBlock;
4349 4350 4351
  pInfo->seqLoadData = true;

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

4353
  pOperator->name         = "ExchangeOperator";
X
Xiaoyu Wang 已提交
4354
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_EXCHANGE;
4355
  pOperator->blockingOptr = false;
4356 4357 4358 4359 4360 4361 4362
  pOperator->status       = OP_NOT_OPENED;
  pOperator->info         = pInfo;
  pOperator->numOfOutput  = pBlock->info.numOfCols;
  pOperator->pTaskInfo    = pTaskInfo;
  pOperator->_openFn      = prepareLoadRemoteData;  // assign a dummy function.
  pOperator->getNextFn    = doLoadRemoteData;
  pOperator->closeFn      = destroyExchangeOperatorInfo;
4363

S
Shengliang Guan 已提交
4364
#if 1
L
Liu Jicong 已提交
4365
  {  // todo refactor
H
Haojun Liao 已提交
4366 4367 4368
    SRpcInit rpcInit;
    memset(&rpcInit, 0, sizeof(rpcInit));
    rpcInit.localPort = 0;
H
Haojun Liao 已提交
4369
    rpcInit.label = "EX";
H
Haojun Liao 已提交
4370
    rpcInit.numOfThreads = 1;
S
Shengliang Guan 已提交
4371
    rpcInit.cfp = qProcessFetchRsp;
S
Shengliang Guan 已提交
4372
    rpcInit.sessions = tsMaxConnections;
H
Haojun Liao 已提交
4373
    rpcInit.connType = TAOS_CONN_CLIENT;
L
Liu Jicong 已提交
4374
    rpcInit.user = (char*)"root";
S
Shengliang Guan 已提交
4375
    rpcInit.idleTime = tsShellActivityTimer * 1000;
H
Haojun Liao 已提交
4376
    rpcInit.ckey = "key";
S
Shengliang Guan 已提交
4377
    rpcInit.spi = 1;
L
Liu Jicong 已提交
4378
    rpcInit.secret = (char*)"dcc5bed04851fec854c035b2e40263b6";
H
Haojun Liao 已提交
4379 4380 4381

    pInfo->pTransporter = rpcOpen(&rpcInit);
    if (pInfo->pTransporter == NULL) {
L
Liu Jicong 已提交
4382
      return NULL;  // todo
H
Haojun Liao 已提交
4383 4384
    }
  }
S
Shengliang 已提交
4385
#endif
4386

4387
  return pOperator;
H
Haojun Liao 已提交
4388

L
Liu Jicong 已提交
4389
_error:
H
Haojun Liao 已提交
4390
  if (pInfo != NULL) {
H
Haojun Liao 已提交
4391
    destroyExchangeOperatorInfo(pInfo, numOfSources);
H
Haojun Liao 已提交
4392 4393
  }

wafwerar's avatar
wafwerar 已提交
4394 4395
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
4396
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
H
Haojun Liao 已提交
4397
  return NULL;
4398 4399
}

4400
static int32_t doInitAggInfoSup(SAggSupporter* pAggSup, SqlFunctionCtx* pCtx, int32_t numOfOutput, size_t keyBufSize, const char* pKey);
H
Haojun Liao 已提交
4401
static void    cleanupAggSup(SAggSupporter* pAggSup);
4402

4403
static void destroySortedMergeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
4404
  SSortedMergeOperatorInfo* pInfo = (SSortedMergeOperatorInfo*)param;
H
Haojun Liao 已提交
4405
  taosArrayDestroy(pInfo->pSortInfo);
4406 4407 4408
  taosArrayDestroy(pInfo->groupInfo);

  if (pInfo->pSortHandle != NULL) {
H
Haojun Liao 已提交
4409
    tsortDestroySortHandle(pInfo->pSortHandle);
4410 4411
  }

H
Haojun Liao 已提交
4412
  blockDataDestroy(pInfo->binfo.pRes);
H
Haojun Liao 已提交
4413
  cleanupAggSup(&pInfo->aggSup);
4414
}
H
Haojun Liao 已提交
4415

X
Xiaoyu Wang 已提交
4416 4417
static void assignExprInfo(SExprInfo* dst, const SExprInfo* src) {
  assert(dst != NULL && src != NULL);
4418

X
Xiaoyu Wang 已提交
4419
  *dst = *src;
4420

X
Xiaoyu Wang 已提交
4421
  dst->pExpr = exprdup(src->pExpr);
wafwerar's avatar
wafwerar 已提交
4422
  dst->base.pParam = taosMemoryCalloc(src->base.numOfParams, sizeof(SColumn));
H
Haojun Liao 已提交
4423
  memcpy(dst->base.pParam, src->base.pParam, sizeof(SColumn) * src->base.numOfParams);
4424

L
Liu Jicong 已提交
4425 4426 4427 4428
  //  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 已提交
4429
}
4430

4431 4432
static SExprInfo* exprArrayDup(SArray* pExprList) {
  size_t numOfOutput = taosArrayGetSize(pExprList);
4433

wafwerar's avatar
wafwerar 已提交
4434
  SExprInfo* p = taosMemoryCalloc(numOfOutput, sizeof(SExprInfo));
4435 4436
  for (int32_t i = 0; i < numOfOutput; ++i) {
    SExprInfo* pExpr = taosArrayGetP(pExprList, i);
H
Haojun Liao 已提交
4437
    assignExprInfo(&p[i], pExpr);
4438 4439
  }

H
Haojun Liao 已提交
4440 4441
  return p;
}
4442

H
Haojun Liao 已提交
4443
// TODO merge aggregate super table
L
Liu Jicong 已提交
4444
static void appendOneRowToDataBlock(SSDataBlock* pBlock, STupleHandle* pTupleHandle) {
4445 4446
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
    SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, i);
4447

H
Haojun Liao 已提交
4448
    bool isNull = tsortIsNullVal(pTupleHandle, i);
H
Haojun Liao 已提交
4449 4450 4451
    if (isNull) {
      colDataAppend(pColInfo, pBlock->info.rows, NULL, true);
    } else {
H
Haojun Liao 已提交
4452
      char* pData = tsortGetValue(pTupleHandle, i);
H
Haojun Liao 已提交
4453 4454
      colDataAppend(pColInfo, pBlock->info.rows, pData, false);
    }
4455 4456
  }

4457 4458
  pBlock->info.rows += 1;
}
4459

H
Haojun Liao 已提交
4460
SSDataBlock* getSortedBlockData(SSortHandle* pHandle, SSDataBlock* pDataBlock, int32_t capacity) {
4461
  blockDataCleanup(pDataBlock);
D
dapan1121 已提交
4462
  blockDataEnsureCapacity(pDataBlock, capacity);
4463

wmmhello's avatar
wmmhello 已提交
4464 4465
  blockDataEnsureCapacity(pDataBlock, capacity);

L
Liu Jicong 已提交
4466
  while (1) {
H
Haojun Liao 已提交
4467
    STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
4468 4469 4470
    if (pTupleHandle == NULL) {
      break;
    }
4471

4472 4473 4474
    appendOneRowToDataBlock(pDataBlock, pTupleHandle);
    if (pDataBlock->info.rows >= capacity) {
      return pDataBlock;
H
Haojun Liao 已提交
4475 4476
    }
  }
4477

L
Liu Jicong 已提交
4478
  return (pDataBlock->info.rows > 0) ? pDataBlock : NULL;
4479
}
4480

4481
SSDataBlock* loadNextDataBlock(void* param) {
L
Liu Jicong 已提交
4482 4483
  SOperatorInfo* pOperator = (SOperatorInfo*)param;
  bool           newgroup = false;
H
Haojun Liao 已提交
4484
  return pOperator->getNextFn(pOperator, &newgroup);
4485 4486
}

L
Liu Jicong 已提交
4487
static bool needToMerge(SSDataBlock* pBlock, SArray* groupInfo, char** buf, int32_t rowIndex) {
4488 4489 4490 4491
  size_t size = taosArrayGetSize(groupInfo);
  if (size == 0) {
    return true;
  }
4492

4493 4494
  for (int32_t i = 0; i < size; ++i) {
    int32_t* index = taosArrayGet(groupInfo, i);
4495

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

4499 4500 4501
    if ((isNull && buf[i] != NULL) || (!isNull && buf[i] == NULL)) {
      return false;
    }
4502

4503 4504 4505 4506 4507 4508 4509 4510 4511 4512 4513 4514 4515
    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;
      }
4516 4517 4518
    }
  }

4519
  return 0;
4520 4521
}

L
Liu Jicong 已提交
4522 4523 4524
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
4525
    pCtx[j].startRow = rowIndex;
4526 4527
  }

4528 4529
  for (int32_t j = 0; j < numOfExpr; ++j) {
    int32_t functionId = pCtx[j].functionId;
L
Liu Jicong 已提交
4530 4531 4532 4533 4534 4535 4536 4537 4538
    //    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]);
    //    }
4539
  }
4540
}
4541

L
Liu Jicong 已提交
4542 4543
static void doFinalizeResultImpl(SqlFunctionCtx* pCtx, int32_t numOfExpr) {
  for (int32_t j = 0; j < numOfExpr; ++j) {
4544 4545 4546 4547
    int32_t functionId = pCtx[j].functionId;
    //    if (functionId == FUNC_TAG_DUMMY || functionId == FUNC_TS_DUMMY) {
    //      continue;
    //    }
4548

4549 4550 4551 4552
    //    if (functionId < 0) {
    //      SUdfInfo* pUdfInfo = taosArrayGet(pInfo->udfInfo, -1 * functionId - 1);
    //      doInvokeUdf(pUdfInfo, &pCtx[j], 0, TSDB_UDF_FUNC_FINALIZE);
    //    } else {
H
Haojun Liao 已提交
4553
    pCtx[j].fpSet.finalize(&pCtx[j]);
4554 4555
  }
}
4556

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

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

4564 4565 4566
    char* data = colDataGetData(pColInfo, rowIndex);
    memcpy(rowColData[i], data, colDataGetLength(pColInfo, rowIndex));
  }
4567

4568 4569
  return true;
}
4570

4571 4572
static void doMergeImpl(SOperatorInfo* pOperator, int32_t numOfExpr, SSDataBlock* pBlock) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
4573

4574
  SqlFunctionCtx* pCtx = pInfo->binfo.pCtx;
L
Liu Jicong 已提交
4575
  for (int32_t i = 0; i < pBlock->info.numOfCols; ++i) {
4576 4577
    pCtx[i].size = 1;
  }
4578

L
Liu Jicong 已提交
4579
  for (int32_t i = 0; i < pBlock->info.rows; ++i) {
4580 4581 4582 4583 4584 4585 4586 4587 4588
    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 已提交
4589
        int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfOutput, NULL);
4590
        //        setTagValueForMultipleRows(pCtx, pOperator->numOfOutput, numOfRows);
4591

4592
        // TODO check for available buffer;
H
Haojun Liao 已提交
4593

4594 4595 4596 4597 4598
        // next group info data
        pInfo->binfo.pRes->info.rows += numOfRows;
        for (int32_t j = 0; j < numOfExpr; ++j) {
          if (pCtx[j].functionId < 0) {
            continue;
4599
          }
4600

H
Haojun Liao 已提交
4601
          pCtx[j].fpSet.process(&pCtx[j]);
4602
        }
4603 4604 4605

        doMergeResultImpl(pInfo, pCtx, numOfExpr, i);
        pInfo->hasGroupVal = saveCurrentTuple(pInfo->groupVal, pInfo->groupInfo, pBlock, i);
H
Haojun Liao 已提交
4606
      }
4607 4608 4609 4610
    }
  }
}

4611 4612
static SSDataBlock* doMerge(SOperatorInfo* pOperator) {
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
4613
  SSortHandle*              pHandle = pInfo->pSortHandle;
4614

4615 4616 4617
  SSDataBlock* pDataBlock = createOneDataBlock(pInfo->binfo.pRes);
  blockDataEnsureCapacity(pDataBlock, pInfo->binfo.capacity);

L
Liu Jicong 已提交
4618
  while (1) {
4619
    blockDataCleanup(pDataBlock);
4620
    while (1) {
H
Haojun Liao 已提交
4621
      STupleHandle* pTupleHandle = tsortNextTuple(pHandle);
4622 4623
      if (pTupleHandle == NULL) {
        break;
4624
      }
4625

4626 4627 4628 4629 4630
      // build datablock for merge for one group
      appendOneRowToDataBlock(pDataBlock, pTupleHandle);
      if (pDataBlock->info.rows >= pInfo->binfo.capacity) {
        break;
      }
4631
    }
4632

4633 4634 4635
    if (pDataBlock->info.rows == 0) {
      break;
    }
4636

4637
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pDataBlock, TSDB_ORDER_ASC);
L
Liu Jicong 已提交
4638 4639
    //  updateOutputBuf(&pInfo->binfo, &pAggInfo->bufCapacity, pBlock->info.rows * pAggInfo->resultRowFactor,
    //  pOperator->pRuntimeEnv, true);
4640 4641 4642
    doMergeImpl(pOperator, pOperator->numOfOutput, pDataBlock);
    // flush to tuple store, and after all data have been handled, return to upstream node or sink node
  }
4643

4644
  doFinalizeResultImpl(pInfo->binfo.pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
4645
  int32_t numOfRows = getNumOfResult(pInfo->binfo.pCtx, pOperator->numOfOutput, NULL);
4646
  //        setTagValueForMultipleRows(pCtx, pOperator->numOfOutput, numOfRows);
4647

4648
  // TODO check for available buffer;
4649

4650 4651
  // next group info data
  pInfo->binfo.pRes->info.rows += numOfRows;
L
Liu Jicong 已提交
4652
  return (pInfo->binfo.pRes->info.rows > 0) ? pInfo->binfo.pRes : NULL;
4653
}
4654

L
Liu Jicong 已提交
4655
static SSDataBlock* doSortedMerge(SOperatorInfo* pOperator, bool* newgroup) {
4656 4657
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
4658 4659
  }

L
Liu Jicong 已提交
4660
  SExecTaskInfo*            pTaskInfo = pOperator->pTaskInfo;
4661
  SSortedMergeOperatorInfo* pInfo = pOperator->info;
H
Haojun Liao 已提交
4662
  if (pOperator->status == OP_RES_TO_RETURN) {
H
Haojun Liao 已提交
4663
    return getSortedBlockData(pInfo->pSortHandle, pInfo->binfo.pRes, pInfo->binfo.capacity);
4664 4665
  }

4666
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
4667
  pInfo->pSortHandle = tsortCreateSortHandle(pInfo->pSortInfo, NULL, SORT_MULTISOURCE_MERGE, pInfo->bufPageSize, numOfBufPage,
L
Liu Jicong 已提交
4668
                                             pInfo->binfo.pRes, "GET_TASKID(pTaskInfo)");
H
Haojun Liao 已提交
4669

H
Haojun Liao 已提交
4670
  tsortSetFetchRawDataFp(pInfo->pSortHandle, loadNextDataBlock);
4671

L
Liu Jicong 已提交
4672
  for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
wmmhello's avatar
wmmhello 已提交
4673
    SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
H
Haojun Liao 已提交
4674
    ps->param = pOperator->pDownstream[i];
H
Haojun Liao 已提交
4675
    tsortAddSource(pInfo->pSortHandle, ps);
4676 4677
  }

H
Haojun Liao 已提交
4678
  int32_t code = tsortOpen(pInfo->pSortHandle);
4679
  if (code != TSDB_CODE_SUCCESS) {
4680
    longjmp(pTaskInfo->env, terrno);
4681 4682
  }

H
Haojun Liao 已提交
4683
  pOperator->status = OP_RES_TO_RETURN;
4684
  return doMerge(pOperator);
4685
}
4686

L
Liu Jicong 已提交
4687 4688
static int32_t initGroupCol(SExprInfo* pExprInfo, int32_t numOfCols, SArray* pGroupInfo,
                            SSortedMergeOperatorInfo* pInfo) {
4689 4690
  if (pGroupInfo == NULL || taosArrayGetSize(pGroupInfo) == 0) {
    return 0;
H
Haojun Liao 已提交
4691 4692
  }

4693 4694 4695 4696 4697 4698 4699 4700
  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 已提交
4701 4702
  size_t numOfGroupCol = taosArrayGetSize(pInfo->groupInfo);
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
4703
    SColumn* pCol = taosArrayGet(pGroupInfo, i);
L
Liu Jicong 已提交
4704
    for (int32_t j = 0; j < numOfCols; ++j) {
H
Haojun Liao 已提交
4705
      SExprInfo* pe = &pExprInfo[j];
4706
      if (pe->base.resSchema.slotId == pCol->colId) {
4707 4708
        taosArrayPush(plist, pCol);
        taosArrayPush(pInfo->groupInfo, &j);
H
Haojun Liao 已提交
4709
        len += pCol->bytes;
4710 4711
        break;
      }
H
Haojun Liao 已提交
4712 4713 4714
    }
  }

4715
  ASSERT(taosArrayGetSize(pGroupInfo) == taosArrayGetSize(plist));
H
Haojun Liao 已提交
4716

wafwerar's avatar
wafwerar 已提交
4717
  pInfo->groupVal = taosMemoryCalloc(1, (POINTER_BYTES * numOfGroupCol + len));
4718 4719 4720 4721
  if (pInfo->groupVal == NULL) {
    taosArrayDestroy(plist);
    return TSDB_CODE_OUT_OF_MEMORY;
  }
H
Haojun Liao 已提交
4722

4723
  int32_t offset = 0;
L
Liu Jicong 已提交
4724 4725
  char*   start = (char*)(pInfo->groupVal + (POINTER_BYTES * numOfGroupCol));
  for (int32_t i = 0; i < numOfGroupCol; ++i) {
4726 4727
    pInfo->groupVal[i] = start + offset;
    SColumn* pCol = taosArrayGet(plist, i);
H
Haojun Liao 已提交
4728
    offset += pCol->bytes;
4729
  }
H
Haojun Liao 已提交
4730

4731
  taosArrayDestroy(plist);
H
Haojun Liao 已提交
4732

4733 4734
  return TSDB_CODE_SUCCESS;
}
H
Haojun Liao 已提交
4735

L
Liu Jicong 已提交
4736 4737 4738
SOperatorInfo* createSortedMergeOperatorInfo(SOperatorInfo** downstream, int32_t numOfDownstream, SExprInfo* pExprInfo,
                                             int32_t num, SArray* pSortInfo, SArray* pGroupInfo,
                                             SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
4739
  SSortedMergeOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSortedMergeOperatorInfo));
L
Liu Jicong 已提交
4740
  SOperatorInfo*            pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
4741
  if (pInfo == NULL || pOperator == NULL) {
4742
    goto _error;
4743
  }
H
Haojun Liao 已提交
4744

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

4748 4749 4750
  if (pInfo->binfo.pCtx == NULL || pInfo->binfo.pRes == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
4751

4752 4753
  size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
  int32_t code = doInitAggInfoSup(&pInfo->aggSup, pInfo->binfo.pCtx, num, keyBufSize, pTaskInfo->id.str);
4754 4755 4756
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
4757

H
Haojun Liao 已提交
4758
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, pTaskInfo);
H
Haojun Liao 已提交
4759
  code = initGroupCol(pExprInfo, num, pGroupInfo, pInfo);
4760 4761 4762
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
H
Haojun Liao 已提交
4763

L
Liu Jicong 已提交
4764 4765 4766 4767 4768
  //  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 已提交
4769

4770
  pInfo->binfo.capacity = blockDataGetCapacityInRow(pInfo->binfo.pRes, pInfo->bufPageSize);
H
Haojun Liao 已提交
4771

L
Liu Jicong 已提交
4772
  pOperator->name = "SortedMerge";
X
Xiaoyu Wang 已提交
4773
  // pOperator->operatorType = OP_SortedMerge;
4774
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
4775 4776 4777 4778
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->numOfOutput = num;
  pOperator->pExpr = pExprInfo;
H
Haojun Liao 已提交
4779

L
Liu Jicong 已提交
4780 4781 4782
  pOperator->pTaskInfo = pTaskInfo;
  pOperator->getNextFn = doSortedMerge;
  pOperator->closeFn = destroySortedMergeOperatorInfo;
H
Haojun Liao 已提交
4783

4784 4785 4786
  code = appendDownstream(pOperator, downstream, numOfDownstream);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
4787
  }
H
Haojun Liao 已提交
4788

4789
  return pOperator;
H
Haojun Liao 已提交
4790

L
Liu Jicong 已提交
4791
_error:
4792
  if (pInfo != NULL) {
H
Haojun Liao 已提交
4793
    destroySortedMergeOperatorInfo(pInfo, num);
H
Haojun Liao 已提交
4794 4795
  }

wafwerar's avatar
wafwerar 已提交
4796 4797
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
4798 4799
  terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
  return NULL;
H
Haojun Liao 已提交
4800 4801
}

L
Liu Jicong 已提交
4802
static SSDataBlock* doSort(SOperatorInfo* pOperator, bool* newgroup) {
4803 4804 4805 4806
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

L
Liu Jicong 已提交
4807
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
4808 4809
  SSortOperatorInfo* pInfo = pOperator->info;

H
Haojun Liao 已提交
4810
  if (pOperator->status == OP_RES_TO_RETURN) {
H
Haojun Liao 已提交
4811
    return getSortedBlockData(pInfo->pSortHandle, pInfo->pDataBlock, pInfo->numOfRowsInRes);
H
Haojun Liao 已提交
4812 4813
  }

4814
  int32_t numOfBufPage = pInfo->sortBufSize / pInfo->bufPageSize;
4815
  pInfo->pSortHandle = tsortCreateSortHandle(pInfo->pSortInfo, pInfo->inputSlotMap, SORT_SINGLESOURCE_SORT, pInfo->bufPageSize, numOfBufPage,
H
Haojun Liao 已提交
4816
                                             pInfo->pDataBlock, pTaskInfo->id.str);
4817

H
Haojun Liao 已提交
4818
  tsortSetFetchRawDataFp(pInfo->pSortHandle, loadNextDataBlock);
H
Haojun Liao 已提交
4819

wmmhello's avatar
wmmhello 已提交
4820
  SSortSource* ps = taosMemoryCalloc(1, sizeof(SSortSource));
H
Haojun Liao 已提交
4821
  ps->param = pOperator->pDownstream[0];
H
Haojun Liao 已提交
4822
  tsortAddSource(pInfo->pSortHandle, ps);
4823

H
Haojun Liao 已提交
4824
  int32_t code = tsortOpen(pInfo->pSortHandle);
4825
  taosMemoryFreeClear(ps);
4826
  if (code != TSDB_CODE_SUCCESS) {
4827
    longjmp(pTaskInfo->env, terrno);
4828
  }
4829

H
Haojun Liao 已提交
4830
  pOperator->status = OP_RES_TO_RETURN;
H
Haojun Liao 已提交
4831
  return getSortedBlockData(pInfo->pSortHandle, pInfo->pDataBlock, pInfo->numOfRowsInRes);
4832 4833
}

4834
SOperatorInfo* createSortOperatorInfo(SOperatorInfo* downstream, SSDataBlock* pResBlock, SArray* pSortInfo, SArray* pIndexMap, SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
4835
  SSortOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSortOperatorInfo));
L
Liu Jicong 已提交
4836
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
4837 4838 4839
  int32_t rowSize = pResBlock->info.rowSize;

  if (pInfo == NULL || pOperator == NULL || rowSize > 100 * 1024 * 1024) {
wafwerar's avatar
wafwerar 已提交
4840 4841
    taosMemoryFreeClear(pInfo);
    taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
4842 4843 4844
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
  }
4845

wmmhello's avatar
wmmhello 已提交
4846
  pInfo->bufPageSize = rowSize < 1024 ? 1024*2 : rowSize*2; // there are headers, so pageSize = rowSize + header
4847

wmmhello's avatar
wmmhello 已提交
4848
  pInfo->sortBufSize = pInfo->bufPageSize * 16;  // TODO dynamic set the available sort buffer
L
Liu Jicong 已提交
4849 4850 4851
  pInfo->numOfRowsInRes = 1024;
  pInfo->pDataBlock = pResBlock;
  pInfo->pSortInfo = pSortInfo;
4852
  pInfo->inputSlotMap = pIndexMap;
H
Haojun Liao 已提交
4853

H
Haojun Liao 已提交
4854
  pOperator->name         = "SortOperator";
L
Liu Jicong 已提交
4855 4856
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SORT;
  pOperator->blockingOptr = true;
H
Haojun Liao 已提交
4857 4858
  pOperator->status       = OP_NOT_OPENED;
  pOperator->info         = pInfo;
4859

H
Haojun Liao 已提交
4860 4861 4862
  pOperator->pTaskInfo    = pTaskInfo;
  pOperator->getNextFn    = doSort;
  pOperator->closeFn      = destroyOrderOperatorInfo;
4863

4864
  int32_t code = appendDownstream(pOperator, &downstream, 1);
4865
  return pOperator;
H
Haojun Liao 已提交
4866 4867 4868 4869 4870 4871

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

L
Liu Jicong 已提交
4874
static int32_t getTableScanOrder(STableScanInfo* pTableScanInfo) { return pTableScanInfo->order; }
4875 4876

// this is a blocking operator
L
Liu Jicong 已提交
4877
static int32_t doOpenAggregateOptr(SOperatorInfo* pOperator) {
H
Haojun Liao 已提交
4878 4879
  if (OPTR_IS_OPENED(pOperator)) {
    return TSDB_CODE_SUCCESS;
4880 4881 4882
  }

  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
4883
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;
4884

H
Haojun Liao 已提交
4885
  int32_t        order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
4886
  SOperatorInfo* downstream = pOperator->pDownstream[0];
4887

H
Haojun Liao 已提交
4888 4889
  bool newgroup = true;
  while (1) {
H
Haojun Liao 已提交
4890
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
4891
    SSDataBlock* pBlock = downstream->getNextFn(downstream, &newgroup);
H
Haojun Liao 已提交
4892
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
4893 4894 4895 4896

    if (pBlock == NULL) {
      break;
    }
H
Haojun Liao 已提交
4897 4898 4899
    //    if (pAggInfo->current != NULL) {
    //      setTagValue(pOperator, pAggInfo->current->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    }
4900 4901 4902

    // the pDataBlock are always the same one, no need to call this again
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order);
H
Haojun Liao 已提交
4903
    doAggregateImpl(pOperator, 0, pInfo->pCtx);
4904

4905
#if 0   // test for encode/decode result info
4906 4907 4908 4909 4910 4911 4912 4913 4914 4915 4916
    if(pOperator->encodeResultRow){
      char *result = NULL;
      int32_t length = 0;
      SAggSupporter   *pSup = &pAggInfo->aggSup;
      pOperator->encodeResultRow(pOperator, pSup, pInfo, &result, &length);
      taosHashClear(pSup->pResultRowHashTable);
      pInfo->resultRowInfo.size = 0;
      pOperator->decodeResultRow(pOperator, pSup, pInfo, result, length);
      if(result){
        taosMemoryFree(result);
      }
4917
    }
4918
#endif
4919 4920
  }

4921
  finalizeQueryResult(pInfo->pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
4922 4923 4924 4925 4926

  OPTR_SET_OPENED(pOperator);
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
4927 4928
static SSDataBlock* getAggregateResult(SOperatorInfo* pOperator, bool* newgroup) {
  SAggOperatorInfo* pAggInfo = pOperator->info;
H
Haojun Liao 已提交
4929 4930 4931 4932 4933 4934
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;

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

L
Liu Jicong 已提交
4935
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
H
Haojun Liao 已提交
4936 4937 4938 4939 4940
  pTaskInfo->code = pOperator->_openFn(pOperator);
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
    return NULL;
  }

H
Haojun Liao 已提交
4941
  getNumOfResult(pInfo->pCtx, pOperator->numOfOutput, pInfo->pRes);
H
Haojun Liao 已提交
4942
  doSetOperatorCompleted(pOperator);
4943

L
Liu Jicong 已提交
4944
  return (blockDataGetNumOfRows(pInfo->pRes) != 0) ? pInfo->pRes : NULL;
4945 4946
}

4947
void aggEncodeResultRow(SOperatorInfo* pOperator, SAggSupporter *pSup, SOptrBasicInfo *pInfo, char **result, int32_t *length) {
wmmhello's avatar
wmmhello 已提交
4948
  int32_t size = taosHashGetSize(pSup->pResultRowHashTable);
4949
  size_t  keyLen = sizeof(uint64_t) * 2;  // estimate the key length
wmmhello's avatar
wmmhello 已提交
4950
  int32_t totalSize = sizeof(int32_t) + size * (sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize);
wafwerar's avatar
wafwerar 已提交
4951
  *result = taosMemoryCalloc(1, totalSize);
L
Liu Jicong 已提交
4952
  if (*result == NULL) {
4953
    longjmp(pOperator->pTaskInfo->env, TSDB_CODE_OUT_OF_MEMORY);
wmmhello's avatar
wmmhello 已提交
4954 4955 4956
  }
  *(int32_t*)(*result) = size;
  int32_t offset = sizeof(int32_t);
4957 4958 4959 4960 4961 4962 4963 4964

  // prepare memory
  SResultRowPosition* pos = &pInfo->resultRowInfo.pPosition[pInfo->resultRowInfo.curPos];
  void* pPage = getBufPage(pSup->pResultBuf, pos->pageId);
  SResultRow* pRow = (SResultRow*)((char*)pPage + pos->offset);
  setBufPageDirty(pPage, true);
  releaseBufPage(pSup->pResultBuf, pPage);

L
Liu Jicong 已提交
4965
  void*   pIter = taosHashIterate(pSup->pResultRowHashTable, NULL);
wmmhello's avatar
wmmhello 已提交
4966
  while (pIter) {
L
Liu Jicong 已提交
4967
    void*        key = taosHashGetKey(pIter, &keyLen);
4968
    SResultRowPosition* p1 = (SResultRowPosition*)pIter;
4969

4970 4971
    pPage = (SFilePage*) getBufPage(pSup->pResultBuf, p1->pageId);
    pRow = (SResultRow*)((char*)pPage + p1->offset);
4972 4973
    setBufPageDirty(pPage, true);
    releaseBufPage(pSup->pResultBuf, pPage);
wmmhello's avatar
wmmhello 已提交
4974 4975 4976

    // recalculate the result size
    int32_t realTotalSize = offset + sizeof(int32_t) + keyLen + sizeof(int32_t) + pSup->resultRowSize;
L
Liu Jicong 已提交
4977 4978 4979
    if (realTotalSize > totalSize) {
      char* tmp = taosMemoryRealloc(*result, realTotalSize);
      if (tmp == NULL) {
wmmhello's avatar
wmmhello 已提交
4980
        terrno = TSDB_CODE_OUT_OF_MEMORY;
wafwerar's avatar
wafwerar 已提交
4981
        taosMemoryFree(*result);
wmmhello's avatar
wmmhello 已提交
4982
        *result = NULL;
4983
        longjmp(pOperator->pTaskInfo->env, TSDB_CODE_OUT_OF_MEMORY);
L
Liu Jicong 已提交
4984
      } else {
wmmhello's avatar
wmmhello 已提交
4985 4986 4987 4988 4989 4990 4991 4992 4993 4994 4995 4996
        *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);
4997
    memcpy(*result + offset, pRow, pSup->resultRowSize);
wmmhello's avatar
wmmhello 已提交
4998 4999 5000 5001 5002
    offset += pSup->resultRowSize;

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

L
Liu Jicong 已提交
5003
  if (length) {
wmmhello's avatar
wmmhello 已提交
5004 5005 5006 5007 5008
    *length = offset;
  }
  return;
}

5009
bool aggDecodeResultRow(SOperatorInfo* pOperator, SAggSupporter *pSup, SOptrBasicInfo *pInfo, char* result, int32_t length) {
L
Liu Jicong 已提交
5010
  if (!result || length <= 0) {
wmmhello's avatar
wmmhello 已提交
5011 5012 5013 5014 5015 5016 5017
    return false;
  }

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

  int32_t offset = sizeof(int32_t);
L
Liu Jicong 已提交
5018
  while (count-- > 0 && length > offset) {
wmmhello's avatar
wmmhello 已提交
5019 5020 5021
    int32_t keyLen = *(int32_t*)(result + offset);
    offset += sizeof(int32_t);

L
Liu Jicong 已提交
5022 5023 5024
    uint64_t    tableGroupId = *(uint64_t*)(result + offset);
    SResultRow* resultRow = getNewResultRow_rv(pSup->pResultBuf, tableGroupId, pSup->resultRowSize);
    if (!resultRow) {
5025
      longjmp(pOperator->pTaskInfo->env, TSDB_CODE_TSC_INVALID_INPUT);
wmmhello's avatar
wmmhello 已提交
5026 5027
    }
    // add a new result set for a new group
5028 5029
    SResultRowPosition pos = {.pageId = resultRow->pageId, .offset = resultRow->offset};
    taosHashPut(pSup->pResultRowHashTable, result + offset, keyLen, &pos, sizeof(SResultRowPosition));
wmmhello's avatar
wmmhello 已提交
5030 5031 5032

    offset += keyLen;
    int32_t valueLen = *(int32_t*)(result + offset);
L
Liu Jicong 已提交
5033
    if (valueLen != pSup->resultRowSize) {
5034
      longjmp(pOperator->pTaskInfo->env, TSDB_CODE_TSC_INVALID_INPUT);
wmmhello's avatar
wmmhello 已提交
5035 5036 5037 5038 5039 5040 5041 5042 5043 5044
    }
    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);
5045 5046 5047
    prepareResultListBuffer(&pInfo->resultRowInfo, pOperator->pTaskInfo->env);
    pInfo->resultRowInfo.curPos = pInfo->resultRowInfo.size;
    pInfo->resultRowInfo.pPosition[pInfo->resultRowInfo.size++] = (SResultRowPosition) {.pageId = resultRow->pageId, .offset = resultRow->offset};
wmmhello's avatar
wmmhello 已提交
5048 5049
  }

L
Liu Jicong 已提交
5050
  if (offset != length) {
5051
    longjmp(pOperator->pTaskInfo->env, TSDB_CODE_TSC_INVALID_INPUT);
wmmhello's avatar
wmmhello 已提交
5052 5053 5054 5055
  }
  return true;
}

L
Liu Jicong 已提交
5056
static SSDataBlock* doMultiTableAggregate(SOperatorInfo* pOperator, bool* newgroup) {
5057 5058 5059 5060 5061
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  SAggOperatorInfo* pAggInfo = pOperator->info;
L
Liu Jicong 已提交
5062 5063
  SOptrBasicInfo*   pInfo = &pAggInfo->binfo;
  SExecTaskInfo*    pTaskInfo = pOperator->pTaskInfo;
5064 5065

  if (pOperator->status == OP_RES_TO_RETURN) {
5066
    toSDatablock(pInfo->pRes, pAggInfo->binfo.capacity, &pAggInfo->groupResInfo, pOperator->pExpr, pAggInfo->pResultBuf, pAggInfo->binfo.rowCellInfoOffset);
H
Haojun Liao 已提交
5067
    if (pInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pAggInfo->groupResInfo)) {
5068 5069 5070 5071 5072 5073
      pOperator->status = OP_EXEC_DONE;
    }

    return pInfo->pRes;
  }

5074
  // table scan order
L
Liu Jicong 已提交
5075
  int32_t        order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
5076
  SOperatorInfo* downstream = pOperator->pDownstream[0];
5077

L
Liu Jicong 已提交
5078
  while (1) {
H
Haojun Liao 已提交
5079
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
5080
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
5081
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5082 5083 5084 5085 5086

    if (pBlock == NULL) {
      break;
    }

L
Liu Jicong 已提交
5087 5088 5089 5090 5091
    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    if (downstream->operatorType == OP_TableScan) {
    //      STableScanInfo* pScanInfo = downstream->info;
    //      order = getTableScanOrder(pScanInfo);
    //    }
5092 5093 5094 5095 5096

    // the pDataBlock are always the same one, no need to call this again
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, order);

    TSKEY key = 0;
5097
    if (order == TSDB_ORDER_ASC) {
5098 5099 5100 5101 5102 5103
      key = pBlock->info.window.ekey;
      TSKEY_MAX_ADD(key, 1);
    } else {
      key = pBlock->info.window.skey;
      TSKEY_MIN_SUB(key, -1);
    }
L
Liu Jicong 已提交
5104 5105 5106

    setExecutionContext(pOperator->numOfOutput, pAggInfo->current->groupIndex, key, pTaskInfo, pAggInfo->current,
                        pAggInfo);
H
Haojun Liao 已提交
5107
    doAggregateImpl(pOperator, 0, pInfo->pCtx);
5108 5109 5110 5111
  }

  pOperator->status = OP_RES_TO_RETURN;
  closeAllResultRows(&pInfo->resultRowInfo);
5112
  updateNumOfRowsInResultRows(pInfo->pCtx, pOperator->numOfOutput, &pInfo->resultRowInfo, pInfo->rowCellInfoOffset);
5113

H
Haojun Liao 已提交
5114
  initGroupResInfo(&pAggInfo->groupResInfo, &pInfo->resultRowInfo);
5115
  toSDatablock(pInfo->pRes, pAggInfo->binfo.capacity, &pAggInfo->groupResInfo, pOperator->pExpr, pAggInfo->pResultBuf, pAggInfo->binfo.rowCellInfoOffset);
5116

H
Haojun Liao 已提交
5117
  if (pInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pAggInfo->groupResInfo)) {
5118 5119 5120 5121 5122 5123
    doSetOperatorCompleted(pOperator);
  }

  return pInfo->pRes;
}

L
Liu Jicong 已提交
5124
static SSDataBlock* doProjectOperation(SOperatorInfo* pOperator, bool* newgroup) {
5125
  SProjectOperatorInfo* pProjectInfo = pOperator->info;
L
Liu Jicong 已提交
5126
  SOptrBasicInfo*       pInfo = &pProjectInfo->binfo;
5127 5128

  SSDataBlock* pRes = pInfo->pRes;
5129
  blockDataCleanup(pRes);
5130
  
H
Haojun Liao 已提交
5131
#if 0
5132 5133 5134 5135 5136 5137
  if (pProjectInfo->existDataBlock) {  // TODO refactor
    SSDataBlock* pBlock = pProjectInfo->existDataBlock;
    pProjectInfo->existDataBlock = NULL;
    *newgroup = true;

    // todo dynamic set tags
L
Liu Jicong 已提交
5138 5139 5140
    //    if (pTableQueryInfo != NULL) {
    //      setTagValue(pOperator, pTableQueryInfo->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    }
5141 5142

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

H
Haojun Liao 已提交
5145
    blockDataEnsureCapacity(pInfo->pRes, pBlock->info.rows);
5146
    projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfOutput);
L
Liu Jicong 已提交
5147
    if (pRes->info.rows >= pProjectInfo->binfo.capacity * 0.8) {
5148 5149 5150 5151 5152
      copyTsColoum(pRes, pInfo->pCtx, pOperator->numOfOutput);
      resetResultRowEntryResult(pInfo->pCtx, pOperator->numOfOutput);
      return pRes;
    }
  }
H
Haojun Liao 已提交
5153
#endif
5154

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

L
Liu Jicong 已提交
5157
  while (1) {
5158 5159
    bool prevVal = *newgroup;

H
Haojun Liao 已提交
5160
    // The downstream exec may change the value of the newgroup, so use a local variable instead.
H
Haojun Liao 已提交
5161 5162 5163
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5164 5165 5166

    if (pBlock == NULL) {
      *newgroup = prevVal;
5167
      setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
5168 5169 5170 5171 5172 5173 5174 5175
      break;
    }

    // Return result of the previous group in the firstly.
    if (*newgroup) {
      if (pRes->info.rows > 0) {
        pProjectInfo->existDataBlock = pBlock;
        break;
L
Liu Jicong 已提交
5176
      } else {  // init output buffer for a new group data
5177 5178 5179 5180 5181
        initCtxOutputBuffer(pInfo->pCtx, pOperator->numOfOutput);
      }
    }

    // todo dynamic set tags
H
Haojun Liao 已提交
5182 5183 5184 5185
    //    STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;
    //    if (pTableQueryInfo != NULL) {
    //      setTagValue(pOperator, pTableQueryInfo->pTable, pInfo->pCtx, pOperator->numOfOutput);
    //    }
5186 5187

    // the pDataBlock are always the same one, no need to call this again
H
Haojun Liao 已提交
5188
    setInputDataBlock(pOperator, pInfo->pCtx, pBlock, TSDB_ORDER_ASC);
5189 5190
    blockDataEnsureCapacity(pInfo->pRes, pInfo->pRes->info.rows + pBlock->info.rows);

H
Haojun Liao 已提交
5191
    projectApplyFunctions(pOperator->pExpr, pInfo->pRes, pBlock, pInfo->pCtx, pOperator->numOfOutput, pProjectInfo->pPseudoColInfo);
H
Haojun Liao 已提交
5192

H
Haojun Liao 已提交
5193 5194 5195 5196 5197 5198 5199 5200 5201 5202 5203 5204 5205 5206 5207 5208 5209 5210 5211 5212 5213 5214 5215 5216 5217 5218 5219 5220 5221 5222 5223 5224
    if (pProjectInfo->curSOffset > 0) {
      if (pProjectInfo->groupId == 0) {  // it is the first group
        pProjectInfo->groupId = pBlock->info.groupId;
        blockDataCleanup(pInfo->pRes);
        continue;
      } else if (pProjectInfo->groupId != pBlock->info.groupId) {
        pProjectInfo->curSOffset -= 1;

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

      pProjectInfo->groupId = pBlock->info.groupId;
    }

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

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

    pProjectInfo->groupId = pBlock->info.groupId;

H
Haojun Liao 已提交
5225
    // todo extract method
H
Haojun Liao 已提交
5226 5227 5228 5229 5230 5231 5232 5233 5234
    if (pProjectInfo->curOffset < pInfo->pRes->info.rows && pProjectInfo->curOffset > 0) {
      blockDataTrimFirstNRows(pInfo->pRes, pProjectInfo->curOffset);
      pProjectInfo->curOffset = 0;
    } else if (pProjectInfo->curOffset >= pInfo->pRes->info.rows) {
      pProjectInfo->curOffset -= pInfo->pRes->info.rows;
      blockDataCleanup(pInfo->pRes);
      continue;
    }

5235
    if (pRes->info.rows >= pOperator->resultInfo.threshold) {
5236 5237 5238
      break;
    }
  }
H
Haojun Liao 已提交
5239
  
H
Haojun Liao 已提交
5240
  if (pProjectInfo->limit.limit > 0 && pProjectInfo->curOutput + pInfo->pRes->info.rows >= pProjectInfo->limit.limit) {
H
Haojun Liao 已提交
5241 5242 5243 5244
    pInfo->pRes->info.rows = (int32_t)(pProjectInfo->limit.limit - pProjectInfo->curOutput);
  }

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

L
Liu Jicong 已提交
5246 5247
  //  copyTsColoum(pRes, pInfo->pCtx, pOperator->numOfOutput);
  return (pInfo->pRes->info.rows > 0) ? pInfo->pRes : NULL;
5248 5249
}

L
Liu Jicong 已提交
5250
static int32_t doOpenIntervalAgg(SOperatorInfo* pOperator) {
5251 5252
  if (OPTR_IS_OPENED(pOperator)) {
    return TSDB_CODE_SUCCESS;
5253 5254
  }

H
Haojun Liao 已提交
5255
  STableIntervalOperatorInfo* pInfo = pOperator->info;
5256

5257
  int32_t order = TSDB_ORDER_ASC;
5258 5259
  //  STimeWindow win = {0};
  bool newgroup = false;
H
Haojun Liao 已提交
5260
  SOperatorInfo* downstream = pOperator->pDownstream[0];
5261

5262
  while (1) {
H
Haojun Liao 已提交
5263
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
5264
    SSDataBlock* pBlock = downstream->getNextFn(downstream, &newgroup);
H
Haojun Liao 已提交
5265
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5266 5267 5268 5269 5270

    if (pBlock == NULL) {
      break;
    }

5271
    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pInfo->pCtx, pOperator->numOfOutput);
5272
    // the pDataBlock are always the same one, no need to call this again
5273
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pBlock, order);
H
Haojun Liao 已提交
5274
    hashIntervalAgg(pOperator, &pInfo->binfo.resultRowInfo, pBlock, 0);
5275

5276
#if 0   // test for encode/decode result info
5277 5278 5279 5280 5281 5282 5283 5284 5285 5286 5287
    if(pOperator->encodeResultRow){
      char *result = NULL;
      int32_t length = 0;
      SAggSupporter   *pSup = &pInfo->aggSup;
      pOperator->encodeResultRow(pOperator, pSup, &pInfo->binfo, &result, &length);
      taosHashClear(pSup->pResultRowHashTable);
      pInfo->binfo.resultRowInfo.size = 0;
      pOperator->decodeResultRow(pOperator, pSup, &pInfo->binfo, result, length);
      if(result){
        taosMemoryFree(result);
      }
5288
    }
5289
#endif
5290 5291
  }

H
Haojun Liao 已提交
5292
  closeAllResultRows(&pInfo->binfo.resultRowInfo);
L
Liu Jicong 已提交
5293 5294
  finalizeMultiTupleQueryResult(pInfo->binfo.pCtx, pOperator->numOfOutput, pInfo->aggSup.pResultBuf,
                                &pInfo->binfo.resultRowInfo, pInfo->binfo.rowCellInfoOffset);
5295

H
Haojun Liao 已提交
5296
  initGroupResInfo(&pInfo->groupResInfo, &pInfo->binfo.resultRowInfo);
5297 5298 5299 5300
  OPTR_SET_OPENED(pOperator);
  return TSDB_CODE_SUCCESS;
}

L
Liu Jicong 已提交
5301
static SSDataBlock* doBuildIntervalResult(SOperatorInfo* pOperator, bool* newgroup) {
5302
  STableIntervalOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
5303
  SExecTaskInfo*              pTaskInfo = pOperator->pTaskInfo;
5304 5305 5306 5307 5308

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

5309 5310 5311 5312
  if (pInfo->execModel == OPTR_EXEC_MODEL_STREAM) {
    return pOperator->getStreamResFn(pOperator, newgroup);
  }

5313 5314 5315 5316 5317 5318
  pTaskInfo->code = pOperator->_openFn(pOperator);
  if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
    return NULL;
  }

  blockDataEnsureCapacity(pInfo->binfo.pRes, pInfo->binfo.capacity);
5319
  toSDatablock(pInfo->binfo.pRes, pInfo->binfo.capacity, &pInfo->groupResInfo, pOperator->pExpr, pInfo->aggSup.pResultBuf, pInfo->binfo.rowCellInfoOffset);
5320

H
Haojun Liao 已提交
5321
  if (pInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
5322 5323 5324
    doSetOperatorCompleted(pOperator);
  }

L
Liu Jicong 已提交
5325
  return pInfo->binfo.pRes->info.rows == 0 ? NULL : pInfo->binfo.pRes;
5326 5327
}

5328
static SSDataBlock* doStreamIntervalAgg(SOperatorInfo *pOperator, bool* newgroup) {
5329 5330 5331 5332 5333 5334 5335 5336
  STableIntervalOperatorInfo* pInfo = pOperator->info;
  int32_t order = TSDB_ORDER_ASC;

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

  if (pOperator->status == OP_RES_TO_RETURN) {
5337
    toSDatablock(pInfo->binfo.pRes, pInfo->binfo.capacity, &pInfo->groupResInfo, pOperator->pExpr, pInfo->aggSup.pResultBuf, pInfo->binfo.rowCellInfoOffset);
5338 5339 5340
    if (pInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
      pOperator->status = OP_EXEC_DONE;
    }
5341
    return pInfo->binfo.pRes->info.rows == 0 ? NULL : pInfo->binfo.pRes;
5342 5343 5344
  }

  //  STimeWindow win = {0};
5345
  *newgroup = false;
5346 5347 5348 5349 5350 5351
  SOperatorInfo* downstream = pOperator->pDownstream[0];

  SArray* pUpdated = NULL;

  while (1) {
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
5352
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
5353 5354 5355 5356 5357 5358 5359 5360 5361 5362 5363 5364 5365 5366 5367 5368 5369
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);

    if (pBlock == NULL) {
      break;
    }

    // The timewindows that overlaps the timestamps of the input pBlock need to be recalculated and return to the caller.
    // Note that all the time window are not close till now.

    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pInfo->pCtx, pOperator->numOfOutput);
    // the pDataBlock are always the same one, no need to call this again
    setInputDataBlock(pOperator, pInfo->binfo.pCtx, pBlock, order);
    pUpdated = hashIntervalAgg(pOperator, &pInfo->binfo.resultRowInfo, pBlock, 0);
  }

  finalizeUpdatedResult(pInfo->binfo.pCtx, pOperator->numOfOutput, pInfo->aggSup.pResultBuf, pUpdated, pInfo->binfo.rowCellInfoOffset);

H
Haojun Liao 已提交
5370
  initMultiResInfoFromArrayList(&pInfo->groupResInfo, pUpdated);
5371
  blockDataEnsureCapacity(pInfo->binfo.pRes, pInfo->binfo.capacity);
5372
  toSDatablock(pInfo->binfo.pRes, pInfo->binfo.capacity, &pInfo->groupResInfo, pOperator->pExpr, pInfo->aggSup.pResultBuf, pInfo->binfo.rowCellInfoOffset);
5373 5374 5375 5376 5377 5378 5379 5380

  ASSERT(pInfo->binfo.pRes->info.rows > 0);
  pOperator->status = OP_RES_TO_RETURN;

  return pInfo->binfo.pRes->info.rows == 0 ? NULL : pInfo->binfo.pRes;
}

static SSDataBlock* doAllIntervalAgg(SOperatorInfo *pOperator, bool* newgroup) {
5381 5382 5383 5384
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

5385
  STimeSliceOperatorInfo* pSliceInfo = pOperator->info;
5386
  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
5387
    //    toSDatablock(&pRuntimeEnv->groupResInfo, pRuntimeEnv, pIntervalInfo->pRes);
5388
    if (pSliceInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pSliceInfo->groupResInfo)) {
5389 5390 5391
      doSetOperatorCompleted(pOperator);
    }

5392
    return pSliceInfo->binfo.pRes;
5393 5394
  }

5395 5396
  int32_t     order = TSDB_ORDER_ASC;
//  STimeWindow win = pQueryAttr->window;
H
Haojun Liao 已提交
5397
  SOperatorInfo* downstream = pOperator->pDownstream[0];
5398

L
Liu Jicong 已提交
5399
  while (1) {
H
Haojun Liao 已提交
5400
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
5401
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
5402
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5403 5404 5405 5406
    if (pBlock == NULL) {
      break;
    }

L
Liu Jicong 已提交
5407
    //    setTagValue(pOperator, pRuntimeEnv->current->pTable, pIntervalInfo->pCtx, pOperator->numOfOutput);
5408
    // the pDataBlock are always the same one, no need to call this again
5409 5410
    setInputDataBlock(pOperator, pSliceInfo->binfo.pCtx, pBlock, order);
    hashAllIntervalAgg(pOperator, &pSliceInfo->binfo.resultRowInfo, pBlock, 0);
5411 5412 5413 5414
  }

  // restore the value
  pOperator->status = OP_RES_TO_RETURN;
5415
  closeAllResultRows(&pSliceInfo->binfo.resultRowInfo);
5416
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
5417
  finalizeQueryResult(pSliceInfo->binfo.pCtx, pOperator->numOfOutput);
5418

5419 5420
  initGroupResInfo(&pSliceInfo->groupResInfo, &pSliceInfo->binfo.resultRowInfo);
  //  toSDatablock(&pRuntimeEnv->groupResInfo, pRuntimeEnv, pSliceInfo->pRes);
5421

5422
  if (pSliceInfo->binfo.pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pSliceInfo->groupResInfo)) {
5423 5424 5425
    pOperator->status = OP_EXEC_DONE;
  }

5426
  return pSliceInfo->binfo.pRes->info.rows == 0 ? NULL : pSliceInfo->binfo.pRes;
5427 5428
}

L
Liu Jicong 已提交
5429
static SSDataBlock* doSTableIntervalAgg(SOperatorInfo* pOperator, bool* newgroup) {
5430 5431 5432 5433 5434
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  STableIntervalOperatorInfo* pIntervalInfo = pOperator->info;
L
Liu Jicong 已提交
5435
  STaskRuntimeEnv*            pRuntimeEnv = pOperator->pRuntimeEnv;
5436 5437 5438 5439

  if (pOperator->status == OP_RES_TO_RETURN) {
    int64_t st = taosGetTimestampUs();

L
Liu Jicong 已提交
5440
    //    copyToSDataBlock(NULL, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
5441
    if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
5442 5443
      doSetOperatorCompleted(pOperator);
    }
H
Haojun Liao 已提交
5444
    return pIntervalInfo->binfo.pRes;
5445 5446
  }

H
Haojun Liao 已提交
5447
  STaskAttr* pQueryAttr = pRuntimeEnv->pQueryAttr;
L
Liu Jicong 已提交
5448
  int32_t    order = pQueryAttr->order.order;
5449

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

L
Liu Jicong 已提交
5452
  while (1) {
H
Haojun Liao 已提交
5453
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
5454
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
5455
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5456 5457 5458 5459 5460 5461 5462 5463

    if (pBlock == NULL) {
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
    STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;

L
Liu Jicong 已提交
5464
    //    setTagValue(pOperator, pTableQueryInfo->pTable, pIntervalInfo->pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
5465
    setInputDataBlock(pOperator, pIntervalInfo->binfo.pCtx, pBlock, pQueryAttr->order.order);
5466 5467 5468 5469 5470 5471
    setIntervalQueryRange(pRuntimeEnv, pBlock->info.window.skey);

    hashIntervalAgg(pOperator, &pTableQueryInfo->resInfo, pBlock, pTableQueryInfo->groupIndex);
  }

  pOperator->status = OP_RES_TO_RETURN;
L
Liu Jicong 已提交
5472
  pQueryAttr->order.order = order;  // TODO : restore the order
5473
  doCloseAllTimeWindow(pRuntimeEnv);
5474
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
5475

L
Liu Jicong 已提交
5476
  //  copyToSDataBlock(pRuntimeEnv, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
5477
  if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
5478 5479 5480
    pOperator->status = OP_EXEC_DONE;
  }

H
Haojun Liao 已提交
5481
  return pIntervalInfo->binfo.pRes;
5482 5483
}

L
Liu Jicong 已提交
5484
static SSDataBlock* doAllSTableIntervalAgg(SOperatorInfo* pOperator, bool* newgroup) {
5485 5486 5487 5488 5489
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

  STableIntervalOperatorInfo* pIntervalInfo = pOperator->info;
L
Liu Jicong 已提交
5490
  STaskRuntimeEnv*            pRuntimeEnv = pOperator->pRuntimeEnv;
5491 5492

  if (pOperator->status == OP_RES_TO_RETURN) {
L
Liu Jicong 已提交
5493
    //    copyToSDataBlock(pRuntimeEnv, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
5494
    if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
5495 5496 5497
      pOperator->status = OP_EXEC_DONE;
    }

H
Haojun Liao 已提交
5498
    return pIntervalInfo->binfo.pRes;
5499 5500
  }

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

L
Liu Jicong 已提交
5503
  while (1) {
H
Haojun Liao 已提交
5504
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
5505
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
5506
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5507 5508 5509 5510 5511 5512 5513 5514

    if (pBlock == NULL) {
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
    STableQueryInfo* pTableQueryInfo = pRuntimeEnv->current;

L
Liu Jicong 已提交
5515
    //    setTagValue(pOperator, pTableQueryInfo->pTable, pIntervalInfo->pCtx, pOperator->numOfOutput);
H
Haojun Liao 已提交
5516
//    setInputDataBlock(pOperator, pIntervalInfo->binfo.pCtx, pBlock, pQueryAttr->order.order);
5517 5518 5519 5520 5521 5522
    setIntervalQueryRange(pRuntimeEnv, pBlock->info.window.skey);

    hashAllIntervalAgg(pOperator, &pTableQueryInfo->resInfo, pBlock, pTableQueryInfo->groupIndex);
  }

  pOperator->status = OP_RES_TO_RETURN;
H
Haojun Liao 已提交
5523
//  pQueryAttr->order.order = order;  // TODO : restore the order
5524
  doCloseAllTimeWindow(pRuntimeEnv);
5525
  setTaskStatus(pOperator->pTaskInfo, TASK_COMPLETED);
5526 5527

  int64_t st = taosGetTimestampUs();
L
Liu Jicong 已提交
5528
  //  copyToSDataBlock(pRuntimeEnv, 3000, pIntervalInfo->pRes, pIntervalInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
5529
  if (pIntervalInfo->binfo.pRes->info.rows == 0 || !hasRemainData(&pRuntimeEnv->groupResInfo)) {
5530 5531 5532
    pOperator->status = OP_EXEC_DONE;
  }

H
Haojun Liao 已提交
5533
  return pIntervalInfo->binfo.pRes;
5534 5535
}

5536 5537
static void doStateWindowAggImpl(SOperatorInfo* pOperator, SStateWindowOperatorInfo* pInfo, SSDataBlock* pBlock) {
  SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
5538 5539
  SOptrBasicInfo* pBInfo = &pInfo->binfo;

5540 5541
  SColumnInfoData* pStateColInfoData = taosArrayGet(pBlock->pDataBlock, pInfo->colIndex);
  int64_t gid = pBlock->info.groupId;
5542

5543 5544 5545 5546 5547 5548 5549 5550
  bool    masterScan = true;
  int32_t numOfOutput = pOperator->numOfOutput;

  int16_t bytes = pStateColInfoData->info.bytes;
  int16_t type = pStateColInfoData->info.type;

  SColumnInfoData* pColInfoData = taosArrayGet(pBlock->pDataBlock, 0);
  TSKEY*           tsList = (TSKEY*)pColInfoData->pData;
5551

5552 5553 5554 5555 5556
  SWindowRowsSup* pRowSup = &pInfo->winSup;
  pRowSup->numOfRows = 0;
  
  for (int32_t j = 0; j < pBlock->info.rows; ++j) {
    if (colDataIsNull(pStateColInfoData, pBlock->info.rows, j, pBlock->pBlockAgg)) {
5557 5558
      continue;
    }
5559 5560 5561 5562 5563 5564 5565 5566 5567 5568 5569 5570 5571

    char* val = colDataGetData(pStateColInfoData, j);

    if (!pInfo->hasKey) {
      memcpy(pInfo->stateKey.pData, val, bytes);
      pInfo->hasKey = true;

      doKeepNewWindowStartInfo(pRowSup, tsList, j);
      doKeepTuple(pRowSup, tsList[j]);
    } else if (memcmp(pInfo->stateKey.pData, val, bytes) == 0) {
      doKeepTuple(pRowSup, tsList[j]);
      if (j == 0 && pRowSup->startRowIndex != 0) {
        pRowSup->startRowIndex = 0;
5572
      }
5573
    } else {  // a new state window started
5574
      SResultRow* pResult = NULL;
5575 5576 5577 5578 5579 5580 5581

      // keep the time window for the closed time window.
      STimeWindow window = pRowSup->win;

      pRowSup->win.ekey = pRowSup->win.skey;
      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);
5582
      if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
5583
        longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
5584 5585
      }

5586 5587 5588 5589 5590 5591
      updateTimeWindowInfo(&pInfo->timeWindowData, &window, false);
      doApplyFunctions(pInfo->binfo.pCtx, &window, &pInfo->timeWindowData, pRowSup->startRowIndex, pRowSup->numOfRows, NULL, pBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);

      // here we start a new session window
      doKeepNewWindowStartInfo(pRowSup, tsList, j);
      doKeepTuple(pRowSup, tsList[j]);
5592 5593 5594 5595
    }
  }

  SResultRow* pResult = NULL;
5596 5597 5598
  pRowSup->win.ekey = tsList[pBlock->info.rows - 1];
  int32_t ret = setResultOutputBufByKey_rv(&pInfo->binfo.resultRowInfo, pBlock->info.uid, &pRowSup->win, masterScan, &pResult,
                                           gid, pInfo->binfo.pCtx, numOfOutput, pInfo->binfo.rowCellInfoOffset, &pInfo->aggSup, pTaskInfo);
5599
  if (ret != TSDB_CODE_SUCCESS) {  // null data, too many state code
5600
    longjmp(pTaskInfo->env, TSDB_CODE_QRY_APP_ERROR);
5601 5602
  }

5603 5604
  updateTimeWindowInfo(&pInfo->timeWindowData, &pRowSup->win, false);
  doApplyFunctions(pInfo->binfo.pCtx, &pRowSup->win, &pInfo->timeWindowData, pRowSup->startRowIndex, pRowSup->numOfRows, NULL, pBlock->info.rows, numOfOutput, TSDB_ORDER_ASC);
5605 5606
}

L
Liu Jicong 已提交
5607
static SSDataBlock* doStateWindowAgg(SOperatorInfo* pOperator, bool* newgroup) {
5608 5609 5610 5611
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

5612
  SStateWindowOperatorInfo* pInfo = pOperator->info;
H
Haojun Liao 已提交
5613
  SExecTaskInfo*  pTaskInfo = pOperator->pTaskInfo;
5614
  SOptrBasicInfo* pBInfo    = &pInfo->binfo;
5615 5616

  if (pOperator->status == OP_RES_TO_RETURN) {
5617
    toSDatablock(pBInfo->pRes, pBInfo->capacity, &pInfo->groupResInfo, pOperator->pExpr, pInfo->aggSup.pResultBuf, pBInfo->rowCellInfoOffset);
5618 5619 5620 5621
    if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
      doSetOperatorCompleted(pOperator);
      return NULL;
    }
5622 5623 5624 5625

    return pBInfo->pRes;
  }

H
Haojun Liao 已提交
5626 5627 5628
  int32_t        order = TSDB_ORDER_ASC;
  STimeWindow    win = pTaskInfo->window;

H
Haojun Liao 已提交
5629
  SOperatorInfo* downstream = pOperator->pDownstream[0];
5630
  while (1) {
H
Haojun Liao 已提交
5631
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
5632
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
5633
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5634 5635 5636 5637

    if (pBlock == NULL) {
      break;
    }
H
Haojun Liao 已提交
5638

5639 5640
    setInputDataBlock(pOperator, pBInfo->pCtx, pBlock, order);
    doStateWindowAggImpl(pOperator, pInfo, pBlock);
5641 5642 5643 5644
  }

  pOperator->status = OP_RES_TO_RETURN;
  closeAllResultRows(&pBInfo->resultRowInfo);
5645
  finalizeMultiTupleQueryResult(pBInfo->pCtx, pOperator->numOfOutput, pInfo->aggSup.pResultBuf, &pBInfo->resultRowInfo, pBInfo->rowCellInfoOffset);
5646

5647 5648
  initGroupResInfo(&pInfo->groupResInfo, &pBInfo->resultRowInfo);
  blockDataEnsureCapacity(pBInfo->pRes, pBInfo->capacity);
5649
  toSDatablock(pBInfo->pRes, pBInfo->capacity, &pInfo->groupResInfo, pOperator->pExpr, pInfo->aggSup.pResultBuf, pBInfo->rowCellInfoOffset);
5650 5651 5652
  if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
    doSetOperatorCompleted(pOperator);
  }
5653

L
Liu Jicong 已提交
5654
  return pBInfo->pRes->info.rows == 0 ? NULL : pBInfo->pRes;
5655 5656
}

L
Liu Jicong 已提交
5657
static SSDataBlock* doSessionWindowAgg(SOperatorInfo* pOperator, bool* newgroup) {
5658 5659 5660 5661
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

5662
  SSessionAggOperatorInfo* pInfo = pOperator->info;
L
Liu Jicong 已提交
5663
  SOptrBasicInfo*          pBInfo = &pInfo->binfo;
5664 5665

  if (pOperator->status == OP_RES_TO_RETURN) {
5666
    toSDatablock(pBInfo->pRes, pBInfo->capacity, &pInfo->groupResInfo, pOperator->pExpr, pInfo->aggSup.pResultBuf, pBInfo->rowCellInfoOffset);
5667 5668 5669
    if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
      doSetOperatorCompleted(pOperator);
      return NULL;
5670 5671 5672 5673 5674
    }

    return pBInfo->pRes;
  }

L
Liu Jicong 已提交
5675
  int32_t        order = TSDB_ORDER_ASC;
H
Haojun Liao 已提交
5676
  SOperatorInfo* downstream = pOperator->pDownstream[0];
5677

L
Liu Jicong 已提交
5678
  while (1) {
H
Haojun Liao 已提交
5679
    publishOperatorProfEvent(downstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
H
Haojun Liao 已提交
5680
    SSDataBlock* pBlock = downstream->getNextFn(downstream, newgroup);
H
Haojun Liao 已提交
5681
    publishOperatorProfEvent(downstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5682 5683 5684 5685 5686
    if (pBlock == NULL) {
      break;
    }

    // the pDataBlock are always the same one, no need to call this again
H
Haojun Liao 已提交
5687
    setInputDataBlock(pOperator, pBInfo->pCtx, pBlock, order);
5688
    doSessionWindowAggImpl(pOperator, pInfo, pBlock);
5689 5690 5691 5692 5693
  }

  // restore the value
  pOperator->status = OP_RES_TO_RETURN;
  closeAllResultRows(&pBInfo->resultRowInfo);
5694
  finalizeMultiTupleQueryResult(pBInfo->pCtx, pOperator->numOfOutput, pInfo->aggSup.pResultBuf, &pBInfo->resultRowInfo, pBInfo->rowCellInfoOffset);
5695

5696 5697
  initGroupResInfo(&pInfo->groupResInfo, &pBInfo->resultRowInfo);
  blockDataEnsureCapacity(pBInfo->pRes, pBInfo->capacity);
5698
  toSDatablock(pBInfo->pRes, pBInfo->capacity, &pInfo->groupResInfo, pOperator->pExpr, pInfo->aggSup.pResultBuf, pBInfo->rowCellInfoOffset);
5699 5700
  if (pBInfo->pRes->info.rows == 0 || !hasRemainDataInCurrentGroup(&pInfo->groupResInfo)) {
    doSetOperatorCompleted(pOperator);
5701 5702
  }

L
Liu Jicong 已提交
5703
  return pBInfo->pRes->info.rows == 0 ? NULL : pBInfo->pRes;
5704 5705
}

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

L
Liu Jicong 已提交
5710 5711
  int64_t ekey = Q_STATUS_EQUAL(pTaskInfo->status, TASK_COMPLETED) ? pTaskInfo->window.ekey
                                                                   : pInfo->existNewGroupBlock->info.window.ekey;
5712 5713
  taosResetFillInfo(pInfo->pFillInfo, getFillInfoStart(pInfo->pFillInfo));

5714
  taosFillSetStartInfo(pInfo->pFillInfo, pInfo->existNewGroupBlock->info.rows, ekey);
5715 5716
  taosFillSetInputDataBlock(pInfo->pFillInfo, pInfo->existNewGroupBlock);

H
Haojun Liao 已提交
5717
  doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, pResultInfo->capacity, pInfo->p);
5718 5719 5720 5721
  pInfo->existNewGroupBlock = NULL;
  *newgroup = true;
}

L
Liu Jicong 已提交
5722 5723
static void doHandleRemainBlockFromNewGroup(SFillOperatorInfo* pInfo, SResultInfo* pResultInfo, bool* newgroup,
                                            SExecTaskInfo* pTaskInfo) {
5724 5725
  if (taosFillHasMoreResults(pInfo->pFillInfo)) {
    *newgroup = false;
H
Haojun Liao 已提交
5726 5727
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, (int32_t)pResultInfo->capacity, pInfo->p);
    if (pInfo->pRes->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult)) {
5728 5729 5730 5731 5732 5733
      return;
    }
  }

  // handle the cached new group data block
  if (pInfo->existNewGroupBlock) {
5734
    doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
5735 5736 5737
  }
}

L
Liu Jicong 已提交
5738 5739 5740
static SSDataBlock* doFill(SOperatorInfo* pOperator, bool* newgroup) {
  SFillOperatorInfo* pInfo = pOperator->info;
  SExecTaskInfo*     pTaskInfo = pOperator->pTaskInfo;
5741

H
Haojun Liao 已提交
5742 5743
  SResultInfo* pResultInfo = &pOperator->resultInfo;
  blockDataCleanup(pInfo->pRes);
5744 5745 5746 5747
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

5748
  doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
H
Haojun Liao 已提交
5749 5750 5751
  if (pInfo->pRes->info.rows > pResultInfo->threshold || (!pInfo->multigroupResult && pInfo->pRes->info.rows > 0)) {
    return pInfo->pRes;
  }
5752

H
Haojun Liao 已提交
5753
  SOperatorInfo* pDownstream = pOperator->pDownstream[0];
L
Liu Jicong 已提交
5754
  while (1) {
H
Haojun Liao 已提交
5755 5756 5757
    publishOperatorProfEvent(pDownstream, QUERY_PROF_BEFORE_OPERATOR_EXEC);
    SSDataBlock* pBlock = pDownstream->getNextFn(pDownstream, newgroup);
    publishOperatorProfEvent(pDownstream, QUERY_PROF_AFTER_OPERATOR_EXEC);
5758 5759 5760 5761 5762 5763 5764 5765 5766 5767 5768

    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
5769
      taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
5770 5771 5772 5773 5774 5775 5776
    } else {
      if (pBlock == NULL) {
        if (pInfo->totalInputRows == 0) {
          pOperator->status = OP_EXEC_DONE;
          return NULL;
        }

5777
        taosFillSetStartInfo(pInfo->pFillInfo, 0, pTaskInfo->window.ekey);
5778 5779 5780 5781 5782 5783 5784
      } else {
        pInfo->totalInputRows += pBlock->info.rows;
        taosFillSetStartInfo(pInfo->pFillInfo, pBlock->info.rows, pBlock->info.window.ekey);
        taosFillSetInputDataBlock(pInfo->pFillInfo, pBlock);
      }
    }

H
Haojun Liao 已提交
5785
    doFillTimeIntervalGapsInResults(pInfo->pFillInfo, pInfo->pRes, pInfo->capacity, pInfo->p);
5786 5787 5788 5789 5790

    // 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 已提交
5791
      if (pInfo->pRes->info.rows > pResultInfo->threshold || pBlock == NULL || (!pInfo->multigroupResult)) {
5792 5793 5794
        return pInfo->pRes;
      }

5795
      doHandleRemainBlockFromNewGroup(pInfo, pResultInfo, newgroup, pTaskInfo);
H
Haojun Liao 已提交
5796
      if (pInfo->pRes->info.rows > pOperator->resultInfo.threshold || pBlock == NULL) {
5797 5798 5799 5800
        return pInfo->pRes;
      }
    } else if (pInfo->existNewGroupBlock) {  // try next group
      assert(pBlock != NULL);
5801
      doHandleRemainBlockForNewGroupImpl(pInfo, pResultInfo, newgroup, pTaskInfo);
H
Haojun Liao 已提交
5802
      if (pInfo->pRes->info.rows > pResultInfo->threshold) {
5803 5804 5805 5806 5807 5808 5809 5810 5811
        return pInfo->pRes;
      }
    } else {
      return NULL;
    }
  }
}

// todo set the attribute of query scan count
H
Haojun Liao 已提交
5812
static int32_t getNumOfScanTimes(STaskAttr* pQueryAttr) {
L
Liu Jicong 已提交
5813
  for (int32_t i = 0; i < pQueryAttr->numOfOutput; ++i) {
5814 5815 5816 5817 5818 5819 5820 5821 5822 5823 5824 5825 5826 5827
    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 已提交
5828 5829
  if (pOperator->closeFn != NULL) {
    pOperator->closeFn(pOperator->info, pOperator->numOfOutput);
5830 5831
  }

H
Haojun Liao 已提交
5832
  if (pOperator->pDownstream != NULL) {
L
Liu Jicong 已提交
5833
    for (int32_t i = 0; i < pOperator->numOfDownstream; ++i) {
H
Haojun Liao 已提交
5834
      destroyOperatorInfo(pOperator->pDownstream[i]);
5835 5836
    }

wafwerar's avatar
wafwerar 已提交
5837
    taosMemoryFreeClear(pOperator->pDownstream);
H
Haojun Liao 已提交
5838
    pOperator->numOfDownstream = 0;
5839 5840
  }

wafwerar's avatar
wafwerar 已提交
5841 5842
  taosMemoryFreeClear(pOperator->info);
  taosMemoryFreeClear(pOperator);
5843 5844
}

5845
int32_t doInitAggInfoSup(SAggSupporter* pAggSup, SqlFunctionCtx* pCtx, int32_t numOfOutput, size_t keyBufSize, const char* pKey) {
5846 5847
  _hash_fn_t hashFn = taosGetDefaultHashFunction(TSDB_DATA_TYPE_BINARY);

H
Haojun Liao 已提交
5848
  pAggSup->resultRowSize       = getResultRowSize(pCtx, numOfOutput);
5849
  pAggSup->keyBuf              = taosMemoryCalloc(1, keyBufSize);
5850
  pAggSup->pResultRowHashTable = taosHashInit(10, hashFn, true, HASH_NO_LOCK);
H
Haojun Liao 已提交
5851
  pAggSup->pResultRowListSet   = taosHashInit(100, hashFn, false, HASH_NO_LOCK);
5852 5853 5854
  pAggSup->pResultRowArrayList = taosArrayInit(10, sizeof(SResultRowCell));

  if (pAggSup->keyBuf == NULL || pAggSup->pResultRowArrayList == NULL || pAggSup->pResultRowListSet == NULL ||
5855
      pAggSup->pResultRowHashTable == NULL) {
5856 5857 5858
    return TSDB_CODE_OUT_OF_MEMORY;
  }

H
Haojun Liao 已提交
5859 5860 5861 5862 5863
  int32_t code = createDiskbasedBuf(&pAggSup->pResultBuf, 4096, 4096 * 256, pKey, "/tmp/");
  if (code != TSDB_CODE_SUCCESS) {
    return code;
  }

5864 5865 5866
  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
5867
static void cleanupAggSup(SAggSupporter* pAggSup) {
wafwerar's avatar
wafwerar 已提交
5868
  taosMemoryFreeClear(pAggSup->keyBuf);
5869 5870 5871
  taosHashCleanup(pAggSup->pResultRowHashTable);
  taosHashCleanup(pAggSup->pResultRowListSet);
  taosArrayDestroy(pAggSup->pResultRowArrayList);
H
Haojun Liao 已提交
5872
  destroyDiskbasedBuf(pAggSup->pResultBuf);
5873 5874
}

H
Haojun Liao 已提交
5875
int32_t initAggInfo(SOptrBasicInfo* pBasicInfo, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols,
5876
                           int32_t numOfRows, SSDataBlock* pResultBlock, size_t keyBufSize, const char* pkey) {
5877
  pBasicInfo->pCtx = createSqlFunctionCtx(pExprInfo, numOfCols, &pBasicInfo->rowCellInfoOffset);
H
Haojun Liao 已提交
5878 5879 5880
  pBasicInfo->pRes = pResultBlock;
  pBasicInfo->capacity = numOfRows;

5881
  doInitAggInfoSup(pAggSup, pBasicInfo->pCtx, numOfCols, keyBufSize, pkey);
5882
  return TSDB_CODE_SUCCESS;
H
Haojun Liao 已提交
5883 5884 5885
}

static STableQueryInfo* initTableQueryInfo(const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
5886
  STableQueryInfo* pTableQueryInfo = taosMemoryCalloc(pTableGroupInfo->numOfTables, sizeof(STableQueryInfo));
H
Haojun Liao 已提交
5887 5888
  if (pTableQueryInfo == NULL) {
    return NULL;
H
Haojun Liao 已提交
5889
  }
H
Haojun Liao 已提交
5890 5891

  int32_t index = 0;
L
Liu Jicong 已提交
5892
  for (int32_t i = 0; i < taosArrayGetSize(pTableGroupInfo->pGroupList); ++i) {
H
Haojun Liao 已提交
5893
    SArray* pa = taosArrayGetP(pTableGroupInfo->pGroupList, i);
L
Liu Jicong 已提交
5894
    for (int32_t j = 0; j < taosArrayGetSize(pa); ++j) {
H
Haojun Liao 已提交
5895 5896
      STableKeyInfo* pk = taosArrayGet(pa, j);

H
Haojun Liao 已提交
5897
      STableQueryInfo* pTQueryInfo = &pTableQueryInfo[index++];
L
Liu Jicong 已提交
5898 5899
      pTQueryInfo->uid = pk->uid;
      pTQueryInfo->lastKey = pk->lastKey;
H
Haojun Liao 已提交
5900 5901 5902
      pTQueryInfo->groupIndex = i;
    }
  }
H
Haojun Liao 已提交
5903 5904

  STimeWindow win = {0, INT64_MAX};
H
Haojun Liao 已提交
5905 5906
  createTableQueryInfo(pTableQueryInfo, false, win);
  return pTableQueryInfo;
H
Haojun Liao 已提交
5907 5908
}

L
Liu Jicong 已提交
5909 5910 5911
SOperatorInfo* createAggregateOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                           SSDataBlock* pResultBlock, SExecTaskInfo* pTaskInfo,
                                           const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
5912
  SAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SAggOperatorInfo));
L
Liu Jicong 已提交
5913
  SOperatorInfo*    pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
5914 5915 5916
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
5917 5918

  //(int32_t)(getRowNumForMultioutput(pQueryAttr, pQueryAttr->topBotQuery, pQueryAttr->stableQuery));
H
Haojun Liao 已提交
5919
  int32_t numOfRows = 1;
5920 5921
  size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
  int32_t code = initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResultBlock, keyBufSize, pTaskInfo->id.str);
H
Haojun Liao 已提交
5922 5923
  pInfo->pTableQueryInfo = initTableQueryInfo(pTableGroupInfo);
  if (code != TSDB_CODE_SUCCESS || pInfo->pTableQueryInfo == NULL) {
H
Haojun Liao 已提交
5924 5925
    goto _error;
  }
H
Haojun Liao 已提交
5926

H
Haojun Liao 已提交
5927
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, pTaskInfo);
5928

H
Haojun Liao 已提交
5929
  pOperator->name         = "TableAggregate";
X
Xiaoyu Wang 已提交
5930
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_AGG;
5931
  pOperator->blockingOptr = true;
H
Haojun Liao 已提交
5932 5933 5934 5935 5936 5937 5938 5939 5940
  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 已提交
5941 5942
  pOperator->encodeResultRow = aggEncodeResultRow;
  pOperator->decodeResultRow = aggDecodeResultRow;
H
Haojun Liao 已提交
5943 5944 5945 5946 5947

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

  return pOperator;
L
Liu Jicong 已提交
5950
_error:
H
Haojun Liao 已提交
5951
  destroyAggOperatorInfo(pInfo, numOfCols);
wafwerar's avatar
wafwerar 已提交
5952 5953
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
5954 5955
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
5956 5957
}

H
Haojun Liao 已提交
5958
void doDestroyBasicInfo(SOptrBasicInfo* pInfo, int32_t numOfOutput) {
5959 5960
  assert(pInfo != NULL);

5961
  destroySqlFunctionCtx(pInfo->pCtx, numOfOutput);
wafwerar's avatar
wafwerar 已提交
5962
  taosMemoryFreeClear(pInfo->rowCellInfoOffset);
5963 5964

  cleanupResultRowInfo(&pInfo->resultRowInfo);
H
Haojun Liao 已提交
5965
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
5966 5967
}

H
Haojun Liao 已提交
5968
void destroyBasicOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5969
  SOptrBasicInfo* pInfo = (SOptrBasicInfo*)param;
5970 5971
  doDestroyBasicInfo(pInfo, numOfOutput);
}
H
Haojun Liao 已提交
5972 5973

void destroyStateWindowOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5974
  SStateWindowOperatorInfo* pInfo = (SStateWindowOperatorInfo*)param;
5975
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
5976
  taosMemoryFreeClear(pInfo->stateKey.pData);
5977
}
H
Haojun Liao 已提交
5978 5979

void destroyAggOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5980
  SAggOperatorInfo* pInfo = (SAggOperatorInfo*)param;
5981 5982
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}
5983

H
Haojun Liao 已提交
5984
void destroyIntervalOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5985
  STableIntervalOperatorInfo* pInfo = (STableIntervalOperatorInfo*)param;
H
Haojun Liao 已提交
5986 5987 5988 5989 5990
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
  cleanupAggSup(&pInfo->aggSup);
}

void destroySWindowOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5991
  SSessionAggOperatorInfo* pInfo = (SSessionAggOperatorInfo*)param;
5992 5993 5994
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}

H
Haojun Liao 已提交
5995
void destroySFillOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
5996
  SFillOperatorInfo* pInfo = (SFillOperatorInfo*)param;
5997
  pInfo->pFillInfo = taosDestroyFillInfo(pInfo->pFillInfo);
H
Haojun Liao 已提交
5998
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
wafwerar's avatar
wafwerar 已提交
5999
  taosMemoryFreeClear(pInfo->p);
6000 6001
}

H
Haojun Liao 已提交
6002
static void destroyProjectOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
6003
  SProjectOperatorInfo* pInfo = (SProjectOperatorInfo*)param;
6004 6005 6006
  doDestroyBasicInfo(&pInfo->binfo, numOfOutput);
}

H
Haojun Liao 已提交
6007
static void destroyTagScanOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
6008
  STagScanInfo* pInfo = (STagScanInfo*)param;
H
Haojun Liao 已提交
6009
  pInfo->pRes = blockDataDestroy(pInfo->pRes);
6010 6011
}

H
Haojun Liao 已提交
6012
static void destroyOrderOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
6013
  SSortOperatorInfo* pInfo = (SSortOperatorInfo*)param;
H
Haojun Liao 已提交
6014 6015
  pInfo->pDataBlock = blockDataDestroy(pInfo->pDataBlock);

H
Haojun Liao 已提交
6016
  taosArrayDestroy(pInfo->pSortInfo);
6017
  taosArrayDestroy(pInfo->inputSlotMap);
6018 6019
}

H
Haojun Liao 已提交
6020
void destroyExchangeOperatorInfo(void* param, int32_t numOfOutput) {
L
Liu Jicong 已提交
6021
  SExchangeInfo* pExInfo = (SExchangeInfo*)param;
H
Haojun Liao 已提交
6022 6023 6024 6025 6026 6027 6028 6029 6030
  taosArrayDestroy(pExInfo->pSources);
  taosArrayDestroy(pExInfo->pSourceDataInfo);
  if (pExInfo->pResult != NULL) {
    blockDataDestroy(pExInfo->pResult);
  }

  tsem_destroy(&pExInfo->ready);
}

L
Liu Jicong 已提交
6031 6032 6033
SOperatorInfo* createMultiTableAggOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                               SSDataBlock* pResBlock, SExecTaskInfo* pTaskInfo,
                                               const STableGroupInfo* pTableGroupInfo) {
wafwerar's avatar
wafwerar 已提交
6034
  SAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SAggOperatorInfo));
6035

H
Haojun Liao 已提交
6036
  int32_t numOfRows = 1;
6037
  size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
L
Liu Jicong 已提交
6038
  int32_t code =
6039
      initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResBlock, keyBufSize, pTaskInfo->id.str);
H
Haojun Liao 已提交
6040 6041 6042 6043
  pInfo->pTableQueryInfo = initTableQueryInfo(pTableGroupInfo);
  if (code != TSDB_CODE_SUCCESS || pInfo->pTableQueryInfo == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
6044

L
Liu Jicong 已提交
6045
  size_t tableGroup = taosArrayGetSize(pTableGroupInfo->pGroupList);
6046
  initResultRowInfo(&pInfo->binfo.resultRowInfo, (int32_t)tableGroup);
6047

wafwerar's avatar
wafwerar 已提交
6048
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
6049
  pOperator->name = "MultiTableAggregate";
X
Xiaoyu Wang 已提交
6050
  // pOperator->operatorType = OP_MultiTableAggregate;
6051
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
6052 6053 6054 6055 6056
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
  pOperator->pExpr = pExprInfo;
  pOperator->numOfOutput = numOfCols;
  pOperator->pTaskInfo = pTaskInfo;
6057

H
Haojun Liao 已提交
6058
  pOperator->getNextFn = doMultiTableAggregate;
H
Haojun Liao 已提交
6059
  pOperator->closeFn = destroyAggOperatorInfo;
H
Haojun Liao 已提交
6060 6061 6062 6063
  code = appendDownstream(pOperator, &downstream, 1);
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
6064 6065

  return pOperator;
H
Haojun Liao 已提交
6066 6067 6068

_error:
  return NULL;
6069 6070
}

H
Haojun Liao 已提交
6071 6072 6073 6074 6075 6076 6077 6078 6079 6080 6081
static SArray* setRowTsColumnOutputInfo(SqlFunctionCtx* pCtx, int32_t numOfCols) {
  SArray* pList = taosArrayInit(4, sizeof(int32_t));
  for(int32_t i = 0; i < numOfCols; ++i) {
    if (fmIsPseudoColumnFunc(pCtx[i].functionId)) {
      taosArrayPush(pList, &i);
    }
  }

  return pList;
}

L
Liu Jicong 已提交
6082
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t num,
H
Haojun Liao 已提交
6083
                                         SSDataBlock* pResBlock, SLimit* pLimit, SLimit* pSlimit, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
6084
  SProjectOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SProjectOperatorInfo));
L
Liu Jicong 已提交
6085
  SOperatorInfo*        pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
6086 6087 6088
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
6089

H
Haojun Liao 已提交
6090
  pInfo->limit      = *pLimit;
H
Haojun Liao 已提交
6091
  pInfo->slimit     = *pSlimit;
H
Haojun Liao 已提交
6092
  pInfo->curOffset  = pLimit->offset;
H
Haojun Liao 已提交
6093 6094
  pInfo->curSOffset = pSlimit->offset;

H
Haojun Liao 已提交
6095
  pInfo->binfo.pRes = pResBlock;
H
Haojun Liao 已提交
6096 6097 6098

  int32_t numOfCols = num;
  int32_t numOfRows = 4096;
6099 6100
  size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
  initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResBlock, keyBufSize, pTaskInfo->id.str);
H
Haojun Liao 已提交
6101
  setFunctionResultOutput(&pInfo->binfo, &pInfo->aggSup, MAIN_SCAN, pTaskInfo);
H
Haojun Liao 已提交
6102
  pInfo->pPseudoColInfo = setRowTsColumnOutputInfo(pInfo->binfo.pCtx, numOfCols);
6103

H
Haojun Liao 已提交
6104
  pOperator->name         = "ProjectOperator";
H
Haojun Liao 已提交
6105
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_PROJECT;
6106
  pOperator->blockingOptr = false;
H
Haojun Liao 已提交
6107 6108 6109 6110 6111 6112 6113
  pOperator->status       = OP_NOT_OPENED;
  pOperator->info         = pInfo;
  pOperator->pExpr        = pExprInfo;
  pOperator->numOfOutput  = num;
  pOperator->_openFn      = operatorDummyOpenFn;
  pOperator->getNextFn    = doProjectOperation;
  pOperator->closeFn      = destroyProjectOperatorInfo;
L
Liu Jicong 已提交
6114 6115

  pOperator->pTaskInfo = pTaskInfo;
6116
  int32_t code = appendDownstream(pOperator, &downstream, 1);
H
Haojun Liao 已提交
6117
  if (code != TSDB_CODE_SUCCESS) {
H
Haojun Liao 已提交
6118 6119
    goto _error;
  }
6120 6121

  return pOperator;
H
Haojun Liao 已提交
6122

L
Liu Jicong 已提交
6123
_error:
H
Haojun Liao 已提交
6124 6125
  pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
  return NULL;
6126 6127
}

L
Liu Jicong 已提交
6128
SOperatorInfo* createIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
6129
                                          SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlot,
6130
                                          const STableGroupInfo* pTableGroupInfo, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
6131
  STableIntervalOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableIntervalOperatorInfo));
L
Liu Jicong 已提交
6132
  SOperatorInfo*              pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
6133 6134 6135
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
6136

H
Haojun Liao 已提交
6137 6138
  pInfo->order     = TSDB_ORDER_ASC;
  pInfo->interval  = *pInterval;
6139 6140 6141
  pInfo->execModel = pTaskInfo->execModel;

  pInfo->win       = pTaskInfo->window;
6142 6143
  pInfo->win.skey  = 0;
  pInfo->win.ekey  = INT64_MAX;
6144

6145 6146
  pInfo->primaryTsIndex = primaryTsSlot;

6147
  int32_t numOfRows = 4096;
6148 6149
  size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
  int32_t code = initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResBlock, keyBufSize, pTaskInfo->id.str);
6150 6151
  initExecTimeWindowInfo(&pInfo->timeWindowData, &pInfo->win);

L
Liu Jicong 已提交
6152 6153
  //  pInfo->pTableQueryInfo = initTableQueryInfo(pTableGroupInfo);
  if (code != TSDB_CODE_SUCCESS /* || pInfo->pTableQueryInfo == NULL*/) {
H
Haojun Liao 已提交
6154 6155
    goto _error;
  }
H
Haojun Liao 已提交
6156 6157

  initResultRowInfo(&pInfo->binfo.resultRowInfo, (int32_t)1);
6158

6159
  pOperator->name         = "TimeIntervalAggOperator";
H
Haojun Liao 已提交
6160
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_INTERVAL;
6161
  pOperator->blockingOptr = true;
H
Haojun Liao 已提交
6162 6163 6164 6165 6166 6167 6168
  pOperator->status       = OP_NOT_OPENED;
  pOperator->pExpr        = pExprInfo;
  pOperator->pTaskInfo    = pTaskInfo;
  pOperator->numOfOutput  = numOfCols;
  pOperator->info         = pInfo;
  pOperator->_openFn      = doOpenIntervalAgg;
  pOperator->getNextFn    = doBuildIntervalResult;
6169
  pOperator->getStreamResFn= doStreamIntervalAgg;
H
Haojun Liao 已提交
6170
  pOperator->closeFn      = destroyIntervalOperatorInfo;
6171 6172
  pOperator->encodeResultRow = aggEncodeResultRow;
  pOperator->decodeResultRow = aggDecodeResultRow;
6173

H
Haojun Liao 已提交
6174
  code = appendDownstream(pOperator, &downstream, 1);
H
Haojun Liao 已提交
6175 6176 6177 6178
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }

6179
  return pOperator;
H
Haojun Liao 已提交
6180

L
Liu Jicong 已提交
6181
_error:
H
Haojun Liao 已提交
6182
  destroyIntervalOperatorInfo(pInfo, numOfCols);
wafwerar's avatar
wafwerar 已提交
6183 6184
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
6185 6186
  pTaskInfo->code = code;
  return NULL;
6187 6188
}

6189 6190 6191 6192 6193 6194
SOperatorInfo* createTimeSliceOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResultBlock, SExecTaskInfo* pTaskInfo) {
  STimeSliceOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STimeSliceOperatorInfo));
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pOperator == NULL || pInfo == NULL) {
    goto _error;
  }
6195

H
Haojun Liao 已提交
6196
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
6197

6198
  pOperator->name         = "TimeSliceOperator";
L
Liu Jicong 已提交
6199
  //  pOperator->operatorType = OP_AllTimeWindow;
6200
  pOperator->blockingOptr = true;
6201 6202 6203 6204 6205 6206 6207
  pOperator->status       = OP_NOT_OPENED;
  pOperator->pExpr        = pExprInfo;
  pOperator->numOfOutput  = numOfCols;
  pOperator->info         = pInfo;
  pOperator->pTaskInfo    = pTaskInfo;
  pOperator->getNextFn    = doAllIntervalAgg;
  pOperator->closeFn      = destroyBasicOperatorInfo;
6208

L
Liu Jicong 已提交
6209
  int32_t code = appendDownstream(pOperator, &downstream, 1);
6210
  return pOperator;
6211 6212 6213 6214 6215 6216

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

H
Haojun Liao 已提交
6219
SOperatorInfo* createStatewindowOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExpr, int32_t numOfCols, SSDataBlock* pResBlock, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
6220
  SStateWindowOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SStateWindowOperatorInfo));
6221 6222 6223 6224
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
6225

L
Liu Jicong 已提交
6226
  pInfo->colIndex = -1;
6227 6228
  size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
  initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExpr, numOfCols, 4096, pResBlock, keyBufSize, pTaskInfo->id.str);
6229
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
6230

6231 6232
  pOperator->name         = "StateWindowOperator";
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_STATE_WINDOW;
6233
  pOperator->blockingOptr = true;
H
Haojun Liao 已提交
6234 6235 6236 6237 6238 6239 6240 6241
  pOperator->status       = OP_NOT_OPENED;
  pOperator->pExpr        = pExpr;
  pOperator->numOfOutput  = numOfCols;

  pOperator->pTaskInfo    = pTaskInfo;
  pOperator->info         = pInfo;
  pOperator->getNextFn    = doStateWindowAgg;
  pOperator->closeFn      = destroyStateWindowOperatorInfo;
6242 6243
  pOperator->encodeResultRow = aggEncodeResultRow;
  pOperator->decodeResultRow = aggDecodeResultRow;
6244

H
Haojun Liao 已提交
6245
  int32_t code = appendDownstream(pOperator, &downstream, 1);
6246
  return pOperator;
6247 6248 6249 6250

  _error:
  pTaskInfo->code = TSDB_CODE_SUCCESS;
  return NULL;
6251 6252
}

L
Liu Jicong 已提交
6253 6254
SOperatorInfo* createSessionAggOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                            SSDataBlock* pResBlock, int64_t gap, SExecTaskInfo* pTaskInfo) {
wafwerar's avatar
wafwerar 已提交
6255
  SSessionAggOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SSessionAggOperatorInfo));
L
Liu Jicong 已提交
6256
  SOperatorInfo*           pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
6257 6258 6259
  if (pInfo == NULL || pOperator == NULL) {
    goto _error;
  }
H
Haojun Liao 已提交
6260

H
Haojun Liao 已提交
6261
  int32_t numOfRows = 4096;
6262 6263
  size_t keyBufSize = sizeof(int64_t) + sizeof(int64_t) + POINTER_BYTES;
  int32_t code = initAggInfo(&pInfo->binfo, &pInfo->aggSup, pExprInfo, numOfCols, numOfRows, pResBlock, keyBufSize, pTaskInfo->id.str);
H
Haojun Liao 已提交
6264 6265 6266
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
6267

H
Haojun Liao 已提交
6268
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
6269
  initExecTimeWindowInfo(&pInfo->timeWindowData, &pTaskInfo->window);
6270

6271 6272
  pInfo->gap              = gap;
  pInfo->binfo.pRes       = pResBlock;
6273
  pInfo->winSup.prevTs    = INT64_MIN;
6274 6275
  pInfo->reptScan         = false;
  pOperator->name         = "SessionWindowAggOperator";
H
Haojun Liao 已提交
6276
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SESSION_WINDOW;
6277
  pOperator->blockingOptr = true;
6278 6279 6280 6281 6282 6283
  pOperator->status       = OP_NOT_OPENED;
  pOperator->pExpr        = pExprInfo;
  pOperator->numOfOutput  = numOfCols;
  pOperator->info         = pInfo;
  pOperator->getNextFn    = doSessionWindowAgg;
  pOperator->closeFn      = destroySWindowOperatorInfo;
6284 6285
  pOperator->encodeResultRow = aggEncodeResultRow;
  pOperator->decodeResultRow = aggDecodeResultRow;
6286
  pOperator->pTaskInfo    = pTaskInfo;
6287

H
Haojun Liao 已提交
6288
  code = appendDownstream(pOperator, &downstream, 1);
6289
  return pOperator;
H
Haojun Liao 已提交
6290

L
Liu Jicong 已提交
6291
_error:
H
Haojun Liao 已提交
6292 6293 6294 6295
  if (pInfo != NULL) {
    destroySWindowOperatorInfo(pInfo, numOfCols);
  }

wafwerar's avatar
wafwerar 已提交
6296 6297
  taosMemoryFreeClear(pInfo);
  taosMemoryFreeClear(pOperator);
H
Haojun Liao 已提交
6298 6299
  pTaskInfo->code = code;
  return NULL;
6300 6301
}

L
Liu Jicong 已提交
6302 6303
SOperatorInfo* createMultiTableTimeIntervalOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream,
                                                        SExprInfo* pExpr, int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
6304
  STableIntervalOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableIntervalOperatorInfo));
6305

H
Haojun Liao 已提交
6306
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
6307

wafwerar's avatar
wafwerar 已提交
6308
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
6309 6310
  pOperator->name = "MultiTableTimeIntervalOperator";
  //  pOperator->operatorType = OP_MultiTableTimeInterval;
6311
  pOperator->blockingOptr = true;
L
Liu Jicong 已提交
6312 6313 6314 6315
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->info = pInfo;
6316

H
Haojun Liao 已提交
6317
  pOperator->getNextFn = doSTableIntervalAgg;
H
Haojun Liao 已提交
6318
  pOperator->closeFn = destroyBasicOperatorInfo;
6319

L
Liu Jicong 已提交
6320
  int32_t code = appendDownstream(pOperator, &downstream, 1);
6321 6322 6323
  return pOperator;
}

L
Liu Jicong 已提交
6324 6325
SOperatorInfo* createAllMultiTableTimeIntervalOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream,
                                                           SExprInfo* pExpr, int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
6326
  STableIntervalOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(STableIntervalOperatorInfo));
6327

H
Haojun Liao 已提交
6328
  initResultRowInfo(&pInfo->binfo.resultRowInfo, 8);
6329

wafwerar's avatar
wafwerar 已提交
6330
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
6331 6332
  pOperator->name = "AllMultiTableTimeIntervalOperator";
  //  pOperator->operatorType = OP_AllMultiTableTimeInterval;
6333
  pOperator->blockingOptr = true;
wmmhello's avatar
wmmhello 已提交
6334

L
Liu Jicong 已提交
6335 6336 6337 6338
  pOperator->status = OP_NOT_OPENED;
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
  pOperator->info = pInfo;
6339

H
Haojun Liao 已提交
6340
  pOperator->getNextFn = doAllSTableIntervalAgg;
H
Haojun Liao 已提交
6341
  pOperator->closeFn = destroyBasicOperatorInfo;
6342

L
Liu Jicong 已提交
6343
  int32_t code = appendDownstream(pOperator, &downstream, 1);
6344 6345 6346 6347

  return pOperator;
}

H
Haojun Liao 已提交
6348
static int32_t initFillInfo(SFillOperatorInfo* pInfo, SExprInfo* pExpr, int32_t numOfCols, int64_t* fillVal,
L
Liu Jicong 已提交
6349
                            STimeWindow win, int32_t capacity, const char* id, SInterval* pInterval, int32_t fillType) {
H
Haojun Liao 已提交
6350 6351 6352 6353 6354 6355 6356 6357 6358 6359
  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 已提交
6360 6361
  pInfo->pFillInfo = taosCreateFillInfo(order, w.skey, 0, capacity, numOfCols, pInterval->sliding,
                                        pInterval->slidingUnit, (int8_t)pInterval->precision, fillType, pColInfo, id);
H
Haojun Liao 已提交
6362

wafwerar's avatar
wafwerar 已提交
6363
  pInfo->p = taosMemoryCalloc(numOfCols, POINTER_BYTES);
H
Haojun Liao 已提交
6364 6365 6366 6367 6368 6369 6370 6371

  if (pInfo->pFillInfo == NULL || pInfo->p == NULL) {
    return TSDB_CODE_OUT_OF_MEMORY;
  } else {
    return TSDB_CODE_SUCCESS;
  }
}

L
Liu Jicong 已提交
6372 6373 6374
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 已提交
6375
  SFillOperatorInfo* pInfo = taosMemoryCalloc(1, sizeof(SFillOperatorInfo));
L
Liu Jicong 已提交
6376
  SOperatorInfo*     pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
H
Haojun Liao 已提交
6377

L
Liu Jicong 已提交
6378
  pInfo->pRes = pResBlock;
6379
  pInfo->multigroupResult = multigroupResult;
L
Liu Jicong 已提交
6380
  pInfo->intervalInfo = *pInterval;
6381

H
Haojun Liao 已提交
6382
  SResultInfo* pResultInfo = &pOperator->resultInfo;
L
Liu Jicong 已提交
6383 6384
  int32_t      code = initFillInfo(pInfo, pExpr, numOfCols, (int64_t*)fillVal, pTaskInfo->window, pResultInfo->capacity,
                                   pTaskInfo->id.str, pInterval, fillType);
6385 6386 6387
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
  }
6388

L
Liu Jicong 已提交
6389
  pOperator->name = "FillOperator";
6390
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
6391 6392 6393 6394 6395 6396 6397 6398
  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;
6399

L
Liu Jicong 已提交
6400
  pOperator->closeFn = destroySFillOperatorInfo;
H
Haojun Liao 已提交
6401

6402
  code = appendDownstream(pOperator, &downstream, 1);
6403
  return pOperator;
H
Haojun Liao 已提交
6404

L
Liu Jicong 已提交
6405
_error:
wafwerar's avatar
wafwerar 已提交
6406 6407
  taosMemoryFreeClear(pOperator);
  taosMemoryFreeClear(pInfo);
H
Haojun Liao 已提交
6408
  return NULL;
6409 6410
}

L
Liu Jicong 已提交
6411
static SSDataBlock* doTagScan(SOperatorInfo* pOperator, bool* newgroup) {
6412 6413 6414 6415 6416 6417
#if 0
  SOperatorInfo* pOperator = (SOperatorInfo*) param;
  if (pOperator->status == OP_EXEC_DONE) {
    return NULL;
  }

H
Haojun Liao 已提交
6418
  int32_t maxNumOfTables = (int32_t)pResultInfo->capacity;
6419 6420 6421 6422 6423 6424 6425 6426 6427 6428 6429 6430 6431 6432 6433 6434 6435 6436 6437 6438 6439

  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
    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) {
6440
      if (pQueryAttr->tagColList[i].colId == pExprInfo->base.pColumns->info.colId) {
6441 6442 6443 6444 6445 6446 6447 6448 6449 6450 6451 6452 6453 6454 6455 6456 6457 6458 6459 6460 6461 6462 6463 6464 6465 6466 6467 6468 6469 6470 6471
        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;
6472
      if (pExprInfo->base.pColumns->info.colId == TSDB_TBNAME_COLUMN_INDEX) {
6473 6474
        data = tsdbGetTableName(item->pTable);
      } else {
6475
        data = tsdbGetTableTagVal(item->pTable, pExprInfo->base.pColumns->info.colId, type, bytes);
6476 6477 6478 6479 6480 6481
      }

      doSetTagValueToResultBuf(output, data, type, bytes);
      count += 1;
    }

6482
    //qDebug("QInfo:0x%"PRIx64" create (tableId, tag) info completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
6483 6484 6485 6486 6487 6488
  } 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;
6489
    //qDebug("QInfo:0x%"PRIx64" create count(tbname) query, res:%d rows:1", GET_TASKID(pRuntimeEnv), count);
6490 6491 6492 6493 6494 6495 6496 6497 6498 6499 6500 6501 6502
  } 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
6503
        if (TSDB_COL_IS_UD_COL(pExprInfo[j].base.pColumns->flag)) {
6504 6505 6506 6507 6508 6509 6510
          continue;
        }

        SColumnInfoData* pColInfo = taosArrayGet(pRes->pDataBlock, j);
        type  = pExprInfo[j].base.resSchema.type;
        bytes = pExprInfo[j].base.resSchema.bytes;

6511
        if (pExprInfo[j].base.pColumns->info.colId == TSDB_TBNAME_COLUMN_INDEX) {
6512 6513
          data = tsdbGetTableName(item->pTable);
        } else {
6514
          data = tsdbGetTableTagVal(item->pTable, pExprInfo[j].base.pColumns->info.colId, type, bytes);
6515 6516 6517 6518 6519 6520 6521 6522 6523 6524 6525 6526 6527
        }

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

6528
    //qDebug("QInfo:0x%"PRIx64" create tag values results completed, rows:%d", GET_TASKID(pRuntimeEnv), count);
6529 6530 6531
  }

  if (pOperator->status == OP_EXEC_DONE) {
6532
    setTaskStatus(pOperator->pRuntimeEnv, TASK_COMPLETED);
6533 6534 6535 6536 6537 6538
  }

  pRes->info.rows = count;
  return (pRes->info.rows == 0)? NULL:pInfo->pRes;

#endif
6539
  return TSDB_CODE_SUCCESS;
6540 6541
}

H
Haojun Liao 已提交
6542
SOperatorInfo* createTagScanOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SExprInfo* pExpr, int32_t numOfOutput) {
wafwerar's avatar
wafwerar 已提交
6543
  STagScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STagScanInfo));
L
Liu Jicong 已提交
6544
  //  pInfo->pRes = createOutputBuf(pExpr, numOfOutput, pResultInfo->capacity);
6545 6546 6547 6548 6549 6550

  size_t numOfGroup = GET_NUM_OF_TABLEGROUP(pRuntimeEnv);
  assert(numOfGroup == 0 || numOfGroup == 1);

  pInfo->curPos = 0;

wafwerar's avatar
wafwerar 已提交
6551
  SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
L
Liu Jicong 已提交
6552
  pOperator->name = "SeqTableTagScan";
X
Xiaoyu Wang 已提交
6553
  pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TAG_SCAN;
6554
  pOperator->blockingOptr = false;
L
Liu Jicong 已提交
6555 6556
  pOperator->status = OP_NOT_OPENED;
  pOperator->info = pInfo;
H
Haojun Liao 已提交
6557
  pOperator->getNextFn = doTagScan;
L
Liu Jicong 已提交
6558 6559
  pOperator->pExpr = pExpr;
  pOperator->numOfOutput = numOfOutput;
H
Haojun Liao 已提交
6560
  pOperator->closeFn = destroyTagScanOperatorInfo;
6561 6562 6563

  return pOperator;
}
H
Haojun Liao 已提交
6564

6565

L
Liu Jicong 已提交
6566
static int32_t getColumnIndexInSource(SQueriedTableInfo* pTableInfo, SExprBasicInfo* pExpr, SColumnInfo* pTagCols) {
6567 6568
  int32_t j = 0;

H
Haojun Liao 已提交
6569 6570
  if (TSDB_COL_IS_TAG(pExpr->pParam[0].pCol->type)) {
    if (pExpr->pParam[0].pCol->colId == TSDB_TBNAME_COLUMN_INDEX) {
6571 6572 6573
      return TSDB_TBNAME_COLUMN_INDEX;
    }

L
Liu Jicong 已提交
6574
    while (j < pTableInfo->numOfTags) {
H
Haojun Liao 已提交
6575
      if (pExpr->pParam[0].pCol->colId == pTagCols[j].colId) {
6576 6577 6578 6579 6580 6581
        return j;
      }

      j += 1;
    }

6582
  } /*else if (TSDB_COL_IS_UD_COL(pExpr->colInfo.flag)) {  // user specified column data
6583 6584 6585 6586 6587 6588 6589 6590 6591
    return TSDB_UD_COLUMN_INDEX;
  } else {
    while (j < pTableInfo->numOfCols) {
      if (pExpr->colInfo.colId == pTableInfo->colList[j].colId) {
        return j;
      }

      j += 1;
    }
6592
  }*/
6593 6594 6595 6596

  return INT32_MIN;  // return a less than TSDB_TBNAME_COLUMN_INDEX value
}

L
Liu Jicong 已提交
6597
bool validateExprColumnInfo(SQueriedTableInfo* pTableInfo, SExprBasicInfo* pExpr, SColumnInfo* pTagCols) {
6598 6599 6600 6601
  int32_t j = getColumnIndexInSource(pTableInfo, pExpr, pTagCols);
  return j != INT32_MIN;
}

L
Liu Jicong 已提交
6602 6603
static SResSchema createResSchema(int32_t type, int32_t bytes, int32_t slotId, int32_t scale, int32_t precision,
                                  const char* name) {
H
Haojun Liao 已提交
6604
  SResSchema s = {0};
6605 6606 6607 6608
  s.scale  = scale;
  s.type   = type;
  s.bytes  = bytes;
  s.slotId = slotId;
H
Haojun Liao 已提交
6609
  s.precision = precision;
H
Haojun Liao 已提交
6610 6611 6612 6613
  strncpy(s.name, name, tListLen(s.name));

  return s;
}
H
Haojun Liao 已提交
6614

H
Haojun Liao 已提交
6615 6616 6617 6618 6619 6620 6621 6622 6623 6624 6625 6626 6627 6628 6629 6630 6631
static SColumn* createColumn(int32_t blockId, int32_t slotId, SDataType* pType) {
  SColumn* pCol = taosMemoryCalloc(1, sizeof(SColumn));
  if (pCol == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

  pCol->slotId      = slotId;
  pCol->bytes       = pType->bytes;
  pCol->type        = pType->type;
  pCol->scale       = pType->scale;
  pCol->precision   = pType->precision;
  pCol->dataBlockId = blockId;

  return pCol;
}

H
Haojun Liao 已提交
6632
SExprInfo* createExprInfo(SNodeList* pNodeList, SNodeList* pGroupKeys, int32_t* numOfExprs) {
H
Haojun Liao 已提交
6633
  int32_t numOfFuncs = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
6634 6635 6636 6637
  int32_t numOfGroupKeys = 0;
  if (pGroupKeys != NULL) {
    numOfGroupKeys = LIST_LENGTH(pGroupKeys);
  }
H
Haojun Liao 已提交
6638

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

L
Liu Jicong 已提交
6642
  for (int32_t i = 0; i < (*numOfExprs); ++i) {
H
Haojun Liao 已提交
6643 6644 6645 6646 6647 6648
    STargetNode* pTargetNode = NULL;
    if (i < numOfFuncs) {
      pTargetNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    } else {
      pTargetNode = (STargetNode*)nodesListGetNode(pGroupKeys, i - numOfFuncs);
    }
H
Haojun Liao 已提交
6649

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

wafwerar's avatar
wafwerar 已提交
6652
    pExp->pExpr = taosMemoryCalloc(1, sizeof(tExprNode));
H
Haojun Liao 已提交
6653
    pExp->pExpr->_function.num = 1;
H
Haojun Liao 已提交
6654
    pExp->pExpr->_function.functionId = -1;
H
Haojun Liao 已提交
6655

H
Haojun Liao 已提交
6656
    // it is a project query, or group by column
H
Haojun Liao 已提交
6657
    if (nodeType(pTargetNode->pExpr) == QUERY_NODE_COLUMN) {
H
Haojun Liao 已提交
6658
      pExp->pExpr->nodeType = QUERY_NODE_COLUMN;
L
Liu Jicong 已提交
6659
      SColumnNode* pColNode = (SColumnNode*)pTargetNode->pExpr;
H
Haojun Liao 已提交
6660

G
Ganlin Zhao 已提交
6661 6662 6663
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;

H
Haojun Liao 已提交
6664 6665
      SDataType* pType = &pColNode->node.resType;
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale, pType->precision, pColNode->colName);
H
Haojun Liao 已提交
6666 6667
      pExp->base.pParam[0].pCol = createColumn(pColNode->dataBlockId, pColNode->slotId, pType);
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
6668
    } else if (nodeType(pTargetNode->pExpr) == QUERY_NODE_VALUE) {
6669 6670 6671 6672 6673 6674 6675 6676 6677 6678
      pExp->pExpr->nodeType = QUERY_NODE_VALUE;
      SValueNode* pValNode = (SValueNode*)pTargetNode->pExpr;

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

      SDataType* pType = &pValNode->node.resType;
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale, pType->precision, pValNode->node.aliasName);
      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_VALUE;
      valueNodeToVariant(pValNode, &pExp->base.pParam[0].param);
6679
    } else if (nodeType(pTargetNode->pExpr) == QUERY_NODE_FUNCTION) {
H
Haojun Liao 已提交
6680
      pExp->pExpr->nodeType = QUERY_NODE_FUNCTION;
H
Haojun Liao 已提交
6681 6682 6683
      SFunctionNode* pFuncNode = (SFunctionNode*)pTargetNode->pExpr;

      SDataType* pType = &pFuncNode->node.resType;
L
Liu Jicong 已提交
6684 6685
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale,
                                             pType->precision, pFuncNode->node.aliasName);
H
Haojun Liao 已提交
6686

H
Haojun Liao 已提交
6687
      pExp->pExpr->_function.functionId = pFuncNode->funcId;
H
Haojun Liao 已提交
6688
      pExp->pExpr->_function.pFunctNode = pFuncNode;
H
Haojun Liao 已提交
6689
      strncpy(pExp->pExpr->_function.functionName, pFuncNode->functionName, tListLen(pExp->pExpr->_function.functionName));
H
Haojun Liao 已提交
6690 6691 6692

      // TODO: value parameter needs to be handled
      int32_t numOfParam = LIST_LENGTH(pFuncNode->pParameterList);
G
Ganlin Zhao 已提交
6693 6694 6695 6696

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

H
Haojun Liao 已提交
6697 6698
      for (int32_t j = 0; j < numOfParam; ++j) {
        SNode*       p1 = nodesListGetNode(pFuncNode->pParameterList, j);
G
Ganlin Zhao 已提交
6699
        if (p1->type == QUERY_NODE_COLUMN) {
H
Haojun Liao 已提交
6700
          SColumnNode* pcn = (SColumnNode*) p1;
G
Ganlin Zhao 已提交
6701 6702

          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_COLUMN;
H
Haojun Liao 已提交
6703
          pExp->base.pParam[j].pCol = createColumn(pcn->dataBlockId, pcn->slotId, &pcn->node.resType);
G
Ganlin Zhao 已提交
6704 6705 6706 6707
        } else if (p1->type == QUERY_NODE_VALUE) {
          SValueNode* pvn = (SValueNode*)p1;
          pExp->base.pParam[j].type = FUNC_PARAM_TYPE_VALUE;
        }
H
Haojun Liao 已提交
6708
      }
6709
    } else if (nodeType(pTargetNode->pExpr) == QUERY_NODE_OPERATOR) {
H
Haojun Liao 已提交
6710
      pExp->pExpr->nodeType = QUERY_NODE_OPERATOR;
L
Liu Jicong 已提交
6711
      SOperatorNode* pNode = (SOperatorNode*)pTargetNode->pExpr;
6712

G
Ganlin Zhao 已提交
6713 6714 6715
      pExp->base.pParam = taosMemoryCalloc(1, sizeof(SFunctParam));
      pExp->base.numOfParams = 1;

6716
      SDataType* pType = &pNode->node.resType;
H
Haojun Liao 已提交
6717
      pExp->base.resSchema = createResSchema(pType->type, pType->bytes, pTargetNode->slotId, pType->scale, pType->precision, pNode->node.aliasName);
6718 6719 6720

      pExp->pExpr->_optrRoot.pRootNode = pTargetNode->pExpr;

H
Haojun Liao 已提交
6721 6722
//      pExp->base.pParam[0].type = FUNC_PARAM_TYPE_COLUMN;
//      pExp->base.pParam[0].pCol = createColumn(pTargetNode->dataBlockId, pTargetNode->slotId, pType);
6723 6724
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
6725 6726 6727
    }
  }

H
Haojun Liao 已提交
6728
  return pExprs;
H
Haojun Liao 已提交
6729 6730
}

6731
static SExecTaskInfo* createExecTaskInfo(uint64_t queryId, uint64_t taskId, EOPTR_EXEC_MODEL model) {
wafwerar's avatar
wafwerar 已提交
6732
  SExecTaskInfo* pTaskInfo = taosMemoryCalloc(1, sizeof(SExecTaskInfo));
6733
  setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
H
Haojun Liao 已提交
6734

6735
  pTaskInfo->cost.created = taosGetTimestampMs();
H
Haojun Liao 已提交
6736
  pTaskInfo->id.queryId = queryId;
6737
  pTaskInfo->execModel  = model;
H
Haojun Liao 已提交
6738

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

6743 6744
  return pTaskInfo;
}
H
Haojun Liao 已提交
6745

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

L
Liu Jicong 已提交
6749 6750
static int32_t doCreateTableGroup(void* metaHandle, int32_t tableType, uint64_t tableUid, STableGroupInfo* pGroupInfo,
                                  uint64_t queryId, uint64_t taskId);
6751 6752
static SArray* extractTableIdList(const STableGroupInfo* pTableGroupInfo);
static SArray* extractScanColumnId(SNodeList* pNodeList);
H
Haojun Liao 已提交
6753
static SArray* extractColumnInfo(SNodeList* pNodeList);
H
Haojun Liao 已提交
6754
static SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols);
6755

6756
static SArray* createSortInfo(SNodeList* pNodeList, SNodeList* pNodeListTarget);
6757
static SArray* createIndexMap(SNodeList* pNodeList);
6758
static SArray* extractPartitionColInfo(SNodeList* pNodeList);
H
Haojun Liao 已提交
6759

H
Haojun Liao 已提交
6760
SOperatorInfo* createOperatorTree(SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, SReadHandle* pHandle,
L
Liu Jicong 已提交
6761
                                        uint64_t queryId, uint64_t taskId, STableGroupInfo* pTableGroupInfo) {
X
Xiaoyu Wang 已提交
6762
  if (pPhyNode->pChildren == NULL || LIST_LENGTH(pPhyNode->pChildren) == 0) {
H
Haojun Liao 已提交
6763 6764
    int32_t type = nodeType(pPhyNode);
    if (QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN == type) {
X
Xiaoyu Wang 已提交
6765
      SScanPhysiNode* pScanPhyNode = (SScanPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
6766

H
Haojun Liao 已提交
6767
      int32_t     numOfCols = 0;
H
Haojun Liao 已提交
6768 6769
      tsdbReaderT pDataReader = doCreateDataReader((STableScanPhysiNode*)pPhyNode, pHandle, pTableGroupInfo, (uint64_t)queryId, taskId);
      SArray* pColList = extractColMatchInfo(pScanPhyNode->pScanCols, pScanPhyNode->node.pOutputDataBlockDesc, &numOfCols);
6770 6771
      SSDataBlock* pResBlock = createOutputBuf_rv1(pScanPhyNode->node.pOutputDataBlockDesc);

L
Liu Jicong 已提交
6772
      return createTableScanOperatorInfo(pDataReader, pScanPhyNode->order, numOfCols, pScanPhyNode->count,
6773
                                         pScanPhyNode->reverse, pColList, pResBlock, pScanPhyNode->node.pConditions, pTaskInfo);
H
Haojun Liao 已提交
6774
    } else if (QUERY_NODE_PHYSICAL_PLAN_EXCHANGE == type) {
H
Haojun Liao 已提交
6775
      SExchangePhysiNode* pExchange = (SExchangePhysiNode*)pPhyNode;
L
Liu Jicong 已提交
6776
      SSDataBlock*        pResBlock = createOutputBuf_rv1(pExchange->node.pOutputDataBlockDesc);
H
Haojun Liao 已提交
6777
      return createExchangeOperatorInfo(pExchange->pSrcEndPoints, pResBlock, pTaskInfo);
H
Haojun Liao 已提交
6778
    } else if (QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN == type) {
H
Haojun Liao 已提交
6779
      SScanPhysiNode* pScanPhyNode = (SScanPhysiNode*)pPhyNode;  // simple child table.
6780

6781
      int32_t code = doCreateTableGroup(pHandle->meta, pScanPhyNode->tableType, pScanPhyNode->uid, pTableGroupInfo, queryId, taskId);
H
Haojun Liao 已提交
6782
      SArray* tableIdList = extractTableIdList(pTableGroupInfo);
H
Haojun Liao 已提交
6783

6784
      SSDataBlock* pResBlock = createOutputBuf_rv1(pScanPhyNode->node.pOutputDataBlockDesc);
6785 6786 6787 6788

      int32_t numOfCols = 0;
      SArray* pColList = extractColMatchInfo(pScanPhyNode->pScanCols, pScanPhyNode->node.pOutputDataBlockDesc, &numOfCols);
      SOperatorInfo* pOperator = createStreamScanOperatorInfo(pHandle->reader, pResBlock, pColList, tableIdList, pTaskInfo);
6789
      taosArrayDestroy(tableIdList);
H
Haojun Liao 已提交
6790
      return pOperator;
H
Haojun Liao 已提交
6791
    } else if (QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN == type) {
L
Liu Jicong 已提交
6792 6793
      SSystemTableScanPhysiNode* pSysScanPhyNode = (SSystemTableScanPhysiNode*)pPhyNode;
      SSDataBlock*               pResBlock = createOutputBuf_rv1(pSysScanPhyNode->scan.node.pOutputDataBlockDesc);
H
Haojun Liao 已提交
6794

6795
      struct SScanPhysiNode* pScanNode = &pSysScanPhyNode->scan;
L
Liu Jicong 已提交
6796
      SArray*                colList = extractScanColumnId(pScanNode->pScanCols);
6797

L
Liu Jicong 已提交
6798 6799 6800
      SOperatorInfo* pOperator = createSysTableScanOperatorInfo(
          pHandle->meta, pResBlock, &pScanNode->tableName, pScanNode->node.pConditions, pSysScanPhyNode->mgmtEpSet,
          colList, pTaskInfo, pSysScanPhyNode->showRewrite, pSysScanPhyNode->accountId);
H
Haojun Liao 已提交
6801 6802 6803
      return pOperator;
    } else {
      ASSERT(0);
H
Haojun Liao 已提交
6804 6805 6806
    }
  }

H
Haojun Liao 已提交
6807 6808 6809
  int32_t type = nodeType(pPhyNode);
  size_t size = LIST_LENGTH(pPhyNode->pChildren);
  ASSERT(size == 1);
H
Haojun Liao 已提交
6810

H
Haojun Liao 已提交
6811 6812
  SPhysiNode*    pChildNode = (SPhysiNode*)nodesListGetNode(pPhyNode->pChildren, 0);
  SOperatorInfo* op = createOperatorTree(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);
6813
  int32_t        num = 0;
H
Haojun Liao 已提交
6814

H
Haojun Liao 已提交
6815
  if (QUERY_NODE_PHYSICAL_PLAN_PROJECT == type) {
H
Haojun Liao 已提交
6816 6817 6818
    SProjectPhysiNode* pProjPhyNode = (SProjectPhysiNode*) pPhyNode;
    SExprInfo*   pExprInfo = createExprInfo(pProjPhyNode->pProjections, NULL, &num);

H
Haojun Liao 已提交
6819
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
H
Haojun Liao 已提交
6820
    SLimit limit = {.limit = pProjPhyNode->limit, .offset = pProjPhyNode->offset};
H
Haojun Liao 已提交
6821 6822
    SLimit slimit = {.limit = pProjPhyNode->slimit, .offset = pProjPhyNode->soffset};
    return createProjectOperatorInfo(op, pExprInfo, num, pResBlock, &limit, &slimit, pTaskInfo);
H
Haojun Liao 已提交
6823 6824 6825 6826
  } else if (QUERY_NODE_PHYSICAL_PLAN_AGG == type) {
    SAggPhysiNode* pAggNode = (SAggPhysiNode*)pPhyNode;
    SExprInfo*     pExprInfo = createExprInfo(pAggNode->pAggFuncs, pAggNode->pGroupKeys, &num);
    SSDataBlock*   pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
H
Haojun Liao 已提交
6827

6828 6829
    SExprInfo* pScalarExprInfo = NULL;
    int32_t numOfScalarExpr = 0;
H
Haojun Liao 已提交
6830 6831
    if (pAggNode->pGroupKeys != NULL) {
      SArray* pColList = extractColumnInfo(pAggNode->pGroupKeys);
6832 6833 6834 6835 6836
      if (pAggNode->pExprs != NULL) {
        pScalarExprInfo = createExprInfo(pAggNode->pExprs, NULL, &numOfScalarExpr);
      }

      return createGroupOperatorInfo(op, pExprInfo, num, pResBlock, pColList, pAggNode->node.pConditions, pScalarExprInfo, numOfScalarExpr, pTaskInfo, NULL);
H
Haojun Liao 已提交
6837 6838
    } else {
      return createAggregateOperatorInfo(op, pExprInfo, num, pResBlock, pTaskInfo, pTableGroupInfo);
H
Haojun Liao 已提交
6839
    }
H
Haojun Liao 已提交
6840 6841
  } else if (QUERY_NODE_PHYSICAL_PLAN_INTERVAL == type) {
    SIntervalPhysiNode* pIntervalPhyNode = (SIntervalPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
6842

H
Haojun Liao 已提交
6843 6844
    SExprInfo*   pExprInfo = createExprInfo(pIntervalPhyNode->window.pFuncs, NULL, &num);
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
H
Haojun Liao 已提交
6845

H
Haojun Liao 已提交
6846 6847 6848 6849 6850 6851 6852 6853 6854
    SInterval interval = {
        .interval     = pIntervalPhyNode->interval,
        .sliding      = pIntervalPhyNode->sliding,
        .intervalUnit = pIntervalPhyNode->intervalUnit,
        .slidingUnit  = pIntervalPhyNode->slidingUnit,
        .offset       = pIntervalPhyNode->offset,
        .precision    = pIntervalPhyNode->precision
    };

6855
    int32_t primaryTsSlotId = ((SColumnNode*) pIntervalPhyNode->window.pTspk)->slotId;
H
Haojun Liao 已提交
6856 6857
    return createIntervalOperatorInfo(op, pExprInfo, num, pResBlock, &interval, primaryTsSlotId, pTableGroupInfo, pTaskInfo);
  } else if (QUERY_NODE_PHYSICAL_PLAN_SORT == type) {
H
Haojun Liao 已提交
6858
    SSortPhysiNode* pSortPhyNode = (SSortPhysiNode*)pPhyNode;
H
Haojun Liao 已提交
6859

H
Haojun Liao 已提交
6860
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
6861
    SArray*      info = createSortInfo(pSortPhyNode->pSortKeys, pSortPhyNode->pTargets);
6862 6863
    SArray*      slotMap = createIndexMap(pSortPhyNode->pTargets);
    return createSortOperatorInfo(op, pResBlock, info, slotMap, pTaskInfo);
H
Haojun Liao 已提交
6864
  } else if (QUERY_NODE_PHYSICAL_PLAN_SESSION_WINDOW == type) {
H
Haojun Liao 已提交
6865 6866 6867 6868 6869
    SSessionWinodwPhysiNode* pSessionNode = (SSessionWinodwPhysiNode*)pPhyNode;

    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 已提交
6870 6871
  } else if (QUERY_NODE_PHYSICAL_PLAN_PARTITION == type) {
    SPartitionPhysiNode* pPartNode = (SPartitionPhysiNode*) pPhyNode;
6872
    SArray* pColList = extractPartitionColInfo(pPartNode->pPartitionKeys);
H
Haojun Liao 已提交
6873
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
6874 6875 6876

    SExprInfo* pExprInfo = createExprInfo(pPartNode->pTargets, NULL, &num);
    return createPartitionOperatorInfo(op, pExprInfo, num, pResBlock, pColList, pTaskInfo, NULL);
6877 6878 6879 6880 6881 6882
  } else if (QUERY_NODE_STATE_WINDOW == type) {
    SStateWinodwPhysiNode* pStateNode = (SStateWinodwPhysiNode*) pPhyNode;

    SExprInfo* pExprInfo = createExprInfo(pStateNode->window.pFuncs, NULL, &num);
    SSDataBlock* pResBlock = createOutputBuf_rv1(pPhyNode->pOutputDataBlockDesc);
    return createStatewindowOperatorInfo(op, pExprInfo, num, pResBlock, pTaskInfo);
H
Haojun Liao 已提交
6883 6884
  } else {
    ASSERT(0);
L
Liu Jicong 已提交
6885 6886 6887 6888 6889 6890
  } /*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);
H
Haojun Liao 已提交
6891
       SOperatorInfo* op = createOperatorTree(pChildNode, pTaskInfo, pHandle, queryId, taskId, pTableGroupInfo);
L
Liu Jicong 已提交
6892 6893 6894
       return createMultiTableAggOperatorInfo(op, pPhyNode->pTargets, pTaskInfo, pTableGroupInfo);
     }
   }*/
6895
  return NULL;
6896
}
H
Haojun Liao 已提交
6897

L
Liu Jicong 已提交
6898 6899
static tsdbReaderT createDataReaderImpl(STableScanPhysiNode* pTableScanNode, STableGroupInfo* pGroupInfo,
                                        void* readHandle, uint64_t queryId, uint64_t taskId) {
H
Haojun Liao 已提交
6900
  STsdbQueryCond cond = {.loadExternalRows = false};
H
Haojun Liao 已提交
6901

L
Liu Jicong 已提交
6902
  cond.order = pTableScanNode->scan.order;
X
Xiaoyu Wang 已提交
6903
  cond.numOfCols = LIST_LENGTH(pTableScanNode->scan.pScanCols);
L
Liu Jicong 已提交
6904
  cond.colList = taosMemoryCalloc(cond.numOfCols, sizeof(SColumnInfo));
H
Haojun Liao 已提交
6905 6906 6907 6908 6909
  if (cond.colList == NULL) {
    terrno = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return NULL;
  }

X
Xiaoyu Wang 已提交
6910
  cond.twindow = pTableScanNode->scanRange;
H
Haojun Liao 已提交
6911
  cond.type = BLOCK_LOAD_OFFSET_SEQ_ORDER;
L
Liu Jicong 已提交
6912
  //  cond.type = pTableScanNode->scanFlag;
H
Haojun Liao 已提交
6913

H
Haojun Liao 已提交
6914
  int32_t j = 0;
H
Haojun Liao 已提交
6915
  for (int32_t i = 0; i < cond.numOfCols; ++i) {
H
Haojun Liao 已提交
6916 6917
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pTableScanNode->scan.pScanCols, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
6918 6919 6920
    if (pColNode->colType == COLUMN_TYPE_TAG) {
      continue;
    }
H
Haojun Liao 已提交
6921

6922
    cond.colList[j].type  = pColNode->node.resType.type;
H
Haojun Liao 已提交
6923 6924 6925
    cond.colList[j].bytes = pColNode->node.resType.bytes;
    cond.colList[j].colId = pColNode->colId;
    j += 1;
H
Haojun Liao 已提交
6926 6927
  }

H
Haojun Liao 已提交
6928
  cond.numOfCols = j;
H
Haojun Liao 已提交
6929
  return tsdbQueryTables(readHandle, &cond, pGroupInfo, queryId, taskId);
H
Haojun Liao 已提交
6930
}
H
Haojun Liao 已提交
6931

6932
SArray* extractScanColumnId(SNodeList* pNodeList) {
L
Liu Jicong 已提交
6933
  size_t  numOfCols = LIST_LENGTH(pNodeList);
6934 6935 6936 6937 6938 6939
  SArray* pList = taosArrayInit(numOfCols, sizeof(int16_t));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
6940
  for (int32_t i = 0; i < numOfCols; ++i) {
X
Xiaoyu Wang 已提交
6941
    for (int32_t j = 0; j < numOfCols; ++j) {
L
Liu Jicong 已提交
6942
      STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, j);
X
Xiaoyu Wang 已提交
6943
      if (pNode->slotId == i) {
L
Liu Jicong 已提交
6944
        SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
X
Xiaoyu Wang 已提交
6945 6946 6947 6948
        taosArrayPush(pList, &pColNode->colId);
        break;
      }
    }
6949 6950 6951 6952 6953
  }

  return pList;
}

H
Haojun Liao 已提交
6954
SArray* extractColumnInfo(SNodeList* pNodeList) {
L
Liu Jicong 已提交
6955
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
6956 6957 6958 6959 6960 6961
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
6962 6963 6964
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
6965

H
Haojun Liao 已提交
6966
    // todo extract method
H
Haojun Liao 已提交
6967 6968
    SColumn c = {0};
    c.slotId = pColNode->slotId;
H
Haojun Liao 已提交
6969 6970 6971
    c.colId  = pColNode->colId;
    c.type   = pColNode->node.resType.type;
    c.bytes  = pColNode->node.resType.bytes;
L
Liu Jicong 已提交
6972 6973
    c.precision = pColNode->node.resType.precision;
    c.scale = pColNode->node.resType.scale;
H
Haojun Liao 已提交
6974 6975 6976 6977 6978 6979 6980

    taosArrayPush(pList, &c);
  }

  return pList;
}

6981 6982 6983 6984 6985 6986 6987 6988 6989 6990 6991 6992 6993 6994 6995 6996 6997 6998 6999 7000 7001 7002 7003 7004 7005 7006
SArray* extractPartitionColInfo(SNodeList* pNodeList) {
  size_t  numOfCols = LIST_LENGTH(pNodeList);
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColumn));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

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

    // todo extract method
    SColumn c = {0};
    c.slotId = pColNode->slotId;
    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;

    taosArrayPush(pList, &c);
  }

  return pList;
}

7007
SArray* createSortInfo(SNodeList* pNodeList, SNodeList* pNodeListTarget) {
L
Liu Jicong 已提交
7008
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
7009 7010 7011 7012 7013 7014
  SArray* pList = taosArrayInit(numOfCols, sizeof(SBlockOrderInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return pList;
  }

L
Liu Jicong 已提交
7015
  for (int32_t i = 0; i < numOfCols; ++i) {
7016
    
7017
    SOrderByExprNode* pSortKey = (SOrderByExprNode*)nodesListGetNode(pNodeList, i);
L
Liu Jicong 已提交
7018 7019
    SBlockOrderInfo   bi = {0};
    bi.order = (pSortKey->order == ORDER_ASC) ? TSDB_ORDER_ASC : TSDB_ORDER_DESC;
H
Haojun Liao 已提交
7020 7021 7022 7023
    bi.nullFirst = (pSortKey->nullOrder == NULL_ORDER_FIRST);

    SColumnNode* pColNode = (SColumnNode*)pSortKey->pExpr;

7024 7025 7026 7027 7028 7029 7030 7031 7032 7033 7034 7035 7036 7037 7038
    bool found = false;
    for (int32_t j = 0; j < LIST_LENGTH(pNodeListTarget); ++j) {
      STargetNode* pTarget = (STargetNode*)nodesListGetNode(pNodeListTarget, j);

      SColumnNode* pColNodeT = (SColumnNode*)pTarget->pExpr;
      if(pColNode->slotId == pColNodeT->slotId){    // to find slotId in PhysiSort OutputDataBlockDesc
        bi.slotId = pTarget->slotId;
        found = true;
        break;
      }
    }

    if(!found){
      qError("sort slot id does not found");
    }
H
Haojun Liao 已提交
7039 7040 7041 7042 7043 7044
    taosArrayPush(pList, &bi);
  }

  return pList;
}

7045 7046 7047 7048 7049 7050 7051 7052 7053 7054 7055 7056 7057 7058 7059 7060 7061 7062
SArray* createIndexMap(SNodeList* pNodeList) {
  size_t  numOfCols = LIST_LENGTH(pNodeList);
  SArray* pList = taosArrayInit(numOfCols, sizeof(int32_t));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return pList;
  }

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

    SColumnNode* pColNode = (SColumnNode*)pTarget->pExpr;
    taosArrayPush(pList, &pColNode->slotId);
  }

  return pList;
}

H
Haojun Liao 已提交
7063
SArray* extractColMatchInfo(SNodeList* pNodeList, SDataBlockDescNode* pOutputNodeList, int32_t* numOfOutputCols) {
L
Liu Jicong 已提交
7064
  size_t  numOfCols = LIST_LENGTH(pNodeList);
H
Haojun Liao 已提交
7065 7066 7067 7068 7069 7070
  SArray* pList = taosArrayInit(numOfCols, sizeof(SColMatchInfo));
  if (pList == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

L
Liu Jicong 已提交
7071 7072 7073
  for (int32_t i = 0; i < numOfCols; ++i) {
    STargetNode* pNode = (STargetNode*)nodesListGetNode(pNodeList, i);
    SColumnNode* pColNode = (SColumnNode*)pNode->pExpr;
H
Haojun Liao 已提交
7074 7075

    SColMatchInfo c = {0};
L
Liu Jicong 已提交
7076
    c.colId = pColNode->colId;
H
Haojun Liao 已提交
7077
    c.targetSlotId = pNode->slotId;
L
Liu Jicong 已提交
7078
    c.output = true;
H
Haojun Liao 已提交
7079 7080 7081
    taosArrayPush(pList, &c);
  }

H
Haojun Liao 已提交
7082 7083
  *numOfOutputCols = 0;
  int32_t num = LIST_LENGTH(pOutputNodeList->pSlots);
L
Liu Jicong 已提交
7084 7085
  for (int32_t i = 0; i < num; ++i) {
    SSlotDescNode* pNode = (SSlotDescNode*)nodesListGetNode(pOutputNodeList->pSlots, i);
7086 7087 7088 7089 7090 7091
    // todo: add reserve flag check
    if (pNode->slotId >= numOfCols) { // it is a column reserved for the arithmetic expression calculation
      (*numOfOutputCols) += 1;
      continue;
    }

H
Haojun Liao 已提交
7092
    SColMatchInfo* info = taosArrayGet(pList, pNode->slotId);
H
Haojun Liao 已提交
7093 7094 7095 7096 7097
    if (pNode->output) {
      (*numOfOutputCols) += 1;
    } else {
      info->output = false;
    }
H
Haojun Liao 已提交
7098 7099
  }

H
Haojun Liao 已提交
7100 7101 7102
  return pList;
}

H
Haojun Liao 已提交
7103
int32_t doCreateTableGroup(void* metaHandle, int32_t tableType, uint64_t tableUid, STableGroupInfo* pGroupInfo, uint64_t queryId, uint64_t taskId) {
7104
  int32_t code = 0;
H
Haojun Liao 已提交
7105
  if (tableType == TSDB_SUPER_TABLE) {
H
Haojun Liao 已提交
7106
    code = tsdbQuerySTableByTagCond(metaHandle, tableUid, 0, NULL, 0, 0, NULL, pGroupInfo, NULL, 0, queryId, taskId);
H
Haojun Liao 已提交
7107
  } else {  // Create one table group.
H
Haojun Liao 已提交
7108
    code = tsdbGetOneTableGroup(metaHandle, tableUid, 0, pGroupInfo);
7109 7110 7111 7112
  }

  return code;
}
H
Haojun Liao 已提交
7113

7114 7115 7116 7117 7118 7119 7120 7121 7122 7123 7124 7125 7126 7127 7128 7129 7130 7131
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 已提交
7132 7133
tsdbReaderT doCreateDataReader(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle,
                               STableGroupInfo* pTableGroupInfo, uint64_t queryId, uint64_t taskId) {
7134
  uint64_t uid = pTableScanNode->scan.uid;
L
Liu Jicong 已提交
7135 7136
  int32_t  code =
      doCreateTableGroup(pHandle->meta, pTableScanNode->scan.tableType, uid, pTableGroupInfo, queryId, taskId);
7137 7138
  if (code != TSDB_CODE_SUCCESS) {
    goto _error;
H
Haojun Liao 已提交
7139
  }
H
Haojun Liao 已提交
7140

H
Haojun Liao 已提交
7141
  if (pTableGroupInfo->numOfTables == 0) {
H
Haojun Liao 已提交
7142
    code = 0;
L
Liu Jicong 已提交
7143
    qDebug("no table qualified for query, TID:0x%" PRIx64 ", QID:0x%" PRIx64, taskId, queryId);
H
Haojun Liao 已提交
7144 7145
    goto _error;
  }
H
Haojun Liao 已提交
7146

H
Haojun Liao 已提交
7147
  return createDataReaderImpl(pTableScanNode, pTableGroupInfo, pHandle->reader, queryId, taskId);
H
Haojun Liao 已提交
7148

L
Liu Jicong 已提交
7149
_error:
H
Haojun Liao 已提交
7150 7151 7152 7153
  terrno = code;
  return NULL;
}

7154
int32_t createExecTaskInfoImpl(SSubplan* pPlan, SExecTaskInfo** pTaskInfo, SReadHandle* pHandle, uint64_t taskId, EOPTR_EXEC_MODEL model) {
H
Haojun Liao 已提交
7155 7156
  uint64_t queryId = pPlan->id.queryId;

H
Haojun Liao 已提交
7157
  int32_t code = TSDB_CODE_SUCCESS;
7158
  *pTaskInfo = createExecTaskInfo(queryId, taskId, model);
H
Haojun Liao 已提交
7159 7160 7161 7162
  if (*pTaskInfo == NULL) {
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    goto _complete;
  }
H
Haojun Liao 已提交
7163

H
Haojun Liao 已提交
7164
  STableGroupInfo group = {0};
H
Haojun Liao 已提交
7165
  (*pTaskInfo)->pRoot = createOperatorTree(pPlan->pNode, *pTaskInfo, pHandle, queryId, taskId, &group);
D
dapan1121 已提交
7166
  if (NULL == (*pTaskInfo)->pRoot) {
L
Liu Jicong 已提交
7167
    code = terrno;
D
dapan1121 已提交
7168 7169
    goto _complete;
  }
H
Haojun Liao 已提交
7170

7171
  if ((*pTaskInfo)->pRoot == NULL) {
H
Haojun Liao 已提交
7172
    code = TSDB_CODE_QRY_OUT_OF_MEMORY;
H
Haojun Liao 已提交
7173
    goto _complete;
7174 7175
  }

H
Haojun Liao 已提交
7176 7177
  return code;

H
Haojun Liao 已提交
7178
_complete:
wafwerar's avatar
wafwerar 已提交
7179
  taosMemoryFreeClear(*pTaskInfo);
H
Haojun Liao 已提交
7180 7181 7182

  terrno = code;
  return code;
H
Haojun Liao 已提交
7183 7184
}

L
Liu Jicong 已提交
7185 7186
static int32_t updateOutputBufForTopBotQuery(SQueriedTableInfo* pTableInfo, SColumnInfo* pTagCols, SExprInfo* pExprs,
                                             int32_t numOfOutput, int32_t tagLen, bool superTable) {
7187 7188 7189 7190 7191 7192 7193 7194 7195
  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 已提交
7196 7197 7198 7199
        //        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);
7200 7201 7202 7203 7204 7205 7206
      }
    }
  }

  return TSDB_CODE_SUCCESS;
}

H
Haojun Liao 已提交
7207
void setResultBufSize(STaskAttr* pQueryAttr, SResultInfo* pResultInfo) {
7208 7209 7210 7211 7212 7213 7214 7215
  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;

H
Haojun Liao 已提交
7216 7217 7218 7219 7220 7221 7222 7223 7224 7225
//  if (isProjQuery(pQueryAttr)) {
//    int32_t numOfRes = DEFAULT_RESULT_MSG_SIZE / pQueryAttr->resultRowSize;
//    if (numOfRes < MIN_ROWS_FOR_PRJ_QUERY) {
//      numOfRes = MIN_ROWS_FOR_PRJ_QUERY;
//    }
//
//    pResultInfo->capacity = numOfRes;
//  } else {  // in case of non-prj query, a smaller output buffer will be used.
//    pResultInfo->capacity = DEFAULT_MIN_ROWS;
//  }
7226 7227

  pResultInfo->threshold = (int32_t)(pResultInfo->capacity * THRESHOLD_RATIO);
H
Haojun Liao 已提交
7228
  pResultInfo->totalRows = 0;
7229 7230
}

L
Liu Jicong 已提交
7231
// TODO refactor
7232
void freeColumnFilterInfo(SColumnFilterInfo* pFilter, int32_t numOfFilters) {
L
Liu Jicong 已提交
7233 7234 7235
  if (pFilter == NULL || numOfFilters == 0) {
    return;
  }
7236

L
Liu Jicong 已提交
7237 7238 7239
  for (int32_t i = 0; i < numOfFilters; i++) {
    if (pFilter[i].filterstr && pFilter[i].pz) {
      taosMemoryFree((void*)(pFilter[i].pz));
7240
    }
L
Liu Jicong 已提交
7241
  }
7242

L
Liu Jicong 已提交
7243
  taosMemoryFree(pFilter);
7244 7245 7246 7247
}

static void doDestroyTableQueryInfo(STableGroupInfo* pTableqinfoGroupInfo) {
  if (pTableqinfoGroupInfo->pGroupList != NULL) {
L
Liu Jicong 已提交
7248
    int32_t numOfGroups = (int32_t)taosArrayGetSize(pTableqinfoGroupInfo->pGroupList);
7249
    for (int32_t i = 0; i < numOfGroups; ++i) {
L
Liu Jicong 已提交
7250
      SArray* p = taosArrayGetP(pTableqinfoGroupInfo->pGroupList, i);
7251 7252

      size_t num = taosArrayGetSize(p);
L
Liu Jicong 已提交
7253
      for (int32_t j = 0; j < num; ++j) {
7254 7255 7256 7257 7258 7259 7260 7261 7262 7263 7264 7265 7266 7267 7268 7269
        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 已提交
7270
void doDestroyTask(SExecTaskInfo* pTaskInfo) {
H
Haojun Liao 已提交
7271 7272
  qDebug("%s execTask is freed", GET_TASKID(pTaskInfo));

H
Haojun Liao 已提交
7273
  doDestroyTableQueryInfo(&pTaskInfo->tableqinfoGroupInfo);
L
Liu Jicong 已提交
7274 7275
  //  taosArrayDestroy(pTaskInfo->summary.queryProfEvents);
  //  taosHashCleanup(pTaskInfo->summary.operatorProfResults);
7276

wafwerar's avatar
wafwerar 已提交
7277 7278 7279
  taosMemoryFreeClear(pTaskInfo->sql);
  taosMemoryFreeClear(pTaskInfo->id.str);
  taosMemoryFreeClear(pTaskInfo);
7280 7281 7282 7283 7284 7285 7286 7287 7288 7289 7290 7291
}

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 已提交
7292
      int32_t len = (varDataLen(val) > maxLen) ? maxLen : varDataLen(val);
7293 7294 7295 7296 7297 7298 7299 7300 7301 7302 7303 7304
      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 已提交
7305 7306
  //  size_t s3 = sizeof(STableCheckInfo);  buffer consumption in tsdb
  return (int64_t)(s1 * 1.5 * numOfTables);
7307 7308 7309 7310 7311 7312 7313
}

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 已提交
7314
    while (1) {
7315 7316 7317 7318 7319 7320 7321 7322 7323 7324 7325 7326 7327 7328 7329 7330 7331 7332 7333 7334 7335 7336 7337 7338 7339 7340
      int64_t s = tsQueryBufferSizeBytes;
      int64_t remain = s - t;
      if (remain >= 0) {
        if (atomic_val_compare_exchange_64(&tsQueryBufferSizeBytes, s, remain) == s) {
          return TSDB_CODE_SUCCESS;
        }
      } else {
        return TSDB_CODE_QRY_NOT_ENOUGH_BUFFER;
      }
    }
  }

  // disable query processing if the value of tsQueryBufferSize is zero.
  return TSDB_CODE_QRY_NOT_ENOUGH_BUFFER;
}

void releaseQueryBuf(size_t numOfTables) {
  if (tsQueryBufferSizeBytes < 0) {
    return;
  }

  int64_t t = getQuerySupportBufSize(numOfTables);

  // restore value is not enough buffer available
  atomic_add_fetch_64(&tsQueryBufferSizeBytes, t);
}
D
dapan1121 已提交
7341

D
dapan1121 已提交
7342
int32_t getOperatorExplainExecInfo(SOperatorInfo *operatorInfo, SExplainExecInfo **pRes, int32_t *capacity, int32_t *resNum) {
D
dapan1121 已提交
7343 7344 7345 7346 7347
  if (*resNum >= *capacity) {
    *capacity += 10;
    
    *pRes = taosMemoryRealloc(*pRes, (*capacity) * sizeof(SExplainExecInfo));
    if (NULL == *pRes) {
D
dapan1121 已提交
7348
      qError("malloc %d failed", (*capacity) * (int32_t)sizeof(SExplainExecInfo));
D
dapan1121 已提交
7349 7350 7351 7352
      return TSDB_CODE_QRY_OUT_OF_MEMORY;
    }
  }

D
dapan1121 已提交
7353 7354 7355 7356 7357 7358 7359 7360 7361 7362 7363
  (*pRes)[*resNum].numOfRows = operatorInfo->resultInfo.totalRows;
  (*pRes)[*resNum].startupCost = operatorInfo->cost.openCost;
  (*pRes)[*resNum].totalCost = operatorInfo->cost.totalCost;

  if (operatorInfo->getExplainFn) {
    int32_t code = (*operatorInfo->getExplainFn)(operatorInfo, &(*pRes)->verboseInfo);
    if (code) {
      qError("operator getExplainFn failed, error:%s", tstrerror(code));
      return code;
    }
  }
D
dapan1121 已提交
7364 7365 7366 7367
  
  ++(*resNum);
  
  int32_t code = 0;
D
dapan1121 已提交
7368 7369
  for (int32_t i = 0; i < operatorInfo->numOfDownstream; ++i) {
    code = getOperatorExplainExecInfo(operatorInfo->pDownstream[i], pRes, capacity, resNum);
D
dapan1121 已提交
7370 7371 7372 7373 7374 7375 7376
    if (code) {
      taosMemoryFreeClear(*pRes);
      return TSDB_CODE_QRY_OUT_OF_MEMORY;
    }
  }

  return TSDB_CODE_SUCCESS;
D
dapan1121 已提交
7377 7378 7379
}