scalar.c 29.7 KB
Newer Older
D
dapan1121 已提交
1 2
#include "function.h"
#include "functionMgt.h"
H
Haojun Liao 已提交
3 4
#include "nodes.h"
#include "querynodes.h"
D
dapan1121 已提交
5
#include "sclInt.h"
H
Haojun Liao 已提交
6 7 8
#include "sclvector.h"
#include "tcommon.h"
#include "tdatablock.h"
9
#include "scalar.h"
S
slzhou 已提交
10
#include "tudf.h"
D
fix bug  
dapan1121 已提交
11
#include "ttime.h"
D
dapan1121 已提交
12

D
dapan1121 已提交
13
int32_t scalarGetOperatorParamNum(EOperatorType type) {
D
dapan1121 已提交
14
  if (OP_TYPE_IS_NULL == type || OP_TYPE_IS_NOT_NULL == type || OP_TYPE_IS_TRUE == type || OP_TYPE_IS_NOT_TRUE == type 
D
dapan1121 已提交
15 16
   || OP_TYPE_IS_FALSE == type || OP_TYPE_IS_NOT_FALSE == type || OP_TYPE_IS_UNKNOWN == type || OP_TYPE_IS_NOT_UNKNOWN == type
   || OP_TYPE_MINUS == type) {
D
dapan1121 已提交
17 18 19 20 21 22
    return 1;
  }

  return 2;
}

D
dapan1121 已提交
23
int32_t sclConvertToTsValueNode(int8_t precision, SValueNode* valueNode) {
D
dapan 已提交
24
  char *timeStr = valueNode->datum.p;
D
dapan1121 已提交
25 26 27
  int32_t code = convertStringToTimestamp(valueNode->node.resType.type, valueNode->datum.p, precision, &valueNode->datum.i);
  if (code != TSDB_CODE_SUCCESS) {
    return code;
D
dapan 已提交
28 29
  }
  taosMemoryFree(timeStr);
D
dapan1121 已提交
30
  valueNode->typeData = valueNode->datum.i;
D
dapan 已提交
31 32 33
  
  valueNode->node.resType.type = TSDB_DATA_TYPE_TIMESTAMP;
  valueNode->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_TIMESTAMP].bytes;
D
dapan1121 已提交
34 35

  return TSDB_CODE_SUCCESS;
D
dapan 已提交
36 37 38
}


39
SColumnInfoData* createColumnInfoData(SDataType* pType, int32_t numOfRows) {
H
Haojun Liao 已提交
40
  SColumnInfoData* pColumnData = taosMemoryCalloc(1, sizeof(SColumnInfoData));
41 42 43 44 45 46 47 48 49 50
  if (pColumnData == NULL) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
    return NULL;
  }

  pColumnData->info.type      = pType->type;
  pColumnData->info.bytes     = pType->bytes;
  pColumnData->info.scale     = pType->scale;
  pColumnData->info.precision = pType->precision;

51
  int32_t code = colInfoDataEnsureCapacity(pColumnData, 0, numOfRows);
52 53
  if (code != TSDB_CODE_SUCCESS) {
    terrno = TSDB_CODE_OUT_OF_MEMORY;
H
Haojun Liao 已提交
54
    taosMemoryFree(pColumnData);
55 56 57 58 59 60
    return NULL;
  } else {
    return pColumnData;
  }
}

61 62 63 64
int32_t doConvertDataType(SValueNode* pValueNode, SScalarParam* out) {
  SScalarParam in = {.numOfRows = 1};
  in.columnData = createColumnInfoData(&pValueNode->node.resType, 1);
  colDataAppend(in.columnData, 0, nodesGetValueFromNode(pValueNode), false);
65

66
  colInfoDataEnsureCapacity(out->columnData, 0, 1);
67 68 69 70
  int32_t code = vectorConvertImpl(&in, out);
  sclFreeParam(&in);

  return code;
71 72
}

D
dapan1121 已提交
73 74 75 76 77 78 79 80 81 82 83 84
int32_t scalarGenerateSetFromList(void **data, void *pNode, uint32_t type) {
  SHashObj *pObj = taosHashInit(256, taosGetDefaultHashFunction(type), true, false);
  if (NULL == pObj) {
    sclError("taosHashInit failed, size:%d", 256);
    SCL_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }

  taosHashSetEqualFp(pObj, taosGetDefaultEqualFunction(type)); 

  int32_t code = 0;
  SNodeListNode *nodeList = (SNodeListNode *)pNode;
  SListCell *cell = nodeList->pNodeList->pHead;
H
Haojun Liao 已提交
85
  SScalarParam out = {.columnData = taosMemoryCalloc(1, sizeof(SColumnInfoData))};
86

D
dapan1121 已提交
87 88 89 90 91
  int32_t len = 0;
  void *buf = NULL;
  
  for (int32_t i = 0; i < nodeList->pNodeList->length; ++i) {
    SValueNode *valueNode = (SValueNode *)cell->pNode;
D
dapan1121 已提交
92
    
D
dapan1121 已提交
93
    if (valueNode->node.resType.type != type) {
94
      out.columnData->info.type = type;
D
dapan1121 已提交
95 96 97 98 99 100 101 102 103
      if (IS_VAR_DATA_TYPE(type)) {
        if (IS_VAR_DATA_TYPE(valueNode->node.resType.type)) {
          out.columnData->info.bytes = valueNode->node.resType.bytes * TSDB_NCHAR_SIZE;
        } else {
          out.columnData->info.bytes = 64 * TSDB_NCHAR_SIZE;
        }
      } else {
        out.columnData->info.bytes = tDataTypes[type].bytes;
      }
104

105 106
      code = doConvertDataType(valueNode, &out);
      if (code != TSDB_CODE_SUCCESS) {
107
//        sclError("convert data from %d to %d failed", in.type, out.type);
D
dapan1121 已提交
108 109 110 111
        SCL_ERR_JRET(code);
      }

      if (IS_VAR_DATA_TYPE(type)) {
D
dapan1121 已提交
112 113 114
        char* data = colDataGetVarData(out.columnData, 0);
        len = varDataLen(data);
        buf = varDataVal(data);
D
dapan1121 已提交
115 116
      } else {
        len = tDataTypes[type].bytes;
117
        buf = out.columnData->pData;
D
dapan1121 已提交
118 119 120
      }
    } else {
      buf = nodesGetValueFromNode(valueNode);
D
dapan1121 已提交
121 122 123 124 125
      if (IS_VAR_DATA_TYPE(type)) {
        len = varDataLen(buf);
        buf = varDataVal(buf);
      } else {
        len = valueNode->node.resType.bytes;
126
      }
D
dapan1121 已提交
127 128
    }
    
129
    if (taosHashPut(pObj, buf, (size_t)len, NULL, 0)) {
D
dapan1121 已提交
130
      sclError("taosHashPut to set failed");
D
dapan1121 已提交
131 132
      SCL_ERR_JRET(TSDB_CODE_QRY_OUT_OF_MEMORY);
    }
D
dapan1121 已提交
133 134

    cell = cell->pNext;
D
dapan1121 已提交
135 136 137 138 139 140 141 142 143 144
  }

  *data = pObj;
  return TSDB_CODE_SUCCESS;

_return:
  taosHashCleanup(pObj);
  SCL_RET(code);
}

