diff --git a/server/stat/src/constant.rs b/server/stat/src/constant.rs new file mode 100644 index 0000000..684445a --- /dev/null +++ b/server/stat/src/constant.rs @@ -0,0 +1,3 @@ +pub mod constant; + +pub use constant::*; \ No newline at end of file diff --git a/server/stat/src/constant/constant.rs b/server/stat/src/constant/constant.rs new file mode 100644 index 0000000..415dff1 --- /dev/null +++ b/server/stat/src/constant/constant.rs @@ -0,0 +1 @@ +pub const MAX_PACKET_LEN: usize = 1024 * 64; \ No newline at end of file diff --git a/server/stat/src/listener/wslistener.rs b/server/stat/src/listener/wslistener.rs index cad9d65..a5bc5d2 100644 --- a/server/stat/src/listener/wslistener.rs +++ b/server/stat/src/listener/wslistener.rs @@ -25,7 +25,7 @@ use actix_web::web::Bytes; use futures::FutureExt; use tokio::io::{AsyncReadExt, Interest}; use crate::app::UserApp; -use crate::{MAX_PACKET_LEN}; +use crate::constant; use tokio::net::TcpStream; use tokio::runtime::Runtime; use crate::app::user_app::DownStreamPack; @@ -78,8 +78,8 @@ impl WsConn { } while true { let mut read_bytes = buf_len - already_read_bytes; - if read_bytes > MAX_PACKET_LEN - self.recv_buf_len { - read_bytes = MAX_PACKET_LEN - self.recv_buf_len; + if read_bytes > constant::MAX_PACKET_LEN - self.recv_buf_len { + read_bytes = constant::MAX_PACKET_LEN - self.recv_buf_len; } if read_bytes > 0 { self.recv_buf[self.recv_buf_len..bin.len()].copy_from_slice(&bin); diff --git a/server/stat/src/main.rs b/server/stat/src/main.rs index 962e640..86d57f7 100755 --- a/server/stat/src/main.rs +++ b/server/stat/src/main.rs @@ -5,9 +5,9 @@ mod listener; mod upstream; mod downstream; mod ss; -use crate::app::UserApp; +mod constant; -const MAX_PACKET_LEN: usize = 1024 * 64; +use crate::app::UserApp; fn main() { App::init(UserApp::instance());