ws-async 0.1.0

Asynchronous WebSocket Library
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

use hyper::client::connect::dns::{GaiResolver, Name};
use hyper::client::HttpConnector;
use hyper::Uri;
use std::net::IpAddr;
use std::task::{Context, Poll};
use tokio::net::TcpStream;
use tower_service::Service;

type StdError = Box<dyn std::error::Error + Send + Sync>;

#[derive(Debug)]
pub struct WsConnector<R = GaiResolver> {
    http: HttpConnector<R>,
}

impl Default for WsConnector {
    fn default() -> Self {
        Self::new_with_resolver(GaiResolver::new())
    }
}

impl WsConnector {
    pub fn new() -> Self {
        Self::default()
    }
}

impl<R> WsConnector<R>
where
    R: Service<Name> + Clone + Send + Sync + 'static,
    R::Response: Iterator<Item = IpAddr>,
    R::Error: Into<StdError>,
    R::Future: Send,
{
    pub fn new_with_resolver(resolver: R) -> Self {
        let mut http = HttpConnector::new_with_resolver(resolver);
        http.enforce_http(false);

        Self::from_http(http)
    }
    pub fn from_http(http: HttpConnector<R>) -> Self {
        Self { http }
    }
}

impl<R> Service<Uri> for WsConnector<R>
where
    R: Service<Name> + Clone + Send + Sync + 'static,
    R::Response: Iterator<Item = IpAddr>,
    R::Error: Into<StdError>,
    R::Future: Send,
{
    type Response = TcpStream;
    type Error = <HttpConnector<R> as Service<Uri>>::Error;
    type Future = <HttpConnector<R> as Service<Uri>>::Future;

    fn poll_ready(&mut self, cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
        self.http.poll_ready(cx)
    }

    fn call(&mut self, dst: Uri) -> Self::Future {
        self.http.call(dst)
    }
}