1use 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)]
25pub enum Message<T> {
27 Item(T),
29 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#[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}