feat: pretty print backtrace

Cargo.toml

@@ -230,6 +230,7 @@ chrono = { version = "0.4.31", features = ["serde"] }
 chrono-tz = { version = "0.8", features = ["serde"] }
 clap = { version = "4.4.2", features = ["derive"] }
 criterion = "0.5"
+ctor = "0.2"
 dashmap = "5.4.0"
 deepsize = { version = "0.2.0" }
 deltalake = "0.17"

src/bendpy/Cargo.toml

@@ -17,7 +17,7 @@ crate-type = ["cdylib"]
 [dependencies]
 arrow = { workspace = true, features = ["pyarrow"] }
 arrow-schema = { workspace = true }
-ctor = "0.2.5"
+ctor = { workspace = true }
 databend-common-config = { workspace = true }
 databend-common-exception = { workspace = true }
 databend-common-expression = { workspace = true }

src/common/exception/Cargo.toml

@@ -11,13 +11,15 @@ doctest = false
 test = true
 
 [dependencies]
-databend-common-arrow = { path = "../arrow" }
-databend-common-ast = { path = "../../query/ast" }
+databend-common-arrow = { workspace = true }
+databend-common-ast = { workspace = true }
 
 anyhow = { workspace = true }
 arrow-schema = { workspace = true }
 backtrace = { workspace = true }
 bincode = { workspace = true }
+color-backtrace = { version = "0.6" }
+ctor = { workspace = true }
 geos = { workspace = true }
 geozero = { workspace = true }
 http = { workspace = true }

src/common/exception/src/lib.rs

@@ -28,3 +28,8 @@ pub use exception_backtrace::USER_SET_ENABLE_BACKTRACE;
 pub use exception_into::SerializedError;
 pub use with_context::ErrorWithContext;
 pub use with_context::WithContext;
+
+#[ctor::ctor]
+fn init_color_backtrace() {
+    color_backtrace::install();
+}

src/common/io/Cargo.toml

@@ -16,7 +16,7 @@ borsh = { workspace = true }
 bytes = { workspace = true }
 chrono = { workspace = true }
 chrono-tz = { workspace = true }
-databend-common-exception = { path = "../exception" }
+databend-common-exception = { workspace = true }
 ethnum = { workspace = true }
 geo = { workspace = true }
 geos = { workspace = true }

src/query/ee/Cargo.toml

@@ -63,7 +63,7 @@ jsonb = { workspace = true }
 tantivy = { workspace = true }
 
 [build-dependencies]
-databend-common-building = { path = "../../common/building" }
+databend-common-building = { workspace = true }
 
 [lints]
 workspace = true

src/query/functions/Cargo.toml

@@ -19,7 +19,7 @@ chrono = { workspace = true }
 chrono-tz = { workspace = true }
 crc32fast = "1.3.2"
 criterion = { workspace = true }
-ctor = "0.1.26"
+ctor = { workspace = true }
 databend-common-arrow = { workspace = true }
 databend-common-base = { workspace = true }
 databend-common-exception = { workspace = true }

src/query/service/Cargo.toml

@@ -49,7 +49,7 @@ byteorder = { workspace = true }
 chrono = { workspace = true }
 chrono-tz = { workspace = true }
 config = { version = "0.13.4", features = [] }
-ctor = "0.1.26"
+ctor = { workspace = true }
 dashmap = { workspace = true }
 databend-common-arrow = { workspace = true }
 databend-common-ast = { workspace = true }

src/query/service/src/servers/mysql/mysql_federated.rs

@@ -14,8 +14,8 @@
 
 use std::collections::HashMap;
 use std::sync::Arc;
+use std::sync::LazyLock;
 
-use ctor::ctor;
 use databend_common_expression::types::StringType;
 use databend_common_expression::utils::FromData;
 use databend_common_expression::DataBlock;
