tfserver/server/
handler.rs1use std::net::SocketAddr;
2use std::sync::{Arc};
3use async_trait::async_trait;
4use tokio::io;
5use tokio::sync::{Mutex};
6use tokio::sync::oneshot::Sender;
7use tokio_util::bytes::{Bytes, BytesMut};
8use tokio_util::codec::{Decoder, Encoder, Framed};
9use crate::codec::codec_trait::TfCodec;
10use crate::structures::s_type::StructureType;
11use crate::structures::traffic_proc::TrafficProcessorHolder;
12use crate::structures::transport::Transport;
13
14#[async_trait]
15pub trait Handler: Send + Sync {
17 type Codec: Encoder<Bytes, Error = io::Error> + Decoder<Item = BytesMut, Error = io::Error> + Clone + Send + Sync + 'static + TfCodec;
18 async fn serve_route(
28 &mut self,
29 client_meta: (SocketAddr, &mut Option<Sender<Arc<Mutex<dyn Handler<Codec = Self::Codec>>>>>),
35 s_type: Box<dyn StructureType>,
36 data: BytesMut,
37 ) -> Result<Vec<u8>, Vec<u8>>;
38
39 async fn accept_stream(&mut self, add: SocketAddr, stream: (Framed<Transport, Self::Codec>, TrafficProcessorHolder<Self::Codec>));
41
42}