executorimpl.h 28.4 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * 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/>.
 */
#ifndef TDENGINE_EXECUTORIMPL_H
#define TDENGINE_EXECUTORIMPL_H

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

22
#include "os.h"
H
Haojun Liao 已提交
23
#include "tsort.h"
S
common  
Shengliang Guan 已提交
24
#include "tcommon.h"
25
#include "tlosertree.h"
26 27 28
#include "ttszip.h"
#include "tvariant.h"

H
Haojun Liao 已提交
29
#include "dataSinkMgt.h"
30
#include "executil.h"
H
Haojun Liao 已提交
31
#include "executor.h"
H
Haojun Liao 已提交
32
#include "planner.h"
D
dapan1121 已提交
33
#include "scalar.h"
34 35
#include "taosdef.h"
#include "tarray.h"
H
Haojun Liao 已提交
36
#include "thash.h"
37
#include "tlockfree.h"
H
Haojun Liao 已提交
38
#include "tpagedbuf.h"
D
dapan1121 已提交
39
#include "tmsg.h"
H
Haojun Liao 已提交
40

41 42 43 44 45
struct SColumnFilterElem;

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

#define IS_QUERY_KILLED(_q) ((_q)->code == TSDB_CODE_TSC_QUERY_CANCELLED)
46
#define Q_STATUS_EQUAL(p, s) (((p) & (s)) != 0u)
47 48
#define QUERY_IS_ASC_QUERY(q) (GET_FORWARD_DIRECTION_FACTOR((q)->order.order) == QUERY_ASC_FORWARD_STEP)

49
#define GET_TABLEGROUP(q, _index) ((SArray*)taosArrayGetP((q)->tableqinfoGroupInfo.pGroupList, (_index)))
50

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

enum {
54 55
  // when this task starts to execute, this status will set
  TASK_NOT_COMPLETED = 0x1u,
56

57
  /* Task is over
58 59 60
   * 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
   */
61
  TASK_COMPLETED = 0x2u,
62 63 64 65

  /* when the result is not completed return to client, this status will be
   * usually used in case of interval query with interpolation option
   */
66
  TASK_OVER = 0x4u,
67 68 69
};

typedef struct SResultRowCell {
70
  uint64_t    groupId;
H
Haojun Liao 已提交
71
  SResultRowPosition pos;
72 73 74 75 76 77
} SResultRowCell;

/**
 * If the number of generated results is greater than this value,
 * query query will be halt and return results to client immediate.
 */
H
Haojun Liao 已提交
78 79 80 81 82 83
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.
} SResultInfo;
84 85

typedef struct STableQueryInfo {
H
Haojun Liao 已提交
86 87
  TSKEY          lastKey;     // last check ts
  uint64_t       uid;         // table uid
88
  int32_t        groupIndex;  // group id in table list
89
//  SVariant       tag;
H
Haojun Liao 已提交
90
  SResultRowInfo resInfo;     // result info
91 92 93 94 95 96 97 98 99 100
} STableQueryInfo;

typedef enum {
  QUERY_PROF_BEFORE_OPERATOR_EXEC = 0,
  QUERY_PROF_AFTER_OPERATOR_EXEC,
  QUERY_PROF_QUERY_ABORT
} EQueryProfEventType;

typedef struct {
  EQueryProfEventType eventType;
101
  int64_t             eventTime;
102 103

  union {
104 105
    uint8_t operatorType;  // for operator event
    int32_t abortCode;     // for query abort event
106 107 108 109 110 111 112 113 114
  };
} SQueryProfEvent;

typedef struct {
  uint8_t operatorType;
  int64_t sumSelfTime;
  int64_t sumRunTimes;
} SOperatorProfResult;

H
Haojun Liao 已提交
115 116 117 118 119
typedef struct SLimit {
  int64_t limit;
  int64_t offset;
} SLimit;

