From b640cbd0ba73fbb2f79fda39813aabbbf4edff2c Mon Sep 17 00:00:00 2001
From: John Jin <ptma@163.com>
Date: Sat, 13 Jun 2026 17:03:27 +0800
Subject: [PATCH] fix: influxdb code removed on merge

Thanks for the fix!
---
 crates/dbx-core/src/schema.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/crates/dbx-core/src/schema.rs b/crates/dbx-core/src/schema.rs
index 3ee5b9c9f..c3ccf6744 100644
--- a/crates/dbx-core/src/schema.rs
+++ b/crates/dbx-core/src/schema.rs
@@ -292,6 +292,10 @@ async fn list_databases_once(state: &AppState, connection_id: &str) -> Result<Ve
             drop(connections);
             return db::clickhouse_driver::list_databases(&client).await;
         }
+        if let Some(client) = extract_pool!(&connections, connection_id, InfluxDb) {
+            drop(connections);
+            return db::influxdb_driver::list_databases(&client).await;
+        }
         try_sqlserver!(connections, connection_id, list_databases);
         if let Some(client) = extract_pool!(&connections, connection_id, Agent) {
             let is_mongo =
@@ -477,6 +481,12 @@ async fn list_tables_once(
                 .await
                 .map(|tables| filter_table_infos(tables, filter, limit, offset));
         }
+        if let Some(client) = extract_pool!(&connections, &pool_key, InfluxDb) {
+            drop(connections);
+            return db::influxdb_driver::list_tables(&client, database)
+                .await
+                .map(|tables| filter_table_infos(tables, filter, limit, offset));
+        }
         try_sqlserver!(connections, &pool_key, list_tables, schema, filter, limit, offset);
         if let Some(client) = extract_pool!(&connections, &pool_key, Agent) {
             let fallback_config = db_config.clone();
@@ -1110,6 +1120,10 @@ pub async fn get_columns_core(
                 .await
                 .map(deduplicate_column_infos);
         }
+        if let Some(client) = extract_pool!(&connections, &pool_key, InfluxDb) {
+            drop(connections);
+            return db::influxdb_driver::get_columns(&client, database, table).await.map(deduplicate_column_infos);
+        }
         try_sqlserver!(connections, &pool_key, get_columns, schema, table);
         if let Some(client) = extract_pool!(&connections, &pool_key, Agent) {
             let fallback_config = db_config.clone();