texpr.h 6.0 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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 已提交
16 17
#ifndef TDENGINE_TEXPR_H
#define TDENGINE_TEXPR_H
H
hzcheng 已提交
18 19 20 21 22

#ifdef __cplusplus
extern "C" {
#endif

23
#include "os.h"
H
hzcheng 已提交
24

S
slguan 已提交
25
#include "taosmsg.h"
H
hzcheng 已提交
26
#include "taosdef.h"
27 28
#include "tskiplist.h"
#include "tbuffer.h"
29
#include "tvariant.h"
H
hzcheng 已提交
30

H
hjxilinx 已提交
31
struct tExprNode;
H
hzcheng 已提交
32 33
struct SSchema;

H
Haojun Liao 已提交
34 35
#define QUERY_COND_REL_PREFIX_IN "IN|"
#define QUERY_COND_REL_PREFIX_LIKE "LIKE|"
36
#define QUERY_COND_REL_PREFIX_MATCH "MATCH|"
37
#define QUERY_COND_REL_PREFIX_NMATCH "NMATCH|"
H
Haojun Liao 已提交
38 39 40

#define QUERY_COND_REL_PREFIX_IN_LEN   3
#define QUERY_COND_REL_PREFIX_LIKE_LEN 5
41
#define QUERY_COND_REL_PREFIX_MATCH_LEN 6
42
#define QUERY_COND_REL_PREFIX_NMATCH_LEN 7
H
Haojun Liao 已提交
43

44
#define TSDB_FUNC_FLAG_SCALAR       0x4000
45
#define TSDB_FUNC_IS_SCALAR(id)     ((((id) > 0)) && (((id) & TSDB_FUNC_FLAG_SCALAR) != 0))
46
#define TSDB_FUNC_SCALAR_INDEX(id)  ((id) & ~TSDB_FUNC_FLAG_SCALAR)
S
shenglian zhou 已提交
47

48 49 50 51
///////////////////////////////////////////
// SCALAR FUNCTIONS
#define TSDB_FUNC_SCALAR_POW          (TSDB_FUNC_FLAG_SCALAR | 0x0000)
#define TSDB_FUNC_SCALAR_LOG          (TSDB_FUNC_FLAG_SCALAR | 0x0001)
52 53 54 55 56 57 58 59
#define TSDB_FUNC_SCALAR_ABS          (TSDB_FUNC_FLAG_SCALAR | 0x0002)
#define TSDB_FUNC_SCALAR_ACOS         (TSDB_FUNC_FLAG_SCALAR | 0x0003)
#define TSDB_FUNC_SCALAR_ASIN         (TSDB_FUNC_FLAG_SCALAR | 0x0004)
#define TSDB_FUNC_SCALAR_ATAN         (TSDB_FUNC_FLAG_SCALAR | 0x0005)
#define TSDB_FUNC_SCALAR_COS          (TSDB_FUNC_FLAG_SCALAR | 0x0006)
#define TSDB_FUNC_SCALAR_SIN          (TSDB_FUNC_FLAG_SCALAR | 0x0007)
#define TSDB_FUNC_SCALAR_TAN          (TSDB_FUNC_FLAG_SCALAR | 0x0008)
#define TSDB_FUNC_SCALAR_SQRT         (TSDB_FUNC_FLAG_SCALAR | 0x0009)
60 61 62 63 64
#define TSDB_FUNC_SCALAR_CEIL         (TSDB_FUNC_FLAG_SCALAR | 0x000A)
#define TSDB_FUNC_SCALAR_FLOOR        (TSDB_FUNC_FLAG_SCALAR | 0x000B)
#define TSDB_FUNC_SCALAR_ROUND        (TSDB_FUNC_FLAG_SCALAR | 0x000C)
#define TSDB_FUNC_SCALAR_CONCAT       (TSDB_FUNC_FLAG_SCALAR | 0x000D)
#define TSDB_FUNC_SCALAR_LENGTH       (TSDB_FUNC_FLAG_SCALAR | 0x000E)
65
#define TSDB_FUNC_SCALAR_CONCAT_WS    (TSDB_FUNC_FLAG_SCALAR | 0x000F)
66
#define TSDB_FUNC_SCALAR_CHAR_LENGTH  (TSDB_FUNC_FLAG_SCALAR | 0x0010)
67
#define TSDB_FUNC_SCALAR_CAST         (TSDB_FUNC_FLAG_SCALAR | 0x0011)
68 69 70 71 72 73
#define TSDB_FUNC_SCALAR_LOWER        (TSDB_FUNC_FLAG_SCALAR | 0x0012)
#define TSDB_FUNC_SCALAR_UPPER        (TSDB_FUNC_FLAG_SCALAR | 0x0013)
#define TSDB_FUNC_SCALAR_LTRIM        (TSDB_FUNC_FLAG_SCALAR | 0x0014)
#define TSDB_FUNC_SCALAR_RTRIM        (TSDB_FUNC_FLAG_SCALAR | 0x0015)
#define TSDB_FUNC_SCALAR_SUBSTR       (TSDB_FUNC_FLAG_SCALAR | 0x0016)
#define TSDB_FUNC_SCALAR_MAX_NUM      23
74

75
#define TSDB_FUNC_SCALAR_NAME_MAX_LEN 16
S
shenglian zhou 已提交
76

77 78 79 80 81 82 83
typedef struct {
  int16_t type;
  int16_t bytes;
  int16_t numOfRows;
  char* data;
} tExprOperandInfo;

84
typedef void (*_expr_scalar_function_t)(int16_t functionId, tExprOperandInfo* pInputs, int32_t numInputs, tExprOperandInfo* pOutput, int32_t order);
85 86 87 88 89 90

_expr_scalar_function_t getExprScalarFunction(uint16_t scalar);

typedef struct tScalarFunctionInfo{
  int16_t functionId; // scalar function id & ~TSDB_FUNC_FLAG_SCALAR == index
  char    name[TSDB_FUNC_SCALAR_NAME_MAX_LEN];
S
shenglian zhou 已提交
91
  _expr_scalar_function_t scalarFunc;
92 93 94 95
} tScalarFunctionInfo;

/* global scalar sql functions array */
extern struct tScalarFunctionInfo aScalarFunctions[TSDB_FUNC_SCALAR_MAX_NUM];
S
shenglian zhou 已提交
96 97


H
Haojun Liao 已提交
98 99 100
typedef bool (*__result_filter_fn_t)(const void *, void *);
typedef void (*__do_filter_suppl_fn_t)(void *, void *);

H
hzcheng 已提交
101
enum {
H
Haojun Liao 已提交
102
  TSQL_NODE_DUMMY = 0x0,
H
hjxilinx 已提交
103 104
  TSQL_NODE_EXPR  = 0x1,
  TSQL_NODE_COL   = 0x2,
H
hzcheng 已提交
105
  TSQL_NODE_VALUE = 0x4,
D
cast  
dapan1121 已提交
106 107
  TSQL_NODE_FUNC  = 0x8,
  TSQL_NODE_TYPE  = 0x10
H
hzcheng 已提交
108 109
};

S
slguan 已提交
110 111 112 113 114
/**
 * this structure is used to filter data in tags, so the offset of filtered tag column in tagdata string is required
 */
typedef struct tQueryInfo {
  uint8_t       optr;     // expression operator
H
hjxilinx 已提交
115
  SSchema       sch;      // schema of tags
H
hjxilinx 已提交
116
  char*         q;
S
slguan 已提交
117
  __compar_fn_t compare;  // filter function
H
Haojun Liao 已提交
118
  bool          indexed;  // indexed columns
S
slguan 已提交
119 120
} tQueryInfo;

H
hjxilinx 已提交
121
typedef struct tExprNode {
H
hzcheng 已提交
122 123
  uint8_t nodeType;
  union {
H
[td-32]  
hjxilinx 已提交
124
    struct {
H
Haojun Liao 已提交
125 126 127
      uint8_t           optr;   // filter operator
      uint8_t           hasPK;  // 0: do not contain primary filter, 1: contain
      void             *info;   // support filter operation on this expression only available for leaf node
H
hjxilinx 已提交
128 129
      struct tExprNode *pLeft;  // left child pointer
      struct tExprNode *pRight; // right child pointer
H
[td-32]  
hjxilinx 已提交
130
    } _node;
H
Haojun Liao 已提交
131 132

    struct SSchema     *pSchema;
133

H
Haojun Liao 已提交
134
    tVariant           *pVal;
135 136 137

    struct {
      int16_t functionId;
138
      int32_t numChildren;
S
shenglian zhou 已提交
139
      struct tExprNode **pChildren;
140
    } _func;
D
cast  
dapan1121 已提交
141 142

    TAOS_FIELD        *pType;
H
hzcheng 已提交
143
  };
144 145
  int16_t resultType;
  int16_t resultBytes;
H
hjxilinx 已提交
146
} tExprNode;
H
hzcheng 已提交
147

H
Haojun Liao 已提交
148 149 150
typedef struct SExprTraverseSupp {
  __result_filter_fn_t   nodeFilterFn;
  __do_filter_suppl_fn_t setupInfoFn;
H
Haojun Liao 已提交
151
  void                  *pExtInfo;
H
Haojun Liao 已提交
152 153 154
} SExprTraverseSupp;

void tExprTreeDestroy(tExprNode *pNode, void (*fp)(void *));
H
hzcheng 已提交
155

D
cast  
dapan1121 已提交
156
int32_t exprTreeValidateTree(char* msgbuf, tExprNode *pExpr);
S
shenglian zhou 已提交
157

H
Haojun Liao 已提交
158
void exprTreeToBinary(SBufferWriter* bw, tExprNode* pExprTree);
H
Haojun Liao 已提交
159
tExprNode* exprTreeFromBinary(const void* data, size_t size);
H
Haojun Liao 已提交
160
tExprNode* exprdup(tExprNode* pTree);
H
hjxilinx 已提交
161

162
void exprTreeToBinary(SBufferWriter* bw, tExprNode* pExprTree);
163

H
Huo Linhe 已提交
164
bool exprTreeApplyFilter(tExprNode *pExpr, const void *pItem, SExprTraverseSupp *param);
H
Haojun Liao 已提交
165

166
void exprTreeNodeTraverse(tExprNode *pExpr, int32_t numOfRows, tExprOperandInfo *output, void *param, int32_t order,
167
                                  char *(*getSourceDataBlock)(void *, const char*, int32_t));
H
Haojun Liao 已提交
168

Y
yihaoDeng 已提交
169 170
void buildFilterSetFromBinary(void **q, const char *buf, int32_t len);

H
hzcheng 已提交
171 172 173 174
#ifdef __cplusplus
}
#endif

H
Haojun Liao 已提交
175
#endif  // TDENGINE_TEXPR_H