Resolve Axum 0.7 regressions (#9291)

* perf: restore axum [postgresql] performance

* fix: axum plaintext regression

* feat: upgrade to axum 0.7.6

Author: andrew-james-dev

frameworks/Rust/axum/Cargo.lock

@@ -39,7 +39,7 @@ simd-json = [
 ]
 
 [dependencies]
-axum = { version = "0.7.5", default-features = false, features = [
+axum = { version = "0.7.6", default-features = false, features = [
     "json",
     "query",
     "http1",

frameworks/Rust/axum/Cargo.toml

@@ -11,11 +11,13 @@ use axum::Json;
 use common::simd_json::Json;
 
 /// Return a plaintext static string.
-pub async fn plaintext() -> impl IntoResponse {
-    (StatusCode::OK, "Hello, World!")
+#[inline(always)]
+pub async fn plaintext() -> &'static str {
+    "Hello, World!"
 }
 
 /// Return a JSON message.
+#[inline(always)]
 pub async fn json() -> impl IntoResponse {
     let message = Message {
         message: "Hello, World!",

frameworks/Rust/axum/src/main.rs

@@ -1,5 +1,6 @@
 mod common;
 mod mongo;
+mod server;
 //mod mongo_raw;
 
 use std::time::Duration;
@@ -24,8 +25,6 @@ use mongodb::{
 use rand::{rngs::SmallRng, thread_rng, Rng, SeedableRng};
 use yarte::Template;
 
-mod server;
-
 use common::{
     get_env,
     utils::{parse_params, Params, Utf8Html},
@@ -117,25 +116,10 @@ async fn fortunes(DatabaseConnection(db): DatabaseConnection) -> impl IntoRespon
 
 fn main() {
     dotenv().ok();
-
-    let rt = tokio::runtime::Builder::new_current_thread()
-        .enable_all()
-        .build()
-        .unwrap();
-
-    for _ in 1..num_cpus::get() {
-        std::thread::spawn(move || {
-            let rt = tokio::runtime::Builder::new_current_thread()
-                .enable_all()
-                .build()
-                .unwrap();
-            rt.block_on(serve());
-        });
-    }
-    rt.block_on(serve());
+    server::start_tokio(serve_app)
 }
 
-async fn serve() {
+async fn serve_app() {
     let database_url: String = get_env("MONGODB_URL");
     let max_pool_size: u32 = get_env("MONGODB_MAX_POOL_SIZE");
     let min_pool_size: u32 = get_env("MONGODB_MIN_POOL_SIZE");

frameworks/Rust/axum/src/main_mongo.rs

@@ -86,25 +86,10 @@ async fn updates(
 
 fn main() {
     dotenv().ok();
-
-    let rt = tokio::runtime::Builder::new_current_thread()
-        .enable_all()
-        .build()
-        .unwrap();
-
-    for _ in 1..num_cpus::get() {
-        std::thread::spawn(move || {
-            let rt = tokio::runtime::Builder::new_current_thread()
-                .enable_all()
-                .build()
-                .unwrap();
-            rt.block_on(serve());
-        });
-    }
-    rt.block_on(serve());
+    server::start_tokio(serve_app)
 }
 
-async fn serve() {
+async fn serve_app() {
     let database_url: String = get_env("MONGODB_URL");
     let max_pool_size: u32 = get_env("MONGODB_MAX_POOL_SIZE");
     let min_pool_size: u32 = get_env("MONGODB_MIN_POOL_SIZE");

frameworks/Rust/axum/src/main_mongo_raw.rs

@@ -78,10 +78,12 @@ async fn updates(
     (StatusCode::OK, Json(worlds))
 }
 
-#[tokio::main]
-async fn main() {
+fn main() {
     dotenv().ok();
+    server::start_tokio(serve_app)
+}
 
+async fn serve_app() {
     let database_url: String = get_env("POSTGRES_URL");
 
     // Create shared database connection

frameworks/Rust/axum/src/main_pg.rs

@@ -1,4 +1,5 @@
 use std::{
+    future::Future,
     io,
     net::{Ipv4Addr, SocketAddr, TcpListener},
 };
@@ -104,3 +105,26 @@ pub async fn serve_hyper(app: Router<()>, port: Option<u16>) {
         });
     }
 }
+
+/// Start a single-threaded tokio runtime on multiple threads.
+#[allow(dead_code)]
+pub fn start_tokio<Fut>(f: fn() -> Fut)
+where
+    Fut: Future<Output = ()> + 'static,
+{
+    let rt = tokio::runtime::Builder::new_current_thread()
+        .enable_all()
+        .build()
+        .unwrap();
+
+    for _ in 1..num_cpus::get() {
+        std::thread::spawn(move || {
+            let rt = tokio::runtime::Builder::new_current_thread()
+                .enable_all()
+                .build()
+                .unwrap();
+            rt.block_on(f());
+        });
+    }
+    rt.block_on(f());
+}