diff --git a/g3proxy/src/serve/http_proxy/task/connect/task.rs b/g3proxy/src/serve/http_proxy/task/connect/task.rs index d2b5c000..316d90c7 100644 --- a/g3proxy/src/serve/http_proxy/task/connect/task.rs +++ b/g3proxy/src/serve/http_proxy/task/connect/task.rs @@ -115,11 +115,7 @@ impl HttpProxyConnectTask { self.ctx .set_custom_header_for_local_reply(&self.tcp_notes, &mut rsp); let should_close = rsp.should_close(); - if should_close { - self.back_to_http = false; - } else { - self.back_to_http = true; - } + self.back_to_http = !should_close; if rsp.reply_err_to_request(clt_w).await.is_err() { self.back_to_http = false; diff --git a/lib/g3-types/src/net/port/ports.rs b/lib/g3-types/src/net/port/ports.rs index cd8ae7e7..527e67f0 100644 --- a/lib/g3-types/src/net/port/ports.rs +++ b/lib/g3-types/src/net/port/ports.rs @@ -21,15 +21,9 @@ use std::str::FromStr; use anyhow::anyhow; use rustc_hash::FxHashSet; -#[derive(Clone, Eq, PartialEq)] +#[derive(Clone, Default, Eq, PartialEq)] pub struct Ports(FxHashSet); -impl Default for Ports { - fn default() -> Self { - Ports(FxHashSet::default()) - } -} - impl Ports { pub fn add_single(&mut self, port: u16) { self.0.insert(port);