1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
use std::net::{TcpListener, TcpStream, Shutdown};
use std::io::Write;
use std::str;
use crate::request::Request;
use crate::response::Response;
use crate::http_method::HttpMethod;
pub trait HttpHandler {
// TODO(andrew): add default behaviour to handlers?
fn do_get(&self, request: Request) -> Response;
fn do_post(&self, request: Request) -> Response;
}
pub struct HttpServer<T: HttpHandler> {
host: String,
port: u16,
socket: TcpListener,
handler: T
}
impl<T> HttpServer<T> where T: HttpHandler {
// TODO(andrew): Add more verbose error handling.
pub fn new(host: &str, port: u16, handler: T) -> Option<HttpServer<T>> {
let addr = format!("{}:{}", host, port);
let sock = TcpListener::bind(addr);
match sock {
Ok(s) => {
let server = HttpServer {
host: String::from(host),
port: port,
socket: s,
handler: handler,
};
return Some(server);
},
Err(_) => return None
}
}
pub fn serve_forever(&self) {
for stream in self.socket.incoming() {
match stream {
Ok(s) => {
println!("Got connection!");
self.handle_client(&s);
// TODO(andrew): handle possible errors.
s.shutdown(Shutdown::Both);
println!("Connection hadled");
},
Err(_) => break
};
}
}
fn handle_client(&self, mut stream: &TcpStream) {
let mut buf: [u8; 1024] = [0; 1024];
// TODO(andrew): read all body, not first 1024 bytes.
stream.peek(&mut buf).expect("Couldn't read from socket");
let request = Request::from(&buf);
// TODO(andrew): remove panic.
let request = match request {
Some(r) => r,
None => panic!("Request parsed with errors")
};
// TODO(andrew): add more methods.
let response = match request.method {
HttpMethod::GET => self.handler.do_get(request),
HttpMethod::POST => self.handler.do_post(request)
};
let response = response.format();
// TODO(andrew): handle possible errors.
stream.write(&response);
}
}
|
