executorimpl.h 40.0 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/>.
 */
H
Hongze Cheng 已提交
15
// clang-format off
16 17 18
#ifndef TDENGINE_EXECUTORIMPL_H
#define TDENGINE_EXECUTORIMPL_H

19 20 21 22
#ifdef __cplusplus
extern "C" {
#endif

23
#include "os.h"
S
common  
Shengliang Guan 已提交
24
#include "tcommon.h"
25
#include "tlosertree.h"
dengyihao's avatar
dengyihao 已提交
26
#include "tsort.h"
27 28 29
#include "ttszip.h"
#include "tvariant.h"

H
Haojun Liao 已提交
30
#include "dataSinkMgt.h"
31
#include "executil.h"
H
Haojun Liao 已提交
32
#include "executor.h"
H
Haojun Liao 已提交
33
#include "planner.h"
D
dapan1121 已提交
34
#include "scalar.h"
35 36
#include "taosdef.h"
#include "tarray.h"
H
Haojun Liao 已提交
37
#include "thash.h"
38
#include "tlockfree.h"
D
dapan1121 已提交
39
#include "tmsg.h"
dengyihao's avatar
dengyihao 已提交
40
#include "tpagedbuf.h"
5
54liuyao 已提交
41
#include "tstreamUpdate.h"
L
Liu Jicong 已提交
42
#include "tstream.h"
H
Haojun Liao 已提交
43

L
Liu Jicong 已提交
44
#include "vnode.h"
45
#include "executorInt.h"
H
Hongze Cheng 已提交
46

47 48
typedef int32_t (*__block_search_fn_t)(char* data, int32_t num, int64_t key, int32_t order);

dengyihao's avatar
dengyihao 已提交
49 50
#define IS_QUERY_KILLED(_q)   ((_q)->code == TSDB_CODE_TSC_QUERY_CANCELLED)
#define Q_STATUS_EQUAL(p, s)  (((p) & (s)) != 0u)
51 52
#define QUERY_IS_ASC_QUERY(q) (GET_FORWARD_DIRECTION_FACTOR((q)->order.order) == QUERY_ASC_FORWARD_STEP)

53
#define NEEDTO_COMPRESS_QUERY(size) ((size) > tsCompressColData ? 1 : 0)
54

L
Liu Jicong 已提交
55 56 57 58
#define START_TS_COLUMN_INDEX       0
#define END_TS_COLUMN_INDEX         1
#define UID_COLUMN_INDEX            2
#define GROUPID_COLUMN_INDEX        UID_COLUMN_INDEX
59 60
#define DELETE_GROUPID_COLUMN_INDEX 2

61
enum {
62 63
  // when this task starts to execute, this status will set
  TASK_NOT_COMPLETED = 0x1u,
64

65
  /* Task is over
66 67 68
   * 1. this status is used in one row result query process, e.g., count/sum/first/last/ avg...etc.
   * 2. when all data within queried time window, it is also denoted as query_completed
   */
69
  TASK_COMPLETED = 0x2u,
70 71 72 73 74 75
};

/**
 * If the number of generated results is greater than this value,
 * query query will be halt and return results to client immediate.
 */
dengyihao's avatar
dengyihao 已提交
76 77 78 79 80
typedef struct SResultInfo {  // TODO refactor
  int64_t totalRows;          // total generated result size in rows
  int64_t totalBytes;         // total results in bytes.
  int32_t capacity;           // capacity of current result output buffer
  int32_t threshold;          // result size threshold in rows.
H
Haojun Liao 已提交
81
} SResultInfo;
82 83

typedef struct STableQueryInfo {
L
Liu Jicong 已提交
84 85
  TSKEY              lastKey;  // last check ts, todo remove it later
  SResultRowPosition pos;      // current active time window
86 87
} STableQueryInfo;

H
Haojun Liao 已提交
88 89 90 91 92
typedef struct SLimit {
  int64_t limit;
  int64_t offset;
} SLimit;

93
typedef struct STableScanAnalyzeInfo SFileBlockLoadRecorder;
94

H
Haojun Liao 已提交
95
typedef struct STaskCostInfo {
96 97
  int64_t  created;
  int64_t  start;
98 99 100 101 102 103 104 105
  uint64_t loadStatisTime;
  uint64_t loadFileBlockTime;
  uint64_t loadDataInCacheTime;
  uint64_t loadStatisSize;
  uint64_t loadFileBlockSize;
  uint64_t loadDataInCacheSize;

  uint64_t loadDataTime;
106 107

  SFileBlockLoadRecorder* pRecoder;
L
Liu Jicong 已提交
108
  uint64_t                elapsedTime;
109

110 111 112 113 114 115 116 117
  uint64_t firstStageMergeTime;
  uint64_t winInfoSize;
  uint64_t tableInfoSize;
  uint64_t hashSize;
  uint64_t numOfTimeWindows;

  SArray*   queryProfEvents;      // SArray<SQueryProfEvent>
  SHashObj* operatorProfResults;  // map<operator_type, SQueryProfEvent>
H
Haojun Liao 已提交
118
} STaskCostInfo;
119

H
Haojun Liao 已提交
120
typedef struct SOperatorCostInfo {
L
Liu Jicong 已提交
121 122
  double openCost;
  double totalCost;
H
Haojun Liao 已提交
123 124
} SOperatorCostInfo;

125 126
struct SOperatorInfo;

wmmhello's avatar
wmmhello 已提交
127
typedef int32_t (*__optr_encode_fn_t)(struct SOperatorInfo* pOperator, char** result, int32_t* length);
128
typedef int32_t (*__optr_decode_fn_t)(struct SOperatorInfo* pOperator, char* result);
129

130
typedef int32_t (*__optr_open_fn_t)(struct SOperatorInfo* pOptr);
131
typedef SSDataBlock* (*__optr_fn_t)(struct SOperatorInfo* pOptr);
H
Haojun Liao 已提交
132
typedef void (*__optr_close_fn_t)(void* param, int32_t num);
133
typedef int32_t (*__optr_explain_fn_t)(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len);
H
Haojun Liao 已提交
134

H
Haojun Liao 已提交
135
typedef struct STaskIdInfo {
136 137 138 139
  uint64_t queryId;  // this is also a request id
  uint64_t subplanId;
  uint64_t templateId;
  char*    str;
H
Haojun Liao 已提交
140 141
} STaskIdInfo;

L
Liu Jicong 已提交
142 143 144 145 146 147
enum {
  STREAM_RECOVER_STEP__NONE = 0,
  STREAM_RECOVER_STEP__PREPARE,
  STREAM_RECOVER_STEP__SCAN,
};

L
Liu Jicong 已提交
148
typedef struct {
L
Liu Jicong 已提交
149
  //TODO remove prepareStatus
L
Liu Jicong 已提交
150 151 152 153 154
  STqOffsetVal   prepareStatus; // for tmq
  STqOffsetVal   lastStatus;    // for tmq
  void*          metaBlk;       // for tmq fetching meta
  SSDataBlock*   pullOverBlk;   // for streaming
  SWalFilterCond cond;
L
Liu Jicong 已提交
155
  int64_t        lastScanUid;
L
Liu Jicong 已提交
156 157 158 159
  int8_t         recoverStep;
  SQueryTableDataCond tableCond;
  int64_t recoverStartVer;
  int64_t recoverEndVer;
L
Liu Jicong 已提交
160 161
} SStreamTaskInfo;

162 163 164 165 166 167 168
typedef struct {
  char*           tablename;
  char*           dbname;
  int32_t         tversion;
  SSchemaWrapper* sw;
} SSchemaInfo;

169
typedef struct SExecTaskInfo {
170 171 172 173 174 175 176 177 178 179 180 181 182 183
  STaskIdInfo      id;
  uint32_t         status;
  STimeWindow      window;
  STaskCostInfo    cost;
  int64_t          owner;  // if it is in execution
  int32_t          code;

  SStreamTaskInfo  streamInfo;
  SSchemaInfo      schemaInfo;
  STableListInfo   tableqinfoList;  // this is a table list
  const char*      sql;             // query sql string
  jmp_buf          env;             // jump to this position when error happens.
  EOPTR_EXEC_MODEL execModel;       // operator execution model [batch model|stream model]
  SSubplan*        pSubplan;
184
  struct SOperatorInfo* pRoot;
185
} SExecTaskInfo;
H
Haojun Liao 已提交
186

187
enum {
188 189
  OP_NOT_OPENED    = 0x0,
  OP_OPENED        = 0x1,
H
Haojun Liao 已提交
190
  OP_RES_TO_RETURN = 0x5,
191
  OP_EXEC_DONE     = 0x9,
192 193
};

194
typedef struct SOperatorFpSet {
L
Liu Jicong 已提交
195 196 197 198 199 200 201 202
  __optr_open_fn_t    _openFn;  // DO NOT invoke this function directly
  __optr_fn_t         getNextFn;
  __optr_fn_t         getStreamResFn;  // execute the aggregate in the stream model, todo remove it
  __optr_fn_t         cleanupFn;       // call this function to release the allocated resources ASAP
  __optr_close_fn_t   closeFn;
  __optr_encode_fn_t  encodeResultRow;
  __optr_decode_fn_t  decodeResultRow;
  __optr_explain_fn_t getExplainFn;
203 204
} SOperatorFpSet;

205 206
typedef struct SExprSupp {
  SExprInfo*      pExprInfo;
L
Liu Jicong 已提交
207
  int32_t         numOfExprs;  // the number of scalar expression in group operator
208 209 210 211
  SqlFunctionCtx* pCtx;
  int32_t*        rowEntryInfoOffset;  // offset value for each row result cell info
} SExprSupp;

212
typedef struct SOperatorInfo {
L
Liu Jicong 已提交
213 214 215 216 217 218 219 220 221 222 223 224
  uint8_t                operatorType;
  bool                   blocking;  // block operator or not
  uint8_t                status;    // denote if current operator is completed
  char*                  name;      // name, for debug purpose
  void*                  info;      // extension attribution
  SExprSupp              exprSupp;
  SExecTaskInfo*         pTaskInfo;
  SOperatorCostInfo      cost;
  SResultInfo            resultInfo;
  struct SOperatorInfo** pDownstream;      // downstram pointer list
  int32_t                numOfDownstream;  // number of downstream. The value is always ONE expect for join operator
  SOperatorFpSet         fpSet;
225 226
} SOperatorInfo;

227 228
typedef enum {
  EX_SOURCE_DATA_NOT_READY = 0x1,
229
  EX_SOURCE_DATA_READY     = 0x2,
230 231
  EX_SOURCE_DATA_EXHAUSTED = 0x3,
} EX_SOURCE_STATUS;
232

233 234 235
#define COL_MATCH_FROM_COL_ID  0x1
#define COL_MATCH_FROM_SLOT_ID 0x2

236
typedef struct SSourceDataInfo {
L
Liu Jicong 已提交
237 238 239 240 241 242
  int32_t            index;
  SRetrieveTableRsp* pRsp;
  uint64_t           totalRows;
  int32_t            code;
  EX_SOURCE_STATUS   status;
  const char*        taskId;
243 244
} SSourceDataInfo;

H
Haojun Liao 已提交
245
typedef struct SLoadRemoteDataInfo {
dengyihao's avatar
dengyihao 已提交
246 247 248
  uint64_t totalSize;     // total load bytes from remote
  uint64_t totalRows;     // total number of rows
  uint64_t totalElapsed;  // total elapsed time
H
Haojun Liao 已提交
249 250
} SLoadRemoteDataInfo;

251
typedef struct SLimitInfo {
252 253 254 255 256 257 258
  SLimit              limit;
  SLimit              slimit;
  uint64_t            currentGroupId;
  int64_t             remainGroupOffset;
  int64_t             numOfOutputGroups;
  int64_t             remainOffset;
  int64_t             numOfOutputRows;
259 260
} SLimitInfo;

261
typedef struct SExchangeInfo {
dengyihao's avatar
dengyihao 已提交
262 263 264 265 266 267 268
  SArray*             pSources;
  SArray*             pSourceDataInfo;
  tsem_t              ready;
  void*               pTransporter;
  SSDataBlock*        pResult;
  bool                seqLoadData;  // sequential load data or not, false by default
  int32_t             current;
H
Haojun Liao 已提交
269
  SLoadRemoteDataInfo loadInfo;
270
  uint64_t            self;
271
  SLimitInfo          limitInfo;
272 273
} SExchangeInfo;

H
Haojun Liao 已提交
274
typedef struct SColMatchInfo {
275
  int32_t srcSlotId;     // source slot id
H
Haojun Liao 已提交
276 277
  int32_t colId;
  int32_t targetSlotId;
278
  bool    output;        // todo remove this?
279
  bool    reserved;
280
  int32_t matchType;     // determinate the source according to col id or slot id
H
Haojun Liao 已提交
281 282
} SColMatchInfo;

283
typedef struct SScanInfo {
284 285
  int32_t         numOfAsc;
  int32_t         numOfDesc;
286 287
} SScanInfo;

288 289 290 291 292
typedef struct SSampleExecInfo {
  double          sampleRatio;  // data block sample ratio, 1 by default
  uint32_t        seed;         // random seed value
} SSampleExecInfo;

L
Liu Jicong 已提交
293 294
enum {
  TABLE_SCAN__TABLE_ORDER = 1,
295
  TABLE_SCAN__BLOCK_ORDER = 2,
L
Liu Jicong 已提交
296 297
};

298
typedef struct STableScanInfo {
H
Hongze Cheng 已提交
299
  STsdbReader*    dataReader;
300 301
  SReadHandle     readHandle;

302
  SFileBlockLoadRecorder readRecorder;
303
  SScanInfo       scanInfo;
304 305
  int32_t         scanTimes;
  SNode*          pFilterNode;  // filter info, which is push down by optimizer
306

307
  SSDataBlock*    pResBlock;
H
Haojun Liao 已提交
308
  SArray*         pColMatchInfo;
309
  SExprSupp       pseudoSup;
310
  SQueryTableDataCond cond;
311
  int32_t         scanFlag;     // table scan flag to denote if it is a repeat/reverse/main scan
312
  int32_t         dataBlockLoadFlag;
313
  SInterval       interval;     // if the upstream is an interval operator, the interval info is also kept here to get the time window to check if current data block needs to be loaded.
314
  SSampleExecInfo sample;       // sample execution info
wmmhello's avatar
wmmhello 已提交
315
  int32_t         currentGroupId;
L
Liu Jicong 已提交
316
  int32_t         currentTable;
317 318
  int8_t          scanMode;
  int8_t          noTable;
319 320 321
} STableScanInfo;

typedef struct STagScanInfo {
322 323 324 325 326
  SColumnInfo     *pCols;
  SSDataBlock     *pRes;
  SArray          *pColMatchInfo;
  int32_t          curPos;
  SReadHandle      readHandle;
wmmhello's avatar
wmmhello 已提交
327
  STableListInfo  *pTableList;
328 329
} STagScanInfo;

330 331 332
typedef struct SLastrowScanInfo {
  SSDataBlock    *pRes;
  SReadHandle     readHandle;
333 334 335
  void           *pLastrowReader;
  SArray         *pColMatchInfo;
  int32_t        *pSlotIds;
336
  SExprSupp       pseudoExprSup;
337 338 339 340 341
  int32_t         retrieveType;
  int32_t         currentGroupIndex;
  SSDataBlock    *pBufferredRes;
  SArray         *pUidList;
  int32_t         indexOfBufferedRes;
342 343
} SLastrowScanInfo;

5
54liuyao 已提交
344 345 346 347
typedef enum EStreamScanMode {
  STREAM_SCAN_FROM_READERHANDLE = 1,
  STREAM_SCAN_FROM_RES,
  STREAM_SCAN_FROM_UPDATERES,
5
54liuyao 已提交
348
  STREAM_SCAN_FROM_DATAREADER, // todo(liuyao) delete it
5
54liuyao 已提交
349
  STREAM_SCAN_FROM_DATAREADER_RETRIEVE,
L
Liu Jicong 已提交
350
  STREAM_SCAN_FROM_DATAREADER_RANGE,
5
54liuyao 已提交
351 352
} EStreamScanMode;

5
54liuyao 已提交
353
typedef struct SCatchSupporter {
L
Liu Jicong 已提交
354 355 356 357
  SHashObj*      pWindowHashTable;  // quick locate the window object for each window
  SDiskbasedBuf* pDataBuf;          // buffer based on blocked-wised disk file
  int32_t        keySize;
  int64_t*       pKeyBuf;
5
54liuyao 已提交
358 359
} SCatchSupporter;

5
54liuyao 已提交
360
typedef struct SStreamAggSupporter {
361 362 363
  SHashObj*      pResultRows;
  SArray*        pCurWins;
  int32_t        valueSize;
5
54liuyao 已提交
364 365 366 367
  int32_t        keySize;
  char*          pKeyBuf;              // window key buffer
  SDiskbasedBuf* pResultBuf;           // query result buffer based on blocked-wised disk file
  int32_t        resultRowSize;        // the result buffer size for each result row, with the meta data size for each row
5
54liuyao 已提交
368
  SArray*        pScanWindow;
5
54liuyao 已提交
369 370 371 372
} SStreamAggSupporter;

typedef struct SessionWindowSupporter {
  SStreamAggSupporter* pStreamAggSup;
L
Liu Jicong 已提交
373 374
  int64_t              gap;
  uint8_t              parentType;
5
54liuyao 已提交
375
} SessionWindowSupporter;
376

5
54liuyao 已提交
377 378 379 380 381 382 383
typedef struct STimeWindowSupp {
  int8_t           calTrigger;
  int64_t          waterMark;
  TSKEY            maxTs;
  SColumnInfoData  timeWindowData;     // query time window info for scalar function execution.
} STimeWindowAggSupp;

384
typedef struct SStreamScanInfo {
L
Liu Jicong 已提交
385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411
  uint64_t    tableUid;  // queried super table uid
  SExprInfo*  pPseudoExpr;
  int32_t     numOfPseudoExpr;
  int32_t     primaryTsIndex;  // primary time stamp slot id
  SReadHandle readHandle;
  SInterval   interval;       // if the upstream is an interval operator, the interval info is also kept here.
  SArray*     pColMatchInfo;  //
  SNode*      pCondition;

  SArray*      pBlockLists;  // multiple SSDatablock.
  SSDataBlock* pRes;         // result SSDataBlock
  SSDataBlock* pUpdateRes;   // update SSDataBlock
  int32_t      updateResIndex;
  int32_t      blockType;        // current block type
  int32_t      validBlockIndex;  // Is current data has returned?
  uint64_t     numOfExec;        // execution times
  STqReader*   tqReader;

  int32_t      tsArrayIndex;
  SArray*      tsArray;
  uint64_t     groupId;
  SUpdateInfo* pUpdateInfo;

  EStreamScanMode        scanMode;
  SOperatorInfo*         pStreamScanOp;
  SOperatorInfo*         pTableScanOp;
  SArray*                childIds;
5
54liuyao 已提交
412
  SessionWindowSupporter sessionSup;
L
Liu Jicong 已提交
413 414 415 416 417 418
  bool                   assignBlockUid;  // assign block uid to groupId, temporarily used for generating rollup SMA.
  int32_t                scanWinIndex;    // for state operator
  int32_t                pullDataResIndex;
  SSDataBlock*           pPullDataRes;    // pull data SSDataBlock
  SSDataBlock*           pDeleteDataRes;  // delete data SSDataBlock
  int32_t                deleteDataIndex;
5
54liuyao 已提交
419
  STimeWindow            updateWin;
5
54liuyao 已提交
420
  STimeWindowAggSupp     twAggSup;
L
Liu Jicong 已提交
421 422

  // status for tmq
L
Liu Jicong 已提交
423
  // SSchemaWrapper schema;
424 425 426
  STqOffset              offset;
  SNode*                 pTagCond;
  SNode*                 pTagIndexCond;
427
} SStreamScanInfo;
H
Haojun Liao 已提交
428

429
typedef struct SSysTableScanInfo {
dengyihao's avatar
dengyihao 已提交
430 431 432 433
  SRetrieveMetaTableRsp* pRsp;
  SRetrieveTableReq      req;
  SEpSet                 epSet;
  tsem_t                 ready;
434 435
  SReadHandle            readHandle;
  int32_t                accountId;
436
  const char*            pUser;
437 438 439 440 441 442 443 444
  bool                   showRewrite;
  SNode*                 pCondition;  // db_name filter condition, to discard data that are not in current database
  SMTbCursor*            pCur;        // cursor for iterate the local table meta store.
  SArray*                scanCols;    // SArray<int16_t> scan column id list
  SName                  name;
  SSDataBlock*           pRes;
  int64_t                numOfBlocks;  // extract basic running information.
  SLoadRemoteDataInfo    loadInfo;
445 446
} SSysTableScanInfo;

447 448 449
typedef struct SBlockDistInfo {
  SSDataBlock* pResBlock;
  void*        pHandle;
450 451
  SReadHandle  readHandle;
  uint64_t     uid;        // table uid
452 453
} SBlockDistInfo;

454
// todo remove this
455
typedef struct SOptrBasicInfo {
dengyihao's avatar
dengyihao 已提交
456 457
  SResultRowInfo  resultRowInfo;
  SSDataBlock*    pRes;
458 459
} SOptrBasicInfo;

460
typedef struct SAggSupporter {
dengyihao's avatar
dengyihao 已提交
461 462 463
  SHashObj*      pResultRowHashTable;  // quick locate the window object for each result
  char*          keyBuf;               // window key buffer
  SDiskbasedBuf* pResultBuf;           // query result buffer based on blocked-wised disk file
H
Haojun Liao 已提交
464
  int32_t        resultRowSize;        // the result buffer size for each result row, with the meta data size for each row
465 466
} SAggSupporter;

467
typedef struct SIntervalAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
468
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
469
  SOptrBasicInfo     binfo;              // basic info
wmmhello's avatar
wmmhello 已提交
470
  SAggSupporter      aggSup;             // aggregate supporter
471
  SExprSupp          scalarSupp;         // supporter for perform scalar function
472 473 474 475 476
  SGroupResInfo      groupResInfo;       // multiple results build supporter
  SInterval          interval;           // interval info
  int32_t            primaryTsIndex;     // primary time stamp slot id from result of downstream operator.
  STimeWindow        win;                // query time range
  bool               timeWindowInterpo;  // interpolation needed or not
477
  SArray*            pInterpCols;        // interpolation columns
478 479 480
  int32_t            order;              // current SSDataBlock scan order
  EOPTR_EXEC_MODEL   execModel;          // operator execution model [batch model|stream model]
  STimeWindowAggSupp twAggSup;
5
54liuyao 已提交
481
  bool               invertible;
482
  SArray*            pPrevValues;        //  SArray<SGroupKeys> used to keep the previous not null value for interpolation.
5
54liuyao 已提交
483
  bool               ignoreExpiredData;
484 485 486 487
  SArray*            pRecycledPages;
  SArray*            pDelWins;           // SWinRes
  int32_t            delIndex;
  SSDataBlock*       pDelRes;
488 489

  SNode *pCondition;
490
} SIntervalAggOperatorInfo;
491

