actix/actix-web
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-28 01:52:57 +01:00
Code Issues Releases Wiki Activity

update websocket-chat example

Browse Source
This commit is contained in:
Nikolay Kim 2018-02-03 08:25:31 -08:00
parent 671ab35cf6
commit d568161852
2 changed files with 61 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
examples/websocket-chat/src/client.rs
View File

@ -27,9 +27,12 @@ fn main() {
Arbiter::handle().spawn(
TcpStream::connect(&addr, Arbiter::handle())
.and_then(|stream| {
let addr: SyncAddress<_> = ChatClient::create_with(
stream.framed(codec::ClientChatCodec),
|_, framed| ChatClient{framed: framed});
let addr: SyncAddress<_> = ChatClient::create(|mut ctx| {
let (reader, writer) =
FramedReader::wrap(stream.framed(codec::ClientChatCodec));
ChatClient::add_stream(reader, &mut ctx);
ChatClient{framed: writer}
});
// start console loop
thread::spawn(move|| {
@ -58,7 +61,7 @@ fn main() {
struct ChatClient {
framed: FramedCell<TcpStream, codec::ClientChatCodec>,
framed: FramedWriter<TcpStream, codec::ClientChatCodec>,
}
#[derive(Message)]
@ -125,12 +128,10 @@ impl Handler<ClientCommand> for ChatClient {
/// Server communication
impl FramedHandler<TcpStream, codec::ClientChatCodec> for ChatClient {
impl StreamHandler<codec::ChatResponse, FramedError<codec::ClientChatCodec>> for ChatClient {
fn handle(&mut self, msg: io::Result<codec::ChatResponse>, ctx: &mut Context<Self>) {
fn handle(&mut self, msg: codec::ChatResponse, _: &mut Context<Self>) {
match msg {
Err(_) => ctx.stop(),
Ok(msg) => match msg {
codec::ChatResponse::Message(ref msg) => {
println!("message: {}", msg);
}
@ -147,5 +148,4 @@ impl FramedHandler<TcpStream, codec::ClientChatCodec> for ChatClient {
_ => (),
}
}
}
}

19
examples/websocket-chat/src/session.rs
View File

@ -1,6 +1,6 @@
//! `ClientSession` is an actor, it manages peer tcp connection and
//! proxies commands from peer to `ChatServer`.
use std::{io, net};
use std::net;
use std::str::FromStr;
use std::time::{Instant, Duration};
use futures::Stream;
@ -28,7 +28,7 @@ pub struct ChatSession {
/// joined room
room: String,
/// Framed wrapper
framed: FramedCell<TcpStream, ChatCodec>,
framed: FramedWriter<TcpStream, ChatCodec>,
}
impl Actor for ChatSession {
@ -63,13 +63,11 @@ impl Actor for ChatSession {
}
/// To use `Framed` we have to define Io type and Codec
impl FramedHandler<TcpStream, ChatCodec> for ChatSession {
impl StreamHandler<ChatRequest, FramedError<ChatCodec>> for ChatSession {
/// This is main event loop for client requests
fn handle(&mut self, msg: io::Result<ChatRequest>, ctx: &mut Context<Self>) {
fn handle(&mut self, msg: ChatRequest, ctx: &mut Context<Self>) {
match msg {
Err(_) => ctx.stop(),
Ok(msg) => match msg {
ChatRequest::List => {
// Send ListRooms message to chat server and wait for response
println!("List rooms");
@ -104,7 +102,6 @@ impl FramedHandler<TcpStream, ChatCodec> for ChatSession {
self.hb = Instant::now(),
}
}
}
}
/// Handler for Message, chat server sends this message, we just send string to peer
@ -121,7 +118,7 @@ impl Handler<Message> for ChatSession {
impl ChatSession {
pub fn new(addr: SyncAddress<ChatServer>,
framed: FramedCell<TcpStream, ChatCodec>) -> ChatSession {
framed: FramedWriter<TcpStream, ChatCodec>) -> ChatSession {
ChatSession {id: 0, addr: addr, hb: Instant::now(),
room: "Main".to_owned(), framed: framed}
}
@ -194,8 +191,10 @@ impl Handler<TcpConnect> for TcpServer {
// For each incoming connection we create `ChatSession` actor
// with out chat server address.
let server = self.chat.clone();
let _: () = ChatSession::create_with(msg.0.framed(ChatCodec), |_, framed| {
ChatSession::new(server, framed)
let _: () = ChatSession::create(|mut ctx| {
let (reader, writer) = FramedReader::wrap(msg.0.framed(ChatCodec));
ChatSession::add_stream(reader, &mut ctx);
ChatSession::new(server, writer)
});
}
}