taosdata / TDengine / #4894

coverage: 65.72% (+0.2%) from 65.57%

Source File
Press 'n' to go to next uncovered line, 'b' for previous

93.95
/source/libs/parser/src/parCalcConst.c
1 
/*
2 
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3 
 *
4 
 * This program is free software: you can use, redistribute, and/or modify
5 
 * it under the terms of the GNU Affero General Public License, version 3
6 
 * or later ("AGPL"), as published by the Free Software Foundation.
7 
 *
8 
 * This program is distributed in the hope that it will be useful, but WITHOUT
9 
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 
 * FITNESS FOR A PARTICULAR PURPOSE.
11 
 *
12 
 * You should have received a copy of the GNU Affero General Public License
13 
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
14 
 */
15 









16



#include "functionMgt.h"










17



#include "parInt.h"










18



#include "scalar.h"










19



#include "ttime.h"










20












21



typedef struct SNodeReplaceContext {










22



  SNode* pTarget;










23



  SNode* pNew;










24



  bool   replaced;










25



} SNodeReplaceContext;










26












27



typedef struct SCalcConstContext {










28



  SParseContext*      pParseCxt;










29



  SNodeReplaceContext replaceCxt;










30



  SMsgBuf             msgBuf;










31



  int32_t             code;










32



} SCalcConstContext;










33












34



static int32_t calcConstQuery(SCalcConstContext* pCxt, SNode* pStmt, bool subquery);










35












36



static int32_t calcConstSubquery(SCalcConstContext* pCxt, STempTableNode* pTempTable) {




12,589,001✔





37



  return calcConstQuery(pCxt, pTempTable->pSubquery, true);




12,589,001✔





38



}










39












40



static int32_t calcConstNode(SNode** pNode) {




803,149,013✔





41



  if (NULL == *pNode) {




803,149,013✔





42



    return TSDB_CODE_SUCCESS;




688,006,135✔





43



  }










44












45



  SNode*  pNew = NULL;




115,143,769✔





46



  int32_t code = scalarCalculateConstants(*pNode, &pNew);




115,143,558✔





47



  if (TSDB_CODE_SUCCESS == code) {




115,143,558✔





48



    *pNode = pNew;




115,143,769✔





49



  }










50



  return code;




115,143,558✔





51



}










52












53



static int32_t calcConstList(SNodeList* pList) {




732,544,250✔





54



  SNode* pNode = NULL;




732,544,250✔





55



  FOREACH(pNode, pList) {




845,802,469✔





56



    SNode*  pNew = NULL;




113,257,961✔





57



    int32_t code = scalarCalculateConstants(pNode, &pNew);




113,258,219✔





58



    if (TSDB_CODE_SUCCESS == code) {




113,257,961✔





59



      REPLACE_NODE(pNew);




113,257,961✔





60



    } else {










61



      return code;




×







62



    }










63



  }










64



  return TSDB_CODE_SUCCESS;




732,545,184✔





65



}










66












67



static bool isCondition(const SNode* pNode) {




144,588,258✔





68



  if (QUERY_NODE_OPERATOR == nodeType(pNode)) {




144,588,258✔





69



    return nodesIsComparisonOp((const SOperatorNode*)pNode);




116,264,052✔





70



  }










71



  return (QUERY_NODE_LOGIC_CONDITION == nodeType(pNode));




28,324,994✔





72



}










73












74



static int32_t rewriteIsTrue(SNode* pSrc, SNode** pIsTrue) {




281,565✔





75



  SOperatorNode* pOp = NULL;




281,565✔





76



  int32_t        code = nodesMakeNode(QUERY_NODE_OPERATOR, (SNode**)&pOp);




281,565✔





77



  if (NULL == pOp) {




281,565✔





78



    return code;




×







79



  }










80



  pOp->opType = OP_TYPE_IS_TRUE;




281,565✔





81



  pOp->pLeft = pSrc;




281,565✔





82



  pOp->node.resType.type = TSDB_DATA_TYPE_BOOL;




281,565✔





83



  pOp->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_BOOL].bytes;




281,565✔





84



  *pIsTrue = (SNode*)pOp;




281,565✔





85



  return TSDB_CODE_SUCCESS;




281,565✔





86



}










87












88



static EDealRes doRewriteCondition(SNode** pNode, void* pContext) {




454,627,892✔





89



  if (QUERY_NODE_LOGIC_CONDITION == nodeType(*pNode)) {




454,627,892✔





90



    SNode* pParam = NULL;




28,049,235✔





91



    FOREACH(pParam, ((SLogicConditionNode*)*pNode)->pParameterList) {




96,500,931✔





92



      if (!isCondition(pParam)) {




68,452,213✔





93



        SNode* pIsTrue = NULL;




265,564✔





94



        if (TSDB_CODE_SUCCESS != rewriteIsTrue(pParam, &pIsTrue)) {




265,564✔





95



          ((SCalcConstContext*)pContext)->code = TSDB_CODE_OUT_OF_MEMORY;




×







96



          return DEAL_RES_ERROR;




×







97



        }










98



        REPLACE_NODE(pIsTrue);




265,564✔





99



      }










100



    }










101



  }










102



  return DEAL_RES_CONTINUE;




454,629,783✔





103



}










104












105



static int32_t rewriteCondition(SCalcConstContext* pCxt, SNode** pNode) {




76,135,834✔





106



  if (!isCondition(*pNode)) {




76,135,834✔





107



    return rewriteIsTrue(*pNode, pNode);




16,001✔





108



  }










109



  nodesRewriteExprPostOrder(pNode, doRewriteCondition, pCxt);




76,120,316✔





110



  return pCxt->code;




76,120,316✔





111



}










