texpr.c 23.8 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/>.
 */

S
slguan 已提交
16
#include "os.h"
H
Haojun Liao 已提交
17 18

#include "exception.h"
19
#include "taosdef.h"
H
hzcheng 已提交
20
#include "taosmsg.h"
H
Haojun Liao 已提交
21 22 23
#include "tarray.h"
#include "tbuffer.h"
#include "tcompare.h"
24
#include "tname.h"
H
Haojun Liao 已提交
25
#include "thash.h"
H
Haojun Liao 已提交
26
#include "tskiplist.h"
H
Haojun Liao 已提交
27
#include "texpr.h"
28
//#include "tarithoperator.h"
H
Haojun Liao 已提交
29
#include "tvariant.h"
H
hzcheng 已提交
30

31 32 33 34 35 36 37 38 39 40 41 42
//static uint8_t UNUSED_FUNC isQueryOnPrimaryKey(const char *primaryColumnName, const tExprNode *pLeft, const tExprNode *pRight) {
//  if (pLeft->nodeType == TEXPR_COL_NODE) {
//    // if left node is the primary column,return true
//    return (strcmp(primaryColumnName, pLeft->pSchema->name) == 0) ? 1 : 0;
//  } else {
//    // if any children have query on primary key, their parents are also keep this value
//    return ((pLeft->nodeType == TEXPR_BINARYEXPR_NODE && pLeft->_node.hasPK == 1) ||
//            (pRight->nodeType == TEXPR_BINARYEXPR_NODE && pRight->_node.hasPK == 1)) == true
//               ? 1
//               : 0;
//  }
//}
H
hzcheng 已提交
43

44 45
static void reverseCopy(char* dest, const char* src, int16_t type, int32_t numOfRows) {
  switch(type) {
46 47
    case TSDB_DATA_TYPE_TINYINT:
    case TSDB_DATA_TYPE_UTINYINT:{
48 49 50 51 52 53
      int8_t* p = (int8_t*) dest;
      int8_t* pSrc = (int8_t*) src;

      for(int32_t i = 0; i < numOfRows; ++i) {
        p[i] = pSrc[numOfRows - i - 1];
      }
54
      return;
55
    }
56 57 58

    case TSDB_DATA_TYPE_SMALLINT:
    case TSDB_DATA_TYPE_USMALLINT:{
59 60 61 62 63 64
      int16_t* p = (int16_t*) dest;
      int16_t* pSrc = (int16_t*) src;

      for(int32_t i = 0; i < numOfRows; ++i) {
        p[i] = pSrc[numOfRows - i - 1];
      }
65
      return;
66
    }
67 68
    case TSDB_DATA_TYPE_INT:
    case TSDB_DATA_TYPE_UINT: {
69 70 71 72 73 74
      int32_t* p = (int32_t*) dest;
      int32_t* pSrc = (int32_t*) src;

      for(int32_t i = 0; i < numOfRows; ++i) {
        p[i] = pSrc[numOfRows - i - 1];
      }
75
      return;
76
    }
77 78
    case TSDB_DATA_TYPE_BIGINT:
    case TSDB_DATA_TYPE_UBIGINT: {
79 80 81 82 83 84
      int64_t* p = (int64_t*) dest;
      int64_t* pSrc = (int64_t*) src;

      for(int32_t i = 0; i < numOfRows; ++i) {
        p[i] = pSrc[numOfRows - i - 1];
      }
85
      return;
86 87 88 89 90 91 92 93
    }
    case TSDB_DATA_TYPE_FLOAT: {
      float* p = (float*) dest;
      float* pSrc = (float*) src;

      for(int32_t i = 0; i < numOfRows; ++i) {
        p[i] = pSrc[numOfRows - i - 1];
      }
94
      return;
95 96 97 98 99 100 101 102
    }
    case TSDB_DATA_TYPE_DOUBLE: {
      double* p = (double*) dest;
      double* pSrc = (double*) src;

      for(int32_t i = 0; i < numOfRows; ++i) {
        p[i] = pSrc[numOfRows - i - 1];
      }
103
      return;
104 105 106 107 108
    }
    default: assert(0);
  }
}

H
Haojun Liao 已提交
109 110 111
static void doExprTreeDestroy(tExprNode **pExpr, void (*fp)(void *));

void tExprTreeDestroy(tExprNode *pNode, void (*fp)(void *)) {
H
Haojun Liao 已提交
112 113 114 115
  if (pNode == NULL) {
    return;
  }

116
  if (pNode->nodeType == TEXPR_BINARYEXPR_NODE || pNode->nodeType == TEXPR_UNARYEXPR_NODE) {
H
Haojun Liao 已提交
117
    doExprTreeDestroy(&pNode, fp);
118
  } else if (pNode->nodeType == TEXPR_VALUE_NODE) {
H
Haojun Liao 已提交
119
    taosVariantDestroy(pNode->pVal);
120
  } else if (pNode->nodeType == TEXPR_COL_NODE) {
W
wpan 已提交
121
    tfree(pNode->pSchema);
H
Haojun Liao 已提交
122 123 124 125 126
  }

  free(pNode);
}