D
dapan1121 已提交
145 146 147 148 149 150 151
void sclFreeRes(SHashObj *res) {
  SScalarParam *p = NULL;
  void *pIter = taosHashIterate(res, NULL);
  while (pIter) {
    p = (SScalarParam *)pIter;

    if (p) {
D
dapan 已提交
152
      sclFreeParam(p);
D
dapan1121 已提交
153 154 155 156 157 158
    }
    pIter = taosHashIterate(res, pIter);
  }
  taosHashCleanup(res);
}

D
dapan1121 已提交
159
void sclFreeParam(SScalarParam *param) {
160 161
  if (param->columnData != NULL) {
    colDataDestroy(param->columnData);
wmmhello's avatar
wmmhello 已提交
162
    taosMemoryFree(param->columnData);
163 164 165 166 167
  }

  if (param->pHashFilter != NULL) {
    taosHashCleanup(param->pHashFilter);
  }
D
dapan1121 已提交
168 169
}

D
dapan1121 已提交
170 171 172 173 174
int32_t sclCopyValueNodeValue(SValueNode *pNode, void **res) {
  if (TSDB_DATA_TYPE_NULL == pNode->node.resType.type) {
    return TSDB_CODE_SUCCESS;
  }
  
wafwerar's avatar
wafwerar 已提交
175
  *res = taosMemoryMalloc(pNode->node.resType.bytes);
D
dapan1121 已提交
176 177 178 179 180 181 182 183 184
  if (NULL == (*res)) {
    sclError("malloc %d failed", pNode->node.resType.bytes);
    SCL_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }

  memcpy(*res, nodesGetValueFromNode(pNode), pNode->node.resType.bytes);
  return TSDB_CODE_SUCCESS;
}

D
dapan1121 已提交
185 186
int32_t sclInitParam(SNode* node, SScalarParam *param, SScalarCtx *ctx, int32_t *rowNum) {
  switch (nodeType(node)) {
D
dapan1121 已提交
187 188 189 190 191
    case QUERY_NODE_LEFT_VALUE: {
      SSDataBlock* pb = taosArrayGetP(ctx->pBlockList, 0);
      param->numOfRows = pb->info.rows;
      break;
    }
D
dapan1121 已提交
192 193
    case QUERY_NODE_VALUE: {
      SValueNode *valueNode = (SValueNode *)node;
194 195 196 197

      param->numOfRows = 1;
      param->columnData = createColumnInfoData(&valueNode->node.resType, 1);
      if (TSDB_DATA_TYPE_NULL == valueNode->node.resType.type) {
198
        colDataAppendNULL(param->columnData, 0);
199 200
      } else {
        colDataAppend(param->columnData, 0, nodesGetValueFromNode(valueNode), false);
D
dapan1121 已提交
201
      }
D
dapan1121 已提交
202 203
      break;
    }
D
dapan1121 已提交
204 205
    case QUERY_NODE_NODE_LIST: {
      SNodeListNode *nodeList = (SNodeListNode *)node;
206 207
      if (LIST_LENGTH(nodeList->pNodeList) <= 0) {
        sclError("invalid length in nodeList, length:%d", LIST_LENGTH(nodeList->pNodeList));
D
dapan1121 已提交
208 209 210
        SCL_RET(TSDB_CODE_QRY_INVALID_INPUT);
      }

D
dapan1121 已提交
211 212 213 214 215 216 217
      int32_t type = vectorGetConvertType(ctx->type.selfType, ctx->type.peerType);
      if (type == 0) {
        type = nodeList->dataType.type;
      }
      
      SCL_ERR_RET(scalarGenerateSetFromList((void **)&param->pHashFilter, node, type));
      param->hashValueType = type;
D
dapan 已提交
218
      if (taosHashPut(ctx->pRes, &node, POINTER_BYTES, param, sizeof(*param))) {
219
        taosHashCleanup(param->pHashFilter);
D
dapan 已提交
220 221 222
        sclError("taosHashPut nodeList failed, size:%d", (int32_t)sizeof(*param));
        return TSDB_CODE_QRY_OUT_OF_MEMORY;
      }   
D
dapan1121 已提交
223 224
      break;
    }
X
Xiaoyu Wang 已提交
225
    case QUERY_NODE_COLUMN: {
D
dapan 已提交
226 227
      if (NULL == ctx->pBlockList) {
        sclError("invalid node type for constant calculating, type:%d, src:%p", nodeType(node), ctx->pBlockList);
D
dapan1121 已提交
228 229 230
        SCL_ERR_RET(TSDB_CODE_QRY_APP_ERROR);
      }
      
X
Xiaoyu Wang 已提交
231
      SColumnNode *ref = (SColumnNode *)node;
232 233 234 235 236 237 238 239 240 241 242

      int32_t index = -1;
      for(int32_t i = 0; i < taosArrayGetSize(ctx->pBlockList); ++i) {
        SSDataBlock* pb = taosArrayGetP(ctx->pBlockList, i);
        if (pb->info.blockId == ref->dataBlockId) {
          index = i;
          break;
        }
      }

      if (index == -1) {
D
dapan1121 已提交
243
        sclError("column tupleId is too big, tupleId:%d, dataBlockNum:%d", ref->dataBlockId, (int32_t)taosArrayGetSize(ctx->pBlockList));
D
dapan 已提交
244 245 246
        SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
      }

247 248
      SSDataBlock *block = *(SSDataBlock **)taosArrayGet(ctx->pBlockList, index);
      if (NULL == block || ref->slotId >= block->info.numOfCols) {
D
dapan 已提交
249
        sclError("column slotId is too big, slodId:%d, dataBlockSize:%d", ref->slotId, (int32_t)taosArrayGetSize(block->pDataBlock));
D
dapan1121 已提交
250 251 252
        SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
      }

D
dapan 已提交
253
      SColumnInfoData *columnData = (SColumnInfoData *)taosArrayGet(block->pDataBlock, ref->slotId);
254 255
      param->numOfRows = block->info.rows;
      param->columnData = columnData;
D
dapan1121 已提交
256 257
      break;
    }
258 259 260
    case QUERY_NODE_FUNCTION:
    case QUERY_NODE_OPERATOR:
    case QUERY_NODE_LOGIC_CONDITION: {
D
dapan1121 已提交
261 262 263 264 265 266 267 268
      SScalarParam *res = (SScalarParam *)taosHashGet(ctx->pRes, &node, POINTER_BYTES);
      if (NULL == res) {
        sclError("no result for node, type:%d, node:%p", nodeType(node), node);
        SCL_ERR_RET(TSDB_CODE_QRY_APP_ERROR);
      }
      *param = *res;
      break;
    }
269 270
    default:
      break;
D
dapan1121 已提交
271 272
  }

273 274 275
  if (param->numOfRows > *rowNum) {
    if ((1 != param->numOfRows) && (1 < *rowNum)) {
      sclError("different row nums, rowNum:%d, newRowNum:%d", *rowNum, param->numOfRows);
D
dapan1121 已提交
276 277 278
      SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
    }
    
279
    *rowNum = param->numOfRows;
D
dapan1121 已提交
280 281
  }

282
  param->param = ctx->param;
D
dapan1121 已提交
283 284 285
  return TSDB_CODE_SUCCESS;
}

