actix/actix-extras
1
0
Fork 0
mirror of https://github.com/actix/actix-extras.git synced 2024-11-24 07:53:00 +01:00
Code Issues Releases Wiki Activity

update websocket example

Browse Source
This commit is contained in:
Nikolay Kim 2018-01-28 08:58:18 -08:00
parent 715ec4ae2f
commit 9835a4537a
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
examples/websocket/src/client.rs
View File

@ -30,7 +30,7 @@ fn main() {
}) })
.map(|(reader, writer)| { .map(|(reader, writer)| {
let addr: SyncAddress<_> = ChatClient::create(|ctx| { let addr: SyncAddress<_> = ChatClient::create(|ctx| {
ctx.add_stream(reader); ChatClient::add_stream(reader, ctx);
ChatClient(writer) ChatClient(writer)
}); });
@ -68,7 +68,7 @@ impl Actor for ChatClient {
} }
fn stopping(&mut self, _: &mut Context<Self>) -> bool { fn stopping(&mut self, _: &mut Context<Self>) -> bool {
println!("Disconnected"); println!("Stopping");
// Stop application on disconnect // Stop application on disconnect
Arbiter::system().send(actix::msgs::SystemExit(0)); Arbiter::system().send(actix::msgs::SystemExit(0));
@ -96,13 +96,21 @@ impl Handler<ClientCommand> for ChatClient {
} }
/// Handle server websocket messages /// Handle server websocket messages
impl Handler<Result<Message, WsClientError>> for ChatClient { impl StreamHandler<Message, WsClientError> for ChatClient {
type Result = ();
fn handle(&mut self, msg: Result<Message, WsClientError>, ctx: &mut Context<Self>) { fn handle(&mut self, msg: Message, ctx: &mut Context<Self>) {
match msg { match msg {
Ok(Message::Text(txt)) => println!("Server: {:?}", txt), Message::Text(txt) => println!("Server: {:?}", txt),
_ => () _ => ()
} }
} }
fn started(&mut self, ctx: &mut Context<Self>, _: SpawnHandle) {
println!("Connected");
}
fn finished(&mut self, err: Option<WsClientError>, ctx: &mut Context<Self>, _: SpawnHandle) {
println!("Server disconnected");
ctx.stop()
}
} }