[OPPRO-233] Add SingularOrList support (#45)

oap-project · Nov 7, 2022 · ab13751 · ab13751
1 parent 8f8f1ef
commit ab13751
Show file tree

Hide file tree

Showing 4 changed files with 278 additions and 166 deletions.
diff --git a/velox/substrait/SubstraitToVeloxExpr.cpp b/velox/substrait/SubstraitToVeloxExpr.cpp
@@ -142,6 +142,19 @@ SubstraitVeloxExprConverter::toVeloxExpr(
       toVeloxType(typeName), std::move(params), veloxFunction);
 }
 
+core::TypedExprPtr SubstraitVeloxExprConverter::toVeloxExpr(
+    const ::substrait::Expression::SingularOrList& singularOrList,
+    const RowTypePtr& inputType) {
+  std::vector<std::shared_ptr<const core::ITypedExpr>> params;
+  auto inLists = singularOrList.options().data()[0];
+  params.reserve(2);
+  // first is the value, second is the list
+  params.emplace_back(toVeloxExpr(singularOrList.value(), inputType));
+  params.emplace_back(toVeloxExpr(*inLists, inputType));
+  return std::make_shared<const core::CallTypedExpr>(
+      BOOLEAN(), std::move(params), "in");
+}
+
 std::shared_ptr<const core::ConstantTypedExpr>
 SubstraitVeloxExprConverter::toVeloxExpr(
     const ::substrait::Expression::Literal& substraitLit) {

diff --git a/velox/substrait/SubstraitToVeloxExpr.h b/velox/substrait/SubstraitToVeloxExpr.h
@@ -54,6 +54,10 @@ class SubstraitVeloxExprConverter {
       const ::substrait::Expression::ScalarFunction& sFunc,
       const RowTypePtr& inputType);
 
+  core::TypedExprPtr toVeloxExpr(
+      const ::substrait::Expression::SingularOrList& singularOrList,
+      const RowTypePtr& inputType);
+
   /// Convert Substrait CastExpression to Velox Expression.
   std::shared_ptr<const core::ITypedExpr> toVeloxExpr(
       const ::substrait::Expression::Cast& castExpr,