Skip to content

Commit d30c823

Browse files
committed
fix: CteScanNode replica set & query done
1 parent e4ab745 commit d30c823

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/CteMaterializer.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ public void materializeCTE(Analysis analysis, MPPQueryContext context) {
8383
if (dataStore == null) {
8484
// CTE query execution failed. Use inline instead of materialization
8585
// in the outer query
86-
query.setMaterialized(false);
86+
query.setDone(false);
8787
return;
8888
}
8989

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/RelationPlanner.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,7 @@ protected RelationPlan visitTable(final Table table, final Void context) {
249249
throw new SemanticException("unexpected recursive cte");
250250
}
251251

252-
if (namedQuery.isMaterialized()) {
252+
if (namedQuery.isMaterialized() && namedQuery.isDone()) {
253253
CteDataStore dataStore = queryContext.getCteDataStore(table);
254254
if (dataStore != null) {
255255
List<Symbol> cteSymbols =

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/node/CteScanNode.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
package org.apache.iotdb.db.queryengine.plan.relational.planner.node;
2323

2424
import org.apache.iotdb.common.rpc.thrift.TRegionReplicaSet;
25+
import org.apache.iotdb.db.queryengine.common.DataNodeEndPoints;
2526
import org.apache.iotdb.db.queryengine.plan.planner.plan.node.PlanNode;
2627
import org.apache.iotdb.db.queryengine.plan.planner.plan.node.PlanNodeId;
2728
import org.apache.iotdb.db.queryengine.plan.planner.plan.node.PlanVisitor;
@@ -81,7 +82,8 @@ public void close() throws Exception {}
8182

8283
@Override
8384
public TRegionReplicaSet getRegionReplicaSet() {
84-
return null;
85+
return new TRegionReplicaSet(
86+
null, ImmutableList.of(DataNodeEndPoints.getLocalDataNodeLocation()));
8587
}
8688

8789
@Override

0 commit comments

Comments
 (0)