Breaks read loop on 'exit'

2 files changed, 16 insertions, 4 deletions

diff --git a/crates/gen_lsp_server/src/lib.rs b/crates/gen_lsp_server/src/lib.rs
index 86b72f663..edbdda6c8 100644
--- a/crates/gen_lsp_server/src/lib.rs
+++ b/crates/gen_lsp_server/src/lib.rs
@@ -22,7 +22,7 @@
 //!         sender,
 //!         main_loop,
 //!     )?;
-//!     io_threads.exit()?;
+//!     io_threads.join()?;
 //!     Ok(())
 //! }
 //!
diff --git a/crates/gen_lsp_server/src/stdio.rs b/crates/gen_lsp_server/src/stdio.rs
index c5fe5b83a..5edfbc39c 100644
--- a/crates/gen_lsp_server/src/stdio.rs
+++ b/crates/gen_lsp_server/src/stdio.rs
@@ -5,6 +5,7 @@ use std::{
 
 use crossbeam_channel::{bounded, Receiver, Sender};
 use failure::bail;
+use lsp_types::notification::Exit;
 
 use crate::{RawMessage, Result};
 
@@ -21,9 +22,18 @@ pub fn stdio_transport() -> (Receiver<RawMessage>, Sender<RawMessage>, Threads)
         let stdin = stdin();
         let mut stdin = stdin.lock();
         while let Some(msg) = RawMessage::read(&mut stdin)? {
+            let is_exit = match &msg {
+                RawMessage::Notification(n) => n.is::<Exit>(),
+                _ => false,
+            };
+
             if let Err(_) = reader_sender.send(msg) {
                 break;
             }
+
+            if is_exit {
+                break;
+            }
         }
         Ok(())
     });
@@ -37,9 +47,11 @@ pub struct Threads {
 }
 
 impl Threads {
-    pub fn exit(self) -> Result<()> {
-        // We can't rely on stdin being closed
-        drop(self.reader);
+    pub fn join(self) -> Result<()> {
+        match self.reader.join() {
+            Ok(r) => r?,
+            Err(_) => bail!("reader panicked"),
+        }
         match self.writer.join() {
             Ok(r) => r,
             Err(_) => bail!("writer panicked"),