H
Haojun Liao 已提交
127
static void doExprTreeDestroy(tExprNode **pExpr, void (*fp)(void *)) {
H
Haojun Liao 已提交
128 129 130 131
  if (*pExpr == NULL) {
    return;
  }
  
132
  if ((*pExpr)->nodeType == TEXPR_BINARYEXPR_NODE) {
H
Haojun Liao 已提交
133 134
    doExprTreeDestroy(&(*pExpr)->_node.pLeft, fp);
    doExprTreeDestroy(&(*pExpr)->_node.pRight, fp);
H
Haojun Liao 已提交
135 136 137 138
  
    if (fp != NULL) {
      fp((*pExpr)->_node.info);
    }
139
  } else if ((*pExpr)->nodeType == TEXPR_VALUE_NODE) {
H
Haojun Liao 已提交
140
    taosVariantDestroy((*pExpr)->pVal);
H
Haojun Liao 已提交
141
    free((*pExpr)->pVal);
142
  } else if ((*pExpr)->nodeType == TEXPR_COL_NODE) {
H
Haojun Liao 已提交
143 144 145 146 147 148 149
    free((*pExpr)->pSchema);
  }

  free(*pExpr);
  *pExpr = NULL;
}

H
Haojun Liao 已提交
150
bool exprTreeApplyFilter(tExprNode *pExpr, const void *pItem, SExprTraverseSupp *param) {
H
Haojun Liao 已提交
151 152 153 154
  tExprNode *pLeft  = pExpr->_node.pLeft;
  tExprNode *pRight = pExpr->_node.pRight;

  //non-leaf nodes, recursively traverse the expression tree in the post-root order
155
  if (pLeft->nodeType == TEXPR_BINARYEXPR_NODE && pRight->nodeType == TEXPR_BINARYEXPR_NODE) {
H
Haojun Liao 已提交
156
    if (pExpr->_node.optr == TSDB_RELATION_OR) {  // or
H
Haojun Liao 已提交
157
      if (exprTreeApplyFilter(pLeft, pItem, param)) {
H
Haojun Liao 已提交
158 159 160 161
        return true;
      }

      // left child does not satisfy the query condition, try right child
H
Haojun Liao 已提交
162
      return exprTreeApplyFilter(pRight, pItem, param);
H
Haojun Liao 已提交
163
    } else {  // and
H
Haojun Liao 已提交
164
      if (!exprTreeApplyFilter(pLeft, pItem, param)) {
H
Haojun Liao 已提交
165 166 167
        return false;
      }

H
Haojun Liao 已提交
168
      return exprTreeApplyFilter(pRight, pItem, param);
H
Haojun Liao 已提交
169 170 171 172 173 174 175 176 177
    }
  }

  // handle the leaf node
  param->setupInfoFn(pExpr, param->pExtInfo);
  return param->nodeFilterFn(pItem, pExpr->_node.info);
}