D
dapan 已提交
286
int32_t sclInitParamList(SScalarParam **pParams, SNodeList* pParamList, SScalarCtx *ctx, int32_t *paramNum, int32_t *rowNum) {  
D
dapan1121 已提交
287
  int32_t code = 0;
D
dapan1121 已提交
288 289 290 291 292 293 294 295
  if (NULL == pParamList) {
    if (ctx->pBlockList) {
      SSDataBlock *pBlock = taosArrayGet(ctx->pBlockList, 0);
      *rowNum = pBlock->info.rows;
    } else {
      *rowNum = 1;
    }

D
dapan 已提交
296
    *paramNum = 1;
D
dapan1121 已提交
297
  } else {
D
dapan 已提交
298
    *paramNum = pParamList->length;
D
dapan1121 已提交
299 300
  }

D
dapan 已提交
301
  SScalarParam *paramList = taosMemoryCalloc(*paramNum, sizeof(SScalarParam));
D
dapan1121 已提交
302
  if (NULL == paramList) {
D
dapan 已提交
303
    sclError("calloc %d failed", (int32_t)((*paramNum) * sizeof(SScalarParam)));
D
dapan1121 已提交
304
    SCL_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
D
dapan1121 已提交
305 306
  }

D
dapan1121 已提交
307 308 309 310 311 312
  if (pParamList) {
    SNode *tnode = NULL;
    int32_t i = 0;
    if (SCL_IS_CONST_CALC(ctx)) {
      WHERE_EACH (tnode, pParamList) { 
        if (!SCL_IS_CONST_NODE(tnode)) {
D
dapan 已提交
313
          WHERE_NEXT;
D
dapan1121 已提交
314 315 316 317 318 319 320 321 322 323 324 325
        } else {
          SCL_ERR_JRET(sclInitParam(tnode, &paramList[i], ctx, rowNum));
          ERASE_NODE(pParamList);
        }
        
        ++i;
      }
    } else {
      FOREACH(tnode, pParamList) { 
        SCL_ERR_JRET(sclInitParam(tnode, &paramList[i], ctx, rowNum));
        ++i;
      }
D
dapan1121 已提交
326
    }
D
dapan1121 已提交
327 328 329
  } else {
    paramList[0].numOfRows = *rowNum;
  }
D
dapan1121 已提交
330

D
dapan1121 已提交
331 332
  if (0 == *rowNum) {
    taosMemoryFreeClear(paramList);    
D
dapan1121 已提交
333
  }
D
dapan1121 已提交
334

D
dapan1121 已提交
335
  *pParams = paramList;
D
dapan1121 已提交
336
  return TSDB_CODE_SUCCESS;
D
dapan1121 已提交
337

D
dapan1121 已提交
338
_return:
wafwerar's avatar
wafwerar 已提交
339
  taosMemoryFreeClear(paramList);
D
dapan1121 已提交
340 341 342
  SCL_RET(code);
}

D
dapan1121 已提交
343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382
int32_t sclGetNodeType(SNode *pNode, SScalarCtx *ctx) {
  if (NULL == pNode) {
    return -1;
  }
  
  switch (nodeType(pNode)) {
    case QUERY_NODE_VALUE: {
      SValueNode *valueNode = (SValueNode *)pNode;
      return valueNode->node.resType.type;
    }
    case QUERY_NODE_NODE_LIST: {
      SNodeListNode *nodeList = (SNodeListNode *)pNode;
      return nodeList->dataType.type;
    }
    case QUERY_NODE_COLUMN: {
      SColumnNode *colNode = (SColumnNode *)pNode;
      return colNode->node.resType.type;
    }
    case QUERY_NODE_FUNCTION:
    case QUERY_NODE_OPERATOR:
    case QUERY_NODE_LOGIC_CONDITION: {
      SScalarParam *res = (SScalarParam *)taosHashGet(ctx->pRes, &pNode, POINTER_BYTES);
      if (NULL == res) {
        sclError("no result for node, type:%d, node:%p", nodeType(pNode), pNode);
        return -1;
      }
      return res->columnData->info.type;
    }
  }

  return -1;
}


void sclSetOperatorValueType(SOperatorNode *node, SScalarCtx *ctx) {
  ctx->type.opResType = node->node.resType.type;
  ctx->type.selfType = sclGetNodeType(node->pLeft, ctx);
  ctx->type.peerType = sclGetNodeType(node->pRight, ctx);
}

