fix: distinct column removed in union all issue

This commit is contained in:
dapan1121 2023-06-16 18:43:36 +08:00
parent bb8840d4ec
commit ad9b26abac
1 changed files with 17 additions and 2 deletions

View File

@ -372,18 +372,33 @@ static bool notRefByOrderBy(SColumnNode* pCol, SNodeList* pOrderByList) {
return !cxt.hasThisCol;
}
static bool isDistinctSubQuery(SNode* pNode) {
if (NULL == pNode) {
return false;
}
switch (nodeType(pNode)) {
case QUERY_NODE_SELECT_STMT:
return ((SSelectStmt*)pNode)->isDistinct;
case QUERY_NODE_SET_OPERATOR:
return isDistinctSubQuery((((SSetOperator*)pNode)->pLeft)) || isDistinctSubQuery((((SSetOperator*)pNode)->pLeft));
default:
break;
}
return false;
}
static bool isSetUselessCol(SSetOperator* pSetOp, int32_t index, SExprNode* pProj) {
if (!isUselessCol(pProj)) {
return false;
}
SNodeList* pLeftProjs = getChildProjection(pSetOp->pLeft);
if (!isUselessCol((SExprNode*)nodesListGetNode(pLeftProjs, index))) {
if (!isUselessCol((SExprNode*)nodesListGetNode(pLeftProjs, index)) || isDistinctSubQuery(pSetOp->pLeft)) {
return false;
}
SNodeList* pRightProjs = getChildProjection(pSetOp->pRight);
if (!isUselessCol((SExprNode*)nodesListGetNode(pRightProjs, index))) {
if (!isUselessCol((SExprNode*)nodesListGetNode(pRightProjs, index)) || isDistinctSubQuery(pSetOp->pLeft)) {
return false;
}