From 19f80aeda9099e5b0400f2dc003461fe98f08a24 Mon Sep 17 00:00:00 2001 From: Chris Rice Date: Thu, 29 Aug 2024 09:18:30 -0400 Subject: [PATCH] feat(isthmus): allow WindowFunctionConverter to use a non-default TypeConverter --- .../isthmus/expression/WindowFunctionConverter.java | 9 +++++++++ .../isthmus/expression/WindowRelFunctionConverter.java | 9 +++++++++ 2 files changed, 18 insertions(+) diff --git a/isthmus/src/main/java/io/substrait/isthmus/expression/WindowFunctionConverter.java b/isthmus/src/main/java/io/substrait/isthmus/expression/WindowFunctionConverter.java index 97ebf202a..2d3b42051 100644 --- a/isthmus/src/main/java/io/substrait/isthmus/expression/WindowFunctionConverter.java +++ b/isthmus/src/main/java/io/substrait/isthmus/expression/WindowFunctionConverter.java @@ -10,6 +10,7 @@ import io.substrait.expression.WindowBound; import io.substrait.extension.SimpleExtension; import io.substrait.isthmus.AggregateFunctions; +import io.substrait.isthmus.TypeConverter; import io.substrait.type.Type; import java.util.Collections; import java.util.List; @@ -39,6 +40,14 @@ public WindowFunctionConverter( super(functions, typeFactory); } + public WindowFunctionConverter( + List functions, + List additionalSignatures, + RelDataTypeFactory typeFactory, + TypeConverter typeConverter) { + super(functions, additionalSignatures, typeFactory, typeConverter); + } + @Override protected Expression.WindowFunctionInvocation generateBinding( WrappedWindowCall call, diff --git a/isthmus/src/main/java/io/substrait/isthmus/expression/WindowRelFunctionConverter.java b/isthmus/src/main/java/io/substrait/isthmus/expression/WindowRelFunctionConverter.java index 1c00362f0..cd254d551 100644 --- a/isthmus/src/main/java/io/substrait/isthmus/expression/WindowRelFunctionConverter.java +++ b/isthmus/src/main/java/io/substrait/isthmus/expression/WindowRelFunctionConverter.java @@ -9,6 +9,7 @@ import io.substrait.expression.WindowBound; import io.substrait.extension.SimpleExtension; import io.substrait.isthmus.AggregateFunctions; +import io.substrait.isthmus.TypeConverter; import io.substrait.relation.ConsistentPartitionWindow; import io.substrait.type.Type; import java.util.List; @@ -38,6 +39,14 @@ public WindowRelFunctionConverter( super(functions, typeFactory); } + public WindowRelFunctionConverter( + List functions, + List additionalSignatures, + RelDataTypeFactory typeFactory, + TypeConverter typeConverter) { + super(functions, additionalSignatures, typeFactory, typeConverter); + } + @Override protected ConsistentPartitionWindow.WindowRelFunctionInvocation generateBinding( WrappedWindowRelCall call,