void arithmeticTreeTraverse(tExprNode *pExprs, int32_t numOfRows, char *pOutput, void *param, int32_t order,
178
                                char *(*getSourceDataBlock)(void *, const char*, int32_t)) {
H
hzcheng 已提交
179 180 181
  if (pExprs == NULL) {
    return;
  }
182
#if 0
H
hjxilinx 已提交
183 184
  tExprNode *pLeft = pExprs->_node.pLeft;
  tExprNode *pRight = pExprs->_node.pRight;
H
hzcheng 已提交
185 186

  /* the left output has result from the left child syntax tree */
L
lihui 已提交
187
  char *pLeftOutput = (char*)malloc(sizeof(int64_t) * numOfRows);
188
  if (pLeft->nodeType == TEXPR_BINARYEXPR_NODE) {
H
Haojun Liao 已提交
189
    arithmeticTreeTraverse(pLeft, numOfRows, pLeftOutput, param, order, getSourceDataBlock);
H
hzcheng 已提交
190 191
  }

192
  // the right output has result from the right child syntax tree
H
hzcheng 已提交
193
  char *pRightOutput = malloc(sizeof(int64_t) * numOfRows);
194 195
  char *pdata = malloc(sizeof(int64_t) * numOfRows);

196
  if (pRight->nodeType == TEXPR_BINARYEXPR_NODE) {
H
Haojun Liao 已提交
197
    arithmeticTreeTraverse(pRight, numOfRows, pRightOutput, param, order, getSourceDataBlock);
H
hzcheng 已提交
198 199
  }

200 201
  if (pLeft->nodeType == TEXPR_BINARYEXPR_NODE) {
    if (pRight->nodeType == TEXPR_BINARYEXPR_NODE) {
S
slguan 已提交
202 203 204 205
      /*
       * exprLeft + exprRight
       * the type of returned value of one expression is always double float precious
       */
206 207
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
      OperatorFn(pLeftOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, pRightOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, pOutput, TSDB_ORDER_ASC);
H
hzcheng 已提交
208

209
    } else if (pRight->nodeType == TEXPR_COL_NODE) {  // exprLeft + columnRight
210
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
211

H
hzcheng 已提交
212
      // set input buffer
H
[td-32]  
hjxilinx 已提交
213
      char *pInputData = getSourceDataBlock(param, pRight->pSchema->name, pRight->pSchema->colId);
214 215
      if (order == TSDB_ORDER_DESC) {
        reverseCopy(pdata, pInputData, pRight->pSchema->type, numOfRows);
216
        OperatorFn(pLeftOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, pdata, numOfRows, pRight->pSchema->type, pOutput, TSDB_ORDER_ASC);
217
      } else {
218
        OperatorFn(pLeftOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, pInputData, numOfRows, pRight->pSchema->type, pOutput, TSDB_ORDER_ASC);
219
      }
H
hzcheng 已提交
220

221
    } else if (pRight->nodeType == TEXPR_VALUE_NODE) {  // exprLeft + 12
222 223
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
      OperatorFn(pLeftOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, &pRight->pVal->i64, 1, pRight->pVal->nType, pOutput, TSDB_ORDER_ASC);
H
hzcheng 已提交
224
    }
225
  } else if (pLeft->nodeType == TEXPR_COL_NODE) {
H
hzcheng 已提交
226
    // column data specified on left-hand-side
H
[td-32]  
hjxilinx 已提交
227
    char *pLeftInputData = getSourceDataBlock(param, pLeft->pSchema->name, pLeft->pSchema->colId);
228
    if (pRight->nodeType == TEXPR_BINARYEXPR_NODE) {  // columnLeft + expr2
229
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
230 231 232

      if (order == TSDB_ORDER_DESC) {
        reverseCopy(pdata, pLeftInputData, pLeft->pSchema->type, numOfRows);
233
        OperatorFn(pdata, numOfRows, pLeft->pSchema->type, pRightOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, pOutput, TSDB_ORDER_ASC);
234
      } else {
235
        OperatorFn(pLeftInputData, numOfRows, pLeft->pSchema->type, pRightOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, pOutput, TSDB_ORDER_ASC);
236
      }
H
hzcheng 已提交
237

238
    } else if (pRight->nodeType == TEXPR_COL_NODE) {  // columnLeft + columnRight
H
hzcheng 已提交
239
      // column data specified on right-hand-side
H
[td-32]  
hjxilinx 已提交
240
      char *pRightInputData = getSourceDataBlock(param, pRight->pSchema->name, pRight->pSchema->colId);
241
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
H
hzcheng 已提交
242

243
      // both columns are descending order, do not reverse the source data
244
      OperatorFn(pLeftInputData, numOfRows, pLeft->pSchema->type, pRightInputData, numOfRows, pRight->pSchema->type, pOutput, order);
245
    } else if (pRight->nodeType == TEXPR_VALUE_NODE) {  // columnLeft + 12
246
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
247 248 249

      if (order == TSDB_ORDER_DESC) {
        reverseCopy(pdata, pLeftInputData, pLeft->pSchema->type, numOfRows);
250
        OperatorFn(pdata, numOfRows, pLeft->pSchema->type, &pRight->pVal->i64, 1, pRight->pVal->nType, pOutput, TSDB_ORDER_ASC);
251
      } else {
252
        OperatorFn(pLeftInputData, numOfRows, pLeft->pSchema->type, &pRight->pVal->i64, 1, pRight->pVal->nType, pOutput, TSDB_ORDER_ASC);
253
      }
H
hzcheng 已提交
254 255 256
    }
  } else {
    // column data specified on left-hand-side
257
    if (pRight->nodeType == TEXPR_BINARYEXPR_NODE) {  // 12 + expr2
258 259
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
      OperatorFn(&pLeft->pVal->i64, 1, pLeft->pVal->nType, pRightOutput, numOfRows, TSDB_DATA_TYPE_DOUBLE, pOutput, TSDB_ORDER_ASC);
H
hzcheng 已提交
260

261
    } else if (pRight->nodeType == TEXPR_COL_NODE) {  // 12 + columnRight
H
hzcheng 已提交
262
      // column data specified on right-hand-side
263
      char *pRightInputData = getSourceDataBlock(param, pRight->pSchema->name, pRight->pSchema->colId);
264
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
265 266 267

      if (order == TSDB_ORDER_DESC) {
        reverseCopy(pdata, pRightInputData, pRight->pSchema->type, numOfRows);
268
        OperatorFn(&pLeft->pVal->i64, 1, pLeft->pVal->nType, pdata, numOfRows, pRight->pSchema->type, pOutput, TSDB_ORDER_ASC);
269
      } else {
270
        OperatorFn(&pLeft->pVal->i64, 1, pLeft->pVal->nType, pRightInputData, numOfRows, pRight->pSchema->type, pOutput, TSDB_ORDER_ASC);
271
      }
H
hzcheng 已提交
272

273
    } else if (pRight->nodeType == TEXPR_VALUE_NODE) {  // 12 + 12
274 275
      _arithmetic_operator_fn_t OperatorFn = getArithmeticOperatorFn(pExprs->_node.optr);
      OperatorFn(&pLeft->pVal->i64, 1, pLeft->pVal->nType, &pRight->pVal->i64, 1, pRight->pVal->nType, pOutput, TSDB_ORDER_ASC);
H
hzcheng 已提交
276 277 278
    }
  }

H
Haojun Liao 已提交
279 280 281
  tfree(pdata);
  tfree(pLeftOutput);
  tfree(pRightOutput);
282 283
#endif

H
hzcheng 已提交
284 285
}