112












113



static int32_t calcConstCondition(SCalcConstContext* pCxt, SNode** pNode) {




4,363,802✔





114



  int32_t code = rewriteCondition(pCxt, pNode);




4,363,802✔





115



  if (TSDB_CODE_SUCCESS == code) {




4,363,802✔





116



    code = calcConstNode(pNode);




4,363,802✔





117



  }










118



  return code;




4,363,802✔





119



}










120












121



static EDealRes rewriteCalcConstValue(SNode** pNode, void* pContext) {




20,508,831✔





122



  int32_t                code = TSDB_CODE_SUCCESS;




20,508,831✔





123



  int32_t                lino = 0;




20,508,831✔





124



  SCalcConstContext* pCtx = (SCalcConstContext*)pContext;




20,508,831✔





125












126



  if (QUERY_NODE_LOGIC_CONDITION == nodeType(*pNode)) {




20,508,831✔





127



    return DEAL_RES_CONTINUE;




3,553,601✔





128



  } else if (QUERY_NODE_OPERATOR == nodeType(*pNode)) {




16,955,230✔





129



    SOperatorNode* pOp = (SOperatorNode*)*pNode;




16,955,230✔





130



    if (OP_TYPE_EQUAL == pOp->opType && (TSDB_DATA_TYPE_TIMESTAMP == ((SExprNode*)pOp->pLeft)->resType.type || TSDB_DATA_TYPE_TIMESTAMP == ((SExprNode*)pOp->pRight)->resType.type)) {




16,955,230✔





131



      code = calcConstNode(&pOp->pLeft);




12,591,428✔





132



      if (TSDB_CODE_SUCCESS == code) {




12,591,428✔





133



        code = calcConstNode(&pOp->pRight);




12,591,428✔





134



      }










135



      










136



      goto _end;




12,591,428✔





137



    }










138



  }










139












140



  if (TSDB_CODE_SUCCESS == code) {




4,363,802✔





141



    code = calcConstCondition(pCtx, pNode);




4,363,802✔





142



  }










143



  










144



_end:




×







145












146



  if (code != TSDB_CODE_SUCCESS) {




16,955,230✔





147



    qError("%s failed at line %d since %s", __func__, lino, tstrerror(code));




×







148



    return DEAL_RES_ERROR;




×







149



  }










150



  










151



  return DEAL_RES_IGNORE_CHILD;




16,955,230✔





152



}










153












154



static int32_t rewriteConditionForFromTable(SCalcConstContext* pCxt, SNode* pTable) {




266,563,103✔





155



  int32_t code = TSDB_CODE_SUCCESS;




266,563,103✔





156



  switch (nodeType(pTable)) {




266,563,103✔





157



    case QUERY_NODE_TEMP_TABLE: {




12,589,001✔





158



      code = calcConstSubquery(pCxt, (STempTableNode*)pTable);




12,589,001✔





159



      break;




12,589,001✔





160



    }










161



    case QUERY_NODE_JOIN_TABLE: {




13,288,176✔





162



      SJoinTableNode* pJoin = (SJoinTableNode*)pTable;




13,288,176✔





163



      SNode* pCond = NULL;




13,288,176✔





164



      code = rewriteConditionForFromTable(pCxt, pJoin->pLeft);




13,288,176✔





165



      if (TSDB_CODE_SUCCESS == code) {




13,287,918✔





166



        code = rewriteConditionForFromTable(pCxt, pJoin->pRight);




13,287,918✔





167



      }










168



      if (TSDB_CODE_SUCCESS == code && NULL != pJoin->pOnCond) {




13,287,918✔





169



        code = rewriteCondition(pCxt, &pJoin->pOnCond);




12,897,377✔





170



      }










171



      if (TSDB_CODE_SUCCESS == code && NULL != pJoin->pOnCond) {




13,287,918✔





172



        nodesRewriteExpr(&pJoin->pOnCond, rewriteCalcConstValue, pCxt);




12,897,377✔





173



      }










174












175



/*










176



      if (TSDB_CODE_SUCCESS == code && NULL != pJoin->pOnCond) {










177



        code = nodesCloneNode(pJoin->pOnCond, &pCond);










178



      }










179



      if (TSDB_CODE_SUCCESS == code && NULL != pJoin->pOnCond) {










180



        code = calcConstCondition(pCxt, &pJoin->pOnCond);










181



      }










182



      if (TSDB_CODE_SUCCESS == code && pJoin->pOnCond && QUERY_NODE_VALUE == nodeType(pJoin->pOnCond)) {










183



        nodesDestroyNode(pJoin->pOnCond);










184



        pJoin->pOnCond = pCond;










185



        pCond = NULL;










186



      }










187



      nodesDestroyNode(pCond);










188



*/










189



      // todo empty table










190



      break;




13,287,918✔





191



    }










192



    default:




240,685,926✔





193



      break;




240,685,926✔





194



  }










195



  return code;




266,562,845✔





196



}










197












198



static int32_t calcConstFromTable(SCalcConstContext* pCxt, SNode* pTable) {




239,987,253✔





199



  return rewriteConditionForFromTable(pCxt, pTable);




239,987,253✔





200



}










201












202



