executorimpl.h 43.7 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
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"
5
54liuyao 已提交
37
#include "tfill.h"
H
Haojun Liao 已提交
38
#include "thash.h"
39
#include "tlockfree.h"
D
dapan1121 已提交
40
#include "tmsg.h"
dengyihao's avatar
dengyihao 已提交
41
#include "tpagedbuf.h"
L
Liu Jicong 已提交
42
#include "tstream.h"
L
Liu Jicong 已提交
43
#include "tstreamUpdate.h"
H
Haojun Liao 已提交
44

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

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

dengyihao's avatar
dengyihao 已提交
50
#define Q_STATUS_EQUAL(p, s)  (((p) & (s)) != 0u)
5
54liuyao 已提交
51 52 53 54 55
#define IS_VALID_SESSION_WIN(winInfo)        ((winInfo).sessionWin.win.skey > 0)
#define SET_SESSION_WIN_INVALID(winInfo)     ((winInfo).sessionWin.win.skey = INT64_MIN)
#define IS_INVALID_SESSION_WIN_KEY(winKey)   ((winKey).win.skey <= 0)
#define SET_SESSION_WIN_KEY_INVALID(pWinKey) ((pWinKey)->win.skey = INT64_MIN)

56
enum {
57 58
  // when this task starts to execute, this status will set
  TASK_NOT_COMPLETED = 0x1u,
59

60
  /* Task is over
61 62 63
   * 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
   */
64
  TASK_COMPLETED = 0x2u,
65 66 67 68 69 70
};