D
dapan1121 已提交
383 384
int32_t sclInitOperatorParams(SScalarParam **pParams, SOperatorNode *node, SScalarCtx *ctx, int32_t *rowNum) {
  int32_t code = 0;
D
dapan1121 已提交
385
  int32_t paramNum = scalarGetOperatorParamNum(node->opType);
D
dapan1121 已提交
386 387 388 389 390
  if (NULL == node->pLeft || (paramNum == 2 && NULL == node->pRight)) {
    sclError("invalid operation node, left:%p, right:%p", node->pLeft, node->pRight);
    SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }
  
wafwerar's avatar
wafwerar 已提交
391
  SScalarParam *paramList = taosMemoryCalloc(paramNum, sizeof(SScalarParam));
D
dapan1121 已提交
392 393
  if (NULL == paramList) {
    sclError("calloc %d failed", (int32_t)(paramNum * sizeof(SScalarParam)));
D
dapan1121 已提交
394 395 396
    SCL_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }

D
dapan1121 已提交
397 398
  sclSetOperatorValueType(node, ctx);

D
dapan1121 已提交
399
  SCL_ERR_JRET(sclInitParam(node->pLeft, &paramList[0], ctx, rowNum));
D
dapan1121 已提交
400
  if (paramNum > 1) {
D
dapan1121 已提交
401
    TSWAP(ctx->type.selfType, ctx->type.peerType);
D
dapan1121 已提交
402
    SCL_ERR_JRET(sclInitParam(node->pRight, &paramList[1], ctx, rowNum));
D
dapan1121 已提交
403 404
  }

D
dapan1121 已提交
405
  *pParams = paramList;
D
dapan1121 已提交
406
  return TSDB_CODE_SUCCESS;
D
dapan1121 已提交
407 408

_return:
wafwerar's avatar
wafwerar 已提交
409
  taosMemoryFreeClear(paramList);
D
dapan1121 已提交
410
  SCL_RET(code);
D
dapan1121 已提交
411 412
}

413
int32_t sclExecFunction(SFunctionNode *node, SScalarCtx *ctx, SScalarParam *output) {
D
dapan1121 已提交
414 415
  SScalarParam *params = NULL;
  int32_t rowNum = 0;
D
dapan 已提交
416
  int32_t paramNum = 0;
D
dapan1121 已提交
417
  int32_t code = 0;
D
dapan 已提交
418
  SCL_ERR_RET(sclInitParamList(&params, node->pParameterList, ctx, &paramNum, &rowNum));
D
dapan1121 已提交
419

D
dapan1121 已提交
420
  if (fmIsUserDefinedFunc(node->funcId)) {
421
    code = callUdfScalarFunc(node->functionName, params, paramNum, output);
422 423 424 425
    if (code != 0) {
      sclError("fmExecFunction error. callUdfScalarFunc. function name: %s, udf code:%d", node->functionName, code);
      goto _return;
    }
D
dapan1121 已提交
426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444
  } else {
    SScalarFuncExecFuncs ffpSet = {0};
    code = fmGetScalarFuncExecFuncs(node->funcId, &ffpSet);
    if (code) {
      sclError("fmGetFuncExecFuncs failed, funcId:%d, code:%s", node->funcId, tstrerror(code));
      SCL_ERR_JRET(code);
    }
  
    output->columnData = createColumnInfoData(&node->node.resType, rowNum);
    if (output->columnData == NULL) {
      sclError("calloc %d failed", (int32_t)(rowNum * output->columnData->info.bytes));
      SCL_ERR_JRET(TSDB_CODE_QRY_OUT_OF_MEMORY);
    }

    code = (*ffpSet.process)(params, paramNum, output);
    if (code) {
      sclError("scalar function exec failed, funcId:%d, code:%s", node->funcId, tstrerror(code));
      SCL_ERR_JRET(code);
    }
D
dapan1121 已提交
445 446 447 448
  }

_return:

D
dapan 已提交
449
  for (int32_t i = 0; i < paramNum; ++i) {
H
Haojun Liao 已提交
450
//    sclFreeParamNoData(params + i);
D
dapan1121 已提交
451 452
  }

wafwerar's avatar
wafwerar 已提交
453
  taosMemoryFreeClear(params);
D
dapan1121 已提交
454 455 456 457 458 459 460 461 462 463 464 465
  SCL_RET(code);
}

int32_t sclExecLogic(SLogicConditionNode *node, SScalarCtx *ctx, SScalarParam *output) {
  if (NULL == node->pParameterList || node->pParameterList->length <= 0) {
    sclError("invalid logic parameter list, list:%p, paramNum:%d", node->pParameterList, node->pParameterList ? node->pParameterList->length : 0);
    SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }

  if (TSDB_DATA_TYPE_BOOL != node->node.resType.type) {
    sclError("invalid logic resType, type:%d", node->node.resType.type);
    SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
D
dapan1121 已提交
466 467
  }

D
dapan1121 已提交
468 469 470 471 472 473 474
  if (LOGIC_COND_TYPE_NOT == node->condType && node->pParameterList->length > 1) {
    sclError("invalid NOT operation parameter number, paramNum:%d", node->pParameterList->length);
    SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }

  SScalarParam *params = NULL;
  int32_t rowNum = 0;
D
dapan 已提交
475
  int32_t paramNum = 0;
D
dapan1121 已提交
476
  int32_t code = 0;
D
dapan 已提交
477
  SCL_ERR_RET(sclInitParamList(&params, node->pParameterList, ctx, &paramNum, &rowNum));
D
dapan1121 已提交
478 479 480 481
  if (NULL == params) {
    output->numOfRows = 0;
    return TSDB_CODE_SUCCESS;
  }
D
dapan1121 已提交
482

483 484 485 486 487 488
  int32_t type = node->node.resType.type;
  output->numOfRows = rowNum;

  SDataType t = {.type = type, .bytes = tDataTypes[type].bytes};
  output->columnData = createColumnInfoData(&t, rowNum);
  if (output->columnData == NULL) {
D
dapan1121 已提交
489
    sclError("calloc %d failed", (int32_t)(rowNum * sizeof(bool)));
D
dapan1121 已提交
490 491
    SCL_ERR_JRET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }
D
dapan1121 已提交
492

D
dapan1121 已提交
493
  bool value = false;
D
dapan 已提交
494
  bool complete = true;
D
dapan1121 已提交
495
  for (int32_t i = 0; i < rowNum; ++i) {
D
dapan 已提交
496 497
    complete = true;
    for (int32_t m = 0; m < paramNum; ++m) {
D
dapan1121 已提交
498
      if (NULL == params[m].columnData) {
D
dapan 已提交
499
        complete = false;
D
dapan1121 已提交
500 501
        continue;
      }
502 503 504
      char* p = colDataGetData(params[m].columnData, i);
      GET_TYPED_DATA(value, bool, params[m].columnData->info.type, p);

D
dapan1121 已提交
505
      if (LOGIC_COND_TYPE_AND == node->condType && (false == value)) {
D
dapan1121 已提交
506
        complete = true;
D
dapan1121 已提交
507 508
        break;
      } else if (LOGIC_COND_TYPE_OR == node->condType && value) {
D
dapan1121 已提交
509
        complete = true;
D
dapan1121 已提交
510 511 512 513 514 515
        break;
      } else if (LOGIC_COND_TYPE_NOT == node->condType) {
        value = !value;
      }
    }

D
dapan 已提交
516 517 518
    if (complete) {
      colDataAppend(output->columnData, i, (char*) &value, false);
    }
D
dapan1121 已提交
519 520
  }

D
dapan1121 已提交
521 522 523 524 525
  if (SCL_IS_CONST_CALC(ctx) && (false == complete)) {
    sclFreeParam(output);
    output->numOfRows = 0;
  }

D
dapan1121 已提交
526
_return:
D
dapan1121 已提交
527

D
dapan 已提交
528
  for (int32_t i = 0; i < paramNum; ++i) {
H
Haojun Liao 已提交
529
//    sclFreeParamNoData(params + i);
D
dapan1121 已提交
530 531
  }

wafwerar's avatar
wafwerar 已提交
532
  taosMemoryFreeClear(params);
D
dapan1121 已提交
533
  SCL_RET(code);
D
dapan1121 已提交
534 535 536 537 538 539
}

