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

use super::{UpgradeHandle, WsService};
use futures::future;
use std::task::{Context, Poll};
use tokio::sync::mpsc::{self, Receiver, Sender};
use tower_service::Service;

pub struct WsFactory {
    tx: Sender<UpgradeHandle>,
}

impl WsFactory {
    pub fn new() -> (Self, Receiver<UpgradeHandle>) {
        let (tx, rx) = mpsc::channel(1);
        (Self { tx }, rx)
    }
}

impl<T> Service<T> for WsFactory {
    type Response = WsService;
    type Error = hyper::Error;
    type Future = future::Ready<hyper::Result<Self::Response>>;

    fn poll_ready(&mut self, _cx: &mut Context) -> Poll<hyper::Result<()>> {
        Ok(()).into()
    }

    fn call(&mut self, _req: T) -> Self::Future {
        future::ready(Ok(WsService::new(self.tx.clone())))
    }
}