From 67cdcc2e12118becb823e20a40cc2687f2b8425a Mon Sep 17 00:00:00 2001 From: Adam Carpenter <53hornet@gmail.com> Date: Wed, 27 Mar 2019 15:32:37 -0400 Subject: Started Rust in Action MEAP. --- hello_server/src/bin/main.rs | 46 ----------------- hello_server/src/lib.rs | 117 ------------------------------------------- 2 files changed, 163 deletions(-) delete mode 100644 hello_server/src/bin/main.rs delete mode 100644 hello_server/src/lib.rs (limited to 'hello_server/src') diff --git a/hello_server/src/bin/main.rs b/hello_server/src/bin/main.rs deleted file mode 100644 index bcf1e46..0000000 --- a/hello_server/src/bin/main.rs +++ /dev/null @@ -1,46 +0,0 @@ -use hello_server::ThreadPool; -use std::thread; -use std::time::Duration; -use std::io::prelude::*; -use std::fs; -use std::net::TcpStream; -use std::net::TcpListener; - -fn main() { - let listener = TcpListener::bind("127.0.0.1:7878").unwrap(); - let pool = ThreadPool::new(4).unwrap(); - - for stream in listener.incoming().take(2) { - let stream = stream.unwrap(); - - pool.execute(|| { - handle_connection(stream); - }); - } - - println!("shutting down..."); - -} - -fn handle_connection(mut stream: TcpStream) { - let mut buffer = [0; 512]; - stream.read(&mut buffer).unwrap(); - let get = b"GET / HTTP/1.1\r\n"; - let sleep = b"GET /sleep HTTP/1.1\r\n"; - - let (status_line, filename) = if buffer.starts_with(get) { - ("HTTP/1.1 200 OK\r\n\r\n", "hello.html") - } - else if buffer.starts_with(sleep) { - thread::sleep(Duration::from_secs(5)); - ("HTTP/1.1 200 OK\r\n\r\n", "hello.html") - } - else { - ("HTTP/1.1 404 NOT FOUND\r\n\r\n", "404.html") - }; - - let contents = fs::read_to_string(filename).unwrap(); - let response = format!("{}{}", status_line, contents); - stream.write(response.as_bytes()).unwrap(); - stream.flush().unwrap(); -} diff --git a/hello_server/src/lib.rs b/hello_server/src/lib.rs deleted file mode 100644 index cd1f616..0000000 --- a/hello_server/src/lib.rs +++ /dev/null @@ -1,117 +0,0 @@ -use std::sync::Arc; -use std::sync::Mutex; -use std::sync::mpsc; -use std::thread; - -enum Message { - NewJob(Job), - Terminate, -} - -trait FnBox { - fn call_box(self: Box); -} - -impl FnBox for F { - fn call_box(self: Box) { - (*self)() - } -} - -type Job = Box; - -pub struct ThreadPool { - workers: Vec, - sender: mpsc::Sender, -} - -impl ThreadPool { - pub fn new(size: usize) -> Result { - if size <= 0 { - return Err("failed to create pool"); - } - - let (sender, receiver) = mpsc::channel(); - - let receiver = Arc::new(Mutex::new(receiver)); - - let mut workers = Vec::with_capacity(size); - - for id in 0..size { - workers.push(Worker::new(id, Arc::clone(&receiver))?); - } - - Ok(ThreadPool { - workers, - sender, - }) - } - - pub fn spawn(f: F) -> thread::JoinHandle - where - F: FnOnce() -> T + Send + 'static, - T: Send + 'static - { - thread::spawn(f) - } - - pub fn execute(&self, f: F) - where - F: FnOnce() + Send + 'static - { - let job = Box::new(f); - self.sender.send(Message::NewJob(job)).unwrap(); - } - -} - -impl Drop for ThreadPool { - fn drop(&mut self) { - for _ in &mut self.workers { - self.sender.send(Message::Terminate).unwrap(); - } - - println!("shutting down all workers..."); - - for worker in &mut self.workers { - println!("shutting down worker {}", worker.id); - - if let Some(thread) = worker.thread.take() { - thread.join().unwrap(); - } - } - } -} - -struct Worker { - id: usize, - thread: Option>, -} - -impl Worker { - pub fn new(id: usize, receiver: Arc>>) - -> Result - { - let thread = thread::spawn(move || { - loop { - let message = receiver.lock().unwrap().recv().unwrap(); - - match message { - Message::NewJob(job) => { - println!("worker {} got job, executing...", id); - job.call_box(); - }, - Message::Terminate => { - println!("worker {} was told to terminate...", id); - break; - }, - } - } - }); - - Ok(Worker { - id, - thread: Some(thread), - }) - } -} -- cgit v1.2.3