feat: add support for Doris, StarRocks, and Redshift databases in connection handling and UI

src-tauri/src/commands/connection.rs

@@ -86,8 +86,8 @@ impl AppState {
         let (host, port) = self.connection_host_port(connection_id, &db_config).await?;
         let url = connection_url_for_endpoint(&db_config, &host, port);
         let pool = match db_config.db_type {
-            DatabaseType::Mysql => PoolKind::Mysql(db::mysql::connect(&url).await?),
-            DatabaseType::Postgres => PoolKind::Postgres(db::postgres::connect(&url).await?),
+            DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks => PoolKind::Mysql(db::mysql::connect(&url).await?),
+            DatabaseType::Postgres | DatabaseType::Redshift => PoolKind::Postgres(db::postgres::connect(&url).await?),
             DatabaseType::Sqlite => PoolKind::Sqlite(db::sqlite::connect(&url).await?),
             DatabaseType::Redis => {
                 let con = db::redis_driver::connect(&url).await?;
@@ -265,14 +265,14 @@ pub async fn test_connection(
         target
     );
     let result = match config.db_type {
-        DatabaseType::Mysql => match db::mysql::connect(&url).await {
+        DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks => match db::mysql::connect(&url).await {
             Ok(pool) => {
                 pool.close().await;
                 Ok("Connection successful".to_string())
             }
             Err(e) => Err(e),
         },
-        DatabaseType::Postgres => match db::postgres::connect(&url).await {
+        DatabaseType::Postgres | DatabaseType::Redshift => match db::postgres::connect(&url).await {
             Ok(pool) => {
                 pool.close().await;
                 Ok("Connection successful".to_string())
@@ -358,8 +358,8 @@ pub async fn connect_db(
     let url = connection_url_for_endpoint(&config, &host, port);
 
     let pool = match config.db_type {
-        DatabaseType::Mysql => PoolKind::Mysql(db::mysql::connect(&url).await?),
-        DatabaseType::Postgres => PoolKind::Postgres(db::postgres::connect(&url).await?),
+        DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks => PoolKind::Mysql(db::mysql::connect(&url).await?),
+        DatabaseType::Postgres | DatabaseType::Redshift => PoolKind::Postgres(db::postgres::connect(&url).await?),
         DatabaseType::Sqlite => PoolKind::Sqlite(db::sqlite::connect(&url).await?),
         DatabaseType::Redis => {
             let con = db::redis_driver::connect(&url).await?;

src-tauri/src/commands/transfer.rs

@@ -52,7 +52,7 @@ pub enum TransferStatus {
 
 fn quote_identifier(name: &str, db_type: &DatabaseType) -> String {
     match db_type {
-        DatabaseType::Mysql | DatabaseType::ClickHouse => format!("`{}`", name.replace('`', "``")),
+        DatabaseType::Mysql | DatabaseType::ClickHouse | DatabaseType::Doris | DatabaseType::StarRocks => format!("`{}`", name.replace('`', "``")),
         DatabaseType::SqlServer => format!("[{}]", name.replace(']', "]]")),
         _ => format!("\"{}\"", name.replace('"', "\"\"")),
     }
@@ -71,7 +71,7 @@ fn escape_value(val: &serde_json::Value, db_type: &DatabaseType) -> String {
     match val {
         serde_json::Value::Null => "NULL".to_string(),
         serde_json::Value::Bool(b) => match db_type {
-            DatabaseType::Mysql | DatabaseType::Sqlite | DatabaseType::DuckDb => {
+            DatabaseType::Mysql | DatabaseType::Sqlite | DatabaseType::DuckDb | DatabaseType::Doris | DatabaseType::StarRocks => {
                 if *b { "1".to_string() } else { "0".to_string() }
             }
             _ => if *b { "TRUE".to_string() } else { "FALSE".to_string() },

src-tauri/src/models/connection.rs

@@ -60,6 +60,10 @@ pub enum DatabaseType {
     Oracle,
     #[serde(rename = "elasticsearch")]
     Elasticsearch,
+    Doris,
+    #[serde(rename = "starrocks")]
+    StarRocks,
+    Redshift,
 }
 
 impl ConnectionConfig {
@@ -88,8 +92,8 @@ impl ConnectionConfig {
                 let scheme = if self.ssl { "rediss" } else { "redis" };
                 format!("{scheme}://{host}:{port}/")
             }
-            DatabaseType::Mysql => format!("mysql://{host}:{port}{db_part}?{params}"),
-            DatabaseType::Postgres => {
+            DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks => format!("mysql://{host}:{port}{db_part}?{params}"),
+            DatabaseType::Postgres | DatabaseType::Redshift => {
                 let suffix = if params.is_empty() {
                     String::new()
                 } else {
@@ -138,11 +142,11 @@ impl ConnectionConfig {
                     format!("{scheme}://{username}:{password}@{host}:{port}/")
                 }
             }
-            DatabaseType::Mysql => format!(
+            DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks => format!(
                 "mysql://{}:{}@{host}:{port}{db_part}?{params}",
                 username, password
             ),
-            DatabaseType::Postgres => {
+            DatabaseType::Postgres | DatabaseType::Redshift => {
                 let suffix = if params.is_empty() {
                     String::new()
                 } else {
@@ -180,7 +184,7 @@ impl ConnectionConfig {
     fn normalized_url_params(&self) -> String {
         let value = self.url_params.as_deref().unwrap_or("").trim();
         match self.db_type {
-            DatabaseType::Mysql => {
+            DatabaseType::Mysql | DatabaseType::Doris | DatabaseType::StarRocks => {
                 let base = "ssl-mode=preferred&charset=utf8mb4";
                 if value.is_empty() {
                     base.to_string()
@@ -192,7 +196,7 @@ impl ConnectionConfig {
                     if v.contains("charset=") { format!("ssl-mode=preferred&{v}") } else { format!("{base}&{v}") }
                 }
             }
-            DatabaseType::Postgres => value.trim_start_matches('?').to_string(),
+            DatabaseType::Postgres | DatabaseType::Redshift => value.trim_start_matches('?').to_string(),
             _ => value.trim_start_matches('?').to_string(),
         }
     }