int32_t sclExecOperator(SOperatorNode *node, SScalarCtx *ctx, SScalarParam *output) {
  SScalarParam *params = NULL;
  int32_t rowNum = 0;
  int32_t code = 0;
540

D
dapan1121 已提交
541
  SCL_ERR_RET(sclInitOperatorParams(&params, node, ctx, &rowNum));
542 543 544
  output->columnData = createColumnInfoData(&node->node.resType, rowNum);
  if (output->columnData == NULL) {
    sclError("calloc failed, size:%d", (int32_t)rowNum * node->node.resType.bytes);
D
dapan1121 已提交
545 546 547 548 549
    SCL_ERR_JRET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }

  _bin_scalar_fn_t OperatorFn = getBinScalarOperatorFn(node->opType);

D
dapan1121 已提交
550
  int32_t paramNum = scalarGetOperatorParamNum(node->opType);
D
dapan1121 已提交
551 552
  SScalarParam* pLeft = &params[0];
  SScalarParam* pRight = paramNum > 1 ? &params[1] : NULL;
553

D
dapan 已提交
554
  OperatorFn(pLeft, pRight, output, TSDB_ORDER_ASC);
D
dapan1121 已提交
555 556

_return:
D
dapan1121 已提交
557
  for (int32_t i = 0; i < paramNum; ++i) {
558
//    sclFreeParam(&params[i]);
D
dapan1121 已提交
559 560
  }

wafwerar's avatar
wafwerar 已提交
561
  taosMemoryFreeClear(params);
D
dapan1121 已提交
562
  SCL_RET(code);
D
dapan1121 已提交
563 564
}

D
dapan1121 已提交
565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586
EDealRes sclRewriteBasedOnOptr(SNode** pNode, SScalarCtx *ctx, EOperatorType opType) {
  if (opType <= OP_TYPE_CALC_MAX) {
    SValueNode *res = (SValueNode *)nodesMakeNode(QUERY_NODE_VALUE);
    if (NULL == res) {
      sclError("make value node failed");    
      ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
      return DEAL_RES_ERROR;
    }
    
    res->node.resType.type = TSDB_DATA_TYPE_NULL;
    
    nodesDestroyNode(*pNode);
    *pNode = (SNode*)res;
  } else {
    SValueNode *res = (SValueNode *)nodesMakeNode(QUERY_NODE_VALUE);
    if (NULL == res) {
      sclError("make value node failed");    
      ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
      return DEAL_RES_ERROR;
    }
    
    res->node.resType.type = TSDB_DATA_TYPE_BOOL;
D
dapan1121 已提交
587
    res->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BOOL].bytes;
D
dapan1121 已提交
588 589 590 591 592 593 594 595 596 597
    res->datum.b = false;
    
    nodesDestroyNode(*pNode);
    *pNode = (SNode*)res;
  }

  return DEAL_RES_CONTINUE;
}


D
dapan 已提交
598
EDealRes sclRewriteNonConstOperator(SNode** pNode, SScalarCtx *ctx) {
D
dapan1121 已提交
599
  SOperatorNode *node = (SOperatorNode *)*pNode;
D
dapan1121 已提交
600
  int32_t code = 0;
D
dapan1121 已提交
601 602 603

  if (node->pLeft && (QUERY_NODE_VALUE == nodeType(node->pLeft))) {
    SValueNode *valueNode = (SValueNode *)node->pLeft;
D
dapan1121 已提交
604
    if (SCL_IS_NULL_VALUE_NODE(valueNode) && (node->opType != OP_TYPE_IS_NULL && node->opType != OP_TYPE_IS_NOT_NULL)) {
D
dapan1121 已提交
605 606
      return sclRewriteBasedOnOptr(pNode, ctx, node->opType);
    }
D
dapan 已提交
607 608 609

    if (IS_STR_DATA_TYPE(valueNode->node.resType.type) && node->pRight && nodesIsExprNode(node->pRight) 
      && ((SExprNode*)node->pRight)->resType.type == TSDB_DATA_TYPE_TIMESTAMP) {
D
dapan1121 已提交
610 611 612 613 614
      code = sclConvertToTsValueNode(((SExprNode*)node->pRight)->resType.precision, valueNode);
      if (code) {
        ctx->code = code;
        return DEAL_RES_ERROR;
      }
D
dapan 已提交
615
    }
D
dapan1121 已提交
616 617 618 619
  }

  if (node->pRight && (QUERY_NODE_VALUE == nodeType(node->pRight))) {
    SValueNode *valueNode = (SValueNode *)node->pRight;
D
dapan1121 已提交
620
    if (SCL_IS_NULL_VALUE_NODE(valueNode) && (node->opType != OP_TYPE_IS_NULL && node->opType != OP_TYPE_IS_NOT_NULL)) {
D
dapan1121 已提交
621 622
      return sclRewriteBasedOnOptr(pNode, ctx, node->opType);
    }
D
dapan 已提交
623 624 625

    if (IS_STR_DATA_TYPE(valueNode->node.resType.type) && node->pLeft && nodesIsExprNode(node->pLeft) 
      && ((SExprNode*)node->pLeft)->resType.type == TSDB_DATA_TYPE_TIMESTAMP) {
D
dapan1121 已提交
626 627 628 629 630
      code = sclConvertToTsValueNode(((SExprNode*)node->pLeft)->resType.precision, valueNode);
      if (code) {
        ctx->code = code;
        return DEAL_RES_ERROR;
      }
D
dapan 已提交
631
    }
D
dapan1121 已提交
632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657
  }

  if (node->pRight && (QUERY_NODE_NODE_LIST == nodeType(node->pRight))) {
    SNodeListNode *listNode = (SNodeListNode *)node->pRight;
    SNode* tnode = NULL;
    WHERE_EACH(tnode, listNode->pNodeList) {
      if (SCL_IS_NULL_VALUE_NODE(tnode)) {
        if (node->opType == OP_TYPE_IN) {
          ERASE_NODE(listNode->pNodeList);
          continue;
        } else { //OP_TYPE_NOT_IN
          return sclRewriteBasedOnOptr(pNode, ctx, node->opType);
        }
      }

      WHERE_NEXT;
    }

    if (listNode->pNodeList->length <= 0) {
      return sclRewriteBasedOnOptr(pNode, ctx, node->opType);
    }
  }

  return DEAL_RES_CONTINUE;
}

