qExecutor.h 7.6 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
/*
 * 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_QUERYEXECUTOR_H
#define TDENGINE_QUERYEXECUTOR_H

#include "os.h"

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

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

H
Haojun Liao 已提交
36 37 38
typedef struct SGroupResInfo {
  int32_t  groupId;
  int32_t  numOfDataPages;
39 40
  int32_t  pageId;
  int32_t  rowId;
H
Haojun Liao 已提交
41 42
} SGroupResInfo;

43
typedef struct SSqlGroupbyExpr {
H
Haojun Liao 已提交
44 45 46 47 48
  int16_t tableIndex;
  SArray* columnInfo;  // SArray<SColIndex>, group by columns information
  int16_t numOfGroupCols;
  int16_t orderIndex;  // order by column index
  int16_t orderType;   // order by type: asc/desc
49 50 51
} SSqlGroupbyExpr;

typedef struct SWindowResult {
52 53 54
  int32_t       pageId;      // pageId & rowId is the position of current result in disk-based output buffer
  int32_t       rowId:15;
  bool          closed:1;    // this result status: closed or opened
55
  uint16_t      numOfRows;   // number of rows of current time window
56
  SResultInfo*  resultInfo;  // For each result column, there is a resultInfo
57
  union {STimeWindow win; char* key;};  // start key of current time window
58 59
} SWindowResult;

H
Haojun Liao 已提交
60 61 62 63
/**
 * If the number of generated results is greater than this value,
 * query query will be halt and return results to client immediate.
 */
64
typedef struct SResultRec {
H
Haojun Liao 已提交
65 66 67 68
  int64_t total;      // total generated result size in rows
  int64_t rows;       // current result set size in rows
  int64_t capacity;   // capacity of current result output buffer
  int32_t threshold;  // result size threshold in rows.
69 70 71 72
} SResultRec;

typedef struct SWindowResInfo {
  SWindowResult* pResult;    // result list
H
Haojun Liao 已提交
73
  SHashObj*      hashList;   // hash list for quick access
74 75 76 77 78 79
  int16_t        type;       // data type for hash key
  int32_t        capacity;   // max capacity
  int32_t        curIndex;   // current start active index
  int32_t        size;       // number of result set
  int64_t        startTime;  // start time of the first time window for sliding query
  int64_t        prevSKey;   // previous (not completed) sliding window start key
H
Haojun Liao 已提交
80
  int64_t        threshold;  // threshold to halt query and return the generated results.
81
  int64_t        interval;   // time window interval
82 83 84 85 86 87 88 89 90
} SWindowResInfo;

typedef struct SColumnFilterElem {
  int16_t           bytes;  // column length
  __filter_func_t   fp;
  SColumnFilterInfo filterInfo;
} SColumnFilterElem;

typedef struct SSingleColumnFilterInfo {
H
Haojun Liao 已提交
91
  void*              pData;
92
  int32_t            numOfFilters;
H
Haojun Liao 已提交
93
  SColumnInfo        info;
94 95 96
  SColumnFilterElem* pFilters;
} SSingleColumnFilterInfo;

H
Haojun Liao 已提交
97
typedef struct STableQueryInfo {
H
hjxilinx 已提交
98
  TSKEY       lastKey;
H
Haojun Liao 已提交
99
  int32_t     groupIndex;     // group id in table list
100
  int16_t     queryRangeSet;  // denote if the query range is set, only available for interval query
101
  tVariant    tag;
H
hjxilinx 已提交
102
  STimeWindow win;
103
  STSCursor   cur;
H
Haojun Liao 已提交
104
  void*       pTable;         // for retrieve the page id list
105 106 107
  SWindowResInfo windowResInfo;
} STableQueryInfo;

H
Haojun Liao 已提交
108 109 110 111 112 113 114 115 116
typedef struct SQueryCostInfo {
  uint64_t loadStatisTime;
  uint64_t loadFileBlockTime;
  uint64_t loadDataInCacheTime;
  uint64_t loadStatisSize;
  uint64_t loadFileBlockSize;
  uint64_t loadDataInCacheSize;
  
  uint64_t loadDataTime;
117 118 119 120
  uint64_t totalRows;
  uint64_t totalCheckedRows;
  uint32_t totalBlocks;
  uint32_t loadBlocks;
H
Haojun Liao 已提交
121 122
  uint32_t loadBlockStatis;
  uint32_t discardBlocks;
123
  uint64_t elapsedTime;
H
Haojun Liao 已提交
124
  uint64_t firstStageMergeTime;
H
Haojun Liao 已提交
125 126
  uint64_t winInfoSize;
  uint64_t tableInfoSize;
127
  uint64_t numOfTimeWindows;
H
Haojun Liao 已提交
128
} SQueryCostInfo;
H
hjxilinx 已提交
129

