mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-27 00:00:12 +02:00
Re-work I/O
* Auto-register interest whenever we see WouldBlock
* Remove implementations of `Stream<Item=Ready>`, no longer needed
* Add explicit `poll_{read,write}` methods, if needed
* Remove all I/O streams, libstd ones suffice
* Update all I/O futures
This commit is contained in:
+26
-8
@@ -5,11 +5,14 @@ extern crate futures_io;
|
||||
extern crate futures_mio;
|
||||
|
||||
use std::env;
|
||||
use std::io::{self, Read, Write};
|
||||
use std::net::SocketAddr;
|
||||
use std::sync::Arc;
|
||||
|
||||
use futures::Future;
|
||||
use futures_io::{copy, TaskIo};
|
||||
use futures::stream::Stream;
|
||||
use futures_io::copy;
|
||||
use futures_mio::TcpStream;
|
||||
|
||||
fn main() {
|
||||
let addr = env::args().nth(1).unwrap_or("127.0.0.1:8080".to_string());
|
||||
@@ -27,17 +30,14 @@ fn main() {
|
||||
|
||||
// Pull out the stream of incoming connections and then for each new
|
||||
// one spin up a new task copying data. We put the `socket` into a
|
||||
// `TaskIo` structure which then allows us to `split` it into the read
|
||||
// and write halves of the socket.
|
||||
// `Arc` structure which then allows us to share it across the
|
||||
// read/write halves with a small shim.
|
||||
//
|
||||
// Finally we use the `io::copy` future to copy all data from the
|
||||
// reading half onto the writing half.
|
||||
socket.incoming().for_each(|(socket, addr)| {
|
||||
let io = TaskIo::new(socket);
|
||||
let pair = io.map(|io| io.split());
|
||||
let amt = pair.and_then(|(reader, writer)| {
|
||||
copy(reader, writer)
|
||||
});
|
||||
let socket = Arc::new(socket);
|
||||
let amt = copy(SocketIo(socket.clone()), SocketIo(socket));
|
||||
|
||||
// Once all that is done we print out how much we wrote, and then
|
||||
// critically we *forget* this future which allows it to run
|
||||
@@ -51,3 +51,21 @@ fn main() {
|
||||
});
|
||||
l.run(done).unwrap();
|
||||
}
|
||||
|
||||
struct SocketIo(Arc<TcpStream>);
|
||||
|
||||
impl Read for SocketIo {
|
||||
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
||||
(&*self.0).read(buf)
|
||||
}
|
||||
}
|
||||
|
||||
impl Write for SocketIo {
|
||||
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
|
||||
(&*self.0).write(buf)
|
||||
}
|
||||
|
||||
fn flush(&mut self) -> io::Result<()> {
|
||||
(&*self.0).flush()
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user