tuxmain/rust-rpxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor

Browse source
This commit is contained in:
Jun Kurihara 2022-07-07 20:31:37 +09:00
commit 5571f19c40
No known key found for this signature in database
GPG key ID: 48ADFD173ED22B03
2 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/proxy/proxy_h3.rs
View file

@ -4,6 +4,7 @@ use bytes::{Buf, Bytes};
use h3::{quic::BidiStream, server::RequestStream};
use hyper::{client::connect::Connect, Body, HeaderMap, Request, Response};
use std::net::SocketAddr;
use tokio::time::Duration;
impl<T> Proxy<T>
where
@ -57,7 +58,7 @@ where
// .await
// .map_err(|e| anyhow!("HTTP/3 accept failed: {}", e))?
while let Some((req, stream)) = match tokio::time::timeout(
tokio::time::Duration::from_millis(H3_CONN_TIMEOUT_MILLIS),
Duration::from_millis(H3_CONN_TIMEOUT_MILLIS),
h3_conn.accept(),
)
.await

6
src/proxy/proxy_tls.rs
View file

@ -5,8 +5,8 @@ use futures::StreamExt;
use futures::{future::FutureExt, select};
use hyper::{client::connect::Connect, server::conn::Http};
use rustls::ServerConfig;
use std::{sync::Arc, time::Duration};
use tokio::net::TcpListener;
use std::sync::Arc;
use tokio::{net::TcpListener, time::Duration};
impl<T> Proxy<T>
where
@ -122,7 +122,7 @@ where
.get(initial_app_name)
.ok_or_else(|| anyhow!(""))?;
while backend_serve.get_tls_server_config().is_none() {
tokio::time::sleep(tokio::time::Duration::from_millis(10)).await;
tokio::time::sleep(Duration::from_millis(10)).await;
}
let server_crypto = backend_serve
.get_tls_server_config()