sark-h2 0.26.2

Fiber-free HTTP/2 protocol primitives and manifold server for sark
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157

use std::convert::Infallible;

use o3::alloc;
use sark_core::{
    decode::{self, request},
    manifold::{
        schedule,
        server::{Event, app, drain, timing},
    },
};

use crate::{
    conn::config,
    server::{
        self,
        runtime::Connection,
        sync::{Book, Dispatch},
    },
};

#[doc(hidden)]
pub struct Route<R> {
    router: R,
    limits: server::Config,
}

impl<R> Route<R> {
    pub(in crate::server) const fn new(router: R, limits: server::Config) -> Self {
        Self { router, limits }
    }

    pub(super) const fn router(&self) -> &R {
        &self.router
    }

    pub(super) const fn limits(&self) -> server::Config {
        self.limits
    }
}

impl<R> app::Spec for Route<R>
where
    R: decode::Decode,
    R::Prepared: request::Synchronous,
{
    type Failure = Infallible;
    type Fields = R::Plan;
    type Egress<'d> = R::Egress<'d>;

    fn layout(&self) -> app::Layout {
        app::Layout::new(
            config::DEFAULT_STREAM_CAPACITY,
            self.limits.max_connection_body_bytes,
        )
    }
}

impl<'d, R, O> app::Application<'d, Connection<'d, Route<R>>, O> for Route<R>
where
    R: Dispatch<'d>,
    O: Copy,
{
    type State = Book<'d, R>;
    type Timing = timing::Off;
    type Build = alloc::Error;

    fn state(&self, capacity: app::Capacity) -> Result<Self::State, alloc::Error> {
        let streams = capacity.streams();
        Book::try_new(streams)
    }

    fn drive<H>(
        &mut self,
        state: &mut Self::State,
        carrier: &mut Connection<'d, Self>,
        turn: &mut app::Turn<'_, '_, '_, 'd, O>,
        egress: schedule::Egress,
        context: &mut H,
        _: impl FnMut(
            &mut H,
            &mut Connection<'d, Self>,
            &mut drain::Events<'_, 'd>,
            &Self::State,
        ) -> Result<Option<Event<Connection<'d, Self>>>, crate::conn::Error>,
        _: impl FnMut(&mut H, schedule::Wake<crate::stream::Key>),
        finish: impl FnOnce(&mut H, &mut Connection<'d, Self>),
        mut report: impl FnMut(&mut H, app::Error<crate::conn::Error, Self::Failure>),
    ) -> Result<drain::Progress, app::Error<crate::conn::Error, Self::Failure>> {
        let mut report = |error| report(context, app::Error::Connection(error));
        if matches!(egress, schedule::Egress::Released) {
            match Self::State::has_input(carrier) {
                true => {
                    self.router.pump::<true, _>(
                        state,
                        carrier,
                        self.limits,
                        egress,
                        || turn.work().permit(),
                        &mut report,
                    );
                }
                false => {
                    self.router.pump::<false, _>(
                        state,
                        carrier,
                        self.limits,
                        egress,
                        || turn.work().permit(),
                        &mut report,
                    );
                }
            }
        }
        let progress = self
            .router
            .drive(state, carrier, self.limits, turn.work(), &mut report);
        if !Self::State::has_events(carrier) {
            self.router.pump::<false, _>(
                state,
                carrier,
                self.limits,
                schedule::Egress::Unchanged,
                || turn.work().permit(),
                &mut report,
            );
        }
        if Self::State::has_events(carrier) || state.has_ready() {
            turn.wake().wake();
        }
        finish(context, carrier);
        Ok(progress)
    }

    fn close(&mut self, state: &mut Self::State, carrier: &mut Connection<'d, Self>, _: O) {
        self.router.clear(state, carrier);
    }

    fn selected(_: &Self::State, _: &Event<Connection<'d, Self>>) -> bool {
        true
    }

    fn wake(state: &mut Self::State, wake: schedule::Wake<crate::stream::Key>) -> bool {
        state.runtime.wake(wake) != 0
    }

    fn released(state: &mut Self::State) {
        state.runtime.wake_egress();
    }

    fn activity(state: &Self::State) -> schedule::Quiescence {
        if state.runtime.is_empty() {
            schedule::Quiescence::Quiet
        } else {
            schedule::Quiescence::Busy
        }
    }
}