executorimpl.h 43.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

55 56 57 58 59 60
#define START_TS_COLUMN_INDEX             0
#define END_TS_COLUMN_INDEX               1
#define UID_COLUMN_INDEX                  2
#define GROUPID_COLUMN_INDEX              3
#define CALCULATE_START_TS_COLUMN_INDEX   4
#define CALCULATE_END_TS_COLUMN_INDEX     5
61

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

66
  /* Task is over
67 68 69
   * 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
   */
70
  TASK_COMPLETED = 0x2u,
71 72 73 74 75 76
};

/**
 * 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 已提交
77 78 79 80 81
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 已提交
82
} SResultInfo;
83 84

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

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

94
typedef struct STableScanAnalyzeInfo SFileBlockLoadRecorder;
95

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

  uint64_t loadDataTime;
107 108

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

111 112 113 114 115 116 117
  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
  STqOffsetVal   prepareStatus; // for tmq
  STqOffsetVal   lastStatus;    // for tmq
wmmhello's avatar
wmmhello 已提交
152
  SMqMetaRsp     metaRsp;       // for tmq fetching meta
L
Liu Jicong 已提交
153 154
  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
typedef struct {
  char*           tablename;
  char*           dbname;
  int32_t         tversion;
  SSchemaWrapper* sw;
167
  SSchemaWrapper* qsw;
168 169
} SSchemaInfo;

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

178
  int64_t          version; // used for stream to record wal version
179 180 181 182 183 184 185
  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;
186
  struct SOperatorInfo* pRoot;
187
} SExecTaskInfo;
H
Haojun Liao 已提交
188

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

196
typedef struct SOperatorFpSet {
L
Liu Jicong 已提交
197 198 199 200 201 202 203 204
  __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;
205 206
} SOperatorFpSet;

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

214
typedef struct SOperatorInfo {
L
Liu Jicong 已提交
215 216 217 218 219 220 221 222 223 224 225 226
  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;
227
  int16_t                resultDataBlockId;
228 229
} SOperatorInfo;

230 231
typedef enum {
  EX_SOURCE_DATA_NOT_READY = 0x1,
232
  EX_SOURCE_DATA_READY     = 0x2,
233 234
  EX_SOURCE_DATA_EXHAUSTED = 0x3,
} EX_SOURCE_STATUS;
235

236 237 238
#define COL_MATCH_FROM_COL_ID  0x1
#define COL_MATCH_FROM_SLOT_ID 0x2

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

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

254
typedef struct SLimitInfo {
255 256 257 258 259 260 261
  SLimit              limit;
  SLimit              slimit;
  uint64_t            currentGroupId;
  int64_t             remainGroupOffset;
  int64_t             numOfOutputGroups;
  int64_t             remainOffset;
  int64_t             numOfOutputRows;
262 263
} SLimitInfo;

264
typedef struct SExchangeInfo {
dengyihao's avatar
dengyihao 已提交
265 266 267 268 269 270 271
  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 已提交
272
  SLoadRemoteDataInfo loadInfo;
273
  uint64_t            self;
274
  SLimitInfo          limitInfo;
275 276
} SExchangeInfo;

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

286
typedef struct SScanInfo {
287 288
  int32_t         numOfAsc;
  int32_t         numOfDesc;
289 290
} SScanInfo;

291 292 293 294 295
typedef struct SSampleExecInfo {
  double          sampleRatio;  // data block sample ratio, 1 by default
  uint32_t        seed;         // random seed value
} SSampleExecInfo;

L
Liu Jicong 已提交
296 297
enum {
  TABLE_SCAN__TABLE_ORDER = 1,
298
  TABLE_SCAN__BLOCK_ORDER = 2,
L
Liu Jicong 已提交
299 300
};

301 302 303 304 305 306 307 308 309 310 311 312 313 314
typedef struct SAggSupporter {
    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
    int32_t        resultRowSize;        // the result buffer size for each result row, with the meta data size for each row
} SAggSupporter;

typedef struct {
    // 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.
 SInterval      interval;
 SAggSupporter *pAggSup;
 SExprSupp     *pExprSup;         // expr supporter of aggregate operator
} SAggOptrPushDownInfo;

315
typedef struct STableScanInfo {
H
Hongze Cheng 已提交
316
  STsdbReader*    dataReader;
317 318
  SReadHandle     readHandle;

319
  SFileBlockLoadRecorder readRecorder;
320
  SScanInfo       scanInfo;
321 322
  int32_t         scanTimes;
  SNode*          pFilterNode;  // filter info, which is push down by optimizer
323

324
  SSDataBlock*    pResBlock;
H
Haojun Liao 已提交
325
  SArray*         pColMatchInfo;
326
  SExprSupp       pseudoSup;
327
  SQueryTableDataCond cond;
328
  int32_t         scanFlag;     // table scan flag to denote if it is a repeat/reverse/main scan
329
  int32_t         dataBlockLoadFlag;
330
//  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.
331
  SSampleExecInfo sample;       // sample execution info
wmmhello's avatar
wmmhello 已提交
332
  int32_t         currentGroupId;
L
Liu Jicong 已提交
333
  int32_t         currentTable;
334 335
  int8_t          scanMode;
  int8_t          noTable;
336
  SAggOptrPushDownInfo pdInfo;
337
  int8_t          assignBlockUid;
338 339
} STableScanInfo;

340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369
typedef struct STableMergeScanInfo {
  STableListInfo* tableListInfo;
  int32_t         tableStartIndex;
  int32_t         tableEndIndex;
  bool            hasGroupId;
  uint64_t        groupId;
  SArray*         dataReaders;  // array of tsdbReaderT*
  SReadHandle     readHandle;
  int32_t         bufPageSize;
  uint32_t        sortBufSize;  // max buffer size for in-memory sort
  SArray*         pSortInfo;
  SSortHandle*    pSortHandle;

  SSDataBlock* pSortInputBlock;
  int64_t      startTs;  // sort start time
  SArray*      sortSourceParams;

  SFileBlockLoadRecorder readRecorder;
  int64_t                numOfRows;
  SScanInfo              scanInfo;
  int32_t                scanTimes;
  SNode*                 pFilterNode;  // filter info, which is push down by optimizer
  SqlFunctionCtx*        pCtx;         // which belongs to the direct upstream operator operator query context
  SResultRowInfo*        pResultRowInfo;
  int32_t*               rowEntryInfoOffset;
  SExprInfo*             pExpr;
  SSDataBlock*           pResBlock;
  SArray*                pColMatchInfo;
  int32_t                numOfOutput;

X
Xiaoyu Wang 已提交
370
  SExprSupp       pseudoSup;
371 372 373 374 375 376 377 378

  SQueryTableDataCond cond;
  int32_t             scanFlag;  // table scan flag to denote if it is a repeat/reverse/main scan
  int32_t             dataBlockLoadFlag;
  // 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.
  SInterval       interval;
  SSampleExecInfo sample;  // sample execution info
379 380

  SSortExecInfo sortExecInfo;
381 382
} STableMergeScanInfo;

383
typedef struct STagScanInfo {
384 385 386 387 388
  SColumnInfo     *pCols;
  SSDataBlock     *pRes;
  SArray          *pColMatchInfo;
  int32_t          curPos;
  SReadHandle      readHandle;
wmmhello's avatar
wmmhello 已提交
389
  STableListInfo  *pTableList;
390 391
} STagScanInfo;

392 393 394
typedef struct SLastrowScanInfo {
  SSDataBlock    *pRes;
  SReadHandle     readHandle;
395 396 397
  void           *pLastrowReader;
  SArray         *pColMatchInfo;
  int32_t        *pSlotIds;
398
  SExprSupp       pseudoExprSup;
399 400 401 402 403
  int32_t         retrieveType;
  int32_t         currentGroupIndex;
  SSDataBlock    *pBufferredRes;
  SArray         *pUidList;
  int32_t         indexOfBufferedRes;
404 405
} SLastrowScanInfo;

5
54liuyao 已提交
406 407 408 409
typedef enum EStreamScanMode {
  STREAM_SCAN_FROM_READERHANDLE = 1,
  STREAM_SCAN_FROM_RES,
  STREAM_SCAN_FROM_UPDATERES,
5
54liuyao 已提交
410
  STREAM_SCAN_FROM_DATAREADER_RETRIEVE,
L
Liu Jicong 已提交
411
  STREAM_SCAN_FROM_DATAREADER_RANGE,
5
54liuyao 已提交
412 413
} EStreamScanMode;

414 415 416 417 418
enum {
  PROJECT_RETRIEVE_CONTINUE = 0x1,
  PROJECT_RETRIEVE_DONE = 0x2,
};

5
54liuyao 已提交
419
typedef struct SCatchSupporter {
L
Liu Jicong 已提交
420 421 422 423
  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 已提交
424 425
} SCatchSupporter;

5
54liuyao 已提交
426
typedef struct SStreamAggSupporter {
427 428 429
  SHashObj*      pResultRows;
  SArray*        pCurWins;
  int32_t        valueSize;
5
54liuyao 已提交
430 431 432 433
  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
434
  SSDataBlock*   pScanBlock;
5
54liuyao 已提交
435 436 437 438
} SStreamAggSupporter;

typedef struct SessionWindowSupporter {
  SStreamAggSupporter* pStreamAggSup;
L
Liu Jicong 已提交
439 440
  int64_t              gap;
  uint8_t              parentType;
441
  SAggSupporter*       pIntervalAggSup;
5
54liuyao 已提交
442
} SessionWindowSupporter;
443

5
54liuyao 已提交
444 445 446 447 448 449 450
typedef struct STimeWindowSupp {
  int8_t           calTrigger;
  int64_t          waterMark;
  TSKEY            maxTs;
  SColumnInfoData  timeWindowData;     // query time window info for scalar function execution.
} STimeWindowAggSupp;

451
typedef struct SStreamScanInfo {
L
Liu Jicong 已提交
452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476
  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;

  uint64_t     groupId;
  SUpdateInfo* pUpdateInfo;

  EStreamScanMode        scanMode;
  SOperatorInfo*         pStreamScanOp;
  SOperatorInfo*         pTableScanOp;
  SArray*                childIds;
5
54liuyao 已提交
477
  SessionWindowSupporter sessionSup;
L
Liu Jicong 已提交
478 479 480 481 482 483
  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 已提交
484
  STimeWindow            updateWin;
5
54liuyao 已提交
485
  STimeWindowAggSupp     twAggSup;
486
  SSDataBlock*           pUpdateDataRes;
L
Liu Jicong 已提交
487
  // status for tmq
L
Liu Jicong 已提交
488
  // SSchemaWrapper schema;
489
  SNodeList*             pGroupTags;
490 491
  SNode*                 pTagCond;
  SNode*                 pTagIndexCond;
492
} SStreamScanInfo;
H
Haojun Liao 已提交
493

494 495 496 497 498 499 500
typedef struct SStreamRawScanInfo{
//  int8_t    subType;
//  bool      withMeta;
//  int64_t   suid;
//  int64_t   snapVersion;
//  void     *metaInfo;
//  void     *dataInfo;
wmmhello's avatar
wmmhello 已提交
501
  SVnode*         vnode;
wmmhello's avatar
wmmhello 已提交
502
  SWalCkHead*     pCkHead;
wmmhello's avatar
wmmhello 已提交
503 504
  bool            needFetchLog;
  bool            hasDataInOneFetchVer;
505 506 507
  SSDataBlock     pRes;         // result SSDataBlock
  STsdbReader*    dataReader;
  SSnapContext*   sContext;
wmmhello's avatar
wmmhello 已提交
508 509
  STqReader*      tqReader;
  SHashObj*       pFilterOutTbUid;
510 511
}SStreamRawScanInfo;

512
typedef struct SSysTableScanInfo {
dengyihao's avatar
dengyihao 已提交
513 514 515 516
  SRetrieveMetaTableRsp* pRsp;
  SRetrieveTableReq      req;
  SEpSet                 epSet;
  tsem_t                 ready;
517 518
  SReadHandle            readHandle;
  int32_t                accountId;
519
  const char*            pUser;
520 521 522 523 524 525 526 527
  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;
528 529
} SSysTableScanInfo;

530 531 532
typedef struct SBlockDistInfo {
  SSDataBlock* pResBlock;
  void*        pHandle;
533 534
  SReadHandle  readHandle;
  uint64_t     uid;        // table uid
535 536
} SBlockDistInfo;

537
// todo remove this
538
typedef struct SOptrBasicInfo {
dengyihao's avatar
dengyihao 已提交
539 540
  SResultRowInfo  resultRowInfo;
  SSDataBlock*    pRes;
541 542
} SOptrBasicInfo;

543
typedef struct SIntervalAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
544
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
545
  SOptrBasicInfo     binfo;              // basic info
wmmhello's avatar
wmmhello 已提交
546
  SAggSupporter      aggSup;             // aggregate supporter
547
  SExprSupp          scalarSupp;         // supporter for perform scalar function
548 549 550 551 552
  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
553
  SArray*            pInterpCols;        // interpolation columns
554
  int32_t            resultTsOrder;      // result timestamp order
555
  int32_t            inputOrder;         // input data ts order
556 557
  EOPTR_EXEC_MODEL   execModel;          // operator execution model [batch model|stream model]
  STimeWindowAggSupp twAggSup;
5
54liuyao 已提交
558
  bool               invertible;
559
  SArray*            pPrevValues;        //  SArray<SGroupKeys> used to keep the previous not null value for interpolation.
5
54liuyao 已提交
560
  bool               ignoreExpiredData;
561 562 563 564
  SArray*            pRecycledPages;
  SArray*            pDelWins;           // SWinRes
  int32_t            delIndex;
  SSDataBlock*       pDelRes;
565
  SNode*             pCondition;
566
} SIntervalAggOperatorInfo;
567

568 569 570 571
typedef struct SMergeAlignedIntervalAggOperatorInfo {
  SIntervalAggOperatorInfo *intervalAggOperatorInfo;

  bool         hasGroupId;
572 573
  uint64_t     groupId;  // current groupId
  int64_t      curTs;    // current ts
574 575 576 577 578 579
  SSDataBlock* prefetchedBlock;
  bool         inputBlocksFinished;

  SNode*       pCondition;
} SMergeAlignedIntervalAggOperatorInfo;

5
54liuyao 已提交
580
typedef struct SStreamFinalIntervalOperatorInfo {
wmmhello's avatar
wmmhello 已提交
581
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
5
54liuyao 已提交
582
  SOptrBasicInfo     binfo;              // basic info
wmmhello's avatar
wmmhello 已提交
583
  SAggSupporter      aggSup;             // aggregate supporter
5
54liuyao 已提交
584
  SExprSupp          scalarSupp;         // supporter for perform scalar function
5
54liuyao 已提交
585 586 587 588 589
  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;
590
  SArray*            pChildren;
5
54liuyao 已提交
591
  SSDataBlock*       pUpdateRes;
5
54liuyao 已提交
592
  bool               returnUpdate;
5
54liuyao 已提交
593
  SPhysiNode*        pPhyNode;           // create new child
5
54liuyao 已提交
594
  bool               isFinal;
5
54liuyao 已提交
595 596 597 598
  SHashObj*          pPullDataMap;
  SArray*            pPullWins;          // SPullWindowInfo
  int32_t            pullIndex;
  SSDataBlock*       pPullDataRes;
5
54liuyao 已提交
599
  bool               ignoreExpiredData;
600 601 602 603
  SArray*            pRecycledPages;
  SArray*            pDelWins;           // SWinRes
  int32_t            delIndex;
  SSDataBlock*       pDelRes;
5
54liuyao 已提交
604 605
} SStreamFinalIntervalOperatorInfo;

606
typedef struct SAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
607
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
H
Haojun Liao 已提交
608 609
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
wmmhello's avatar
wmmhello 已提交
610

H
Haojun Liao 已提交
611
  STableQueryInfo   *current;
H
Haojun Liao 已提交
612
  uint64_t           groupId;
H
Haojun Liao 已提交
613
  SGroupResInfo      groupResInfo;
614
  SExprSupp          scalarExprSup;
S
slzhou 已提交
615 616

  SNode             *pCondition;
617 618 619
} SAggOperatorInfo;

typedef struct SProjectOperatorInfo {
wmmhello's avatar
wmmhello 已提交
620
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
621 622
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
623
  SNode*             pFilterNode;  // filter info, which is push down by optimizer
624
  SArray*            pPseudoColInfo;
625
  SLimitInfo         limitInfo;
H
Haojun Liao 已提交
626 627
  bool               mergeDataBlocks;
  SSDataBlock*       pFinalRes;
628
  SNode*             pCondition;
629 630
} SProjectOperatorInfo;

H
Haojun Liao 已提交
631 632 633 634
typedef struct SIndefOperatorInfo {
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
  SArray*            pPseudoColInfo;
635
  SExprSupp          scalarSup;
S
slzhou 已提交
636
  SNode*             pCondition;
637 638 639
  uint64_t           groupId;

  SSDataBlock*       pNextGroupRes;
H
Haojun Liao 已提交
640 641
} SIndefOperatorInfo;

642
typedef struct SFillOperatorInfo {
643 644 645 646 647 648
  struct SFillInfo* pFillInfo;
  SSDataBlock*      pRes;
  int64_t           totalInputRows;
  void**            p;
  SSDataBlock*      existNewGroupBlock;
  bool              multigroupResult;
649
  STimeWindow       win;
S
slzhou 已提交
650
  SNode*            pCondition;
651
  SArray*           pColMatchColInfo;
H
Haojun Liao 已提交
652
  int32_t           primaryTsCol;
653
  uint64_t          curGroupId;       // current handled group id
654 655 656
} SFillOperatorInfo;

typedef struct SGroupbyOperatorInfo {
wmmhello's avatar
wmmhello 已提交
657
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
dengyihao's avatar
dengyihao 已提交
658
  SOptrBasicInfo  binfo;
wmmhello's avatar
wmmhello 已提交
659 660
  SAggSupporter   aggSup;

661
  SArray*         pGroupCols;     // group by columns, SArray<SColumn>
dengyihao's avatar
dengyihao 已提交
662 663 664 665 666 667
  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;
668
  SExprSupp       scalarSup;
669 670
} SGroupbyOperatorInfo;

H
Haojun Liao 已提交
671
typedef struct SDataGroupInfo {
672 673 674
  uint64_t        groupId;
  int64_t         numOfRows;
  SArray*         pPageList;
H
Haojun Liao 已提交
675 676 677 678
} SDataGroupInfo;

// The sort in partition may be needed later.
typedef struct SPartitionOperatorInfo {
dengyihao's avatar
dengyihao 已提交
679 680 681 682 683 684 685 686 687 688
  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
689 690
  SArray*        sortedGroupArray;   // SDataGroupInfo sorted by group id
  int32_t        groupIndex;  // group index
691
  int32_t        pageIndex;   // page index of current group
692
  SSDataBlock*   pUpdateRes;
693
  SExprSupp      scalarSup;
H
Haojun Liao 已提交
694
} SPartitionOperatorInfo;
695

696
typedef struct SWindowRowsSup {
dengyihao's avatar
dengyihao 已提交
697 698 699 700
  STimeWindow win;
  TSKEY       prevTs;
  int32_t     startRowIndex;
  int32_t     numOfRows;
701 702
} SWindowRowsSup;

H
Haojun Liao 已提交
703
typedef struct SSessionAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
704
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
705 706
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
wmmhello's avatar
wmmhello 已提交
707

708 709 710 711
  SGroupResInfo      groupResInfo;
  SWindowRowsSup     winSup;
  bool               reptScan;        // next round scan
  int64_t            gap;             // session window gap
712
  int32_t            tsSlotId;        // primary timestamp slot id
713
  STimeWindowAggSupp twAggSup;
L
Liu Jicong 已提交
714
  const SNode* pCondition;
H
Haojun Liao 已提交
715
} SSessionAggOperatorInfo;
716

5
54liuyao 已提交
717 718 719 720
typedef struct SResultWindowInfo {
  SResultRowPosition pos;
  STimeWindow win;
  bool isOutput;
5
54liuyao 已提交
721
  bool isClosed;
5
54liuyao 已提交
722 723
} SResultWindowInfo;

5
54liuyao 已提交
724 725 726 727 728
typedef struct SStateWindowInfo {
  SResultWindowInfo winInfo;
  SStateKeys stateKey;
} SStateWindowInfo;

5
54liuyao 已提交
729 730 731
typedef struct SStreamSessionAggOperatorInfo {
  SOptrBasicInfo       binfo;
  SStreamAggSupporter  streamAggSup;
5
54liuyao 已提交
732
  SExprSupp            scalarSupp;         // supporter for perform scalar function
5
54liuyao 已提交
733 734 735
  SGroupResInfo        groupResInfo;
  int64_t              gap;             // session window gap
  int32_t              primaryTsIndex;  // primary timestamp slot id
5
54liuyao 已提交
736
  int32_t              endTsIndex;      // window end timestamp slot id
5
54liuyao 已提交
737 738 739 740
  int32_t              order;           // current SSDataBlock scan order
  STimeWindowAggSupp   twAggSup;
  SSDataBlock*         pWinBlock;       // window result
  SqlFunctionCtx*      pDummyCtx;       // for combine
5
54liuyao 已提交
741
  SSDataBlock*         pDelRes;         // delete result
5
54liuyao 已提交
742
  bool                 returnDelete;
5
54liuyao 已提交
743
  SSDataBlock*         pUpdateRes;      // update window
5
54liuyao 已提交
744 745
  SHashObj*            pStDeleted;
  void*                pDelIterator;
5
54liuyao 已提交
746
  SArray*              pChildren;       // cache for children's result; final stream operator
5
54liuyao 已提交
747 748
  SPhysiNode*          pPhyNode;           // create new child
  bool                 isFinal;
5
54liuyao 已提交
749
  bool                 ignoreExpiredData;
5
54liuyao 已提交
750 751
} SStreamSessionAggOperatorInfo;

752
typedef struct STimeSliceOperatorInfo {
753
  SSDataBlock*   pRes;
H
Haojun Liao 已提交
754
  STimeWindow    win;
dengyihao's avatar
dengyihao 已提交
755
  SInterval      interval;
H
Haojun Liao 已提交
756 757
  int64_t        current;
  SArray*        pPrevRow;      // SArray<SGroupValue>
758 759 760
  SArray*        pNextRow;      // SArray<SGroupValue>
  bool           isPrevRowSet;
  bool           isNextRowSet;
H
Haojun Liao 已提交
761
  int32_t        fillType;      // fill type
762 763
  SColumn        tsCol;         // primary timestamp column
  SExprSupp      scalarSup;     // scalar calculation
H
Haojun Liao 已提交
764
  struct SFillColInfo*  pFillColInfo;  // fill column info
765 766
} STimeSliceOperatorInfo;

767
typedef struct SStateWindowOperatorInfo {
wmmhello's avatar
wmmhello 已提交
768
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
769 770
  SOptrBasicInfo     binfo;
  SAggSupporter      aggSup;
wmmhello's avatar
wmmhello 已提交
771

772 773
  SGroupResInfo      groupResInfo;
  SWindowRowsSup     winSup;
774
  SColumn            stateCol;  // start row index
775 776
  bool               hasKey;
  SStateKeys         stateKey;
777
  int32_t            tsSlotId;  // primary timestamp column slot id
778
  STimeWindowAggSupp twAggSup;
dengyihao's avatar
dengyihao 已提交
779
  //  bool             reptScan;
L
Liu Jicong 已提交
780
  const SNode* pCondition;
781 782
} SStateWindowOperatorInfo;

5
54liuyao 已提交
783 784 785
typedef struct SStreamStateAggOperatorInfo {
  SOptrBasicInfo       binfo;
  SStreamAggSupporter  streamAggSup;
5
54liuyao 已提交
786
  SExprSupp            scalarSupp;      // supporter for perform scalar function
5
54liuyao 已提交
787 788 789 790
  SGroupResInfo        groupResInfo;
  int32_t              primaryTsIndex;  // primary timestamp slot id
  int32_t              order;           // current SSDataBlock scan order
  STimeWindowAggSupp   twAggSup;
5
54liuyao 已提交
791
  SColumn              stateCol;
5
54liuyao 已提交
792 793 794 795 796
  SqlFunctionCtx*      pDummyCtx;       // for combine
  SSDataBlock*         pDelRes;
  SHashObj*            pSeDeleted;
  void*                pDelIterator;
  SArray*              pChildren;       // cache for children's result;
5
54liuyao 已提交
797
  bool                 ignoreExpiredData;
5
54liuyao 已提交
798 799
} SStreamStateAggOperatorInfo;

800
typedef struct SSortedMergeOperatorInfo {
wmmhello's avatar
wmmhello 已提交
801
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
802
  SOptrBasicInfo   binfo;
wmmhello's avatar
wmmhello 已提交
803 804
  SAggSupporter    aggSup;

805 806 807 808 809 810 811 812 813 814 815
  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;
816
} SSortedMergeOperatorInfo;
H
Haojun Liao 已提交
817

H
Haojun Liao 已提交
818
typedef struct SSortOperatorInfo {
819
  SOptrBasicInfo binfo;
820
  uint32_t     sortBufSize;    // max buffer size for in-memory sort
dengyihao's avatar
dengyihao 已提交
821 822
  SArray*      pSortInfo;
  SSortHandle* pSortHandle;
823
  SArray*      pColMatchInfo;  // for index map from table scan output
dengyihao's avatar
dengyihao 已提交
824
  int32_t      bufPageSize;
H
Haojun Liao 已提交
825

826 827
  int64_t      startTs;       // sort start time
  uint64_t     sortElapsed;   // sort elapsed time, time to flush to disk not included.
828 829
  SLimitInfo   limitInfo;
  SNode*       pCondition;
H
Haojun Liao 已提交
830
} SSortOperatorInfo;
831

dengyihao's avatar
dengyihao 已提交
832 833 834
typedef struct STagFilterOperatorInfo {
  SOptrBasicInfo binfo;
} STagFilterOperatorInfo;
835

836 837 838
typedef struct SJoinOperatorInfo {
  SSDataBlock       *pRes;
  int32_t            joinType;
839
  int32_t            inputOrder;
840 841 842 843 844 845 846 847

  SSDataBlock       *pLeft;
  int32_t            leftPos;
  SColumnInfo        leftCol;

  SSDataBlock       *pRight;
  int32_t            rightPos;
  SColumnInfo        rightCol;
848
  SNode             *pCondAfterMerge;
849 850
} SJoinOperatorInfo;

851 852 853
#define OPTR_IS_OPENED(_optr)  (((_optr)->status & OP_OPENED) == OP_OPENED)
#define OPTR_SET_OPENED(_optr) ((_optr)->status |= OP_OPENED)

854 855
void doDestroyExchangeOperatorInfo(void* param);

856 857
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,
858
    __optr_decode_fn_t decode, __optr_explain_fn_t explain);
859

H
Haojun Liao 已提交
860
int32_t operatorDummyOpenFn(SOperatorInfo* pOperator);
dengyihao's avatar
dengyihao 已提交
861
void    operatorDummyCloseFn(void* param, int32_t numOfCols);
H
Haojun Liao 已提交
862
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num);
863 864 865

void    initBasicInfo(SOptrBasicInfo* pInfo, SSDataBlock* pBlock);
void    cleanupBasicInfo(SOptrBasicInfo* pInfo);
866
int32_t initExprSupp(SExprSupp* pSup, SExprInfo* pExprInfo, int32_t numOfExpr);
867
void    cleanupExprSupp(SExprSupp* pSup);
868
void    destroyExprInfo(SExprInfo* pExpr, int32_t numOfExprs);
869 870
int32_t initAggInfo(SExprSupp *pSup, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols, size_t keyBufSize,
                    const char* pkey);
871
void    initResultSizeInfo(SResultInfo * pResultInfo, int32_t numOfRows);
872
void    doBuildResultDatablock(SOperatorInfo* pOperator, SOptrBasicInfo* pbInfo, SGroupResInfo* pGroupResInfo, SDiskbasedBuf* pBuf);
873 874 875
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);
876

877
void    doApplyFunctions(SExecTaskInfo* taskInfo, SqlFunctionCtx* pCtx, STimeWindow* pWin, SColumnInfoData* pTimeWindowData, int32_t offset,
dengyihao's avatar
dengyihao 已提交
878
                         int32_t forwardStep, TSKEY* tsCol, int32_t numOfTotal, int32_t numOfOutput, int32_t order);
879 880

int32_t extractDataBlockFromFetchRsp(SSDataBlock* pRes, SLoadRemoteDataInfo* pLoadInfo, int32_t numOfRows, char* pData,
dengyihao's avatar
dengyihao 已提交
881 882
                                  int32_t compLen, int32_t numOfOutput, int64_t startTs, uint64_t* total,
                                  SArray* pColList);
883 884
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order);

885
int32_t getTableScanInfo(SOperatorInfo* pOperator, int32_t *order, int32_t* scanFlag);
886
int32_t getBufferPgSize(int32_t rowSize, uint32_t* defaultPgsz, uint32_t* defaultBufsz);
887

dengyihao's avatar
dengyihao 已提交
888
void    doSetOperatorCompleted(SOperatorInfo* pOperator);
889
void    doFilter(const SNode* pFilterNode, SSDataBlock* pBlock, const SArray* pColMatchInfo);
890 891
int32_t addTagPseudoColumnData(SReadHandle* pHandle, SExprInfo* pPseudoExpr, int32_t numOfPseudoExpr,
                               SSDataBlock* pBlock, const char* idStr);
892

893
void    cleanupAggSup(SAggSupporter* pAggSup);
894 895
void    destroyBasicOperatorInfo(void* param, int32_t numOfOutput);
void    appendOneRowToDataBlock(SSDataBlock* pBlock, STupleHandle* pTupleHandle);
896
void    setTbNameColData(void* pMeta, const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId);
897

L
Liu Jicong 已提交
898
int32_t doPrepareScan(SOperatorInfo* pOperator, uint64_t uid, int64_t ts);
899
int32_t doGetScanStatus(SOperatorInfo* pOperator, uint64_t* uid, int64_t* ts);
900

901
SSDataBlock* loadNextDataBlock(void* param);
902

903
void setResultRowInitCtx(SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowEntryInfoOffset);
904 905 906 907

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

909 910
SOperatorInfo* createExchangeOperatorInfo(void* pTransporter, SExchangePhysiNode* pExNode, SExecTaskInfo* pTaskInfo);

H
Haojun Liao 已提交
911
SOperatorInfo* createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle, SExecTaskInfo* pTaskInfo);
dengyihao's avatar
dengyihao 已提交
912 913
SOperatorInfo* createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pPhyNode,
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo);
914
SOperatorInfo* createSysTableScanOperatorInfo(void* readHandle, SSystemTableScanPhysiNode *pScanPhyNode, const char* pUser, SExecTaskInfo* pTaskInfo);
915

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

H
Haojun Liao 已提交
919
SOperatorInfo* createIndefinitOutputOperatorInfo(SOperatorInfo* downstream, SPhysiNode *pNode, SExecTaskInfo* pTaskInfo);
920
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SProjectPhysiNode* pProjPhyNode, SExecTaskInfo* pTaskInfo);
921
SOperatorInfo* createSortOperatorInfo(SOperatorInfo* downstream, SSortPhysiNode* pSortNode, SExecTaskInfo* pTaskInfo);
922
SOperatorInfo* createMultiwayMergeOperatorInfo(SOperatorInfo** dowStreams, size_t numStreams, SMergePhysiNode* pMergePhysiNode, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
923
SOperatorInfo* createSortedMergeOperatorInfo(SOperatorInfo** downstream, int32_t numOfDownstream, SExprInfo* pExprInfo, int32_t num, SArray* pSortInfo, SArray* pGroupInfo, SExecTaskInfo* pTaskInfo);
924
SOperatorInfo* createLastrowScanOperator(SLastRowScanPhysiNode* pTableScanNode, SReadHandle* readHandle, SExecTaskInfo* pTaskInfo);
925

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

930 931 932 933
SOperatorInfo* createMergeIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                          SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlotId,
                                          SExecTaskInfo* pTaskInfo);

934
SOperatorInfo* createMergeAlignedIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
S
shenglian zhou 已提交
935
                                          SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlotId,
936
                                          SNode* pCondition, SExecTaskInfo* pTaskInfo);
S
shenglian zhou 已提交
937

5
54liuyao 已提交
938 939
SOperatorInfo* createStreamFinalIntervalOperatorInfo(SOperatorInfo* downstream,
    SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, int32_t numOfChild);
940 941
SOperatorInfo* createStreamIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                                SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlotId,
wmmhello's avatar
wmmhello 已提交
942
                                                STimeWindowAggSupp *pTwAggSupp, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
943 944
SOperatorInfo* createSessionAggOperatorInfo(SOperatorInfo* downstream, SSessionWinodwPhysiNode* pSessionNode,
                                            SExecTaskInfo* pTaskInfo);
dengyihao's avatar
dengyihao 已提交
945 946
SOperatorInfo* createGroupOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols,
                                       SSDataBlock* pResultBlock, SArray* pGroupColList, SNode* pCondition,
wmmhello's avatar
wmmhello 已提交
947
                                       SExprInfo* pScalarExprInfo, int32_t numOfScalarExpr, SExecTaskInfo* pTaskInfo);
948 949
SOperatorInfo* createDataBlockInfoScanOperator(void* dataReader, SReadHandle* readHandle, uint64_t uid, SBlockDistScanPhysiNode* pBlockScanNode,
                                               SExecTaskInfo* pTaskInfo);
950

951
SOperatorInfo* createStreamScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode, SNode* pTagCond,
952
                                            SExecTaskInfo* pTaskInfo);
953

954 955
SOperatorInfo* createRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo);

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

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

962
SOperatorInfo* createPartitionOperatorInfo(SOperatorInfo* downstream, SPartitionPhysiNode* pPartNode, SExecTaskInfo* pTaskInfo);
963

964
SOperatorInfo* createTimeSliceOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pNode, SExecTaskInfo* pTaskInfo);
965

966
SOperatorInfo* createMergeJoinOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream, SSortMergeJoinPhysiNode* pJoinNode,
967
                                           SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
968

969 970 971 972
SOperatorInfo* createStreamSessionAggOperatorInfo(SOperatorInfo* downstream,
    SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo);
SOperatorInfo* createStreamFinalSessionAggOperatorInfo(SOperatorInfo* downstream,
    SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo, int32_t numOfChild);
5
54liuyao 已提交
973

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

H
Haojun Liao 已提交
976
#if 0
H
Haojun Liao 已提交
977
SOperatorInfo* createTableSeqScanOperatorInfo(void* pTsdbReadHandle, STaskRuntimeEnv* pRuntimeEnv);
H
Haojun Liao 已提交
978
#endif
979

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

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

985
bool    isTaskKilled(SExecTaskInfo* pTaskInfo);
986 987
int32_t checkForQueryBuf(size_t numOfTables);

988 989
void    setTaskKilled(SExecTaskInfo* pTaskInfo);
void    queryCostStatis(SExecTaskInfo* pTaskInfo);
990

dengyihao's avatar
dengyihao 已提交
991
void    doDestroyTask(SExecTaskInfo* pTaskInfo);
992 993
int32_t getMaximumIdleDurationSec();

wmmhello's avatar
wmmhello 已提交
994 995 996 997
/*
 * 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 已提交
998
 * nOptrWithVal: *nOptrWithVal save the number of optr with value
wmmhello's avatar
wmmhello 已提交
999 1000
 * return:  result code, 0 means success
 */