/**
 * 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 已提交
71 72 73 74 75
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 已提交
76
} SResultInfo;
77 78

typedef struct STableQueryInfo {
L
Liu Jicong 已提交
79 80
  TSKEY              lastKey;  // last check ts, todo remove it later
  SResultRowPosition pos;      // current active time window
81 82
} STableQueryInfo;

H
Haojun Liao 已提交
83 84 85 86 87
typedef struct SLimit {
  int64_t limit;
  int64_t offset;
} SLimit;

88
typedef struct STableScanAnalyzeInfo SFileBlockLoadRecorder;
89

H
Haojun Liao 已提交
90
typedef struct STaskCostInfo {
91 92
  int64_t  created;
  int64_t  start;
93 94 95 96 97 98 99 100
  uint64_t loadStatisTime;
  uint64_t loadFileBlockTime;
  uint64_t loadDataInCacheTime;
  uint64_t loadStatisSize;
  uint64_t loadFileBlockSize;
  uint64_t loadDataInCacheSize;

  uint64_t loadDataTime;
101 102

  SFileBlockLoadRecorder* pRecoder;
L
Liu Jicong 已提交
103
  uint64_t                elapsedTime;
104

105 106 107 108 109 110 111
  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 已提交
112
} STaskCostInfo;
113

H
Haojun Liao 已提交
114
typedef struct SOperatorCostInfo {
L
Liu Jicong 已提交
115 116
  double openCost;
  double totalCost;
H
Haojun Liao 已提交
117 118
} SOperatorCostInfo;

119 120
struct SOperatorInfo;

wmmhello's avatar
wmmhello 已提交
121
typedef int32_t (*__optr_encode_fn_t)(struct SOperatorInfo* pOperator, char** result, int32_t* length);
122
typedef int32_t (*__optr_decode_fn_t)(struct SOperatorInfo* pOperator, char* result);
123

124
typedef int32_t (*__optr_open_fn_t)(struct SOperatorInfo* pOptr);
125
typedef SSDataBlock* (*__optr_fn_t)(struct SOperatorInfo* pOptr);
126
typedef void (*__optr_close_fn_t)(void* param);
127
typedef int32_t (*__optr_explain_fn_t)(struct SOperatorInfo* pOptr, void** pOptrExplain, uint32_t* len);
H
Haojun Liao 已提交
128

H
Haojun Liao 已提交
129
typedef struct STaskIdInfo {
130 131 132 133
  uint64_t queryId;  // this is also a request id
  uint64_t subplanId;
  uint64_t templateId;
  char*    str;
H
Haojun Liao 已提交
134 135
} STaskIdInfo;

L
Liu Jicong 已提交
136 137 138 139 140 141
enum {
  STREAM_RECOVER_STEP__NONE = 0,
  STREAM_RECOVER_STEP__PREPARE,
  STREAM_RECOVER_STEP__SCAN,
};

L
Liu Jicong 已提交
142
typedef struct {
L
Liu Jicong 已提交
143 144 145 146 147 148
  // TODO remove prepareStatus
  STqOffsetVal prepareStatus;  // for tmq
  STqOffsetVal lastStatus;     // for tmq
  SMqMetaRsp   metaRsp;        // for tmq fetching meta
  int8_t       returned;
  int64_t      snapshotVer;
L
Liu Jicong 已提交
149
  const SSubmitReq*  pReq;
L
Liu Jicong 已提交
150 151 152 153 154 155 156

  SSchemaWrapper*     schema;
  char                tbName[TSDB_TABLE_NAME_LEN];
  SSDataBlock*        pullOverBlk;  // for streaming
  SWalFilterCond      cond;
  int64_t             lastScanUid;
  int8_t              recoverStep;
L
Liu Jicong 已提交
157
  SQueryTableDataCond tableCond;
L
Liu Jicong 已提交
158 159
  int64_t             recoverStartVer;
  int64_t             recoverEndVer;
L
Liu Jicong 已提交
160 161
  int64_t             fillHistoryVer1;
  int64_t             fillHistoryVer2;
L
Liu Jicong 已提交
162
  SStreamState*       pState;
L
Liu Jicong 已提交
163 164
} SStreamTaskInfo;

165 166 167 168 169
typedef struct {
  char*           tablename;
  char*           dbname;
  int32_t         tversion;
  SSchemaWrapper* sw;
170
  SSchemaWrapper* qsw;
171 172
} SSchemaInfo;

173
typedef struct SExecTaskInfo {
L
Liu Jicong 已提交
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
  STaskIdInfo   id;
  uint32_t      status;
  STimeWindow   window;
  STaskCostInfo cost;
  int64_t       owner;  // if it is in execution
  int32_t       code;

  int64_t               version;  // used for stream to record wal version
  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;
189
  struct SOperatorInfo* pRoot;
D
dapan1121 已提交
190
  SLocalFetch      localFetch;
191
} SExecTaskInfo;
H
Haojun Liao 已提交
192

193
enum {
L
Liu Jicong 已提交
194 195
  OP_NOT_OPENED = 0x0,
  OP_OPENED = 0x1,
H
Haojun Liao 已提交
196
  OP_RES_TO_RETURN = 0x5,
L
Liu Jicong 已提交
197
  OP_EXEC_DONE = 0x9,
L
Liu Jicong 已提交
198
  OP_EXEC_RECV = 0x11,
199 200
};

201
typedef struct SOperatorFpSet {
L
Liu Jicong 已提交
202 203 204 205 206 207 208 209
  __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;
210 211
} SOperatorFpSet;

212 213
typedef struct SExprSupp {
  SExprInfo*      pExprInfo;
L
Liu Jicong 已提交
214
  int32_t         numOfExprs;  // the number of scalar expression in group operator
215 216
  SqlFunctionCtx* pCtx;
  int32_t*        rowEntryInfoOffset;  // offset value for each row result cell info
H
Haojun Liao 已提交
217
  SFilterInfo*    pFilterInfo;
218 219
} SExprSupp;

220
typedef struct SOperatorInfo {
221
  uint16_t               operatorType;
222
  int16_t                resultDataBlockId;
L
Liu Jicong 已提交
223 224 225 226 227 228 229 230 231 232 233
  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;
234 235
} SOperatorInfo;

236 237
typedef enum {
  EX_SOURCE_DATA_NOT_READY = 0x1,
H
Haojun Liao 已提交
238
  EX_SOURCE_DATA_READY     = 0x2,
239 240
  EX_SOURCE_DATA_EXHAUSTED = 0x3,
} EX_SOURCE_STATUS;
241

242 243 244
#define COL_MATCH_FROM_COL_ID  0x1
#define COL_MATCH_FROM_SLOT_ID 0x2

245
typedef struct SSourceDataInfo {
L
Liu Jicong 已提交
246 247 248 249 250 251
  int32_t            index;
  SRetrieveTableRsp* pRsp;
  uint64_t           totalRows;
  int32_t            code;
  EX_SOURCE_STATUS   status;
  const char*        taskId;
252 253
} SSourceDataInfo;

H
Haojun Liao 已提交
254
typedef struct SLoadRemoteDataInfo {
dengyihao's avatar
dengyihao 已提交
255 256 257
  uint64_t totalSize;     // total load bytes from remote
  uint64_t totalRows;     // total number of rows
  uint64_t totalElapsed;  // total elapsed time
H
Haojun Liao 已提交
258 259
} SLoadRemoteDataInfo;

260
typedef struct SLimitInfo {
L
Liu Jicong 已提交
261 262 263 264 265 266 267
  SLimit   limit;
  SLimit   slimit;
  uint64_t currentGroupId;
  int64_t  remainGroupOffset;
  int64_t  numOfOutputGroups;
  int64_t  remainOffset;
  int64_t  numOfOutputRows;
268 269
} SLimitInfo;

270
typedef struct SExchangeInfo {
L
Liu Jicong 已提交
271 272 273 274
  SArray* pSources;
  SArray* pSourceDataInfo;
  tsem_t  ready;
  void*   pTransporter;
275 276 277
  // SArray<SSDataBlock*>, result block list, used to keep the multi-block that
  // passed by downstream operator
  SArray*             pResultBlockList;
L
Liu Jicong 已提交
278 279 280
  int32_t             rspBlockIndex;  // indicate the return block index in pResultBlockList
  SSDataBlock*        pDummyBlock;    // dummy block, not keep data
  bool                seqLoadData;    // sequential load data or not, false by default
dengyihao's avatar
dengyihao 已提交
281
  int32_t             current;
H
Haojun Liao 已提交
282
  SLoadRemoteDataInfo loadInfo;
283
  uint64_t            self;
284
  SLimitInfo          limitInfo;
285 286
} SExchangeInfo;

287
typedef struct SScanInfo {
L
Liu Jicong 已提交
288 289
  int32_t numOfAsc;
  int32_t numOfDesc;
290 291
} SScanInfo;

292
typedef struct SSampleExecInfo {
L
Liu Jicong 已提交
293 294
  double   sampleRatio;  // data block sample ratio, 1 by default
  uint32_t seed;         // random seed value
295 296
} SSampleExecInfo;

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

302
typedef struct SAggSupporter {
H
Haojun Liao 已提交
303
  SSHashObj*     pResultRowHashTable;  // quick locate the window object for each result
304 305
  char*          keyBuf;               // window key buffer
  SDiskbasedBuf* pResultBuf;           // query result buffer based on blocked-wised disk file
L
Liu Jicong 已提交
306 307
  int32_t        resultRowSize;  // the result buffer size for each result row, with the meta data size for each row
  int32_t        currentPageId;  // current write page id
308 309 310
} SAggSupporter;

typedef struct {
L
Liu Jicong 已提交
311 312 313 314 315
  // 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
316 317
} SAggOptrPushDownInfo;

318
typedef struct STableScanInfo {
L
Liu Jicong 已提交
319 320
  STsdbReader* dataReader;
  SReadHandle  readHandle;
321

322
  SFileBlockLoadRecorder readRecorder;
L
Liu Jicong 已提交
323 324 325
  SScanInfo              scanInfo;
  int32_t                scanTimes;
  SNode*                 pFilterNode;  // filter info, which is push down by optimizer
326

L
Liu Jicong 已提交
327
  SSDataBlock*         pResBlock;
H
Haojun Liao 已提交
328
  SColMatchInfo        matchInfo;
L
Liu Jicong 已提交
329 330 331 332 333 334 335 336 337
  SExprSupp            pseudoSup;
  SQueryTableDataCond  cond;
  int32_t              scanFlag;  // table scan flag to denote if it is a repeat/reverse/main scan
  int32_t              dataBlockLoadFlag;
  SSampleExecInfo      sample;  // sample execution info
  int32_t              currentGroupId;
  int32_t              currentTable;
  int8_t               scanMode;
  int8_t               noTable;
338
  SAggOptrPushDownInfo pdInfo;
L
Liu Jicong 已提交
339
  int8_t               assignBlockUid;
340 341
} STableScanInfo;

342 343 344 345 346 347 348 349 350 351 352 353
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;
H
Haojun Liao 已提交
354 355 356
  SSDataBlock*    pSortInputBlock;
  int64_t         startTs;  // sort start time
  SArray*         sortSourceParams;
357 358 359 360 361 362 363 364 365 366 367

  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;
H
Haojun Liao 已提交
368
  SColMatchInfo          matchInfo;
369
  int32_t                numOfOutput;
H
Haojun Liao 已提交
370 371 372 373
  SExprSupp              pseudoSup;
  SQueryTableDataCond    cond;
  int32_t                scanFlag;  // table scan flag to denote if it is a repeat/reverse/main scan
  int32_t                dataBlockLoadFlag;
374 375
  // 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.
H
Haojun Liao 已提交
376 377 378
  SInterval              interval;
  SSampleExecInfo        sample;  // sample execution info
  SSortExecInfo          sortExecInfo;
379 380
} STableMergeScanInfo;

381
typedef struct STagScanInfo {
L
Liu Jicong 已提交
382 383
  SColumnInfo*    pCols;
  SSDataBlock*    pRes;
H
Haojun Liao 已提交
384
  SColMatchInfo   matchInfo;
L
Liu Jicong 已提交
385 386 387
  int32_t         curPos;
  SReadHandle     readHandle;
  STableListInfo* pTableList;
388 389
} STagScanInfo;

390
typedef struct SLastrowScanInfo {
H
Haojun Liao 已提交
391 392 393 394 395 396 397 398 399 400 401
  SSDataBlock*   pRes;
  SReadHandle    readHandle;
  void*          pLastrowReader;
  SColMatchInfo  matchInfo;
  int32_t*       pSlotIds;
  SExprSupp      pseudoExprSup;
  int32_t        retrieveType;
  int32_t        currentGroupIndex;
  SSDataBlock*   pBufferredRes;
  SArray*        pUidList;
  int32_t        indexOfBufferedRes;
402 403
} SLastrowScanInfo;

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

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

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

5
54liuyao 已提交
425
typedef struct SStreamAggSupporter {
5
54liuyao 已提交
426 427 428 429 430 431 432 433 434
  int32_t         resultRowSize;  // the result buffer size for each result row, with the meta data size for each row
  SSDataBlock*    pScanBlock;
  SStreamState*   pState;
  int64_t         gap;            // stream session window gap
  SqlFunctionCtx* pDummyCtx;      // for combine
  SSHashObj*      pResultRows;
  int32_t         stateKeySize;
  int16_t         stateKeyType;
  SDiskbasedBuf*  pResultBuf;
5
54liuyao 已提交
435 436
} SStreamAggSupporter;

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

444
typedef struct SPartitionBySupporter {
L
Liu Jicong 已提交
445 446 447 448
  SArray* pGroupCols;     // group by columns, SArray<SColumn>
  SArray* pGroupColVals;  // current group column values, SArray<SGroupKeys>
  char*   keyBuf;         // group by keys for hash
  bool    needCalc;       // partition by column
449 450 451
} SPartitionBySupporter;

typedef struct SPartitionDataInfo {
L
Liu Jicong 已提交
452
  uint64_t groupId;
453 454
  char*    tbname;
  SArray*  tags;
L
Liu Jicong 已提交
455
  SArray*  rowIds;
456
} SPartitionDataInfo;
457

5
54liuyao 已提交
458
typedef struct STimeWindowAggSupp {
L
Liu Jicong 已提交
459 460
  int8_t          calTrigger;
  int64_t         waterMark;
5
54liuyao 已提交
461
  int64_t         deleteMark;
L
Liu Jicong 已提交
462
  TSKEY           maxTs;
5
54liuyao 已提交
463
  TSKEY           minTs;
L
Liu Jicong 已提交
464
  SColumnInfoData timeWindowData;  // query time window info for scalar function execution.
5
54liuyao 已提交
465 466
} STimeWindowAggSupp;

467
typedef struct SStreamScanInfo {
H
Haojun Liao 已提交
468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489
  uint64_t              tableUid;  // queried super table uid
  SExprInfo*            pPseudoExpr;
  int32_t               numOfPseudoExpr;
  SExprSupp             tbnameCalSup;
  SExprSupp             tagCalSup;
  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.
  SColMatchInfo         matchInfo;
  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;
L
Liu Jicong 已提交
490

L
Liu Jicong 已提交
491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506
  EStreamScanMode       scanMode;
  SOperatorInfo*        pStreamScanOp;
  SOperatorInfo*        pTableScanOp;
  SArray*               childIds;
  SWindowSupporter      windowSup;
  SPartitionBySupporter partitionSup;
  SExprSupp*            pPartScalarSup;
  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;
  STimeWindow           updateWin;
  STimeWindowAggSupp    twAggSup;
  SSDataBlock*          pUpdateDataRes;
L
Liu Jicong 已提交
507
  // status for tmq
L
Liu Jicong 已提交
508 509 510
  SNodeList* pGroupTags;
  SNode*     pTagCond;
  SNode*     pTagIndexCond;
511
} SStreamScanInfo;
H
Haojun Liao 已提交
512

L
Liu Jicong 已提交
513 514 515 516 517 518 519 520 521 522 523 524
typedef struct {
  //  int8_t    subType;
  //  bool      withMeta;
  //  int64_t   suid;
  //  int64_t   snapVersion;
  //  void     *metaInfo;
  //  void     *dataInfo;
  SVnode*       vnode;
  SSDataBlock   pRes;  // result SSDataBlock
  STsdbReader*  dataReader;
  SSnapContext* sContext;
} SStreamRawScanInfo;
525

dengyihao's avatar
dengyihao 已提交
526 527 528 529 530 531
typedef struct SSysTableIndex {
  int8_t init;
  SArray *uids; 
  int32_t lastIdx; 
} SSysTableIndex;

532
typedef struct SSysTableScanInfo {
dengyihao's avatar
dengyihao 已提交
533 534 535 536
  SRetrieveMetaTableRsp* pRsp;
  SRetrieveTableReq      req;
  SEpSet                 epSet;
  tsem_t                 ready;
537 538
  SReadHandle            readHandle;
  int32_t                accountId;
539
  const char*            pUser;
540
  bool                   sysInfo;
541 542 543
  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.
H
Haojun Liao 已提交
544 545
  SSysTableIndex*        pIdx;         // idx for local table meta
  SColMatchInfo          matchInfo;
546 547 548 549
  SName                  name;
  SSDataBlock*           pRes;
  int64_t                numOfBlocks;  // extract basic running information.
  SLoadRemoteDataInfo    loadInfo;
550 551
} SSysTableScanInfo;

552 553 554
typedef struct SBlockDistInfo {
  SSDataBlock* pResBlock;
  void*        pHandle;
555
  SReadHandle  readHandle;
L
Liu Jicong 已提交
556
  uint64_t     uid;  // table uid
557 558
} SBlockDistInfo;

559
// todo remove this
560
typedef struct SOptrBasicInfo {
L
Liu Jicong 已提交
561 562 563
  SResultRowInfo resultRowInfo;
  SSDataBlock*   pRes;
  bool           mergeResultBlock;
564 565
} SOptrBasicInfo;

566
typedef struct SIntervalAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
567
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
568
  SOptrBasicInfo     binfo;              // basic info
wmmhello's avatar
wmmhello 已提交
569
  SAggSupporter      aggSup;             // aggregate supporter
570
  SExprSupp          scalarSupp;         // supporter for perform scalar function
571 572 573 574 575
  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
576
  SArray*            pInterpCols;        // interpolation columns
577
  int32_t            resultTsOrder;      // result timestamp order
578
  int32_t            inputOrder;         // input data ts order
579 580
  EOPTR_EXEC_MODEL   execModel;          // operator execution model [batch model|stream model]
  STimeWindowAggSupp twAggSup;
L
Liu Jicong 已提交
581
  SArray*            pPrevValues;  //  SArray<SGroupKeys> used to keep the previous not null value for interpolation.
582
  SNode*             pCondition;
583
} SIntervalAggOperatorInfo;
584

585
typedef struct SMergeAlignedIntervalAggOperatorInfo {
L
Liu Jicong 已提交
586
  SIntervalAggOperatorInfo* intervalAggOperatorInfo;
587

588
//  bool         hasGroupId;
589 590
  uint64_t     groupId;  // current groupId
  int64_t      curTs;    // current ts
591 592
  SSDataBlock* prefetchedBlock;
  SNode*       pCondition;
593
  SResultRow*  pResultRow;
594 595
} SMergeAlignedIntervalAggOperatorInfo;

596
typedef struct SStreamIntervalOperatorInfo {
wmmhello's avatar
wmmhello 已提交
597
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
5
54liuyao 已提交
598
  SOptrBasicInfo     binfo;              // basic info
wmmhello's avatar
wmmhello 已提交
599
  SAggSupporter      aggSup;             // aggregate supporter
5
54liuyao 已提交
600
  SExprSupp          scalarSupp;         // supporter for perform scalar function
5
54liuyao 已提交
601 602 603
  SGroupResInfo      groupResInfo;       // multiple results build supporter
  SInterval          interval;           // interval info
  int32_t            primaryTsIndex;     // primary time stamp slot id from result of downstream operator.
604 605 606 607 608 609
  STimeWindowAggSupp twAggSup;
  bool               invertible;
  bool               ignoreExpiredData;
  SArray*            pDelWins;           // SWinRes
  int32_t            delIndex;
  SSDataBlock*       pDelRes;
610
  SPhysiNode*        pPhyNode;           // create new child
5
54liuyao 已提交
611
  SHashObj*          pPullDataMap;
612
  SArray*            pPullWins;          // SPullWindowInfo
5
54liuyao 已提交
613 614
  int32_t            pullIndex;
  SSDataBlock*       pPullDataRes;
615 616 617 618
  bool               isFinal;
  SArray*            pChildren;
  SStreamState*      pState;
  SWinKey            delKey;
619
  SHashObj*          pGroupIdTbNameMap;  // uint64_t -> char[TSDB_TABLE_NAME_LEN]
620
} SStreamIntervalOperatorInfo;
5
54liuyao 已提交
621

622
typedef struct SAggOperatorInfo {
wmmhello's avatar
wmmhello 已提交
623
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
L
Liu Jicong 已提交
624 625
  SOptrBasicInfo binfo;
  SAggSupporter  aggSup;
wmmhello's avatar
wmmhello 已提交
626

L
Liu Jicong 已提交
627 628 629 630 631
  STableQueryInfo* current;
  uint64_t         groupId;
  SGroupResInfo    groupResInfo;
  SExprSupp        scalarExprSup;
  SNode*           pCondition;
632 633 634
} SAggOperatorInfo;

typedef struct SProjectOperatorInfo {
L
Liu Jicong 已提交
635 636 637 638 639 640 641 642
  SOptrBasicInfo binfo;
  SAggSupporter  aggSup;
  SNode*         pFilterNode;  // filter info, which is push down by optimizer
  SArray*        pPseudoColInfo;
  SLimitInfo     limitInfo;
  bool           mergeDataBlocks;
  SSDataBlock*   pFinalRes;
  SNode*         pCondition;
643 644
} SProjectOperatorInfo;

H
Haojun Liao 已提交
645
typedef struct SIndefOperatorInfo {
L
Liu Jicong 已提交
646 647 648 649 650 651
  SOptrBasicInfo binfo;
  SAggSupporter  aggSup;
  SArray*        pPseudoColInfo;
  SExprSupp      scalarSup;
  SNode*         pCondition;
  uint64_t       groupId;
652

L
Liu Jicong 已提交
653
  SSDataBlock* pNextGroupRes;
H
Haojun Liao 已提交
654 655
} SIndefOperatorInfo;

656
typedef struct SFillOperatorInfo {
657 658
  struct SFillInfo* pFillInfo;
  SSDataBlock*      pRes;
H
Haojun Liao 已提交
659
  SSDataBlock*      pFinalRes;
660 661 662
  int64_t           totalInputRows;
  void**            p;
  SSDataBlock*      existNewGroupBlock;
663
  STimeWindow       win;
S
slzhou 已提交
664
  SNode*            pCondition;
H
Haojun Liao 已提交
665
  SColMatchInfo     matchInfo;
H
Haojun Liao 已提交
666
  int32_t           primaryTsCol;
H
Haojun Liao 已提交
667
  int32_t           primarySrcSlotId;
L
Liu Jicong 已提交
668
  uint64_t          curGroupId;  // current handled group id
669 670
  SExprInfo*        pExprInfo;
  int32_t           numOfExpr;
671
  SExprSupp         noFillExprSupp;
672 673 674
} SFillOperatorInfo;

typedef struct SGroupbyOperatorInfo {
L
Liu Jicong 已提交
675 676
  SOptrBasicInfo binfo;
  SAggSupporter  aggSup;
H
Haojun Liao 已提交
677 678 679 680 681 682 683 684
  SArray*        pGroupCols;     // group by columns, SArray<SColumn>
  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;
  SExprSupp      scalarSup;
685 686
} SGroupbyOperatorInfo;

H
Haojun Liao 已提交
687
typedef struct SDataGroupInfo {
L
Liu Jicong 已提交
688 689 690
  uint64_t groupId;
  int64_t  numOfRows;
  SArray*  pPageList;
H
Haojun Liao 已提交
691 692 693 694
} SDataGroupInfo;

// The sort in partition may be needed later.
typedef struct SPartitionOperatorInfo {
dengyihao's avatar
dengyihao 已提交
695 696 697 698 699 700 701
  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

L
Liu Jicong 已提交
702 703 704 705 706 707
  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
  SArray*        sortedGroupArray;  // SDataGroupInfo sorted by group id
  int32_t        groupIndex;        // group index
  int32_t        pageIndex;         // page index of current group
708
  SExprSupp      scalarSup;
H
Haojun Liao 已提交
709
} SPartitionOperatorInfo;
710

711
typedef struct SWindowRowsSup {
dengyihao's avatar
dengyihao 已提交
712 713 714 715
  STimeWindow win;
  TSKEY       prevTs;
  int32_t     startRowIndex;
  int32_t     numOfRows;
716
  uint64_t    groupId;
717 718
} SWindowRowsSup;

H
Haojun Liao 已提交
719
typedef struct SSessionAggOperatorInfo {
L
Liu Jicong 已提交
720 721
  SOptrBasicInfo binfo;
  SAggSupporter  aggSup;
wmmhello's avatar
wmmhello 已提交
722

723 724
  SGroupResInfo      groupResInfo;
  SWindowRowsSup     winSup;
L
Liu Jicong 已提交
725 726 727
  bool               reptScan;  // next round scan
  int64_t            gap;       // session window gap
  int32_t            tsSlotId;  // primary timestamp slot id
728
  STimeWindowAggSupp twAggSup;
L
Liu Jicong 已提交
729
  const SNode*       pCondition;
H
Haojun Liao 已提交
730
} SSessionAggOperatorInfo;
731

5
54liuyao 已提交
732
typedef struct SResultWindowInfo {
5
54liuyao 已提交
733 734
  void*              pOutputBuf;
  SSessionKey        sessionWin;
L
Liu Jicong 已提交
735
  bool               isOutput;
5
54liuyao 已提交
736 737
} SResultWindowInfo;

5
54liuyao 已提交
738 739
typedef struct SStateWindowInfo {
  SResultWindowInfo winInfo;
5
54liuyao 已提交
740
  SStateKeys*       pStateKey;
5
54liuyao 已提交
741 742
} SStateWindowInfo;

5
54liuyao 已提交
743
typedef struct SStreamSessionAggOperatorInfo {
L
Liu Jicong 已提交
744 745
  SOptrBasicInfo      binfo;
  SStreamAggSupporter streamAggSup;
5
54liuyao 已提交
746
  SExprSupp           scalarSupp;      // supporter for perform scalar function
L
Liu Jicong 已提交
747 748 749 750 751
  SGroupResInfo       groupResInfo;
  int32_t             primaryTsIndex;  // primary timestamp slot id
  int32_t             endTsIndex;      // window end timestamp slot id
  int32_t             order;           // current SSDataBlock scan order
  STimeWindowAggSupp  twAggSup;
5
54liuyao 已提交
752 753 754
  SSDataBlock*        pWinBlock;       // window result
  SSDataBlock*        pDelRes;         // delete result
  SSDataBlock*        pUpdateRes;      // update window
L
Liu Jicong 已提交
755
  bool                returnUpdate;
5
54liuyao 已提交
756
  SSHashObj*          pStDeleted;
L
Liu Jicong 已提交
757
  void*               pDelIterator;
5
54liuyao 已提交
758 759
  SArray*             pChildren;       // cache for children's result; final stream operator
  SPhysiNode*         pPhyNode;        // create new child
L
Liu Jicong 已提交
760 761
  bool                isFinal;
  bool                ignoreExpiredData;
762
  SHashObj*           pGroupIdTbNameMap;
5
54liuyao 已提交
763 764
} SStreamSessionAggOperatorInfo;

5
54liuyao 已提交
765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780
typedef struct SStreamStateAggOperatorInfo {
  SOptrBasicInfo      binfo;
  SStreamAggSupporter streamAggSup;
  SExprSupp           scalarSupp;      // supporter for perform scalar function
  SGroupResInfo       groupResInfo;
  int32_t             primaryTsIndex;  // primary timestamp slot id
  STimeWindowAggSupp  twAggSup;
  SColumn             stateCol;
  SSDataBlock*        pDelRes;
  SSHashObj*          pSeDeleted;
  void*               pDelIterator;
  SArray*             pChildren;       // cache for children's result;
  bool                ignoreExpiredData;
  SHashObj*           pGroupIdTbNameMap;
} SStreamStateAggOperatorInfo;

781 782 783 784
typedef struct SStreamPartitionOperatorInfo {
  SOptrBasicInfo        binfo;
  SPartitionBySupporter partitionSup;
  SExprSupp             scalarSup;
785
  SExprSupp             tbnameCalSup;
L
Liu Jicong 已提交
786
  SExprSupp             tagCalSup;
787 788 789 790
  SHashObj*             pPartitions;
  void*                 parIte;
  SSDataBlock*          pInputDataBlock;
  int32_t               tsColIndex;
791
  SSDataBlock*          pDelRes;
792 793
} SStreamPartitionOperatorInfo;

5
54liuyao 已提交
794 795 796 797 798 799 800 801 802 803
typedef struct SStreamFillOperatorInfo {
  SStreamFillSupporter* pFillSup;
  SSDataBlock*      pRes;
  SSDataBlock*      pSrcBlock;
  int32_t           srcRowIndex;
  SSDataBlock*      pPrevSrcBlock;
  SSDataBlock*      pSrcDelBlock;
  int32_t           srcDelRowIndex;
  SSDataBlock*      pDelRes;
  SNode*            pCondition;
H
Haojun Liao 已提交
804
  SColMatchInfo     matchInfo;
5
54liuyao 已提交
805 806 807 808 809
  int32_t           primaryTsCol;
  int32_t           primarySrcSlotId;
  SStreamFillInfo*  pFillInfo;
} SStreamFillOperatorInfo;

810
typedef struct STimeSliceOperatorInfo {
L
Liu Jicong 已提交
811 812 813 814 815 816 817 818 819 820 821 822 823 824
  SSDataBlock*         pRes;
  STimeWindow          win;
  SInterval            interval;
  int64_t              current;
  SArray*              pPrevRow;     // SArray<SGroupValue>
  SArray*              pNextRow;     // SArray<SGroupValue>
  SArray*              pLinearInfo;  // SArray<SFillLinearInfo>
  bool                 fillLastPoint;
  bool                 isPrevRowSet;
  bool                 isNextRowSet;
  int32_t              fillType;      // fill type
  SColumn              tsCol;         // primary timestamp column
  SExprSupp            scalarSup;     // scalar calculation
  struct SFillColInfo* pFillColInfo;  // fill column info
825 826
} STimeSliceOperatorInfo;

827
typedef struct SStateWindowOperatorInfo {
wmmhello's avatar
wmmhello 已提交
828
  // SOptrBasicInfo should be first, SAggSupporter should be second for stream encode
L
Liu Jicong 已提交
829 830
  SOptrBasicInfo binfo;
  SAggSupporter  aggSup;
831
  SExprSupp      scalarSup;
wmmhello's avatar
wmmhello 已提交
832

833 834
  SGroupResInfo      groupResInfo;
  SWindowRowsSup     winSup;
835
  SColumn            stateCol;  // start row index
836 837
  bool               hasKey;
  SStateKeys         stateKey;
838
  int32_t            tsSlotId;  // primary timestamp column slot id
839
  STimeWindowAggSupp twAggSup;
L
Liu Jicong 已提交
840
  const SNode* pCondition;
841 842
} SStateWindowOperatorInfo;

H
Haojun Liao 已提交
843
typedef struct SSortOperatorInfo {
844
  SOptrBasicInfo binfo;
L
Liu Jicong 已提交
845 846 847
  uint32_t       sortBufSize;  // max buffer size for in-memory sort
  SArray*        pSortInfo;
  SSortHandle*   pSortHandle;
H
Haojun Liao 已提交
848
  SColMatchInfo  matchInfo;
L
Liu Jicong 已提交
849
  int32_t        bufPageSize;
850 851 852 853
  int64_t        startTs;      // sort start time
  uint64_t       sortElapsed;  // sort elapsed time, time to flush to disk not included.
  SLimitInfo     limitInfo;
  SNode*         pCondition;
H
Haojun Liao 已提交
854
} SSortOperatorInfo;
855

dengyihao's avatar
dengyihao 已提交
856 857 858
typedef struct STagFilterOperatorInfo {
  SOptrBasicInfo binfo;
} STagFilterOperatorInfo;
859

860
typedef struct SJoinOperatorInfo {
L
Liu Jicong 已提交
861 862 863 864 865 866 867 868 869 870 871 872
  SSDataBlock* pRes;
  int32_t      joinType;
  int32_t      inputOrder;

  SSDataBlock* pLeft;
  int32_t      leftPos;
  SColumnInfo  leftCol;

  SSDataBlock* pRight;
  int32_t      rightPos;
  SColumnInfo  rightCol;
  SNode*       pCondAfterMerge;
873 874
} SJoinOperatorInfo;

875 876 877
#define OPTR_IS_OPENED(_optr)  (((_optr)->status & OP_OPENED) == OP_OPENED)
#define OPTR_SET_OPENED(_optr) ((_optr)->status |= OP_OPENED)

878 879
void doDestroyExchangeOperatorInfo(void* param);

880
SOperatorFpSet createOperatorFpSet(__optr_open_fn_t openFn, __optr_fn_t nextFn, __optr_fn_t streamFn,
881
                                   __optr_fn_t cleanup, __optr_close_fn_t closeFn, __optr_explain_fn_t explain);
882

H
Haojun Liao 已提交
883 884
int32_t operatorDummyOpenFn(SOperatorInfo* pOperator);
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num);
885 886 887

void    initBasicInfo(SOptrBasicInfo* pInfo, SSDataBlock* pBlock);
void    cleanupBasicInfo(SOptrBasicInfo* pInfo);
888
int32_t initExprSupp(SExprSupp* pSup, SExprInfo* pExprInfo, int32_t numOfExpr);
889
void    cleanupExprSupp(SExprSupp* pSup);
890
void    destroyExprInfo(SExprInfo* pExpr, int32_t numOfExprs);
L
Liu Jicong 已提交
891
int32_t initAggInfo(SExprSupp* pSup, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols, size_t keyBufSize,
892
                    const char* pkey);
L
Liu Jicong 已提交
893
void    initResultSizeInfo(SResultInfo* pResultInfo, int32_t numOfRows);
894 895 896

void doBuildStreamResBlock(SOperatorInfo* pOperator, SOptrBasicInfo* pbInfo, SGroupResInfo* pGroupResInfo,
                           SDiskbasedBuf* pBuf);
L
Liu Jicong 已提交
897 898
void    doBuildResultDatablock(SOperatorInfo* pOperator, SOptrBasicInfo* pbInfo, SGroupResInfo* pGroupResInfo,
                               SDiskbasedBuf* pBuf);
899

L
Liu Jicong 已提交
900
int32_t handleLimitOffset(SOperatorInfo* pOperator, SLimitInfo* pLimitInfo, SSDataBlock* pBlock, bool holdDataInBuf);
901 902
bool    hasLimitOffsetInfo(SLimitInfo* pLimitInfo);
void    initLimitInfo(const SNode* pLimit, const SNode* pSLimit, SLimitInfo* pLimitInfo);
903

L
Liu Jicong 已提交
904 905
void doApplyFunctions(SExecTaskInfo* taskInfo, SqlFunctionCtx* pCtx, SColumnInfoData* pTimeWindowData, int32_t offset,
                      int32_t forwardStep, int32_t numOfTotal, int32_t numOfOutput);
906

H
Haojun Liao 已提交
907
int32_t extractDataBlockFromFetchRsp(SSDataBlock* pRes, char* pData, SArray* pColList, char** pNextStart);
L
Liu Jicong 已提交
908 909
void    updateLoadRemoteInfo(SLoadRemoteDataInfo* pInfo, int32_t numOfRows, int32_t dataLen, int64_t startTs,
                             SOperatorInfo* pOperator);
910

911 912
STimeWindow getFirstQualifiedTimeWindow(int64_t ts, STimeWindow* pWindow, SInterval* pInterval, int32_t order);

L
Liu Jicong 已提交
913
int32_t getTableScanInfo(SOperatorInfo* pOperator, int32_t* order, int32_t* scanFlag);
914
int32_t getBufferPgSize(int32_t rowSize, uint32_t* defaultPgsz, uint32_t* defaultBufsz);
915

dengyihao's avatar
dengyihao 已提交
916
void    doSetOperatorCompleted(SOperatorInfo* pOperator);
H
Haojun Liao 已提交
917
void    doFilter(const SNode* pFilterNode, SSDataBlock* pBlock, SColMatchInfo* pColMatchInfo, SFilterInfo* pFilterInfo);
918 919
int32_t addTagPseudoColumnData(SReadHandle* pHandle, SExprInfo* pPseudoExpr, int32_t numOfPseudoExpr,
                               SSDataBlock* pBlock, const char* idStr);
920

L
Liu Jicong 已提交
921 922 923
void cleanupAggSup(SAggSupporter* pAggSup);
void appendOneRowToDataBlock(SSDataBlock* pBlock, STupleHandle* pTupleHandle);
void setTbNameColData(void* pMeta, const SSDataBlock* pBlock, SColumnInfoData* pColInfoData, int32_t functionId);
924

L
Liu Jicong 已提交
925
int32_t doPrepareScan(SOperatorInfo* pOperator, uint64_t uid, int64_t ts);
926
int32_t doGetScanStatus(SOperatorInfo* pOperator, uint64_t* uid, int64_t* ts);
927

928
SSDataBlock* loadNextDataBlock(void* param);
929

930
void setResultRowInitCtx(SResultRow* pResult, SqlFunctionCtx* pCtx, int32_t numOfOutput, int32_t* rowEntryInfoOffset);
931

L
Liu Jicong 已提交
932 933 934
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 已提交
935

936 937
SOperatorInfo* createExchangeOperatorInfo(void* pTransporter, SExchangePhysiNode* pExNode, SExecTaskInfo* pTaskInfo);

L
Liu Jicong 已提交
938 939
SOperatorInfo* createTableScanOperatorInfo(STableScanPhysiNode* pTableScanNode, SReadHandle* pHandle,
                                           SExecTaskInfo* pTaskInfo);
dengyihao's avatar
dengyihao 已提交
940 941
SOperatorInfo* createTagScanOperatorInfo(SReadHandle* pReadHandle, STagScanPhysiNode* pPhyNode,
                                         STableListInfo* pTableListInfo, SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
942 943
SOperatorInfo* createSysTableScanOperatorInfo(void* readHandle, SSystemTableScanPhysiNode* pScanPhyNode,
                                              const char* pUser, SExecTaskInfo* pTaskInfo);
944

H
Haojun Liao 已提交
945
SOperatorInfo* createAggregateOperatorInfo(SOperatorInfo* downstream, SAggPhysiNode* pNode, SExecTaskInfo* pTaskInfo);
946

L
Liu Jicong 已提交
947 948 949 950
SOperatorInfo* createIndefinitOutputOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pNode,
                                                 SExecTaskInfo* pTaskInfo);
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SProjectPhysiNode* pProjPhyNode,
                                         SExecTaskInfo* pTaskInfo);
951
SOperatorInfo* createSortOperatorInfo(SOperatorInfo* downstream, SSortPhysiNode* pSortNode, SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
952 953 954 955
SOperatorInfo* createMultiwayMergeOperatorInfo(SOperatorInfo** dowStreams, size_t numStreams,
                                               SMergePhysiNode* pMergePhysiNode, SExecTaskInfo* pTaskInfo);
SOperatorInfo* createCacherowsScanOperator(SLastRowScanPhysiNode* pTableScanNode, SReadHandle* readHandle,
                                           SExecTaskInfo* pTaskInfo);
956

H
Haojun Liao 已提交
957
SOperatorInfo* createIntervalOperatorInfo(SOperatorInfo* downstream, SIntervalPhysiNode* pPhyNode,
L
Liu Jicong 已提交
958
                                          SExecTaskInfo* pTaskInfo, bool isStream);
959 960 961 962
SOperatorInfo* createMergeIntervalOperatorInfo(SOperatorInfo* downstream, SMergeIntervalPhysiNode* pIntervalPhyNode,
                                               SExecTaskInfo* pTaskInfo);
SOperatorInfo* createMergeAlignedIntervalOperatorInfo(SOperatorInfo* downstream, SMergeAlignedIntervalPhysiNode* pNode,
                                                      SExecTaskInfo* pTaskInfo);
963 964
SOperatorInfo* createStreamFinalIntervalOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pPhyNode,
                                                     SExecTaskInfo* pTaskInfo, int32_t numOfChild);
H
Haojun Liao 已提交
965 966
SOperatorInfo* createSessionAggOperatorInfo(SOperatorInfo* downstream, SSessionWinodwPhysiNode* pSessionNode,
                                            SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
967
SOperatorInfo* createGroupOperatorInfo(SOperatorInfo* downstream, SAggPhysiNode *pAggNode, SExecTaskInfo* pTaskInfo);
968 969
SOperatorInfo* createDataBlockInfoScanOperator(void* dataReader, SReadHandle* readHandle, uint64_t uid,
                                               SBlockDistScanPhysiNode* pBlockScanNode, SExecTaskInfo* pTaskInfo);
970

971
SOperatorInfo* createStreamScanOperatorInfo(SReadHandle* pHandle, STableScanPhysiNode* pTableScanNode, SNode* pTagCond,
972
                                            SExecTaskInfo* pTaskInfo);
973

974 975
SOperatorInfo* createRawScanOperatorInfo(SReadHandle* pHandle, SExecTaskInfo* pTaskInfo);

L
Liu Jicong 已提交
976 977 978 979 980
SOperatorInfo* createFillOperatorInfo(SOperatorInfo* downstream, SFillPhysiNode* pPhyFillNode,
                                      SExecTaskInfo* pTaskInfo);
SOperatorInfo* createStatewindowOperatorInfo(SOperatorInfo* downstream, SStateWinodwPhysiNode* pStateNode,
                                             SExecTaskInfo* pTaskInfo);
SOperatorInfo* createPartitionOperatorInfo(SOperatorInfo* downstream, SPartitionPhysiNode* pPartNode,
981
                                           SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
982

983
SOperatorInfo* createStreamPartitionOperatorInfo(SOperatorInfo* downstream, SStreamPartitionPhysiNode* pPartNode,
L
Liu Jicong 已提交
984
                                                 SExecTaskInfo* pTaskInfo);
985

986
SOperatorInfo* createTimeSliceOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pNode, SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
987 988
SOperatorInfo* createMergeJoinOperatorInfo(SOperatorInfo** pDownstream, int32_t numOfDownstream,
                                           SSortMergeJoinPhysiNode* pJoinNode, SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
989

L
Liu Jicong 已提交
990 991 992 993
SOperatorInfo* createStreamSessionAggOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pPhyNode,
                                                  SExecTaskInfo* pTaskInfo);
SOperatorInfo* createStreamFinalSessionAggOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pPhyNode,
                                                       SExecTaskInfo* pTaskInfo, int32_t numOfChild);
994 995
SOperatorInfo* createStreamIntervalOperatorInfo(SOperatorInfo* downstream,
                                                SPhysiNode* pPhyNode, SExecTaskInfo* pTaskInfo);
5
54liuyao 已提交
996

L
Liu Jicong 已提交
997 998
SOperatorInfo* createStreamStateAggOperatorInfo(SOperatorInfo* downstream, SPhysiNode* pPhyNode,
                                                SExecTaskInfo* pTaskInfo);
5
54liuyao 已提交
999 1000
SOperatorInfo* createStreamFillOperatorInfo(SOperatorInfo* downstream, SStreamFillPhysiNode* pPhyFillNode,
                                            SExecTaskInfo* pTaskInfo);
5
54liuyao 已提交
1001

1002
int32_t projectApplyFunctions(SExprInfo* pExpr, SSDataBlock* pResult, SSDataBlock* pSrcBlock, SqlFunctionCtx* pCtx,
L
Liu Jicong 已提交
1003
                              int32_t numOfOutput, SArray* pPseudoList);
1004

1005
void setInputDataBlock(SExprSupp* pExprSupp, SSDataBlock* pBlock, int32_t order, int32_t scanFlag, bool createDummyCol);
X
Xiaoyu Wang 已提交
1006

1007
bool    isTaskKilled(SExecTaskInfo* pTaskInfo);
1008 1009
int32_t checkForQueryBuf(size_t numOfTables);

L
Liu Jicong 已提交
1010 1011
void setTaskKilled(SExecTaskInfo* pTaskInfo);
void queryCostStatis(SExecTaskInfo* pTaskInfo);
1012

dengyihao's avatar
dengyihao 已提交
1013
void    doDestroyTask(SExecTaskInfo* pTaskInfo);
1014 1015
int32_t getMaximumIdleDurationSec();

wmmhello's avatar
wmmhello 已提交
1016 1017 1018 1019
/*
 * 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 已提交
1020
 * nOptrWithVal: *nOptrWithVal save the number of optr with value
wmmhello's avatar
wmmhello 已提交
1021 1022
 * return:  result code, 0 means success
 */