492 493 494 495 496 497 498 499 500 501 502
typedef struct SMergeAlignedIntervalAggOperatorInfo {
  SIntervalAggOperatorInfo *intervalAggOperatorInfo;

  bool         hasGroupId;
  uint64_t     groupId;
  SSDataBlock* prefetchedBlock;
  bool         inputBlocksFinished;

  SNode*       pCondition;
} SMergeAlignedIntervalAggOperatorInfo;

5
54liuyao 已提交
503
typedef struct SStreamFinalIntervalOperatorInfo {
wmmhello's avatar
wmmhello 已提交
504
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
5
54liuyao 已提交
505
  SOptrBasicInfo     binfo;              // basic info
wmmhello's avatar
wmmhello 已提交
506
  SAggSupporter      aggSup;             // aggregate supporter
5
54liuyao 已提交
507
  SExprSupp          scalarSupp;         // supporter for perform scalar function
5
54liuyao 已提交
508 509 510 511 512
  SGroupResInfo      groupResInfo;       // multiple results build supporter
  SInterval          interval;           // interval info
  int32_t            primaryTsIndex;     // primary time stamp slot id from result of downstream operator.
  int32_t            order;              // current SSDataBlock scan order
  STimeWindowAggSupp twAggSup;
513
  SArray*            pChildren;
5
54liuyao 已提交
514
  SSDataBlock*       pUpdateRes;
5
54liuyao 已提交
515
  bool               returnUpdate;
5
54liuyao 已提交
516
  SPhysiNode*        pPhyNode;           // create new child
5
54liuyao 已提交
517
  bool               isFinal;
5
54liuyao 已提交
518 519 520 521
  SHashObj*          pPullDataMap;
  SArray*            pPullWins;          // SPullWindowInfo
  int32_t            pullIndex;
  SSDataBlock*       pPullDataRes;
5
54liuyao 已提交
522
  bool               ignoreExpiredData;
523 524 525 526
  SArray*            pRecycledPages;
  SArray*            pDelWins;           // SWinRes
  int32_t            delIndex;
  SSDataBlock*       pDelRes;
5
54liuyao 已提交
527 528
} SStreamFinalIntervalOperatorInfo;