static void rewriteConstCondition(SNode** pCond, bool* pAlwaysFalse) {




58,874,927✔





203



  if (QUERY_NODE_VALUE != nodeType(*pCond)) {




58,874,927✔





204



    return;




57,086,484✔





205



  }










206



  if (((SValueNode*)*pCond)->datum.b) {




1,788,443✔





207



    nodesDestroyNode(*pCond);




1,751,799✔





208



    *pCond = NULL;




1,751,799✔





209



  } else {










210



    *pAlwaysFalse = true;




36,644✔





211



  }










212



}










213












214



static int32_t calcConstStmtCondition(SCalcConstContext* pCxt, SNode** pCond, bool* pAlwaysFalse) {




478,091,384✔





215



  if (NULL == *pCond) {




478,091,384✔





216



    return TSDB_CODE_SUCCESS;




419,216,668✔





217



  }










218












219



  int32_t code = rewriteCondition(pCxt, pCond);




58,875,120✔





220



  if (TSDB_CODE_SUCCESS == code) {




58,875,138✔





221



    code = calcConstNode(pCond);




58,875,138✔





222



  }










223



  if (TSDB_CODE_SUCCESS == code) {




58,875,138✔





224



    rewriteConstCondition(pCond, pAlwaysFalse);




58,875,138✔





225



  }










226



  return code;




58,875,138✔





227



}










228












229



static EDealRes doFindAndReplaceNode(SNode** pNode, void* pContext) {




4,266✔





230



  SCalcConstContext* pCxt = pContext;




4,266✔





231



  if (pCxt->replaceCxt.pTarget == *pNode) {




4,266✔





232



    char aliasName[TSDB_COL_NAME_LEN] = {0};




4,266✔





233



    tstrncpy(aliasName, ((SExprNode*)*pNode)->aliasName, TSDB_COL_NAME_LEN);




4,266✔





234



    nodesDestroyNode(*pNode);




4,266✔





235



    *pNode = NULL;




4,266✔





236



    pCxt->code = nodesCloneNode(pCxt->replaceCxt.pNew, pNode);




4,266✔





237



    if (NULL == *pNode) {




4,266✔





238



      return DEAL_RES_ERROR;




×







239



    }










240



    tstrncpy(((SExprNode*)*pNode)->aliasName, aliasName, TSDB_COL_NAME_LEN);




4,266✔





241












242



    pCxt->replaceCxt.replaced = true;




4,266✔





243



    return DEAL_RES_END;




4,266✔





244



  }










245



  return DEAL_RES_CONTINUE;




×







246



}










247












248



static int32_t findAndReplaceNode(SCalcConstContext* pCxt, SNode** pRoot, SNode* pTarget, SNode* pNew, bool strict) {




4,266✔





249



  pCxt->replaceCxt.pNew = pNew;




4,266✔





250



  pCxt->replaceCxt.pTarget = pTarget;




4,266✔





251












252



  nodesRewriteExprPostOrder(pRoot, doFindAndReplaceNode, pCxt);




4,266✔





253



  if (TSDB_CODE_SUCCESS == pCxt->code && strict && !pCxt->replaceCxt.replaced) {




4,266✔





254



    parserError("target replace node not found, %p", pTarget);




×







255



    return TSDB_CODE_PAR_INTERNAL_ERROR;




×







256



  }










257



  return pCxt->code;




4,266✔





258



}










259












260



static int32_t calcConstProject(SCalcConstContext* pCxt, SNode* pProject, bool dual, SNode** pNew) {




660,701,223✔





261



  SArray* pAssociation = NULL;




660,701,223✔





262



  










263



  if (NULL != ((SExprNode*)pProject)->pAssociation) {




660,701,223✔





264



    pAssociation = taosArrayDup(((SExprNode*)pProject)->pAssociation, NULL);




24,877,642✔





265



    if (NULL == pAssociation) {




24,877,642✔





266



      return terrno;




×







267



    }










268



  }










269












270



  char    aliasName[TSDB_COL_NAME_LEN] = {0}, srcTable[TSDB_TABLE_NAME_LEN] = {0};




660,701,950✔





271



  int32_t code = TSDB_CODE_SUCCESS;




660,701,953✔





272



  if (dual) {




660,701,953✔





273



    code = scalarCalculateConstantsFromDual(pProject, pNew);




2,944,379✔





274



  } else {










275



    code = scalarCalculateConstants(pProject, pNew);




657,757,574✔





276



  }










277



  if (TSDB_CODE_SUCCESS == code) {




660,702,392✔





278



    if (QUERY_NODE_VALUE == nodeType(*pNew) && NULL != pAssociation) {




660,698,166✔





279



      int32_t size = taosArrayGetSize(pAssociation);




1,267,838✔





280



      for (int32_t i = 0; i < size; ++i) {




2,537,292✔





281



        SAssociationNode* pAssNode = taosArrayGet(pAssociation, i);




1,269,454✔





282



        SNode**           pCol = pAssNode->pPlace;




1,269,454✔





283



        if (((SExprNode*)pAssNode->pAssociationNode)->joinSrc) {




1,269,454✔





284



          //((SExprNode*)pAssNode->pAssociationNode)->constValue = true;










285



          continue;




×







286



        }










287



        










288



        if (*pCol == pAssNode->pAssociationNode) {




1,269,454✔





289



          tstrncpy(aliasName, ((SExprNode*)*pCol)->aliasName, TSDB_COL_NAME_LEN);




1,265,188✔





290



          if (QUERY_NODE_COLUMN == nodeType(*pCol)) {




1,265,188✔





291



            tstrncpy(srcTable, ((SColumnNode*)*pCol)->tableAlias, TSDB_TABLE_NAME_LEN);




1,265,188✔





292



          }










293



          SArray* pOrigAss = NULL;




1,265,188✔





294



          TSWAP(((SExprNode*)*pCol)->pAssociation, pOrigAss);




1,265,188✔





295



          nodesDestroyNode(*pCol);




1,265,188✔





296



          *pCol = NULL;




1,265,188✔





297



          code = nodesCloneNode(*pNew, pCol);




1,265,188✔





298



          if (TSDB_CODE_SUCCESS == code) {




1,265,188✔





299



            tstrncpy(((SExprNode*)*pCol)->aliasName, aliasName, TSDB_COL_NAME_LEN);




1,265,188✔





300



            if (srcTable[0]) {




1,265,188✔





301



              tstrncpy(((SExprNode*)*pCol)->srcTable, srcTable, TSDB_TABLE_NAME_LEN);




1,265,188✔





302



            }










303



            TSWAP(pOrigAss, ((SExprNode*)*pCol)->pAssociation);




1,265,188✔





304



          }










305



          taosArrayDestroy(pOrigAss);




1,265,188✔





306



          if (TSDB_CODE_SUCCESS != code) {




1,265,188✔





307



            break;




×







308



          }










309



        } else {










310



          code = findAndReplaceNode(pCxt, pCol, pAssNode->pAssociationNode, *pNew, true);




4,266✔





311



          if (TSDB_CODE_SUCCESS != code) {




4,266✔





312



            break;




×







313



          }










314



        }










315



      }










316



    }










317



  }










318



  taosArrayDestroy(pAssociation);




660,702,392✔





319



  return code;




660,702,392✔





320



}










