From 24364ab8b4acf48f19d72c1b6d06c20478635eaf Mon Sep 17 00:00:00 2001 From: hozan23 Date: Fri, 10 Nov 2023 11:29:30 +0300 Subject: p2p: use VecDeque for connection queue --- p2p/src/net/connection_queue.rs | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'p2p/src/net') diff --git a/p2p/src/net/connection_queue.rs b/p2p/src/net/connection_queue.rs index fbc4bfc..4c0de28 100644 --- a/p2p/src/net/connection_queue.rs +++ b/p2p/src/net/connection_queue.rs @@ -1,4 +1,4 @@ -use std::sync::Arc; +use std::{collections::VecDeque, sync::Arc}; use smol::{channel::Sender, lock::Mutex}; @@ -16,14 +16,14 @@ pub struct NewConn { /// Connection queue pub struct ConnQueue { - queue: Mutex>, + queue: Mutex>, conn_available: CondVar, } impl ConnQueue { pub fn new() -> Arc { Arc::new(Self { - queue: Mutex::new(Vec::new()), + queue: Mutex::new(VecDeque::new()), conn_available: CondVar::new(), }) } @@ -36,7 +36,7 @@ impl ConnQueue { conn, disconnect_signal, }; - self.queue.lock().await.push(new_conn); + self.queue.lock().await.push_back(new_conn); self.conn_available.signal(); let _ = chan.recv().await; } @@ -47,6 +47,6 @@ impl ConnQueue { while queue.is_empty() { queue = self.conn_available.wait(queue).await; } - queue.pop().unwrap() + queue.pop_front().unwrap() } } -- cgit v1.2.3