286 287
static void exprTreeToBinaryImpl(SBufferWriter* bw, tExprNode* expr) {
  tbufWriteUint8(bw, expr->nodeType);
288
  
289
  if (expr->nodeType == TEXPR_VALUE_NODE) {
H
Haojun Liao 已提交
290
    SVariant* pVal = expr->pVal;
291
    
292
    tbufWriteUint32(bw, pVal->nType);
293
    if (pVal->nType == TSDB_DATA_TYPE_BINARY) {
294 295
      tbufWriteInt32(bw, pVal->nLen);
      tbufWrite(bw, pVal->pz, pVal->nLen);
296
    } else {
297
      tbufWriteInt64(bw, pVal->i64);
298 299
    }
    
300
  } else if (expr->nodeType == TEXPR_COL_NODE) {
301 302 303 304 305
    SSchema* pSchema = expr->pSchema;
    tbufWriteInt16(bw, pSchema->colId);
    tbufWriteInt16(bw, pSchema->bytes);
    tbufWriteUint8(bw, pSchema->type);
    tbufWriteString(bw, pSchema->name);
306
    
307
  } else if (expr->nodeType == TEXPR_BINARYEXPR_NODE) {
308 309 310
    tbufWriteUint8(bw, expr->_node.optr);
    exprTreeToBinaryImpl(bw, expr->_node.pLeft);
    exprTreeToBinaryImpl(bw, expr->_node.pRight);
311 312 313
  }
}

314 315 316
void exprTreeToBinary(SBufferWriter* bw, tExprNode* expr) {
  if (expr != NULL) {
    exprTreeToBinaryImpl(bw, expr);
317
  }
318 319 320 321 322 323
}

// TODO: these three functions should be made global
static void* exception_calloc(size_t nmemb, size_t size) {
  void* p = calloc(nmemb, size);
  if (p == NULL) {
324
    THROW(TSDB_CODE_QRY_OUT_OF_MEMORY);
325
  }
326 327 328 329 330 331
  return p;
}

static void* exception_malloc(size_t size) {
  void* p = malloc(size);
  if (p == NULL) {
332
    THROW(TSDB_CODE_QRY_OUT_OF_MEMORY);
333 334
  }
  return p;
335 336
}

337
static UNUSED_FUNC char* exception_strdup(const char* str) {
338 339
  char* p = strdup(str);
  if (p == NULL) {
340
    THROW(TSDB_CODE_QRY_OUT_OF_MEMORY);
341 342 343 344 345 346
  }
  return p;
}

static tExprNode* exprTreeFromBinaryImpl(SBufferReader* br) {
  int32_t anchor = CLEANUP_GET_ANCHOR();
dengyihao's avatar
TD-816  
dengyihao 已提交
347 348 349 350
  if (CLEANUP_EXCEED_LIMIT()) {
    THROW(TSDB_CODE_QRY_EXCEED_TAGS_LIMIT);
    return NULL;
  }
351 352

  tExprNode* pExpr = exception_calloc(1, sizeof(tExprNode));
H
Haojun Liao 已提交
353
  CLEANUP_PUSH_VOID_PTR_PTR(true, tExprTreeDestroy, pExpr, NULL);
354
  pExpr->nodeType = tbufReadUint8(br);
355
  
356
  if (pExpr->nodeType == TEXPR_VALUE_NODE) {
H
Haojun Liao 已提交
357
    SVariant* pVal = exception_calloc(1, sizeof(SVariant));
weixin_48148422's avatar
weixin_48148422 已提交
358
    pExpr->pVal = pVal;
359
  
360
    pVal->nType = tbufReadUint32(br);
361
    if (pVal->nType == TSDB_DATA_TYPE_BINARY) {
362
      tbufReadToBuffer(br, &pVal->nLen, sizeof(pVal->nLen));
363
      pVal->pz = calloc(1, pVal->nLen + 1);
364
      tbufReadToBuffer(br, pVal->pz, pVal->nLen);
365
    } else {
366
      pVal->i64 = tbufReadInt64(br);
367 368
    }
    
369
  } else if (pExpr->nodeType == TEXPR_COL_NODE) {
370
    SSchema* pSchema = exception_calloc(1, sizeof(SSchema));
weixin_48148422's avatar
weixin_48148422 已提交
371 372
    pExpr->pSchema = pSchema;

373 374 375 376
    pSchema->colId = tbufReadInt16(br);
    pSchema->bytes = tbufReadInt16(br);
    pSchema->type = tbufReadUint8(br);
    tbufReadToString(br, pSchema->name, TSDB_COL_NAME_LEN);
377
    
378
  } else if (pExpr->nodeType == TEXPR_BINARYEXPR_NODE) {
379 380 381
    pExpr->_node.optr = tbufReadUint8(br);
    pExpr->_node.pLeft = exprTreeFromBinaryImpl(br);
    pExpr->_node.pRight = exprTreeFromBinaryImpl(br);
382 383 384
    assert(pExpr->_node.pLeft != NULL && pExpr->_node.pRight != NULL);
  }
  
385
  CLEANUP_EXECUTE_TO(anchor, false);
weixin_48148422's avatar
weixin_48148422 已提交
386
  return pExpr;
387 388
}

