Browse Source

Use Response in IncomingMessage for uniformity.

wip
Titouan Rigoudy 9 years ago
parent
commit
0b849ce8d2
2 changed files with 9 additions and 5 deletions
  1. +6
    -3
      src/client.rs
  2. +3
    -2
      src/handler.rs

+ 6
- 3
src/client.rs View File

@ -4,11 +4,11 @@ use mio;
use config;
use control::{ControlRequest, ControlResponse};
use proto::{Request, Response};
use proto::{Response, Request};
use proto::server::*;
pub enum IncomingMessage {
ServerResponse(ServerResponse),
ProtoResponse(Response),
ControlRequest(ControlRequest),
}
@ -56,12 +56,15 @@ impl Client {
loop {
match self.rx.recv() {
Ok(IncomingMessage::ServerResponse(server_response)) => {
Ok(IncomingMessage::ProtoResponse(
Response::ServerResponse(server_response))) => {
self.handle_server_response(server_response);
},
Ok(IncomingMessage::ControlRequest(control_request)) => {
warn!("Unhandled control request: {:?}", control_request);
},
Err(e) => {
error!("Error receiving response: {}", e);
},


+ 3
- 2
src/handler.rs View File

@ -6,7 +6,7 @@ use mio::{EventLoop, EventSet, Handler, PollOpt, Token};
use mio::tcp::TcpStream;
use client::IncomingMessage;
use proto::{Packet, PacketStream, Request};
use proto::{Packet, PacketStream, Request, Response};
use proto::server::*;
struct TokenCounter {
@ -95,7 +95,8 @@ impl ConnectionHandler {
};
let server_response = try!(ServerResponse::from_packet(packet));
let message = IncomingMessage::ServerResponse(server_response);
let message = IncomingMessage::ProtoResponse(
Response::ServerResponse(server_response));
match self.client_tx.send(message) {
Ok(()) => Ok(true),
Err(e) => Err(io::Error::new(


Loading…
Cancel
Save