kcpserver 0.3.2

use super::super::kcp_module::{Kcp, KcpResult};
use crate::udp::{RecvType, TokenStore};
use async_mutex::Mutex;
use log::*;
use std::cell::RefCell;
use std::net::SocketAddr;
use std::sync::atomic::{AtomicI64, AtomicU32, Ordering};
use tokio::sync::mpsc::Sender;

/// KCP LOCK
/// 将KCP 对象操作完全上锁,以保证内存安全 通知简化调用
pub struct KcpLock(Mutex<Kcp>);
unsafe impl Send for KcpLock {}
unsafe impl Sync for KcpLock {}

impl KcpLock {
    #[inline]
    pub async fn peeksize(&self) -> KcpResult<usize> {
        self.0.lock().await.peeksize()
    }

    #[inline]
    pub async fn check(&self, current: u32) -> u32 {
        self.0.lock().await.check(current)
    }

    #[inline]
    pub async fn input(&self, buf: &[u8]) -> KcpResult<usize> {
        self.0.lock().await.input(buf)
    }

    #[inline]
    pub async fn recv(&self, buf: &mut [u8]) -> KcpResult<usize> {
        self.0.lock().await.recv(buf)
    }

    #[inline]
    pub async fn send(&self, buf: &[u8]) -> KcpResult<usize> {
        self.0.lock().await.send(buf)
    }

    #[inline]
    pub async fn update(&self, current: u32) -> KcpResult<u32> {
        let mut p = self.0.lock().await;
        p.update(current)?;
        Ok(p.check(current))
    }

    #[inline]
    pub async fn flush(&self) -> KcpResult<()> {
        self.0.lock().await.flush()
    }

    #[inline]
    pub async fn flush_async(&self) -> KcpResult<()> {
        self.0.lock().await.flush_async()
    }
}

impl Kcp {
    #[inline]
    pub fn get_lock(self) -> KcpLock {
        let recv = Mutex::new(self);
        KcpLock(recv)
    }
}

pub type DisconnectFnStore = Mutex<Option<Box<dyn FnOnce(u32)>>>;

/// KCP Peer
/// UDP的包进入 KCP PEER 经过KCP 处理后 输出
/// 输入的包 进入KCP PEER处理,然后 输出到UDP PEER SEND TO
/// 同时还需要一个UPDATE 线程 去10MS 一次的运行KCP UPDATE
/// token 用于扩赞逻辑上下文
pub struct KcpPeer<T> {
    pub kcp: KcpLock,
    pub conv: u32,
    pub addr: SocketAddr,
    pub token: RefCell<TokenStore<T>>,
    pub last_rev_time: AtomicI64,
    pub next_update_time: AtomicU32,
    pub disconnect_event: DisconnectFnStore,
    pub(crate) main_tx: Sender<RecvType>,
}

impl<T> Drop for KcpPeer<T> {
    fn drop(&mut self) {
        debug!("kcp_peer:{} is Drop", self.conv);
    }
}

unsafe impl<T> Send for KcpPeer<T> {}
unsafe impl<T> Sync for KcpPeer<T> {}

/// 简化KCP PEER 函数
impl<T: Send> KcpPeer<T> {
    pub fn disconnect(&self) {
        if let Some(mut call_value)=self.disconnect_event.try_lock() {
            if let Some(call) = call_value.take() {
                call(self.conv);
            }
        }
    }

    pub(crate) async fn peeksize(&self) -> KcpResult<usize> {
        self.kcp.peeksize().await
    }

    pub(crate) async fn input(&self, buf: &[u8]) -> KcpResult<usize> {
        self.next_update_time.store(0, Ordering::Release);
        self.kcp.input(buf).await
    }

    pub(crate) async fn recv(&self, buf: &mut [u8]) -> KcpResult<usize> {
        self.kcp.recv(buf).await
    }

    pub async fn send(&self, buf: &[u8]) -> KcpResult<usize> {
        self.next_update_time.store(0, Ordering::Release);
        self.kcp.send(buf).await
    }

    pub(crate) async fn update(&self, current: u32) -> KcpResult<()> {
        let next = self.kcp.update(current).await?;
        self.next_update_time
            .store(next + current, Ordering::Release);
        Ok(())
    }

    pub async fn flush(&self) -> KcpResult<()> {
        self.kcp.flush().await
    }

    pub async fn flush_async(&self) -> KcpResult<()> {
        self.kcp.flush_async().await
    }

    pub fn get_main_tx(&self) -> Sender<RecvType> {
        self.main_tx.clone()
    }
}