qExecutor.h 23.8 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
Haojun Liao 已提交
15 16
#ifndef TDENGINE_QEXECUTOR_H
#define TDENGINE_QEXECUTOR_H
17 18 19 20

#include "os.h"

#include "hash.h"
H
Haojun Liao 已提交
21
#include "qAggMain.h"
H
Haojun Liao 已提交
22 23
#include "qFill.h"
#include "qResultbuf.h"
24
#include "qSqlparser.h"
H
Haojun Liao 已提交
25
#include "qTableMeta.h"
H
Haojun Liao 已提交
26
#include "qTsbuf.h"
27
#include "query.h"
28
#include "taosdef.h"
H
Haojun Liao 已提交
29
#include "tarray.h"
B
Bomin Zhang 已提交
30
#include "tlockfree.h"
H
Haojun Liao 已提交
31
#include "tsdb.h"
H
Haojun Liao 已提交
32
#include "qUdf.h"
33 34

struct SColumnFilterElem;
35
typedef bool (*__filter_func_t)(struct SColumnFilterElem* pFilter, const char* val1, const char* val2, int16_t type);
H
hjxilinx 已提交
36
typedef int32_t (*__block_search_fn_t)(char* data, int32_t num, int64_t key, int32_t order);
37

H
Haojun Liao 已提交
38 39 40 41 42 43
#define IS_QUERY_KILLED(_q) ((_q)->code == TSDB_CODE_TSC_QUERY_CANCELLED)
#define Q_STATUS_EQUAL(p, s)  (((p) & (s)) != 0u)
#define QUERY_IS_ASC_QUERY(q) (GET_FORWARD_DIRECTION_FACTOR((q)->order.order) == QUERY_ASC_FORWARD_STEP)

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

H
Haojun Liao 已提交
44 45
#define GET_NUM_OF_RESULTS(_r) (((_r)->outputBuf) == NULL? 0:((_r)->outputBuf)->info.rows)

H
Haojun Liao 已提交
46 47 48 49 50 51 52 53
enum {
  // when query starts to execute, this status will set
      QUERY_NOT_COMPLETED = 0x1u,

  /* query is over
   * 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
   */
H
Haojun Liao 已提交
54
      QUERY_COMPLETED = 0x2u,
H
Haojun Liao 已提交
55 56 57 58

  /* when the result is not completed return to client, this status will be
   * usually used in case of interval query with interpolation option
   */
H
Haojun Liao 已提交
59
      QUERY_OVER = 0x4u,
H
Haojun Liao 已提交
60 61
};

H
Haojun Liao 已提交
62
typedef struct SResultRowPool {
63 64 65 66 67 68 69 70 71 72
  int32_t elemSize;
  int32_t blockSize;
  int32_t numOfElemPerBlock;

  struct {
    int32_t blockIndex;
    int32_t pos;
  } position;

  SArray* pData;    // SArray<void*>
H
Haojun Liao 已提交
73
} SResultRowPool;
74

H
Haojun Liao 已提交
75
typedef struct SResultRow {
76
  int32_t       pageId;      // pageId & rowId is the position of current result in disk-based output buffer
77
  int32_t       offset:29;   // row index in buffer page
78 79 80 81
  bool          startInterp; // the time window start timestamp has done the interpolation already.
  bool          endInterp;   // the time window end timestamp has done the interpolation already.
  bool          closed;      // this result status: closed or opened
  uint32_t      numOfRows;   // number of rows of current time window
H
Haojun Liao 已提交
82
  SResultRowCellInfo*  pCellInfo;  // For each result column, there is a resultInfo
83 84
  STimeWindow   win;
  char         *key;               // start key of current result row
H
Haojun Liao 已提交
85
} SResultRow;
86

87
typedef struct SGroupResInfo {
H
Haojun Liao 已提交
88 89
  int32_t totalGroup;
  int32_t currentGroup;
90 91 92 93
  int32_t index;
  SArray* pRows;      // SArray<SResultRow*>
} SGroupResInfo;

