fix: execute statement reliable

Author: juleswritescode

crates/pgt_lsp/src/handlers/code_actions.rs

@@ -1,3 +1,5 @@
+use std::sync::atomic::Ordering;
+
 use crate::{adapters::get_cursor_position, session::Session};
 use anyhow::{Result, anyhow};
 use tower_lsp::lsp_types::{
@@ -91,6 +93,8 @@ pub async fn execute_command(
                     path,
                 })?;
 
+            std::sync::atomic::fence(Ordering::SeqCst);
+
             /*
              * Updating all diagnostics: the changes caused by the statement execution
              * might affect many files.

crates/pgt_typecheck/src/lib.rs

@@ -42,7 +42,14 @@ pub async fn check_sql(params: TypecheckParams<'_>) -> Option<TypecheckDiagnosti
         return None;
     }
 
-    let res = params.conn.prepare(params.sql).await;
+    let mut conn = match params.conn.acquire().await {
+        Ok(c) => c,
+        Err(_) => return None,
+    };
+
+    conn.close_on_drop();
+
+    let res = conn.prepare(params.sql).await;
 
     match res {
         Ok(_) => None,