VPN/TrustTunnel
1
0
Fork 0
mirror of https://github.com/TrustTunnel/TrustTunnel.git synced 2026-04-26 20:36:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Consume deferred headers in flush method and utilize WaitingWritable event for it

Browse Source
This commit is contained in:
boommy
2026-03-03 17:05:41 +04:00
parent 08ccb5b216
commit 13a420a4a7
1 changed files with 13 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
lib/src/http3_codec.rs
View File

@@ -430,22 +430,6 @@ impl StreamSink {
Ok(())
}
async fn consume_pending_response(&mut self) -> io::Result<()> {
while self.pending_response.is_some() {
self.try_send_pending_response()?;
if self.pending_response.is_some() {
self.codec_tx
.send(StreamMessage::WaitingWritable(self.stream_id))
.map_err(|_| io::Error::from(ErrorKind::UnexpectedEof))?;
match self.writable_event_rx.recv().await {
None => return Err(io::Error::from(ErrorKind::UnexpectedEof)),
Some(_) => continue,
}
}
}
Ok(())
}
async fn wait_body_capacity(&mut self) -> io::Result<()> {
loop {
match self.socket.stream_capacity(self.stream_id) {
@@ -543,6 +527,19 @@ impl pipe::Sink for StreamSink {
async fn wait_writable(&mut self) -> io::Result<()> {
self.wait_body_capacity().await
}
async fn flush(&mut self) -> io::Result<()> {
while self.pending_response.is_some() {
self.codec_tx
.send(StreamMessage::WaitingWritable(self.stream_id))
.map_err(|_| io::Error::from(ErrorKind::UnexpectedEof))?;
match self.writable_event_rx.recv().await {
None => return Err(io::Error::from(ErrorKind::UnexpectedEof)),
Some(_) => self.try_send_pending_response()?,
}
}
Ok(())
}
}
impl http_codec::DroppingSink for StreamSink {