zed-industries/zed
1
1
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-11-08 07:35:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Avoid retaining executor when using Connection::in_memory

Browse Source
This commit is contained in:
Max Brunsfeld 2022-03-01 18:02:12 -08:00
parent 95b2f4fb16
commit acf7ef3d61
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
crates/rpc/src/conn.rs
View File

@ -59,18 +59,21 @@ impl Connection {
) {
use futures::channel::mpsc;
use io::{Error, ErrorKind};
use std::sync::Arc;
let (tx, rx) = mpsc::unbounded::<WebSocketMessage>();
let tx = tx
.sink_map_err(|e| WebSocketError::from(Error::new(ErrorKind::Other, e)))
.with({
let executor = Arc::downgrade(&executor);
let kill_rx = kill_rx.clone();
let executor = executor.clone();
move |msg| {
let kill_rx = kill_rx.clone();
let executor = executor.clone();
Box::pin(async move {
executor.simulate_random_delay().await;
if let Some(executor) = executor.upgrade() {
executor.simulate_random_delay().await;
}
if kill_rx.borrow().is_none() {
Ok(msg)
} else {
@ -80,9 +83,11 @@ impl Connection {
}
});
let rx = rx.then(move |msg| {
let executor = executor.clone();
let executor = Arc::downgrade(&executor);
Box::pin(async move {
executor.simulate_random_delay().await;
if let Some(executor) = executor.upgrade() {
executor.simulate_random_delay().await;
}
msg
})
});