529
typedef struct SAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
530
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
H
Haojun Liao 已提交
531 532
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
wmmhello's avatar
wmmhello 已提交
533

H
Haojun Liao 已提交
534
  STableQueryInfo   *current;
H
Haojun Liao 已提交
535
  uint64_t           groupId;
H
Haojun Liao 已提交
536
  SGroupResInfo      groupResInfo;
537
  SExprSupp          scalarExprSup;
S
slzhou 已提交
538 539

  SNode             *pCondition;
540 541 542
} SAggOperatorInfo;

typedef struct SProjectOperatorInfo {
wmmhello's avatar
wmmhello 已提交
543
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
544 545
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
546
  SNode*             pFilterNode;  // filter info, which is push down by optimizer
547 548
  SSDataBlock*       existDataBlock;
  SArray*            pPseudoColInfo;
549
  SLimitInfo         limitInfo;
550 551
} SProjectOperatorInfo;

H
Haojun Liao 已提交
552 553 554 555
typedef struct SIndefOperatorInfo {
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
  SArray*            pPseudoColInfo;
556
  SExprSupp          scalarSup;
S
slzhou 已提交
557
  SNode*             pCondition;
558 559 560
  uint64_t           groupId;

  SSDataBlock*       pNextGroupRes;
H
Haojun Liao 已提交
561 562
} SIndefOperatorInfo;