130
typedef struct SQuery {
H
Haojun Liao 已提交
131 132 133 134
  int16_t          numOfCols;
  int16_t          numOfTags;
  SOrderVal        order;
  STimeWindow      window;
135
  SInterval        interval;
136
  int16_t          precision;
H
Haojun Liao 已提交
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153
  int16_t          numOfOutput;
  int16_t          fillType;
  int16_t          checkBuffer;  // check if the buffer is full during scan each block
  SLimitVal        limit;
  int32_t          rowSize;
  SSqlGroupbyExpr* pGroupbyExpr;
  SExprInfo*       pSelectExpr;
  SColumnInfo*     colList;
  SColumnInfo*     tagColList;
  int32_t          numOfFilterCols;
  int64_t*         fillVal;
  uint32_t         status;  // query status
  SResultRec       rec;
  int32_t          pos;
  tFilePage**      sdata;
  STableQueryInfo* current;

154 155 156 157
  SSingleColumnFilterInfo* pFilterInfo;
} SQuery;

typedef struct SQueryRuntimeEnv {
H
Haojun Liao 已提交
158
  jmp_buf              env;
H
Haojun Liao 已提交
159
  SResultInfo*         resultInfo;       // todo refactor to merge with SWindowResInfo
H
Haojun Liao 已提交
160 161
  SQuery*              pQuery;
  SQLFunctionCtx*      pCtx;
162
  int32_t              numOfRowsPerPage;
H
Haojun Liao 已提交
163
  int16_t              offset[TSDB_MAX_COLUMNS];
H
Haojun Liao 已提交
164
  uint16_t             scanFlag;         // denotes reversed scan of data or not
H
Haojun Liao 已提交
165 166 167 168
  SFillInfo*           pFillInfo;
  SWindowResInfo       windowResInfo;
  STSBuf*              pTSBuf;
  STSCursor            cur;
169
  SQueryCostInfo       summary;
H
Haojun Liao 已提交
170 171
  void*                pQueryHandle;
  void*                pSecQueryHandle;  // another thread for
H
Haojun Liao 已提交
172
  bool                 stableQuery;      // super table query or not
H
Haojun Liao 已提交
173
  bool                 topBotQuery;      // false
H
Haojun Liao 已提交
174 175
  bool                 groupbyNormalCol; // denote if this is a groupby normal column query
  bool                 hasTagResults;    // if there are tag values in final result or not
H
Haojun Liao 已提交
176
  int32_t              interBufSize;     // intermediate buffer sizse
H
Haojun Liao 已提交
177
  int32_t              prevGroupId;      // previous executed group id
H
Haojun Liao 已提交
178
  SDiskbasedResultBuf* pResultBuf;       // query result buffer based on blocked-wised disk file
179 180
} SQueryRuntimeEnv;

181 182 183 184 185
enum {
  QUERY_RESULT_NOT_READY = 1,
  QUERY_RESULT_READY     = 2,
};

Y
TD-1733  
yihaoDeng 已提交
186 187 188 189 190 191
typedef struct SMemRef {
  int32_t ref; 
  void *mem;
  void *imem;
} SMemRef;

192
typedef struct SQInfo {
H
Haojun Liao 已提交
193
  void*            signature;
H
Haojun Liao 已提交
194
  int32_t          code;   // error code to returned to client
195
  int64_t          owner; // if it is in execution
H
Haojun Liao 已提交
196
  void*            tsdb;
Y
TD-1733  
yihaoDeng 已提交
197
  SMemRef          memRef; 
H
Haojun Liao 已提交
198
  int32_t          vgId;
H
Haojun Liao 已提交
199
  STableGroupInfo  tableGroupInfo;       // table <tid, last_key> list  SArray<STableKeyInfo>
200
  STableGroupInfo  tableqinfoGroupInfo;  // this is a group array list, including SArray<STableQueryInfo*> structure
201
  SQueryRuntimeEnv runtimeEnv;
weixin_48148422's avatar
weixin_48148422 已提交
202
  SArray*          arrTableIdInfo;
H
Haojun Liao 已提交
203
  int32_t          groupIndex;
H
Haojun Liao 已提交
204

205 206 207 208
  /*
   * the query is executed position on which meter of the whole list.
   * when the index reaches the last one of the list, it means the query is completed.
   */
H
Haojun Liao 已提交
209
  int32_t          tableIndex;
H
Haojun Liao 已提交
210
  SGroupResInfo    groupResInfo;
211
  void*            pBuf;        // allocated buffer for STableQueryInfo, sizeof(STableQueryInfo)*numOfTables;
H
Haojun Liao 已提交
212

213
  pthread_mutex_t  lock;        // used to synchronize the rsp/query threads
H
Haojun Liao 已提交
214
  tsem_t           ready;
215 216
  int32_t          dataReady;   // denote if query result is ready or not
  void*            rspContext;  // response context
217 218 219
} SQInfo;

#endif  // TDENGINE_QUERYEXECUTOR_H