language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-09-28 21:05:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Use jod_thread

Browse source
This commit is contained in:
Edwin Cheng 2020-03-28 20:52:45 +08:00
parent 39706a5786
commit 7f7a16675d
3 changed files with 16 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

31
crates/ra_proc_macro/src/process.rs
View file

@ -12,7 +12,6 @@ use std::{
io::{self, Write},
path::{Path, PathBuf},
process::{Child, Command, Stdio},
thread::{spawn, JoinHandle},
};
#[derive(Debug, Default)]
@ -22,8 +21,18 @@ pub(crate) struct ProcMacroProcessSrv {
#[derive(Debug)]
pub(crate) struct ProcMacroProcessThread {
handle: Option<JoinHandle<()>>,
sender: Sender<Task>,
// XXX: drop order is significant
sender: SenderGuard,
handle: jod_thread::JoinHandle<()>,
}
#[derive(Debug)]
struct SenderGuard(pub Sender<Task>);
impl std::ops::Drop for SenderGuard {
fn drop(&mut self) {
let _ = self.0.send(Task::Close);
}
}
enum Task {
@ -62,18 +71,6 @@ impl Process {
}
}
impl std::ops::Drop for ProcMacroProcessThread {
fn drop(&mut self) {
if let Some(handle) = self.handle.take() {
let _ = self.sender.send(Task::Close);
// Join the thread, it should finish shortly. We don't really care
// whether it panicked, so it is safe to ignore the result
let _ = handle.join();
}
}
}
impl ProcMacroProcessSrv {
pub fn run(
process_path: &Path,
@ -81,12 +78,12 @@ impl ProcMacroProcessSrv {
let process = Process::run(process_path)?;
let (task_tx, task_rx) = bounded(0);
let handle = spawn(move || {
let handle = jod_thread::spawn(move || {
client_loop(task_rx, process);
});
let srv = ProcMacroProcessSrv { inner: Some(task_tx.clone()) };
let thread = ProcMacroProcessThread { handle: Some(handle), sender: task_tx };
let thread = ProcMacroProcessThread { handle, sender: SenderGuard(task_tx) };
Ok((thread, srv))
}