dadada/portfs
dadada/portfs
1
0
Fork 0
mirror of https://github.com/dadada/portfs.git synced 2025-06-07 09:33:56 +02:00
Code Issues Releases Wiki Activity

Refactoring

Browse source
This commit is contained in:
Tim Schubert 2019-06-01 19:58:44 +02:00
parent c872351556
commit d9135bf33a
1 changed files with 28 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

54
src/main.rs
View file

@ -6,39 +6,41 @@ use tokio::io::copy;
use tokio::net::{TcpListener, TcpStream};
use tokio::prelude::*;
use tokio::prelude::{AsyncRead, Future};
use tokio::executor::Spawn;
fn handle_client(sock: TcpStream) -> Spawn {
let addr = sock.peer_addr().unwrap();
println!("Received connection from {:?}", addr.port());
let filename = addr.port().to_string();
let path = Path::new(&filename);
let (reader, writer) = sock.split();
if path.exists() {
let task = File::open(filename)
.and_then(|file| copy(file, writer))
.map(|res| println!("{:?}", res))
.map_err(|err| eprintln!("IO error: {:?}", err));
tokio::spawn(task)
} else {
let task = File::create(filename)
.and_then(|file| copy(reader, file))
.map(|res| println!("{:?}", res))
.map_err(|err| eprintln!("IO error: {:?}", err));
tokio::spawn(task)
}
}
fn main() {
let addr = "127.0.0.1:8080".parse().unwrap();
let listener = TcpListener::bind(&addr).expect("unable to bind TCP listener");
let server = listener
let task = listener
.incoming()
.map_err(|e| eprintln!("accept failed = {:?}", e))
.for_each(|sock| {
let addr = sock.peer_addr().unwrap();
println!("Received connection from {:?}", addr.port());
let filename = addr.port().to_string();
let path = Path::new(&filename);
let (reader, writer) = sock.split();
if path.exists() {
let task = File::open(filename)
.and_then(|file| copy(file, writer))
.map(|res| println!("{:?}", res))
.map_err(|err| eprintln!("IO error: {:?}", err));
tokio::spawn(task)
} else {
let task = File::create(filename)
.and_then(|file| copy(reader, file))
.map(|res| println!("{:?}", res))
.map_err(|err| eprintln!("IO error: {:?}", err));
tokio::spawn(task)
}
});
.for_each(|sock| handle_client(sock));
// Start the Tokio runtime
tokio::run(server);
tokio::run(task);
}