src/doc/book/listings/ch20-web-server/no-listing-01-define-threadpool-struct/src/bin/main.rs

   1 // ANCHOR: here
   2 use hello::ThreadPool;
   3 // ANCHOR_END: here
   4 use std::fs;
   5 use std::io::prelude::*;
   6 use std::net::TcpListener;
   7 use std::net::TcpStream;
   8 use std::thread;
   9 use std::time::Duration;
  10
  11 fn main() {
  12     let listener = TcpListener::bind("127.0.0.1:7878").unwrap();
  13     let pool = ThreadPool::new(4);
  14
  15     for stream in listener.incoming() {
  16         let stream = stream.unwrap();
  17
  18         pool.execute(|| {
  19             handle_connection(stream);
  20         });
  21     }
  22 }
  23
  24 fn handle_connection(mut stream: TcpStream) {
  25     let mut buffer = [0; 1024];
  26     stream.read(&mut buffer).unwrap();
  27
  28     let get = b"GET / HTTP/1.1\r\n";
  29     let sleep = b"GET /sleep HTTP/1.1\r\n";
  30
  31     let (status_line, filename) = if buffer.starts_with(get) {
  32         ("HTTP/1.1 200 OK", "hello.html")
  33     } else if buffer.starts_with(sleep) {
  34         thread::sleep(Duration::from_secs(5));
  35         ("HTTP/1.1 200 OK", "hello.html")
  36     } else {
  37         ("HTTP/1.1 404 NOT FOUND", "404.html")
  38     };
  39
  40     let contents = fs::read_to_string(filename).unwrap();
  41
  42     let response = format!(
  43         "{}\r\nContent-Length: {}\r\n\r\n{}",
  44         status_line,
  45         contents.len(),
  46         contents
  47     );
  48
  49     stream.write(response.as_bytes()).unwrap();
  50     stream.flush().unwrap();
  51 }