feat(query): bloom index support prune expr with simple casts (#17533)

* try_eliminate_cast

* fold_for_prune

* fix

* domain

* refine test

* visit_expr_column_eq_constant

* filter_index_field

* x

* ControlFlow

* test

* update

* RewriteVisitor

* clean

* fix

* refine

Author: forsaken628

Cargo.lock

@@ -17,6 +17,7 @@ bstr = { workspace = true }
 bumpalo = { workspace = true }
 crc32fast = { workspace = true }
 ctor = { workspace = true }
+databend-common-ast = { workspace = true }
 databend-common-base = { workspace = true }
 databend-common-column = { workspace = true }
 databend-common-exception = { workspace = true }

src/query/functions/Cargo.toml

@@ -15,14 +15,12 @@
 #[macro_use]
 extern crate criterion;
 
-#[path = "../tests/it/scalars/parser.rs"]
-mod parser;
-
 use criterion::Criterion;
 use databend_common_expression::type_check;
 use databend_common_expression::DataBlock;
 use databend_common_expression::Evaluator;
 use databend_common_expression::FunctionContext;
+use databend_common_functions::test_utils as parser;
 use databend_common_functions::BUILTIN_FUNCTIONS;
 
 fn bench(c: &mut Criterion) {

src/query/functions/benches/bench.rs

@@ -33,6 +33,7 @@ pub mod aggregates;
 mod cast_rules;
 pub mod scalars;
 pub mod srfs;
+pub mod test_utils;
 
 pub fn is_builtin_function(name: &str) -> bool {
     let name = Ascii::new(name);

src/query/functions/src/lib.rs

@@ -1,4 +1,4 @@
-// Copyright 2022 Datafuse Labs.
+// Copyright 2021 Datafuse Labs
 //
 // Licensed under the Apache License, Version 2.0 (the "License");
 // you may not use this file except in compliance with the License.
@@ -37,7 +37,8 @@ use databend_common_expression::ConstantFolder;
 use databend_common_expression::FunctionContext;
 use databend_common_expression::RawExpr;
 use databend_common_expression::Scalar;
-use databend_common_functions::BUILTIN_FUNCTIONS;
+
+use crate::BUILTIN_FUNCTIONS;
 
 pub fn parse_raw_expr(text: &str, columns: &[(&str, DataType)]) -> RawExpr {
     let tokens = tokenize_sql(text).unwrap();
@@ -107,7 +108,7 @@ macro_rules! transform_interval_add_sub {
     };
 }
 
-pub fn transform_expr(ast: AExpr, columns: &[(&str, DataType)]) -> RawExpr {
+fn transform_expr(ast: AExpr, columns: &[(&str, DataType)]) -> RawExpr {
     match ast {
         AExpr::Literal { span, value } => RawExpr::Constant {
             span,
@@ -631,7 +632,7 @@ fn transform_data_type(target_type: databend_common_ast::ast::TypeName) -> DataT
     }
 }
 
-pub fn transform_literal(lit: ASTLiteral) -> Scalar {
+fn transform_literal(lit: ASTLiteral) -> Scalar {
     let scalar = match lit {
         ASTLiteral::UInt64(u) => Scalar::Number(NumberScalar::UInt64(u)),
         ASTLiteral::Decimal256 {

src/query/functions/tests/it/scalars/parser.rs renamed to src/query/functions/src/test_utils.rs

@@ -51,13 +51,14 @@ mod math;
 mod misc;
 mod obfuscator;
 mod other;
-pub(crate) mod parser;
 mod regexp;
 mod string;
 mod tuple;
 mod variant;
 mod vector;
 
+pub use databend_common_functions::test_utils as parser;
+
 pub fn run_ast(file: &mut impl Write, text: impl AsRef<str>, columns: &[(&str, Column)]) {
     let text = text.as_ref();
     let result: Result<_> = try {

src/query/functions/tests/it/scalars/mod.rs

@@ -23,6 +23,7 @@ databend-common-expression = { workspace = true }
 databend-common-functions = { workspace = true }
 databend-storages-common-table-meta = { workspace = true }
 fastrace = { workspace = true }
+goldenfile = { workspace = true }
 jsonb = { workspace = true }
 levenshtein_automata = { workspace = true }
 log = { workspace = true }