proxy

proxy.rs

use std::io::TcpListener;
use std::io::{Acceptor, Listener};

fn main() {
    let listener = TcpListener::bind("0.0.0.0", 9999);
    let mut acceptor = listener.listen();

    for stream in acceptor.incoming() {
        match stream {
            Err(e) => println!("connect failed, {}", e),
            Ok(stream) => spawn(proc() {
                handler::proxy(stream);
            })
        }
    }

    drop(acceptor);
}

mod handler {
    use std::io::TcpStream;
    use std::io::util::copy;
    pub fn proxy(mut stream: TcpStream) {
        let mut redis_stream = match TcpStream::connect("127.0.0.1", 12345) {
            Err(e) => {
                println!("connect failed, {}", e);
                return
            },
            Ok(redis_stream) => redis_stream,
        };
        let mut r_stream = stream.clone();
        let mut r_redis_stream = redis_stream.clone();
        spawn(proc() {
            copy(&mut r_stream, &mut r_redis_stream);
        });
        copy(&mut redis_stream, &mut stream);
        drop(redis_stream);
        drop(stream)
    }
}