Fix:lock unwrap cause panic

If streams cannot be locked, it will cause a panic Signed-off-by: jokemanfire <[email protected]>
containerd · Sep 25, 2024 · bf021aa · bf021aa
1 parent 152ac12
commit bf021aa
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/asynchronous/client.rs b/src/asynchronous/client.rs
@@ -74,8 +74,9 @@ impl Client {
 
         let (tx, mut rx): (ResultSender, ResultReceiver) = mpsc::channel(100);
 
-        // TODO: check return.
-        self.streams.lock().unwrap().insert(stream_id, tx);
+        self.streams.lock().map(|mut x| {
+            x.insert(stream_id, tx);
+        }).map_err(|_| Error::Others("Failed to acquire lock on streams".to_string()))?;
 
         self.req_tx
             .send(msg)
@@ -136,8 +137,10 @@ impl Client {
         }
 
         let (tx, rx): (ResultSender, ResultReceiver) = mpsc::channel(100);
-        // TODO: check return
-        self.streams.lock().unwrap().insert(stream_id, tx);
+        self.streams.lock().map(|mut x| {
+            x.insert(stream_id, tx);
+        }).map_err(|_| Error::Others("Failed to acquire lock on streams".to_string()))?;
+
         self.req_tx
             .send(msg)
             .await