D
dapan 已提交
658
EDealRes sclRewriteFunction(SNode** pNode, SScalarCtx *ctx) {
D
dapan1121 已提交
659
  SFunctionNode *node = (SFunctionNode *)*pNode;
D
dapan1121 已提交
660
  SNode* tnode = NULL;
661
  if (!fmIsScalarFunc(node->funcId)) {
D
dapan1121 已提交
662 663
    return DEAL_RES_CONTINUE;
  }
664

D
dapan1121 已提交
665 666 667 668 669 670
  FOREACH(tnode, node->pParameterList) {
    if (!SCL_IS_CONST_NODE(tnode)) {
      return DEAL_RES_CONTINUE;
    }
  }

D
dapan1121 已提交
671
  SScalarParam output = {0};
672

673
  ctx->code = sclExecFunction(node, ctx, &output);
D
dapan 已提交
674
  if (ctx->code) {
D
dapan1121 已提交
675 676 677
    return DEAL_RES_ERROR;
  }

D
dapan1121 已提交
678
  SValueNode *res = (SValueNode *)nodesMakeNode(QUERY_NODE_VALUE);
D
dapan1121 已提交
679 680
  if (NULL == res) {
    sclError("make value node failed");
D
dapan1121 已提交
681
    sclFreeParam(&output);
D
dapan 已提交
682
    ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
D
dapan1121 已提交
683 684 685
    return DEAL_RES_ERROR;
  }

686 687
  res->translate = true;

688 689
  if (colDataIsNull_s(output.columnData, 0)) {
    res->node.resType.type = TSDB_DATA_TYPE_NULL;
D
dapan1121 已提交
690
  } else {
D
dapan1121 已提交
691 692 693 694
    res->node.resType.type = output.columnData->info.type;
    res->node.resType.bytes = output.columnData->info.bytes;
    res->node.resType.scale = output.columnData->info.scale;
    res->node.resType.precision = output.columnData->info.precision;
695 696
    int32_t type = output.columnData->info.type;
    if (IS_VAR_DATA_TYPE(type)) {
697
      res->datum.p = taosMemoryCalloc(res->node.resType.bytes + VARSTR_HEADER_SIZE + 1, 1);
698
      memcpy(res->datum.p, output.columnData->pData, varDataTLen(output.columnData->pData));
699
    } else {
D
dapan1121 已提交
700
      nodesSetValueNodeValue(res, output.columnData->pData);
701
    }
D
dapan1121 已提交
702
  }
703

D
dapan1121 已提交
704 705 706
  nodesDestroyNode(*pNode);
  *pNode = (SNode*)res;

D
dapan1121 已提交
707
  sclFreeParam(&output);
D
dapan1121 已提交
708 709 710
  return DEAL_RES_CONTINUE;
}

D
dapan 已提交
711
EDealRes sclRewriteLogic(SNode** pNode, SScalarCtx *ctx) {
D
dapan1121 已提交
712 713
  SLogicConditionNode *node = (SLogicConditionNode *)*pNode;

H
Haojun Liao 已提交
714
  SScalarParam output = {0};
D
dapan 已提交
715 716
  ctx->code = sclExecLogic(node, ctx, &output);
  if (ctx->code) {
D
dapan1121 已提交
717 718 719
    return DEAL_RES_ERROR;
  }

D
dapan1121 已提交
720 721 722 723
  if (0 == output.numOfRows) {
    return DEAL_RES_CONTINUE;
  }

D
dapan1121 已提交
724
  SValueNode *res = (SValueNode *)nodesMakeNode(QUERY_NODE_VALUE);
D
dapan1121 已提交
725 726
  if (NULL == res) {
    sclError("make value node failed");
727
    sclFreeParam(&output);
D
dapan 已提交
728
    ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
D
dapan1121 已提交
729 730 731 732
    return DEAL_RES_ERROR;
  }

  res->node.resType = node->node.resType;
733
  res->translate = true;
D
dapan1121 已提交
734

735 736 737 738
  int32_t type = output.columnData->info.type;
  if (IS_VAR_DATA_TYPE(type)) {
    res->datum.p = output.columnData->pData;
    output.columnData->pData = NULL;
D
dapan1121 已提交
739
  } else {
D
dapan1121 已提交
740
    nodesSetValueNodeValue(res, output.columnData->pData);
D
dapan1121 已提交
741
  }
D
dapan1121 已提交
742 743 744 745

  nodesDestroyNode(*pNode);
  *pNode = (SNode*)res;

D
dapan1121 已提交
746
  sclFreeParam(&output);
D
dapan1121 已提交
747 748 749
  return DEAL_RES_CONTINUE;
}