389
tExprNode* exprTreeFromBinary(const void* data, size_t size) {
weixin_48148422's avatar
weixin_48148422 已提交
390 391 392
  if (size == 0) {
    return NULL;
  }
H
Haojun Liao 已提交
393

394 395
  SBufferReader br = tbufInitReader(data, size, false);
  return exprTreeFromBinaryImpl(&br);
396 397
}

weixin_48148422's avatar
weixin_48148422 已提交
398 399 400 401 402
tExprNode* exprTreeFromTableName(const char* tbnameCond) {
  if (!tbnameCond) {
    return NULL;
  }

403 404 405
  int32_t anchor = CLEANUP_GET_ANCHOR();

  tExprNode* expr = exception_calloc(1, sizeof(tExprNode));
H
Haojun Liao 已提交
406
  CLEANUP_PUSH_VOID_PTR_PTR(true, tExprTreeDestroy, expr, NULL);
407

408
  expr->nodeType = TEXPR_BINARYEXPR_NODE;
weixin_48148422's avatar
weixin_48148422 已提交
409

410
  tExprNode* left = exception_calloc(1, sizeof(tExprNode));
weixin_48148422's avatar
weixin_48148422 已提交
411 412
  expr->_node.pLeft = left;

413
  left->nodeType = TEXPR_COL_NODE;
414
  SSchema* pSchema = exception_calloc(1, sizeof(SSchema));
weixin_48148422's avatar
weixin_48148422 已提交
415 416
  left->pSchema = pSchema;

H
Haojun Liao 已提交
417
//  *pSchema = NULL;//*tGetTbnameColumnSchema();
weixin_48148422's avatar
weixin_48148422 已提交
418

419
  tExprNode* right = exception_calloc(1, sizeof(tExprNode));
weixin_48148422's avatar
weixin_48148422 已提交
420 421 422
  expr->_node.pRight = right;

  if (strncmp(tbnameCond, QUERY_COND_REL_PREFIX_LIKE, QUERY_COND_REL_PREFIX_LIKE_LEN) == 0) {
423
    right->nodeType = TEXPR_VALUE_NODE;
weixin_48148422's avatar
weixin_48148422 已提交
424
    expr->_node.optr = TSDB_RELATION_LIKE;
H
Haojun Liao 已提交
425
    SVariant* pVal = exception_calloc(1, sizeof(SVariant));
weixin_48148422's avatar
weixin_48148422 已提交
426 427
    right->pVal = pVal;
    size_t len = strlen(tbnameCond + QUERY_COND_REL_PREFIX_LIKE_LEN) + 1;
428
    pVal->pz = exception_malloc(len);
weixin_48148422's avatar
weixin_48148422 已提交
429
    memcpy(pVal->pz, tbnameCond + QUERY_COND_REL_PREFIX_LIKE_LEN, len);
430
    pVal->nType = TSDB_DATA_TYPE_BINARY;
431
    pVal->nLen = (int32_t)len;
weixin_48148422's avatar
weixin_48148422 已提交
432

433
  } else if (strncmp(tbnameCond, QUERY_COND_REL_PREFIX_MATCH, QUERY_COND_REL_PREFIX_MATCH_LEN) == 0) {
434
    right->nodeType = TEXPR_VALUE_NODE;
435
    expr->_node.optr = TSDB_RELATION_MATCH;
H
Haojun Liao 已提交
436
    SVariant* pVal = exception_calloc(1, sizeof(SVariant));
437 438 439 440 441 442
    right->pVal = pVal;
    size_t len = strlen(tbnameCond + QUERY_COND_REL_PREFIX_MATCH_LEN) + 1;
    pVal->pz = exception_malloc(len);
    memcpy(pVal->pz, tbnameCond + QUERY_COND_REL_PREFIX_MATCH_LEN, len);
    pVal->nType = TSDB_DATA_TYPE_BINARY;
    pVal->nLen = (int32_t)len;
443
  } else if (strncmp(tbnameCond, QUERY_COND_REL_PREFIX_NMATCH, QUERY_COND_REL_PREFIX_NMATCH_LEN) == 0) {
444
    right->nodeType = TEXPR_VALUE_NODE;
445
    expr->_node.optr = TSDB_RELATION_NMATCH;
H
Haojun Liao 已提交
446
    SVariant* pVal = exception_calloc(1, sizeof(SVariant));
447 448 449 450 451 452
    right->pVal = pVal;
    size_t len = strlen(tbnameCond + QUERY_COND_REL_PREFIX_NMATCH_LEN) + 1;
    pVal->pz = exception_malloc(len);
    memcpy(pVal->pz, tbnameCond + QUERY_COND_REL_PREFIX_NMATCH_LEN, len);
    pVal->nType = TSDB_DATA_TYPE_BINARY;
    pVal->nLen = (int32_t)len;
weixin_48148422's avatar
weixin_48148422 已提交
453
  } else if (strncmp(tbnameCond, QUERY_COND_REL_PREFIX_IN, QUERY_COND_REL_PREFIX_IN_LEN) == 0) {
454
    right->nodeType = TEXPR_VALUE_NODE;
weixin_48148422's avatar
weixin_48148422 已提交
455
    expr->_node.optr = TSDB_RELATION_IN;
H
Haojun Liao 已提交
456
    SVariant* pVal = exception_calloc(1, sizeof(SVariant));
weixin_48148422's avatar
weixin_48148422 已提交
457
    right->pVal = pVal;
W
wpan 已提交
458
    pVal->nType = TSDB_DATA_TYPE_POINTER_ARRAY;
459
    pVal->arr = taosArrayInit(2, POINTER_BYTES);
weixin_48148422's avatar
weixin_48148422 已提交
460 461 462

    const char* cond = tbnameCond + QUERY_COND_REL_PREFIX_IN_LEN;
    for (const char *e = cond; *e != 0; e++) {
463 464 465
      if (*e == TS_PATH_DELIMITER[0]) {
        cond = e + 1;
      } else if (*e == ',') {
466 467
        size_t len = e - cond;
        char* p = exception_malloc(len + VARSTR_HEADER_SIZE);
S
TD-1057  
Shengliang Guan 已提交
468
        STR_WITH_SIZE_TO_VARSTR(p, cond, (VarDataLenT)len);
weixin_48148422's avatar
weixin_48148422 已提交
469
        cond += len;
weixin_48148422's avatar
weixin_48148422 已提交
470
        taosArrayPush(pVal->arr, &p);
weixin_48148422's avatar
weixin_48148422 已提交
471 472 473 474
      }
    }

    if (*cond != 0) {
475 476 477
      size_t len = strlen(cond) + VARSTR_HEADER_SIZE;
      
      char* p = exception_malloc(len);
S
TD-1057  
Shengliang Guan 已提交
478
      STR_WITH_SIZE_TO_VARSTR(p, cond, (VarDataLenT)(len - VARSTR_HEADER_SIZE));
479
      taosArrayPush(pVal->arr, &p);
weixin_48148422's avatar
weixin_48148422 已提交
480 481
    }

482
    taosArraySortString(pVal->arr, taosArrayCompareString);
weixin_48148422's avatar
weixin_48148422 已提交
483 484
  }

485
  CLEANUP_EXECUTE_TO(anchor, false);
weixin_48148422's avatar
weixin_48148422 已提交
486
  return expr;
dengyihao's avatar
dengyihao 已提交
487
}
H
Haojun Liao 已提交
488

