openethereum/util/src/io/mod.rs

116 lines
3.7 KiB
Rust
Raw Normal View History

2016-01-12 17:33:40 +01:00
/// General IO module.
///
/// Example usage for craeting a network service and adding an IO handler:
///
/// ```rust
2016-01-13 13:56:48 +01:00
/// extern crate ethcore_util;
/// use ethcore_util::*;
2016-01-12 17:33:40 +01:00
///
/// struct MyHandler;
///
2016-01-13 13:56:48 +01:00
/// struct MyMessage {
/// data: u32
/// }
2016-01-12 17:33:40 +01:00
///
2016-01-13 13:56:48 +01:00
/// impl IoHandler<MyMessage> for MyHandler {
/// fn initialize(&mut self, io: &mut IoContext<MyMessage>) {
/// io.register_timer(1000).unwrap();
2016-01-12 17:33:40 +01:00
/// }
///
2016-01-13 13:56:48 +01:00
/// fn timeout(&mut self, _io: &mut IoContext<MyMessage>, timer: TimerToken) {
2016-01-12 17:33:40 +01:00
/// println!("Timeout {}", timer);
/// }
///
2016-01-13 13:56:48 +01:00
/// fn message(&mut self, _io: &mut IoContext<MyMessage>, message: &mut MyMessage) {
/// println!("Message {}", message.data);
2016-01-12 17:33:40 +01:00
/// }
2016-01-13 13:56:48 +01:00
/// }
2016-01-12 17:33:40 +01:00
///
/// fn main () {
2016-01-13 13:56:48 +01:00
/// let mut service = IoService::<MyMessage>::start().expect("Error creating network service");
/// service.register_handler(Box::new(MyHandler)).unwrap();
2016-01-12 17:33:40 +01:00
///
/// // Wait for quit condition
/// // ...
/// // Drop the service
/// }
/// ```
mod service;
2016-01-21 16:48:37 +01:00
mod worker;
use mio::{EventLoop, Token};
2016-01-12 17:33:40 +01:00
#[derive(Debug)]
pub enum IoError {
Mio(::std::io::Error),
}
2016-01-21 16:48:37 +01:00
impl<Message> From<::mio::NotifyError<service::IoMessage<Message>>> for IoError where Message: Send + Clone {
2016-01-13 13:56:48 +01:00
fn from(_err: ::mio::NotifyError<service::IoMessage<Message>>) -> IoError {
2016-01-12 17:33:40 +01:00
IoError::Mio(::std::io::Error::new(::std::io::ErrorKind::ConnectionAborted, "Network IO notification error"))
}
}
/// Generic IO handler.
/// All the handler function are called from within IO event loop.
/// `Message` type is used as notification data
2016-01-21 16:48:37 +01:00
pub trait IoHandler<Message>: Send + Sync where Message: Send + Sync + Clone + 'static {
/// Initialize the handler
2016-01-21 16:48:37 +01:00
fn initialize(&self, _io: &IoContext<Message>) {}
/// Timer function called after a timeout created with `HandlerIo::timeout`.
2016-01-21 16:48:37 +01:00
fn timeout(&self, _io: &IoContext<Message>, _timer: TimerToken) {}
/// Called when a broadcasted message is received. The message can only be sent from a different IO handler.
2016-01-21 16:48:37 +01:00
fn message(&self, _io: &IoContext<Message>, _message: &Message) {}
/// Called when an IO stream gets closed
2016-01-21 16:48:37 +01:00
fn stream_hup(&self, _io: &IoContext<Message>, _stream: StreamToken) {}
/// Called when an IO stream can be read from
2016-01-21 16:48:37 +01:00
fn stream_readable(&self, _io: &IoContext<Message>, _stream: StreamToken) {}
/// Called when an IO stream can be written to
2016-01-21 16:48:37 +01:00
fn stream_writable(&self, _io: &IoContext<Message>, _stream: StreamToken) {}
/// Register a new stream with the event loop
fn register_stream(&self, _stream: StreamToken, _reg: Token, _event_loop: &mut EventLoop<IoManager<Message>>) {}
/// Re-register a stream with the event loop
fn update_stream(&self, _stream: StreamToken, _reg: Token, _event_loop: &mut EventLoop<IoManager<Message>>) {}
}
2016-01-12 17:33:40 +01:00
pub type TimerToken = service::TimerToken;
pub type StreamToken = service::StreamToken;
2016-01-21 16:48:37 +01:00
pub use io::service::IoContext;
pub type IoService<Message> = service::IoService<Message>;
pub type IoChannel<Message> = service::IoChannel<Message>;
pub type IoManager<Message> = service::IoManager<Message>;
pub const TOKENS_PER_HANDLER: usize = service::TOKENS_PER_HANDLER;
2016-01-12 17:33:40 +01:00
2016-01-13 13:56:48 +01:00
#[cfg(test)]
mod tests {
use io::*;
struct MyHandler;
struct MyMessage {
data: u32
}
impl IoHandler<MyMessage> for MyHandler {
fn initialize(&mut self, io: &mut IoContext<MyMessage>) {
io.register_timer(1000).unwrap();
}
fn timeout(&mut self, _io: &mut IoContext<MyMessage>, timer: TimerToken) {
println!("Timeout {}", timer);
}
fn message(&mut self, _io: &mut IoContext<MyMessage>, message: &mut MyMessage) {
println!("Message {}", message.data);
}
}
#[test]
fn test_service_register_handler () {
let mut service = IoService::<MyMessage>::start().expect("Error creating network service");
service.register_handler(Box::new(MyHandler)).unwrap();
}
}