fix: influxdb code removed on merge

Thanks for the fix!
This commit is contained in:
John Jin authored and GitHub committed 2026-06-13 17:03:27 +08:00
1 parent 48ba76bc9a
commit b640cbd0ba
1 file changed
+14
+14
View File
@@ -292,6 +292,10 @@ async fn list_databases_once(state: &AppState, connection_id: &str) -> Result<Ve
drop(connections); drop(connections);
return db::clickhouse_driver::list_databases(&client).await; 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); try_sqlserver!(connections, connection_id, list_databases);
if let Some(client) = extract_pool!(&connections, connection_id, Agent) { if let Some(client) = extract_pool!(&connections, connection_id, Agent) {
let is_mongo = let is_mongo =
@@ -477,6 +481,12 @@ async fn list_tables_once(
.await .await
.map(|tables| filter_table_infos(tables, filter, limit, offset)); .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); try_sqlserver!(connections, &pool_key, list_tables, schema, filter, limit, offset);
if let Some(client) = extract_pool!(&connections, &pool_key, Agent) { if let Some(client) = extract_pool!(&connections, &pool_key, Agent) {
let fallback_config = db_config.clone(); let fallback_config = db_config.clone();
@@ -1110,6 +1120,10 @@ pub async fn get_columns_core(
.await .await
.map(deduplicate_column_infos); .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); try_sqlserver!(connections, &pool_key, get_columns, schema, table);
if let Some(client) = extract_pool!(&connections, &pool_key, Agent) { if let Some(client) = extract_pool!(&connections, &pool_key, Agent) {
let fallback_config = db_config.clone(); let fallback_config = db_config.clone();