563
typedef struct SFillOperatorInfo {
564 565 566 567 568 569
  struct SFillInfo* pFillInfo;
  SSDataBlock*      pRes;
  int64_t           totalInputRows;
  void**            p;
  SSDataBlock*      existNewGroupBlock;
  bool              multigroupResult;
570
  STimeWindow       win;
S
slzhou 已提交
571
  SNode*            pCondition;
572
  SArray*           pColMatchColInfo;
H
Haojun Liao 已提交
573
  int32_t           primaryTsCol;
574
  uint64_t          curGroupId;       // current handled group id
575 576 577
} SFillOperatorInfo;

typedef struct SGroupbyOperatorInfo {
wmmhello's avatar
wmmhello 已提交
578
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
dengyihao's avatar
dengyihao 已提交
579
  SOptrBasicInfo  binfo;
wmmhello's avatar
wmmhello 已提交
580 581
  SAggSupporter   aggSup;

582
  SArray*         pGroupCols;     // group by columns, SArray<SColumn>
dengyihao's avatar
dengyihao 已提交
583 584 585 586 587 588
  SArray*         pGroupColVals;  // current group column values, SArray<SGroupKeys>
  SNode*          pCondition;
  bool            isInit;       // denote if current val is initialized or not
  char*           keyBuf;       // group by keys for hash
  int32_t         groupKeyLen;  // total group by column width
  SGroupResInfo   groupResInfo;
589
  SExprSupp       scalarSup;
590 591
} SGroupbyOperatorInfo;

