Skip to content

Commit 5ab6abd

Browse files
committed
refactor: rename SubstraitRelNodeConverter to SubstraitToCalciteVisitor
1 parent 73f5232 commit 5ab6abd

File tree

8 files changed

+27
-27
lines changed

8 files changed

+27
-27
lines changed

isthmus/src/main/java/io/substrait/isthmus/SubstraitToCalcite.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
/**
2626
* Converts between Substrait {@link Rel}s and Calcite {@link RelNode}s.
2727
*
28-
* <p>Can be extended to customize the {@link RelBuilder} and {@link SubstraitRelNodeConverter} used
28+
* <p>Can be extended to customize the {@link RelBuilder} and {@link SubstraitToCalciteVisitor} used
2929
* in the conversion.
3030
*/
3131
public class SubstraitToCalcite {
@@ -68,28 +68,28 @@ protected RelBuilder createRelBuilder(CalciteSchema schema) {
6868
}
6969

7070
/**
71-
* Creates a {@link SubstraitRelNodeConverter} from the {@link RelBuilder}
71+
* Creates a {@link SubstraitToCalciteVisitor} from the {@link RelBuilder}
7272
*
73-
* <p>Override this method to customize the {@link SubstraitRelNodeConverter}.
73+
* <p>Override this method to customize the {@link SubstraitToCalciteVisitor}.
7474
*/
75-
protected SubstraitRelNodeConverter createSubstraitRelNodeConverter(RelBuilder relBuilder) {
76-
return new SubstraitRelNodeConverter(extensions, typeFactory, relBuilder);
75+
protected SubstraitToCalciteVisitor createSubstraitRelNodeConverter(RelBuilder relBuilder) {
76+
return new SubstraitToCalciteVisitor(extensions, typeFactory, relBuilder);
7777
}
7878

7979
/**
8080
* Converts a Substrait {@link Rel} to a Calcite {@link RelNode}
8181
*
8282
* <p>Generates a {@link CalciteSchema} based on the contents of the {@link Rel}, which will be
8383
* used to construct a {@link RelBuilder} with the required schema information to build {@link
84-
* RelNode}s, and a then a {@link SubstraitRelNodeConverter} to perform the actual conversion.
84+
* RelNode}s, and a then a {@link SubstraitToCalciteVisitor} to perform the actual conversion.
8585
*
8686
* @param rel {@link Rel} to convert
8787
* @return {@link RelNode}
8888
*/
8989
public RelNode convert(Rel rel) {
9090
CalciteSchema rootSchema = toSchema(rel);
9191
RelBuilder relBuilder = createRelBuilder(rootSchema);
92-
SubstraitRelNodeConverter converter = createSubstraitRelNodeConverter(relBuilder);
92+
SubstraitToCalciteVisitor converter = createSubstraitRelNodeConverter(relBuilder);
9393
return rel.accept(converter);
9494
}
9595

isthmus/src/main/java/io/substrait/isthmus/SubstraitRelNodeConverter.java renamed to isthmus/src/main/java/io/substrait/isthmus/SubstraitToCalciteVisitor.java

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
* RelVisitor to convert Substrait Rel plan to Calcite RelNode plan. Unsupported Rel node will call
5757
* visitFallback and throw UnsupportedOperationException.
5858
*/
59-
public class SubstraitRelNodeConverter extends AbstractRelVisitor<RelNode, RuntimeException> {
59+
public class SubstraitToCalciteVisitor extends AbstractRelVisitor<RelNode, RuntimeException> {
6060

6161
protected final RelDataTypeFactory typeFactory;
6262

@@ -68,7 +68,7 @@ public class SubstraitRelNodeConverter extends AbstractRelVisitor<RelNode, Runti
6868
protected final RexBuilder rexBuilder;
6969
private final TypeConverter typeConverter;
7070

71-
public SubstraitRelNodeConverter(
71+
public SubstraitToCalciteVisitor(
7272
SimpleExtension.ExtensionCollection extensions,
7373
RelDataTypeFactory typeFactory,
7474
RelBuilder relBuilder) {
@@ -81,7 +81,7 @@ public SubstraitRelNodeConverter(
8181
TypeConverter.DEFAULT);
8282
}
8383

84-
public SubstraitRelNodeConverter(
84+
public SubstraitToCalciteVisitor(
8585
RelDataTypeFactory typeFactory,
8686
RelBuilder relBuilder,
8787
ScalarFunctionConverter scalarFunctionConverter,
@@ -99,7 +99,7 @@ public SubstraitRelNodeConverter(
9999
typeFactory, scalarFunctionConverter, windowFunctionConverter, typeConverter));
100100
}
101101

102-
public SubstraitRelNodeConverter(
102+
public SubstraitToCalciteVisitor(
103103
RelDataTypeFactory typeFactory,
104104
RelBuilder relBuilder,
105105
ScalarFunctionConverter scalarFunctionConverter,
@@ -132,7 +132,7 @@ public static RelNode convert(
132132
.build());
133133

134134
return relRoot.accept(
135-
new SubstraitRelNodeConverter(
135+
new SubstraitToCalciteVisitor(
136136
EXTENSION_COLLECTION, relOptCluster.getTypeFactory(), relBuilder));
137137
}
138138

isthmus/src/main/java/io/substrait/isthmus/SubstraitToSql.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,10 @@ public SubstraitToSql() {
1616
public RelNode substraitRelToCalciteRel(Rel relRoot, List<String> tables)
1717
throws SqlParseException {
1818
CalciteCatalogReader catalogReader = registerCreateTables(tables);
19-
return SubstraitRelNodeConverter.convert(relRoot, relOptCluster, catalogReader, parserConfig);
19+
return SubstraitToCalciteVisitor.convert(relRoot, relOptCluster, catalogReader, parserConfig);
2020
}
2121

2222
public RelNode substraitRelToCalciteRel(Rel relRoot, Prepare.CatalogReader catalog) {
23-
return SubstraitRelNodeConverter.convert(relRoot, relOptCluster, catalog, parserConfig);
23+
return SubstraitToCalciteVisitor.convert(relRoot, relOptCluster, catalog, parserConfig);
2424
}
2525
}

isthmus/src/main/java/io/substrait/isthmus/expression/ExpressionRexConverter.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
import io.substrait.expression.FunctionArg;
1717
import io.substrait.expression.WindowBound;
1818
import io.substrait.extension.SimpleExtension;
19-
import io.substrait.isthmus.SubstraitRelNodeConverter;
19+
import io.substrait.isthmus.SubstraitToCalciteVisitor;
2020
import io.substrait.isthmus.TypeConverter;
2121
import io.substrait.type.StringTypeVisitor;
2222
import io.substrait.type.Type;
@@ -60,7 +60,7 @@ public class ExpressionRexConverter extends AbstractExpressionVisitor<RexNode, R
6060
protected final RexBuilder rexBuilder;
6161
protected final ScalarFunctionConverter scalarFunctionConverter;
6262
protected final WindowFunctionConverter windowFunctionConverter;
63-
protected SubstraitRelNodeConverter relNodeConverter;
63+
protected SubstraitToCalciteVisitor relNodeConverter;
6464

6565
private static final SqlIntervalQualifier YEAR_MONTH_INTERVAL =
6666
new SqlIntervalQualifier(
@@ -90,8 +90,8 @@ public ExpressionRexConverter(
9090
this.windowFunctionConverter = windowFunctionConverter;
9191
}
9292

93-
public void setRelNodeConverter(final SubstraitRelNodeConverter substraitRelNodeConverter) {
94-
this.relNodeConverter = substraitRelNodeConverter;
93+
public void setRelNodeConverter(final SubstraitToCalciteVisitor substraitToCalciteVisitor) {
94+
this.relNodeConverter = substraitToCalciteVisitor;
9595
}
9696

9797
@Override

isthmus/src/test/java/io/substrait/isthmus/CustomFunctionTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -260,8 +260,8 @@ public CustomSubstraitToCalcite(
260260
}
261261

262262
@Override
263-
protected SubstraitRelNodeConverter createSubstraitRelNodeConverter(RelBuilder relBuilder) {
264-
return new SubstraitRelNodeConverter(
263+
protected SubstraitToCalciteVisitor createSubstraitRelNodeConverter(RelBuilder relBuilder) {
264+
return new SubstraitToCalciteVisitor(
265265
typeFactory,
266266
relBuilder,
267267
scalarFunctionConverter,

isthmus/src/test/java/io/substrait/isthmus/RelExtensionRoundtripTest.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ void roundtrip(Rel pojo1) {
6868

6969
// Substrait POJO 2 -> Calcite
7070
var calcite =
71-
pojo2.accept(new CustomSubstraitRelNodeConverter(extensions, typeFactory, builder));
71+
pojo2.accept(new CustomSubstraitToCalciteVisitor(extensions, typeFactory, builder));
7272

7373
// Calcite -> Substrait POJO 3
7474
var pojo3 = (new CustomCalciteToSubstraitVisitor(typeFactory, extensions)).apply(calcite);
@@ -180,12 +180,12 @@ protected Extension.MultiRelDetail detailFromExtensionMultiRel(Any any) {
180180
}
181181

182182
/**
183-
* Extends the standard {@link SubstraitRelNodeConverter} to handle Extension relations containing
183+
* Extends the standard {@link SubstraitToCalciteVisitor} to handle Extension relations containing
184184
* {@link ColumnAppendDetail}
185185
*/
186-
static class CustomSubstraitRelNodeConverter extends SubstraitRelNodeConverter {
186+
static class CustomSubstraitToCalciteVisitor extends SubstraitToCalciteVisitor {
187187

188-
public CustomSubstraitRelNodeConverter(
188+
public CustomSubstraitToCalciteVisitor(
189189
SimpleExtension.ExtensionCollection extensions,
190190
RelDataTypeFactory typeFactory,
191191
RelBuilder relBuilder) {

isthmus/src/test/java/io/substrait/isthmus/SubstraitExpressionConverterTest.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,8 @@ public class SubstraitExpressionConverterTest extends PlanTestBase {
3737
final Rel commonTable =
3838
b.namedScan(List.of("example"), List.of("a", "b", "c", "d"), commonTableType);
3939

40-
final SubstraitRelNodeConverter relNodeConverter =
41-
new SubstraitRelNodeConverter(extensions, typeFactory, builder);
40+
final SubstraitToCalciteVisitor relNodeConverter =
41+
new SubstraitToCalciteVisitor(extensions, typeFactory, builder);
4242

4343
public SubstraitExpressionConverterTest() {
4444
converter = relNodeConverter.expressionRexConverter;

isthmus/src/test/java/io/substrait/isthmus/SubstraitRelNodeConverterTest.java renamed to isthmus/src/test/java/io/substrait/isthmus/SubstraitToCalciteVisitorTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import org.junit.jupiter.api.Nested;
1616
import org.junit.jupiter.api.Test;
1717

18-
public class SubstraitRelNodeConverterTest extends PlanTestBase {
18+
public class SubstraitToCalciteVisitorTest extends PlanTestBase {
1919

2020
static final TypeCreator R = TypeCreator.of(false);
2121
static final TypeCreator N = TypeCreator.of(true);

0 commit comments

Comments
 (0)