L
Liu Jicong 已提交
1023
int32_t encodeOperator(SOperatorInfo* ops, char** data, int32_t* length, int32_t* nOptrWithVal);
wmmhello's avatar
wmmhello 已提交
1024 1025 1026 1027 1028 1029 1030

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

1033
void    setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status);
dengyihao's avatar
dengyihao 已提交
1034
int32_t createExecTaskInfoImpl(SSubplan* pPlan, SExecTaskInfo** pTaskInfo, SReadHandle* pHandle, uint64_t taskId,
D
dapan1121 已提交
1035
                               char* sql, EOPTR_EXEC_MODEL model);
L
Liu Jicong 已提交
1036
int32_t createDataSinkParam(SDataSinkNode* pNode, void** pParam, qTaskInfo_t* pTaskInfo, SReadHandle* readHandle);
H
Haojun Liao 已提交
1037
int32_t getOperatorExplainExecInfo(SOperatorInfo* operatorInfo, SArray* pExecInfoList);
dengyihao's avatar
dengyihao 已提交
1038

1039
int32_t aggDecodeResultRow(SOperatorInfo* pOperator, char* result);
wmmhello's avatar
wmmhello 已提交
1040 1041
int32_t aggEncodeResultRow(SOperatorInfo* pOperator, char** result, int32_t* length);

