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

[feature](vec) Support update stmt in vec query engine (#8296)

Browse Source
This commit is contained in:
HappenLee
2022-03-07 14:03:55 +08:00
committed by GitHub
parent 3b159a9820
commit 477b87cb28
5 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fe/fe-core/src/main/java/org/apache/doris/load/update/UpdatePlanner.java
View File

@ -30,6 +30,7 @@ import org.apache.doris.catalog.OlapTable;
import org.apache.doris.common.AnalysisException;
import org.apache.doris.common.IdGenerator;
import org.apache.doris.common.UserException;
import org.apache.doris.common.util.VectorizedUtil;
import org.apache.doris.planner.DataPartition;
import org.apache.doris.planner.OlapScanNode;
import org.apache.doris.planner.OlapTableSink;
@ -86,6 +87,9 @@ public class UpdatePlanner extends Planner {
/* END */
olapScanNode.init(analyzer);
olapScanNode.finalize(analyzer);
if (VectorizedUtil.isVectorized()) {
olapScanNode.convertToVectoriezd();
}
scanNodeList.add(olapScanNode);
// 2. gen olap table sink
OlapTableSink olapTableSink = new OlapTableSink(targetTable, computeTargetTupleDesc(), null);

2
fe/fe-core/src/main/java/org/apache/doris/load/update/UpdateStmtExecutor.java
View File

@ -34,6 +34,7 @@ import org.apache.doris.common.QuotaExceedException;
import org.apache.doris.common.UserException;
import org.apache.doris.common.util.DebugUtil;
import org.apache.doris.common.util.TimeUtils;
import org.apache.doris.common.util.VectorizedUtil;
import org.apache.doris.metric.MetricRepo;
import org.apache.doris.qe.Coordinator;
import org.apache.doris.qe.QeProcessorImpl;
@ -141,6 +142,7 @@ public class UpdateStmtExecutor {
coordinator = new Coordinator(Catalog.getCurrentCatalog().getNextId(), queryId, analyzer.getDescTbl(),
updatePlanner.getFragments(), updatePlanner.getScanNodes(), TimeUtils.DEFAULT_TIME_ZONE, false);
coordinator.setQueryType(TQueryType.LOAD);
coordinator.setExecVecEngine(VectorizedUtil.isVectorized());
QeProcessorImpl.INSTANCE.registerQuery(queryId, coordinator);
analyzer.getContext().getExecutor().setCoord(coordinator);

2
fe/fe-core/src/main/java/org/apache/doris/planner/HashJoinNode.java
View File

@ -669,7 +669,7 @@ public class HashJoinNode extends PlanNode {
}
@Override
void convertToVectoriezd() {
public void convertToVectoriezd() {
if (!otherJoinConjuncts.isEmpty()) {
votherJoinConjunct = convertConjunctsToAndCompoundPredicate(otherJoinConjuncts);
initCompoundPredicate(votherJoinConjunct);

2
fe/fe-core/src/main/java/org/apache/doris/planner/PlanNode.java
View File

@ -860,7 +860,7 @@ abstract public class PlanNode extends TreeNode<PlanNode> {
return sb.toString();
}
void convertToVectoriezd() {
public void convertToVectoriezd() {
if (!conjuncts.isEmpty()) {
vconjunct = convertConjunctsToAndCompoundPredicate(conjuncts);
initCompoundPredicate(vconjunct);

4
fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java
View File

@ -301,6 +301,10 @@ public class Coordinator {
this.queryOptions.setQueryType(type);
}
public void setExecVecEngine(boolean vec) {
this.queryOptions.setEnableVectorizedEngine(vec);
}
public Status getExecStatus() {
return queryStatus;
}