H
Haojun Liao 已提交
94 95 96 97
/**
 * 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 已提交
98
typedef struct SRspResultInfo {
H
Haojun Liao 已提交
99
  int64_t total;      // total generated result size in rows
H
Haojun Liao 已提交
100
  int32_t capacity;   // capacity of current result output buffer
H
Haojun Liao 已提交
101
  int32_t threshold;  // result size threshold in rows.
H
Haojun Liao 已提交
102
} SRspResultInfo;
103

H
Haojun Liao 已提交
104
typedef struct SResultRowInfo {
H
Haojun Liao 已提交
105 106 107 108
  SResultRow** pResult;    // result list
  int16_t      type:8;     // data type for hash key
  int32_t      size:24;    // number of result set
  int32_t      capacity;   // max capacity
109
  int32_t      curPos;     // current active result row index of pResult list
H
Haojun Liao 已提交
110
} SResultRowInfo;
111 112 113 114 115

typedef struct SColumnFilterElem {
  int16_t           bytes;  // column length
  __filter_func_t   fp;
  SColumnFilterInfo filterInfo;
Y
yihaoDeng 已提交
116
  void              *q;
117 118 119
} SColumnFilterElem;

typedef struct SSingleColumnFilterInfo {
H
Haojun Liao 已提交
120
  void*              pData;
W
wpan 已提交
121
  void*              pData2;  //used for nchar column
122
  int32_t            numOfFilters;
H
Haojun Liao 已提交
123
  SColumnInfo        info;
124 125 126
  SColumnFilterElem* pFilters;
} SSingleColumnFilterInfo;

H
Haojun Liao 已提交
127
typedef struct STableQueryInfo {
H
hjxilinx 已提交
128
  TSKEY       lastKey;
H
Haojun Liao 已提交
129
  int32_t     groupIndex;     // group id in table list
130
  tVariant    tag;
H
hjxilinx 已提交
131
  STimeWindow win;
132
  STSCursor   cur;
H
Haojun Liao 已提交
133
  void*       pTable;         // for retrieve the page id list
H
Haojun Liao 已提交
134
  SResultRowInfo resInfo;
135 136
} STableQueryInfo;

137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
typedef enum {
  QUERY_PROF_BEFORE_OPERATOR_EXEC = 0,
  QUERY_PROF_AFTER_OPERATOR_EXEC,
  QUERY_PROF_QUERY_ABORT
} EQueryProfEventType;

typedef struct {
  EQueryProfEventType eventType;
  int64_t eventTime;

  union {
    uint8_t operatorType; //for operator event
    int32_t abortCode; //for query abort event
  };
} SQueryProfEvent;

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

H
Haojun Liao 已提交
159 160 161 162 163 164 165 166 167
typedef struct SQueryCostInfo {
  uint64_t loadStatisTime;
  uint64_t loadFileBlockTime;
  uint64_t loadDataInCacheTime;
  uint64_t loadStatisSize;
  uint64_t loadFileBlockSize;
  uint64_t loadDataInCacheSize;
  
  uint64_t loadDataTime;
168 169 170 171
  uint64_t totalRows;
  uint64_t totalCheckedRows;
  uint32_t totalBlocks;
  uint32_t loadBlocks;
H
Haojun Liao 已提交
172 173
  uint32_t loadBlockStatis;
  uint32_t discardBlocks;
174
  uint64_t elapsedTime;
H
Haojun Liao 已提交
175
  uint64_t firstStageMergeTime;
H
Haojun Liao 已提交
176 177
  uint64_t winInfoSize;
  uint64_t tableInfoSize;
H
Haojun Liao 已提交
178
  uint64_t hashSize;
179
  uint64_t numOfTimeWindows;
180 181 182

  SArray*   queryProfEvents;  //SArray<SQueryProfEvent>
  SHashObj* operatorProfResults; //map<operator_type, SQueryProfEvent>
H
Haojun Liao 已提交
183
} SQueryCostInfo;
H
hjxilinx 已提交
184

H
Haojun Liao 已提交
185 186 187 188 189
typedef struct {
  int64_t vgroupLimit;
  int64_t ts;
} SOrderedPrjQueryInfo;

190 191 192 193 194
typedef struct {
  char*   tags;
  SArray* pResult;  // SArray<SStddevInterResult>
} SInterResult;

H
Haojun Liao 已提交
195 196 197 198 199 200
typedef struct SSDataBlock {
  SDataStatis *pBlockStatis;
  SArray      *pDataBlock;
  SDataBlockInfo info;
} SSDataBlock;

H
Haojun Liao 已提交
201
// The basic query information extracted from the SQueryInfo tree to support the
H
Haojun Liao 已提交
202
// execution of query in a data node.
H
Haojun Liao 已提交
203
typedef struct SQueryAttr {
H
Haojun Liao 已提交
204
  SLimitVal        limit;
H
Haojun Liao 已提交
205
  SLimitVal        slimit;
H
Haojun Liao 已提交
206

H
Haojun Liao 已提交
207 208 209 210 211 212 213
  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             hasTagResults;    // if there are tag values in final result or not
  bool             timeWindowInterpo;// if the time window start/end required interpolation
  bool             queryBlockDist;    // if query data block distribution
  bool             stabledev;        // super table stddev query
H
Haojun Liao 已提交
214
  bool             tsCompQuery;      // is tscomp query
215
  bool             diffQuery;        // is diff query
H
Haojun Liao 已提交
216 217 218
  bool             simpleAgg;
  bool             pointInterpQuery; // point interpolation query
  bool             needReverseScan;  // need reverse scan
219
  bool             distinctTag;      // distinct tag query
220
  bool             stateWindow;       // window State on sub/normal table
W
fix bug  
wpan 已提交
221
  bool             createFilterOperator; // if filter operator is needed
H
Haojun Liao 已提交
222
  int32_t          interBufSize;     // intermediate buffer sizse
H
Haojun Liao 已提交
223

D
dapan1121 已提交
224 225
  int32_t          havingNum;        // having expr number

H
Haojun Liao 已提交
226
  SOrderVal        order;
H
Haojun Liao 已提交
227 228
  int16_t          numOfCols;
  int16_t          numOfTags;
H
Haojun Liao 已提交
229

H
Haojun Liao 已提交
230
  STimeWindow      window;
231
  SInterval        interval;
232
  SSessionWindow   sw;
233
  int16_t          precision;
H
Haojun Liao 已提交
234 235
  int16_t          numOfOutput;
  int16_t          fillType;
236 237 238

  int32_t          srcRowSize;       // todo extract struct
  int32_t          resultRowSize;
H
Haojun Liao 已提交
239
  int32_t          intermediateResultRowSize; // intermediate result row size, in case of top-k query.
H
Haojun Liao 已提交
240
  int32_t          maxTableColumnWidth;
241
  int32_t          tagLen;           // tag value length of current query
H
Haojun Liao 已提交
242
  SGroupbyExpr    *pGroupbyExpr;
H
Haojun Liao 已提交
243

H
Haojun Liao 已提交
244
  SExprInfo*       pExpr1;
H
Haojun Liao 已提交
245 246
  SExprInfo*       pExpr2;
  int32_t          numOfExpr2;
H
Haojun Liao 已提交
247 248 249
  SExprInfo*       pExpr3;
  int32_t          numOfExpr3;

H
Haojun Liao 已提交
250
  SColumnInfo*     tableCols;
H
Haojun Liao 已提交
251 252 253
  SColumnInfo*     tagColList;
  int32_t          numOfFilterCols;
  int64_t*         fillVal;
254
  SOrderedPrjQueryInfo prjInfo;        // limit value for each vgroup, only available in global order projection query.
255
  SSingleColumnFilterInfo* pFilterInfo;
H
Haojun Liao 已提交
256 257 258

  void*            tsdb;
  SMemRef          memRef;
H
Haojun Liao 已提交
259
  STableGroupInfo  tableGroupInfo;       // table <tid, last_key> list  SArray<STableKeyInfo>
H
Haojun Liao 已提交
260
  int32_t          vgId;
261
  SArray          *pUdfInfo;             // no need to free
H
Haojun Liao 已提交
262
} SQueryAttr;
263

H
Haojun Liao 已提交
264
typedef SSDataBlock* (*__operator_fn_t)(void* param, bool* newgroup);
H
Haojun Liao 已提交
265
typedef void (*__optr_cleanup_fn_t)(void* param, int32_t num);
H
Haojun Liao 已提交
266

H
Haojun Liao 已提交
267
struct SOperatorInfo;
H
Haojun Liao 已提交
268

269
typedef struct SQueryRuntimeEnv {
H
Haojun Liao 已提交
270
  jmp_buf               env;
H
Haojun Liao 已提交
271
  SQueryAttr*           pQueryAttr;
H
Haojun Liao 已提交
272
  uint32_t              status;           // query status
H
Haojun Liao 已提交
273
  void*                 qinfo;
H
Haojun Liao 已提交
274
  uint8_t               scanFlag;         // denotes reversed scan of data or not
H
Haojun Liao 已提交
275 276 277
  void*                 pQueryHandle;

  int32_t               prevGroupId;      // previous executed group id
278
  bool                  enableGroupData;
H
Haojun Liao 已提交
279 280
  SDiskbasedResultBuf*  pResultBuf;       // query result buffer based on blocked-wised disk file
  SHashObj*             pResultRowHashTable; // quick locate the window object for each result
281
  SHashObj*             pResultRowListSet;   // used to check if current ResultRowInfo has ResultRow object or not
H
Haojun Liao 已提交
282 283 284 285 286 287 288 289 290 291 292 293 294
  char*                 keyBuf;           // window key buffer
  SResultRowPool*       pool;             // window result object pool
  char**                prevRow;

  SArray*               prevResult;       // intermediate result, SArray<SInterResult>
  STSBuf*               pTsBuf;           // timestamp filter list
  STSCursor             cur;

  char*                 tagVal;           // tag value of current data block
  SArithmeticSupport   *sasArray;

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

H
Haojun Liao 已提交
299
  STableQueryInfo      *current;
H
Haojun Liao 已提交
300
  SRspResultInfo        resultInfo;
H
Haojun Liao 已提交
301
  SHashObj             *pTableRetrieveTsMap;
D
dapan1121 已提交
302
  SUdfInfo             *pUdfInfo;
303 304
} SQueryRuntimeEnv;

H
Haojun Liao 已提交
305 306 307 308 309 310
enum {
  OP_IN_EXECUTING   = 1,
  OP_RES_TO_RETURN  = 2,
  OP_EXEC_DONE      = 3,
};

H
Haojun Liao 已提交
311 312 313 314 315
enum OPERATOR_TYPE_E {
  OP_TableScan         = 1,
  OP_DataBlocksOptScan = 2,
  OP_TableSeqScan      = 3,
  OP_TagScan           = 4,
H
Haojun Liao 已提交
316 317
  OP_TableBlockInfoScan= 5,
  OP_Aggregate         = 6,
318
  OP_Project           = 7,
H
Haojun Liao 已提交
319 320
  OP_Groupby           = 8,
  OP_Limit             = 9,
H
Haojun Liao 已提交
321
  OP_SLimit            = 10,
322 323 324 325 326
  OP_TimeWindow        = 11,
  OP_SessionWindow     = 12,
  OP_Fill              = 13,
  OP_MultiTableAggregate     = 14,
  OP_MultiTableTimeInterval  = 15,
H
Haojun Liao 已提交
327
  OP_DummyInput        = 16,   //TODO remove it after fully refactor.
H
Haojun Liao 已提交
328
  OP_MultiwayMergeSort = 17,   // multi-way data merge into one input stream.
H
Haojun Liao 已提交
329
  OP_GlobalAggregate   = 18,   // global merge for the multi-way data sources.
330 331
  OP_Filter            = 19,
  OP_Distinct          = 20,
H
Haojun Liao 已提交
332
  OP_Join              = 21,
Y
yihaoDeng 已提交
333
  OP_StateWindow       = 22,
H
Haojun Liao 已提交
334 335
};

H
Haojun Liao 已提交
336
typedef struct SOperatorInfo {
H
Haojun Liao 已提交
337 338 339 340 341 342 343 344
  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;
  SQueryRuntimeEnv     *pRuntimeEnv;
H
Haojun Liao 已提交
345

H
Haojun Liao 已提交
346 347
  struct SOperatorInfo **upstream;     // upstream pointer list
  int32_t               numOfUpstream; // number of upstream. The value is always ONE expect for join operator
H
Haojun Liao 已提交
348
  __operator_fn_t       exec;
H
Haojun Liao 已提交
349
  __optr_cleanup_fn_t   cleanup;
H
Haojun Liao 已提交
350 351
} SOperatorInfo;

352 353 354 355 356
enum {
  QUERY_RESULT_NOT_READY = 1,
  QUERY_RESULT_READY     = 2,
};

H
Haojun Liao 已提交
357 358 359 360 361 362
typedef struct {
  int32_t      numOfTags;
  int32_t      numOfCols;
  SColumnInfo *colList;
} SQueriedTableInfo;

363
typedef struct SQInfo {
H
Haojun Liao 已提交
364
  void*            signature;
D
dapan1121 已提交
365
  uint64_t         qId;
H
Haojun Liao 已提交
366 367
  int32_t          code;        // error code to returned to client
  int64_t          owner;       // if it is in execution
H
Haojun Liao 已提交
368

369
  SQueryRuntimeEnv runtimeEnv;
H
Haojun Liao 已提交
370
  SQueryAttr       query;
371
  void*            pBuf;        // allocated buffer for STableQueryInfo, sizeof(STableQueryInfo)*numOfTables;
H
Haojun Liao 已提交
372

373
  pthread_mutex_t  lock;        // used to synchronize the rsp/query threads
H
Haojun Liao 已提交
374
  tsem_t           ready;
375 376
  int32_t          dataReady;   // denote if query result is ready or not
  void*            rspContext;  // response context
377
  int64_t          startExecTs; // start to exec timestamp
378
  char*            sql;         // query sql string
H
Haojun Liao 已提交
379
  SQueryCostInfo   summary;
380 381
} SQInfo;

H
Haojun Liao 已提交
382 383 384 385 386 387
typedef struct SQueryParam {
  char            *sql;
  char            *tagCond;
  char            *tbnameCond;
  char            *prevResult;
  SArray          *pTableIdList;
H
Haojun Liao 已提交
388 389
  SSqlExpr       **pExpr;
  SSqlExpr       **pSecExpr;
H
Haojun Liao 已提交
390 391 392 393 394
  SExprInfo       *pExprs;
  SExprInfo       *pSecExprs;

  SColIndex       *pGroupColIndex;
  SColumnInfo     *pTagColumnInfo;
H
Haojun Liao 已提交
395
  SGroupbyExpr *pGroupbyExpr;
H
Haojun Liao 已提交
396 397
  int32_t          tableScanOperator;
  SArray          *pOperator;
H
Haojun Liao 已提交
398
  SUdfInfo        *pUdfInfo;
H
Haojun Liao 已提交
399 400
} SQueryParam;

H
Haojun Liao 已提交
401
typedef struct STableScanInfo {
H
Haojun Liao 已提交
402 403 404 405 406 407 408 409 410 411 412 413
  void           *pQueryHandle;
  int32_t         numOfBlocks;
  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

  SQLFunctionCtx *pCtx;         // next operator query context
H
Haojun Liao 已提交
414
  SResultRowInfo *pResultRowInfo;
H
Haojun Liao 已提交
415 416
  int32_t        *rowCellInfoOffset;
  SExprInfo      *pExpr;
H
Haojun Liao 已提交
417 418
  SSDataBlock     block;
  bool            loadExternalRows; // load external rows (prev & next rows)
H
Haojun Liao 已提交
419
  int32_t         numOfOutput;
H
Haojun Liao 已提交
420
  int64_t         elapsedTime;
H
Haojun Liao 已提交
421

H
Haojun Liao 已提交
422
  int32_t         tableIndex;
423
  int32_t         prevGroupId;     // previous table group id
H
Haojun Liao 已提交
424 425
} STableScanInfo;

426 427 428
typedef struct STagScanInfo {
  SColumnInfo* pCols;
  SSDataBlock* pRes;
H
Haojun Liao 已提交
429
  int32_t      totalTables;
430
  int32_t      curPos;
431 432
} STagScanInfo;

H
Haojun Liao 已提交
433
typedef struct SOptrBasicInfo {
H
Haojun Liao 已提交
434
  SResultRowInfo    resultRowInfo;
H
Haojun Liao 已提交
435
  int32_t          *rowCellInfoOffset;  // offset value for each row result cell info
H
Haojun Liao 已提交
436
  SQLFunctionCtx   *pCtx;
H
Haojun Liao 已提交
437
  SSDataBlock      *pRes;
H
Haojun Liao 已提交
438 439
} SOptrBasicInfo;

H
Haojun Liao 已提交
440 441 442 443 444 445
typedef struct SOptrBasicInfo STableIntervalOperatorInfo;

typedef struct SAggOperatorInfo {
  SOptrBasicInfo binfo;
  uint32_t       seed;
} SAggOperatorInfo;
H
Haojun Liao 已提交
446

447
typedef struct SProjectOperatorInfo {
H
Haojun Liao 已提交
448 449
  SOptrBasicInfo binfo;
  int32_t        bufCapacity;
H
Haojun Liao 已提交
450
  uint32_t       seed;
H
Haojun Liao 已提交
451 452

  SSDataBlock   *existDataBlock;
453
} SProjectOperatorInfo;
H
Haojun Liao 已提交
454

H
Haojun Liao 已提交
455
typedef struct SLimitOperatorInfo {
H
Haojun Liao 已提交
456 457
  int64_t   limit;
  int64_t   total;
H
Haojun Liao 已提交
458 459
} SLimitOperatorInfo;

H
Haojun Liao 已提交
460 461 462 463 464 465 466 467
typedef struct SSLimitOperatorInfo {
  int64_t   groupTotal;
  int64_t   currentGroupOffset;

  int64_t   rowsTotal;
  int64_t   currentOffset;
  SLimitVal limit;
  SLimitVal slimit;
H
Haojun Liao 已提交
468

H
Haojun Liao 已提交
469 470 471
  char    **prevRow;
  SArray   *orderColumnList;
} SSLimitOperatorInfo;
D
dapan1121 已提交
472

H
Haojun Liao 已提交
473
typedef struct SFilterOperatorInfo {
474 475
  SSingleColumnFilterInfo *pFilterInfo;
  int32_t numOfFilterCols;
H
Haojun Liao 已提交
476
} SFilterOperatorInfo;
D
dapan1121 已提交
477

H
Haojun Liao 已提交
478
typedef struct SFillOperatorInfo {
H
Haojun Liao 已提交
479
  SFillInfo   *pFillInfo;
H
Haojun Liao 已提交
480 481
  SSDataBlock *pRes;
  int64_t      totalInputRows;
H
Haojun Liao 已提交
482 483

  SSDataBlock *existNewGroupBlock;
H
Haojun Liao 已提交
484 485
} SFillOperatorInfo;

H
Haojun Liao 已提交
486 487 488
typedef struct SGroupbyOperatorInfo {
  SOptrBasicInfo binfo;
  int32_t        colIndex;
H
Haojun Liao 已提交
489
  char          *prevData;   // previous group by value
H
Haojun Liao 已提交
490
} SGroupbyOperatorInfo;
H
Haojun Liao 已提交
491

492 493 494 495 496 497
typedef struct SSWindowOperatorInfo {
  SOptrBasicInfo binfo;
  STimeWindow    curWindow;  // current time window
  TSKEY          prevTs;     // previous timestamp
  int32_t        numOfRows;  // number of rows
  int32_t        start;      // start row index
498
  bool           reptScan;    // next round scan
499 500
} SSWindowOperatorInfo;

Y
TD-2570  
yihaoDeng 已提交
501 502 503 504 505 506 507
typedef struct SStateWindowOperatorInfo {
  SOptrBasicInfo binfo;
  STimeWindow    curWindow;  // current time window
  int32_t        numOfRows;  // number of rows
  int32_t        colIndex;      // start row index
  int32_t        start;
  char*          prevData;    // previous data 
508
  bool           reptScan;
Y
TD-2570  
yihaoDeng 已提交
509 510
} SStateWindowOperatorInfo ;

511 512 513 514 515 516 517 518
typedef struct SDistinctOperatorInfo {
  SHashObj         *pSet;
  SSDataBlock      *pRes;
  bool              recordNullVal;  //has already record the null value, no need to try again
  int64_t           threshold;
  int64_t           outputCapacity;
} SDistinctOperatorInfo;

H
Haojun Liao 已提交
519
struct SGlobalMerger;
H
Haojun Liao 已提交
520 521

typedef struct SMultiwayMergeInfo {
H
Haojun Liao 已提交
522
  struct SGlobalMerger *pMerge;
H
Haojun Liao 已提交
523
  SOptrBasicInfo       binfo;
H
Haojun Liao 已提交
524
  int32_t              bufCapacity;
H
Haojun Liao 已提交
525 526 527
  int64_t              seed;
  char               **prevRow;
  SArray              *orderColumnList;
528
  int32_t              resultRowFactor;
H
Haojun Liao 已提交
529

H
Haojun Liao 已提交
530 531
  bool                 hasGroupColData;
  char               **currentGroupColData;
H
Haojun Liao 已提交
532 533 534 535 536 537
  SArray              *groupColumnList;
  bool                 hasDataBlockForNewGroup;
  SSDataBlock         *pExistBlock;

  bool                 hasPrev;
  bool                 groupMix;
H
Haojun Liao 已提交
538
  SArray              *udfInfo;
H
Haojun Liao 已提交
539 540
} SMultiwayMergeInfo;

H
Haojun Liao 已提交
541 542
void appendUpstream(SOperatorInfo* p, SOperatorInfo* pUpstream);

H
Haojun Liao 已提交
543 544 545 546 547
SOperatorInfo* createDataBlocksOptScanInfo(void* pTsdbQueryHandle, SQueryRuntimeEnv* pRuntimeEnv, int32_t repeatTime, int32_t reverseTime);
SOperatorInfo* createTableScanOperator(void* pTsdbQueryHandle, SQueryRuntimeEnv* pRuntimeEnv, int32_t repeatTime);
SOperatorInfo* createTableSeqScanOperator(void* pTsdbQueryHandle, SQueryRuntimeEnv* pRuntimeEnv);

SOperatorInfo* createAggregateOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
548
SOperatorInfo* createProjectOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
H
Haojun Liao 已提交
549 550 551 552 553 554 555 556
SOperatorInfo* createLimitOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream);
SOperatorInfo* createTimeIntervalOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
SOperatorInfo* createSWindowOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
SOperatorInfo* createFillOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
SOperatorInfo* createGroupbyOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
SOperatorInfo* createMultiTableAggOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
SOperatorInfo* createMultiTableTimeIntervalOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
SOperatorInfo* createTagScanOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SExprInfo* pExpr, int32_t numOfOutput);
557
SOperatorInfo* createDistinctOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
H
Haojun Liao 已提交
558
SOperatorInfo* createTableBlockInfoScanOperator(void* pTsdbQueryHandle, SQueryRuntimeEnv* pRuntimeEnv);
H
Haojun Liao 已提交
559
SOperatorInfo* createMultiwaySortOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SExprInfo* pExpr, int32_t numOfOutput,
H
Haojun Liao 已提交
560
                                              int32_t numOfRows, void* merger, bool groupMix);
561
SOperatorInfo* createGlobalAggregateOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput, void* param, SArray* pUdfInfo);
Y
TD-2570  
yihaoDeng 已提交
562
SOperatorInfo* createStatewindowOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput);
H
Haojun Liao 已提交
563
SOperatorInfo* createSLimitOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr, int32_t numOfOutput, void* merger);
H
Haojun Liao 已提交
564 565
SOperatorInfo* createFilterOperatorInfo(SQueryRuntimeEnv* pRuntimeEnv, SOperatorInfo* upstream, SExprInfo* pExpr,
                                        int32_t numOfOutput, SColumnInfo* pCols, int32_t numOfFilter);
H
Haojun Liao 已提交
566

567
SOperatorInfo* createJoinOperatorInfo(SOperatorInfo** pUpstream, int32_t numOfUpstream, SSchema* pSchema, int32_t numOfOutput);
H
Haojun Liao 已提交
568

H
Haojun Liao 已提交
569
SSDataBlock* doGlobalAggregate(void* param, bool* newgroup);
H
Haojun Liao 已提交
570
SSDataBlock* doMultiwayMergeSort(void* param, bool* newgroup);
H
Haojun Liao 已提交
571
SSDataBlock* doSLimit(void* param, bool* newgroup);
H
Haojun Liao 已提交
572

573 574 575 576 577
int32_t doCreateFilterInfo(SColumnInfo* pCols, int32_t numOfCols, int32_t numOfFilterCols, SSingleColumnFilterInfo** pFilterInfo, uint64_t qId);
void doSetFilterColumnInfo(SSingleColumnFilterInfo* pFilterInfo, int32_t numOfFilterCols, SSDataBlock* pBlock);
bool doFilterDataBlock(SSingleColumnFilterInfo* pFilterInfo, int32_t numOfFilterCols, int32_t numOfRows, int8_t* p);
void doCompactSDataBlock(SSDataBlock* pBlock, int32_t numOfRows, int8_t* p);

H
Haojun Liao 已提交
578
SSDataBlock* createOutputBuf(SExprInfo* pExpr, int32_t numOfOutput, int32_t numOfRows);
579
void* destroyOutputBuf(SSDataBlock* pBlock);
580
void* doDestroyFilterInfo(SSingleColumnFilterInfo* pFilterInfo, int32_t numOfFilterCols);
581

H
Haojun Liao 已提交
582 583 584
void setInputDataBlock(SOperatorInfo* pOperator, SQLFunctionCtx* pCtx, SSDataBlock* pBlock, int32_t order);
int32_t getNumOfResult(SQueryRuntimeEnv *pRuntimeEnv, SQLFunctionCtx* pCtx, int32_t numOfOutput);
void finalizeQueryResult(SOperatorInfo* pOperator, SQLFunctionCtx* pCtx, SResultRowInfo* pResultRowInfo, int32_t* rowCellInfoOffset);
H
Haojun Liao 已提交
585
void updateOutputBuf(SOptrBasicInfo* pBInfo, int32_t *bufCapacity, int32_t numOfInputRows);
586
void clearOutputBuf(SOptrBasicInfo* pBInfo, int32_t *bufCapacity);
H
Haojun Liao 已提交
587

H
Haojun Liao 已提交
588 589
void freeParam(SQueryParam *param);
int32_t convertQueryMsg(SQueryTableMsg *pQueryMsg, SQueryParam* param);
H
Haojun Liao 已提交
590
int32_t createQueryFunc(SQueriedTableInfo* pTableInfo, int32_t numOfOutput, SExprInfo** pExprInfo,
H
Haojun Liao 已提交
591
                        SSqlExpr** pExprMsg, SColumnInfo* pTagCols, int32_t queryType, void* pMsg, SUdfInfo* pUdfInfo);
H
Haojun Liao 已提交
592

H
Haojun Liao 已提交
593
int32_t createIndirectQueryFuncExprFromMsg(SQueryTableMsg *pQueryMsg, int32_t numOfOutput, SExprInfo **pExprInfo,
H
Haojun Liao 已提交
594
                                           SSqlExpr **pExpr, SExprInfo *prevExpr, SUdfInfo *pUdfInfo);
H
Haojun Liao 已提交
595

H
Haojun Liao 已提交
596 597
SGroupbyExpr *createGroupbyExprFromMsg(SQueryTableMsg *pQueryMsg, SColIndex *pColIndex, int32_t *code);
SQInfo *createQInfoImpl(SQueryTableMsg *pQueryMsg, SGroupbyExpr *pGroupbyExpr, SExprInfo *pExprs,
H
Haojun Liao 已提交
598
                        SExprInfo *pSecExprs, STableGroupInfo *pTableGroupInfo, SColumnInfo* pTagCols, int32_t vgId, char* sql, uint64_t qId, SUdfInfo* pUdfInfo);
H
Haojun Liao 已提交
599

H
Haojun Liao 已提交
600
int32_t initQInfo(STsBufInfo* pTsBufInfo, void* tsdb, void* sourceOptr, SQInfo* pQInfo, SQueryParam* param, char* start,
H
Haojun Liao 已提交
601
                  int32_t prevResultLen, void* merger);
H
Haojun Liao 已提交
602

H
Haojun Liao 已提交
603
int32_t createFilterInfo(SQueryAttr* pQueryAttr, uint64_t qId);
H
Haojun Liao 已提交
604 605
void freeColumnFilterInfo(SColumnFilterInfo* pFilter, int32_t numOfFilters);

H
Haojun Liao 已提交
606
STableQueryInfo *createTableQueryInfo(SQueryAttr* pQueryAttr, void* pTable, bool groupbyColumn, STimeWindow win, void* buf);
H
Haojun Liao 已提交
607 608
STableQueryInfo* createTmpTableQueryInfo(STimeWindow win);

609
int32_t buildArithmeticExprFromMsg(SExprInfo *pArithExprInfo, void *pQueryMsg);
H
Haojun Liao 已提交
610

H
Haojun Liao 已提交
611 612 613
bool isQueryKilled(SQInfo *pQInfo);
int32_t checkForQueryBuf(size_t numOfTables);
bool doBuildResCheck(SQInfo* pQInfo);
H
Haojun Liao 已提交
614
void setQueryStatus(SQueryRuntimeEnv *pRuntimeEnv, int8_t status);
H
Haojun Liao 已提交
615

H
Haojun Liao 已提交
616
bool onlyQueryTags(SQueryAttr* pQueryAttr);
D
dapan1121 已提交
617 618
void destroyUdfInfo(SUdfInfo* pUdfInfo);

H
Haojun Liao 已提交
619 620 621 622 623 624
bool isValidQInfo(void *param);

int32_t doDumpQueryResult(SQInfo *pQInfo, char *data);

size_t getResultSize(SQInfo *pQInfo, int64_t *numOfRows);
void setQueryKilled(SQInfo *pQInfo);
625 626 627 628

void publishOperatorProfEvent(SOperatorInfo* operatorInfo, EQueryProfEventType eventType);
void publishQueryAbortEvent(SQInfo* pQInfo, int32_t code);
void calculateOperatorProfResults(SQInfo* pQInfo);
H
Haojun Liao 已提交
629
void queryCostStatis(SQInfo *pQInfo);
630

H
Haojun Liao 已提交
631
void freeQInfo(SQInfo *pQInfo);
632
void freeQueryAttr(SQueryAttr *pQuery);
H
Haojun Liao 已提交
633 634 635

int32_t getMaximumIdleDurationSec();

636 637
void doInvokeUdf(SUdfInfo* pUdfInfo, SQLFunctionCtx *pCtx, int32_t idx, int32_t type);

H
Haojun Liao 已提交
638
#endif  // TDENGINE_QEXECUTOR_H