This commit is contained in:
azw 2024-05-12 18:06:50 +08:00
parent 58fa1b92e5
commit 8668cf7686
2 changed files with 8 additions and 7 deletions

View File

@ -1,6 +1,7 @@
use std::sync::Arc; use std::sync::Arc;
use std::sync::Mutex; use std::sync::Mutex;
use actix::dev::MessageResponse; use actix::dev::MessageResponse;
use actix::Message;
pub struct DownStreamPack { pub struct DownStreamPack {
holder: Option<Arc::<std::sync::Mutex::<Self>>>, holder: Option<Arc::<std::sync::Mutex::<Self>>>,
@ -9,6 +10,12 @@ pub struct DownStreamPack {
pub entry: Arc::<std::sync::Mutex::<r9::ListHeadLock<Self>>>, pub entry: Arc::<std::sync::Mutex::<r9::ListHeadLock<Self>>>,
} }
#[derive(Message)]
#[rtype(result = "()")]
pub struct DownStreamMessage {
pub data: actix_web::web::Bytes,
}
impl DownStreamPack { impl DownStreamPack {
pub fn new(socket_handle: u16, data: bytes::BytesMut) -> Arc<Mutex<Self>> { pub fn new(socket_handle: u16, data: bytes::BytesMut) -> Arc<Mutex<Self>> {

View File

@ -21,7 +21,7 @@ use crate::app::UserApp;
use crate::constant; use crate::constant;
use tokio::net::TcpStream; use tokio::net::TcpStream;
use tokio::runtime::Runtime; use tokio::runtime::Runtime;
use crate::common::types::DownStreamPack; use crate::common::types::{DownStreamPack, DownStreamMessage};
use tokio::io::{AsyncWriteExt}; use tokio::io::{AsyncWriteExt};
use std::sync::OnceLock; use std::sync::OnceLock;
use std::sync::atomic::{AtomicU16, Ordering}; use std::sync::atomic::{AtomicU16, Ordering};
@ -42,12 +42,6 @@ pub struct WsListener {
static WsListenerInstance: OnceLock<WsListener> = OnceLock::new(); static WsListenerInstance: OnceLock<WsListener> = OnceLock::new();
#[derive(Message)]
#[rtype(result = "()")]
pub struct DownStreamMessage {
data: actix_web::web::Bytes,
}
/// Define HTTP actor /// Define HTTP actor
pub struct WsConn { pub struct WsConn {
socket_handle: u16, socket_handle: u16,