twisty 0.3.1

Example WebSocket Echo client implemented with twist
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

use client::{self, Client};
use futures::Future;
use futures::future::{self, FutureResult, Loop};
use std::io;
use tokio_service::Service;
use twist::WebSocketFrame;

pub struct LineReader {
    client: Client,
    current_line: String,
}

impl LineReader {
    pub fn new(client: Client) -> LineReader {
        LineReader {
            client: client,
            current_line: String::new(),
        }
    }

    pub fn current_line(&self) -> String {
        self.current_line.clone()
    }

    pub fn read_line(mut self) -> FutureResult<LineReader, io::Error> {
        let mut buf = String::new();
        match io::stdin().read_line(&mut buf) {
            Ok(_) => {
                self.current_line = buf;
                if self.current_line.starts_with("exit") {
                    future::err(client::other("exiting"))
                } else {
                    future::done(Ok(self))
                }
            }
            Err(e) => future::err(e),
        }
    }

    pub fn call(self, frame: WebSocketFrame) -> Box<Future<Item = Loop<LineReader, LineReader>, Error = io::Error>> {
        Box::new(self.client.call(frame)
            .and_then(|resp| -> Result<Loop<LineReader, LineReader>, io::Error> {
                          Ok(Loop::Continue(self))
                      })
            .or_else(|e| -> Result<Loop<LineReader, LineReader>, io::Error> {
                         println!("error: {}", e);
                         Err(e)
                     }))
    }
}