D
dapan 已提交
750
EDealRes sclRewriteOperator(SNode** pNode, SScalarCtx *ctx) {
D
dapan1121 已提交
751
  SOperatorNode *node = (SOperatorNode *)*pNode;
D
dapan1121 已提交
752

D
dapan1121 已提交
753
  if ((!SCL_IS_CONST_NODE(node->pLeft)) || (!SCL_IS_CONST_NODE(node->pRight))) {
D
dapan 已提交
754
    return sclRewriteNonConstOperator(pNode, ctx);
D
dapan1121 已提交
755 756
  }

H
Haojun Liao 已提交
757
  SScalarParam output = {.columnData = taosMemoryCalloc(1, sizeof(SColumnInfoData))};
D
dapan 已提交
758 759
  ctx->code = sclExecOperator(node, ctx, &output);
  if (ctx->code) {
D
dapan1121 已提交
760 761 762
    return DEAL_RES_ERROR;
  }

D
dapan1121 已提交
763
  SValueNode *res = (SValueNode *)nodesMakeNode(QUERY_NODE_VALUE);
D
dapan1121 已提交
764
  if (NULL == res) {
765 766
    sclError("make value node failed");
    sclFreeParam(&output);
D
dapan 已提交
767
    ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
D
dapan1121 已提交
768 769 770
    return DEAL_RES_ERROR;
  }

771
  res->translate = true;
D
dapan1121 已提交
772

773 774
  if (colDataIsNull_s(output.columnData, 0)) {
    res->node.resType.type = TSDB_DATA_TYPE_NULL;
D
dapan1121 已提交
775
  } else {
776 777 778 779 780 781
    res->node.resType = node->node.resType;
    int32_t type = output.columnData->info.type;
    if (IS_VAR_DATA_TYPE(type)) {  // todo refactor
      res->datum.p = output.columnData->pData;
      output.columnData->pData = NULL;
    } else {
D
dapan1121 已提交
782
      nodesSetValueNodeValue(res, output.columnData->pData);    
783
    }
D
dapan1121 已提交
784
  }
D
dapan1121 已提交
785 786 787 788

  nodesDestroyNode(*pNode);
  *pNode = (SNode*)res;

H
Haojun Liao 已提交
789
  sclFreeParam(&output);
D
dapan1121 已提交
790 791 792 793
  return DEAL_RES_CONTINUE;
}

EDealRes sclConstantsRewriter(SNode** pNode, void* pContext) {
D
dapan 已提交
794 795
  SScalarCtx *ctx = (SScalarCtx *)pContext;

D
dapan1121 已提交
796
  if (QUERY_NODE_FUNCTION == nodeType(*pNode)) {
D
dapan 已提交
797
    return sclRewriteFunction(pNode, ctx);
D
dapan1121 已提交
798 799 800
  }

  if (QUERY_NODE_LOGIC_CONDITION == nodeType(*pNode)) {
D
dapan 已提交
801
    return sclRewriteLogic(pNode, ctx);
D
dapan1121 已提交
802 803
  }

D
dapan1121 已提交
804
  if (QUERY_NODE_OPERATOR == nodeType(*pNode)) {
D
dapan 已提交
805
    return sclRewriteOperator(pNode, ctx);
806
  }
807 808

  return DEAL_RES_CONTINUE;
D
dapan1121 已提交
809 810
}

D
dapan 已提交
811
EDealRes sclWalkFunction(SNode* pNode, SScalarCtx *ctx) {
D
dapan1121 已提交
812
  SFunctionNode *node = (SFunctionNode *)pNode;
D
dapan1121 已提交
813
  SScalarParam output = {0};
814

815
  ctx->code = sclExecFunction(node, ctx, &output);
D
dapan1121 已提交
816 817
  if (ctx->code) {
    return DEAL_RES_ERROR;
D
dapan1121 已提交
818 819
  }

D
dapan1121 已提交
820
  if (taosHashPut(ctx->pRes, &pNode, POINTER_BYTES, &output, sizeof(output))) {
D
dapan1121 已提交
821 822
    ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
    return DEAL_RES_ERROR;
D
dapan1121 已提交
823 824
  }

D
dapan1121 已提交
825 826 827
  return DEAL_RES_CONTINUE;
}

D
dapan 已提交
828
EDealRes sclWalkLogic(SNode* pNode, SScalarCtx *ctx) {
D
dapan1121 已提交
829
  SLogicConditionNode *node = (SLogicConditionNode *)pNode;
D
dapan1121 已提交
830
  SScalarParam output = {0};
831

D
dapan1121 已提交
832 833 834
  ctx->code = sclExecLogic(node, ctx, &output);
  if (ctx->code) {
    return DEAL_RES_ERROR;
D
dapan1121 已提交
835 836
  }

D
dapan1121 已提交
837
  if (taosHashPut(ctx->pRes, &pNode, POINTER_BYTES, &output, sizeof(output))) {
D
dapan1121 已提交
838
    ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
D
dapan1121 已提交
839
    return DEAL_RES_ERROR;
D
dapan1121 已提交
840 841 842 843 844
  }

  return DEAL_RES_CONTINUE;
}

D
dapan 已提交
845
EDealRes sclWalkOperator(SNode* pNode, SScalarCtx *ctx) {
D
dapan1121 已提交
846
  SOperatorNode *node = (SOperatorNode *)pNode;
D
dapan1121 已提交
847
  SScalarParam output = {0};
D
dapan1121 已提交
848
  
D
dapan1121 已提交
849 850
  ctx->code = sclExecOperator(node, ctx, &output);
  if (ctx->code) {
D
dapan1121 已提交
851 852
    return DEAL_RES_ERROR;
  }
D
dapan1121 已提交
853

D
dapan1121 已提交
854
  if (taosHashPut(ctx->pRes, &pNode, POINTER_BYTES, &output, sizeof(output))) {
D
dapan1121 已提交
855
    ctx->code = TSDB_CODE_QRY_OUT_OF_MEMORY;
D
dapan1121 已提交
856 857 858
    return DEAL_RES_ERROR;
  }

D
dapan1121 已提交
859 860 861
  return DEAL_RES_CONTINUE;
}

862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910
EDealRes sclWalkTarget(SNode* pNode, SScalarCtx *ctx) {
  STargetNode *target = (STargetNode *)pNode;

  if (target->dataBlockId >= taosArrayGetSize(ctx->pBlockList)) {
    sclError("target tupleId is too big, tupleId:%d, dataBlockNum:%d", target->dataBlockId, (int32_t)taosArrayGetSize(ctx->pBlockList));
    ctx->code = TSDB_CODE_QRY_INVALID_INPUT;
    return DEAL_RES_ERROR;
  }

  int32_t index = -1;
  for(int32_t i = 0; i < taosArrayGetSize(ctx->pBlockList); ++i) {
    SSDataBlock* pb = taosArrayGetP(ctx->pBlockList, i);
    if (pb->info.blockId == target->dataBlockId) {
      index = i;
      break;
    }
  }

  if (index == -1) {
    sclError("column tupleId is too big, tupleId:%d, dataBlockNum:%d", target->dataBlockId, (int32_t)taosArrayGetSize(ctx->pBlockList));
    ctx->code = TSDB_CODE_QRY_INVALID_INPUT;
    return DEAL_RES_ERROR;
  }

  SSDataBlock *block = *(SSDataBlock **)taosArrayGet(ctx->pBlockList, index);

  if (target->slotId >= taosArrayGetSize(block->pDataBlock)) {
    sclError("target slot not exist, dataBlockId:%d, slotId:%d, dataBlockNum:%d", target->dataBlockId, target->slotId, (int32_t)taosArrayGetSize(block->pDataBlock));
    ctx->code = TSDB_CODE_QRY_INVALID_INPUT;
    return DEAL_RES_ERROR;
  }

  // if block->pDataBlock is not enough, there are problems if target->slotId bigger than the size of block->pDataBlock,
  SColumnInfoData *col = taosArrayGet(block->pDataBlock, target->slotId);

  SScalarParam *res = (SScalarParam *)taosHashGet(ctx->pRes, (void *)&target->pExpr, POINTER_BYTES);
  if (NULL == res) {
    sclError("no valid res in hash, node:%p, type:%d", target->pExpr, nodeType(target->pExpr));
    ctx->code = TSDB_CODE_QRY_APP_ERROR;
    return DEAL_RES_ERROR;
  }

  colDataAssign(col, res->columnData, res->numOfRows);
  block->info.rows = res->numOfRows;

  sclFreeParam(res);
  taosHashRemove(ctx->pRes, (void *)&target->pExpr, POINTER_BYTES);
  return DEAL_RES_CONTINUE;
}
D
dapan 已提交
911