321












322



typedef struct SIsUselessColCtx {










323



  bool isUseless;










324



} SIsUselessColCtx;










325












326



EDealRes checkUselessCol(SNode* pNode, void* pContext) {




95,086,858✔





327



  SIsUselessColCtx* ctx = (SIsUselessColCtx*)pContext;




95,086,858✔





328



  if (QUERY_NODE_FUNCTION == nodeType(pNode) && !fmIsScalarFunc(((SFunctionNode*)pNode)->funcId) &&




95,086,858✔





329



      !fmIsPseudoColumnFunc(((SFunctionNode*)pNode)->funcId)) {




7,629,529✔





330



    ctx->isUseless = false;




6,359,520✔





331



    return DEAL_RES_END;




6,359,520✔





332



  }










333












334



  return DEAL_RES_CONTINUE;




88,727,338✔





335



}










336












337



static bool isUselessCol(SExprNode* pProj) {




79,462,517✔





338



  SIsUselessColCtx ctx = {.isUseless = true};




79,462,517✔





339



  nodesWalkExpr((SNode*)pProj, checkUselessCol, (void*)&ctx);




79,462,517✔





340



  if (!ctx.isUseless) {




79,462,517✔





341



    return false;




6,359,520✔





342



  }










343



  return NULL == ((SExprNode*)pProj)->pAssociation;




73,102,997✔





344



}










345












346



static int32_t createConstantValue(SValueNode** ppNode) {




110,334✔





347



  SValueNode* pVal = NULL;




110,334✔





348



  int32_t     code = nodesMakeNode(QUERY_NODE_VALUE, (SNode**)&pVal);




110,334✔





349



  if (NULL == pVal) {




110,334✔





350



    return code;




×







351



  }










352



  pVal->node.resType.type = TSDB_DATA_TYPE_INT;




110,334✔





353



  pVal->node.resType.bytes = tDataTypes[TSDB_DATA_TYPE_INT].bytes;




110,334✔





354



  const int32_t val = 1;




110,334✔





355



  code = nodesSetValueNodeValue(pVal, (void*)&val);




110,334✔





356



  if (TSDB_CODE_SUCCESS == code) {




110,334✔





357



    pVal->translate = true;




110,334✔





358



    *ppNode = pVal;




110,334✔





359



  } else {










360



    nodesDestroyNode((SNode*)pVal);




×







361



  }










362



  return code;




110,334✔





363



}










364












365



static int32_t calcConstProjections(SCalcConstContext* pCxt, SSelectStmt* pSelect, bool subquery) {




240,438,770✔





366



  SNode* pProj = NULL;




240,438,770✔





367



  WHERE_EACH(pProj, pSelect->pProjectionList) {




952,358,658✔





368



    if (subquery && !pSelect->isDistinct && !pSelect->tagScan && isUselessCol((SExprNode*)pProj) && (pSelect->pFromTable && nodeType(pSelect->pFromTable) != QUERY_NODE_VIRTUAL_TABLE)) {




711,923,478✔





369



      ERASE_NODE(pSelect->pProjectionList);




51,221,725✔





370



      continue;




51,221,725✔





371



    }










372



    SNode*  pNew = NULL;




660,701,753✔





373



    int32_t code = calcConstProject(pCxt, pProj, (NULL == pSelect->pFromTable), &pNew);




660,701,495✔





374



    if (TSDB_CODE_SUCCESS == code) {




660,701,879✔





375



      REPLACE_NODE(pNew);




660,697,653✔





376



    } else {










377



      return code;




4,226✔





378



    }










379



    WHERE_NEXT;




660,697,653✔





380



  }










381



  if (0 == LIST_LENGTH(pSelect->pProjectionList)) {




240,435,241✔





382



    SValueNode* pVal = NULL;




106,078✔





383



    int32_t     code = createConstantValue(&pVal);




105,789✔





384



    if (TSDB_CODE_SUCCESS == code) {




105,789✔





385



      return nodesListStrictAppend(pSelect->pProjectionList, (SNode*)pVal);




105,789✔





386



    }










387



  }










388



  return TSDB_CODE_SUCCESS;




240,328,952✔





389



}










