the-fourth-server 0.3.0

A lightweight tcp server/client pair for network programming
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
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use crate::codec::codec_trait::TfCodec;
use crate::structures::transport::Transport;
use async_trait::async_trait;
use tokio_util::bytes::{BytesMut};
use tokio_util::codec::{ Framed};

#[async_trait]
///A traffic processor trait, that applied to all streams. Processes all stream. If you need setup by each specific stream, use codecs instead
pub trait TrafficProcess: Send + Sync {
    type Codec;
    ///The routine that defines if we can connect stream or not
    async fn initial_connect(&mut self, source: &mut Transport) -> bool;
    ///The routine that defines if we can connect stream or not, but when framed was setted up
    async fn initial_framed_connect(&mut self, source: &mut Framed<Transport, Self::Codec>)
    -> bool;
    ///Process every traffic that is handled by server
    async fn post_process_traffic(&mut self, data: Vec<u8>) -> Vec<u8>;
    ///Process every traffic that is handled by server
    async fn pre_process_traffic(&mut self, data: BytesMut) -> BytesMut;
    fn clone(&self) -> Box<dyn TrafficProcess<Codec = Self::Codec>>;
}

pub struct TrafficProcessorHolder<C>
where
    C: TfCodec,
{
    processors: Vec<Box<dyn TrafficProcess<Codec = C>>>,
}

impl<C> Clone for TrafficProcessorHolder<C>
where
    C: TfCodec,
{
    fn clone(&self) -> Self {
        let mut processors = Vec::new();
        self.processors
            .iter()
            .for_each(|p| processors.push(p.as_ref().clone()));

        Self { processors }
    }
}

impl<C> TrafficProcessorHolder<C>
where
    C: TfCodec,
{
    pub fn new() -> Self {
        TrafficProcessorHolder { processors: vec![] }
    }
    pub fn register_processor(&mut self, processor: Box<dyn TrafficProcess<Codec = C>>) {
        self.processors.push(processor);
    }

    pub async fn initial_connect(&mut self, source: &mut Transport) -> bool {
        for processor in self.processors.iter_mut() {
            if !processor.as_mut().initial_connect(source).await {
                return false;
            }
        }
        true
    }

    pub async fn initial_framed_connect(&mut self, source: &mut Framed<Transport, C>) -> bool {
        for processor in self.processors.iter_mut() {
            if !processor.as_mut().initial_framed_connect(source).await {
                return false;
            }
        }
        true
    }

    pub async fn post_process_traffic(&mut self, mut data: Vec<u8>) -> Vec<u8> {
        for proc in self.processors.iter_mut() {
            data = proc.post_process_traffic(data).await;
        }
        data
    }

    pub async fn pre_process_traffic(&mut self, mut data: BytesMut) -> BytesMut {
        for proc in self.processors.iter_mut() {
            data = proc.pre_process_traffic(data).await;
        }
        data
    }
}