H
Haojun Liao 已提交
592
typedef struct SDataGroupInfo {
593 594 595
  uint64_t        groupId;
  int64_t         numOfRows;
  SArray*         pPageList;
H
Haojun Liao 已提交
596 597 598 599
} SDataGroupInfo;

// The sort in partition may be needed later.
typedef struct SPartitionOperatorInfo {
dengyihao's avatar
dengyihao 已提交
600 601 602 603 604 605 606 607 608 609
  SOptrBasicInfo binfo;
  SArray*        pGroupCols;
  SArray*        pGroupColVals;  // current group column values, SArray<SGroupKeys>
  char*          keyBuf;         // group by keys for hash
  int32_t        groupKeyLen;    // total group by column width
  SHashObj*      pGroupSet;      // quick locate the window object for each result

  SDiskbasedBuf* pBuf;          // query result buffer based on blocked-wised disk file
  int32_t        rowCapacity;   // maximum number of rows for each buffer page
  int32_t*       columnOffset;  // start position for each column data
610 611
  SArray*        sortedGroupArray;   // SDataGroupInfo sorted by group id
  int32_t        groupIndex;  // group index
612
  int32_t        pageIndex;   // page index of current group
613
  SSDataBlock*   pUpdateRes;
614
  SExprSupp      scalarSup;
H
Haojun Liao 已提交
615
} SPartitionOperatorInfo;
616

617
typedef struct SWindowRowsSup {
dengyihao's avatar
dengyihao 已提交
618 619 620 621
  STimeWindow win;
  TSKEY       prevTs;
  int32_t     startRowIndex;
  int32_t     numOfRows;
622 623
} SWindowRowsSup;

H
Haojun Liao 已提交
624
typedef struct SSessionAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
625
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
626 627
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
wmmhello's avatar
wmmhello 已提交
628

629 630 631 632
  SGroupResInfo      groupResInfo;
  SWindowRowsSup     winSup;
  bool               reptScan;        // next round scan
  int64_t            gap;             // session window gap
633
  int32_t            tsSlotId;        // primary timestamp slot id
634
  STimeWindowAggSupp twAggSup;
L
Liu Jicong 已提交
635
  const SNode* pCondition;
H
Haojun Liao 已提交
636
} SSessionAggOperatorInfo;
637

5
54liuyao 已提交
638 639 640 641
typedef struct SResultWindowInfo {
  SResultRowPosition pos;
  STimeWindow win;
  bool isOutput;
5
54liuyao 已提交
642
  bool isClosed;
5
54liuyao 已提交
643 644
} SResultWindowInfo;

5
54liuyao 已提交
645 646 647 648 649
typedef struct SStateWindowInfo {
  SResultWindowInfo winInfo;
  SStateKeys stateKey;
} SStateWindowInfo;

5
54liuyao 已提交
650 651 652
typedef struct SStreamSessionAggOperatorInfo {
  SOptrBasicInfo       binfo;
  SStreamAggSupporter  streamAggSup;
5
54liuyao 已提交
653
  SExprSupp            scalarSupp;         // supporter for perform scalar function
5
54liuyao 已提交
654 655 656
  SGroupResInfo        groupResInfo;
  int64_t              gap;             // session window gap
  int32_t              primaryTsIndex;  // primary timestamp slot id
5
54liuyao 已提交
657
  int32_t              endTsIndex;      // window end timestamp slot id
5
54liuyao 已提交
658 659 660 661
  int32_t              order;           // current SSDataBlock scan order
  STimeWindowAggSupp   twAggSup;
  SSDataBlock*         pWinBlock;       // window result
  SqlFunctionCtx*      pDummyCtx;       // for combine
5
54liuyao 已提交
662
  SSDataBlock*         pDelRes;         // delete result
5
54liuyao 已提交
663
  bool                 returnDelete;
5
54liuyao 已提交
664
  SSDataBlock*         pUpdateRes;      // update window
5
54liuyao 已提交
665 666
  SHashObj*            pStDeleted;
  void*                pDelIterator;
5
54liuyao 已提交
667
  SArray*              pChildren;       // cache for children's result; final stream operator
5
54liuyao 已提交
668 669
  SPhysiNode*          pPhyNode;           // create new child
  bool                 isFinal;
5
54liuyao 已提交
670
  bool                 ignoreExpiredData;
5
54liuyao 已提交
671 672
} SStreamSessionAggOperatorInfo;

673
typedef struct STimeSliceOperatorInfo {
674
  SSDataBlock*   pRes;
H
Haojun Liao 已提交
675
  STimeWindow    win;
dengyihao's avatar
dengyihao 已提交
676
  SInterval      interval;
H
Haojun Liao 已提交
677 678
  int64_t        current;
  SArray*        pPrevRow;      // SArray<SGroupValue>
H
Haojun Liao 已提交
679
  int32_t        fillType;      // fill type
680 681
  SColumn        tsCol;         // primary timestamp column
  SExprSupp      scalarSup;     // scalar calculation
H
Haojun Liao 已提交
682
  struct SFillColInfo*  pFillColInfo;  // fill column info
683 684
} STimeSliceOperatorInfo;

685
typedef struct SStateWindowOperatorInfo {
wmmhello's avatar
wmmhello 已提交
686
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
687 688
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
wmmhello's avatar
wmmhello 已提交
689

690 691
  SGroupResInfo      groupResInfo;
  SWindowRowsSup     winSup;
692
  SColumn            stateCol;  // start row index
693 694
  bool               hasKey;
  SStateKeys         stateKey;
695
  int32_t            tsSlotId;  // primary timestamp column slot id
696
  STimeWindowAggSupp twAggSup;
dengyihao's avatar
dengyihao 已提交
697
  //  bool             reptScan;
L
Liu Jicong 已提交
698
  const SNode* pCondition;
699 700
} SStateWindowOperatorInfo;

