backend: use mimalloc and set nodelay to increase performance (#542)

vnghia · Dec 3, 2024 · 05a7927 · 05a7927
1 parent b7f47f4
commit 05a7927
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/nghe-backend/Cargo.toml b/nghe-backend/Cargo.toml
@@ -64,6 +64,7 @@ indexmap = { version = "2.6.0" }
 libaes = { version = "0.7.0" }
 lofty = { version = "0.21.1" }
 loole = { version = "0.4.0" }
+mimalloc = { version = "0.1.43" }
 o2o = { version = "0.5.1-beta1", default-features = false, features = ["syn2"] }
 rsmpeg = { version = "0.15.1", default-features = false, features = [
   "ffmpeg7",

diff --git a/nghe-backend/src/lib.rs b/nghe-backend/src/lib.rs
@@ -34,10 +34,14 @@ use axum::body::Body;
 use axum::http::Request;
 use axum::Router;
 use error::Error;
+use mimalloc::MiMalloc;
 use tower_http::compression::CompressionLayer;
 use tower_http::cors::CorsLayer;
 use tower_http::trace::TraceLayer;
 
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;
+
 pub async fn build(config: config::Config) -> Router {
     let filesystem =
         filesystem::Filesystem::new(&config.filesystem.tls, &config.filesystem.s3).await;

diff --git a/nghe-backend/src/main.rs b/nghe-backend/src/main.rs
@@ -22,5 +22,5 @@ async fn main() {
     migration::run(&config.database.url).await;
 
     let listener = tokio::net::TcpListener::bind(config.server.to_socket_addr()).await.unwrap();
-    axum::serve(listener, build(config).await).await.unwrap();
+    axum::serve(listener, build(config).await).tcp_nodelay(true).await.unwrap();
 }