1042
STimeWindow getActiveTimeWindow(SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int64_t ts, SInterval* pInterval,
1043
                                int32_t order);
1044
int32_t getNumOfRowsInTimeWindow(SDataBlockInfo* pDataBlockInfo, TSKEY* pPrimaryColumn, int32_t startPos, TSKEY ekey,
L
Liu Jicong 已提交
1045
                                 __block_search_fn_t searchFn, STableQueryInfo* item, int32_t order);
5
54liuyao 已提交
1046
int32_t binarySearchForKey(char* pValue, int num, TSKEY key, int order);
L
Liu Jicong 已提交
1047
SResultRow*        getNewResultRow(SDiskbasedBuf* pResultBuf, int32_t* currentPageId, int32_t interBufSize);
5
54liuyao 已提交
1048
void getCurSessionWindow(SStreamAggSupporter* pAggSup, TSKEY startTs, TSKEY endTs, uint64_t groupId, SSessionKey* pKey);
L
Liu Jicong 已提交
1049 1050 1051 1052 1053
bool               isInTimeWindow(STimeWindow* pWin, TSKEY ts, int64_t gap);
bool               functionNeedToExecute(SqlFunctionCtx* pCtx);
bool               isOverdue(TSKEY ts, STimeWindowAggSupp* pSup);
bool               isCloseWindow(STimeWindow* pWin, STimeWindowAggSupp* pSup);
bool               isDeletedWindow(STimeWindow* pWin, uint64_t groupId, SAggSupporter* pSup);
1054
bool               isDeletedStreamWindow(STimeWindow* pWin, uint64_t groupId, SStreamState* pState, STimeWindowAggSupp* pTwSup);
1055
void               appendOneRowToStreamSpecialBlock(SSDataBlock* pBlock, TSKEY* pStartTs, TSKEY* pEndTs, uint64_t* pUid, uint64_t* pGp, void* pTbName);
L
Liu Jicong 已提交
1056
void               printDataBlock(SSDataBlock* pBlock, const char* flag);
1057
uint64_t calGroupIdByData(SPartitionBySupporter* pParSup, SExprSupp* pExprSup, SSDataBlock* pBlock, int32_t rowId);
1058

