shutdown logs

Author: juleswritescode

crates/pg_lsp/src/db_connection.rs

@@ -65,6 +65,7 @@ impl DbConnection {
         connection_string == self.connection_string
     }
 
+    #[tracing::instrument(name = "Closing DB Pool", skip(self))]
     pub(crate) async fn close(self) {
         let _ = self.close_tx.send(());
         let _ = self.schema_update_handle.await;

crates/pg_lsp/src/server.rs

@@ -227,18 +227,16 @@ impl LanguageServer for LspServer {
 
     #[tracing::instrument(name = "shutdown", skip(self))]
     async fn shutdown(&self) -> jsonrpc::Result<()> {
-        tracing::info!("Shutting down session...");
         self.session.shutdown().await;
-
-        tracing::info!("Shutting down debouncer...");
         self.debouncer.shutdown().await;
 
         self.client
-            .log_message(MessageType::INFO, "Postgres LSP terminated.")
+            .send_notification::<ShowMessage>(ShowMessageParams {
+                message: "Shutdown successful.".into(),
+                typ: MessageType::INFO,
+            })
             .await;
 
-        tracing::info!("Shutdown successful.");
-
         Ok(())
     }
 

crates/pg_lsp/src/session.rs

@@ -28,6 +28,7 @@ impl Session {
         }
     }
 
+    #[tracing::instrument(name = "Shutting down Session", skip(self))]
     pub async fn shutdown(&self) {
         let mut db = self.db.write().await;
         let db = db.take();