language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-11-19 19:44:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

lsp-server: Drop outgoing messages on background thread

Browse source 
Closes #18961
This commit is contained in:
Piotr Osiewicz 2025-01-18 17:30:30 +01:00
parent 8364ef2997
commit b57157ea94
3 changed files with 42 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

23
lib/lsp-server/src/socket.rs
View file

@ -15,8 +15,11 @@ pub(crate) fn socket_transport(
stream: TcpStream,
) -> (Sender<Message>, Receiver<Message>, IoThreads) {
let (reader_receiver, reader) = make_reader(stream.try_clone().unwrap());
let (writer_sender, writer) = make_write(stream);
let io_threads = make_io_threads(reader, writer);
let (writer_sender, writer, messages_to_drop) = make_write(stream);
let dropper = std::thread::spawn(move || {
messages_to_drop.into_iter().for_each(drop);
});
let io_threads = make_io_threads(reader, writer, dropper);
(writer_sender, reader_receiver, io_threads)
}
@ -36,11 +39,21 @@ fn make_reader(stream: TcpStream) -> (Receiver<Message>, thread::JoinHandle<io::
(reader_receiver, reader)
}
fn make_write(mut stream: TcpStream) -> (Sender<Message>, thread::JoinHandle<io::Result<()>>) {
fn make_write(
mut stream: TcpStream,
) -> (Sender<Message>, thread::JoinHandle<io::Result<()>>, Receiver<Message>) {
let (writer_sender, writer_receiver) = bounded::<Message>(0);
let (drop_sender, drop_receiver) = bounded::<Message>(0);
let writer = thread::spawn(move || {
writer_receiver.into_iter().try_for_each(|it| it.write(&mut stream)).unwrap();
writer_receiver
.into_iter()
.try_for_each(|it| {
let result = it.write(&mut stream);
let _ = drop_sender.send(it);
result
})
.unwrap();
Ok(())
});
(writer_sender, writer)
(writer_sender, writer, drop_receiver)
}