390












391



static int32_t calcConstGroupBy(SCalcConstContext* pCxt, SSelectStmt* pSelect) {




238,242,290✔





392



  int32_t code = calcConstList(pSelect->pGroupByList);




238,242,290✔





393



  if (TSDB_CODE_SUCCESS == code) {




238,242,548✔





394



    SNode* pNode = NULL;




238,242,504✔





395



    FOREACH(pNode, pSelect->pGroupByList) {




263,386,891✔





396



      bool   hasNotValue = false;




25,144,387✔





397



      SNode* pGroupPara = NULL;




25,144,387✔





398



      FOREACH(pGroupPara, ((SGroupingSetNode*)pNode)->pParameterList) {




25,923,439✔





399



        if (QUERY_NODE_VALUE != nodeType(pGroupPara)) {




25,144,387✔





400



          hasNotValue = true;




24,365,335✔





401



          break;




24,365,335✔





402



        }










403



      }










404



      if (!hasNotValue) {




25,144,387✔





405



        if (pSelect->hasAggFuncs) {




779,052✔





406



          ERASE_NODE(pSelect->pGroupByList);




37,956✔





407



        } else {










408



          if (!cell->pPrev && !cell->pNext) continue;




741,096✔





409



          ERASE_NODE(pSelect->pGroupByList);




×







410



        }










411



      }










412



    }










413



  }










414



  return code;




238,242,562✔





415



}










416












417



static int32_t calcConstSelectWithoutFrom(SCalcConstContext* pCxt, SSelectStmt* pSelect, bool subquery) {




2,196,187✔





418



  return calcConstProjections(pCxt, pSelect, subquery);




2,196,187✔





419



}










420












421



static int32_t calcConstSelectFrom(SCalcConstContext* pCxt, SSelectStmt* pSelect, bool subquery) {




238,295,757✔





422



  int32_t code = calcConstFromTable(pCxt, pSelect->pFromTable);




238,295,757✔





423



  if (TSDB_CODE_SUCCESS == code && QUERY_NODE_TEMP_TABLE == nodeType(pSelect->pFromTable) &&




238,295,496✔





424



      ((STempTableNode*)pSelect->pFromTable)->pSubquery != NULL &&




12,194,471✔





425



      QUERY_NODE_SELECT_STMT == nodeType(((STempTableNode*)pSelect->pFromTable)->pSubquery) &&




12,194,471✔





426



      ((SSelectStmt*)((STempTableNode*)pSelect->pFromTable)->pSubquery)->isEmptyResult) {




10,663,250✔





427



    pSelect->isEmptyResult = true;




52,211✔





428



    return code;




52,211✔





429



  }










430



  if (pSelect->mixSysTableAndActualTable) {




238,243,027✔





431



    return code;




705✔





432



  }










433



  if (TSDB_CODE_SUCCESS == code) {




238,241,642✔





434



    code = calcConstProjections(pCxt, pSelect, subquery);




238,242,885✔





435



  }










436



  if (TSDB_CODE_SUCCESS == code) {




238,241,748✔





437



    code = calcConstStmtCondition(pCxt, &pSelect->pWhere, &pSelect->isEmptyResult);




238,242,991✔





438



  }










439



  if (TSDB_CODE_SUCCESS == code) {




238,242,808✔





440



    code = calcConstList(pSelect->pPartitionByList);




238,242,808✔





441



  }










442



  if (TSDB_CODE_SUCCESS == code) {




238,242,293✔





443



    code = calcConstNode(&pSelect->pWindow);




238,242,293✔





444



  }










445



  if (TSDB_CODE_SUCCESS == code) {




238,242,501✔





446



    code = calcConstGroupBy(pCxt, pSelect);




238,242,501✔





447



  }










448



  if (TSDB_CODE_SUCCESS == code) {




238,242,501✔





449



    code = calcConstStmtCondition(pCxt, &pSelect->pHaving, &pSelect->isEmptyResult);




238,242,759✔





450



  }










451



  if (TSDB_CODE_SUCCESS == code) {




238,242,501✔





452



    code = calcConstList(pSelect->pOrderByList);




238,242,501✔





453



  }










454



  if (TSDB_CODE_SUCCESS == code) {




238,242,715✔





455



    code = calcConstNode(&pSelect->pFill);




238,242,715✔





456



  }










457



  if (TSDB_CODE_SUCCESS == code) {




238,242,991✔





458



    code = calcConstNode(&pSelect->pTimeRange);




238,242,991✔





459



  }










460



  return code;




238,242,991✔





461



}










462












463












464