H
Haojun Liao 已提交
120
typedef struct STaskCostInfo {
121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
  int64_t created;
  int64_t start;
  int64_t end;

  uint64_t loadStatisTime;
  uint64_t loadFileBlockTime;
  uint64_t loadDataInCacheTime;
  uint64_t loadStatisSize;
  uint64_t loadFileBlockSize;
  uint64_t loadDataInCacheSize;

  uint64_t loadDataTime;
  uint64_t totalRows;
  uint64_t totalCheckedRows;
  uint32_t totalBlocks;
  uint32_t loadBlocks;
  uint32_t loadBlockStatis;
  uint32_t discardBlocks;
  uint64_t elapsedTime;
  uint64_t firstStageMergeTime;
  uint64_t winInfoSize;
  uint64_t tableInfoSize;
  uint64_t hashSize;
  uint64_t numOfTimeWindows;

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

H
Haojun Liao 已提交
150 151
typedef struct SOperatorCostInfo {
  uint64_t openCost;
D
dapan1121 已提交
152
  uint64_t totalCost;
H
Haojun Liao 已提交
153 154
} SOperatorCostInfo;

155 156
// The basic query information extracted from the SQueryInfo tree to support the
// execution of query in a data node.
H
Haojun Liao 已提交
157
typedef struct STaskAttr {
158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181
  SLimit          limit;
  SLimit          slimit;
  bool            stableQuery;        // super table query or not
  bool            topBotQuery;        // TODO used bitwise flag
  bool            groupbyColumn;      // denote if this is a groupby normal column query
  bool            timeWindowInterpo;  // if the time window start/end required interpolation
  bool            tsCompQuery;        // is tscomp query
  bool            diffQuery;          // is diff query
  bool            pointInterpQuery;      // point interpolation query
  int32_t         havingNum;  // having expr number
  int16_t         numOfCols;
  int16_t         numOfTags;
  STimeWindow     window;
  SInterval       interval;
  int16_t         precision;
  int16_t         numOfOutput;
  int16_t         fillType;
  int32_t         resultRowSize;
  int32_t         tagLen;  // tag value length of current query

  SExprInfo      *pExpr1;
  SColumnInfo*    tagColList;
  int32_t         numOfFilterCols;
  int64_t*        fillVal;
182 183 184
  void*           tsdb;
  STableGroupInfo tableGroupInfo;  // table <tid, last_key> list  SArray<STableKeyInfo>
  int32_t         vgId;
H
Haojun Liao 已提交
185
} STaskAttr;
186 187

struct SOperatorInfo;
188 189
struct SAggSupporter;
struct SOptrBasicInfo;
190

191 192
typedef void (*__optr_encode_fn_t)(struct SOperatorInfo* pOperator, struct SAggSupporter *pSup, struct SOptrBasicInfo *pInfo, char **result, int32_t *length);
typedef bool (*__optr_decode_fn_t)(struct SOperatorInfo* pOperator, struct SAggSupporter *pSup, struct SOptrBasicInfo *pInfo, char *result, int32_t length);
193

194 195
typedef int32_t (*__optr_open_fn_t)(struct SOperatorInfo* pOptr);
typedef SSDataBlock* (*__optr_fn_t)(struct SOperatorInfo* pOptr, bool* newgroup);
H
Haojun Liao 已提交
196
typedef void (*__optr_close_fn_t)(void* param, int32_t num);
D
dapan1121 已提交
197
typedef int32_t (*__optr_get_explain_fn_t)(struct SOperatorInfo* pOptr, void **pOptrExplain);
H
Haojun Liao 已提交
198

H
Haojun Liao 已提交
199
typedef struct STaskIdInfo {
200 201 202 203
  uint64_t queryId;  // this is also a request id
  uint64_t subplanId;
  uint64_t templateId;
  char*    str;
H
Haojun Liao 已提交
204 205
} STaskIdInfo;

206
typedef struct SExecTaskInfo {
H
Haojun Liao 已提交
207 208 209 210
  STaskIdInfo     id;
  uint32_t        status;
  STimeWindow     window;
  STaskCostInfo   cost;
211
  int64_t         owner;  // if it is in execution
212
  int32_t         code;
213
  uint64_t        totalRows;            // total number of rows
H
Haojun Liao 已提交
214
  STableGroupInfo tableqinfoGroupInfo;  // this is a group array list, including SArray<STableQueryInfo*> structure
215
  char*           sql;                  // query sql string
216
  jmp_buf         env;                  // jump to this position when error happens.
217
  EOPTR_EXEC_MODEL execModel;           // operator execution model [batch model|stream model]
218
  struct SOperatorInfo* pRoot;
219
} SExecTaskInfo;
H
Haojun Liao 已提交
220

H
Haojun Liao 已提交
221
typedef struct STaskRuntimeEnv {
H
Haojun Liao 已提交
222 223 224 225 226 227 228 229 230 231 232 233 234 235
  jmp_buf         env;
  STaskAttr*      pQueryAttr;
  uint32_t        status;  // query status
  void*           qinfo;
  uint8_t         scanFlag;  // denotes reversed scan of data or not
  void*           pTsdbReadHandle;

  int32_t         prevGroupId;  // previous executed group id
  bool            enableGroupData;
  SDiskbasedBuf*  pResultBuf;           // query result buffer based on blocked-wised disk file
  SHashObj*       pResultRowHashTable;  // quick locate the window object for each result
  SHashObj*       pResultRowListSet;    // used to check if current ResultRowInfo has ResultRow object or not
  SArray*         pResultRowArrayList;  // The array list that contains the Result rows
  char*           keyBuf;               // window key buffer
H
Haojun Liao 已提交
236
  // The window result objects pool, all the resultRow Objects are allocated and managed by this object.
H
Haojun Liao 已提交
237 238 239 240
  char**          prevRow;
  SArray*         prevResult;  // intermediate result, SArray<SInterResult>
  STSBuf*         pTsBuf;      // timestamp filter list
  STSCursor       cur;
241

H
Haojun Liao 已提交
242
  char*           tagVal;  // tag value of current data block
243 244 245 246 247
  struct SScalarFunctionSupport* scalarSup;

  SSDataBlock*    outputBuf;
  STableGroupInfo tableqinfoGroupInfo;  // this is a group array list, including SArray<STableQueryInfo*> structure
  struct SOperatorInfo* proot;
H
Haojun Liao 已提交
248 249
  SGroupResInfo   groupResInfo;
  int64_t         currentOffset;  // dynamic offset value
250

251
  STableQueryInfo* current;
252
  SResultInfo      resultInfo;
253 254
  SHashObj*        pTableRetrieveTsMap;
  struct SUdfInfo* pUdfInfo;
H
Haojun Liao 已提交
255
} STaskRuntimeEnv;
256 257

enum {
H
Haojun Liao 已提交
258 259 260 261
  OP_NOT_OPENED    = 0x0,
  OP_OPENED        = 0x1,
  OP_RES_TO_RETURN = 0x5,
  OP_EXEC_DONE     = 0x9,
262 263 264
};

typedef struct SOperatorInfo {
D
dapan1121 已提交
265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285
  uint8_t                 operatorType;
  bool                    blockingOptr;  // block operator or not
  uint8_t                 status;        // denote if current operator is completed
  int32_t                 numOfOutput;   // number of columns of the current operator results
  char*                   name;          // name, used to show the query execution plan
  void*                   info;          // extension attribution
  SExprInfo*              pExpr;
  STaskRuntimeEnv*        pRuntimeEnv;   // todo remove it
  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
  __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.
  __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_get_explain_fn_t getExplainFn;
286 287 288 289 290
} SOperatorInfo;

typedef struct {
  int32_t      numOfTags;
  int32_t      numOfCols;
291
  SColumnInfo* colList;
292 293
} SQueriedTableInfo;

294 295 296 297 298
typedef enum {
  EX_SOURCE_DATA_NOT_READY = 0x1,
  EX_SOURCE_DATA_READY     = 0x2,
  EX_SOURCE_DATA_EXHAUSTED = 0x3,
} EX_SOURCE_STATUS;
299 300 301

typedef struct SSourceDataInfo {
  struct SExchangeInfo *pEx;
302 303 304
  int32_t               index;
  SRetrieveTableRsp    *pRsp;
  uint64_t              totalRows;
H
Haojun Liao 已提交
305
  int32_t               code;
306
  EX_SOURCE_STATUS      status;
307 308
} SSourceDataInfo;

H
Haojun Liao 已提交
309 310 311 312 313 314
typedef struct SLoadRemoteDataInfo {
  uint64_t           totalSize;     // total load bytes from remote
  uint64_t           totalRows;     // total number of rows
  uint64_t           totalElapsed;  // total elapsed time
} SLoadRemoteDataInfo;

315
typedef struct SExchangeInfo {
316
  SArray*            pSources;
317
  SArray*            pSourceDataInfo;
H
Haojun Liao 已提交
318
  tsem_t             ready;
319 320
  void*              pTransporter;
  SSDataBlock*       pResult;
H
Haojun Liao 已提交
321
  bool               seqLoadData;   // sequential load data or not, false by default
322
  int32_t            current;
H
Haojun Liao 已提交
323
  SLoadRemoteDataInfo loadInfo;
324 325
} SExchangeInfo;

H
Haojun Liao 已提交
326 327 328 329 330 331
typedef struct SColMatchInfo {
  int32_t colId;
  int32_t targetSlotId;
  bool    output;
} SColMatchInfo;

332
typedef struct STableScanInfo {
H
Haojun Liao 已提交
333
  void*           dataReader;
334 335 336 337 338 339 340 341
  int32_t         numOfBlocks;  // extract basic running information.
  int32_t         numOfSkipped;
  int32_t         numOfBlockStatis;
  int64_t         numOfRows;
  int32_t         order;  // scan order
  int32_t         times;  // repeat counts
  int32_t         current;
  int32_t         reverseTimes;  // 0 by default
H
Haojun Liao 已提交
342
  SNode*          pFilterNode;   // filter operator info
343 344 345 346
  SqlFunctionCtx* pCtx;  // next operator query context
  SResultRowInfo* pResultRowInfo;
  int32_t*        rowCellInfoOffset;
  SExprInfo*      pExpr;
347
  SSDataBlock*    pResBlock;
H
Haojun Liao 已提交
348
  SArray*         pColMatchInfo;
349 350
  int32_t         numOfOutput;
  int64_t         elapsedTime;
H
Haojun Liao 已提交
351
  int32_t         prevGroupId;  // previous table group id
352
  int32_t         scanFlag;  // table scan flag to denote if it is a repeat/reverse/main scan
353 354 355 356 357 358 359 360 361
} STableScanInfo;

typedef struct STagScanInfo {
  SColumnInfo* pCols;
  SSDataBlock* pRes;
  int32_t      totalTables;
  int32_t      curPos;
} STagScanInfo;

H
Haojun Liao 已提交
362
typedef struct SStreamBlockScanInfo {
H
Haojun Liao 已提交
363
  SArray*      pBlockLists;   // multiple SSDatablock.
364
  SSDataBlock* pRes;          // result SSDataBlock
365
  int32_t      blockType;     // current block type
H
Haojun Liao 已提交
366
  int32_t      validBlockIndex;    // Is current data has returned?
367 368 369 370
  SColumnInfo* pCols;         // the output column info
  uint64_t     numOfRows;     // total scanned rows
  uint64_t     numOfExec;     // execution times
  void*        readerHandle;  // stream block reader handle
371
  SArray*      pColMatchInfo; //
H
Haojun Liao 已提交
372 373
} SStreamBlockScanInfo;

374
typedef struct SSysTableScanInfo {
375 376 377 378 379
  union {
    void* pTransporter;
    void* readHandle;
  };

H
Haojun Liao 已提交
380
  SRetrieveMetaTableRsp *pRsp;
H
Haojun Liao 已提交
381 382 383
  SRetrieveTableReq   req;
  SEpSet              epSet;
  tsem_t              ready;
H
Haojun Liao 已提交
384

D
dapan1121 已提交
385 386
  int32_t             accountId;
  bool                showRewrite;
387
  SNode              *pCondition; // db_name filter condition, to discard data that are not in current database
388 389 390 391
  void               *pCur;       // cursor for iterate the local table meta store.
  SArray             *scanCols;   // SArray<int16_t> scan column id list

  int32_t             type;       // show type, TODO remove it
H
Haojun Liao 已提交
392
  SName               name;
393
  SSDataBlock        *pRes;
H
Haojun Liao 已提交
394 395
  int32_t             capacity;
  int64_t             numOfBlocks;  // extract basic running information.
H
Haojun Liao 已提交
396
  SLoadRemoteDataInfo loadInfo;
397 398
} SSysTableScanInfo;

399
typedef struct SOptrBasicInfo {
H
Haojun Liao 已提交
400 401 402 403
  SResultRowInfo     resultRowInfo;
  int32_t*           rowCellInfoOffset;  // offset value for each row result cell info
  SqlFunctionCtx*    pCtx;
  SSDataBlock*       pRes;
404
  int32_t            capacity;  // TODO remove it
405 406
} SOptrBasicInfo;

407
//TODO move the resultrowsiz together with SOptrBasicInfo:rowCellInfoOffset
408
typedef struct SAggSupporter {
H
Haojun Liao 已提交
409 410 411 412
  SHashObj*          pResultRowHashTable;  // quick locate the window object for each result
  SHashObj*          pResultRowListSet;    // used to check if current ResultRowInfo has ResultRow object or not
  SArray*            pResultRowArrayList;  // The array list that contains the Result rows
  char*              keyBuf;               // window key buffer
H
Haojun Liao 已提交
413
  SDiskbasedBuf     *pResultBuf;           // query result buffer based on blocked-wised disk file
H
Haojun Liao 已提交
414
  int32_t            resultRowSize;        // the result buffer size for each result row, with the meta data size for each row
415 416
} SAggSupporter;

H
Haojun Liao 已提交
417
typedef struct STableIntervalOperatorInfo {
418 419 420
  SOptrBasicInfo     binfo;                // basic info
  SGroupResInfo      groupResInfo;         // multiple results build supporter
  SInterval          interval;             // interval info
421
  int32_t            primaryTsIndex;       // primary time stamp slot id from result of downstream operator.
422 423 424 425 426 427
  STimeWindow        win;                  // query time range
  bool               timeWindowInterpo;    // interpolation needed or not
  char             **pRow;                 // previous row/tuple of already processed datablock
  SAggSupporter      aggSup;               // aggregate supporter
  STableQueryInfo   *pCurrent;             // current tableQueryInfo struct
  int32_t            order;                // current SSDataBlock scan order
428
  EOPTR_EXEC_MODEL   execModel;            // operator execution model [batch model|stream model]
429
  SArray            *pUpdatedWindow;       // updated time window due to the input data block from the downstream operator.
430
  SColumnInfoData    timeWindowData;       // query time window info for scalar function execution.
H
Haojun Liao 已提交
431
} STableIntervalOperatorInfo;
432 433

typedef struct SAggOperatorInfo {
H
Haojun Liao 已提交
434 435 436 437 438 439 440
  SOptrBasicInfo     binfo;
  SDiskbasedBuf     *pResultBuf;           // query result buffer based on blocked-wised disk file
  SAggSupporter      aggSup;
  STableQueryInfo   *current;
  uint32_t           groupId;
  SGroupResInfo      groupResInfo;
  STableQueryInfo   *pTableQueryInfo;
441 442 443 444
} SAggOperatorInfo;

typedef struct SProjectOperatorInfo {
  SOptrBasicInfo binfo;
H
Haojun Liao 已提交
445
  SAggSupporter  aggSup;
H
Haojun Liao 已提交
446
  SSDataBlock   *existDataBlock;
H
Haojun Liao 已提交
447
  SArray        *pPseudoColInfo;
H
Haojun Liao 已提交
448
  SLimit         limit;
H
Haojun Liao 已提交
449 450 451 452 453 454
  SLimit         slimit;

  uint64_t       groupId;
  int64_t        curSOffset;
  int64_t        curGroupOutput;

H
Haojun Liao 已提交
455 456
  int64_t        curOffset;
  int64_t        curOutput;
457 458 459
} SProjectOperatorInfo;

typedef struct SFillOperatorInfo {
460 461 462 463 464 465
  struct SFillInfo* pFillInfo;
  SSDataBlock*      pRes;
  int64_t           totalInputRows;
  void**            p;
  SSDataBlock*      existNewGroupBlock;
  bool              multigroupResult;
H
Haojun Liao 已提交
466 467
  SInterval         intervalInfo;
  int32_t           capacity;
468 469
} SFillOperatorInfo;

470
typedef struct {
471 472 473 474
  char             *pData;
  bool              isNull;
  int16_t           type;
  int32_t           bytes;
475
} SGroupKeys, SStateKeys;
476

477 478
typedef struct SGroupbyOperatorInfo {
  SOptrBasicInfo binfo;
H
Haojun Liao 已提交
479
  SArray*        pGroupCols;
480
  SArray*        pGroupColVals; // current group column values, SArray<SGroupKeys>
481
  SNode*         pCondition;
482 483 484
  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
H
Haojun Liao 已提交
485
  SGroupResInfo  groupResInfo;
H
Haojun Liao 已提交
486
  SAggSupporter  aggSup;
487 488 489
  SExprInfo*     pScalarExprInfo;
  int32_t        numOfScalarExpr;// the number of scalar expression in group operator
  SqlFunctionCtx*pScalarFuncCtx;
490 491
} SGroupbyOperatorInfo;

H
Haojun Liao 已提交
492 493 494 495 496 497 498 499
typedef struct SDataGroupInfo {
  uint64_t groupId;
  int64_t  numOfRows;
  SArray  *pPageList;
} SDataGroupInfo;

// The sort in partition may be needed later.
typedef struct SPartitionOperatorInfo {
500 501 502 503 504 505 506 507 508 509 510 511 512
  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

  void*            pGroupIter;    // group iterator
  int32_t          pageIndex;     // page index of current group
H
Haojun Liao 已提交
513
} SPartitionOperatorInfo;
514

515 516 517 518 519 520 521
typedef struct SWindowRowsSup {
  STimeWindow      win;
  TSKEY            prevTs;
  int32_t          startRowIndex;
  int32_t          numOfRows;
} SWindowRowsSup;

H
Haojun Liao 已提交
522
typedef struct SSessionAggOperatorInfo {
523 524 525
  SOptrBasicInfo   binfo;
  SAggSupporter    aggSup;
  SGroupResInfo    groupResInfo;
526
  SWindowRowsSup   winSup;
527 528 529
  bool             reptScan;         // next round scan
  int64_t          gap;              // session window gap
  SColumnInfoData  timeWindowData;   // query time window info for scalar function execution.
H
Haojun Liao 已提交
530
} SSessionAggOperatorInfo;
531

532 533 534 535 536 537
typedef struct STimeSliceOperatorInfo {
  SOptrBasicInfo   binfo;
  SInterval        interval;
  SGroupResInfo    groupResInfo;         // multiple results build supporter
} STimeSliceOperatorInfo;

538
typedef struct SStateWindowOperatorInfo {
539 540 541 542 543 544 545 546 547
  SOptrBasicInfo   binfo;
  SAggSupporter    aggSup;
  SGroupResInfo    groupResInfo;
  SWindowRowsSup   winSup;
  int32_t          colIndex;   // start row index
  bool             hasKey;
  SStateKeys       stateKey;
  SColumnInfoData  timeWindowData;   // query time window info for scalar function execution.
//  bool             reptScan;
548 549
} SStateWindowOperatorInfo;

550
typedef struct SSortedMergeOperatorInfo {
551 552
  SOptrBasicInfo   binfo;
  bool             hasVarCol;
553
  
554 555 556 557 558 559 560 561 562 563 564 565
  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;
  SAggSupporter    aggSup;
566
} SSortedMergeOperatorInfo;
H
Haojun Liao 已提交
567

H
Haojun Liao 已提交
568
typedef struct SSortOperatorInfo {
H
Haojun Liao 已提交
569 570
  uint32_t           sortBufSize;  // max buffer size for in-memory sort
  SSDataBlock       *pDataBlock;
H
Haojun Liao 已提交
571
  SArray*            pSortInfo;
H
Haojun Liao 已提交
572
  SSortHandle       *pSortHandle;
573
  SArray*            inputSlotMap;  // for index map from table scan output
H
Haojun Liao 已提交
574 575
  int32_t            bufPageSize;
  int32_t            numOfRowsInRes;
H
Haojun Liao 已提交
576

H
Haojun Liao 已提交
577
  // TODO extact struct
H
Haojun Liao 已提交
578 579 580 581 582
  int64_t            startTs;       // sort start time
  uint64_t           sortElapsed;   // sort elapsed time, time to flush to disk not included.
  uint64_t           totalSize;     // total load bytes from remote
  uint64_t           totalRows;     // total number of rows
  uint64_t           totalElapsed;  // total elapsed time
H
Haojun Liao 已提交
583
} SSortOperatorInfo;
584

H
Haojun Liao 已提交
585 586 587 588
int32_t operatorDummyOpenFn(SOperatorInfo* pOperator);
void operatorDummyCloseFn(void* param, int32_t numOfCols);
int32_t appendDownstream(SOperatorInfo* p, SOperatorInfo** pDownstream, int32_t num);
int32_t initAggInfo(SOptrBasicInfo* pBasicInfo, SAggSupporter* pAggSup, SExprInfo* pExprInfo, int32_t numOfCols,
589
                    int32_t numOfRows, SSDataBlock* pResultBlock, size_t keyBufSize, const char* pkey);
590
void toSDatablock(SSDataBlock* pBlock, int32_t rowCapacity, SGroupResInfo* pGroupResInfo, SExprInfo* pExprInfo, SDiskbasedBuf* pBuf, int32_t* rowCellOffset);
H
Haojun Liao 已提交
591 592
void finalizeMultiTupleQueryResult(SqlFunctionCtx* pCtx, int32_t numOfOutput, SDiskbasedBuf* pBuf, SResultRowInfo* pResultRowInfo, int32_t* rowCellInfoOffset);
void doApplyFunctions(SqlFunctionCtx* pCtx, STimeWindow* pWin, SColumnInfoData* pTimeWindowData, int32_t offset, int32_t forwardStep, TSKEY* tsCol, int32_t numOfTotal, int32_t numOfOutput, int32_t order);
593
int32_t setGroupResultOutputBuf(SOptrBasicInfo* binfo, int32_t numOfCols, char* pData, int16_t type, int16_t bytes, int32_t groupId, SDiskbasedBuf* pBuf, SExecTaskInfo* pTaskInfo, SAggSupporter* pAggSup);
H
Haojun Liao 已提交
594 595 596 597
void doDestroyBasicInfo(SOptrBasicInfo* pInfo, int32_t numOfOutput);
int32_t setSDataBlockFromFetchRsp(SSDataBlock* pRes, SLoadRemoteDataInfo* pLoadInfo, int32_t numOfRows,
                                         char* pData, int32_t compLen, int32_t numOfOutput, int64_t startTs,
                                         uint64_t* total, SArray* pColList);
598 599
void doSetOperatorCompleted(SOperatorInfo* pOperator);
void doFilter(const SNode* pFilterNode, SSDataBlock* pBlock);
600
SqlFunctionCtx* createSqlFunctionCtx(SExprInfo* pExprInfo, int32_t numOfOutput, int32_t** rowCellInfoOffset);
H
Haojun Liao 已提交
601

H
Haojun Liao 已提交
602
SOperatorInfo* createExchangeOperatorInfo(const SNodeList* pSources, SSDataBlock* pBlock, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
603
SOperatorInfo* createTableScanOperatorInfo(void* pTsdbReadHandle, int32_t order, int32_t numOfCols, int32_t repeatTime,
604
                                           int32_t reverseTime, SArray* pColMatchInfo, SSDataBlock* pResBlock, SNode* pCondition, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
605 606 607
SOperatorInfo* createAggregateOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResultBlock,
                                           SExecTaskInfo* pTaskInfo, const STableGroupInfo* pTableGroupInfo);
SOperatorInfo* createMultiTableAggOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResBlock, SExecTaskInfo* pTaskInfo, const STableGroupInfo* pTableGroupInfo);
608

wmmhello's avatar
wmmhello 已提交
609
SOperatorInfo* createProjectOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t num, SSDataBlock* pResBlock, SLimit* pLimit, SLimit* pSlimit, SExecTaskInfo* pTaskInfo);
610
SOperatorInfo *createSortOperatorInfo(SOperatorInfo* downstream, SSDataBlock* pResBlock, SArray* pSortInfo, SArray* pIndexMap, SExecTaskInfo* pTaskInfo);
611

