aboutsummaryrefslogtreecommitdiff
path: root/p2p/src/conn_queue.rs
blob: 9a153f3dd9f4c14d011c54366b27d512bbae8597 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
use std::{collections::VecDeque, fmt, sync::Arc};

use async_channel::Sender;

use karyon_core::{async_runtime::lock::Mutex, async_util::CondVar};
use karyon_net::Conn;

use crate::{message::NetMsg, Result};

/// Defines the direction of a network connection.
#[derive(Clone, Debug)]
pub enum ConnDirection {
    Inbound,
    Outbound,
}

impl fmt::Display for ConnDirection {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            ConnDirection::Inbound => write!(f, "Inbound"),
            ConnDirection::Outbound => write!(f, "Outbound"),
        }
    }
}

pub struct NewConn {
    pub direction: ConnDirection,
    pub conn: Conn<NetMsg>,
    pub disconnect_signal: Sender<Result<()>>,
}

/// Connection queue
pub struct ConnQueue {
    queue: Mutex<VecDeque<NewConn>>,
    conn_available: CondVar,
}

impl ConnQueue {
    pub fn new() -> Arc<Self> {
        Arc::new(Self {
            queue: Mutex::new(VecDeque::new()),
            conn_available: CondVar::new(),
        })
    }

    /// Push a connection into the queue and wait for the disconnect signal
    pub async fn handle(&self, conn: Conn<NetMsg>, direction: ConnDirection) -> Result<()> {
        let (disconnect_signal, chan) = async_channel::bounded(1);
        let new_conn = NewConn {
            direction,
            conn,
            disconnect_signal,
        };
        self.queue.lock().await.push_back(new_conn);
        self.conn_available.signal();
        if let Ok(result) = chan.recv().await {
            return result;
        }
        Ok(())
    }

    /// Receive the next connection in the queue
    pub async fn next(&self) -> NewConn {
        let mut queue = self.queue.lock().await;
        while queue.is_empty() {
            queue = self.conn_available.wait(queue).await;
        }
        queue.pop_front().unwrap()
    }
}