5
54liuyao 已提交
701 702 703
typedef struct SStreamStateAggOperatorInfo {
  SOptrBasicInfo       binfo;
  SStreamAggSupporter  streamAggSup;
5
54liuyao 已提交
704
  SExprSupp            scalarSupp;      // supporter for perform scalar function
5
54liuyao 已提交
705 706 707 708
  SGroupResInfo        groupResInfo;
  int32_t              primaryTsIndex;  // primary timestamp slot id
  int32_t              order;           // current SSDataBlock scan order
  STimeWindowAggSupp   twAggSup;
5
54liuyao 已提交
709
  SColumn              stateCol;
5
54liuyao 已提交
710 711 712 713 714 715
  SqlFunctionCtx*      pDummyCtx;       // for combine
  SSDataBlock*         pDelRes;
  SHashObj*            pSeDeleted;
  void*                pDelIterator;
  SArray*              pScanWindow;
  SArray*              pChildren;       // cache for children's result;
5
54liuyao 已提交
716
  bool                 ignoreExpiredData;
5
54liuyao 已提交
717 718
} SStreamStateAggOperatorInfo;

719
typedef struct SSortedMergeOperatorInfo {
wmmhello's avatar
wmmhello 已提交
720
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
721
  SOptrBasicInfo   binfo;
wmmhello's avatar
wmmhello 已提交
722 723
  SAggSupporter    aggSup;

724 725 726 727 728 729 730 731 732 733 734
  SArray*          pSortInfo;
  int32_t          numOfSources;
  SSortHandle     *pSortHandle;
  int32_t          bufPageSize;
  uint32_t         sortBufSize;  // max buffer size for in-memory sort
  int32_t          resultRowFactor;
  bool             hasGroupVal;
  SDiskbasedBuf   *pTupleStore;  // keep the final results
  int32_t          numOfResPerPage;
  char**           groupVal;
  SArray          *groupInfo;
735
} SSortedMergeOperatorInfo;
H
Haojun Liao 已提交
736

H
Haojun Liao 已提交
737
typedef struct SSortOperatorInfo {
738
  SOptrBasicInfo binfo;
739
  uint32_t     sortBufSize;    // max buffer size for in-memory sort
dengyihao's avatar
dengyihao 已提交
740 741
  SArray*      pSortInfo;
  SSortHandle* pSortHandle;
742
  SArray*      pColMatchInfo;  // for index map from table scan output
dengyihao's avatar
dengyihao 已提交
743
  int32_t      bufPageSize;
H
Haojun Liao 已提交
744

745 746
  int64_t      startTs;       // sort start time
  uint64_t     sortElapsed;   // sort elapsed time, time to flush to disk not included.
S
slzhou 已提交
747 748

  SNode*      pCondition;
H
Haojun Liao 已提交
749
} SSortOperatorInfo;
750

dengyihao's avatar
dengyihao 已提交
751 752 753
typedef struct STagFilterOperatorInfo {
  SOptrBasicInfo binfo;
} STagFilterOperatorInfo;
754

755 756 757 758 759 760 761 762 763 764 765
typedef struct SJoinOperatorInfo {
  SSDataBlock       *pRes;
  int32_t            joinType;

  SSDataBlock       *pLeft;
  int32_t            leftPos;
  SColumnInfo        leftCol;

  SSDataBlock       *pRight;
  int32_t            rightPos;
  SColumnInfo        rightCol;
766
  SNode             *pCondAfterMerge;
767 768
} SJoinOperatorInfo;

769 770 771
#define OPTR_IS_OPENED(_optr)  (((_optr)->status & OP_OPENED) == OP_OPENED)
#define OPTR_SET_OPENED(_optr) ((_optr)->status |= OP_OPENED)

772 773
void doDestroyExchangeOperatorInfo(void* param);

774 775
SOperatorFpSet createOperatorFpSet(__optr_open_fn_t openFn, __optr_fn_t nextFn, __optr_fn_t streamFn,
    __optr_fn_t cleanup, __optr_close_fn_t closeFn, __optr_encode_fn_t encode,
776
    __optr_decode_fn_t decode, __optr_explain_fn_t explain);
777

H
Haojun Liao 已提交
778
int32_t operatorDummyOpenFn(SOperatorInfo* pOperator);
dengyihao's avatar
dengyihao 已提交
779
void    operatorDummyCloseFn(void* param, int32_t numOfCols);
H
Haojun Liao 已提交
780
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num);
781 782 783

void    initBasicInfo(SOptrBasicInfo* pInfo, SSDataBlock* pBlock);
void    cleanupBasicInfo(SOptrBasicInfo* pInfo);
784
int32_t initExprSupp(SExprSupp* pSup, SExprInfo* pExprInfo, int32_t numOfExpr);
785
void    cleanupExprSupp(SExprSupp* pSup);
786 787
int32_t initAggInfo(SExprSupp *pSup, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols, size_t keyBufSize,
                    const char* pkey);
788
void    initResultSizeInfo(SOperatorInfo* pOperator, int32_t numOfRows);
789
void    doBuildResultDatablock(SOperatorInfo* pOperator, SOptrBasicInfo* pbInfo, SGroupResInfo* pGroupResInfo, SDiskbasedBuf* pBuf);
790 791 792
int32_t handleLimitOffset(SOperatorInfo *pOperator, SLimitInfo* pLimitInfo, SSDataBlock* pBlock, bool holdDataInBuf);
bool    hasLimitOffsetInfo(SLimitInfo* pLimitInfo);
void    initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo);
793

794
void    doApplyFunctions(SExecTaskInfo* taskInfo, SqlFunctionCtx* pCtx, STimeWindow* pWin, SColumnInfoData* pTimeWindowData, int32_t offset,
dengyihao's avatar
dengyihao 已提交
795
                         int32_t forwardStep, TSKEY* tsCol, int32_t numOfTotal, int32_t numOfOutput, int32_t order);
796 797

int32_t extractDataBlockFromFetchRsp(SSDataBlock* pRes, SLoadRemoteDataInfo* pLoadInfo, int32_t numOfRows, char* pData,
dengyihao's avatar
dengyihao 已提交
798 799
                                  int32_t compLen, int32_t numOfOutput, int64_t startTs, uint64_t* total,
                                  SArray* pColList);
800
void    getAlignQueryTimeWindow(SInterval* pInterval, int32_t precision, int64_t key, STimeWindow* win);
801 802
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order);

803
int32_t getTableScanInfo(SOperatorInfo* pOperator, int32_t *order, int32_t* scanFlag);
804
int32_t getBufferPgSize(int32_t rowSize, uint32_t* defaultPgsz, uint32_t* defaultBufsz);
805

