mirror/dble
1
0
Fork 0
mirror of https://github.com/actiontech/dble.git synced 2026-02-14 01:29:01 -06:00
Code Issues Packages Projects Releases Wiki Activity

fix er relationship error inner 1816 (#3315)

Browse Source 
* fix er relationship error inner 1816

fix

fix

* modify parameter name
This commit is contained in:
ylinzhu
2022-08-01 10:23:32 +08:00
committed by GitHub
parent f5c7378620
commit a2ff1bc91e
1 changed files with 42 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
src/main/java/com/actiontech/dble/plan/optimizer/JoinStrategyChooser.java
View File

@@ -15,6 +15,7 @@ import com.actiontech.dble.plan.node.TableNode;
import com.actiontech.dble.plan.util.PlanUtil;
import com.actiontech.dble.util.StringUtil;
import com.google.common.base.Strings;
import org.jetbrains.annotations.Nullable;
import java.util.*;
@@ -149,25 +150,14 @@ public class JoinStrategyChooser {
if (isSmallTable((TableNode) node) || canDoAsMerge(joinNode) || !innerJoin) {
return;
}
List<ItemFuncEqual> joinFilter = joinNode.getJoinFilter();
for (ItemFuncEqual itemFuncEqual : joinFilter) {
List<Item> arguments = itemFuncEqual.arguments();
Item item = arguments.stream().filter(argument -> !StringUtil.equals(getTableName((TableNode) node), argument.getTableName())).findFirst().get();
PlanNode dependNode = nodeMap.get(item.getTableName());
if (isSmallTable((TableNode) dependNode) && innerJoin) {
joinNode.setStrategy(Strategy.ALWAYS_NEST_LOOP);
node.setNestLoopFilters(new ArrayList<>());
node.setNestLoopDependNode(dependNode);
List<PlanNode> nodeList = Optional.ofNullable(dependNode.getNestLoopDependOnNodeList()).orElse(new ArrayList<>());
nodeList.add(nodeList.size(), node);
dependNode.setNestLoopDependOnNodeList(nodeList);
return;
}
}
joinNode.setStrategy(JoinNode.Strategy.ALWAYS_NEST_LOOP);
PlanNode dependedNode = findDependedNode(node, innerJoin);
if (Objects.isNull(dependedNode)) {
return;
}
handlerDependedNode(dependedNode, node);
node.setNestLoopFilters(new ArrayList<>());
node.setNestLoopDependNode(findDependNode(node));
node.setNestLoopDependNode(dependedNode);
}
private String getTableName(TableNode node) {
@@ -178,20 +168,47 @@ public class JoinStrategyChooser {
return node.getTableName();
}
private PlanNode findDependNode(PlanNode node) {
private PlanNode findDependedNode(PlanNode node, boolean innerJoin) {
JoinNode joinNode = (JoinNode) node.getParent();
String firstTableName = null;
PlanNode firstNode = null;
List<ItemFuncEqual> joinFilter = joinNode.getJoinFilter();
for (ItemFuncEqual itemFuncEqual : joinFilter) {
List<Item> arguments = itemFuncEqual.arguments();
String tableName = arguments.get(0).getTableName();
firstTableName = Optional.ofNullable(firstTableName).orElse(tableName);
Item item = arguments.stream().filter(argument -> !StringUtil.equals(getTableName((TableNode) node), argument.getTableName())).findFirst().get();
PlanNode dependedNode = nodeMap.get(item.getTableName());
if (Objects.isNull(firstNode)) {
firstNode = dependedNode;
}
if (isSmallTable((TableNode) dependedNode) && innerJoin) {
return dependedNode;
}
}
PlanNode dependNode = nodeMap.get(firstTableName);
List<PlanNode> nodeList = Optional.ofNullable(dependNode.getNestLoopDependOnNodeList()).orElse(new ArrayList<>());
return firstNode;
}
@Nullable
private void handlerDependedNode(PlanNode dependedNode, PlanNode node) {
setNestLoopDependOnNodeList(dependedNode, node);
PlanNode parent = dependedNode.getParent();
setNestLoopDependOnNodeList(dependedNode, node);
boolean isCurrentNode = true;
while (Objects.nonNull(parent)) {
if (!(parent instanceof JoinNode) || !canDoAsMerge((JoinNode) parent)) {
break;
}
dependedNode = parent;
isCurrentNode = false;
parent = parent.getParent();
}
if (!isCurrentNode) {
setNestLoopDependOnNodeList(dependedNode, node);
}
}
private void setNestLoopDependOnNodeList(PlanNode dependedNode, PlanNode node) {
List<PlanNode> nodeList = Optional.ofNullable(dependedNode.getNestLoopDependOnNodeList()).orElse(new ArrayList<>());
nodeList.add(nodeList.size(), node);
dependNode.setNestLoopDependOnNodeList(nodeList);
return nodeMap.get(firstTableName);
dependedNode.setNestLoopDependOnNodeList(nodeList);
}
private boolean buildNodeMap(JoinNode joinNode) {