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