1059 1060
int32_t finalizeResultRows(SDiskbasedBuf* pBuf, SResultRowPosition* resultRowPosition,
                                             SExprSupp* pSup, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo);
1061

L
Liu Jicong 已提交
1062 1063 1064
int32_t createScanTableListInfo(SScanPhysiNode* pScanNode, SNodeList* pGroupTags, bool groupSort, SReadHandle* pHandle,
                                STableListInfo* pTableListInfo, SNode* pTagCond, SNode* pTagIndexCond,
                                const char* idstr);
1065

S
shenglian zhou 已提交
1066 1067
SOperatorInfo* createGroupSortOperatorInfo(SOperatorInfo* downstream, SGroupSortPhysiNode* pSortPhyNode,
                                           SExecTaskInfo* pTaskInfo);
L
Liu Jicong 已提交
1068
SOperatorInfo* createTableMergeScanOperatorInfo(STableScanPhysiNode* pTableScanNode, STableListInfo* pTableListInfo,
1069
                                                SReadHandle* readHandle, SExecTaskInfo* pTaskInfo);
1070

1071
void copyUpdateDataBlock(SSDataBlock* pDest, SSDataBlock* pSource, int32_t tsColIndex);
1072

L
Liu Jicong 已提交
1073
bool    groupbyTbname(SNodeList* pGroupList);
wmmhello's avatar
wmmhello 已提交
1074
int32_t generateGroupIdMap(STableListInfo* pTableListInfo, SReadHandle* pHandle, SNodeList* groupKey);
L
Liu Jicong 已提交
1075
void*   destroySqlFunctionCtx(SqlFunctionCtx* pCtx, int32_t numOfOutput);
1076
int32_t buildDataBlockFromGroupRes(SOperatorInfo* pOperator, SStreamState* pState, SSDataBlock* pBlock, SExprSupp* pSup,
1077
                                   SGroupResInfo* pGroupResInfo);
5
54liuyao 已提交
1078
int32_t saveSessionDiscBuf(SStreamState* pState, SSessionKey* key, void* buf, int32_t size);
1079
int32_t buildSessionResultDataBlock(SOperatorInfo* pOperator, SStreamState* pState, SSDataBlock* pBlock,
5
54liuyao 已提交
1080
                                    SExprSupp* pSup, SGroupResInfo* pGroupResInfo);
1081 1082 1083 1084
int32_t setOutputBuf(SStreamState* pState, STimeWindow* win, SResultRow** pResult, int64_t tableGroupId, SqlFunctionCtx* pCtx,
                     int32_t numOfOutput, int32_t* rowEntryInfoOffset, SAggSupporter* pAggSup);
int32_t releaseOutputBuf(SStreamState* pState, SWinKey* pKey, SResultRow* pResult);
int32_t saveOutputBuf(SStreamState* pState, SWinKey* pKey, SResultRow* pResult, int32_t resSize);
5
54liuyao 已提交
1085
void    getNextIntervalWindow(SInterval* pInterval, STimeWindow* tw, int32_t order);
1086

1087 1088 1089 1090
#ifdef __cplusplus
}
#endif

1091
#endif  // TDENGINE_EXECUTORIMPL_H