Merge pull request #7514 from ariesdevil/num_func_v2

feat(function-v2): migrate math functions to function-v2

Author: mergify[bot]

Cargo.lock

@@ -0,0 +1,24 @@
+// Copyright 2022 Datafuse Labs.
+//
+// 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.
+
+// Concat two arrays of different lengths
+pub const fn concat<T, const A: usize, const B: usize>(a: &[T; A], b: &[T; B]) -> [T; A + B] {
+    let mut result = std::mem::MaybeUninit::uninit();
+    let dest = result.as_mut_ptr() as *mut T;
+    unsafe {
+        std::ptr::copy_nonoverlapping(a.as_ptr(), dest, A);
+        std::ptr::copy_nonoverlapping(b.as_ptr(), dest.add(A), B);
+        result.assume_init()
+    }
+}

src/common/base/src/containers/array.rs

@@ -12,9 +12,11 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+mod array;
 mod pool;
 mod ttlhmap;
 
+pub use array::concat;
 pub use pool::ItemManager;
 pub use pool::Pool;
 pub use ttlhmap::CleanPolicy;

src/common/base/src/containers/mod.rs

@@ -14,6 +14,10 @@
 
 #![feature(backtrace)]
 #![feature(thread_local)]
+#![allow(incomplete_features)]
+#![feature(generic_const_exprs)]
+#![feature(const_maybe_uninit_as_mut_ptr)]
+#![feature(const_mut_refs)]
 
 pub mod base;
 pub mod containers;

src/common/base/src/lib.rs

@@ -12,15 +12,18 @@ doctest = false
 [dependencies] # In alphabetical order
 # Workspace dependencies
 common-arrow = { path = "../../common/arrow" }
+common-base = { path = "../../common/base" }
 common-expression = { path = "../expression" }
 
 # Crates.io dependencies
 base64 = "0.13.0"
 bstr = "0.2.17"
+crc32fast = "1.3.2"
 hex = "0.4.3"
 itertools = "0.10.3"
 match-template = "0.0.1"
 num-traits = "0.2.15"
+rand = { version = "0.8.5", features = ["small_rng"] }
 strength_reduce = "0.2.3"
 
 [dev-dependencies]

src/query/functions-v2/Cargo.toml

@@ -22,6 +22,7 @@ use common_expression::FunctionProperty;
 use common_expression::FunctionRegistry;
 
 use super::arithmetic_modulo::vectorize_modulo;
+use crate::scalars::ALL_NUMERICS_TYPES;
 
 pub fn register(registry: &mut FunctionRegistry) {
     registry.register_aliases("plus", &["add"]);
@@ -31,21 +32,8 @@ pub fn register(registry: &mut FunctionRegistry) {
     // TODO support modulo
     // registry.register_aliases("%", &["mod", "modulo"]);
 
-    let all_numerics_types = &[
-        DataType::UInt8,
-        DataType::UInt16,
-        DataType::UInt32,
-        DataType::UInt64,
-        DataType::Int8,
-        DataType::Int16,
-        DataType::Int32,
-        DataType::Int64,
-        DataType::Float32,
-        DataType::Float64,
-    ];
-
     // Unary OP for minus and plus
-    for left in all_numerics_types {
+    for left in ALL_NUMERICS_TYPES {
         with_number_mapped_type!(L, match left {
             DataType::L => {
                 type T = <L as ResultTypeOfUnary>::Negate;
@@ -78,8 +66,8 @@ pub fn register(registry: &mut FunctionRegistry) {
         });
     }
 
-    for left in all_numerics_types {
-        for right in all_numerics_types {
+    for left in ALL_NUMERICS_TYPES {
+        for right in ALL_NUMERICS_TYPES {
             with_number_mapped_type!(L, match left {
                 DataType::L => with_number_mapped_type!(R, match right {
                     DataType::R => {