tokio_proto_listen.rs

extern crate futures;
extern crate tokio_proto;
#[macro_use]
extern crate tokio_core;

use std::io::{self, Read};

use futures::{Future, Poll};
use tokio_proto::server;
use tokio_core::{Loop, TcpStream};

struct Connection {
    stream: TcpStream,
    buf: Box<[u8]>,
}

impl Connection {
    fn new(stream: TcpStream) -> Connection {
        let buf = vec![0; 1024];

        Connection {
            stream: stream,
            buf: buf.into_boxed_slice(),
        }
    }
}

impl Future for Connection {
    type Item = ();
    type Error = io::Error;

    fn poll(&mut self) -> Poll<(), io::Error> {
        loop {
            let n = try_nb!(self.stream.read(&mut self.buf));
            println!("read {} bytes", n);

            if n == 0 {
                // Socket closed, shutdown
                return Poll::Ok(())
            }
        }

        Poll::NotReady
    }
}

fn main() {
    let mut lp = Loop::new().unwrap();

    // Launch the server
    server::listen(lp.handle(),
                   "0.0.0.0:3245".parse().unwrap(),
                   |stream| Ok(Connection::new(stream)));

    // Run the reactor
    lp.run(futures::empty::<(), ()>()).unwrap();
}