static bool isEmptyResultQuery(SNode* pStmt) {




304,885,124✔





465



  bool isEmptyResult = false;




304,885,124✔





466



  switch (nodeType(pStmt)) {




304,885,124✔





467



    case QUERY_NODE_SELECT_STMT:




209,996,972✔





468



      isEmptyResult = ((SSelectStmt*)pStmt)->isEmptyResult;




209,996,972✔





469



      break;




209,997,213✔





470



    case QUERY_NODE_EXPLAIN_STMT:




4,806,415✔





471



      isEmptyResult = isEmptyResultQuery(((SExplainStmt*)pStmt)->pQuery);




4,806,415✔





472



      break;




4,808,131✔





473



    case QUERY_NODE_SET_OPERATOR: {




16,059,522✔





474



      SSetOperator* pSetOp = (SSetOperator*)pStmt;




16,059,522✔





475



      isEmptyResult = isEmptyResultQuery(pSetOp->pLeft);




16,059,522✔





476



      if (isEmptyResult) {




16,059,522✔





477



        isEmptyResult = isEmptyResultQuery(pSetOp->pRight);




595✔





478



      }










479



      break;




16,059,522✔





480



    }










481



    case QUERY_NODE_DELETE_STMT:




1,609,416✔





482



      isEmptyResult = ((SDeleteStmt*)pStmt)->deleteZeroRows;




1,609,416✔





483



      break;




1,606,283✔





484



    default:




72,417,119✔





485



      break;




72,417,119✔





486



  }










487



  return isEmptyResult;




304,888,268✔





488



}










489












490



static int32_t calcSubQueries(SCalcConstContext* pCxt, SNodeList* pSubQueries) {




30,377,363✔





491



  int32_t code = TSDB_CODE_SUCCESS;




30,377,363✔





492



  SNode* pNode = NULL;




30,377,363✔





493



  FOREACH(pNode, pSubQueries) {




120,966,427✔





494



    code = calcConstQuery(pCxt, pNode, false);




90,589,064✔





495



    if (code) {




90,589,064✔





496



      break;




×







497



    }










498



    if (isEmptyResultQuery(pNode) && nodesIsScalarSubQuery(pNode)) {




90,589,064✔





499



      // TODO










500



/*










501



      parserError("%" PRIx64 " scalar subquery got empty result", pCxt->pParseCxt->requestId);










502



      code = TSDB_CODE_PAR_INVALID_SCALAR_SUBQ_RES_ROWS;










503



*/










504



      break;




×







505



    }










506



  }










507












508



  return code;




30,377,363✔





509



}










510












511



static int32_t calcConstSelect(SCalcConstContext* pCxt, SSelectStmt* pSelect, bool subquery) {




240,491,944✔





512



  int32_t code = TSDB_CODE_SUCCESS;




240,491,944✔





513



  










514



  if (NULL == pSelect->pFromTable) {




240,491,944✔





515



    code = calcConstSelectWithoutFrom(pCxt, pSelect, subquery);




2,196,187✔





516



  } else {










517



    code = calcConstSelectFrom(pCxt, pSelect, subquery);




238,295,968✔





518



  }










519












520



  if (TSDB_CODE_SUCCESS == code && pSelect->pSubQueries && pSelect->pSubQueries->length > 0) {




240,491,349✔





521



    code = calcSubQueries(pCxt, pSelect->pSubQueries);




29,149,328✔





522



  }










523












524



  return code;




240,491,621✔





525



}










526












527



static int32_t calcConstDelete(SCalcConstContext* pCxt, SDeleteStmt* pDelete) {




1,606,283✔





528



  int32_t code = calcConstFromTable(pCxt, pDelete->pFromTable);




1,606,283✔





529



  if (TSDB_CODE_SUCCESS == code) {




1,606,283✔





530



    code = calcConstStmtCondition(pCxt, &pDelete->pWhere, &pDelete->deleteZeroRows);




1,606,283✔





531



  }










532



  if (code == TSDB_CODE_SUCCESS && pDelete->timeRange.skey > pDelete->timeRange.ekey) {




1,606,283✔





533



    pDelete->deleteZeroRows = true;




209,153✔





534



  }










535



  return code;




1,606,283✔





536



}










537












538



static int32_t calcConstInsert(SCalcConstContext* pCxt, SInsertStmt* pInsert) {




85,227✔





539



  int32_t code = calcConstFromTable(pCxt, pInsert->pTable);




85,227✔





540



  if (TSDB_CODE_SUCCESS == code) {




85,227✔





541



    code = calcConstQuery(pCxt, pInsert->pQuery, false);




85,227✔





542



  }










543



  return code;




85,227✔





544



}










545












546



static SNodeList* getChildProjection(SNode* pStmt) {




2,394,814✔





547



  switch (nodeType(pStmt)) {




2,394,814✔





548



    case QUERY_NODE_SELECT_STMT:




2,384,130✔





549



      return ((SSelectStmt*)pStmt)->pProjectionList;




2,384,130✔





550



    case QUERY_NODE_SET_OPERATOR:




10,684✔





551



      return ((SSetOperator*)pStmt)->pProjectionList;




10,684✔





552



    default:




×







553



      break;




×







554



  }










555



  return NULL;




×







556



}










557












558



static void eraseSetOpChildProjection(SSetOperator* pSetOp, int32_t index) {




595,828✔





559



  SNodeList* pLeftProjs = getChildProjection(pSetOp->pLeft);




595,828✔





560



  (void)nodesListErase(pLeftProjs, nodesListGetCell(pLeftProjs, index));




595,828✔





561



  if (QUERY_NODE_SET_OPERATOR == nodeType(pSetOp->pLeft)) {




595,828✔





562



    eraseSetOpChildProjection((SSetOperator*)pSetOp->pLeft, index);




5,272✔





563



  }










564



  SNodeList* pRightProjs = getChildProjection(pSetOp->pRight);




595,828✔





565



  (void)nodesListErase(pRightProjs, nodesListGetCell(pRightProjs, index));




595,828✔





566



  if (QUERY_NODE_SET_OPERATOR == nodeType(pSetOp->pRight)) {




595,828✔





567



    eraseSetOpChildProjection((SSetOperator*)pSetOp->pRight, index);




×







568



  }










569



}