@@ -124,54 +124,62 @@ impl MySQLFederated {
 
     // Check SELECT @@variable, @@variable
     fn federated_select_variable_check(&self, query: &str) -> Option<(TableSchemaRef, DataBlock)> {
-        #[ctor]
-        static SELECT_VARIABLES_LAZY_RULES: Vec<(Regex, LazyBlockFunc)> = vec![
-            (
-                Regex::new("(?i)^(SELECT @@(.*))").unwrap(),
-                MySQLFederated::select_variable_data_block,
-            ),
-            (
-                Regex::new("(?i)^(/\\* mysql-connector-java(.*))").unwrap(),
-                MySQLFederated::select_variable_data_block,
-            ),
-        ];
+        static SELECT_VARIABLES_LAZY_RULES: LazyLock<Vec<(Regex, LazyBlockFunc)>> =
+            LazyLock::new(|| {
+                vec![
+                    (
+                        Regex::new("(?i)^(SELECT @@(.*))").unwrap(),
+                        MySQLFederated::select_variable_data_block,
+                    ),
+                    (
+                        Regex::new("(?i)^(/\\* mysql-connector-java(.*))").unwrap(),
+                        MySQLFederated::select_variable_data_block,
+                    ),
+                ]
+            });
 
         FederatedHelper::lazy_block_match_rule(query, &SELECT_VARIABLES_LAZY_RULES)
     }
 
     // Check SHOW VARIABLES LIKE.
     fn federated_show_variables_check(&self, query: &str) -> Option<(TableSchemaRef, DataBlock)> {
-        #[ctor]
-        static SHOW_VARIABLES_RULES: Vec<(Regex, Option<(TableSchemaRef, DataBlock)>)> = vec![
-            (
-                // sqlalchemy < 1.4.30
-                Regex::new("(?i)^(SHOW VARIABLES LIKE 'sql_mode'(.*))").unwrap(),
-                MySQLFederated::show_variables_block(
-                    "sql_mode",
-                    "ONLY_FULL_GROUP_BY STRICT_TRANS_TABLES NO_ZERO_IN_DATE NO_ZERO_DATE ERROR_FOR_DIVISION_BY_ZERO NO_ENGINE_SUBSTITUTION",
-                ),
-            ),
-            (
-                Regex::new("(?i)^(SHOW VARIABLES LIKE 'lower_case_table_names'(.*))").unwrap(),
-                MySQLFederated::show_variables_block("lower_case_table_names", "0"),
-            ),
-            (
-                Regex::new("(?i)^(show collation where(.*))").unwrap(),
-                MySQLFederated::show_variables_block("", ""),
-            ),
-            (
-                Regex::new("(?i)^(SHOW VARIABLES(.*))").unwrap(),
-                MySQLFederated::show_variables_block("", ""),
-            ),
-        ];
+        #![allow(clippy::type_complexity)]
+        static SHOW_VARIABLES_RULES: LazyLock<Vec<(Regex, Option<(TableSchemaRef, DataBlock)>)>> =
+            LazyLock::new(|| {
+                vec![
+                    (
+                        // sqlalchemy < 1.4.30
+                        Regex::new("(?i)^(SHOW VARIABLES LIKE 'sql_mode'(.*))").unwrap(),
+                        MySQLFederated::show_variables_block(
+                            "sql_mode",
+                            "ONLY_FULL_GROUP_BY STRICT_TRANS_TABLES NO_ZERO_IN_DATE NO_ZERO_DATE ERROR_FOR_DIVISION_BY_ZERO NO_ENGINE_SUBSTITUTION",
+                        ),
+                    ),
+                    (
+                        Regex::new("(?i)^(SHOW VARIABLES LIKE 'lower_case_table_names'(.*))")
+                            .unwrap(),
+                        MySQLFederated::show_variables_block("lower_case_table_names", "0"),
+                    ),
+                    (
+                        Regex::new("(?i)^(show collation where(.*))").unwrap(),
+                        MySQLFederated::show_variables_block("", ""),
+                    ),
+                    (
+                        Regex::new("(?i)^(SHOW VARIABLES(.*))").unwrap(),
+                        MySQLFederated::show_variables_block("", ""),
+                    ),
+                ]
+            });
 
         FederatedHelper::block_match_rule(query, &SHOW_VARIABLES_RULES)
     }
 
     // Check for SET or others query, this is the final check of the federated query.
     fn federated_mixed_check(&self, query: &str) -> Option<(TableSchemaRef, DataBlock)> {
-        #[ctor]
-        static MIXED_RULES: Vec<(Regex, Option<(TableSchemaRef, DataBlock)>)> = vec![
+        #![allow(clippy::type_complexity)]
+        static MIXED_RULES: LazyLock<Vec<(Regex, Option<(TableSchemaRef, DataBlock)>)>> =
+            LazyLock::new(|| {
+                vec![
             // Txn.
             (Regex::new("(?i)^(START(.*))").unwrap(), None),
             (Regex::new("(?i)^(SET NAMES(.*))").unwrap(), None),
@@ -235,7 +243,8 @@ impl MySQLFederated {
             (Regex::new("(?i)^(/\\*!40014 SET(.*) \\*/)$").unwrap(), None),
             (Regex::new("(?i)^(/\\*!40000 SET(.*) \\*/)$").unwrap(), None),
             (Regex::new("(?i)^(/\\*!40000 ALTER(.*) \\*/)$").unwrap(), None),
-        ];
+        ]
+            });
 
         FederatedHelper::block_match_rule(query, &MIXED_RULES)
     }

src/query/sql/Cargo.toml

@@ -24,7 +24,7 @@ chrono = { workspace = true }
 chrono-tz = { workspace = true }
 cidr = { version = "0.2.2" }
 cron = "0.12.0"
-ctor = "0.1.26"
+ctor = { workspace = true }
 dashmap = { workspace = true }
 databend-common-ast = { workspace = true }
 databend-common-async-functions = { workspace = true }