fix: execute statement reliable (#291)

Author: juleswritescode

crates/pgt_lsp/src/handlers/code_actions.rs

@@ -94,8 +94,6 @@ pub async fn execute_command(
             /*
              * Updating all diagnostics: the changes caused by the statement execution
              * might affect many files.
-             *
-             * TODO: in test.sql, this seems to work after create table, but not after drop table.
              */
             session.update_all_diagnostics().await;
 

crates/pgt_typecheck/src/lib.rs

@@ -42,7 +42,18 @@ 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,
+    };
+
+    // Postgres caches prepared statements within the current DB session (connection).
+    // This can cause issues if the underlying table schema changes while statements
+    // are cached. By closing the connection after use, we ensure a fresh state for
+    // each typecheck operation.
+    conn.close_on_drop();
+
+    let res = conn.prepare(params.sql).await;
 
     match res {
         Ok(_) => None,