kcpserver 0.3.0

fast kcp server frame.
Documentation 
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

use std::net::SocketAddr;
use std::option::Option::Some;
use tokio::sync::mpsc::{unbounded_channel, UnboundedReceiver, UnboundedSender};
use tokio::net::UdpSocket;
use std::sync::Arc;

pub type SendUDP = UnboundedSender<(Vec<u8>, SocketAddr)>;

pub struct SendPool {
    mpsc_sender: SendUDP,
}

unsafe impl Send for SendPool {}
unsafe impl Sync for SendPool {}

impl SendPool {
    pub fn new(udp_send: Arc<UdpSocket>) -> SendPool {
        let (tx, rx) = unbounded_channel();
        Self::recv(rx, udp_send);
        SendPool { mpsc_sender: tx }
    }

    pub fn get_tx(&self) -> SendUDP {
        self.mpsc_sender.clone()
    }

    fn recv(mut mpsc_receiver: UnboundedReceiver<(Vec<u8>, SocketAddr)>, udp_send: Arc<UdpSocket>) {
        tokio::spawn(async move {
            while let Some((data, addr)) = mpsc_receiver.recv().await {
                let _ = udp_send.send_to(&data, &addr).await;
            }
        });
    }
}