starred/windmill
1
0
Fork 0
mirror of https://github.com/windmill-labs/windmill.git synced 2026-08-24 16:01:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: move RESET ALL inside result_f async block

Browse Source 
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Alexander Petric
2026-04-03 20:33:09 +00:00
parent 3f55f226c3
commit 43d2cf4522
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/windmill-worker/src/pg_executor.rs
+6 -5
View File
@@ -357,11 +357,6 @@ pub async fn do_postgresql(
(client, Some(handle))
} else {
let (_, client) = mtex.as_ref().unwrap().as_ref().unwrap();
// Reset all session state (role, search_path, statement_timeout, etc.) to defaults
// to prevent state leaking between unrelated script executions sharing this cached connection.
if let Err(e) = client.simple_query("RESET ALL").await {
tracing::warn!("Failed to RESET ALL on cached connection: {}", e);
}
(client, None)
};
@@ -374,6 +369,12 @@ pub async fn do_postgresql(
let size = AtomicUsize::new(0);
let size_ref = &size;
let result_f = async move {
// Reset all session state (role, search_path, statement_timeout, etc.) to defaults
// to prevent state leaking between unrelated script executions sharing this cached connection.
if has_cached_con {
let _ = client.simple_query("RESET ALL").await;
}
let mut results = vec![];
for (i, query) in queries.iter().enumerate() {
if annotations.prepare {