remove unwraps

This commit is contained in:
lc
2025-11-13 23:06:40 +08:00
parent 3e1e58747a
commit f5f78c84d5
2 changed files with 21 additions and 7 deletions
+7 -5
View File
@@ -102,13 +102,15 @@ async fn read_loop(mut stream: hbb_common::Stream) -> Result<()> {
println!("WebRTC stream closed; Exit the read_loop"); println!("WebRTC stream closed; Exit the read_loop");
return Ok(()); return Ok(());
}; };
if res.is_err() { match res {
println!("WebRTC stream read error: {}; Exit the read_loop", res.err().unwrap()); Err(e) => {
println!("WebRTC stream read error: {}; Exit the read_loop", e);
return Ok(()); return Ok(());
} }
println!("Message from stream: {}", Ok(data) => {
String::from_utf8(res.unwrap().to_vec())? println!("Message from stream: {}", String::from_utf8(data.to_vec())?);
); }
}
} }
} }
+13 -1
View File
@@ -105,7 +105,19 @@ impl WebRTCStream {
let contains = lock.contains_key(&key); let contains = lock.contains_key(&key);
if contains { if contains {
log::debug!("Start webrtc with cached peer"); log::debug!("Start webrtc with cached peer");
return Ok(lock.get(&key).unwrap().clone()); return Ok(lock[&key].clone());
}
// ...existing code...
Self::get_remote_offer(remote_endpoint)?
};
let key = remote_offer.clone();
let mut lock = SESSIONS.lock().await;
if let Some(cached_stream) = lock.get(&key) {
if !key.is_empty() {
log::debug!("Start webrtc with cached peer");
return Ok(cached_stream.clone());
}
} }
// Create a SettingEngine and enable Detach // Create a SettingEngine and enable Detach