H
Haojun Liao 已提交
612
SOperatorInfo* createSortedMergeOperatorInfo(SOperatorInfo** downstream, int32_t numOfDownstream, SExprInfo* pExprInfo, int32_t num, SArray* pSortInfo, SArray* pGroupInfo, SExecTaskInfo* pTaskInfo);
X
Xiaoyu Wang 已提交
613
SOperatorInfo* createSysTableScanOperatorInfo(void* pSysTableReadHandle, SSDataBlock* pResBlock, const SName* pName,
D
dapan1121 已提交
614
                                              SNode* pCondition, SEpSet epset, SArray* colList, SExecTaskInfo* pTaskInfo, bool showRewrite, int32_t accountId);
615
SOperatorInfo* createIntervalOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResBlock, SInterval* pInterval, int32_t primaryTsSlot,
H
Haojun Liao 已提交
616
                                          const STableGroupInfo* pTableGroupInfo, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
617
SOperatorInfo* createSessionAggOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResBlock, int64_t gap, SExecTaskInfo* pTaskInfo);
618 619
SOperatorInfo* createGroupOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResultBlock, SArray* pGroupColList,
                                       SNode* pCondition, SExprInfo* pScalarExprInfo, int32_t numOfScalarExpr, SExecTaskInfo* pTaskInfo, const STableGroupInfo* pTableGroupInfo);
