websocket_server_async/
builder.rs

1use crate::peer::WSPeer;
2use crate::stream::MaybeRustlsStream;
3use crate::websocket_server::{ConnectEventType, WebSocketServer};
4use anyhow::Result;
5use aqueue::Actor;
6use futures_util::stream::SplitStream;
7use std::future::Future;
8use std::marker::PhantomData;
9use std::sync::Arc;
10use tokio::net::{TcpStream, ToSocketAddrs};
11use tokio_rustls::TlsAcceptor;
12use tokio_tungstenite::tungstenite::protocol::WebSocketConfig;
13use tokio_tungstenite::WebSocketStream;
14
15/// websocket server builder
16pub struct Builder<I, R, A, T> {
17    input: Option<I>,
18    tls_acceptor: Option<TlsAcceptor>,
19    connect_event: Option<ConnectEventType>,
20    addr: A,
21    config: Option<WebSocketConfig>,
22    load_timeout_secs: u64,
23    _phantom1: PhantomData<R>,
24    _phantom2: PhantomData<T>,
25}
26
27impl<I, R, A, T> Builder<I, R, A, T>
28where
29    I: Fn(SplitStream<WebSocketStream<MaybeRustlsStream<TcpStream>>>, Arc<Actor<WSPeer>>, T) -> R
30        + Send
31        + Sync
32        + 'static,
33    R: Future<Output = Result<()>> + Send + 'static,
34    A: ToSocketAddrs,
35    T: Clone + Send + 'static,
36{
37    pub fn new(addr: A) -> Builder<I, R, A, T> {
38        Builder {
39            input: None,
40            tls_acceptor: None,
41            connect_event: None,
42            addr,
43            config: None,
44            load_timeout_secs: 60,
45            _phantom1: Default::default(),
46            _phantom2: Default::default(),
47        }
48    }
49
50    /// 设置websocket server 输入事件
51    pub fn set_input_event(mut self, f: I) -> Self {
52        self.input = Some(f);
53        self
54    }
55
56    /// 设置TCP server 连接事件
57    pub fn set_connect_event(mut self, c: ConnectEventType) -> Self {
58        self.connect_event = Some(c);
59        self
60    }
61
62    /// 设置TCP server 连接事件
63    pub fn set_tls(mut self, tls: Option<TlsAcceptor>) -> Self {
64        self.tls_acceptor = tls;
65        self
66    }
67
68    /// 设置config
69    pub fn set_config(mut self, config: WebSocketConfig) -> Self {
70        self.config = Some(config);
71        self
72    }
73
74    /// 设置等待websocket hand accept 加载 等待时间
75    pub fn set_load_timeout(mut self, load_timeout_secs: u64) -> Self {
76        self.load_timeout_secs = load_timeout_secs;
77        self
78    }
79
80    /// 生成TCPSERVER,如果没有设置 tcp input 将报错
81    pub async fn build(mut self) -> Arc<Actor<WebSocketServer<I, R, T>>> {
82        if let Some(input) = self.input.take() {
83            return WebSocketServer::new(
84                self.addr,
85                input,
86                self.connect_event,
87                self.config,
88                self.tls_acceptor,
89                self.load_timeout_secs,
90            )
91            .await
92            .unwrap();
93        }
94        panic!("input event is no settings,please use set_input_event function set input event.");
95    }
96}
websocket_server_async/builder.rs

websocket_server_async/
builder.rs
