Register lpad/rpad functions for Spark SQL (facebookincubator#63)

rui-mo · Jan 3, 2023 · 5a027f1 · 5a027f1
1 parent d927e5d
commit 5a027f1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/velox/functions/sparksql/Register.cpp b/velox/functions/sparksql/Register.cpp
@@ -81,6 +81,10 @@ void registerFunctions(const std::string& prefix) {
   registerFunction<sparksql::ChrFunction, Varchar, int64_t>({prefix + "chr"});
   registerFunction<AsciiFunction, int32_t, Varchar>({prefix + "ascii"});
 
+  registerFunction<LPadFunction, Varchar, Varchar, int32_t, Varchar>(
+      {prefix + "lpad"});
+  registerFunction<RPadFunction, Varchar, Varchar, int32_t, Varchar>(
+      {prefix + "rpad"});
   registerFunction<SubstrFunction, Varchar, Varchar, int32_t>(
       {prefix + "substring"});
   registerFunction<SubstrFunction, Varchar, Varchar, int32_t, int32_t>(