feat: helpers for conversion functions

apache · Jun 3, 2022 · b28e5b1 · b28e5b1
1 parent 23f37f5
commit b28e5b1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cpp/src/arrow/engine/substrait/expression_internal.cc b/cpp/src/arrow/engine/substrait/expression_internal.cc
@@ -164,6 +164,15 @@ Result<compute::Expression> FromProto(const substrait::Expression& expr,
       return arrow_function(scalar_fn);                      
     }
 
+    case substrait::Expression::kEnum: {
+      auto enum_expr = expr.enum_();
+      if(enum_expr.has_specified()){
+      return compute::literal(std::move(enum_expr.specified()));
+    } else {
+      return Status::Invalid("Substrait Enum value not specified");
+    }
+    }
+
     default:
       break;
   }