H
Haojun Liao 已提交
620
SOperatorInfo* createDataBlockInfoScanOperator(void* dataReader, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
621
SOperatorInfo* createStreamScanOperatorInfo(void* streamReadHandle, SSDataBlock* pResBlock, SArray* pColList, SArray* pTableIdList, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
622

H
Haojun Liao 已提交
623 624
SOperatorInfo* createFillOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExpr, int32_t numOfCols, SInterval* pInterval, SSDataBlock* pResBlock,
                                      int32_t fillType, char* fillVal, bool multigroupResult, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
625
SOperatorInfo* createStatewindowOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExpr, int32_t numOfCols, SSDataBlock* pResBlock, SExecTaskInfo* pTaskInfo);
H
Haojun Liao 已提交
626

627 628
SOperatorInfo* createPartitionOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResultBlock, SArray* pGroupColList,
                                           SExecTaskInfo* pTaskInfo, const STableGroupInfo* pTableGroupInfo);
629 630
SOperatorInfo* createTimeSliceOperatorInfo(SOperatorInfo* downstream, SExprInfo* pExprInfo, int32_t numOfCols, SSDataBlock* pResultBlock, SExecTaskInfo* pTaskInfo);

H
Haojun Liao 已提交
631
#if 0
H
Haojun Liao 已提交
632
SOperatorInfo* createTableSeqScanOperatorInfo(void* pTsdbReadHandle, STaskRuntimeEnv* pRuntimeEnv);
633 634 635 636
SOperatorInfo* createMultiTableTimeIntervalOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream,
                                                        SExprInfo* pExpr, int32_t numOfOutput);
