Skip to content

Commit d182ec5

Browse files
committed
refactor: rename SubstraitRelNodeConverter to SubstraitToCalciteVisitor
1 parent a2e2d2a commit d182ec5

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
@@ -29,7 +29,7 @@
2929
/**
3030
* Converts between Substrait {@link Rel}s and Calcite {@link RelNode}s.
3131
*
32-
* <p>Can be extended to customize the {@link RelBuilder} and {@link SubstraitRelNodeConverter} used
32+
* <p>Can be extended to customize the {@link RelBuilder} and {@link SubstraitToCalciteVisitor} used
3333
* in the conversion.
3434
*/
3535
public class SubstraitToCalcite {
@@ -82,28 +82,28 @@ protected RelBuilder createRelBuilder(CalciteSchema schema) {
8282
}
8383

8484
/**
85-
* Creates a {@link SubstraitRelNodeConverter} from the {@link RelBuilder}
85+
* Creates a {@link SubstraitToCalciteVisitor} from the {@link RelBuilder}
8686
*
87-
* <p>Override this method to customize the {@link SubstraitRelNodeConverter}.
87+
* <p>Override this method to customize the {@link SubstraitToCalciteVisitor}.
8888
*/
89-
protected SubstraitRelNodeConverter createSubstraitRelNodeConverter(RelBuilder relBuilder) {
90-
return new SubstraitRelNodeConverter(extensions, typeFactory, relBuilder);
89+
protected SubstraitToCalciteVisitor createSubstraitRelNodeConverter(RelBuilder relBuilder) {
90+
return new SubstraitToCalciteVisitor(extensions, typeFactory, relBuilder);
9191
}
9292

9393
/**
9494
* Converts a Substrait {@link Rel} to a Calcite {@link RelNode}
9595
*
9696
* <p>Generates a {@link CalciteSchema} based on the contents of the {@link Rel}, which will be
9797
* used to construct a {@link RelBuilder} with the required schema information to build {@link
98-
* RelNode}s, and a then a {@link SubstraitRelNodeConverter} to perform the actual conversion.
98+
* RelNode}s, and a then a {@link SubstraitToCalciteVisitor} to perform the actual conversion.
9999
*
100100
* @param rel {@link Rel} to convert
101101
* @return {@link RelNode}
102102
*/
103103
public RelNode convert(Rel rel) {
104104
CalciteSchema rootSchema = toSchema(rel);
105105
RelBuilder relBuilder = createRelBuilder(rootSchema);
106-
SubstraitRelNodeConverter converter = createSubstraitRelNodeConverter(relBuilder);
106+
SubstraitToCalciteVisitor converter = createSubstraitRelNodeConverter(relBuilder);
107107
return rel.accept(converter);
108108
}
109109

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
@@ -13,7 +13,7 @@
1313
import io.substrait.expression.FunctionArg;
1414
import io.substrait.expression.WindowBound;
1515
import io.substrait.extension.SimpleExtension;
16-
import io.substrait.isthmus.SubstraitRelNodeConverter;
16+
import io.substrait.isthmus.SubstraitToCalciteVisitor;
1717
import io.substrait.isthmus.TypeConverter;
1818
import io.substrait.type.StringTypeVisitor;
1919
import io.substrait.type.Type;
@@ -57,7 +57,7 @@ public class ExpressionRexConverter extends AbstractExpressionVisitor<RexNode, R
5757
protected final RexBuilder rexBuilder;
5858
protected final ScalarFunctionConverter scalarFunctionConverter;
5959
protected final WindowFunctionConverter windowFunctionConverter;
60-
protected SubstraitRelNodeConverter relNodeConverter;
60+
protected SubstraitToCalciteVisitor relNodeConverter;
6161

6262
private static final SqlIntervalQualifier YEAR_MONTH_INTERVAL =
6363
new SqlIntervalQualifier(
@@ -87,8 +87,8 @@ public ExpressionRexConverter(
8787
this.windowFunctionConverter = windowFunctionConverter;
8888
}
8989

90-
public void setRelNodeConverter(final SubstraitRelNodeConverter substraitRelNodeConverter) {
91-
this.relNodeConverter = substraitRelNodeConverter;
90+
public void setRelNodeConverter(final SubstraitToCalciteVisitor substraitToCalciteVisitor) {
91+
this.relNodeConverter = substraitToCalciteVisitor;
9292
}
9393

9494
@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
@@ -33,8 +33,8 @@ public class SubstraitExpressionConverterTest extends PlanTestBase {
3333
final Rel commonTable =
3434
b.namedScan(List.of("example"), List.of("a", "b", "c", "d"), commonTableType);
3535

36-
final SubstraitRelNodeConverter relNodeConverter =
37-
new SubstraitRelNodeConverter(extensions, typeFactory, builder);
36+
final SubstraitToCalciteVisitor relNodeConverter =
37+
new SubstraitToCalciteVisitor(extensions, typeFactory, builder);
3838

3939
public SubstraitExpressionConverterTest() {
4040
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)