595,828✔





570












571



typedef struct SNotRefByOrderByCxt {










572



  SColumnNode* pCol;










573



  bool         hasThisCol;










574



} SNotRefByOrderByCxt;










575












576



static EDealRes notRefByOrderByImpl(SNode* pNode, void* pContext) {




3,837,594✔





577



  if (QUERY_NODE_COLUMN == nodeType(pNode)) {




3,837,594✔





578



    SNotRefByOrderByCxt* pCxt = (SNotRefByOrderByCxt*)pContext;




1,918,797✔





579



    if (nodesEqualNode((SNode*)pCxt->pCol, pNode)) {




1,918,797✔





580



      pCxt->hasThisCol = true;




621,636✔





581



      return DEAL_RES_END;




621,636✔





582



    }










583



  }










584



  return DEAL_RES_CONTINUE;




3,215,958✔





585



}










586












587



static bool notRefByOrderBy(SColumnNode* pCol, SNodeList* pOrderByList) {




2,292,642✔





588



  SNotRefByOrderByCxt cxt = {.pCol = pCol, .hasThisCol = false};




2,292,642✔





589



  nodesWalkExprs(pOrderByList, notRefByOrderByImpl, &cxt);




2,292,642✔





590



  return !cxt.hasThisCol;




2,292,642✔





591



}










592












593



static bool isDistinctSubQuery(SNode* pNode) {




1,225,720✔





594



  if (NULL == pNode) {




1,225,720✔





595



    return false;




×







596



  }










597



  switch (nodeType(pNode)) {




1,225,720✔





598



    case QUERY_NODE_SELECT_STMT:




1,203,696✔





599



      return ((SSelectStmt*)pNode)->isDistinct;




1,203,696✔





600



    case QUERY_NODE_SET_OPERATOR:




22,024✔





601



      return isDistinctSubQuery((((SSetOperator*)pNode)->pLeft)) || isDistinctSubQuery((((SSetOperator*)pNode)->pLeft));




22,024✔





602



    default:




×







603



      break;




×







604



  }










605



  return false;




×







606



}










607












608



static bool isSetUselessCol(SSetOperator* pSetOp, int32_t index, SExprNode* pProj) {




1,671,006✔





609



  if (!isUselessCol(pProj)) {




1,671,006✔





610



    return false;




1,058,964✔





611



  }










612












613



  SNodeList* pLeftProjs = getChildProjection(pSetOp->pLeft);




612,042✔





614



  if (!isUselessCol((SExprNode*)nodesListGetNode(pLeftProjs, index)) || isDistinctSubQuery(pSetOp->pLeft)) {




612,042✔





615



    return false;




20,926✔





616



  }










617












618



  SNodeList* pRightProjs = getChildProjection(pSetOp->pRight);




591,116✔





619



  if (!isUselessCol((SExprNode*)nodesListGetNode(pRightProjs, index)) || isDistinctSubQuery(pSetOp->pLeft)) {




591,116✔





620



    return false;




560✔





621



  }










622












623



  return true;




590,556✔





624



}










625












626



static int32_t calcConstSetOpProjections(SCalcConstContext* pCxt, SSetOperator* pSetOp, bool subquery) {




17,816,869✔





627



  if (subquery && pSetOp->opType == SET_OP_TYPE_UNION) {




17,816,869✔





628



    return TSDB_CODE_SUCCESS;




755,313✔





629



  }










630



  int32_t index = 0;




17,061,556✔





631



  SNode*  pProj = NULL;




17,061,556✔





632



  WHERE_EACH(pProj, pSetOp->pProjectionList) {




38,149,574✔





633



    if (subquery && notRefByOrderBy((SColumnNode*)pProj, pSetOp->pOrderByList) &&




22,759,024✔





634



        isSetUselessCol(pSetOp, index, (SExprNode*)pProj)) {




1,671,006✔





635



      ERASE_NODE(pSetOp->pProjectionList);




590,556✔





636



      eraseSetOpChildProjection(pSetOp, index);




590,556✔





637



      continue;




590,556✔





638



    }










639



    ++index;




20,497,462✔





640



    WHERE_NEXT;




20,497,462✔





641



  }










642



  if (0 == LIST_LENGTH(pSetOp->pProjectionList)) {




17,061,556✔





643



    SValueNode* pVal = NULL;




4,545✔





644



    int32_t     code = createConstantValue(&pVal);




4,545✔





645



    if (TSDB_CODE_SUCCESS == code) {




4,545✔





646



      return nodesListStrictAppend(pSetOp->pProjectionList, (SNode*)pVal);




4,545✔





647



    }










648



  }










649



  return TSDB_CODE_SUCCESS;




17,057,011✔





650



}










651












652



static int32_t calcConstSetOperator(SCalcConstContext* pCxt, SSetOperator* pSetOp, bool subquery) {




17,816,869✔





653



  int32_t code = calcConstSetOpProjections(pCxt, pSetOp, subquery);




17,816,869✔





654



  if (TSDB_CODE_SUCCESS == code) {




17,816,869✔





655



    code = calcConstQuery(pCxt, pSetOp->pLeft, false);




17,816,869✔





656



  }










657



  if (TSDB_CODE_SUCCESS == code) {




17,816,869✔





658



    code = calcConstQuery(pCxt, pSetOp->pRight, false);




17,816,869✔





659



  }










660



  if (TSDB_CODE_SUCCESS == code) {




17,816,869✔





661



    code = calcConstList(pSetOp->pOrderByList);




17,816,869✔





662



  }










663



  if (TSDB_CODE_SUCCESS == code && pSetOp->pSubQueries && pSetOp->pSubQueries->length > 0) {




17,816,869✔





664



    code = calcSubQueries(pCxt, pSetOp->pSubQueries);




1,228,035✔





665



  }










666



  










667



  return code;




17,816,869✔





668



}