Y
yihaoDeng 已提交
489 490 491 492
void buildFilterSetFromBinary(void **q, const char *buf, int32_t len) {
  SBufferReader br = tbufInitReader(buf, len, false); 
  uint32_t type  = tbufReadUint32(&br);     
  SHashObj *pObj = taosHashInit(256, taosGetDefaultHashFunction(type), true, false);
493
  
H
Haojun Liao 已提交
494 495
//  taosHashSetEqualFp(pObj, taosGetDefaultEqualFunction(type));

Y
yihaoDeng 已提交
496 497 498
  int dummy = -1;
  int32_t sz = tbufReadInt32(&br);
  for (int32_t i = 0; i < sz; i++) {
499
    if (type == TSDB_DATA_TYPE_BOOL || IS_SIGNED_NUMERIC_TYPE(type)) {
500 501
      int64_t val = tbufReadInt64(&br); 
      taosHashPut(pObj, (char *)&val, sizeof(val),  &dummy, sizeof(dummy));
502 503 504 505 506
    } else if (IS_UNSIGNED_NUMERIC_TYPE(type)) {
      uint64_t val = tbufReadUint64(&br); 
      taosHashPut(pObj, (char *)&val, sizeof(val),  &dummy, sizeof(dummy));
    }
    else if (type == TSDB_DATA_TYPE_TIMESTAMP) {
Y
yihaoDeng 已提交
507 508 509 510 511
      int64_t val = tbufReadInt64(&br); 
      taosHashPut(pObj, (char *)&val, sizeof(val),  &dummy, sizeof(dummy));
    } else if (type == TSDB_DATA_TYPE_DOUBLE || type == TSDB_DATA_TYPE_FLOAT) {
      double  val = tbufReadDouble(&br);
      taosHashPut(pObj, (char *)&val, sizeof(val), &dummy, sizeof(dummy));
Y
yihaoDeng 已提交
512
    } else if (type == TSDB_DATA_TYPE_BINARY) {
Y
yihaoDeng 已提交
513 514 515
      size_t  t = 0;
      const char *val = tbufReadBinary(&br, &t);
      taosHashPut(pObj, (char *)val, t, &dummy, sizeof(dummy));
Y
yihaoDeng 已提交
516
    } else if (type == TSDB_DATA_TYPE_NCHAR) {
Y
yihaoDeng 已提交
517 518 519
      size_t  t = 0;
      const char *val = tbufReadBinary(&br, &t);      
      taosHashPut(pObj, (char *)val, t, &dummy, sizeof(dummy));
Y
yihaoDeng 已提交
520 521 522 523 524
    }
  } 
  *q = (void *)pObj;
}

