diff --git a/server/stat/src/common/types.rs b/server/stat/src/common/types.rs index 82669a9..d146ac7 100644 --- a/server/stat/src/common/types.rs +++ b/server/stat/src/common/types.rs @@ -1,6 +1,7 @@ use std::sync::Arc; use std::sync::Mutex; use actix::dev::MessageResponse; +use actix::Message; pub struct DownStreamPack { holder: Option>>, @@ -9,6 +10,12 @@ pub struct DownStreamPack { pub entry: Arc::>>, } +#[derive(Message)] +#[rtype(result = "()")] +pub struct DownStreamMessage { + pub data: actix_web::web::Bytes, +} + impl DownStreamPack { pub fn new(socket_handle: u16, data: bytes::BytesMut) -> Arc> { diff --git a/server/stat/src/listener/wslistener.rs b/server/stat/src/listener/wslistener.rs index dec907c..bbac0a4 100644 --- a/server/stat/src/listener/wslistener.rs +++ b/server/stat/src/listener/wslistener.rs @@ -21,7 +21,7 @@ use crate::app::UserApp; use crate::constant; use tokio::net::TcpStream; use tokio::runtime::Runtime; -use crate::common::types::DownStreamPack; +use crate::common::types::{DownStreamPack, DownStreamMessage}; use tokio::io::{AsyncWriteExt}; use std::sync::OnceLock; use std::sync::atomic::{AtomicU16, Ordering}; @@ -42,12 +42,6 @@ pub struct WsListener { static WsListenerInstance: OnceLock = OnceLock::new(); -#[derive(Message)] -#[rtype(result = "()")] -pub struct DownStreamMessage { - data: actix_web::web::Bytes, -} - /// Define HTTP actor pub struct WsConn { socket_handle: u16,