with shutdown?

Author: juleswritescode

crates/pg_lsp/src/debouncer.rs

@@ -80,6 +80,7 @@ impl SimpleTokioDebouncer {
     pub async fn shutdown(&self) {
         self.shutdown_flag
             .store(true, std::sync::atomic::Ordering::Relaxed);
+
         let _ = self.handle.abort(); //  we don't care about any errors during shutdown
     }
 }

crates/pg_lsp/src/server.rs

@@ -227,13 +227,18 @@ impl LanguageServer for LspServer {
 
     #[tracing::instrument(name = "shutdown", skip(self))]
     async fn shutdown(&self) -> jsonrpc::Result<()> {
-        // self.session.shutdown().await;
-        // self.debouncer.shutdown().await;
+        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.")
             .await;
 
+        tracing::info!("Shutdown successful.");
+
         Ok(())
     }