feat(server): use the num_cpus crate to configure the number of threa…

…ds for the pool
steveklabnik · Oct 19, 2017 · c8255a6 · c8255a6
1 parent e784b0f
commit c8255a6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/Cargo.toml b/Cargo.toml
@@ -15,6 +15,7 @@ travis-ci = { repository = "steveklabnik/simple-server" }
 http = "0.1.0"
 httparse = "1.2.3"
 log = "0.3"
+num_cpus = "1"
 scoped_threadpool = "0.1.7"
 
 [dev-dependencies]

diff --git a/src/lib.rs b/src/lib.rs
@@ -28,6 +28,7 @@ extern crate log;
 
 extern crate http;
 extern crate httparse;
+extern crate num_cpus;
 extern crate scoped_threadpool;
 
 pub use http::Request;
@@ -117,7 +118,8 @@ impl Server {
     /// }
     /// ```
     pub fn listen(&self, host: &str, port: &str) {
-        let mut pool = Pool::new(4);
+        let logical_cores = num_cpus::get() as u32;
+        let mut pool = Pool::new(logical_cores);
         let listener =
             TcpListener::bind(format!("{}:{}", host, port)).expect("Error starting the server.");