SOperatorInfo* createAllMultiTableTimeIntervalOperatorInfo(STaskRuntimeEnv* pRuntimeEnv, SOperatorInfo* downstream,
                                                           SExprInfo* pExpr, int32_t numOfOutput);
637
SOperatorInfo* createTagScanOperatorInfo(SReaderHandle* pReaderHandle, SExprInfo* pExpr, int32_t numOfOutput);
H
Haojun Liao 已提交
638

639 640
SOperatorInfo* createJoinOperatorInfo(SOperatorInfo** pdownstream, int32_t numOfDownstream, SSchema* pSchema,
                                      int32_t numOfOutput);
H
Haojun Liao 已提交
641
#endif
642

643 644
void projectApplyFunctions(SExprInfo* pExpr, SSDataBlock* pResult, SSDataBlock* pSrcBlock, SqlFunctionCtx* pCtx, int32_t numOfOutput, SArray* pPseudoList);

H
Haojun Liao 已提交
645
void setInputDataBlock(SOperatorInfo* pOperator, SqlFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order);
X
Xiaoyu Wang 已提交
646

647
void finalizeQueryResult(SqlFunctionCtx* pCtx, int32_t numOfOutput);
H
Haojun Liao 已提交
648
void copyTsColoum(SSDataBlock* pRes, SqlFunctionCtx* pCtx, int32_t numOfOutput);
649

