tonic-mt thread limit

rust_tonic_mt_bench/.cargo/config

@@ -0,0 +1,2 @@
+[build]
+rustflags = [ "-C", "target-cpu=native" ]

rust_tonic_mt_bench/src/main.rs

@@ -23,8 +23,24 @@ impl Greeter for MyGreeter {
     }
 }
 
-#[tokio::main]
-async fn main() -> Result<(), Box<dyn std::error::Error>> {
+fn main() -> Result<(), Box<dyn std::error::Error>> {
+    let cpus = std::env::var("GRPC_SERVER_CPUS")
+        .map(|v| v.parse().unwrap())
+        .unwrap_or(1);
+
+    println!("Running with {} threads", cpus);
+
+    // Esentially the same as tokio::main, but with number of threads set to 
+    // avoid thrashing when cggroup limits are applied by Docker.
+    tokio::runtime::Builder::new_multi_thread()
+        .worker_threads(cpus)
+        .enable_all()
+        .build()
+        .unwrap()
+        .block_on(serve())
+}
+
+async fn serve() -> Result<(), Box<dyn std::error::Error>> {
     let addr = "0.0.0.0:50051".parse().unwrap();
     let greeter = MyGreeter::default();
 
@@ -36,4 +52,4 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         .await?;
 
     Ok(())
-}
+}

rust_tonic_st_bench/.cargo/config

@@ -0,0 +1,2 @@
+[build]
+rustflags = [ "-C", "target-cpu=native" ]