dengyihao's avatar
dengyihao 已提交
806
void    doSetOperatorCompleted(SOperatorInfo* pOperator);
807
void    doFilter(const SNode* pFilterNode, SSDataBlock* pBlock);
808 809
int32_t addTagPseudoColumnData(SReadHandle* pHandle, SExprInfo* pPseudoExpr, int32_t numOfPseudoExpr,
                               SSDataBlock* pBlock, const char* idStr);
810

811
void    cleanupAggSup(SAggSupporter* pAggSup);
812 813
void    destroyBasicOperatorInfo(void* param, int32_t numOfOutput);
void    appendOneRowToDataBlock(SSDataBlock* pBlock, STupleHandle* pTupleHandle);
814
void    setTbNameColData(void* pMeta, const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId);
815

L
Liu Jicong 已提交
816
int32_t doPrepareScan(SOperatorInfo* pOperator, uint64_t uid, int64_t ts);
817
int32_t doGetScanStatus(SOperatorInfo* pOperator, uint64_t* uid, int64_t* ts);
818

819
SSDataBlock* loadNextDataBlock(void* param);
820

821
void setResultRowInitCtx(SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowEntryInfoOffset);
822 823 824 825

SResultRow* doSetResultOutBufByKey(SDiskbasedBuf* pResultBuf, SResultRowInfo* pResultRowInfo,
                                   char* pData, int16_t bytes, bool masterscan, uint64_t groupId,
                                   SExecTaskInfo* pTaskInfo, bool isIntervalQuery, SAggSupporter* pSup);
H
Haojun Liao 已提交
826

827 828
SOperatorInfo* createExchangeOperatorInfo(void* pTransporter, SExchangePhysiNode* pExNode, SExecTaskInfo* pTaskInfo);

H
Haojun Liao 已提交
829
SOperatorInfo* createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle, SExecTaskInfo* pTaskInfo);
dengyihao's avatar
dengyihao 已提交
830 831
SOperatorInfo* createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pPhyNode,
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo);
832
SOperatorInfo* createSysTableScanOperatorInfo(void* readHandle, SSystemTableScanPhysiNode *pScanPhyNode, const char* pUser, SExecTaskInfo* pTaskInfo);
833

S
slzhou 已提交
834
SOperatorInfo* createAggregateOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResultBlock, SNode* pCondition, SExprInfo* pScalarExprInfo,
wmmhello's avatar
wmmhello 已提交
835
                                           int32_t numOfScalarExpr, SExecTaskInfo* pTaskInfo);
836

H
Haojun Liao 已提交
837
SOperatorInfo* createIndefinitOutputOperatorInfo(SOperatorInfo* downstream, SPhysiNode *pNode, SExecTaskInfo* pTaskInfo);
838
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SProjectPhysiNode* pProjPhyNode, SExecTaskInfo* pTaskInfo);
839
SOperatorInfo* createSortOperatorInfo(SOperatorInfo* downstream, SSortPhysiNode* pSortNode, SExecTaskInfo* pTaskInfo);
840
SOperatorInfo* createMultiwayMergeOperatorInfo(SOperatorInfo** dowStreams, size_t numStreams, SMergePhysiNode* pMergePhysiNode, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
841
SOperatorInfo* createSortedMergeOperatorInfo(SOperatorInfo** downstream, int32_t numOfDownstream, SExprInfo* pExprInfo, int32_t num, SArray* pSortInfo, SArray* pGroupInfo, SExecTaskInfo* pTaskInfo);
842
SOperatorInfo* createLastrowScanOperator(SLastRowScanPhysiNode* pTableScanNode, SReadHandle* readHandle, SExecTaskInfo* pTaskInfo);
843

dengyihao's avatar
dengyihao 已提交
844
SOperatorInfo* createIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
845
                                          SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlotId,
846
                                          STimeWindowAggSupp* pTwAggSupp, SIntervalPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, bool isStream);
S
shenglian zhou 已提交
847

848 849 850 851
SOperatorInfo* createMergeIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                          SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlotId,
                                          SExecTaskInfo* pTaskInfo);

852
SOperatorInfo* createMergeAlignedIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
S
shenglian zhou 已提交
853
                                          SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlotId,
854
                                          SNode* pCondition, SExecTaskInfo* pTaskInfo);
S
shenglian zhou 已提交
855

5
54liuyao 已提交
856 857
SOperatorInfo* createStreamFinalIntervalOperatorInfo(SOperatorInfo* downstream,
    SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, int32_t numOfChild);
858 859
SOperatorInfo* createStreamIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                                SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlotId,
wmmhello's avatar
wmmhello 已提交
860
                                                STimeWindowAggSupp *pTwAggSupp, SExecTaskInfo* pTaskInfo);
dengyihao's avatar
dengyihao 已提交
861
SOperatorInfo* createSessionAggOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
862
                                            SSDataBlock* pResBlock, int64_t gap, int32_t tsSlotId, STimeWindowAggSupp* pTwAggSupp,
863
                                            SNode* pCondition, SExecTaskInfo* pTaskInfo);
dengyihao's avatar
dengyihao 已提交
864 865
SOperatorInfo* createGroupOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                       SSDataBlock* pResultBlock, SArray* pGroupColList, SNode* pCondition,
wmmhello's avatar
wmmhello 已提交
866
                                       SExprInfo* pScalarExprInfo, int32_t numOfScalarExpr, SExecTaskInfo* pTaskInfo);
867 868
SOperatorInfo* createDataBlockInfoScanOperator(void* dataReader, SReadHandle* readHandle, uint64_t uid, SBlockDistScanPhysiNode* pBlockScanNode,
                                               SExecTaskInfo* pTaskInfo);
869

870 871
SOperatorInfo* createStreamScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode, SNode* pTagCond,
                                            SExecTaskInfo* pTaskInfo, STimeWindowAggSupp* pTwSup);
872

H
Haojun Liao 已提交
873
SOperatorInfo* createFillOperatorInfo(SOperatorInfo* downstream, SFillPhysiNode* pPhyFillNode, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
874

dengyihao's avatar
dengyihao 已提交
875
SOperatorInfo* createStatewindowOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExpr, int32_t numOfCols,
876 877
                                             SSDataBlock* pResBlock, STimeWindowAggSupp *pTwAggSupp, int32_t tsSlotId,
                                             SColumn* pStateKeyCol, SNode* pCondition, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
878

879
SOperatorInfo* createPartitionOperatorInfo(SOperatorInfo* downstream, SPartitionPhysiNode* pPartNode, SExecTaskInfo* pTaskInfo);
880

881 882
SOperatorInfo* createTimeSliceOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pNode, /*SExprInfo* pExprInfo, int32_t numOfCols,
                                           SSDataBlock* pResultBlock, const SNodeListNode* pValNode, */SExecTaskInfo* pTaskInfo);
883