W
wpan 已提交
525 526 527 528 529
void convertFilterSetFromBinary(void **q, const char *buf, int32_t len, uint32_t tType) {
  SBufferReader br = tbufInitReader(buf, len, false); 
  uint32_t sType  = tbufReadUint32(&br);     
  SHashObj *pObj = taosHashInit(256, taosGetDefaultHashFunction(tType), true, false);
  
H
Haojun Liao 已提交
530
//  taosHashSetEqualFp(pObj, taosGetDefaultEqualFunction(tType));
W
wpan 已提交
531 532
  
  int dummy = -1;
H
Haojun Liao 已提交
533
  SVariant tmpVar = {0};  
W
wpan 已提交
534 535
  size_t  t = 0;
  int32_t sz = tbufReadInt32(&br);
W
wpan 已提交
536 537
  void *pvar = NULL;  
  int64_t val = 0;
W
wpan 已提交
538 539 540 541 542 543 544 545 546 547 548 549
  int32_t bufLen = 0;
  if (IS_NUMERIC_TYPE(sType)) {
    bufLen = 60;  // The maximum length of string that a number is converted to.
  } else {
    bufLen = 128;
  }

  char *tmp = calloc(1, bufLen * TSDB_NCHAR_SIZE);
    
  for (int32_t i = 0; i < sz; i++) {
    switch (sType) {
    case TSDB_DATA_TYPE_BOOL:
W
wpan 已提交
550
    case TSDB_DATA_TYPE_UTINYINT:
W
wpan 已提交
551
    case TSDB_DATA_TYPE_TINYINT: {
W
wpan 已提交
552
      *(uint8_t *)&val = (uint8_t)tbufReadInt64(&br); 
W
wpan 已提交
553 554 555 556
      t = sizeof(val);
      pvar = &val;
      break;
    }
W
wpan 已提交
557
    case TSDB_DATA_TYPE_USMALLINT:
W
wpan 已提交
558
    case TSDB_DATA_TYPE_SMALLINT: {
W
wpan 已提交
559
      *(uint16_t *)&val = (uint16_t)tbufReadInt64(&br); 
W
wpan 已提交
560 561 562 563
      t = sizeof(val);
      pvar = &val;
      break;
    }
W
wpan 已提交
564
    case TSDB_DATA_TYPE_UINT:
W
wpan 已提交
565
    case TSDB_DATA_TYPE_INT: {
W
wpan 已提交
566
      *(uint32_t *)&val = (uint32_t)tbufReadInt64(&br); 
W
wpan 已提交
567 568 569 570
      t = sizeof(val);
      pvar = &val;
      break;
    }
W
wpan 已提交
571
    case TSDB_DATA_TYPE_TIMESTAMP:
W
wpan 已提交
572
    case TSDB_DATA_TYPE_UBIGINT:
W
wpan 已提交
573
    case TSDB_DATA_TYPE_BIGINT: {
W
wpan 已提交
574
      *(uint64_t *)&val = (uint64_t)tbufReadInt64(&br); 
W
wpan 已提交
575 576 577 578 579
      t = sizeof(val);
      pvar = &val;
      break;
    }
    case TSDB_DATA_TYPE_DOUBLE: {
W
wpan 已提交
580
      *(double *)&val = tbufReadDouble(&br);
W
wpan 已提交
581 582 583 584 585
      t = sizeof(val);
      pvar = &val;
      break;
    }
    case TSDB_DATA_TYPE_FLOAT: {
W
wpan 已提交
586
      *(float *)&val = (float)tbufReadDouble(&br);
W
wpan 已提交
587 588 589 590 591
      t = sizeof(val);
      pvar = &val;
      break;
    }
    case TSDB_DATA_TYPE_BINARY: {
W
wpan 已提交
592
      pvar = (char *)tbufReadBinary(&br, &t);
W
wpan 已提交
593 594 595
      break;
    }
    case TSDB_DATA_TYPE_NCHAR: {
W
wpan 已提交
596
      pvar = (char *)tbufReadBinary(&br, &t);      
W
wpan 已提交
597 598 599 600 601 602 603 604
      break;
    }
    default:
      taosHashCleanup(pObj);
      *q = NULL;
      return;
    }
    
H
Haojun Liao 已提交
605
    taosVariantCreateFromBinary(&tmpVar, (char *)pvar, t, sType);
W
wpan 已提交
606 607 608

    if (bufLen < t) {
      tmp = realloc(tmp, t * TSDB_NCHAR_SIZE);
W
wpan 已提交
609
      bufLen = (int32_t)t;
W
wpan 已提交
610 611 612 613
    }

    switch (tType) {
      case TSDB_DATA_TYPE_BOOL:
W
wpan 已提交
614
      case TSDB_DATA_TYPE_UTINYINT:
W
wpan 已提交
615
      case TSDB_DATA_TYPE_TINYINT: {
H
Haojun Liao 已提交
616
        if (taosVariantDump(&tmpVar, (char *)&val, tType, false)) {
W
wpan 已提交
617 618 619 620 621 622
          goto err_ret;
        }
        pvar = &val;
        t = sizeof(val);
        break;
      }
W
wpan 已提交
623
      case TSDB_DATA_TYPE_USMALLINT:
W
wpan 已提交
624
      case TSDB_DATA_TYPE_SMALLINT: {
H
Haojun Liao 已提交
625
        if (taosVariantDump(&tmpVar, (char *)&val, tType, false)) {
W
wpan 已提交
626 627 628 629 630 631
          goto err_ret;
        }
        pvar = &val;
        t = sizeof(val);
        break;
      }
W
wpan 已提交
632
      case TSDB_DATA_TYPE_UINT:
W
wpan 已提交
633
      case TSDB_DATA_TYPE_INT: {
H
Haojun Liao 已提交
634
        if (taosVariantDump(&tmpVar, (char *)&val, tType, false)) {
W
wpan 已提交
635 636 637 638 639 640
          goto err_ret;
        }
        pvar = &val;
        t = sizeof(val);
        break;
      }
W
wpan 已提交
641
      case TSDB_DATA_TYPE_TIMESTAMP:
W
wpan 已提交
642
      case TSDB_DATA_TYPE_UBIGINT:
W
wpan 已提交
643
      case TSDB_DATA_TYPE_BIGINT: {
H
Haojun Liao 已提交
644
        if (taosVariantDump(&tmpVar, (char *)&val, tType, false)) {
W
wpan 已提交
645 646 647 648 649 650 651
          goto err_ret;
        }
        pvar = &val;
        t = sizeof(val);
        break;
      }
      case TSDB_DATA_TYPE_DOUBLE: {
H
Haojun Liao 已提交
652
        if (taosVariantDump(&tmpVar, (char *)&val, tType, false)) {
W
wpan 已提交
653 654 655 656 657 658 659
          goto err_ret;
        }
        pvar = &val;
        t = sizeof(val);
        break;
      }
      case TSDB_DATA_TYPE_FLOAT: {
H
Haojun Liao 已提交
660
        if (taosVariantDump(&tmpVar, (char *)&val, tType, false)) {
W
wpan 已提交
661 662 663 664 665 666 667
          goto err_ret;
        }
        pvar = &val;
        t = sizeof(val);
        break;
      }
      case TSDB_DATA_TYPE_BINARY: {
H
Haojun Liao 已提交
668
        if (taosVariantDump(&tmpVar, tmp, tType, true)) {
W
wpan 已提交
669 670 671 672 673 674 675
          goto err_ret;
        }
        t = varDataLen(tmp);
        pvar = varDataVal(tmp);
        break;
      }
      case TSDB_DATA_TYPE_NCHAR: {
H
Haojun Liao 已提交
676
        if (taosVariantDump(&tmpVar, tmp, tType, true)) {
W
wpan 已提交
677 678 679 680 681 682 683 684 685 686 687
          goto err_ret;
        }
        t = varDataLen(tmp);
        pvar = varDataVal(tmp);        
        break;
      }
      default:
        goto err_ret;
    }
    
    taosHashPut(pObj, (char *)pvar, t,  &dummy, sizeof(dummy));
H
Haojun Liao 已提交
688
    taosVariantDestroy(&tmpVar);
W
wpan 已提交
689
    memset(&tmpVar, 0, sizeof(tmpVar));
W
wpan 已提交
690 691 692 693 694 695
  } 

  *q = (void *)pObj;
  pObj = NULL;
  
err_ret:  
H
Haojun Liao 已提交
696
  taosVariantDestroy(&tmpVar);
W
wpan 已提交
697 698 699 700 701
  taosHashCleanup(pObj);
  tfree(tmp);
}