D
dapan1121 已提交
912
EDealRes sclCalcWalker(SNode* pNode, void* pContext) {
D
dapan1121 已提交
913
  if (QUERY_NODE_VALUE == nodeType(pNode) || QUERY_NODE_NODE_LIST == nodeType(pNode) || QUERY_NODE_COLUMN == nodeType(pNode)|| QUERY_NODE_LEFT_VALUE == nodeType(pNode)) {
D
dapan1121 已提交
914
    return DEAL_RES_CONTINUE;
D
dapan1121 已提交
915
  }
D
dapan 已提交
916 917

  SScalarCtx *ctx = (SScalarCtx *)pContext;
D
dapan1121 已提交
918
  if (QUERY_NODE_FUNCTION == nodeType(pNode)) {
D
dapan 已提交
919
    return sclWalkFunction(pNode, ctx);
D
dapan1121 已提交
920
  }
D
dapan1121 已提交
921

D
dapan1121 已提交
922
  if (QUERY_NODE_LOGIC_CONDITION == nodeType(pNode)) {
D
dapan 已提交
923
    return sclWalkLogic(pNode, ctx);
D
dapan1121 已提交
924
  }
D
dapan1121 已提交
925

D
dapan1121 已提交
926
  if (QUERY_NODE_OPERATOR == nodeType(pNode)) {
D
dapan 已提交
927
    return sclWalkOperator(pNode, ctx);
D
dapan1121 已提交
928
  }
D
dapan1121 已提交
929

930 931 932
  if (QUERY_NODE_TARGET == nodeType(pNode)) {
    return sclWalkTarget(pNode, ctx);
  }
D
dapan1121 已提交
933

D
dapan 已提交
934
  sclError("invalid node type for scalar calculating, type:%d", nodeType(pNode));
D
dapan1121 已提交
935 936
  ctx->code = TSDB_CODE_QRY_INVALID_INPUT;
  return DEAL_RES_ERROR;
D
dapan1121 已提交
937 938 939 940 941 942 943 944
}

int32_t scalarCalculateConstants(SNode *pNode, SNode **pRes) {
  if (NULL == pNode) {
    SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }

  int32_t code = 0;
D
dapan 已提交
945
  SScalarCtx ctx = {0};
946
  ctx.pRes = taosHashInit(SCL_DEFAULT_OP_NUM, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
D
dapan 已提交
947
  if (NULL == ctx.pRes) {
948
    sclError("taosHashInit failed, num:%d", SCL_DEFAULT_OP_NUM);
D
dapan 已提交
949 950
    SCL_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }
951
  
X
Xiaoyu Wang 已提交
952
  nodesRewriteExprPostOrder(&pNode, sclConstantsRewriter, (void *)&ctx);
D
dapan 已提交
953
  SCL_ERR_JRET(ctx.code);
D
dapan1121 已提交
954 955
  *pRes = pNode;

D
dapan 已提交
956 957 958
_return:
  sclFreeRes(ctx.pRes);
  return code;
D
dapan1121 已提交
959 960
}

D
dapan 已提交
961
int32_t scalarCalculate(SNode *pNode, SArray *pBlockList, SScalarParam *pDst) {
D
dapan1121 已提交
962
  if (NULL == pNode || NULL == pBlockList) {
D
dapan1121 已提交
963 964 965 966
    SCL_ERR_RET(TSDB_CODE_QRY_INVALID_INPUT);
  }

  int32_t code = 0;
wmmhello's avatar
wmmhello 已提交
967
  SScalarCtx ctx = {.code = 0, .pBlockList = pBlockList, .param = pDst ? pDst->param : NULL};
968

969
  // TODO: OPT performance
970
  ctx.pRes = taosHashInit(SCL_DEFAULT_OP_NUM, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
D
dapan1121 已提交
971
  if (NULL == ctx.pRes) {
972
    sclError("taosHashInit failed, num:%d", SCL_DEFAULT_OP_NUM);
D
dapan1121 已提交
973 974
    SCL_ERR_RET(TSDB_CODE_QRY_OUT_OF_MEMORY);
  }
975
  
X
Xiaoyu Wang 已提交
976
  nodesWalkExprPostOrder(pNode, sclCalcWalker, (void *)&ctx);
D
dapan 已提交
977
  SCL_ERR_JRET(ctx.code);
D
dapan1121 已提交
978

D
dapan1121 已提交
979 980 981 982 983 984 985
  if (pDst) {
    SScalarParam *res = (SScalarParam *)taosHashGet(ctx.pRes, (void *)&pNode, POINTER_BYTES);
    if (NULL == res) {
      sclError("no valid res in hash, node:%p, type:%d", pNode, nodeType(pNode));
      SCL_ERR_JRET(TSDB_CODE_QRY_APP_ERROR);
    }
    
986 987
    colDataAssign(pDst->columnData, res->columnData, res->numOfRows);
    pDst->numOfRows = res->numOfRows;
D
dapan1121 已提交
988
    taosHashRemove(ctx.pRes, (void *)&pNode, POINTER_BYTES);
D
dapan1121 已提交
989
  }
D
dapan1121 已提交
990

D
dapan 已提交
991
_return:
D
dapan1121 已提交
992
  //nodesDestroyNode(pNode);
D
dapan 已提交
993 994
  sclFreeRes(ctx.pRes);
  return code;
D
dapan1121 已提交
995
}