mirror of
https://github.com/tokio-rs/tokio.git
synced 2026-08-25 00:00:18 +02:00
Alex Crichton
d0b911189c
* 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
43 lines
1.2 KiB
Rust
43 lines
1.2 KiB
Rust
//! A small server that writes as many nul bytes on all connections it receives.
|
|
//!
|
|
//! There is no concurrency in this server, only one connection is written to at
|
|
//! a time.
|
|
|
|
#[macro_use]
|
|
extern crate futures;
|
|
extern crate futures_io;
|
|
extern crate futures_mio;
|
|
|
|
use std::env;
|
|
use std::net::SocketAddr;
|
|
|
|
use futures::Future;
|
|
use futures::stream::Stream;
|
|
use futures_io::IoFuture;
|
|
|
|
fn main() {
|
|
let addr = env::args().nth(1).unwrap_or("127.0.0.1:8080".to_string());
|
|
let addr = addr.parse::<SocketAddr>().unwrap();
|
|
|
|
let mut l = futures_mio::Loop::new().unwrap();
|
|
let server = l.handle().tcp_listen(&addr).and_then(|socket| {
|
|
socket.incoming().and_then(|(socket, addr)| {
|
|
println!("got a socket: {}", addr);
|
|
write(socket).or_else(|_| Ok(()))
|
|
}).for_each(|()| {
|
|
println!("lost the socket");
|
|
Ok(())
|
|
})
|
|
});
|
|
println!("Listenering on: {}", addr);
|
|
l.run(server).unwrap();
|
|
}
|
|
|
|
// TODO: this blows the stack...
|
|
fn write(socket: futures_mio::TcpStream) -> IoFuture<()> {
|
|
static BUF: &'static [u8] = &[0; 1 * 1024 * 1024];
|
|
futures_io::write_all(socket, BUF).and_then(|(socket, _)| {
|
|
write(socket)
|
|
}).boxed()
|
|
}
|