H
Haojun Liao 已提交
702 703
tExprNode* exprdup(tExprNode* pNode) {
  if (pNode == NULL) {
H
Haojun Liao 已提交
704 705 706
    return NULL;
  }

H
Haojun Liao 已提交
707
  tExprNode* pCloned = calloc(1, sizeof(tExprNode));
708
  if (pNode->nodeType == TEXPR_BINARYEXPR_NODE) {
H
Haojun Liao 已提交
709 710 711 712 713 714
    tExprNode* pLeft  = exprdup(pNode->_node.pLeft);
    tExprNode* pRight = exprdup(pNode->_node.pRight);

    pCloned->_node.pLeft  = pLeft;
    pCloned->_node.pRight = pRight;
    pCloned->_node.optr  = pNode->_node.optr;
715
  } else if (pNode->nodeType == TEXPR_VALUE_NODE) {
H
Haojun Liao 已提交
716
    pCloned->pVal = calloc(1, sizeof(SVariant));
H
Haojun Liao 已提交
717
    taosVariantAssign(pCloned->pVal, pNode->pVal);
718
  } else if (pNode->nodeType == TEXPR_COL_NODE) {
H
Haojun Liao 已提交
719 720
    pCloned->pSchema = calloc(1, sizeof(SSchema));
    *pCloned->pSchema = *pNode->pSchema;
H
Haojun Liao 已提交
721 722
  }

H
Haojun Liao 已提交
723 724
  pCloned->nodeType = pNode->nodeType;
  return pCloned;
H
Haojun Liao 已提交
725 726
}