actori_http/h1/
mod.rs

1//! HTTP/1 implementation
2use bytes::{Bytes, BytesMut};
3
4mod client;
5mod codec;
6mod decoder;
7mod dispatcher;
8mod encoder;
9mod expect;
10mod payload;
11mod service;
12mod upgrade;
13mod utils;
14
15pub use self::client::{ClientCodec, ClientPayloadCodec};
16pub use self::codec::Codec;
17pub use self::dispatcher::Dispatcher;
18pub use self::expect::ExpectHandler;
19pub use self::payload::Payload;
20pub use self::service::{H1Service, H1ServiceHandler, OneRequest};
21pub use self::upgrade::UpgradeHandler;
22pub use self::utils::SendResponse;
23
24#[derive(Debug)]
25/// Codec message
26pub enum Message<T> {
27    /// Http message
28    Item(T),
29    /// Payload chunk
30    Chunk(Option<Bytes>),
31}
32
33impl<T> From<T> for Message<T> {
34    fn from(item: T) -> Self {
35        Message::Item(item)
36    }
37}
38
39/// Incoming request type
40#[derive(Debug, Clone, Copy, PartialEq, Eq)]
41pub enum MessageType {
42    None,
43    Payload,
44    Stream,
45}
46
47const LW: usize = 2 * 1024;
48const HW: usize = 32 * 1024;
49
50pub(crate) fn reserve_readbuf(src: &mut BytesMut) {
51    let cap = src.capacity();
52    if cap < LW {
53        src.reserve(HW - cap);
54    }
55}
56
57#[cfg(test)]
58mod tests {
59    use super::*;
60    use crate::request::Request;
61
62    impl Message<Request> {
63        pub fn message(self) -> Request {
64            match self {
65                Message::Item(req) => req,
66                _ => panic!("error"),
67            }
68        }
69
70        pub fn chunk(self) -> Bytes {
71            match self {
72                Message::Chunk(Some(data)) => data,
73                _ => panic!("error"),
74            }
75        }
76
77        pub fn eof(self) -> bool {
78            match self {
79                Message::Chunk(None) => true,
80                Message::Chunk(Some(_)) => false,
81                _ => panic!("error"),
82            }
83        }
84    }
85}