texpr.h 5.4 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 65
#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)
#define TSDB_FUNC_SCALAR_MAX_NUM      15
66

67
#define TSDB_FUNC_SCALAR_NAME_MAX_LEN 16
S
shenglian zhou 已提交
68

69 70 71 72 73 74 75
typedef struct {
  int16_t type;
  int16_t bytes;
  int16_t numOfRows;
  char* data;
} tExprOperandInfo;

76
typedef void (*_expr_scalar_function_t)(int16_t functionId, tExprOperandInfo* pInputs, int32_t numInputs, tExprOperandInfo* pOutput, int32_t order);
77 78 79 80 81 82

_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 已提交
83
  _expr_scalar_function_t scalarFunc;
84 85 86 87
} tScalarFunctionInfo;

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


H
Haojun Liao 已提交
90 91 92
typedef bool (*__result_filter_fn_t)(const void *, void *);
typedef void (*__do_filter_suppl_fn_t)(void *, void *);

H
hzcheng 已提交
93
enum {
H
Haojun Liao 已提交
94
  TSQL_NODE_DUMMY = 0x0,
H
hjxilinx 已提交
95 96
  TSQL_NODE_EXPR  = 0x1,
  TSQL_NODE_COL   = 0x2,
H
hzcheng 已提交
97
  TSQL_NODE_VALUE = 0x4,
98
  TSQL_NODE_FUNC  = 0x8
H
hzcheng 已提交
99 100
};

S
slguan 已提交
101 102 103 104 105
/**
 * 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 已提交
106
  SSchema       sch;      // schema of tags
H
hjxilinx 已提交
107
  char*         q;
S
slguan 已提交
108
  __compar_fn_t compare;  // filter function
H
Haojun Liao 已提交
109
  bool          indexed;  // indexed columns
S
slguan 已提交
110 111
} tQueryInfo;

H
hjxilinx 已提交
112
typedef struct tExprNode {
H
hzcheng 已提交
113 114
  uint8_t nodeType;
  union {
H
[td-32]  
hjxilinx 已提交
115
    struct {
H
Haojun Liao 已提交
116 117 118
      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 已提交
119 120
      struct tExprNode *pLeft;  // left child pointer
      struct tExprNode *pRight; // right child pointer
H
[td-32]  
hjxilinx 已提交
121
    } _node;
H
Haojun Liao 已提交
122 123

    struct SSchema     *pSchema;
124

H
Haojun Liao 已提交
125
    tVariant           *pVal;
126 127 128

    struct {
      int16_t functionId;
129
      int32_t numChildren;
S
shenglian zhou 已提交
130
      struct tExprNode **pChildren;
131
    } _func;
H
hzcheng 已提交
132
  };
133 134
  int16_t resultType;
  int16_t resultBytes;
H
hjxilinx 已提交
135
} tExprNode;
H
hzcheng 已提交
136

H
Haojun Liao 已提交
137 138 139
typedef struct SExprTraverseSupp {
  __result_filter_fn_t   nodeFilterFn;
  __do_filter_suppl_fn_t setupInfoFn;
H
Haojun Liao 已提交
140
  void                  *pExtInfo;
H
Haojun Liao 已提交
141 142 143
} SExprTraverseSupp;

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

S
shenglian zhou 已提交
145 146
int32_t exprTreeValidateTree(tExprNode *pExpr);

H
Haojun Liao 已提交
147
void exprTreeToBinary(SBufferWriter* bw, tExprNode* pExprTree);
H
Haojun Liao 已提交
148
tExprNode* exprTreeFromBinary(const void* data, size_t size);
H
Haojun Liao 已提交
149
tExprNode* exprdup(tExprNode* pTree);
H
hjxilinx 已提交
150

151
void exprTreeToBinary(SBufferWriter* bw, tExprNode* pExprTree);
152

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

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

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

H
hzcheng 已提交
160 161 162 163
#ifdef __cplusplus
}
#endif

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