rs-netty 0.2.3

A Tokio-native typed TCP/UDP pipeline framework inspired by Netty.
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

use std::future::Future;

use crate::{
    context::{BusinessContext, InboundContext, OutboundContext},
    traits::{Business, Flow, Inbound, Outbound},
    Result,
};

pub struct Identity;

pub struct Then<A, B> {
    pub a: A,
    pub b: B,
}

impl<A, B> Then<A, B> {
    pub fn new(a: A, b: B) -> Self {
        Self { a, b }
    }
}

pub trait InboundPipe<I>: Send + 'static {
    type Out: Send + 'static;

    fn process<'ctx>(
        &'ctx mut self,
        ctx: &'ctx mut InboundContext,
        msg: I,
    ) -> impl Future<Output = Result<Flow<Self::Out>>> + Send + 'ctx;
}

impl<I: Send + 'static> InboundPipe<I> for Identity {
    type Out = I;

    async fn process(&mut self, _ctx: &mut InboundContext, msg: I) -> Result<Flow<Self::Out>> {
        Ok(Flow::Next(msg))
    }
}

impl<I, A, B> InboundPipe<I> for Then<A, B>
where
    I: Send + 'static,
    A: InboundPipe<I>,
    B: Inbound<A::Out>,
{
    type Out = B::Out;

    async fn process(&mut self, ctx: &mut InboundContext, msg: I) -> Result<Flow<Self::Out>> {
        match self.a.process(ctx, msg).await? {
            Flow::Next(mid) => self.b.read(ctx, mid).await,
            Flow::Stop => Ok(Flow::Stop),
        }
    }
}

pub trait BusinessPipe<I>: Send + 'static {
    type Out: Send + 'static;

    fn process<'ctx>(
        &'ctx mut self,
        ctx: &'ctx mut BusinessContext,
        msg: I,
    ) -> impl Future<Output = Result<Flow<Self::Out>>> + Send + 'ctx;
}

impl<I: Send + 'static> BusinessPipe<I> for Identity {
    type Out = I;

    async fn process(&mut self, _ctx: &mut BusinessContext, msg: I) -> Result<Flow<Self::Out>> {
        Ok(Flow::Next(msg))
    }
}

impl<I, A, B> BusinessPipe<I> for Then<A, B>
where
    I: Send + 'static,
    A: BusinessPipe<I>,
    B: Business<A::Out>,
{
    type Out = B::Out;

    async fn process(&mut self, ctx: &mut BusinessContext, msg: I) -> Result<Flow<Self::Out>> {
        match self.a.process(ctx, msg).await? {
            Flow::Next(mid) => self.b.handle(ctx, mid).await,
            Flow::Stop => Ok(Flow::Stop),
        }
    }
}

pub trait OutboundPipe<I>: Send + 'static {
    type Out: Send + 'static;

    fn process<'ctx>(
        &'ctx mut self,
        ctx: &'ctx mut OutboundContext,
        msg: I,
    ) -> impl Future<Output = Result<Flow<Self::Out>>> + Send + 'ctx;
}

impl<I: Send + 'static> OutboundPipe<I> for Identity {
    type Out = I;

    async fn process(&mut self, _ctx: &mut OutboundContext, msg: I) -> Result<Flow<Self::Out>> {
        Ok(Flow::Next(msg))
    }
}

impl<I, A, B> OutboundPipe<I> for Then<A, B>
where
    I: Send + 'static,
    A: OutboundPipe<I>,
    B: Outbound<A::Out>,
{
    type Out = B::Out;

    async fn process(&mut self, ctx: &mut OutboundContext, msg: I) -> Result<Flow<Self::Out>> {
        match self.a.process(ctx, msg).await? {
            Flow::Next(mid) => self.b.write(ctx, mid).await,
            Flow::Stop => Ok(Flow::Stop),
        }
    }
}