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
|
use crate::{server::Server, Id};
use async_std::{
net::UdpSocket,
sync::{Arc, RwLock},
task,
};
use std::{
collections::BTreeMap,
net::SocketAddr,
sync::atomic::{AtomicBool, Ordering},
};
pub struct Endpoint {
running: AtomicBool,
socket: UdpSocket,
bind: String,
clients: RwLock<BTreeMap<Id, Client>>,
}
pub struct Client {}
impl Endpoint {
pub async fn new(bind: &str) -> Arc<Self> {
let socket = UdpSocket::bind(bind).await.unwrap();
Arc::new(Self {
running: true.into(),
socket,
bind: bind.into(),
clients: Default::default(),
})
}
/// Stop the endpoint
pub fn stop(self: &Arc<Self>) {
self.running.store(false, Ordering::Relaxed);
}
pub async fn listen(self: &Arc<Self>, serv: Arc<Server>) {
let mut buf = vec![0; 1024];
info!("Listening for connections on {}", self.bind);
while self.running.load(Ordering::Relaxed) {
let (int, peer) = self.socket.recv_from(&mut buf).await.unwrap();
if int > 1024 {
warn!("Read a larger chunk than buffer?");
}
}
}
}
|