H
Haojun Liao 已提交
650
STableQueryInfo* createTableQueryInfo(void* buf, bool groupbyColumn, STimeWindow win);
651

652
bool    isTaskKilled(SExecTaskInfo* pTaskInfo);
653 654
int32_t checkForQueryBuf(size_t numOfTables);

655
void   setTaskKilled(SExecTaskInfo* pTaskInfo);
656 657

void publishOperatorProfEvent(SOperatorInfo* operatorInfo, EQueryProfEventType eventType);
658
void publishQueryAbortEvent(SExecTaskInfo* pTaskInfo, int32_t code);
659

660
void queryCostStatis(SExecTaskInfo* pTaskInfo);
661

662
void doDestroyTask(SExecTaskInfo* pTaskInfo);
663 664
int32_t getMaximumIdleDurationSec();

665 666
void    doInvokeUdf(struct SUdfInfo* pUdfInfo, SqlFunctionCtx* pCtx, int32_t idx, int32_t type);
void    setTaskStatus(SExecTaskInfo* pTaskInfo, int8_t status);
667
int32_t createExecTaskInfoImpl(SSubplan* pPlan, SExecTaskInfo** pTaskInfo, SReadHandle* pHandle, uint64_t taskId, EOPTR_EXEC_MODEL model);
D
dapan1121 已提交
668
int32_t getOperatorExplainExecInfo(SOperatorInfo *operatorInfo, SExplainExecInfo **pRes, int32_t *capacity, int32_t *resNum);
669

670 671 672
bool aggDecodeResultRow(SOperatorInfo* pOperator, SAggSupporter *pSup, SOptrBasicInfo *pInfo, char* result, int32_t length);
void aggEncodeResultRow(SOperatorInfo* pOperator, SAggSupporter *pSup, SOptrBasicInfo *pInfo, char **result, int32_t *length);

673 674 675 676
#ifdef __cplusplus
}
#endif

677
#endif  // TDENGINE_EXECUTORIMPL_H