C
Cary Xu 已提交
1001
int32_t encodeOperator(SOperatorInfo* ops, char** data, int32_t *length, int32_t *nOptrWithVal);
wmmhello's avatar
wmmhello 已提交
1002 1003 1004 1005 1006 1007 1008

/*
 * 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 已提交
1009
int32_t decodeOperator(SOperatorInfo* ops, const char* data, int32_t length);
wmmhello's avatar
wmmhello 已提交
1010

1011
void    setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status);
dengyihao's avatar
dengyihao 已提交
1012
int32_t createExecTaskInfoImpl(SSubplan* pPlan, SExecTaskInfo** pTaskInfo, SReadHandle* pHandle, uint64_t taskId,
D
dapan1121 已提交
1013
                               char* sql, EOPTR_EXEC_MODEL model);
H
Haojun Liao 已提交
1014 1015
int32_t createDataSinkParam(SDataSinkNode *pNode, void **pParam, qTaskInfo_t* pTaskInfo, SReadHandle* readHandle);
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SArray* pExecInfoList);
dengyihao's avatar
dengyihao 已提交
1016

1017
int32_t aggDecodeResultRow(SOperatorInfo* pOperator, char* result);
wmmhello's avatar
wmmhello 已提交
1018 1019
int32_t aggEncodeResultRow(SOperatorInfo* pOperator, char** result, int32_t* length);

1020
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
1021
                                int32_t order);
1022 1023
int32_t getNumOfRowsInTimeWindow(SDataBlockInfo* pDataBlockInfo, TSKEY* pPrimaryColumn, int32_t startPos, TSKEY ekey,
    __block_search_fn_t searchFn, STableQueryInfo* item, int32_t order);
5
54liuyao 已提交
1024
int32_t binarySearchForKey(char* pValue, int num, TSKEY key, int order);
1025 1026
int32_t initStreamAggSupporter(SStreamAggSupporter* pSup, const char* pKey, SqlFunctionCtx* pCtx, int32_t numOfOutput,
    int32_t size);
1027
SResultRow* getNewResultRow(SDiskbasedBuf* pResultBuf, int64_t tableGroupId, int32_t interBufSize);
5
54liuyao 已提交
1028 1029
SResultWindowInfo* getSessionTimeWindow(SStreamAggSupporter* pAggSup, TSKEY startTs,
    TSKEY endTs, uint64_t groupId, int64_t gap, int32_t* pIndex);
5
54liuyao 已提交
1030 1031 1032
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 已提交
1033 1034
int32_t updateSessionWindowInfo(SResultWindowInfo* pWinInfo, TSKEY* pStartTs,
    TSKEY* pEndTs, int32_t rows, int32_t start, int64_t gap, SHashObj* pStDeleted);
5
54liuyao 已提交
1035
bool functionNeedToExecute(SqlFunctionCtx* pCtx);
5
54liuyao 已提交
1036
bool isCloseWindow(STimeWindow* pWin, STimeWindowAggSupp* pSup);
1037
bool isDeletedWindow(STimeWindow* pWin, uint64_t groupId, SAggSupporter* pSup);
1038
void appendOneRow(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, uint64_t* pUid);
5
54liuyao 已提交
1039
void printDataBlock(SSDataBlock* pBlock, const char* flag);
1040

1041
int32_t finalizeResultRowIntoResultDataBlock(SDiskbasedBuf* pBuf, SResultRowPosition* resultRowPosition,
1042 1043
                                       SqlFunctionCtx* pCtx, SExprInfo* pExprInfo, int32_t numOfExprs, const int32_t* rowCellOffset,
                                       SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo);
1044

1045
int32_t createScanTableListInfo(SScanPhysiNode *pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
1046
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond, const char* idstr);
1047

S
shenglian zhou 已提交
1048 1049
SOperatorInfo* createGroupSortOperatorInfo(SOperatorInfo* downstream, SGroupSortPhysiNode* pSortPhyNode,
                                           SExecTaskInfo* pTaskInfo);
1050
SOperatorInfo* createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, STableListInfo *pTableListInfo,
1051
                                                SReadHandle* readHandle, SExecTaskInfo* pTaskInfo);
1052

1053
void copyUpdateDataBlock(SSDataBlock* pDest, SSDataBlock* pSource, int32_t tsColIndex);
1054

1055
bool groupbyTbname(SNodeList* pGroupList);
wmmhello's avatar
wmmhello 已提交
1056
int32_t generateGroupIdMap(STableListInfo* pTableListInfo, SReadHandle* pHandle, SNodeList* groupKey);
1057
SSDataBlock* createSpecialDataBlock(EStreamType type);
5
54liuyao 已提交
1058
void* destroySqlFunctionCtx(SqlFunctionCtx* pCtx, int32_t numOfOutput);
1059

1060 1061 1062 1063
#ifdef __cplusplus
}
#endif

1064
#endif  // TDENGINE_EXECUTORIMPL_H