actix/actix-web
1
0
Fork 0
mirror of https://github.com/fafhrd91/actix-web synced 2024-11-24 00:21:08 +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)| {
let addr: SyncAddress<_> = ChatClient::create(|ctx| {
ctx.add_stream(reader);
ChatClient::add_stream(reader, ctx);
ChatClient(writer)
});
@ -68,7 +68,7 @@ impl Actor for ChatClient {
}
fn stopping(&mut self, _: &mut Context<Self>) -> bool {
println!("Disconnected");
println!("Stopping");
// Stop application on disconnect
Arbiter::system().send(actix::msgs::SystemExit(0));
@ -96,13 +96,21 @@ impl Handler<ClientCommand> for ChatClient {
}
/// Handle server websocket messages
impl Handler<Result<Message, WsClientError>> for ChatClient {
type Result = ();
impl StreamHandler<Message, WsClientError> for ChatClient {
fn handle(&mut self, msg: Result<Message, WsClientError>, ctx: &mut Context<Self>) {
fn handle(&mut self, msg: Message, ctx: &mut Context<Self>) {
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()
}
}