fix: report db connection errors (#348)

* fix: report db connection errors

* Update lib.rs

Author: psteinroe

crates/pgt_typecheck/src/lib.rs

@@ -3,10 +3,9 @@ mod diagnostics;
 pub use diagnostics::TypecheckDiagnostic;
 use diagnostics::create_type_error;
 use pgt_text_size::TextRange;
-use sqlx::Executor;
-use sqlx::PgPool;
 use sqlx::postgres::PgDatabaseError;
 pub use sqlx::postgres::PgSeverity;
+use sqlx::{Executor, PgPool};
 
 #[derive(Debug)]
 pub struct TypecheckParams<'a> {
@@ -29,7 +28,9 @@ pub struct TypeError {
     pub constraint: Option<String>,
 }
 
-pub async fn check_sql(params: TypecheckParams<'_>) -> Option<TypecheckDiagnostic> {
+pub async fn check_sql(
+    params: TypecheckParams<'_>,
+) -> Result<Option<TypecheckDiagnostic>, sqlx::Error> {
     // Check if the AST is not a supported statement type
     if !matches!(
         params.ast,
@@ -39,13 +40,10 @@ pub async fn check_sql(params: TypecheckParams<'_>) -> Option<TypecheckDiagnosti
             | pgt_query_ext::NodeEnum::DeleteStmt(_)
             | pgt_query_ext::NodeEnum::CommonTableExpr(_)
     ) {
-        return None;
+        return Ok(None);
     }
 
-    let mut conn = match params.conn.acquire().await {
-        Ok(c) => c,
-        Err(_) => return None,
-    };
+    let mut conn = params.conn.acquire().await?;
 
     // Postgres caches prepared statements within the current DB session (connection).
     // This can cause issues if the underlying table schema changes while statements
@@ -56,11 +54,11 @@ pub async fn check_sql(params: TypecheckParams<'_>) -> Option<TypecheckDiagnosti
     let res = conn.prepare(params.sql).await;
 
     match res {
-        Ok(_) => None,
+        Ok(_) => Ok(None),
         Err(sqlx::Error::Database(err)) => {
             let pg_err = err.downcast_ref::<PgDatabaseError>();
-            Some(create_type_error(pg_err, params.tree))
+            Ok(Some(create_type_error(pg_err, params.tree)))
         }
-        Err(_) => None,
+        Err(err) => Err(err),
     }
 }

crates/pgt_typecheck/tests/diagnostics.rs

@@ -37,7 +37,7 @@ async fn test(name: &str, query: &str, setup: &str) {
 
     Formatter::new(&mut writer)
         .write_markup(markup! {
-            {PrintDiagnostic::simple(&result.unwrap())}
+            {PrintDiagnostic::simple(&result.unwrap().unwrap())}
         })
         .unwrap();
 

crates/pgt_workspace/src/workspace/server.rs

@@ -360,8 +360,6 @@ impl Workspace for WorkspaceServer {
 
         let mut diagnostics: Vec<SDiagnostic> = parser.document_diagnostics().to_vec();
 
-        // TODO: run this in parallel with rayon based on rayon.count()
-
         if let Some(pool) = self
             .connection
             .read()
@@ -385,13 +383,15 @@ impl Workspace for WorkspaceServer {
                                 })
                                 .await
                                 .map(|d| {
-                                    let r = d.location().span.map(|span| span + range.start());
+                                    d.map(|d| {
+                                        let r = d.location().span.map(|span| span + range.start());
 
-                                    d.with_file_path(path.as_path().display().to_string())
-                                        .with_file_span(r.unwrap_or(range))
+                                        d.with_file_path(path.as_path().display().to_string())
+                                            .with_file_span(r.unwrap_or(range))
+                                    })
                                 })
                             } else {
-                                None
+                                Ok(None)
                             }
                         }
                     })
@@ -400,8 +400,11 @@ impl Workspace for WorkspaceServer {
                     .await
             })?;
 
-            for result in async_results.into_iter().flatten() {
-                diagnostics.push(SDiagnostic::new(result));
+            for result in async_results.into_iter() {
+                let result = result?;
+                if let Some(diag) = result {
+                    diagnostics.push(SDiagnostic::new(diag));
+                }
             }
         }