Noratrieb/haesli
1
0
Fork 0
mirror of https://github.com/Noratrieb/haesli.git synced 2026-01-16 04:35:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

route topics

Browse source
This commit is contained in:
nora 2022-03-20 18:31:41 +01:00
parent 58de7f1e2d
commit 43d0ce05dc
10 changed files with 98 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

13
haesli_transport/src/connection.rs
View file

@ -15,9 +15,8 @@ use haesli_core::{
ConnectionCloseOk, ConnectionOpen, ConnectionOpenOk, ConnectionStart, ConnectionStartOk,
ConnectionTune, ConnectionTuneOk, FieldValue, Longstr, Method, ReplyCode, ReplyText, Table,
},
GlobalData,
GlobalData, SingleVec,
};
use tinyvec::TinyVec;
use tokio::{
io::{AsyncReadExt, AsyncWriteExt},
net::TcpStream,
@ -75,7 +74,7 @@ const DEFAULT_TIMEOUT: Duration = Duration::from_secs(30);
enum ChannelStatus {
Default,
NeedHeader(u16, Box<Method>),
NeedsBody(Box<Method>, ContentHeader, TinyVec<[Bytes; 1]>),
NeedsBody(Box<Method>, ContentHeader, SingleVec<Bytes>),
}
impl ChannelStatus {
@ -166,7 +165,7 @@ impl TransportConnection {
channel: ChannelNum,
method: &Method,
header: ContentHeader,
body: &TinyVec<[Bytes; 1]>,
body: &SingleVec<Bytes>,
) -> Result<()> {
self.send_method(channel, method).await?;
@ -177,7 +176,7 @@ impl TransportConnection {
self.send_bodies(channel, body).await
}
async fn send_bodies(&mut self, channel: ChannelNum, body: &TinyVec<[Bytes; 1]>) -> Result<()> {
async fn send_bodies(&mut self, channel: ChannelNum, body: &SingleVec<Bytes>) -> Result<()> {
// this is inefficient if it's a huge message sent by a client with big frames to one with
// small frames
// we assume that this won't happen that that the first branch will be taken in most cases,
@ -438,7 +437,7 @@ impl TransportConnection {
let header = parse_content_header(&frame.payload)?;
ensure_conn(header.class_id == class_id)?;
channel.status = ChannelStatus::NeedsBody(method, header, TinyVec::new());
channel.status = ChannelStatus::NeedsBody(method, header, SingleVec::new());
Ok(())
}
ChannelStatus::NeedsBody(_, _, _) => {
@ -485,7 +484,7 @@ impl TransportConnection {
&mut self,
method: Method,
header: ContentHeader,
payloads: TinyVec<[Bytes; 1]>,
payloads: SingleVec<Bytes>,
channel: ChannelNum,
) -> Result<()> {
// The only method with content that is sent to the server is Basic.Publish.