github/dbx
4
0 Fork 0
mirror of https://github.com/t8y2/dbx.git synced 2026-11-13 22:47:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(query): set schema context (search_path) before query execution (#113)

Browse Source 
Add schema selector to the editor toolbar for schema-aware databases
(PostgreSQL, SQL Server, Oracle). When a schema is selected, the backend
sets the appropriate search_path before executing queries so users no
longer need to write schema-qualified table names.
This commit is contained in:
t8y2 committed 2026-05-06 18:22:41 +08:00
1 parent 0d1b99e3ee
commit b6a39c1fa0
15 files changed
+279 -32
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src-tauri/src/commands/query.rs
+16 -4
View File
@@ -14,13 +14,15 @@ pub async fn execute_query(
connection_id: String,
database: String,
sql: String,
schema: Option<String>,
execution_id: Option<String>,
) -> Result<db::QueryResult, String> {
let registered_query =
execution_id.as_ref().filter(|id| !id.trim().is_empty()).map(|id| state.running_queries.register(id.clone()));
let cancel_token = registered_query.as_ref().map(|query| query.token());
dbx_core::query::execute_sql_statement(&state, &connection_id, &database, &sql, cancel_token).await
dbx_core::query::execute_sql_statement(&state, &connection_id, &database, &sql, schema.as_deref(), cancel_token)
.await
}
#[tauri::command]
@@ -29,13 +31,14 @@ pub async fn execute_multi(
connection_id: String,
database: String,
sql: String,
schema: Option<String>,
execution_id: Option<String>,
) -> Result<Vec<db::QueryResult>, String> {
let registered_query =
execution_id.as_ref().filter(|id| !id.trim().is_empty()).map(|id| state.running_queries.register(id.clone()));
let cancel_token = registered_query.as_ref().map(|query| query.token());
dbx_core::query::execute_multi_core(&state, &connection_id, &database, &sql, cancel_token).await
dbx_core::query::execute_multi_core(&state, &connection_id, &database, &sql, schema.as_deref(), cancel_token).await
}
#[tauri::command]
@@ -49,8 +52,9 @@ pub async fn execute_batch(
connection_id: String,
database: String,
statements: Vec<String>,
schema: Option<String>,
) -> Result<db::QueryResult, String> {
dbx_core::query::execute_statements(&state, &connection_id, &database, &statements).await
dbx_core::query::execute_statements(&state, &connection_id, &database, &statements, schema.as_deref()).await
}
#[tauri::command]
@@ -59,6 +63,14 @@ pub async fn execute_script(
connection_id: String,
database: String,
sql: String,
schema: Option<String>,
) -> Result<db::QueryResult, String> {
dbx_core::query::execute_statements(&state, &connection_id, &database, &split_sql_statements(&sql)).await
dbx_core::query::execute_statements(
&state,
&connection_id,
&database,
&split_sql_statements(&sql),
schema.as_deref(),
)
.await
}