refactor server message handling

imgbot
Dmitry Sharshakov 3 years ago committed by Blaž Hrastnik
parent 541f7a0514
commit 9678df1c62

@ -167,36 +167,51 @@ impl Transport {
client_tx: &UnboundedSender<Payload>, client_tx: &UnboundedSender<Payload>,
msg: Payload, msg: Payload,
) -> Result<()> { ) -> Result<()> {
let (id, result) = match msg { match msg {
Payload::Response(Response { Payload::Response(Response {
success: true, ref success,
seq, ref seq,
request_seq,
..
}) => {
info!("<- DAP success ({}, in response to {})", seq, request_seq);
if let Payload::Response(val) = msg {
(request_seq, Ok(val))
} else {
unreachable!();
}
}
Payload::Response(Response {
success: false,
message,
body,
request_seq, request_seq,
command, ref command,
ref message,
ref body,
.. ..
}) => { }) => {
error!( let result = match success {
"<- DAP error {:?} ({:?}) for command #{} {}", true => {
message, body, request_seq, command info!("<- DAP success ({}, in response to {})", seq, request_seq);
); if let Payload::Response(val) = msg {
( Ok(val)
request_seq, } else {
Err(Error::Other(anyhow::format_err!("{:?}", body))), unreachable!();
) }
}
false => {
error!(
"<- DAP error {:?} ({:?}) for command #{} {}",
message, body, request_seq, command
);
Err(Error::Other(anyhow::format_err!("{:?}", body)))
}
};
let tx = self
.pending_requests
.lock()
.await
.remove(&request_seq)
.expect("pending_request with id not found!");
match tx.send(result).await {
Ok(_) => (),
Err(_) => error!(
"Tried sending response into a closed channel (id={:?}), original request likely timed out",
request_seq
),
};
Ok(())
} }
Payload::Request(Request { Payload::Request(Request {
ref command, ref command,
@ -205,33 +220,16 @@ impl Transport {
}) => { }) => {
info!("<- DAP request {} #{}", command, seq); info!("<- DAP request {} #{}", command, seq);
client_tx.send(msg).expect("Failed to send"); client_tx.send(msg).expect("Failed to send");
return Ok(()); Ok(())
} }
Payload::Event(Event { Payload::Event(Event {
ref event, ref seq, .. ref event, ref seq, ..
}) => { }) => {
info!("<- DAP event {} #{}", event, seq); info!("<- DAP event {} #{}", event, seq);
client_tx.send(msg).expect("Failed to send"); client_tx.send(msg).expect("Failed to send");
return Ok(()); Ok(())
} }
}; }
let tx = self
.pending_requests
.lock()
.await
.remove(&id)
.expect("pending_request with id not found!");
match tx.send(result).await {
Ok(_) => (),
Err(_) => error!(
"Tried sending response into a closed channel (id={:?}), original request likely timed out",
id
),
};
Ok(())
} }
async fn recv( async fn recv(

Loading…
Cancel
Save