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(),
         }
     }

src/components/connection/ConnectionDialog.vue

@@ -9,8 +9,9 @@ import { Button } from "@/components/ui/button";
 import { Input } from "@/components/ui/input";
 import { Label } from "@/components/ui/label";
 import {
-  Select, SelectContent, SelectGroup, SelectItem, SelectLabel, SelectTrigger, SelectValue,
-} from "@/components/ui/select";
+  DropdownMenu, DropdownMenuContent, DropdownMenuTrigger,
+  DropdownMenuLabel,
+} from "@/components/ui/dropdown-menu";
 import type { ConnectionConfig, DatabaseType } from "@/types/database";
 import { useConnectionStore } from "@/stores/connectionStore";
 import DatabaseIcon from "@/components/icons/DatabaseIcon.vue";
@@ -91,8 +92,13 @@ const driverProfiles: Record<string, { type: DatabaseType; port: number; user: s
   gaussdb:    { type: "postgres",   port: 5432,  user: "gaussdb",  label: "GaussDB",     icon: "gaussdb" },
   kingbase:   { type: "postgres",   port: 54321, user: "system",   label: "KingBase",    icon: "kingbase" },
   vastbase:   { type: "postgres",   port: 5432,  user: "vastbase", label: "Vastbase",    icon: "vastbase" },
-  custom_mysql:    { type: "mysql",    port: 3306, user: "root",     label: "Custom MySQL-compatible",       icon: "mysql",    urlParams: "" },
-  custom_postgres: { type: "postgres", port: 5432, user: "postgres", label: "Custom PostgreSQL-compatible",  icon: "postgres", urlParams: "" },
+  doris:      { type: "mysql",      port: 9030,  user: "root",     label: "Doris",       icon: "doris" },
+  starrocks:  { type: "mysql",      port: 9030,  user: "root",     label: "StarRocks",   icon: "starrocks" },
+  redshift:   { type: "postgres",   port: 5439,  user: "awsuser",  label: "Redshift",    icon: "redshift" },
+  cockroachdb:{ type: "postgres",   port: 26257, user: "root",     label: "CockroachDB", icon: "cockroachdb" },
+  tdengine:   { type: "mysql",      port: 6030,  user: "root",     label: "TDengine",    icon: "tdengine" },
+  custom_mysql:    { type: "mysql",    port: 3306, user: "root",     label: "Custom",       icon: "mysql",    urlParams: "" },
+  custom_postgres: { type: "postgres", port: 5432, user: "postgres", label: "Custom",  icon: "postgres", urlParams: "" },
 };
 
 function profileForConfig(config: ConnectionConfig) {
@@ -166,9 +172,12 @@ watch(() => props.editConfig, (config) => {
 const isEditing = ref(false);
 watch(() => editingId.value, (v) => { isEditing.value = !!v; });
 
+const dbTypeMenuOpen = ref(false);
+
 function onDbTypeChange(val: string) {
   customDriverName.value = "";
   applyProfile(val, !!editingId.value);
+  dbTypeMenuOpen.value = false;
 }
 
 const iconTypeMap: Record<string, string> = {
@@ -178,6 +187,8 @@ const iconTypeMap: Record<string, string> = {
   elasticsearch: "elasticsearch",
   mariadb: "mariadb", tidb: "tidb", oceanbase: "oceanbase", goldendb: "goldendb",
   opengauss: "opengauss", gaussdb: "gaussdb", kingbase: "kingbase", vastbase: "vastbase",
+  doris: "doris", starrocks: "starrocks", redshift: "redshift",
+  cockroachdb: "cockroachdb", tdengine: "tdengine",
   custom_mysql: "mysql", custom_postgres: "postgres",
 };
 
@@ -199,7 +210,10 @@ const mysqlCompat = [
   { value: "tidb", label: "TiDB" },
   { value: "oceanbase", label: "OceanBase" },
   { value: "goldendb", label: "GoldenDB" },
-  { value: "custom_mysql", label: "Custom MySQL-compatible" },
+  { value: "doris", label: "Doris" },
+  { value: "starrocks", label: "StarRocks" },
+  { value: "tdengine", label: "TDengine" },
+  { value: "custom_mysql", label: "Custom" },
 ];
 
 const pgCompat = [
@@ -207,7 +221,9 @@ const pgCompat = [
   { value: "gaussdb", label: "GaussDB" },
   { value: "kingbase", label: "KingBase" },
   { value: "vastbase", label: "Vastbase" },
-  { value: "custom_postgres", label: "Custom PostgreSQL-compatible" },
+  { value: "redshift", label: "Redshift" },
+  { value: "cockroachdb", label: "CockroachDB" },
+  { value: "custom_postgres", label: "Custom" },
 ];
 
 watch(customDriverName, (value) => {
@@ -295,42 +311,57 @@ watch([() => editingId.value, () => open.value], () => {
 
         <div class="grid grid-cols-4 items-center gap-4">
           <Label class="text-right">{{ t('connection.type') }}</Label>
-          <Select :model-value="selectedType" @update:model-value="(val: any) => onDbTypeChange(String(val))">
-            <SelectTrigger class="col-span-3">
-              <div class="flex items-center gap-2">
+          <DropdownMenu v-model:open="dbTypeMenuOpen">
+            <DropdownMenuTrigger as-child>
+              <Button variant="outline" class="col-span-3 justify-start gap-2 font-normal">
                 <DatabaseIcon :db-type="iconTypeMap[selectedType] || selectedType" class="w-4 h-4" />
-                <SelectValue />
+                {{ selectedProfile().label }}
+              </Button>
+            </DropdownMenuTrigger>
+            <DropdownMenuContent class="w-auto p-3" align="start">
+              <div class="grid grid-cols-3 gap-3">
+                <div>
+                  <DropdownMenuLabel class="text-xs text-muted-foreground px-1 pb-1">{{ t('connection.mainstream') }}</DropdownMenuLabel>
+                  <button
+                    v-for="opt in dbOptions"
+                    :key="opt.value"
+                    class="flex w-full items-center gap-2 rounded-md px-2 py-1.5 text-[13px] hover:bg-accent cursor-default"
+                    :class="{ 'bg-accent': selectedType === opt.value }"
+                    @click="onDbTypeChange(opt.value)"
+                  >
+                    <DatabaseIcon :db-type="iconTypeMap[opt.value]" class="w-3.5 h-3.5" />
+                    {{ opt.label }}
+                  </button>
+                </div>
+                <div>
+                  <DropdownMenuLabel class="text-xs text-muted-foreground px-1 pb-1">MySQL {{ t('connection.compatible') }}</DropdownMenuLabel>
+                  <button
+                    v-for="opt in mysqlCompat"
+                    :key="opt.value"
+                    class="flex w-full items-center gap-2 rounded-md px-2 py-1.5 text-[13px] hover:bg-accent cursor-default"
+                    :class="{ 'bg-accent': selectedType === opt.value }"
+                    @click="onDbTypeChange(opt.value)"
+                  >
+                    <DatabaseIcon :db-type="iconTypeMap[opt.value]" class="w-3.5 h-3.5" />
+                    {{ opt.label }}
+                  </button>
+                </div>
+                <div>
+                  <DropdownMenuLabel class="text-xs text-muted-foreground px-1 pb-1">PostgreSQL {{ t('connection.compatible') }}</DropdownMenuLabel>
+                  <button
+                    v-for="opt in pgCompat"
+                    :key="opt.value"
+                    class="flex w-full items-center gap-2 rounded-md px-2 py-1.5 text-[13px] hover:bg-accent cursor-default"
+                    :class="{ 'bg-accent': selectedType === opt.value }"
+                    @click="onDbTypeChange(opt.value)"
+                  >
+                    <DatabaseIcon :db-type="iconTypeMap[opt.value]" class="w-3.5 h-3.5" />
+                    {{ opt.label }}
+                  </button>
+                </div>
               </div>
-            </SelectTrigger>
-            <SelectContent>
-              <SelectGroup>
-                <SelectItem v-for="opt in dbOptions" :key="opt.value" :value="opt.value">
-                  <div class="flex items-center gap-2">
-                    <DatabaseIcon :db-type="iconTypeMap[opt.value]" class="w-3.5 h-3.5" />
-                    {{ opt.label }}
-                  </div>
-                </SelectItem>
-              </SelectGroup>
-              <SelectGroup>
-                <SelectLabel class="text-xs text-muted-foreground">MySQL {{ t('connection.compatible') }}</SelectLabel>
-                <SelectItem v-for="opt in mysqlCompat" :key="opt.value" :value="opt.value">
-                  <div class="flex items-center gap-2">
-                    <DatabaseIcon :db-type="iconTypeMap[opt.value]" class="w-3.5 h-3.5" />
-                    {{ opt.label }}
-                  </div>
-                </SelectItem>
-              </SelectGroup>
-              <SelectGroup>
-                <SelectLabel class="text-xs text-muted-foreground">PostgreSQL {{ t('connection.compatible') }}</SelectLabel>
-                <SelectItem v-for="opt in pgCompat" :key="opt.value" :value="opt.value">
-                  <div class="flex items-center gap-2">
-                    <DatabaseIcon :db-type="iconTypeMap[opt.value]" class="w-3.5 h-3.5" />
-                    {{ opt.label }}
-                  </div>
-                </SelectItem>
-              </SelectGroup>
-            </SelectContent>
-          </Select>
+            </DropdownMenuContent>
+          </DropdownMenu>
         </div>
 
         <div v-if="isCustomCompatibleProfile()" class="grid grid-cols-4 items-center gap-4">

src/i18n/locales/en.ts

@@ -59,6 +59,7 @@ export default {
     sshPasswordPlaceholder: "Leave empty to use key",
     sshKeyPath: "Key Path",
     compatible: "Compatible",
+    mainstream: "Popular",
     color: "Color",
     colorNone: "No color",
     colorGreen: "Green",

src/i18n/locales/zh-CN.ts

@@ -61,6 +61,7 @@ export default {
     sshPasswordPlaceholder: "留空则使用密钥",
     sshKeyPath: "密钥路径",
     compatible: "兼容",
+    mainstream: "主流",
     color: "颜色",
     colorNone: "无颜色",
     colorGreen: "绿色",

src/stores/connectionStore.ts

@@ -42,6 +42,9 @@ export const useConnectionStore = defineStore("connection", () => {
       mongodb: "MongoDB",
       oracle: "Oracle",
       elasticsearch: "Elasticsearch",
+      doris: "Doris",
+      starrocks: "StarRocks",
+      redshift: "Redshift",
     };
     return {
       ...config,

src/types/database.ts

@@ -1,4 +1,4 @@
-export type DatabaseType = "mysql" | "postgres" | "sqlite" | "redis" | "duckdb" | "clickhouse" | "sqlserver" | "mongodb" | "oracle" | "elasticsearch";
+export type DatabaseType = "mysql" | "postgres" | "sqlite" | "redis" | "duckdb" | "clickhouse" | "sqlserver" | "mongodb" | "oracle" | "elasticsearch" | "doris" | "starrocks" | "redshift";
 
 export interface ConnectionConfig {
   id: string;