Fix Peer request map

Author: Rigidity

crates/chia-sdk-client/src/request_map.rs

@@ -1,12 +1,14 @@
-use std::{collections::HashMap, sync::Arc};
+use std::{
+    collections::HashMap,
+    sync::atomic::{AtomicU16, Ordering},
+};
 
 use chia_protocol::Message;
-use tokio::sync::{oneshot, Mutex, OwnedSemaphorePermit, Semaphore};
+use tokio::sync::{oneshot, Mutex};
 
 #[derive(Debug)]
 pub(crate) struct Request {
     sender: oneshot::Sender<Message>,
-    _permit: OwnedSemaphorePermit,
 }
 
 impl Request {
@@ -17,41 +19,27 @@ impl Request {
 
 #[derive(Debug)]
 pub(crate) struct RequestMap {
+    next_id: AtomicU16,
     items: Mutex<HashMap<u16, Request>>,
-    semaphore: Arc<Semaphore>,
 }
 
 impl RequestMap {
     pub(crate) fn new() -> Self {
         Self {
+            next_id: AtomicU16::new(0),
             items: Mutex::new(HashMap::new()),
-            semaphore: Arc::new(Semaphore::new(u16::MAX as usize)),
         }
     }
 
     pub(crate) async fn insert(&self, sender: oneshot::Sender<Message>) -> u16 {
-        let permit = self
-            .semaphore
-            .clone()
-            .acquire_owned()
-            .await
-            .expect("semaphore closed");
-
         let mut items = self.items.lock().await;
 
         items.retain(|_, v| !v.sender.is_closed());
 
-        let index = (0..=u16::MAX)
-            .find(|i| !items.contains_key(i))
-            .expect("exceeded expected number of requests");
-
-        items.insert(
-            index,
-            Request {
-                sender,
-                _permit: permit,
-            },
-        );
+        let index = self.next_id.fetch_add(0, Ordering::SeqCst);
+
+        items.insert(index, Request { sender });
+
         index
     }