sparksql add unscaled_value function (facebookincubator#4)

velox/functions/sparksql/Decimal.h

@@ -0,0 +1,32 @@
+/*
+ * Copyright (c) Facebook, Inc. and its affiliates.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+#include <velox/type/UnscaledShortDecimal.h>
+#include "velox/functions/Macros.h"
+#include "velox/type/Type.h"
+
+namespace facebook::velox::functions {
+
+template <typename T>
+struct UnscaledValueFunction {
+  VELOX_DEFINE_FUNCTION_TYPES(T);
+
+  FOLLY_ALWAYS_INLINE void call(
+      int64_t& result,
+      const arg_type<UnscaledShortDecimal>& shortDecimal) {
+    result = shortDecimal.unscaledValue();
+  }
+};
+} // namespace facebook::velox::functions

velox/functions/sparksql/Register.cpp

@@ -26,6 +26,7 @@
 #include "velox/functions/sparksql/CompareFunctionsNullSafe.h"
 #include "velox/functions/sparksql/DateTime.h"
 #include "velox/functions/sparksql/DateTimeFunctions.h"
+#include "velox/functions/sparksql/Decimal.h"
 #include "velox/functions/sparksql/Hash.h"
 #include "velox/functions/sparksql/In.h"
 #include "velox/functions/sparksql/LeastGreatest.h"
@@ -220,8 +221,9 @@ void registerFunctions(const std::string& prefix) {
   registerFunction<DateAddFunction, Date, Date, int32_t>({"date_add"});
   registerFunction<DateAddFunction, Date, Date, int16_t>({"date_add"});
   registerFunction<DateAddFunction, Date, Date, int8_t>({"date_add"});
-  registerFunction<DateDiffFunction, int32_t, Date, Date>(
-      {"date_diff"});
+  registerFunction<DateDiffFunction, int32_t, Date, Date>({"date_diff"});
+  registerFunction<UnscaledValueFunction, int64_t, UnscaledShortDecimal>(
+      {prefix + "unscaled_value"});
 }
 
 } // namespace sparksql