actix-web/examples/websocket-chat/src/client.rs

extern crate actix;
extern crate bytes;
extern crate byteorder;
extern crate futures;
extern crate tokio_io;
extern crate tokio_core;
extern crate serde;
extern crate serde_json;
#[macro_use] extern crate serde_derive;

use std::{io, net, process, thread};
use std::str::FromStr;
use std::time::Duration;
use futures::Future;
use tokio_core::net::TcpStream;
use actix::prelude::*;

mod codec;


fn main() {
    let sys = actix::System::new("chat-client");

    // Connect to server
    let addr = net::SocketAddr::from_str("127.0.0.1:12345").unwrap();
    Arbiter::handle().spawn(
        TcpStream::connect(&addr, Arbiter::handle())
            .and_then(|stream| {
                let addr: SyncAddress<_> = ChatClient.framed(stream, codec::ClientChatCodec);

                // start console loop
                thread::spawn(move|| {
                    loop {
                        let mut cmd = String::new();
                        if io::stdin().read_line(&mut cmd).is_err() {
                            println!("error");
                            return
                        }

                        addr.send(ClientCommand(cmd));
                    }
                });

                futures::future::ok(())
            })
            .map_err(|e| {
                println!("Can not connect to server: {}", e);
                process::exit(1)
            })
    );

    println!("Running chat client");
    sys.run();
}


struct ChatClient;

struct ClientCommand(String);

impl ResponseType for ClientCommand {
    type Item = ();
    type Error = ();
}

impl Actor for ChatClient {
    type Context = FramedContext<Self>;

    fn started(&mut self, ctx: &mut FramedContext<Self>) {
        // start heartbeats otherwise server will disconnect after 10 seconds
        self.hb(ctx)
    }
}

impl ChatClient {
    fn hb(&self, ctx: &mut FramedContext<Self>) {
        ctx.run_later(Duration::new(1, 0), |act, ctx| {
            if ctx.send(codec::ChatRequest::Ping).is_ok() {
                act.hb(ctx);
            }
        });
    }
}

/// Handle stdin commands
impl Handler<ClientCommand> for ChatClient
{
    fn handle(&mut self, msg: ClientCommand, ctx: &mut FramedContext<Self>)
              -> Response<Self, ClientCommand>
    {
        let m = msg.0.trim();
        if m.is_empty() {
            return Self::empty()
        }

        // we check for /sss type of messages
        if m.starts_with('/') {
            let v: Vec<&str> = m.splitn(2, ' ').collect();
            match v[0] {
                "/list" => {
                    let _ = ctx.send(codec::ChatRequest::List);
                },
                "/join" => {
                    if v.len() == 2 {
                        let _ = ctx.send(codec::ChatRequest::Join(v[1].to_owned()));
                    } else {
                        println!("!!! room name is required");
                    }
                },
                _ => println!("!!! unknown command"),
            }
        } else {
            let _ = ctx.send(codec::ChatRequest::Message(m.to_owned()));
        }

        Self::empty()
    }
}

/// Server communication

impl FramedActor for ChatClient {
    type Io = TcpStream;
    type Codec = codec::ClientChatCodec;
}

impl StreamHandler<codec::ChatResponse, io::Error> for ChatClient {

    fn finished(&mut self, _: &mut FramedContext<Self>) {
        println!("Disconnected");

        // Stop application on disconnect
        Arbiter::system().send(msgs::SystemExit(0));
    }
}

impl Handler<codec::ChatResponse, io::Error> for ChatClient {

    fn handle(&mut self, msg: codec::ChatResponse, _: &mut FramedContext<Self>)
              -> Response<Self, codec::ChatResponse>
    {
        match msg {
            codec::ChatResponse::Message(ref msg) => {
                println!("message: {}", msg);
            }
            codec::ChatResponse::Joined(ref msg) => {
                println!("!!! joined: {}", msg);
            }
            codec::ChatResponse::Rooms(rooms) => {
                println!("\n!!! Available rooms:");
                for room in rooms {
                    println!("{}", room);
                }
                println!("");
            }
            _ => (),
        }

        Self::empty()
    }
}