669












670



static int32_t calcConstQuery(SCalcConstContext* pCxt, SNode* pStmt, bool subquery) {




337,139,355✔





671



  int32_t code = TSDB_CODE_SUCCESS;




337,139,355✔





672



  switch (nodeType(pStmt)) {




337,139,355✔





673



    case QUERY_NODE_SELECT_STMT:




240,491,941✔





674



      code = calcConstSelect(pCxt, (SSelectStmt*)pStmt, subquery);




240,491,941✔





675



      break;




240,491,832✔





676



    case QUERY_NODE_EXPLAIN_STMT:




4,806,415✔





677



      code = calcConstQuery(pCxt, ((SExplainStmt*)pStmt)->pQuery, subquery);




4,806,415✔





678



      break;




4,808,180✔





679



    case QUERY_NODE_SET_OPERATOR: {




17,816,869✔





680



      code = calcConstSetOperator(pCxt, (SSetOperator*)pStmt, subquery);




17,816,869✔





681



      break;




17,816,869✔





682



    }










683



    case QUERY_NODE_DELETE_STMT:




1,606,283✔





684



      code = calcConstDelete(pCxt, (SDeleteStmt*)pStmt);




1,606,283✔





685



      break;




1,606,283✔





686



    case QUERY_NODE_INSERT_STMT:




87,138✔





687



      code = calcConstInsert(pCxt, (SInsertStmt*)pStmt);




87,138✔





688



      break;




85,227✔





689



    default:




72,332,900✔





690



      break;




72,332,900✔





691



  }










692



  return code;




337,141,291✔





693



}










694












695












696



static void resetProjectNullTypeImpl(SNodeList* pProjects) {




120,356,972✔





697



  SNode* pProj = NULL;




120,356,972✔





698



  FOREACH(pProj, pProjects) {




632,029,683✔





699



    SExprNode* pExpr = (SExprNode*)pProj;




511,673,912✔





700



    if (TSDB_DATA_TYPE_NULL == pExpr->resType.type) {




511,673,912✔





701



      pExpr->resType.type = TSDB_DATA_TYPE_VARCHAR;




455,498✔





702



      pExpr->resType.bytes = VARSTR_HEADER_SIZE;




455,498✔





703



      if (QUERY_NODE_VALUE == nodeType(pExpr)) {




455,498✔





704



        ((SValueNode*)pExpr)->isNull = true;




19,871✔





705



      }










706



    }










707



  }










708



}




120,356,967✔





709












710



static void resetProjectNullType(SNode* pStmt) {




199,174,952✔





711



  switch (nodeType(pStmt)) {




199,174,952✔





712



    case QUERY_NODE_SELECT_STMT:




117,485,351✔





713



      resetProjectNullTypeImpl(((SSelectStmt*)pStmt)->pProjectionList);




117,485,351✔





714



      break;




117,484,589✔





715



    case QUERY_NODE_SET_OPERATOR: {




2,872,794✔





716



      SSetOperator* pSetOp = (SSetOperator*)pStmt;




2,872,794✔





717



      resetProjectNullTypeImpl(pSetOp->pProjectionList);




2,872,794✔





718



      if (pSetOp->pLeft)




2,872,378✔





719



        resetProjectNullType(pSetOp->pLeft);




2,872,378✔





720



      if (pSetOp->pRight)




2,872,378✔





721



        resetProjectNullType(pSetOp->pRight);




2,872,378✔





722



      break;




2,872,378✔





723



    }










724



    default:




78,818,560✔





725



      break;




78,818,560✔





726



  }










727



}




199,175,527✔





728












729



static int32_t rewriteScalarSubQResValue(SParseContext* pParseCxt, SQuery* pQuery) {




193,428,877✔





730



  // TODO










731



  return TSDB_CODE_SUCCESS;




193,428,877✔





732



}










733












734



int32_t calculateConstant(SParseContext* pParseCxt, SQuery* pQuery) {




193,436,968✔





735



  SCalcConstContext cxt = {.pParseCxt = pParseCxt,




193,436,968✔





736



                           .msgBuf.buf = pParseCxt->pMsg,




193,444,240✔





737



                           .msgBuf.len = pParseCxt->msgLen,




193,439,204✔





738



                           .code = TSDB_CODE_SUCCESS};










739












740



  int32_t code = TSDB_CODE_SUCCESS;




193,443,221✔





741












742



  if (TSDB_CODE_SUCCESS == code) {




193,443,221✔





743



    code = calcConstQuery(&cxt, pQuery->pRoot, false);




193,440,104✔





744



  }










745



  if (TSDB_CODE_SUCCESS == code) {




193,438,580✔





746



    resetProjectNullType(pQuery->pRoot);




193,434,422✔





747



    if (isEmptyResultQuery(pQuery->pRoot)) {




193,431,688✔





748



      pQuery->execMode = QUERY_EXEC_MODE_EMPTY_RESULT;




382,122✔





749



    }










750



    code = rewriteScalarSubQResValue(pParseCxt, pQuery);




193,432,067✔





751



  }










752



  










753



  return code;




193,434,696✔





754



}
























    

  • 

    •