fix: dangling process on exit

Author: LastLeaf

src/lib.rs

@@ -527,6 +527,9 @@ async fn serve() -> anyhow::Result<()> {
                         // log::warn!("Missing LSP response handler for {:?}", id);
                     }
                     Message::Notification(note) => {
+                        if note.method == "exit" {
+                            break;
+                        }
                         if let Err(err) = handle_notification(ctx, note).await {
                             log::error!("{}", err);
                         }
@@ -536,7 +539,13 @@ async fn serve() -> anyhow::Result<()> {
         });
         tokio::task::spawn_blocking(move || {
             while let Ok(msg) = lsp_receiver.recv() {
+                let need_exit = if let Message::Notification(note) = &msg {
+                    note.method == "exit"
+                } else {
+                    false
+                };
                 sender.send(msg).unwrap();
+                if need_exit { break; }
             }
         })
     };

src/main.rs

@@ -10,6 +10,10 @@ fn main() -> anyhow::Result<()> {
         .enable_all()
         .build()
         .unwrap();
-    runtime.block_on(async { glass_easel_analyzer::run().await })?;
-    Ok(())
+    if let Err(err) = runtime.block_on(async { glass_easel_analyzer::run().await }) {
+        eprintln!("{}", err);
+        std::process::exit(1);
+    } else {
+        std::process::exit(0);
+    }
 }

vscode-extension/src/client.ts

@@ -83,11 +83,7 @@ export class Client {
     const run: Executable = {
       command,
       args,
-      options: {
-        env: {
-          RUST_BACKTRACE: '1',
-        },
-      },
+      options: {},
     }
     const debug: Executable = {
       command,
@@ -155,5 +151,6 @@ export class Client {
     this.tsServerHost?.destroy()
     this.tsServerHost = null
     await this.client?.stop()
+    await this.client?.dispose()
   }
 }