884 885
SOperatorInfo* createMergeJoinOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream, SJoinPhysiNode* pJoinNode,
                                           SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
886

887 888 889 890
SOperatorInfo* createStreamSessionAggOperatorInfo(SOperatorInfo* downstream,
    SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo);
SOperatorInfo* createStreamFinalSessionAggOperatorInfo(SOperatorInfo* downstream,
    SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, int32_t numOfChild);
5
54liuyao 已提交
891

892
SOperatorInfo* createStreamStateAggOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo);
5
54liuyao 已提交
893

H
Haojun Liao 已提交
894
#if 0
H
Haojun Liao 已提交
895
SOperatorInfo* createTableSeqScanOperatorInfo(void* pTsdbReadHandle, STaskRuntimeEnv* pRuntimeEnv);
H
Haojun Liao 已提交
896
#endif
897

898
int32_t projectApplyFunctions(SExprInfo* pExpr, SSDataBlock* pResult, SSDataBlock* pSrcBlock, SqlFunctionCtx* pCtx,
dengyihao's avatar
dengyihao 已提交
899
                           int32_t numOfOutput, SArray* pPseudoList);
900

901
void setInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order, int32_t scanFlag, bool createDummyCol);
X
Xiaoyu Wang 已提交
902

903
bool    isTaskKilled(SExecTaskInfo* pTaskInfo);
904 905
int32_t checkForQueryBuf(size_t numOfTables);

906 907
void    setTaskKilled(SExecTaskInfo* pTaskInfo);
void    queryCostStatis(SExecTaskInfo* pTaskInfo);
908

dengyihao's avatar
dengyihao 已提交
909
void    doDestroyTask(SExecTaskInfo* pTaskInfo);
910 911
int32_t getMaximumIdleDurationSec();

wmmhello's avatar
wmmhello 已提交
912 913 914 915
/*
 * ops:     root operator
 * data:    *data save the result of encode, need to be freed by caller
 * length:  *length save the length of *data
C
Cary Xu 已提交
916
 * nOptrWithVal: *nOptrWithVal save the number of optr with value
wmmhello's avatar
wmmhello 已提交
917 918
 * return:  result code, 0 means success
 */
C
Cary Xu 已提交
919
int32_t encodeOperator(SOperatorInfo* ops, char** data, int32_t *length, int32_t *nOptrWithVal);
wmmhello's avatar
wmmhello 已提交
920 921 922 923 924 925 926

/*
 * ops:    root operator, created by caller
 * data:   save the result of decode
 * length: the length of data
 * return: result code, 0 means success
 */
H
Haojun Liao 已提交
927
int32_t decodeOperator(SOperatorInfo* ops, const char* data, int32_t length);
wmmhello's avatar
wmmhello 已提交
928

929
void    setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status);
dengyihao's avatar
dengyihao 已提交
930
int32_t createExecTaskInfoImpl(SSubplan* pPlan, SExecTaskInfo** pTaskInfo, SReadHandle* pHandle, uint64_t taskId,
931
                               const char* sql, EOPTR_EXEC_MODEL model);
D
dapan1121 已提交
932
int32_t createDataSinkParam(SDataSinkNode *pNode, void **pParam, qTaskInfo_t* pTaskInfo, SReadHandle* readHandle);                               
dengyihao's avatar
dengyihao 已提交
933 934 935
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SExplainExecInfo** pRes, int32_t* capacity,
                                   int32_t* resNum);

936
int32_t aggDecodeResultRow(SOperatorInfo* pOperator, char* result);
wmmhello's avatar
wmmhello 已提交
937 938
int32_t aggEncodeResultRow(SOperatorInfo* pOperator, char** result, int32_t* length);

939
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
940
                                int32_t order);
941 942
int32_t getNumOfRowsInTimeWindow(SDataBlockInfo* pDataBlockInfo, TSKEY* pPrimaryColumn, int32_t startPos, TSKEY ekey,
    __block_search_fn_t searchFn, STableQueryInfo* item, int32_t order);
5
54liuyao 已提交
943
int32_t binarySearchForKey(char* pValue, int num, TSKEY key, int order);
944 945
int32_t initStreamAggSupporter(SStreamAggSupporter* pSup, const char* pKey, SqlFunctionCtx* pCtx, int32_t numOfOutput,
    int32_t size);
946
SResultRow* getNewResultRow(SDiskbasedBuf* pResultBuf, int64_t tableGroupId, int32_t interBufSize);
5
54liuyao 已提交
947 948
SResultWindowInfo* getSessionTimeWindow(SStreamAggSupporter* pAggSup, TSKEY startTs,
    TSKEY endTs, uint64_t groupId, int64_t gap, int32_t* pIndex);
5
54liuyao 已提交
949 950 951
SResultWindowInfo* getCurSessionWindow(SStreamAggSupporter* pAggSup, TSKEY startTs,
    TSKEY endTs, uint64_t groupId, int64_t gap, int32_t* pIndex);
bool isInTimeWindow(STimeWindow* pWin, TSKEY ts, int64_t gap);
5
54liuyao 已提交
952 953
int32_t updateSessionWindowInfo(SResultWindowInfo* pWinInfo, TSKEY* pStartTs,
    TSKEY* pEndTs, int32_t rows, int32_t start, int64_t gap, SHashObj* pStDeleted);
5
54liuyao 已提交
954
bool functionNeedToExecute(SqlFunctionCtx* pCtx);
5
54liuyao 已提交
955
bool isCloseWindow(STimeWindow* pWin, STimeWindowAggSupp* pSup);
956

957
int32_t finalizeResultRowIntoResultDataBlock(SDiskbasedBuf* pBuf, SResultRowPosition* resultRowPosition,
958 959
                                       SqlFunctionCtx* pCtx, SExprInfo* pExprInfo, int32_t numOfExprs, const int32_t* rowCellOffset,
                                       SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo);
960

961
int32_t createScanTableListInfo(SScanPhysiNode *pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
962
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond, const char* idstr);
963

S
shenglian zhou 已提交
964 965
SOperatorInfo* createGroupSortOperatorInfo(SOperatorInfo* downstream, SGroupSortPhysiNode* pSortPhyNode,
                                           SExecTaskInfo* pTaskInfo);
966
SOperatorInfo* createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, STableListInfo *pTableListInfo,
967
                                                SReadHandle* readHandle, SExecTaskInfo* pTaskInfo);
968

969
void copyUpdateDataBlock(SSDataBlock* pDest, SSDataBlock* pSource, int32_t tsColIndex);
970

wmmhello's avatar
wmmhello 已提交
971
int32_t generateGroupIdMap(STableListInfo* pTableListInfo, SReadHandle* pHandle, SNodeList* groupKey);
5
54liuyao 已提交
972
SSDataBlock* createPullDataBlock();
973

974 975 976 977
#ifdef __cplusplus
}
#endif

978
#endif  // TDENGINE_EXECUTORIMPL_H