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