database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[fix](nereids) the order exprs in sort node should be slotRef in its tupleDesc (#16363)

Browse Source
This commit is contained in:
starocean999
2023-02-03 13:28:08 +08:00
committed by GitHub
parent a9177569c6
commit dfb610d7ec
2 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
fe/fe-core/src/main/java/org/apache/doris/nereids/glue/translator/PhysicalPlanTranslator.java
View File

@ -646,14 +646,10 @@ public class PhysicalPlanTranslator extends DefaultPlanVisitor<PlanFragment, Pla
outputList.forEach(k -> {
sortTupleOutputList.add(ExpressionTranslator.translate(k, context));
});
// 2. Generate new Tuple
TupleDescriptor tupleDesc = generateTupleDesc(outputList, orderKeyList, context, null);
// 3. Get current slotRef
// 2. Generate new Tuple and get current slotRef for newOrderingExprList
List<Expr> newOrderingExprList = Lists.newArrayList();
orderKeyList.forEach(k -> {
newOrderingExprList.add(ExpressionTranslator.translate(k.getExpr(), context));
});
// 4. fill in SortInfo members
TupleDescriptor tupleDesc = generateTupleDesc(outputList, orderKeyList, newOrderingExprList, context, null);
// 3. fill in SortInfo members
SortInfo sortInfo = new SortInfo(newOrderingExprList, ascOrderList, nullsFirstParamList, tupleDesc);
PlanNode childNode = childFragment.getPlanRoot();
SortNode sortNode = new SortNode(context.nextPlanNodeId(), childNode, sortInfo, true);
@ -1445,6 +1441,7 @@ public class PhysicalPlanTranslator extends DefaultPlanVisitor<PlanFragment, Pla
}
private TupleDescriptor generateTupleDesc(List<Slot> slotList, List<OrderKey> orderKeyList,
List<Expr> newOrderingExprList,
PlanTranslatorContext context, Table table) {
TupleDescriptor tupleDescriptor = context.generateTupleDesc();
Set<ExprId> alreadyExists = Sets.newHashSet();
@ -1458,9 +1455,11 @@ public class PhysicalPlanTranslator extends DefaultPlanVisitor<PlanFragment, Pla
}
// TODO: trick here, we need semanticEquals to remove redundant expression
if (alreadyExists.contains(slotReference.getExprId())) {
newOrderingExprList.add(context.findSlotRef(slotReference.getExprId()));
continue;
}
context.createSlotDesc(tupleDescriptor, slotReference);
newOrderingExprList.add(context.findSlotRef(slotReference.getExprId()));
alreadyExists.add(slotReference.getExprId());
}
for (Slot slot : slotList) {