Merge pull request #7556 from andylokandy/codegen

refactor(expr): codegen function registers

Author: mergify[bot]

Cargo.lock

@@ -18,5 +18,6 @@ path = "src/bin/codegen.rs"
 common-datavalues = { path = "../datavalues" }
 common-expression = { path = "../expression" }
 
+itertools = "0.10"
 serde = { version = "1.0.137", features = ["derive"] }
 serde_json = "1.0.81"

src/query/codegen/Cargo.toml

@@ -12,10 +12,8 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-use codegen::writes::codegen_arithmetic_type;
-use codegen::writes::codegen_arithmetic_type_v2;
-
 fn main() {
-    codegen_arithmetic_type();
-    codegen_arithmetic_type_v2();
+    codegen::writes::codegen_arithmetic_type();
+    codegen::writes::codegen_arithmetic_type_v2();
+    codegen::writes::codegen_register();
 }

src/query/codegen/src/bin/codegen.rs

@@ -12,4 +12,6 @@
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
+#![feature(exit_status_error)]
+
 pub mod writes;

src/query/codegen/src/lib.rs

@@ -25,7 +25,7 @@ pub fn codegen_arithmetic_type() {
     use DataValueBinaryOperator::*;
     use DataValueUnaryOperator::*;
 
-    let dest = Path::new("src/common/datavalues/src/types");
+    let dest = Path::new("src/query/datavalues/src/types");
     let path = dest.join("arithmetics_type.rs");
 
     let mut file = File::create(&path).expect("open");

src/query/codegen/src/writes/arithmetics_type.rs

@@ -35,7 +35,9 @@ pub fn codegen_arithmetic_type_v2() {
     let path = dest.join("arithmetics_type.rs");
 
     let mut file = File::create(&path).expect("open");
+
     // Write the head.
+    let codegen_src = file!();
     writeln!(
         file,
         "// Copyright 2021 Datafuse Labs.
@@ -52,7 +54,7 @@ pub fn codegen_arithmetic_type_v2() {
 // See the License for the specific language governing permissions and
 // limitations under the License.
 
-// This code is generated by common/codegen. DO NOT EDIT.
+// This code is generated by {codegen_src}. DO NOT EDIT.
 use super::number::Number;
 
 pub trait ResultTypeOfBinary: Sized {{

src/query/codegen/src/writes/arithmetics_type_v2.rs

@@ -14,6 +14,8 @@
 
 mod arithmetics_type;
 mod arithmetics_type_v2;
+mod register;
 
 pub use arithmetics_type::*;
 pub use arithmetics_type_v2::*;
+pub use register::*;