I wrote a multithreaded asynchronous HTTP server in Rust using mio. When I run a load test (using siege) the server works fine on the first load test but when the load test is done all requests to the server start failing.
Using some logging, I noticed that every new connection I get with accept()
receives a hangup event immediately. The server connection itself doesn't get any error or hangup events.
I am running Rust 1.12.0 with mio 0.6 on OS X 10.11 El Capitan
Here's the main event loop of my server:
pub fn run(self) {
let poll = Poll::new().unwrap();
let server = TcpListener::bind(&SocketAddr::from_str(&self.host).unwrap()).unwrap();
poll.register(&server, SERVER, Ready::readable(), PollOpt::edge()).unwrap();
let mut events = Events::with_capacity(1024);
let mut next_conn: usize = 1;
let mut workers = Vec::new();
// Create worker threads.
for _ in 0..self.num_workers {
let (tx, rx) = channel();
let worker_handler = self.event_handler.duplicate();
thread::spawn(move || {
Self::process_events(rx, worker_handler);
});
workers.push(tx);
}
loop {
println!("Polling...");
match poll.poll(&mut events, None) {
Err(e) => panic!("Error during poll(): {}", e),
Ok(_) => {}
}
for event in events.iter() {
match event.token() {
SERVER => {
println!("Accepting..");
match server.accept() {
Ok((stream, _)) => {
println!("Registering new connection...");
match poll.register(&stream,
Token(next_conn),
Ready::readable(),
PollOpt::edge()) {
Err(e) => panic!("Error during register(): {}", e),
Ok(_) => {
println!("New connection on worker {} ",
next_conn % self.num_workers);
workers[next_conn % self.num_workers]
.send(Msg::NewConn(next_conn, stream))
.unwrap();
next_conn += 1;
}
}
}
Err(e) => panic!("Error during accept() : {}", e),
}
}
Token(id) => {
println!("Sending event on conn {} to worker {}",
id,
id % self.num_workers);
workers[id % self.num_workers]
.send(Msg::ConnEvent(id, event.kind()))
.unwrap();
}
}
}
}
}
fn process_events(channel: Receiver<Msg>, mut event_handler: Box<EventHandler>) {
loop {
let msg = channel.recv().unwrap();
match msg {
Msg::NewConn(id, conn) => {
event_handler.new_conn(id, conn);
}
Msg::ConnEvent(id, event) => {
event_handler.conn_event(id, event);
}
}
}
}
Full code with the example webapp I am using is available on GitHub.
Load test command:
siege -b -c10 -d10 -t20S http://localhost:8080