lynn_tcp 1.1.17

Lightweight asynchronous TCP framework
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
35
36
37
38
39
40

use crate::{app::AsyncFunc, handler::IntoSystem};
use std::{
    collections::HashMap,
    sync::{Arc, Mutex},
};

pub(super) struct LynnRouter {
    pub(crate) map: *mut HashMap<u16, Arc<AsyncFunc>>,
    lock: Mutex<()>,
}

unsafe impl Send for LynnRouter {}
unsafe impl Sync for LynnRouter {}

impl LynnRouter {
    pub(super) fn new() -> Self {
        let map = HashMap::new();
        LynnRouter {
            map: Box::into_raw(Box::new(map)),
            lock: Mutex::new(()),
        }
    }

    pub(super) fn add_router<Param>(&self, method_id: u16, handler: impl IntoSystem<Param>) {
        if let Ok(_mutex) = self.lock.lock() {
            if let Some(map) = unsafe { self.map.as_mut() } {
                map.insert(method_id, Arc::new(Box::new(handler.to_system())));
            }
        }
    }

    pub(crate) fn get_handler_by_method_id(&self, method_id: &u16) -> Option<Arc<AsyncFunc>> {
        if let Some(map) = unsafe { self.map.as_ref() } {
            if map.contains_key(method_id) {
                return Some(map.get(method_id).unwrap().clone());
            }
        }
        None
    }
}